77 lines
2 KiB
Go
77 lines
2 KiB
Go
|
package crypto
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/ProtonMail/gopenpgp/constants"
|
||
|
|
)
|
||
|
|
|
||
|
|
var testRandomToken []byte
|
||
|
|
func TestRandomToken(t *testing.T) {
|
||
|
|
var err error
|
||
|
|
testRandomToken, err = pgp.RandomToken()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Expected no error while generating random token, got:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Len(t, testRandomToken, 32)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRandomTokenWith(t *testing.T) {
|
||
|
|
token, err := pgp.RandomTokenWith(40)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Expected no error while generating random token, got:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Len(t, token, 40)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAsymmetricKeyPacket(t *testing.T) {
|
||
|
|
symmetricKey := &SymmetricKey{
|
||
|
|
Key: testRandomToken,
|
||
|
|
Algo: constants.AES256,
|
||
|
|
}
|
||
|
|
|
||
|
|
privateKeyRing, _ := ReadArmoredKeyRing(strings.NewReader(readTestFile("keyring_privateKey", false)))
|
||
|
|
publicKey, _ := testPrivateKeyRing.GetArmoredPublicKey()
|
||
|
|
|
||
|
|
keyPacket, err := pgp.KeyPacketWithPublicKey(symmetricKey, publicKey)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Expected no error while generating key packet, got:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Password defined in keyring_test
|
||
|
|
outputSymmetricKey, err := pgp.GetSessionFromKeyPacket(keyPacket, privateKeyRing, testMailboxPassword)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Expected no error while decrypting key packet, got:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Exactly(t, symmetricKey, outputSymmetricKey)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSymmetricKeyPacket(t *testing.T) {
|
||
|
|
symmetricKey := &SymmetricKey{
|
||
|
|
Key: testRandomToken,
|
||
|
|
Algo: constants.AES256,
|
||
|
|
}
|
||
|
|
|
||
|
|
password := "I like encryption"
|
||
|
|
|
||
|
|
keyPacket, err := pgp.SymmetricKeyPacketWithPassword(symmetricKey, password)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Expected no error while generating key packet, got:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = pgp.GetSessionFromSymmetricPacket(keyPacket, "Wrong password")
|
||
|
|
assert.EqualError(t, err, "password incorrect")
|
||
|
|
|
||
|
|
outputSymmetricKey, err := pgp.GetSessionFromSymmetricPacket(keyPacket, password)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal("Expected no error while decrypting key packet, got:", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Exactly(t, symmetricKey, outputSymmetricKey)
|
||
|
|
}
|