mirror of
git://scm.dxcluster.org/scm/spider
synced 2024-09-22 00:07:10 +00:00
345 lines
7.9 KiB
Perl
345 lines
7.9 KiB
Perl
#!/usr/bin/perl
|
|
#
|
|
# This module impliments the user facing command mode for a dx cluster
|
|
#
|
|
# Copyright (c) 1998 Dirk Koopman G1TLH
|
|
#
|
|
# $Id$
|
|
#
|
|
|
|
package DXCommandmode;
|
|
|
|
@ISA = qw(DXChannel);
|
|
|
|
use DXUtil;
|
|
use DXChannel;
|
|
use DXUser;
|
|
use DXVars;
|
|
use DXDebug;
|
|
use DXM;
|
|
use Carp;
|
|
|
|
use strict;
|
|
use vars qw(%Cache %cmd_cache);
|
|
|
|
%Cache = (); # cache of dynamically loaded routine's mod times
|
|
%cmd_cache = (); # cache of short names
|
|
|
|
#
|
|
# obtain a new connection this is derived from dxchannel
|
|
#
|
|
|
|
sub new
|
|
{
|
|
my $self = DXChannel::alloc(@_);
|
|
$self->{sort} = 'U'; # in absence of how to find out what sort of an object I am
|
|
return $self;
|
|
}
|
|
|
|
# this is how a a connection starts, you get a hello message and the motd with
|
|
# possibly some other messages asking you to set various things up if you are
|
|
# new (or nearly new and slacking) user.
|
|
|
|
sub start
|
|
{
|
|
my ($self, $line) = @_;
|
|
my $user = $self->{user};
|
|
my $call = $self->{call};
|
|
my $name = $user->{name};
|
|
|
|
$self->{name} = $name ? $name : $call;
|
|
$self->msg('l2',$self->{name});
|
|
$self->send_file($main::motd) if (-e $main::motd);
|
|
$self->msg('pr', $call);
|
|
$self->state('prompt'); # a bit of room for further expansion, passwords etc
|
|
$self->{priv} = $user->priv;
|
|
$self->{priv} = 0 if $line =~ /^(ax|te)/; # set the connection priv to 0 - can be upgraded later
|
|
$self->{consort} = $line; # save the connection type
|
|
|
|
# set some necessary flags on the user if they are connecting
|
|
$self->{wwv} = $self->{talk} = $self->{ann} = $self->{here} = $self->{dx} = 1;
|
|
$self->prompt() if $self->{state} =~ /^prompt/o;
|
|
|
|
}
|
|
|
|
#
|
|
# This is the normal command prompt driver
|
|
#
|
|
sub normal
|
|
{
|
|
my $self = shift;
|
|
my $user = $self->{user};
|
|
my $call = $self->{call};
|
|
my $cmdline = shift;
|
|
|
|
# strip out //
|
|
$cmdline =~ s|//|/|og;
|
|
|
|
# split the command line up into parts, the first part is the command
|
|
my ($cmd, $args) = $cmdline =~ /^([\w\/]+)\s*(.*)/o;
|
|
|
|
if ($cmd) {
|
|
|
|
my ($path, $fcmd);
|
|
|
|
# first expand out the entry to a command
|
|
($path, $fcmd) = search($main::localcmd, $cmd, "pl");
|
|
($path, $fcmd) = search($main::cmd, $cmd, "pl") if !$path || !$fcmd;
|
|
|
|
my @ans = $self->eval_file($path, $fcmd, $args) if $path && $fcmd;
|
|
# @ans = $self->eval_file($main::cmd, $cmd, $args) if !$ans[0];
|
|
if ($ans[0]) {
|
|
shift @ans;
|
|
$self->send(@ans) if @ans > 0;
|
|
} else {
|
|
shift @ans;
|
|
if (@ans > 0) {
|
|
$self->msg('e2', @ans);
|
|
} else {
|
|
$self->msg('e1');
|
|
}
|
|
}
|
|
} else {
|
|
$self->msg('e1');
|
|
}
|
|
|
|
# send a prompt only if we are in a prompt state
|
|
$self->prompt() if $self->{state} =~ /^prompt/o;
|
|
}
|
|
|
|
#
|
|
# This is called from inside the main cluster processing loop and is used
|
|
# for despatching commands that are doing some long processing job
|
|
#
|
|
sub process
|
|
{
|
|
my $t = time;
|
|
my @chan = DXChannel->get_all();
|
|
my $chan;
|
|
|
|
foreach $chan (@chan) {
|
|
next if $chan->sort ne 'U';
|
|
|
|
# send a prompt if no activity out on this channel
|
|
if ($t >= $chan->t + $main::user_interval) {
|
|
$chan->prompt() if $chan->{state} =~ /^prompt/o;
|
|
$chan->t($t);
|
|
}
|
|
}
|
|
}
|
|
|
|
#
|
|
# finish up a user context
|
|
#
|
|
sub finish
|
|
{
|
|
|
|
}
|
|
|
|
#
|
|
# short cut to output a prompt
|
|
#
|
|
|
|
sub prompt
|
|
{
|
|
my $self = shift;
|
|
my $call = $self->{call};
|
|
DXChannel::msg($self, 'pr', $call);
|
|
}
|
|
|
|
# broadcast a message to all users [except those mentioned after buffer]
|
|
sub broadcast
|
|
{
|
|
my $pkg = shift; # ignored
|
|
my $s = shift; # the line to be rebroadcast
|
|
my @except = @_; # to all channels EXCEPT these (dxchannel refs)
|
|
my @list = DXChannel->get_all(); # just in case we are called from some funny object
|
|
my ($chan, $except);
|
|
|
|
L: foreach $chan (@list) {
|
|
next if !$chan->sort eq 'U'; # only interested in user channels
|
|
foreach $except (@except) {
|
|
next L if $except == $chan; # ignore channels in the 'except' list
|
|
}
|
|
chan->send($s); # send it
|
|
}
|
|
}
|
|
|
|
# gimme all the users
|
|
sub get_all
|
|
{
|
|
my @list = DXChannel->get_all();
|
|
my $ref;
|
|
my @out;
|
|
foreach $ref (@list) {
|
|
push @out, $ref if $ref->sort eq 'U';
|
|
}
|
|
return @out;
|
|
}
|
|
|
|
#
|
|
# search for the command in the cache of short->long form commands
|
|
#
|
|
|
|
sub search
|
|
{
|
|
my ($path, $short_cmd, $suffix) = @_;
|
|
my ($apath, $acmd);
|
|
|
|
# commands are lower case
|
|
$short_cmd = lc $short_cmd;
|
|
dbg('command', "command: $path $short_cmd\n");
|
|
|
|
# return immediately if we have it
|
|
my ($apath, $acmd) = split ',', $cmd_cache{$short_cmd};
|
|
if ($apath && $acmd) {
|
|
dbg('command', "cached $short_cmd = ($apath, $acmd)\n");
|
|
return ($apath, $acmd) if $apath;
|
|
}
|
|
|
|
# if not guess
|
|
my @parts = split '/', $short_cmd;
|
|
my $dirfn;
|
|
my $curdir = $path;
|
|
my $p;
|
|
my $i;
|
|
|
|
for ($i = 0; $i < @parts; $i++) {
|
|
my $p = $parts[$i];
|
|
opendir(D, $curdir) or confess "can't open $curdir $!";
|
|
my @ls = readdir D;
|
|
closedir D;
|
|
my $l;
|
|
foreach $l (sort @ls) {
|
|
next if $l =~ /^\./;
|
|
if ($i < $#parts) { # we are dealing with directories
|
|
if ((-d "$curdir/$l") && $p eq substr($l, 0, length $p)) {
|
|
dbg('command', "got dir: $curdir/$l\n");
|
|
$dirfn .= "$l/";
|
|
$curdir .= "/$l";
|
|
last;
|
|
}
|
|
} else { # we are dealing with commands
|
|
next if !$l =~ /\.$suffix$/; # only look for .$suffix files
|
|
if ($p eq substr($l, 0, length $p)) {
|
|
$l =~ s/\.$suffix$//; # remove the suffix
|
|
chop $dirfn; # remove trailing /
|
|
$cmd_cache{"$short_cmd"} = join(',', ($path, "$dirfn/$l")); # cache it
|
|
dbg('command', "got path: $path cmd: $dirfn/$l\n");
|
|
return ($path, "$dirfn/$l");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return ();
|
|
}
|
|
|
|
# clear the command name cache
|
|
sub clear_cmd_cache
|
|
{
|
|
%cmd_cache = ();
|
|
}
|
|
|
|
#
|
|
# the persistant execution of things from the command directories
|
|
#
|
|
#
|
|
# This allows perl programs to call functions dynamically
|
|
#
|
|
# This has been nicked directly from the perlembed pages
|
|
#
|
|
|
|
#require Devel::Symdump;
|
|
|
|
sub valid_package_name {
|
|
my($string) = @_;
|
|
$string =~ s/([^A-Za-z0-9\/])/sprintf("_%2x",unpack("C",$1))/eg;
|
|
|
|
#second pass only for words starting with a digit
|
|
$string =~ s|/(\d)|sprintf("/_%2x",unpack("C",$1))|eg;
|
|
|
|
#Dress it up as a real package name
|
|
$string =~ s|/|_|g;
|
|
return "Emb_" . $string;
|
|
}
|
|
|
|
#borrowed from Safe.pm
|
|
sub delete_package {
|
|
my $pkg = shift;
|
|
my ($stem, $leaf);
|
|
|
|
no strict 'refs';
|
|
$pkg = "DXChannel::$pkg\::"; # expand to full symbol table name
|
|
($stem, $leaf) = $pkg =~ m/(.*::)(\w+::)$/;
|
|
|
|
my $stem_symtab = *{$stem}{HASH};
|
|
|
|
delete $stem_symtab->{$leaf};
|
|
}
|
|
|
|
sub eval_file {
|
|
my $self = shift;
|
|
my $path = shift;
|
|
my $cmdname = shift;
|
|
my $package = valid_package_name($cmdname);
|
|
my $filename = "$path/$cmdname.pl";
|
|
my $mtime = -M $filename;
|
|
|
|
# return if we can't find it
|
|
return (0, DXM::msg('e1')) if !defined $mtime;
|
|
|
|
if(defined $Cache{$package}{mtime} && $Cache{$package}{mtime } <= $mtime) {
|
|
#we have compiled this subroutine already,
|
|
#it has not been updated on disk, nothing left to do
|
|
#print STDERR "already compiled $package->handler\n";
|
|
;
|
|
} else {
|
|
local *FH;
|
|
if (!open FH, $filename) {
|
|
return (0, "Syserr: can't open '$filename' $!");
|
|
};
|
|
local($/) = undef;
|
|
my $sub = <FH>;
|
|
close FH;
|
|
|
|
#wrap the code into a subroutine inside our unique package
|
|
my $eval = qq{package DXChannel; sub $package { $sub; }};
|
|
if (isdbg('eval')) {
|
|
my @list = split /\n/, $eval;
|
|
my $line;
|
|
foreach (@list) {
|
|
dbg('eval', $_, "\n");
|
|
}
|
|
}
|
|
#print "eval $eval\n";
|
|
{
|
|
#hide our variables within this block
|
|
my($filename,$mtime,$package,$sub);
|
|
eval $eval;
|
|
}
|
|
if ($@) {
|
|
delete_package($package);
|
|
return (0, "Syserr: Eval err $@ on $package");
|
|
}
|
|
|
|
#cache it unless we're cleaning out each time
|
|
$Cache{$package}{mtime} = $mtime;
|
|
}
|
|
|
|
my @r;
|
|
my $c = qq{ \@r = \$self->$package(\@_); };
|
|
dbg('eval', "cluster cmd = $c\n");
|
|
eval $c; ;
|
|
if ($@) {
|
|
delete_package($package);
|
|
return (0, "Syserr: Eval err $@ on cached $package");
|
|
}
|
|
|
|
#take a look if you want
|
|
#print Devel::Symdump->rnew($package)->as_string, $/;
|
|
return @r;
|
|
}
|
|
|
|
1;
|
|
__END__
|