package Catalyst::Engine::CGI;
-use strict;
-use base 'Catalyst::Engine';
-use NEXT;
-use URI;
-use URI::Query;
+use Moose;
+extends 'Catalyst::Engine';
+
+has env => (is => 'rw');
+has _header_buf => (is => 'rw', clearer => '_clear_header_buf', predicate => '_has_header_buf');
=head1 NAME
This class overloads some methods from C<Catalyst::Engine>.
-=over 4
-
-=item $self->finalize_headers($c)
+=head2 $self->finalize_headers($c)
=cut
$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 ) = @_;
-
- $c->request->address( $ENV{REMOTE_ADDR} );
-
- PROXY_CHECK:
+ local (*ENV) = $self->env || \%ENV;
+
+ my $request = $c->request;
+ $request->address( $ENV{REMOTE_ADDR} );
+
+ PROXY_CHECK:
{
unless ( $c->config->{using_frontend_proxy} ) {
last PROXY_CHECK if $ENV{REMOTE_ADDR} ne '127.0.0.1';
last PROXY_CHECK if $c->config->{ignore_frontend_proxy};
}
last PROXY_CHECK unless $ENV{HTTP_X_FORWARDED_FOR};
-
+
# 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 $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:
+ PROXY_CHECK:
{
unless ( $c->config->{using_frontend_proxy} ) {
last PROXY_CHECK if $host !~ /localhost|127.0.0.1/;
last PROXY_CHECK unless $ENV{HTTP_X_FORWARDED_HOST};
$host = $ENV{HTTP_X_FORWARDED_HOST};
- # backend could be on any port, so
+
+ # backend could be on any port, so
# assume frontend is on the default port
$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{^/+}{};
- 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};
+ # 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
- # sanitize the URI
- $uri = $uri->canonical;
- $c->request->uri( $uri );
+ my $query = $ENV{QUERY_STRING} ? '?' . $ENV{QUERY_STRING} : '';
+ my $uri = $scheme . '://' . $host . '/' . $path . $query;
+
+ $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 ) = @_;
-
- my $u = URI::Query->new( $ENV{QUERY_STRING} );
- $c->request->query_parameters( { $u->hash } );
+ local (*ENV) = $self->env || \%ENV;
+
+ if ( $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
- $c->response->handle->autoflush(1);
-
- $self->NEXT::prepare_write( $c );
-}
+around prepare_write => sub {
+ *STDOUT->autoflush(1);
+ return shift->(@_);
+};
-=item $self->read_chunk($c, $buffer, $length)
+=head2 $self->write($c, $buffer)
+
+Writes the buffer to the client.
=cut
-sub read_chunk { shift; shift->request->handle->sysread( @_ ); }
+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 );
+};
-=item $self->run
+=head2 $self->read_chunk($c, $buffer, $length)
=cut
-sub run { shift; shift->handle_request(@_) }
+sub read_chunk { shift; shift; *STDIN->sysread(@_); }
-=back
+=head2 $self->run
-=head1 SEE ALSO
+=cut
-L<Catalyst> L<Catalyst::Engine>.
+sub run { shift; shift->handle_request(@_) }
-=head1 AUTHORS
+=head1 SEE ALSO
-Sebastian Riedel, <sri@cpan.org>
+L<Catalyst>, L<Catalyst::Engine>
-Christian Hansen, <ch@ngmedia.com>
+=head1 AUTHORS
-Andy Grundman, <andy@hybridized.org>
+Catalyst Contributors, see Catalyst.pm
=head1 COPYRIGHT
the same terms as Perl itself.
=cut
+no Moose;
1;