use base 'Class::Accessor::Fast';
use Carp;
+use HTTP::Response;
use IO::Handle;
use IO::File;
__PACKAGE__->mk_accessors(qw[ enviroment request stdin stdout stderr ]);
-our $VERSION = 0.1;
+our $VERSION = 0.5_02;
sub new {
my $class = shift;
my $request = shift;
-
+
unless ( @_ % 2 == 0 && eval { $request->isa('HTTP::Request') } ) {
croak(qq/usage: $class->new( \$request [, key => value] )/);
}
- my $self = {
- request => $request,
- restored => 0,
- setuped => 0,
- stdin => IO::File->new_tmpfile,
- stdout => IO::File->new_tmpfile
- };
+ my $self = $class->SUPER::new( { restored => 0, setuped => 0 } );
+ $self->request($request);
+ $self->stdin( IO::File->new_tmpfile );
+ $self->stdout( IO::File->new_tmpfile );
my $host = $request->header('Host');
my $uri = $request->uri->clone;
$uri->port(80) unless $uri->port;
$uri->host_port($host) unless !$host || ( $host eq $uri->host_port );
- $self->{enviroment} = {
+ $uri = $uri->canonical;
+
+ my $enviroment = {
GATEWAY_INTERFACE => 'CGI/1.1',
HTTP_HOST => $uri->host_port,
HTTPS => ( $uri->scheme eq 'https' ) ? 'ON' : 'OFF', # not in RFC 3875
REMOTE_ADDR => '127.0.0.1',
REMOTE_HOST => 'localhost',
REMOTE_PORT => int( rand(64000) + 1000 ), # not in RFC 3875
- REQUEST_URI => $uri->path_query || '/', # not in RFC 3875
+ REQUEST_URI => $uri->path_query, # not in RFC 3875
REQUEST_METHOD => $request->method,
@_
};
foreach my $field ( $request->headers->header_field_names ) {
- my $key = uc($field);
+ my $key = uc("HTTP_$field");
$key =~ tr/-/_/;
- $key = 'HTTP_' . $key unless $field =~ /^Content-(Length|Type)$/;
+ $key =~ s/^HTTP_// if $field =~ /^Content-(Length|Type)$/;
- unless ( exists $self->{enviroment}->{$key} ) {
- $self->{enviroment}->{$key} = $request->headers->header($field);
+ unless ( exists $enviroment->{$key} ) {
+ $enviroment->{$key} = $request->headers->header($field);
}
}
- return $class->SUPER::new($self);
+ unless ( $enviroment->{SCRIPT_NAME} eq '/' && $enviroment->{PATH_INFO} ) {
+ $enviroment->{PATH_INFO} =~ s/^\Q$enviroment->{SCRIPT_NAME}\E/\//;
+ $enviroment->{PATH_INFO} =~ s/^\/+/\//;
+ }
+
+ $self->enviroment($enviroment);
+
+ return $self;
}
sub setup {
$self->{restore}->{enviroment} = {%ENV};
- open( $self->{restore}->{stdin}, '>&', STDIN->fileno )
- or croak("Can't dup stdin: $!");
-
- open( STDIN, '<&=', $self->stdin->fileno )
- or croak("Can't open stdin: $!");
-
binmode( $self->stdin );
- binmode( STDIN );
if ( $self->request->content_length ) {
or croak("Can't seek stdin handle: $!");
}
+ open( $self->{restore}->{stdin}, '<&', STDIN->fileno )
+ or croak("Can't dup stdin: $!");
+
+ open( STDIN, '<&=', $self->stdin->fileno )
+ or croak("Can't open stdin: $!");
+
+ binmode( STDIN );
+
if ( $self->stdout ) {
open( $self->{restore}->{stdout}, '>&', STDOUT->fileno )
no warnings 'uninitialized';
%ENV = %{ $self->enviroment };
}
-
+
if ( $INC{'CGI.pm'} ) {
CGI::initialize_globals();
- }
+ }
$self->{setuped}++;
sub response {
my ( $self, $callback ) = @_;
- return undef unless $self->{setuped};
- return undef unless $self->{restored};
- return undef unless $self->{restore}->{stdout};
-
- require HTTP::Response;
+ return undef unless $self->stdout;
seek( $self->stdout, 0, SEEK_SET )
or croak("Can't seek stdout handle: $!");
- my $message;
+ my $headers;
while ( my $line = $self->stdout->getline ) {
- $message .= $line;
- last if $message =~ /\x0d?\x0a\x0d?\x0a$/;
+ $headers .= $line;
+ last if $headers =~ /\x0d?\x0a\x0d?\x0a$/;
}
-
- unless ( $message =~ /^HTTP/ ) {
- $message = "HTTP/1.1 200 OK\x0d\x0a" . $message;
+
+ unless ( defined $headers ) {
+ $headers = "HTTP/1.1 500 Internal Server Error\x0d\x0a";
}
- my $response = HTTP::Response->new;
- my @headers = split( /\x0d?\x0a/, $message );
- my $status = shift(@headers);
-
- unless ( $status =~ s/^(HTTP\/\d\.\d) (\d{3}) (.*)$// ) {
- croak( "Invalid Status-Line: '$status'" );
+ unless ( $headers =~ /^HTTP/ ) {
+ $headers = "HTTP/1.1 200 OK\x0d\x0a" . $headers;
}
- $response->protocol($1);
- $response->code($2);
- $response->message($3);
+ my $response = HTTP::Response->parse($headers);
+ $response->date( time() ) unless $response->date;
- my $token = qr/[^][\x00-\x1f\x7f()<>@,;:\\"\/?={} \t]+/;
+ my $message = $response->message;
+ my $status = $response->header('Status');
- foreach my $header (@headers) {
+ if ( $message && $message =~ /^(.+)\x0d$/ ) {
+ $response->message($1);
+ }
- unless( $header =~ s/^($token):[\t ]*// ) {
- croak( "Invalid header field name : '$header'" );
- }
+ if ( $status && $status =~ /^(\d\d\d)\s?(.+)?$/ ) {
- $response->push_header( $1 => $header );
- }
+ my $code = $1;
+ my $message = $2 || HTTP::Status::status_message($code);
- if ( my $code = $response->header('Status') ) {
$response->code($code);
- $response->message( HTTP::Status::status_message($code) );
+ $response->message($message);
}
+
+ my $length = ( stat( $self->stdout ) )[7] - tell( $self->stdout );
+
+ if ( $response->code == 500 && !$length ) {
- $response->headers->date( time() );
+ $response->content( $response->error_as_HTML );
+ $response->content_type('text/html');
+
+ return $response;
+ }
if ($callback) {
+
+ my $handle = $self->stdout;
+
$response->content( sub {
- if ( $self->stdout->read( my $buffer, 4096 ) ) {
+
+ if ( $handle->read( my $buffer, 4096 ) ) {
return $buffer;
}
+
return undef;
});
}
else {
+
my $length = 0;
+
while ( $self->stdout->read( my $buffer, 4096 ) ) {
$length += length($buffer);
$response->add_content($buffer);
}
-
+
if ( $length && !$response->content_length ) {
$response->content_length($length);
}
sub restore {
my $self = shift;
- %ENV = %{ $self->{restore}->{enviroment} };
+ {
+ no warnings 'uninitialized';
+ %ENV = %{ $self->{restore}->{enviroment} };
+ }
- open( STDIN, '>&', $self->{restore}->{stdin} )
+ open( STDIN, '<&', $self->{restore}->{stdin} )
or croak("Can't restore stdin: $!");
sysseek( $self->stdin, 0, SEEK_SET )
=item new ( $request [, key => value ] )
Contructor, first argument must be a instance of HTTP::Request
-followed by optional pairs of environment keys and values.
+followed by optional pairs of environment key and value.
=item enviroment
=back
+=head1 SEE ALSO
+
+=over 4
+
+=item examples directory in this distribution.
+
+=item L<WWW::Mechanize::CGI>
+
+=item L<Test::WWW::Mechanize::CGI>
+
+=back
+
=head1 THANKS TO
Thomas L. Shinnick for his valuable win32 testing.