storj/satellite/orders/encryptionkey_test.go

216 lines
7.4 KiB
Go
Raw Normal View History

// Copyright (C) 2020 Storj Labs, Inc.
// See LICENSE for copying information.
package orders_test
import (
"encoding/hex"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"storj.io/common/storj"
"storj.io/common/testrand"
"storj.io/storj/satellite/orders"
)
func TestEncryptionKey_Set_Valid(t *testing.T) {
type Test struct {
Hex string
Key orders.EncryptionKey
}
tests := []Test{
{
Hex: `0100000000000000=0100000000000000000000000000000000000000000000000000000000000000`,
Key: orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x01},
Key: storj.Key{0x01},
},
},
{
Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF`,
Key: orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF},
Key: storj.Key{
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF,
},
},
},
}
for _, test := range tests {
var got orders.EncryptionKey
err := got.Set(test.Hex)
assert.NoError(t, err, test.Hex)
assert.Equal(t, test.Key, got, test.Hex)
}
}
func TestEncryptionKey_Set_Invalid(t *testing.T) {
type Test struct {
Hex string
}
tests := []Test{
{Hex: ``},
{Hex: `=`},
{Hex: `01=`},
{Hex: `=01`},
{Hex: `1=1`},
{Hex: `=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `112233445566778=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `1122334455667788=`},
{Hex: `1122334455667788=112233445566778811223344556677881122334455667788112233445566778`},
{Hex: `11223344556677QQ=11223344556677881122334455667788112233445566778811223344556677QQ`},
}
for _, test := range tests {
var got orders.EncryptionKey
err := got.Set(test.Hex)
assert.Error(t, err, test.Hex)
}
}
func TestEncryptionKeys_Set_Valid(t *testing.T) {
var keys orders.EncryptionKeys
err := keys.Set(`11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,0100000000000000=0100000000000000000000000000000000000000000000000000000000000000`)
require.NoError(t, err)
first := orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF},
Key: storj.Key{
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0xFF,
},
}
second := orders.EncryptionKey{
ID: orders.EncryptionKeyID{0x01},
Key: storj.Key{0x01},
}
assert.Equal(t, first, keys.Default)
assert.EqualValues(t, []orders.EncryptionKey{first, second}, keys.List)
assert.Equal(t, first.Key, keys.KeyByID[first.ID])
assert.Equal(t, second.Key, keys.KeyByID[second.ID])
}
func TestEncryptionKeys_Set_Invalid(t *testing.T) {
type Test struct {
Hex string
}
tests := []Test{
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,=`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,01=`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,=01`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,1=1`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,112233445566778=1122334455667788112233445566778811223344556677881122334455667788`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,1122334455667788=`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,1122334455667788=112233445566778811223344556677881122334455667788112233445566778`},
{Hex: `11223344556677FF=11223344556677881122334455667788112233445566778811223344556677FF,11223344556677QQ=11223344556677881122334455667788112233445566778811223344556677QQ`},
}
for _, test := range tests {
var got orders.EncryptionKeys
err := got.Set(test.Hex)
assert.Error(t, err, test.Hex)
}
}
func randEncryptionKey() orders.EncryptionKey {
k := orders.EncryptionKey{}
r := testrand.Nonce()
copy(k.ID[:], r[:])
k.Key = testrand.Key()
return k
}
func TestEncryptionKey_EncryptDecrypt(t *testing.T) {
for i := 0; i < 10; i++ {
key := randEncryptionKey()
data := testrand.BytesInt(16)
serial := testrand.SerialNumber()
encrypted := key.Encrypt(data, serial)
require.NotEqual(t, encrypted, data)
decrypted, err := key.Decrypt(encrypted, serial)
require.NoError(t, err)
require.Equal(t, data, decrypted)
}
}
func TestEncryptionKey_BackwardsCompatibility(t *testing.T) {
type Test struct {
Key string
DataHex string
Serial storj.SerialNumber
EncryptedHex string
}
tests := []Test{
{
Key: "1d729566c74d1003=0d86d1e91e00167939cb6694d2c422acd208a0072939487f6999eb9d18a44784",
DataHex: "045d87f3c67cf22746e995af5a253679",
Serial: storj.SerialNumber{81, 186, 162, 255, 108, 212, 113, 196, 131, 241, 95, 185, 11, 173, 179, 124},
EncryptedHex: "5fe4c9fc734baed429afe24502ffe9d4c6efd99de5d5ae07433f7449bc55e4e1",
}, {
Key: "5821b6d95526a41a=86216325253fec738dd7a9e28bf921119c160f0702448615bbda08313f6a8eb6",
DataHex: "68d20bf5059875921e668a5bdf2c7fc4",
Serial: storj.SerialNumber{132, 69, 146, 210, 87, 43, 205, 6, 104, 210, 214, 197, 47, 80, 84, 226},
EncryptedHex: "335eb0ab85e0624862b63c6bad78c4bdd1d52db621332773aef29b977626bcd0",
}, {
Key: "d0836bf84c7174cb=358b0c3b525da1786f9fff094279db1944ebd7a19d0f7bbacbe0255aa5b7d44b",
DataHex: "ec40f84c892b9bffd43629b0223beea5",
Serial: storj.SerialNumber{244, 247, 67, 145, 244, 69, 209, 90, 253, 66, 148, 4, 3, 116, 246, 146},
EncryptedHex: "2615a21b9968f4fa7fdc5eb423f708a0630ec25295852e513c658eecaee0493e",
}, {
Key: "4b98cbf8713f8d96=b586b14323a6bc8f9e7df1d929333ff993933bea6f5b3af6de0374366c4719e4",
DataHex: "3a1b067d89bc7f01f1f573981659a44f",
Serial: storj.SerialNumber{241, 122, 76, 114, 21, 163, 181, 57, 235, 30, 88, 73, 198, 7, 125, 187},
EncryptedHex: "e57f0b0e936bff96895aae6f963e0ff0a14350d2f787dda47116de0e1a7bf390",
}, {
Key: "5722f5717a289a26=5e82ed6f4125c8fa7311e4d7defa922daae7786667f7e936cd4f24abf7df866b",
DataHex: "aa56038367ad6145de1ee8f4a8b0993e",
Serial: storj.SerialNumber{189, 248, 136, 58, 10, 216, 190, 156, 57, 120, 176, 72, 131, 229, 106, 21},
EncryptedHex: "8a238f268df2ae5a58134e2f52a813ffc157e0f12d8430268f55712c6bf8aae9",
},
}
for _, test := range tests {
var key orders.EncryptionKey
require.NoError(t, key.Set(test.Key))
data, err := hex.DecodeString(test.DataHex)
require.NoError(t, err)
encrypted, err := hex.DecodeString(test.EncryptedHex)
require.NoError(t, err)
decrypted, err := key.Decrypt(encrypted, test.Serial)
require.NoError(t, err)
require.Equal(t, data, decrypted)
}
}