Fix nix builds: go dependencies are awful
This commit is contained in:
47
netforward/stream.go
Normal file
47
netforward/stream.go
Normal file
@@ -0,0 +1,47 @@
|
||||
// From https://github.com/goburrow/netforward
|
||||
package netforward
|
||||
|
||||
import (
|
||||
"io"
|
||||
"log"
|
||||
"net"
|
||||
)
|
||||
|
||||
func Forward(remote Dialer, local net.Listener) error {
|
||||
for {
|
||||
conn, err := local.Accept()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
go forward(remote, conn)
|
||||
}
|
||||
}
|
||||
|
||||
func forward(remote Dialer, conn io.ReadWriteCloser) {
|
||||
defer conn.Close()
|
||||
|
||||
remoteConn, err := remote.Dial()
|
||||
if err != nil {
|
||||
log.Printf("dial failed: %v", err)
|
||||
return
|
||||
}
|
||||
defer remoteConn.Close()
|
||||
|
||||
// remote -> local
|
||||
go func() {
|
||||
buf := getBuffer()
|
||||
defer releaseBuffer(buf)
|
||||
_, err := io.CopyBuffer(remoteConn, conn, buf)
|
||||
if err != nil {
|
||||
log.Printf("forward failed: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// local -> remote
|
||||
buf := getBuffer()
|
||||
defer releaseBuffer(buf)
|
||||
_, err = io.CopyBuffer(conn, remoteConn, buf)
|
||||
if err != nil {
|
||||
log.Printf("forward failed: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user