fosrl/newt
1
0
Fork 0
mirror of https://github.com/fosrl/newt.git synced 2025-05-13 05:30:39 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix remove target error

Browse source
This commit is contained in:
Owen Schwartz 2024-11-23 16:52:31 -05:00
parent 94a94613eb
commit 4df89c208b
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
2 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
main.go
View file

@ -348,11 +348,10 @@ func updateTargets(pm *proxy.ProxyManager, action string, tunnelIP string, proto
if action == "add" { if action == "add" {
target := parts[1] + ":" + parts[2] target := parts[1] + ":" + parts[2]
pm.RemoveTarget(proto, tunnelIP, port) // remove it first incase this is an update. we are kind of using the internal port as the "targetId" in the proxy pm.RemoveTarget(proto, tunnelIP, port) // remove it first in case this is an update. we are kind of using the internal port as the "targetId" in the proxy
pm.AddTarget(proto, tunnelIP, port, target) pm.AddTarget(proto, tunnelIP, port, target)
// log the target
log.Printf("Added target: %s:%d -> %s", tunnelIP, port, target)
} else if action == "remove" { } else if action == "remove" {
log.Printf("Removing target with port %d", port)
pm.RemoveTarget(proto, tunnelIP, port) pm.RemoveTarget(proto, tunnelIP, port)
} }
} }

2
proxy/manager.go
View file

@ -46,7 +46,7 @@ func (pm *ProxyManager) RemoveTarget(protocol, listen string, port int) error {
target.Port == port && target.Port == port &&
strings.ToLower(target.Protocol) == protocol { strings.ToLower(target.Protocol) == protocol {
// Signal the serving goroutine to stop // Signal the serving goroutine to stop
close(target.cancel) // close(target.cancel)
// Close the appropriate listener/connection based on protocol // Close the appropriate listener/connection based on protocol
target.Lock() target.Lock()