Search in sources :

Example 16 with StreamHost

use of org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost in project Smack by igniterealtime.

the class Socks5ClientForInitiatorTest method shouldSuccessfullyConnectThroughLocalSocks5Proxy.

/**
     * Initiator and target should successfully connect to the local SOCKS5 proxy.
     * 
     * @throws Exception should not happen
     */
@Test
public void shouldSuccessfullyConnectThroughLocalSocks5Proxy() throws Exception {
    // start a local SOCKS5 proxy
    Socks5Proxy.setLocalSocks5ProxyPort(proxyPort);
    Socks5Proxy socks5Proxy = Socks5Proxy.getSocks5Proxy();
    socks5Proxy.start();
    // test data
    final byte[] data = new byte[] { 1, 2, 3 };
    // create digest
    final String digest = Socks5Utils.createDigest(sessionID, initiatorJID, targetJID);
    // allow connection of target with this digest
    socks5Proxy.addTransfer(digest);
    // build stream host information
    final StreamHost streamHost = new StreamHost(connection.getUser(), loopbackAddress, socks5Proxy.getPort());
    // target connects to local SOCKS5 proxy
    Thread targetThread = new Thread() {

        @Override
        public void run() {
            try {
                Socks5Client targetClient = new Socks5Client(streamHost, digest);
                Socket socket = targetClient.getSocket(10000);
                socket.getOutputStream().write(data);
            } catch (Exception e) {
                fail(e.getMessage());
            }
        }
    };
    targetThread.start();
    Thread.sleep(200);
    // initiator connects
    Socks5ClientForInitiator socks5Client = new Socks5ClientForInitiator(streamHost, digest, connection, sessionID, targetJID);
    Socket socket = socks5Client.getSocket(10000);
    // verify test data
    InputStream in = socket.getInputStream();
    for (int i = 0; i < data.length; i++) {
        assertEquals(data[i], in.read());
    }
    targetThread.join();
    // assert no XMPP messages were sent
    protocol.verifyAll();
    socks5Proxy.removeTransfer(digest);
    socks5Proxy.stop();
}
Also used : InputStream(java.io.InputStream) StreamHost(org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost) Socket(java.net.Socket) SmackException(org.jivesoftware.smack.SmackException) XMPPErrorException(org.jivesoftware.smack.XMPPException.XMPPErrorException) XMPPException(org.jivesoftware.smack.XMPPException) Test(org.junit.Test)

Example 17 with StreamHost

use of org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost in project Smack by igniterealtime.

the class Socks5ClientForInitiatorTest method shouldFailIfTargetIsNotConnectedToLocalSocks5Proxy.

/**
     * If the target is not connected to the local SOCKS5 proxy an exception should be thrown.
     * 
     * @throws Exception should not happen
     */
@Test
public void shouldFailIfTargetIsNotConnectedToLocalSocks5Proxy() throws Exception {
    // start a local SOCKS5 proxy
    Socks5Proxy.setLocalSocks5ProxyPort(proxyPort);
    Socks5Proxy socks5Proxy = Socks5Proxy.getSocks5Proxy();
    socks5Proxy.start();
    // build stream host information for local SOCKS5 proxy
    StreamHost streamHost = new StreamHost(connection.getUser(), loopbackAddress, socks5Proxy.getPort());
    // create digest to get the socket opened by target
    String digest = Socks5Utils.createDigest(sessionID, initiatorJID, targetJID);
    Socks5ClientForInitiator socks5Client = new Socks5ClientForInitiator(streamHost, digest, connection, sessionID, targetJID);
    try {
        socks5Client.getSocket(10000);
        fail("exception should be thrown");
    } catch (SmackException e) {
        assertTrue(e.getMessage().contains("target is not connected to SOCKS5 proxy"));
        // assert no XMPP messages were sent
        protocol.verifyAll();
    }
    socks5Proxy.stop();
}
Also used : SmackException(org.jivesoftware.smack.SmackException) StreamHost(org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost) Test(org.junit.Test)

Example 18 with StreamHost

use of org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost in project Smack by igniterealtime.

the class Socks5ClientTest method shouldCloseSocketIfServerRepliesInUnsupportedWay.

