server/ext/http.ext

#! perl # mandatory

# http server on base port

use Coro::AnyEvent;

our @WEBSOCKETS_FORWARDER = qw(slashflash.pl 13327);

sub send {
   my $self = $_[0];

   $self->{wbuf} .= $_[1];

   $self->{ww} ||= AE::io $self->{fh}, 1, sub {
      my $len = syswrite $self->{fh}, $self->{wbuf};
      substr $self->{wbuf}, 0, $len, "";

      delete $self->{ww} unless length $self->{wbuf};
   };
}

sub fatal {
   my ($self) = @_;

   $self->send ("HTTP/1.1 500 internal error\015\012");
   delete $self->{rw};
}

sub respond {
   $_[0]->send ("HTTP/1.1 $_[1]\015\012content-length: " . (0 + length $_[2]) . "\015\012$_[3]\015\012$_[2]");
}

my $cache_headers = "cache-control: max-age=8640000\015\012"
                    . "etag: \"0\"\015\012";

sub content_type {
   return "content-type: image/png\015\012"   if $_[0] =~ /^\x89PNG/;
   return "content-type: image/jpeg\015\012"  if $_[0] =~ /^......JFIF/s;
   return "content-type: audio/wav\015\012"   if $_[0] =~ /^RIFF/;
   return "content-type: audio/x-ogg\015\012" if $_[0] =~ /^OggS/;

   "content-type: text/plain\015\012"
}

sub copy {
   my ($a, $b, $buf) = @_;

   # ultra-lame
   Coro::async {
      while () {
         while (length $buf) {
            my $len = syswrite $b, $buf
               or $! == Errno::EINTR
               or return;

            substr $buf, 0, $len, "";

            Coro::AnyEvent::writable $b;
         }

         Coro::AnyEvent::readable $a;
         sysread $a, $buf, 4096
            or return;
      }
   };
}

sub handle_req {
   my ($self) = @_;

   while ($self->{rbuf} =~ s/^( (?: [^\015]+ | . )+? )\015\012\015\012//xs) {
      my $req = $1;
      my $orig_req = $req;

      # we ignore headers atm.

      $req =~ s%^GET (\S+) HTTP/[0-9.]+\015\012%%
         or return $self->fatal;

      my $uri = $1;

      $uri =~ s%^http://[^/]*%%i; # just in case

      cf::debug "HTTP GET: $self->{id} $uri";

      if ($uri =~ m%^/(M?)([0-9a-f]+)$%) { # faces
         my $want_meta = $1;
         my $idx = $cf::FACEHASH{pack "H*", $2};

         $idx
            or do { $self->respond ("404 illegal face name"), next };

         if ($req =~ /if-none-match/i) { # dirtiest hack evar
            $self->respond ("304 not modified", "", $cache_headers);
            next;
         }

         my $type = cf::face::get_type $idx, 1;
         my $data = cf::face::get_data $idx, 1;

         (my $meta, $data) = unpack "(w/a*)*", $data
            if $type & 1;

         if ($want_meta) {
            if ($type & 1) {
               $self->respond ("200 OK", $meta, "content-type: text/plain\015\012" . $cache_headers);
            } else {
               $self->respond ("404 type $type has no metadata");
            }
         } else {
            $self->respond ("200 OK", $data, (content_type $data) . $cache_headers);
         }

      } elsif ($uri eq "/debug") { # for debugging
         my $body = "<html><body>";

         for my $type (6, 5, 4, 3, 2, 1, 0) {
            $body .= "<h1>$type</h1>";

            for (1 .. cf::face::faces_size - 1) {
               next if $type != cf::face::get_type $_;
               my $name = cf::face::get_name $_;
               my $id   = unpack "H*", cf::face::get_chksum $_, 1;
               $body .= "$_ <a href='$id'>$name ($id)</a>";
               $body .= " <a href='M$id'>(meta)</a>" if $type & 1;
               $body .= "<br>";
            }
         }

         $body .= "</body></html>";

         $self->respond ("200 OK", $body, "Content-Type: text/html\015\012");
      } elsif ($uri eq "/ws") {
         my $fh  = $self->{fh};
         my $buf = $orig_req . "\015\012\015\012" . $self->{rbuf};

         %$self = ();

         &AnyEvent::Socket::tcp_connect (@WEBSOCKETS_FORWARDER, sub {
            my ($fh2) = shift;

            Coro::async {
               if ($fh2) {
                  my $a = copy $fh, $fh2, $buf;
                  my $b = copy $fh2, $fh;

                  $a->join;
                  $b->join;

                  close $fh2;
               }

               close $fh;
            };
         });

      } else {
         $self->respond ("404 not found");
      }
   }
}

