MetzNet
/
pnyx
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Filled out session packet tests

live
noah metz 2024-04-07 13:35:01 -06:00
parent 6c2fb20b1c
commit a617d74102
2 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
packet.go
Unescape Escape View File

@ -245,7 +245,7 @@ func ParseSessionData(session *Session, encrypted []byte) ([]byte, error) {
func NewSessionClose(session *Session) []byte { func NewSessionClose(session *Session) []byte {
packet := make([]byte, COMMAND_LENGTH + ID_LENGTH + SESSION_CLOSE_LENGTH) packet := make([]byte, COMMAND_LENGTH + ID_LENGTH + SESSION_CLOSE_LENGTH)
packet[0] = COMMAND_LENGTH packet[0] = byte(SESSION_CLOSE)
copy(packet[1:], session.ID[:]) copy(packet[1:], session.ID[:])
hmac := sha512.Sum512(append(session.ID[:], session.secret...)) hmac := sha512.Sum512(append(session.ID[:], session.secret...))

27
packet_test.go
Unescape Escape View File

@ -25,6 +25,9 @@ func TestSessionOpen(t *testing.T) {
client_so, client_ecdh, err := NewSessionOpen(client_key) client_so, client_ecdh, err := NewSessionOpen(client_key)
fatalErr(t, err) fatalErr(t, err)
if client_so[0] != byte(SESSION_OPEN) {
t.Fatalf("Session open command byte mismatch(%x != %x)", client_so[0], SESSION_OPEN)
}
server_so, server_ecdh, err := NewSessionOpen(server_key) server_so, server_ecdh, err := NewSessionOpen(server_key)
fatalErr(t, err) fatalErr(t, err)
@ -71,7 +74,14 @@ func TestSessionConnect(t *testing.T) {
test_addr, err := net.ResolveUDPAddr("udp", "127.0.0.1:8080") test_addr, err := net.ResolveUDPAddr("udp", "127.0.0.1:8080")
fatalErr(t, err) fatalErr(t, err)
session_id := ID[SessionID](secret)
session_connect := NewSessionConnect(test_addr, secret) session_connect := NewSessionConnect(test_addr, secret)
if session_connect[0] != byte(SESSION_CONNECT) {
t.Fatalf("Session open command byte mismatch(%x != %x)", session_connect[0], SESSION_CONNECT)
} else if slices.Compare(session_connect[COMMAND_LENGTH:COMMAND_LENGTH+ID_LENGTH], session_id[:]) != 0 {
t.Fatal("Session open ID mismatch")
}
parsed_addr, err := ParseSessionConnect(session_connect[COMMAND_LENGTH + ID_LENGTH:], secret) parsed_addr, err := ParseSessionConnect(session_connect[COMMAND_LENGTH + ID_LENGTH:], secret)
fatalErr(t, err) fatalErr(t, err)
@ -100,10 +110,16 @@ func TestSessionData(t *testing.T) {
fatalErr(t, err) fatalErr(t, err)
message := []byte("hello") message := []byte("hello")
server_hello, err := NewSessionData(&server_session, message) session_data, err := NewSessionData(&server_session, message)
fatalErr(t, err) fatalErr(t, err)
parsed_message, err := ParseSessionData(&client_session, server_hello[COMMAND_LENGTH+ID_LENGTH:]) if session_data[0] != byte(SESSION_DATA) {
t.Fatalf("Session data command byte mismatch(%x != %x)", session_data[0], SESSION_DATA)
} else if slices.Compare(session_data[COMMAND_LENGTH:COMMAND_LENGTH+ID_LENGTH], server_session.ID[:]) != 0 {
t.Fatal("Session data ID mismatch")
}
parsed_message, err := ParseSessionData(&client_session, session_data[COMMAND_LENGTH+ID_LENGTH:])
fatalErr(t, err) fatalErr(t, err)
if slices.Compare(message, parsed_message) != 0 { if slices.Compare(message, parsed_message) != 0 {
@ -131,5 +147,12 @@ func TestSessionClose(t *testing.T) {
fatalErr(t, err) fatalErr(t, err)
session_close := NewSessionClose(&client_session) session_close := NewSessionClose(&client_session)
if session_close[0] != byte(SESSION_CLOSE) {
t.Fatalf("Session close command byte mismatch(%x != %x)", session_close[0], SESSION_CLOSE)
} else if slices.Compare(session_close[COMMAND_LENGTH:COMMAND_LENGTH+ID_LENGTH], server_session.ID[:]) != 0 {
t.Fatal("Session close ID mismatch")
}
fatalErr(t, ParseSessionClose(&server_session, session_close[COMMAND_LENGTH+ID_LENGTH:])) fatalErr(t, ParseSessionClose(&server_session, session_close[COMMAND_LENGTH+ID_LENGTH:]))
} }