61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package bson
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/keep-network/paillier"
|
|
"gopkg.in/mgo.v2/bson"
|
|
)
|
|
|
|
type SerializablePublicKey paillier.PublicKey
|
|
|
|
// Serializes PublicKey to BSON
|
|
func SerializePublicKey(publicKey *paillier.PublicKey) ([]byte, error) {
|
|
return bson.Marshal(toSerializablePublicKey(publicKey))
|
|
}
|
|
|
|
// Deserializes BSON to PublicKey
|
|
func DeserializePublicKey(data []byte) (*paillier.PublicKey, error) {
|
|
serializable := new(SerializablePublicKey)
|
|
if err := bson.Unmarshal(data, serializable); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return toOriginalPublicKey(serializable), nil
|
|
}
|
|
|
|
func toSerializablePublicKey(publicKey *paillier.PublicKey) *SerializablePublicKey {
|
|
serializable := SerializablePublicKey(*publicKey)
|
|
return &serializable
|
|
}
|
|
|
|
func toOriginalPublicKey(serializable *SerializablePublicKey) *paillier.PublicKey {
|
|
original := paillier.PublicKey(*serializable)
|
|
return &original
|
|
}
|
|
|
|
type dbPublicKey struct {
|
|
N string `bson:",omitempty"`
|
|
}
|
|
|
|
func (publicKey *SerializablePublicKey) GetBSON() (interface{}, error) {
|
|
m := make(map[string]string)
|
|
m["n"] = fmt.Sprintf("%x", publicKey.N)
|
|
return m, nil
|
|
}
|
|
|
|
func (publicKey *SerializablePublicKey) SetBSON(raw bson.Raw) error {
|
|
var err error = nil
|
|
c := new(dbPublicKey)
|
|
raw.Unmarshal(c)
|
|
|
|
if c.N != "" {
|
|
publicKey.N, err = fromHex(c.N)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|