# dirty hack: called directly from tcp.ext
sub server {
   my $self = bless {
      id   => $_[0],
      fh   => $_[1],
      rbuf => $_[2],
      wbuf => "",
   };

   $self->{rw} = AE::io $self->{fh}, 0, sub {
      my $len = sysread $self->{fh}, $self->{rbuf}, 4096, length $self->{rbuf};

      if ($len == 0) {
         delete $self->{rw};
      } else {
         $self->handle_req;

         delete $self->{rw} if length $self->{rbuf} > 8192; # headers too long
      }
   };

   $self->handle_req; # in the unlikely case of the buffer already forming a valid request
}

cf::register_exticmd http_faceurl => sub {
   my ($ns) = @_;

   "/"
};

Revision:	1.9
Committed:	Wed Oct 31 19:41:32 2012 UTC (11 years, 6 months ago) by root
Branch:	MAIN
Changes since 1.8:	+52 -2 lines
Log Message:	most inefficient copying hack - proof of concept
#	User	Rev	Content
1	root	1.1	#! perl # mandatory
2
3			# http server on base port
4
5			use Coro::AnyEvent;
6
7	root	1.9	our @WEBSOCKETS_FORWARDER = qw(slashflash.pl 13327);
8
9	root	1.1	sub send {
10			my $self = $_[0];
11
12			$self->{wbuf} .= $_[1];
13
14			$self->{ww} \|\|= AE::io $self->{fh}, 1, sub {
15			my $len = syswrite $self->{fh}, $self->{wbuf};
16			substr $self->{wbuf}, 0, $len, "";
17
18			delete $self->{ww} unless length $self->{wbuf};
19			};
20			}
21
22			sub fatal {
23			my ($self) = @_;
24
25			$self->send ("HTTP/1.1 500 internal error\015\012");
26			delete $self->{rw};
27			}
28
29			sub respond {
30	root	1.6	$_[0]->send ("HTTP/1.1 $_[1]\015\012content-length: " . (0 + length $_[2]) . "\015\012$_[3]\015\012$_[2]");
31	root	1.1	}
32
33	root	1.6	my $cache_headers = "cache-control: max-age=8640000\015\012"
34			. "etag: \"0\"\015\012";
35
36	root	1.8	sub content_type {
37			return "content-type: image/png\015\012" if $_[0] =~ /^\x89PNG/;
38			return "content-type: image/jpeg\015\012" if $_[0] =~ /^......JFIF/s;
39			return "content-type: audio/wav\015\012" if $_[0] =~ /^RIFF/;
40			return "content-type: audio/x-ogg\015\012" if $_[0] =~ /^OggS/;
41
42			"content-type: text/plain\015\012"
43			}
44
45	root	1.9	sub copy {
46			my ($a, $b, $buf) = @_;
47
48			# ultra-lame
49			Coro::async {
50			while () {
51			while (length $buf) {
52			my $len = syswrite $b, $buf
53			or $! == Errno::EINTR
54			or return;
55
56			substr $buf, 0, $len, "";
57
58			Coro::AnyEvent::writable $b;
59			}
60
61			Coro::AnyEvent::readable $a;
62			sysread $a, $buf, 4096
63			or return;
64			}
65			};
66			}
67
68	root	1.1	sub handle_req {
69			my ($self) = @_;
70
71	root	1.9	while ($self->{rbuf} =~ s/^( (?: [^\015]+ \| . )+? )\015\012\015\012//xs) {
72	root	1.4	my $req = $1;
73	root	1.9	my $orig_req = $req;
74	root	1.1
75	root	1.4	# we ignore headers atm.
76	root	1.1
77	root	1.4	$req =~ s%^GET (\S+) HTTP/[0-9.]+\015\012%%
78			or return $self->fatal;
79	root	1.1
80	root	1.4	my $uri = $1;
81	root	1.1
82	root	1.4	$uri =~ s%^http://[^/]*%%i; # just in case
83	root	1.1
84	root	1.4	cf::debug "HTTP GET: $self->{id} $uri";
85	root	1.2
86	root	1.8	if ($uri =~ m%^/(M?)([0-9a-f]+)$%) { # faces
87			my $want_meta = $1;
88			my $idx = $cf::FACEHASH{pack "H*", $2};
89	root	1.2
90	root	1.4	$idx
91	root	1.7	or do { $self->respond ("404 illegal face name"), next };
92	root	1.1
93	root	1.4	if ($req =~ /if-none-match/i) { # dirtiest hack evar
94	root	1.6	$self->respond ("304 not modified", "", $cache_headers);
95	root	1.7	next;
96	root	1.4	}
97	root	1.1
98	root	1.4	my $type = cf::face::get_type $idx, 1;
99	root	1.8	my $data = cf::face::get_data $idx, 1;
100	root	1.1
101	root	1.8	(my $meta, $data) = unpack "(w/a)", $data
102			if $type & 1;
103
104			if ($want_meta) {
105			if ($type & 1) {
106			$self->respond ("200 OK", $meta, "content-type: text/plain\015\012" . $cache_headers);
107	root	1.4	} else {
108	root	1.8	$self->respond ("404 type $type has no metadata");
109	root	1.4	}
110	root	1.8	} else {
111			$self->respond ("200 OK", $data, (content_type $data) . $cache_headers);
112	root	1.1	}
113	root	1.8
114			} elsif ($uri eq "/debug") { # for debugging
115	root	1.4	my $body = "<html><body>";
116	root	1.2
117	root	1.8	for my $type (6, 5, 4, 3, 2, 1, 0) {
118			$body .= "<h1>$type</h1>";
119
120			for (1 .. cf::face::faces_size - 1) {
121			next if $type != cf::face::get_type $_;
122			my $name = cf::face::get_name $_;
123			my $id = unpack "H*", cf::face::get_chksum $_, 1;
124	root	1.9	$body .= "$_ <a href='$id'>$name ($id)</a>";
125	root	1.8	$body .= " <a href='M$id'>(meta)</a>" if $type & 1;
126			$body .= "<br>";
127			}
128	root	1.4	}
129	root	1.2
130	root	1.4	$body .= "</body></html>";
131	root	1.2
132	root	1.4	$self->respond ("200 OK", $body, "Content-Type: text/html\015\012");
133	root	1.9	} elsif ($uri eq "/ws") {
134			my $fh = $self->{fh};
135			my $buf = $orig_req . "\015\012\015\012" . $self->{rbuf};
136
137			%$self = ();
138
139			&AnyEvent::Socket::tcp_connect (@WEBSOCKETS_FORWARDER, sub {
140			my ($fh2) = shift;
141
142			Coro::async {
143			if ($fh2) {
144			my $a = copy $fh, $fh2, $buf;
145			my $b = copy $fh2, $fh;
146
147			$a->join;
148			$b->join;
149
150			close $fh2;
151			}
152
153			close $fh;
154			};
155			});
156
157	root	1.4	} else {
158			$self->respond ("404 not found");
159			}
160	root	1.1	}
161			}
162
163			# dirty hack: called directly from tcp.ext
164			sub server {
165			my $self = bless {
166	root	1.2	id => $_[0],
167			fh => $_[1],
168			rbuf => $_[2],
169	root	1.1	wbuf => "",
170			};
171
172			$self->{rw} = AE::io $self->{fh}, 0, sub {
173			my $len = sysread $self->{fh}, $self->{rbuf}, 4096, length $self->{rbuf};
174
175			if ($len == 0) {
176			delete $self->{rw};
177			} else {
178			$self->handle_req;
179
180			delete $self->{rw} if length $self->{rbuf} > 8192; # headers too long
181			}
182			};
183
184			$self->handle_req; # in the unlikely case of the buffer already forming a valid request
185			}
186
187			cf::register_exticmd http_faceurl => sub {
188			my ($ns) = @_;
189
190			"/"
191			};
192