/**
     * 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, serverPort);
            Socks5Client socks5Client = new Socks5Client(streamHost, digest);
            try {
                socks5Client.getSocket(10000);
                fail("exception should be thrown");
            } catch (SmackException e) {
                assertTrue(e.getMessage().contains("Unsupported SOCKS5 address type"));
            } 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
    // version
    assertEquals((byte) 0x05, (byte) in.read());
    // number of supported auth methods
    assertEquals((byte) 0x01, (byte) in.read());
    // no-authentication method
    assertEquals((byte) 0x00, (byte) in.read());
    // 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());
}
Also used : DataOutputStream(java.io.DataOutputStream) SmackException(org.jivesoftware.smack.SmackException) DataInputStream(java.io.DataInputStream) StreamHost(org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost) SmackException(org.jivesoftware.smack.SmackException) Socket(java.net.Socket) ServerSocket(java.net.ServerSocket) Test(org.junit.Test)

Example 19 with StreamHost

use of org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost in project Smack by igniterealtime.

the class Socks5ClientTest method shouldSuccessfullyConnectToSocks5Server.

/**
     * 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, 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
    // version
    assertEquals((byte) 0x05, (byte) in.read());
    // number of supported auth methods
    assertEquals((byte) 0x01, (byte) in.read());
    // no-authentication method
    assertEquals((byte) 0x00, (byte) in.read());
    // respond that no no-authentication method is used
    out.write(new byte[] { (byte) 0x05, (byte) 0x00 });
    out.flush();
    byte[] address = digest.getBytes(StringUtils.UTF8);
    // version
    assertEquals((byte) 0x05, (byte) in.read());
    // connect request
    assertEquals((byte) 0x01, (byte) in.read());
    // reserved byte (always 0)
    assertEquals((byte) 0x00, (byte) in.read());
    // address type (domain)
    assertEquals((byte) 0x03, (byte) in.read());
    // address length
    assertEquals(address.length, (byte) in.read());
    for (int i = 0; i < address.length; i++) {
        // address
        assertEquals(address[i], (byte) in.read());
    }
    // port
    assertEquals((byte) 0x00, (byte) in.read());
    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());
}
Also used : DataOutputStream(java.io.DataOutputStream) DataInputStream(java.io.DataInputStream) StreamHost(org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost) Socket(java.net.Socket) ServerSocket(java.net.ServerSocket) SmackException(org.jivesoftware.smack.SmackException) Test(org.junit.Test)

Example 20 with StreamHost

use of org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost in project Smack by igniterealtime.

the class Socks5ByteStreamManagerTest method shouldNegotiateSocks5BytestreamAndTransferData.

/**
     * Invoking {@link Socks5BytestreamManager#establishSession(org.jxmpp.jid.Jid, String)} should successfully
     * negotiate and return a SOCKS5 Bytestream connection.
     * 
     * @throws Exception should not happen
     */
