Add rust kext to the mono repo
This commit is contained in:
Vladimir Stoilov
246
windows_kext/kext_interface/protocol_test.go
Normal file
246
windows_kext/kext_interface/protocol_test.go
Normal file
@@ -0,0 +1,246 @@
|
||||
package kext_interface
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"math/rand"
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestRustInfoFile(t *testing.T) {
|
||||
file, err := os.Open("../protocol/rust_info_test.bin")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer file.Close()
|
||||
for {
|
||||
info, err := RecvInfo(file)
|
||||
if err != nil {
|
||||
if err != io.EOF {
|
||||
t.Errorf("unexpected error: %s\n", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
if info.LogLine != nil {
|
||||
if info.LogLine.Severity != 1 {
|
||||
t.Errorf("unexpected Log severity: %d\n", info.LogLine.Severity)
|
||||
}
|
||||
|
||||
if info.LogLine.Line != "prefix: test log" {
|
||||
t.Errorf("unexpected Log line: %s\n", info.LogLine.Line)
|
||||
}
|
||||
} else if info.ConnectionV4 != nil {
|
||||
conn := info.ConnectionV4
|
||||
expected := connectionV4Internal{
|
||||
Id: 1,
|
||||
ProcessId: 2,
|
||||
Direction: 3,
|
||||
Protocol: 4,
|
||||
LocalIp: [4]byte{1, 2, 3, 4},
|
||||
RemoteIp: [4]byte{2, 3, 4, 5},
|
||||
LocalPort: 5,
|
||||
RemotePort: 6,
|
||||
PayloadLayer: 7,
|
||||
}
|
||||
if conn.connectionV4Internal != expected {
|
||||
t.Errorf("unexpected ConnectionV4: %+v\n", conn)
|
||||
}
|
||||
if !bytes.Equal(conn.Payload, []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}) {
|
||||
t.Errorf("unexpected ConnectionV4 payload: %+v\n", conn.Payload)
|
||||
}
|
||||
} else if info.ConnectionV6 != nil {
|
||||
conn := info.ConnectionV6
|
||||
expected := connectionV6Internal{
|
||||
Id: 1,
|
||||
ProcessId: 2,
|
||||
Direction: 3,
|
||||
Protocol: 4,
|
||||
LocalIp: [16]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16},
|
||||
RemoteIp: [16]byte{2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17},
|
||||
LocalPort: 5,
|
||||
RemotePort: 6,
|
||||
PayloadLayer: 7,
|
||||
}
|
||||
if conn.connectionV6Internal != expected {
|
||||
t.Errorf("unexpected ConnectionV6: %+v\n", conn)
|
||||
}
|
||||
if !bytes.Equal(conn.Payload, []byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}) {
|
||||
t.Errorf("unexpected ConnectionV6 payload: %+v\n", conn.Payload)
|
||||
}
|
||||
} else if info.ConnectionEndV4 != nil {
|
||||
endEvent := info.ConnectionEndV4
|
||||
expected := ConnectionEndV4{
|
||||
ProcessId: 1,
|
||||
Direction: 2,
|
||||
Protocol: 3,
|
||||
LocalIp: [4]byte{1, 2, 3, 4},
|
||||
RemoteIp: [4]byte{2, 3, 4, 5},
|
||||
LocalPort: 4,
|
||||
RemotePort: 5,
|
||||
}
|
||||
if *endEvent != expected {
|
||||
t.Errorf("unexpected ConnectionEndV4: %+v\n", endEvent)
|
||||
}
|
||||
} else if info.ConnectionEndV6 != nil {
|
||||
endEvent := info.ConnectionEndV6
|
||||
expected := ConnectionEndV6{
|
||||
ProcessId: 1,
|
||||
Direction: 2,
|
||||
Protocol: 3,
|
||||
LocalIp: [16]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16},
|
||||
RemoteIp: [16]byte{2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17},
|
||||
LocalPort: 4,
|
||||
RemotePort: 5,
|
||||
}
|
||||
if *endEvent != expected {
|
||||
t.Errorf("unexpected ConnectionEndV6: %+v\n", endEvent)
|
||||
}
|
||||
} else if info.BandwidthStats != nil {
|
||||
stats := info.BandwidthStats
|
||||
if stats.Protocol != 1 {
|
||||
t.Errorf("unexpected Bandwidth stats protocol: %d\n", stats.Protocol)
|
||||
}
|
||||
|
||||
if stats.ValuesV4 != nil {
|
||||
if len(stats.ValuesV4) != 2 {
|
||||
t.Errorf("unexpected Bandwidth stats value length: %d\n", len(stats.ValuesV4))
|
||||
}
|
||||
expected1 := BandwidthValueV4{
|
||||
LocalIP: [4]byte{1, 2, 3, 4},
|
||||
LocalPort: 1,
|
||||
RemoteIP: [4]byte{2, 3, 4, 5},
|
||||
RemotePort: 2,
|
||||
TransmittedBytes: 3,
|
||||
ReceivedBytes: 4,
|
||||
}
|
||||
if stats.ValuesV4[0] != expected1 {
|
||||
t.Errorf("unexpected Bandwidth stats value: %+v expected: %+v\n", stats.ValuesV4[0], expected1)
|
||||
}
|
||||
expected2 := BandwidthValueV4{
|
||||
LocalIP: [4]byte{1, 2, 3, 4},
|
||||
LocalPort: 5,
|
||||
RemoteIP: [4]byte{2, 3, 4, 5},
|
||||
RemotePort: 6,
|
||||
TransmittedBytes: 7,
|
||||
ReceivedBytes: 8,
|
||||
}
|
||||
if stats.ValuesV4[1] != expected2 {
|
||||
t.Errorf("unexpected Bandwidth stats value: %+v expected: %+v\n", stats.ValuesV4[1], expected2)
|
||||
}
|
||||
|
||||
} else if stats.ValuesV6 != nil {
|
||||
if len(stats.ValuesV6) != 2 {
|
||||
t.Errorf("unexpected Bandwidth stats value length: %d\n", len(stats.ValuesV6))
|
||||
}
|
||||
|
||||
expected1 := BandwidthValueV6{
|
||||
LocalIP: [16]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16},
|
||||
LocalPort: 1,
|
||||
RemoteIP: [16]byte{2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17},
|
||||
RemotePort: 2,
|
||||
TransmittedBytes: 3,
|
||||
ReceivedBytes: 4,
|
||||
}
|
||||
if stats.ValuesV6[0] != expected1 {
|
||||
t.Errorf("unexpected Bandwidth stats value: %+v expected: %+v\n", stats.ValuesV6[0], expected1)
|
||||
}
|
||||
expected2 := BandwidthValueV6{
|
||||
LocalIP: [16]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16},
|
||||
LocalPort: 5,
|
||||
RemoteIP: [16]byte{2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17},
|
||||
RemotePort: 6,
|
||||
TransmittedBytes: 7,
|
||||
ReceivedBytes: 8,
|
||||
}
|
||||
if stats.ValuesV6[1] != expected2 {
|
||||
t.Errorf("unexpected Bandwidth stats value: %+v expected: %+v\n", stats.ValuesV6[1], expected2)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGenerateCommandFile(t *testing.T) {
|
||||
file, err := os.Create("go_command_test.bin")
|
||||
if err != nil {
|
||||
t.Errorf("failed to create file: %s", err)
|
||||
}
|
||||
defer file.Close()
|
||||
enums := []byte{
|
||||
CommandShutdown,
|
||||
CommandVerdict,
|
||||
CommandUpdateV4,
|
||||
CommandUpdateV6,
|
||||
CommandClearCache,
|
||||
CommandGetLogs,
|
||||
CommandBandwidthStats,
|
||||
CommandCleanEndedConnections,
|
||||
}
|
||||
|
||||
selected := make([]byte, 5000)
|
||||
for i := range selected {
|
||||
selected[i] = enums[rand.Intn(len(enums))]
|
||||
}
|
||||
|
||||
for _, value := range selected {
|
||||
switch value {
|
||||
case CommandShutdown:
|
||||
{
|
||||
SendShutdownCommand(file)
|
||||
}
|
||||
case CommandVerdict:
|
||||
{
|
||||
SendVerdictCommand(file, Verdict{
|
||||
Id: 1,
|
||||
Verdict: 2,
|
||||
})
|
||||
}
|
||||
case CommandUpdateV4:
|
||||
{
|
||||
SendUpdateV4Command(file, UpdateV4{
|
||||
Protocol: 1,
|
||||
LocalAddress: [4]byte{1, 2, 3, 4},
|
||||
LocalPort: 2,
|
||||
RemoteAddress: [4]byte{2, 3, 4, 5},
|
||||
RemotePort: 3,
|
||||
Verdict: 4,
|
||||
})
|
||||
}
|
||||
case CommandUpdateV6:
|
||||
{
|
||||
|
||||
SendUpdateV6Command(file, UpdateV6{
|
||||
Protocol: 1,
|
||||
LocalAddress: [16]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16},
|
||||
LocalPort: 2,
|
||||
RemoteAddress: [16]byte{2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17},
|
||||
RemotePort: 3,
|
||||
Verdict: 4,
|
||||
})
|
||||
}
|
||||
case CommandClearCache:
|
||||
{
|
||||
SendClearCacheCommand(file)
|
||||
}
|
||||
case CommandGetLogs:
|
||||
{
|
||||
SendGetLogsCommand(file)
|
||||
}
|
||||
case CommandBandwidthStats:
|
||||
{
|
||||
SendGetBandwidthStatsCommand(file)
|
||||
}
|
||||
case CommandPrintMemoryStats:
|
||||
{
|
||||
SendPrintMemoryStatsCommand(file)
|
||||
}
|
||||
case CommandCleanEndedConnections:
|
||||
{
|
||||
SendCleanEndedConnectionsCommand(file)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user