diff options
| -rw-r--r-- | core/crypto/dkg/dkg.go | 14 | ||||
| -rw-r--r-- | core/crypto/dkg/dkg_test.go | 12 |
2 files changed, 26 insertions, 0 deletions
diff --git a/core/crypto/dkg/dkg.go b/core/crypto/dkg/dkg.go index 8d0e088..8ca511d 100644 --- a/core/crypto/dkg/dkg.go +++ b/core/crypto/dkg/dkg.go @@ -58,6 +58,20 @@ type PrivateKey struct { publicKey PublicKey } +// EncodeRLP implements rlp.Encoder +func (prv *PrivateKey) EncodeRLP(w io.Writer) error { + return rlp.Encode(w, prv.Bytes()) +} + +// DecodeRLP implements rlp.Decoder +func (prv *PrivateKey) DecodeRLP(s *rlp.Stream) error { + var b []byte + if err := s.Decode(&b); err != nil { + return err + } + return prv.SetBytes(b) +} + // MarshalJSON implements json.Marshaller. func (prv *PrivateKey) MarshalJSON() ([]byte, error) { return json.Marshal(&prv.privateKey) diff --git a/core/crypto/dkg/dkg_test.go b/core/crypto/dkg/dkg_test.go index fe7ec73..44e68d8 100644 --- a/core/crypto/dkg/dkg_test.go +++ b/core/crypto/dkg/dkg_test.go @@ -292,6 +292,18 @@ func (s *DKGTestSuite) TestSignature() { s.False(pubKey.VerifySignature(hash, sig)) } +func (s *DKGTestSuite) TestPrivateKeyRLPEncodeDecode() { + k := NewPrivateKey() + b, err := rlp.EncodeToBytes(k) + s.Require().NoError(err) + + var kk PrivateKey + err = rlp.DecodeBytes(b, &kk) + s.Require().NoError(err) + + s.Require().True(reflect.DeepEqual(*k, kk)) +} + func (s *DKGTestSuite) TestPublicKeySharesRLPEncodeDecode() { p := NewEmptyPublicKeyShares() for i, id := range s.genID(1) { |