update func calls

ginuerzh · Feb 9, 2020 · 8121e20 · 8121e20
1 parent ece7994
commit 8121e20
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/gost.go b/gost.go
@@ -20,7 +20,7 @@ import (
 )
 
 // Version is the gost version.
-const Version = "2.10.1"
+const Version = "2.10.1-dev"
 
 // Debug is a flag that enables the debug log.
 var Debug bool

diff --git a/redirect.go b/redirect.go
@@ -53,7 +53,8 @@ func (h *tcpRedirectHandler) Handle(c net.Conn) {
 
 	log.Logf("[red-tcp] %s -> %s", srcAddr, dstAddr)
 
-	cc, err := h.options.Chain.Dial(dstAddr.String(),
+	cc, err := h.options.Chain.DialContext(context.Background(),
+		"tcp", dstAddr.String(),
 		RetryChainOption(h.options.Retries),
 		TimeoutChainOption(h.options.Timeout),
 	)

diff --git a/udp.go b/udp.go
@@ -345,7 +345,7 @@ func (c *udpClientConn) WriteTo(b []byte, addr net.Addr) (int, error) {
 }
 
 func (c *udpClientConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
-	n, err = c.Read(b)
+	n, err = c.UDPConn.Read(b)
 	addr = c.RemoteAddr()
 	return
 }