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>