nxshock/gonx
1
0
Fork 0
mirror of https://github.com/nxshock/gonx.git synced 2025-07-01 23:53:35 +05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add unix socket support

Browse source
This commit is contained in:
NXShock 2025-06-02 17:51:28 +05:00
parent ea03e2c4fb
commit 23fbce06c9
2 changed files with 41 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

42
mapping.go
View file

@ -39,7 +39,18 @@ func (h HostMapping) Add(host, outputUrlStr string) error {
slog.Debug(err.Error())
continue
}
go func() { _ = handleProxy(conn.(*tls.Conn), pd.Output) }()
go func() { _ = handleTcp(conn.(*tls.Conn), pd.Output) }()
}
}(pd)
case "unix":
go func(pd ProxyDirection) {
for {
conn, err := pd.listener.Accept()
if err != nil {
slog.Debug(err.Error())
continue
}
go func() { _ = handleUnix(conn.(*tls.Conn), pd.Output) }()
}
}(pd)
default:
@ -68,7 +79,7 @@ func handleTlsConn(conn *tls.Conn, hosts HostMapping) error {
return nil
}
func handleProxy(conn *tls.Conn, outputUrl *url.URL) error {
func handleTcp(conn *tls.Conn, outputUrl *url.URL) error {
c, err := net.Dial(outputUrl.Scheme, outputUrl.Host)
if err != nil {
return fmt.Errorf("dial: %v", err)
@ -94,3 +105,30 @@ func handleProxy(conn *tls.Conn, outputUrl *url.URL) error {
return nil
}
func handleUnix(conn *tls.Conn, outputUrl *url.URL) error {
c, err := net.Dial(outputUrl.Scheme, outputUrl.Host+outputUrl.Path)
if err != nil {
return fmt.Errorf("dial: %v", err)
}
defer c.Close()
wg := new(sync.WaitGroup)
wg.Add(2)
go func() {
defer wg.Done()
_, _ = io.Copy(conn, c)
}()
go func() {
defer wg.Done()
_, _ = io.Copy(c, conn)
}()
wg.Wait()
return nil
}