Annotation of embedaddon/bmon/src/utils.c, revision 1.1.1.1
1.1 misho 1: /*
2: * utils.c General purpose utilities
3: *
4: * Copyright (c) 2001-2005 Thomas Graf <tgraf@suug.ch>
5: *
6: * Permission is hereby granted, free of charge, to any person obtaining a
7: * copy of this software and associated documentation files (the "Software"),
8: * to deal in the Software without restriction, including without limitation
9: * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10: * and/or sell copies of the Software, and to permit persons to whom the
11: * Software is furnished to do so, subject to the following conditions:
12: *
13: * The above copyright notice and this permission notice shall be included
14: * in all copies or substantial portions of the Software.
15: *
16: * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
17: * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18: * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19: * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20: * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21: * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22: * DEALINGS IN THE SOFTWARE.
23: */
24:
25: #include <bmon/bmon.h>
26: #include <bmon/conf.h>
27: #include <bmon/utils.h>
28:
29: void * xcalloc(size_t n, size_t s)
30: {
31: void *d = calloc(n, s);
32:
33: if (NULL == d) {
34: fprintf(stderr, "xalloc: Out of memory\n");
35: exit(ENOMEM);
36: }
37:
38: return d;
39: }
40:
41: void * xrealloc(void *p, size_t s)
42: {
43: void *d = realloc(p, s);
44:
45: if (NULL == d) {
46: fprintf(stderr, "xrealloc: Out of memory!\n");
47: exit(ENOMEM);
48: }
49:
50: return d;
51: }
52:
53: void xfree(void *d)
54: {
55: if (d)
56: free(d);
57: }
58:
59: inline float ts_to_float(timestamp_t *src)
60: {
61: return (float) src->tv_sec + ((float) src->tv_usec / 1000000.0f);
62: }
63:
64: inline void float_to_ts(timestamp_t *dst, float src)
65: {
66: dst->tv_sec = (time_t) src;
67: dst->tv_usec = (src - ((float) ((time_t) src))) * 1000000.0f;
68: }
69:
70: inline void ts_add(timestamp_t *dst, timestamp_t *src1, timestamp_t *src2)
71: {
72: dst->tv_sec = src1->tv_sec + src2->tv_sec;
73: dst->tv_usec = src1->tv_usec + src2->tv_usec;
74:
75: if (dst->tv_usec >= 1000000) {
76: dst->tv_sec++;
77: dst->tv_usec -= 1000000;
78: }
79: }
80:
81: inline void ts_sub(timestamp_t *dst, timestamp_t *src1, timestamp_t *src2)
82: {
83: dst->tv_sec = src1->tv_sec - src2->tv_sec;
84: dst->tv_usec = src1->tv_usec - src2->tv_usec;
85: if (dst->tv_usec <= -1000000) {
86: dst->tv_sec--;
87: dst->tv_usec += 1000000;
88: }
89: }
90:
91: inline int ts_le(timestamp_t *a, timestamp_t *b)
92: {
93: if (a->tv_sec > b->tv_sec)
94: return 0;
95:
96: if (a->tv_sec < b->tv_sec || a->tv_usec <= b->tv_usec)
97: return 1;
98:
99: return 0;
100: }
101:
102: inline void update_ts(timestamp_t *dst)
103: {
104: struct timeval tv;
105:
106: gettimeofday(&tv, NULL);
107:
108: dst->tv_sec = tv.tv_sec;
109: dst->tv_usec = tv.tv_usec;
110: }
111:
112:
113: float time_diff(timestamp_t *t1, timestamp_t *t2)
114: {
115: timestamp_t ts;
116: ts_sub(&ts, t2, t1);
117: return ts_to_float(&ts);
118: }
119:
120: float diff_now(timestamp_t *t1)
121: {
122: timestamp_t now;
123: update_ts(&now);
124: return time_diff(t1, &now);
125: }
126:
127: static inline b_cnt_t __divisor(int type, int exp)
128: {
129: static b_cnt_t cache[2][32];
130:
131: if (exp) {
132: if (!cache[get_use_si()][exp]) {
133: cache[get_use_si()][exp] =
134: (b_cnt_t) pow(get_use_si() ||
135: type == U_NUMBER ? 1000.0f : 1024.0f, exp);
136: }
137: return cache[get_use_si()][exp];
138: } else
139: return 1;
140: }
141:
142: static inline char * __unit(int type, int exp)
143: {
144: static char *u[2][3][32] = {
145: [0] = { /* IEC */
146: [U_BITS] = { " b ", "Kib", "Mib", "Gib", "Tib" },
147: [U_BYTES] = { " B ", "KiB", "MiB", "GiB", "TiB" },
148: [U_NUMBER] = { " ", "K", "M", "G", "T" },
149: },
150: [1] = { /* SI */
151: [U_BITS] = { " b", "Kb ", "Mb ", "Gb ", "Tb " },
152: [U_BYTES] = { " B ", "KB ", "MB ", "GB ", "TB " },
153: [U_NUMBER] = { " ", "K", "M", "G", "T" },
154: }
155: };
156:
157: return u[get_use_si()][type][exp];
158:
159: }
160:
161: b_cnt_t get_divisor(b_cnt_t hint, int unit, char **dst_unit, int *prec)
162: {
163: int yunit = get_y_unit();
164:
165: if (prec)
166: *prec = 2;
167:
168: if (yunit == Y_DYNAMIC) {
169: if (hint >= __divisor(unit, 3)) {
170: *dst_unit = __unit(unit, 3);
171: return __divisor(unit, 3);
172: } else if (hint >= __divisor(unit, 2)) {
173: *dst_unit = __unit(unit, 2);
174: return __divisor(unit, 2);
175: } else if (hint >= __divisor(unit, 1)) {
176: *dst_unit = __unit(unit, 1);
177: return __divisor(unit, 1);
178: } else {
179: *dst_unit = __unit(unit, 0);
180: if (prec)
181: *prec = 0;
182: return 1;
183: }
184: } else {
185: *dst_unit = __unit(unit, yunit);
186: return __divisor(unit, yunit);
187: }
188:
189: return 1;
190: }
191:
192: double cancel_down(b_cnt_t l, int unit, char **dst_unit, int *prec)
193: {
194: return ((double) l / (double) get_divisor(l, unit, dst_unit, prec));
195: }
196:
197: const char * xinet_ntop(struct sockaddr *src, char *dst, socklen_t cnt)
198: {
199: void *s;
200: int family;
201:
202: if (src->sa_family == AF_INET6) {
203: s = &((struct sockaddr_in6 *) src)->sin6_addr;
204: family = AF_INET6;
205: } else if (src->sa_family == AF_INET) {
206: s = &((struct sockaddr_in *) src)->sin_addr;
207: family = AF_INET;
208: } else
209: return NULL;
210:
211: return inet_ntop(family, s, dst, cnt);
212: }
213:
214: b_cnt_t parse_size(const char *str)
215: {
216: char *p;
217: b_cnt_t l = strtol(str, &p, 0);
218: if (p == str)
219: return -1;
220:
221: if (*p) {
222: if (!strcasecmp(p, "kb") || !strcasecmp(p, "k"))
223: l *= 1024;
224: else if (!strcasecmp(p, "gb") || !strcasecmp(p, "g"))
225: l *= 1024*1024*1024;
226: else if (!strcasecmp(p, "gbit"))
227: l *= 1024*1024*1024/8;
228: else if (!strcasecmp(p, "mb") || !strcasecmp(p, "m"))
229: l *= 1024*1024;
230: else if (!strcasecmp(p, "mbit"))
231: l *= 1024*1024/8;
232: else if (!strcasecmp(p, "kbit"))
233: l *= 1024/8;
234: else if (strcasecmp(p, "b") != 0)
235: return -1;
236: }
237:
238: return l;
239: }
240:
241: #ifndef HAVE_STRDUP
242: char *strdup(const char *s)
243: {
244: char *t = xcalloc(1, strlen(s) + 1);
245: memcpy(t, s, strlen(s));
246: return s;
247: }
248: #endif
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>