You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
7 years ago
|
package gnutls
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"crypto/aes"
|
||
|
"crypto/cipher"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestEncryptDecrypt(t *testing.T) {
|
||
|
key := []byte("0123456789abcdef")
|
||
|
iv := []byte("abcdefg123456789")
|
||
|
c, err := NewCipher(GNUTLS_CIPHER_AES_128_CBC, key, iv)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
defer c.Close()
|
||
|
|
||
|
c1, err := NewCipher(GNUTLS_CIPHER_AES_128_CBC, key, iv)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
defer c1.Close()
|
||
|
|
||
|
data := []byte("1234012121212121")
|
||
|
if c == nil {
|
||
|
t.Fatal("new ciphoer failed")
|
||
|
}
|
||
|
cdata, err := c.Encrypt(data)
|
||
|
if err != nil {
|
||
|
t.Fatal("encrypt failed", err)
|
||
|
}
|
||
|
data1, err := c1.Decrypt(cdata)
|
||
|
if err != nil {
|
||
|
t.Fatal("decrypt failed", err)
|
||
|
}
|
||
|
if !bytes.Equal(data, data1) {
|
||
|
t.Fatal("encrypt/decrypt failed", string(data), string(data1))
|
||
|
}
|
||
|
block, _ := aes.NewCipher(key)
|
||
|
mode := cipher.NewCBCEncrypter(block, iv)
|
||
|
dst := make([]byte, len(data))
|
||
|
mode.CryptBlocks(dst, data)
|
||
|
if !bytes.Equal(dst, cdata) {
|
||
|
t.Fatal("cipher text not equal to cypto/aes")
|
||
|
}
|
||
|
}
|