mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-03-30 15:08:33 +00:00
08749deaeb
* [NOD-1538] Correct messages.proto. * [NOD-1538] Fix invalid message handling. * [NOD-1538] Fix mempool not wrapping consensus errors. * [NOD-1538] Extract wrapping logic to a separate function. * [NOD-1538] Extract wrapping logic to an even better separate function.
111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package protowire
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/hashes"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/subnetworks"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/transactionid"
|
|
"math"
|
|
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/util/mstime"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func (x *Hash) toDomain() (*externalapi.DomainHash, error) {
|
|
return hashes.FromBytes(x.Bytes)
|
|
}
|
|
|
|
func protoHashesToDomain(protoHashes []*Hash) ([]*externalapi.DomainHash, error) {
|
|
domainHashes := make([]*externalapi.DomainHash, len(protoHashes))
|
|
for i, protoHash := range protoHashes {
|
|
var err error
|
|
domainHashes[i], err = protoHash.toDomain()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return domainHashes, nil
|
|
}
|
|
|
|
func domainHashToProto(hash *externalapi.DomainHash) *Hash {
|
|
return &Hash{
|
|
Bytes: hash[:],
|
|
}
|
|
}
|
|
|
|
func domainHashesToProto(hashes []*externalapi.DomainHash) []*Hash {
|
|
protoHashes := make([]*Hash, len(hashes))
|
|
for i, hash := range hashes {
|
|
protoHashes[i] = domainHashToProto(hash)
|
|
}
|
|
return protoHashes
|
|
}
|
|
|
|
func (x *TransactionId) toDomain() (*externalapi.DomainTransactionID, error) {
|
|
return transactionid.FromBytes(x.Bytes)
|
|
}
|
|
|
|
func protoTransactionIDsToDomain(protoIDs []*TransactionId) ([]*externalapi.DomainTransactionID, error) {
|
|
txIDs := make([]*externalapi.DomainTransactionID, len(protoIDs))
|
|
for i, protoID := range protoIDs {
|
|
var err error
|
|
txIDs[i], err = protoID.toDomain()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return txIDs, nil
|
|
}
|
|
|
|
func domainTransactionIDToProto(id *externalapi.DomainTransactionID) *TransactionId {
|
|
return &TransactionId{
|
|
Bytes: id[:],
|
|
}
|
|
}
|
|
|
|
func wireTransactionIDsToProto(ids []*externalapi.DomainTransactionID) []*TransactionId {
|
|
protoIDs := make([]*TransactionId, len(ids))
|
|
for i, hash := range ids {
|
|
protoIDs[i] = domainTransactionIDToProto(hash)
|
|
}
|
|
return protoIDs
|
|
}
|
|
|
|
func (x *SubnetworkId) toDomain() (*externalapi.DomainSubnetworkID, error) {
|
|
if x == nil {
|
|
return nil, nil
|
|
}
|
|
return subnetworks.FromBytes(x.Bytes)
|
|
}
|
|
|
|
func domainSubnetworkIDToProto(id *externalapi.DomainSubnetworkID) *SubnetworkId {
|
|
if id == nil {
|
|
return nil
|
|
}
|
|
return &SubnetworkId{
|
|
Bytes: id[:],
|
|
}
|
|
}
|
|
|
|
func (x *NetAddress) toAppMessage() (*appmessage.NetAddress, error) {
|
|
if x.Port > math.MaxUint16 {
|
|
return nil, errors.Errorf("port number is larger than %d", math.MaxUint16)
|
|
}
|
|
return &appmessage.NetAddress{
|
|
Timestamp: mstime.UnixMilliseconds(x.Timestamp),
|
|
Services: appmessage.ServiceFlag(x.Services),
|
|
IP: x.Ip,
|
|
Port: uint16(x.Port),
|
|
}, nil
|
|
}
|
|
|
|
func appMessageNetAddressToProto(address *appmessage.NetAddress) *NetAddress {
|
|
return &NetAddress{
|
|
Timestamp: address.Timestamp.UnixMilliseconds(),
|
|
Services: uint64(address.Services),
|
|
Ip: address.IP,
|
|
Port: uint32(address.Port),
|
|
}
|
|
}
|