protocol/reuse.go: Redo UPnP before Dialing

hzyitc · May 25, 2021 · f9b3b38 · f9b3b38
1 parent 00be6f7
commit f9b3b38
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/TCPMode/reuse.go b/TCPMode/reuse.go
@@ -44,6 +44,8 @@ func NewReuse(rpfc routerPortForward.Config, port int) (Interface, error) {
 }
 
 func (s *reuse) Dial(addr string) (net.Conn, error) {
+	s.rpf.Redo()
+
 	if s.conn != nil {
 		return nil, errors.New("double dial")
 	}

diff --git a/routerPortForward/upnp.go b/routerPortForward/upnp.go
@@ -7,6 +7,7 @@ type Config struct {
 }
 
 type Interface interface {
+	Redo()
 	Close()
 }
 
@@ -44,6 +45,12 @@ func New(config Config, port int) (Interface, error) {
 	}, nil
 }
 
+func (s *upnpImpl) Redo() {
+	if s.config.Enable {
+		s.d.Forward(s.port, "mnh")
+	}
+}
+
 func (s *upnpImpl) Close() {
 	if s.config.Enable {
 		s.d.Clear(s.port)