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;

      # we ignore headers atm.

      $req =~ m%^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 = "$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.10
Committed:	Wed Oct 31 20:04:44 2012 UTC (11 years, 6 months ago) by root
Branch:	MAIN
Changes since 1.9:	+2 -3 lines
Log Message:	* empty log message *
#	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.1
74	root	1.4	# we ignore headers atm.
75	root	1.1
76	root	1.10	$req =~ m%^GET (\S+) HTTP/[0-9.]+\015\012%
77	root	1.4	or return $self->fatal;
78	root	1.1
79	root	1.4	my $uri = $1;
80	root	1.1
81	root	1.4	$uri =~ s%^http://[^/]*%%i; # just in case
82	root	1.1
83	root	1.4	cf::debug "HTTP GET: $self->{id} $uri";
84	root	1.2
85	root	1.8	if ($uri =~ m%^/(M?)([0-9a-f]+)$%) { # faces
86			my $want_meta = $1;
87			my $idx = $cf::FACEHASH{pack "H*", $2};
88	root	1.2
89	root	1.4	$idx
90	root	1.7	or do { $self->respond ("404 illegal face name"), next };
91	root	1.1
92	root	1.4	if ($req =~ /if-none-match/i) { # dirtiest hack evar
93	root	1.6	$self->respond ("304 not modified", "", $cache_headers);
94	root	1.7	next;
95	root	1.4	}
96	root	1.1
97	root	1.4	my $type = cf::face::get_type $idx, 1;
98	root	1.8	my $data = cf::face::get_data $idx, 1;
99	root	1.1
100	root	1.8	(my $meta, $data) = unpack "(w/a)", $data
101			if $type & 1;
102
103			if ($want_meta) {
104			if ($type & 1) {
105			$self->respond ("200 OK", $meta, "content-type: text/plain\015\012" . $cache_headers);
106	root	1.4	} else {
107	root	1.8	$self->respond ("404 type $type has no metadata");
108	root	1.4	}
109	root	1.8	} else {
110			$self->respond ("200 OK", $data, (content_type $data) . $cache_headers);
111	root	1.1	}
112	root	1.8
113			} elsif ($uri eq "/debug") { # for debugging
114	root	1.4	my $body = "<html><body>";
115	root	1.2
116	root	1.8	for my $type (6, 5, 4, 3, 2, 1, 0) {
117			$body .= "<h1>$type</h1>";
118
119			for (1 .. cf::face::faces_size - 1) {
120			next if $type != cf::face::get_type $_;
121			my $name = cf::face::get_name $_;
122			my $id = unpack "H*", cf::face::get_chksum $_, 1;
123	root	1.9	$body .= "$_ <a href='$id'>$name ($id)</a>";
124	root	1.8	$body .= " <a href='M$id'>(meta)</a>" if $type & 1;
125			$body .= "<br>";
126			}
127	root	1.4	}
128	root	1.2
129	root	1.4	$body .= "</body></html>";
130	root	1.2
131	root	1.4	$self->respond ("200 OK", $body, "Content-Type: text/html\015\012");
132	root	1.9	} elsif ($uri eq "/ws") {
133			my $fh = $self->{fh};
134	root	1.10	my $buf = "$req\015\012\015\012$self->{rbuf}";
135	root	1.9
136			%$self = ();
137
138			&AnyEvent::Socket::tcp_connect (@WEBSOCKETS_FORWARDER, sub {
139			my ($fh2) = shift;
140
141			Coro::async {
142			if ($fh2) {
143			my $a = copy $fh, $fh2, $buf;
144			my $b = copy $fh2, $fh;
145
146			$a->join;
147			$b->join;
148
149			close $fh2;
150			}
151
152			close $fh;
153			};
154			});
155
156	root	1.4	} else {
157			$self->respond ("404 not found");
158			}
159	root	1.1	}
160			}
161
162			# dirty hack: called directly from tcp.ext
163			sub server {
164			my $self = bless {
165	root	1.2	id => $_[0],
166			fh => $_[1],
167			rbuf => $_[2],
168	root	1.1	wbuf => "",
169			};
170
171			$self->{rw} = AE::io $self->{fh}, 0, sub {
172			my $len = sysread $self->{fh}, $self->{rbuf}, 4096, length $self->{rbuf};
173
174			if ($len == 0) {
175			delete $self->{rw};
176			} else {
177			$self->handle_req;
178
179			delete $self->{rw} if length $self->{rbuf} > 8192; # headers too long
180			}
181			};
182
183			$self->handle_req; # in the unlikely case of the buffer already forming a valid request
184			}
185
186			cf::register_exticmd http_faceurl => sub {
187			my ($ns) = @_;
188
189			"/"
190			};
191