Coro/myhttpd/shell.pl

# a server command shell

use Coro;
use Coro::Handle;
use Coro::Socket;
use Event;
use Time::HiRes 'time';

use Text::Abbrev;

my $last_ts = time;

my %complete;
my @commands = qw(quit squit refresh country restart block info print);

abbrev \%complete, @commands;

sub shell {
   my $fh = shift;

   while (defined (print $fh "cmd> "), $_ = <$fh>) {
      s/\015?\012$//;
      if (s/^(\S+)\s*// && (my $cmd = $complete{$1})) {
         if ($cmd eq "quit") {
            print $fh "bye bye.\n";#d#
            last;
         } elsif ($cmd eq "squit") {
            print $fh "server quit.\n";#d#
            Event::unloop;
            last;
         } elsif ($cmd eq "print") {
            my @res = eval $_;
            print $fh "eval: $@\n" if $@;
            print $fh "RES = ", (join " : ", @res), "\n";
         } elsif ($cmd eq "block") {
            print "blocked '$_'\n";#d#
            $conn::blocked{$_} = time + $::BLOCKTIME;
         } elsif ($cmd eq "info") {
            $::NOW = time+1e-6;
            my @data;
            for (values %conn::conn) {
               for (values %$_) {
                  next unless $_;
                  my $rate = sprintf "%.1f", $_->{written} / ($::NOW - $_->{time});
                  push @data, "$_->{country}/$_->{remote_addr} $_->{written} $rate $_->{method} $_->{uri}\n";
               }
            }
            print $fh sort @data;
            print $fh scalar@data, " ($::conns) connections\n";#d#
            print $fh "$::written bytes written in the last ",$::NOW - $last_ts, " seconds\n";
            printf $fh "(%.1f bytes/s)\n", $::written / ($::NOW - $last_ts);
            ($last_ts, $::written) = ($::NOW, 0);
         } elsif ($cmd eq "refresh") {
            do "config.pl";
            print $fh "config.pl: $@\n" if $@;
            %statdata_cache = ();
            read_blockuri;
            read_blockref;
         } elsif ($cmd eq "restart") {
            $::RESTART = 1;
            unloop;
            print $fh "restarting, cu!\n";
            last;
         } elsif ($cmd eq "country") {
            print $fh netgeo::ip_request($_), "\n";
         }
      } else {
         print $fh "try one of @commands\n";
      }
   }
}

# bind to tcp port
if ($CMDSHELL_PORT) {
   my $port = new Coro::Socket
        #LocalAddr => "127.0.0.1",
        LocalPort => $CMDSHELL_PORT,
        ReuseAddr => 1,
        Listen => 1,
   or die "unable to bind cmdshell port: $!";

   push @listen_sockets, $port;

   async {
      while () {
         async \&shell, scalar $port->accept;
      }
   };
}

# bind to stdin (debug)
if (1) {
   my $tty;
   open $tty, "+</dev/tty"
      and async \&shell, unblock $tty;
}

Revision:	1.17
Committed:	Tue Aug 28 02:30:49 2001 UTC (22 years, 10 months ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.16:	+1 -1 lines
Log Message:	* empty log message *
#	User	Rev	Content
1	root	1.1	# a server command shell
2
3			use Coro;
4			use Coro::Handle;
5			use Coro::Socket;
6			use Event;
7	root	1.8	use Time::HiRes 'time';
8
9	root	1.10	use Text::Abbrev;
10
11	root	1.8	my $last_ts = time;
12	root	1.1
13	root	1.10	my %complete;
14	root	1.12	my @commands = qw(quit squit refresh country restart block info print);
15	root	1.10
16			abbrev \%complete, @commands;
17
18	root	1.1	sub shell {
19			my $fh = shift;
20
21			while (defined (print $fh "cmd> "), $_ = <$fh>) {
22	root	1.2	s/\015?\012$//;
23	root	1.10	if (s/^(\S+)\s*// && (my $cmd = $complete{$1})) {
24			if ($cmd eq "quit") {
25	root	1.13	print $fh "bye bye.\n";#d#
26	root	1.10	last;
27			} elsif ($cmd eq "squit") {
28	root	1.13	print $fh "server quit.\n";#d#
29	root	1.10	Event::unloop;
30	root	1.11	last;
31	root	1.12	} elsif ($cmd eq "print") {
32	root	1.11	my @res = eval $_;
33			print $fh "eval: $@\n" if $@;
34			print $fh "RES = ", (join " : ", @res), "\n";
35	root	1.10	} elsif ($cmd eq "block") {
36			print "blocked '$_'\n";#d#
37			$conn::blocked{$_} = time + $::BLOCKTIME;
38			} elsif ($cmd eq "info") {
39			$::NOW = time+1e-6;
40			my @data;
41			for (values %conn::conn) {
42			for (values %$_) {
43			next unless $_;
44			my $rate = sprintf "%.1f", $_->{written} / ($::NOW - $_->{time});
45			push @data, "$_->{country}/$_->{remote_addr} $_->{written} $rate $_->{method} $_->{uri}\n";
46			}
47	root	1.1	}
48	root	1.10	print $fh sort @data;
49			print $fh scalar@data, " ($::conns) connections\n";#d#
50			print $fh "$::written bytes written in the last ",$::NOW - $last_ts, " seconds\n";
51			printf $fh "(%.1f bytes/s)\n", $::written / ($::NOW - $last_ts);
52			($last_ts, $::written) = ($::NOW, 0);
53			} elsif ($cmd eq "refresh") {
54	root	1.11	do "config.pl";
55			print $fh "config.pl: $@\n" if $@;
56	root	1.16	%statdata_cache = ();
57	root	1.15	read_blockuri;
58			read_blockref;
59	root	1.10	} elsif ($cmd eq "restart") {
60			$::RESTART = 1;
61			unloop;
62			print $fh "restarting, cu!\n";
63			last;
64			} elsif ($cmd eq "country") {
65	root	1.17	print $fh netgeo::ip_request($_), "\n";
66	root	1.1	}
67			} else {
68	root	1.10	print $fh "try one of @commands\n";
69	root	1.1	}
70			}
71			}
72
73			# bind to tcp port
74			if ($CMDSHELL_PORT) {
75			my $port = new Coro::Socket
76	root	1.14	#LocalAddr => "127.0.0.1",
77	root	1.1	LocalPort => $CMDSHELL_PORT,
78			ReuseAddr => 1,
79			Listen => 1,
80			or die "unable to bind cmdshell port: $!";
81	root	1.2
82			push @listen_sockets, $port;
83	root	1.1
84			async {
85	root	1.7	while () {
86			async \&shell, scalar $port->accept;
87			}
88	root	1.1	};
89			}
90
91			# bind to stdin (debug)
92			if (1) {
93			my $tty;
94			open $tty, "+</dev/tty"
95			and async \&shell, unblock $tty;
96			}
97