From 84a565bae6a0eb2c17a78c7c756bc094bc1c33c4 Mon Sep 17 00:00:00 2001
From: idk <problemsolver@openmailbox.org>
Date: Wed, 26 Dec 2018 13:30:42 -0500
Subject: [PATCH] Conns aren't being saved

---
 socks/manager.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/socks/manager.go b/socks/manager.go
index 79d9151..f685e62 100644
--- a/socks/manager.go
+++ b/socks/manager.go
@@ -19,6 +19,7 @@ type Manager struct {
 	resolver.Resolver
 	socks5.Config
 	*sam3.SAM
+	listen  net.Listener
 	conns   []conn.Conn
 	datadir string
 	host    string
@@ -33,11 +34,10 @@ func (m Manager) Serve() error {
 	if err != nil {
 		return err
 	}
-	var listen net.Listener
-	if err := net.Listen("tcp", m.host+":"+m.port); err != nil {
+	if m.listen, err = net.Listen("tcp", m.host+":"+m.port); err != nil {
 		return err
 	}
-	if err := server.ListenAndServe(listen); err != nil {
+	if err := server.Serve(listen); err != nil {
 		return err
 	}
 	return nil
@@ -77,6 +77,7 @@ func (m Manager) Cleanup() error {
 		}
 	}
 	log.Println("Finished cleanup.")
+	m.listen.Close()
 	return nil
 }