mirrors/wireproxy
1
0
Fork 0
mirror of https://github.com/whyvl/wireproxy.git synced 2025-04-29 19:01:42 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use io.Copy instead of io.CopyBuffer

Browse source
This commit is contained in:
François Guerraz 2023-08-17 11:55:02 +01:00
parent d2f7768c51
commit d8bf236a75
1 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
routine.go
View file

@ -162,9 +162,8 @@ func (c CredentialValidator) Valid(username, password string) bool {
} }
// connForward copy data from `from` to `to`, then close both stream. // connForward copy data from `from` to `to`, then close both stream.
func connForward(bufSize int, from io.ReadWriteCloser, to io.ReadWriteCloser) { func connForward(from io.ReadWriteCloser, to io.ReadWriteCloser) {
buf := make([]byte, bufSize) _, err := io.Copy(to, from)
_, err := io.CopyBuffer(to, from, buf)
if err != nil { if err != nil {
errorLogger.Printf("Cannot forward traffic: %s\n", err.Error()) errorLogger.Printf("Cannot forward traffic: %s\n", err.Error())
} }
@ -188,8 +187,8 @@ func tcpClientForward(vt *VirtualTun, raddr *addressPort, conn net.Conn) {
return return
} }
go connForward(1024, sconn, conn) go connForward(sconn, conn)
go connForward(1024, conn, sconn) go connForward(conn, sconn)
} }
// STDIOTcpForward starts a new connection via wireguard and forward traffic from `conn` // STDIOTcpForward starts a new connection via wireguard and forward traffic from `conn`
@ -214,8 +213,8 @@ func STDIOTcpForward(vt *VirtualTun, raddr *addressPort) {
return return
} }
go connForward(1024, os.Stdin, sconn) go connForward(os.Stdin, sconn)
go connForward(1024, sconn, stdout) go connForward(sconn, stdout)
} }
// SpawnRoutine spawns a local TCP server which acts as a proxy to the specified target // SpawnRoutine spawns a local TCP server which acts as a proxy to the specified target
@ -265,8 +264,8 @@ func tcpServerForward(vt *VirtualTun, raddr *addressPort, conn net.Conn) {
return return
} }
go connForward(1024, sconn, conn) go connForward(sconn, conn)
go connForward(1024, conn, sconn) go connForward(conn, sconn)
} }
// SpawnRoutine spawns a TCP server on wireguard which acts as a proxy to the specified target // SpawnRoutine spawns a TCP server on wireguard which acts as a proxy to the specified target