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