// TestRSAGenerateEnDe
func TestRSAGenerateEnDe(t *testing.T) {
privBytes, pubBytes, err := utils.GenerateRSAKeyPair(1024)
assert.Nil(t, err, "Fail to genereate RSA Key Pair")
testData := []byte("This is the testdata for encrypt and decryp")
encrypted, err := utils.RSAEncrypt(pubBytes, testData)
assert.Nil(t, err, "Fail to encrypt data")
decrypted, err := utils.RSADecrypt(privBytes, encrypted)
assert.Nil(t, err, "Fail to decrypt data")
assert.Equal(t, testData, decrypted, "Fail to get correct data after en/de")
}
func generateKey(keyDir string) error {
privBytes, pubBytes, err := utils.GenerateRSAKeyPair(defaultBitsSize)
if err != nil {
return err
}
if !utils.IsDirExist(keyDir) {
err := os.MkdirAll(keyDir, 0755)
if err != nil {
return err
}
}
if err := ioutil.WriteFile(filepath.Join(keyDir, defaultPrivateKey), privBytes, 0600); err != nil {
return err
}
if err := ioutil.WriteFile(filepath.Join(keyDir, defaultPublicKey), pubBytes, 0644); err != nil {
return err
}
return nil
}
