lib/PXElator/httpd.pm

package httpd;

use warnings;
use strict;
use autodie;

=head1 httpd

Start with:

  perl -Ilib/PXElator -Mhttpd -e httpd::start

=cut

use Data::Dump qw/dump/;
use Carp qw/confess/;
use File::Slurp;
#use JSON;
use IO::Socket::INET;
use Module::Refresh;

our $pids = { httpd => $$ };

sub DESTROY {
        warn "pids ",dump( $pids );
        foreach ( values %$pids ) {
                warn "kill $_";
                kill 1,$_ || kill 9, $_;
        }
}

our $port = 7777;

use server;
our $debug = server::debug;
our $url = "http://$server::ip:$port";

use html;

sub static {
        my ($client,$path) = @_;

        my $full = "$server::base_dir/tftp/$path";

        return if ! -f $full;

        my $type = 'application/octet-stream';
        $type = 'text/html' if $path =~ m{\.htm};
        $type = 'application/javascript' if $path =~ m{\.js};
        $type = 'text/plain' if $path =~ m{\.txt};

        my $size = -s $full || return;

        print $client "HTTP/1.0 200 OK\r\nContent-Type: $type\r\nContent-Length: $size\r\nConnection: close\r\n\r\n";

        open(my $fh, $full);
        print "static $path $type $size\n";

        my $block = 8192;
        my $buff;
        my $pos = 0;

        while( my $len = read $fh, $buff, $block ) {
                print $client $buff;
                $pos += $len;
                printf "%s %d/%d %.2f%%\r", $path, $pos, $size, $pos * 100 / $size;
        }
        close($fh);
        close($client);

        print "$path $pos == $size OK\n";

        return $path;
}

use boolean;

use screen;
use kvm;

$SIG{CHLD} = 'IGNORE';

sub start_stop {
        my $daemon = shift;
        my $pid = $pids->{$daemon};

        warn "start_stop $daemon $pid pids: ",dump( $pids );

        if ( $pid =~ m{^\d+$} ) {
                warn "kill 9 $pid";
                kill 9, $pid;
                $pids->{$daemon} = 'stopped';
                return qq|$daemon pid $pid stopped|;
        } else {
                if ( $pid = fork ) {
                        # parent
                        $pids->{$daemon} = $pid;
                        warn "forked $daemon $pid";
                        return qq|$daemon pid $pid started|;
                } elsif ( defined $pid ) {
                        # child
                        my $eval = $daemon . '::start(' . ( @_ ? dump(@_) : '' ) . ')';
                        warn "eval $eval";
                        eval $eval;
                        warn "can't start $daemon: $@" if $@;
                        exit;
                } else {
                        die "fork error $!";
                }
        }
}

my $ok =       qq|HTTP/1.0 200 OK\r\nContent-Type: text/html\r\nConnection: close\r\n\r\n|;
my $redirect = qq|HTTP/1.1 302 Found\r\nContent-type: text/html\r\nLocation: $url\r\n\r\n|;

