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.
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package service_test
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"crypto/rsa"
|
|
"git.cijber.net/zer.ooo/service"
|
|
"testing"
|
|
)
|
|
|
|
func TestPadding(t *testing.T) {
|
|
x := []byte{0, 0, 0}
|
|
x = append(x, service.Pad(x)...)
|
|
|
|
padLen := service.GetPad(x)
|
|
|
|
if padLen != 5 {
|
|
t.Errorf("Failed adding/measuring padding, Pad(%v, len=%d)", x, padLen)
|
|
}
|
|
|
|
y := service.Unpad(x)
|
|
if len(y) != 3 {
|
|
t.Errorf("Failed padding and unpadding, Pad(%v), Unpad(%v)", x, y)
|
|
}
|
|
}
|
|
|
|
func TestSingleRound(t *testing.T) {
|
|
keyA, _ := rsa.GenerateKey(rand.Reader, 4096)
|
|
keyB, _ := rsa.GenerateKey(rand.Reader, 4096)
|
|
signature := []byte("Hello world!")
|
|
v := &service.UpdateOpenVPNConfigRequest{
|
|
BaseMessage: &service.BaseMessage{""},
|
|
Config: "Hello world!",
|
|
}
|
|
x := make([]byte, 0)
|
|
b := bytes.NewBuffer(x)
|
|
err := service.EncryptAndSign(v, keyA, &keyB.PublicKey, signature, b)
|
|
|
|
if err != nil {
|
|
t.Errorf("Failed encrypting: %s", err)
|
|
}
|
|
|
|
ov := &service.UpdateOpenVPNConfigRequest{
|
|
BaseMessage: &service.BaseMessage{""},
|
|
}
|
|
err = service.DecryptAndVerify(b.Bytes(), keyB, &keyA.PublicKey, signature, ov)
|
|
|
|
if err != nil {
|
|
t.Errorf("Failed decrypting: %s", err)
|
|
}
|
|
|
|
if ov.Config != "Hello world!" {
|
|
t.Errorf("Config scrambled: %s", ov.Config)
|
|
}
|
|
}
|