Robot
/
croc
mirror of https://github.com/schollz/croc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Need to close connections correctly

This commit is contained in:
Zack Scholl 2017-10-21 16:19:34 -06:00
parent 23e19b009c
commit 0ee7f75a22
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
relay.go
View File

@ -114,8 +114,6 @@ func (r *Relay) listener(id int) (err error) {
}
func (r *Relay) clientCommuncation(id int, connection net.Conn) {
defer connection.Close()
logger := log.WithFields(log.Fields{
"id": id,
"ip": connection.RemoteAddr().String(),
@ -171,6 +169,10 @@ func (r *Relay) clientCommuncation(id int, connection net.Conn) {
Pipe(con1, con2)
logger.Debug("done piping")
r.connections.Lock()
// close connections
r.connections.sender[key].Close()
r.connections.receiver[key].Close()
// delete connctions
delete(r.connections.sender, key)
delete(r.connections.receiver, key)
delete(r.connections.metadata, key)