feat: 新增 net pids 端口占用查询并补齐 README
- kopsc net pids <端口> 等价 lsof -ti:<端口>，输出 PID 升序去重、每行一个 - Linux 回退链 lsof -> ss -> /proc；macOS 使用 lsof；Windows 解析 netstat -ano - 端口 1-65535 校验，无占用时无输出并正常退出 - README 新增 net.pids 小节，修正过时的匹配示例、-n 例外说明与目录结构
This commit is contained in:
@@ -0,0 +1,188 @@
|
||||
package executor
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestParsePortArgs(t *testing.T) {
|
||||
cases := []struct {
|
||||
name string
|
||||
args []string
|
||||
want int
|
||||
wantErr string
|
||||
}{
|
||||
{"positional", []string{"8080"}, 8080, ""},
|
||||
{"port flag", []string{"--port", "443"}, 443, ""},
|
||||
{"port flag equals", []string{"--port=443"}, 443, ""},
|
||||
{"missing", nil, 0, "必须提供端口"},
|
||||
{"zero", []string{"0"}, 0, "1-65535"},
|
||||
{"too large", []string{"65536"}, 0, "1-65535"},
|
||||
{"not a number", []string{"abc"}, 0, "1-65535"},
|
||||
{"two ports", []string{"80", "81"}, 0, "一次只支持一个端口"},
|
||||
{"unknown option", []string{"--wat"}, 0, "未知选项"},
|
||||
}
|
||||
|
||||
for _, tc := range cases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
port, err := parsePortArgs(tc.args)
|
||||
if tc.wantErr != "" {
|
||||
if err == nil || !strings.Contains(err.Error(), tc.wantErr) {
|
||||
t.Fatalf("err = %v, want %q", err, tc.wantErr)
|
||||
}
|
||||
return
|
||||
}
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if port != tc.want {
|
||||
t.Errorf("port = %d, want %d", port, tc.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseNetstatPIDs(t *testing.T) {
|
||||
output := `Active Connections
|
||||
|
||||
Proto Local Address Foreign Address State PID
|
||||
TCP 0.0.0.0:8080 0.0.0.0:0 LISTENING 1234
|
||||
TCP [::]:8080 [::]:0 LISTENING 1234
|
||||
TCP 127.0.0.1:8080 127.0.0.1:51515 ESTABLISHED 5678
|
||||
TCP 0.0.0.0:8081 0.0.0.0:0 LISTENING 9999
|
||||
UDP 0.0.0.0:8080 *:* 4321
|
||||
`
|
||||
|
||||
got := parseNetstatPIDs(output, 8080)
|
||||
want := []string{"1234", "4321", "5678"}
|
||||
if strings.Join(got, ",") != strings.Join(want, ",") {
|
||||
t.Fatalf("pids = %v, want %v", got, want)
|
||||
}
|
||||
|
||||
if got := parseNetstatPIDs(output, 9999); len(got) != 0 {
|
||||
t.Errorf("port 9999 should have no match, got %v", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNetPIDsUnixWithFakeLsof(t *testing.T) {
|
||||
if runtime.GOOS == "windows" {
|
||||
t.Skip("lsof fixture is not executable on Windows")
|
||||
}
|
||||
|
||||
dir := t.TempDir()
|
||||
script := filepath.Join(dir, "lsof")
|
||||
if err := os.WriteFile(script, []byte("#!/bin/sh\necho 111\necho 222\n"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Setenv("PATH", dir+string(os.PathListSeparator)+os.Getenv("PATH"))
|
||||
|
||||
var out bytes.Buffer
|
||||
if err := runNetPIDs(context.Background(), BuiltinRequest{Args: []string{"8080"}, Stdout: &out}); err != nil {
|
||||
t.Fatalf("runNetPIDs error: %v", err)
|
||||
}
|
||||
if out.String() != "111\n222\n" {
|
||||
t.Fatalf("output = %q", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestNetPIDsUnixNoMatchIsNotAnError(t *testing.T) {
|
||||
if runtime.GOOS == "windows" {
|
||||
t.Skip("lsof fixture is not executable on Windows")
|
||||
}
|
||||
|
||||
dir := t.TempDir()
|
||||
script := filepath.Join(dir, "lsof")
|
||||
if err := os.WriteFile(script, []byte("#!/bin/sh\nexit 1\n"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
t.Setenv("PATH", dir+string(os.PathListSeparator)+os.Getenv("PATH"))
|
||||
|
||||
var out bytes.Buffer
|
||||
if err := runNetPIDs(context.Background(), BuiltinRequest{Args: []string{"8080"}, Stdout: &out}); err != nil {
|
||||
t.Fatalf("no-match should not fail: %v", err)
|
||||
}
|
||||
if out.Len() != 0 {
|
||||
t.Fatalf("expected no output, got %q", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestNetPIDsRejectsInvalidPortBeforeRunning(t *testing.T) {
|
||||
var out bytes.Buffer
|
||||
err := runNetPIDs(context.Background(), BuiltinRequest{Args: []string{"70000"}, Stdout: &out})
|
||||
if err == nil || !strings.Contains(err.Error(), "1-65535") {
|
||||
t.Fatalf("err = %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestParseSSPIDs(t *testing.T) {
|
||||
output := `tcp ESTAB 0 0 127.0.0.1:8080 127.0.0.1:51515 users:(("app",pid=4321,fd=6))
|
||||
udp UNCONN 0 0 0.0.0.0:8080 0.0.0.0:* users:(("dns",pid=123,fd=5))
|
||||
tcp LISTEN 0 128 0.0.0.0:8081 0.0.0.0:* users:(("other",pid=9999,fd=3))`
|
||||
|
||||
got := parseSSPIDs(output, 8080)
|
||||
want := []string{"123", "4321"}
|
||||
if strings.Join(got, ",") != strings.Join(want, ",") {
|
||||
t.Fatalf("pids = %v, want %v", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNetPIDsUnixFallsBackToSS(t *testing.T) {
|
||||
if runtime.GOOS == "windows" {
|
||||
t.Skip("shell fixtures are not executable on Windows")
|
||||
}
|
||||
|
||||
dir := t.TempDir()
|
||||
body := "#!/bin/sh\n" +
|
||||
"echo 'tcp ESTAB 0 0 127.0.0.1:8080 127.0.0.1:51515 users:((\"app\",pid=4321,fd=6))'\n" +
|
||||
"echo 'tcp LISTEN 0 128 0.0.0.0:8081 0.0.0.0:* users:((\"other\",pid=9999,fd=3))'\n"
|
||||
if err := os.WriteFile(filepath.Join(dir, "ss"), []byte(body), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
// Only the fixture directory: there is no lsof, so ss must be used.
|
||||
t.Setenv("PATH", dir)
|
||||
|
||||
var out bytes.Buffer
|
||||
if err := runNetPIDs(context.Background(), BuiltinRequest{Args: []string{"8080"}, Stdout: &out}); err != nil {
|
||||
t.Fatalf("runNetPIDs error: %v", err)
|
||||
}
|
||||
if out.String() != "4321\n" {
|
||||
t.Fatalf("output = %q, want %q", out.String(), "4321\n")
|
||||
}
|
||||
}
|
||||
|
||||
func TestProcPIDs(t *testing.T) {
|
||||
if runtime.GOOS != "linux" {
|
||||
t.Skip("procPIDs is Linux-only")
|
||||
}
|
||||
|
||||
root := t.TempDir()
|
||||
if err := os.MkdirAll(filepath.Join(root, "net"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
tcp := " sl local_address rem_address st tx_queue rx_queue tr tm->when retrnsmt uid timeout inode\n" +
|
||||
" 0: 0100007F:1F90 00000000:0000 0A 00000000:00000000 00:00000000 00000000 1000 0 12345 1\n"
|
||||
if err := os.WriteFile(filepath.Join(root, "net", "tcp"), []byte(tcp), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
fdDir := filepath.Join(root, "4242", "fd")
|
||||
if err := os.MkdirAll(fdDir, 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.Symlink("socket:[12345]", filepath.Join(fdDir, "3")); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
got, err := procPIDs(8080, root) // 0x1F90 == 8080
|
||||
if err != nil {
|
||||
t.Fatalf("procPIDs error: %v", err)
|
||||
}
|
||||
if strings.Join(got, ",") != "4242" {
|
||||
t.Fatalf("pids = %v, want [4242]", got)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user