72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package crypto
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/ProtonMail/go-crypto/ed25519"
|
|
"github.com/ProtonMail/go-crypto/openpgp/ecdh"
|
|
"github.com/ProtonMail/go-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)
|
|
}
|