6883b7122851d4418388fd77ac99688020b9c916
[srvx.git] / src / tools.c
1 /* tools.c - miscellaneous utility functions
2  * Copyright 2000-2004 srvx Development Team
3  *
4  * This file is part of srvx.
5  *
6  * srvx is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with srvx; if not, write to the Free Software Foundation,
18  * Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
19  */
20
21 #include "helpfile.h"
22 #include "log.h"
23 #include "nickserv.h"
24 #include "recdb.h"
25
26 #ifdef HAVE_NETDB_H
27 #include <netdb.h>
28 #endif
29 #ifdef HAVE_SYS_SOCKET_H
30 #include <sys/socket.h>
31 #endif
32 #ifdef HAVE_ARPA_INET_H
33 #include <arpa/inet.h>
34 #endif
35
36 #define NUMNICKLOG 6
37 #define NUMNICKBASE (1 << NUMNICKLOG)
38 #define NUMNICKMASK (NUMNICKBASE - 1)
39
40 /* Yes, P10's encoding here is almost-but-not-quite MIME Base64.  Yay
41  * for gratuitous incompatibilities. */
42 static const char convert2y[256] = {
43   'A','B','C','D','E','F','G','H','I','J','K','L','M','N','O','P',
44   'Q','R','S','T','U','V','W','X','Y','Z','a','b','c','d','e','f',
45   'g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v',
46   'w','x','y','z','0','1','2','3','4','5','6','7','8','9','[',']'
47 };
48
49 static const unsigned char convert2n[256] = {
50    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
51    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
52    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
53   52,53,54,55,56,57,58,59,60,61, 0, 0, 0, 0, 0, 0,
54    0, 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,62, 0,63, 0, 0,
56    0,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, 0, 0, 0, 0, 0
58 };
59
60 static const unsigned char ctype[256] = {
61    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
62    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
63    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
64    0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 0, 0, 0, 0, 0,
65    0,10,11,12,13,14,15, 0, 0, 0, 0, 0, 0, 0, 0, 0,
66    0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
67    0,10,11,12,13,14,15, 0, 0, 0, 0, 0, 0, 0, 0, 0,
68 };
69
70 unsigned long int
71 base64toint(const char* s, int count)
72 {
73     unsigned int i = 0;
74     while (*s && count) {
75         i = (i << NUMNICKLOG) + convert2n[(unsigned char)*s++];
76         count--;
77     }
78     return i;
79 }
80
81 const char* inttobase64(char* buf, unsigned int v, unsigned int count)
82 {
83   buf[count] = '\0';
84   while (count > 0) {
85       buf[--count] = convert2y[(unsigned char)(v & NUMNICKMASK)];
86       v >>= NUMNICKLOG;
87   }
88   return buf;
89 }
90
91 unsigned int
92 irc_ntop(char *output, unsigned int out_size, const irc_in_addr_t *addr)
93 {
94     static const char hexdigits[] = "0123456789abcdef";
95     unsigned int pos;
96
97     assert(output);
98     assert(addr);
99
100     if (irc_in_addr_is_ipv4(*addr)) {
101         unsigned int ip4;
102
103         ip4 = (ntohs(addr->in6[6]) << 16) | ntohs(addr->in6[7]);
104         pos = snprintf(output, out_size, "%u.%u.%u.%u", (ip4 >> 24), (ip4 >> 16) & 255, (ip4 >> 8) & 255, ip4 & 255);
105    } else {
106         unsigned int part, max_start, max_zeros, curr_zeros, ii;
107
108         /* Find longest run of zeros. */
109         for (max_start = max_zeros = curr_zeros = ii = 0; ii < 8; ++ii) {
110             if (!addr->in6[ii])
111                 curr_zeros++;
112             else if (curr_zeros > max_zeros) {
113                 max_start = ii - curr_zeros;
114                 max_zeros = curr_zeros;
115                 curr_zeros = 0;
116             }
117         }
118         if (curr_zeros > max_zeros) {
119             max_start = ii - curr_zeros;
120             max_zeros = curr_zeros;
121         }
122
123         /* Print out address. */
124 #define APPEND(CH) do { if (pos < out_size) output[pos] = (CH); pos++; } while (0)
125         for (pos = 0, ii = 0; ii < 8; ++ii) {
126             if ((max_zeros > 0) && (ii == max_start)) {
127                 if (ii == 0)
128                     APPEND(':');
129                 APPEND(':');
130                 ii += max_zeros - 1;
131                 continue;
132             }
133             part = ntohs(addr->in6[ii]);
134             if (part >= 0x1000)
135                 APPEND(hexdigits[part >> 12]);
136             if (part >= 0x100)
137                 APPEND(hexdigits[(part >> 8) & 15]);
138             if (part >= 0x10)
139                 APPEND(hexdigits[(part >> 4) & 15]);
140             APPEND(hexdigits[part & 15]);
141             if (ii < 7)
142                 APPEND(':');
143         }
144 #undef APPEND
145         output[pos < out_size ? pos : out_size - 1] = '\0';
146     }
147
148     return pos;
149 }
150
151 unsigned int
152 irc_ntop_mask(char *output, unsigned int out_size, const irc_in_addr_t *addr, unsigned char bits)
153 {
154     char base_addr[IRC_NTOP_MAX_SIZE];
155     int len;
156
157     if (bits >= 128)
158         return irc_ntop(output, out_size, addr);
159     if (!irc_ntop(base_addr, sizeof(base_addr), addr))
160         return 0;
161     len = snprintf(output, out_size, "%s/%d", base_addr, bits);
162     if ((unsigned int)len >= out_size)
163         return 0;
164     return len;
165 }
166
167 static unsigned int
168 irc_pton_ip4(const char *input, unsigned char *pbits, uint32_t *output)
169 {
170     unsigned int dots = 0, pos = 0, part = 0, ip = 0, bits = 32;
171
172     /* Intentionally no support for bizarre IPv4 formats (plain
173      * integers, octal or hex components) -- only vanilla dotted
174      * decimal quads, optionally with trailing /nn.
175      */
176     if (input[0] == '.')
177         return 0;
178     while (1) switch (input[pos]) {
179     default:
180         if (dots < 3)
181             return 0;
182     out:
183         ip |= part << (24 - 8 * dots++);
184         *output = htonl(ip);
185         if (pbits)
186             *pbits = bits;
187         return pos;
188     case '.':
189         if (input[++pos] == '.')
190             return 0;
191         ip |= part << (24 - 8 * dots++);
192         part = 0;
193         if (input[pos] == '*') {
194             while (input[++pos] == '*') ;
195             if (input[pos] != '\0')
196                 return 0;
197             if (pbits)
198                 *pbits = dots * 8;
199             *output = htonl(ip);
200             return pos;
201         }
202         break;
203     case '/':
204         if (!pbits || !isdigit(input[pos + 1]))
205             return 0;
206         for (bits = 0; isdigit(input[++pos]); )
207             bits = bits * 10 + input[pos] - '0';
208         if (bits > 32)
209             return 0;
210         goto out;
211     case '0': case '1': case '2': case '3': case '4':
212     case '5': case '6': case '7': case '8': case '9':
213         part = part * 10 + input[pos++] - '0';
214         if (part > 255)
215             return 0;
216         break;
217     }
218 }
219
220 unsigned int
221 irc_pton(irc_in_addr_t *addr, unsigned char *bits, const char *input)
222 {
223     const char *part_start = NULL;
224     char *colon;
225     char *dot;
226     unsigned int part = 0, pos = 0, ii = 0, cpos = 8;
227
228     assert(input);
229     memset(addr, 0, sizeof(*addr));
230     colon = strchr(input, ':');
231     dot = strchr(input, '.');
232
233     if (colon && (!dot || (dot > colon))) {
234         /* Parse IPv6, possibly like ::127.0.0.1.
235          * This is pretty straightforward; the only trick is borrowed
236          * from Paul Vixie (BIND): when it sees a "::" continue as if
237          * it were a single ":", but note where it happened, and fill
238          * with zeros afterwards.
239          */
240         if (input[pos] == ':') {
241             if ((input[pos+1] != ':') || (input[pos+2] == ':'))
242                 return 0;
243             cpos = 0;
244             pos += 2;
245             part_start = input + pos;
246         }
247         while (ii < 8) switch (input[pos]) {
248         case '0': case '1': case '2': case '3': case '4':
249         case '5': case '6': case '7': case '8': case '9':
250         case 'A': case 'B': case 'C': case 'D': case 'E': case 'F':
251         case 'a': case 'b': case 'c': case 'd': case 'e': case 'f':
252             part = (part << 4) | (ctype[(unsigned char)input[pos++]] & 15);
253             if (part > 0xffff)
254                 return 0;
255             break;
256         case ':':
257             part_start = input + ++pos;
258             if (input[pos] == '.')
259                 return 0;
260             addr->in6[ii++] = htons(part);
261             part = 0;
262             if (input[pos] == ':') {
263                 if (cpos < 8)
264                     return 0;
265                 cpos = ii;
266             }
267             break;
268         case '.': {
269             uint32_t ip4;
270             unsigned int len;
271             len = irc_pton_ip4(part_start, bits, &ip4);
272             if (!len || (ii > 6))
273                 return 0;
274             memcpy(addr->in6 + ii, &ip4, sizeof(ip4));
275             if (bits)
276                 *bits += 96;
277             ii += 2;
278             pos = part_start + len - input;
279             goto finish;
280         }
281         case '/':
282             if (!bits || !isdigit(input[pos + 1]))
283                 return 0;
284             addr->in6[ii++] = htons(part);
285             for (part = 0; isdigit(input[++pos]); )
286                 part = part * 10 + input[pos] - '0';
287             if (part > 128)
288                 return 0;
289             *bits = part;
290             goto finish;
291         case '*':
292             while (input[++pos] == '*') ;
293             if (input[pos] != '\0' || cpos < 8)
294                 return 0;
295             if (bits)
296                 *bits = ii * 16;
297             return pos;
298         default:
299             addr->in6[ii++] = htons(part);
300             if (cpos == 8 && ii < 8)
301                 return 0;
302             if (bits)
303                 *bits = 128;
304             goto finish;
305         }
306     finish:
307         /* Shift stuff after "::" up and fill middle with zeros. */
308         if (cpos < 8) {
309             unsigned int jj;
310             for (jj = 0; jj < ii - cpos; jj++)
311                 addr->in6[7 - jj] = addr->in6[ii - jj - 1];
312             for (jj = 0; jj < 8 - ii; jj++)
313                 addr->in6[cpos + jj] = 0;
314         }
315     } else if (dot) {
316         unsigned int ip4;
317         pos = irc_pton_ip4(input, bits, &ip4);
318         if (pos) {
319             addr->in6[5] = htons(65535);
320             addr->in6[6] = htons(ntohl(ip4) >> 16);
321             addr->in6[7] = htons(ntohl(ip4) & 65535);
322             if (bits)
323                 *bits += 96;
324         }
325     } else if (input[0] == '*') {
326         while (input[++pos] == '*') ;
327         if (input[pos] != '\0')
328             return 0;
329         if (bits)
330             *bits = 0;
331     }
332     return pos;
333 }
334
335 const char *irc_ntoa(const irc_in_addr_t *addr)
336 {
337     static char ntoa[IRC_NTOP_MAX_SIZE];
338     irc_ntop(ntoa, sizeof(ntoa), addr);
339     return ntoa;
340 }
341
342 unsigned int
343 irc_check_mask(const irc_in_addr_t *check, const irc_in_addr_t *mask, unsigned char bits)
344 {
345     unsigned int ii;
346
347     for (ii = 0; (ii < 8) && (bits > 16); bits -= 16, ++ii)
348         if (check->in6[ii] != mask->in6[ii])
349             return 0;
350     if (ii < 8 && bits > 0
351         && (ntohs(check->in6[ii] ^ mask->in6[ii]) >> (16 - bits)))
352         return 0;
353     return 1;
354 }
355
356 static char irc_tolower[256];
357 #undef tolower
358 #define tolower(X) irc_tolower[(unsigned char)(X)]
359
360 int
361 irccasecmp(const char *stra, const char *strb) {
362     while (*stra && (tolower(*stra) == tolower(*strb)))
363         stra++, strb++;
364     return tolower(*stra) - tolower(*strb);
365 }
366
367 int
368 ircncasecmp(const char *stra, const char *strb, unsigned int len) {
369     len--;
370     while (*stra && (tolower(*stra) == tolower(*strb)) && len)
371         stra++, strb++, len--;
372     return tolower(*stra) - tolower(*strb);
373 }
374
375 const char *
376 irccasestr(const char *haystack, const char *needle) {
377     unsigned int hay_len = strlen(haystack), needle_len = strlen(needle), pos;
378     if (hay_len < needle_len)
379         return NULL;
380     for (pos=0; pos<hay_len+1-needle_len; ++pos) {
381         if ((tolower(haystack[pos]) == tolower(*needle))
382             && !ircncasecmp(haystack+pos, needle, needle_len))
383             return haystack+pos;
384     }
385     return NULL;
386 }
387
388 int
389 split_line(char *line, int irc_colon, int argv_size, char *argv[])
390 {
391     int argc = 0;
392     int n;
393     while (*line && (argc < argv_size)) {
394         while (*line == ' ')
395             *line++ = 0;
396         if (*line == ':' && irc_colon && argc > 0) {
397             /* the rest is a single parameter */
398             argv[argc++] = line + 1;
399             break;
400         }
401         if (!*line)
402             break;
403         argv[argc++] = line;
404         if (argc >= argv_size)
405             break;
406         while (*line != ' ' && *line)
407             line++;
408     }
409 #ifdef NDEBUG
410     n = 0;
411 #else
412     for (n=argc; n<argv_size; n++)
413         argv[n] = (char*)0xFEEDBEEF;
414 #endif
415     return argc;
416 }
417
418 /* This is ircu's mmatch() function, from match.c. */
419 int mmatch(const char *old_mask, const char *new_mask)
420 {
421   register const char *m = old_mask;
422   register const char *n = new_mask;
423   const char *ma = m;
424   const char *na = n;
425   int wild = 0;
426   int mq = 0, nq = 0;
427
428   while (1)
429   {
430     if (*m == '*')
431     {
432       while (*m == '*')
433         m++;
434       wild = 1;
435       ma = m;
436       na = n;
437     }
438
439     if (!*m)
440     {
441       if (!*n)
442         return 0;
443       for (m--; (m > old_mask) && (*m == '?'); m--)
444         ;
445       if ((*m == '*') && (m > old_mask) && (m[-1] != '\\'))
446         return 0;
447       if (!wild)
448         return 1;
449       m = ma;
450
451       /* Added to `mmatch' : Because '\?' and '\*' now is one character: */
452       if ((*na == '\\') && ((na[1] == '*') || (na[1] == '?')))
453         ++na;
454
455       n = ++na;
456     }
457     else if (!*n)
458     {
459       while (*m == '*')
460         m++;
461       return (*m != 0);
462     }
463     if ((*m == '\\') && ((m[1] == '*') || (m[1] == '?')))
464     {
465       m++;
466       mq = 1;
467     }
468     else
469       mq = 0;
470
471     /* Added to `mmatch' : Because '\?' and '\*' now is one character: */
472     if ((*n == '\\') && ((n[1] == '*') || (n[1] == '?')))
473     {
474       n++;
475       nq = 1;
476     }
477     else
478       nq = 0;
479
480 /*
481  * This `if' has been changed compared to match() to do the following:
482  * Match when:
483  *   old (m)         new (n)         boolean expression
484  *    *               any             (*m == '*' && !mq) ||
485  *    ?               any except '*'  (*m == '?' && !mq && (*n != '*' || nq)) ||
486  * any except * or ?  same as m       (!((*m == '*' || *m == '?') && !mq) &&
487  *                                      toLower(*m) == toLower(*n) &&
488  *                                        !((mq && !nq) || (!mq && nq)))
489  *
490  * Here `any' also includes \* and \? !
491  *
492  * After reworking the boolean expressions, we get:
493  * (Optimized to use boolean shortcircuits, with most frequently occuring
494  *  cases upfront (which took 2 hours!)).
495  */
496     if ((*m == '*' && !mq) ||
497         ((!mq || nq) && tolower(*m) == tolower(*n)) ||
498         (*m == '?' && !mq && (*n != '*' || nq)))
499     {
500       if (*m)
501         m++;
502       if (*n)
503         n++;
504     }
505     else
506     {
507       if (!wild)
508         return 1;
509       m = ma;
510
511       /* Added to `mmatch' : Because '\?' and '\*' now is one character: */
512       if ((*na == '\\') && ((na[1] == '*') || (na[1] == '?')))
513         ++na;
514
515       n = ++na;
516     }
517   }
518 }
519
520 int
521 match_ircglob(const char *text, const char *glob)
522 {
523     const char *m = glob, *n = text;
524     const char *m_tmp = glob, *n_tmp = text;
525     int star_p;
526
527     for (;;) switch (*m) {
528     case '\0':
529         if (!*n)
530             return 1;
531     backtrack:
532         if (m_tmp == glob)
533             return 0;
534         m = m_tmp;
535         n = ++n_tmp;
536         break;
537     case '\\':
538         m++;
539         /* allow escaping to force capitalization */
540         if (*m++ != *n++)
541             return 0;
542         break;
543     case '*': case '?':
544         for (star_p = 0; ; m++) {
545             if (*m == '*')
546                 star_p = 1;
547             else if (*m == '?') {
548                 if (!*n++)
549                     goto backtrack;
550             } else break;
551         }
552         if (star_p) {
553             if (!*m)
554                 return 1;
555             else if (*m == '\\') {
556                 m_tmp = ++m;
557                 if (!*m)
558                     return 0;
559                 for (n_tmp = n; *n && *n != *m; n++) ;
560             } else {
561                 m_tmp = m;
562                 for (n_tmp = n; *n && tolower(*n) != tolower(*m); n++) ;
563             }
564         }
565         /* and fall through */
566     default:
567         if (!*n)
568             return *m == '\0';
569         if (tolower(*m) != tolower(*n))
570             goto backtrack;
571         m++;
572         n++;
573         break;
574     }
575 }
576
577 extern const char *hidden_host_suffix;
578
579 int
580 user_matches_glob(struct userNode *user, const char *orig_glob, int flags)
581 {
582     char *glob, *marker;
583
584     /* Make a writable copy of the glob */
585     glob = alloca(strlen(orig_glob)+1);
586     strcpy(glob, orig_glob);
587     /* Check the nick, if it's present */
588     if (flags & MATCH_USENICK) {
589         if (!(marker = strchr(glob, '!'))) {
590             log_module(MAIN_LOG, LOG_ERROR, "user_matches_glob(\"%s\", \"%s\", %d) called, and glob doesn't include a '!'", user->nick, orig_glob, flags);
591             return 0;
592         }
593         *marker = 0;
594         if (!match_ircglob(user->nick, glob)) return 0;
595         glob = marker + 1;
596     }
597     /* Check the ident */
598     if (!(marker = strchr(glob, '@'))) {
599         log_module(MAIN_LOG, LOG_ERROR, "user_matches_glob(\"%s\", \"%s\", %d) called, and glob doesn't include an '@'", user->nick, orig_glob, flags);
600         return 0;
601     }
602     *marker = 0;
603     if (!match_ircglob(user->ident, glob))
604         return 0;
605     glob = marker + 1;
606     /* Check for a fakehost match. */
607     if (IsFakeHost(user) && match_ircglob(user->fakehost, glob))
608         return 1;
609     /* Check for an account match. */
610     if (hidden_host_suffix && user->handle_info) {
611         char hidden_host[HOSTLEN+1];
612         snprintf(hidden_host, sizeof(hidden_host), "%s.%s", user->handle_info->handle, hidden_host_suffix);
613         if (match_ircglob(hidden_host, glob))
614             return 1;
615     }
616     /* If only matching the visible hostnames, bail early. */
617     if ((flags & MATCH_VISIBLE) && (IsFakeHost(user) || IsHiddenHost(user)))
618         return 0;
619     /* If it might be an IP glob, test that. */
620     if (!glob[strspn(glob, "0123456789./*?")]
621         && match_ircglob(irc_ntoa(&user->ip), glob))
622         return 1;
623     /* None of the above; could only be a hostname match. */
624     return match_ircglob(user->hostname, glob);
625 }
626
627 int
628 is_ircmask(const char *text)
629 {
630     while (*text && (isalnum((char)*text) || strchr("-_[]|\\`^{}?*", *text)))
631         text++;
632     if (*text++ != '!')
633         return 0;
634     while (*text && *text != '@' && !isspace((char)*text))
635         text++;
636     if (*text++ != '@')
637         return 0;
638     while (*text && !isspace((char)*text))
639         text++;
640     return !*text;
641 }
642
643 int
644 is_gline(const char *text)
645 {
646     if (*text == '@')
647         return 0;
648     text += strcspn(text, "@!% \t\r\n");
649     if (*text++ != '@')
650         return 0;
651     if (!*text)
652         return 0;
653     while (*text && (isalnum((char)*text) || strchr(".-?*:", *text)))
654         text++;
655     return !*text;
656 }
657
658 int
659 split_ircmask(char *text, char **nick, char **ident, char **host)
660 {
661     char *start;
662
663     start = text;
664     while (isalnum((char)*text) || strchr("=[]\\`^{}?*", *text))
665         text++;
666     if (*text != '!' || ((text - start) > NICKLEN))
667         return 0;
668     *text = 0;
669     if (nick)
670         *nick = start;
671
672     start = ++text;
673     while (*text && *text != '@' && !isspace((char)*text))
674         text++;
675     if (*text != '@' || ((text - start) > USERLEN))
676         return 0;
677     *text = 0;
678     if (ident)
679         *ident = start;
680
681     start = ++text;
682     while (*text && (isalnum((char)*text) || strchr(".-?*:", *text)))
683         text++;
684     if (host)
685         *host = start;
686     return !*text && ((text - start) <= HOSTLEN) && nick && ident && host;
687 }
688
689 char *
690 sanitize_ircmask(char *input)
691 {
692     unsigned int length, flag;
693     char *mask, *start, *output;
694
695     /* Sanitize everything in place; input *must* be a valid
696        hostmask. */
697     output = input;
698     flag = 0;
699
700     /* The nick is truncated at the end. */
701     length = 0;
702     mask = input;
703     while(*input++ != '!')
704     {
705         length++;
706     }
707     if(length > NICKLEN)
708     {
709         mask += NICKLEN;
710         *mask++ = '!';
711
712         /* This flag is used to indicate following parts should
713            be shifted. */
714         flag = 1;
715     }
716     else
717     {
718         mask = input;
719     }
720
721     /* The ident and host must be truncated at the beginning and
722        replaced with a '*' to be compatible with ircu. */
723     length = 0;
724     start = input;
725     while(*input++ != '@')
726     {
727         length++;
728     }
729     if(length > USERLEN || flag)
730     {
731         if(length > USERLEN)
732         {
733             start = input - USERLEN;
734             *mask++ = '*';
735         }
736         while(*start != '@')
737         {
738             *mask++ = *start++;
739         }
740         *mask++ = '@';
741
742         flag = 1;
743     }
744     else
745     {
746         mask = input;
747     }
748
749     length = 0;
750     start = input;
751     while(*input++)
752     {
753         length++;
754     }
755     if(length > HOSTLEN || flag)
756     {
757         if(length > HOSTLEN)
758         {
759             start = input - HOSTLEN;
760             *mask++ = '*';
761         }
762         while(*start)
763         {
764             *mask++ = *start++;
765         }
766         *mask = '\0';
767     }
768
769     return output;
770 }
771
772 static long
773 TypeLength(char type)
774 {
775     switch (type) {
776     case 'y': return 365*24*60*60;
777     case 'M': return 31*24*60*60;
778     case 'w': return 7*24*60*60;
779     case 'd': return 24*60*60;
780     case 'h': return 60*60;
781     case 'm': return 60;
782     case 's': return 1;
783     default: return 0;
784     }
785 }
786
787 unsigned long
788 ParseInterval(const char *interval)
789 {
790     unsigned long seconds = 0;
791     int partial = 0;
792     char c;
793
794     /* process the string, resetting the count if we find a unit character */
795     while ((c = *interval++)) {
796         if (isdigit((int)c)) {
797             partial = partial*10 + c - '0';
798         } else {
799             seconds += TypeLength(c) * partial;
800             partial = 0;
801         }
802     }
803     /* assume the last chunk is seconds (the normal case) */
804     return seconds + partial;
805 }
806
807 static long
808 GetSizeMultiplier(char type)
809 {
810     switch (type) {
811     case 'G': case 'g': return 1024*1024*1024;
812     case 'M': case 'm': return 1024*1024;
813     case 'K': case 'k': return 1024;
814     case 'B': case 'b': return 1;
815     default: return 0;
816     }
817 }
818
819 unsigned long
820 ParseVolume(const char *volume)
821 {
822     unsigned long accum = 0, partial = 0;
823     char c;
824     while ((c = *volume++)) {
825         if (isdigit((int)c)) {
826             partial = partial*10 + c - '0';
827         } else {
828             accum += GetSizeMultiplier(c) * partial;
829             partial = 0;
830         }
831     }
832     return accum + partial;
833 }
834
835 char *
836 unsplit_string(char *set[], unsigned int max, char *dest)
837 {
838     static char unsplit_buffer[MAXLEN*2];
839     unsigned int ii, jj, pos;
840
841     if (!dest)
842         dest = unsplit_buffer;
843     for (ii=pos=0; ii<max; ii++) {
844         for (jj=0; set[ii][jj]; jj++)
845             dest[pos++] = set[ii][jj];
846         dest[pos++] = ' ';
847     }
848     dest[--pos] = 0;
849     return dest;
850 }
851
852 char *
853 intervalString(char *output, time_t interval, struct handle_info *hi)
854 {
855     static const struct {
856         const char *msg_single;
857         const char *msg_plural;
858         long length;
859     } unit[] = {
860         { "MSG_YEAR",   "MSG_YEARS", 365 * 24 * 60 * 60 },
861         { "MSG_WEEK",   "MSG_WEEKS",   7 * 24 * 60 * 60 },
862         { "MSG_DAY",    "MSG_DAYS",        24 * 60 * 60 },
863         { "MSG_HOUR",   "MSG_HOURS",            60 * 60 },
864         { "MSG_MINUTE", "MSG_MINUTES",               60 },
865         { "MSG_SECOND", "MSG_SECONDS",                1 }
866     };
867     struct language *lang;
868     const char *msg;
869     unsigned int type, words, pos, count;
870
871     lang = hi ? hi->language : lang_C;
872     if(!interval)
873     {
874         msg = language_find_message(lang, "MSG_0_SECONDS");
875         return strcpy(output, msg);
876     }
877
878     for (type = 0, words = pos = 0;
879          interval && (words < 2) && (type < ArrayLength(unit));
880          type++) {
881         if (interval < unit[type].length)
882             continue;
883         count = interval / unit[type].length;
884         interval = interval % unit[type].length;
885
886         if (words++ == 1) {
887             msg = language_find_message(lang, "MSG_AND");
888             pos += sprintf(output + pos, " %s ", msg);
889         }
890         if (count == 1)
891             msg = language_find_message(lang, unit[type].msg_single);
892         else
893             msg = language_find_message(lang, unit[type].msg_plural);
894         pos += sprintf(output + pos, "%d %s", count, msg);
895     }
896
897     output[pos] = 0;
898     return output;
899 }
900
901 int
902 getipbyname(const char *name, unsigned long *ip)
903 {
904     struct hostent *he = gethostbyname(name);
905     if (!he)
906         return 0;
907     if (he->h_addrtype != AF_INET)
908         return 0;
909     memcpy(ip, he->h_addr_list[0], sizeof(*ip));
910     return 1;
911 }
912
913 DEFINE_LIST(string_buffer, char)
914
915 void
916 string_buffer_append_substring(struct string_buffer *buf, const char *tail, unsigned int len)
917 {
918     while (buf->used + len >= buf->size) {
919         if (!buf->size)
920             buf->size = 16;
921         else
922             buf->size <<= 1;
923         buf->list = realloc(buf->list, buf->size*sizeof(buf->list[0]));
924     }
925     memcpy(buf->list + buf->used, tail, len+1);
926     buf->used += len;
927 }
928
929 void
930 string_buffer_append_string(struct string_buffer *buf, const char *tail)
931 {
932     string_buffer_append_substring(buf, tail, strlen(tail));
933 }
934
935 void
936 string_buffer_append_vprintf(struct string_buffer *buf, const char *fmt, va_list args)
937 {
938     va_list working;
939     unsigned int len;
940     int ret;
941
942     VA_COPY(working, args);
943     len = strlen(fmt);
944     if (!buf->list || ((buf->used + buf->size) < len)) {
945         buf->size = buf->used + len;
946         buf->list = realloc(buf->list, buf->size);
947     }
948     ret = vsnprintf(buf->list + buf->used, buf->size - buf->used, fmt, working);
949     if (ret <= 0) {
950         /* pre-C99 behavior; double buffer size until it is big enough */
951         va_end(working);
952         VA_COPY(working, args);
953         while ((ret = vsnprintf(buf->list + buf->used, buf->size - buf->used, fmt, working)) <= 0) {
954             buf->size += len;
955             buf->list = realloc(buf->list, buf->size);
956             va_end(working);
957             VA_COPY(working, args);
958         }
959         buf->used += ret;
960     } else if (buf->used + ret < buf->size) {
961         /* no need to increase allocation size */
962         buf->used += ret;
963     } else {
964         /* now we know exactly how much space we need */
965         if (buf->size <= buf->used + ret) {
966             buf->size = buf->used + ret + 1;
967             buf->list = realloc(buf->list, buf->size);
968         }
969         va_end(working);
970         VA_COPY(working, args);
971         buf->used += vsnprintf(buf->list + buf->used, buf->size, fmt, working);
972     }
973     va_end(working);
974     va_end(args);
975 }
976
977 void string_buffer_append_printf(struct string_buffer *buf, const char *fmt, ...)
978 {
979     va_list args;
980     va_start(args, fmt);
981     string_buffer_append_vprintf(buf, fmt, args);
982 }
983
984 void
985 string_buffer_replace(struct string_buffer *buf, unsigned int from, unsigned int len, const char *repl)
986 {
987     unsigned int repl_len = strlen(repl);
988     if (from > buf->used)
989         return;
990     if (len + from > buf->used)
991         len = buf->used - from;
992     buf->used = buf->used + repl_len - len;
993     if (buf->size <= buf->used) {
994         while (buf->used >= buf->size)
995             buf->size <<= 1;
996         buf->list = realloc(buf->list, buf->size*sizeof(buf->list[0]));
997     }
998     memmove(buf->list+from+repl_len, buf->list+from+len, strlen(buf->list+from+len));
999     strcpy(buf->list+from, repl);
1000 }
1001
1002 struct string_list str_tab;
1003
1004 const char *
1005 strtab(unsigned int ii) {
1006     if (ii > 65536)
1007         return NULL;
1008     if (ii > str_tab.size) {
1009         unsigned int old_size = str_tab.size;
1010         while (ii >= str_tab.size)
1011             str_tab.size <<= 1;
1012         str_tab.list = realloc(str_tab.list, str_tab.size*sizeof(str_tab.list[0]));
1013         memset(str_tab.list+old_size, 0, (str_tab.size-old_size)*sizeof(str_tab.list[0]));
1014     }
1015     if (!str_tab.list[ii]) {
1016         str_tab.list[ii] = malloc(12);
1017         sprintf(str_tab.list[ii], "%u", ii);
1018     }
1019     return str_tab.list[ii];
1020 }
1021
1022 void
1023 tools_init(void)
1024 {
1025     unsigned int upr, lwr;
1026     for (lwr=0; lwr<256; ++lwr)
1027         tolower(lwr) = lwr;
1028     for (upr='A', lwr='a'; lwr <= 'z'; ++upr, ++lwr)
1029         tolower(upr) = lwr;
1030 #ifdef WITH_PROTOCOL_P10
1031     for (upr='[', lwr='{'; lwr <= '~'; ++upr, ++lwr)
1032         tolower(upr) = lwr;
1033     for (upr=0xc0, lwr=0xe0; lwr <= 0xf6; ++upr, ++lwr)
1034         tolower(upr) = lwr;
1035     for (upr=0xd8, lwr=0xf8; lwr <= 0xfe; ++upr, ++lwr)
1036         tolower(upr) = lwr;
1037 #endif
1038     str_tab.size = 1001;
1039     str_tab.list = calloc(str_tab.size, sizeof(str_tab.list[0]));
1040 }
1041
1042 void
1043 tools_cleanup(void)
1044 {
1045     unsigned int ii;
1046     for (ii=0; ii<str_tab.size; ++ii)
1047         free(str_tab.list[ii]);
1048     free(str_tab.list);
1049 }