Skip to content

Commit

Permalink
Add the pruning point proof to the wire package.
Browse files Browse the repository at this point in the history
  • Loading branch information
stasatdaglabs committed Sep 5, 2021
1 parent de36396 commit 1a66498
Show file tree
Hide file tree
Showing 9 changed files with 985 additions and 730 deletions.
1,427 changes: 734 additions & 693 deletions infrastructure/network/netadapter/server/grpcserver/protowire/messages.pb.go

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,8 @@ message KaspadMessage {
RequestHeadersMessage requestHeaders = 45;
RequestBlockLocatorMessage requestBlockLocator = 46;
PruningPointsMessage pruningPoints = 47;
RequestPruningPointProofMessage requestPruningPointProof = 48;
PruningPointProofMessage pruningPointProof = 49;

GetCurrentNetworkRequestMessage getCurrentNetworkRequest = 1001;
GetCurrentNetworkResponseMessage getCurrentNetworkResponse = 1002;
Expand Down

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

148 changes: 131 additions & 17 deletions infrastructure/network/netadapter/server/grpcserver/protowire/p2p.pb.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
Expand Up @@ -248,4 +248,11 @@ message BlockBlueWorkMessage {

message PruningPointsMessage {
repeated BlockHeader headers = 1;
}
}

message RequestPruningPointProofMessage {
}

message PruningPointProofMessage {
repeated BlockHeader headers = 1;
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
package protowire

import (
"github.com/kaspanet/kaspad/app/appmessage"
"github.com/pkg/errors"
)

func (x *KaspadMessage_PruningPointProof) toAppMessage() (appmessage.Message, error) {
if x == nil {
return nil, errors.Wrapf(errorNil, "KaspadMessage_PruningPointProof is nil")
}

if x.PruningPointProof == nil {
return nil, errors.Wrapf(errorNil, "x.PruningPointProof is nil")
}

blockHeaders := make([]*appmessage.MsgBlockHeader, len(x.PruningPointProof.Headers))
for i, blockHeader := range x.PruningPointProof.Headers {
var err error
blockHeaders[i], err = blockHeader.toAppMessage()
if err != nil {
return nil, err
}
}
return &appmessage.MsgPruningPointProof{
Headers: blockHeaders,
}, nil
}

func (x *KaspadMessage_PruningPointProof) fromAppMessage(msgPruningPointProof *appmessage.MsgPruningPointProof) error {
blockHeaders := make([]*BlockHeader, len(msgPruningPointProof.Headers))
for i, blockHeader := range msgPruningPointProof.Headers {
blockHeaders[i] = &BlockHeader{}
err := blockHeaders[i].fromAppMessage(blockHeader)
if err != nil {
return err
}
}

x.PruningPointProof = &PruningPointProofMessage{
Headers: blockHeaders,
}
return nil
}
Loading

0 comments on commit 1a66498

Please sign in to comment.