Do not export internal ThresholdKey functions

This commit is contained in:
Piotr Dyraga committed 2018-05-30 14:48:28 +02:00
1 parent e25bc358e8
commit adcb812a08
2 files changed
+26 -28

No files matched your search

+15 -17
View File
@@ -17,17 +17,17 @@ type ThresholdKey struct {
} }
// returns the value of (4*delta**2)** -1 mod n // returns the value of (4*delta**2)** -1 mod n
func (this *ThresholdKey) CombineSharesConstant() *big.Int { func (this *ThresholdKey) combineSharesConstant() *big.Int {
tmp := new(big.Int).Mul(FOUR, new(big.Int).Mul(this.Delta(), this.Delta())) tmp := new(big.Int).Mul(FOUR, new(big.Int).Mul(this.delta(), this.delta()))
return (&big.Int{}).ModInverse(tmp, this.N) return (&big.Int{}).ModInverse(tmp, this.N)
} }
// returns the factorial of the number of TotalNumberOfDecryptionServers // returns the factorial of the number of TotalNumberOfDecryptionServers
func (this *ThresholdKey) Delta() *big.Int { func (this *ThresholdKey) delta() *big.Int {
return Factorial(this.TotalNumberOfDecryptionServers) return Factorial(this.TotalNumberOfDecryptionServers)
} }
func (this *ThresholdKey) MakeVerificationBeforeCombiningPartialDecryptions(shares []*PartialDecryption) error { func (this *ThresholdKey) makeVerificationBeforeCombiningPartialDecryptions(shares []*PartialDecryption) error {
if len(shares) < this.Threshold { if len(shares) < this.Threshold {
return errors.New("Threshold not meet") return errors.New("Threshold not meet")
} }
@@ -41,17 +41,17 @@ func (this *ThresholdKey) MakeVerificationBeforeCombiningPartialDecryptions(shar
return nil return nil
} }
func (this *ThresholdKey) UpdateLambda(share1, share2 *PartialDecryption, lambda *big.Int) *big.Int { func (this *ThresholdKey) updateLambda(share1, share2 *PartialDecryption, lambda *big.Int) *big.Int {
num := new(big.Int).Mul(lambda, big.NewInt(int64(-share2.Id))) num := new(big.Int).Mul(lambda, big.NewInt(int64(-share2.Id)))
denom := big.NewInt(int64(share1.Id - share2.Id)) denom := big.NewInt(int64(share1.Id - share2.Id))
return new(big.Int).Div(num, denom) return new(big.Int).Div(num, denom)
} }
func (this *ThresholdKey) ComputeLambda(share *PartialDecryption, shares []*PartialDecryption) *big.Int { func (this *ThresholdKey) computeLambda(share *PartialDecryption, shares []*PartialDecryption) *big.Int {
lambda := this.Delta() lambda := this.delta()
for _, share2 := range shares { for _, share2 := range shares {
if share2.Id != share.Id { if share2.Id != share.Id {
lambda = this.UpdateLambda(share, share2, lambda) lambda = this.updateLambda(share, share2, lambda)
} }
} }
return lambda return lambda
@@ -85,17 +85,17 @@ func (this *ThresholdKey) exp(a, b, c *big.Int) *big.Int {
func (this *ThresholdKey) computeDecryption(cprime *big.Int) *big.Int { func (this *ThresholdKey) computeDecryption(cprime *big.Int) *big.Int {
l := L(cprime, this.N) l := L(cprime, this.N)
return new(big.Int).Mod(new(big.Int).Mul(this.CombineSharesConstant(), l), this.N) return new(big.Int).Mod(new(big.Int).Mul(this.combineSharesConstant(), l), this.N)
} }
func (this *ThresholdKey) CombinePartialDecryptions(shares []*PartialDecryption) (*big.Int, error) { func (this *ThresholdKey) CombinePartialDecryptions(shares []*PartialDecryption) (*big.Int, error) {
if err := this.MakeVerificationBeforeCombiningPartialDecryptions(shares); err != nil { if err := this.makeVerificationBeforeCombiningPartialDecryptions(shares); err != nil {
return nil, err return nil, err
} }
cprime := ONE cprime := ONE
for _, share := range shares { for _, share := range shares {
lambda := this.ComputeLambda(share, shares) lambda := this.computeLambda(share, shares)
cprime = this.updateCprime(cprime, lambda, share) cprime = this.updateCprime(cprime, lambda, share)
} }
@@ -143,7 +143,7 @@ type ThresholdPrivateKey struct {
func (this *ThresholdPrivateKey) Decrypt(c *big.Int) *PartialDecryption { func (this *ThresholdPrivateKey) Decrypt(c *big.Int) *PartialDecryption {
ret := new(PartialDecryption) ret := new(PartialDecryption)
ret.Id = this.Id ret.Id = this.Id
exp := new(big.Int).Mul(this.Share, new(big.Int).Mul(TWO, this.Delta())) exp := new(big.Int).Mul(this.Share, new(big.Int).Mul(TWO, this.delta()))
ret.Decryption = new(big.Int).Exp(c, exp, this.GetNSquare()) ret.Decryption = new(big.Int).Exp(c, exp, this.GetNSquare())
return ret return ret
@@ -167,11 +167,10 @@ func (this *ThresholdPrivateKey) GetThresholdKey() *ThresholdKey {
return ret return ret
} }
func (this *ThresholdPrivateKey) ComputeZ(r, e *big.Int) *big.Int { func (this *ThresholdPrivateKey) computeZ(r, e *big.Int) *big.Int {
tmp := new(big.Int).Mul(e, this.Delta()) tmp := new(big.Int).Mul(e, this.delta())
tmp = new(big.Int).Mul(tmp, this.Share) tmp = new(big.Int).Mul(tmp, this.Share)
return new(big.Int).Add(r, tmp) return new(big.Int).Add(r, tmp)
} }
func (this *ThresholdPrivateKey) computeHash(a, b, c4, ci2 *big.Int) *big.Int { func (this *ThresholdPrivateKey) computeHash(a, b, c4, ci2 *big.Int) *big.Int {
@@ -181,7 +180,6 @@ func (this *ThresholdPrivateKey) computeHash(a, b, c4, ci2 *big.Int) *big.Int {
hash.Write(c4.Bytes()) hash.Write(c4.Bytes())
hash.Write(ci2.Bytes()) hash.Write(ci2.Bytes())
return new(big.Int).SetBytes(hash.Sum([]byte{})) return new(big.Int).SetBytes(hash.Sum([]byte{}))
} }
func (this *ThresholdPrivateKey) DecryptAndProduceZNP(c *big.Int, random io.Reader) (*PartialDecryptionZKP, error) { func (this *ThresholdPrivateKey) DecryptAndProduceZNP(c *big.Int, random io.Reader) (*PartialDecryptionZKP, error) {
@@ -208,7 +206,7 @@ func (this *ThresholdPrivateKey) DecryptAndProduceZNP(c *big.Int, random io.Read
pd.E = this.computeHash(a, b, c4, ci2) pd.E = this.computeHash(a, b, c4, ci2)
pd.Z = this.ComputeZ(r, pd.E) pd.Z = this.computeZ(r, pd.E)
return pd, nil return pd, nil
} }
+11 -11
View File
@@ -7,7 +7,7 @@ import (
"testing" "testing"
) )
func GetThresholdPrivateKey() *ThresholdPrivateKey { func getThresholdPrivateKey() *ThresholdPrivateKey {
tkh := GetThresholdKeyGenerator(32, 10, 6, rand.Reader) tkh := GetThresholdKeyGenerator(32, 10, 6, rand.Reader)
tpks, err := tkh.Generate() tpks, err := tkh.Generate()
if err != nil { if err != nil {
@@ -19,7 +19,7 @@ func GetThresholdPrivateKey() *ThresholdPrivateKey {
func TestDelta(t *testing.T) { func TestDelta(t *testing.T) {
tk := new(ThresholdKey) tk := new(ThresholdKey)
tk.TotalNumberOfDecryptionServers = 6 tk.TotalNumberOfDecryptionServers = 6
if delta := tk.Delta(); 720 != n(delta) { if delta := tk.delta(); 720 != n(delta) {
t.Error("Delta is not 720 but", delta) t.Error("Delta is not 720 but", delta)
} }
} }
@@ -29,7 +29,7 @@ func TestCombineSharesConstant(t *testing.T) {
tk.N = big.NewInt(101 * 103) tk.N = big.NewInt(101 * 103)
tk.TotalNumberOfDecryptionServers = 6 tk.TotalNumberOfDecryptionServers = 6
if c := tk.CombineSharesConstant(); !reflect.DeepEqual(big.NewInt(4558), c) { if c := tk.combineSharesConstant(); !reflect.DeepEqual(big.NewInt(4558), c) {
t.Error("wrong combined key. ", c) t.Error("wrong combined key. ", c)
} }
} }
@@ -66,7 +66,7 @@ func TestCopyVi(t *testing.T) {
} }
func TestEncryptWithThresholdKey(t *testing.T) { func TestEncryptWithThresholdKey(t *testing.T) {
pd := GetThresholdPrivateKey() pd := getThresholdPrivateKey()
_, err := pd.Encrypt(big.NewInt(876), rand.Reader) _, err := pd.Encrypt(big.NewInt(876), rand.Reader)
if err != nil { if err != nil {
t.Fail() t.Fail()
@@ -74,7 +74,7 @@ func TestEncryptWithThresholdKey(t *testing.T) {
} }
func TestDecryptWithThresholdKey(t *testing.T) { func TestDecryptWithThresholdKey(t *testing.T) {
pd := GetThresholdPrivateKey() pd := getThresholdPrivateKey()
c, err := pd.Encrypt(big.NewInt(876), rand.Reader) c, err := pd.Encrypt(big.NewInt(876), rand.Reader)
if err != nil { if err != nil {
t.Fail() t.Fail()
@@ -111,7 +111,7 @@ func TestVerifyPart2(t *testing.T) {
} }
func TestDecryptAndProduceZNP(t *testing.T) { func TestDecryptAndProduceZNP(t *testing.T) {
pd := GetThresholdPrivateKey() pd := getThresholdPrivateKey()
c, err := pd.Encrypt(big.NewInt(876), rand.Reader) c, err := pd.Encrypt(big.NewInt(876), rand.Reader)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
@@ -131,16 +131,16 @@ func TestDecryptAndProduceZNP(t *testing.T) {
func TestMakeVerificationBeforeCombiningPartialDecryptions(t *testing.T) { func TestMakeVerificationBeforeCombiningPartialDecryptions(t *testing.T) {
tk := new(ThresholdKey) tk := new(ThresholdKey)
tk.Threshold = 2 tk.Threshold = 2
if tk.MakeVerificationBeforeCombiningPartialDecryptions([]*PartialDecryption{}) == nil { if tk.makeVerificationBeforeCombiningPartialDecryptions([]*PartialDecryption{}) == nil {
t.Fail() t.Fail()
} }
prms := []*PartialDecryption{new(PartialDecryption), new(PartialDecryption)} prms := []*PartialDecryption{new(PartialDecryption), new(PartialDecryption)}
prms[1].Id = 1 prms[1].Id = 1
if tk.MakeVerificationBeforeCombiningPartialDecryptions(prms) != nil { if tk.makeVerificationBeforeCombiningPartialDecryptions(prms) != nil {
t.Fail() t.Fail()
} }
prms[1].Id = 0 prms[1].Id = 0
if tk.MakeVerificationBeforeCombiningPartialDecryptions(prms) == nil { if tk.makeVerificationBeforeCombiningPartialDecryptions(prms) == nil {
t.Fail() t.Fail()
} }
} }
@@ -150,7 +150,7 @@ func TestUpdateLambda(t *testing.T) {
lambda := b(11) lambda := b(11)
share1 := &PartialDecryption{3, b(5)} share1 := &PartialDecryption{3, b(5)}
share2 := &PartialDecryption{7, b(3)} share2 := &PartialDecryption{7, b(3)}
res := tk.UpdateLambda(share1, share2, lambda) res := tk.updateLambda(share1, share2, lambda)
if n(res) != 20 { if n(res) != 20 {
t.Error("wrong lambda", n(res)) t.Error("wrong lambda", n(res))
} }
@@ -250,7 +250,7 @@ func TestDivide(t *testing.T) {
} }
func TestValidate(t *testing.T) { func TestValidate(t *testing.T) {
pk := GetThresholdPrivateKey() pk := getThresholdPrivateKey()
if err := pk.Validate(rand.Reader); err != nil { if err := pk.Validate(rand.Reader); err != nil {
t.Error(err) t.Error(err)
} }