Annotation of embedaddon/bmon/src/utils.c, revision 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>