logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-02 12:53:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

try to extract the key in order to decide whether to attach the key

Browse Source
This commit is contained in:
vyzo 2018-10-14 02:16:41 +03:00
parent bd15183543
commit 4dc796359e
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pubsub.go
View File

@ -677,7 +677,7 @@ func (p *PubSub) Publish(topic string, data []byte) error {
}
if p.signKey != nil {
m.From = []byte(p.signID)
err := signMessage(p.signKey, m)
err := signMessage(p.signID, p.signKey, m)
if err != nil {
return err
}

8
sign.go
View File

@ -68,7 +68,7 @@ func messagePubKey(m *pb.Message) (crypto.PubKey, error) {
return pubk, nil
}
func signMessage(key crypto.PrivKey, m *pb.Message) error {
func signMessage(pid peer.ID, key crypto.PrivKey, m *pb.Message) error {
bytes, err := m.Marshal()
if err != nil {
return err
@ -82,14 +82,16 @@ func signMessage(key crypto.PrivKey, m *pb.Message) error {
}
m.Signature = sig
switch key.Type() {
case crypto.RSA:
pk, _ := pid.ExtractPublicKey()
if pk == nil {
pubk, err := key.GetPublic().Bytes()
if err != nil {
return err
}
m.Key = pubk
}
return nil
}

2
sign_test.go
View File

@ -34,7 +34,7 @@ func testSignVerify(t *testing.T, privk crypto.PrivKey) {
From: []byte(id),
Seqno: []byte("123"),
}
signMessage(privk, &m)
signMessage(id, privk, &m)
err = verifyMessageSignature(&m)
if err != nil {
t.Fatal(err)