Diff for /loncom/lonmaxima between versions 1.1 and 1.12

version 1.1, 2006/03/03 16:07:34 version 1.12, 2006/03/08 03:18:42
Line 29 Line 29
 #   # 
 # http://www.lon-capa.org/  # http://www.lon-capa.org/
 #  #
   
     
 use IPC::Open3;  use IPC::Open3;
 use IO::Select;  use IO::Select;
 # Scary: cannot use strict!!!  use IO::Socket;
 ##### use strict;  use IO::File;
   use Symbol;
   use POSIX;
   use lib '/home/httpd/lib/perl/';
   use LONCAPA::Configuration;
    
   use strict;
   
   # global variables
   my $STARTPORT              = 5664;     # port for first child's server
   my $PREFORK                = 5;        # number of children to maintain
   my $MAX_CLIENTS_PER_CHILD  = 5;        # number of clients each child should process
   my %children               = ();       # keys are current child process IDs
   my %usedmaximaports        = ();       # keys are the used maximaports
   my $children               = 0;        # current number of children
   my $status;                            # string for current status
   my $pidfile;                           # file containg parent process pid
   my $port;                              # path to UNIX socket file
   my %perlvar;                           # configuration file info
   my $lastlog;                           # last string that was logged
   use vars qw($PREFORK $MAX_CLIENTS_PER_CHILD %children $children %usedmaximaports $status
       $pidfile $port %perlvar $lastlog);
    
 sub maximareply {  sub maximareply {
     my $cmd=shift;      my ($cmd,$maximaclient,$maximapid) = @_;
     my $reply='';      my $reply='';
     my $error='';      while (my $line=<$maximaclient>) {
     my $exitstatus='';         $reply.=$line;
       }
     unless ($cmd=~/\;\n$/) { $cmd.=";\n"; }      unless ($cmd=~/\;\n$/) { $cmd.=";\n"; }
     my $pid = open3($cmd_in, $cmd_out, $cmd_err, 'maxima');      print $maximaclient $cmd;
           &status("Command sent to $maximapid");
     $SIG{CHLD} = sub {      while (my $line=<$maximaclient>) {
  $exitstatus="$? on $pid\n" if waitpid($pid, 0) > 0;         $reply.=$line;
     };      }
       &status("Command processed by $maximapid");
       return $reply;
   }
    
   # ------------------------------------------------------------ Service routines 
   sub REAPER {                        # takes care of dead children 
                                       # and MAXIMA processes
       $SIG{CHLD} = \&REAPER;
       my $pid = wait;
       $children--;
       delete($usedmaximaports{$children{$pid}});
       delete($children{$pid});
   }
    
   sub HUNTSMAN {                      # signal handler for SIGINT
       local($SIG{CHLD}) = 'IGNORE';   # we're going to kill our children
       kill('INT' => keys(%children));
       unlink($pidfile);
       unlink($port);
       &logthis('---- Shutdown ----');
       exit;                           # clean up with dignity
   }
   
     print $cmd_in $cmd;  
     close $cmd_in;  
   
     my $selector = IO::Select->new( );   
     $selector->add($cmd_err, $cmd_out);  # --------------------------------------------------------------------- Logging
    
   sub logthis {
       my ($message)=@_;
       my $execdir=$perlvar{'lonDaemons'};
       my $fh=IO::File->new(">>$execdir/logs/lonmaxima.log");
       my $now=time;
       my $local=localtime($now);
       $lastlog=$local.': '.$message;
       print $fh "$local ($$): $message\n";
   }
    
   # -------------------------------------------------------------- Status setting
    
   sub status {
       my ($what)=@_;
       my $now=time;
       my $local=localtime($now);
       $status=$local.': '.$what;
       $0='lonmaxima: '.$what.' '.$local;
   }
    
   # -------------------------------------------------------- Escape Special Chars
    
   sub escape {
       my ($str)=@_;
       $str =~ s/(\W)/"%".unpack('H2',$1)/eg;
       return $str;
   }
    
   # ----------------------------------------------------- Un-Escape Special Chars
    
   sub unescape {
       my ($str)=@_;
       $str =~ s/%([a-fA-F0-9][a-fA-F0-9])/pack("C",hex($1))/eg;
       return $str;
   }
    
   # ------------------------ grabs exception and records it to log before exiting
   sub catchexception {
       my ($signal)=@_;
       $SIG{QUIT}='DEFAULT';
       $SIG{__DIE__}='DEFAULT';
       chomp($signal);
       &logthis("<font color=\"red\">CRITICAL: "
        ."ABNORMAL EXIT. Child $$ died through "
        ."\"$signal\"</font>");
       die("Signal abend");
   }
   
   
   
   # ---------------------------------------------------------------- Main program
   # -------------------------------- Set signal handlers to record abnormal exits
    
    
   $SIG{'QUIT'}=\&catchexception;
   $SIG{__DIE__}=\&catchexception;
    
   # ---------------------------------- Read loncapa_apache.conf and loncapa.conf
   &status("Read loncapa.conf and loncapa_apache.conf");
   %perlvar=%{&LONCAPA::Configuration::read_conf('loncapa.conf')};
    
   # ----------------------------- Make sure this process is running from user=www
   my $wwwid=getpwnam('www');
   if ($wwwid!=$<) {
       my $emailto="$perlvar{'lonAdmEMail'},$perlvar{'lonSysEMail'}";
       my $subj="LON: User ID mismatch";
       system("echo 'User ID mismatch.  lonmaxima must be run as user www.' |\
    mailto $emailto -s '$subj' > /dev/null");
       exit 1;
   }
    
   # --------------------------------------------- Check if other instance running
    
   $pidfile="$perlvar{'lonDaemons'}/logs/lonmaxima.pid";
    
   if (-e $pidfile) {
       my $lfh=IO::File->new("$pidfile");
       my $pide=<$lfh>;
       chomp($pide);
       if (kill(0 => $pide)) { die "already running"; }
   }
   
   # ------------------------------------------------------- Listen to UNIX socket
   &status("Opening socket");
    
   $port = "$perlvar{'lonSockDir'}/maximasock";
    
   unlink($port);
    
   
   my $server = IO::Socket::UNIX->new(Local  => $port,
      Type   => SOCK_STREAM,
      Listen => 10 );
   if (!$server) {
       my $st=120+int(rand(240));
   
       &logthis("<font color=blue>WARNING: ".
        "Can't make server socket ($st secs):  .. exiting</font>");
   
       sleep($st);
       exit;
   }
           
     while (my @ready = $selector->can_read) {   
  foreach my $fh (@ready) {  # ---------------------------------------------------- Fork once and dissociate
     if (fileno($fh) == fileno($cmd_err)) {   
  $error.=<$cmd_err>;  my $fpid=fork;
     } else {  exit if $fpid;
  my $line = scalar <$cmd_out>;  die("Couldn't fork: $!") unless defined($fpid);
                 if ($line=~/^(\(\%o|\s)/) {   
                    $line=~s/^\(.*\)/     /;   POSIX::setsid() or die "Can't start new session: $!";
                    $reply.=$line;    
        }  # ------------------------------------------------------- Write our PID on disk
     }   
     $selector->remove($fh) if eof($fh);  my $execdir=$perlvar{'lonDaemons'};
  }  open(PIDSAVE,">$execdir/logs/lonmaxima.pid");
   print PIDSAVE "$$\n";
   close(PIDSAVE);
   &logthis("<font color='red'>CRITICAL: ---------- Starting ----------</font>");
   &status('Starting');
        
   
   # Install signal handlers.
   $SIG{CHLD} = \&REAPER;
   $SIG{INT}  = $SIG{TERM} = \&HUNTSMAN;
   
   my $maximaport=$STARTPORT; 
   # Fork off our children.
   for (1 .. $PREFORK) {
       &make_new_child($server,$maximaport);
       $maximaport++;
   }
    
   # And maintain the population.
   while (1) {
       &status('Parent process, sleeping');
       sleep;                          # wait for a signal (i.e., child's death)
       for (my $i = $children; $i < $PREFORK; $i++) {
           &status('Parent process, starting child');
           my $newport;
           foreach $newport ($STARTPORT .. $STARTPORT+$PREFORK-1) {
               if (!defined($usedmaximaports{$newport})) { last; }
           }
           &make_new_child($server,$newport);           # top up the child pool
     }      }
     close $cmd_out;  
     close $cmd_err;  
     return ($reply,$error,$exitstatus);  
 }  }
                                                                                   
   sub make_new_child {
       my ($server,$maximaport) = @_;
   
       # block signal for fork
       my $sigset = POSIX::SigSet->new(SIGINT);
       sigprocmask(SIG_BLOCK, $sigset)
           or die("Can't block SIGINT for fork: $!\n");
        
       die("fork: $!") unless defined(my $pid = fork);
        
       if ($pid) {
           # Parent records the child's birth and returns.
           sigprocmask(SIG_UNBLOCK, $sigset)
               or die("Can't unblock SIGINT for fork: $!\n");
           $children{$pid} = $maximaport;
           $children++;
           return;
       } else {
           # Child can *not* return from this subroutine.
           $SIG{INT} = 'DEFAULT';      # make SIGINT kill us as it did before
        
           # unblock signals
           sigprocmask(SIG_UNBLOCK, $sigset)
               or die("Can't unblock SIGINT for fork: $!\n");
           
           # open the MAXIMA port
           my $maximaserver = IO::Socket::INET->new(LocalPort => $maximaport,
                                                 Type      => SOCK_STREAM,
                                                 Proto     => 'tcp',
                                                 Reuse     => 1,
                                                 Listen    => 10 )
                           or die "making socket: $@\n";
   
 print join("\n----\n",&maximareply('1234'));          # open MAXIMA to talk to that port
 print join("\n----\n",&maximareply('x0: 5;x1: 7;integrate (x^2, x, x0, x1);'));          my ($cmd_in, $cmd_out, $cmd_err);
           my $maximapid = open3($cmd_in, $cmd_out, $cmd_err, "maxima -s $maximaport");
           $children{$maximapid} = 1;
   
           my $prompt=<$cmd_out>;
           &logthis("Maxima $maximapid: $prompt");
   
           &status("Waiting $maximapid on $maximaport");
           # Hopefully, MAXIMA calls us back
           my $maximaclient=$maximaserver->accept();
           &status("$maximapid on $maximaport connected.");
           &logthis("Maxima $maximapid on port $maximaport connected.");
   
           # Absorb initial prompts     
           &logthis(&maximareply("0;\n",$maximaclient,$maximapid));
    
           # Ready for action  
   
           &process_requests($server,$maximaclient,$maximapid);
   
           # tidy up gracefully and finish
   
           if (ref($cmd_out)) { close($cmd_out); }
           if (ref($cmd_err)) { close($cmd_err); }
           if (ref($cmd_in)) { close($cmd_in); }
   
           # this exit is VERY important, otherwise the child will become
           # a producer of more and more children, forking yourself into
           # process death.
           exit;
       }
   }
   
   sub process_requests {
       my ($server,$maximaclient,$maximapid) = @_;
       # handle connections until we've reached $MAX_CLIENTS_PER_CHILD
       for (my $i=0; $i < $MAX_CLIENTS_PER_CHILD; $i++) {
    &status('Accepting connections for '.$maximapid.' on '.$maximaport);     
    my $client = $server->accept()     or last;
    while (my $cmd=<$client>) {
       &status('Processing command by '.$maximapid.' on '.$maximaport);
       print $client &escape((&maximareply(&unescape($cmd),$maximaclient,$maximapid))[0])."\n";
    }
       }    
   }

Removed from v.1.1  
changed lines
  Added in v.1.12


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>