/
remote_forwarded_ports.go
56 lines (42 loc) · 1.15 KB
/
remote_forwarded_ports.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.
package tunnels
import "sync"
type remoteForwardedPorts struct {
portsMu sync.RWMutex
ports map[uint16]bool
notify chan remoteForwardedPortNotification
}
type remoteForwardedPortNotification struct {
port uint16
notificationType remoteForwardedPortNotificationType
}
type remoteForwardedPortNotificationType int
const (
remoteForwardedPortNotificationTypeAdd remoteForwardedPortNotificationType = iota
remoteForwardedPortNotificationTypeRemove
)
func newRemoteForwardedPorts() *remoteForwardedPorts {
return &remoteForwardedPorts{
ports: make(map[uint16]bool),
notify: make(chan remoteForwardedPortNotification),
}
}
func (r *remoteForwardedPorts) Add(port uint16) {
r.portsMu.Lock()
defer r.portsMu.Unlock()
r.ports[port] = true
notification := remoteForwardedPortNotification{
port: port,
notificationType: remoteForwardedPortNotificationTypeAdd,
}
select {
case r.notify <- notification:
default:
}
}
func (r *remoteForwardedPorts) hasPort(port uint16) bool {
r.portsMu.RLock()
defer r.portsMu.RUnlock()
return r.ports[port]
}