From fa1b158cc47e7f5d8958823ac51504c41a4b440e Mon Sep 17 00:00:00 2001 From: Bluesaxorcist Date: Thu, 8 Aug 2019 17:50:48 -0500 Subject: Modified dialer to return conn and error --- modes/pt_socks5/pt_socks5.go | 4 ++-- modes/stun_udp/stun_udp.go | 4 ++-- modes/transparent_udp/transparent_udp.go | 4 ++-- shapeshifter-dispatcher/shapeshifter-dispatcher.go | 8 ++++---- 4 files changed, 10 insertions(+), 10 deletions(-) diff --git a/modes/pt_socks5/pt_socks5.go b/modes/pt_socks5/pt_socks5.go index 067a036..14b55e1 100644 --- a/modes/pt_socks5/pt_socks5.go +++ b/modes/pt_socks5/pt_socks5.go @@ -107,7 +107,7 @@ func clientHandler(target string, termMon *termmon.TermMonitor, name string, con } } - var dialer func(address string) net.Conn + var dialer func(address string) (net.Conn, error) // Deal with arguments. switch name { @@ -160,7 +160,7 @@ func clientHandler(target string, termMon *termmon.TermMonitor, name string, con f := dialer - remote := f(socksReq.Target) + remote, _ := f(socksReq.Target) if err != nil { log.Errorf("%s(%s) - outgoing connection failed: %s", name, addrStr, log.ElideError(err)) socksReq.Reply(socks5.ErrorToReplyCode(err)) diff --git a/modes/stun_udp/stun_udp.go b/modes/stun_udp/stun_udp.go index e80526b..7f0cdea 100644 --- a/modes/stun_udp/stun_udp.go +++ b/modes/stun_udp/stun_udp.go @@ -168,7 +168,7 @@ func dialConn(tracker *ConnTracker, addr string, target string, name string, opt return } - var f func(address string) net.Conn + var f func(address string) (net.Conn, error) // Deal with arguments. switch name { @@ -200,7 +200,7 @@ func dialConn(tracker *ConnTracker, addr string, target string, name string, opt } fmt.Println("Dialing ", target) - remote := f(target) + remote, _ := f(target) // if err != nil { // fmt.Println("outgoing connection failed", err) // log.Errorf("(%s) - outgoing connection failed: %s", target, log.ElideError(err)) diff --git a/modes/transparent_udp/transparent_udp.go b/modes/transparent_udp/transparent_udp.go index 9cceb82..6e573c2 100644 --- a/modes/transparent_udp/transparent_udp.go +++ b/modes/transparent_udp/transparent_udp.go @@ -173,7 +173,7 @@ func dialConn(tracker *ConnTracker, addr string, target string, name string, opt fmt.Println("Dialing....") - var dialer func(address string) net.Conn + var dialer func(address string) (net.Conn, error) args, argsErr := pt.ParsePT2ClientParameters(options) if argsErr != nil { @@ -212,7 +212,7 @@ func dialConn(tracker *ConnTracker, addr string, target string, name string, opt f := dialer fmt.Println("Dialing ", target) - remote := f(target) + remote, _ := f(target) // if err != nil { // fmt.Println("outgoing connection failed", err) // log.Errorf("(%s) - outgoing connection failed: %s", target, log.ElideError(err)) diff --git a/shapeshifter-dispatcher/shapeshifter-dispatcher.go b/shapeshifter-dispatcher/shapeshifter-dispatcher.go index 643df18..03923a6 100644 --- a/shapeshifter-dispatcher/shapeshifter-dispatcher.go +++ b/shapeshifter-dispatcher/shapeshifter-dispatcher.go @@ -261,10 +261,10 @@ func main() { } termMon.Wait(true) - - for { - // FIXME - block because termMon.Wait is not blocking - } + // FIXME - block because termMon.Wait is not blocking + //for { + // + //} } func checkIsClient(client bool, server bool) (bool, error) { -- cgit v1.2.3