sub get_request {
        my ( $client, $path, $param ) = @_;

        warn "get_request $client $path ",dump( $param );

        if ( my $found = static( $client,$path ) ) {
                warn "static $found" if $debug;
        } elsif ( $path eq '/' ) {

                my $screen = $pids->{screen} ? qq|stop <tt>$pids->{screen}</tt>|        : 'start';
                my $kvm    = $pids->{kvm}    ? qq|stop <tt>$pids->{kvm}</tt>|           :
                                         $pids->{screen} ? qq|start|                                                    : qq|start screen first|;

                my @rows = (
                        'ip',           html::tt( $server::ip ),
                        'netmask',      html::tt( $server::netmask ),

                        'debug',        qq|<a href=/our/debug/| . boolean::toggle($debug) . qq|>$debug</a>|,
                );
                foreach my $name ( %$pids ) {
                        my $pid = $pids->{$name} || next;
                        my $proc = "/proc/$pid/status";

                        my $html = qq|<a href=/$name>$pid</a>|;
                        $html   .= qq|<pre style="font-size: 10%">|
                                        .  ( $debug && -e $proc ? read_file($proc) : '' )
                                        .  qq|</pre>| if $debug;

                        push @rows, ( $name => $html );
                }

                print $client $ok, html::table( 2, @rows );

        } elsif ( $path =~ m{^/our/(\w+)/(\S+)} ) {
                eval 'our $' . $1 . ' = ' . $2;
                warn $@ if $@;
                print $client $redirect, qq|<big>$1 = $2</big><br>Location: <a href="$url">$url</a>|;
                server::debug( $debug ) if $1 eq 'debug';
        } elsif ( $path =~ m{^/(screen|kvm)} ) {
                print $client $redirect, start_stop($1);
        } elsif ( $path eq '/exit' ) {
#               DESTROY;
                exit 0;
        } elsif ( $path =~ m{/boot} ) {
                print $client qq{$ok
#!gpxe
imgfree
login
chain http://$server::ip:$httpd::port/

                        };
        } else {
                print $client "HTTP/1.0 404 $path\r\nConnection: close\r\nContent-type: text/html\r\n\r\n<big>404 $path</big>";
                warn "404 $path";
        }

}

use browser;

sub start {

        my $server = IO::Socket::INET->new(
                        Proto     => 'tcp',
                        LocalPort => $httpd::port,
                        Listen    => SOMAXCONN,
                        Reuse     => 1
        ) || die "can't start server on $url: $!";

        print "url $url\n";

        start_stop 'browser', $url;
        start_stop 'screen';
        start_stop 'kvm';

        while (my $client = $server->accept()) {
                $client->autoflush(1);
                my $request = <$client>;

                warn "request $request\n" if $debug;

                Module::Refresh->refresh;

                if ($request =~ m{^GET (/.*) HTTP/1.[01]}) {
                        my $path = $1;
                        my $param;
                        if ( $path =~ s{\?(.+)}{} ) {
                                foreach my $p ( split(/[&;]/, $1) ) {
                                        my ($n,$v) = split(/=/, $p, 2);
                                        $param->{$n} = $v;
                                }
                                warn "param: ",dump( $param ) if $debug;
                        }
                        warn "path $path param: ",dump( $param );
                        get_request $client, $path, $param;
                } else {
                        print $client "HTTP/1.0 500 No method\r\nConnection: close\r\nContent-type: text/plain\r\n\r\n500 $request";
                        warn "500 $request";
                }

                print $client qq{
                <div style="font-size: 80%; color: #888">
                <a href="">reload</a>
                <a href=/>index</a>
                <a href=/exit>exit</a>
                </div>
                } if $client->connected;

        }

        die "server died";
}

warn "loaded";

