Ensure channel is set before returning from connect().

src/main/java/org/spacehq/packetlib/tcp/TcpClientSession.java

@@ -93,13 +93,20 @@ public class TcpClientSession extends TcpSession {
                         future.addListener(new ChannelFutureListener() {
                             @Override
                             public void operationComplete(ChannelFuture future) throws Exception {
-                                System.err.println("Future start.");
                                 if(!future.isSuccess()) {
                                     exceptionCaught(null, future.cause());
                                 }
-                                System.err.println("Future end.");
                             }
                         }).await();
+
+                        if(future.isSuccess()) {
+                            while(!isConnected()) {
+                                try {
+                                    Thread.sleep(5);
+                                } catch(InterruptedException e) {
+                                }
+                            }
+                        }
                     } catch(Throwable t) {
                         exceptionCaught(null, t);
                     }
@@ -108,7 +115,6 @@ public class TcpClientSession extends TcpSession {
 
             if(wait) {
                 connectTask.run();
-                System.err.println("Connect completed.");
             } else {
                 new Thread(connectTask).start();
             }

src/main/java/org/spacehq/packetlib/tcp/TcpConnectionListener.java

@@ -85,8 +85,6 @@ public class TcpConnectionListener implements ConnectionListener {
                 pipeline.addLast("sizer", new TcpPacketSizer(session));
                 pipeline.addLast("codec", new TcpPacketCodec(session));
                 pipeline.addLast("manager", session);
-
-                server.addSession(session);
             }
         }).group(this.group).localAddress(this.host, this.port).bind();
 

src/main/java/org/spacehq/packetlib/tcp/TcpServerSession.java

@@ -21,6 +21,13 @@ public class TcpServerSession extends TcpSession {
         return ret;
     }
 
+    @Override
+    public void channelActive(ChannelHandlerContext ctx) throws Exception {
+        super.channelActive(ctx);
+
+        this.server.addSession(this);
+    }
+
     @Override
     public void channelInactive(ChannelHandlerContext ctx) throws Exception {
         super.channelInactive(ctx);

src/main/java/org/spacehq/packetlib/tcp/TcpSession.java

@@ -308,18 +308,13 @@ public abstract class TcpSession extends SimpleChannelInboundHandler<Packet> imp
 
     @Override
     public void channelActive(ChannelHandlerContext ctx) throws Exception {
-        System.err.println("Channel active?");
         if(this.disconnected || this.channel != null) {
             ctx.channel().close();
             return;
         }
 
-        System.err.println("Channel active.");
-
         this.channel = ctx.channel();
 
-        System.err.println("Channel set.");
-
         this.packetHandleThread = new Thread(new Runnable() {
             @Override
             public void run() {