[3102] | 1 | package nu.fw.jeti.plugins.filetransfer.socks5.jsocks; |
---|
| 2 | |
---|
| 3 | import java.net.*; |
---|
| 4 | import java.io.*; |
---|
| 5 | //TODO remove if bind not necessary |
---|
| 6 | /** |
---|
| 7 | SocksServerSocket allows to accept connections from one particular |
---|
| 8 | host through the SOCKS4 or SOCKS5 proxy. |
---|
| 9 | */ |
---|
| 10 | public class SocksServerSocket extends ServerSocket{ |
---|
| 11 | //Data members |
---|
| 12 | protected Proxy proxy; |
---|
| 13 | protected String localHost; |
---|
| 14 | protected InetAddress localIP; |
---|
| 15 | protected int localPort; |
---|
| 16 | |
---|
| 17 | boolean doing_direct = false; |
---|
| 18 | InetAddress remoteAddr; |
---|
| 19 | |
---|
| 20 | /** |
---|
| 21 | * Creates ServerSocket capable of accepting one connection |
---|
| 22 | * through the firewall, uses default Proxy. |
---|
| 23 | *@param host Host from which the connection should be recieved. |
---|
| 24 | *@param port Port number of the primary connection. |
---|
| 25 | */ |
---|
| 26 | public SocksServerSocket(String host,int port) |
---|
| 27 | throws SocksException,UnknownHostException,IOException{ |
---|
| 28 | this(Proxy.defaultProxy,host,port); |
---|
| 29 | } |
---|
| 30 | /** |
---|
| 31 | *Creates ServerSocket capable of accepting one connection |
---|
| 32 | *through the firewall, uses given proxy. |
---|
| 33 | *@param p Proxy object to use. |
---|
| 34 | *@param host Host from which the connection should be recieved. |
---|
| 35 | *@param port Port number of the primary connection. |
---|
| 36 | */ |
---|
| 37 | public SocksServerSocket(Proxy p,String host,int port) |
---|
| 38 | throws SocksException,UnknownHostException,IOException{ |
---|
| 39 | |
---|
| 40 | |
---|
| 41 | super(0); |
---|
| 42 | if(p == null) throw new SocksException(Proxy.SOCKS_NO_PROXY); |
---|
| 43 | //proxy=p; |
---|
| 44 | proxy = p.copy(); |
---|
| 45 | if(proxy.isDirect(host)){ |
---|
| 46 | remoteAddr = InetAddress.getByName(host); |
---|
| 47 | proxy = null; |
---|
| 48 | doDirect(); |
---|
| 49 | }else{ |
---|
| 50 | processReply(proxy.bind(host,port)); |
---|
| 51 | } |
---|
| 52 | } |
---|
| 53 | |
---|
| 54 | /** |
---|
| 55 | * Creates ServerSocket capable of accepting one connection |
---|
| 56 | * through the firewall, uses default Proxy. |
---|
| 57 | *@param ip Host from which the connection should be recieved. |
---|
| 58 | *@param port Port number of the primary connection. |
---|
| 59 | */ |
---|
| 60 | public SocksServerSocket(InetAddress ip, int port) throws SocksException, |
---|
| 61 | IOException{ |
---|
| 62 | this(Proxy.defaultProxy,ip,port); |
---|
| 63 | } |
---|
| 64 | |
---|
| 65 | /** |
---|
| 66 | *Creates ServerSocket capable of accepting one connection |
---|
| 67 | *through the firewall, uses given proxy. |
---|
| 68 | *@param p Proxy object to use. |
---|
| 69 | *@param ip Host from which the connection should be recieved. |
---|
| 70 | *@param port Port number of the primary connection. |
---|
| 71 | */ |
---|
| 72 | public SocksServerSocket(Proxy p,InetAddress ip, int port) |
---|
| 73 | throws SocksException,IOException{ |
---|
| 74 | super(0); |
---|
| 75 | |
---|
| 76 | if(p == null) throw new SocksException(Proxy.SOCKS_NO_PROXY); |
---|
| 77 | this.proxy = p.copy(); |
---|
| 78 | |
---|
| 79 | if(proxy.isDirect(ip)){ |
---|
| 80 | remoteAddr = ip; |
---|
| 81 | doDirect(); |
---|
| 82 | }else{ |
---|
| 83 | processReply(proxy.bind(ip,port)); |
---|
| 84 | } |
---|
| 85 | } |
---|
| 86 | |
---|
| 87 | |
---|
| 88 | /** |
---|
| 89 | * Accepts the incoming connection. |
---|
| 90 | */ |
---|
| 91 | public Socket accept() throws IOException{ |
---|
| 92 | Socket s; |
---|
| 93 | |
---|
| 94 | if(!doing_direct){ |
---|
| 95 | if(proxy == null) return null; |
---|
| 96 | |
---|
| 97 | ProxyMessage msg = proxy.accept(); |
---|
| 98 | s = msg.ip == null? new SocksSocket(msg.host,msg.port,proxy) |
---|
| 99 | : new SocksSocket(msg.ip,msg.port,proxy); |
---|
| 100 | //Set timeout back to 0 |
---|
| 101 | proxy.proxySocket.setSoTimeout(0); |
---|
| 102 | }else{ //Direct Connection |
---|
| 103 | |
---|
| 104 | //Mimic the proxy behaviour, |
---|
| 105 | //only accept connections from the speciefed host. |
---|
| 106 | while(true){ |
---|
| 107 | s = super.accept(); |
---|
| 108 | if(s.getInetAddress().equals(remoteAddr)){ |
---|
| 109 | //got the connection from the right host |
---|
| 110 | //Close listenning socket. |
---|
| 111 | break; |
---|
| 112 | }else |
---|
| 113 | s.close(); //Drop all connections from other hosts |
---|
| 114 | } |
---|
| 115 | |
---|
| 116 | } |
---|
| 117 | proxy = null; |
---|
| 118 | //Return accepted socket |
---|
| 119 | return s; |
---|
| 120 | } |
---|
| 121 | |
---|
| 122 | /** |
---|
| 123 | * Closes the connection to proxy if socket have not been accepted, if |
---|
| 124 | * the direct connection is used, closes direct ServerSocket. If the |
---|
| 125 | * client socket have been allready accepted, does nothing. |
---|
| 126 | */ |
---|
| 127 | public void close() throws IOException{ |
---|
| 128 | super.close(); |
---|
| 129 | if(proxy != null) proxy.endSession(); |
---|
| 130 | proxy = null; |
---|
| 131 | } |
---|
| 132 | |
---|
| 133 | /** |
---|
| 134 | Get the name of the host proxy is using to listen for incoming |
---|
| 135 | connection. |
---|
| 136 | <P> |
---|
| 137 | Usefull when address is returned by proxy as the hostname. |
---|
| 138 | @return the hostname of the address proxy is using to listen |
---|
| 139 | for incoming connection. |
---|
| 140 | */ |
---|
| 141 | public String getHost(){ |
---|
| 142 | return localHost; |
---|
| 143 | } |
---|
| 144 | |
---|
| 145 | /** |
---|
| 146 | * Get address assigned by proxy to listen for incomming |
---|
| 147 | * connections, or the local machine address if doing direct |
---|
| 148 | * connection. |
---|
| 149 | */ |
---|
| 150 | public InetAddress getInetAddress(){ |
---|
| 151 | if(localIP == null){ |
---|
| 152 | try{ |
---|
| 153 | localIP = InetAddress.getByName(localHost); |
---|
| 154 | }catch(UnknownHostException e){ |
---|
| 155 | return null; |
---|
| 156 | } |
---|
| 157 | } |
---|
| 158 | return localIP; |
---|
| 159 | } |
---|
| 160 | |
---|
| 161 | /** |
---|
| 162 | * Get port assigned by proxy to listen for incoming connections, or |
---|
| 163 | the port chosen by local system, if accepting directly. |
---|
| 164 | */ |
---|
| 165 | public int getLocalPort(){ |
---|
| 166 | return localPort; |
---|
| 167 | } |
---|
| 168 | |
---|
| 169 | /** |
---|
| 170 | Set Timeout. |
---|
| 171 | |
---|
| 172 | @param timeout Amount of time in milliseconds, accept should wait for |
---|
| 173 | incoming connection before failing with exception. |
---|
| 174 | Zero timeout implies infinity. |
---|
| 175 | */ |
---|
| 176 | public void setSoTimeout(int timeout) throws SocketException{ |
---|
| 177 | super.setSoTimeout(timeout); |
---|
| 178 | if(!doing_direct) proxy.proxySocket.setSoTimeout(timeout); |
---|
| 179 | } |
---|
| 180 | |
---|
| 181 | |
---|
| 182 | //Private Methods |
---|
| 183 | ////////////////// |
---|
| 184 | |
---|
| 185 | private void processReply(ProxyMessage reply)throws SocksException{ |
---|
| 186 | localPort = reply.port; |
---|
| 187 | /* |
---|
| 188 | * If the server have assigned same host as it was contacted on |
---|
| 189 | * it might return an address of all zeros |
---|
| 190 | */ |
---|
| 191 | if(reply.host.equals("0.0.0.0")){ |
---|
| 192 | localIP = proxy.proxyIP; |
---|
| 193 | localHost = localIP.getHostName(); |
---|
| 194 | }else{ |
---|
| 195 | localHost = reply.host; |
---|
| 196 | localIP = reply.ip; |
---|
| 197 | } |
---|
| 198 | } |
---|
| 199 | |
---|
| 200 | private void doDirect(){ |
---|
| 201 | doing_direct = true; |
---|
| 202 | localPort = super.getLocalPort(); |
---|
| 203 | localIP = super.getInetAddress(); |
---|
| 204 | localHost = localIP.getHostName(); |
---|
| 205 | } |
---|
| 206 | |
---|
| 207 | } |
---|