mirror of
https://github.com/cbeuw/Cloak.git
synced 2024-10-23 15:46:25 +00:00
cfbf0dfcb6
* Fix critical bugs in session opening for TCP and UDP in case of Singleplex mode. - In case of TCP, don't open the session in the listener accept thread. This causes resource exhaustion of the tcp listener backlog queue in case of internet connection disruption or timeout. - In case of UDP, don't create a new session for every UDP packet. * Fix race in integration test. * Fix race condition in session maker * Code style improvements * Explicit session.Close() call is indeed needed Co-authored-by: Andy Wang <cbeuw.andy@gmail.com>
137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
package client
|
|
|
|
import (
|
|
"github.com/cbeuw/Cloak/internal/common"
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
mux "github.com/cbeuw/Cloak/internal/multiplex"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func RouteUDP(bindFunc func() (*net.UDPConn, error), streamTimeout time.Duration, singleplex bool, newSeshFunc func() *mux.Session) {
|
|
var sesh *mux.Session
|
|
localConn, err := bindFunc()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
streams := make(map[string]*mux.Stream)
|
|
|
|
data := make([]byte, 8192)
|
|
for {
|
|
i, addr, err := localConn.ReadFrom(data)
|
|
if err != nil {
|
|
log.Errorf("Failed to read first packet from proxy client: %v", err)
|
|
continue
|
|
}
|
|
|
|
if !singleplex && (sesh == nil || sesh.IsClosed()) {
|
|
sesh = newSeshFunc()
|
|
}
|
|
|
|
stream, ok := streams[addr.String()]
|
|
if !ok {
|
|
if singleplex {
|
|
sesh = newSeshFunc()
|
|
}
|
|
|
|
stream, err = sesh.OpenStream()
|
|
if err != nil {
|
|
if singleplex {
|
|
sesh.Close()
|
|
}
|
|
log.Errorf("Failed to open stream: %v", err)
|
|
continue
|
|
}
|
|
|
|
streams[addr.String()] = stream
|
|
proxyAddr := addr
|
|
go func(stream *mux.Stream, localConn *net.UDPConn) {
|
|
buf := make([]byte, 8192)
|
|
for {
|
|
n, err := stream.Read(buf)
|
|
if err != nil {
|
|
log.Tracef("copying stream to proxy client: %v", err)
|
|
stream.Close()
|
|
return
|
|
}
|
|
|
|
_, err = localConn.WriteTo(buf[:n], proxyAddr)
|
|
if err != nil {
|
|
log.Tracef("copying stream to proxy client: %v", err)
|
|
stream.Close()
|
|
return
|
|
}
|
|
}
|
|
}(stream, localConn)
|
|
}
|
|
|
|
_, err = stream.Write(data[:i])
|
|
if err != nil {
|
|
log.Tracef("copying proxy client to stream: %v", err)
|
|
delete(streams, addr.String())
|
|
stream.Close()
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
func RouteTCP(listener net.Listener, streamTimeout time.Duration, singleplex bool, newSeshFunc func() *mux.Session) {
|
|
var sesh *mux.Session
|
|
for {
|
|
localConn, err := listener.Accept()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
continue
|
|
}
|
|
if !singleplex && (sesh == nil || sesh.IsClosed()) {
|
|
sesh = newSeshFunc()
|
|
}
|
|
go func(sesh *mux.Session, localConn net.Conn, timeout time.Duration) {
|
|
if singleplex {
|
|
sesh = newSeshFunc()
|
|
}
|
|
|
|
data := make([]byte, 10240)
|
|
_ = localConn.SetReadDeadline(time.Now().Add(streamTimeout))
|
|
i, err := io.ReadAtLeast(localConn, data, 1)
|
|
if err != nil {
|
|
log.Errorf("Failed to read first packet from proxy client: %v", err)
|
|
localConn.Close()
|
|
return
|
|
}
|
|
var zeroTime time.Time
|
|
_ = localConn.SetReadDeadline(zeroTime)
|
|
|
|
stream, err := sesh.OpenStream()
|
|
if err != nil {
|
|
log.Errorf("Failed to open stream: %v", err)
|
|
localConn.Close()
|
|
if singleplex {
|
|
sesh.Close()
|
|
}
|
|
return
|
|
}
|
|
|
|
_, err = stream.Write(data[:i])
|
|
if err != nil {
|
|
log.Errorf("Failed to write to stream: %v", err)
|
|
localConn.Close()
|
|
stream.Close()
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
if _, err := common.Copy(localConn, stream); err != nil {
|
|
log.Tracef("copying stream to proxy client: %v", err)
|
|
}
|
|
}()
|
|
if _, err = common.Copy(stream, localConn); err != nil {
|
|
log.Tracef("copying proxy client to stream: %v", err)
|
|
}
|
|
}(sesh, localConn, streamTimeout)
|
|
}
|
|
}
|