mirror of
https://github.com/Mezeporta/Erupe.git
synced 2025-12-12 23:14:36 +01:00
105 lines
3.8 KiB
Go
105 lines
3.8 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,
|
|
},
|
|
{
|
|
[]byte{0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x03, 0x02, 0x00, 0x6C, 0x6C, 0x00, 0x00, 0x00, 0x12, 0x00, 0xDE, 0x00, 0x03, 0x00, 0x00, 0x00, 0x30, 0x00, 0x02, 0x01, 0x00, 0x00, 0x00, 0x00, 0x03, 0x20, 0x18, 0x46, 0x00, 0x00, 0x80, 0x3F, 0xDC, 0xE4, 0x0A, 0x46, 0x10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x67, 0xD3, 0x5B, 0x00, 0x77, 0x01, 0x78, 0x00, 0x77, 0x01, 0x4F, 0x01, 0x5B, 0x6F, 0x76, 0xC5, 0x30, 0x00, 0x02, 0x02, 0x00, 0x00, 0x00, 0x00, 0x2A, 0xDD, 0x17, 0x46, 0x00, 0x00, 0x80, 0x3F, 0x9E, 0x11, 0x0C, 0x46, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x18, 0x00, 0x00, 0x00, 0x00, 0x0C, 0x37, 0x64, 0x00, 0x2C, 0x01, 0x64, 0x00, 0x2C, 0x01, 0x4F, 0x01, 0x5B, 0x6F, 0x76, 0xC5, 0x00, 0x10},
|
|
2000476,
|
|
[]byte{0x2E, 0x52, 0x24, 0xE3, 0x05, 0x2B, 0xFC, 0x04, 0x0B, 0x26, 0x90, 0xEA, 0x61, 0xDB, 0x8D, 0x27, 0xCB, 0xB1, 0x69, 0xA1, 0x77, 0x80, 0x4A, 0xC2, 0xA0, 0xBD, 0x50, 0x54, 0xF5, 0xC2, 0x94, 0x66, 0xBB, 0xCE, 0x53, 0x29, 0xEE, 0xB4, 0xFA, 0xF6, 0x5F, 0x8D, 0x80, 0x3E, 0x5D, 0x5F, 0xB0, 0x53, 0xE6, 0x92, 0x17, 0x80, 0xE7, 0xED, 0xE7, 0xDC, 0x61, 0xF0, 0xCD, 0xE4, 0x41, 0x82, 0x21, 0xBA, 0x47, 0xAB, 0x58, 0xFF, 0x30, 0x76, 0x80, 0x2D, 0x38, 0xF4, 0xDF, 0x86, 0x8C, 0x6C, 0x8D, 0x33, 0x4C, 0x37, 0xA3, 0xDA, 0x01, 0x3C, 0x98, 0x66, 0x1F, 0xB9, 0xE2, 0xEA, 0xF0, 0x84, 0xE8, 0xAA, 0x00, 0x3D, 0x4A, 0xB6, 0xF2, 0x3D, 0x91, 0x58, 0x4B, 0x0B, 0xE2, 0xD5, 0xC7, 0x39, 0x4D, 0x59, 0xED, 0xC3, 0x61, 0x6F, 0x6E, 0x69, 0x9B, 0x3C},
|
|
0xCFF8, 0x086B, 0x3BAE, 0x4057,
|
|
},
|
|
}
|
|
|
|
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 := Crypto(tt.decryptedData, tt.key, true, 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 := Crypto(tt.decryptedData, tt.key, false, 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))
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|