]> sjero.net Git - wget/blobdiff - tests/HTTPServer.pm
[svn] New OO Architecture for Wget Test Suite
[wget] / tests / HTTPServer.pm
index c7de835fd22e7ba972427b993427141b0ac0de3a..072cd39ee8d4ef43b83d9f33215427750567df5b 100755 (executable)
@@ -1,30 +1,32 @@
 #!/usr/bin/perl -w
 
+package HTTPServer;
+
+use strict;
+
 use HTTP::Daemon;
 use HTTP::Status;
 use HTTP::Headers;
 use HTTP::Response;
 
-use strict;
+our @ISA=qw(HTTP::Daemon);
 
-package HTTPServer;
+my $CRLF = "\015\012"; # "\r\n" is not portable
 
-sub run_daemon {
-    my %urls = @_;
-    my $server = HTTP::Daemon->new (LocalAddr => 'localhost', 
-                                    LocalPort => '8080',
-                                    ReuseAddr => 1) or die "Cannot create server!!!";
+sub run {
+    my ($self, $urls) = @_;
                                 
-    while (my $con = $server->accept) {
+    while (my $con = $self->accept) {
         while (my $req = $con->get_request) {
             # print STDERR "method: ", $req->method, "\n";
-            if ($req->method eq "GET" and $urls{$req->url->path}) {
+            if (exists($urls->{$req->url->path})) {
+                next unless ($req->method eq "HEAD" || $req->method eq "GET");
                 # print STDERR "requested URL: ", $req->url->path, "\n";
-                
+                    
                 # create response
-                my $tmp = $urls{$req->url->path};
-                my $resp = HTTP::Response->new ($tmp->{'code'},
-                                                $tmp->{'msg'});
+                my $tmp = $urls->{$req->url->path};
+                my $resp = HTTP::Response->new ($tmp->{code},
+                                                $tmp->{msg});
                 # print STDERR "HTTP::Response: \n", $resp->as_string;
                 
                 # fill in headers
@@ -33,15 +35,26 @@ sub run_daemon {
                     $resp->header($name => $value);
                 }
                 # print STDERR "HTTP::Response with headers: \n", $resp->as_string;
-
-                # fill in content
-                $resp->content($tmp->{content});
-                # print STDERR "HTTP::Response with content: \n", $resp->as_string;
+                
+                if ($req->method eq "GET") {
+                    if (exists($tmp->{headers}{"Content-Length"})) {
+                        # Content-Length and length($tmp->{content}) don't match
+                        # manually prepare the HTTP response
+                        $con->send_basic_header($tmp->{code}, $resp->message, $resp->protocol);
+                        print $con $resp->headers_as_string($CRLF);
+                        print $con $CRLF;
+                        print $con $tmp->{content};                                
+                        next;
+                    }
+                    # fill in content
+                    $resp->content($tmp->{content});
+                    # print STDERR "HTTP::Response with content: \n", $resp->as_string;
+                }
                 
                 $con->send_response($resp);
                 # print STDERR "HTTP::Response sent: \n", $resp->as_string;
             } else {
-                print STDERR "requested wrong URL: ", $req->url->path, "\n";
+                print STDERR "requested wrong URL: ", $req->url->path, "\n";
                 $con->send_error($HTTP::Status::RC_FORBIDDEN);
             }
         }
@@ -50,16 +63,6 @@ sub run_daemon {
     }
 }
 
-sub run {
-    my $pid = fork();
-
-    if($pid == 0) {
-        run_daemon(@_);
-    }
-
-    return $pid;
-}
-
 1;
 
 # vim: et ts=4 sw=4