6 typedef struct Dict Dict;
7 typedef struct Piece Piece;
8 typedef struct File File;
9 typedef struct Stats Stats;
17 char typ; // i, d, s, l
55 char *deftrack = "http://exodus.desync.com/announce";
56 char *mntweb = "/mnt/web";
74 return nhavepieces >= npieces;
89 bparse(char *s, char *e, Dict **dp)
111 s = bparse(s, e, dp);
115 d = *dp = mallocz(sizeof(*d), 1);
119 s = bparse(s, e, &d->val);
130 if((s = memchr(x, 'e', e - x)) == nil)
136 if((x = memchr(s, ':', e - s)) == nil)
139 if((n = atoi(s)) < 0)
142 if((s > e) || (s < x)){
148 d = mallocz(sizeof(*d) + n+1, 1);
150 memmove(d->str, x, d->len = n);
159 if(d && (d->typ == 's' || d->typ == 'i'))
165 dlook(Dict *d, char *s)
167 for(; d && d->typ == 'd'; d = d->next)
168 if(d->len && strcmp(d->str, s) == 0)
174 readall(int fd, char **p)
180 while(*p = realloc(*p, n+1024)){
181 if((r = read(fd, *p+n, 1024)) <= 0)
189 rwpiece(int wr, int index, uchar *data, int len, int poff)
195 if(len <= 0 || poff >= pieces[index].len)
197 if(len+poff > pieces[index].len)
198 len = pieces[index].len - poff;
199 off = (vlong)index * blocksize;
201 for(f = files; f; f = f->next)
202 if((f->off+f->len) > off)
205 n = ((off + len) > f->len) ? f->len - off : len;
206 if((n = (wr ? pwrite(f->fd, data, n, off) : pread(f->fd, data, n, off))) <= 0)
208 if((m = rwpiece(wr, index, data + n, len - n, poff + n)) < 0)
216 uchar *p, m, hash[20];
220 if(havemap[x>>3] & m)
222 p = malloc(blocksize);
224 if(rwpiece(0, x, p, n, 0) != n){
228 sha1(p, n, hash, nil);
230 if(memcmp(hash, pieces[x].hash, 20))
233 if((havemap[x>>3] & m) == 0){
236 stats.left -= pieces[x].len;
243 pickpiece(uchar *map)
250 for(i = 0; i<nhavemap; i++){
253 for(x = i<<3, m = 0x80; m; m >>= 1, x++){
254 if((~map[i] | havemap[i]) & m)
264 unpack(uchar *s, int n, char *fmt, ...)
278 if(s+1 > e) goto Err;
279 *va_arg(arg, int*) = *s++;
282 if(s+2 > e) goto Err;
283 *va_arg(arg, int*) = s[0]<<8 | s[1];
287 if(s+4 > e) goto Err;
288 *va_arg(arg, int*) = s[0]<<24 | s[1]<<16 | s[2]<<8 | s[3];
292 if(s+4 > e) goto Err;
293 *va_arg(arg, vlong*) =
314 pack(uchar *s, int n, char *fmt, ...)
330 i = va_arg(arg, int);
332 if(s+1 > e) goto Err;
336 i = va_arg(arg, int);
337 if(s+2 > e) goto Err;
338 *s++ = (i>>8) & 0xFF;
342 i = va_arg(arg, int);
343 if(s+4 > e) goto Err;
344 *s++ = (i>>24) & 0xFF;
345 *s++ = (i>>16) & 0xFF;
346 *s++ = (i>>8) & 0xFF;
350 v = va_arg(arg, vlong);
351 if(s+8 > e) goto Err;
352 *s++ = (v>>56) & 0xFF;
353 *s++ = (v>>48) & 0xFF;
354 *s++ = (v>>40) & 0xFF;
355 *s++ = (v>>32) & 0xFF;
356 *s++ = (v>>24) & 0xFF;
357 *s++ = (v>>16) & 0xFF;
358 *s++ = (v>>8) & 0xFF;
362 i = va_arg(arg, int);
363 if(s+i > e) goto Err;
364 memmove(s, va_arg(arg, void*), i);
379 peer(int fd, int incoming, char *addr)
381 uchar buf[64+MAXIO], *map, *told, *p, m;
382 int mechoking, hechoking;
387 if(debug) fprint(2, "peer %s: %s connected\n", addr, incoming ? "incoming" : "outgoing");
390 if((incoming && i) || (!incoming && !i)){
391 if(debug) fprint(2, "peer %s: -> handshake\n", addr);
392 n = pack(buf, sizeof(buf), "*________**",
393 20, "\x13BitTorrent protocol",
394 sizeof(infohash), infohash,
395 sizeof(peerid), peerid);
396 if(write(fd, buf, n) != n)
399 if((incoming && !i) || (!incoming && i)){
400 n = 20 + 8 + sizeof(infohash);
401 if((n = readn(fd, buf, n)) != n)
403 if(memcmp(buf, "\x13BitTorrent protocol", 20))
405 if(memcmp(infohash, buf + 20 + 8, sizeof(infohash)))
407 if(debug) fprint(2, "peer %s: <- handshake\n", addr);
410 if(readn(fd, buf, sizeof(peerid)) != sizeof(peerid))
412 if(memcmp(peerid, buf, sizeof(peerid)) == 0)
414 if(debug) fprint(2, "peer %s: peerid %.*s\n", addr, sizeof(peerid), (char*)buf);
421 map = mallocz(nhavemap, 1);
422 told = malloc(nhavemap);
424 if(debug) fprint(2, "peer %s: -> bitfield %d\n", addr, nhavemap);
425 memmove(told, havemap, nhavemap);
426 n = pack(buf, sizeof(buf), "lb*", nhavemap+1, 0x05, nhavemap, told);
427 if(write(fd, buf, n) != n)
431 for(i=0; i<nhavemap; i++){
432 if(told[i] != havemap[i]){
433 for(x = i<<3, m = 0x80; m; m >>= 1, x++){
434 if((~havemap[i] | told[i] | map[i]) & m)
437 if(debug) fprint(2, "peer %s: -> have %d\n", addr, x);
438 n = pack(buf, sizeof(buf), "lbl", 1+4, 0x04, x);
439 if(write(fd, buf, n) != n)
443 if(!mewant && (map[i] & ~havemap[i])){
445 if(debug) fprint(2, "peer %s: -> interested\n", addr);
446 n = pack(buf, sizeof(buf), "lb", 1, 0x02);
447 if(write(fd, buf, n) != n)
451 if(!hechoking && mewant){
453 if(x >= 0 && pieces[x].brk < pieces[x].len)
455 else x = pickpiece(map);
458 l = pieces[x].len - o;
461 if(debug) fprint(2, "peer %s: -> request %d %d %d\n", addr, x, o, l);
462 n = pack(buf, sizeof(buf), "lblll", 1+4+4+4, 0x06, x, o, l);
463 if(write(fd, buf, n) != n)
468 if(mechoking && hewant){
470 if(debug) fprint(2, "peer %s: -> unchoke\n", addr);
471 n = pack(buf, sizeof(buf), "lb", 1, 0x01);
472 if(write(fd, buf, n) != n)
476 if(readn(fd, buf, 4) != 4)
478 unpack(buf, 4, "l", &n);
479 if(n < 0 || n > sizeof(buf))
483 if(readn(fd, buf, n) != n)
492 if(debug) fprint(2, "peer %s: <- choke\n", addr);
494 case 0x01: // Unchoke
496 if(debug) fprint(2, "peer %s: <- unchoke\n", addr);
498 case 0x02: // Interested
500 if(debug) fprint(2, "peer %s: <- interested\n", addr);
502 case 0x03: // Notinterested
504 if(debug) fprint(2, "peer %s: <- notinterested\n", addr);
506 case 0x04: // Have <piceindex>
507 if(unpack(p, n, "l", &x) < 0)
509 if(debug) fprint(2, "peer %s: <- have %d\n", addr, x);
510 if(x < 0 || x >= npieces)
512 map[x>>3] |= 0x80>>(x&7);
514 case 0x05: // Bitfield
515 if(debug) fprint(2, "peer %s: <- bitfield %d\n", addr, n);
520 case 0x06: // Request <index> <begin> <length>
521 if(unpack(p, n, "lll", &x, &o, &l) < 0)
523 if(debug) fprint(2, "peer %s: <- request %d %d %d\n", addr, x, o, l);
524 if(x < 0 || x >= npieces)
526 if(!hewant || mechoking || (~havemap[x>>3]&(0x80>>(x&7))))
528 if(debug) fprint(2, "peer %s: -> piece %d %d\n", addr, x, o);
532 if((l = rwpiece(0, x, buf + n, l, o)) <= 0)
534 n = pack(buf, sizeof(buf), "lbll", 1+4+4+l, 0x07, x, o);
536 if(write(fd, buf, n) != n)
542 case 0x07: // Piece <index> <begin> <block>
543 if(unpack(p, n, "ll", &x, &o) != 8)
550 if(debug) fprint(2, "peer %s: <- piece %d %d %d\n", addr, x, o, n);
551 if(x < 0 || x >= npieces)
553 if((pieces[x].brk != o) || (havemap[x>>3]&(0x80>>(x&7))))
555 if(rwpiece(1, x, p, n, o) == n){
556 if((pieces[x].brk = o+n) == pieces[x].len){
562 case 0x08: // Cancel <index> <begin> <length>
563 if(unpack(p, n, "lll", &x, &o, &l) < 0)
565 if(debug) fprint(2, "peer %s: <- cancel %d %d %d\n", addr, x, o, l);
567 case 0x09: // Port <port>
568 if(unpack(p, n, "l", &x) < 0)
570 if(debug) fprint(2, "peer %s: <- port %d\n", addr, x);
584 char addr[64], adir[40], ldir[40];
585 int afd, lfd, dfd, pid, nprocs;
590 for(port=6881; port<6890; port++){
591 snprint(addr, sizeof(addr), "tcp!*!%d", port);
592 if((afd = announce(addr, adir)) >= 0)
596 fprint(2, "announce: %r");
599 if(rfork(RFFDG|RFPROC|RFMEM))
602 if((lfd = listen(adir, ldir)) < 0){
603 fprint(2, "listen: %r");
606 while(nprocs >= SRVPROCS)
610 if(pid = rfork(RFFDG|RFPROC|RFMEM)){
616 if((dfd = accept(lfd, ldir)) < 0){
617 fprint(2, "accept: %r");
620 ni = getnetconninfo(ldir, dfd);
621 peer(dfd, 1, ni ? ni->raddr : "???");
622 if(ni) freenetconninfo(ni);
629 client(char *ip, char *port)
631 static Dict *peerqh, *peerqt;
632 static QLock peerslk;
638 if(ip == nil || port == nil)
641 d = mallocz(sizeof(*d) + 64, 1);
642 snprint(addr = d->str, 64, "tcp!%s!%s", ip, port);
644 if(dlook(peerqh, addr)){
649 d->len = strlen(addr);
652 /* enqueue to front */
653 if((d->next = peerqh) == nil)
656 if(nprocs >= CLIPROCS){
662 if(rfork(RFFDG|RFPROC|RFMEM|RFNOWAIT))
667 /* dequeue and put to tail */
669 if((peerqh = d->next) == nil)
683 if(debug) fprint(2, "client %s\n", addr);
684 if((fd = dial(addr, nil, nil, nil)) >= 0){
688 sleep(1000+nrand(5000));
693 hopen(char *url, ...)
695 int conn, ctlfd, fd, n;
699 snprint(buf, sizeof buf, "%s/clone", mntweb);
700 if((ctlfd = open(buf, ORDWR)) < 0)
702 if((n = read(ctlfd, buf, sizeof buf-1)) <= 0){
710 n = 4+vsnprint(buf+4, sizeof(buf)-4, url, arg);
712 if(write(ctlfd, buf, n) != n){
717 snprint(buf, sizeof buf, "%s/%d/body", mntweb, conn);
718 if((fd = open(buf, OREAD)) < 0)
725 webseed(Dict *w, File *f)
727 int fd, err, n, m, o, p, x, y;
733 if(w == nil || f == nil || finished())
735 if(rfork(RFPROC|RFMEM))
739 if(debug) fprint(2, "webseed %s %s\n", w->str, f->name);
740 s = strrchr(w->str, '/');
742 fd = hopen("%s%s", w->str, f->name);
744 fd = hopen("%s", w->str);
747 if(debug) fprint(2, "webseed %s %s: %r\n", w->str, f->name);
750 if((w = w->next) == w0)
758 while(len > 0 && !finished()){
762 if((n = read(fd, buf, m)) <= 0)
766 p = off - (vlong)x*blocksize;
773 m = pieces[x].len - p;
776 if((havemap[x>>3] & (0x80>>(x&7))) == 0)
777 rwpiece(1, x, buf+o, m, p);
787 werrstr("file corrupted");
792 havepiece(off / blocksize);
793 havepiece(f->off / blocksize);
799 clients4(uchar *p, int len)
801 char ip[16], port[6];
805 snprint(ip, sizeof(ip), "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
806 snprint(port, sizeof(port), "%d", p[4]<<8 | p[5]);
813 webtracker(char *url)
819 if(rfork(RFPROC|RFMEM))
821 if(debug) fprint(2, "webtracker %s\n", url);
823 event = "&event=started";
825 vlong up, down, left;
834 if((fd = hopen("%s?info_hash=%.*H&peer_id=%.*H&port=%d&"
835 "uploaded=%lld&downloaded=%lld&left=%lld&compact=1&no_peer_id=1%s",
836 url, sizeof(infohash), infohash, sizeof(peerid), peerid, port,
837 up, down, left, event)) >= 0){
843 } else if(debug) fprint(2, "tracker %s: %r\n", url);
844 if(l = dlook(d, "peers")){
846 clients4((uchar*)l->str, l->len);
847 else for(; l && l->typ == 'l'; l = l->next)
848 client(dstr(dlook(l->val, "ip")), dstr(dlook(l->val, "port")));
851 if(p = dstr(dlook(d, "interval")))
853 if(n < 10 | n > 60*60)
856 sleep(n * 1000 + nrand(5000));
861 udpaddr(char addr[64], int naddr, char *url)
866 if((url = strchr(url, ':')) == nil)
871 if(x = strchr(url, ':')){
875 if((x = strchr(url, '/')) == nil)
878 snprint(addr, naddr, "udp!%.*s!%d", (int)(x-url), url, port);
883 udptracker(char *url)
885 int fd, event, n, m, a, i;
886 int transid, interval;
891 if(udpaddr(addr, sizeof(addr), url) < 0)
893 if(rfork(RFPROC|RFMEM))
895 if(debug) fprint(2, "udptracker %s\n", addr);
900 if((fd = dial(addr, 0, 0, 0)) < 0)
905 n = pack(buf, sizeof(buf), "vll", 0x41727101980LL, 0, transid);
906 if(write(fd, buf, n) != n)
909 if((n = read(fd, buf, sizeof(buf))) <= 0)
911 if(unpack(buf, n, "llv", &a, &i, &connid) < 0)
913 if(a == 0 && i == transid)
921 n = pack(buf, sizeof(buf), "vll**vvvl____llw",
923 sizeof(infohash), infohash,
924 sizeof(peerid), peerid,
935 if(write(fd, buf, n) != n)
938 if((n = read(fd, buf, sizeof(buf))) <= 0)
940 if((m = unpack(buf, n, "lll________", &a, &i, &interval)) < 0)
942 if(a == 1 && i == transid){
943 clients4(buf+m, n - m);
952 if(interval < 10 | interval > 60*60)
954 sleep(interval * 1000 + nrand(5000));
961 static Dict *trackers;
962 static QLock trackerslk;
969 if(dlook(trackers, url)){
970 qunlock(&trackerslk);
974 d = mallocz(sizeof(*d) + n+1, 1);
982 qunlock(&trackerslk);
983 if(!cistrncmp(url, "udp:", 4))
993 s = va_arg(f->args, uchar*);
994 if(f->flags & FmtPrec)
997 e = s + strlen((char*)s);
999 if(fmtprint(f, *s && ((*s >= '0' && *s <= '9') ||
1000 (*s >= 'a' && *s <= 'z') ||
1001 (*s >= 'A' && *s <= 'Z') ||
1002 strchr(".-_~", *s)) ? "%c" : "%%%.2x", *s) < 0)
1008 mktorrent(int fd, Dict *alist, Dict *wlist)
1014 if((d = dirfstat(fd)) == nil)
1016 if(d->qid.type & QTDIR){
1018 werrstr("file is a directory");
1023 werrstr("empty file");
1027 for(blocksize = 256*1024;;blocksize<<=1){
1028 npieces = (d->length + blocksize-1) / blocksize;
1029 if(npieces <= 8*1024 || blocksize >= 2*1024*1024)
1034 * keys in dictionaries have to be ordered alphabetically
1036 print("d8:announce%ld:%s", strlen(alist->str), alist->str);
1038 print("13:announce-listl");
1039 print("l%ld:%se", strlen(alist->str), alist->str);
1040 for(alist = alist->next; alist; alist = alist->next)
1041 print("l%ld:%se", strlen(alist->str), alist->str);
1046 print("6:lengthi%llde", d->length);
1047 print("4:name%ld:%s", strlen(d->name), d->name);
1048 print("12:piece lengthi%de", blocksize);
1049 print("6:pieces%d:", npieces*sizeof(h));
1051 b = malloc(blocksize);
1052 while((n = readn(fd, b, blocksize)) > 0){
1054 if(write(1, h, sizeof(h)) != sizeof(h)){
1061 werrstr("read failed: %r");
1069 print("8:url-listl");
1070 for(; wlist; wlist = wlist->next)
1071 print("%ld:%s", strlen(wlist->str), wlist->str);
1074 print("8:url-list%ld:%s", strlen(wlist->str), wlist->str);
1087 if(access(s, AEXIST) == 0)
1089 for(p=strchr(s+1, '/'); p; p=strchr(p+1, '/')){
1091 if(access(s, AEXIST)){
1092 if((f = create(s, OREAD, DMDIR | 0777)) < 0){
1114 s += chartorune(&r, s);
1117 if((n + runelen(r)) >= l){
1121 n += runetochar(d + n, &r);
1124 return cleanname(d);
1128 catch(void *, char *msg)
1130 if(strstr(msg, "alarm"))
1132 postnote(PNGROUP, killgroup, "kill");
1139 fprint(2, "usage: %s [ -vsdpc ] [ -m mtpt ] [ -t tracker-url ] "
1140 "[ -w webseed-url ] [ file ]\n", argv0);
1145 scons(char *s, Dict *t)
1151 for(l = t; l; l = l->next)
1152 if(strcmp(l->str, s) == 0)
1154 l = mallocz(sizeof(*l) + strlen(s)+1, 1);
1161 main(int argc, char *argv[])
1163 int sflag, pflag, vflag, cflag, fd, i, n;
1164 Dict *alist, *wlist, *info, *torrent, *d, *l;
1169 fmtinstall('H', Hfmt);
1170 alist = wlist = nil;
1171 sflag = pflag = vflag = cflag = 0;
1174 mntweb = EARGF(usage());
1177 alist = scons(EARGF(usage()), alist);
1180 wlist = scons(EARGF(usage()), wlist);
1201 if((s = getenv("NPROC")) != 0){
1202 if((nproc = atoi(s)) <= 0)
1209 if((fd = open(*argv, OREAD)) < 0)
1210 sysfatal("open: %r");
1213 alist = scons(deftrack, alist);
1214 if(mktorrent(fd, alist, wlist) < 0)
1218 if((n = readall(fd, &p)) <= 0)
1219 sysfatal("read torrent: %r");
1220 bparse(p, p+n, &torrent);
1222 alist = scons(dstr(dlook(torrent, "announce")), alist);
1223 for(d = dlook(torrent, "announce-list"); d && d->typ == 'l'; d = d->next)
1224 for(l = d->val; l && l->typ == 'l'; l = l->next)
1225 alist = scons(dstr(l->val), alist);
1227 if(d = dlook(torrent, "url-list")){
1229 wlist = scons(dstr(d->val), wlist);
1230 else for(l = d; l && l->typ == 'l'; l = l->next)
1231 wlist = scons(dstr(l->val), wlist);
1232 /* make wlist into a ring */
1233 for(l = wlist; l && l->next; l = l->next)
1235 if(l) l->next = wlist;
1238 if(alist == nil && wlist == nil)
1239 sysfatal("no trackers or webseeds in torrent");
1241 if((d = info = dlook(torrent, "info")) == nil)
1242 sysfatal("no meta info in torrent");
1243 for(s = e = d->start; d && d->typ == 'd'; d = d->next)
1245 sha1((uchar*)s, e - s, (uchar*)infohash, nil);
1249 if(d = dlook(info, "files")){
1250 for(; d && d->typ == 'l'; d = d->next){
1253 if((s = dstr(dlook(d->val, "length"))) == nil)
1255 f = mallocz(sizeof(*f), 1);
1257 f->name = dstr(dlook(info, "name"));
1258 for(di = dlook(d->val, "path"); di && di->typ == 'l'; di = di->next)
1259 if(s = dstr(di->val))
1260 f->name = f->name ? smprint("%s/%s", f->name, s) : s;
1264 } else if(s = dstr(dlook(info, "length"))){
1265 f = mallocz(sizeof(*f), 1);
1267 f->name = dstr(dlook(info, "name"));
1271 for(f = files; f; f = f->next){
1272 if(f->name == nil || f->len <= 0)
1273 sysfatal("bogus file entry in meta info");
1274 s = fixnamedup(f->name);
1275 if(vflag) fprint(pflag ? 2 : 1, "%s\n", s);
1276 if((f->fd = open(s, ORDWR)) < 0){
1278 sysfatal("mkdirs: %r");
1279 if((f->fd = create(s, ORDWR, 0666)) < 0)
1280 sysfatal("create: %r");
1286 sysfatal("no files in torrent");
1288 if((s = dstr(dlook(info, "piece length"))) == nil)
1289 sysfatal("missing piece length in meta info");
1290 if((blocksize = atoi(s)) <= 0)
1291 sysfatal("bogus piece length in meta info");
1292 d = dlook(info, "pieces");
1293 if(d == nil || d->typ != 's' || d->len <= 0 || d->len % 20)
1294 sysfatal("bad or no pices in meta info");
1295 npieces = d->len / 20;
1296 pieces = mallocz(sizeof(Piece) * npieces, 1);
1297 nhavemap = (npieces+7) / 8;
1298 havemap = mallocz(nhavemap, 1);
1299 for(i = 0; i<npieces; i++){
1300 pieces[i].hash = (uchar*)d->str + i*20;
1302 pieces[i].len = len;
1304 pieces[i].len = blocksize;
1305 len -= pieces[i].len;
1306 stats.left += pieces[i].len;
1309 sysfatal("pieces do not match file length");
1311 for(i=0; i<nproc; i++){
1312 switch(rfork(RFPROC|RFMEM)){
1314 sysfatal("fork: %r");
1316 for(; i<npieces; i+=nproc)
1321 while(waitpid() >= 0)
1326 switch(i = rfork(RFPROC|RFMEM|RFNOTEG)){
1328 sysfatal("fork: %r");
1330 memmove(peerid, "-NF9001-", 8);
1331 for(i=8; i<sizeof(peerid); i++)
1332 peerid[i] = nrand(10)+'0';
1334 for(; alist; alist = alist->next)
1335 tracker(alist->str);
1336 for(f = files, l = wlist; f && l; f = f->next, l = l->next)
1338 while(waitpid() != -1)
1346 print("%d %d\n", nhavepieces, npieces);
1347 } while(!finished() || sflag);
1349 postnote(PNGROUP, killgroup, "kill");