Skip to content

Commit

Permalink
simplify socks transport
Browse files Browse the repository at this point in the history
  • Loading branch information
DarienRaymond committed Aug 28, 2016
1 parent bd4ca7b commit 9b72d3c
Showing 1 changed file with 6 additions and 13 deletions.
19 changes: 6 additions & 13 deletions proxy/socks/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -295,28 +295,21 @@ func (this *Server) transport(reader io.Reader, writer io.Writer, session *proxy
input := ray.InboundInput()
output := ray.InboundOutput()

var inputFinish, outputFinish sync.Mutex
inputFinish.Lock()
outputFinish.Lock()
defer input.Close()
defer output.Release()

go func() {
v2reader := v2io.NewAdaptiveReader(reader)
defer v2reader.Release()

v2io.Pipe(v2reader, input)
inputFinish.Unlock()
input.Close()
}()

go func() {
v2writer := v2io.NewAdaptiveWriter(writer)
defer v2writer.Release()
v2writer := v2io.NewAdaptiveWriter(writer)
defer v2writer.Release()

v2io.Pipe(output, v2writer)
outputFinish.Unlock()
output.Release()
}()
outputFinish.Lock()
v2io.Pipe(output, v2writer)
output.Release()
}

type ServerFactory struct{}
Expand Down

0 comments on commit 9b72d3c

Please sign in to comment.