X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FCatalyst%2FEngine%2FHTTP.pm;h=808cbca4cb50e3ae1dd51a40b991315ccfd1629e;hb=00c9932455ecd6ca565d60d76180a2d850b7317b;hp=8d5cf4415792b62c0683222f14f8deb857fcbba4;hpb=4eeca0f22817aebad0ddfbcd1ce2ed0a446443d6;p=catagits%2FCatalyst-Runtime.git diff --git a/lib/Catalyst/Engine/HTTP.pm b/lib/Catalyst/Engine/HTTP.pm index 8d5cf44..808cbca 100644 --- a/lib/Catalyst/Engine/HTTP.pm +++ b/lib/Catalyst/Engine/HTTP.pm @@ -3,12 +3,15 @@ package Catalyst::Engine::HTTP; use strict; use base 'Catalyst::Engine::CGI'; use Errno 'EWOULDBLOCK'; -use FindBin; -use File::Find; -use File::Spec; use HTTP::Status; use NEXT; use Socket; +use IO::Socket::INET (); +use IO::Select (); + +# For PAR +require Catalyst::Engine::HTTP::Restarter; +require Catalyst::Engine::HTTP::Restarter::Watcher; =head1 NAME @@ -34,9 +37,7 @@ This is the Catalyst engine specialized for development and testing. =head1 METHODS -=over 4 - -=item $self->finalize_headers($c) +=head2 $self->finalize_headers($c) =cut @@ -47,10 +48,17 @@ sub finalize_headers { my $message = status_message($status); print "$protocol $status $message\015\012"; $c->response->headers->date(time); - $self->NEXT::finalize_headers($c); + $c->response->headers->header( + Connection => $self->_keep_alive ? 'keep-alive' : 'close' ); + + $c->response->header( Status => $c->response->status ); + + # Avoid 'print() on closed filehandle Remote' warnings when using IE + print $c->response->headers->as_string("\015\012") if *STDOUT->opened(); + print "\015\012" if *STDOUT->opened(); } -=item $self->finalize_read($c) +=head2 $self->finalize_read($c) =cut @@ -64,7 +72,7 @@ sub finalize_read { return $self->NEXT::finalize_read($c); } -=item $self->prepare_read($c) +=head2 $self->prepare_read($c) =cut @@ -77,7 +85,7 @@ sub prepare_read { return $self->NEXT::prepare_read($c); } -=item $self->read_chunk($c, $buffer, $length) +=head2 $self->read_chunk($c, $buffer, $length) =cut @@ -103,7 +111,20 @@ sub read_chunk { } } -=item run +=head2 $self->write($c, $buffer) + +Writes the buffer to the client. Can only be called once for a request. + +=cut + +sub write { + # Avoid 'print() on closed filehandle Remote' warnings when using IE + return unless *STDOUT->opened(); + + shift->NEXT::write( @_ ); +} + +=head2 run =cut @@ -113,126 +134,158 @@ sub run { $options ||= {}; - our $GOT_HUP; - local $GOT_HUP = 0; + if ($options->{background}) { + my $child = fork; + die "Can't fork: $!" unless defined($child); + exit if $child; + } - local $SIG{HUP} = sub { $GOT_HUP = 1; }; + my $restart = 0; local $SIG{CHLD} = 'IGNORE'; - # Setup restarter - my $restarter; - if ( $options->{restart} ) { - my $parent = $$; - unless ( $restarter = fork ) { - - # Prepare - close STDIN; - close STDOUT; - - # Index parent directory - my $dir = File::Spec->catdir( $FindBin::Bin, '..' ); - - my $regex = $options->{restart_regex}; - my $one = _index( $dir, $regex ); - RESTART: while (1) { - sleep $options->{restart_delay}; - my $two = _index( $dir, $regex ); - my $changes = _compare_index( $one, $two ); - if (@$changes) { - $one = $two; - - # Test modified pm's - for my $file (@$changes) { - next unless $file =~ /\.pm$/; - if ( my $error = _test($file) ) { - print STDERR - qq/File "$file" modified, not restarting\n\n/; - print STDERR '*' x 80, "\n"; - print STDERR $error; - print STDERR '*' x 80, "\n"; - next RESTART; - } - } - - # Restart - my $files = join ', ', @$changes; - print STDERR qq/File(s) "$files" modified, restarting\n\n/; - kill( 1, $parent ); - exit; - } - } - } + my $allowed = $options->{allowed} || { '127.0.0.1' => '255.255.255.255' }; + my $addr = $host ? inet_aton($host) : INADDR_ANY; + if ( $addr eq INADDR_ANY ) { + require Sys::Hostname; + $host = lc Sys::Hostname::hostname(); + } + else { + $host = gethostbyaddr( $addr, AF_INET ) || inet_ntoa($addr); } # Handle requests # Setup socket - $host = $host ? inet_aton($host) : INADDR_ANY; - socket( HTTPDaemon, PF_INET, SOCK_STREAM, getprotobyname('tcp') ); - setsockopt( HTTPDaemon, SOL_SOCKET, SO_REUSEADDR, pack( "l", 1 ) ); - bind( HTTPDaemon, sockaddr_in( $port, $host ) ); - listen( HTTPDaemon, SOMAXCONN ); - my $url = 'http://'; - if ( $host eq INADDR_ANY ) { - require Sys::Hostname; - $url .= lc Sys::Hostname::hostname(); - } - else { - $url .= gethostbyaddr( $host, AF_INET ) || inet_ntoa($host); - } - $url .= ":$port"; + my $daemon = IO::Socket::INET->new( + Listen => SOMAXCONN, + LocalAddr => inet_ntoa($addr), + LocalPort => $port, + Proto => 'tcp', + ReuseAddr => 1, + Type => SOCK_STREAM, + ) + or die "Couldn't create daemon: $!"; + + my $url = "http://$host"; + $url .= ":$port" unless $port == 80; + print "You can connect to your server at $url\n"; - my $pid = undef; - while ( accept( Remote, HTTPDaemon ) ) { - # Fork - if ( $options->{fork} ) { next if $pid = fork } + if ($options->{background}) { + open STDIN, "+&STDIN" or die $!; + open STDERR, ">&STDIN" or die $!; + if ( $^O !~ /MSWin32/ ) { + require POSIX; + POSIX::setsid() + or die "Can't start a new session: $!"; + } + } - close HTTPDaemon if defined $pid; + if (my $pidfile = $options->{pidfile}) { + if (! open PIDFILE, "> $pidfile") { + warn("Cannot open: $pidfile: $!"); + } + print PIDFILE "$$\n"; + close PIDFILE; + } - # Ignore broken pipes as an HTTP server should - local $SIG{PIPE} = sub { close Remote }; - local $SIG{HUP} = ( defined $pid ? 'IGNORE' : $SIG{HUP} ); + $self->_keep_alive( $options->{keepalive} || 0 ); - local *STDIN = \*Remote; - local *STDOUT = \*Remote; - select STDOUT; + my $pid = undef; + while ( accept( Remote, $daemon ) ) + { # TODO: get while ( my $remote = $daemon->accept ) to work + + select Remote; # Request data - my $remote_sockaddr = getpeername( \*Remote ); - my ( undef, $iaddr ) = sockaddr_in($remote_sockaddr); - my $peername = gethostbyaddr( $iaddr, AF_INET ) || "localhost"; - my $peeraddr = inet_ntoa($iaddr) || "127.0.0.1"; - my $local_sockaddr = getsockname( \*Remote ); - my ( undef, $localiaddr ) = sockaddr_in($local_sockaddr); - my $localname = gethostbyaddr( $localiaddr, AF_INET ) - || "localhost"; - my $localaddr = inet_ntoa($localiaddr) || "127.0.0.1"; - - STDIN->blocking(1); - - # Parse request line - my $line = $self->_get_line( \*STDIN ); + + Remote->blocking(1); + next unless my ( $method, $uri, $protocol ) = - $line =~ m/\A(\w+)\s+(\S+)(?:\s+HTTP\/(\d+(?:\.\d+)?))?\z/; + $self->_parse_request_line( \*Remote ); + + unless ( uc($method) eq 'RESTART' ) { + + # Fork + if ( $options->{fork} ) { next if $pid = fork } + + $self->_handler( $class, $port, $method, $uri, $protocol ); + + $daemon->close if defined $pid; + + } + else { + my $sockdata = $self->_socket_data( \*Remote ); + my $ipaddr = _inet_addr( $sockdata->{peeraddr} ); + my $ready = 0; + foreach my $ip ( keys %$allowed ) { + my $mask = $allowed->{$ip}; + $ready = ( $ipaddr & _inet_addr($mask) ) == _inet_addr($ip); + last if $ready; + } + if ($ready) { + $restart = 1; + last; + } + } + + exit if defined $pid; + } + continue { + close Remote; + } + $daemon->close; + + if ($restart) { + $SIG{CHLD} = 'DEFAULT'; + wait; + + ### if the standalone server was invoked with perl -I .. we will loose + ### those include dirs upon re-exec. So add them to PERL5LIB, so they + ### are available again for the exec'ed process --kane + use Config; + $ENV{PERL5LIB} .= join $Config{path_sep}, @INC; + + exec $^X . ' "' . $0 . '" ' . join( ' ', @{ $options->{argv} } ); + } + + exit; +} + +sub _handler { + my ( $self, $class, $port, $method, $uri, $protocol ) = @_; + + # Ignore broken pipes as an HTTP server should + local $SIG{PIPE} = sub { close Remote }; - # We better be careful and just use 1.0 - $protocol = '1.0'; + local *STDIN = \*Remote; + local *STDOUT = \*Remote; + # We better be careful and just use 1.0 + $protocol = '1.0'; + + my $sockdata = $self->_socket_data( \*Remote ); + my %copy_of_env = %ENV; + + my $sel = IO::Select->new; + $sel->add( \*STDIN ); + + while (1) { my ( $path, $query_string ) = split /\?/, $uri, 2; # Initialize CGI environment local %ENV = ( - PATH_INFO => $path || '', - QUERY_STRING => $query_string || '', - REMOTE_ADDR => $peeraddr, - REMOTE_HOST => $peername, - REQUEST_METHOD => $method || '', - SERVER_NAME => $localname, - SERVER_PORT => $port, + PATH_INFO => $path || '', + QUERY_STRING => $query_string || '', + REMOTE_ADDR => $sockdata->{peeraddr}, + REMOTE_HOST => $sockdata->{peername}, + REQUEST_METHOD => $method || '', + SERVER_NAME => $sockdata->{localname}, + SERVER_PORT => $port, SERVER_PROTOCOL => "HTTP/$protocol", - %ENV, + %copy_of_env, ); # Parse headers @@ -260,32 +313,67 @@ sub run { # Pass flow control to Catalyst $class->handle_request; - exit if defined $pid; - } - continue { - close Remote; - } - close HTTPDaemon; - if ($GOT_HUP) { - $SIG{CHLD} = 'DEFAULT'; - wait; - exec {$0}( ( ( -x $0 ) ? () : ($^X) ), $0, @{ $options->{argv} } ); + my $connection = lc $ENV{HTTP_CONNECTION}; + last + unless $self->_keep_alive() + && index( $connection, 'keep-alive' ) > -1 + && index( $connection, 'te' ) == -1 # opera stuff + && $sel->can_read(5); + + last + unless ( $method, $uri, $protocol ) = + $self->_parse_request_line( \*STDIN ); } + + close Remote; } -sub _compare_index { - my ( $one, $two ) = @_; - my %clone = %$two; - my @changes; - while ( my ( $key, $val ) = each %$one ) { - if ( !$clone{$key} || ( $clone{$key} ne $val ) ) { - push @changes, $key; - } - delete $clone{$key}; - } - for my $key ( keys %clone ) { push @changes, $key } - return \@changes; +sub _keep_alive { + my ( $self, $keepalive ) = @_; + + my $r = $self->{_keepalive} || 0; + $self->{_keepalive} = $keepalive if defined $keepalive; + + return $r; + +} + +sub _parse_request_line { + my ( $self, $handle ) = @_; + + # Parse request line + my $line = $self->_get_line($handle); + return () + unless my ( $method, $uri, $protocol ) = + $line =~ m/\A(\w+)\s+(\S+)(?:\s+HTTP\/(\d+(?:\.\d+)?))?\z/; + return ( $method, $uri, $protocol ); +} + +sub _socket_data { + my ( $self, $handle ) = @_; + + my $remote_sockaddr = getpeername($handle); + my ( undef, $iaddr ) = $remote_sockaddr + ? sockaddr_in($remote_sockaddr) + : (undef, undef); + + my $local_sockaddr = getsockname($handle); + my ( undef, $localiaddr ) = sockaddr_in($local_sockaddr); + + # This mess is necessary to keep IE from crashing the server + my $data = { + peername => $iaddr + ? ( gethostbyaddr( $iaddr, AF_INET ) || 'localhost' ) + : 'localhost', + peeraddr => $iaddr + ? ( inet_ntoa($iaddr) || '127.0.0.1' ) + : '127.0.0.1', + localname => gethostbyaddr( $localiaddr, AF_INET ) || 'localhost', + localaddr => inet_ntoa($localiaddr) || '127.0.0.1', + }; + + return $data; } sub _get_line { @@ -303,38 +391,7 @@ sub _get_line { return $line; } -sub _index { - my ( $dir, $regex ) = @_; - my %index; - finddepth( - { - wanted => sub { - my $file = File::Spec->rel2abs($File::Find::name); - return unless $file =~ /$regex/; - return unless -f $file; - my $time = ( stat $file )[9]; - $index{$file} = $time; - }, - no_chdir => 1 - }, - $dir - ); - return \%index; -} - -sub _test { - my $file = shift; - delete $INC{$file}; - local $SIG{__WARN__} = sub { }; - open my $olderr, '>&STDERR'; - open STDERR, '>', File::Spec->devnull; - eval "require '$file'"; - open STDERR, '>&', $olderr; - return $@ if $@; - return 0; -} - -=back +sub _inet_addr { unpack "N*", inet_aton( $_[0] ) } =head1 SEE ALSO @@ -346,6 +403,8 @@ Sebastian Riedel, Dan Kubb, +Sascha Kiefer, + =head1 THANKS Many parts are ripped out of C by Jesse Vincent.