mirror of https://github.com/coder/coder.git
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
//go:build linux || (windows && amd64)
|
|
|
|
package agent
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/cakturk/go-netstat/netstat"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/codersdk/workspacesdk"
|
|
)
|
|
|
|
func (lp *listeningPortsHandler) getListeningPorts() ([]codersdk.WorkspaceAgentListeningPort, error) {
|
|
lp.mut.Lock()
|
|
defer lp.mut.Unlock()
|
|
|
|
if time.Since(lp.mtime) < lp.cacheDuration {
|
|
// copy
|
|
ports := make([]codersdk.WorkspaceAgentListeningPort, len(lp.ports))
|
|
copy(ports, lp.ports)
|
|
return ports, nil
|
|
}
|
|
|
|
tabs, err := netstat.TCPSocks(func(s *netstat.SockTabEntry) bool {
|
|
return s.State == netstat.Listen
|
|
})
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("scan listening ports: %w", err)
|
|
}
|
|
|
|
seen := make(map[uint16]struct{}, len(tabs))
|
|
ports := []codersdk.WorkspaceAgentListeningPort{}
|
|
for _, tab := range tabs {
|
|
if tab.LocalAddr == nil || tab.LocalAddr.Port < workspacesdk.AgentMinimumListeningPort {
|
|
continue
|
|
}
|
|
|
|
// Ignore ports that we've been told to ignore.
|
|
if _, ok := lp.ignorePorts[int(tab.LocalAddr.Port)]; ok {
|
|
continue
|
|
}
|
|
|
|
// Don't include ports that we've already seen. This can happen on
|
|
// Windows, and maybe on Linux if you're using a shared listener socket.
|
|
if _, ok := seen[tab.LocalAddr.Port]; ok {
|
|
continue
|
|
}
|
|
seen[tab.LocalAddr.Port] = struct{}{}
|
|
|
|
procName := ""
|
|
if tab.Process != nil {
|
|
procName = tab.Process.Name
|
|
}
|
|
ports = append(ports, codersdk.WorkspaceAgentListeningPort{
|
|
ProcessName: procName,
|
|
Network: "tcp",
|
|
Port: tab.LocalAddr.Port,
|
|
})
|
|
}
|
|
|
|
lp.ports = ports
|
|
lp.mtime = time.Now()
|
|
|
|
// copy
|
|
ports = make([]codersdk.WorkspaceAgentListeningPort, len(lp.ports))
|
|
copy(ports, lp.ports)
|
|
return ports, nil
|
|
}
|