update tap

listener/tap/conn.go

@@ -1,18 +1,20 @@
 package tap
 
 import (
+	"context"
 	"errors"
+	"io"
 	"net"
 	"time"
 
 	mdata "github.com/go-gost/core/metadata"
-	"github.com/songgao/water"
 )
 
 type conn struct {
-	ifce  *water.Interface
-	laddr net.Addr
-	raddr net.Addr
+	ifce   io.ReadWriteCloser
+	laddr  net.Addr
+	raddr  net.Addr
+	cancel context.CancelFunc
 }
 
 func (c *conn) Read(b []byte) (n int, err error) {
@@ -44,6 +46,9 @@ func (c *conn) SetWriteDeadline(t time.Time) error {
 }
 
 func (c *conn) Close() (err error) {
+	if c.cancel != nil {
+		c.cancel()
+	}
 	return c.ifce.Close()
 }