@Test
public void shouldNegotiateSocks5BytestreamAndTransferData() throws Exception {
    // disable clients local SOCKS5 proxy
    Socks5Proxy.setLocalSocks5ProxyEnabled(false);
    // get Socks5ByteStreamManager for connection
    Socks5BytestreamManager byteStreamManager = Socks5BytestreamManager.getBytestreamManager(connection);
    /**
         * create responses in the order they should be queried specified by the XEP-0065
         * specification
         */
    // build discover info that supports the SOCKS5 feature
    DiscoverInfo discoverInfo = Socks5PacketUtils.createDiscoverInfo(targetJID, initiatorJID);
    discoverInfo.addFeature(Bytestream.NAMESPACE);
    // return that SOCKS5 is supported if target is queried
    protocol.addResponse(discoverInfo, Verification.correspondingSenderReceiver, Verification.requestTypeGET);
    // build discover items containing a proxy item
    DiscoverItems discoverItems = Socks5PacketUtils.createDiscoverItems(xmppServer, initiatorJID);
    Item item = new Item(proxyJID);
    discoverItems.addItem(item);
    // return the proxy item if XMPP server is queried
    protocol.addResponse(discoverItems, Verification.correspondingSenderReceiver, Verification.requestTypeGET);
    // build discover info for proxy containing information about being a SOCKS5 proxy
    DiscoverInfo proxyInfo = Socks5PacketUtils.createDiscoverInfo(proxyJID, initiatorJID);
    Identity identity = new Identity("proxy", proxyJID.toString(), "bytestreams");
    proxyInfo.addIdentity(identity);
    // return the socks5 bytestream proxy identity if proxy is queried
    protocol.addResponse(proxyInfo, Verification.correspondingSenderReceiver, Verification.requestTypeGET);
    // build a socks5 stream host info containing the address and the port of the
    // proxy
    Bytestream streamHostInfo = Socks5PacketUtils.createBytestreamResponse(proxyJID, initiatorJID);
    streamHostInfo.addStreamHost(proxyJID, proxyAddress, 7778);
    // return stream host info if it is queried
    protocol.addResponse(streamHostInfo, Verification.correspondingSenderReceiver, Verification.requestTypeGET);
    // build used stream host response
    Bytestream streamHostUsedPacket = Socks5PacketUtils.createBytestreamResponse(targetJID, initiatorJID);
    streamHostUsedPacket.setSessionID(sessionID);
    streamHostUsedPacket.setUsedHost(proxyJID);
    // return used stream host info as response to the bytestream initiation
    protocol.addResponse(streamHostUsedPacket, new Verification<Bytestream, Bytestream>() {

        @Override
        public void verify(Bytestream request, Bytestream response) {
            assertEquals(response.getSessionID(), request.getSessionID());
            assertEquals(1, request.getStreamHosts().size());
            StreamHost streamHost = (StreamHost) request.getStreamHosts().toArray()[0];
            assertEquals(response.getUsedHost().getJID(), streamHost.getJID());
        }
    }, Verification.correspondingSenderReceiver, Verification.requestTypeSET);
    // build response to proxy activation
    IQ activationResponse = Socks5PacketUtils.createActivationConfirmation(proxyJID, initiatorJID);
    // return proxy activation response if proxy should be activated
    protocol.addResponse(activationResponse, new Verification<Bytestream, IQ>() {

        @Override
        public void verify(Bytestream request, IQ response) {
            assertEquals(targetJID, request.getToActivate().getTarget());
        }
    }, Verification.correspondingSenderReceiver, Verification.requestTypeSET);
    // start a local SOCKS5 proxy
    Socks5TestProxy socks5Proxy = Socks5TestProxy.getProxy(7778);
    socks5Proxy.start();
    // create digest to get the socket opened by target
    String digest = Socks5Utils.createDigest(sessionID, initiatorJID, targetJID);
    // finally call the method that should be tested
    OutputStream outputStream = byteStreamManager.establishSession(targetJID, sessionID).getOutputStream();
    // test the established bytestream
    InputStream inputStream = socks5Proxy.getSocket(digest).getInputStream();
    byte[] data = new byte[] { 1, 2, 3 };
    outputStream.write(data);
    byte[] result = new byte[3];
    inputStream.read(result);
    assertArrayEquals(data, result);
    protocol.verifyAll();
}
Also used : DiscoverInfo(org.jivesoftware.smackx.disco.packet.DiscoverInfo) InputStream(java.io.InputStream) OutputStream(java.io.OutputStream) ErrorIQ(org.jivesoftware.smack.packet.ErrorIQ) IQ(org.jivesoftware.smack.packet.IQ) DiscoverItems(org.jivesoftware.smackx.disco.packet.DiscoverItems) Item(org.jivesoftware.smackx.disco.packet.DiscoverItems.Item) Bytestream(org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream) Identity(org.jivesoftware.smackx.disco.packet.DiscoverInfo.Identity) StreamHost(org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost) Test(org.junit.Test)

Aggregations

StreamHost (org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream.StreamHost)18 Test (org.junit.Test)13 Bytestream (org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream)12 SmackException (org.jivesoftware.smack.SmackException)10 Socket (java.net.Socket)8 InputStream (java.io.InputStream)6 OutputStream (java.io.OutputStream)5 XMPPErrorException (org.jivesoftware.smack.XMPPException.XMPPErrorException)5 DataInputStream (java.io.DataInputStream)4 DataOutputStream (java.io.DataOutputStream)4 IOException (java.io.IOException)4 ServerSocket (java.net.ServerSocket)4 XMPPException (org.jivesoftware.smack.XMPPException)4 ArrayList (java.util.ArrayList)3 TimeoutException (java.util.concurrent.TimeoutException)3 FeatureNotSupportedException (org.jivesoftware.smack.SmackException.FeatureNotSupportedException)3 XMPPConnection (org.jivesoftware.smack.XMPPConnection)3 ErrorIQ (org.jivesoftware.smack.packet.ErrorIQ)3 IQ (org.jivesoftware.smack.packet.IQ)3 DiscoverInfo (org.jivesoftware.smackx.disco.packet.DiscoverInfo)3