1;
1	dpavlin	42	package httpd;
2
3			use warnings;
4			use strict;
5			use autodie;
6
7			=head1 httpd
8
9			Start with:
10
11			perl -Ilib/PXElator -Mhttpd -e httpd::start
12
13			=cut
14
15			use Data::Dump qw/dump/;
16			use Carp qw/confess/;
17			use File::Slurp;
18			#use JSON;
19			use IO::Socket::INET;
20	dpavlin	43	use Module::Refresh;
21	dpavlin	42
22	dpavlin	66	our $pids = { httpd => $$ };
23
24			sub DESTROY {
25			warn "pids ",dump( $pids );
26			foreach ( values %$pids ) {
27			warn "kill $_";
28			kill 1,$_ \|\| kill 9, $_;
29			}
30			}
31
32	dpavlin	42	our $port = 7777;
33
34			use server;
35	dpavlin	67	our $debug = server::debug;
36	dpavlin	64	our $url = "http://$server::ip:$port";
37	dpavlin	42
38	dpavlin	43	use html;
39
40	dpavlin	42	sub static {
41			my ($client,$path) = @_;
42
43	dpavlin	65	my $full = "$server::base_dir/tftp/$path";
44	dpavlin	42
45	dpavlin	65	return if ! -f $full;
46	dpavlin	42
47	dpavlin	65	my $type = 'application/octet-stream';
48	dpavlin	42	$type = 'text/html' if $path =~ m{\.htm};
49			$type = 'application/javascript' if $path =~ m{\.js};
50	dpavlin	65	$type = 'text/plain' if $path =~ m{\.txt};
51	dpavlin	42
52	dpavlin	65	my $size = -s $full \|\| return;
53
54			print $client "HTTP/1.0 200 OK\r\nContent-Type: $type\r\nContent-Length: $size\r\nConnection: close\r\n\r\n";
55
56			open(my $fh, $full);
57			print "static $path $type $size\n";
58
59			my $block = 8192;
60			my $buff;
61			my $pos = 0;
62
63			while( my $len = read $fh, $buff, $block ) {
64			print $client $buff;
65			$pos += $len;
66			printf "%s %d/%d %.2f%%\r", $path, $pos, $size, $pos * 100 / $size;
67	dpavlin	42	}
68	dpavlin	65	close($fh);
69			close($client);
70	dpavlin	42
71	dpavlin	65	print "$path $pos == $size OK\n";
72
73	dpavlin	42	return $path;
74			}
75
76	dpavlin	43	use boolean;
77	dpavlin	57
78	dpavlin	53	use screen;
79	dpavlin	57	use kvm;
80	dpavlin	42
81	dpavlin	57	$SIG{CHLD} = 'IGNORE';
82
83			sub start_stop {
84			my $daemon = shift;
85			my $pid = $pids->{$daemon};
86
87	dpavlin	67	warn "start_stop $daemon $pid pids: ",dump( $pids );
88	dpavlin	57
89	dpavlin	67	if ( $pid =~ m{^\d+$} ) {
90	dpavlin	57	warn "kill 9 $pid";
91			kill 9, $pid;
92	dpavlin	66	$pids->{$daemon} = 'stopped';
93	dpavlin	57	return qq\|$daemon pid $pid stopped\|;
94			} else {
95			if ( $pid = fork ) {
96			# parent
97			$pids->{$daemon} = $pid;
98			warn "forked $daemon $pid";
99			return qq\|$daemon pid $pid started\|;
100			} elsif ( defined $pid ) {
101			# child
102	dpavlin	67	my $eval = $daemon . '::start(' . ( @_ ? dump(@_) : '' ) . ')';
103	dpavlin	57	warn "eval $eval";
104			eval $eval;
105			warn "can't start $daemon: $@" if $@;
106			exit;
107			} else {
108			die "fork error $!";
109			}
110			}
111			}
112
113	dpavlin	67	my $ok = qq\|HTTP/1.0 200 OK\r\nContent-Type: text/html\r\nConnection: close\r\n\r\n\|;
114			my $redirect = qq\|HTTP/1.1 302 Found\r\nContent-type: text/html\r\nLocation: $url\r\n\r\n\|;
115
116	dpavlin	43	sub get_request {
117			my ( $client, $path, $param ) = @_;
118
119	dpavlin	53	warn "get_request $client $path ",dump( $param );
120	dpavlin	43
121			if ( my $found = static( $client,$path ) ) {
122			warn "static $found" if $debug;
123			} elsif ( $path eq '/' ) {
124	dpavlin	53
125	dpavlin	57	my $screen = $pids->{screen} ? qq\|stop <tt>$pids->{screen}</tt>\| : 'start';
126			my $kvm = $pids->{kvm} ? qq\|stop <tt>$pids->{kvm}</tt>\| :
127			$pids->{screen} ? qq\|start\| : qq\|start screen first\|;
128	dpavlin	53
129	dpavlin	66	my @rows = (
130	dpavlin	64	'ip', html::tt( $server::ip ),
131			'netmask', html::tt( $server::netmask ),
132	dpavlin	66
133	dpavlin	64	'debug', qq\|<a href=/our/debug/\| . boolean::toggle($debug) . qq\|>$debug</a>\|,
134	dpavlin	43	);
135	dpavlin	66	foreach my $name ( %$pids ) {
136			my $pid = $pids->{$name} \|\| next;
137	dpavlin	67	my $proc = "/proc/$pid/status";
138	dpavlin	43
139	dpavlin	66	my $html = qq\|<a href=/$name>$pid</a>\|;
140			$html .= qq\|<pre style="font-size: 10%">\|
141	dpavlin	67	. ( $debug && -e $proc ? read_file($proc) : '' )
142	dpavlin	66	. qq\|</pre>\| if $debug;
143
144			push @rows, ( $name => $html );
145			}
146
147			print $client $ok, html::table( 2, @rows );
148
149	dpavlin	43	} elsif ( $path =~ m{^/our/(\w+)/(\S+)} ) {
150			eval 'our $' . $1 . ' = ' . $2;
151			warn $@ if $@;
152	dpavlin	67	print $client $redirect, qq\|<big>$1 = $2</big><br>Location: <a href="$url">$url</a>\|;
153			server::debug( $debug ) if $1 eq 'debug';
154	dpavlin	57	} elsif ( $path =~ m{^/(screen\|kvm)} ) {
155	dpavlin	67	print $client $redirect, start_stop($1);
156	dpavlin	66	} elsif ( $path eq '/exit' ) {
157			# DESTROY;
158			exit 0;
159	dpavlin	43	} elsif ( $path =~ m{/boot} ) {
160			print $client qq{$ok
161			#!gpxe
162			imgfree
163			login
164			chain http://$server::ip:$httpd::port/
165
166			};
167			} else {
168			print $client "HTTP/1.0 404 $path\r\nConnection: close\r\nContent-type: text/html\r\n\r\n<big>404 $path</big>";
169			warn "404 $path";
170			}
171
172			}
173	dpavlin	53
174	dpavlin	66	use browser;
175
176	dpavlin	42	sub start {
177
178			my $server = IO::Socket::INET->new(
179			Proto => 'tcp',
180			LocalPort => $httpd::port,
181			Listen => SOMAXCONN,
182			Reuse => 1
183	dpavlin	43	) \|\| die "can't start server on $url: $!";
184	dpavlin	42
185			print "url $url\n";
186
187	dpavlin	66	start_stop 'browser', $url;
188			start_stop 'screen';
189			start_stop 'kvm';
190	dpavlin	42
191			while (my $client = $server->accept()) {
192			$client->autoflush(1);
193			my $request = <$client>;
194
195			warn "request $request\n" if $debug;
196
197	dpavlin	43	Module::Refresh->refresh;
198
199	dpavlin	42	if ($request =~ m{^GET (/.*) HTTP/1.[01]}) {
200	dpavlin	43	my $path = $1;
201	dpavlin	42	my $param;
202	dpavlin	43	if ( $path =~ s{\?(.+)}{} ) {
203	dpavlin	42	foreach my $p ( split(/[&;]/, $1) ) {
204			my ($n,$v) = split(/=/, $p, 2);
205			$param->{$n} = $v;
206			}
207			warn "param: ",dump( $param ) if $debug;
208			}
209	dpavlin	43	warn "path $path param: ",dump( $param );
210			get_request $client, $path, $param;
211	dpavlin	42	} else {
212			print $client "HTTP/1.0 500 No method\r\nConnection: close\r\nContent-type: text/plain\r\n\r\n500 $request";
213			warn "500 $request";
214			}
215
216			print $client qq{
217	dpavlin	43	<div style="font-size: 80%; color: #888">
218			<a href="">reload</a>
219	dpavlin	66	<a href=/>index</a>
220			<a href=/exit>exit</a>
221	dpavlin	43	</div>
222			} if $client->connected;
223	dpavlin	42
224			}
225
226			die "server died";
227			}
228
229	dpavlin	45	warn "loaded";
230
231	dpavlin	42	1;