ipn/proxies: pause and resume fns

ignoramous · ignoramous · commit 928668fe1b12 · 2025-08-10T02:44:41.000+05:30
diff --git a/intra/backend/ipn_proxies.go b/intra/backend/ipn_proxies.go
@@ -69,6 +69,8 @@ const ( // see ipn/proxies.go
 
 	// status of proxies
 
+	// proxy paused until resumed; will not dial
+	TPU = 3
 	// proxy UP but not responding
 	TNT = 2
 	// proxy idle
@@ -140,6 +142,10 @@ type Proxy interface {
 	Status() int
 	// Ping pings this proxy.
 	Ping() bool
+	// Pause pauses this proxy.
+	Pause() bool
+	// Resume resumes this proxy.
+	Resume() bool
 	// Stop stops this proxy.
 	Stop() error
 	// Refresh re-registers this proxy, if necessary.
diff --git a/intra/ipn/auto.go b/intra/ipn/auto.go
@@ -29,6 +29,7 @@ type auto struct {
 	NoDNS
 	ProtoAgnostic
 	SkipRefresh
+	CantPause
 	GW
 	pxr  ProxyProvider
 	addr string
diff --git a/intra/ipn/base.go b/intra/ipn/base.go
@@ -25,6 +25,7 @@ type base struct {
 	NoDNS
 	ProtoAgnostic
 	SkipRefresh
+	CantPause
 	GW
 	id       string
 	addr     string
diff --git a/intra/ipn/exit.go b/intra/ipn/exit.go
@@ -31,6 +31,7 @@ type exit struct {
 	NoDNS
 	ProtoAgnostic
 	SkipRefresh
+	CantPause
 	GWNoVia
 	id       string
 	addr     string
diff --git a/intra/ipn/exit64.go b/intra/ipn/exit64.go
@@ -210,6 +210,32 @@ func (h *exit64) Status() int {
 	return h.status.Load()
 }
 
+// Since implements x.Proxy.
+func (h *exit64) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("proxy: exit64: pause called when stopped")
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("proxy: exit64: paused? %t", ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *exit64) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("proxy: exit64: resume called when not paused; status %d", st)
+		return false
+	}
+
+	ok := h.status.Cas(st, TUP)
+	log.I("proxy: exit64: resumed? %t", ok)
+	return ok
+}
+
 // Stop implements Proxy.
 func (h *exit64) Stop() error {
 	h.status.Store(END)
diff --git a/intra/ipn/ground.go b/intra/ipn/ground.go
@@ -18,6 +18,7 @@ type ground struct {
 	ProtoAgnostic
 	SkipRefresh
 	GWNoVia
+	CantPause
 	addr string
 }
 
diff --git a/intra/ipn/http1.go b/intra/ipn/http1.go
@@ -209,6 +209,32 @@ func (h *http1) Status() int {
 	return s
 }
 
+// Pause implements x.Proxy.
+func (h *http1) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("proxy: http1: pause called when stopped")
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("proxy: http1: paused? %t", ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *http1) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("proxy: http1: resume called when not paused; status %d", st)
+		return false
+	}
+
+	ok := h.status.Cas(st, TUP)
+	log.I("proxy: http1: resumed? %t", ok)
+	return ok
+}
+
 // Stop implements Proxy.
 func (h *http1) Stop() error {
 	h.status.Store(END)
diff --git a/intra/ipn/nop.go b/intra/ipn/nop.go
@@ -103,6 +103,14 @@ func (SkipRefresh) onNotOK() (didRefresh bool, allOK bool) { return false, true
 // Ping implements Proxy.
 func (SkipRefresh) Ping() bool { return false }
 
+type CantPause struct{}
+
+// Pause implements Proxy.
+func (CantPause) Pause() bool { return false }
+
+// Resume implements Proxy.
+func (CantPause) Resume() bool { return false }
+
 // NoFwd is a proxy that does not support listening or forwarding.
 type NoFwd struct{}
 
@@ -139,6 +147,7 @@ type NoProxy struct {
 	ProtoAgnostic
 	SkipRefresh
 	NoFwd
+	CantPause
 	GWNoVia
 }
 
diff --git a/intra/ipn/piph2.go b/intra/ipn/piph2.go
@@ -338,6 +338,32 @@ func (t *piph2) Status() int {
 	return st
 }
 
+// Since implements x.Proxy.
+func (h *piph2) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("proxy: piph2: pause called when stopped")
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("proxy: piph2: paused? %t", ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *piph2) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("proxy: piph2: resume called when not paused; status %d", st)
+		return false
+	}
+
+	ok := h.status.Cas(st, TUP)
+	log.I("proxy: piph2: resumed? %t", ok)
+	return ok
+}
+
 // Scenario 4: privacypass.github.io/protocol
 func (t *piph2) claim(msg string) []string {
 	if len(t.token) == 0 || len(t.toksig) == 0 {
diff --git a/intra/ipn/pipws.go b/intra/ipn/pipws.go
@@ -336,6 +336,32 @@ func (t *pipws) Status() int {
 	return s
 }
 
+// Pause implements x.Proxy.
+func (h *pipws) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("proxy: pipws: pause called when stopped")
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("proxy: pipws: paused? %t", ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *pipws) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("proxy: pipws: resume called when not paused; status %d", st)
+		return false
+	}
+
+	ok := h.status.Cas(st, TUP)
+	log.I("proxy: pipws: resumed? %t", ok)
+	return ok
+}
+
 // Scenario 4: privacypass.github.io/protocol
 func (t *pipws) claim(msg string) []string {
 	if len(t.token) == 0 || len(t.toksig) == 0 {
diff --git a/intra/ipn/proxies.go b/intra/ipn/proxies.go
@@ -55,6 +55,7 @@ const (
 	INTERNET = x.INTERNET
 	RPN      = x.RPN
 
+	TPU = x.TPU
 	TNT = x.TNT
 	TZZ = x.TZZ
 	TUP = x.TUP
@@ -102,6 +103,7 @@ var (
 	errNoProxyConn        = errors.New("proxy: not a tcp/udp conn")
 	errNotUDPConn         = errors.New("proxy: not a udp conn")
 	errProxyStopped       = errors.New("proxy: stopped")
+	errProxyPaused        = errors.New("proxy: paused")
 	errProxyConfig        = errors.New("proxy: invalid config")
 	errProxyReadd         = errors.New("proxy: cannot update; readd config")
 	errNoProxyResponse    = errors.New("proxy: no response from upstream")
diff --git a/intra/ipn/seproxy.go b/intra/ipn/seproxy.go
@@ -460,6 +460,32 @@ func (h *seproxy) Status() int {
 	return h.status.Load()
 }
 
+// Pause implements x.Proxy.
+func (h *seproxy) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("proxy: se: pause called when stopped")
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("proxy: se: paused? %t", ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *seproxy) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("proxy: se: resume called when not paused; status %d", st)
+		return false
+	}
+
+	ok := h.status.Cas(st, TUP)
+	log.I("proxy: se: resumed? %t", ok)
+	return ok
+}
+
 // Stop implements Proxy.
 func (h *seproxy) Stop() error {
 	h.status.Store(END)
diff --git a/intra/ipn/socks5.go b/intra/ipn/socks5.go
@@ -321,6 +321,32 @@ func (h *socks5) Status() int {
 	return s
 }
 
+// Pause implements x.Proxy.
+func (h *socks5) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("proxy: socks5: pause called when stopped")
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("proxy: socks5: paused? %t", ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *socks5) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("proxy: socks5: resume called when not paused; status %d", st)
+		return false
+	}
+
+	ok := h.status.Cas(st, TUP)
+	log.I("proxy: socks5: resumed? %t", ok)
+	return ok
+}
+
 // Stop implements Proxy.
 func (h *socks5) Stop() error {
 	h.status.Store(END)
diff --git a/intra/ipn/wgproxy.go b/intra/ipn/wgproxy.go
@@ -77,6 +77,8 @@ const (
 
 	removeViaOnErrors = false
 	resetDeviceOnTNT  = false
+	// reset device if it is in TUP state (resuming...)
+	resetDeviceOnTUP = false
 
 	FAST = x.WGFAST
 
@@ -321,7 +323,8 @@ func (w *wgproxy) Refresh() (err error) {
 		return errProxyStopped
 	}
 
-	resetDevice := resetDeviceOnTNT && status == TNT
+	resetDevice := (resetDeviceOnTNT && status == TNT) ||
+		(resetDeviceOnTUP && status == TUP)
 
 	w.latestPing.Store(0) // reset latest ping time
 
@@ -340,6 +343,8 @@ func (w *wgproxy) Refresh() (err error) {
 
 	if err = w.resetMtu(via); err == nil {
 		err = w.Device.Down()
+
+		// for now, never reset since resetDeviceOnTNT is false
 		resetDevice = resetDevice && w.wgtun.ignoreTUNClose.CompareAndSwap(false, true)
 		if resetDevice && err == nil {
 			var newdev *device.Device
@@ -1291,6 +1296,33 @@ func (h *wgtun) Status() int {
 	return h.status.Load()
 }
 
+// Pause implements x.Proxy.
+func (h *wgproxy) Pause() bool {
+	st := h.status.Load()
+	if st == END {
+		log.W("wg: %s pause called when stopped", h.tag())
+		return false
+	}
+
+	ok := h.status.Cas(st, TPU)
+	log.I("wg: %s paused? %t", h.tag(), ok)
+	return ok
+}
+
+// Resume implements x.Proxy.
+func (h *wgproxy) Resume() bool {
+	st := h.status.Load()
+	if st != TPU {
+		log.W("wg: %s resume called when not paused; status %d", h.tag(), st)
+		return false
+	}
+
+	go h.Refresh() // reconnect
+	ok := h.status.Cas(st, TUP)
+	log.I("wg: %s resumed? %t", h.tag(), ok)
+	return ok
+}
+
 // DNS implements x.Proxy.
 func (h *wgtun) DNS() *x.Gostr {
 	return x.StrOf(h.dnsResolvers())

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,7 @@ type ground struct {`
`18`	`18`	`ProtoAgnostic`
`19`	`19`	`SkipRefresh`
`20`	`20`	`GWNoVia`
	`21`	`+ CantPause`
`21`	`22`	`addr string`
`22`	`23`	`}`
`23`	`24`