Use another thread for sending frames.

Fixes NetworkOnMainThreadException on Android 4.1 (Jelly Bean).

Author: codebutler

src/com/codebutler/android_websockets/HybiParser.java

@@ -128,7 +128,7 @@ public void start(HappyDataInputStream stream) throws IOException {
                     break;
             }
         }
-        mClient.getHandler().onDisconnect(0, "EOF");
+        mClient.getListener().onDisconnect(0, "EOF");
     }
 
     private void parseOpcode(byte data) throws ProtocolError {
@@ -262,25 +262,25 @@ private void emitFrame() throws IOException {
             if (mFinal) {
                 byte[] message = mBuffer.toByteArray();
                 if (mMode == MODE_TEXT) {
-                    mClient.getHandler().onMessage(encode(message));
+                    mClient.getListener().onMessage(encode(message));
                 } else {
-                    mClient.getHandler().onMessage(message);
+                    mClient.getListener().onMessage(message);
                 }
                 reset();
             }
 
         } else if (opcode == OP_TEXT) {
             if (mFinal) {
                 String messageText = encode(payload);
-                mClient.getHandler().onMessage(messageText);
+                mClient.getListener().onMessage(messageText);
             } else {
                 mMode = MODE_TEXT;
                 mBuffer.write(payload);
             }
 
         } else if (opcode == OP_BINARY) {
             if (mFinal) {
-                mClient.getHandler().onMessage(payload);
+                mClient.getListener().onMessage(payload);
             } else {
                 mMode = MODE_BINARY;
                 mBuffer.write(payload);
@@ -290,7 +290,7 @@ private void emitFrame() throws IOException {
             int    code   = (payload.length >= 2) ? 256 * payload[0] + payload[1] : 0;
             String reason = (payload.length >  2) ? encode(slice(payload, 2))     : null;
             Log.d(TAG, "Got close op! " + code + " " + reason);
-            mClient.getHandler().onDisconnect(code, reason);
+            mClient.getListener().onDisconnect(code, reason);
 
         } else if (opcode == OP_PING) {
             if (payload.length > 125) { throw new ProtocolError("Ping payload too large"); }

src/com/codebutler/android_websockets/WebSocketClient.java

@@ -1,5 +1,7 @@
 package com.codebutler.android_websockets;
 
+import android.os.Handler;
+import android.os.HandlerThread;
 import android.text.TextUtils;
 import android.util.Base64;
 import android.util.Log;
@@ -27,9 +29,11 @@ public class WebSocketClient {
     private static final String TAG = "WebSocketClient";
 
     private URI                      mURI;
-    private Handler                  mHandler;
+    private Listener                 mListener;
     private Socket                   mSocket;
     private Thread                   mThread;
+    private HandlerThread            mHandlerThread;
+    private Handler                  mHandler;
     private List<BasicNameValuePair> mExtraHeaders;
     private HybiParser               mParser;
 
@@ -41,15 +45,19 @@ public static void setTrustManagers(TrustManager[] tm) {
         sTrustManagers = tm;
     }
 
-    public WebSocketClient(URI uri, Handler handler, List<BasicNameValuePair> extraHeaders) {
+    public WebSocketClient(URI uri, Listener listener, List<BasicNameValuePair> extraHeaders) {
         mURI          = uri;
-        mHandler      = handler;
+        mListener = listener;
         mExtraHeaders = extraHeaders;
         mParser       = new HybiParser(this);
+
+        mHandlerThread = new HandlerThread("websocket-thread");
+        mHandlerThread.start();
+        mHandler = new Handler(mHandlerThread.getLooper());
     }
 
-    public Handler getHandler() {
-        return mHandler;
+    public Listener getListener() {
+        return mListener;
     }
 
     public void connect() {
@@ -69,7 +77,7 @@ public void run() {
                     }
 
                     String originScheme = mURI.getScheme().equals("wss") ? "https" : "http";
-                    URI origin = new URI(originScheme, mURI.getSchemeSpecificPart(), null);
+                    URI origin = new URI(originScheme, "//" + mURI.getHost(), null);
 
                     SocketFactory factory = mURI.getScheme().equals("wss") ? getSSLSocketFactory() : SocketFactory.getDefault();
                     mSocket = factory.createSocket(mURI.getHost(), port);
@@ -109,32 +117,42 @@ public void run() {
                         }
                     }
 
-                    mHandler.onConnect();
+                    mListener.onConnect();
 
                     // Now decode websocket frames.
                     mParser.start(stream);
 
                 } catch (EOFException ex) {
                     Log.d(TAG, "WebSocket EOF!", ex);
-                    mHandler.onDisconnect(0, "EOF");
+                    mListener.onDisconnect(0, "EOF");
 
                 } catch (SSLException ex) {
                     // Connection reset by peer
                     Log.d(TAG, "Websocket SSL error!", ex);
-                    mHandler.onDisconnect(0, "SSL");
+                    mListener.onDisconnect(0, "SSL");
 
                 } catch (Exception ex) {
-                    mHandler.onError(ex);
+                    mListener.onError(ex);
                 }
             }
         });
         mThread.start();
     }
 
-    public void disconnect() throws IOException {
+    public void disconnect() {
         if (mSocket != null) {
-            mSocket.close();
-            mSocket = null;
+            mHandler.post(new Runnable() {
+                @Override
+                public void run() {
+                    try {
+                        mSocket.close();
+                        mSocket = null;
+                    } catch (IOException ex) {
+                        Log.d(TAG, "Error while disconnecting", ex);
+                        mListener.onError(ex);
+                    }
+                }
+            });
         }
     }
 
@@ -154,7 +172,7 @@ private StatusLine parseStatusLine(String line) {
     }
 
     private Header parseHeader(String line) {
-        return BasicLineParser.parseHeader(line,  new BasicLineParser());
+        return BasicLineParser.parseHeader(line, new BasicLineParser());
     }
 
     // Can't use BufferedReader because it buffers past the HTTP data.
@@ -185,19 +203,24 @@ private String createSecret() {
         return Base64.encodeToString(nonce, Base64.DEFAULT).trim();
     }
 
-    void sendFrame(byte[] frame) {
-        try {
-            synchronized (mSendLock) {
-                OutputStream outputStream = mSocket.getOutputStream();
-                outputStream.write(frame);
-                outputStream.flush();
+    void sendFrame(final byte[] frame) {
+        mHandler.post(new Runnable() {
+            @Override
+            public void run() {
+                try {
+                    synchronized (mSendLock) {
+                        OutputStream outputStream = mSocket.getOutputStream();
+                        outputStream.write(frame);
+                        outputStream.flush();
+                    }
+                } catch (IOException e) {
+                    mListener.onError(e);
+                }
             }
-        } catch (IOException e) {
-            mHandler.onError(e);
-        }
+        });
     }
 
-    public interface Handler {
+    public interface Listener {
         public void onConnect();
         public void onMessage(String message);
         public void onMessage(byte[] data);