54f45d0471
* Change keyring unlock functionalities * Add keyring#Lock, keyring#CheckIntegrity, tests * Update helpers, fix bugs * Update go.mod with ProtonMail/crypto commit * Change key management system * Clear keys from memory + tests * Create SessionKey with direct encryption for datapackets. Move symmetrickey to password. * Fix upstream dependencies * Update module to V2, documentation * Add linter * Add v2 folder to .gitignore * Minor changes to KeyID getters * Remove old changelog * Improve docs, remove compilation script
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package crypto
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
"golang.org/x/crypto/ed25519"
|
|
"golang.org/x/crypto/openpgp/ecdh"
|
|
"golang.org/x/crypto/rsa"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const testTime = 1557754627 // 2019-05-13T13:37:07+00:00
|
|
|
|
func readTestFile(name string, trimNewlines bool) string {
|
|
data, err := ioutil.ReadFile("testdata/" + name) //nolint
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if trimNewlines {
|
|
return strings.TrimRight(string(data), "\n")
|
|
}
|
|
return string(data)
|
|
}
|
|
|
|
func init() {
|
|
UpdateTime(testTime) // 2019-05-13T13:37:07+00:00
|
|
|
|
initGenerateKeys()
|
|
initArmoredKeys()
|
|
initKeyRings()
|
|
}
|
|
|
|
func assertBigIntCleared(t *testing.T, x *big.Int) {
|
|
w := x.Bits()
|
|
for k := range w {
|
|
assert.Exactly(t, big.Word(0x00), w[k])
|
|
}
|
|
}
|
|
|
|
func assertMemCleared(t *testing.T, b []byte) {
|
|
for k := range b {
|
|
assert.Exactly(t, uint8(0x00), b[k])
|
|
}
|
|
}
|
|
|
|
func assertRSACleared(t *testing.T, rsaPriv *rsa.PrivateKey) {
|
|
assertBigIntCleared(t, rsaPriv.D)
|
|
for idx := range rsaPriv.Primes {
|
|
assertBigIntCleared(t, rsaPriv.Primes[idx])
|
|
}
|
|
assertBigIntCleared(t, rsaPriv.Precomputed.Qinv)
|
|
assertBigIntCleared(t, rsaPriv.Precomputed.Dp)
|
|
assertBigIntCleared(t, rsaPriv.Precomputed.Dq)
|
|
|
|
for idx := range rsaPriv.Precomputed.CRTValues {
|
|
assertBigIntCleared(t, rsaPriv.Precomputed.CRTValues[idx].Exp)
|
|
assertBigIntCleared(t, rsaPriv.Precomputed.CRTValues[idx].Coeff)
|
|
assertBigIntCleared(t, rsaPriv.Precomputed.CRTValues[idx].R)
|
|
}
|
|
}
|
|
|
|
func assertEdDSACleared(t *testing.T, priv ed25519.PrivateKey) {
|
|
assertMemCleared(t, priv)
|
|
}
|
|
|
|
func assertECDHCleared(t *testing.T, priv *ecdh.PrivateKey) {
|
|
assertMemCleared(t, priv.D)
|
|
}
|