2 * Dolda Connect - Modular multiuser Direct Connect-style client
3 * Copyright (C) 2004 Fredrik Tolf <fredrik@dolda2000.com>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
31 #include <netinet/in.h>
48 static char *base64set = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
49 static int base64rev[] = {
50 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
51 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
52 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63,
53 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1,
54 -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
55 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
56 -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
57 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1,
58 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
59 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
60 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
61 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
62 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
63 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
64 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
65 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
67 static char *base32set = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567";
68 static int base32rev[] = {
69 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
70 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
71 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
72 -1, -1, 26, 27, 28, 29, 30, 31, -1, -1, -1, -1, -1, -1, -1, -1,
73 -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
74 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
75 -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
76 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
77 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
78 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
79 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
80 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
81 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
82 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
83 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
84 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
87 char *vsprintf2(char *format, va_list al)
94 ret = vsnprintf(NULL, 0, format, al2);
96 if((buf = malloc(ret + 1)) == NULL)
102 vsnprintf(buf, ret + 1, format, al2);
107 char *sprintf2(char *format, ...)
112 va_start(args, format);
113 buf = vsprintf2(format, args);
118 wchar_t *vswprintf2(wchar_t *format, va_list al)
125 buf = smalloc(sizeof(wchar_t) * (bufsize = 1024));
129 ret = vswprintf(buf, bufsize, format, al2);
133 buf = srealloc(buf, sizeof(wchar_t) * (bufsize *= 2));
135 if(bufsize > ret + 1)
136 buf = srealloc(buf, sizeof(wchar_t) * (ret + 1));
140 wchar_t *swprintf2(wchar_t *format, ...)
145 va_start(args, format);
146 buf = vswprintf2(format, args);
151 int havecharset(char *charset)
155 if((cd = iconv_open("wchar_t", charset)) == (iconv_t)-1)
158 if((cd = iconv_open(charset, "wchar_t")) == (iconv_t)-1)
164 wchar_t *icmbstowcs(char *mbs, char *charset)
169 size_t len1, len2, bufsize, data;
172 len1 = strlen(mbs) + 1;
173 bufsize = len2 = len1 * sizeof(wchar_t);
174 if((buf = malloc(bufsize)) == NULL)
180 charset = nl_langinfo(CODESET);
181 if((cd = iconv_open("wchar_t", charset)) == (iconv_t)-1)
184 flog(LOG_ERR, "icmbstowcs: could not open iconv structure for %s: %s", charset, strerror(errno));
192 ret = iconv(cd, &mbs, &len1, &p, &len2);
200 if((p2 = realloc(buf, bufsize)) == NULL)
217 buf = realloc(buf, p - buf);
219 return((wchar_t *)buf);
222 wchar_t *icsmbstowcs(char *mbs, char *charset, wchar_t *def)
224 static wchar_t *buf = NULL;
228 if((buf = icmbstowcs(mbs, charset)) == NULL)
230 if((def != NULL) && (*def == L'~'))
233 flog(LOG_WARNING, "icsmbstowcs: could not convert wcs string into charset %s: %s", charset, strerror(errno));
242 char *icwcstombs(wchar_t *wcs, char *charset)
247 size_t len1, len2, bufsize, data;
250 len1 = sizeof(wchar_t) * (wcslen(wcs) + 1);
251 bufsize = len2 = len1;
252 if((buf = malloc(bufsize)) == NULL)
260 charset = nl_langinfo(CODESET);
261 if((cd = iconv_open(charset, "wchar_t")) == (iconv_t)-1)
264 flog(LOG_ERR, "icwcstombs: could not open iconv structure for %s: %s", charset, strerror(errno));
272 ret = iconv(cd, (char **)&wcs, &len1, &p, &len2);
280 if((p2 = realloc(buf, bufsize)) == NULL)
297 buf = realloc(buf, p - buf);
302 char *icswcstombs(wchar_t *wcs, char *charset, char *def)
304 static char *buf = NULL;
308 if((buf = icwcstombs(wcs, charset)) == NULL)
310 if((def != NULL) && (*def == '~'))
313 flog(LOG_WARNING, "icswcstombs: could not convert mbs string from charset %s: %s", charset, strerror(errno));
322 wchar_t *wcstolower(wchar_t *wcs)
326 for(p = wcs; *p != L'\0'; p++)
331 wchar_t ucptowc(int ucp)
334 unsigned long ucpbuf;
338 size_t len1, len2, bufsize, data;
342 mbsp = (char *)&ucpbuf;
344 bufsize = len2 = len1 * sizeof(wchar_t);
345 if((buf = malloc(bufsize)) == NULL)
350 if((cd = iconv_open("wchar_t", "UCS-4BE")) == (iconv_t)-1)
353 flog(LOG_ERR, "ucptowc: could not open iconv structure for UCS-4BE: %s", strerror(errno));
361 ret = iconv(cd, &mbsp, &len1, &p, &len2);
369 if((p2 = realloc(buf, bufsize)) == NULL)
386 buf = realloc(buf, p - buf);
388 res = *(wchar_t *)buf;
393 void _sizebuf(void **buf, size_t *bufsize, size_t reqsize, size_t elsize, int algo)
395 if(*bufsize >= reqsize)
400 *buf = srealloc(*buf, elsize * ((*bufsize) = reqsize));
405 while(*bufsize < reqsize)
407 *buf = srealloc(*buf, elsize * (*bufsize));
416 gettimeofday(&tv, NULL);
417 return((double)tv.tv_sec + ((double)tv.tv_usec / 1000000.0));
420 int wcsexists(wchar_t *h, wchar_t *n)
425 ln = alloca(sizeof(*ln) * (nl = wcslen(n)));
426 for(i = 0; i < nl; i++)
427 ln[i] = towlower(n[i]);
428 lh = alloca(sizeof(*lh) * (hl = wcslen(h)));
431 for(i = 0; i < nl; i++)
432 lh[i] = towlower(h[i]);
436 for(o = 0; o < nl; o++)
438 if(lh[i + o] != ln[o])
445 lh[i + nl] = towlower(h[i + nl]);
450 #ifndef HAVE_WCSCASECMP
451 int wcscasecmp(const wchar_t *s1, const wchar_t *s2)
453 for(; (towlower(*s1) == towlower(*s2)) && (*s1 != L'\0'); s1++, s2++);
454 return(towlower(*s1) - towlower(*s2));
458 char *hexencode(char *data, size_t datalen)
461 size_t bufsize, bufdata;
465 bufsize = bufdata = 0;
466 for(; datalen > 0; datalen--, data++)
468 dig = (*data & 0xF0) >> 4;
470 this = 'A' + dig - 10;
476 this = 'A' + dig - 10;
485 char *hexdecode(char *data, size_t *len)
487 char *buf, this, bit;
488 size_t bufsize, bufdata;
491 bufsize = bufdata = 0;
492 for(bit = 4, this = 0; *data; data++)
494 if((*data >= 'A') && (*data <= 'F'))
496 this |= (this & 0x0F) | ((*data - 'A' + 10) << bit);
497 } else if((*data >= 'a') && (*data <= 'f')) {
498 this |= (this & 0x0F) | ((*data - 'a' + 10) << bit);
499 } else if((*data >= '0') && (*data <= '9')) {
500 this |= (this & 0x0F) | ((*data - '0') << bit);
501 } else if(*data == '\n') {
527 char *base64encode(char *data, size_t datalen)
530 size_t bufsize, bufdata;
535 bufsize = bufdata = 0;
538 addtobuf(buf, base64set[(data[0] & 0xfc) >> 2]);
539 addtobuf(buf, base64set[((data[0] & 0x03) << 4) | ((data[1] & 0xf0) >> 4)]);
540 addtobuf(buf, base64set[((data[1] & 0x0f) << 2) | ((data[2] & 0xc0) >> 6)]);
541 addtobuf(buf, base64set[data[2] & 0x3f]);
547 addtobuf(buf, base64set[(data[0] & 0xfc) >> 2]);
548 addtobuf(buf, base64set[(data[0] & 0x03) << 4]);
549 bufcat(buf, "==", 2);
553 addtobuf(buf, base64set[(data[0] & 0xfc) >> 2]);
554 addtobuf(buf, base64set[((data[0] & 0x03) << 4) | ((data[1] & 0xf0) >> 4)]);
555 addtobuf(buf, base64set[(data[1] & 0x0f) << 2]);
562 char *base64decode(char *data, size_t *datalen)
566 size_t bufsize, bufdata;
569 bufsize = bufdata = 0;
572 for(; *data > 0; data++)
574 c = (int)(unsigned char)*data;
579 if(base64rev[c] == -1)
588 cur |= base64rev[c] >> -b;
593 cur |= base64rev[c] << b;
601 char *base32encode(char *data, size_t datalen)
604 size_t bufsize, bufdata;
609 bufsize = bufdata = 0;
612 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
613 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
614 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
615 addtobuf(buf, base32set[((data[1] & 0x01) << 4) | ((data[2] & 0xf0) >> 4)]);
616 addtobuf(buf, base32set[((data[2] & 0x0f) << 1) | ((data[3] & 0x80) >> 7)]);
617 addtobuf(buf, base32set[((data[3] & 0x7c) >> 2)]);
618 addtobuf(buf, base32set[((data[3] & 0x03) << 3) | ((data[4] & 0xe0) >> 5)]);
619 addtobuf(buf, base32set[data[4] & 0x1f]);
625 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
626 addtobuf(buf, base32set[((data[0] & 0x07) << 2)]);
627 bufcat(buf, "======", 6);
631 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
632 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
633 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
634 addtobuf(buf, base32set[((data[1] & 0x01) << 4)]);
635 bufcat(buf, "====", 4);
639 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
640 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
641 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
642 addtobuf(buf, base32set[((data[1] & 0x01) << 4) | ((data[2] & 0xf0) >> 4)]);
643 addtobuf(buf, base32set[((data[2] & 0x0f) << 1)]);
644 bufcat(buf, "===", 3);
648 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
649 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
650 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
651 addtobuf(buf, base32set[((data[1] & 0x01) << 4) | ((data[2] & 0xf0) >> 4)]);
652 addtobuf(buf, base32set[((data[2] & 0x0f) << 1) | ((data[3] & 0x80) >> 7)]);
653 addtobuf(buf, base32set[((data[3] & 0x7c) >> 2)]);
654 addtobuf(buf, base32set[((data[3] & 0x03) << 3)]);
661 char *base32decode(char *data, size_t *datalen)
665 size_t bufsize, bufdata;
668 bufsize = bufdata = 0;
671 for(; *data > 0; data++)
673 c = (int)(unsigned char)*data;
678 if(base32rev[c] == -1)
687 cur |= base32rev[c] >> -b;
692 cur |= base32rev[c] << b;
700 void _freeparr(void **arr)
706 for(buf = arr; *buf != NULL; buf++)
711 int _parrlen(void **arr)
717 for(i = 0; *arr != NULL; arr++)
722 char *getetcpath(char *binpath)
726 size_t etcpathsize, etcpathdata;
729 etcpathsize = etcpathdata = 0;
737 for(p = binpath; *p && (*p != ':'); p++);
738 for(; (p >= binpath) && (*p != '/'); p--);
742 addtobuf(etcpath, ':');
743 bufcat(etcpath, binpath, p - binpath + 1);
744 bufcat(etcpath, "etc", 3);
746 } while((binpath = strchr(binpath, ':')) != NULL);
747 addtobuf(etcpath, 0);
751 char *findfile(char *name, char *homedir, int filldef)
753 char *path, *binpath, *etcpath, *p;
760 mode = R_OK | (filldef ? W_OK : 0);
761 homeonly = homedir != NULL;
763 if(!strchr(name, '/'))
766 homedir = getenv("HOME");
767 if((homedir == NULL) && ((pw = getpwuid(getuid())) != NULL))
768 homedir = pw->pw_dir;
769 if((homedir != NULL) && ((path = sprintf2("%s/.%s", homedir, name)) != NULL))
771 if(!access(path, mode))
779 if(strchr(name, '/') != NULL)
781 if(!access(name, mode))
782 return(sstrdup(name));
784 if((binpath = getenv("PATH")) == NULL)
785 etcpath = sstrdup("/usr/local/etc:/etc:/usr/etc");
787 etcpath = getetcpath(binpath);
788 for(p = strtok(etcpath, ":"); p != NULL; p = strtok(NULL, ":"))
790 if((path = sprintf2("%s/%s", p, name)) != NULL)
792 if(!access(path, mode))
806 return(sprintf2("%s/.%s", homedir, name));
807 return(sprintf2("/etc/%s", name));
813 struct strpair *newstrpair(char *key, char *val, struct strpair **list)
815 struct strpair *pair;
817 pair = smalloc(sizeof(*pair));
818 memset(pair, 0, sizeof(*pair));
820 pair->key = sstrdup(key);
822 pair->val = sstrdup(val);
831 void freestrpair(struct strpair *pair, struct strpair **list)
835 for(cur = *list; cur != NULL; list = &(cur->next), cur = cur->next)
848 char *spfind(struct strpair *list, char *key)
850 for(; list != NULL; list = list->next)
852 if(!strcmp(list->key, key))
858 struct wcspair *newwcspair(wchar_t *key, wchar_t *val, struct wcspair **list)
860 struct wcspair *pair;
862 pair = smalloc(sizeof(*pair));
863 memset(pair, 0, sizeof(*pair));
865 pair->key = swcsdup(key);
867 pair->val = swcsdup(val);
876 void freewcspair(struct wcspair *pair, struct wcspair **list)
880 for(cur = *list; cur != NULL; list = &(cur->next), cur = cur->next)
893 wchar_t *wpfind(struct wcspair *list, wchar_t *key)
895 for(; list != NULL; list = list->next)
897 if(!wcscmp(list->key, key))
903 static int btheight(struct btree *tree)
910 static void btsetheight(struct btree *tree)
916 lh = btheight(tree->l);
917 rh = btheight(tree->r);
918 tree->h = ((lh > rh)?lh:rh) + 1;
921 static void bbtrl(struct btree **tree);
923 static void bbtrr(struct btree **tree)
925 struct btree *m, *l, *r;
927 if(btheight((*tree)->l->r) > btheight((*tree)->l->l))
939 static void bbtrl(struct btree **tree)
941 struct btree *m, *l, *r;
943 if(btheight((*tree)->r->l) > btheight((*tree)->r->r))
955 int bbtreedel(struct btree **tree, void *item, int (*cmp)(void *, void *))
958 struct btree *s, **sp, *o;
962 if((c = cmp(item, (*tree)->d)) < 0) {
963 r = bbtreedel(&(*tree)->l, item, cmp);
965 r = bbtreedel(&(*tree)->r, item, cmp);
969 if(((*tree)->r != NULL) && ((*tree)->l != NULL)) {
972 while(s->l != NULL) {
980 } else if((*tree)->l != NULL) {
982 } else if((*tree)->r != NULL) {
991 if(btheight((*tree)->l) > btheight((*tree)->r) + 1)
993 if(btheight((*tree)->r) > btheight((*tree)->l) + 1)
999 int bbtreeput(struct btree **tree, void *item, int (*cmp)(void *, void *))
1004 *tree = smalloc(sizeof(**tree));
1005 (*tree)->l = (*tree)->r = NULL;
1010 if((c = cmp(item, (*tree)->d)) < 0)
1011 r = bbtreeput(&(*tree)->l, item, cmp);
1013 r = bbtreeput(&(*tree)->r, item, cmp);
1017 if(btheight((*tree)->l) > btheight((*tree)->r) + 1)
1019 if(btheight((*tree)->r) > btheight((*tree)->l) + 1)
1024 void *btreeget(struct btree *tree, void *key, int (*cmp)(void *, void *))
1031 c = cmp(key, tree->d);
1041 void btreefree(struct btree *tree)
1050 static void *btreenext(void **iterp)
1052 struct treeiter *iter;
1056 if((iter = *iterp) == NULL)
1059 s = iter->st[iter->sp].s;
1060 n = iter->st[iter->sp].n;
1062 iter->st[iter->sp].s = 1;
1065 sizebuf2(iter->st, iter->sp + 1, 1);
1066 iter->st[iter->sp].s = 0;
1067 iter->st[iter->sp].n = n->l;
1070 iter->st[iter->sp].s = 2;
1071 return(iter->st[iter->sp].n->d);
1073 iter->st[iter->sp].s = 3;
1076 sizebuf2(iter->st, iter->sp + 1, 1);
1077 iter->st[iter->sp].s = 0;
1078 iter->st[iter->sp].n = n->r;
1092 static void *btreefirst(struct btree *tree, void **iterp)
1094 struct treeiter *iter;
1100 *iterp = iter = memset(smalloc(sizeof(*iter)), 0, sizeof(*iter));
1101 sizebuf2(iter->st, 1, 1);
1102 iter->st[0].n = tree;
1104 return(btreenext(iterp));
1107 static void btreestop(void **iterp)
1109 struct treeiter *iter;
1111 if((iter = *iterp) == NULL)
1113 if(iter->st != NULL)
1119 void *btreeiter(struct btree *tree)
1121 static void *iter = NULL;
1124 return(btreenext(&iter));
1128 return(btreefirst(tree, &iter));