fs_sesmon/fs_session_server

   1 #!/usr/bin/perl -Tw
   2 #
   3 # fs_session_server
   4 #
   5
   6 use strict;
   7 use vars qw( $opt $Debug );
   8 use IO::Handle;
   9 use Net::SSH qw(sshopen3)
  10 use FS::UID qw(adminsuidsetup);
  11 use FS::Record qw( qsearch qsearchs );
  12 #use FS::cust_main_county;
  13 #use FS::cust_main;
  14 use FS::session;
  15 use FS::port;
  16 use FS::svc_acct;
  17
  18 #require "configfile";
  19 $Debug = 1;
  20
  21 my $user = shift or die &usage;
  22 &adminsuidsetup( $user );
  23
  24 my $machine = shift or die &usage;
  25
  26 my $fs_sessiond = "/usr/local/sbin/fs_sessiond";
  27
  28 my $me = "[fs_session_server]";
  29
  30 while (1) {
  31   my($reader, $writer) = (new IO::Handle, new IO::Handle);
  32   $writer->autoflush(1);
  33   warn "$me Connecting to $machine\n" if $Debug;
  34   sshopen2($machine,$reader,$writer,$fs_signupd);
  35
  36   warn "$me Entering main loop\n" if $Debug;
  37   while (1) {
  38     warn "$me Reading (waiting for) data\n" if $Debug;
  39     my $command = scalar(<$reader));
  40     #DoS protection here too, to protect against a compromised client?  *sigh*
  41     while ( ( my $key = scalar(<$reader>) ) != "END\n" ) {
  42       chomp $key;
  43       chomp( $hash{$key} = scalar(<$reader>) );
  44     }
  45
  46     if ( $command eq 'login' ) {
  47       $error = &login(\%hash);
  48       print $writer "$error\n";
  49     } elsif ( $command eq 'logoff' ) {
  50       $error = &logoff(\%hash);
  51       print $writer "$error\n";
  52     } elsif ( $command eq 'portnum' ) {
  53       if ( exists $hash{'ip'} ) {
  54         $hash{'ip'} =~ /^([\d\.]+)$/ or $1='nomatch';
  55         $port = qsearchs('port', { 'ip' => $1 } );
  56       } else {
  57         $hash{'nasnum'} =~ /^(\d+)$/ and my $nasnum = $1;
  58         $hash{'nasport'} =~ /^(\d+)$/ and my $nasport = $1;
  59         $port = qsearchs('port', { 'nasnum'=>$nasnum, 'nasport'=>$nasport } );
  60       }
  61       print $writer ( $port ? $port->portnum : '' ), "\n";
  62     } else {
  63       warn "$me WARNING: unrecognized command";
  64     }
  65   }
  66   #won't ever reach without code above to throw out of loop, but...
  67   close $writer;
  68   close $reader;
  69   warn "connection to $machine lost!\n"
  70   sleep 5;
  71   warn "reconnecting...\n";
  72 }
  73
  74 sub login {
  75   my $href = shift;
  76   $href->{'username'} =~ /^([a-z0-9_\-\.]+)$/ or return "Illegal username";
  77   my $username = $1;
  78   $svc_acct = qsearchs('svc_acct', { 'username' => $username } )
  79     or return "Unknown user";
  80   return "Incorrect password"
  81     if defined($href->{'password'})
  82        && $href->{'password'} ne $svc_acct->_password;
  83   my $session = new FS::session {
  84     'portnum' => $href->{'portnum'},
  85     'svcnum'  => $svc_acct->svcnum,
  86     'login'   => $href->{'login'},
  87   };
  88   $session->insert;
  89 }
  90
  91 sub logout {
  92   my $href = shift;
  93   $href->{'username'} =~ /^([a-z0-9_\-\.]+)$/ or return "Illegal username";
  94   my $username = $1;
  95   $svc_acct = qsearchs('svc_acct', { 'username' => $username } )
  96     or return "Unknown user";
  97   return "Incorrect password"
  98     if defined($href->{'password'})
  99        && $href->{'password'} ne $svc_acct->_password;
 100   my $session = qsearchs FS::session {
 101     'portnum' => $href->{'portnum'},
 102     'svcnum'  => $svc_acct->svcnum,
 103     'logoff'  => '',
 104   };
 105   return "No currently open sessios found for that user/port!" unless $session;
 106   my $nsession = new FS::session ( { $old->hash } );
 107   $nsession->replace($session);
 108 }
 109
 110 sub usage {
 111   die "Usage:\n\n  fs_session_server user machine\n";
 112 }
 113