Differentiate between timeouts and disconnects

1 files changed, 4 insertions, 1 deletions

diff --git a/connect.go b/connect.go
index 7ecf30a..4be538a 100644
--- a/connect.go
+++ b/connect.go
@@ -6,10 +6,13 @@ import (
 	"github.com/anon55555/mt"
 )
 
-func connect(conn net.Conn) *serverConn {
+func connect(conn net.Conn, cc *clientConn) *serverConn {
 	sc := &serverConn{
 		Peer: mt.Connect(conn),
+		clt:  cc,
 	}
+	sc.log("-->", "connect")
+	cc.srv = sc
 
 	go handleSrv(sc)
 	return sc