1 | use Coro; |
1 | use Coro; |
2 | use Coro::Semaphore; |
2 | use Coro::Semaphore; |
3 | use Coro::Event; |
3 | use Coro::EV; |
4 | use Coro::Socket; |
4 | use Coro::Socket; |
5 | use Coro::Signal; |
5 | use Coro::Signal; |
6 | use Coro::AIO (); |
6 | use Coro::AIO (); |
7 | |
7 | |
8 | use HTTP::Date; |
8 | use HTTP::Date; |
… | |
… | |
26 | our $HTTP_NOW; |
26 | our $HTTP_NOW; |
27 | |
27 | |
28 | our $ERROR_LOG; |
28 | our $ERROR_LOG; |
29 | our $ACCESS_LOG; |
29 | our $ACCESS_LOG; |
30 | |
30 | |
31 | Event->timer(interval => 1, hard => 1, cb => sub { |
31 | our $update_time = EV::periodic 0, 1, undef, sub { |
32 | $NOW = time; |
32 | $NOW = time; |
33 | $HTTP_NOW = time2str $NOW; |
33 | $HTTP_NOW = time2str $NOW; |
34 | })->now; |
34 | }; |
|
|
35 | $update_time->invoke; |
35 | |
36 | |
36 | if ($ERROR_LOG) { |
37 | if ($ERROR_LOG) { |
37 | use IO::Handle; |
38 | use IO::Handle; |
38 | open $errorlog, ">>$ERROR_LOG" |
39 | open $errorlog, ">>$ERROR_LOG" |
39 | or die "$ERROR_LOG: $!"; |
40 | or die "$ERROR_LOG: $!"; |
… | |
… | |
150 | use Convert::Scalar 'weaken'; |
151 | use Convert::Scalar 'weaken'; |
151 | use IO::AIO; |
152 | use IO::AIO; |
152 | |
153 | |
153 | IO::AIO::min_parallel $::AIO_PARALLEL; |
154 | IO::AIO::min_parallel $::AIO_PARALLEL; |
154 | |
155 | |
155 | Event->io (fd => IO::AIO::poll_fileno, |
156 | our $AIO_WATCHER = EV::io IO::AIO::poll_fileno, EV::READ, \&IO::AIO::poll_cb; |
156 | poll => 'r', async => 1, |
|
|
157 | cb => \&IO::AIO::poll_cb); |
|
|
158 | |
157 | |
159 | our %conn; # $conn{ip}{self} => connobj |
158 | our %conn; # $conn{ip}{self} => connobj |
160 | our %uri; # $uri{ip}{uri}{self} |
159 | our %uri; # $uri{ip}{uri}{self} |
161 | our %blocked; |
160 | our %blocked; |
162 | our %mimetype; |
161 | our %mimetype; |
… | |
… | |
197 | } |
196 | } |
198 | |
197 | |
199 | sub DESTROY { |
198 | sub DESTROY { |
200 | my $self = shift; |
199 | my $self = shift; |
201 | |
200 | |
202 | close $self->{fh}; # workaround |
|
|
203 | --$::conns; |
201 | --$::conns; |
204 | } |
202 | } |
205 | |
203 | |
206 | sub prune_cache { |
204 | sub prune_cache { |
207 | my $hash = $_[0]; |
205 | my $hash = $_[0]; |
… | |
… | |
223 | for (keys %blocked) { |
221 | for (keys %blocked) { |
224 | delete $blocked{$_} unless $blocked{$_}[0] > $::NOW; |
222 | delete $blocked{$_} unless $blocked{$_}[0] > $::NOW; |
225 | } |
223 | } |
226 | } |
224 | } |
227 | |
225 | |
228 | Event->timer (interval => 60, cb => \&prune_caches); |
226 | our $PRUNE_WATCHER = EV::timer 60, 60, \&prune_caches; |
229 | |
227 | |
230 | sub slog { |
228 | sub slog { |
231 | my $self = shift; |
229 | my $self = shift; |
232 | main::slog($_[0], "$self->{remote_id}> $_[1]"); |
230 | main::slog($_[0], "$self->{remote_id}> $_[1]"); |
233 | } |
231 | } |
… | |
… | |
426 | sub map_uri { |
424 | sub map_uri { |
427 | my $self = shift; |
425 | my $self = shift; |
428 | my $host = $self->{server_name}; |
426 | my $host = $self->{server_name}; |
429 | my $uri = $self->{uri}; |
427 | my $uri = $self->{uri}; |
430 | |
428 | |
|
|
429 | $host =~ /[\/\\]/ |
|
|
430 | and $self->err(400, "bad request"); |
|
|
431 | |
431 | # some massaging, also makes it more secure |
432 | # some massaging, also makes it more secure |
432 | $uri =~ s/%([0-9a-fA-F][0-9a-fA-F])/chr hex $1/ge; |
433 | $uri =~ s/%([0-9a-fA-F][0-9a-fA-F])/chr hex $1/ge; |
433 | $uri =~ s%//+%/%g; |
434 | $uri =~ s%//+%/%g; |
434 | $uri =~ s%/\.(?=/|$)%%g; |
435 | $uri =~ s%/\.(?=/|$)%%g; |
435 | 1 while $uri =~ s%/[^/]+/\.\.(?=/|$)%%; |
436 | 1 while $uri =~ s%/[^/]+/\.\.(?=/|$)%%; |
… | |
… | |
554 | }; |
555 | }; |
555 | |
556 | |
556 | my @code = (200, "ok"); |
557 | my @code = (200, "ok"); |
557 | my ($l, $h); |
558 | my ($l, $h); |
558 | |
559 | |
559 | if ($self->{h}{range} =~ /^bytes=(.*)$/) { |
560 | if ($self->{h}{range} =~ /^bytes=(.*)$/i) { |
560 | for (split /,/, $1) { |
561 | for (split /,/, $1) { |
561 | if (/^-(\d+)$/) { |
562 | if (/^-(\d+)$/) { |
562 | ($l, $h) = ($length - $1, $length - 1); |
563 | ($l, $h) = ($length - $1, $length - 1); |
563 | } elsif (/^(\d+)-(\d*)$/) { |
564 | } elsif (/^(\d+)-(\d*)$/) { |
564 | ($l, $h) = ($1, ($2 ne "" || $2 >= $length) ? $2 : $length - 1); |
565 | ($l, $h) = ($1, ($2 ne "" || $2 >= $length) ? $2 : $length - 1); |
… | |
… | |
630 | |
631 | |
631 | Coro::AIO::aio_read $fh, $l, ($h > $bufsize ? $bufsize : $h), my $buf, 0 |
632 | Coro::AIO::aio_read $fh, $l, ($h > $bufsize ? $bufsize : $h), my $buf, 0 |
632 | or last; |
633 | or last; |
633 | |
634 | |
634 | $tbf->request (length $buf); |
635 | $tbf->request (length $buf); |
635 | my $w = syswrite $self->{fh}, $buf |
636 | my $w = $self->{fh}->syswrite ($buf) |
636 | or last; |
637 | or last; |
637 | $::written += $w; |
638 | $::written += $w; |
638 | $self->{written} += $w; |
639 | $self->{written} += $w; |
639 | $l += $w; |
640 | $l += $w; |
640 | } |
641 | } |