Skip to content

chore: replace wsconncache with a single tailnet #8176

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 26 commits into from
Jul 12, 2023
Merged

chore: replace wsconncache with a single tailnet #8176

Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
7222135
chore: replace wsconncache with a single tailnet
coadler Jun 26, 2023
d4181b0
remove duplicate AwaitReachable
coadler Jun 26, 2023
2a133d1
properly release net.Conn
coadler Jun 26, 2023
aa6bcb6
tailnetTransport
coadler Jun 26, 2023
f9040fc
link to issue in deprecation notice
coadler Jun 26, 2023
a988fef
address review comments
coadler Jun 28, 2023
2e201dc
fixup! address review comments
coadler Jun 28, 2023
5901f68
Merge branch 'main' into colin/rm-wsconncache2
coadler Jun 28, 2023
311ea2b
second round
coadler Jun 28, 2023
30aefcb
fixup! second round
coadler Jun 28, 2023
e55e146
support swapping coordinators
coadler Jun 29, 2023
457470d
use multiagents for all clients
coadler Jun 29, 2023
2c32434
Merge branch 'main' into colin/rm-wsconncache2
coadler Jun 29, 2023
11f2805
fixups
coadler Jun 30, 2023
a88be66
fixup! fixups
coadler Jun 30, 2023
49a8364
Merge branch 'main' into colin/rm-wsconncache2
coadler Jun 30, 2023
5e4b631
fixes
coadler Jun 30, 2023
8ef2fb7
Merge branch 'main' into colin/rm-wsconncache2
coadler Jun 30, 2023
9864d89
Merge branch 'main' into colin/rm-wsconncache2
coadler Jul 7, 2023
dd3cc15
comment
coadler Jul 7, 2023
8896ae4
fixup! comment
coadler Jul 7, 2023
be4db71
fixup! comment
coadler Jul 7, 2023
7bfac9b
add feature flag
coadler Jul 11, 2023
10d44fe
Merge branch 'main' into colin/rm-wsconncache2
coadler Jul 11, 2023
ef44092
fixes
coadler Jul 12, 2023
9a29d85
Merge branch 'main' into colin/rm-wsconncache2
coadler Jul 12, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 25 additions & 3 deletions agent/agent.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ type Options struct {
SSHMaxTimeout time.Duration
TailnetListenPort uint16
Subsystem codersdk.AgentSubsystem
Addresses []netip.Prefix

PrometheusRegistry *prometheus.Registry
}
Expand Down Expand Up @@ -131,6 +132,7 @@ func New(options Options) Agent {
connStatsChan: make(chan *agentsdk.Stats, 1),
sshMaxTimeout: options.SSHMaxTimeout,
subsystem: options.Subsystem,
addresses: options.Addresses,

prometheusRegistry: prometheusRegistry,
metrics: newAgentMetrics(prometheusRegistry),
Expand Down Expand Up @@ -174,6 +176,7 @@ type agent struct {
lifecycleStates []agentsdk.PostLifecycleRequest

network *tailnet.Conn
addresses []netip.Prefix
connStatsChan chan *agentsdk.Stats
latestStat atomic.Pointer[agentsdk.Stats]

Expand Down Expand Up @@ -592,7 +595,7 @@ func (a *agent) run(ctx context.Context) error {
network := a.network
a.closeMutex.Unlock()
if network == nil {
network, err = a.createTailnet(ctx, manifest.DERPMap, manifest.DisableDirectConnections)
network, err = a.createTailnet(ctx, manifest.AgentID, manifest.DERPMap, manifest.DisableDirectConnections)
if err != nil {
return xerrors.Errorf("create tailnet: %w", err)
}
Expand All @@ -610,6 +613,11 @@ func (a *agent) run(ctx context.Context) error {

a.startReportingConnectionStats(ctx)
} else {
// Update the wireguard IPs if the agent ID changed.
err := network.SetAddresses(a.wireguardAddresses(manifest.AgentID))
if err != nil {
a.logger.Error(ctx, "update tailnet addresses", slog.Error(err))
}
// Update the DERP map and allow/disallow direct connections.
network.SetDERPMap(manifest.DERPMap)
network.SetBlockEndpoints(manifest.DisableDirectConnections)
Expand All @@ -623,6 +631,20 @@ func (a *agent) run(ctx context.Context) error {
return nil
}

func (a *agent) wireguardAddresses(agentID uuid.UUID) []netip.Prefix {
if len(a.addresses) == 0 {
return []netip.Prefix{
// This is the IP that should be used primarily.
netip.PrefixFrom(tailnet.IPFromUUID(agentID), 128),
// We also listen on the legacy codersdk.WorkspaceAgentIP. This
// allows for a transition away from wsconncache.
netip.PrefixFrom(codersdk.WorkspaceAgentIP, 128),
}
}

return a.addresses
}

func (a *agent) trackConnGoroutine(fn func()) error {
a.closeMutex.Lock()
defer a.closeMutex.Unlock()
Expand All @@ -637,9 +659,9 @@ func (a *agent) trackConnGoroutine(fn func()) error {
return nil
}

func (a *agent) createTailnet(ctx context.Context, derpMap *tailcfg.DERPMap, disableDirectConnections bool) (_ *tailnet.Conn, err error) {
func (a *agent) createTailnet(ctx context.Context, agentID uuid.UUID, derpMap *tailcfg.DERPMap, disableDirectConnections bool) (_ *tailnet.Conn, err error) {
network, err := tailnet.NewConn(&tailnet.Options{
Addresses: []netip.Prefix{netip.PrefixFrom(codersdk.WorkspaceAgentIP, 128)},
Addresses: a.wireguardAddresses(agentID),
DERPMap: derpMap,
Logger: a.logger.Named("tailnet"),
ListenPort: a.tailnetListenPort,
Expand Down
Loading