File:  [LON-CAPA] / loncom / Attic / lonc
Revision 1.35: download - view: text, annotated - select for diffs
Tue Mar 26 04:37:59 2002 UTC (22 years, 1 month ago) by foxr
Branches: MAIN
CVS tags: HEAD
Inserted support functions to allow restructuring the lonc main loop.
Note that no actual logic changes have been implemented with this
edit.

    1: #!/usr/bin/perl
    2: 
    3: # The LearningOnline Network
    4: # lonc - LON TCP-Client Domain-Socket-Server
    5: # provides persistent TCP connections to the other servers in the network
    6: # through multiplexed domain sockets
    7: #
    8: # $Id: lonc,v 1.35 2002/03/26 04:37:59 foxr Exp $
    9: #
   10: # Copyright Michigan State University Board of Trustees
   11: #
   12: # This file is part of the LearningOnline Network with CAPA (LON-CAPA).
   13: #
   14: # LON-CAPA is free software; you can redistribute it and/or modify
   15: # it under the terms of the GNU General Public License as published by
   16: # the Free Software Foundation; either version 2 of the License, or
   17: # (at your option) any later version.
   18: #
   19: # LON-CAPA is distributed in the hope that it will be useful,
   20: # but WITHOUT ANY WARRANTY; without even the implied warranty of
   21: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   22: # GNU General Public License for more details.
   23: #
   24: # You should have received a copy of the GNU General Public License
   25: # along with LON-CAPA; if not, write to the Free Software
   26: # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
   27: #
   28: # /home/httpd/html/adm/gpl.txt
   29: #
   30: # http://www.lon-capa.org/
   31: #
   32: # PID in subdir logs/lonc.pid
   33: # kill kills
   34: # HUP restarts
   35: # USR1 tries to open connections again
   36: 
   37: # 6/4/99,6/5,6/7,6/8,6/9,6/10,6/11,6/12,7/14,7/19,
   38: # 10/8,10/9,10/15,11/18,12/22,
   39: # 2/8,7/25 Gerd Kortemeyer
   40: # 12/05 Scott Harrison
   41: # 12/05 Gerd Kortemeyer
   42: # YEAR=2001
   43: # 01/10/01 Scott Harrison
   44: # 03/14/01,03/15,06/12,11/26,11/27,11/28 Gerd Kortemeyer
   45: # 12/20 Scott Harrison
   46: # YEAR=2002
   47: # 2/19/02,02/22/02,02/25/02 Gerd Kortemeyer
   48: # 3/07/02 Ron Fox 
   49: # based on nonforker from Perl Cookbook
   50: # - server who multiplexes without forking
   51: 
   52: use POSIX;
   53: use IO::Socket;
   54: use IO::Select;
   55: use IO::File;
   56: use Socket;
   57: use Fcntl;
   58: use Tie::RefHash;
   59: use Crypt::IDEA;
   60: #use Net::Ping;
   61: use LWP::UserAgent();
   62: 
   63: $status='';
   64: $lastlog='';
   65: $conserver='SHELL';
   66: $DEBUG = 0;			# Set to 1 for annoyingly complete logs.
   67: 
   68: # -------------------------------- Set signal handlers to record abnormal exits
   69: 
   70: &status("Init exception handlers");
   71: $SIG{QUIT}=\&catchexception;
   72: $SIG{__DIE__}=\&catchexception;
   73: 
   74: # ------------------------------------ Read httpd access.conf and get variables
   75: &status("Read access.conf");
   76: open (CONFIG,"/etc/httpd/conf/access.conf") || die "Can't read access.conf";
   77: 
   78: while ($configline=<CONFIG>) {
   79:     if ($configline =~ /PerlSetVar/) {
   80: 	my ($dummy,$varname,$varvalue)=split(/\s+/,$configline);
   81:         chomp($varvalue);
   82:         $perlvar{$varname}=$varvalue;
   83:     }
   84: }
   85: close(CONFIG);
   86: 
   87: # ----------------------------- Make sure this process is running from user=www
   88: &status("Check user ID");
   89: my $wwwid=getpwnam('www');
   90: if ($wwwid!=$<) {
   91:    $emailto="$perlvar{'lonAdmEMail'},$perlvar{'lonSysEMail'}";
   92:    $subj="LON: $perlvar{'lonHostID'} User ID mismatch";
   93:    system("echo 'User ID mismatch.  lonc must be run as user www.' |\
   94:  mailto $emailto -s '$subj' > /dev/null");
   95:    exit 1;
   96: }
   97: 
   98: # --------------------------------------------- Check if other instance running
   99: 
  100: my $pidfile="$perlvar{'lonDaemons'}/logs/lonc.pid";
  101: 
  102: if (-e $pidfile) {
  103:    my $lfh=IO::File->new("$pidfile");
  104:    my $pide=<$lfh>;
  105:    chomp($pide);
  106:    if (kill 0 => $pide) { die "already running"; }
  107: }
  108: 
  109: # ------------------------------------------------------------- Read hosts file
  110: 
  111: open (CONFIG,"$perlvar{'lonTabDir'}/hosts.tab") || die "Can't read host file";
  112: 
  113: while ($configline=<CONFIG>) {
  114:     my ($id,$domain,$role,$name,$ip)=split(/:/,$configline);
  115:     chomp($ip);
  116:     if ($ip) {
  117:      $hostip{$id}=$ip;
  118:      $hostname{$id}=$name;
  119:     }
  120: }
  121: 
  122: close(CONFIG);
  123: 
  124: # -------------------------------------------------------- Routines for forking
  125: 
  126: %children               = ();       # keys are current child process IDs,
  127:                                     # values are hosts
  128: %childpid               = ();       # the other way around
  129: 
  130: %childatt               = ();       # number of attempts to start server
  131:                                     # for ID
  132: 
  133: $childmaxattempts=5;
  134: 
  135: # ---------------------------------------------------- Fork once and dissociate
  136: &status("Fork and dissociate");
  137: $fpid=fork;
  138: exit if $fpid;
  139: die "Couldn't fork: $!" unless defined ($fpid);
  140: 
  141: POSIX::setsid() or die "Can't start new session: $!";
  142: 
  143: $conserver='PARENT';
  144: 
  145: # ------------------------------------------------------- Write our PID on disk
  146: &status("Write PID");
  147: $execdir=$perlvar{'lonDaemons'};
  148: open (PIDSAVE,">$execdir/logs/lonc.pid");
  149: print PIDSAVE "$$\n";
  150: close(PIDSAVE);
  151: &logthis("<font color=red>CRITICAL: ---------- Starting ----------</font>");
  152: 
  153: # ----------------------------- Ignore signals generated during initial startup
  154: $SIG{HUP}=$SIG{USR1}='IGNORE';
  155: # ------------------------------------------------------- Now we are on our own
  156:     
  157: # Fork off our children, one for every server
  158: 
  159: &status("Forking ...");
  160: 
  161: foreach $thisserver (keys %hostip) {
  162:     #if (&online($hostname{$thisserver})) {
  163:        make_new_child($thisserver);
  164:     #}
  165: }
  166: 
  167: &logthis("Done starting initial servers");
  168: # ----------------------------------------------------- Install signal handlers
  169: 
  170: 
  171: $SIG{INT}  = $SIG{TERM} = \&HUNTSMAN;
  172: $SIG{HUP}  = \&HUPSMAN;
  173: $SIG{USR1} = \&USRMAN;
  174: 
  175: # And maintain the population.
  176: while (1) {
  177:     my $deadpid = wait;		# Wait for the next child to die.
  178:                                     # See who died and start new one
  179:     &status("Woke up");
  180:     my $skipping='';
  181: 
  182:     if(exists($children{$deadpid})) {
  183: 
  184: 	$thisserver = $children{$deadpid}; # Look name of dead guy's peer.
  185: 
  186: 	delete($children{$deadpid}); # Get rid of dead hash entry.
  187: 
  188: 	if($childatt{$thisserver} < $childmaxattempts) {
  189: 	    $childatt{$thisserver}++;
  190: 	    &logthis(
  191: 	       "<font color=yellow>INFO: Trying to reconnect for $thisserver "
  192:             ."($childatt{$thisserver} of $childmaxattempts attempts)</font>"); 
  193: 	    make_new_child($thisserver);
  194: 	
  195: 	}
  196: 	else {
  197: 	    $skipping .= $thisserver.' ';
  198: 	}
  199: 	if($skipping) {
  200: 	    &logthis("<font color=blue>WARNING: Skipped $skipping</font>");
  201:   
  202: 	}
  203:     }
  204: 
  205: }
  206: 
  207: 
  208: 
  209: sub make_new_child {
  210:    
  211:     $newserver=shift;
  212:     my $pid;
  213:     my $sigset;
  214:     &logthis("Attempting to start child for server $newserver");
  215:     # block signal for fork
  216:     $sigset = POSIX::SigSet->new(SIGINT);
  217:     sigprocmask(SIG_BLOCK, $sigset)
  218:         or die "Can't block SIGINT for fork: $!\n";
  219:     
  220:     die "fork: $!" unless defined ($pid = fork);
  221:     
  222:     if ($pid) {
  223:         # Parent records the child's birth and returns.
  224:         sigprocmask(SIG_UNBLOCK, $sigset)
  225:             or die "Can't unblock SIGINT for fork: $!\n";
  226:         $children{$pid} = $newserver;
  227:         $childpid{$newserver} = $pid;
  228:         return;
  229:     } else {
  230:         $conserver=$newserver;
  231:         # Child can *not* return from this subroutine.
  232:         $SIG{INT} = 'DEFAULT';      # make SIGINT kill us as it did before
  233:         $SIG{USR1}= \&logstatus;
  234:    
  235:         # unblock signals
  236:         sigprocmask(SIG_UNBLOCK, $sigset)
  237:             or die "Can't unblock SIGINT for fork: $!\n";
  238: 
  239: # ----------------------------- This is the modified main program of non-forker
  240: 
  241: $port = "$perlvar{'lonSockDir'}/$conserver";
  242: 
  243: unlink($port);
  244: 
  245: # -------------------------------------------------------------- Open other end
  246: 
  247: &openremote($conserver);
  248: 	&logthis("<font color=green> Connection to $conserver open </font>");
  249: # ----------------------------------------- We're online, send delayed messages
  250:     &status("Checking for delayed messages");
  251: 
  252:     my @allbuffered;
  253:     my $path="$perlvar{'lonSockDir'}/delayed";
  254:     opendir(DIRHANDLE,$path);
  255:     @allbuffered=grep /\.$conserver$/, readdir DIRHANDLE;
  256:     closedir(DIRHANDLE);
  257:     my $dfname;
  258:     foreach (@allbuffered) {
  259:         &status("Sending delayed: $_");
  260:         $dfname="$path/$_";
  261:         if($DEBUG) { &logthis('Sending '.$dfname); }
  262:         my $wcmd;
  263:         {
  264:          my $dfh=IO::File->new($dfname);
  265:          $cmd=<$dfh>;
  266:         }
  267:         chomp($cmd);
  268:         my $bcmd=$cmd;
  269:         if ($cmd =~ /^encrypt\:/) {
  270: 	    my $rcmd=$cmd;
  271:             $rcmd =~ s/^encrypt\://;
  272:             chomp($rcmd);
  273:             my $cmdlength=length($rcmd);
  274:             $rcmd.="         ";
  275:             my $encrequest='';
  276:             for (my $encidx=0;$encidx<=$cmdlength;$encidx+=8) {
  277:                 $encrequest.=
  278:                     unpack("H16",$cipher->encrypt(substr($rcmd,$encidx,8)));
  279:             }
  280:             $cmd="enc:$cmdlength:$encrequest\n";
  281:         }
  282: 	$answer = londtransaction($remotesock, $cmd, 60);
  283: 	chomp($answer);
  284: 
  285:         if (($answer ne '') && ($@!~/timeout/)) {
  286: 	    unlink("$dfname");
  287:             &logthis("Delayed $cmd: >$answer<");
  288:             &logperm("S:$conserver:$bcmd");
  289:         }        
  290:     }
  291: 	if($DEBUG) { &logthis("<font color=green> Delayed transactions sent"); }
  292: 
  293: # ------------------------------------------------------- Listen to UNIX socket
  294: &status("Opening socket");
  295: unless (
  296:   $server = IO::Socket::UNIX->new(Local  => $port,
  297:                                   Type   => SOCK_STREAM,
  298:                                   Listen => 10 )
  299:    ) { 
  300:        my $st=120+int(rand(240));
  301:        &logthis(
  302:          "<font color=blue>WARNING: ".
  303:          "Can't make server socket ($st secs):  .. exiting</font>");
  304:        sleep($st);
  305:        exit; 
  306:      };
  307:    
  308: # -----------------------------------------------------------------------------
  309: 
  310: &logthis("<font color=green>$conserver online</font>");
  311: 
  312: # -----------------------------------------------------------------------------
  313: # begin with empty buffers
  314: %inbuffer  = ();
  315: %outbuffer = ();
  316: %ready     = ();
  317: %servers   = ();	# To be compatible with make filevector.  indexed by
  318: 			# File descriptors, values are file descriptors.
  319: 			# note that the accept socket is omitted.
  320: 
  321: tie %ready, 'Tie::RefHash';
  322: 
  323: nonblock($server);
  324: $select = IO::Select->new($server);
  325: 
  326: # Main loop: check reads/accepts, check writes, check ready to process
  327: while (1) {
  328:     my $client;
  329:     my $rv;
  330:     my $data;
  331: 
  332:     my $infdset;		# bit vec of fd's to select on input.
  333:     my $inreadyset;		# Bit vec of fd's ready for input.
  334: 
  335:     my $outfdset;		# Bit vec of fd's to select on output.
  336:     my $outreadyset;		# bit vec of fds ready for output.
  337: 
  338: 
  339:     $infdset = MakeFileVector(\%servers);
  340:     $outfdset= MakeFileVector(\%outbuffer);
  341: 
  342:     # check for new information on the connections we have
  343:     # anything to read or accept?
  344: 
  345:     foreach $client ($select->can_read(00.10)) {
  346:         if ($client == $server) {
  347:             # accept a new connection
  348:             &status("Accept new connection: $conserver");
  349:             $client = $server->accept();
  350:             $select->add($client);
  351:             nonblock($client);
  352:         } else {
  353:             # read data
  354:             $data = '';
  355:             $rv   = $client->recv($data, POSIX::BUFSIZ, 0);
  356: 
  357:             unless (defined($rv) && length $data) {
  358:                 # This would be the end of file, so close the client
  359:                 delete $inbuffer{$client};
  360:                 delete $outbuffer{$client};
  361:                 delete $ready{$client};
  362: 
  363:                 &status("Idle");
  364:                 $select->remove($client);
  365:                 close $client;
  366:                 next;
  367:             }
  368: 
  369:             $inbuffer{$client} .= $data;
  370: 
  371: 
  372:             # test whether the data in the buffer or the data we
  373:             # just read means there is a complete request waiting
  374:             # to be fulfilled.  If there is, set $ready{$client}
  375:             # to the requests waiting to be fulfilled.
  376:             while ($inbuffer{$client} =~ s/(.*\n)//) {
  377:                 push( @{$ready{$client}}, $1 );
  378:             }
  379:         }
  380:     }
  381:     
  382:     # Any complete requests to process?
  383:     foreach $client (keys %ready) {
  384:         handle($client);
  385:     }
  386:  
  387:     # Buffers to flush?
  388:     foreach $client ($select->can_write(1)) {
  389:         # Skip this client if we have nothing to say
  390:         next unless exists $outbuffer{$client};
  391: 
  392:         $rv = $client->send($outbuffer{$client}, 0);
  393: 
  394:       unless ($outbuffer{$client} eq "con_lost\n") {
  395:         unless (defined $rv) {
  396:             # Whine, but move on.
  397:             &logthis("I was told I could write, but I can't.\n");
  398:             next;
  399:         }
  400:         $errno=$!;
  401:         if (($rv == length $outbuffer{$client}) ||
  402:             ($errno == POSIX::EWOULDBLOCK) || ($errno == 0)) {
  403:             substr($outbuffer{$client}, 0, $rv) = '';
  404:             delete $outbuffer{$client} unless length $outbuffer{$client};
  405:         } else {
  406:             # Couldn't write all the data, and it wasn't because
  407:             # it would have blocked.  Shutdown and move on.
  408: 
  409: 	    &logthis("Dropping data with ".$errno.": ".
  410:                      length($outbuffer{$client}).", $rv");
  411: 
  412:             delete $inbuffer{$client};
  413:             delete $outbuffer{$client};
  414:             delete $ready{$client};
  415: 
  416:             $select->remove($client);
  417:             close($client);
  418:             next;
  419:         }
  420:       } else {
  421: # -------------------------------------------------------- Wow, connection lost
  422:          &logthis(
  423:      "<font color=red>CRITICAL: Closing connection</font>");
  424: 	 &status("Connection lost");
  425:          $remotesock->shutdown(2);
  426:          &logthis("Attempting to open new connection");
  427:          &openremote($conserver);          
  428:       }
  429:     }
  430:    
  431: }
  432: }
  433: 
  434: # ------------------------------------------------------- End of make_new_child
  435: 
  436: 
  437: #
  438: #  Make a vector of file descriptors to wait for in a select.
  439: #  parameters:
  440: #     \%fdhash  -reference to a hash which has IO::Socket's as indices.  
  441: #                We only care about the indices, not the values.
  442: #  A select vector is created from all indices of the hash.
  443: 
  444: sub MakeFileVector
  445: {
  446:     my $fdhash = shift;
  447:     my $selvar = "";
  448: 
  449:     foreach $socket (keys %fdhash) {
  450: 	vec($selvar, ($fdhash->{$socket})->fileno, 1) = 1;
  451:     }
  452:     return $selvar;
  453: }
  454: 
  455: 
  456: #
  457: #  HandleOutput:
  458: #    Processes output on a buffered set of file descriptors which are
  459: #    ready to be read.
  460: #  Parameters:
  461: #    $selvector - Vector of writable file descriptors which are writable.
  462: #    \%sockets  - Vector of socket references indexed by socket.
  463: #    \%buffers  - Reference to a hash containing output buffers.
  464: #                 Hashes are indexed by sockets.  The file descriptors of some
  465: #                 of those sockets will be present in $selvector.
  466: #                 For each one of those, we will attempt to write the output
  467: #                 buffer to the socket.  Note that we will assume that
  468: #                 the sockets are being run in non blocking mode.
  469: #   \%inbufs    - Reference to hash containing input buffers.
  470: #   \%readys    - Reference to hash containing flags for items with complete
  471: #                 requests.
  472: #
  473: sub HandleOutput
  474: {
  475:     my $selvector = shift;
  476:     my $sockets   = shift;
  477:     my $buffers   = shift;
  478:     my $inbufs    = shift;
  479:     my $readys    = shift;
  480: 
  481:     foreach $sock (keys %buffers) {
  482: 	my $socket = $sockets->{$sock};
  483: 	if(vec($selvector, $$socket->fileno, 1)) { # $socket is writable.
  484: 	    my $rv = $$socket->send($buffers->{$sock}, 0);
  485: 	    $errno = $!;
  486: 	    unless ($buffers->{$sock} eq "con_lost\n") {
  487: 		unless (defined $rv) { # Write failed... could be EINTR
  488: 		    unless ($errno == POSIX::EINTR) {
  489: 			&logthis("Write failed on writable socket");
  490: 		    }		# EINTR is not an error .. just retry.
  491: 		    next;
  492: 		}
  493: 		if( ($rv == length $buffers->{$sock})    ||
  494: 		    ($errno == POSIX::EWOULDBLOCK)       ||
  495: 		    ($errno == POSIX::EAGAIN)            || # same as above.
  496: 		    ($errno == POSIX::EINTR)             || # signal during IO
  497: 		    ($errno == 0)) {
  498: 		    substr($buffers->{$sock}, 0, $rv)=""; # delete written part
  499: 		    delete $buffers->{$sock} unless length $buffers->{$sock};
  500: 		} else {
  501: 		    # For some reason the write failed with an error code
  502: 		    # we didn't look for.  Shutdown the socket.
  503: 		    &logthis("Unable to write data with ".$errno.": ".
  504: 			     "Dropping data: ".length($buffers->{$sock}).
  505: 			     ", $rv");
  506: 		    #
  507: 		    # kill off the buffers in the hash:
  508: 
  509: 		    delete $buffers->{$sock};
  510: 		    delete $inbufs->{$sock};
  511: 		    delete $readys->{$sock};
  512: 
  513: 		    close($$socket); # Close the client socket.
  514: 		    next;
  515: 		}
  516: 	    } else {		# Kludgy way to mark lond connection lost.
  517: 		&logthis(
  518: 		 "<font color=red>CRITICAL lond connection lost</font>");
  519: 		status("Connection lost");
  520: 		$remotesock->shutdown(2);
  521: 		&logthis("Attempting to open a new connection");
  522: 		&openremot($conserver);
  523: 	    }
  524: 		   
  525: 	}
  526:     }
  527: 
  528: }
  529: #
  530: #   HandleInput - Deals with input on client sockets.
  531: #                 Each socket has an associated input buffer.
  532: #                 For each readable socket, the currently available
  533: #                 data is appended to this buffer.
  534: #                 If necessary, the buffer is created.
  535: #                 On various failures, we may shutdown the client.
  536: #  Parameters:
  537: #     $selvec   - Vector of readable sockets.
  538: #     \%sockets - Refers to the  Hash of sockets indexed by sockets.  
  539: #                 Each of these may or may not have it's fd bit set 
  540: #                 in the $selvec.
  541: #     \%ibufs   - Refers to the hash of input buffers indexed by socket.
  542: #     \%obufs   - Hash of output buffers indexed by socket. 
  543: #     \%ready   - Hash of ready flags indicating the existence of a completed
  544: #                 Request.
  545: sub HandleInput 
  546: {
  547: 
  548:     # Marshall the parameters.   Note that the hashes are actually
  549:     # references not values.
  550: 
  551:     my $selvec  = shift;
  552:     my $sockets = shift;
  553:     my $ibufs   = shift;
  554:     my $obufs   = shift;
  555:     my $ready   = shift;
  556: 
  557:     foreach $sock (keys %sockets) {
  558: 	my $socket = $sockets->{$sock};
  559: 	if(vec($selvec, $$socket->fileno, 1)) { # Socket which is readable.
  560: 
  561: 	    #  Attempt to read the data and do error management.
  562: 	    my $data = '';
  563: 	    my $rv = $$socket->recv($data, POSIX::BUFSIZ, 0);
  564: 	    unless (defined($rv) && length $data) {
  565: 
  566: 		# Read an end of file.. this is a disconnect from the peer.
  567: 
  568: 		delete $sockets->{$sock};
  569: 		delete $ibufs->{$sock};
  570: 		delete $obufs->{$sock};
  571: 		delete $ready->{$sock};
  572: 
  573: 		status("Idle");
  574: 		close $$socket;
  575: 		next;
  576: 	    }
  577: 	    #  Append the read data to the input buffer. If the buffer
  578: 	    # now contains a \n the request is complete and we can 
  579: 	    # mark this in the $ready hash (one request for each \n.)
  580: 
  581: 	    $ibufs->{$sock} .= $data;
  582: 	    while($ibufs->{$sock} =~ s/(.*\n)//) {
  583: 		push(@{$ready->{$sock}}, $1);
  584: 	    }
  585: 	    
  586: 	}
  587:     }
  588:     #  Now handle any requests which are ready:
  589: 
  590:     foreach $client (keys %ready) {
  591: 	handle($client);
  592:     }
  593: }
  594: 
  595: # handle($socket) deals with all pending requests for $client
  596: #
  597: sub handle {
  598:     # requests are in $ready{$client}
  599:     # send output to $outbuffer{$client}
  600:     my $client = shift;
  601:     my $request;
  602:     foreach $request (@{$ready{$client}}) {
  603: # ============================================================= Process request
  604:         # $request is the text of the request
  605:         # put text of reply into $outbuffer{$client}
  606: # ------------------------------------------------------------ Is this the end?
  607: 	chomp($request);
  608: 	if($DEBUG) {
  609:      &logthis("<font color=green> Request $request processing starts</font>");
  610:         }
  611:         if ($request eq "close_connection_exit\n") {
  612: 	    &status("Request close connection");
  613:            &logthis(
  614:      "<font color=red>CRITICAL: Request Close Connection ... exiting</font>");
  615:            $remotesock->shutdown(2);
  616:            $server->close();
  617:            exit;
  618:         }
  619: # -----------------------------------------------------------------------------
  620:         if ($request =~ /^encrypt\:/) {
  621: 	    my $cmd=$request;
  622:             $cmd =~ s/^encrypt\://;
  623:             chomp($cmd);
  624:             my $cmdlength=length($cmd);
  625:             $cmd.="         ";
  626:             my $encrequest='';
  627:             for (my $encidx=0;$encidx<=$cmdlength;$encidx+=8) {
  628:                 $encrequest.=
  629:                     unpack("H16",$cipher->encrypt(substr($cmd,$encidx,8)));
  630:             }
  631:             $request="enc:$cmdlength:$encrequest";
  632:         }
  633: # --------------------------------------------------------------- Main exchange
  634: 	$answer = londtransaction($remotesock, $request, 300);
  635: 
  636: 	if($DEBUG) { 
  637: 	    &logthis("<font color=green> Request data exchange complete");
  638: 	}
  639: 	if ($@=~/timeout/) { 
  640: 	    $answer='';
  641: 	    &logthis(
  642: 		     "<font color=red>CRITICAL: Timeout: $request</font>");
  643: 	}  
  644: 
  645: 
  646:         if ($answer) {
  647: 	   if ($answer =~ /^enc/) {
  648:                my ($cmd,$cmdlength,$encinput)=split(/:/,$answer);
  649:                chomp($encinput);
  650: 	       $answer='';
  651:                for (my $encidx=0;$encidx<length($encinput);$encidx+=16) {
  652:                   $answer.=$cipher->decrypt(
  653:                    pack("H16",substr($encinput,$encidx,16))
  654:                   );
  655: 	       }
  656: 	      $answer=substr($answer,0,$cmdlength);
  657: 	      $answer.="\n";
  658: 	   }
  659: 	   if($DEBUG) {
  660: 	       &logthis("sending $answer to client\n");
  661: 	   }
  662:            $outbuffer{$client} .= $answer;
  663:         } else {
  664:            $outbuffer{$client} .= "con_lost\n";
  665:         }
  666: 
  667:      &status("Completed: $request");
  668: 	if($DEBUG) {
  669: 	    &logthis("<font color=green> Request processing complete</font>");
  670: 	}
  671: # ===================================================== Done processing request
  672:     }
  673:     delete $ready{$client};
  674: # -------------------------------------------------------------- End non-forker
  675:     if($DEBUG) {
  676: 	&logthis("<font color=green> requests for child handled</font>");
  677:     }
  678: }
  679: # ---------------------------------------------------------- End make_new_child
  680: }
  681: 
  682: # nonblock($socket) puts socket into nonblocking mode
  683: sub nonblock {
  684:     my $socket = shift;
  685:     my $flags;
  686: 
  687:     
  688:     $flags = fcntl($socket, F_GETFL, 0)
  689:             or die "Can't get flags for socket: $!\n";
  690:     fcntl($socket, F_SETFL, $flags | O_NONBLOCK)
  691:             or die "Can't make socket nonblocking: $!\n";
  692: }
  693: 
  694: 
  695: sub openremote {
  696: # ---------------------------------------------------- Client to network server
  697: 
  698:     my $conserver=shift;
  699: 
  700: &status("Opening TCP");
  701:     my $st=120+int(rand(240)); # Sleep before opening:
  702: 
  703: unless (
  704:   $remotesock = IO::Socket::INET->new(PeerAddr => $hostip{$conserver},
  705:                                       PeerPort => $perlvar{'londPort'},
  706:                                       Proto    => "tcp",
  707:                                       Type     => SOCK_STREAM)
  708:    ) { 
  709: 
  710:        &logthis(
  711: "<font color=blue>WARNING: Couldn't connect to $conserver ($st secs): </font>");
  712:        sleep($st);
  713:        exit; 
  714:      };
  715: # ----------------------------------------------------------------- Init dialog
  716: 
  717: &logthis("<font color=green>INFO Connected to $conserver, initing </font>");
  718: &status("Init dialogue: $conserver");
  719: 
  720:     $answer = londtransaction($remotesock, "init", 60);
  721:     chomp($answer);
  722:     $answer = londtransaction($remotesock, $answer, 60);
  723:     chomp($answer);
  724:  
  725:      if ($@=~/timeout/) {
  726: 	 &logthis("Timed out during init.. exiting");
  727:          exit;
  728:      }
  729: 
  730: if ($answer ne 'ok') {
  731:        &logthis("Init reply: >$answer<");
  732:        my $st=120+int(rand(240));
  733:        &logthis(
  734: "<font color=blue>WARNING: Init failed ($st secs)</font>");
  735:        sleep($st);
  736:        exit; 
  737: }
  738: 
  739: sleep 5;
  740: &status("Ponging");
  741: print $remotesock "pong\n";
  742: $answer=<$remotesock>;
  743: chomp($answer);
  744: if ($answer!~/^$conserver/) {
  745:    &logthis("Pong reply: >$answer<");
  746: }
  747: # ----------------------------------------------------------- Initialize cipher
  748: 
  749: &status("Initialize cipher");
  750: print $remotesock "ekey\n";
  751: my $buildkey=<$remotesock>;
  752: my $key=$conserver.$perlvar{'lonHostID'};
  753: $key=~tr/a-z/A-Z/;
  754: $key=~tr/G-P/0-9/;
  755: $key=~tr/Q-Z/0-9/;
  756: $key=$key.$buildkey.$key.$buildkey.$key.$buildkey;
  757: $key=substr($key,0,32);
  758: my $cipherkey=pack("H32",$key);
  759: if ($cipher=new IDEA $cipherkey) {
  760:    &logthis("Secure connection initialized");
  761: } else {
  762:    my $st=120+int(rand(240));
  763:    &logthis(
  764:      "<font color=blue>WARNING: ".
  765:      "Could not establish secure connection ($st secs)!</font>");
  766:    sleep($st);
  767:    exit;
  768: }
  769:     &logthis("<font color=green> Remote open success </font>");
  770: }
  771: 
  772: 
  773: 
  774: # grabs exception and records it to log before exiting
  775: sub catchexception {
  776:     my ($signal)=@_;
  777:     $SIG{QUIT}='DEFAULT';
  778:     $SIG{__DIE__}='DEFAULT';
  779:     chomp($signal);
  780:     &logthis("<font color=red>CRITICAL: "
  781:      ."ABNORMAL EXIT. Child $$ for server [$wasserver] died through "
  782:      ."\"$signal\" with parameter </font>");
  783:     die("Signal abend");
  784: }
  785: 
  786: # -------------------------------------- Routines to see if other box available
  787: 
  788: #sub online {
  789: #    my $host=shift;
  790: #    &status("Pinging ".$host);
  791: #    my $p=Net::Ping->new("tcp",20);
  792: #    my $online=$p->ping("$host");
  793: #    $p->close();
  794: #    undef ($p);
  795: #    return $online;
  796: #}
  797: 
  798: sub connected {
  799:     my ($local,$remote)=@_;
  800:     &status("Checking connection $local to $remote");
  801:     $local=~s/\W//g;
  802:     $remote=~s/\W//g;
  803: 
  804:     unless ($hostname{$local}) { return 'local_unknown'; }
  805:     unless ($hostname{$remote}) { return 'remote_unknown'; }
  806: 
  807:     #unless (&online($hostname{$local})) { return 'local_offline'; }
  808: 
  809:     my $ua=new LWP::UserAgent;
  810:     
  811:     my $request=new HTTP::Request('GET',
  812:       "http://".$hostname{$local}.'/cgi-bin/ping.pl?'.$remote);
  813: 
  814:     my $response=$ua->request($request);
  815: 
  816:     unless ($response->is_success) { return 'local_error'; }
  817: 
  818:     my $reply=$response->content;
  819:     $reply=(split("\n",$reply))[0];
  820:     $reply=~s/\W//g;
  821:     if ($reply ne $remote) { return $reply; }
  822:     return 'ok';
  823: }
  824: 
  825: 
  826: 
  827: sub hangup {
  828:     foreach (keys %children) {
  829:         $wasserver=$children{$_};
  830:         &status("Closing $wasserver");
  831:         &logthis('Closing '.$wasserver.': '.&subreply('exit',$wasserver));
  832:         &status("Kill PID $_ for $wasserver");
  833: 	kill ('INT',$_);
  834:     }
  835: }
  836: 
  837: sub HUNTSMAN {                      # signal handler for SIGINT
  838:     local($SIG{CHLD}) = 'IGNORE';   # we're going to kill our children
  839:     &hangup();
  840:     my $execdir=$perlvar{'lonDaemons'};
  841:     unlink("$execdir/logs/lonc.pid");
  842:     &logthis("<font color=red>CRITICAL: Shutting down</font>");
  843:     exit;                           # clean up with dignity
  844: }
  845: 
  846: sub HUPSMAN {                      # signal handler for SIGHUP
  847:     local($SIG{CHLD}) = 'IGNORE';  # we're going to kill our children
  848:     &hangup();
  849:     &logthis("<font color=red>CRITICAL: Restarting</font>");
  850:     unlink("$execdir/logs/lonc.pid");
  851:     my $execdir=$perlvar{'lonDaemons'};
  852:     exec("$execdir/lonc");         # here we go again
  853: }
  854: 
  855: sub checkchildren {
  856:     &initnewstatus();
  857:     &logstatus();
  858:     &logthis('Going to check on the children');
  859:     foreach (sort keys %children) {
  860: 	sleep 1;
  861:         unless (kill 'USR1' => $_) {
  862: 	    &logthis ('<font color=red>CRITICAL: Child '.$_.' is dead</font>');
  863:             &logstatus($$.' is dead');
  864:         } 
  865:     }
  866: }
  867: 
  868: sub USRMAN {
  869:     &logthis("USR1: Trying to establish connections again");
  870:     %childatt=();
  871:     &checkchildren();
  872: }
  873: 
  874: # -------------------------------------------------- Non-critical communication
  875: sub subreply { 
  876:  my ($cmd,$server)=@_;
  877:  my $answer='';
  878:  if ($server ne $perlvar{'lonHostID'}) { 
  879:     my $peerfile="$perlvar{'lonSockDir'}/$server";
  880:     my $sclient=IO::Socket::UNIX->new(Peer    =>"$peerfile",
  881:                                       Type    => SOCK_STREAM,
  882:                                       Timeout => 10)
  883:        or return "con_lost";
  884: 
  885: 
  886:     $answer = londtransaction($sclient, $cmd, 10);
  887: 
  888:     if ((!$answer) || ($@=~/timeout/)) { $answer="con_lost"; }
  889:     $SIG{ALRM}='DEFAULT';
  890:     $SIG{__DIE__}=\&catchexception;
  891:  } else { $answer='self_reply'; }
  892:  return $answer;
  893: }
  894: 
  895: # --------------------------------------------------------------------- Logging
  896: 
  897: sub logthis {
  898:     my $message=shift;
  899:     my $execdir=$perlvar{'lonDaemons'};
  900:     my $fh=IO::File->new(">>$execdir/logs/lonc.log");
  901:     my $now=time;
  902:     my $local=localtime($now);
  903:     $lastlog=$local.': '.$message;
  904:     print $fh "$local ($$) [$conserver] [$status]: $message\n";
  905: }
  906: 
  907: #--------------------------------------  londtransaction:
  908: #  
  909: #  Performs a transaction with lond with timeout support.
  910: #    result = londtransaction(socket,request,timeout)
  911: #
  912: sub londtransaction {
  913:     my ($socket, $request, $tmo) = @_;
  914: 
  915:     if($DEBUG) {
  916: 	&logthis("londtransaction request: $request");
  917:     }
  918: 
  919:     # Set the signal handlers: ALRM for timeout and disble the others.
  920: 
  921:     $SIG{ALRM} = sub { die "timeout" };
  922:     $SIG{__DIE__} = 'DEFAULT';
  923:     
  924:     # Disable all but alarm so that only that can interupt the
  925:     # send /receive.
  926:     #
  927:     my $sigset = POSIX::SigSet->new(QUIT, USR1, HUP, INT, TERM);
  928:     my $priorsigs = POSIX::SigSet->new;
  929:     unless (defined sigprocmask(SIG_BLOCK, $sigset, $priorsigs)) {
  930: 	&logthis("<font color=red> CRITICAL -- londtransaction ".
  931: 		"failed to block signals </font>");
  932: 	die "could not block signals in londtransaction";
  933:     }
  934:     $answer = '';
  935:     #
  936:     #  Send request to lond.
  937:     #
  938:     eval { 
  939: 	alarm($tmo);
  940: 	print $socket "$request\n";
  941: 	alarm(0);
  942:     };
  943:     #  If request didn't timeout, try for the response.
  944:     #
  945: 
  946:     if ($@!~/timeout/) {
  947: 	eval {
  948: 	    alarm($tmo);
  949: 	    $answer = <$socket>;
  950: 	    if($DEBUG) {
  951: 		&logthis("Received $answer in londtransaction");
  952: 	    }
  953: 	    alarm(0);
  954: 	};
  955:     } else {
  956: 	if($DEBUG) {
  957: 	    &logthis("Timeout on send in londtransaction");
  958: 	}
  959:     }
  960:     if( ($@ =~ /timeout/)  && ($DEBUG)) {
  961: 	&logthis("Timeout on receive in londtransaction");
  962:     }
  963:     #
  964:     # Restore the initial sigmask set.
  965:     #
  966:     unless (defined sigprocmask(SIG_UNBLOCK, $priorsigs)) {
  967: 	&logthis("<font color=red> CRITICAL -- londtransaction ".
  968: 		"failed to re-enable signal processing. </font>");
  969: 	die "londtransaction failed to re-enable signals";
  970:     }
  971:     #
  972:     # go back to the prior handler set.
  973:     #
  974:     $SIG{ALRM} = 'DEFAULT';
  975:     $SIG{__DIE__} = \&cathcexception;
  976: 
  977:     #    chomp $answer;
  978:     if ($DEBUG) {
  979: 	&logthis("Returning $answer in londtransaction");
  980:     }
  981:     return $answer;
  982: 
  983: }
  984: 
  985: sub logperm {
  986:     my $message=shift;
  987:     my $execdir=$perlvar{'lonDaemons'};
  988:     my $now=time;
  989:     my $local=localtime($now);
  990:     my $fh=IO::File->new(">>$execdir/logs/lonnet.perm.log");
  991:     print $fh "$now:$message:$local\n";
  992: }
  993: # ------------------------------------------------------------------ Log status
  994: 
  995: sub logstatus {
  996:     my $docdir=$perlvar{'lonDocRoot'};
  997:     my $fh=IO::File->new(">>$docdir/lon-status/loncstatus.txt");
  998:     print $fh $$."\t".$conserver."\t".$status."\t".$lastlog."\n";
  999: }
 1000: 
 1001: sub initnewstatus {
 1002:     my $docdir=$perlvar{'lonDocRoot'};
 1003:     my $fh=IO::File->new(">$docdir/lon-status/loncstatus.txt");
 1004:     my $now=time;
 1005:     my $local=localtime($now);
 1006:     print $fh "LONC status $local - parent $$\n\n";
 1007: }
 1008: 
 1009: # -------------------------------------------------------------- Status setting
 1010: 
 1011: sub status {
 1012:     my $what=shift;
 1013:     my $now=time;
 1014:     my $local=localtime($now);
 1015:     $status=$local.': '.$what;
 1016: }
 1017: 
 1018: 
 1019: 
 1020: # ----------------------------------- POD (plain old documentation, CPAN style)
 1021: 
 1022: =head1 NAME
 1023: 
 1024: lonc - LON TCP-MySQL-Server Daemon for handling database requests.
 1025: 
 1026: =head1 SYNOPSIS
 1027: 
 1028: Usage: B<lonc>
 1029: 
 1030: Should only be run as user=www.  This is a command-line script which
 1031: is invoked by B<loncron>.  There is no expectation that a typical user
 1032: will manually start B<lonc> from the command-line.  (In other words,
 1033: DO NOT START B<lonc> YOURSELF.)
 1034: 
 1035: =head1 DESCRIPTION
 1036: 
 1037: Provides persistent TCP connections to the other servers in the network
 1038: through multiplexed domain sockets
 1039: 
 1040: B<lonc> forks off children processes that correspond to the other servers
 1041: in the network.  Management of these processes can be done at the
 1042: parent process level or the child process level.
 1043: 
 1044:   After forking off the children, B<lonc> the B<parent> 
 1045: executes a main loop which simply waits for processes to exit.
 1046: As a process exits, a new process managing a link to the same
 1047: peer as the exiting process is created.  
 1048: 
 1049: B<logs/lonc.log> is the location of log messages.
 1050: 
 1051: The process management is now explained in terms of linux shell commands,
 1052: subroutines internal to this code, and signal assignments:
 1053: 
 1054: =over 4
 1055: 
 1056: =item *
 1057: 
 1058: PID is stored in B<logs/lonc.pid>
 1059: 
 1060: This is the process id number of the parent B<lonc> process.
 1061: 
 1062: =item *
 1063: 
 1064: SIGTERM and SIGINT
 1065: 
 1066: Parent signal assignment:
 1067:  $SIG{INT}  = $SIG{TERM} = \&HUNTSMAN;
 1068: 
 1069: Child signal assignment:
 1070:  $SIG{INT}  = 'DEFAULT'; (and SIGTERM is DEFAULT also)
 1071: (The child dies and a SIGALRM is sent to parent, awaking parent from slumber
 1072:  to restart a new child.)
 1073: 
 1074: Command-line invocations:
 1075:  B<kill> B<-s> SIGTERM I<PID>
 1076:  B<kill> B<-s> SIGINT I<PID>
 1077: 
 1078: Subroutine B<HUNTSMAN>:
 1079:  This is only invoked for the B<lonc> parent I<PID>.
 1080: This kills all the children, and then the parent.
 1081: The B<lonc.pid> file is cleared.
 1082: 
 1083: =item *
 1084: 
 1085: SIGHUP
 1086: 
 1087: Current bug:
 1088:  This signal can only be processed the first time
 1089: on the parent process.  Subsequent SIGHUP signals
 1090: have no effect.
 1091: 
 1092: Parent signal assignment:
 1093:  $SIG{HUP}  = \&HUPSMAN;
 1094: 
 1095: Child signal assignment:
 1096:  none (nothing happens)
 1097: 
 1098: Command-line invocations:
 1099:  B<kill> B<-s> SIGHUP I<PID>
 1100: 
 1101: Subroutine B<HUPSMAN>:
 1102:  This is only invoked for the B<lonc> parent I<PID>,
 1103: This kills all the children, and then the parent.
 1104: The B<lonc.pid> file is cleared.
 1105: 
 1106: =item *
 1107: 
 1108: SIGUSR1
 1109: 
 1110: Parent signal assignment:
 1111:  $SIG{USR1} = \&USRMAN;
 1112: 
 1113: Child signal assignment:
 1114:  $SIG{USR1}= \&logstatus;
 1115: 
 1116: Command-line invocations:
 1117:  B<kill> B<-s> SIGUSR1 I<PID>
 1118: 
 1119: Subroutine B<USRMAN>:
 1120:  When invoked for the B<lonc> parent I<PID>,
 1121: SIGUSR1 is sent to all the children, and the status of
 1122: each connection is logged.
 1123: 
 1124: 
 1125: =back
 1126: 
 1127: =head1 PREREQUISITES
 1128: 
 1129: POSIX
 1130: IO::Socket
 1131: IO::Select
 1132: IO::File
 1133: Socket
 1134: Fcntl
 1135: Tie::RefHash
 1136: Crypt::IDEA
 1137: 
 1138: =head1 COREQUISITES
 1139: 
 1140: =head1 OSNAMES
 1141: 
 1142: linux
 1143: 
 1144: =head1 SCRIPT CATEGORIES
 1145: 
 1146: Server/Process
 1147: 
 1148: =cut

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