Browse Source

hooks into scanning functions built

dynamic-accesslists
erdgeist 18 years ago
parent
commit
83e339bbe0
  1. 477
      opentracker.c

477
opentracker.c

@ -36,272 +36,267 @@ struct http_data {
int header_complete(struct http_data* r) int header_complete(struct http_data* r)
{ {
long i; long i;
long l = array_bytes(&r->r); long l = array_bytes(&r->r);
const char* c = array_start(&r->r); const char* c = array_start(&r->r);
for (i=0; i+1<l; ++i) for (i=0; i+1<l; ++i)
{ {
if (c[i]=='\n' && c[i+1]=='\n') if (c[i]=='\n' && c[i+1]=='\n')
return i+2; return i+2;
if (i+3<l && c[i]=='\r' && c[i+1]=='\n' && c[i+2]=='\r' && c[i+3]=='\n') if (i+3<l && c[i]=='\r' && c[i+1]=='\n' && c[i+2]=='\r' && c[i+3]=='\n')
return i+4; return i+4;
} }
return 0; return 0;
} }
void httperror(struct http_data* r,const char* title,const char* message) void httperror(struct http_data* r,const char* title,const char* message)
{ {
char* c; char* c;
c=r->hdrbuf=(char*)malloc(strlen(message)+strlen(title)+200); c=r->hdrbuf=(char*)malloc(strlen(message)+strlen(title)+200);
if (!c) if (!c)
{ {
r->hdrbuf="HTTP/1.0 500 internal error\r\nContent-Type: text/plain\r\nConnection: close\r\n\r\nout of memory\n"; r->hdrbuf="HTTP/1.0 500 internal error\r\nContent-Type: text/plain\r\nConnection: close\r\n\r\nout of memory\n";
r->hlen=strlen(r->hdrbuf); r->hlen=strlen(r->hdrbuf);
} }
else else
{ {
c+=fmt_str(c,"HTTP/1.0 "); c+=fmt_str(c,"HTTP/1.0 ");
c+=fmt_str(c,title); c+=fmt_str(c,title);
c+=fmt_str(c,"\r\nContent-Type: text/html\r\nConnection: close\r\nContent-Length: "); c+=fmt_str(c,"\r\nContent-Type: text/html\r\nConnection: close\r\nContent-Length: ");
c+=fmt_ulong(c,strlen(message)+strlen(title)+16-4); c+=fmt_ulong(c,strlen(message)+strlen(title)+16-4);
c+=fmt_str(c,"\r\n\r\n<title>"); c+=fmt_str(c,"\r\n\r\n<title>");
c+=fmt_str(c,title+4); c+=fmt_str(c,title+4);
c+=fmt_str(c,"</title>\n"); c+=fmt_str(c,"</title>\n");
r->hlen=c - r->hdrbuf; r->hlen=c - r->hdrbuf;
} }
iob_addbuf(&r->iob,r->hdrbuf,r->hlen); iob_addbuf(&r->iob,r->hdrbuf,r->hlen);
} }
// bestimmten http parameter auslesen und adresse zurueckgeben // bestimmten http parameter auslesen und adresse zurueckgeben
const char* http_header(struct http_data* r,const char* h) const char* http_header(struct http_data* r,const char* h)
{ {
long i; long i;
long l = array_bytes(&r->r); long l = array_bytes(&r->r);
long sl = strlen(h); long sl = strlen(h);
const char* c = array_start(&r->r); const char* c = array_start(&r->r);
for (i=0; i+sl+2<l; ++i) for (i=0; i+sl+2<l; ++i)
{ {
if (c[i]=='\n' && case_equalb(c+i+1,sl,h) && c[i+sl+1]==':') if (c[i]=='\n' && case_equalb(c+i+1,sl,h) && c[i+sl+1]==':')
{ {
c+=i+sl+1; c+=i+sl+1;
if (*c==' ' || *c=='\t') ++c; if (*c==' ' || *c=='\t') ++c;
return c; return c;
} }
return 0; return 0;
} }
} }
void httpresponse(struct http_data* h,int64 s) void httpresponse(struct http_data* h,int64 s)
{ {
char* c; char* c;
array_cat0(&h->r); array_cat0(&h->r);
c = array_start(&h->r); c = array_start(&h->r);
if (byte_diff(c,4,"GET ")) if (byte_diff(c,4,"GET "))
{ {
e400: e400:
httperror(h,"400 Invalid Request","This server only understands GET."); httperror(h,"400 Invalid Request","This server only understands GET.");
} }
else else
{ {
char *d; char *d;
int64 fd; int64 fd;
struct stat s; struct stat s;
// expect 'GET /uri?nnbjhg HTTP/1.*' // expect 'GET /uri?nnbjhg HTTP/1.*'
c+=4; c+=4;
for (d=c; *d!=' '&&*d!='\t'&&*d!='\n'&&*d!='\r'; ++d) ; for (d=c; *d!=' '&&*d!='\t'&&*d!='\n'&&*d!='\r'; ++d) ;
if (*d!=' ') goto e400; if (*d!=' ') goto e400;
*d=0; *d=0;
if (c[0]!='/') goto e404; if (c[0]!='/') goto e404;
while (c[1]=='/') ++c; while (c[1]=='/') ++c;
if (!byte_diff(c,9,"announce?")) data = c;
{ switch( scan_urlencoded_query( &c, data, SCAN_PATH ) ) {
// info_hash, left, port, numwant, compact case 6: /* scrape ? */
struct ot_peer peer; if (!byte_diff(c,6,"scrape"))
goto 404;
break;
case 9:
if( !byte_diff(c,8,"announce"))
goto 404;
else {
// info_hash, left, port, numwant, compact
struct ot_peer peer;
ot_hash hash; ot_hash hash;
byte_copy( peer.ip, h->ip, 4); byte_copy( peer.ip, h->ip, 4);
peer.port = 6881; peer.port = 6881;
c+=9; while( NOCHAMSCANNEN ) {
while( *c!=' ' ) { data = c;
if(!byte_diff(c,10,"info_hash=")) switch( scan_urlencoded_query( &c, data, SCAN_SEARCHPATH_PARAM ) ) {
{ case -1: /* error */
size_t destlen = 20; httperror(h,"404 Not Found","No such file or directory.");
// String is expected to be URL encoded, so expect goto e404;
// (%[0-9A-F][0-9A-F]){20} case 4:
if(!byte_diff(c,4,"port"))
int s = scan_urlencoded( c+10, hash, &destlen); /* scan int */
if( c[s+10] != '&' ) else if(!byte_diff(c,4,"left"))
goto e_parse; /* scan int */
break;
} case 7:
else if(!byte_diff(c,8,"numwant=")) if(!byte_diff(c,7,"numwant"))
{ /* scan int */
else if(!byte_diff(c,7,"compact"))
} /* scan flag */
else if(!byte_diff(c,8,"compact=")) break;
{ case 9: /* info_hash */
if(!byte_diff(c,9,"info_hash"))
} /* scan 20 bytes */
else if(!byte_diff(c,5,"port=")) break;
{ }
}
} break;
else if(!byte_diff(c,5,"left=")) default: /* neither scrape nor announce */
{ httperror(h,"404 Not Found","No such file or directory.");
goto e404;
} }
while( *++c!=' ' && *c!='&'); c=h->hdrbuf=(char*)malloc(500);
} c+=fmt_str(c,"HTTP/1.1 Coming Up\r\nContent-Type: text/plain");
} c+=fmt_str(c,"\r\nContent-Length: ");
else if (!byte_diff(c,7,"scrape?")) /* ANSWER SIZE*/
{ c+=fmt_ulonglong(c,s.st_size);
c+=fmt_str(c,"\r\nLast-Modified: ");
} /* MODIFY DATE */
else c+=fmt_httpdate(c,s.st_mtime);
{ c+=fmt_str(c,"\r\nConnection: close\r\n\r\n");
httperror(h,"404 Not Found","No such file or directory."); iob_addbuf(&h->iob,h->hdrbuf,c - h->hdrbuf);
goto e404; iob_addbuf(&h->iob,tracker_answer, tzracker_answer_size);
} }
c=h->hdrbuf=(char*)malloc(500);
c+=fmt_str(c,"HTTP/1.1 Coming Up\r\nContent-Type: text/plain");
c+=fmt_str(c,"\r\nContent-Length: ");
/* ANSWER SIZE*/
c+=fmt_ulonglong(c,s.st_size);
c+=fmt_str(c,"\r\nLast-Modified: ");
/* MODIFY DATE */
c+=fmt_httpdate(c,s.st_mtime);
c+=fmt_str(c,"\r\nConnection: close\r\n\r\n");
iob_addbuf(&h->iob,h->hdrbuf,c - h->hdrbuf);
iob_addbuf(&h->iob,tracker_answer, tzracker_answer_size);
}
e404: e404:
io_dontwantread(s); io_dontwantread(s);
io_wantwrite(s); io_wantwrite(s);
} }
int main() int main()
{ {
int s=socket_tcp6(); int s=socket_tcp6();
uint32 scope_id; uint32 scope_id;
char ip[16]; char ip[16];
uint16 port; uint16 port;
if (socket_bind6_reuse(s,V6any,8000,0)==-1) if (socket_bind6_reuse(s,V6any,8000,0)==-1)
panic("socket_bind6_reuse"); panic("socket_bind6_reuse");
if (socket_listen(s,16)==-1) if (socket_listen(s,16)==-1)
panic("socket_listen"); panic("socket_listen");
if (!io_fd(s)) if (!io_fd(s))
panic("io_fd"); panic("io_fd");
io_wantread(s); io_wantread(s);
for (;;) for (;;)
{ {
int64 i; int64 i;
io_wait(); io_wait();
while ((i=io_canread())!=-1) while ((i=io_canread())!=-1)
{ {
if (i==s) // ist es der serversocket? if (i==s) // ist es der serversocket?
{ {
int n; int n;
while ((n=socket_accept6(s,ip,&port,&scope_id))!=-1) while ((n=socket_accept6(s,ip,&port,&scope_id))!=-1)
{ {
if (io_fd(n)) if (io_fd(n))
{ {
struct http_data* h=(struct http_data*)malloc(sizeof(struct http_data)); struct http_data* h=(struct http_data*)malloc(sizeof(struct http_data));
io_wantread(n); io_wantread(n);
if (h) if (h)
{ {
byte_zero(h,sizeof(struct http_data)); byte_zero(h,sizeof(struct http_data));
byte_copy(h->ip,ip,sizeof(ip)); byte_copy(h->ip,ip,sizeof(ip));
io_setcookie(n,h); io_setcookie(n,h);
} else } else
io_close(n); io_close(n);
} else } else
io_close(n); io_close(n);
buffer_putnlflush(buffer_2); buffer_putnlflush(buffer_2);
} }
if (errno==EAGAIN) if (errno==EAGAIN)
io_eagain(s); io_eagain(s);
else else
carp("socket_accept6"); carp("socket_accept6");
} }
else else
{ {
char buf[8192]; char buf[8192];
struct http_data* h=io_getcookie(i); struct http_data* h=io_getcookie(i);
int l=io_tryread(i,buf,sizeof buf); int l=io_tryread(i,buf,sizeof buf);
if (l<=0) if (l<=0)
{ {
if (h) if (h)
{ {
array_reset(&h->r); array_reset(&h->r);
iob_reset(&h->iob); iob_reset(&h->iob);
free(h->hdrbuf); h->hdrbuf=0; free(h->hdrbuf); h->hdrbuf=0;
} }
io_close(i); io_close(i);
} }
else else
{ {
array_catb(&h->r,buf,l); array_catb(&h->r,buf,l);
if (array_failed(&h->r)) if (array_failed(&h->r))
{ {
httperror(h,"500 Server Error","request too long."); httperror(h,"500 Server Error","request too long.");
emerge: emerge:
io_dontwantread(i); io_dontwantread(i);
io_wantwrite(i); io_wantwrite(i);
} }
else if (array_bytes(&h->r)>8192) else if (array_bytes(&h->r)>8192)
{ {
httperror(h,"500 request too long","You sent too much headers"); httperror(h,"500 request too long","You sent too much headers");
goto emerge; goto emerge;
} }
else if ((l=header_complete(h))) else if ((l=header_complete(h)))
{ {
httpresponse(h,i); httpresponse(h,i);
} }
} }
} }
} }
while ((i=io_canwrite())!=-1) while ((i=io_canwrite())!=-1)
{ {
struct http_data* h=io_getcookie(i); struct http_data* h=io_getcookie(i);
int64 r=iob_send(i,&h->iob); int64 r=iob_send(i,&h->iob);
if (r==-1) if (r==-1)
io_eagain(i); io_eagain(i);
else else
if (r<=0) if (r<=0)
{ {
array_trunc(&h->r); array_trunc(&h->r);
iob_reset(&h->iob); iob_reset(&h->iob);
free(h->hdrbuf); h->hdrbuf=0; free(h->hdrbuf); h->hdrbuf=0;
io_close(i); io_close(i);
} }
} }
} }
return 0; return 0;
} }

Loading…
Cancel
Save