home *** CD-ROM | disk | FTP | other *** search
/ ftp.madoka.org / 2014.12.ftp.madoka.org.tar / ftp.madoka.org / pub / plum / plum2_33_1.lzh / support / multiproxy < prev    next >
Text File  |  1999-03-24  |  5KB  |  207 lines

  1. #!/bin/perl -w
  2. # $Id: multiproxy,v 2.9 1999/01/20 15:04:08 hasegawa Exp $
  3. # copyright (c)1998-1999 Yoshinori Hasegawa <hasegawa@madoka.org>
  4.  
  5. if ($] < 5) {
  6.   foreach $inc (@INC) {
  7.     if (-r "$inc/sys/socket.ph") {
  8.       eval 'require "sys/socket.ph"';
  9.       $SOCKET = "$inc/sys/socket.ph" unless $@;
  10.       last;
  11.     }
  12.     if (-r "$inc/socket.ph") {
  13.       eval 'require "socket.ph"';
  14.       $SOCKET = "$inc/socket.ph" unless $@;
  15.       last;
  16.     }
  17.   }
  18. } else {
  19.   eval 'use Socket';
  20.   $SOCKET = 'Socket.pm' unless $@;
  21. }
  22.  
  23. $NIL = $;;
  24.  
  25. $READSIZE = 1024;
  26. $SOCKADDR = 'S n N x8';
  27.  
  28. $PROTO = (getprotobyname('tcp'))[2];
  29.  
  30. $AF_INET = eval '&AF_INET' || 2;
  31. $PF_INET = eval '&PF_INET' || 2;
  32. $SOCK_STREAM = eval '&SOCK_STREAM' || 1;
  33. $SOMAXCONN = eval '&SOMAXCONN' || 16;
  34. $INADDR_ANY = eval '&INADDR_ANY' || "\0\0\0\0";
  35. $SOL_SOCKET = eval '&SOL_SOCKET';
  36. $SO_REUSEADDR = eval '&SO_REUSEADDR';
  37.  
  38. $'rin = '';
  39.  
  40. $handle = 0;
  41.  
  42. $SIG{'PIPE'} = 'IGNORE' if &'exist(&'list(keys(%SIG)), 'PIPE');
  43.  
  44. &main();
  45.  
  46. sub main {
  47.   local($line, $llist, $slist, $rout, $host, $port, $lport, $nf, $tmp, $socket, $sno, $cno);
  48.   while (defined($line = <>)) {
  49.     $line =~ tr/\r\n//d;
  50.     ($host, $port, $lport) = split(/\s+/, $line);
  51.     $remotehost{$lport} = $host;
  52.     $remoteport{$lport} = $port;
  53.     $portlist = &'add($portlist, $lport);
  54.   }
  55.   foreach $port (&'array($portlist)) {
  56.     $lno = &'listen($port, 0) || die "cannot listen port\n";
  57.     $llist = &'add($llist, $lno);
  58.   }
  59.   for (;;) {
  60.     $nf = select($rout = $'rin, undef, undef, undef);
  61.     die "error in select\n" if $nf < 0;
  62.     foreach $no (&'array($slist)) {
  63.       next unless vec($rout, $no, 1);
  64.       $tmp = '';
  65.       if (sysread($'socket[$no], $tmp, $READSIZE)) {
  66.         $socket = $'socket[$peer[$no]];
  67.         print $socket $tmp;
  68.       } else {
  69.         &'close($no);
  70.         &'close($peer[$no]);
  71.         $slist = &'remove($slist, $no, $peer[$no]);
  72.       }
  73.     }
  74.     foreach $lno (&'array($llist)) {
  75.       next unless vec($rout, $lno, 1);
  76.       $port = (&'sockname($lno))[0];
  77.       if ($cno = &'accept($lno)) {
  78.         if ($sno = &'connect($remotehost{$port}, $remoteport{$port})) {
  79.           $peer[$cno] = $sno;
  80.           $peer[$sno] = $cno;
  81.           $slist = &'add($slist, $cno, $sno);
  82.         } else {
  83.           &'close($cno);
  84.         }
  85.       }
  86.     }
  87.   }
  88. }
  89.  
  90. sub 'connect {
  91.   local($host, $port) = @_;
  92.   local($serverno, $socket, $ip, @addr, $name);
  93.   if ($host =~ /^\d+$/) {
  94.     $ip = $host;
  95.   } elsif ($host =~ /^[\d\.]+$/) {
  96.     @addr = split(/\./, $host);
  97.     $ip = unpack('N', pack('C4', @addr, 0, 0, 0));
  98.   } else {
  99.     $ip = unpack('N', (gethostbyname($host))[4] || "\0\0\0\0");
  100.   }
  101.   return 0 unless $ip;
  102.   $socket = '\'S' . ++$handle;
  103.   socket($socket, $PF_INET, $SOCK_STREAM, $PROTO) || return 0;
  104.   $name = pack($SOCKADDR, $AF_INET, $port, $ip);
  105.   connect($socket, $name) || return 0;
  106.   binmode($socket);
  107.   $serverno = fileno($socket);
  108.   vec($'rin, $serverno, 1) = 1;
  109.   $'socket[$serverno] = $socket;
  110.   select((select($socket), $| = 1)[0]);
  111.   $'access[$serverno] = time();
  112.   return $serverno;
  113. }
  114.  
  115. sub 'listen {
  116.   local($port, $count) = @_;
  117.   local($listenno, $socket, $name);
  118.   $socket = '\'L' . ++$handle;
  119.   socket($socket, $PF_INET, $SOCK_STREAM, $PROTO) || return 0;
  120.   if (defined($SOL_SOCKET) && defined($SO_REUSEADDR)) {
  121.     setsockopt($socket, $SOL_SOCKET, $SO_REUSEADDR, pack('l', 1));
  122.   }
  123.   $name = pack($SOCKADDR, $AF_INET, $port, unpack('N', $INADDR_ANY));
  124.   bind($socket, $name) || return 0;
  125.   listen($socket, $count || $SOMAXCONN) || return 0;
  126.   $listenno = fileno($socket);
  127.   vec($'rin, $listenno, 1) = 1;
  128.   $'socket[$listenno] = $socket;
  129.   select((select($socket), $| = 1)[0]);
  130.   $'access[$listenno] = time();
  131.   return $listenno;
  132. }
  133.  
  134. sub 'accept {
  135.   local($listenno) = @_;
  136.   local($clientno, $socket);
  137.   $socket = '\'C' . ++$handle;
  138.   accept($socket, $'socket[$listenno]) || return 0;
  139.   binmode($socket);
  140.   $clientno = fileno($socket);
  141.   vec($'rin, $clientno, 1) = 1;
  142.   $'socket[$clientno] = $socket;
  143.   select((select($socket), $| = 1)[0]);
  144.   $'access[$clientno] = time();
  145.   return $clientno;
  146. }
  147.  
  148. sub 'close {
  149.   local($no) = @_;
  150.   close($'socket[$no]);
  151.   vec($'rin, $no, 1) = 0;
  152. }
  153.  
  154. sub 'sockname {
  155.   local($no) = @_;
  156.   local($port, $ip, $host);
  157.   ($port, $ip) = (unpack($SOCKADDR, getsockname($'socket[$no])))[1, 2];
  158.   $host = (gethostbyaddr(pack('N', $ip), $AF_INET))[0];
  159.   return ($port, $ip, $host);
  160. }
  161.  
  162. sub 'add {
  163.   local($list, @items) = @_;
  164.   $list = '' unless $list;
  165.   foreach $item (@items) {
  166.     next if &'exist($list, $item);
  167.     $list .= $NIL . $item;
  168.   }
  169.   return $list;
  170. }
  171.  
  172. sub 'remove {
  173.   local($list, @items) = @_;
  174.   local($idx);
  175.   $list = '' unless $list;
  176.   $list .= $NIL;
  177.   foreach $item (@items) {
  178.     $idx = index("\L$list\E", $NIL . "\L$item\E" . $NIL);
  179.     next if $idx == -1;
  180.     substr($list, $idx, length($NIL . $item . $NIL)) = $NIL;
  181.   }
  182.   return substr($list, 0, length($list) - 1);
  183. }
  184.  
  185. sub 'exist {
  186.   local($list, @items) = @_;
  187.   return 0 unless $list;
  188.   $list .= $NIL;
  189.   foreach $item (@items) {
  190.     return 1 if index("\L$list\E", $NIL . "\L$item\E" . $NIL) != -1;
  191.   }
  192.   return 0;
  193. }
  194.  
  195. sub 'list {
  196.   local(@array) = @_;
  197.   return join($NIL, '', @array);
  198. }
  199.  
  200. sub 'array {
  201.   local($list) = @_;
  202.   return () unless $list;
  203.   $list = substr($list, 1);
  204.   return () unless $list;
  205.   return split(/$NIL/, $list);
  206. }
  207.