# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
package Apache::TestClient;
#this module provides some fallback for when libwww-perl is not installed
#it is by no means an LWP replacement, just enough for very simple requests
#this module does not and will never support certain features such as:
#file upload, http/1.1 (byteranges, keepalive, etc.), following redirects,
#authentication, GET body callbacks, SSL, etc.
use strict;
use warnings FATAL => 'all';
use Apache::TestRequest ();
my $CRLF = "\015\012";
sub request {
my($method, $url, @headers) = @_;
my @real_headers = ();
my $content;
for (my $i = 0; $i < scalar @headers; $i += 2) {
if ($headers[$i] =~ /^content$/i) {
$content = $headers[$i+1];
}
else {
push @real_headers, ($headers[$i], $headers[$i+1]);
}
}
## XXX:
## This is not a FULL URL encode mapping
## space ' '; however is very common, so this
## is useful to convert
$url =~ s/ /%20/g;
my $config = Apache::Test::config();
$method ||= 'GET';
$url ||= '/';
my %headers = ();
my $hostport = Apache::TestRequest::hostport($config);
$headers{Host} = (split ':', $hostport)[0];
my $s = Apache::TestRequest::vhost_socket();
unless ($s) {
warn "cannot connect to $hostport: $!";
return undef;
}
if ($content) {
$headers{'Content-Length'} ||= length $content;
$headers{'Content-Type'} ||= 'application/x-www-form-urlencoded';
}
#for modules/setenvif
$headers{'User-Agent'} ||= 'libwww-perl/0.00';
my $request = join $CRLF,
"$method $url HTTP/1.0",
(map { "$_: $headers{$_}" } keys %headers);
$request .= $CRLF;
for (my $i = 0; $i < scalar @real_headers; $i += 2) {
$request .= "$real_headers[$i]: $real_headers[$i+1]$CRLF";
}
$request .= $CRLF;
# using send() avoids the need to use SIGPIPE if the server aborts
# the connection
$s->send($request);
$s->send($content) if $content;
$request =~ s/\015//g; #for as_string
my $res = {
request => (bless {
headers_as_string => $request,
content => $content || '',
}, 'Apache::TestClientRequest'),
headers_as_string => '',
method => $method,
code => -1, # unknown
};
my($response_line, $header_term);
my $eol = "\015?\012";
local $_;
while (<$s>) {
$res->{headers_as_string} .= $_;
if (m:^(HTTP/\d+\.\d+)[ \t]+(\d+)[ \t]*(.*?)$eol:io) {
$res->{protocol} = $1;
$res->{code} = $2;
$res->{message} = $3;
$response_line = 1;
}
elsif (/^([a-zA-Z0-9_\-]+)\s*:\s*(.*?)$eol/o) {
$res->{headers}->{lc $1} = $2;
}
elsif (/^$eol$/o) {
$header_term = 1;
last;
}
}
unless ($response_line and $header_term) {
warn "malformed response";
}
{
local $/;
$res->{content} = <$s>;
}
close $s;
# an empty body is a valid response
$res->{content} = ''
unless exists $res->{content} and defined $res->{content};
$res->{headers_as_string} =~ s/\015//g; #for as_string
bless $res, 'Apache::TestClientResponse';
}
for my $method (qw(GET HEAD POST PUT)) {
no strict 'refs';
*$method = sub {
my $url = shift;
request($method, $url, @_);
};
}
package Apache::TestClientResponse;
sub header {
my($self, $key) = @_;
$self->{headers}->{lc $key};
}
my @headers = qw(Last-Modified Content-Type);
for my $header (@headers) {
no strict 'refs';
(my $method = lc $header) =~ s/-/_/g;
*$method = sub { shift->{headers}->{lc $header} };
}
sub is_success {
my $code = shift->{code};
return 0 unless defined $code && $code;
$code >= 200 && $code < 300;
}
sub status_line {
my $self = shift;
"$self->{code} $self->{message}";
}
sub as_string {
my $self = shift;
$self->{headers_as_string} . ($self->{content} || '');
}
my @methods = qw(
request protocol code message method
headers_as_string headers content
);
for my $method (@methods) {
no strict 'refs';
*$method = sub {
my($self, $val) = @_;
$self->{$method} = $val if $val;
$self->{$method};
};
}
#inherit headers_as_string, as_string, protocol, content, etc. methods
@Apache::TestClientRequest::ISA = qw(Apache::TestClientResponse);
1;