};
int debug;
+int nproc = 1;
int killgroup = -1;
int port = 6881;
char *deftrack = "http://exodus.desync.com/announce";
char *mntweb = "/mnt/web";
+char *useragent = "torrent";
uchar infohash[20];
uchar peerid[20];
int blocksize;
int n, m;
File *f;
- if(len <= 0 || poff >= pieces[index].len)
+ if(len <= 0 || poff < 0 || poff >= pieces[index].len)
return 0;
if(len+poff > pieces[index].len)
len = pieces[index].len - poff;
- off = (vlong)index * blocksize;
+ off = (vlong)index * (vlong)blocksize;
off += poff;
for(f = files; f; f = f->next)
if((f->off+f->len) > off)
}
if(!hechoking && mewant){
x = workpiece;
- if(x >= 0 && pieces[x].brk < pieces[x].len)
- {}
- else x = pickpiece(map);
+ if(x < 0 || (havemap[x>>3]&(0x80>>(x&7))) != 0)
+ x = pickpiece(map);
if(x >= 0){
+ workpiece = x;
o = pieces[x].brk;
+ if(o < 0 || o >= pieces[x].len){
+ pieces[x].brk = 0;
+ o = 0;
+ }
l = pieces[x].len - o;
if(l > MAXIO)
l = MAXIO;
n = pack(buf, sizeof(buf), "lblll", 1+4+4+4, 0x06, x, o, l);
if(write(fd, buf, n) != n)
goto Out;
- workpiece = x;
}
}
if(mechoking && hewant){
if(debug) fprint(2, "peer %s: <- piece %d %d %d\n", addr, x, o, n);
if(x < 0 || x >= npieces)
continue;
- if((pieces[x].brk != o) || (havemap[x>>3]&(0x80>>(x&7))))
+ if((havemap[x>>3]&(0x80>>(x&7))) != 0)
continue;
- if(rwpiece(1, x, p, n, o) == n){
- if((pieces[x].brk = o+n) == pieces[x].len){
- if(!havepiece(x))
- pieces[x].brk = 0;
- }
+ if(o < 0 || o >= pieces[x].len)
+ continue;
+ if(o+n > pieces[x].len)
+ n = o - pieces[x].len;
+ if((o > pieces[x].brk) || (o+n <= pieces[x].brk))
+ continue;
+ n = rwpiece(1, x, p, n, o);
+ if(n <= 0)
+ continue;
+ pieces[x].brk = o+n;
+ if(o+n >= pieces[x].len && !havepiece(x)){
+ /* backoff from this piece for a while */
+ if(x == workpiece)
+ workpiece = -1;
}
break;
case 0x08: // Cancel <index> <begin> <length>
void
client(char *ip, char *port)
{
- static Dict *peers, *peerqh, *peerqt;
+ static Dict *peerqh, *peerqt;
static QLock peerslk;
static int nprocs;
- int try, fd;
char *addr;
Dict *d;
+ int fd;
if(ip == nil || port == nil)
return;
d = mallocz(sizeof(*d) + 64, 1);
snprint(addr = d->str, 64, "tcp!%s!%s", ip, port);
qlock(&peerslk);
- if(dlook(peers, addr) || dlook(peerqh, addr)){
+ if(dlook(peerqh, addr)){
qunlock(&peerslk);
free(d);
return;
d->len = strlen(addr);
d->typ = 'd';
d->val = d;
- d->next = nil;
- if(peerqt == nil)
- peerqh = d;
- else
- peerqt->next = d;
- peerqt = d;
+ /* enqueue to front */
+ if((d->next = peerqh) == nil)
+ peerqt = d;
+ peerqh = d;
if(nprocs >= CLIPROCS){
qunlock(&peerslk);
return;
}
nprocs++;
qunlock(&peerslk);
-
if(rfork(RFFDG|RFPROC|RFMEM|RFNOWAIT))
return;
+
for(;;){
qlock(&peerslk);
+ /* dequeue and put to tail */
if(d = peerqh){
if((peerqh = d->next) == nil)
peerqt = nil;
- d->next = peers;
- peers = d;
+ d->next = nil;
+ if(peerqt)
+ peerqt->next = d;
+ else
+ peerqh = d;
+ peerqt = d;
} else
nprocs--;
qunlock(&peerslk);
exits(0);
addr = d->str;
if(debug) fprint(2, "client %s\n", addr);
- for(try = 0; try < 5; try++){
- if((fd = dial(addr, nil, nil, nil)) >= 0){
- if(!peer(fd, 0, addr))
- break;
- close(fd);
- }
- sleep((1000<<try)+nrand(5000));
+ if((fd = dial(addr, nil, nil, nil)) >= 0){
+ peer(fd, 0, addr);
+ close(fd);
}
+ sleep(1000+nrand(5000));
}
}
close(ctlfd);
return -1;
}
+ if(useragent != nil && useragent[0] != '\0'){
+ n = snprint(buf, sizeof buf, "useragent %s", useragent);
+ write(ctlfd, buf, n);
+ }
snprint(buf, sizeof buf, "%s/%d/body", mntweb, conn);
if((fd = open(buf, OREAD)) < 0)
goto ErrOut;
bparse(p, p+n, &d);
free(p);
} else if(debug) fprint(2, "tracker %s: %r\n", url);
+ /* check errors and warnings */
+ if(p = dstr(dlook(d, "failure reason"))) {
+ if(debug)
+ fprint(2, "tracker failure: %s\n", p);
+ exits(0);
+ }
+ if(p = dstr(dlook(d, "warning message")))
+ if(debug)
+ fprint(2, "tracker warning: %s\n", p);
if(l = dlook(d, "peers")){
if(l->typ == 's')
clients4((uchar*)l->str, l->len);
werrstr("empty file");
return -1;
}
- npieces = 1;
for(blocksize = 256*1024;;blocksize<<=1){
npieces = (d->length + blocksize-1) / blocksize;
if(npieces <= 8*1024 || blocksize >= 2*1024*1024)
usage(void)
{
fprint(2, "usage: %s [ -vsdpc ] [ -m mtpt ] [ -t tracker-url ] "
- "[ -w webseed-url ] [ file ]\n", argv0);
+ "[ -w webseed-url ] [ -i peerid ] [ -A useragent ] [ file ]\n", argv0);
exits("usage");
}
case 'd':
debug++;
break;
+ case 'i':
+ strncpy((char*)peerid, EARGF(usage()), sizeof(peerid));
+ break;
+ case 'A':
+ useragent = EARGF(usage());
+ break;
default:
usage();
} ARGEND;
+ if((s = getenv("NPROC")) != 0){
+ if((nproc = atoi(s)) <= 0)
+ nproc = 1;
+ free(s);
+ }
+
fd = 0;
if(*argv)
if((fd = open(*argv, OREAD)) < 0)
if(len)
sysfatal("pieces do not match file length");
- for(i = 0; i<npieces; i++)
- havepiece(i);
+ for(i=0; i<nproc; i++){
+ switch(rfork(RFPROC|RFMEM)){
+ case -1:
+ sysfatal("fork: %r");
+ case 0:
+ for(; i<npieces; i+=nproc)
+ havepiece(i);
+ exits(0);
+ }
+ }
+ while(waitpid() >= 0)
+ ;
- srand(time(0));
+ srand(truerand());
atnotify(catch, 1);
switch(i = rfork(RFPROC|RFMEM|RFNOTEG)){
case -1:
sysfatal("fork: %r");
case 0:
- memmove(peerid, "-NF9001-", 8);
- for(i=8; i<sizeof(peerid); i++)
+ if(peerid[0] == 0)
+ strncpy((char*)peerid, "-NF9001-", 9);
+ for(i=sizeof(peerid)-1; i >= 0 && peerid[i] == 0; i--)
peerid[i] = nrand(10)+'0';
server();
for(; alist; alist = alist->next)