X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?p=catagits%2FCatalyst-Runtime.git;a=blobdiff_plain;f=lib%2FCatalyst%2FEngine%2FCGI.pm;h=fa2e23ed244eae0b773c15e50ec13d0aa7a4b10f;hp=950cc2d709cdddd972c6b47d8f421001677baf5f;hb=ae29b412955743885e80350085167b54b69672da;hpb=f81097665c9a71971e5355c9d8331d9a9f9ba305 diff --git a/lib/Catalyst/Engine/CGI.pm b/lib/Catalyst/Engine/CGI.pm index 950cc2d..fa2e23e 100644 --- a/lib/Catalyst/Engine/CGI.pm +++ b/lib/Catalyst/Engine/CGI.pm @@ -1,9 +1,10 @@ package Catalyst::Engine::CGI; -use strict; -use base 'Catalyst::Engine'; -use NEXT; -use URI; +use Moose; +extends 'Catalyst::Engine'; + +has env => (is => 'rw'); +has _header_buf => (is => 'rw', clearer => '_clear_header_buf', predicate => '_has_header_buf'); =head1 NAME @@ -32,9 +33,7 @@ This is the Catalyst engine specialized for the CGI environment. This class overloads some methods from C. -=over 4 - -=item $self->finalize_headers($c) +=head2 $self->finalize_headers($c) =cut @@ -43,18 +42,19 @@ sub finalize_headers { $c->response->header( Status => $c->response->status ); - print $c->response->headers->as_string("\015\012"); - print "\015\012"; + $self->_header_buf($c->response->headers->as_string("\015\012") . "\015\012"); } -=item $self->prepare_connection($c) +=head2 $self->prepare_connection($c) =cut sub prepare_connection { my ( $self, $c ) = @_; + local (*ENV) = $self->env || \%ENV; - $c->request->address( $ENV{REMOTE_ADDR} ); + my $request = $c->request; + $request->address( $ENV{REMOTE_ADDR} ); PROXY_CHECK: { @@ -67,49 +67,58 @@ sub prepare_connection { # If we are running as a backend server, the user will always appear # as 127.0.0.1. Select the most recent upstream IP (last in the list) my ($ip) = $ENV{HTTP_X_FORWARDED_FOR} =~ /([^,\s]+)$/; - $c->request->address($ip); + $request->address($ip); } - $c->request->hostname( $ENV{REMOTE_HOST} ); - $c->request->protocol( $ENV{SERVER_PROTOCOL} ); - $c->request->user( $ENV{REMOTE_USER} ); - $c->request->method( $ENV{REQUEST_METHOD} ); + $request->hostname( $ENV{REMOTE_HOST} ) if exists $ENV{REMOTE_HOST}; + $request->protocol( $ENV{SERVER_PROTOCOL} ); + $request->user( $ENV{REMOTE_USER} ); + $request->method( $ENV{REQUEST_METHOD} ); if ( $ENV{HTTPS} && uc( $ENV{HTTPS} ) eq 'ON' ) { - $c->request->secure(1); + $request->secure(1); } if ( $ENV{SERVER_PORT} == 443 ) { - $c->request->secure(1); + $request->secure(1); } } -=item $self->prepare_headers($c) +=head2 $self->prepare_headers($c) =cut sub prepare_headers { my ( $self, $c ) = @_; - + local (*ENV) = $self->env || \%ENV; + my $headers = $c->request->headers; # Read headers from %ENV - while ( my ( $header, $value ) = each %ENV ) { + foreach my $header ( keys %ENV ) { next unless $header =~ /^(?:HTTP|CONTENT|COOKIE)/i; ( my $field = $header ) =~ s/^HTTPS?_//; - $c->req->headers->header( $field => $value ); + $headers->header( $field => $ENV{$header} ); } } -=item $self->prepare_path($c) +=head2 $self->prepare_path($c) =cut sub prepare_path { my ( $self, $c ) = @_; + local (*ENV) = $self->env || \%ENV; my $scheme = $c->request->secure ? 'https' : 'http'; my $host = $ENV{HTTP_HOST} || $ENV{SERVER_NAME}; my $port = $ENV{SERVER_PORT} || 80; - my $base_path = $ENV{SCRIPT_NAME} || '/'; + my $base_path; + if ( exists $ENV{REDIRECT_URL} ) { + $base_path = $ENV{REDIRECT_URL}; + $base_path =~ s/$ENV{PATH_INFO}$//; + } + else { + $base_path = $ENV{SCRIPT_NAME} || '/'; + } # If we are running as a backend proxy, get the true hostname PROXY_CHECK: @@ -127,80 +136,112 @@ sub prepare_path { $port = $c->request->secure ? 443 : 80; } - my $path = $base_path . $ENV{PATH_INFO}; + # set the request URI + my $path = $base_path . ( $ENV{PATH_INFO} || '' ); $path =~ s{^/+}{}; + + # Using URI directly is way too slow, so we construct the URLs manually + my $uri_class = "URI::$scheme"; + + # HTTP_HOST will include the port even if it's 80/443 + $host =~ s/:(?:80|443)$//; + + if ( $port !~ /^(?:80|443)$/ && $host !~ /:/ ) { + $host .= ":$port"; + } + + # Escape the path + $path =~ s/([^$URI::uric])/$URI::Escape::escapes{$1}/go; + $path =~ s/\?/%3F/g; # STUPID STUPID SPECIAL CASE + + my $query = $ENV{QUERY_STRING} ? '?' . $ENV{QUERY_STRING} : ''; + my $uri = $scheme . '://' . $host . '/' . $path . $query; - my $uri = URI->new; - $uri->scheme($scheme); - $uri->host($host); - $uri->port($port); - $uri->path($path); - $uri->query( $ENV{QUERY_STRING} ) if $ENV{QUERY_STRING}; - - # sanitize the URI - $uri = $uri->canonical; - $c->request->uri($uri); + $c->request->uri( bless \$uri, $uri_class ); # set the base URI # base must end in a slash - $base_path .= '/' unless ( $base_path =~ /\/$/ ); - my $base = $uri->clone; - $base->path_query($base_path); - $c->request->base($base); + $base_path .= '/' unless $base_path =~ m{/$}; + + my $base_uri = $scheme . '://' . $host . $base_path; + + $c->request->base( bless \$base_uri, $uri_class ); } -=item $self->prepare_query_parameters($c) +=head2 $self->prepare_query_parameters($c) =cut -sub prepare_query_parameters { +around prepare_query_parameters => sub { + my $orig = shift; my ( $self, $c ) = @_; - + local (*ENV) = $self->env || \%ENV; + if ( $ENV{QUERY_STRING} ) { - $self->SUPER::prepare_query_parameters( $c, $ENV{QUERY_STRING} ); + $self->$orig( $c, $ENV{QUERY_STRING} ); + } +}; + +=head2 $self->prepare_request($c, (env => \%env)) + +=cut + +sub prepare_request { + my ( $self, $c, %args ) = @_; + + if ( $args{env} ) { + $self->env( $args{env} ); } } -=item $self->prepare_write($c) +=head2 $self->prepare_write($c) Enable autoflush on the output handle for CGI-based engines. =cut -sub prepare_write { - my ( $self, $c ) = @_; - - # Set the output handle to autoflush +around prepare_write => sub { *STDOUT->autoflush(1); + return shift->(@_); +}; - $self->NEXT::prepare_write($c); -} +=head2 $self->write($c, $buffer) -=item $self->read_chunk($c, $buffer, $length) +Writes the buffer to the client. + +=cut + +around write => sub { + my $orig = shift; + my ( $self, $c, $buffer ) = @_; + + # Prepend the headers if they have not yet been sent + if ( $self->_has_header_buf ) { + $buffer = $self->_clear_header_buf . $buffer; + } + + return $self->$orig( $c, $buffer ); +}; + +=head2 $self->read_chunk($c, $buffer, $length) =cut sub read_chunk { shift; shift; *STDIN->sysread(@_); } -=item $self->run +=head2 $self->run =cut sub run { shift; shift->handle_request(@_) } -=back - =head1 SEE ALSO -L L. +L, L =head1 AUTHORS -Sebastian Riedel, - -Christian Hansen, - -Andy Grundman, +Catalyst Contributors, see Catalyst.pm =head1 COPYRIGHT @@ -208,5 +249,6 @@ This program is free software, you can redistribute it and/or modify it under the same terms as Perl itself. =cut +no Moose; 1;