ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/cvsroot/Coro/myhttpd/httpd.pl
Revision: 1.53
Committed: Fri Nov 30 05:14:12 2001 UTC (22 years, 7 months ago) by root
Content type: text/plain
Branch: MAIN
Changes since 1.52: +1 -0 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 root 1.1 use Coro;
2     use Coro::Semaphore;
3     use Coro::Event;
4     use Coro::Socket;
5 root 1.44 use Coro::Signal;
6 root 1.1
7 root 1.32 use HTTP::Date;
8 root 1.47 use POSIX ();
9 root 1.32
10 root 1.1 no utf8;
11     use bytes;
12    
13     # at least on my machine, this thingy serves files
14     # quite a bit faster than apache, ;)
15     # and quite a bit slower than thttpd :(
16    
17     $SIG{PIPE} = 'IGNORE';
18 root 1.27
19     our $accesslog;
20    
21     if ($ACCESS_LOG) {
22     use IO::Handle;
23     open $accesslog, ">>$ACCESS_LOG"
24     or die "$ACCESS_LOG: $!";
25     $accesslog->autoflush(1);
26     }
27    
28 root 1.1 sub slog {
29     my $level = shift;
30     my $format = shift;
31     printf "---: $format\n", @_;
32     }
33    
34 root 1.32 our $connections = new Coro::Semaphore $MAX_CONNECTS || 250;
35 root 1.44 our $httpevent = new Coro::Signal;
36 root 1.34
37     our $wait_factor = 0.95;
38    
39 root 1.52 our $queue_small = new transferqueue $MAX_TRANSFERS_SMALL;
40     our $queue_large = new transferqueue $MAX_TRANSFERS_LARGE;
41     our $queue_index = new transferqueue 5;
42 root 1.1
43 root 1.6 my @newcons;
44 root 1.1 my @pool;
45    
46 root 1.2 # one "execution thread"
47 root 1.1 sub handler {
48     while () {
49 root 1.38 if (@newcons) {
50 root 1.1 eval {
51 root 1.38 conn->new(@{pop @newcons})->handle;
52 root 1.1 };
53     slog 1, "$@" if $@ && !ref $@;
54     $connections->up;
55     } else {
56     last if @pool >= $MAX_POOL;
57     push @pool, $Coro::current;
58     schedule;
59     }
60     }
61     }
62    
63 root 1.40 sub listen_on {
64     my $listen = $_[0];
65    
66     push @listen_sockets, $listen;
67    
68     # the "main thread"
69     async {
70     slog 1, "accepting connections";
71     while () {
72     $connections->down;
73     push @newcons, [$listen->accept];
74     #slog 3, "accepted @$connections ".scalar(@pool);
75     if (@pool) {
76     (pop @pool)->ready;
77     } else {
78     async \&handler;
79     }
80    
81     }
82     };
83     }
84    
85 root 1.4 my $http_port = new Coro::Socket
86     LocalAddr => $SERVER_HOST,
87     LocalPort => $SERVER_PORT,
88     ReuseAddr => 1,
89 root 1.13 Listen => 50,
90 root 1.4 or die "unable to start server";
91    
92 root 1.40 listen_on $http_port;
93    
94 root 1.41 if ($SERVER_PORT2) {
95     my $http_port = new Coro::Socket
96     LocalAddr => $SERVER_HOST,
97     LocalPort => $SERVER_PORT2,
98     ReuseAddr => 1,
99     Listen => 50,
100     or die "unable to start server";
101 root 1.40
102 root 1.41 listen_on $http_port;
103     }
104 root 1.4
105 root 1.32 our $NOW;
106     our $HTTP_NOW;
107    
108     Event->timer(interval => 1, hard => 1, cb => sub {
109     $NOW = time;
110     $HTTP_NOW = time2str $NOW;
111 root 1.34 })->now;
112 root 1.1
113     package conn;
114    
115     use Socket;
116     use HTTP::Date;
117 root 1.2 use Convert::Scalar 'weaken';
118 root 1.16 use Linux::AIO;
119    
120     Linux::AIO::min_parallel $::AIO_PARALLEL;
121    
122     Event->io(fd => Linux::AIO::poll_fileno,
123 root 1.17 poll => 'r', async => 1,
124 root 1.21 cb => \&Linux::AIO::poll_cb);
125 root 1.16
126 root 1.26 our %conn; # $conn{ip}{self} => connobj
127     our %uri; # $uri{ip}{uri}{self}
128 root 1.3 our %blocked;
129 root 1.9 our %mimetype;
130    
131     sub read_mimetypes {
132     local *M;
133 root 1.10 if (open M, "<mime_types") {
134 root 1.9 while (<M>) {
135     if (/^([^#]\S+)\t+(\S+)$/) {
136     $mimetype{lc $1} = $2;
137     }
138     }
139     } else {
140 root 1.10 print "cannot open mime_types\n";
141 root 1.9 }
142     }
143 root 1.1
144 root 1.10 read_mimetypes;
145    
146 root 1.1 sub new {
147     my $class = shift;
148 root 1.42 my $fh = shift;
149 root 1.6 my $peername = shift;
150 root 1.2 my $self = bless { fh => $fh }, $class;
151 root 1.6 my (undef, $iaddr) = unpack_sockaddr_in $peername
152     or $self->err(500, "unable to decode peername");
153 root 1.7
154 root 1.50 $self->{remote_addr} =
155     $self->{remote_id} = inet_ntoa $iaddr;
156 root 1.11 $self->{time} = $::NOW;
157 root 1.2
158 root 1.49 weaken ($Coro::current->{conn} = $self);
159    
160 root 1.13 $::conns++;
161 root 1.53 $::maxconns = $::conns if $::conns > $::maxconns;
162 root 1.13
163 root 1.2 $self;
164     }
165    
166     sub DESTROY {
167     my $self = shift;
168 root 1.13 $::conns--;
169 root 1.19 $self->eoconn;
170     }
171    
172     # end of connection
173     sub eoconn {
174 root 1.26 my $self = shift;
175 root 1.36
176     # clean up hints
177     delete $conn{$self->{remote_id}}{$self*1};
178     delete $uri{$self->{remote_id}}{$self->{uri}}{$self*1};
179 root 1.44
180     $httpevent->broadcast;
181 root 1.1 }
182    
183     sub slog {
184 root 1.4 my $self = shift;
185 root 1.50 main::slog($_[0], "$self->{remote_id}> $_[1]");
186 root 1.1 }
187    
188 root 1.4 sub response {
189 root 1.1 my ($self, $code, $msg, $hdr, $content) = @_;
190 root 1.17 my $res = "HTTP/1.1 $code $msg\015\012";
191 root 1.1
192 root 1.46 if (exists $hdr->{Connection}) {
193     if ($hdr->{Connection} =~ /close/) {
194     $self->{h}{connection} = "close"
195     }
196     } else {
197     if ($self->{version} < 1.1) {
198     if ($self->{h}{connection} =~ /keep-alive/i) {
199     $hdr->{Connection} = "Keep-Alive";
200     } else {
201     $self->{h}{connection} = "close"
202     }
203     }
204     }
205 root 1.28
206 root 1.32 $res .= "Date: $HTTP_NOW\015\012";
207 root 1.1
208     while (my ($h, $v) = each %$hdr) {
209     $res .= "$h: $v\015\012"
210     }
211 root 1.10 $res .= "\015\012";
212 root 1.4
213 root 1.13 $res .= $content if defined $content and $self->{method} ne "HEAD";
214 root 1.1
215 root 1.47 my $log = (POSIX::strftime "%Y-%m-%d %H:%M:%S", gmtime $NOW).
216 root 1.50 " $self->{remote_id} \"$self->{uri}\" $code ".$hdr->{"Content-Length"}.
217     " \"$self->{h}{referer}\"\n";
218 root 1.27
219     print $accesslog $log if $accesslog;
220     print STDERR $log;
221 root 1.2
222 root 1.11 $self->{written} +=
223     print {$self->{fh}} $res;
224 root 1.1 }
225    
226     sub err {
227     my $self = shift;
228     my ($code, $msg, $hdr, $content) = @_;
229    
230     unless (defined $content) {
231 root 1.35 $content = "$code $msg\n";
232 root 1.1 $hdr->{"Content-Type"} = "text/plain";
233     $hdr->{"Content-Length"} = length $content;
234     }
235 root 1.17 $hdr->{"Connection"} = "close";
236 root 1.1
237 root 1.4 $self->response($code, $msg, $hdr, $content);
238 root 1.1
239     die bless {}, err::;
240     }
241    
242     sub handle {
243     my $self = shift;
244     my $fh = $self->{fh};
245    
246 root 1.29 my $host;
247    
248 root 1.17 $fh->timeout($::REQ_TIMEOUT);
249     while() {
250     $self->{reqs}++;
251 root 1.1
252     # read request and parse first line
253     my $req = $fh->readline("\015\012\015\012");
254    
255 root 1.17 unless (defined $req) {
256     if (exists $self->{version}) {
257     last;
258     } else {
259     $self->err(408, "request timeout");
260     }
261     }
262    
263     $self->{h} = {};
264 root 1.1
265 root 1.17 $fh->timeout($::RES_TIMEOUT);
266 root 1.3
267 root 1.1 $req =~ /^(?:\015\012)?
268     (GET|HEAD) \040+
269     ([^\040]+) \040+
270     HTTP\/([0-9]+\.[0-9]+)
271     \015\012/gx
272 root 1.14 or $self->err(405, "method not allowed", { Allow => "GET,HEAD" });
273 root 1.1
274     $self->{method} = $1;
275     $self->{uri} = $2;
276 root 1.17 $self->{version} = $3;
277    
278 root 1.20 $3 =~ /^1\./
279 root 1.17 or $self->err(506, "http protocol version $3 not supported");
280 root 1.1
281     # parse headers
282     {
283     my (%hdr, $h, $v);
284    
285     $hdr{lc $1} .= ",$2"
286     while $req =~ /\G
287     ([^:\000-\040]+):
288     [\008\040]*
289     ((?: [^\015\012]+ | \015\012[\008\040] )*)
290     \015\012
291     /gxc;
292    
293     $req =~ /\G\015\012$/
294     or $self->err(400, "bad request");
295    
296     $self->{h}{$h} = substr $v, 1
297     while ($h, $v) = each %hdr;
298     }
299    
300 root 1.36 # remote id should be unique per user
301     my $id = $self->{remote_addr};
302    
303     if (exists $self->{h}{"client-ip"}) {
304     $id .= "[".$self->{h}{"client-ip"}."]";
305     } elsif (exists $self->{h}{"x-forwarded-for"}) {
306     $id .= "[".$self->{h}{"x-forwarded-for"}."]";
307     }
308    
309     $self->{remote_id} = $id;
310    
311     if ($blocked{$id}) {
312     $self->err_blocked($blocked{$id})
313     if $blocked{$id} > $::NOW;
314    
315     delete $blocked{$id};
316     }
317    
318     if (%{$conn{$id}} >= $::MAX_CONN_IP) {
319 root 1.44 my $delay = $::PER_TIMEOUT + $::NOW + 15;
320 root 1.36 while (%{$conn{$id}} >= $::MAX_CONN_IP) {
321 root 1.44 if ($delay < $::NOW) {
322 root 1.36 $self->slog(2, "blocked ip $id");
323     $self->err_blocked;
324     } else {
325 root 1.44 $httpevent->wait;
326 root 1.36 }
327     }
328     }
329    
330 root 1.29 # find out server name and port
331     if ($self->{uri} =~ s/^http:\/\/([^\/?#]*)//i) {
332     $host = $1;
333     } else {
334     $host = $self->{h}{host};
335     }
336    
337     if (defined $host) {
338     $self->{server_port} = $host =~ s/:([0-9]+)$// ? $1 : 80;
339     } else {
340     ($self->{server_port}, $host)
341 root 1.43 = unpack_sockaddr_in $self->{fh}->sockname
342 root 1.29 or $self->err(500, "unable to get socket name");
343     $host = inet_ntoa $host;
344     }
345    
346     $self->{server_name} = $host;
347    
348 root 1.36 # enter ourselves into various lists
349     weaken ($conn{$id}{$self*1} = $self);
350     weaken ($uri{$id}{$self->{uri}}{$self*1} = $self);
351 root 1.1
352 root 1.24 eval {
353     $self->map_uri;
354     $self->respond;
355     };
356    
357 root 1.26 $self->eoconn;
358    
359 root 1.24 die if $@ && !ref $@;
360 root 1.17
361 root 1.46 last if $self->{h}{connection} =~ /close/;
362 root 1.17
363 root 1.44 $httpevent->broadcast;
364    
365 root 1.17 $fh->timeout($::PER_TIMEOUT);
366     }
367 root 1.1 }
368    
369     # uri => path mapping
370     sub map_uri {
371     my $self = shift;
372 root 1.29 my $host = $self->{server_name};
373 root 1.1 my $uri = $self->{uri};
374    
375     # some massaging, also makes it more secure
376     $uri =~ s/%([0-9a-fA-F][0-9a-fA-F])/chr hex $1/ge;
377     $uri =~ s%//+%/%g;
378     $uri =~ s%/\.(?=/|$)%%g;
379     1 while $uri =~ s%/[^/]+/\.\.(?=/|$)%%;
380    
381     $uri =~ m%^/?\.\.(?=/|$)%
382     and $self->err(400, "bad request");
383    
384     $self->{name} = $uri;
385    
386     # now do the path mapping
387     $self->{path} = "$::DOCROOT/$host$uri";
388 root 1.7
389     $self->access_check;
390 root 1.1 }
391    
392     sub _cgi {
393     my $self = shift;
394     my $path = shift;
395     my $fh;
396    
397     # no two-way xxx supported
398     if (0 == fork) {
399     open STDOUT, ">&".fileno($self->{fh});
400     if (chdir $::DOCROOT) {
401     $ENV{SERVER_SOFTWARE} = "thttpd-myhttpd"; # we are thttpd-alike
402 root 1.29 $ENV{HTTP_HOST} = $self->{server_name};
403     $ENV{HTTP_PORT} = $self->{server_port};
404 root 1.1 $ENV{SCRIPT_NAME} = $self->{name};
405 root 1.10 exec $path;
406 root 1.1 }
407     Coro::State::_exit(0);
408     } else {
409 root 1.29 die;
410 root 1.1 }
411     }
412    
413 root 1.29 sub server_hostport {
414     $_[0]{server_port} == 80
415     ? $_[0]{server_name}
416     : "$_[0]{server_name}:$_[0]{server_port}";
417     }
418    
419 root 1.1 sub respond {
420     my $self = shift;
421     my $path = $self->{path};
422    
423 root 1.49 if ($self->{name} =~ s%^/internal/([^/]+)%%) {
424     if ($::internal{$1}) {
425     $::internal{$1}->($self);
426     } else {
427     $self->err(404, "not found");
428     }
429     } else {
430 root 1.1
431 root 1.49 stat $path
432     or $self->err(404, "not found");
433 root 1.10
434 root 1.49 $self->{stat} = [stat _];
435 root 1.1
436 root 1.49 # idiotic netscape sends idiotic headers AGAIN
437     my $ims = $self->{h}{"if-modified-since"} =~ /^([^;]+)/
438     ? str2time $1 : 0;
439    
440     if (-d _ && -r _) {
441     # directory
442     if ($path !~ /\/$/) {
443     # create a redirect to get the trailing "/"
444     # we don't try to avoid the :80
445     $self->err(301, "moved permanently", { Location => "http://".$self->server_hostport."$self->{uri}/" });
446 root 1.25 } else {
447 root 1.49 $ims < $self->{stat}[9]
448     or $self->err(304, "not modified");
449    
450     if (-r "$path/index.html") {
451     # replace directory "size" by index.html filesize
452     $self->{stat}[7] = (stat ($self->{path} .= "/index.html"))[7];
453 root 1.52 $self->handle_file($queue_index);
454 root 1.49 } else {
455     $self->handle_dir;
456     }
457 root 1.1 }
458 root 1.49 } elsif (-f _ && -r _) {
459     -x _ and $self->err(403, "forbidden");
460 root 1.52 $self->handle_file(-s _ >= $::TRANSFER_SMALL ? $queue_large : $queue_small);
461 root 1.49 } else {
462     $self->err(404, "not found");
463 root 1.1 }
464     }
465     }
466    
467     sub handle_dir {
468     my $self = shift;
469 root 1.10 my $idx = $self->diridx;
470    
471     $self->response(200, "ok",
472     {
473     "Content-Type" => "text/html",
474     "Content-Length" => length $idx,
475 root 1.50 #d# directories change all the time, so X-
476     "X-Last-Modified" => time2str ((stat _)[9]),
477 root 1.10 },
478     $idx);
479 root 1.1 }
480    
481     sub handle_file {
482 root 1.52 my ($self, $queue) = @_;
483 root 1.34 my $length = $self->{stat}[7];
484 root 1.1 my $hdr = {
485     "Last-Modified" => time2str ((stat _)[9]),
486     };
487    
488     my @code = (200, "ok");
489     my ($l, $h);
490    
491     if ($self->{h}{range} =~ /^bytes=(.*)$/) {
492     for (split /,/, $1) {
493     if (/^-(\d+)$/) {
494     ($l, $h) = ($length - $1, $length - 1);
495     } elsif (/^(\d+)-(\d*)$/) {
496     ($l, $h) = ($1, ($2 ne "" || $2 >= $length) ? $2 : $length - 1);
497     } else {
498     ($l, $h) = (0, $length - 1);
499     goto ignore;
500     }
501 root 1.26 goto satisfiable if $l >= 0 && $l < $length && $h >= 0 && $h >= $l;
502 root 1.1 }
503     $hdr->{"Content-Range"} = "bytes */$length";
504 root 1.24 $hdr->{"Content-Length"} = $length;
505     $self->err(416, "not satisfiable", $hdr, "");
506 root 1.1
507     satisfiable:
508 root 1.4 # check for segmented downloads
509 root 1.10 if ($l && $::NO_SEGMENTED) {
510 root 1.44 my $delay = $::NOW + $::PER_TIMEOUT + 15;
511 root 1.36 while (%{$uri{$self->{remote_id}}{$self->{uri}}} > 1) {
512 root 1.44 if ($delay <= $::NOW) {
513 root 1.30 $self->err_segmented_download;
514 root 1.29 } else {
515 root 1.46 $httpevent->wait;
516 root 1.29 }
517 root 1.4 }
518     }
519    
520 root 1.1 $hdr->{"Content-Range"} = "bytes $l-$h/$length";
521     @code = (206, "partial content");
522     $length = $h - $l + 1;
523    
524     ignore:
525     } else {
526     ($l, $h) = (0, $length - 1);
527     }
528    
529 root 1.9 $self->{path} =~ /\.([^.]+)$/;
530     $hdr->{"Content-Type"} = $mimetype{lc $1} || "application/octet-stream";
531 root 1.1 $hdr->{"Content-Length"} = $length;
532    
533 root 1.4 $self->response(@code, $hdr, "");
534 root 1.1
535     if ($self->{method} eq "GET") {
536 root 1.32 $self->{time} = $::NOW;
537    
538 root 1.49 my $current = $Coro::current;
539 root 1.35
540 root 1.49 my ($fh, $buf, $r);
541 root 1.32
542 root 1.1 open $fh, "<", $self->{path}
543     or die "$self->{path}: late open failure ($!)";
544    
545     $h -= $l - 1;
546    
547 root 1.49 if (0) { # !AIO
548 root 1.19 if ($l) {
549     sysseek $fh, $l, 0;
550     }
551     }
552 root 1.48
553 root 1.51 my $transfer = $queue->start_transfer;
554     my $locked;
555 root 1.48 my $bufsize = $::WAIT_BUFSIZE; # initial buffer size
556 root 1.19
557 root 1.1 while ($h > 0) {
558 root 1.51 unless ($locked) {
559     if ($locked ||= $transfer->try($::WAIT_INTERVAL)) {
560 root 1.48 $bufsize = $::BUFSIZE;
561     $self->{time} = $::NOW;
562     }
563     }
564    
565 root 1.49 if (0) { # !AIO
566 root 1.48 sysread $fh, $buf, $h > $bufsize ? $bufsize : $h
567 root 1.19 or last;
568     } else {
569 root 1.48 aio_read($fh, $l, ($h > $bufsize ? $bufsize : $h),
570 root 1.19 $buf, 0, sub {
571     $r = $_[0];
572 root 1.37 Coro::ready($current);
573 root 1.19 });
574     &Coro::schedule;
575     last unless $r;
576     }
577 root 1.37 my $w = syswrite $self->{fh}, $buf
578 root 1.1 or last;
579 root 1.11 $::written += $w;
580     $self->{written} += $w;
581 root 1.16 $l += $r;
582 root 1.1 }
583 root 1.32
584     close $fh;
585 root 1.1 }
586 root 1.7 }
587    
588 root 1.2 1;