Dolda2000 GitWeb / doldaconnect.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Fixed a bug in btreenext.
[doldaconnect.git] / common / utils.c
CommitLineData
d3372da9 1/*
2 * Dolda Connect - Modular multiuser Direct Connect-style client
302a2600 3 * Copyright (C) 2004 Fredrik Tolf <fredrik@dolda2000.com>
d3372da9 4 *
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.
9 *
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.
14 *
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
18*/
b5010caa 19#include <stdlib.h>
d3372da9 20#include <stdarg.h>
21#include <stdio.h>
22#include <wchar.h>
23#include <iconv.h>
24#include <errno.h>
25#include <string.h>
26#include <wctype.h>
27#include <langinfo.h>
28#include <pwd.h>
29#include <unistd.h>
30#include <sys/time.h>
31#include <netinet/in.h>
e304af88 32#include <alloca.h>
d3372da9 33
34#ifdef HAVE_CONFIG_H
35#include <config.h>
36#endif
8b17e919 37#include <utils.h>
38#include <log.h>
d3372da9 39
a1e6d478
FT		 40struct treeiter {
41 struct {
42 struct btree *n;
43 int s;
44 } *st;
0e238ce6
FT		 45 size_t stsize;
46 int sp;
a1e6d478
FT		 47};
48
d3372da9 49static char *base64set = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
50static int base64rev[] = {
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, -1, -1, -1, -1, -1,
53 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63,
54 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1,
55 -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
56 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
57 -1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
58 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -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,
66 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
67};
68static char *base32set = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567";
69static int base32rev[] = {
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, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
73 -1, -1, 26, 27, 28, 29, 30, 31, -1, -1, -1, -1, -1, -1, -1, -1,
74 -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
75 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
76 -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
77 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -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,
85 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
86};
87
88char *vsprintf2(char *format, va_list al)
89{
90 int ret;
91 char *buf;
4dffb251 92 va_list al2;
d3372da9 93
4dffb251 94 va_copy(al2, al);
fe8c6c8e 95 ret = vsnprintf(NULL, 0, format, al2);
96 va_end(al2);
d3372da9 97 if((buf = malloc(ret + 1)) == NULL)
98 {
99 LOGOOM(ret + 1);
100 return(NULL);
101 }
4dffb251 102 va_copy(al2, al);
fe8c6c8e 103 vsnprintf(buf, ret + 1, format, al2);
104 va_end(al2);
d3372da9 105 return(buf);
106}
107
108char *sprintf2(char *format, ...)
109{
110 va_list args;
111 char *buf;
112
113 va_start(args, format);
114 buf = vsprintf2(format, args);
115 va_end(args);
116 return(buf);
117}
118
119wchar_t *vswprintf2(wchar_t *format, va_list al)
120{
121 int ret;
122 wchar_t *buf;
123 size_t bufsize;
4dffb251 124 va_list al2;
d3372da9 125
126 buf = smalloc(sizeof(wchar_t) * (bufsize = 1024));
4dffb251 127 while(1)
128 {
129 va_copy(al2, al);
130 ret = vswprintf(buf, bufsize, format, al2);
fe8c6c8e 131 va_end(al2);
4dffb251 132 if(ret >= 0)
133 break;
d3372da9 134 buf = srealloc(buf, sizeof(wchar_t) * (bufsize *= 2));
4dffb251 135 }
d3372da9 136 if(bufsize > ret + 1)
137 buf = srealloc(buf, sizeof(wchar_t) * (ret + 1));
138 return(buf);
139}
140
141wchar_t *swprintf2(wchar_t *format, ...)
142{
143 va_list args;
144 wchar_t *buf;
145
146 va_start(args, format);
147 buf = vswprintf2(format, args);
148 va_end(args);
149 return(buf);
150}
151
edccab15 152int havecharset(char *charset)
153{
154 iconv_t cd;
155
156 if((cd = iconv_open("wchar_t", charset)) == (iconv_t)-1)
157 return(0);
158 iconv_close(cd);
159 if((cd = iconv_open(charset, "wchar_t")) == (iconv_t)-1)
160 return(0);
161 iconv_close(cd);
162 return(1);
163}
164
d3372da9 165wchar_t *icmbstowcs(char *mbs, char *charset)
166{
167 int ret;
168 char *buf;
169 char *p, *p2;
170 size_t len1, len2, bufsize, data;
171 iconv_t cd;
172
173 len1 = strlen(mbs) + 1;
174 bufsize = len2 = len1 * sizeof(wchar_t);
175 if((buf = malloc(bufsize)) == NULL)
176 {
177 LOGOOM(bufsize);
178 return(NULL);
179 }
180 if(charset == NULL)
181 charset = nl_langinfo(CODESET);
182 if((cd = iconv_open("wchar_t", charset)) == (iconv_t)-1)
183 {
8b17e919 184#ifdef DAEMON
d3372da9 185 flog(LOG_ERR, "icmbstowcs: could not open iconv structure for %s: %s", charset, strerror(errno));
8b17e919 186#endif
d3372da9 187 free(buf);
188 return(NULL);
189 }
190 p = buf;
191 while(len1 > 0)
192 {
193 ret = iconv(cd, &mbs, &len1, &p, &len2);
194 if(ret < 0)
195 {
196 if(errno == E2BIG)
197 {
198 data = p - buf;
199 len2 += 128;
200 bufsize += 128;
201 if((p2 = realloc(buf, bufsize)) == NULL)
202 {
203 LOGOOM(bufsize);
204 free(buf);
205 iconv_close(cd);
206 return(NULL);
207 }
208 buf = p2;
209 p = buf + data;
210 } else {
211 free(buf);
212 iconv_close(cd);
213 return(NULL);
214 }
215 }
216 }
217 if(len2 > 0)
218 buf = realloc(buf, p - buf);
219 iconv_close(cd);
220 return((wchar_t *)buf);
221}
222
223wchar_t *icsmbstowcs(char *mbs, char *charset, wchar_t *def)
224{
225 static wchar_t *buf = NULL;
226
227 if(buf != NULL)
228 free(buf);
229 if((buf = icmbstowcs(mbs, charset)) == NULL)
230 {
96cf7d1f 231 if((def != NULL) && (*def == L'~'))
d3372da9 232 {
8b17e919 233#ifdef DAEMON
d3372da9 234 flog(LOG_WARNING, "icsmbstowcs: could not convert wcs string into charset %s: %s", charset, strerror(errno));
8b17e919 235#endif
d3372da9 236 def++;
237 }
238 return(def);
239 }
240 return(buf);
241}
242
243char *icwcstombs(wchar_t *wcs, char *charset)
244{
245 int ret;
246 char *buf;
247 char *p, *p2;
248 size_t len1, len2, bufsize, data;
249 iconv_t cd;
250
251 len1 = sizeof(wchar_t) * (wcslen(wcs) + 1);
252 bufsize = len2 = len1;
253 if((buf = malloc(bufsize)) == NULL)
254 {
8b17e919 255#ifdef DAEMON
d3372da9 256 LOGOOM(bufsize);
8b17e919 257#endif
d3372da9 258 return(NULL);
259 }
260 if(charset == NULL)
261 charset = nl_langinfo(CODESET);
262 if((cd = iconv_open(charset, "wchar_t")) == (iconv_t)-1)
263 {
8b17e919 264#ifdef DAEMON
d3372da9 265 flog(LOG_ERR, "icwcstombs: could not open iconv structure for %s: %s", charset, strerror(errno));
8b17e919 266#endif
d3372da9 267 free(buf);
268 return(NULL);
269 }
270 p = buf;
271 while(len1 > 0)
272 {
273 ret = iconv(cd, (char **)&wcs, &len1, &p, &len2);
274 if(ret < 0)
275 {
276 if(errno == E2BIG)
277 {
278 data = p - buf;
279 len2 += 128;
280 bufsize += 128;
281 if((p2 = realloc(buf, bufsize)) == NULL)
282 {
283 LOGOOM(bufsize);
284 free(buf);
285 iconv_close(cd);
286 return(NULL);
287 }
288 buf = p2;
289 p = buf + data;
290 } else {
291 free(buf);
292 iconv_close(cd);
293 return(NULL);
294 }
295 }
296 }
297 if(len2 > 0)
298 buf = realloc(buf, p - buf);
299 iconv_close(cd);
300 return(buf);
301}
302
303char *icswcstombs(wchar_t *wcs, char *charset, char *def)
304{
305 static char *buf = NULL;
306
307 if(buf != NULL)
308 free(buf);
309 if((buf = icwcstombs(wcs, charset)) == NULL)
310 {
96cf7d1f 311 if((def != NULL) && (*def == '~'))
d3372da9 312 {
8b17e919 313#ifdef DAEMON
d3372da9 314 flog(LOG_WARNING, "icswcstombs: could not convert mbs string from charset %s: %s", charset, strerror(errno));
8b17e919 315#endif
d3372da9 316 def++;
317 }
318 return(def);
319 }
320 return(buf);
321}
322
323wchar_t *wcstolower(wchar_t *wcs)
324{
325 wchar_t *p;
326
327 for(p = wcs; *p != L'\0'; p++)
328 *p = towlower(*p);
329 return(wcs);
330}
331
332wchar_t ucptowc(int ucp)
333{
334 int ret;
335 unsigned long ucpbuf;
336 char *buf;
337 char *mbsp, *p, *p2;
338 wchar_t res;
339 size_t len1, len2, bufsize, data;
340 iconv_t cd;
341
342 ucpbuf = htonl(ucp);
343 mbsp = (char *)&ucpbuf;
344 len1 = 4;
345 bufsize = len2 = len1 * sizeof(wchar_t);
346 if((buf = malloc(bufsize)) == NULL)
347 {
348 LOGOOM(bufsize);
349 return(L'\0');
350 }
351 if((cd = iconv_open("wchar_t", "UCS-4BE")) == (iconv_t)-1)
352 {
8b17e919 353#ifdef DAEMON
d3372da9 354 flog(LOG_ERR, "ucptowc: could not open iconv structure for UCS-4BE: %s", strerror(errno));
8b17e919 355#endif
d3372da9 356 free(buf);
357 return(L'\0');
358 }
359 p = buf;
360 while(len1 > 0)
361 {
362 ret = iconv(cd, &mbsp, &len1, &p, &len2);
363 if(ret < 0)
364 {
365 if(errno == E2BIG)
366 {
367 data = p - buf;
368 len2 += 128;
369 bufsize += 128;
370 if((p2 = realloc(buf, bufsize)) == NULL)
371 {
372 LOGOOM(bufsize);
373 free(buf);
374 iconv_close(cd);
375 return(L'\0');
376 }
377 buf = p2;
378 p = buf + data;
379 } else {
380 free(buf);
381 iconv_close(cd);
382 return(L'\0');
383 }
384 }
385 }
386 if(len2 > 0)
387 buf = realloc(buf, p - buf);
388 iconv_close(cd);
389 res = *(wchar_t *)buf;
390 free(buf);
391 return(res);
392}
393
394void _sizebuf(void **buf, size_t *bufsize, size_t reqsize, size_t elsize, int algo)
395{
396 if(*bufsize >= reqsize)
397 return;
398 switch(algo)
399 {
400 case 0:
401 *buf = srealloc(*buf, elsize * ((*bufsize) = reqsize));
402 break;
403 case 1:
404 if(*bufsize == 0)
405 *bufsize = 1;
406 while(*bufsize < reqsize)
407 *bufsize <<= 1;
408 *buf = srealloc(*buf, elsize * (*bufsize));
409 break;
410 }
411}
412
413double ntime(void)
414{
415 struct timeval tv;
416
417 gettimeofday(&tv, NULL);
418 return((double)tv.tv_sec + ((double)tv.tv_usec / 1000000.0));
419}
420
421int wcsexists(wchar_t *h, wchar_t *n)
422{
423 int i, o, nl, hl;
424 wchar_t *ln, *lh;
425
426 ln = alloca(sizeof(*ln) * (nl = wcslen(n)));
427 for(i = 0; i < nl; i++)
428 ln[i] = towlower(n[i]);
429 lh = alloca(sizeof(*lh) * (hl = wcslen(h)));
430 if(nl > hl)
431 return(0);
432 for(i = 0; i < nl; i++)
433 lh[i] = towlower(h[i]);
434 i = 0;
435 while(1)
436 {
437 for(o = 0; o < nl; o++)
438 {
439 if(lh[i + o] != ln[o])
440 break;
441 }
442 if(o == nl)
443 return(1);
444 if(i == hl - nl)
445 return(0);
446 lh[i + nl] = towlower(h[i + nl]);
447 i++;
448 }
449}
450
451#ifndef HAVE_WCSCASECMP
452int wcscasecmp(const wchar_t *s1, const wchar_t *s2)
453{
2b1082a6 454 for(; (towlower(*s1) == towlower(*s2)) && (*s1 != L'\0'); s1++, s2++);
d3372da9 455 return(towlower(*s1) - towlower(*s2));
456}
457#endif
458
459char *hexencode(char *data, size_t datalen)
460{
461 char *buf, this;
462 size_t bufsize, bufdata;
463 int dig;
464
465 buf = NULL;
466 bufsize = bufdata = 0;
467 for(; datalen > 0; datalen--, data++)
468 {
469 dig = (*data & 0xF0) >> 4;
470 if(dig > 9)
471 this = 'A' + dig - 10;
472 else
473 this = dig + '0';
474 addtobuf(buf, this);
475 dig = *data & 0x0F;
476 if(dig > 9)
477 this = 'A' + dig - 10;
478 else
479 this = dig + '0';
480 addtobuf(buf, this);
481 }
482 addtobuf(buf, 0);
483 return(buf);
484}
485
486char *hexdecode(char *data, size_t *len)
487{
b9707562 488 char *buf, this, bit;
d3372da9 489 size_t bufsize, bufdata;
490
491 buf = NULL;
492 bufsize = bufdata = 0;
b9707562 493 for(bit = 4, this = 0; *data; data++)
d3372da9 494 {
495 if((*data >= 'A') && (*data <= 'F'))
496 {
b9707562 497 this |= (this & 0x0F) | ((*data - 'A' + 10) << bit);
498 } else if((*data >= 'a') && (*data <= 'f')) {
499 this |= (this & 0x0F) | ((*data - 'a' + 10) << bit);
d3372da9 500 } else if((*data >= '0') && (*data <= '9')) {
b9707562 501 this |= (this & 0x0F) | ((*data - '0') << bit);
502 } else if(*data == '\n') {
503 continue;
d3372da9 504 } else {
505 if(buf != NULL)
506 free(buf);
507 return(NULL);
508 }
b9707562 509 if(bit == 4) {
510 bit = 0;
d3372da9 511 } else {
b9707562 512 bit = 4;
513 addtobuf(buf, this);
514 this = 0;
d3372da9 515 }
b9707562 516 }
517 if(bit != 4) {
518 if(buf != NULL)
519 free(buf);
520 return(NULL);
d3372da9 521 }
522 addtobuf(buf, 0);
523 if(len != NULL)
524 *len = bufdata - 1;
525 return(buf);
526}
527
528char *base64encode(char *data, size_t datalen)
529{
530 char *buf;
531 size_t bufsize, bufdata;
532
533 if(datalen == 0)
534 return(sstrdup(""));
535 buf = NULL;
536 bufsize = bufdata = 0;
537 while(datalen >= 3)
538 {
539 addtobuf(buf, base64set[(data[0] & 0xfc) >> 2]);
540 addtobuf(buf, base64set[((data[0] & 0x03) << 4) | ((data[1] & 0xf0) >> 4)]);
541 addtobuf(buf, base64set[((data[1] & 0x0f) << 2) | ((data[2] & 0xc0) >> 6)]);
542 addtobuf(buf, base64set[data[2] & 0x3f]);
543 datalen -= 3;
544 data += 3;
545 }
546 if(datalen == 1)
547 {
548 addtobuf(buf, base64set[(data[0] & 0xfc) >> 2]);
549 addtobuf(buf, base64set[(data[0] & 0x03) << 4]);
550 bufcat(buf, "==", 2);
551 }
552 if(datalen == 2)
553 {
554 addtobuf(buf, base64set[(data[0] & 0xfc) >> 2]);
555 addtobuf(buf, base64set[((data[0] & 0x03) << 4) | ((data[1] & 0xf0) >> 4)]);
556 addtobuf(buf, base64set[(data[1] & 0x0f) << 2]);
557 addtobuf(buf, '=');
558 }
559 addtobuf(buf, 0);
560 return(buf);
561}
562
563char *base64decode(char *data, size_t *datalen)
564{
565 int b, c;
566 char *buf, cur;
567 size_t bufsize, bufdata;
568
569 buf = NULL;
570 bufsize = bufdata = 0;
571 cur = 0;
572 b = 8;
573 for(; *data > 0; data++)
574 {
575 c = (int)(unsigned char)*data;
576 if(c == '=')
577 break;
65399bc2 578 if(c == '\n')
579 continue;
d3372da9 580 if(base64rev[c] == -1)
581 {
582 if(buf != NULL)
583 free(buf);
584 return(NULL);
585 }
586 b -= 6;
587 if(b <= 0)
588 {
589 cur |= base64rev[c] >> -b;
590 addtobuf(buf, cur);
591 b += 8;
592 cur = 0;
593 }
594 cur |= base64rev[c] << b;
595 }
596 if(datalen != NULL)
597 *datalen = bufdata;
598 addtobuf(buf, 0);
599 return(buf);
600}
601
602char *base32encode(char *data, size_t datalen)
603{
604 char *buf;
605 size_t bufsize, bufdata;
606
607 if(datalen == 0)
608 return(sstrdup(""));
609 buf = NULL;
610 bufsize = bufdata = 0;
611 while(datalen >= 5)
612 {
613 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
614 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
615 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
616 addtobuf(buf, base32set[((data[1] & 0x01) << 4) | ((data[2] & 0xf0) >> 4)]);
617 addtobuf(buf, base32set[((data[2] & 0x0f) << 1) | ((data[3] & 0x80) >> 7)]);
618 addtobuf(buf, base32set[((data[3] & 0x7c) >> 2)]);
619 addtobuf(buf, base32set[((data[3] & 0x03) << 3) | ((data[4] & 0xe0) >> 5)]);
620 addtobuf(buf, base32set[data[4] & 0x1f]);
621 datalen -= 5;
622 data += 5;
623 }
624 if(datalen == 1)
625 {
626 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
627 addtobuf(buf, base32set[((data[0] & 0x07) << 2)]);
628 bufcat(buf, "======", 6);
629 }
630 if(datalen == 2)
631 {
632 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
633 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
634 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
635 addtobuf(buf, base32set[((data[1] & 0x01) << 4)]);
636 bufcat(buf, "====", 4);
637 }
638 if(datalen == 3)
639 {
640 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
641 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
642 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
643 addtobuf(buf, base32set[((data[1] & 0x01) << 4) | ((data[2] & 0xf0) >> 4)]);
644 addtobuf(buf, base32set[((data[2] & 0x0f) << 1)]);
645 bufcat(buf, "===", 3);
646 }
647 if(datalen == 4)
648 {
649 addtobuf(buf, base32set[((data[0] & 0xf8) >> 3)]);
650 addtobuf(buf, base32set[((data[0] & 0x07) << 2) | ((data[1] & 0xc0) >> 6)]);
651 addtobuf(buf, base32set[((data[1] & 0x3e) >> 1)]);
652 addtobuf(buf, base32set[((data[1] & 0x01) << 4) | ((data[2] & 0xf0) >> 4)]);
653 addtobuf(buf, base32set[((data[2] & 0x0f) << 1) | ((data[3] & 0x80) >> 7)]);
654 addtobuf(buf, base32set[((data[3] & 0x7c) >> 2)]);
655 addtobuf(buf, base32set[((data[3] & 0x03) << 3)]);
656 bufcat(buf, "=", 1);
657 }
658 addtobuf(buf, 0);
659 return(buf);
660}
661
662char *base32decode(char *data, size_t *datalen)
663{
664 int b, c;
665 char *buf, cur;
666 size_t bufsize, bufdata;
667
668 buf = NULL;
669 bufsize = bufdata = 0;
670 cur = 0;
671 b = 8;
672 for(; *data > 0; data++)
673 {
674 c = (int)(unsigned char)*data;
675 if(c == '=')
676 break;
65399bc2 677 if(c == '\n')
678 continue;
d3372da9 679 if(base32rev[c] == -1)
680 {
681 if(buf != NULL)
682 free(buf);
683 return(NULL);
684 }
685 b -= 5;
686 if(b <= 0)
687 {
688 cur |= base32rev[c] >> -b;
689 addtobuf(buf, cur);
690 b += 8;
691 cur = 0;
692 }
693 cur |= base32rev[c] << b;
694 }
695 if(datalen != NULL)
696 *datalen = bufdata;
697 addtobuf(buf, 0);
698 return(buf);
699}
700
701void _freeparr(void **arr)
702{
703 void **buf;
704
705 if(arr == NULL)
706 return;
707 for(buf = arr; *buf != NULL; buf++)
708 free(*buf);
709 free(arr);
710}
711
712int _parrlen(void **arr)
713{
714 int i;
715
716 if(arr == NULL)
717 return(0);
718 for(i = 0; *arr != NULL; arr++)
719 i++;
720 return(i);
721}
722
723char *getetcpath(char *binpath)
724{
73247114 725 int f;
d3372da9 726 char *etcpath, *p;
727 size_t etcpathsize, etcpathdata;
728
729 etcpath = NULL;
730 etcpathsize = etcpathdata = 0;
73247114 731 f = 1;
d3372da9 732 do
733 {
73247114 734 if(f)
735 f = 0;
736 else
737 binpath++;
d3372da9 738 for(p = binpath; *p && (*p != ':'); p++);
739 for(; (p >= binpath) && (*p != '/'); p--);
740 if(p >= binpath)
741 {
742 if(etcpathdata > 0)
743 addtobuf(etcpath, ':');
744 bufcat(etcpath, binpath, p - binpath + 1);
745 bufcat(etcpath, "etc", 3);
746 }
747 } while((binpath = strchr(binpath, ':')) != NULL);
748 addtobuf(etcpath, 0);
749 return(etcpath);
750}
751
1df8650c 752char *findfile(char *name, char *homedir, int filldef)
d3372da9 753{
754 char *path, *binpath, *etcpath, *p;
48b2d6d1 755 struct passwd *pw;
1df8650c 756 int mode, homeonly;
d3372da9 757
1df8650c 758 if(name == NULL)
759 return(NULL);
760
41ac4371 761 mode = R_OK | (filldef ? W_OK : 0);
1df8650c 762 homeonly = homedir != NULL;
763
764 if(!strchr(name, '/'))
765 {
41ac4371 766 if(homedir == NULL)
767 homedir = getenv("HOME");
48b2d6d1 768 if((homedir == NULL) && ((pw = getpwuid(getuid())) != NULL))
769 homedir = pw->pw_dir;
1df8650c 770 if((homedir != NULL) && ((path = sprintf2("%s/.%s", homedir, name)) != NULL))
48b2d6d1 771 {
41ac4371 772 if(!access(path, mode))
48b2d6d1 773 return(path);
774 free(path);
775 }
d3372da9 776 }
1df8650c 777
778 if(!homeonly)
d3372da9 779 {
1df8650c 780 if(strchr(name, '/') != NULL)
d3372da9 781 {
1df8650c 782 if(!access(name, mode))
783 return(sstrdup(name));
d3372da9 784 } else {
785 if((binpath = getenv("PATH")) == NULL)
786 etcpath = sstrdup("/usr/local/etc:/etc:/usr/etc");
787 else
788 etcpath = getetcpath(binpath);
789 for(p = strtok(etcpath, ":"); p != NULL; p = strtok(NULL, ":"))
790 {
1df8650c 791 if((path = sprintf2("%s/%s", p, name)) != NULL)
d3372da9 792 {
41ac4371 793 if(!access(path, mode))
d3372da9 794 {
795 free(etcpath);
796 return(path);
797 }
798 free(path);
799 }
800 }
801 free(etcpath);
802 }
803 }
1df8650c 804
48b2d6d1 805 if(filldef) {
1df8650c 806 if(homedir)
807 return(sprintf2("%s/.%s", homedir, name));
808 return(sprintf2("/etc/%s", name));
48b2d6d1 809 } else {
810 return(NULL);
811 }
d3372da9 812}
9ec790e8 813
ba087e49
FT		 814struct strpair *newstrpair(char *key, char *val, struct strpair **list)
815{
816 struct strpair *pair;
817
818 pair = smalloc(sizeof(*pair));
819 memset(pair, 0, sizeof(*pair));
820 if(key != NULL)
821 pair->key = sstrdup(key);
822 if(val != NULL)
823 pair->val = sstrdup(val);
824 if(list != NULL)
825 {
826 pair->next = *list;
827 *list = pair;
828 }
829 return(pair);
830}
831
832void freestrpair(struct strpair *pair, struct strpair **list)
833{
834 struct strpair *cur;
835
836 for(cur = *list; cur != NULL; list = &(cur->next), cur = cur->next)
837 {
838 if(cur == pair)
839 {
840 *list = cur->next;
841 break;
842 }
843 }
844 free(pair->key);
845 free(pair->val);
846 free(pair);
847}
848
849char *spfind(struct strpair *list, char *key)
850{
851 for(; list != NULL; list = list->next)
852 {
853 if(!strcmp(list->key, key))
854 return(list->val);
855 }
856 return(NULL);
857}
858
9ec790e8 859struct wcspair *newwcspair(wchar_t *key, wchar_t *val, struct wcspair **list)
860{
861 struct wcspair *pair;
862
863 pair = smalloc(sizeof(*pair));
864 memset(pair, 0, sizeof(*pair));
865 if(key != NULL)
866 pair->key = swcsdup(key);
867 if(val != NULL)
868 pair->val = swcsdup(val);
ba087e49 869 if(list != NULL)
9ec790e8 870 {
9ec790e8 871 pair->next = *list;
872 *list = pair;
873 }
874 return(pair);
875}
876
877void freewcspair(struct wcspair *pair, struct wcspair **list)
878{
497fe07b 879 struct wcspair *cur;
880
881 for(cur = *list; cur != NULL; list = &(cur->next), cur = cur->next)
9ec790e8 882 {
497fe07b 883 if(cur == pair)
884 {
885 *list = cur->next;
886 break;
887 }
9ec790e8 888 }
889 free(pair->key);
890 free(pair->val);
891 free(pair);
892}
a2761258 893
894wchar_t *wpfind(struct wcspair *list, wchar_t *key)
895{
896 for(; list != NULL; list = list->next)
897 {
898 if(!wcscmp(list->key, key))
899 return(list->val);
900 }
901 return(NULL);
902}
ffb8ed40
FT		 903
904static int btheight(struct btree *tree)
905{
906 if(tree == NULL)
907 return(0);
908 return(tree->h);
909}
910
911static void btsetheight(struct btree *tree)
912{
913 int lh, rh;
914
915 if(tree == NULL)
916 return;
917 lh = btheight(tree->l);
918 rh = btheight(tree->r);
919 tree->h = ((lh > rh)?lh:rh) + 1;
920}
921
922static void bbtrl(struct btree **tree);
923
924static void bbtrr(struct btree **tree)
925{
926 struct btree *m, *l, *r;
927
928 if(btheight((*tree)->l->r) > btheight((*tree)->l->l))
929 bbtrl(&(*tree)->l);
930 r = (*tree);
931 l = r->l;
932 m = l->r;
933 r->l = m;
934 btsetheight(r);
935 l->r = r;
936 btsetheight(l);
937 *tree = l;
938}
939
940static void bbtrl(struct btree **tree)
941{
942 struct btree *m, *l, *r;
943
944 if(btheight((*tree)->r->l) > btheight((*tree)->r->r))
945 bbtrr(&(*tree)->r);
946 l = (*tree);
947 r = l->r;
948 m = r->l;
949 l->r = m;
950 btsetheight(l);
951 r->l = l;
952 btsetheight(r);
953 *tree = r;
954}
955
956int bbtreedel(struct btree **tree, void *item, int (*cmp)(void *, void *))
957{
958 int c, r;
959 struct btree *s, **sp, *o;
960
961 if(*tree == NULL)
962 return(0);
963 if((c = cmp(item, (*tree)->d)) < 0) {
964 r = bbtreedel(&(*tree)->l, item, cmp);
965 } else if(c > 0) {
966 r = bbtreedel(&(*tree)->r, item, cmp);
967 } else {
968 r = 1;
969 o = *tree;
970 if(((*tree)->r != NULL) && ((*tree)->l != NULL)) {
971 sp = &(*tree)->r;
972 s = *sp;
973 while(s->l != NULL) {
974 sp = &(s->l);
975 s = *sp;
976 }
977 *sp = NULL;
978 s->l = (*tree)->l;
979 s->r = (*tree)->r;
980 *tree = s;
981 } else if((*tree)->l != NULL) {
982 *tree = (*tree)->l;
983 } else if((*tree)->r != NULL) {
984 *tree = (*tree)->r;
985 } else {
986 *tree = NULL;
987 }
988 free(o);
989 }
990 if(*tree != NULL) {
991 btsetheight(*tree);
992 if(btheight((*tree)->l) > btheight((*tree)->r) + 1)
993 bbtrr(tree);
994 if(btheight((*tree)->r) > btheight((*tree)->l) + 1)
995 bbtrl(tree);
996 }
997 return(r);
998}
999
1000int bbtreeput(struct btree **tree, void *item, int (*cmp)(void *, void *))
1001{
1002 int c, r;
1003
1004 if(*tree == NULL) {
1005 *tree = smalloc(sizeof(**tree));
1006 (*tree)->l = (*tree)->r = NULL;
1007 (*tree)->d = item;
1008 (*tree)->h = 1;
1009 return(1);
1010 }
1011 if((c = cmp(item, (*tree)->d)) < 0)
1012 r = bbtreeput(&(*tree)->l, item, cmp);
1013 else if(c > 0)
1014 r = bbtreeput(&(*tree)->r, item, cmp);
1015 else
1016 return(0);
1017 btsetheight(*tree);
1018 if(btheight((*tree)->l) > btheight((*tree)->r) + 1)
1019 bbtrr(tree);
1020 if(btheight((*tree)->r) > btheight((*tree)->l) + 1)
1021 bbtrl(tree);
1022 return(r);
1023}
1024
1025void *btreeget(struct btree *tree, void *key, int (*cmp)(void *, void *))
1026{
1027 int c;
1028
1029 while(1) {
1030 if(tree == NULL)
1031 return(NULL);
1032 c = cmp(key, tree->d);
1033 if(c < 0)
1034 tree = tree->l;
1035 else if(c > 0)
1036 tree = tree->r;
1037 else
1038 return(tree->d);
1039 }
1040}
1041
a1e6d478
FT		 1042void btreefree(struct btree *tree)
1043{
1044 if(tree == NULL)
1045 return;
1046 btreefree(tree->l);
1047 btreefree(tree->r);
1048 free(tree);
1049}
1050
1051static void *btreenext(void **iterp)
1052{
1053 struct treeiter *iter;
1054 int s;
1055 struct btree *n;
1056
1057 if((iter = *iterp) == NULL)
1058 return(NULL);
1059 while(1) {
1060 s = iter->st[iter->sp].s;
1061 n = iter->st[iter->sp].n;
1062 if(s == 0) {
1063 iter->st[iter->sp].s = 1;
1064 if(n->l != NULL) {
1065 iter->sp++;
1066 sizebuf2(iter->st, iter->sp + 1, 1);
1067 iter->st[iter->sp].s = 0;
1068 iter->st[iter->sp].n = n->l;
1069 }
1070 } else if(s == 1) {
1071 iter->st[iter->sp].s = 2;
1072 return(iter->st[iter->sp].n->d);
1073 } else if(s == 2) {
1074 iter->st[iter->sp].s = 3;
1075 if(n->r != NULL) {
1076 iter->sp++;
1077 sizebuf2(iter->st, iter->sp + 1, 1);
1078 iter->st[iter->sp].s = 0;
1079 iter->st[iter->sp].n = n->r;
1080 }
1081 } else {
1082 iter->sp--;
1083 if(iter->sp < 0) {
1084 free(iter->st);
1085 free(iter);
1086 *iterp = NULL;
1087 return(NULL);
1088 }
1089 }
1090 }
1091}
1092
1093static void *btreefirst(struct btree *tree, void **iterp)
1094{
1095 struct treeiter *iter;
1096
1097 if(tree == NULL) {
1098 *iterp = NULL;
1099 return(NULL);
1100 }
1101 *iterp = iter = memset(smalloc(sizeof(*iter)), 0, sizeof(*iter));
1102 sizebuf2(iter->st, 1, 1);
1103 iter->st[0].n = tree;
1104 iter->st[0].s = 0;
1105 return(btreenext(iterp));
1106}
1107
1108static void btreestop(void **iterp)
1109{
1110 struct treeiter *iter;
1111
1112 if((iter = *iterp) == NULL)
1113 return;
1114 if(iter->st != NULL)
1115 free(iter->st);
1116 free(iter);
1117 *iterp = NULL;
1118}
1119
1120void *btreeiter(struct btree *tree)
1121{
1122 static void *iter = NULL;
1123
1124 if(tree == NULL) {
1125 return(btreenext(&iter));
1126 } else {
1127 if(iter != NULL)
1128 btreestop(&iter);
1129 return(btreefirst(tree, &iter));
1130 }
1131}
Dolda Connect