2 * Dolda Connect - Modular multiuser Direct Connect-style client
3 * Copyright (C) 2004 Fredrik Tolf <fredrik@dolda2000.com>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25 #include <netinet/in.h>
26 #include <arpa/inet.h>
43 #include "sysevents.h"
48 * The Direct Connect protocol is extremely ugly. Thus, this code must
49 * also be a bit ugly in certain places. Please forgive me. =/
51 * This also means that there might be some pieces of the code that
52 * look completely illogical to you, and you think that you may make
53 * them much neater/better. However, in many cases it might just be
54 * that it's required to cope with some oddity of the protocol, such
55 * as the amazingly ugly delayed key response in the peer protocol.
58 /* I assume this is the correct character set to use for DC,
59 * considering it was developed without i18n support under Windows */
60 #define DCCHARSET "windows-1252"
62 #ifdef DCPP_MASQUERADE
64 * I honestly don't want to pretend being a client that I'm not, but
65 * there are so many hubs that simply do not accept any clients
66 * outside their whitelists, for no obvious reasons, so I feel that I
67 * am left with little choice. Anyhow, as long as I actually support
68 * all the features that my faked DC++ version does, there should be
69 * very little harm done.
72 #define DCIDTAGV "0.674"
73 #define DCIDFULL "DC++ 0.674"
75 #define DCIDTAG "Dolda"
76 #define DCIDTAGV VERSION
77 #define DCIDFULL "DoldaConnect " VERSION
92 void (*handler)(struct socket *sk, void *data, char *cmd, char *args);
98 struct qcommand *next;
104 struct qcommand *f, *l;
111 size_t inbufdata, inbufsize;
112 struct qcmdqueue queue;
113 int extended, dcppemu;
122 struct dcexppeer *next, *prev;
125 struct timer *expire;
130 struct dcpeer *next, *prev;
134 size_t inbufdata, inbufsize;
135 off_t curread, totalsize;
137 struct timer *timeout;
138 struct qcmdqueue queue;
139 struct transfer *transfer;
141 int ptclose; /* Close after transfer is complete */
142 int accepted; /* If false, we connected, otherwise, we accepted */
143 int extended, dcppemu;
144 int direction; /* Using the constants from transfer.h */
146 int hascurpos, fetchingtthl, notthl;
147 struct tigertreehash tth;
156 static struct fnet dcnet;
157 static struct transferiface dctransfer;
158 static struct socket *udpsock = NULL;
159 static struct socket *tcpsock = NULL;
160 static struct dcpeer *peers = NULL;
162 static struct dcexppeer *expected = NULL;
163 static char *hmlistname = NULL;
164 static char *xmllistname = NULL;
165 static char *xmlbz2listname = NULL;
166 static struct timer *listwritetimer = NULL;
168 static void peerconnect(struct socket *sk, int err, struct fnetnode *fn);
169 static void freedcpeer(struct dcpeer *peer);
170 static void transread(struct socket *sk, struct dcpeer *peer);
171 static void transerr(struct socket *sk, int err, struct dcpeer *peer);
172 static void transwrite(struct socket *sk, struct dcpeer *peer);
173 static void updatehmlist(void);
174 static void updatexmllist(void);
175 static void updatexmlbz2list(void);
176 static void requestfile(struct dcpeer *peer);
177 static void updatelists(int now);
179 static int reservedchar(unsigned char c)
181 return((c == 0) || (c == 5) || (c == 124) || (c == 96) || (c == 126) || (c == 36));
184 /* Oh, how I despise having to do this... */
185 static char *dcmakekey(char *lock)
191 buf = smalloc(strlen(lock));
194 for(i = 0; lock[i]; i++)
196 buf[i] = lock[i] ^ save;
197 buf[i] = ((buf[i] & 0x0F) << 4) | ((buf[i] & 0xF0) >> 4);
199 if((i != 0) && reservedchar(buf[i]))
204 buf[0] ^= buf[i - 1];
205 if(reservedchar(buf[0]))
209 key = smalloc(len + 1);
211 for(i = 0; lock[i] != 0; i++)
213 if(reservedchar(buf[i]))
214 offset += sprintf(key + offset, "/%%DCN%03i%%/", buf[i]);
216 key[offset++] = buf[i];
223 static char *pathnmdc2adc(char *path)
226 size_t retsize, retdata;
228 if(!strcmp(path, "files.xml") || !strcmp(path, "files.xml.bz2") || !strcmp(path, "MyList.DcLst"))
229 return(sstrdup(path));
231 retsize = retdata = 0;
238 addtobuf(ret, *path);
244 static int isdchash(struct hash *hash)
246 if(wcscmp(hash->algo, L"TTH"))
254 * Uncomment when used!
256 static int hubsupports(struct dchub *hub, char *cap)
260 if(hub->supports == NULL)
262 for(p = hub->supports; *p != NULL; p++)
264 if(!strcasecmp(*p, cap))
271 static int supports(struct dcpeer *peer, char *cap)
275 if(peer->supports == NULL)
277 for(p = peer->supports; *p != NULL; p++)
279 if(!strcasecmp(*p, cap))
285 static void endcompress(struct dcpeer *peer)
287 if(peer->compress == CPRS_ZLIB)
289 deflateEnd(peer->cprsdata);
290 free(peer->cprsdata);
292 peer->compress = CPRS_NONE;
295 static void initcompress(struct dcpeer *peer, int algo)
300 peer->compress = algo;
301 if(algo == CPRS_ZLIB)
303 peer->cprsdata = smalloc(sizeof(z_stream));
304 memset(peer->cprsdata, 0, sizeof(z_stream));
305 if((ret = deflateInit(peer->cprsdata, 3)) != Z_OK)
307 flog(LOG_CRIT, "Aiya! zlib refuses to init (%i)!", ret);
313 static void unquote(wchar_t *in)
317 for(p = in; *p != L'\0'; p++)
321 for(p2 = p + 1; (*p2 != L'\0') && (*p2 != L';') && (*p2 != L'&'); p2++);
328 if(!wcscmp(p + 1, L"amp"))
331 } else if(p[1] == L'#') {
332 nc = ucptowc(wcstol(p + 2, NULL, 10));
341 memmove(p + 1, p2 + 1, (wcslen(p2 + 1) + 1) * sizeof(wchar_t));
346 static void freeexppeer(struct dcexppeer *ep)
349 ep->next->prev = ep->prev;
351 ep->prev->next = ep->next;
356 if(ep->expire != NULL)
357 canceltimer(ep->expire);
361 static void exppeerexpire(int cancelled, struct dcexppeer *ep)
368 static struct dcexppeer *expectpeer(char *nick, struct fnetnode *fn)
370 struct dcexppeer *ep;
372 ep = smalloc(sizeof(*ep));
373 ep->nick = sstrdup(nick);
374 getfnetnode(ep->fn = fn);
375 ep->expire = timercallback(ntime() + 300, (void (*)(int, void *))exppeerexpire, ep);
384 static struct qcommand *newqcmd(struct qcmdqueue *queue, char *string)
386 struct qcommand *new;
388 new = smalloc(sizeof(*new));
389 new->string = sstrdup(string);
394 queue->l->next = new;
399 static struct qcommand *ulqcmd(struct qcmdqueue *queue)
401 struct qcommand *qcmd;
403 if((qcmd = queue->f) == NULL)
405 if((queue->f = qcmd->next) == NULL)
410 static void freeqcmd(struct qcommand *qcmd)
416 static void hubrecvchat(struct socket *sk, struct fnetnode *fn, char *from, char *string)
418 wchar_t *chat, *wfrom, *wpeer;
420 struct fnetpeer *peer;
424 end = string + strlen(string);
425 while((p = strchr(string, 13)) != NULL)
426 memmove(p, p + 1, (end-- - p));
429 if((strlen(string) > strlen(from) + 2) && (*string == '<') && !memcmp(string + 1, from, strlen(from)) && (*(string + strlen(from) + 1) == '>'))
430 string += strlen(from) + 2;
431 if((wfrom = icmbstowcs(from, hub->charset)) == NULL)
433 wpeer = swcsdup(wfrom);
439 for(p = string + 1; *p; p++)
441 if((*p == ' ') || (*p == '>'))
449 if((wpeer = icmbstowcs(string + 1, hub->charset)) == NULL)
455 wpeer = swcsdup(L"");
457 if((chat = icmbstowcs(string, hub->charset)) == NULL)
467 if((peer = fnetfindpeer(fn, wfrom)) == NULL) /* Assume public chat */
468 fnethandlechat(fn, 1, wfrom, wpeer, chat);
470 fnethandlechat(fn, 0, wfrom, wpeer, chat);
472 fnethandlechat(fn, 1, L"", wpeer, chat);
480 static void peertimeout(int cancelled, struct dcpeer *peer)
482 peer->timeout = NULL;
488 static void sendadc(struct socket *sk, char *arg)
491 size_t bufsize, bufdata;
494 bufsize = bufdata = 0;
500 bufcat(buf, "\\s", 2);
501 } else if(*arg == '\n') {
502 bufcat(buf, "\\n", 2);
503 } else if(*arg == '\\') {
504 bufcat(buf, "\\\\", 2);
509 sockqueue(sk, buf, bufdata);
513 #if defined(__GNUC__)
514 static void __attribute__ ((format (printf, 2, 3))) sendadcf(struct socket *sk, char *arg, ...)
516 static void sendadcf(struct socket *sk, char *arg, ...)
523 buf = vsprintf2(arg, args);
531 static char **parseadc(char *args)
534 size_t retbufsize, retbufdata;
536 size_t bufsize, bufdata;
541 retbufsize = retbufdata = bufsize = bufdata = 0;
550 else if(*args != ' ')
554 if((*args == ' ') || (*args == 0))
557 addtobuf(retbuf, buf);
559 bufsize = bufdata = 0;
562 } else if(*args == '\\') {
565 addtobuf(buf, *args);
574 addtobuf(retbuf, NULL);
577 } else if((*args == 's') || (*args == ' ')) {
579 } else if(*args == 'n') {
581 } else if(*args == '\\') {
591 addtobuf(retbuf, NULL);
595 /* Macros useful in command handlers */
596 #define skipspace(s) ({if(((s) = strchr((s), ' ')) == NULL) return; else (s)++;})
597 #define qstr(sk, str) sockqueue(sk, str, strlen(str))
598 #define qstrf(sk, strandargs...) \
601 if((__buf__ = sprintf2(strandargs)) != NULL) { \
602 sockqueue(sk, __buf__, strlen(__buf__)); \
607 static char *tr(char *str, char *trans)
611 for(; *trans; trans += 2)
613 for(p = strchr(str, trans[0]); p != NULL; p = strchr(p, trans[0]))
619 static char *getadcid(struct dcpeer *peer)
626 if(!wcscmp(peer->transfer->path, L"files.xml") || !wcscmp(peer->transfer->path, L"files.xml.bz2") || !wcscmp(peer->transfer->path, L"MyList.DcLst"))
628 if(!isfilelist && (peer->transfer->hash != NULL) && isdchash(peer->transfer->hash) && supports(peer, "tthf"))
630 buf = base32encode(peer->transfer->hash->buf, 24);
631 ret = sprintf2("TTH/%.39s", buf);
634 if((buf = icwcstombs(peer->transfer->path, "UTF-8")) == NULL)
636 ret = pathnmdc2adc(buf);
643 static int trresumecb(struct transfer *transfer, wchar_t *cmd, wchar_t *arg, struct dcpeer *peer)
645 if(!wcscmp(cmd, L"resume"))
649 flog(LOG_WARNING, "filter returned no position for \"resume\" on transfer %i", transfer->id);
652 transfer->curpos = wcstoll(arg, NULL, 10);
661 static void sendmynick(struct dcpeer *peer)
667 qstrf(peer->sk, "$MyNick %s|", icswcstombs(confgetstr("cli", "defnick"), peer->charset, "DoldaConnectUser-IN"));
669 qstrf(peer->sk, "$MyNick %s|", icswcstombs(fn->mynick, peer->charset, "DoldaConnectUser-IN"));
672 static void sendpeerlock(struct dcpeer *peer)
675 qstrf(peer->sk, "$Lock EXTENDEDPROTOCOLABCABCABCABCABCABC Pk=DCPLUSPLUS0.674ABCABC|");
677 qstrf(peer->sk, "$Lock EXTENDEDPROTOCOLABCABCABCABCABCABC Pk=DOLDA%sABCABCABC|", VERSION);
680 static void sendsupports(struct dcpeer *peer)
683 qstr(peer->sk, "$Supports MiniSlots XmlBZList ADCGet TTHL TTHF GetZBlock ZLIG |");
685 qstr(peer->sk, "$Supports MiniSlots XmlBZList ADCGet TTHL TTHF");
686 if(!confgetint("dc", "hidedeflate"))
687 qstr(peer->sk, " GetZBlock ZLIG");
692 static void requestfile(struct dcpeer *peer)
696 if(peer->transfer->size == -1)
698 /* Use DCCHARSET for $Get paths until further researched... */
699 if((buf = icswcstombs(peer->transfer->path, DCCHARSET, NULL)) == NULL)
701 transferseterror(peer->transfer, TRNSE_NOTFOUND);
705 /* The transfer will be restarted later from
706 * cmd_filelength when it detects that the sizes
708 qstrf(peer->sk, "$Get %s$1|", buf);
711 if((peer->transfer->hash == NULL) && !peer->notthl)
713 if(supports(peer, "adcget") && supports(peer, "tthl"))
715 qstr(peer->sk, "$ADCGET");
716 sendadc(peer->sk, "tthl");
717 if((buf = getadcid(peer)) == NULL)
719 transferseterror(peer->transfer, TRNSE_NOTFOUND);
723 sendadc(peer->sk, buf);
725 sendadc(peer->sk, "0");
726 sendadc(peer->sk, "-1");
728 peer->fetchingtthl = 1;
734 if(forkfilter(peer->transfer))
736 flog(LOG_WARNING, "could not fork filter for transfer %i: %s", peer->transfer->id, strerror(errno));
740 CBREG(peer->transfer, trans_filterout, (int (*)(struct transfer *, wchar_t *, wchar_t *, void *))trresumecb, NULL, peer);
743 if(supports(peer, "adcget"))
745 qstr(peer->sk, "$ADCGET");
746 sendadc(peer->sk, "file");
747 if((buf = getadcid(peer)) == NULL)
749 transferseterror(peer->transfer, TRNSE_NOTFOUND);
753 sendadc(peer->sk, buf);
755 sendadcf(peer->sk, "%ji", (intmax_t)peer->transfer->curpos);
756 sendadcf(peer->sk, "%ji", (intmax_t)(peer->transfer->size - peer->transfer->curpos));
758 } else if(supports(peer, "xmlbzlist")) {
759 if((buf = icswcstombs(peer->transfer->path, "UTF-8", NULL)) == NULL)
761 transferseterror(peer->transfer, TRNSE_NOTFOUND);
765 qstrf(peer->sk, "$UGetBlock %ji %ji %s|", (intmax_t)peer->transfer->curpos, (intmax_t)(peer->transfer->size - peer->transfer->curpos), buf);
767 /* Use DCCHARSET for $Get paths until further researched... */
768 if((buf = icswcstombs(peer->transfer->path, DCCHARSET, NULL)) == NULL)
770 transferseterror(peer->transfer, TRNSE_NOTFOUND);
774 qstrf(peer->sk, "$Get %s$%ji|", buf, (intmax_t)peer->transfer->curpos + 1);
778 static void sendmyinfo(struct socket *sk, struct fnetnode *fn)
782 struct fnetnode *cfn;
786 qstrf(sk, "$MyINFO $ALL %s ", hub->nativenick);
787 buf = tr(icswcstombs(confgetstr("dc", "desc"), hub->charset, "Charset_conv_failure"), "$_|_");
788 qstrf(sk, "%s", buf);
790 for(cfn = fnetnodes; cfn != NULL; cfn = cfn->next)
792 if((cfn->state == FNN_EST) || (cfn->state == FNN_HS))
794 if(cfn->regstatus == FNNS_OP)
796 else if(cfn->regstatus == FNNS_REG)
802 qstrf(sk, "<%s V:%s,M:%c,H:%i/%i/%i,S:%i>",
803 (hub->dcppemu)?"++":"Dolda",
804 (hub->dcppemu)?"0.674":VERSION,
805 (tcpsock == NULL)?'P':'A',
807 confgetint("transfer", "slots")
810 buf = tr(icswcstombs(confgetstr("dc", "speedstring"), hub->charset, "Charset_conv_failure"), "$_|_");
811 qstrf(sk, "%s\x01$", buf);
812 buf = tr(icswcstombs(confgetstr("dc", "email"), hub->charset, "Charset_conv_failure"), "$_|_");
813 qstrf(sk, "%s$", buf);
814 qstrf(sk, "%llu$|", sharesize);
817 static void hubhandleaction(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
822 if(!strcmp(cmd, "$Lock"))
824 qstrf(sk, "$ValidateNick %s|", hub->nativenick);
825 } else if(!strcmp(cmd, "$Hello")) {
826 if(fn->state == FNN_HS)
828 qstrf(sk, "$Version 1,0091|");
829 qstrf(sk, "$GetNickList|");
831 fnetsetstate(fn, FNN_EST);
833 qstrf(sk, "$GetINFO %s %s|", args, hub->nativenick);
838 static void cmd_lock(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
845 if(!strncmp(args, "EXTENDEDPROTOCOL", 16))
847 if((p = strchr(args, ' ')) != NULL)
852 qstrf(sk, "$Supports UserCommand NoGetINFO NoHello UserIP2 TTHSearch GetZBlock |");
854 qstrf(sk, "$Supports UserCommand NoGetINFO NoHello UserIP2 TTHSearch");
855 if(!confgetint("dc", "hidedeflate"))
856 qstr(sk, " GetZBlock");
860 key = dcmakekey(args);
861 qstrf(sk, "$Key %s|", key);
863 hubhandleaction(sk, fn, cmd, args);
866 static void cmd_hubname(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
872 if(hub->nativename == NULL)
873 free(hub->nativename);
874 hub->nativename = sstrdup(args);
875 buf = icmbstowcs(args, hub->charset);
876 fnetsetname(fn, (buf == NULL)?L"Hubname conv error":buf);
879 hubhandleaction(sk, fn, cmd, args);
882 static void cmd_hello(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
888 if((nick = icmbstowcs(args, hub->charset)) == NULL)
890 if(strcmp(args, hub->nativenick) && (fnetfindpeer(fn, nick) == NULL))
891 fnetaddpeer(fn, nick, nick);
893 hubhandleaction(sk, fn, cmd, args);
896 static void cmd_quit(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
899 struct fnetpeer *peer;
903 if((nick = icmbstowcs(args, hub->charset)) == NULL)
905 if((peer = fnetfindpeer(fn, nick)) != NULL)
908 hubhandleaction(sk, fn, cmd, args);
911 static void cmd_nicklist(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
916 struct fnetpeer *peer;
919 for(peer = btreeiter(fn->peers); peer != NULL; peer = btreeiter(NULL))
920 peer->flags.b.delete = 1;
921 while((p = strstr(args, "$$")) != NULL)
924 if((buf = icmbstowcs(args, hub->charset)) != NULL)
926 if((peer = fnetfindpeer(fn, buf)) == NULL)
927 peer = fnetaddpeer(fn, buf, buf);
929 peer->flags.b.delete = 0;
931 qstrf(sk, "$GetINFO %s %s|", args, hub->nativenick);
936 hubhandleaction(sk, fn, cmd, args);
939 static void cmd_oplist(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
944 struct fnetpeer *peer;
947 for(peer = btreeiter(fn->peers); peer != NULL; peer = btreeiter(NULL))
948 peer->flags.b.op = 0;
949 while((p = strstr(args, "$$")) != NULL)
952 if((buf = icmbstowcs(args, hub->charset)) != NULL)
954 if((peer = fnetfindpeer(fn, buf)) != NULL)
955 peer->flags.b.op = 1;
960 hubhandleaction(sk, fn, cmd, args);
963 static void cmd_myinfo(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
966 wchar_t *buf, *wp, *wp2;
968 struct fnetpeer *peer;
973 if(strncmp(p, "$ALL ", 5))
976 if((p2 = strchr(p, ' ')) == NULL)
979 if((buf = icmbstowcs(p, hub->charset)) == NULL)
981 if((peer = fnetfindpeer(fn, buf)) == NULL)
982 peer = fnetaddpeer(fn, buf, buf);
985 if((p2 = strstr(p, "$ $")) == NULL)
988 if((buf = icmbstowcs(p, hub->charset)) == NULL)
990 if((wcslen(buf) > 0) && (buf[wcslen(buf) - 1] == L'>') && ((wp = wcsrchr(buf, L'<')) != NULL))
992 buf[wcslen(buf) - 1] = L'\0';
994 if((wp2 = wcschr(wp, L' ')) != NULL)
997 fnetpeersetstr(peer, L"dc-client", wp);
1001 if((wp2 = wcschr(wp, L',')) != NULL)
1005 swprintf(abuf, 10, L"dc-tag-%lc", wp[0]);
1006 fnetpeersetstr(peer, abuf, wp + 2);
1008 } while(wp2 != NULL);
1011 fnetpeersetstr(peer, L"descr", buf);
1014 if((p2 = strchr(p, '$')) == NULL)
1017 if((buf = icmbstowcs(p, hub->charset)) == NULL)
1019 fnetpeersetstr(peer, L"dc-speed", buf);
1022 if((p2 = strchr(p, '$')) == NULL)
1025 if((buf = icmbstowcs(p, hub->charset)) == NULL)
1027 fnetpeersetstr(peer, L"email", buf);
1032 fnetpeersetlnum(peer, L"share", strtoll(p, NULL, 10));
1033 hubhandleaction(sk, fn, cmd, args);
1036 /* I do not implement the fully in that I do not disconnect from the
1037 * old hub. I do this since I believe that if the hub owner really
1038 * wants to get rid of us, then it is his responsibility to disconnect
1040 static void cmd_forcemove(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1043 struct fnetnode *newfn;
1047 if(strchr(args, ':') == NULL)
1049 args = strcpy(smalloc(strlen(args) + 5), args);
1050 strcat(args, ":411");
1055 if((newfn = fnetinitconnect(L"dc", fn->owner, args, NULL)) != NULL)
1057 linkfnetnode(newfn);
1060 hubhandleaction(sk, fn, cmd, args);
1065 static char *getdcpath(struct sharecache *node, size_t *retlen, char *charset)
1070 if(node->parent == NULL)
1072 if(node->parent == shareroot)
1074 if((buf = icwcstombs(node->name, charset)) == NULL)
1077 *retlen = strlen(buf);
1080 if((buf2 = icwcstombs(node->name, charset)) == NULL)
1082 if((buf = getdcpath(node->parent, &len, charset)) == NULL)
1087 len2 = strlen(buf2);
1088 buf = srealloc(buf, len + 1 + len2 + 1);
1090 strcpy(buf + len, buf2);
1091 buf[len + len2] = 0;
1094 *retlen = len + len2;
1100 * This is the main share searching function for Direct Connect
1101 * peers. Feel free to optimize it if you feel the need for it. I
1102 * haven't ever seen it take any noticable CPU time anyway, so I
1103 * haven't felt that need.
1105 * It may feel dubious to just return the directory when all terms are
1106 * satisfied rather than all the files in it, but it really benefits
1107 * everyone: Less cycles used for us, less UDP squelching for active
1108 * searchers, and less bandwidth waste for hubs when serving passive
1111 static void cmd_search(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1116 char *prefix, *infix, *postfix, *buf, *buf2;
1118 struct sockaddr_in addr;
1119 struct sharecache *node;
1120 int minsize, maxsize;
1123 int termnum, satisfied, skipcheck;
1124 int level, tersat[32];
1129 if((p = strchr(args, ' ')) == NULL)
1133 memset(terms, 0, sizeof(terms));
1134 prefix = infix = postfix = NULL;
1138 if(!strncmp(args, "Hub:", 4))
1140 if(!strcmp(cmd, "$MultiSearch"))
1142 if(!strcmp(args + 4, hub->nativenick))
1144 prefix = sprintf2("$SR %s ", hub->nativenick);
1145 infix = sprintf2(" %i/%i\005", slotsleft(), confgetint("transfer", "slots"));
1146 postfix = sprintf2(" (%s)\005%s|", formataddress(hub->sk->remote, hub->sk->remotelen), args + 4);
1150 if((p2 = strchr(args, ':')) == NULL)
1153 addr.sin_family = AF_INET;
1154 if(!inet_aton(args, &addr.sin_addr))
1156 addr.sin_port = htons(atoi(p2));
1157 prefix = sprintf2("$SR %s ", hub->nativenick);
1158 infix = sprintf2(" %i/%i\005", slotsleft(), confgetint("transfer", "slots"));
1159 postfix = sprintf2(" (%s)|", formataddress(hub->sk->remote, hub->sk->remotelen));
1160 netdgramconn(dsk = netdupsock(udpsock), (struct sockaddr *)&addr, sizeof(addr));
1163 minsize = maxsize = -1;
1172 if((*(p++) == 'T') && (minsize == 0))
1179 if((p2 = strchr(p, '?')) == NULL)
1194 if((*p2 == 0) || (*p2 == '$'))
1202 if(!dotth && !strncmp(p, "TTH:", 4))
1205 if(((buf = base32decode(p + 4, &buflen)) == NULL) || (buflen != 24))
1210 memcpy(hashtth, buf, 24);
1213 if((terms[termnum] = icmbstowcs(p, hub->charset)) != NULL)
1224 node = shareroot->child;
1226 for(i = 0; i < termnum; i++)
1232 if(node->f.b.type == FILE_REG)
1234 if((minsize >= 0) && (node->size < minsize))
1236 if((maxsize >= 0) && (node->size > maxsize))
1239 if(!skipcheck && dotth)
1241 if((node->f.b.type != FILE_REG) || (node->f.b.hastth && memcmp(hashtth, node->hashtth, 24)))
1246 for(i = 0; i < termnum; i++)
1250 if(wcsexists(node->name, terms[i]))
1254 } else if(node->child == NULL) {
1259 if(!skipcheck && (satisfied == termnum))
1261 /* Use DCCHARSET in $Get paths until further researched... */
1262 if((buf = getdcpath(node, NULL, DCCHARSET)) != NULL)
1264 if(node->f.b.hastth)
1266 buf2 = base32encode(node->hashtth, 24);
1267 qstrf(dsk, "%s%s\005%ji%sTTH:%.39s%s", prefix, buf, (intmax_t)node->size, infix, buf2, postfix);
1270 qstrf(dsk, "%s%s\005%ji%s%s%s", prefix, buf, (intmax_t)node->size, infix, hub->nativename, postfix);
1275 if((!skipcheck && (satisfied == termnum)) || (node->child == NULL))
1277 while(node->next == NULL)
1279 if((node = node->parent) == shareroot)
1283 if(node == shareroot)
1285 for(i = 0; i < termnum; i++)
1287 if(tersat[i] >= level)
1300 hubhandleaction(sk, fn, cmd, args);
1311 for(i = 0; (i < 32) && (terms[i] != NULL); i++)
1315 static void cmd_connecttome(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1319 struct socket *newsk;
1320 struct sockaddr_in addr;
1323 if((p = strchr(args, ' ')) == NULL)
1326 if(strcmp(args, hub->nativenick))
1328 addr.sin_family = AF_INET;
1330 if((p = strchr(args, ':')) == NULL)
1333 addr.sin_port = htons(atoi(p));
1334 if(!inet_aton(args, &addr.sin_addr))
1336 newsk = netcsconn((struct sockaddr *)&addr, sizeof(addr), (void (*)(struct socket *, int, void *))peerconnect, fn);
1338 hubhandleaction(sk, fn, cmd, args);
1341 static void sendctm(struct socket *sk, char *nick)
1343 struct sockaddr *addr;
1348 if(sockgetremotename2(tcpsock, sk, &addr, &addrlen) < 0)
1350 if(addr->sa_family == AF_INET)
1351 qstrf(sk, "$ConnectToMe %s %s|", nick, formataddress(addr, addrlen));
1353 flog(LOG_WARNING, "Direct Connect TCP socket is suddenly not AF_INET, but %i", addr->sa_family);
1357 static void cmd_revconnecttome(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1363 if((p = strchr(args, ' ')) == NULL)
1366 if(strcmp(p, hub->nativenick))
1369 expectpeer(args, fn);
1370 hubhandleaction(sk, fn, cmd, args);
1373 static void cmd_getnetinfo(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1376 struct fnetnode *node;
1381 for(node = fnetnodes; node != NULL; node = node->next)
1383 if(node->state == FNN_EST)
1386 qstrf(sk, "$NetInfo %i$%i$%c$0|", confgetint("transfer", "slots"), numhubs, (tcpsock == NULL)?'P':'A');
1387 hubhandleaction(sk, fn, cmd, args);
1390 static void cmd_to(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1397 if((p2 = strchr(p, ' ')) == NULL)
1401 if((p2 = strchr(p, ' ')) == NULL)
1404 if(strcmp(p, "From:"))
1407 if((p2 = strstr(p, " $")) == NULL)
1411 hubrecvchat(hub->sk, fn, p, p2);
1412 hubhandleaction(sk, fn, cmd, args);
1415 static void cmd_sr(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1419 char *nick, *filename, *hubname;
1424 wchar_t *wnick, *wfile;
1428 if((p2 = strchr(p, ' ')) == NULL)
1433 if((p2 = strchr(p, 5)) == NULL)
1437 if((p2 = strchr(p, ' ')) == NULL)
1440 size = strtoll(p, NULL, 10);
1442 if((p2 = strchr(p, '/')) == NULL)
1447 if((p2 = strchr(p, 5)) == NULL)
1451 if((p2 = strstr(p, " (")) == NULL)
1454 if((wnick = icmbstowcs(nick, hub->charset)) == NULL)
1456 /* Use DCCHARSET in $Get paths until further researched... */
1457 if((wfile = icmbstowcs(filename, DCCHARSET)) == NULL)
1462 sr = newsrchres(&dcnet, wfile, wnick);
1463 if(sr->peernick != NULL)
1465 sr->peernick = swcsdup(wnick);
1470 if(!strncmp(hubname, "TTH:", 4))
1472 if((buf = base32decode(hubname + 4, &buflen)) != NULL)
1475 sr->hash = newhash(L"TTH", 24, buf);
1479 getfnetnode(sr->fn = fn);
1482 hubhandleaction(sk, fn, cmd, args);
1485 static void cmd_usercommand(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1487 /* Do nothing for now. */
1490 static void cmd_getpass(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1497 pw = wpfind(fn->args, L"password");
1498 if((pw == NULL) || ((mbspw = icwcstombs(pw, hub->charset)) == NULL))
1503 qstrf(sk, "$MyPass %s|", mbspw);
1505 fn->regstatus = FNNS_REG;
1506 hubhandleaction(sk, fn, cmd, args);
1509 static void cmd_logedin(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1514 fn->regstatus = FNNS_OP;
1515 hubhandleaction(sk, fn, cmd, args);
1518 static void cmd_hubsupports(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
1524 size_t arrsize, arrdata;
1527 if(hub->supports != NULL)
1529 for(i = 0; hub->supports[i] != NULL; i++)
1530 free(hub->supports[i]);
1531 free(hub->supports);
1534 arrsize = arrdata = 0;
1538 if((p2 = strchr(p, ' ')) != NULL)
1542 addtobuf(arr, sstrdup(p));
1543 } while((p = p2) != NULL);
1544 addtobuf(arr, NULL);
1545 hub->supports = arr;
1548 static void cmd_mynick(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1550 struct dcexppeer *expect;
1553 if(peer->nativename != NULL)
1554 free(peer->nativename);
1555 peer->nativename = sstrdup(args);
1556 if(peer->wcsname != NULL)
1557 free(peer->wcsname);
1558 if((peer->wcsname = icmbstowcs(peer->nativename, peer->charset)) == NULL)
1565 for(expect = expected; expect != NULL; expect = expect->next)
1567 if(!strcmp(expect->nick, args))
1574 hub = expect->fn->data;
1575 peer->fn = expect->fn;
1576 getfnetnode(peer->fn);
1577 peer->dcppemu = hub->dcppemu;
1578 freeexppeer(expect);
1583 static void cmd_direction(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1587 struct transfer *transfer;
1589 if((p = strchr(args, ' ')) == NULL)
1592 if(!strcmp(args, "Upload"))
1594 if(!strcmp(args, "Download"))
1598 if((peer->transfer == NULL) || (mydir != peer->direction))
1603 if(peer->direction == TRNSD_DOWN)
1606 if(peer->wcsname == NULL)
1611 peer->direction = mydir;
1612 if(peer->direction == TRNSD_UP)
1614 if(confgetint("transfer", "ulquota") && hasupload(&dcnet, peer->wcsname))
1619 transfer = newupload(peer->fn, &dcnet, peer->wcsname, &dctransfer, peer);
1621 if((transfer = finddownload(peer->wcsname)) == NULL)
1626 transferattach(transfer, &dctransfer, peer);
1627 transfersetstate(transfer, TRNS_HS);
1629 transfersetnick(transfer, peer->wcsname);
1630 peer->transfer = transfer;
1633 qstrf(sk, "$Direction %s %i|", (peer->direction == TRNSD_UP)?"Upload":"Download", rand() % 10000);
1634 if(peer->key != NULL)
1635 qstrf(sk, "$Key %s|", peer->key);
1636 if(peer->direction == TRNSD_DOWN)
1641 static void cmd_peerlock(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1644 struct transfer *transfer;
1646 if((p = strchr(args, ' ')) == NULL)
1649 if(!strncmp(args, "EXTENDEDPROTOCOL", 16))
1651 key = dcmakekey(args);
1654 if(peer->wcsname == NULL)
1663 if((transfer = finddownload(peer->wcsname)) == NULL)
1665 if(confgetint("transfer", "ulquota") && hasupload(&dcnet, peer->wcsname))
1670 peer->direction = TRNSD_UP;
1671 transfer = newupload(peer->fn, &dcnet, peer->wcsname, &dctransfer, peer);
1673 peer->direction = TRNSD_DOWN;
1674 transferattach(transfer, &dctransfer, peer);
1675 transfersetstate(transfer, TRNS_HS);
1677 transfersetnick(transfer, peer->wcsname);
1678 peer->transfer = transfer;
1679 qstrf(sk, "$Direction %s %i|", (peer->direction == TRNSD_UP)?"Upload":"Download", rand() % 10000);
1680 qstrf(sk, "$Key %s|", key);
1683 if(peer->key != NULL)
1689 static void cmd_key(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1694 static void startdl(struct dcpeer *peer)
1696 if(peer->timeout != NULL)
1697 canceltimer(peer->timeout);
1698 peer->state = PEER_TRNS;
1699 transferstartdl(peer->transfer, peer->sk);
1700 peer->sk->readcb = (void (*)(struct socket *, void *))transread;
1701 peer->sk->errcb = (void (*)(struct socket *, int, void *))transerr;
1704 static void startul(struct dcpeer *peer)
1706 if(peer->timeout != NULL)
1707 canceltimer(peer->timeout);
1708 peer->state = PEER_TRNS;
1709 transferstartul(peer->transfer, peer->sk);
1710 peer->sk->writecb = (void (*)(struct socket *, void *))transwrite;
1713 static void cmd_filelength(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1716 struct transfer *transfer;
1718 if(peer->transfer == NULL)
1723 size = strtoll(args, NULL, 10);
1724 if(peer->transfer->size != size)
1726 transfersetsize(peer->transfer, size);
1727 transfer = peer->transfer;
1729 trytransferbypeer(transfer->fnet, transfer->peerid);
1733 qstr(peer->sk, "$Send|");
1736 static void cmd_error(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1738 if(peer->fetchingtthl)
1740 peer->fetchingtthl = 0;
1745 if((peer->transfer != NULL) && (peer->transfer->dir == TRNSD_DOWN))
1747 transferseterror(peer->transfer, TRNSE_NOTFOUND);
1748 resettransfer(peer->transfer);
1754 static void cmd_maxedout(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1756 if((peer->transfer != NULL) && (peer->transfer->dir == TRNSD_DOWN))
1758 transferseterror(peer->transfer, TRNSE_NOSLOTS);
1759 resettransfer(peer->transfer);
1772 {"MyList.DcLst", &hmlistname, updatehmlist},
1773 {"files.xml", &xmllistname, updatexmllist},
1774 {"files.xml.bz2", &xmlbz2listname, updatexmlbz2list},
1778 static int openfilelist(char *name)
1783 for(i = 0; lists[i].name != NULL; i++)
1785 if(!strcmp(name, lists[i].name))
1789 if(lists[i].name == NULL)
1792 if((*lists[i].file == NULL) || ((fd = open(*lists[i].file, O_RDONLY)) < 0))
1794 if((fd < 0) && ((*lists[i].file == NULL) || ((fd = open(*lists[i].file, O_RDONLY)) < 0)))
1797 flog(LOG_ERR, "could not open filelist tempfile: %s", strerror(errno));
1804 static struct sharecache *findbytth(char *tth32)
1808 struct sharecache *node;
1810 if((buf = base32decode(tth32, &buflen)) == NULL)
1817 for(node = shareroot->child; node != NULL; node = nextscnode(node))
1819 if(node->f.b.hastth && !memcmp(node->hashtth, buf, 24))
1826 static struct sharecache *resdcpath(char *path, char *charset, char sep)
1828 struct sharecache *node;
1835 if((p2 = strchr(p, sep)) != NULL)
1837 if((node = findcache(node, icsmbstowcs(p, charset, L""))) == NULL)
1844 static void cmd_get(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1849 struct sharecache *node;
1850 struct socket *lesk;
1854 if(peer->transfer == NULL)
1859 if((p = strchr(args, '$')) == NULL)
1865 if((offset = (strtoll(p, NULL, 10) - 1)) < 0)
1870 if(((fd = openfilelist(args)) < 0) && (errno != 0))
1872 qstr(sk, "$Error Could not send file list|");
1875 } else if(fd >= 0) {
1876 if((buf2 = icsmbstowcs(args, DCCHARSET, NULL)) != NULL)
1877 transfersetpath(peer->transfer, buf2);
1878 peer->transfer->flags.b.minislot = 1;
1882 /* Use DCCHARSET in $Get paths until further researched... */
1883 if((node = resdcpath(args, DCCHARSET, '\\')) == NULL)
1885 qstrf(sk, "$Error File not in share|");
1889 if((fd = opensharecache(node)) < 0)
1891 qstrf(sk, "$Error %s|", strerror(errno));
1895 buf = getfspath(node);
1896 if((buf2 = icsmbstowcs(buf, NULL, NULL)) == NULL)
1897 flog(LOG_WARNING, "could not convert native path into a wcs (%s): %s", buf, strerror(errno));
1899 transfersetpath(peer->transfer, buf2);
1902 if(fstat(fd, &sb) < 0)
1905 flog(LOG_WARNING, "could not stat file %ls: %s", node->name, strerror(errno));
1906 qstrf(sk, "$Error|");
1910 if(sb.st_size < 65536)
1911 peer->transfer->flags.b.minislot = 1;
1912 if(!peer->transfer->flags.b.minislot && (slotsleft() < 1)) {
1914 qstr(sk, "$MaxedOut|");
1918 if((offset != 0) && (lseek(fd, offset, SEEK_SET) < 0))
1921 qstrf(sk, "$Error Offset out of range|");
1925 lesk = wrapsock(fd);
1926 transferprepul(peer->transfer, sb.st_size, offset, -1, lesk);
1928 qstrf(sk, "$FileLength %ji|", (intmax_t)peer->transfer->size);
1931 static void cmd_send(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1933 if(peer->transfer == NULL)
1938 if(peer->transfer->localend == NULL)
1947 static void cmd_supports(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1952 size_t arrsize, arrdata;
1954 if(peer->supports != NULL)
1956 for(i = 0; peer->supports[i] != NULL; i++)
1957 free(peer->supports[i]);
1958 free(peer->supports);
1961 arrsize = arrdata = 0;
1965 if((p2 = strchr(p, ' ')) != NULL)
1969 addtobuf(arr, sstrdup(p));
1970 } while((p = p2) != NULL);
1971 addtobuf(arr, NULL);
1972 peer->supports = arr;
1975 static void cmd_getblock(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
1979 off_t start, numbytes;
1980 char *charset, *buf;
1982 struct sharecache *node;
1984 struct socket *lesk;
1986 if(peer->transfer == NULL)
1992 if((p2 = strchr(p, ' ')) == NULL)
1998 start = strtoll(p, NULL, 10);
2000 if((p2 = strchr(p, ' ')) == NULL)
2006 numbytes = strtoll(p, NULL, 10);
2008 if(!strcmp(cmd, "$UGetBlock") || !strcmp(cmd, "$UGetZBlock"))
2011 /* Use DCCHARSET in $Get paths until further researched... */
2012 charset = DCCHARSET;
2013 if(!strcmp(cmd, "$GetZBlock") || !strcmp(cmd, "$UGetZBlock"))
2014 initcompress(peer, CPRS_ZLIB);
2015 if(((fd = openfilelist(p)) < 0) && (errno != 0))
2017 qstr(sk, "$Error Could not send file list|");
2019 } else if(fd >= 0) {
2020 if((buf2 = icsmbstowcs(args, charset, NULL)) != NULL)
2021 transfersetpath(peer->transfer, buf2);
2022 peer->transfer->flags.b.minislot = 1;
2026 if((node = resdcpath(p, charset, '\\')) == NULL)
2028 qstr(sk, "$Error File not in cache|");
2031 if((fd = opensharecache(node)) < 0)
2033 qstrf(sk, "$Error %s|", strerror(errno));
2036 buf = getfspath(node);
2037 if((buf2 = icsmbstowcs(buf, NULL, NULL)) == NULL)
2038 flog(LOG_WARNING, "could not convert native path into a wcs (%s): %s", buf, strerror(errno));
2040 transfersetpath(peer->transfer, buf2);
2043 if(fstat(fd, &sb) < 0)
2046 flog(LOG_WARNING, "could not stat file %ls: %s", node->name, strerror(errno));
2047 qstr(sk, "$Error|");
2050 if(sb.st_size < 65536)
2051 peer->transfer->flags.b.minislot = 1;
2052 if(!peer->transfer->flags.b.minislot && (slotsleft() < 1)) {
2054 qstr(sk, "$MaxedOut|");
2057 if((start != 0) && ((start >= sb.st_size) || (lseek(fd, start, SEEK_SET) < 0)))
2060 qstr(sk, "$Error Offset out of range|");
2063 if((numbytes < 0) || (start + numbytes > sb.st_size))
2064 numbytes = sb.st_size - start;
2065 lesk = wrapsock(fd);
2066 transferprepul(peer->transfer, sb.st_size, start, start + numbytes, lesk);
2068 qstrf(sk, "$Sending %ji|", (intmax_t)numbytes);
2072 static void cmd_adcget(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2076 off_t start, numbytes;
2077 struct sharecache *node;
2079 struct socket *lesk;
2083 if(peer->transfer == NULL)
2088 if((argv = parseadc(args)) == NULL)
2093 if(parrlen(argv) < 4)
2098 start = strtoll(argv[2], NULL, 10);
2099 numbytes = strtoll(argv[3], NULL, 10);
2102 if(((fd = openfilelist(argv[1])) < 0) && (errno != 0))
2104 qstr(sk, "$Error Could not send file list|");
2106 } else if(fd >= 0) {
2107 if((wbuf = icsmbstowcs(argv[1], "UTF-8", NULL)) != NULL)
2108 transfersetpath(peer->transfer, wbuf);
2109 peer->transfer->flags.b.minislot = 1;
2113 if(!strncmp(argv[1], "TTH/", 4))
2115 if((node = findbytth(argv[1] + 4)) == NULL)
2117 qstr(sk, "$Error File not in cache|");
2121 if((node = resdcpath((argv[1][0] == '/')?(argv[1] + 1):(argv[1]), "UTF-8", '/')) == NULL)
2123 qstr(sk, "$Error File not in cache|");
2127 if((fd = opensharecache(node)) < 0)
2129 qstrf(sk, "$Error %s|", strerror(errno));
2132 buf = getfspath(node);
2133 if((wbuf = icsmbstowcs(buf, NULL, NULL)) == NULL)
2134 flog(LOG_WARNING, "could not convert native path into a wcs (%s): %s", buf, strerror(errno));
2136 transfersetpath(peer->transfer, wbuf);
2139 if(!strcmp(argv[0], "file"))
2141 for(i = 4; argv[i] != NULL; i++)
2143 if(!strcmp(argv[i], "ZL1"))
2144 initcompress(peer, CPRS_ZLIB);
2146 if(fstat(fd, &sb) < 0)
2148 flog(LOG_WARNING, "could not stat file %ls: %s", node->name, strerror(errno));
2149 qstr(sk, "$Error|");
2152 if(sb.st_size < 65536)
2153 peer->transfer->flags.b.minislot = 1;
2154 if(!peer->transfer->flags.b.minislot && (slotsleft() < 1)) {
2155 qstr(sk, "$MaxedOut|");
2158 if((start != 0) && ((start >= sb.st_size) || (lseek(fd, start, SEEK_SET) < 0)))
2160 qstr(sk, "$Error Offset out of range|");
2163 if((numbytes < 0) || (start + numbytes > sb.st_size))
2164 numbytes = sb.st_size - start;
2165 lesk = wrapsock(fd);
2166 transferprepul(peer->transfer, sb.st_size, start, start + numbytes, lesk);
2169 qstr(sk, "$ADCSND");
2170 sendadc(sk, "file");
2171 sendadc(sk, argv[1]);
2172 sendadcf(sk, "%ji", (intmax_t)start);
2173 sendadcf(sk, "%ji", (intmax_t)numbytes);
2174 if(peer->compress == CPRS_ZLIB)
2178 } else if(!strcmp(argv[0], "tthl")) {
2180 * XXX: Implement full TTHL support.
2182 * In the meantime, this is better than nothing, since it at
2183 * least allows fetching of the TTH root if it isn't already
2188 qstr(sk, "$Error no TTHL data for virtual files|");
2191 qstr(sk, "$ADCSND");
2192 sendadc(sk, "tthl");
2193 sendadc(sk, argv[1]);
2197 sockqueue(sk, node->hashtth, 24);
2199 qstr(sk, "$Error Namespace not implemented|");
2209 static void handletthl(struct dcpeer *peer)
2213 while(peer->inbufdata >= 24)
2215 pushtigertree(&peer->tth, peer->inbuf);
2216 memmove(peer->inbuf, peer->inbuf + 24, peer->inbufdata -= 24);
2217 peer->curread += 24;
2219 if(peer->curread >= peer->totalsize)
2221 if(peer->timeout == NULL)
2222 peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
2223 peer->state = PEER_CMD;
2224 synctigertree(&peer->tth);
2225 restigertree(&peer->tth, buf);
2226 transfersethash(peer->transfer, newhash(L"TTH", 24, buf));
2231 static void cmd_adcsnd(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2234 off_t start, numbytes;
2236 if(peer->transfer == NULL)
2241 if((argv = parseadc(args)) == NULL)
2246 if(parrlen(argv) < 4)
2251 start = strtoll(argv[2], NULL, 10);
2252 numbytes = strtoll(argv[3], NULL, 10);
2253 if(!strcmp(argv[0], "tthl"))
2255 if((start != 0) || (numbytes % 24 != 0))
2257 /* Weird. Bail out. */
2261 if(peer->timeout != NULL)
2262 canceltimer(peer->timeout);
2263 peer->state = PEER_TTHL;
2264 peer->totalsize = numbytes;
2266 peer->fetchingtthl = 0;
2267 inittigertree(&peer->tth);
2269 } else if(!strcmp(argv[0], "file")) {
2270 if(start != peer->transfer->curpos)
2275 if(start + numbytes != peer->transfer->size)
2277 transfersetsize(peer->transfer, start + numbytes);
2282 if(peer->inbufdata > 0)
2284 sockpushdata(sk, peer->inbuf, peer->inbufdata);
2285 peer->inbufdata = 0;
2286 transread(sk, peer);
2289 /* We certainly didn't request this...*/
2298 static void cmd_sending(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2302 if(peer->transfer == NULL)
2307 numbytes = strtoll(args, NULL, 10);
2308 if(peer->transfer->size - peer->transfer->curpos != numbytes)
2310 transfersetsize(peer->transfer, peer->transfer->curpos + numbytes);
2315 if(peer->inbufdata > 0)
2317 sockpushdata(sk, peer->inbuf, peer->inbufdata);
2318 peer->inbufdata = 0;
2319 transread(sk, peer);
2325 static void cmd_(struct socket *sk, struct fnetnode *fn, char *cmd, char *args)
2330 hubhandleaction(sk, fn, cmd, args);
2334 static void cmd_(struct socket *sk, struct dcpeer *peer, char *cmd, char *args)
2340 static int hubreqconn(struct fnetpeer *peer)
2345 if((peer->fn->state != FNN_EST) || (peer->fn->fnet != &dcnet))
2350 if((hub = peer->fn->data) == NULL)
2355 if((mbsnick = icwcstombs(peer->id, hub->charset)) == NULL)
2356 return(1); /* Shouldn't happen, of course, but who knows... */
2359 sendctm(hub->sk, mbsnick);
2360 expectpeer(mbsnick, peer->fn);
2362 qstrf(hub->sk, "$RevConnectToMe %s %s|", hub->nativenick, mbsnick);
2368 static int hubsendchat(struct fnetnode *fn, int public, wchar_t *to, wchar_t *string)
2371 char *mbsstring, *mbsto;
2374 if((mbsto = icwcstombs(to, hub->charset)) == NULL)
2379 if((mbsstring = icwcstombs(string, hub->charset)) == NULL)
2384 if((strchr(mbsto, '|') != NULL) || (strchr(mbsto, ' ') != NULL))
2391 if(strchr(mbsstring, '|') != NULL)
2402 qstrf(hub->sk, "<%s> %s|", hub->nativenick, mbsstring);
2404 qstrf(hub->sk, "$To: %s From: %s $<%s> %s|", mbsto, hub->nativenick, hub->nativenick, mbsstring);
2407 qstrf(hub->sk, "$To: %s From: %s $<%s> %s|", mbsto, hub->nativenick, hub->nativenick, mbsstring);
2414 static void findsizelimit(struct sexpr *sexpr, int *min, int *max)
2416 int minl, maxl, minr, maxr, retmin, retmax;
2421 findsizelimit(sexpr->l, &minl, &maxl);
2422 findsizelimit(sexpr->r, &minr, &maxr);
2423 retmin = (minl > minr)?minl:minr;
2424 if((maxl != -1) && (maxr != -1))
2425 retmax = (maxl < maxr)?maxl:maxr;
2434 findsizelimit(sexpr->l, &minl, &maxl);
2435 findsizelimit(sexpr->r, &minr, &maxr);
2436 retmin = (minl < minr)?minl:minr;
2437 if((maxl == -1) || (maxr == -1))
2440 retmax = (maxl > maxr)?maxl:maxr;
2443 findsizelimit(sexpr->l, &minl, &maxl);
2444 if((minl == 0) && (maxl == -1)) /* Interval is unspecified */
2448 } else if((minl == 0) && (maxl != -1)) {
2451 } else if((minl != 0) && (maxl == -1)) {
2454 } else { /* This would yield two seperate intervals, which DC cannot handle */
2460 retmax = sexpr->d.n - 1;
2463 retmin = sexpr->d.n;
2464 retmax = sexpr->d.n;
2467 retmin = sexpr->d.n + 1;
2481 static struct hash *findsehash(struct sexpr *sexpr)
2483 struct hash *h1, *h2;
2488 if((h1 = findsehash(sexpr->l)) != NULL)
2490 if((h1 = findsehash(sexpr->r)) != NULL)
2494 if((h1 = findsehash(sexpr->l)) == NULL)
2496 if((h2 = findsehash(sexpr->r)) == NULL)
2502 if(!wcscmp(sexpr->d.hash->algo, L"TTH"))
2503 return(sexpr->d.hash);
2510 static int hubsearch(struct fnetnode *fn, struct search *srch, struct srchfnnlist *ln)
2513 struct wcslist *list, *cur;
2514 char *sstr, *buf, *p;
2515 size_t sstrsize, sstrdata;
2516 struct sockaddr *name;
2518 int minsize, maxsize;
2522 if((fn->state != FNN_EST) || (hub->sk == NULL) || (hub->sk->state != SOCK_EST))
2524 list = findsexprstrs(srch->sexpr);
2525 findsizelimit(srch->sexpr, &minsize, &maxsize);
2526 hash = findsehash(srch->sexpr);
2527 if((minsize != 0) && (maxsize != -1))
2529 /* Choose either minsize or maxsize by trying to determine
2530 * which choice will be most restrictive. The result can be
2531 * approximative at best anyway, so don't try too hard... */
2532 if((50000000 - maxsize) > (minsize - 50000000))
2538 sstrsize = sstrdata = 0;
2539 if((hash != NULL) && (hash->len == 24))
2541 /* Prioritize hash searches above all else */
2542 bufcat(sstr, "F?T?0?9?TTH:", 12);
2543 buf = base32encode(hash->buf, hash->len);
2545 bufcat(sstr, buf, 39);
2550 sizebuf2(sstr, sstrdata + 32, 1);
2551 sstrdata += snprintf(sstr + sstrdata, sstrsize - sstrdata, "T?F?%i?1?", minsize);
2552 } else if(maxsize != -1) {
2553 sizebuf2(sstr, sstrdata + 32, 1);
2554 sstrdata += snprintf(sstr + sstrdata, sstrsize - sstrdata, "T?T?%i?1?", maxsize);
2556 bufcat(sstr, "F?F?0?1?", 8);
2560 for(cur = list; cur != NULL; cur = cur->next)
2562 /* Use DCCHARSET in $Get paths until further researched... */
2563 if((buf = icwcstombs(cur->str, DCCHARSET)) == NULL)
2565 /* Can't find anything anyway if the search expression
2566 * requires characters outside DC's charset. There's
2567 * nothing technically wrong with the search itself,
2568 * however, so return success. This should be
2569 * considered as an optimization. */
2576 addtobuf(sstr, '$');
2578 * It probably doesn't hurt if buf contains any extra
2579 * dollar signs - it will just result in extra search
2580 * terms, and the extraneous results will be filtered by
2581 * the search layer anyway. It hurts if it contains any
2582 * pipes, though, so let's sell them for money.
2584 for(p = buf; *p; p++)
2589 bufcat(sstr, buf, strlen(buf));
2593 /* Will match all files... :-/ */
2594 addtobuf(sstr, '.');
2600 if(sockgetremotename2(udpsock, hub->sk, &name, &namelen) < 0)
2602 flog(LOG_WARNING, "cannot get address of UDP socket");
2604 qstrf(hub->sk, "$Search %s %s|", formataddress(name, namelen), sstr);
2608 qstrf(hub->sk, "$Search Hub:%s %s|", hub->nativenick, sstr);
2612 fn->lastsrch = time(NULL);
2620 #define cc(c) ((void (*)(struct socket *, void *, char *, char *))(c))
2621 static struct command hubcmds[] =
2623 {"$Lock", cc(cmd_lock)},
2624 {"$HubName", cc(cmd_hubname)},
2625 {"$Hello", cc(cmd_hello)},
2626 {"$Quit", cc(cmd_quit)},
2627 {"$NickList", cc(cmd_nicklist)},
2628 {"$OpList", cc(cmd_oplist)},
2629 {"$MyINFO", cc(cmd_myinfo)},
2630 {"$ForceMove", cc(cmd_forcemove)},
2631 {"$Search", cc(cmd_search)},
2632 {"$MultiSearch", cc(cmd_search)},
2633 {"$ConnectToMe", cc(cmd_connecttome)},
2634 {"$RevConnectToMe", cc(cmd_revconnecttome)},
2635 {"$GetNetInfo", cc(cmd_getnetinfo)},
2636 {"$To:", cc(cmd_to)},
2637 {"$SR", cc(cmd_sr)},
2638 {"$UserCommand", cc(cmd_usercommand)},
2639 {"$GetPass", cc(cmd_getpass)},
2640 {"$LogedIn", cc(cmd_logedin)}, /* sic */
2641 {"$Supports", cc(cmd_hubsupports)},
2645 static struct command peercmds[] =
2647 {"$MyNick", cc(cmd_mynick)},
2648 {"$Lock", cc(cmd_peerlock)},
2649 {"$Direction", cc(cmd_direction)},
2650 {"$Key", cc(cmd_key)},
2651 {"$FileLength", cc(cmd_filelength)},
2652 {"$Error", cc(cmd_error)},
2653 {"$MaxedOut", cc(cmd_maxedout)},
2654 {"$Get", cc(cmd_get)},
2655 {"$Send", cc(cmd_send)},
2656 {"$Supports", cc(cmd_supports)},
2657 {"$GetBlock", cc(cmd_getblock)},
2658 {"$UGetBlock", cc(cmd_getblock)},
2659 {"$GetZBlock", cc(cmd_getblock)},
2660 {"$UGetZBlock", cc(cmd_getblock)},
2661 {"$ADCGET", cc(cmd_adcget)},
2662 {"$ADCSND", cc(cmd_adcsnd), 1},
2663 {"$Sending", cc(cmd_sending), 1},
2668 static void dctransdetach(struct transfer *transfer, struct dcpeer *peer)
2670 CBUNREG(transfer, trans_filterout, peer);
2673 peer->transfer = NULL;
2677 static void dctransgotdata(struct transfer *transfer, struct dcpeer *peer)
2681 unsigned char outbuf[1024];
2685 if((peer->state == PEER_TRNS) || (peer->state == PEER_SYNC))
2687 if(sockqueuesize(peer->sk) < 65536)
2689 if((buf = transfergetdata(transfer, &bufsize)) != NULL)
2691 if(peer->compress == CPRS_NONE)
2693 sockqueue(peer->sk, buf, bufsize);
2694 } else if(peer->compress == CPRS_ZLIB) {
2695 cstr = peer->cprsdata;
2696 cstr->next_in = buf;
2697 cstr->avail_in = bufsize;
2698 while(cstr->avail_in > 0)
2700 cstr->next_out = outbuf;
2701 cstr->avail_out = sizeof(outbuf);
2702 if((ret = deflate(cstr, 0)) != Z_OK)
2704 flog(LOG_WARNING, "bug? deflate() did not return Z_OK (but rather %i)", ret);
2708 sockqueue(peer->sk, outbuf, sizeof(outbuf) - cstr->avail_out);
2713 if(peer->state == PEER_SYNC)
2715 if(peer->compress == CPRS_ZLIB)
2717 cstr = peer->cprsdata;
2718 cstr->next_in = NULL;
2722 cstr->next_out = outbuf;
2723 cstr->avail_out = sizeof(outbuf);
2724 ret = deflate(cstr, Z_FINISH);
2725 if((ret != Z_OK) && (ret != Z_STREAM_END))
2727 flog(LOG_WARNING, "bug? deflate(Z_FINISH) did not return Z_OK (but rather %i)", ret);
2731 sockqueue(peer->sk, outbuf, sizeof(outbuf) - cstr->avail_out);
2732 } while(ret != Z_STREAM_END);
2738 if(peer->timeout == NULL)
2739 peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
2740 peer->state = PEER_CMD;
2742 transfersetstate(transfer, TRNS_HS);
2743 socksettos(peer->sk, confgetint("fnet", "fnptos"));
2744 transfer->flags.b.minislot = 0;
2745 peer->sk->writecb = NULL;
2752 static void dctransendofdata(struct transfer *transfer, struct dcpeer *peer)
2754 peer->state = PEER_SYNC;
2755 dctransgotdata(transfer, peer);
2758 static void dcwantdata(struct transfer *transfer, struct dcpeer *peer)
2760 if(transferdatasize(transfer) < 65536)
2761 peer->sk->ignread = 0;
2764 static void transread(struct socket *sk, struct dcpeer *peer)
2768 struct transfer *transfer;
2770 if((buf = sockgetinbuf(sk, &bufsize)) == NULL)
2772 if(peer->transfer == NULL)
2778 transferputdata(peer->transfer, buf, bufsize);
2780 if(peer->transfer->curpos >= peer->transfer->size)
2782 transfer = peer->transfer;
2783 transferdetach(transfer);
2784 transferendofdata(transfer);
2787 if(transferdatasize(peer->transfer) > 65535)
2791 static void transerr(struct socket *sk, int err, struct dcpeer *peer)
2793 struct transfer *transfer;
2795 if((transfer = peer->transfer) == NULL)
2800 transferdetach(transfer);
2801 transferendofdata(transfer);
2804 static void transwrite(struct socket *sk, struct dcpeer *peer)
2806 if((peer->state != PEER_TRNS) && (peer->state != PEER_SYNC))
2808 if(peer->transfer == NULL)
2813 dctransgotdata(peer->transfer, peer);
2816 static void udpread(struct socket *sk, void *data)
2818 char *buf, *p, *p2, *hashbuf;
2819 size_t buflen, hashlen;
2820 char *nick, *filename, *hubname;
2821 struct sockaddr_in hubaddr;
2824 struct fnetnode *fn, *myfn;
2827 wchar_t *wnick, *wfile;
2830 if((buf = sockgetinbuf(sk, &buflen)) == NULL)
2832 buf = srealloc(buf, buflen + 1);
2834 if(!strncmp(buf, "$SR ", 4))
2838 if((p2 = strchr(p, ' ')) == NULL)
2846 if((p2 = strchr(p, 5)) == NULL)
2853 if((p2 = strchr(p, ' ')) == NULL)
2859 size = strtoll(p, NULL, 10);
2861 if((p2 = strchr(p, '/')) == NULL)
2869 if((p2 = strchr(p, 5)) == NULL)
2876 if((p2 = strstr(p, " (")) == NULL)
2883 if((p2 = strchr(p, ':')) == NULL)
2889 hubaddr.sin_family = AF_INET;
2890 if(!inet_aton(p, &hubaddr.sin_addr))
2896 if((p2 = strchr(p, ')')) == NULL)
2902 hubaddr.sin_port = htons(atoi(p));
2903 /* Use DCCHARSET in $Get paths until further researched... */
2904 if((wfile = icmbstowcs(filename, DCCHARSET)) == NULL)
2911 if(!strncmp(hubname, "TTH:", 4))
2913 if((hashbuf = base32decode(hubname + 4, &hashlen)) != NULL)
2916 hash = newhash(L"TTH", 24, hashbuf);
2920 for(fn = fnetnodes; fn != NULL; fn = fn->next)
2922 if((fn->fnet == &dcnet) && ((hub = fn->data) != NULL))
2924 if((hub->nativename != NULL) && !strcmp(hub->nativename, hubname))
2939 for(fn = fnetnodes; fn != NULL; fn = fn->next)
2941 if((fn->fnet == &dcnet) && ((hub = fn->data) != NULL))
2943 if((hub->sk != NULL) && addreq(hub->sk->remote, (struct sockaddr *)&hubaddr))
2954 if((wnick = icmbstowcs(nick, (hub == NULL)?DCCHARSET:(hub->charset))) == NULL)
2959 sr = newsrchres(&dcnet, wfile, wnick);
2960 if(sr->peernick != NULL)
2962 sr->peernick = swcsdup(wnick);
2968 getfnetnode(sr->fn = myfn);
2977 static void hubread(struct socket *sk, struct fnetnode *fn)
2984 hub = (struct dchub *)fn->data;
2985 if((newbuf = sockgetinbuf(sk, &datalen)) == NULL)
2987 if(hub->inbufdata > 500000) /* Discard possible malicious data */
2989 sizebuf2(hub->inbuf, hub->inbufdata + datalen, 1);
2990 memcpy(hub->inbuf + hub->inbufdata, newbuf, datalen);
2992 p = hub->inbuf + hub->inbufdata;
2993 hub->inbufdata += datalen;
2994 while((datalen > 0) && ((p = memchr(p, '|', datalen)) != NULL))
2997 newqcmd(&hub->queue, hub->inbuf);
2998 memmove(hub->inbuf, p, hub->inbufdata -= p - hub->inbuf);
2999 datalen = hub->inbufdata;
3004 static void huberr(struct socket *sk, int err, struct fnetnode *fn)
3009 static int hubsetnick(struct fnetnode *fn, wchar_t *newnick)
3015 if((buf = icwcstombs(newnick, (hub == NULL)?DCCHARSET:(hub->charset))) == NULL)
3017 if((strchr(buf, ' ') != NULL) || (strchr(buf, '|') != NULL) || (strchr(buf, '$') != NULL))
3022 if(hub == NULL) /* Not yet connected */
3027 if(hub->nativenick != NULL)
3028 free(hub->nativenick);
3029 hub->nativenick = buf;
3033 static struct dchub *newdchub(struct fnetnode *fn)
3040 new = smalloc(sizeof(*new));
3041 memset(new, 0, sizeof(*new));
3043 if(confgetint("dc", "dcppemu"))
3045 if((emu = wpfind(fn->args, L"dcppemu")) != NULL)
3053 if((wcharset = wpfind(fn->args, L"charset")) != NULL)
3055 if((charset = icwcstombs(wcharset, "US-ASCII")) != NULL)
3057 if(!havecharset(charset))
3065 new->charset = charset;
3067 new->charset = sstrdup(DCCHARSET);
3068 if(hubsetnick(fn, fn->mynick))
3069 fnetsetnick(fn, L"DoldaConnectUser-IN");
3070 /* IN as in Invalid Nick */
3074 static struct dcpeer *newdcpeer(struct socket *sk)
3078 new = smalloc(sizeof(*new));
3079 memset(new, 0, sizeof(*new));
3080 new->transfer = NULL;
3083 if(confgetint("dc", "dcppemu"))
3094 static void freedcpeer(struct dcpeer *peer)
3097 struct qcommand *qcmd;
3102 if(peer->next != NULL)
3103 peer->next->prev = peer->prev;
3104 if(peer->prev != NULL)
3105 peer->prev->next = peer->next;
3106 if(peer->transfer != NULL)
3108 if(peer->transfer->dir == TRNSD_UP)
3109 peer->transfer->close = 1;
3110 if(peer->transfer->dir == TRNSD_DOWN)
3111 resettransfer(peer->transfer);
3112 transferdetach(peer->transfer);
3114 if(peer->timeout != NULL)
3115 canceltimer(peer->timeout);
3116 if(peer->sk->data == peer)
3117 peer->sk->data = NULL;
3118 peer->sk->readcb = NULL;
3119 peer->sk->writecb = NULL;
3120 peer->sk->errcb = NULL;
3123 if(peer->supports != NULL)
3125 for(i = 0; peer->supports[i] != NULL; i++)
3126 free(peer->supports[i]);
3127 free(peer->supports);
3129 if(peer->inbuf != NULL)
3131 if(peer->key != NULL)
3133 if(peer->wcsname != NULL)
3134 free(peer->wcsname);
3135 if(peer->nativename != NULL)
3136 free(peer->nativename);
3137 if(peer->charset != NULL)
3138 free(peer->charset);
3139 if(peer->fn != NULL)
3140 putfnetnode(peer->fn);
3141 while((qcmd = ulqcmd(&peer->queue)) != NULL)
3147 static void hubconnect(struct fnetnode *fn, struct socket *sk)
3151 sk->readcb = (void (*)(struct socket *, void *))hubread;
3152 sk->errcb = (void (*)(struct socket *, int, void *))huberr;
3153 fn->data = hub = newdchub(fn);
3155 getsock(hub->sk = sk);
3159 static void hubdestroy(struct fnetnode *fn)
3163 struct qcommand *qcmd;
3165 hub = (struct dchub *)fn->data;
3167 while((qcmd = ulqcmd(&hub->queue)) != NULL)
3169 if(hub->supports != NULL)
3171 for(i = 0; hub->supports[i] != NULL; i++)
3172 free(hub->supports[i]);
3173 free(hub->supports);
3175 if(hub->nativename != NULL)
3176 free(hub->nativename);
3177 if(hub->nativenick != NULL)
3178 free(hub->nativenick);
3179 if(hub->charset != NULL)
3181 if(hub->inbuf != NULL)
3186 static void hubkill(struct fnetnode *fn)
3190 hub = (struct dchub *)fn->data;
3194 static wchar_t *dcbasename(wchar_t *filename)
3198 if((ret = wcsrchr(filename, L'\\')) != NULL)
3203 static struct transferiface dctransfer =
3205 .detach = (void (*)(struct transfer *, void *))dctransdetach,
3206 .gotdata = (void (*)(struct transfer *, void *))dctransgotdata,
3207 .endofdata = (void (*)(struct transfer *, void *))dctransendofdata,
3208 .wantdata = (void (*)(struct transfer *, void *))dcwantdata
3211 static struct fnet dcnet =
3214 .connect = hubconnect,
3215 .destroy = hubdestroy,
3217 .setnick = hubsetnick,
3218 .reqconn = hubreqconn,
3219 .sendchat = hubsendchat,
3220 .search = hubsearch,
3221 .filebasename = dcbasename
3224 static void peerread(struct socket *sk, struct dcpeer *peer)
3228 struct command *cmd;
3230 if((newbuf = sockgetinbuf(sk, &datalen)) == NULL)
3232 sizebuf2(peer->inbuf, peer->inbufdata + datalen, 1);
3233 memcpy(peer->inbuf + peer->inbufdata, newbuf, datalen);
3235 peer->inbufdata += datalen;
3236 if(peer->state == PEER_CMD)
3239 while((peer->inbufdata > 0) && (p = memchr(peer->inbuf, '|', peer->inbufdata)) != NULL)
3242 newqcmd(&peer->queue, peer->inbuf);
3243 for(cmd = peercmds; cmd->handler != NULL; cmd++)
3245 if(!memcmp(peer->inbuf, cmd->name, strlen(cmd->name)) && ((peer->inbuf[strlen(cmd->name)] == ' ') || (peer->inbuf[strlen(cmd->name)] == '|')))
3248 memmove(peer->inbuf, p, peer->inbufdata -= p - peer->inbuf);
3251 peer->state = PEER_STOP;
3255 } else if(peer->state == PEER_TTHL) {
3260 static void peererror(struct socket *sk, int err, struct dcpeer *peer)
3265 static void peerconnect(struct socket *sk, int err, struct fnetnode *fn)
3267 struct dcpeer *peer;
3277 peer = newdcpeer(sk);
3280 peer->dcppemu = hub->dcppemu;
3281 sk->readcb = (void (*)(struct socket *, void *))peerread;
3282 sk->errcb = (void (*)(struct socket *, int, void *))peererror;
3284 socksettos(sk, confgetint("fnet", "fnptos"));
3286 peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
3291 static void peeraccept(struct socket *sk, struct socket *newsk, void *data)
3293 struct dcpeer *peer;
3295 peer = newdcpeer(newsk);
3297 newsk->readcb = (void (*)(struct socket *, void *))peerread;
3298 newsk->errcb = (void (*)(struct socket *, int, void *))peererror;
3300 socksettos(newsk, confgetint("fnet", "fnptos"));
3301 peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
3304 static void updatehmlist(void)
3306 int i, lev, ic, ret;
3307 struct sharecache *node;
3308 char *buf, *buf2, numbuf[32];
3309 size_t bufsize, bufdata;
3313 buf = smalloc(bufsize = 65536);
3314 node = shareroot->child;
3319 /* Use DCCHARSET in $Get paths until further researched... */
3320 if((buf2 = icwcstombs(node->name, DCCHARSET)) != NULL)
3322 for(i = 0; i < lev; i++)
3324 bufcat(buf, buf2, strlen(buf2));
3326 if(node->f.b.type == FILE_REG)
3329 sprintf(numbuf, "%ji", (intmax_t)node->size);
3330 bufcat(buf, numbuf, strlen(numbuf));
3337 if((node->child != NULL) && !ic)
3341 } else if(node->next != NULL) {
3344 while(node->next == NULL)
3347 node = node->parent;
3348 if(node == shareroot)
3351 if(node == shareroot)
3356 if(hmlistname != NULL)
3361 hmlistname = sstrdup("/tmp/dc-filelist-hm-XXXXXX");
3362 if((fd = mkstemp(hmlistname)) < 0)
3364 flog(LOG_WARNING, "could not create HM file list tempfile: %s", strerror(errno));
3369 * I do not want to implement a good Huffman encoder, and it's not
3370 * like Huffman encoding actually yields any impressive results
3371 * for DC file lists anyway, so I'll just output a bogus
3372 * tree. Implement a good encoder if you want to.
3374 write(fd, "HE3\r\0", 5);
3375 write(fd, &bufdata, 4);
3377 write(fd, &ibuf, 2);
3379 for(i = 0; i < 256; i++)
3382 write(fd, &ibuf, 1);
3384 for(i = 0; i < 256; i++)
3386 for(buf2 = buf; bufdata > 0;)
3388 if((ret = write(fd, buf2, bufdata)) < 0)
3390 flog(LOG_WARNING, "could not write file list: %s", strerror(errno));
3401 static struct xmlent
3408 /* {L'\'', L"""}, Emulate DC++ escaping by omitting this. */
3412 static wchar_t *escapexml(wchar_t *src)
3414 static wchar_t *buf = NULL;;
3417 size_t bufsize, bufdata;
3423 bufsize = bufdata = 0;
3424 for(; *src != L'\0'; src++)
3427 if((c > 0) && (c < 32))
3429 bufcat(buf, L"&#", 2);
3430 ret = swprintf(nbuf, (sizeof(nbuf) / sizeof(*nbuf)), L"%i", c);
3431 bufcat(buf, nbuf, ret);
3432 addtobuf(buf, L';');
3434 for(ent = entities; ent->ent != NULL; ent++)
3438 bufcat(buf, ent->ent, wcslen(ent->ent));
3442 if(ent->ent == NULL)
3443 addtobuf(buf, *src);
3446 addtobuf(buf, L'\0');
3450 static void updatexmllist(void)
3454 char cidbuf[14], *namebuf;
3456 struct sharecache *node;
3458 if(xmllistname != NULL)
3460 unlink(xmllistname);
3463 xmllistname = sstrdup("/tmp/dc-filelist-dcxml-XXXXXX");
3464 if((fd = mkstemp(xmllistname)) < 0)
3466 flog(LOG_WARNING, "could not create XML file list tempfile: %s", strerror(errno));
3471 if((fs = fdopen(fd, "w")) == NULL)
3473 flog(LOG_WARNING, "could not fdopen XML list fd %i: %s", fd, strerror(errno));
3474 unlink(xmllistname);
3480 fprintf(fs, "<?xml version=\"1.0\" encoding=\"utf-8\" standalone=\"yes\"?>\r\n");
3481 for(i = 0; i < sizeof(cidbuf) - 1; i++)
3482 cidbuf[i] = (rand() % ('Z' - 'A' + 1)) + 'A';
3484 if(confgetint("dc", "dcppemu"))
3485 fprintf(fs, "<FileListing Version=\"1\" CID=\"%s\" Base=\"/\" Generator=\"DC++ 0.674\">\r\n", cidbuf);
3487 fprintf(fs, "<FileListing Version=\"1\" CID=\"%s\" Base=\"/\" Generator=\"%s\">\r\n", cidbuf, "DoldaConnect" VERSION);
3489 node = shareroot->child;
3493 if((namebuf = icswcstombs(escapexml(node->name), "UTF-8", NULL)) != NULL)
3495 for(i = 0; i < lev; i++)
3497 if(node->child != NULL)
3499 fprintf(fs, "<Directory Name=\"%s\">\r\n", namebuf);
3504 fprintf(fs, "<File Name=\"%s\" Size=\"%ji\"", namebuf, (intmax_t)node->size);
3505 if(node->f.b.hastth)
3507 hashbuf = base32encode(node->hashtth, 24);
3508 fprintf(fs, " TTH=\"%.39s\"", hashbuf);
3511 fprintf(fs, "/>\r\n");
3513 while(node->next == NULL)
3515 node = node->parent;
3516 if(node == shareroot)
3521 for(i = 0; i < lev; i++)
3523 fprintf(fs, "</Directory>\r\n");
3526 if(node == shareroot)
3532 if(confgetint("dc", "dcppemu"))
3533 fprintf(fs, "</FileListing>");
3535 fprintf(fs, "</FileListing>\r\n");
3539 static void updatexmlbz2list(void)
3548 if(xmllistname == NULL)
3550 if((in = fopen(xmllistname, "r")) == NULL)
3552 flog(LOG_WARNING, "could not open XML file list for bzipping: %s", strerror(errno));
3555 if(xmlbz2listname != NULL)
3557 unlink(xmlbz2listname);
3558 free(xmlbz2listname);
3560 xmlbz2listname = sstrdup("/tmp/dc-filelist-dcxmlbz2-XXXXXX");
3561 if((fd = mkstemp(xmlbz2listname)) < 0)
3563 flog(LOG_WARNING, "could not create bzipped XML file list tempfile: %s", strerror(errno));
3564 free(xmlbz2listname);
3565 xmlbz2listname = NULL;
3569 if((real = fdopen(fd, "w")) == NULL)
3571 flog(LOG_WARNING, "could not fdopen bzipped XML list fd %i: %s", fd, strerror(errno));
3573 unlink(xmlbz2listname);
3574 free(xmlbz2listname);
3575 xmlbz2listname = NULL;
3579 out = BZ2_bzWriteOpen(&err, real, 9, 0, 0);
3582 flog(LOG_WARNING, "could not open bzip2 stream from XML list");
3584 unlink(xmlbz2listname);
3585 free(xmlbz2listname);
3586 xmlbz2listname = NULL;
3592 bufdata = fread(buf, 1, sizeof(buf), in);
3593 BZ2_bzWrite(&err, out, buf, bufdata);
3596 BZ2_bzWriteClose(&err, out, 0, NULL, NULL);
3600 static void listtimercb(int cancelled, void *uudata)
3602 listwritetimer = NULL;
3607 static void updatelists(int now)
3609 if((hmlistname == NULL) || (xmllistname == NULL) || (xmlbz2listname == NULL))
3613 if(listwritetimer == NULL)
3614 listwritetimer = timercallback(ntime() + confgetint("cli", "hashwritedelay"), listtimercb, NULL);
3617 if(listwritetimer != NULL)
3618 canceltimer(listwritetimer);
3624 static int shareupdate(unsigned long long uusharesize, void *data)
3630 static char *quotestr(char *str)
3634 size_t bufsize, bufdata;
3636 static char *enc = NULL;
3637 size_t encsize, encdata;
3640 bufsize = bufdata = 0;
3641 for(p = (unsigned char *)str; *p; p++)
3644 bufcat(buf, "\\b", 2);
3646 bufcat(buf, "\\t", 2);
3648 bufcat(buf, "\\n", 2);
3650 bufcat(buf, "\\r", 2);
3652 bufcat(buf, "\\\\", 2);
3656 bprintf(buf, "\\x%02x", *p);
3662 if((wbuf = icmbstowcs((char *)buf, DCCHARSET)) != NULL)
3664 enc = icwcstombs(wbuf, NULL);
3669 encsize = encdata = 0;
3670 for(p = buf; *p; p++) {
3674 bprintf(buf, "\\x%x", *p);
3681 static void logunimpl(char *list, char *cmd, char *args)
3685 if((log = fopen("/tmp/dc-unimpl", "a")) == NULL)
3687 flog(LOG_WARNING, "could not open unimpl log: %s", strerror(errno));
3692 fputs(quotestr(cmd), log);
3696 fputs(quotestr(args), log);
3702 static void dispatchcommand(struct qcommand *qcmd, struct command *cmdlist, struct socket *sk, void *data)
3705 struct command *cmd;
3707 if((p = strchr(qcmd->string, ' ')) != NULL)
3709 for(cmd = cmdlist; cmd->handler != NULL; cmd++)
3711 if(!strcmp(cmd->name, qcmd->string))
3714 if(cmd->handler != NULL)
3716 cmd->handler(sk, data, qcmd->string, p);
3717 } else if(confgetint("dc", "logunimpl")) {
3718 if(cmdlist == hubcmds)
3719 logunimpl("hub", qcmd->string, p);
3720 else if(cmdlist == peercmds)
3721 logunimpl("peer", qcmd->string, p);
3723 logunimpl("other?!", qcmd->string, p);
3727 static int run(void)
3729 struct fnetnode *fn, *nextfn;
3731 struct dcpeer *peer, *nextpeer;
3732 struct qcommand *qcmd;
3736 for(fn = fnetnodes; fn != NULL; fn = nextfn)
3739 if(fn->fnet != &dcnet)
3741 if(fn->data == NULL)
3743 hub = (struct dchub *)fn->data;
3744 if((qcmd = ulqcmd(&hub->queue)) != NULL)
3746 if(*qcmd->string == '$')
3748 if((hub->sk != NULL) && (hub->sk->state == SOCK_EST))
3749 dispatchcommand(qcmd, hubcmds, hub->sk, fn);
3750 } else if(*qcmd->string != 0) {
3751 hubrecvchat(hub->sk, fn, NULL, qcmd->string);
3758 for(peer = peers; peer != NULL; peer = nextpeer)
3760 nextpeer = peer->next;
3761 if((qcmd = ulqcmd(&peer->queue)) != NULL)
3763 if(peer->timeout != NULL)
3764 canceltimer(peer->timeout);
3765 peer->timeout = timercallback(ntime() + 180, (void (*)(int, void *))peertimeout, peer);
3766 if(*qcmd->string == '$')
3767 dispatchcommand(qcmd, peercmds, peer->sk, peer);
3776 static void preinit(int hup)
3783 static int updateudpport(struct configvar *var, void *uudata)
3785 struct sockaddr_in addr;
3786 struct socket *newsock;
3788 memset(&addr, 0, sizeof(addr));
3789 addr.sin_family = AF_INET;
3790 addr.sin_port = htons(var->val.num);
3791 if((newsock = netcsdgram((struct sockaddr *)&addr, sizeof(addr))) == NULL)
3793 flog(LOG_WARNING, "could not create new DC UDP socket, reverting to old: %s", strerror(errno));
3796 newsock->readcb = udpread;
3803 static int updatetcpport(struct configvar *var, void *uudata)
3805 struct sockaddr_in addr;
3806 struct socket *newsock;
3808 memset(&addr, 0, sizeof(addr));
3809 addr.sin_family = AF_INET;
3810 addr.sin_port = htons(var->val.num);
3811 if((newsock = netcslisten(SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr), peeraccept, NULL)) == NULL)
3812 flog(LOG_INFO, "could not listen to a remote address, going into passive mode");
3819 static int init(int hup)
3821 struct sockaddr_in addr;
3825 GCBREG(sharechangecb, shareupdate, NULL);
3830 addr.sin_family = AF_INET;
3831 memset(&addr.sin_addr, 0, sizeof(addr.sin_addr));
3832 addr.sin_port = htons(confgetint("dc", "udpport"));
3833 if((udpsock = netcsdgram((struct sockaddr *)&addr, sizeof(addr))) == NULL)
3835 flog(LOG_CRIT, "could not create DC UDP socket: %s", strerror(errno));
3838 udpsock->readcb = udpread;
3839 addr.sin_port = htons(confgetint("dc", "tcpport"));
3840 if((tcpsock = netcslisten(SOCK_STREAM, (struct sockaddr *)&addr, sizeof(addr), peeraccept, NULL)) == NULL)
3841 flog(LOG_INFO, "could not listen to a remote address, going into passive mode");
3842 CBREG(confgetvar("dc", "udpport"), conf_update, updateudpport, NULL, NULL);
3843 CBREG(confgetvar("dc", "tcpport"), conf_update, updatetcpport, NULL, NULL);
3844 CBREG(confgetvar("net", "mode"), conf_update, updatetcpport, NULL, NULL);
3849 static void terminate(void)
3851 if(hmlistname != NULL)
3856 if(xmllistname != NULL)
3858 unlink(xmllistname);
3861 if(xmlbz2listname != NULL)
3863 unlink(xmlbz2listname);
3864 free(xmlbz2listname);
3868 static struct configvar myvars[] =
3870 /** Specifies the share description reported to other DC users. */
3871 {CONF_VAR_STRING, "desc", {.str = L""}},
3872 /** Specifies the speed reported to other DC users. Normal values
3873 * are 28.8Kbps, 33.6Kbps, 56Kbps, Satellite, ISDN, DSL, Cable,
3874 * LAN(T1) or LAN(T3)*/
3875 {CONF_VAR_STRING, "speedstring", {.str = L"LAN(T1)"}},
3876 /** The e-mail address to report to other DC users. */
3877 {CONF_VAR_STRING, "email", {.str = L"spam@spam.org"}},
3878 /** Specifies a specific UDP port to use for DC search results. If
3879 * left unspecified, a port is allocated dynamically. Useful for
3880 * NAT routers (see also the net.visibleipv4 address for those
3882 {CONF_VAR_INT, "udpport", {.num = 0}},
3883 /** Specifies a specific TCP port to use for DC peer
3884 * connections. If left unspecified, a port is allocated
3885 * dynamically. Useful for NAT routers (see also the
3886 * net.visibleipv4 address for those cases). */
3887 {CONF_VAR_INT, "tcpport", {.num = 0}},
3888 /** If set to true, doldacond will do its best to emulate DC++
3889 * (currently v0.674). This should be left off if at all possible,
3890 * since turning it on will violate the rules of most hubs and
3891 * thus give hub owners an actual reason to kick you if it is
3892 * detected. It might be needed for some of the more bone-headed
3893 * hub owners, though. Note that DC++ emulation can also be turned
3894 * on or off for individual hubs, overriding this setting. */
3895 {CONF_VAR_BOOL, "dcppemu", {.num = 0}},
3896 /** Use for debugging. If set to true, doldacond will log all
3897 * unknown commands it receives, and their arguments, to
3898 * /tmp/dc-unimpl. */
3899 {CONF_VAR_BOOL, "logunimpl", {.num = 0}},
3900 /** If set to true, doldacond will hide its support for deflate
3901 * compression of transfers from other clients, so that they will
3902 * not request compressed uploads. Compressed transfers may
3903 * consume a non-trivial amount of CPU time on slower machines. */
3904 {CONF_VAR_BOOL, "hidedeflate", {.num = 0}},
3908 static struct module me =
3917 .terminate = terminate,