Function TestConn_BlockEndpoints

tailnet/conn_test.go:415–470

(t *testing.T)

}

func TestConn_BlockEndpoints(t *testing.T) {
	t.Parallel()
	logger := testutil.Logger(t)

	derpMap, _ := tailnettest.RunDERPAndSTUN(t)

	// Setup conn 1.
	ip1 := tailnet.TailscaleServicePrefix.RandomAddr()
	conn1, err := tailnet.NewConn(&tailnet.Options{
		Addresses:      []netip.Prefix{netip.PrefixFrom(ip1, 128)},
		Logger:         logger.Named("w1"),
		DERPMap:        derpMap,
		BlockEndpoints: true,
	})
	require.NoError(t, err)
	t.Cleanup(func() {
		err := conn1.Close()
		assert.NoError(t, err)
	})

	// Setup conn 2.
	ip2 := tailnet.TailscaleServicePrefix.RandomAddr()
	conn2, err := tailnet.NewConn(&tailnet.Options{
		Addresses:      []netip.Prefix{netip.PrefixFrom(ip2, 128)},
		Logger:         logger.Named("w2"),
		DERPMap:        derpMap,
		BlockEndpoints: true,
	})
	require.NoError(t, err)
	t.Cleanup(func() {
		err := conn2.Close()
		assert.NoError(t, err)
	})

	// Connect them together and wait for them to be reachable.
	stitch(t, conn2, conn1)
	stitch(t, conn1, conn2)
	awaitReachableCtx, awaitReachableCancel := context.WithTimeout(context.Background(), testutil.WaitShort)
	t.Cleanup(awaitReachableCancel)
	require.True(t, conn1.AwaitReachable(awaitReachableCtx, ip2))

	// Wait 10s for endpoints to potentially be sent over Disco. There's no way
	// to force Disco to send endpoints immediately.
	time.Sleep(10 * time.Second)

	// Double check that both peers don't have endpoints for the other peer
	// according to magicsock.
	conn1Status, ok := conn1.Status().Peer[conn2.Node().Key]
	require.True(t, ok)
	require.Empty(t, conn1Status.Addrs)
	require.Empty(t, conn1Status.CurAddr)
	conn2Status, ok := conn2.Status().Peer[conn1.Node().Key]
	require.True(t, ok)
	require.Empty(t, conn2Status.Addrs)
	require.Empty(t, conn2Status.CurAddr)
}

// stitch sends node updates from src Conn as peer updates to dst Conn.  Sort of