mirror of
https://github.com/ipfs/kubo.git
synced 2025-05-21 00:47:22 +08:00
f4d7369c4a
This change adds the /ipfs/bitswap/1.1.0 protocol. The new protocol adds a 'payload' field to the protobuf message and deprecates the existing 'blocks' field. The 'payload' field is an array of pairs of cid prefixes and block data. The cid prefixes are used to ensure the correct codecs and hash functions are used to handle the block on the receiving end. License: MIT Signed-off-by: Jeromy <why@ipfs.io>
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package merkledag
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
|
|
pb "github.com/ipfs/go-ipfs/merkledag/pb"
|
|
|
|
cid "gx/ipfs/QmXUuRadqDq5BuFWzVU6VuKaSjTcNm1gNCtLvvP1TJCW4z/go-cid"
|
|
mh "gx/ipfs/QmYDds3421prZgqKbLpEK7T9Aa2eVdQ7o3YarX1LVLdP2J/go-multihash"
|
|
u "gx/ipfs/Qmb912gdngC1UWwTkhuW8knyRbcWeu5kqkxBpveLmW8bSr/go-ipfs-util"
|
|
)
|
|
|
|
// for now, we use a PBNode intermediate thing.
|
|
// because native go objects are nice.
|
|
|
|
// unmarshal decodes raw data into a *Node instance.
|
|
// The conversion uses an intermediate PBNode.
|
|
func (n *Node) unmarshal(encoded []byte) error {
|
|
var pbn pb.PBNode
|
|
if err := pbn.Unmarshal(encoded); err != nil {
|
|
return fmt.Errorf("Unmarshal failed. %v", err)
|
|
}
|
|
|
|
pbnl := pbn.GetLinks()
|
|
n.Links = make([]*Link, len(pbnl))
|
|
for i, l := range pbnl {
|
|
n.Links[i] = &Link{Name: l.GetName(), Size: l.GetTsize()}
|
|
h, err := mh.Cast(l.GetHash())
|
|
if err != nil {
|
|
return fmt.Errorf("Link hash #%d is not valid multihash. %v", i, err)
|
|
}
|
|
n.Links[i].Hash = h
|
|
}
|
|
sort.Stable(LinkSlice(n.Links)) // keep links sorted
|
|
|
|
n.data = pbn.GetData()
|
|
n.encoded = encoded
|
|
return nil
|
|
}
|
|
|
|
// Marshal encodes a *Node instance into a new byte slice.
|
|
// The conversion uses an intermediate PBNode.
|
|
func (n *Node) Marshal() ([]byte, error) {
|
|
pbn := n.getPBNode()
|
|
data, err := pbn.Marshal()
|
|
if err != nil {
|
|
return data, fmt.Errorf("Marshal failed. %v", err)
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func (n *Node) getPBNode() *pb.PBNode {
|
|
pbn := &pb.PBNode{}
|
|
if len(n.Links) > 0 {
|
|
pbn.Links = make([]*pb.PBLink, len(n.Links))
|
|
}
|
|
|
|
sort.Stable(LinkSlice(n.Links)) // keep links sorted
|
|
for i, l := range n.Links {
|
|
pbn.Links[i] = &pb.PBLink{}
|
|
pbn.Links[i].Name = &l.Name
|
|
pbn.Links[i].Tsize = &l.Size
|
|
pbn.Links[i].Hash = []byte(l.Hash)
|
|
}
|
|
|
|
if len(n.data) > 0 {
|
|
pbn.Data = n.data
|
|
}
|
|
return pbn
|
|
}
|
|
|
|
// EncodeProtobuf returns the encoded raw data version of a Node instance.
|
|
// It may use a cached encoded version, unless the force flag is given.
|
|
func (n *Node) EncodeProtobuf(force bool) ([]byte, error) {
|
|
sort.Stable(LinkSlice(n.Links)) // keep links sorted
|
|
if n.encoded == nil || force {
|
|
n.cached = nil
|
|
var err error
|
|
n.encoded, err = n.Marshal()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
if n.cached == nil {
|
|
n.cached = cid.NewCidV0(u.Hash(n.encoded))
|
|
}
|
|
|
|
return n.encoded, nil
|
|
}
|
|
|
|
// Decoded decodes raw data and returns a new Node instance.
|
|
func DecodeProtobuf(encoded []byte) (*Node, error) {
|
|
n := new(Node)
|
|
err := n.unmarshal(encoded)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("incorrectly formatted merkledag node: %s", err)
|
|
}
|
|
return n, nil
|
|
}
|