Merged 5.49_01 (r1339) from refactored branch to trunk
[catagits/Catalyst-Runtime.git] / lib / Catalyst / Engine / HTTP.pm
index 4ec9fb3..27873f2 100644 (file)
@@ -1,7 +1,11 @@
 package Catalyst::Engine::HTTP;
 
 use strict;
-use base 'Catalyst::Engine::HTTP::Daemon';
+use base 'Catalyst::Engine::CGI';
+use Errno 'EWOULDBLOCK';
+use HTTP::Status;
+use NEXT;
+use Socket;
 
 =head1 NAME
 
@@ -25,14 +29,228 @@ A script using the Catalyst::Engine::HTTP module might look like:
 
 This is the Catalyst engine specialized for development and testing.
 
+=head1 METHODS
+
+=over 4
+
+=item $self->finalize_headers($c)
+
+=cut
+
+sub finalize_headers {
+    my ( $self, $c ) = @_;
+    my $protocol = $c->request->protocol;
+    my $status   = $c->response->status;
+    my $message  = status_message($status);
+    print "$protocol $status $message\015\012";
+    $c->response->headers->date(time);
+    $self->NEXT::finalize_headers($c);
+}
+
+=item $self->finalize_read($c)
+
+=cut
+
+sub finalize_read {
+    my ( $self, $c ) = @_;
+
+    # Never ever remove this, it would result in random length output
+    # streams if STDIN eq STDOUT (like in the HTTP engine)
+    $c->request->handle->blocking(1);
+
+    return $self->NEXT::finalize_read($c);
+}
+
+=item $self->prepare_read($c)
+
+=cut
+
+sub prepare_read {
+    my ( $self, $c ) = @_;
+
+    # Set the input handle to non-blocking
+    $c->request->handle->blocking(0);
+
+    return $self->NEXT::prepare_read($c);
+}
+
+=item $self->read_chunk($c, $buffer, $length)
+
+=cut
+
+sub read_chunk {
+    my $self = shift;
+    my $c    = shift;
+
+    # support for non-blocking IO
+    my $handle = $c->request->handle;
+    my $rin    = '';
+    vec( $rin, $handle->fileno, 1 ) = 1;
+
+  READ:
+    {
+        select( $rin, undef, undef, undef );
+        my $rc = $handle->sysread(@_);
+        if ( defined $rc ) {
+            return $rc;
+        }
+        else {
+            next READ if $! == EWOULDBLOCK;
+            return;
+        }
+    }
+}
+
+=item run
+
+=cut
+
+# A very very simple HTTP server that initializes a CGI environment
+sub run {
+    my ( $self, $class, $port, $host, $fork ) = @_;
+
+    our $GOT_HUP;
+    local $GOT_HUP = 0;
+
+    local $SIG{HUP} = sub { $GOT_HUP = 1; };
+
+    local $SIG{CHLD} = 'IGNORE';
+
+    # 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";
+    print "You can connect to your server at $url\n";
+    my $pid = undef;
+    while ( accept( Remote, HTTPDaemon ) ) {
+
+        # Fork
+        if ($fork) { next if $pid = fork }
+
+        close HTTPDaemon if defined $pid;
+
+        # Ignore broken pipes as an HTTP server should
+        local $SIG{PIPE} = sub { close Remote };
+        local $SIG{HUP} = (defined $pid ? 'IGNORE' : $SIG{HUP});
+
+        local *STDIN  = \*Remote;
+        local *STDOUT = \*Remote;
+        select STDOUT;
+
+        # 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 );
+        next
+          unless my ( $method, $uri, $protocol ) =
+          $line =~ m/\A(\w+)\s+(\S+)(?:\s+HTTP\/(\d+(?:\.\d+)?))?\z/;
+
+        # We better be careful and just use 1.0
+        $protocol = '1.0';
+
+        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,
+            SERVER_PROTOCOL => "HTTP/$protocol",
+            %ENV,
+        );
+
+        # Parse headers
+        if ( $protocol >= 1 ) {
+            while (1) {
+                my $line = $self->_get_line( \*STDIN );
+                last if $line eq '';
+                next
+                  unless my ( $name, $value ) =
+                  $line =~ m/\A(\w(?:-?\w+)*):\s(.+)\z/;
+
+                $name = uc $name;
+                $name = 'COOKIE' if $name eq 'COOKIES';
+                $name =~ tr/-/_/;
+                $name = 'HTTP_' . $name
+                  unless $name =~ m/\A(?:CONTENT_(?:LENGTH|TYPE)|COOKIE)\z/;
+                if ( exists $ENV{$name} ) {
+                    $ENV{$name} .= "; $value";
+                }
+                else {
+                    $ENV{$name} = $value;
+                }
+            }
+        }
+
+        # Pass flow control to Catalyst
+        $class->handle_request;
+        exit if defined $pid;
+    }
+    continue {
+        close Remote;
+    }
+    close HTTPDaemon;
+    exec {$0}( ( ( -x $0 ) ? () : ($^X) ), $0, @ARGV ) if $GOT_HUP;
+}
+
+sub _get_line {
+    my ( $self, $handle ) = @_;
+
+    my $line = '';
+
+    while ( sysread( $handle, my $byte, 1 ) ) {
+        last if $byte eq "\012";    # eol
+        $line .= $byte;
+    }
+
+    1 while $line =~ s/\s\z//;
+
+    return $line;
+}
+
+=back
+
 =head1 SEE ALSO
 
-L<Catalyst>, L<Catalyst::Engine>, L<Catalyst::Engine::HTTP::Daemon>.
+L<Catalyst>, L<Catalyst::Engine>.
+
+=head1 AUTHORS
+
+Sebastian Riedel, <sri@cpan.org>
+
+Dan Kubb, <dan.kubb-cpan@onautopilot.com>
 
-=head1 AUTHOR
+=head1 THANKS
 
-Sebastian Riedel, C<sri@cpan.org>
-Christian Hansen, C<ch@ngmedia.com>
+Many parts are ripped out of C<HTTP::Server::Simple> by Jesse Vincent.
 
 =head1 COPYRIGHT