1 /*-
2 * Copyright (c) 2002-2007 Sam Leffler, Errno Consulting
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer,
10 * without modification.
11 * 2. Redistributions in binary form must reproduce at minimum a disclaimer
12 * similar to the "NO WARRANTY" disclaimer below ("Disclaimer") and any
13 * redistribution must be conditioned upon including a substantially
14 * similar Disclaimer requirement for further binary redistribution.
15 *
16 * NO WARRANTY
17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19 * LIMITED TO, THE IMPLIED WARRANTIES OF NONINFRINGEMENT, MERCHANTABILITY
20 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
21 * THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY,
22 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
27 * THE POSSIBILITY OF SUCH DAMAGES.
28 *
29 * $FreeBSD: src/tools/tools/net80211/wlanstats/statfoo.c,v 1.4 2008/04/30 19:47:31 sam Exp $
30 */
31
32 #include <stdio.h>
33 #include <string.h>
34
35 #include "statfoo.h"
36
37 static void
statfoo_setfmt(struct statfoo * sf,const char * fmt0)38 statfoo_setfmt(struct statfoo *sf, const char *fmt0)
39 {
40 #define N(a) (sizeof(a)/sizeof(a[0]))
41 char fmt[4096];
42 char *fp, *tok;
43 int i, j, field;
44
45 j = field = 0;
46 strlcpy(fmt, fmt0, sizeof(fmt));
47 for (fp = fmt; (tok = strsep(&fp, ", ")) != NULL;) {
48 for (i = 0; i < sf->nstats; i++)
49 if (strcasecmp(tok, sf->stats[i].name) == 0)
50 break;
51 if (i >= sf->nstats) {
52 fprintf(stderr, "%s: unknown statistic name \"%s\" "
53 "skipped\n", sf->name, tok);
54 continue;
55 }
56 if (j+3 > sizeof(sf->fmts)) {
57 fprintf(stderr, "%s: not enough room for all stats; "
58 "stopped at %s\n", sf->name, tok);
59 break;
60 }
61 if (field > 127) {
62 fprintf(stderr, "%s: too many fields; "
63 "stopped at %s\n", sf->name, tok);
64 break;
65 }
66 if (j != 0)
67 sf->fmts[j++] = ' ';
68 sf->fmts[j++] = 0x80 | field;
69 sf->fields[field++] = i;
70 }
71 sf->fmts[j] = '\0';
72 #undef N
73 }
74
75 static void
statfoo_collect(struct statfoo * sf)76 statfoo_collect(struct statfoo *sf)
77 {
78 fprintf(stderr, "%s: don't know how to collect data\n", sf->name);
79 }
80
81 static void
statfoo_update_tot(struct statfoo * sf)82 statfoo_update_tot(struct statfoo *sf)
83 {
84 fprintf(stderr, "%s: don't know how to update total data\n", sf->name);
85 }
86
87 static int
statfoo_get(struct statfoo * sf,int s,char b[],size_t bs)88 statfoo_get(struct statfoo *sf, int s, char b[], size_t bs)
89 {
90 fprintf(stderr, "%s: don't know how to get stat #%u\n", sf->name, s);
91 return 0;
92 }
93
94 static void
statfoo_print_header(struct statfoo * sf,FILE * fd)95 statfoo_print_header(struct statfoo *sf, FILE *fd)
96 {
97 const unsigned char *cp;
98
99 for (cp = sf->fmts; *cp != '\0'; cp++) {
100 if (*cp & 0x80) {
101 int six = sf->fields[*cp &~ 0x80];
102 const struct fmt *f = &sf->stats[six];
103 fprintf(fd, "%*s", f->width, f->label);
104 } else
105 putc(*cp, fd);
106 }
107 putc('\n', fd);
108 }
109
110 static void
statfoo_print_current(struct statfoo * sf,FILE * fd)111 statfoo_print_current(struct statfoo *sf, FILE *fd)
112 {
113 char buf[32];
114 const unsigned char *cp;
115
116 for (cp = sf->fmts; *cp != '\0'; cp++) {
117 if (*cp & 0x80) {
118 int six = sf->fields[*cp &~ 0x80];
119 const struct fmt *f = &sf->stats[six];
120 if (sf->get_curstat(sf, six, buf, sizeof(buf)))
121 fprintf(fd, "%*s", f->width, buf);
122 } else
123 putc(*cp, fd);
124 }
125 putc('\n', fd);
126 }
127
128 static void
statfoo_print_total(struct statfoo * sf,FILE * fd)129 statfoo_print_total(struct statfoo *sf, FILE *fd)
130 {
131 char buf[32];
132 const unsigned char *cp;
133
134 for (cp = sf->fmts; *cp != '\0'; cp++) {
135 if (*cp & 0x80) {
136 int six = sf->fields[*cp &~ 0x80];
137 const struct fmt *f = &sf->stats[six];
138 if (sf->get_totstat(sf, six, buf, sizeof(buf)))
139 fprintf(fd, "%*s", f->width, buf);
140 } else
141 putc(*cp, fd);
142 }
143 putc('\n', fd);
144 }
145
146 static void
statfoo_print_verbose(struct statfoo * sf,FILE * fd)147 statfoo_print_verbose(struct statfoo *sf, FILE *fd)
148 {
149 const struct fmt *f;
150 char s[32];
151 int i, width;
152
153 width = 0;
154 for (i = 0; i < sf->nstats; i++) {
155 f = &sf->stats[i];
156 if (f->width > width)
157 width = f->width;
158 }
159 for (i = 0; i < sf->nstats; i++) {
160 f = &sf->stats[i];
161 if (sf->get_totstat(sf, i, s, sizeof(s)) && strcmp(s, "0"))
162 fprintf(fd, "%-*s %s\n", width, s, f->desc);
163 }
164 }
165
166 static void
statfoo_print_fields(struct statfoo * sf,FILE * fd)167 statfoo_print_fields(struct statfoo *sf, FILE *fd)
168 {
169 int i, w, width;
170
171 width = 0;
172 for (i = 0; i < sf->nstats; i++) {
173 w = strlen(sf->stats[i].name);
174 if (w > width)
175 width = w;
176 }
177 for (i = 0; i < sf->nstats; i++) {
178 const struct fmt *f = &sf->stats[i];
179 if (f->width != 0)
180 fprintf(fd, "%-*s %s\n", width, f->name, f->desc);
181 }
182 }
183
184 void
statfoo_init(struct statfoo * sf,const char * name,const struct fmt * stats,int nstats)185 statfoo_init(struct statfoo *sf, const char *name, const struct fmt *stats, int nstats)
186 {
187 sf->name = name;
188 sf->stats = stats;
189 sf->nstats = nstats;
190 sf->setfmt = statfoo_setfmt;
191 sf->collect_cur = statfoo_collect;
192 sf->collect_tot = statfoo_collect;
193 sf->update_tot = statfoo_update_tot;
194 sf->get_curstat = statfoo_get;
195 sf->get_totstat = statfoo_get;
196 sf->print_header = statfoo_print_header;
197 sf->print_current = statfoo_print_current;
198 sf->print_total = statfoo_print_total;
199 sf->print_verbose = statfoo_print_verbose;
200 sf->print_fields = statfoo_print_fields;
201 }
202