wamblee.org Git - utils/blob - socketproxy/src/main/java/org/wamblee/socketproxy/SocketProxy.java

   1 package org.wamblee.socketproxy;
   2
   3 /*
   4  * Created on Apr 5, 2005
   5  */
   6
   7 import java.io.IOException;
   8 import java.io.InputStream;
   9 import java.io.OutputStream;
  10 import java.net.ServerSocket;
  11 import java.net.Socket;
  12
  13 /**
  14  * @author erik
  15  *
  16  * TODO To change the template for this generated type comment go to Window -
  17  * Preferences - Java - Code Style - Code Templates
  18  */
  19 public class SocketProxy {
  20
  21     public static void main( final String[] args ) throws IOException {
  22         for ( int i = 0; i < args.length; i++ ) {
  23             // System.out.println(i + " " + args[i]);
  24             String[] fields = args[i].split( ":" );
  25             final int localPort = Integer.parseInt( fields[0] );
  26             final String host = fields[1];
  27             final int remotePort = Integer.parseInt( fields[2] );
  28             runSocketProxy( localPort, host, remotePort );
  29         }
  30     }
  31
  32     /**
  33      * @param localPort
  34      * @param host
  35      * @param remotePort
  36      */
  37     private static void runSocketProxy( final int localPort,
  38             final String host, final int remotePort ) {
  39         new Thread( new Runnable( ) {
  40             public void run( ) {
  41                 try {
  42                     new SocketProxy( localPort, host, remotePort );
  43                 } catch ( IOException e ) {
  44                     System.out.println( "Problem with socket " + localPort
  45                             + ":" + host + ":" + remotePort );
  46                     e.printStackTrace( );
  47                 }
  48             }
  49         } ).start( );
  50     }
  51
  52     public SocketProxy( int localPort, String remoteHost, int remotePort )
  53             throws IOException {
  54         System.out.println( "Listening on port " + localPort );
  55         ServerSocket server = new ServerSocket( localPort );
  56         for ( ;; ) {
  57             Socket socket = server.accept( );
  58             System.out.println( "Got local connection on port "
  59                     + localPort );
  60             InputStream localIs = socket.getInputStream( );
  61             OutputStream localOs = socket.getOutputStream( );
  62             Socket clientSocket = new Socket( remoteHost, remotePort );
  63             final String description = "Port forwarding: " + localPort
  64                     + " -> " + remoteHost + ":" + remotePort;
  65             System.out.println( description + " established." );
  66             InputStream serverIs = clientSocket.getInputStream( );
  67             OutputStream serverOs = clientSocket.getOutputStream( );
  68             Barrier barrier = new Barrier(2);
  69             final Thread t1 = runForwarder( barrier, "> ", localIs, serverOs );
  70             final Thread t2 = runForwarder( barrier, "< ", serverIs, localOs );
  71             waitForConnectionClose( description, t1, t2 );
  72         }
  73     }
  74
  75     /**
  76      * @param description
  77      * @param t1
  78      * @param t2
  79      */
  80     private void waitForConnectionClose( final String description,
  81             final Thread t1, final Thread t2 ) {
  82         new Thread( new Runnable( ) {
  83             public void run( ) {
  84                 try {
  85                     t1.join( );
  86                     t2.join( );
  87                 } catch ( InterruptedException e ) {
  88                     e.printStackTrace( );
  89                 }
  90                 System.out.println( description + " closed" );
  91             }
  92         } ).start( );
  93     }
  94
  95     private Thread runForwarder( final Barrier barrier, final String prefix,
  96             final InputStream is, final OutputStream os ) {
  97         Thread t = new ForwarderThread(prefix, barrier, is, os);
  98         t.start( );
  99         return t;
 100     }
 101 }