server/ext/dm-support.ext

#! perl

use Coro::Debug;
use IO::Socket;

my %global; # for use by eval'ed commands

sub tcp_serve($) {
   my ($fh) = @_;

   binmode $fh, ":raw:perlio:utf8";

   my $buf;
   my ($a, $b, $c, $d, $e, $f, $g, $h, $i, $j, $k, $l, $m, $n, $o, $p, $q, $r, $s, $t, $u, $v, $w, $x, $y, $z);
   my (%l, @l); # for use by shell users

   print $fh <<EOF;
Welcome!
Remember that everything entered here will be in the main coro context within cf::!
You can freely use \$a .. \$z and \@l and \%l

Useful commands (note the '&'):

reload_perl &
reload_config &
reload_regions &
reload_facedata &
reload_treasures &
reload_archetypes &
ext::help::reload &
ext::books::reload &
ext::map_tags::reload &
ext::map_world::reload &
EOF
   print $fh "\n> ";

   Event->io (fd => $fh, poll => 'r', data => 0, cb => sub {
      if (defined (my $cmd = <$fh>)) {
         $cmd =~ s/\s+$//;

         if ($cmd =~ /^\s*exit\b/i) {
            print $fh "will not exit() server.\n";
         } elsif ($cmd =~ s/^coro\s+// or $cmd =~ /^(?:ps|bt\s)/) {
            select $fh;
            Coro::Debug::command $cmd;
            select STDOUT;
         } else {
            my $sub = sub {
               package cf;
               select $fh;

               # compile first, the execute, as Coro does not support switching in eval string
               my $cb = eval "sub { $cmd }";

               my $t1 = Time::HiRes::time;
               my @res = $@ ? () : eval { $cb->() };
               my $t2 = Time::HiRes::time;

               print "\n",
                     "command: '$cmd'\n",
                     "execution time: ", $t2 - $t1, "\n";
               warn "evaluation error: $@" if $@;
               print "evaluation error: $@\n" if $@;
               print "result:\n", cf::dumpval @res > 1 ? \@res : $res[0] if @res;
               print "\n> ";

               select STDOUT;
            };

            if ($cmd =~ s/\s*&$//) {
               cf::async {
                  $Coro::current->desc ($cmd);
                  $sub->()
               };
            } else {
               $sub->();
            }
         }

         print $fh "\n> ";
      } else {
         $_[0]->w->cancel;
      }
   });
}

# now a shell listening on a tcp-port - let the firewall decide access rights
if ($cf::CFG{perl_shell}) {
   if (my $listen = new IO::Socket::INET LocalAddr => $cf::CFG{perl_shell}, Listen => 1, ReuseAddr => 1) {
      Event->io (fd => $listen, poll => 'r', data => cf::WF_AUTOCANCEL, cb => sub { tcp_serve $listen->accept });
   }
}


Revision:	1.14
Committed:	Fri Oct 12 19:13:26 2007 UTC (16 years, 7 months ago) by root
Branch:	MAIN
Changes since 1.13:	+0 -1 lines
Log Message:	slightly better can_merge
#	User	Rev	Content
1	root	1.1	#! perl
2
3	root	1.10	use Coro::Debug;
4	root	1.1	use IO::Socket;
5
6			my %global; # for use by eval'ed commands
7
8			sub tcp_serve($) {
9			my ($fh) = @_;
10
11			binmode $fh, ":raw:perlio:utf8";
12
13			my $buf;
14	root	1.2	my ($a, $b, $c, $d, $e, $f, $g, $h, $i, $j, $k, $l, $m, $n, $o, $p, $q, $r, $s, $t, $u, $v, $w, $x, $y, $z);
15			my (%l, @l); # for use by shell users
16	root	1.1
17	root	1.12	print $fh <<EOF;
18	root	1.4	Welcome!
19			Remember that everything entered here will be in the main coro context within cf::!
20			You can freely use \$a .. \$z and \@l and \%l
21
22			Useful commands (note the '&'):
23
24			reload_perl &
25	root	1.8	reload_config &
26	root	1.4	reload_regions &
27			reload_facedata &
28			reload_treasures &
29			reload_archetypes &
30	root	1.6	ext::help::reload &
31			ext::books::reload &
32	root	1.9	ext::map_tags::reload &
33	root	1.5	ext::map_world::reload &
34	root	1.4	EOF
35	root	1.12	print $fh "\n> ";
36	root	1.1
37	root	1.3	Event->io (fd => $fh, poll => 'r', data => 0, cb => sub {
38	root	1.1	if (defined (my $cmd = <$fh>)) {
39	root	1.4	$cmd =~ s/\s+$//;
40	root	1.1
41			if ($cmd =~ /^\s*exit\b/i) {
42	root	1.13	print $fh "will not exit() server.\n";
43	root	1.10	} elsif ($cmd =~ s/^coro\s+// or $cmd =~ /^(?:ps\|bt\s)/) {
44	root	1.13	select $fh;
45	root	1.10	Coro::Debug::command $cmd;
46	root	1.13	select STDOUT;
47	root	1.1	} else {
48	root	1.4	my $sub = sub {
49			package cf;
50			select $fh;
51
52			# compile first, the execute, as Coro does not support switching in eval string
53			my $cb = eval "sub { $cmd }";
54
55			my $t1 = Time::HiRes::time;
56			my @res = $@ ? () : eval { $cb->() };
57			my $t2 = Time::HiRes::time;
58
59			print "\n",
60			"command: '$cmd'\n",
61			"execution time: ", $t2 - $t1, "\n";
62			warn "evaluation error: $@" if $@;
63			print "evaluation error: $@\n" if $@;
64			print "result:\n", cf::dumpval @res > 1 ? \@res : $res[0] if @res;
65			print "\n> ";
66	root	1.12
67			select STDOUT;
68	root	1.4	};
69
70			if ($cmd =~ s/\s*&$//) {
71	root	1.11	cf::async {
72			$Coro::current->desc ($cmd);
73			$sub->()
74			};
75	root	1.4	} else {
76			$sub->();
77			}
78	root	1.1	}
79
80	root	1.13	print $fh "\n> ";
81	root	1.1	} else {
82			$_[0]->w->cancel;
83			}
84			});
85			}
86
87			# now a shell listening on a tcp-port - let the firewall decide access rights
88			if ($cf::CFG{perl_shell}) {
89			if (my $listen = new IO::Socket::INET LocalAddr => $cf::CFG{perl_shell}, Listen => 1, ReuseAddr => 1) {
90			Event->io (fd => $listen, poll => 'r', data => cf::WF_AUTOCANCEL, cb => sub { tcp_serve $listen->accept });
91			}
92			}
93	root	1.10
94