mirror of
https://github.com/Mezeporta/Erupe.git
synced 2025-12-14 07:55:33 +01:00
99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
package crypto
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
var commonTestData = []byte{0x74, 0x65, 0x73, 0x74}
|
|
var tests = []struct {
|
|
decryptedData []byte
|
|
key uint32
|
|
encryptedData []byte
|
|
ecc, ec0, ec1, ec2 uint16
|
|
}{
|
|
{
|
|
commonTestData,
|
|
0,
|
|
[]byte{0x46, 0x53, 0x28, 0x5E},
|
|
0x2976, 0x06ea, 0x0215, 0x08FB3,
|
|
},
|
|
{
|
|
commonTestData,
|
|
3,
|
|
[]byte{0x46, 0x95, 0x88, 0xEA},
|
|
0x2AE4, 0x0A56, 0x01CD, 0x08FB3,
|
|
},
|
|
/*
|
|
// TODO(Andoryuuta): This case fails. Debug the client and figure out if this is valid expected data.
|
|
{
|
|
commonTestData,
|
|
995117,
|
|
[]byte{0x46, 0x28, 0xFF, 0xAA},
|
|
0x2A22, 0x09D4, 0x014C, 0x08FB3,
|
|
},
|
|
*/
|
|
{
|
|
commonTestData,
|
|
0x7FFFFFFF,
|
|
[]byte{0x46, 0x53, 0x28, 0x5E},
|
|
0x2976, 0x06ea, 0x0215, 0x08FB3,
|
|
},
|
|
{
|
|
commonTestData,
|
|
0x80000000,
|
|
[]byte{0x46, 0x95, 0x88, 0xEA},
|
|
0x2AE4, 0x0A56, 0x01CD, 0x08FB3,
|
|
},
|
|
{
|
|
commonTestData,
|
|
0xFFFFFFFF,
|
|
[]byte{0x46, 0xB5, 0xDC, 0xB2},
|
|
0x2ADD, 0x09A6, 0x021E, 0x08FB3,
|
|
},
|
|
}
|
|
|
|
func TestEncrypt(t *testing.T) {
|
|
for k, tt := range tests {
|
|
testname := fmt.Sprintf("encrypt_test_%d", k)
|
|
t.Run(testname, func(t *testing.T) {
|
|
out, cc, c0, c1, c2 := Encrypt(tt.decryptedData, tt.key, nil)
|
|
if cc != tt.ecc {
|
|
t.Errorf("got cc 0x%X, want 0x%X", cc, tt.ecc)
|
|
} else if c0 != tt.ec0 {
|
|
t.Errorf("got c0 0x%X, want 0x%X", c0, tt.ec0)
|
|
} else if c1 != tt.ec1 {
|
|
t.Errorf("got c1 0x%X, want 0x%X", c1, tt.ec1)
|
|
} else if c2 != tt.ec2 {
|
|
t.Errorf("got c2 0x%X, want 0x%X", c2, tt.ec2)
|
|
} else if !bytes.Equal(out, tt.encryptedData) {
|
|
t.Errorf("got out\n\t%s\nwant\n\t%s", hex.Dump(out), hex.Dump(tt.encryptedData))
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func TestDecrypt(t *testing.T) {
|
|
for k, tt := range tests {
|
|
testname := fmt.Sprintf("decrypt_test_%d", k)
|
|
t.Run(testname, func(t *testing.T) {
|
|
out, cc, c0, c1, c2 := Decrypt(tt.encryptedData, tt.key, nil)
|
|
if cc != tt.ecc {
|
|
t.Errorf("got cc 0x%X, want 0x%X", cc, tt.ecc)
|
|
} else if c0 != tt.ec0 {
|
|
t.Errorf("got c0 0x%X, want 0x%X", c0, tt.ec0)
|
|
} else if c1 != tt.ec1 {
|
|
t.Errorf("got c1 0x%X, want 0x%X", c1, tt.ec1)
|
|
} else if c2 != tt.ec2 {
|
|
t.Errorf("got c2 0x%X, want 0x%X", c2, tt.ec2)
|
|
} else if !bytes.Equal(out, tt.decryptedData) {
|
|
t.Errorf("got out\n\t%s\nwant\n\t%s", hex.Dump(out), hex.Dump(tt.decryptedData))
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|