Merge pull request #7 from keep-network/cypher-multiplication

Added support for a Cypher multiplication

paillier.go

@@ -81,6 +81,19 @@ func (pk *PublicKey) Add(cypher ...*Cypher) *Cypher {
 	}
 }
 
+// Mul returns a product of `cypher` and `scalar` without decrypting `cypher`.
+//
+// It's possible because Paillier is a homomorphic encryption scheme, where
+// an encrypted plaintext `m` raised to an integer `k` will decrypt to the
+// product of the plaintext `m` and `k`:
+//
+// D( E(m)^k mod N^2 ) = km mod N
+func (pk *PublicKey) Mul(cypher *Cypher, scalar *big.Int) *Cypher {
+	return &Cypher{
+		C: new(big.Int).Exp(cypher.C, scalar, pk.GetNSquare()),
+	}
+}
+
 type PrivateKey struct {
 	PublicKey
 	Lambda *big.Int

paillier_test.go

@@ -3,7 +3,6 @@ package paillier
 import (
 	"crypto/rand"
 	"math/big"
-	"reflect"
 	"testing"
 )
 
@@ -14,7 +13,7 @@ func TestComputeL(t *testing.T) {
 	expected := big.NewInt(6)
 	actual := L(u, n)
 
-	if !reflect.DeepEqual(expected, actual) {
+	if expected.Cmp(actual) != 0 {
 		t.Errorf("Unexpected L function result [%v]", actual)
 	}
 }
@@ -26,7 +25,7 @@ func TestComputePhi(t *testing.T) {
 	expected := big.NewInt(24)
 	actual := computePhi(a, b)
 
-	if !reflect.DeepEqual(expected, actual) {
+	if expected.Cmp(actual) != 0 {
 		t.Errorf("Unexpected phi value [%v]", actual)
 	}
 }
@@ -37,11 +36,11 @@ func TestCreatePrivateKey(t *testing.T) {
 
 	privateKey := CreatePrivateKey(p, q)
 
-	if !reflect.DeepEqual(privateKey.N, big.NewInt(292153)) {
+	if privateKey.N.Cmp(big.NewInt(292153)) != 0 {
 		t.Errorf("Unexpected N PublicKey value [%v]", privateKey.N)
 	}
 
-	if !reflect.DeepEqual(privateKey.Lambda, big.NewInt(291060)) {
+	if privateKey.Lambda.Cmp(big.NewInt(291060)) != 0 {
 		t.Errorf("Unexpected Lambda Public key value [%v]", privateKey.Lambda)
 	}
 }
@@ -52,14 +51,14 @@ func TestEncryptDecryptSmall(t *testing.T) {
 	for i := 1; i < 10; i++ {
 		privateKey := CreatePrivateKey(p, q)
 
-		inicialValue := big.NewInt(100)
+		initialValue := big.NewInt(100)
-		cypher, err := privateKey.Encrypt(inicialValue, rand.Reader)
+		cypher, err := privateKey.Encrypt(initialValue, rand.Reader)
 		if err != nil {
 			t.Error(err)
 		}
 		returnedValue := privateKey.Decrypt(cypher)
-		if !reflect.DeepEqual(inicialValue, returnedValue) {
+		if initialValue.Cmp(returnedValue) != 0 {
-			t.Error("wrong decryption ", returnedValue, " is not ", inicialValue)
+			t.Error("wrong decryption ", returnedValue, " is not ", initialValue)
 		}
 	}
 }
@@ -74,7 +73,7 @@ func TestAddCyphers(t *testing.T) {
 	cypher5 := privateKey.Add(cypher1, cypher2, cypher3, cypher4)
 
 	m := privateKey.Decrypt(cypher5)
-	if !reflect.DeepEqual(m, big.NewInt(26)) {
+	if m.Cmp(big.NewInt(26)) != 0 {
 		t.Errorf("Unexpected decrypted value [%v]", m)
 	}
 }
@@ -88,7 +87,41 @@ func TestAddCypherWithSmallKeyModulus(t *testing.T) {
 	cypher4 := privateKey.Add(cypher1, cypher2, cypher3)
 
 	m := privateKey.Decrypt(cypher4)
-	if !reflect.DeepEqual(m, big.NewInt(34)) {
+	if m.Cmp(big.NewInt(34)) != 0 {
 		t.Errorf("Unexpected decrypted value [%v]", m)
 	}
 }
+
+func TestMulCypher(t *testing.T) {
+	privateKey := CreatePrivateKey(big.NewInt(17), big.NewInt(13))
+
+	cypher, err := privateKey.Encrypt(big.NewInt(3), rand.Reader)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cypherMultiple := privateKey.Mul(cypher, big.NewInt(7))
+	multiple := privateKey.Decrypt(cypherMultiple)
+
+	// 3 * 7 = 21
+	if multiple.Cmp(big.NewInt(21)) != 0 {
+		t.Errorf("Unexpected decrypted value [%v]", multiple)
+	}
+}
+
+func TestMulCypherWithSmallKeyModulus(t *testing.T) {
+	privateKey := CreatePrivateKey(big.NewInt(7), big.NewInt(5))
+
+	cypher, err := privateKey.Encrypt(big.NewInt(30), rand.Reader)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cypherMultiple := privateKey.Mul(cypher, big.NewInt(93))
+	multiple := privateKey.Decrypt(cypherMultiple)
+
+	// (30*93) mod (7*5) = 25
+	if multiple.Cmp(big.NewInt(25)) != 0 {
+		t.Errorf("Unexpected decrypted value [%v]", multiple)
+	}
+}