@ -5,23 +5,23 @@ package nat
import (
"context"
"e ncoding/json "
"e rrors "
"fmt"
"io"
"net"
"net/http"
"net/netip"
"net/url"
"os"
"os/exec"
"path/filepath"
"strings"
"sync"
"testing"
"time"
"golang.org/x/sync/errgroup"
"tailscale.com/client/tailscale"
"tailscale.com/ipn/ipnstate"
"tailscale.com/tailcfg"
"tailscale.com/tstest/natlab/vnet"
)
@ -36,7 +36,7 @@ func newNatTest(tb testing.TB) *natTest {
nt := & natTest {
tb : tb ,
tempDir : tb . TempDir ( ) ,
base : "/Users/ bradfitz/src /tailscale.com/gokrazy/tsapp.qcow2",
base : "/Users/ maisem/dev /tailscale.com/gokrazy/tsapp.qcow2",
}
if _ , err := os . Stat ( nt . base ) ; err != nil {
@ -116,7 +116,7 @@ func (nt *natTest) runTest(node1, node2 addNodeFunc) {
"-M" , "microvm,isa-serial=off" ,
"-m" , "1G" ,
"-nodefaults" , "-no-user-config" , "-nographic" ,
"-kernel" , "/Users/ bradfitz/src /github.com/tailscale/gokrazy-kernel/vmlinuz",
"-kernel" , "/Users/ maisem/dev /github.com/tailscale/gokrazy-kernel/vmlinuz",
"-append" , "console=hvc0 root=PARTUUID=60c24cc1-f3f9-427a-8199-dd02023b0001/PARTNROFF=1 ro init=/gokrazy/init panic=10 oops=panic pci=off nousb tsc=unstable clocksource=hpet tailscale-tta=1" ,
"-drive" , "id=blk0,file=" + disk + ",format=qcow2" ,
"-device" , "virtio-blk-device,drive=blk0" ,
@ -142,15 +142,16 @@ func (nt *natTest) runTest(node1, node2 addNodeFunc) {
ctx , cancel := context . WithTimeout ( context . Background ( ) , 60 * time . Second )
defer cancel ( )
c1 := & http . Client { Transport : nt . vnet . NodeAgentRoundTripper ( nodes [ 0 ] ) }
c2 := & http . Client { Transport : nt . vnet . NodeAgentRoundTripper ( nodes [ 1 ] ) }
lc1 := nt . vnet . NodeAgentClient ( nodes [ 0 ] )
lc2 := nt . vnet . NodeAgentClient ( nodes [ 1 ] )
clients := [ ] * vnet . NodeAgentClient { lc1 , lc2 }
var eg errgroup . Group
var sts [ 2 ] * ipnstate . Status
for i , c := range [ ] * http . Client { c1 , c2 } {
for i , c := range clients {
i , c := i , c
eg . Go ( func ( ) error {
st , err := status ( ctx , c )
st , err := c . Status ( ctx )
if err != nil {
return fmt . Errorf ( "node%d status: %w" , i , err )
}
@ -159,7 +160,7 @@ func (nt *natTest) runTest(node1, node2 addNodeFunc) {
return fmt . Errorf ( "node%d up: %w" , i , err )
}
t . Logf ( "node%d up!" , i )
st , err = status ( ctx , c )
st , err = c . Status ( ctx )
if err != nil {
return fmt . Errorf ( "node%d status: %w" , i , err )
}
@ -176,85 +177,40 @@ func (nt *natTest) runTest(node1, node2 addNodeFunc) {
t . Fatalf ( "initial setup: %v" , err )
}
route , err := ping ( ctx , c1, sts [ 1 ] . Self . TailscaleIPs [ 0 ] . String ( ) )
route , err := ping ( ctx , l c1, sts [ 1 ] . Self . TailscaleIPs [ 0 ] )
t . Logf ( "ping route: %v, %v" , route , err )
}
func status ( ctx context . Context , c * http . Client ) ( * ipnstate . Status , error ) {
req , err := http . NewRequestWithContext ( ctx , "GET" , "http://unused/status" , nil )
if err != nil {
return nil , err
}
res , err := c . Do ( req )
if err != nil {
return nil , err
}
defer res . Body . Close ( )
all , err := io . ReadAll ( res . Body )
if err != nil {
return nil , fmt . Errorf ( "ReadAll: %w" , err )
}
var st ipnstate . Status
if err := json . Unmarshal ( all , & st ) ; err != nil {
return nil , fmt . Errorf ( "JSON marshal error: %v; body was %q" , err , all )
}
return & st , nil
}
type routeType string
const (
routeDirect routeType = "direct"
routeDERP routeType = "derp"
routeLAN routeType = "lan"
)
func ping ( ctx context . Context , c * http . Client , target string ) ( routeType , error ) {
req , err := http . NewRequestWithContext ( ctx , "POST" , "http://unused/ping?target=" + url . QueryEscape ( target ) , nil )
if err != nil {
return "" , err
}
res , err := c . Do ( req )
if err != nil {
return "" , err
}
defer res . Body . Close ( )
if res . StatusCode != 200 {
return "" , fmt . Errorf ( "unexpected status code %v" , res . Status )
}
all , _ := io . ReadAll ( res . Body )
var route routeType
for _ , line := range strings . Split ( string ( all ) , "\n" ) {
if strings . Contains ( line , " via DERP" ) {
route = routeDERP
continue
}
// pong from foo (100.82.3.4) via ADDR:PORT in 69ms
if _ , rest , ok := strings . Cut ( line , " via " ) ; ok {
ipPorStr , _ , _ := strings . Cut ( rest , " in " )
ipPort , err := netip . ParseAddrPort ( ipPorStr )
if err == nil {
if ipPort . Addr ( ) . IsPrivate ( ) {
route = routeLAN
} else {
route = routeDirect
}
continue
func ping ( ctx context . Context , c * vnet . NodeAgentClient , target netip . Addr ) ( * ipnstate . PingResult , error ) {
n := 0
var res * ipnstate . PingResult
anyPong := false
for {
n ++
pr , err := c . PingWithOpts ( ctx , target , tailcfg . PingDisco , tailscale . PingOpts { } )
if err != nil {
if anyPong {
return res , nil
}
return nil , err
}
if pr . Err != "" {
return nil , errors . New ( pr . Err )
}
if pr . DERPRegionID == 0 {
return pr , nil
}
time . Sleep ( time . Second )
res = pr
}
if route == "" {
return routeType ( all ) , nil
}
return route , nil
}
func up ( ctx context . Context , c * http . Client ) error {
func up ( ctx context . Context , c * vnet . NodeAgentClient ) error {
req , err := http . NewRequestWithContext ( ctx , "GET" , "http://unused/up" , nil )
if err != nil {
return err
}
res , err := c . Do ( req )
res , err := c . HTTPClient . Do ( req )
if err != nil {
return err
}
@ -272,6 +228,7 @@ func TestEasyEasy(t *testing.T) {
}
func TestEasyHard ( t * testing . T ) {
t . Skip ( )
nt := newNatTest ( t )
nt . runTest ( easy , hard )
}