1 files changed, 22 insertions, 8 deletions
diff --git a/src/com/isode/stroke/filetransfer/S5BTransportSession.java b/src/com/isode/stroke/filetransfer/S5BTransportSession.java
index 768354f..4cb2a02 100644
--- a/src/com/isode/stroke/filetransfer/S5BTransportSession.java
+++ b/src/com/isode/stroke/filetransfer/S5BTransportSession.java
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2015 Isode Limited.
+ * Copyright (c) 2015-2016 Isode Limited.
  * All rights reserved.
  * See the COPYING file for more information.
  */
@@ -11,11 +11,20 @@
 
 package com.isode.stroke.filetransfer;
 
-import com.isode.stroke.signals.Signal1;
 import com.isode.stroke.signals.SignalConnection;
 
-public class S5BTransportSession<T> extends TransportSession {
+/**
+ * S5BTransportSession
+ *
+ * @param <T> Type of {@link SOCKS5AbstractBytestreamSession} to use.
+ */
+public class S5BTransportSession<T extends SOCKS5AbstractBytestreamSession> extends TransportSession {
 
+    /**
+     * Constructor for a read byte stream
+     * @param session byte stream session.  Should not be {@code null}.
+     * @param readStream read byte stream.  Should not be {@code null}.
+     */
 	public S5BTransportSession(
 		T session,
 		ReadBytestream readStream) {
@@ -24,6 +33,11 @@ public class S5BTransportSession<T> extends TransportSession {
 			initialize();
 	}
 
+	/**
+	 * Constructor for a write byte stream
+	 * @param session byte stream session.  Should not be {@code null}.
+	 * @param writeStream write byte stream.  Should not be {@code null}
+	 */
 	public S5BTransportSession(
 		T session,
 		WriteBytestream writeStream) {
@@ -34,20 +48,20 @@ public class S5BTransportSession<T> extends TransportSession {
 
 	public void start() {
 		if (readStream != null) {
-			//session.startSending(readStream);
+			session.startSending(readStream);
 		}
 		else {
-			//session.startReceiving(writeStream);
+			session.startReceiving(writeStream);
 		}
 	}
 
 	public void stop() {
-		//session.stop();
+		session.stop();
 	}
 
 	private void initialize() {
-		//finishedConnection = session.onFinished.connect(onFinished);
-		//bytesSentConnection = session.onBytesSent.connect(onBytesSent);
+		finishedConnection = session.onFinished.connect(onFinished);
+		bytesSentConnection = session.onBytesSent.connect(onBytesSent);
 	}
 
 	private T session;