/** * All rights reserved. Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.jivesoftware.smackx.socks5bytestream; import static org.junit.Assert.*; import java.io.DataInputStream; import java.io.DataOutputStream; import java.net.ServerSocket; import java.net.Socket; import org.jivesoftware.smack.XMPPException; import org.jivesoftware.smackx.socks5bytestream.packet.Bytestream.StreamHost; import org.junit.After; import org.junit.Before; import org.junit.Test; /** * Test for Socks5Client class. * * @author Henning Staib */ public class Socks5ClientTest { // settings private String serverAddress = "127.0.0.1"; private int serverPort = 7890; private String proxyJID = "proxy.xmpp-server"; private String digest = "digest"; private ServerSocket serverSocket; /** * Initialize fields used in the tests. * * @throws Exception should not happen */ @Before public void setup() throws Exception { // create SOCKS5 proxy server socket serverSocket = new ServerSocket(serverPort); } /** * A SOCKS5 client MUST close connection if server doesn't accept any of the given * authentication methods. (See RFC1928 Section 3) * * @throws Exception should not happen */ @Test public void shouldCloseSocketIfServerDoesNotAcceptAuthenticationMethod() throws Exception { // start thread to connect to SOCKS5 proxy Thread serverThread = new Thread() { @Override public void run() { StreamHost streamHost = new StreamHost(proxyJID, serverAddress); streamHost.setPort(serverPort); Socks5Client socks5Client = new Socks5Client(streamHost, digest); try { socks5Client.getSocket(10000); fail("exception should be thrown"); } catch (XMPPException e) { assertTrue(e.getMessage().contains( "establishing connection to SOCKS5 proxy failed")); } catch (Exception e) { fail(e.getMessage()); } } }; serverThread.start(); // accept connection form client Socket socket = serverSocket.accept(); DataInputStream in = new DataInputStream(socket.getInputStream()); DataOutputStream out = new DataOutputStream(socket.getOutputStream()); // validate authentication request assertEquals((byte) 0x05, (byte) in.read()); // version assertEquals((byte) 0x01, (byte) in.read()); // number of supported auth methods assertEquals((byte) 0x00, (byte) in.read()); // no-authentication method // respond that no authentication method is accepted out.write(new byte[] { (byte) 0x05, (byte) 0xFF }); out.flush(); // wait for client to shutdown serverThread.join(); // assert socket is closed assertEquals(-1, in.read()); } /** * The SOCKS5 client should close connection if server replies in an unsupported way. * * @throws Exception should not happen */ @Test public void shouldCloseSocketIfServerRepliesInUnsupportedWay() throws Exception { // start thread to connect to SOCKS5 proxy Thread serverThread = new Thread() { @Override public void run() { StreamHost streamHost = new StreamHost(proxyJID, serverAddress); streamHost.setPort(serverPort); Socks5Client socks5Client = new Socks5Client(streamHost, digest); try { socks5Client.getSocket(10000); fail("exception should be thrown"); } catch (XMPPException e) { assertTrue(e.getMessage().contains( "establishing connection to SOCKS5 proxy failed")); } catch (Exception e) { fail(e.getMessage()); } } }; serverThread.start(); // accept connection from client Socket socket = serverSocket.accept(); DataInputStream in = new DataInputStream(socket.getInputStream()); DataOutputStream out = new DataOutputStream(socket.getOutputStream()); // validate authentication request assertEquals((byte) 0x05, (byte) in.read()); // version assertEquals((byte) 0x01, (byte) in.read()); // number of supported auth methods assertEquals((byte) 0x00, (byte) in.read()); // no-authentication method // respond that no no-authentication method is used out.write(new byte[] { (byte) 0x05, (byte) 0x00 }); out.flush(); Socks5Utils.receiveSocks5Message(in); // reply with unsupported address type out.write(new byte[] { (byte) 0x05, (byte) 0x00, (byte) 0x00, (byte) 0x01, (byte) 0x00 }); out.flush(); // wait for client to shutdown serverThread.join(); // assert socket is closed assertEquals(-1, in.read()); } /** * The SOCKS5 client should close connection if server replies with an error. * * @throws Exception should not happen */ @Test public void shouldCloseSocketIfServerRepliesWithError() throws Exception { // start thread to connect to SOCKS5 proxy Thread serverThread = new Thread() { @Override public void run() { StreamHost streamHost = new StreamHost(proxyJID, serverAddress); streamHost.setPort(serverPort); Socks5Client socks5Client = new Socks5Client(streamHost, digest); try { socks5Client.getSocket(10000); fail("exception should be thrown"); } catch (XMPPException e) { assertTrue(e.getMessage().contains( "establishing connection to SOCKS5 proxy failed")); } catch (Exception e) { fail(e.getMessage()); } } }; serverThread.start(); Socket socket = serverSocket.accept(); DataInputStream in = new DataInputStream(socket.getInputStream()); DataOutputStream out = new DataOutputStream(socket.getOutputStream()); // validate authentication request assertEquals((byte) 0x05, (byte) in.read()); // version assertEquals((byte) 0x01, (byte) in.read()); // number of supported auth methods assertEquals((byte) 0x00, (byte) in.read()); // no-authentication method // respond that no no-authentication method is used out.write(new byte[] { (byte) 0x05, (byte) 0x00 }); out.flush(); Socks5Utils.receiveSocks5Message(in); // reply with full SOCKS5 message with an error code (01 = general SOCKS server // failure) out.write(new byte[] { (byte) 0x05, (byte) 0x01, (byte) 0x00, (byte) 0x03 }); byte[] address = digest.getBytes(); out.write(address.length); out.write(address); out.write(new byte[] { (byte) 0x00, (byte) 0x00 }); out.flush(); // wait for client to shutdown serverThread.join(); // assert socket is closed assertEquals(-1, in.read()); } /** * The SOCKS5 client should successfully connect to the SOCKS5 server * * @throws Exception should not happen */ @Test public void shouldSuccessfullyConnectToSocks5Server() throws Exception { // start thread to connect to SOCKS5 proxy Thread serverThread = new Thread() { @Override public void run() { StreamHost streamHost = new StreamHost(proxyJID, serverAddress); streamHost.setPort(serverPort); Socks5Client socks5Client = new Socks5Client(streamHost, digest); try { Socket socket = socks5Client.getSocket(10000); assertNotNull(socket); socket.getOutputStream().write(123); socket.close(); } catch (Exception e) { fail(e.getMessage()); } } }; serverThread.start(); Socket socket = serverSocket.accept(); DataInputStream in = new DataInputStream(socket.getInputStream()); DataOutputStream out = new DataOutputStream(socket.getOutputStream()); // validate authentication request assertEquals((byte) 0x05, (byte) in.read()); // version assertEquals((byte) 0x01, (byte) in.read()); // number of supported auth methods assertEquals((byte) 0x00, (byte) in.read()); // no-authentication method // respond that no no-authentication method is used out.write(new byte[] { (byte) 0x05, (byte) 0x00 }); out.flush(); byte[] address = digest.getBytes(); assertEquals((byte) 0x05, (byte) in.read()); // version assertEquals((byte) 0x01, (byte) in.read()); // connect request assertEquals((byte) 0x00, (byte) in.read()); // reserved byte (always 0) assertEquals((byte) 0x03, (byte) in.read()); // address type (domain) assertEquals(address.length, (byte) in.read()); // address length for (int i = 0; i < address.length; i++) { assertEquals(address[i], (byte) in.read()); // address } assertEquals((byte) 0x00, (byte) in.read()); // port assertEquals((byte) 0x00, (byte) in.read()); // reply with success SOCKS5 message out.write(new byte[] { (byte) 0x05, (byte) 0x00, (byte) 0x00, (byte) 0x03 }); out.write(address.length); out.write(address); out.write(new byte[] { (byte) 0x00, (byte) 0x00 }); out.flush(); // wait for client to shutdown serverThread.join(); // verify data sent from client assertEquals(123, in.read()); // assert socket is closed assertEquals(-1, in.read()); } /** * Close fake SOCKS5 proxy. * * @throws Exception should not happen */ @After public void cleanup() throws Exception { serverSocket.close(); } }