diff --git a/intel/entity.go b/intel/entity.go index f5fc9979..16d83707 100644 --- a/intel/entity.go +++ b/intel/entity.go @@ -95,12 +95,12 @@ func (e *Entity) Init() *Entity { return e } -// AddDstPort adds the destination port. -func (e *Entity) AddDstPort(dstPort uint16) { +// SetDstPort sets the destination port. +func (e *Entity) SetDstPort(dstPort uint16) { e.dstPort = dstPort } -// AddDstPort adds the destination port. +// DstPort returns the destination port. func (e *Entity) DstPort() uint16 { return e.dstPort } diff --git a/netenv/location_windows.go b/netenv/location_windows.go index bbfb088b..92d34291 100644 --- a/netenv/location_windows.go +++ b/netenv/location_windows.go @@ -10,7 +10,7 @@ import ( ) // Windows specific constants for the WSAIoctl interface. -//nolint:golint +//nolint:golint,stylecheck const ( SIO_RCVALL = syscall.IOC_IN | syscall.IOC_VENDOR | 1 diff --git a/network/connection.go b/network/connection.go index 2ac3cb2b..2839ded3 100644 --- a/network/connection.go +++ b/network/connection.go @@ -128,7 +128,7 @@ func NewConnectionFromFirstPacket(pkt packet.Packet) *Connection { Protocol: uint8(pkt.Info().Protocol), Port: pkt.Info().SrcPort, } - entity.AddDstPort(pkt.Info().DstPort) + entity.SetDstPort(pkt.Info().DstPort) } else { @@ -138,7 +138,7 @@ func NewConnectionFromFirstPacket(pkt packet.Packet) *Connection { Protocol: uint8(pkt.Info().Protocol), Port: pkt.Info().DstPort, } - entity.AddDstPort(entity.Port) + entity.SetDstPort(entity.Port) // check if we can find a domain for that IP ipinfo, err := resolver.GetIPInfo(proc.LocalProfileKey, pkt.Info().Dst.String()) diff --git a/profile/endpoints/endpoints_test.go b/profile/endpoints/endpoints_test.go index c50aca90..9b28e131 100644 --- a/profile/endpoints/endpoints_test.go +++ b/profile/endpoints/endpoints_test.go @@ -17,6 +17,8 @@ func TestMain(m *testing.M) { } func testEndpointMatch(t *testing.T, ep Endpoint, entity *intel.Entity, expectedResult EPResult) { + entity.SetDstPort(entity.Port) + result, _ := ep.Matches(context.TODO(), entity) if result != expectedResult { t.Errorf(