devpod/pkg/netstat/netstat_util_test.go at 80dac5dea5971b6c97d06e614db1f0febbaf2016 · YassineElbouchaibi/devpod · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package netstat

import (
	"os"
	"path/filepath"
	"runtime"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"
)

type NetstatUtilTestSuite struct {
	suite.Suite
}

func TestNetstatUtilSuite(t *testing.T) {
	suite.Run(t, new(NetstatUtilTestSuite))
}

func (s *NetstatUtilTestSuite) TestDoNetstat_MissingFileReturnsEmpty() {
	missing := filepath.Join(s.T().TempDir(), "does-not-exist")

	entries, err := doNetstat(missing, NoopFilter)

	assert.NoError(s.T(), err, "missing procfile must not error")
	assert.Empty(s.T(), entries, "missing procfile must yield zero entries")
}

func (s *NetstatUtilTestSuite) TestDoNetstat_ParsesValidFile() {
	// Synthetic /proc/net/tcp6 entry: [::]:22 (ssh), state 0A=Listen.
	fields := []string{
		"0:",
		"00000000000000000000000000000000:0016",
		"00000000000000000000000000000000:0000",
		"0A",
		"00000000:00000000",
		"00:00000000",
		"00000000",
		"0",
		"0",
		"12345",
		"1",
		"0000000000000000",
		"100",
		"0",
		"0",
		"10",
		"0",
	}
	content := "header line discarded by parseSocktab\n" + strings.Join(fields, " ") + "\n"

	path := filepath.Join(s.T().TempDir(), "tcp6")
	require.NoError(s.T(), os.WriteFile(path, []byte(content), 0o600))

	entries, err := doNetstat(path, func(e *SockTabEntry) bool {
		return e.State == Listen
	})

	require.NoError(s.T(), err)
	require.Len(s.T(), entries, 1)
	assert.Equal(s.T(), uint16(22), entries[0].LocalAddr.Port)
	assert.Equal(s.T(), Listen, entries[0].State)
	assert.Equal(s.T(), uint32(0), entries[0].UID)
}

func (s *NetstatUtilTestSuite) TestDoNetstat_PropagatesNonNotExistOpenError() {
	if runtime.GOOS == "windows" {
		s.T().Skip("permission semantics differ on Windows")
	}
	if os.Geteuid() == 0 {
		s.T().Skip("root bypasses file-mode permission checks")
	}

	path := filepath.Join(s.T().TempDir(), "unreadable")
	require.NoError(s.T(), os.WriteFile(path, []byte("irrelevant"), 0o000))
	s.T().Cleanup(func() { _ = os.Chmod(path, 0o600) })

	_, err := doNetstat(path, NoopFilter)

	require.Error(s.T(), err, "permission errors must propagate")
	assert.False(s.T(), os.IsNotExist(err), "error should not be IsNotExist")
}

func (s *NetstatUtilTestSuite) TestDoNetstat_PropagatesParseError() {
	path := filepath.Join(s.T().TempDir(), "garbage")
	require.NoError(s.T(), os.WriteFile(path, []byte("header\nnot enough fields here\n"), 0o600))

	_, err := doNetstat(path, NoopFilter)

	assert.Error(s.T(), err, "parser errors must propagate")
}