a20aa0567b0e3ef1ce9733d2dd70604332345745
[doldaconnect.git] / daemon / fnet-dc.c
1 /*
2  *  Dolda Connect - Modular multiuser Direct Connect-style client
3  *  Copyright (C) 2004 Fredrik Tolf (fredrik@dolda2000.com)
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 */
19 #include <stdlib.h>
20 #include <stdio.h>
21 #include <wchar.h>
22 #include <malloc.h>
23 #include <unistd.h>
24 #include <fcntl.h>
25 #include <string.h>
26 #include <netinet/in.h>
27 #include <arpa/inet.h>
28 #include <alloca.h>
29 #include <wctype.h>
30 #include <time.h>
31 #include <errno.h>
32 #include <bzlib.h>
33 #include <zlib.h>
34 #include <sys/stat.h>
35
36 #ifdef HAVE_CONFIG_H
37 #include <config.h>
38 #endif
39 #include "filenet.h"
40 #include "log.h"
41 #include "module.h"
42 #include "utils.h"
43 #include "client.h"
44 #include "transfer.h"
45 #include "sysevents.h"
46 #include "net.h"
47 #include "tiger.h"
48
49 /*
50  * The Direct Connect protocol is extremely ugly. Thus, this code must
51  * also be a bit ugly in certain places. Please forgive me. =/
52  *
53  * This also means that there might be some pieces of the code that
54  * look completely illogical to you, and you think that you may make
55  * them much neater/better. However, in many cases it might just be
56  * that it's required to cope with some oddity of the protocol, such
57  * as the amazingly ugly delayed key response in the peer protocol.
58  */
59
60 /* I assume this is the correct character set to use for DC,
61  * considering it was developed without i18n support under Windows */
62 #define DCCHARSET "windows-1252"
63
64 #ifdef DCPP_MASQUERADE
65 /*
66  * I honestly don't want to pretend being a client that I'm not, but
67  * there are so many hubs that simply do not accept any clients
68  * outside their whitelists, for no obvious reasons, so I feel that I
69  * am left with little choice. Anyhow, as long as I actually support
70  * all the features that my faked DC++ version does, there should be
71  * very little harm done.
72  */
73 #define DCIDTAG "++"
74 #define DCIDTAGV "0.674"
75 #define DCIDFULL "DC++ 0.674"
76 #else
77 #define DCIDTAG "Dolda"
78 #define DCIDTAGV VERSION
79 #define DCIDFULL "DoldaConnect " VERSION
80 #endif
81
82 #define PEER_CMD 0
83 #define PEER_STOP 1
84 #define PEER_TRNS 2
85 #define PEER_SYNC 3
86 #define PEER_TTHL 4
87
88 #define CPRS_NONE 0
89 #define CPRS_ZLIB 1
90
91 struct command
92 {
93     char *name;
94     void (*handler)(struct socket *sk, void *data, char *cmd, char *args);
95     int stop;
96 };
97
98 struct qcommand
99 {
100     struct qcommand *next;
101     char *string;
102 };
103
104 struct dchub
105 {
106     char *inbuf;
107     size_t inbufdata, inbufsize;
108     struct qcommand *queue;
109     int extended, dcppemu;
110     char *charset;
111     char *nativename;
112     char *nativenick;
113 };
114
115 struct dcexppeer
116 {
117     struct dcexppeer *next, *prev;
118     char *nick;
119     struct fnetnode *fn;
120     struct timer *expire;
121 };
122
123 struct dcpeer
124 {
125     struct dcpeer *next, *prev;
126     struct socket *sk;
127     struct fnetnode *fn;
128     char *inbuf;
129     size_t inbufdata, inbufsize;
130     size_t curread, totalsize;
131     int freeing;
132     struct timer *timeout;
133     struct qcommand *queue;
134     struct transfer *transfer;
135     int state;
136     int ptclose;      /* Close after transfer is complete */
137     int accepted;     /* If false, we connected, otherwise, we accepted */
138     int extended, dcppemu;
139     int direction;    /* Using the constants from transfer.h */
140     int compress;
141     int hascurpos, fetchingtthl, notthl;
142     struct tigertreehash tth;
143     char *charset;
144     void *cprsdata;
145     char *key;
146     char *nativename;
147     char **supports;
148     wchar_t *wcsname;
149 };
150
151 static struct fnet dcnet;
152 static struct transferiface dctransfer;
153 static struct socket *udpsock = NULL;
154 static struct socket *tcpsock = NULL;
155 static struct dcpeer *peers = NULL;
156 int numdcpeers = 0;
157 static struct dcexppeer *expected = NULL;
158 static char *hmlistname = NULL;
159 static char *xmllistname = NULL;
160 static char *xmlbz2listname = NULL;
161
162 static void peerconnect(struct socket *sk, int err, struct fnetnode *fn);
163 static void freedcpeer(struct dcpeer *peer);
164 static void transread(struct socket *sk, struct dcpeer *peer);
165 static void transerr(struct socket *sk, int err, struct dcpeer *peer);
166 static void transwrite(struct socket *sk, struct dcpeer *peer);
167 static void updatehmlist(void);
168 static void updatexmllist(void);
169 static void updatexmlbz2list(void);
170 static void requestfile(struct dcpeer *peer);
171
172 static int reservedchar(unsigned char c)
173 {
174     return((c == 0) || (c == 5) || (c == 124) || (c == 96) || (c == 126) || (c == 36));
175 }
176
177 /* Oh, how I despise having to do this... */
178 static char *dcmakekey(char *lock)
179 {
180     int i, len, offset;
181     char *buf, *key;
182     char save;
183     
184     buf = smalloc(strlen(lock));
185     save = 5;
186     len = 0;
187     for(i = 0; lock[i]; i++)
188     {
189         buf[i] = lock[i] ^ save;
190         buf[i] = ((buf[i] & 0x0F) << 4) | ((buf[i] & 0xF0) >> 4);
191         save = lock[i];
192         if((i != 0) && reservedchar(buf[i]))
193             len += 10;
194         else
195             len++;
196     }
197     buf[0] ^= buf[i - 1];
198     if(reservedchar(buf[0]))
199         len += 10;
200     else
201         len++;
202     key = smalloc(len + 1);
203     offset = 0;
204     for(i = 0; lock[i] != 0; i++)
205     {
206         if(reservedchar(buf[i]))
207             offset += sprintf(key + offset, "/%%DCN%03i%%/", buf[i]);
208         else
209             key[offset++] = buf[i];
210     }
211     key[offset] = 0;
212     free(buf);
213     return(key);
214 }
215
216 static char *pathnmdc2adc(char *path)
217 {
218     char *ret;
219     size_t retsize, retdata;
220     
221     if(!strcmp(path, "files.xml") || !strcmp(path, "files.xml.bz2") || !strcmp(path, "MyList.DcLst"))
222         return(sstrdup(path));
223     ret = NULL;
224     retsize = retdata = 0;
225     addtobuf(ret, '/');
226     for(; *path; path++)
227     {
228         if(*path == '\\')
229             addtobuf(ret, '/');
230         else
231             addtobuf(ret, *path);
232     }
233     addtobuf(ret, 0);
234     return(ret);
235 }
236
237 static int isdchash(struct hash *hash)
238 {
239     if(wcscmp(hash->algo, L"TTH"))
240         return(0);
241     if(hash->len != 24)
242         return(0);
243     return(1);
244 }
245
246 static int supports(struct dcpeer *peer, char *cap)
247 {
248     char **p;
249     
250     if(peer->supports == NULL)
251         return(0);
252     for(p = peer->supports; *p != NULL; p++)
253     {
254         if(!strcasecmp(*p, cap))
255             return(1);
256     }
257     return(0);
258 }
259
260 static void endcompress(struct dcpeer *peer)
261 {
262     if(peer->compress == CPRS_ZLIB)
263     {
264         deflateEnd(peer->cprsdata);
265         free(peer->cprsdata);
266     }
267     peer->compress = CPRS_NONE;
268 }
269
270 static void initcompress(struct dcpeer *peer, int algo)
271 {
272     int ret;
273     
274     endcompress(peer);
275     peer->compress = algo;
276     if(algo == CPRS_ZLIB)
277     {
278         peer->cprsdata = smalloc(sizeof(z_stream));
279         memset(peer->cprsdata, 0, sizeof(z_stream));
280         if((ret = deflateInit(peer->cprsdata, 3)) != Z_OK)
281         {
282             flog(LOG_CRIT, "Aiya! zlib refuses to init (%i)!", ret);
283             abort();
284         }
285     }
286 }
287
288 static void unquote(wchar_t *in)
289 {
290     wchar_t *p, *p2, nc;
291     
292     for(p = in; *p != L'\0'; p++)
293     {
294         if(*p == L'&')
295         {
296             for(p2 = p + 1; (*p2 != L'\0') && (*p2 != L';') && (*p2 != L'&'); p2++);
297             if(*p2 == L'&')
298                 continue;
299             if(*p2 == L'\0')
300                 return;
301             *p2 = L'\0';
302             nc = L'\0';
303             if(!wcscmp(p + 1, L"amp"))
304             {
305                 nc = L'&';
306             } else if(p[1] == L'#') {
307                 nc = ucptowc(wcstol(p + 2, NULL, 10));
308             }
309             if(nc == L'\0')
310             {
311                 *p2 = L';';
312                 p = p2;
313                 continue;
314             }
315             *p = nc;
316             memmove(p + 1, p2 + 1, (wcslen(p2 + 1) + 1) * sizeof(wchar_t));
317         }
318     }
319 }
320
321 static void freeexppeer(struct dcexppeer *ep)
322 {
323     if(ep->next != NULL)
324         ep->next->prev = ep->prev;
325     if(ep->prev != NULL)
326         ep->prev->next = ep->next;
327     if(ep == expected)
328         expected = ep->next;
329     free(ep->nick);
330     putfnetnode(ep->fn);
331     if(ep->expire != NULL)
332         canceltimer(ep->expire);
333     free(ep);
334 }
335
336 static void exppeerexpire(int cancelled, struct dcexppeer *ep)
337 {
338     ep->expire = NULL;
339     if(!cancelled)
340         freeexppeer(ep);
341 }
342
343 static struct dcexppeer *expectpeer(char *nick, struct fnetnode *fn)
344 {
345     struct dcexppeer *ep;
346     
347     ep = smalloc(sizeof(*ep));
348     ep->nick = sstrdup(nick);
349     getfnetnode(ep->fn = fn);
350     ep->expire = timercallback(ntime() + 300, (void (*)(int, void *))exppeerexpire, ep);
351     ep->next = expected;
352     ep->prev = NULL;
353     if(expected != NULL)
354         expected->prev = ep;
355     expected = ep;
356     return(ep);
357 }
358
359 static struct qcommand *newqcmd(struct qcommand **queue, char *string)
360 {
361     struct qcommand *new;
362     
363     while(*queue != NULL)
364         queue = &(*queue)->next;
365     new = smalloc(sizeof(*new));
366     new->string = sstrdup(string);
367     new->next = *queue;
368     *queue = new;
369     return(new);
370 }
371
372 static struct qcommand *ulqcmd(struct qcommand **queue)
373 {
374     struct qcommand *qcmd;
375     
376     if((qcmd = *queue) == NULL)
377         return(NULL);
378     *queue = qcmd->next;
379     return(qcmd);
380 }
381
382 static void freeqcmd(struct qcommand *qcmd)
383 {
384     free(qcmd->string);
385     free(qcmd);
386 }
387
388 static void hubrecvchat(struct socket *sk, struct fnetnode *fn, char *from, char *string)
389 {
390     wchar_t *chat, *wfrom, *wpeer;
391     char *p, *end;
392     struct fnetpeer *peer;
393     struct dchub *hub;
394     
395     hub = fn->data;
396     end = string + strlen(string);
397     while((p = strchr(string, 13)) != NULL)
398         memmove(p, p + 1, (end-- - p));
399     if(from != NULL)
400     {
401         if((strlen(string) > strlen(from) + 2) && (*string == '<') && !memcmp(string + 1, from, strlen(from)) && (*(string + strlen(from) + 1) == '>'))
402             string += strlen(from) + 2;
403         if((wfrom = icmbstowcs(from, hub->charset)) == NULL)
404             return;
405         wpeer = swcsdup(wfrom);
406     } else {
407         wfrom = NULL;
408         wpeer = NULL;
409         if(*string == '<')
410         {
411             for(p = string + 1; *p; p++)
412             {
413                 if((*p == ' ') || (*p == '>'))
414                     break;
415             }
416             if(*p == '>')
417             {
418                 *(p++) = 0;
419                 if(*p == ' ')
420                     p++;
421                 if((wpeer = icmbstowcs(string + 1, hub->charset)) == NULL)
422                     return;
423                 string = p;
424             }
425         }
426         if(wpeer == NULL)
427             wpeer = swcsdup(L"");
428     }
429     if((chat = icmbstowcs(string, hub->charset)) == NULL)
430     {
431         if(wfrom != NULL)
432             free(wfrom);
433         free(wpeer);
434         return;
435     }
436     unquote(chat);
437     if(wfrom != NULL)
438     {
439         if((peer = fnetfindpeer(fn, wfrom)) == NULL) /* Assume public chat */
440             fnethandlechat(fn, 1, wfrom, wpeer, chat);
441         else
442             fnethandlechat(fn, 0, wfrom, wpeer, chat);
443     } else {
444         fnethandlechat(fn, 1, L"", wpeer, chat);
445     }
446     if(wfrom != NULL)
447         free(wfrom);
448     free(wpeer);
449     free(chat);
450 }
451
452 static void peertimeout(int cancelled, struct dcpeer *peer)
453 {
454     peer->timeout = NULL;
455     if(cancelled)
456         return;
457     freedcpeer(peer);
458 }
459
460 static void sendadc(struct socket *sk, char *arg)
461 {
462     char *buf;
463     size_t bufsize, bufdata;
464     
465     buf = NULL;
466     bufsize = bufdata = 0;
467     addtobuf(buf, ' ');
468     for(; *arg; arg++)
469     {
470         if(*arg == ' ')
471         {
472             bufcat(buf, "\\s", 2);
473         } else if(*arg == '\n') {
474             bufcat(buf, "\\n", 2);
475         } else if(*arg == '\\') {
476             bufcat(buf, "\\\\", 2);
477         } else {
478             addtobuf(buf, *arg);
479         }
480     }
481     sockqueue(sk, buf, bufdata);
482     free(buf);
483 }
484
485 static void sendadcf(struct socket *sk, char *arg, ...)
486 {
487     char *buf;
488     va_list args;
489     
490     va_start(args, arg);
491     buf = vsprintf2(arg, args);
492     va_end(args);
493     if(buf == NULL)
494         return;
495     sendadc(sk, buf);
496     free(buf);
497 }
498
499 static char **parseadc(char *args)
500 {
501     char **retbuf;
502     size_t retbufsize, retbufdata;
503     char *buf;
504     size_t bufsize, bufdata;
505     int state;
506     
507     retbuf = NULL;
508     buf = NULL;
509     retbufsize = retbufdata = bufsize = bufdata = 0;
510     state = 0;
511     while(state != 3)
512     {
513         switch(state)
514         {
515         case 0:
516             if(*args == 0)
517                 state = 3;
518             else if(*args != ' ')
519                 state = 1;
520             break;
521         case 1:
522             if((*args == ' ') || (*args == 0))
523             {
524                 addtobuf(buf, 0);
525                 addtobuf(retbuf, buf);
526                 buf = NULL;
527                 bufsize = bufdata = 0;
528                 if(*args == 0)
529                     state = 3;
530             } else if(*args == '\\') {
531                 state = 2;
532             } else {
533                 addtobuf(buf, *args);
534             }
535             args++;
536             break;
537         case 2:
538             if(*args == 0)
539             {
540                 if(buf != NULL)
541                     free(buf);
542                 addtobuf(retbuf, NULL);
543                 freeparr(retbuf);
544                 return(NULL);
545             } else if((*args == 's') || (*args == ' ')) {
546                 addtobuf(buf, ' ');
547             } else if(*args == 'n') {
548                 addtobuf(buf, '\n');
549             } else if(*args == '\\') {
550                 addtobuf(buf, '\\');
551             }
552             args++;
553             state = 1;
554             break;
555         }
556     }
557     if(buf != NULL)
558         free(buf);
559     addtobuf(retbuf, NULL);
560     return(retbuf);
561 }
562
563 /* Macros useful in command handlers */
564 #define skipspace(s) ({if(((s) = strchr((s), ' ')) == NULL) return; else (s)++;})
565 #define qstr(sk, str) sockqueue(sk, str, strlen(str))
566 #define qstrf(sk, strandargs...) \
567 do { \
568     char *__buf__; \
569     if((__buf__ = sprintf2(strandargs)) != NULL) { \
570         sockqueue(sk, __buf__, strlen(__buf__)); \
571         free(__buf__); \
572     } \
573 } while(0)
574
575 static char *tr(char *str, char *trans)
576 {
577     char *p;
578     
579     for(; *trans; trans += 2)
580     {
581         for(p = strchr(str, trans[0]); p != NULL; p = strchr(p, trans[0]))
582             *p = trans[1];
583     }
584     return(str);
585 }
586
587 static char *getadcid(struct dcpeer *peer)
588 {
589     char *buf;
590     char *ret;
591     int isfilelist;
592     
593     if(!wcscmp(peer->transfer->path, L"files.xml") || !wcscmp(peer->transfer->path, L"files.xml.bz2") || !wcscmp(peer->transfer->path, L"MyList.DcLst"))
594         isfilelist = 1;
595     if(!isfilelist && (peer->transfer->hash != NULL) && isdchash(peer->transfer->hash) && supports(peer, "tthf"))
596     {
597         buf = base32encode(peer->transfer->hash->buf, 24);
598         ret = sprintf2("TTH/%.39s", buf);
599         free(buf);
600     } else {
601         if((buf = icwcstombs(peer->transfer->path, "UTF-8")) == NULL)
602             return(NULL);
603         ret = pathnmdc2adc(buf);
604         free(buf);
605     }
606     return(ret);
607 }
608
609
610 static int trresumecb(struct transfer *transfer, wchar_t *cmd, wchar_t *arg, struct dcpeer *peer)
611 {
612     if(!wcscmp(cmd, L"resume"))
613     {
614         if(arg == NULL)
615         {
616             flog(LOG_WARNING, "filter returned no position for \"resume\" on transfer %i", transfer->id);
617             freedcpeer(peer);
618         } else {
619             transfer->curpos = wcstol(arg, NULL, 10);
620             peer->hascurpos = 1;
621             requestfile(peer);
622         }
623         return(1);
624     }
625     return(0);
626 }
627
628 static void sendmynick(struct dcpeer *peer)
629 {
630     struct fnetnode *fn;
631     
632     fn = peer->fn;
633     if(fn == NULL)
634         qstrf(peer->sk, "$MyNick %s|", icswcstombs(confgetstr("cli", "defnick"), peer->charset, "DoldaConnectUser-IN"));
635     else
636         qstrf(peer->sk, "$MyNick %s|", icswcstombs(fn->mynick, peer->charset, "DoldaConnectUser-IN"));
637 }
638
639 static void sendpeerlock(struct dcpeer *peer)
640 {
641     if(peer->dcppemu)
642         qstrf(peer->sk, "$Lock EXTENDEDPROTOCOLABCABCABCABCABCABC Pk=DCPLUSPLUS0.674ABCABC|");
643     else
644         qstrf(peer->sk, "$Lock EXTENDEDPROTOCOLABCABCABCABCABCABC Pk=DOLDA%sABCABCABC|", VERSION);
645 }
646
647 static void sendsupports(struct dcpeer *peer)
648 {
649     if(peer->dcppemu)
650         qstr(peer->sk, "$Supports MiniSlots XmlBZList ADCGet TTHL TTHF GetZBlock ZLIG |");
651     else
652         qstr(peer->sk, "$Supports MiniSlots XmlBZList ADCGet TTHL TTHF GetZBlock ZLIG|");
653 }
654
655 static void requestfile(struct dcpeer *peer)
656 {
657     char *buf;
658     
659     if(peer->transfer->size == -1)
660     {
661         /* Use DCCHARSET for $Get paths until further researched... */
662         if((buf = icswcstombs(peer->transfer->path, DCCHARSET, NULL)) == NULL)
663         {
664             transferseterror(peer->transfer, TRNSE_NOTFOUND);
665             freedcpeer(peer);
666             return;
667         }
668         /* The transfer will be restarted later from
669          * cmd_filelength when it detects that the sizes
670          * don't match. */
671         qstrf(peer->sk, "$Get %s$1|", buf);
672         return;
673     }
674     if((peer->transfer->hash == NULL) && !peer->notthl)
675     {
676         if(supports(peer, "adcget") && supports(peer, "tthl"))
677         {
678             qstr(peer->sk, "$ADCGET");
679             sendadc(peer->sk, "tthl");
680             if((buf = getadcid(peer)) == NULL)
681             {
682                 transferseterror(peer->transfer, TRNSE_NOTFOUND);
683                 freedcpeer(peer);
684                 return;
685             }
686             sendadc(peer->sk, buf);
687             free(buf);
688             sendadc(peer->sk, "0");
689             sendadc(peer->sk, "-1");
690             qstr(peer->sk, "|");
691             peer->fetchingtthl = 1;
692             return;
693         }
694     }
695     if(!peer->hascurpos)
696     {
697         if(forkfilter(peer->transfer))
698         {
699             flog(LOG_WARNING, "could not fork filter for transfer %i: %s", peer->transfer->id, strerror(errno));
700             freedcpeer(peer);
701             return;
702         }
703         CBREG(peer->transfer, trans_filterout, (int (*)(struct transfer *, wchar_t *, wchar_t *, void *))trresumecb, NULL, peer);
704         return;
705     }
706     if(supports(peer, "adcget"))
707     {
708         qstr(peer->sk, "$ADCGET");
709         sendadc(peer->sk, "file");
710         if((buf = getadcid(peer)) == NULL)
711         {
712             transferseterror(peer->transfer, TRNSE_NOTFOUND);
713             freedcpeer(peer);
714             return;
715         }
716         sendadc(peer->sk, buf);
717         free(buf);
718         sendadcf(peer->sk, "%i", peer->transfer->curpos);
719         sendadcf(peer->sk, "%i", peer->transfer->size - peer->transfer->curpos);
720         qstr(peer->sk, "|");
721     } else if(supports(peer, "xmlbzlist")) {
722         if((buf = icswcstombs(peer->transfer->path, "UTF-8", NULL)) == NULL)
723         {
724             transferseterror(peer->transfer, TRNSE_NOTFOUND);
725             freedcpeer(peer);
726             return;
727         }
728         qstrf(peer->sk, "$UGetBlock %i %i %s|", peer->transfer->curpos, peer->transfer->size - peer->transfer->curpos, buf);
729     } else {
730         /* Use DCCHARSET for $Get paths until further researched... */
731         if((buf = icswcstombs(peer->transfer->path, DCCHARSET, NULL)) == NULL)
732         {
733             transferseterror(peer->transfer, TRNSE_NOTFOUND);
734             freedcpeer(peer);
735             return;
736         }
737         qstrf(peer->sk, "$Get %s$%i|", buf, peer->transfer->curpos + 1);
738     }
739 }
740
741 static void sendmyinfo(struct socket *sk, struct fnetnode *fn)
742 {
743     struct dchub *hub;
744     char *buf;
745     struct fnetnode *cfn;
746     int hn1, hn2, hn3;
747     
748     hub = fn->data;
749     qstrf(sk, "$MyINFO $ALL %s ", hub->nativenick);
750     buf = tr(icswcstombs(confgetstr("dc", "desc"), hub->charset, "Charset_conv_failure"), "$_|_");
751     qstrf(sk, "%s", buf);
752     hn1 = hn2 = hn3 = 0;
753     for(cfn = fnetnodes; cfn != NULL; cfn = cfn->next)
754     {
755         if((cfn->state == FNN_EST) || (cfn->state == FNN_HS))
756         {
757             if(cfn->regstatus == FNNS_OP)
758                 hn3++;
759             else if(cfn->regstatus == FNNS_REG)
760                 hn2++;
761             else
762                 hn1++;
763         }
764     }
765     qstrf(sk, "<%s V:%s,M:%c,H:%i/%i/%i,S:%i>",
766           (hub->dcppemu)?"++":"Dolda",
767           (hub->dcppemu)?"0.674":VERSION,
768           (tcpsock == NULL)?'P':'A',
769           hn1, hn2, hn3,
770           confgetint("transfer", "slots")
771           );
772     qstrf(sk, "$ $");
773     buf = tr(icswcstombs(confgetstr("dc", "speedstring"), hub->charset, "Charset_conv_failure"), "$_|_");
774     qstrf(sk, "%s\x01$", buf);
775     buf = tr(icswcstombs(confgetstr("dc", "email"), hub->charset, "Charset_conv_failure"), "$_|_");
776     qstrf(sk, "%s$", buf);
777     qstrf(sk, "%llu$|", sharesize);
778 }
779
780 static void hubhandleaction(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
781 {
782     struct dchub *hub;
783     
784     hub = fn->data;
785     if(!strcmp(cmd, "$Lock"))
786     {
787         qstrf(sk, "$ValidateNick %s|", hub->nativenick);
788     } else if(!strcmp(cmd, "$Hello")) {
789         if(fn->state == FNN_HS)
790         {
791             qstrf(sk, "$Version 1,0091|");
792             qstrf(sk, "$GetNickList|");
793             sendmyinfo(sk, fn);
794             fnetsetstate(fn, FNN_EST);
795         } else {
796             qstrf(sk, "$GetINFO %s %s|", args, hub->nativenick);
797         }
798     }
799 }
800
801 static void cmd_lock(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
802 {
803     struct dchub *hub;
804     char *key;
805     char *p;
806     
807     hub = fn->data;
808     if(!strncmp(args, "EXTENDEDPROTOCOL", 16))
809         hub->extended = 1;
810     if((p = strchr(args, ' ')) != NULL)
811         *(p++) = 0;
812     if(hub->extended)
813     {
814         if(hub->dcppemu)
815             qstrf(sk, "$Supports UserCommand NoGetINFO NoHello UserIP2 TTHSearch GetZBlock |");
816         else
817             qstrf(sk, "$Supports UserCommand NoGetINFO NoHello UserIP2 TTHSearch GetZBlock|");
818     }
819     key = dcmakekey(args);
820     qstrf(sk, "$Key %s|", key);
821     free(key);
822     hubhandleaction(sk, fn, cmd, args);
823 }
824
825 static void cmd_hubname(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
826 {
827     wchar_t *buf;
828     struct dchub *hub;
829     
830     hub = fn->data;
831     if(hub->nativename == NULL)
832         free(hub->nativename);
833     hub->nativename = sstrdup(args);
834     buf = icmbstowcs(args, hub->charset);
835     fnetsetname(fn, (buf == NULL)?L"Hubname conv error":buf);
836     if(buf != NULL)
837         free(buf);
838     hubhandleaction(sk, fn, cmd, args);
839 }
840
841 static void cmd_hello(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
842 {
843     wchar_t *nick;
844     struct dchub *hub;
845     
846     hub = fn->data;
847     if((nick = icmbstowcs(args, hub->charset)) == NULL)
848         return;
849     if(strcmp(args, hub->nativenick) && (fnetfindpeer(fn, nick) == NULL))
850         fnetaddpeer(fn, nick, nick);
851     free(nick);
852     hubhandleaction(sk, fn, cmd, args);
853 }
854
855 static void cmd_quit(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
856 {
857     wchar_t *nick;
858     struct fnetpeer *peer;
859     struct dchub *hub;
860     
861     hub = fn->data;
862     if((nick = icmbstowcs(args, hub->charset)) == NULL)
863         return;
864     if((peer = fnetfindpeer(fn, nick)) != NULL)
865         fnetdelpeer(peer);
866     free(nick);
867     hubhandleaction(sk, fn, cmd, args);
868 }
869
870 static void cmd_nicklist(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
871 {
872     struct dchub *hub;
873     char *p;
874     wchar_t *buf;
875     struct fnetpeer *peer, *npeer;
876     
877     hub = fn->data;
878     for(peer = fn->peers; peer != NULL; peer = peer->next)
879         peer->flags.b.delete = 1;
880     while((p = strstr(args, "$$")) != NULL)
881     {
882         *p = 0;
883         if((buf = icmbstowcs(args, hub->charset)) != NULL)
884         {
885             if((peer = fnetfindpeer(fn, buf)) == NULL)
886                 peer = fnetaddpeer(fn, buf, buf);
887             else
888                 peer->flags.b.delete = 0;
889             free(buf);
890             qstrf(sk, "$GetINFO %s %s|", args, hub->nativenick);
891         }
892         args = p + 2;
893     }
894     for(peer = fn->peers; peer != NULL; peer = npeer)
895     {
896         npeer = peer->next;
897         if(peer->flags.b.delete)
898             fnetdelpeer(peer);
899     }
900     hubhandleaction(sk, fn, cmd, args);
901 }
902
903 static void cmd_oplist(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
904 {
905     struct dchub *hub;
906     char *p;
907     wchar_t *buf;
908     struct fnetpeer *peer;
909     
910     hub = fn->data;
911     for(peer = fn->peers; peer != NULL; peer = peer->next)
912         peer->flags.b.op = 0;
913     while((p = strstr(args, "$$")) != NULL)
914     {
915         *p = 0;
916         if((buf = icmbstowcs(args, hub->charset)) != NULL)
917         {
918             if((peer = fnetfindpeer(fn, buf)) != NULL)
919                 peer->flags.b.op = 1;
920             free(buf);
921         }
922         args = p + 2;
923     }
924     hubhandleaction(sk, fn, cmd, args);
925 }
926
927 static void cmd_myinfo(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
928 {
929     char *p, *p2;
930     wchar_t *buf, *wp, *wp2;
931     wchar_t abuf[10];
932     struct fnetpeer *peer;
933     struct dchub *hub;
934     
935     hub = fn->data;
936     p = args;
937     if(strncmp(p, "$ALL ", 5))
938         return;
939     p += 5;
940     if((p2 = strchr(p, ' ')) == NULL)
941         return;
942     *p2 = 0;
943     if((buf = icmbstowcs(p, hub->charset)) == NULL)
944         return;
945     if((peer = fnetfindpeer(fn, buf)) == NULL)
946         peer = fnetaddpeer(fn, buf, buf);
947     free(buf);
948     p = p2 + 1;
949     if((p2 = strstr(p, "$ $")) == NULL)
950         return;
951     *p2 = 0;
952     if((buf = icmbstowcs(p, hub->charset)) == NULL)
953         return;
954     if((wcslen(buf) > 0) && (buf[wcslen(buf) - 1] == L'>') && ((wp = wcsrchr(buf, L'<')) != NULL))
955     {
956         buf[wcslen(buf) - 1] = L'\0';
957         *(wp++) = L'\0';
958         if((wp2 = wcschr(wp, L' ')) != NULL)
959         {
960             *(wp2++) = L'\0';
961             fnetpeersetstr(peer, L"dc-client", wp);
962             wp = wp2;
963             do
964             {
965                 if((wp2 = wcschr(wp, L',')) != NULL)
966                     *(wp2++) = L'\0';
967                 if(wp[1] != L':')
968                     continue;
969                 swprintf(abuf, 10, L"dc-tag-%lc", wp[0]);
970                 fnetpeersetstr(peer, abuf, wp + 2);
971                 wp = wp2;
972             } while(wp2 != NULL);
973         }
974     }
975     fnetpeersetstr(peer, L"descr", buf);
976     free(buf);
977     p = p2 + 3;
978     if((p2 = strchr(p, '$')) == NULL)
979         return;
980     *(p2 - 1) = 0;
981     if((buf = icmbstowcs(p, hub->charset)) == NULL)
982         return;
983     fnetpeersetstr(peer, L"dc-speed", buf);
984     free(buf);
985     p = p2 + 1;
986     if((p2 = strchr(p, '$')) == NULL)
987         return;
988     *p2 = 0;
989     if((buf = icmbstowcs(p, hub->charset)) == NULL)
990         return;
991     fnetpeersetstr(peer, L"email", buf);
992     free(buf);
993     p = p2 + 1;
994     if(strlen(p) < 1)
995         return;
996     fnetpeersetlnum(peer, L"share", strtoll(p, NULL, 10));
997     hubhandleaction(sk, fn, cmd, args);
998 }
999
1000 /* I do not implement the fully in that I do not disconnect from the
1001  * old hub. I do this since I believe that if the hub owner really
1002  * wants to get rid of us, then it is his responsibility to disconnect
1003  * us. */
1004 static void cmd_forcemove(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1005 {
1006     struct dchub *hub;
1007     struct fnetnode *newfn;
1008     int freeargs;
1009     
1010     hub = fn->data;
1011     if(strchr(args, ':') == NULL)
1012     {
1013         args = strcpy(smalloc(strlen(args) + 5), args);
1014         strcat(args, ":411");
1015         freeargs = 1;
1016     } else {
1017         freeargs = 0;
1018     }
1019     if((newfn = fnetinitconnect(L"dc", args, NULL)) != NULL)
1020     {
1021         linkfnetnode(newfn);
1022         putfnetnode(newfn);
1023     }
1024     hubhandleaction(sk, fn, cmd, args);
1025     if(freeargs)
1026         free(args);
1027 }
1028
1029 static char *getdcpath(struct sharecache *node, size_t *retlen, char *charset)
1030 {
1031     char *buf, *buf2;
1032     size_t len, len2;
1033     
1034     if(node->parent == NULL)
1035         return(NULL);
1036     if(node->parent == shareroot)
1037     {
1038         if((buf = icwcstombs(node->name, charset)) == NULL)
1039             return(NULL);
1040         if(retlen != NULL)
1041             *retlen = strlen(buf);
1042         return(buf);
1043     } else {
1044         if((buf2 = icwcstombs(node->name, charset)) == NULL)
1045             return(NULL);
1046         if((buf = getdcpath(node->parent, &len, charset)) == NULL)
1047         {
1048             free(buf2);
1049             return(NULL);
1050         }
1051         len2 = strlen(buf2);
1052         buf = srealloc(buf, len + 1 + len2 + 1);
1053         buf[len++] = '\\';
1054         strcpy(buf + len, buf2);
1055         buf[len + len2] = 0;
1056         free(buf2);
1057         if(retlen != NULL)
1058             *retlen = len + len2;
1059         return(buf);
1060     }
1061 }
1062
1063 /*
1064  * This is the main share searching function for Direct Connect
1065  * peers. Feel free to optimize it if you feel the need for it. I
1066  * haven't ever seen it take any noticable CPU time anyway, so I
1067  * haven't felt that need.
1068  *
1069  * It may feel dubious to just return the directory when all terms are
1070  * satisfied rather than all the files in it, but it really benefits
1071  * everyone: Less cycles used for us, less UDP squelching for active
1072  * searchers, and less bandwidth waste for hubs when serving passive
1073  * searchers.
1074  */
1075 static void cmd_search(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1076 {
1077     int i, done;
1078     struct dchub *hub;
1079     char *p, *p2;
1080     char *prefix, *infix, *postfix, *buf, *buf2;
1081     struct socket *dsk;
1082     struct sockaddr_in addr;
1083     struct sharecache *node;
1084     int minsize, maxsize;
1085     int dotth, buflen;
1086     int termnum, satisfied, skipcheck;
1087     int level, tersat[32];
1088     wchar_t *terms[32];
1089     char hashtth[24];
1090     
1091     hub = fn->data;
1092     if((p = strchr(args, ' ')) == NULL)
1093         return;
1094     *(p++) = 0;
1095     
1096     memset(terms, 0, sizeof(terms));
1097     prefix = infix = postfix = NULL;
1098     dsk = NULL;
1099     dotth = 0;
1100     
1101     if(!strncmp(args, "Hub:", 4))
1102     {
1103         if(!strcmp(cmd, "$MultiSearch"))
1104             goto out;
1105         if(!strcmp(args + 4, hub->nativenick))
1106             goto out;
1107         prefix = sprintf2("$SR %s ", hub->nativenick);
1108         infix = sprintf2(" %i/%i\005", slotsleft(), confgetint("transfer", "slots"));
1109         postfix = sprintf2(" (%s)\005%s|", formataddress(fn->sk->remote, fn->sk->remotelen), args + 4);
1110         dsk = sk;
1111         getsock(dsk);
1112     } else {
1113         if((p2 = strchr(args, ':')) == NULL)
1114             goto out;
1115         *(p2++) = 0;
1116         addr.sin_family = AF_INET;
1117         if(!inet_aton(args, &addr.sin_addr))
1118             goto out;
1119         addr.sin_port = htons(atoi(p2));
1120         prefix = sprintf2("$SR %s ", hub->nativenick);
1121         infix = sprintf2(" %i/%i\005", slotsleft(), confgetint("transfer", "slots"));
1122         postfix = sprintf2(" (%s)|", formataddress(fn->sk->remote, fn->sk->remotelen));
1123         netdgramconn(dsk = netdupsock(udpsock), (struct sockaddr *)&addr, sizeof(addr));
1124     }
1125     
1126     minsize = maxsize = -1;
1127     if(*p == 0)
1128         goto out;
1129     if(*(p++) == 'T')
1130         minsize = 0;
1131     if(*(p++) != '?')
1132         goto out;
1133     if(*p == 0)
1134         goto out;
1135     if((*(p++) == 'T') && (minsize == 0))
1136     {
1137         maxsize = 0;
1138         minsize = -1;
1139     }
1140     if(*(p++) != '?')
1141         goto out;
1142     if((p2 = strchr(p, '?')) == NULL)
1143         goto out;
1144     *(p2++) = 0;
1145     if(minsize == 0)
1146         minsize = atoi(p);
1147     if(maxsize == 0)
1148         maxsize = atoi(p);
1149     p = p2 + 1;
1150     if(*(p++) != '?')
1151         goto out;
1152     termnum = 0;
1153     p2 = p;
1154     done = 0;
1155     while(!done)
1156     {
1157         if((*p2 == 0) || (*p2 == '$'))
1158         {
1159             if(*p2 == 0)
1160                 done = 1;
1161             else
1162                 *p2 = 0;
1163             if(*p)
1164             {
1165                 if(!dotth && !strncmp(p, "TTH:", 4))
1166                 {
1167                     dotth = 1;
1168                     if((buf = base32decode(p + 4, &buflen)) == NULL)
1169                         goto out;
1170                     if(buflen != 24)
1171                         goto out;
1172                     memcpy(hashtth, buf, 24);
1173                     free(buf);
1174                 } else {
1175                     if((terms[termnum] = icmbstowcs(p, hub->charset)) != NULL)
1176                         termnum++;
1177                 }
1178             }
1179             p = p2 + 1;
1180             if(termnum == 32)
1181                 break;
1182         }
1183         p2++;
1184     }
1185     
1186     node = shareroot->child;
1187     level = 0;
1188     for(i = 0; i < termnum; i++)
1189         tersat[i] = -1;
1190     satisfied = 0;
1191     while(1)
1192     {
1193         skipcheck = 0;
1194         if(node->f.b.type == FILE_REG)
1195         {
1196             if((minsize >= 0) && (node->size < minsize))
1197                 skipcheck = 1;
1198             if((maxsize >= 0) && (node->size > maxsize))
1199                 skipcheck = 1;
1200         }
1201         if(!skipcheck && dotth)
1202         {
1203             if((node->f.b.type != FILE_REG) || (node->f.b.hastth && memcmp(hashtth, node->hashtth, 24)))
1204                 skipcheck = 1;
1205         }
1206         if(!skipcheck)
1207         {
1208             for(i = 0; i < termnum; i++)
1209             {
1210                 if(tersat[i] >= 0)
1211                     continue;
1212                 if(wcsexists(node->name, terms[i]))
1213                 {
1214                     tersat[i] = level;
1215                     satisfied++;
1216                 } else if(node->child == NULL) {
1217                     break;
1218                 }
1219             }
1220         }
1221         if(!skipcheck && (satisfied == termnum))
1222         {
1223             /* Use DCCHARSET in $Get paths until further researched... */
1224             if((buf = getdcpath(node, NULL, DCCHARSET)) != NULL)
1225             {
1226                 if(node->f.b.hastth)
1227                 {
1228                     buf2 = base32encode(node->hashtth, 24);
1229                     qstrf(dsk, "%s%s\005%i%sTTH:%.39s%s", prefix, buf, node->size, infix, buf2, postfix);
1230                     free(buf2);
1231                 } else {
1232                     qstrf(dsk, "%s%s\005%i%s%s%s", prefix, buf, node->size, infix, hub->nativename, postfix);
1233                 }
1234                 free(buf);
1235             }
1236         }
1237         if((!skipcheck && (satisfied == termnum)) || (node->child == NULL))
1238         {
1239             while(node->next == NULL)
1240             {
1241                 if((node = node->parent) == shareroot)
1242                     break;
1243                 level--;
1244             }
1245             if(node == shareroot)
1246                 break;
1247             for(i = 0; i < termnum; i++)
1248             {
1249                 if(tersat[i] >= level)
1250                 {
1251                     tersat[i] = -1;
1252                     satisfied--;
1253                 }
1254             }
1255             node = node->next;
1256         } else {
1257             node = node->child;
1258             level++;
1259         }
1260     }
1261
1262     hubhandleaction(sk, fn, cmd, args);
1263     
1264  out:
1265     if(dsk != NULL)
1266         putsock(dsk);
1267     if(prefix != NULL)
1268         free(prefix);
1269     if(infix != NULL)
1270         free(infix);
1271     if(postfix != NULL)
1272         free(postfix);
1273     for(i = 0; (i < 32) && (terms[i] != NULL); i++)
1274         free(terms[i]);
1275 }
1276
1277 static void cmd_connecttome(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1278 {
1279     char *p;
1280     struct dchub *hub;
1281     struct socket *newsk;
1282     struct sockaddr_in addr;
1283     
1284     hub = fn->data;
1285     if((p = strchr(args, ' ')) == NULL)
1286         return;
1287     *(p++) = 0;
1288     if(strcmp(args, hub->nativenick))
1289         return;
1290     addr.sin_family = AF_INET;
1291     args = p;
1292     if((p = strchr(args, ':')) == NULL)
1293         return;
1294     *(p++) = 0;
1295     addr.sin_port = htons(atoi(p));
1296     if(!inet_aton(args, &addr.sin_addr))
1297         return;
1298     newsk = netcsconn((struct sockaddr *)&addr, sizeof(addr), (void (*)(struct socket *, int, void *))peerconnect, fn);
1299     getfnetnode(fn);
1300     hubhandleaction(sk, fn, cmd, args);
1301 }
1302
1303 static void sendctm(struct socket *sk, char *nick)
1304 {
1305     struct sockaddr *addr;
1306     socklen_t addrlen;
1307     
1308     if(tcpsock == NULL)
1309         return;
1310     if(sockgetremotename(tcpsock, &addr, &addrlen) < 0)
1311         return;
1312     if(addr->sa_family == AF_INET)
1313         qstrf(sk, "$ConnectToMe %s %s|", nick, formataddress(addr, addrlen));
1314     else
1315         flog(LOG_WARNING, "Direct Connect TCP socket is suddenly not AF_INET, but %i", addr->sa_family);
1316     free(addr);
1317 }
1318
1319 static void cmd_revconnecttome(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1320 {
1321     struct dchub *hub;
1322     char *p;
1323     
1324     hub = fn->data;
1325     if((p = strchr(args, ' ')) == NULL)
1326         return;
1327     *(p++) = 0;
1328     if(strcmp(p, hub->nativenick))
1329         return;
1330     sendctm(sk, args);
1331     expectpeer(args, fn);
1332     hubhandleaction(sk, fn, cmd, args);
1333 }
1334
1335 static void cmd_getnetinfo(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1336 {
1337     struct dchub *hub;
1338     struct fnetnode *node;
1339     int numhubs;
1340     
1341     hub = fn->data;
1342     numhubs = 0;
1343     for(node = fnetnodes; node != NULL; node = node->next)
1344     {
1345         if(node->state == FNN_EST)
1346             numhubs++;
1347     }
1348     qstrf(sk, "$NetInfo %i$%i$%c$0|", confgetint("transfer", "slots"), numhubs, (tcpsock == NULL)?'P':'A');
1349     hubhandleaction(sk, fn, cmd, args);
1350 }
1351
1352 static void cmd_to(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1353 {
1354     struct dchub *hub;
1355     char *p, *p2;
1356     
1357     hub = fn->data;
1358     p = args;
1359     if((p2 = strchr(p, ' ')) == NULL)
1360         return;
1361     *(p2++) = 0;
1362     p = p2;
1363     if((p2 = strchr(p, ' ')) == NULL)
1364         return;
1365     *(p2++) = 0;
1366     if(strcmp(p, "From:"))
1367         return;
1368     p = p2;
1369     if((p2 = strstr(p, " $")) == NULL)
1370         return;
1371     *p2 = 0;
1372     p2 += 2;
1373     hubrecvchat(fn->sk, fn, p, p2);
1374     hubhandleaction(sk, fn, cmd, args);
1375 }
1376
1377 static void cmd_sr(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1378 {
1379     struct dchub *hub;
1380     char *p, *p2, *buf;
1381     char *nick, *filename, *hubname;
1382     int size, slots;
1383     size_t buflen;
1384     struct srchres *sr;
1385     wchar_t *wnick, *wfile;
1386     
1387     hub = fn->data;
1388     nick = p = args;
1389     if((p2 = strchr(p, ' ')) == NULL)
1390         return;
1391     *p2 = 0;
1392     p = p2 + 1;
1393     filename = p;
1394     if((p2 = strchr(p, 5)) == NULL)
1395         return;
1396     *p2 = 0;
1397     p = p2 + 1;
1398     if((p2 = strchr(p, ' ')) == NULL)
1399         return;
1400     *p2 = 0;
1401     size = atoi(p);
1402     p = p2 + 1;
1403     if((p2 = strchr(p, '/')) == NULL)
1404         return;
1405     *p2 = 0;
1406     slots = atoi(p);
1407     p = p2 + 1;
1408     if((p2 = strchr(p, 5)) == NULL)
1409         return;
1410     p = p2 + 1;
1411     hubname = p;
1412     if((p2 = strstr(p, " (")) == NULL)
1413         return;
1414     *p2 = 0;
1415     if((wnick = icmbstowcs(nick, hub->charset)) == NULL)
1416         return;
1417     /* Use DCCHARSET in $Get paths until further researched... */
1418     if((wfile = icmbstowcs(filename, DCCHARSET)) == NULL)
1419     {
1420         free(wnick);
1421         return;
1422     }
1423     sr = newsrchres(&dcnet, wfile, wnick);
1424     if(sr->peernick != NULL)
1425         free(sr->peernick);
1426     sr->peernick = swcsdup(wnick);
1427     sr->size = size;
1428     sr->slots = slots;
1429     free(wfile);
1430     free(wnick);
1431     if(!strncmp(hubname, "TTH:", 4))
1432     {
1433         if((buf = base32decode(hubname + 4, &buflen)) != NULL)
1434         {
1435             if(buflen == 24)
1436                 sr->hash = newhash(L"TTH", 24, buf);
1437             free(buf);
1438         }
1439     }
1440     getfnetnode(sr->fn = fn);
1441     submitsrchres(sr);
1442     freesrchres(sr);
1443     hubhandleaction(sk, fn, cmd, args);
1444 }
1445
1446 static void cmd_usercommand(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1447 {
1448     /* Do nothing for now. */
1449 }
1450
1451 static void cmd_getpass(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1452 {
1453     struct dchub *hub;
1454     wchar_t *pw;
1455     char *mbspw;
1456     
1457     hub = fn->data;
1458     pw = wpfind(fn->args, L"password");
1459     if((pw == NULL) || ((mbspw = icwcstombs(pw, hub->charset)) == NULL))
1460     {
1461         killfnetnode(fn);
1462         return;
1463     }
1464     qstrf(sk, "$MyPass %s|", mbspw);
1465     free(mbspw);
1466     fn->regstatus = FNNS_REG;
1467     hubhandleaction(sk, fn, cmd, args);
1468 }
1469
1470 static void cmd_logedin(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1471 {
1472     struct dchub *hub;
1473     
1474     hub = fn->data;
1475     fn->regstatus = FNNS_OP;
1476     hubhandleaction(sk, fn, cmd, args);
1477 }
1478
1479 static void cmd_mynick(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1480 {
1481     struct dcexppeer *expect;
1482     struct dchub *hub;
1483
1484     if(peer->nativename != NULL)
1485         free(peer->nativename);
1486     peer->nativename = sstrdup(args);
1487     if(peer->wcsname != NULL)
1488         free(peer->wcsname);
1489     if((peer->wcsname = icmbstowcs(peer->nativename, peer->charset)) == NULL)
1490     {
1491         freedcpeer(peer);
1492         return;
1493     }
1494     if(peer->accepted)
1495     {
1496         for(expect = expected; expect != NULL; expect = expect->next)
1497         {
1498             if(!strcmp(expect->nick, args))
1499                 break;
1500         }
1501         if(expect == NULL)
1502         {
1503             peer->fn = NULL;
1504         } else {
1505             hub = expect->fn->data;
1506             peer->fn = expect->fn;
1507             getfnetnode(peer->fn);
1508             peer->dcppemu = hub->dcppemu;
1509             freeexppeer(expect);
1510         }
1511     }
1512 }
1513
1514 static void cmd_direction(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1515 {
1516     char *p;
1517     int mydir;
1518     struct transfer *transfer;
1519     
1520     if((p = strchr(args, ' ')) == NULL)
1521         return;
1522     *p = 0;
1523     if(!strcmp(args, "Upload"))
1524         mydir = TRNSD_DOWN;
1525     if(!strcmp(args, "Download"))
1526         mydir = TRNSD_UP;
1527     if(peer->accepted)
1528     {
1529         if((peer->transfer == NULL) || (mydir != peer->direction))
1530         {
1531             freedcpeer(peer);
1532             return;
1533         }
1534         if(peer->direction == TRNSD_DOWN)
1535             requestfile(peer);
1536     } else {
1537         if(peer->wcsname == NULL)
1538         {
1539             freedcpeer(peer);
1540             return;
1541         }
1542         peer->direction = mydir;
1543         if(peer->direction == TRNSD_UP)
1544         {
1545             transfer = newupload(peer->fn, &dcnet, peer->wcsname, &dctransfer, peer);
1546         } else {
1547             if((transfer = finddownload(peer->wcsname)) == NULL)
1548             {
1549                 freedcpeer(peer);
1550                 return;
1551             }
1552             transferattach(transfer, &dctransfer, peer);
1553             transfersetstate(transfer, TRNS_HS);
1554         }
1555         transfersetnick(transfer, peer->wcsname);
1556         peer->transfer = transfer;
1557         if(peer->extended)
1558             sendsupports(peer);
1559         qstrf(sk, "$Direction %s %i|", (peer->direction == TRNSD_UP)?"Upload":"Download", rand() % 10000);
1560         if(peer->key != NULL)
1561             qstrf(sk, "$Key %s|", peer->key);
1562         if(peer->direction == TRNSD_DOWN)
1563             requestfile(peer);
1564     }
1565 }
1566
1567 static void cmd_peerlock(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1568 {
1569     char *p, *key;
1570     struct transfer *transfer;
1571     
1572     if((p = strchr(args, ' ')) == NULL)
1573         return;
1574     *p = 0;
1575     if(!strncmp(args, "EXTENDEDPROTOCOL", 16))
1576         peer->extended = 1;
1577     key = dcmakekey(args);
1578     if(peer->accepted)
1579     {
1580         if(peer->wcsname == NULL)
1581         {
1582             freedcpeer(peer);
1583             return;
1584         }
1585         sendmynick(peer);
1586         sendpeerlock(peer);
1587         if(peer->extended)
1588             sendsupports(peer);
1589         if((transfer = finddownload(peer->wcsname)) == NULL)
1590         {
1591             peer->direction = TRNSD_UP;
1592             transfer = newupload(peer->fn, &dcnet, peer->wcsname, &dctransfer, peer);
1593         } else {
1594             peer->direction = TRNSD_DOWN;
1595             transferattach(transfer, &dctransfer, peer);
1596             transfersetstate(transfer, TRNS_HS);
1597         }
1598         transfersetnick(transfer, peer->wcsname);
1599         peer->transfer = transfer;
1600         qstrf(sk, "$Direction %s %i|", (peer->direction == TRNSD_UP)?"Upload":"Download", rand() % 10000);
1601         qstrf(sk, "$Key %s|", key);
1602     } else {
1603         if(peer->key != NULL)
1604             free(peer->key);
1605         peer->key = key;
1606     }
1607 }
1608
1609 static void cmd_key(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1610 {
1611     /* NOP */
1612 }
1613
1614 static void startdl(struct dcpeer *peer)
1615 {
1616     if(peer->timeout != NULL)
1617         canceltimer(peer->timeout);
1618     peer->state = PEER_TRNS;
1619     transferstartdl(peer->transfer, peer->sk);
1620     peer->sk->readcb = (void (*)(struct socket *, void *))transread;
1621     peer->sk->errcb = (void (*)(struct socket *, int, void *))transerr;
1622 }
1623
1624 static void startul(struct dcpeer *peer)
1625 {
1626     if(peer->timeout != NULL)
1627         canceltimer(peer->timeout);
1628     peer->state = PEER_TRNS;
1629     transferstartul(peer->transfer, peer->sk);
1630     peer->sk->writecb = (void (*)(struct socket *, void *))transwrite;
1631 }
1632
1633 static void cmd_filelength(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1634 {
1635     int size;
1636     struct transfer *transfer;
1637     
1638     if(peer->transfer == NULL)
1639     {
1640         freedcpeer(peer);
1641         return;
1642     }
1643     size = atoi(args);
1644     if(peer->transfer->size != size)
1645     {
1646         transfersetsize(peer->transfer, size);
1647         transfer = peer->transfer;
1648         freedcpeer(peer);
1649         trytransferbypeer(transfer->fnet, transfer->peerid);
1650         return;
1651     }
1652     startdl(peer);
1653     qstr(peer->sk, "$Send|");
1654 }
1655
1656 static void cmd_error(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1657 {
1658     if(peer->fetchingtthl)
1659     {
1660         peer->fetchingtthl = 0;
1661         peer->notthl = 1;
1662         requestfile(peer);
1663         return;
1664     }
1665     if((peer->transfer != NULL) && (peer->transfer->dir == TRNSD_DOWN))
1666     {
1667         transferseterror(peer->transfer, TRNSE_NOTFOUND);
1668         resettransfer(peer->transfer);
1669         return;
1670     }
1671     freedcpeer(peer);
1672 }
1673
1674 static void cmd_maxedout(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1675 {
1676     if((peer->transfer != NULL) && (peer->transfer->dir == TRNSD_DOWN))
1677     {
1678         transferseterror(peer->transfer, TRNSE_NOSLOTS);
1679         resettransfer(peer->transfer);
1680         return;
1681     }
1682     freedcpeer(peer);
1683 }
1684
1685 static struct
1686 {
1687     char *name;
1688     char **file;
1689     void (*func)(void);
1690 } lists[] =
1691 {
1692     {"MyList.DcLst", &hmlistname, updatehmlist},
1693     {"files.xml", &xmllistname, updatexmllist},
1694     {"files.xml.bz2", &xmlbz2listname, updatexmlbz2list},
1695     {NULL, NULL}
1696 };
1697
1698 static int openfilelist(char *name)
1699 {
1700     int i, fd;
1701     int errnobak;
1702     
1703     for(i = 0; lists[i].name != NULL; i++)
1704     {
1705         if(!strcmp(name, lists[i].name))
1706             break;
1707     }
1708     errno = 0;
1709     if(lists[i].name == NULL)
1710         return(-1);
1711     fd = -1;
1712     if((*lists[i].file == NULL) || ((fd = open(*lists[i].file, O_RDONLY)) < 0))
1713         lists[i].func();
1714     if((fd < 0) && ((*lists[i].file == NULL) || ((fd = open(*lists[i].file, O_RDONLY)) < 0)))
1715     {
1716         errnobak = errno;
1717         flog(LOG_ERR, "could not open filelist tempfile: %s", strerror(errno));
1718         errno = errnobak;
1719         return(-1);
1720     }
1721     return(fd);
1722 }
1723
1724 static struct sharecache *findbytth(char *tth32)
1725 {
1726     char *buf;
1727     size_t buflen;
1728     struct sharecache *node;
1729     
1730     if((buf = base32decode(tth32, &buflen)) == NULL)
1731         return(NULL);
1732     if(buflen != 24)
1733     {
1734         free(buf);
1735         return(NULL);
1736     }
1737     for(node = shareroot->child; node != NULL; node = nextscnode(node))
1738     {
1739         if(node->f.b.hastth && !memcmp(node->hashtth, buf, 24))
1740             break;
1741     }
1742     free(buf);
1743     return(node);
1744 }
1745
1746 static struct sharecache *resdcpath(char *path, char *charset, char sep)
1747 {
1748     struct sharecache *node;
1749     char *p, *p2;
1750
1751     node = shareroot;
1752     p = path;
1753     while(p != NULL)
1754     {
1755         if((p2 = strchr(p, sep)) != NULL)
1756             *(p2++) = 0;
1757         if((node = findcache(node, icsmbstowcs(p, charset, L""))) == NULL)
1758             return(NULL);
1759         p = p2;
1760     }
1761     return(node);
1762 }
1763
1764 static void cmd_get(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1765 {
1766     int offset;
1767     char *p, *buf;
1768     wchar_t *buf2;
1769     struct sharecache *node;
1770     struct socket *lesk;
1771     int fd;
1772     struct stat sb;
1773     
1774     if(peer->transfer == NULL)
1775     {
1776         freedcpeer(peer);
1777         return;
1778     }
1779     if((p = strchr(args, '$')) == NULL)
1780     {
1781         freedcpeer(peer);
1782         return;
1783     }
1784     *(p++) = 0;
1785     if((offset = (atoi(p) - 1)) < 0)
1786     {
1787         freedcpeer(peer);
1788         return;
1789     }
1790     if(((fd = openfilelist(args)) < 0) && (errno != 0))
1791     {
1792         qstr(sk, "$Error Could not send file list|");
1793         freedcpeer(peer);
1794         return;
1795     } else if(fd >= 0) {
1796         if((buf2 = icsmbstowcs(args, DCCHARSET, NULL)) != NULL)
1797             transfersetpath(peer->transfer, buf2);
1798         peer->transfer->flags.b.minislot = 1;
1799     }
1800     if(fd < 0)
1801     {
1802         /* Use DCCHARSET in $Get paths until further researched... */
1803         if((node = resdcpath(args, DCCHARSET, '\\')) == NULL)
1804         {
1805             qstrf(sk, "$Error File not in share|");
1806             freedcpeer(peer);
1807             return;
1808         }
1809         if((fd = opensharecache(node)) < 0)
1810         {
1811             qstrf(sk, "$Error %s|", strerror(errno));
1812             freedcpeer(peer);
1813             return;
1814         }
1815         buf = getfspath(node);
1816         if((buf2 = icsmbstowcs(buf, NULL, NULL)) == NULL)
1817             flog(LOG_WARNING, "could not convert native path into a wcs (%s): %s", buf, strerror(errno));
1818         else
1819             transfersetpath(peer->transfer, buf2);
1820         free(buf);
1821     }
1822     if(fstat(fd, &sb) < 0)
1823     {
1824         close(fd);
1825         flog(LOG_WARNING, "could not stat file %ls: %s", node->name, strerror(errno));
1826         qstrf(sk, "$Error|");
1827         freedcpeer(peer);
1828         return;
1829     }
1830     if(sb.st_size < 65536)
1831         peer->transfer->flags.b.minislot = 1;
1832     if(!peer->transfer->flags.b.minislot && (slotsleft() < 1)) {
1833         close(fd);
1834         qstr(sk, "$MaxedOut|");
1835         freedcpeer(peer);
1836         return;
1837     }
1838     if((offset != 0) && (lseek(fd, offset, SEEK_SET) < 0))
1839     {
1840         close(fd);
1841         qstrf(sk, "$Error Offset out of range|");
1842         freedcpeer(peer);
1843         return;
1844     }
1845     lesk = wrapsock(fd);
1846     transferprepul(peer->transfer, sb.st_size, offset, -1, lesk);
1847     putsock(lesk);
1848     qstrf(sk, "$FileLength %i|", peer->transfer->size);
1849 }
1850
1851 static void cmd_send(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1852 {
1853     if(peer->transfer == NULL)
1854     {
1855         freedcpeer(peer);
1856         return;
1857     }
1858     if(peer->transfer->localend == NULL)
1859     {
1860         freedcpeer(peer);
1861         return;
1862     }
1863     peer->ptclose = 1;
1864     startul(peer);
1865 }
1866
1867 static void cmd_supports(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1868 {
1869     int i;
1870     char *p, *p2;
1871     char **arr;
1872     size_t arrsize, arrdata;
1873     
1874     if(peer->supports != NULL)
1875     {
1876         for(i = 0; peer->supports[i] != NULL; i++)
1877             free(peer->supports[i]);
1878         free(peer->supports);
1879     }
1880     arr = NULL;
1881     arrsize = arrdata = 0;
1882     p = args;
1883     do
1884     {
1885         if((p2 = strchr(p, ' ')) != NULL)
1886             *(p2++) = 0;
1887         if(*p == 0)
1888             continue;
1889         addtobuf(arr, sstrdup(p));
1890     } while((p = p2) != NULL);
1891     addtobuf(arr, NULL);
1892     peer->supports = arr;
1893 }
1894
1895 static void cmd_getblock(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1896 {
1897     int fd;
1898     char *p, *p2;
1899     int start, numbytes;
1900     char *charset, *buf;
1901     wchar_t *buf2;
1902     struct sharecache *node;
1903     struct stat sb;
1904     struct socket *lesk;
1905     
1906     if(peer->transfer == NULL)
1907     {
1908         freedcpeer(peer);
1909         return;
1910     }
1911     p = args;
1912     if((p2 = strchr(p, ' ')) == NULL)
1913     {
1914         freedcpeer(peer);
1915         return;
1916     }
1917     *(p2++) = 0;
1918     start = atoi(p);
1919     p = p2;
1920     if((p2 = strchr(p, ' ')) == NULL)
1921     {
1922         freedcpeer(peer);
1923         return;
1924     }
1925     *(p2++) = 0;
1926     numbytes = atoi(p);
1927     p = p2;
1928     if(!strcmp(cmd, "$UGetBlock") || !strcmp(cmd, "$UGetZBlock"))
1929         charset = "UTF-8";
1930     else
1931         /* Use DCCHARSET in $Get paths until further researched... */
1932         charset = DCCHARSET;
1933     if(!strcmp(cmd, "$GetZBlock") || !strcmp(cmd, "$UGetZBlock"))
1934         initcompress(peer, CPRS_ZLIB);
1935     if(((fd = openfilelist(p)) < 0) && (errno != 0))
1936     {
1937         qstr(sk, "$Error Could not send file list|");
1938         return;
1939     } else if(fd >= 0) {
1940         if((buf2 = icsmbstowcs(args, charset, NULL)) != NULL)
1941             transfersetpath(peer->transfer, buf2);
1942         peer->transfer->flags.b.minislot = 1;
1943     }
1944     if(fd < 0)
1945     {
1946         if((node = resdcpath(p, charset, '\\')) == NULL)
1947         {
1948             qstr(sk, "$Error File not in cache|");
1949             return;
1950         }
1951         if((fd = opensharecache(node)) < 0)
1952         {
1953             qstrf(sk, "$Error %s|", strerror(errno));
1954             return;
1955         }
1956         buf = getfspath(node);
1957         if((buf2 = icsmbstowcs(buf, NULL, NULL)) == NULL)
1958             flog(LOG_WARNING, "could not convert native path into a wcs (%s): %s", buf, strerror(errno));
1959         else
1960             transfersetpath(peer->transfer, buf2);
1961         free(buf);
1962     }
1963     if(fstat(fd, &sb) < 0)
1964     {
1965         close(fd);
1966         flog(LOG_WARNING, "could not stat file %ls: %s", node->name, strerror(errno));
1967         qstr(sk, "$Error|");
1968         return;
1969     }
1970     if(sb.st_size < 65536)
1971         peer->transfer->flags.b.minislot = 1;
1972     if(!peer->transfer->flags.b.minislot && (slotsleft() < 1)) {
1973         close(fd);
1974         qstr(sk, "$MaxedOut|");
1975         return;
1976     }
1977     if((start != 0) && ((start >= sb.st_size) || (lseek(fd, start, SEEK_SET) < 0)))
1978     {
1979         close(fd);
1980         qstr(sk, "$Error Offset out of range|");
1981         return;
1982     }
1983     if((numbytes < 0) || (start + numbytes > sb.st_size))
1984         numbytes = sb.st_size - start;
1985     lesk = wrapsock(fd);
1986     transferprepul(peer->transfer, sb.st_size, start, start + numbytes, lesk);
1987     putsock(lesk);
1988     qstrf(sk, "$Sending %i|", numbytes);
1989     startul(peer);
1990 }
1991
1992 static void cmd_adcget(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1993 {
1994     int i;
1995     char **argv, *buf;
1996     int start, numbytes;
1997     struct sharecache *node;
1998     struct stat sb;
1999     struct socket *lesk;
2000     wchar_t *wbuf;
2001     int fd;
2002     
2003     if(peer->transfer == NULL)
2004     {
2005         freedcpeer(peer);
2006         return;
2007     }
2008     if((argv = parseadc(args)) == NULL)
2009     {
2010         freedcpeer(peer);
2011         return;
2012     }
2013     if(parrlen(argv) < 4)
2014     {
2015         freedcpeer(peer);
2016         goto out;
2017     }
2018     start = atoi(argv[2]);
2019     numbytes = atoi(argv[3]);
2020     node = NULL;
2021     fd = -1;
2022     if(((fd = openfilelist(argv[1])) < 0) && (errno != 0))
2023     {
2024         qstr(sk, "$Error Could not send file list|");
2025         goto out;
2026     } else if(fd >= 0) {
2027         if((wbuf = icsmbstowcs(argv[1], "UTF-8", NULL)) != NULL)
2028             transfersetpath(peer->transfer, wbuf);
2029         peer->transfer->flags.b.minislot = 1;
2030     }
2031     if(fd < 0)
2032     {
2033         if(!strncmp(argv[1], "TTH/", 4))
2034         {
2035             if((node = findbytth(argv[1] + 4)) == NULL)
2036             {
2037                 qstr(sk, "$Error File not in cache|");
2038                 goto out;
2039             }
2040         } else {
2041             if((node = resdcpath((argv[1][0] == '/')?(argv[1] + 1):(argv[1]), "UTF-8", '/')) == NULL)
2042             {
2043                 qstr(sk, "$Error File not in cache|");
2044                 goto out;
2045             }
2046         }
2047         if((fd = opensharecache(node)) < 0)
2048         {
2049             qstrf(sk, "$Error %s|", strerror(errno));
2050             goto out;
2051         }
2052         buf = getfspath(node);
2053         if((wbuf = icsmbstowcs(buf, NULL, NULL)) == NULL)
2054             flog(LOG_WARNING, "could not convert native path into a wcs (%s): %s", buf, strerror(errno));
2055         else
2056             transfersetpath(peer->transfer, wbuf);
2057         free(buf);
2058     }
2059     if(!strcmp(argv[0], "file"))
2060     {
2061         for(i = 4; argv[i] != NULL; i++)
2062         {
2063             if(!strcmp(argv[i], "ZL1"))
2064                 initcompress(peer, CPRS_ZLIB);
2065         }
2066         if(fstat(fd, &sb) < 0)
2067         {
2068             flog(LOG_WARNING, "could not stat file %ls: %s", node->name, strerror(errno));
2069             qstr(sk, "$Error|");
2070             goto out;
2071         }
2072         if(sb.st_size < 65536)
2073             peer->transfer->flags.b.minislot = 1;
2074         if(!peer->transfer->flags.b.minislot && (slotsleft() < 1)) {
2075             qstr(sk, "$MaxedOut|");
2076             goto out;
2077         }
2078         if((start != 0) && ((start >= sb.st_size) || (lseek(fd, start, SEEK_SET) < 0)))
2079         {
2080             qstr(sk, "$Error Offset out of range|");
2081             goto out;
2082         }
2083         if((numbytes < 0) || (start + numbytes > sb.st_size))
2084             numbytes = sb.st_size - start;
2085         lesk = wrapsock(fd);
2086         transferprepul(peer->transfer, sb.st_size, start, start + numbytes, lesk);
2087         putsock(lesk);
2088         fd = -1;
2089         qstr(sk, "$ADCSND");
2090         sendadc(sk, "file");
2091         sendadc(sk, argv[1]);
2092         sendadcf(sk, "%i", start);
2093         sendadcf(sk, "%i", numbytes);
2094         if(peer->compress == CPRS_ZLIB)
2095             sendadc(sk, "ZL1");
2096         qstr(sk, "|");
2097         startul(peer);
2098     } else if(!strcmp(argv[0], "tthl")) {
2099         /*
2100          * XXX: Implement full TTHL support.
2101          *
2102          * In the meantime, this is better than nothing, since it at
2103          * least allows fetching of the TTH root if it isn't already
2104          * known.
2105          */
2106         if(node == NULL)
2107         {
2108             qstr(sk, "$Error no TTHL data for virtual files|");
2109             goto out;
2110         }
2111         qstr(sk, "$ADCSND");
2112         sendadc(sk, "tthl");
2113         sendadc(sk, argv[1]);
2114         sendadc(sk, "0");
2115         sendadc(sk, "24");
2116         qstr(sk, "|");
2117         sockqueue(sk, node->hashtth, 24);
2118     } else {
2119         qstr(sk, "$Error Namespace not implemented|");
2120         goto out;
2121     }
2122     
2123  out:
2124     if(fd >= 0)
2125         close(fd);
2126     freeparr(argv);
2127 }
2128
2129 static void handletthl(struct dcpeer *peer)
2130 {
2131     char buf[24];
2132     
2133     while(peer->inbufdata >= 24)
2134     {
2135         pushtigertree(&peer->tth, peer->inbuf);
2136         memmove(peer->inbuf, peer->inbuf + 24, peer->inbufdata -= 24);
2137         peer->curread += 24;
2138     }
2139     if(peer->curread >= peer->totalsize)
2140     {
2141         if(peer->timeout == NULL)
2142             peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
2143         peer->state = PEER_CMD;
2144         synctigertree(&peer->tth);
2145         restigertree(&peer->tth, buf);
2146         transfersethash(peer->transfer, newhash(L"TTH", 24, buf));
2147         requestfile(peer);
2148     }
2149 }
2150
2151 static void cmd_adcsnd(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2152 {
2153     char **argv;
2154     int start, numbytes;
2155     
2156     if(peer->transfer == NULL)
2157     {
2158         freedcpeer(peer);
2159         return;
2160     }
2161     if((argv = parseadc(args)) == NULL)
2162     {
2163         freedcpeer(peer);
2164         return;
2165     }
2166     if(parrlen(argv) < 4)
2167     {
2168         freedcpeer(peer);
2169         goto out;
2170     }
2171     start = atoi(argv[2]);
2172     numbytes = atoi(argv[3]);
2173     if(!strcmp(argv[0], "tthl"))
2174     {
2175         if((start != 0) || (numbytes % 24 != 0))
2176         {
2177             /* Weird. Bail out. */
2178             freedcpeer(peer);
2179             goto out;
2180         }
2181         if(peer->timeout != NULL)
2182             canceltimer(peer->timeout);
2183         peer->state = PEER_TTHL;
2184         peer->totalsize = numbytes;
2185         peer->curread = 0;
2186         peer->fetchingtthl = 0;
2187         inittigertree(&peer->tth);
2188         handletthl(peer);
2189     } else if(!strcmp(argv[0], "file")) {
2190         if(start != peer->transfer->curpos)
2191         {
2192             freedcpeer(peer);
2193             goto out;
2194         }
2195         if(start + numbytes != peer->transfer->size)
2196         {
2197             transfersetsize(peer->transfer, start + numbytes);
2198             freedcpeer(peer);
2199             goto out;
2200         }
2201         startdl(peer);
2202         if(peer->inbufdata > 0)
2203         {
2204             sockpushdata(sk, peer->inbuf, peer->inbufdata);
2205             peer->inbufdata = 0;
2206             transread(sk, peer);
2207         }
2208     } else {
2209         /* We certainly didn't request this...*/
2210         freedcpeer(peer);
2211         goto out;
2212     }
2213     
2214  out:
2215     freeparr(argv);
2216 }
2217
2218 static void cmd_sending(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2219 {
2220     int numbytes;
2221     
2222     if(peer->transfer == NULL)
2223     {
2224         freedcpeer(peer);
2225         return;
2226     }
2227     numbytes = atoi(args);
2228     if(peer->transfer->size - peer->transfer->curpos != numbytes)
2229     {
2230         transfersetsize(peer->transfer, peer->transfer->curpos + numbytes);
2231         freedcpeer(peer);
2232         return;
2233     }
2234     startdl(peer);
2235     if(peer->inbufdata > 0)
2236     {
2237         sockpushdata(sk, peer->inbuf, peer->inbufdata);
2238         peer->inbufdata = 0;
2239         transread(sk, peer);
2240     }
2241 }
2242
2243 /*
2244 Hub skeleton:
2245 static void cmd_(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
2246 {
2247     struct dchub *hub;
2248     
2249     hub = fn->data;
2250     hubhandleaction(sk, fn, cmd, args);
2251 }
2252
2253 Peer skeleton:
2254 static void cmd_(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2255 {
2256 }
2257
2258 */
2259
2260 static int hubreqconn(struct fnetpeer *peer)
2261 {
2262     struct dchub *hub;
2263     char *mbsnick;
2264     
2265     if((peer->fn->state != FNN_EST) || (peer->fn->fnet != &dcnet))
2266     {
2267         errno = EINVAL;
2268         return(1);
2269     }
2270     if((hub = peer->fn->data) == NULL)
2271     {
2272         errno = EFAULT;
2273         return(1);
2274     }
2275     if((mbsnick = icwcstombs(peer->id, hub->charset)) == NULL)
2276         return(1); /* Shouldn't happen, of course, but who knows... */
2277     if(tcpsock != NULL)
2278     {
2279         sendctm(peer->fn->sk, mbsnick);
2280         expectpeer(mbsnick, peer->fn);
2281     } else {
2282         qstrf(peer->fn->sk, "$RevConnectToMe %s %s|", hub->nativenick, mbsnick);
2283     }
2284     free(mbsnick);
2285     return(0);
2286 }
2287
2288 static int hubsendchat(struct fnetnode *fn, int public, wchar_t *to, wchar_t *string)
2289 {
2290     struct dchub *hub;
2291     char *mbsstring, *mbsto;
2292     
2293     hub = fn->data;
2294     if((mbsto = icwcstombs(to, hub->charset)) == NULL)
2295     {
2296         errno = EILSEQ;
2297         return(1);
2298     }
2299     if((mbsstring = icwcstombs(string, hub->charset)) == NULL)
2300     {
2301         errno = EILSEQ;
2302         return(1);
2303     }
2304     if((strchr(mbsto, '|') != NULL) || (strchr(mbsto, ' ') != NULL))
2305     {
2306         free(mbsto);
2307         free(mbsstring);
2308         errno = ESRCH;
2309         return(1);
2310     }
2311     if(strchr(mbsstring, '|') != NULL)
2312     {
2313         free(mbsto);
2314         free(mbsstring);
2315         errno = EILSEQ;
2316         return(1);
2317     }
2318     if(public)
2319     {
2320         if(*to == L'\0')
2321         {
2322             qstrf(fn->sk, "<%s> %s|", hub->nativenick, mbsstring);
2323         } else {
2324             qstrf(fn->sk, "$To: %s From: %s $<%s> %s|", mbsto, hub->nativenick, hub->nativenick, mbsstring);
2325         }
2326     } else {
2327         qstrf(fn->sk, "$To: %s From: %s $<%s> %s|", mbsto, hub->nativenick, hub->nativenick, mbsstring);
2328     }
2329     free(mbsto);
2330     free(mbsstring);
2331     return(0);
2332 }
2333
2334 static void findsizelimit(struct sexpr *sexpr, int *min, int *max)
2335 {
2336     int minl, maxl, minr, maxr, retmin, retmax;
2337     
2338     switch(sexpr->op)
2339     {
2340     case SOP_AND:
2341         findsizelimit(sexpr->l, &minl, &maxl);
2342         findsizelimit(sexpr->r, &minr, &maxr);
2343         retmin = (minl > minr)?minl:minr;
2344         if((maxl != -1) && (maxr != -1))
2345             retmax = (maxl < maxr)?maxl:maxr;
2346         else if(maxl != -1)
2347             retmax = maxl;
2348         else if(maxr != -1)
2349             retmax = maxr;
2350         else
2351             retmax = -1;
2352         break;
2353     case SOP_OR:
2354         findsizelimit(sexpr->l, &minl, &maxl);
2355         findsizelimit(sexpr->r, &minr, &maxr);
2356         retmin = (minl < minr)?minl:minr;
2357         if((maxl == -1) || (maxr == -1))
2358             retmax = -1;
2359         else
2360             retmax = (maxl > maxr)?maxl:maxr;
2361         break;
2362     case SOP_NOT:
2363         findsizelimit(sexpr->l, &minl, &maxl);
2364         if((minl == 0) && (maxl == -1)) /* Interval is unspecified */
2365         {
2366             retmin = 0;
2367             retmax = -1;
2368         } else if((minl == 0) && (maxl != -1)) {
2369             retmin = maxl + 1;
2370             retmax = -1;
2371         } else if((minl != 0) && (maxl == -1)) {
2372             retmin = 0;
2373             retmax = minl - 1;
2374         } else { /* This would yield two seperate intervals, which DC cannot handle */
2375             retmin = 0;
2376             retmax = -1;
2377         }
2378     case SOP_SIZELT:
2379         retmin = 0;
2380         retmax = sexpr->d.n - 1;
2381         break;
2382     case SOP_SIZEEQ:
2383         retmin = sexpr->d.n;
2384         retmax = sexpr->d.n;
2385         break;
2386     case SOP_SIZEGT:
2387         retmin = sexpr->d.n + 1;
2388         retmax = -1;
2389         break;
2390     default:
2391         retmin = 0;
2392         retmax = -1;
2393         break;
2394     }
2395     if(min != NULL)
2396         *min = retmin;
2397     if(max != NULL)
2398         *max = retmax;
2399 }
2400
2401 static struct hash *findsehash(struct sexpr *sexpr)
2402 {
2403     struct hash *h1, *h2;
2404     
2405     switch(sexpr->op)
2406     {
2407     case SOP_AND:
2408         if((h1 = findsehash(sexpr->l)) != NULL)
2409             return(h1);
2410         if((h1 = findsehash(sexpr->r)) != NULL)
2411             return(h1);
2412         break;
2413     case SOP_OR:
2414         h1 = findsehash(sexpr->l);
2415         h2 = findsehash(sexpr->r);
2416         if(hashcmp(h1, h2))
2417             return(h1);
2418         break;
2419     case SOP_HASHIS:
2420         if(!wcscmp(sexpr->d.hash->algo, L"TTH"))
2421             return(sexpr->d.hash);
2422     default:
2423         break;
2424     }
2425     return(NULL);
2426 }
2427
2428 static int hubsearch(struct fnetnode *fn, struct search *srch, struct srchfnnlist *ln)
2429 {
2430     struct dchub *hub;
2431     struct wcslist *list, *cur;
2432     char *sstr, *buf, *p;
2433     size_t sstrsize, sstrdata;
2434     struct sockaddr *name;
2435     socklen_t namelen;
2436     int minsize, maxsize;
2437     struct hash *hash;
2438     
2439     hub = fn->data;
2440     if((fn->state != FNN_EST) || (fn->sk == NULL) || (fn->sk->state != SOCK_EST))
2441         return(1);
2442     list = findsexprstrs(srch->sexpr);
2443     findsizelimit(srch->sexpr, &minsize, &maxsize);
2444     hash = findsehash(srch->sexpr);
2445     if((minsize != 0) && (maxsize != -1))
2446     {
2447         /* Choose either minsize or maxsize by trying to determine
2448          * which choice will be most restrictive. The result can be
2449          * approximative at best anyway, so don't try too hard... */
2450         if((50000000 - maxsize) > (minsize - 50000000))
2451             minsize = 0;
2452         else
2453             maxsize = -1;
2454     }
2455     sstr = NULL;
2456     sstrsize = sstrdata = 0;
2457     if((hash != NULL) && (hash->len == 24))
2458     {
2459         /* Prioritize hash searches above all else */
2460         bufcat(sstr, "F?T?0?9?TTH:", 12);
2461         buf = base32encode(hash->buf, hash->len);
2462         
2463         bufcat(sstr, buf, 39);
2464         free(buf);
2465     } else {
2466         if(minsize != 0)
2467         {
2468             sizebuf2(sstr, sstrdata + 32, 1);
2469             sstrdata += snprintf(sstr + sstrdata, sstrsize - sstrdata, "T?F?%i?1?", minsize);
2470         } else if(maxsize != -1) {
2471             sizebuf2(sstr, sstrdata + 32, 1);
2472             sstrdata += snprintf(sstr + sstrdata, sstrsize - sstrdata, "T?T?%i?1?", maxsize);
2473         } else {
2474             bufcat(sstr, "F?F?0?1?", 8);
2475         }
2476         if(list != NULL)
2477         {
2478             for(cur = list; cur != NULL; cur = cur->next)
2479             {
2480                 /* Use DCCHARSET in $Get paths until further researched... */
2481                 if((buf = icwcstombs(cur->str, DCCHARSET)) == NULL)
2482                 {
2483                     /* Can't find anything anyway if the search expression
2484                      * requires characters outside DC's charset. There's
2485                      * nothing technically wrong with the search itself,
2486                      * however, so return success. This should be
2487                      * considered as an optimization. */
2488                     freesl(&list);
2489                     if(sstr != NULL)
2490                         free(sstr);
2491                     return(0);
2492                 }
2493                 if(cur != list)
2494                     addtobuf(sstr, '$');
2495                 /*
2496                  * It probably doesn't hurt if buf contains any extra
2497                  * dollar signs - it will just result in extra search
2498                  * terms, and the extraneous results will be filtered by
2499                  * the search layer anyway. It hurts if it contains any
2500                  * pipes, though, so let's sell them for money.
2501                  */
2502                 for(p = buf; *p; p++)
2503                 {
2504                     if(*p == '|')
2505                         *p = '$';
2506                 }
2507                 bufcat(sstr, buf, strlen(buf));
2508                 free(buf);
2509             }
2510         } else {
2511             /* Will match all files... :-/ */
2512             addtobuf(sstr, '.');
2513         }
2514     }
2515     addtobuf(sstr, 0);
2516     if(tcpsock != NULL)
2517     {
2518         if(sockgetremotename(udpsock, &name, &namelen) < 0)
2519         {
2520             flog(LOG_WARNING, "cannot get address of UDP socket");
2521         } else {
2522             qstrf(fn->sk, "$Search %s %s|", formataddress(name, namelen), sstr);
2523             free(name);
2524         }
2525     } else {
2526         qstrf(fn->sk, "$Search Hub:%s %s|", hub->nativenick, sstr);
2527     }
2528     free(sstr);
2529     freesl(&list);
2530     fn->lastsrch = time(NULL);
2531     return(0);
2532 }
2533
2534 #undef skipcmd
2535 #undef qstr
2536 #undef qstrf
2537
2538 #define cc(c) ((void (*)(struct socket *, void *, char *, char *))(c))
2539 struct command hubcmds[] =
2540 {
2541     {"$Lock", cc(cmd_lock)},
2542     {"$HubName", cc(cmd_hubname)},
2543     {"$Hello", cc(cmd_hello)},
2544     {"$Quit", cc(cmd_quit)},
2545     {"$NickList", cc(cmd_nicklist)},
2546     {"$OpList", cc(cmd_oplist)},
2547     {"$MyINFO", cc(cmd_myinfo)},
2548     {"$ForceMove", cc(cmd_forcemove)},
2549     {"$Search", cc(cmd_search)},
2550     {"$MultiSearch", cc(cmd_search)},
2551     {"$ConnectToMe", cc(cmd_connecttome)},
2552     {"$RevConnectToMe", cc(cmd_revconnecttome)},
2553     {"$GetNetInfo", cc(cmd_getnetinfo)},
2554     {"$To:", cc(cmd_to)},
2555     {"$SR", cc(cmd_sr)},
2556     {"$UserCommand", cc(cmd_usercommand)},
2557     {"$GetPass", cc(cmd_getpass)},
2558     {"$LogedIn", cc(cmd_logedin)}, /* sic */
2559     {NULL, NULL}
2560 };
2561
2562 struct command peercmds[] =
2563 {
2564     {"$MyNick", cc(cmd_mynick)},
2565     {"$Lock", cc(cmd_peerlock)},
2566     {"$Direction", cc(cmd_direction)},
2567     {"$Key", cc(cmd_key)},
2568     {"$FileLength", cc(cmd_filelength)},
2569     {"$Error", cc(cmd_error)},
2570     {"$MaxedOut", cc(cmd_maxedout)},
2571     {"$Get", cc(cmd_get)},
2572     {"$Send", cc(cmd_send)},
2573     {"$Supports", cc(cmd_supports)},
2574     {"$GetBlock", cc(cmd_getblock)},
2575     {"$UGetBlock", cc(cmd_getblock)},
2576     {"$GetZBlock", cc(cmd_getblock)},
2577     {"$UGetZBlock", cc(cmd_getblock)},
2578     {"$ADCGET", cc(cmd_adcget)},
2579     {"$ADCSND", cc(cmd_adcsnd), 1},
2580     {"$Sending", cc(cmd_sending), 1},
2581     {NULL, NULL}
2582 };
2583 #undef cc
2584
2585 static void dctransdetach(struct transfer *transfer, struct dcpeer *peer)
2586 {
2587     CBUNREG(transfer, trans_filterout, peer);
2588     if(peer->freeing)
2589         return;
2590     peer->transfer = NULL;
2591     freedcpeer(peer);
2592 }
2593
2594 static void dctransgotdata(struct transfer *transfer, struct dcpeer *peer)
2595 {
2596     int ret;
2597     void *buf;
2598     char outbuf[1024];
2599     z_stream *cstr;
2600     size_t bufsize;
2601     
2602     if((peer->state == PEER_TRNS) || (peer->state == PEER_SYNC))
2603     {
2604         if(sockqueuesize(peer->sk) < 65536)
2605         {
2606             if((buf = transfergetdata(transfer, &bufsize)) != NULL)
2607             {
2608                 if(peer->compress == CPRS_NONE)
2609                 {
2610                     sockqueue(peer->sk, buf, bufsize);
2611                 } else if(peer->compress == CPRS_ZLIB) {
2612                     cstr = peer->cprsdata;
2613                     cstr->next_in = buf;
2614                     cstr->avail_in = bufsize;
2615                     while(cstr->avail_in > 0)
2616                     {
2617                         cstr->next_out = outbuf;
2618                         cstr->avail_out = sizeof(outbuf);
2619                         if((ret = deflate(cstr, 0)) != Z_OK)
2620                         {
2621                             flog(LOG_WARNING, "bug? deflate() did not return Z_OK (but rather %i)", ret);
2622                             freedcpeer(peer);
2623                             return;
2624                         }
2625                         sockqueue(peer->sk, outbuf, sizeof(outbuf) - cstr->avail_out);
2626                     }
2627                 }
2628                 free(buf);
2629             }
2630             if(peer->state == PEER_SYNC)
2631             {
2632                 if(peer->compress == CPRS_ZLIB)
2633                 {
2634                     cstr = peer->cprsdata;
2635                     cstr->next_in = NULL;
2636                     cstr->avail_in = 0;
2637                     do
2638                     {
2639                         cstr->next_out = outbuf;
2640                         cstr->avail_out = sizeof(outbuf);
2641                         ret = deflate(cstr, Z_FINISH);
2642                         if((ret != Z_OK) && (ret != Z_STREAM_END))
2643                         {
2644                             flog(LOG_WARNING, "bug? deflate(Z_FINISH) did not return Z_OK (but rather %i)", ret);
2645                             freedcpeer(peer);
2646                             return;
2647                         }
2648                         sockqueue(peer->sk, outbuf, sizeof(outbuf) - cstr->avail_out);
2649                     } while(ret != Z_STREAM_END);
2650                 }
2651                 if(peer->ptclose)
2652                 {
2653                     freedcpeer(peer);
2654                 } else {
2655                     if(peer->timeout == NULL)
2656                         peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
2657                     peer->state = PEER_CMD;
2658                     endcompress(peer);
2659                     transfersetstate(transfer, TRNS_HS);
2660                     socksettos(peer->sk, confgetint("fnet", "fnptos"));
2661                     transfer->flags.b.minislot = 0;
2662                     peer->sk->writecb = NULL;
2663                 }
2664             }
2665         }
2666     }
2667 }
2668
2669 static void dctransendofdata(struct transfer *transfer, struct dcpeer *peer)
2670 {
2671     peer->state = PEER_SYNC;
2672     dctransgotdata(transfer, peer);
2673 }
2674
2675 static void dcwantdata(struct transfer *transfer, struct dcpeer *peer)
2676 {
2677     if(transferdatasize(transfer) < 65536)
2678         peer->sk->ignread = 0;
2679 }
2680
2681 static void transread(struct socket *sk, struct dcpeer *peer)
2682 {
2683     void *buf;
2684     size_t bufsize;
2685     struct transfer *transfer;
2686     
2687     if((buf = sockgetinbuf(sk, &bufsize)) == NULL)
2688         return;
2689     if(peer->transfer == NULL)
2690     {
2691         free(buf);
2692         freedcpeer(peer);
2693         return;
2694     }
2695     transferputdata(peer->transfer, buf, bufsize);
2696     free(buf);
2697     if(peer->transfer->curpos >= peer->transfer->size)
2698     {
2699         transfer = peer->transfer;
2700         transferdetach(transfer);
2701         transferendofdata(transfer);
2702         return;
2703     }
2704     if(transferdatasize(peer->transfer) > 65535)
2705         sk->ignread = 1;
2706 }
2707
2708 static void transerr(struct socket *sk, int err, struct dcpeer *peer)
2709 {
2710     struct transfer *transfer;
2711
2712     if((transfer = peer->transfer) == NULL)
2713     {
2714         freedcpeer(peer);
2715         return;
2716     }
2717     transferdetach(transfer);
2718     transferendofdata(transfer);
2719 }
2720
2721 static void transwrite(struct socket *sk, struct dcpeer *peer)
2722 {
2723     if((peer->state != PEER_TRNS) && (peer->state != PEER_SYNC))
2724         return;
2725     if(peer->transfer == NULL)
2726     {
2727         freedcpeer(peer);
2728         return;
2729     }
2730     dctransgotdata(peer->transfer, peer);
2731 }
2732
2733 static void udpread(struct socket *sk, void *data)
2734 {
2735     char *buf, *p, *p2, *hashbuf;
2736     size_t buflen, hashlen;
2737     char *nick, *filename, *hubname;
2738     struct sockaddr_in hubaddr;
2739     int size, slots;
2740     struct fnetnode *fn, *myfn;
2741     struct dchub *hub;
2742     struct srchres *sr;
2743     wchar_t *wnick, *wfile;
2744     struct hash *hash;
2745     
2746     if((buf = sockgetinbuf(sk, &buflen)) == NULL)
2747         return;
2748     buf = srealloc(buf, buflen + 1);
2749     buf[buflen] = 0;
2750     if(!strncmp(buf, "$SR ", 4))
2751     {
2752         p = buf + 4;
2753         nick = p;
2754         if((p2 = strchr(p, ' ')) == NULL)
2755         {
2756             free(buf);
2757             return;
2758         }
2759         *p2 = 0;
2760         p = p2 + 1;
2761         filename = p;
2762         if((p2 = strchr(p, 5)) == NULL)
2763         {
2764             free(buf);
2765             return;
2766         }
2767         *p2 = 0;
2768         p = p2 + 1;
2769         if((p2 = strchr(p, ' ')) == NULL)
2770         {
2771             free(buf);
2772             return;
2773         }
2774         *p2 = 0;
2775         size = atoi(p);
2776         p = p2 + 1;
2777         if((p2 = strchr(p, '/')) == NULL)
2778         {
2779             free(buf);
2780             return;
2781         }
2782         *p2 = 0;
2783         slots = atoi(p);
2784         p = p2 + 1;
2785         if((p2 = strchr(p, 5)) == NULL)
2786         {
2787             free(buf);
2788             return;
2789         }
2790         p = p2 + 1;
2791         hubname = p;
2792         if((p2 = strstr(p, " (")) == NULL)
2793         {
2794             free(buf);
2795             return;
2796         }
2797         *p2 = 0;
2798         p = p2 + 2;
2799         if((p2 = strchr(p, ':')) == NULL)
2800         {
2801             free(buf);
2802             return;
2803         }
2804         *(p2++) = 0;
2805         hubaddr.sin_family = AF_INET;
2806         if(!inet_aton(p, &hubaddr.sin_addr))
2807         {
2808             free(buf);
2809             return;
2810         }
2811         p = p2;
2812         if((p2 = strchr(p, ')')) == NULL)
2813         {
2814             free(buf);
2815             return;
2816         }
2817         *p2 = 0;
2818         hubaddr.sin_port = htons(atoi(p));
2819         /* Use DCCHARSET in $Get paths until further researched... */
2820         if((wfile = icmbstowcs(filename, DCCHARSET)) == NULL)
2821         {
2822             free(buf);
2823             return;
2824         }
2825         myfn = NULL;
2826         hash = NULL;
2827         if(!strncmp(hubname, "TTH:", 4))
2828         {
2829             if((hashbuf = base32decode(hubname + 4, &hashlen)) != NULL)
2830             {
2831                 if(hashlen == 24)
2832                     hash = newhash(L"TTH", 24, hashbuf);
2833                 free(hashbuf);
2834             }
2835         } else {
2836             for(fn = fnetnodes; fn != NULL; fn = fn->next)
2837             {
2838                 if((fn->fnet == &dcnet) && ((hub = fn->data) != NULL))
2839                 {
2840                     if((hub->nativename != NULL) && !strcmp(hub->nativename, hubname))
2841                     {
2842                         if(myfn == NULL)
2843                         {
2844                             myfn = fn;
2845                         } else {
2846                             myfn = NULL;
2847                             break;
2848                         }
2849                     }
2850                 }
2851             }
2852         }
2853         if(myfn == NULL)
2854         {
2855             for(fn = fnetnodes; fn != NULL; fn = fn->next)
2856             {
2857                 if((fn->fnet == &dcnet) && (fn->sk != NULL) && addreq(fn->sk->remote, (struct sockaddr *)&hubaddr))
2858                 {
2859                     myfn = fn;
2860                     break;
2861                 }
2862             }
2863         }
2864         hub = NULL;
2865         if(myfn != NULL)
2866             hub = myfn->data;
2867         if((wnick = icmbstowcs(nick, (hub == NULL)?DCCHARSET:(hub->charset))) == NULL)
2868         {
2869             free(buf);
2870             return;
2871         }
2872         sr = newsrchres(&dcnet, wfile, wnick);
2873         if(sr->peernick != NULL)
2874             free(sr->peernick);
2875         sr->peernick = swcsdup(wnick);
2876         sr->size = size;
2877         sr->slots = slots;
2878         free(wfile);
2879         free(wnick);
2880         if(myfn != NULL)
2881             getfnetnode(sr->fn = myfn);
2882         if(hash != NULL)
2883             sr->hash = hash;
2884         submitsrchres(sr);
2885         freesrchres(sr);
2886     }
2887     free(buf);
2888 }
2889
2890 static void hubread(struct socket *sk, struct fnetnode *fn)
2891 {
2892     struct dchub *hub;
2893     char *newbuf;
2894     size_t datalen;
2895     char *p;
2896     
2897     hub = (struct dchub *)fn->data;
2898     if((newbuf = sockgetinbuf(sk, &datalen)) == NULL)
2899         return;
2900     if(hub->inbufdata > 500000) /* Discard possible malicious data */
2901         hub->inbufdata = 0;
2902     sizebuf2(hub->inbuf, hub->inbufdata + datalen, 1);
2903     memcpy(hub->inbuf + hub->inbufdata, newbuf, datalen);
2904     free(newbuf);
2905     p = hub->inbuf + hub->inbufdata;
2906     hub->inbufdata += datalen;
2907     while((datalen > 0) && ((p = memchr(p, '|', datalen)) != NULL))
2908     {
2909         *(p++) = 0;
2910         newqcmd(&hub->queue, hub->inbuf);
2911         memmove(hub->inbuf, p, hub->inbufdata -= p - hub->inbuf);
2912         datalen = hub->inbufdata;
2913         p = hub->inbuf;
2914     }
2915 }
2916
2917 static void huberr(struct socket *sk, int err, struct fnetnode *fn)
2918 {
2919     killfnetnode(fn);
2920 }
2921
2922 static int hubsetnick(struct fnetnode *fn, wchar_t *newnick)
2923 {
2924     struct dchub *hub;
2925     char *buf;
2926     
2927     hub = fn->data;
2928     if((buf = icwcstombs(newnick, (hub == NULL)?DCCHARSET:(hub->charset))) == NULL)
2929         return(1);
2930     if((strchr(buf, ' ') != NULL) || (strchr(buf, '|') != NULL) || (strchr(buf, '$') != NULL))
2931     {
2932         free(buf);
2933         return(1);
2934     }
2935     if(hub == NULL) /* Not yet connected */
2936     {
2937         free(buf);
2938         return(0);
2939     }
2940     if(hub->nativenick != NULL)
2941         free(hub->nativenick);
2942     hub->nativenick = buf;
2943     return(0);
2944 }
2945
2946 static struct dchub *newdchub(struct fnetnode *fn)
2947 {
2948     struct dchub *new;
2949     wchar_t *emu;
2950     wchar_t *wcharset;
2951     char *charset;
2952     
2953     new = smalloc(sizeof(*new));
2954     memset(new, 0, sizeof(*new));
2955     fn->data = new;
2956     if(confgetint("dc", "dcppemu"))
2957         new->dcppemu = 1;
2958     if((emu = wpfind(fn->args, L"dcppemu")) != NULL)
2959     {
2960         if(*emu == L'y')
2961             new->dcppemu = 1;
2962         if(*emu == L'n')
2963             new->dcppemu = 0;
2964     }
2965     charset = NULL;
2966     if((wcharset = wpfind(fn->args, L"charset")) != NULL)
2967     {
2968         if((charset = icwcstombs(wcharset, "US-ASCII")) != NULL)
2969         {
2970             if(!havecharset(charset))
2971             {
2972                 free(charset);
2973                 charset = NULL;
2974             }
2975         }
2976     }
2977     if(charset != NULL)
2978         new->charset = charset;
2979     else
2980         new->charset = sstrdup(DCCHARSET);
2981     if(hubsetnick(fn, fn->mynick))
2982         fnetsetnick(fn, L"DoldaConnectUser-IN");
2983     /* IN as in Invalid Nick */
2984     return(new);
2985 }
2986
2987 static struct dcpeer *newdcpeer(struct socket *sk)
2988 {
2989     struct dcpeer *new;
2990     
2991     new = smalloc(sizeof(*new));
2992     memset(new, 0, sizeof(*new));
2993     new->transfer = NULL;
2994     getsock(sk);
2995     new->sk = sk;
2996     if(confgetint("dc", "dcppemu"))
2997         new->dcppemu = 1;
2998     new->next = peers;
2999     new->prev = NULL;
3000     if(peers != NULL)
3001         peers->prev = new;
3002     peers = new;
3003     numdcpeers++;
3004     return(new);
3005 }
3006
3007 static void freedcpeer(struct dcpeer *peer)
3008 {
3009     int i;
3010     struct qcommand *qcmd;
3011     
3012     peer->freeing = 1;
3013     if(peers == peer)
3014         peers = peer->next;
3015     if(peer->next != NULL)
3016         peer->next->prev = peer->prev;
3017     if(peer->prev != NULL)
3018         peer->prev->next = peer->next;
3019     if(peer->transfer != NULL)
3020     {
3021         if(peer->transfer->dir == TRNSD_UP)
3022             peer->transfer->close = 1;
3023         if(peer->transfer->dir == TRNSD_DOWN)
3024             resettransfer(peer->transfer);
3025         transferdetach(peer->transfer);
3026     }
3027     if(peer->timeout != NULL)
3028         canceltimer(peer->timeout);
3029     if(peer->sk->data == peer)
3030         peer->sk->data = NULL;
3031     peer->sk->readcb = NULL;
3032     peer->sk->writecb = NULL;
3033     peer->sk->errcb = NULL;
3034     putsock(peer->sk);
3035     endcompress(peer);
3036     if(peer->supports != NULL)
3037     {
3038         for(i = 0; peer->supports[i] != NULL; i++)
3039             free(peer->supports[i]);
3040         free(peer->supports);
3041     }
3042     if(peer->inbuf != NULL)
3043         free(peer->inbuf);
3044     if(peer->key != NULL)
3045         free(peer->key);
3046     if(peer->wcsname != NULL)
3047         free(peer->wcsname);
3048     if(peer->nativename != NULL)
3049         free(peer->nativename);
3050     if(peer->charset != NULL)
3051         free(peer->charset);
3052     if(peer->fn != NULL)
3053         putfnetnode(peer->fn);
3054     while((qcmd = ulqcmd(&peer->queue)) != NULL)
3055         freeqcmd(qcmd);
3056     free(peer);
3057     numdcpeers--;
3058 }
3059
3060 static void hubconnect(struct fnetnode *fn)
3061 {
3062     fn->sk->readcb = (void (*)(struct socket *, void *))hubread;
3063     fn->sk->errcb = (void (*)(struct socket *, int, void *))huberr;
3064     getfnetnode(fn);
3065     fn->data = newdchub(fn);
3066     fn->sk->data = fn;
3067     return;
3068 }
3069
3070 static void hubdestroy(struct fnetnode *fn)
3071 {
3072     struct dchub *hub;
3073     struct qcommand *qcmd;
3074     
3075     hub = (struct dchub *)fn->data;
3076     if(fn->sk != NULL)
3077     {
3078         if(fn->sk->data == fn)
3079         {
3080             fn->sk->data = NULL;
3081             putfnetnode(fn);
3082         }
3083     }
3084     if(hub == NULL)
3085         return;
3086     while((qcmd = ulqcmd(&hub->queue)) != NULL)
3087         freeqcmd(qcmd);
3088     if(hub->nativename != NULL)
3089         free(hub->nativename);
3090     if(hub->nativenick != NULL)
3091         free(hub->nativenick);
3092     if(hub->charset != NULL)
3093         free(hub->charset);
3094     if(hub->inbuf != NULL)
3095         free(hub->inbuf);
3096     free(hub);
3097 }
3098
3099 static wchar_t *dcbasename(wchar_t *filename)
3100 {
3101     wchar_t *ret;
3102     
3103     if((ret = wcsrchr(filename, L'\\')) != NULL)
3104         return(ret + 1);
3105     return(filename);
3106 }
3107
3108 static struct transferiface dctransfer =
3109 {
3110     .detach = (void (*)(struct transfer *, void *))dctransdetach,
3111     .gotdata = (void (*)(struct transfer *, void *))dctransgotdata,
3112     .endofdata = (void (*)(struct transfer *, void *))dctransendofdata,
3113     .wantdata = (void (*)(struct transfer *, void *))dcwantdata
3114 };
3115
3116 static struct fnet dcnet =
3117 {
3118     .name = L"dc",
3119     .connect = hubconnect,
3120     .destroy = hubdestroy,
3121     .setnick = hubsetnick,
3122     .reqconn = hubreqconn,
3123     .sendchat = hubsendchat,
3124     .search = hubsearch,
3125     .filebasename = dcbasename
3126 };
3127
3128 static void peerread(struct socket *sk, struct dcpeer *peer)
3129 {
3130     char *newbuf, *p;
3131     size_t datalen;
3132     struct command *cmd;
3133
3134     if((newbuf = sockgetinbuf(sk, &datalen)) == NULL)
3135         return;
3136     sizebuf2(peer->inbuf, peer->inbufdata + datalen, 1);
3137     memcpy(peer->inbuf + peer->inbufdata, newbuf, datalen);
3138     free(newbuf);
3139     peer->inbufdata += datalen;
3140     if(peer->state == PEER_CMD)
3141     {
3142         p = peer->inbuf;
3143         while((peer->inbufdata > 0) && (p = memchr(peer->inbuf, '|', peer->inbufdata)) != NULL)
3144         {
3145             *(p++) = 0;
3146             newqcmd(&peer->queue, peer->inbuf);
3147             for(cmd = peercmds; cmd->handler != NULL; cmd++)
3148             {
3149                 if(!memcmp(peer->inbuf, cmd->name, strlen(cmd->name)) && ((peer->inbuf[strlen(cmd->name)] == ' ') || (peer->inbuf[strlen(cmd->name)] == '|')))
3150                     break;
3151             }
3152             memmove(peer->inbuf, p, peer->inbufdata -= p - peer->inbuf);
3153             if(cmd->stop)
3154             {
3155                 peer->state = PEER_STOP;
3156                 break;
3157             }
3158         }
3159     } else if(peer->state == PEER_TTHL) {
3160         handletthl(peer);
3161     }
3162 }
3163
3164 static void peererror(struct socket *sk, int err, struct dcpeer *peer)
3165 {
3166     freedcpeer(peer);
3167 }
3168
3169 static void peerconnect(struct socket *sk, int err, struct fnetnode *fn)
3170 {
3171     struct dcpeer *peer;
3172     struct dchub *hub;
3173     
3174     if(err != 0)
3175     {
3176         putfnetnode(fn);
3177         return;
3178     }
3179     hub = fn->data;
3180     peer = newdcpeer(sk);
3181     peer->fn = fn;
3182     peer->accepted = 0;
3183     peer->dcppemu = hub->dcppemu;
3184     sk->readcb = (void (*)(struct socket *, void *))peerread;
3185     sk->errcb = (void (*)(struct socket *, int, void *))peererror;
3186     sk->data = peer;
3187     socksettos(sk, confgetint("fnet", "fnptos"));
3188     putsock(sk);
3189     peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
3190     sendmynick(peer);
3191     sendpeerlock(peer);
3192 }
3193
3194 static void peeraccept(struct socket *sk, struct socket *newsk, void *data)
3195 {
3196     struct dcpeer *peer;
3197     
3198     peer = newdcpeer(newsk);
3199     peer->accepted = 1;
3200     newsk->readcb = (void (*)(struct socket *, void *))peerread;
3201     newsk->errcb = (void (*)(struct socket *, int, void *))peererror;
3202     newsk->data = peer;
3203     socksettos(newsk, confgetint("fnet", "fnptos"));
3204     peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
3205 }
3206
3207 static void updatehmlist(void)
3208 {
3209     int i, lev, ic, ret;
3210     struct sharecache *node;
3211     char *buf, *buf2, numbuf[32];
3212     size_t bufsize, bufdata;
3213     int fd, ibuf;
3214     
3215     bufdata = 0;
3216     buf = smalloc(bufsize = 65536);
3217     node = shareroot->child;
3218     lev = 0;
3219     while(1)
3220     {
3221         ic = 0;
3222         /* Use DCCHARSET in $Get paths until further researched... */
3223         if((buf2 = icwcstombs(node->name, DCCHARSET)) != NULL)
3224         {
3225             for(i = 0; i < lev; i++)
3226                 addtobuf(buf, 9);
3227             bufcat(buf, buf2, strlen(buf2));
3228             free(buf2);
3229             if(node->f.b.type == FILE_REG)
3230             {
3231                 addtobuf(buf, '|');
3232                 sprintf(numbuf, "%i", node->size);
3233                 bufcat(buf, numbuf, strlen(numbuf));
3234             }
3235             addtobuf(buf, 13);
3236             addtobuf(buf, 10);
3237         } else {
3238             ic = 1;
3239         }
3240         if((node->child != NULL) && !ic)
3241         {
3242             lev++;
3243             node = node->child;
3244         } else if(node->next != NULL) {
3245             node = node->next;
3246         } else {
3247             while(node->next == NULL)
3248             {
3249                 lev--;
3250                 node = node->parent;
3251                 if(node == shareroot)
3252                     break;
3253             }
3254             if(node == shareroot)
3255                 break;
3256             node = node->next;
3257         }
3258     }
3259     if(hmlistname != NULL)
3260     {
3261         unlink(hmlistname);
3262         free(hmlistname);
3263     }
3264     hmlistname = sstrdup("/tmp/dc-filelist-hm-XXXXXX");
3265     if((fd = mkstemp(hmlistname)) < 0)
3266     {
3267         flog(LOG_WARNING, "could not create HM file list tempfile: %s", strerror(errno));
3268         free(hmlistname);
3269         hmlistname = NULL;
3270     } else {
3271         /*
3272          * I do not want to implement a good Huffman encoder, and it's not
3273          * like Huffman encoding actually yields any impressive results
3274          * for DC file lists anyway, so I'll just output a bogus
3275          * tree. Implement a good encoder if you want to.
3276          */
3277         write(fd, "HE3\r\0", 5);
3278         write(fd, &bufdata, 4);
3279         ibuf = 256;
3280         write(fd, &ibuf, 2);
3281         ibuf = 8;
3282         for(i = 0; i < 256; i++)
3283         {
3284             write(fd, &i, 1);
3285             write(fd, &ibuf, 1);
3286         }
3287         for(i = 0; i < 256; i++)
3288             write(fd, &i, 1);
3289         for(buf2 = buf; bufdata > 0;)
3290         {
3291             if((ret = write(fd, buf2, bufdata)) < 0)
3292             {
3293                 flog(LOG_WARNING, "could not write file list: %s", strerror(errno));
3294                 break;
3295             }
3296             bufdata -= ret;
3297             buf2 += ret;
3298         }
3299         close(fd);
3300     }
3301     free(buf);
3302 }
3303
3304 static struct xmlent
3305 {
3306     wchar_t c;
3307     wchar_t *ent;
3308 } entities[] = {
3309     {L'&', L"&amp;"},
3310     {L'\"', L"&#34;"},
3311 /*  {L'\'', L"&quot;"},  Emulate DC++ escaping by omitting this. */
3312     {L'\0', NULL}
3313 };
3314
3315 static wchar_t *escapexml(wchar_t *src)
3316 {
3317     static wchar_t *buf = NULL;;
3318     int ret, c;
3319     wchar_t nbuf[32];
3320     size_t bufsize, bufdata;
3321     struct xmlent *ent;
3322     
3323     if(buf != NULL)
3324         free(buf);
3325     buf = NULL;
3326     bufsize = bufdata = 0;
3327     for(; *src != L'\0'; src++)
3328     {
3329         c = wctob(*src);
3330         if((c > 0) && (c < 32))
3331         {
3332             bufcat(buf, L"&#", 2);
3333             ret = swprintf(nbuf, (sizeof(nbuf) / sizeof(*nbuf)), L"%i", c);
3334             bufcat(buf, nbuf, ret);
3335             addtobuf(buf, L';');
3336         } else {
3337             for(ent = entities; ent->ent != NULL; ent++)
3338             {
3339                 if(ent->c == *src)
3340                 {
3341                     bufcat(buf, ent->ent, wcslen(ent->ent));
3342                     break;
3343                 }
3344             }
3345             if(ent->ent == NULL)
3346                 addtobuf(buf, *src);
3347         }
3348     }
3349     addtobuf(buf, L'\0');
3350     return(buf);
3351 }
3352
3353 static void updatexmllist(void)
3354 {
3355     int i, fd, lev;
3356     FILE *fs;
3357     char cidbuf[14], *namebuf;
3358     char *hashbuf;
3359     struct sharecache *node;
3360     
3361     if(xmllistname != NULL)
3362     {
3363         unlink(xmllistname);
3364         free(xmllistname);
3365     }
3366     xmllistname = sstrdup("/tmp/dc-filelist-dcxml-XXXXXX");
3367     if((fd = mkstemp(xmllistname)) < 0)
3368     {
3369         flog(LOG_WARNING, "could not create XML file list tempfile: %s", strerror(errno));
3370         free(xmllistname);
3371         xmllistname = NULL;
3372         return;
3373     }
3374     if((fs = fdopen(fd, "w")) == NULL)
3375     {
3376         flog(LOG_WARNING, "could not fdopen XML list fd %i: %s", fd, strerror(errno));
3377         unlink(xmllistname);
3378         free(xmllistname);
3379         xmllistname = NULL;
3380         close(fd);
3381         return;
3382     }
3383     fprintf(fs, "<?xml version=\"1.0\" encoding=\"utf-8\" standalone=\"yes\"?>\r\n");
3384     for(i = 0; i < sizeof(cidbuf) - 1; i++)
3385         cidbuf[i] = (rand() % ('Z' - 'A' + 1)) + 'A';
3386     cidbuf[i] = 0;
3387     if(confgetint("dc", "dcppemu"))
3388         fprintf(fs, "<FileListing Version=\"1\" CID=\"%s\" Base=\"/\" Generator=\"DC++ 0.674\">\r\n", cidbuf);
3389     else
3390         fprintf(fs, "<FileListing Version=\"1\" CID=\"%s\" Base=\"/\" Generator=\"%s\">\r\n", cidbuf, "DoldaConnect" VERSION);
3391     
3392     node = shareroot->child;
3393     lev = 0;
3394     while(1)
3395     {
3396         if((namebuf = icswcstombs(escapexml(node->name), "UTF-8", NULL)) != NULL)
3397         {
3398             for(i = 0; i < lev; i++)
3399                 fputc('\t', fs);
3400             if(node->child != NULL)
3401             {
3402                 fprintf(fs, "<Directory Name=\"%s\">\r\n", namebuf);
3403                 node = node->child;
3404                 lev++;
3405                 continue;
3406             } else {
3407                 fprintf(fs, "<File Name=\"%s\" Size=\"%i\"", namebuf, node->size);
3408                 if(node->f.b.hastth)
3409                 {
3410                     hashbuf = base32encode(node->hashtth, 24);
3411                     fprintf(fs, " TTH=\"%.39s\"", hashbuf);
3412                     free(hashbuf);
3413                 }
3414                 fprintf(fs, "/>\r\n");
3415             }
3416             while(node->next == NULL)
3417             {
3418                 node = node->parent;
3419                 if(node == shareroot)
3420                 {
3421                     break;
3422                 } else {
3423                     lev--;
3424                     for(i = 0; i < lev; i++)
3425                         fputc('\t', fs);
3426                     fprintf(fs, "</Directory>\r\n");
3427                 }
3428             }
3429             if(node == shareroot)
3430                 break;
3431             node = node->next;
3432         }
3433     }
3434     
3435     if(confgetint("dc", "dcppemu"))
3436         fprintf(fs, "</FileListing>");
3437     else
3438         fprintf(fs, "</FileListing>\r\n");
3439     fclose(fs);
3440 }
3441
3442 static void updatexmlbz2list(void)
3443 {
3444     int err, fd;
3445     FILE *in;
3446     FILE *real;
3447     BZFILE *out;
3448     char buf[1024];
3449     size_t bufdata;
3450     
3451     if(xmllistname == NULL)
3452         return;
3453     if((in = fopen(xmllistname, "r")) == NULL)
3454     {
3455         flog(LOG_WARNING, "could not open XML file list for bzipping: %s", strerror(errno));
3456         return;
3457     }
3458     if(xmlbz2listname != NULL)
3459     {
3460         unlink(xmlbz2listname);
3461         free(xmlbz2listname);
3462     }
3463     xmlbz2listname = sstrdup("/tmp/dc-filelist-dcxmlbz2-XXXXXX");
3464     if((fd = mkstemp(xmlbz2listname)) < 0)
3465     {
3466         flog(LOG_WARNING, "could not create bzipped XML file list tempfile: %s", strerror(errno));
3467         free(xmlbz2listname);
3468         xmlbz2listname = NULL;
3469         fclose(in);
3470         return;
3471     }
3472     if((real = fdopen(fd, "w")) == NULL)
3473     {
3474         flog(LOG_WARNING, "could not fdopen bzipped XML list fd %i: %s", fd, strerror(errno));
3475         close(fd);
3476         unlink(xmlbz2listname);
3477         free(xmlbz2listname);
3478         xmlbz2listname = NULL;
3479         fclose(in);
3480         return;
3481     }
3482     out = BZ2_bzWriteOpen(&err, real, 9, 0, 0);
3483     if(err != BZ_OK)
3484     {
3485         flog(LOG_WARNING, "could not open bzip2 stream from XML list");
3486         fclose(real);
3487         unlink(xmlbz2listname);
3488         free(xmlbz2listname);
3489         xmlbz2listname = NULL;
3490         fclose(in);
3491         return;
3492     }
3493     while(!feof(in))
3494     {
3495         bufdata = fread(buf, 1, sizeof(buf), in);
3496         BZ2_bzWrite(&err, out, buf, bufdata);
3497     }
3498     fclose(in);
3499     BZ2_bzWriteClose(&err, out, 0, NULL, NULL);
3500     fclose(real);
3501 }
3502
3503 static int shareupdate(unsigned long long uusharesize, void *data)
3504 {
3505     updatehmlist();
3506     updatexmllist();
3507     updatexmlbz2list();
3508     return(0);
3509 }
3510
3511 static void dispatchcommand(struct qcommand *qcmd, struct command *cmdlist, struct socket *sk, void *data)
3512 {
3513     char *p;
3514     struct command *cmd;
3515     
3516     if((p = strchr(qcmd->string, ' ')) != NULL)
3517         *(p++) = 0;
3518     for(cmd = cmdlist; cmd->handler != NULL; cmd++)
3519     {
3520         if(!strcmp(cmd->name, qcmd->string))
3521             break;
3522     }
3523     if(cmd->handler != NULL)
3524         cmd->handler(sk, data, qcmd->string, p);
3525 /*
3526     else
3527         flog(LOG_DEBUG, "Unimplemented DC command: %s \"%s\"", qcmd->string, p?p:"noargs");
3528 */
3529 }
3530
3531 static int run(void)
3532 {
3533     struct fnetnode *fn, *nextfn;
3534     struct dchub *hub;
3535     struct dcpeer *peer, *nextpeer;
3536     struct qcommand *qcmd;
3537     int ret;
3538     
3539     ret = 0;
3540     for(fn = fnetnodes; fn != NULL; fn = nextfn)
3541     {
3542         nextfn = fn->next;
3543         if(fn->fnet != &dcnet)
3544             continue;
3545         if(fn->data == NULL)
3546             continue;
3547         hub = (struct dchub *)fn->data;
3548         if((qcmd = ulqcmd(&hub->queue)) != NULL)
3549         {
3550             if(*qcmd->string == '$')
3551             {
3552                 if((fn->sk != NULL) && (fn->sk->state == SOCK_EST))
3553                     dispatchcommand(qcmd, hubcmds, fn->sk, fn);
3554             } else if(*qcmd->string != 0) {
3555                 hubrecvchat(fn->sk, fn, NULL, qcmd->string);
3556             }
3557             freeqcmd(qcmd);
3558             ret = 1;
3559             break;
3560         }
3561     }
3562     for(peer = peers; peer != NULL; peer = nextpeer)
3563     {
3564         nextpeer = peer->next;
3565         if((qcmd = ulqcmd(&peer->queue)) != NULL)
3566         {
3567             if(peer->timeout != NULL)
3568                 canceltimer(peer->timeout);
3569             peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
3570             if(*qcmd->string == '$')
3571                 dispatchcommand(qcmd, peercmds, peer->sk, peer);
3572             freeqcmd(qcmd);
3573             ret = 1;
3574             break;
3575         }
3576     }
3577     return(ret);
3578 }
3579
3580 static void preinit(int hup)
3581 {
3582     if(hup)
3583         return;
3584     regfnet(&dcnet);
3585 }
3586
3587 static int updateudpport(struct configvar *var, void *uudata)
3588 {
3589     struct sockaddr_in addr;
3590     struct socket *newsock;
3591     
3592     memset(&addr, 0, sizeof(addr));
3593     addr.sin_family = AF_INET;
3594     addr.sin_port = htons(var->val.num);
3595     if((newsock = netcsdgram((struct sockaddr *)&addr, sizeof(addr))) == NULL)
3596     {
3597         flog(LOG_WARNING, "could not create new DC UDP socket, reverting to old: %s", strerror(errno));
3598         return(0);
3599     }
3600     newsock->readcb = udpread;
3601     if(udpsock != NULL)
3602         putsock(udpsock);
3603     udpsock = newsock;
3604     return(0);
3605 }
3606
3607 static int updatetcpport(struct configvar *var, void *uudata)
3608 {
3609     struct sockaddr_in addr;
3610     struct socket *newsock;
3611     
3612     memset(&addr, 0, sizeof(addr));
3613     addr.sin_family = AF_INET;
3614     addr.sin_port = htons(var->val.num);
3615     if((newsock = netcslisten(SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr), peeraccept, NULL)) == NULL)
3616         flog(LOG_INFO, "could not listen to a remote address, going into passive mode");
3617     if(tcpsock != NULL)
3618         putsock(tcpsock);
3619     tcpsock = newsock;
3620     return(0);
3621 }
3622
3623 static int init(int hup)
3624 {
3625     struct sockaddr_in addr;
3626     
3627     if(!hup)
3628     {
3629         GCBREG(sharechangecb, shareupdate, NULL);
3630         if(udpsock != NULL)
3631             putsock(udpsock);
3632         if(tcpsock != NULL)
3633             putsock(tcpsock);
3634         addr.sin_family = AF_INET;
3635         memset(&addr.sin_addr, 0, sizeof(addr.sin_addr));
3636         addr.sin_port = htons(confgetint("dc", "udpport"));
3637         if((udpsock = netcsdgram((struct sockaddr *)&addr, sizeof(addr))) == NULL)
3638         {
3639             flog(LOG_CRIT, "could not create DC UDP socket: %s", strerror(errno));
3640             return(1);
3641         }
3642         udpsock->readcb = udpread;
3643         addr.sin_port = htons(confgetint("dc", "tcpport"));
3644         if((tcpsock = netcslisten(SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr), peeraccept, NULL)) == NULL)
3645             flog(LOG_INFO, "could not listen to a remote address, going into passive mode");
3646         CBREG(confgetvar("dc", "udpport"), conf_update, updateudpport, NULL, NULL);
3647         CBREG(confgetvar("dc", "tcpport"), conf_update, updatetcpport, NULL, NULL);
3648         CBREG(confgetvar("net", "mode"), conf_update, updatetcpport, NULL, NULL);
3649     }
3650     return(0);
3651 }
3652
3653 static void terminate(void)
3654 {
3655     if(hmlistname != NULL)
3656     {
3657         unlink(hmlistname);
3658         free(hmlistname);
3659     }
3660     if(xmllistname != NULL)
3661     {
3662         unlink(xmllistname);
3663         free(xmllistname);
3664     }
3665     if(xmlbz2listname != NULL)
3666     {
3667         unlink(xmlbz2listname);
3668         free(xmlbz2listname);
3669     }
3670 }
3671
3672 static struct configvar myvars[] =
3673 {
3674     {CONF_VAR_STRING, "desc", {.str = L""}},
3675     {CONF_VAR_STRING, "speedstring", {.str = L"LAN(T1)"}},
3676     {CONF_VAR_STRING, "email", {.str = L"spam@spam.org"}},
3677     {CONF_VAR_INT, "udpport", {.num = 0}},
3678     {CONF_VAR_INT, "tcpport", {.num = 0}},
3679     {CONF_VAR_BOOL, "dcppemu", {.num = 0}},
3680     {CONF_VAR_END}
3681 };
3682
3683 static struct module me =
3684 {
3685     .conf =
3686     {
3687         .vars = myvars
3688     },
3689     .preinit = preinit,
3690     .init = init,
3691     .run = run,
3692     .terminate = terminate,
3693     .name = "dc"
3694 };
3695
3696 MODULE(me)