// Copyright (c) 2020 Tailscale Inc & AUTHORS All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. //go:build darwin && !ios package portlist import ( "bufio" "encoding/json" "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) { want := List{ Port{"tcp", 23, ""}, Port{"tcp", 24, ""}, Port{"udp", 104, ""}, Port{"udp", 106, ""}, Port{"udp", 146, ""}, Port{"tcp", 8185, ""}, // but not 8186, 8187, 8188 on localhost } pl, err := appendParsePortsNetstat(nil, bufio.NewReader(strings.NewReader(netstatOutput))) 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) } } }