(Java) WebSocket through SSH Tunnel

This example shows how to establish a WebSocket connection through an SSH tunnel. The WebSocket protocol communications will be encapsulated within an SSH tunnel.

Chilkat Java Downloads Java Libs for Windows, MacOS, Linux, Alpine Linux, Solaris Java Libs for Android

import com.chilkatsoft.*;

public class ChilkatExample {

  static {
    try {
        System.loadLibrary("chilkat");
    } catch (UnsatisfiedLinkError e) {
      System.err.println("Native code library failed to load.\n" + e);
      System.exit(1);
    }
  }

  public static void main(String argv[])
  {
    //  This example requires the Chilkat API to have been previously unlocked.
    //  See Global Unlock Sample for sample code.

    //  --------------------------------------------------
    //  This example borrows the code from the REST through SSH Tunnel example.
    //  We first use the Chilkat Socket object to establish a connection to the WebSocket server through an SSH Tunnel.
    //  Next, the Rest object uses the Socket object for its connection.
    //  Finally, the WebSocket object uses the Rest object for its connection.
    // 
    //  Hopefully the flexibility of this architecture is easy to see.  All of the HTTP functionality of the Rest object,
    //  such as HTTP authentication, custom headers, etc. is available to the WebSocket.  Likewise, all of the advanced functionality
    //  of the Socket object is in turn available to the Rest object.

    //  The high-level steps for accomplishing the task of running the WebSocket protocol through an SSH accomplished as follows:
    //  1) Create the SSH tunnel using Chilkat Socket.
    //  2) Open a port-forwarding channel (to the WebSocket server) within the tunnel.
    //  2) Tell Rest to use the Socket object.
    //  3) Tell WebSocket to use the Rest object.

    CkSocket tunnel = new CkSocket();

    String sshHostname = "www.my-ssh-server.com";
    int sshPort = 22;

    //  Connect to an SSH server and establish the SSH tunnel:
    boolean success = tunnel.SshOpenTunnel(sshHostname,sshPort);
    if (success != true) {
        System.out.println(tunnel.lastErrorText());
        return;
        }

    //  Authenticate with the SSH server via a login/password
    //  or with a public key.
    //  This example demonstrates SSH password authentication.
    success = tunnel.SshAuthenticatePw("mySshLogin","mySshPassword");
    if (success != true) {
        System.out.println(tunnel.lastErrorText());
        return;
        }

    //   OK, the SSH tunnel is setup.  Now open a channel within the tunnel.
    CkSocket channel;

    boolean bTls = true;
    int port = 443;
    int maxWaitMs = 5000;

    //  This returns a socket object that is a single channel within the SSH tunnel.
    //  The SSH channel is our logical port-forwarded connection through the SSH tunnel.
    //  Note: This example establishes a TLS connection to the target WebSocket server.
    //  (The TLS protocol will run on the logical channel within the SSH tunnel.)
    //  Your application can just as easily make a non-TLS connection by changing the arguments
    //  passed to SshOpenChannel.
    channel = tunnel.SshOpenChannel("some-websocket-server.com",port,bTls,maxWaitMs);
    if (tunnel.get_LastMethodSuccess() != true) {
        System.out.println(tunnel.lastErrorText());
        return;
        }

    //  Create a REST object and tell it to use the SSH channel.
    //  This connection is a TLS running on an SSH channel through an SSH tunnel.
    //  In other words, TLS is wrapped within the SSH tunnel.
    CkRest rest = new CkRest();
    success = rest.UseConnection(channel,false);
    if (success != true) {
        System.out.println(rest.lastErrorText());

        return;
        }

    //  Finally, tell our WebSocket object to use the Rest object..
    CkWebSocket ws = new CkWebSocket();

    //  Tell the WebSocket to use this connection.
    success = ws.UseConnection(rest);
    if (success != true) {
        System.out.println(ws.lastErrorText());
        return;
        }

    //  Add the standard WebSocket open handshake headers that will be needed.
    //  (This adds the required HTTP request headers to the rest object.)
    ws.AddClientHeaders();

    //  Add any additional headers that might be desired.
    //  Two common WebSocketSpecific headers are "Sec-WebSocket-Protocol" and "Origin".
    rest.AddHeader("Sec-WebSocket-Protocol","x-some-websocket-subprotocol");
    rest.AddHeader("Origin","http://some-websocket-server.com");

    //  Do the open handshake.
    String responseBody = rest.fullRequestNoBody("GET","/something");
    if (rest.get_LastMethodSuccess() != true) {
        System.out.println(rest.lastErrorText());
        return;
        }

    //  If successful, the HTTP response status code should be 101,
    //  and the response body will be empty. (If it failed, we'll have a look
    //  at the response body..)
    int statusCode = rest.get_ResponseStatusCode();
    System.out.println("Response status code: " + statusCode);

    if (statusCode != 101) {
        System.out.println(responseBody);
        System.out.println("-- Failed because of unexpected response status code.");
        return;
        }

    //  We have the expected 101 response, so let's now validate the
    //  contents of the response, such as the value sent by the server in the
    //  Sec-WebSocket-Accept header.
    success = ws.ValidateServerHandshake();
    if (success != true) {
        System.out.println(ws.lastErrorText());
        return;
        }

    System.out.println("WebSocket connection successful.");

    //  The application may now begin sending and receiving frames on the WebSocket connection.
    //  (At this point, we're done with the rest object...)
  }
}