| 1 | #ifndef _UTILS_H |
| 2 | #define _UTILS_H |
| 3 | |
| 4 | #include <stdio.h> |
| 5 | #include <stdarg.h> |
| 6 | #include <sys/types.h> |
| 7 | |
| 8 | #define max(a, b) (((b) > (a))?(b):(a)) |
| 9 | #define min(a, b) (((b) < (a))?(b):(a)) |
| 10 | |
| 11 | #define smalloc(size) ({void *__result__; ((__result__ = malloc(size)) == NULL)?({abort(); (void *)0;}):__result__;}) |
| 12 | #define srealloc(ptr, size) ({void *__result__; ((__result__ = realloc((ptr), (size))) == NULL)?({abort(); (void *)0;}):__result__;}) |
| 13 | #define szmalloc(size) memset(smalloc(size), 0, size) |
| 14 | #define sstrdup(str) ({const char *__strbuf__ = (str); strcpy(smalloc(strlen(__strbuf__) + 1), __strbuf__);}) |
| 15 | #define omalloc(o) ((o) = szmalloc(sizeof(*(o)))) |
| 16 | |
| 17 | #define bufinit(buf) memset(&(buf), 0, sizeof(buf)) |
| 18 | #define buffree(buf) do { if((buf).b != NULL) {free((buf).b);} bufinit(buf); } while(0) |
| 19 | #define sizebuf(buf, wanted) (_sizebuf((struct buffer *)&(buf), (wanted), sizeof(*((buf).b)))) |
| 20 | #define bufdel(buf, i) (memmove((buf).b + (i), (buf).b + (i) + 1, (--((buf).d) - (i)) * sizeof(*((buf).b)))) |
| 21 | #define bufadd(buf, new) \ |
| 22 | do { \ |
| 23 | _sizebuf((struct buffer *)&(buf), (buf).d + 1, sizeof(*((buf).b))); \ |
| 24 | (buf).b[(buf).d++] = (new); \ |
| 25 | } while(0) |
| 26 | #define bufcat(buf, new, size) \ |
| 27 | do { \ |
| 28 | size_t __bufcat_size__; \ |
| 29 | __bufcat_size__ = (size); \ |
| 30 | _sizebuf((struct buffer *)&(buf), (buf).d + __bufcat_size__, sizeof((buf).b)); \ |
| 31 | memcpy((buf).b + (buf).d, (new), (__bufcat_size__) * sizeof(*((buf).b))); \ |
| 32 | (buf).d += __bufcat_size__; \ |
| 33 | } while(0) |
| 34 | #define bufcatstr(buf, str) \ |
| 35 | do { \ |
| 36 | char *__buf__; \ |
| 37 | __buf__ = (str); \ |
| 38 | bufcat((buf), __buf__, strlen(__buf__)); \ |
| 39 | } while(0) |
| 40 | #define bufcatstr2(buf, str) \ |
| 41 | do { \ |
| 42 | char *__buf__; \ |
| 43 | __buf__ = (str); \ |
| 44 | bufcat((buf), __buf__, strlen(__buf__) + 1); \ |
| 45 | } while(0) |
| 46 | #define bufeat(buf, len) memmove((buf).b, (buf).b + (len), (buf).d -= (len)) |
| 47 | |
| 48 | struct buffer { |
| 49 | void *b; |
| 50 | size_t s, d; |
| 51 | }; |
| 52 | |
| 53 | #define typedbuf(type) struct {type *b; size_t s, d;} |
| 54 | |
| 55 | struct charbuf { |
| 56 | char *b; |
| 57 | size_t s, d; |
| 58 | }; |
| 59 | |
| 60 | struct charvbuf { |
| 61 | char **b; |
| 62 | size_t s, d; |
| 63 | }; |
| 64 | |
| 65 | struct btree { |
| 66 | struct btree *l, *r; |
| 67 | int h; |
| 68 | void *d; |
| 69 | }; |
| 70 | |
| 71 | void _sizebuf(struct buffer *buf, size_t wanted, size_t el); |
| 72 | char *decstr(char **p, size_t *len); |
| 73 | char *vsprintf2(char *format, va_list al); |
| 74 | char *sprintf2(char *format, ...); |
| 75 | char *sprintf3(char *format, ...); |
| 76 | off_t atoo(char *n); |
| 77 | char **tokenize(char *src); |
| 78 | void freeca(char **ca); |
| 79 | int calen(char **a); |
| 80 | void bvprintf(struct charbuf *buf, char *format, va_list al); |
| 81 | void bprintf(struct charbuf *buf, char *format, ...); |
| 82 | void replstr(char **p, char *n); |
| 83 | char *base64encode(char *data, size_t datalen); |
| 84 | char *base64decode(char *data, size_t *datalen); |
| 85 | int hexdigit(char c); |
| 86 | int bbtreedel(struct btree **tree, void *item, int (*cmp)(void *, void *)); |
| 87 | void freebtree(struct btree **tree, void (*ffunc)(void *)); |
| 88 | int bbtreeput(struct btree **tree, void *item, int (*cmp)(void *, void *)); |
| 89 | void *btreeget(struct btree *tree, void *key, int (*cmp)(void *, void *)); |
| 90 | FILE *funstdio(void *pdata, |
| 91 | ssize_t (*read)(void *pdata, void *buf, size_t len), |
| 92 | ssize_t (*write)(void *pdata, const void *buf, size_t len), |
| 93 | off_t (*seek)(void *pdata, off_t offset, int whence), |
| 94 | int (*close)(void *pdata)); |
| 95 | |
| 96 | #endif |