mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-04 23:45:34 +00:00
3c333f6341
This logs additional information about what mapping(s) are obtained during the creation process, including whether we return an existing cached mapping. Updates #10597 Signed-off-by: Andrew Dunham <andrew@du.nham.ca> Change-Id: I9ff25071f064c91691db9ab0b9365ccc5f948d6e
139 lines
3.4 KiB
Go
139 lines
3.4 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package portmapper
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"reflect"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"tailscale.com/control/controlknobs"
|
|
)
|
|
|
|
func TestCreateOrGetMapping(t *testing.T) {
|
|
if v, _ := strconv.ParseBool(os.Getenv("HIT_NETWORK")); !v {
|
|
t.Skip("skipping test without HIT_NETWORK=1")
|
|
}
|
|
c := NewClient(t.Logf, nil, nil, new(controlknobs.Knobs), nil)
|
|
defer c.Close()
|
|
c.SetLocalPort(1234)
|
|
for i := 0; i < 2; i++ {
|
|
if i > 0 {
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
ext, err := c.createOrGetMapping(context.Background())
|
|
t.Logf("Got: %v, %v", ext, err)
|
|
}
|
|
}
|
|
|
|
func TestClientProbe(t *testing.T) {
|
|
if v, _ := strconv.ParseBool(os.Getenv("HIT_NETWORK")); !v {
|
|
t.Skip("skipping test without HIT_NETWORK=1")
|
|
}
|
|
c := NewClient(t.Logf, nil, nil, new(controlknobs.Knobs), nil)
|
|
defer c.Close()
|
|
for i := 0; i < 3; i++ {
|
|
if i > 0 {
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
res, err := c.Probe(context.Background())
|
|
t.Logf("Got(t=%dms): %+v, %v", i*100, res, err)
|
|
}
|
|
}
|
|
|
|
func TestClientProbeThenMap(t *testing.T) {
|
|
if v, _ := strconv.ParseBool(os.Getenv("HIT_NETWORK")); !v {
|
|
t.Skip("skipping test without HIT_NETWORK=1")
|
|
}
|
|
c := NewClient(t.Logf, nil, nil, new(controlknobs.Knobs), nil)
|
|
defer c.Close()
|
|
c.debug.VerboseLogs = true
|
|
c.SetLocalPort(1234)
|
|
res, err := c.Probe(context.Background())
|
|
t.Logf("Probe: %+v, %v", res, err)
|
|
ext, err := c.createOrGetMapping(context.Background())
|
|
t.Logf("createOrGetMapping: %v, %v", ext, err)
|
|
}
|
|
|
|
func TestProbeIntegration(t *testing.T) {
|
|
igd, err := NewTestIGD(t.Logf, TestIGDOptions{PMP: true, PCP: true, UPnP: true})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer igd.Close()
|
|
|
|
c := newTestClient(t, igd)
|
|
t.Logf("Listening on pxp=%v, upnp=%v", c.testPxPPort, c.testUPnPPort)
|
|
defer c.Close()
|
|
|
|
res, err := c.Probe(context.Background())
|
|
if err != nil {
|
|
t.Fatalf("Probe: %v", err)
|
|
}
|
|
if !res.UPnP {
|
|
t.Errorf("didn't detect UPnP")
|
|
}
|
|
st := igd.stats()
|
|
want := igdCounters{
|
|
numUPnPDiscoRecv: 1,
|
|
numPMPRecv: 1,
|
|
numPCPRecv: 1,
|
|
numPCPDiscoRecv: 1,
|
|
numPMPPublicAddrRecv: 1,
|
|
}
|
|
if !reflect.DeepEqual(st, want) {
|
|
t.Errorf("unexpected stats:\n got: %+v\nwant: %+v", st, want)
|
|
}
|
|
|
|
t.Logf("Probe: %+v", res)
|
|
t.Logf("IGD stats: %+v", st)
|
|
// TODO(bradfitz): finish
|
|
}
|
|
|
|
func TestPCPIntegration(t *testing.T) {
|
|
igd, err := NewTestIGD(t.Logf, TestIGDOptions{PMP: false, PCP: true, UPnP: false})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer igd.Close()
|
|
|
|
c := newTestClient(t, igd)
|
|
defer c.Close()
|
|
res, err := c.Probe(context.Background())
|
|
if err != nil {
|
|
t.Fatalf("probe failed: %v", err)
|
|
}
|
|
if res.UPnP || res.PMP {
|
|
t.Errorf("probe unexpectedly saw upnp or pmp: %+v", res)
|
|
}
|
|
if !res.PCP {
|
|
t.Fatalf("probe did not see pcp: %+v", res)
|
|
}
|
|
|
|
external, err := c.createOrGetMapping(context.Background())
|
|
if err != nil {
|
|
t.Fatalf("failed to get mapping: %v", err)
|
|
}
|
|
if !external.IsValid() {
|
|
t.Errorf("got zero IP, expected non-zero")
|
|
}
|
|
if c.mapping == nil {
|
|
t.Errorf("got nil mapping after successful createOrGetMapping")
|
|
}
|
|
}
|
|
|
|
// Test to ensure that metric names generated by this function do not contain
|
|
// invalid characters.
|
|
//
|
|
// See https://github.com/tailscale/tailscale/issues/9551
|
|
func TestGetUPnPErrorsMetric(t *testing.T) {
|
|
// This will panic if the metric name is invalid.
|
|
getUPnPErrorsMetric(100)
|
|
getUPnPErrorsMetric(0)
|
|
getUPnPErrorsMetric(-100)
|
|
}
|