mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-29 04:55:31 +00:00
e32e5c0d0c
This PR parameterizes receiving loopback updates from the portlist package. Callers can now include services bound to localhost if they want. Note that this option is off by default still. Fixes #8171 Signed-off-by: Marwan Sulaiman <marwan@tailscale.com>
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
//go:build darwin && !ios
|
|
|
|
package portlist
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"go4.org/mem"
|
|
)
|
|
|
|
func TestParsePort(t *testing.T) {
|
|
type InOut struct {
|
|
in string
|
|
expect int
|
|
}
|
|
tests := []InOut{
|
|
{"1.2.3.4:5678", 5678},
|
|
{"0.0.0.0.999", 999},
|
|
{"1.2.3.4:*", 0},
|
|
{"5.5.5.5:0", 0},
|
|
{"[1::2]:5", 5},
|
|
{"[1::2].5", 5},
|
|
{"gibberish", -1},
|
|
}
|
|
|
|
for _, io := range tests {
|
|
got := parsePort(mem.S(io.in))
|
|
if got != io.expect {
|
|
t.Fatalf("input:%#v expect:%v got:%v\n", io.in, io.expect, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
const netstatOutput = `
|
|
// macOS
|
|
tcp4 0 0 *.23 *.* LISTEN
|
|
tcp6 0 0 *.24 *.* LISTEN
|
|
tcp4 0 0 *.8185 *.* LISTEN
|
|
tcp4 0 0 127.0.0.1.8186 *.* LISTEN
|
|
tcp6 0 0 ::1.8187 *.* LISTEN
|
|
tcp4 0 0 127.1.2.3.8188 *.* LISTEN
|
|
|
|
udp6 0 0 *.106 *.*
|
|
udp4 0 0 *.104 *.*
|
|
udp46 0 0 *.146 *.*
|
|
`
|
|
|
|
func TestParsePortsNetstat(t *testing.T) {
|
|
for _, loopBack := range [...]bool{false, true} {
|
|
t.Run(fmt.Sprintf("loopback_%v", loopBack), func(t *testing.T) {
|
|
want := List{
|
|
{"tcp", 23, "", 0},
|
|
{"tcp", 24, "", 0},
|
|
{"udp", 104, "", 0},
|
|
{"udp", 106, "", 0},
|
|
{"udp", 146, "", 0},
|
|
{"tcp", 8185, "", 0}, // but not 8186, 8187, 8188 on localhost, when loopback is false
|
|
}
|
|
if loopBack {
|
|
want = append(want,
|
|
Port{"tcp", 8186, "", 0},
|
|
Port{"tcp", 8187, "", 0},
|
|
Port{"tcp", 8188, "", 0},
|
|
)
|
|
}
|
|
pl, err := appendParsePortsNetstat(nil, bufio.NewReader(strings.NewReader(netstatOutput)), loopBack)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
pl = sortAndDedup(pl)
|
|
jgot, _ := json.MarshalIndent(pl, "", "\t")
|
|
jwant, _ := json.MarshalIndent(want, "", "\t")
|
|
if len(pl) != len(want) {
|
|
t.Fatalf("Got:\n%s\n\nWant:\n%s\n", jgot, jwant)
|
|
}
|
|
for i := range pl {
|
|
if pl[i] != want[i] {
|
|
t.Errorf("row#%d\n got: %+v\n\nwant: %+v\n",
|
|
i, pl[i], want[i])
|
|
t.Fatalf("Got:\n%s\n\nWant:\n%s\n", jgot, jwant)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|