-
Notifications
You must be signed in to change notification settings - Fork 459
/
builtin_actors_bundle.go
82 lines (69 loc) · 2.28 KB
/
builtin_actors_bundle.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package actors
import (
"bytes"
"context"
"fmt"
"io"
"os"
actorstypes "github.com/filecoin-project/go-state-types/actors"
"github.com/filecoin-project/venus/venus-shared/blockstore"
cid "github.com/ipfs/go-cid"
"github.com/ipld/go-car"
)
func LoadBundleFromFile(ctx context.Context, bs blockstore.Blockstore, path string) (cid.Cid, error) {
f, err := os.Open(path)
if err != nil {
return cid.Undef, fmt.Errorf("error opening bundle %q for builtin-actors: %w", path, err)
}
defer f.Close() //nolint
return LoadBundle(ctx, bs, f)
}
func LoadBundle(ctx context.Context, bs blockstore.Blockstore, r io.Reader) (cid.Cid, error) {
hdr, err := car.LoadCar(ctx, bs, r)
if err != nil {
return cid.Undef, fmt.Errorf("error loading builtin actors bundle: %w", err)
}
if len(hdr.Roots) != 1 {
return cid.Undef, fmt.Errorf("expected one root when loading actors bundle, got %d", len(hdr.Roots))
}
return hdr.Roots[0], nil
}
// LoadBundles loads the bundles for the specified actor versions into the passed blockstore, if and
// only if the bundle's manifest is not already present in the blockstore.
func LoadBundles(ctx context.Context, bs blockstore.Blockstore, versions ...actorstypes.Version) error {
for _, av := range versions {
// No bundles before version 8.
if av < actorstypes.Version8 {
continue
}
manifestCid, ok := GetManifest(av)
if !ok {
// All manifests are registered on start, so this must succeed.
return fmt.Errorf("unknown actor version v%d", av)
}
if haveManifest, err := bs.Has(ctx, manifestCid); err != nil {
return fmt.Errorf("blockstore error when loading manifest %s: %w", manifestCid, err)
} else if haveManifest {
// We already have the manifest, and therefore everything under it.
continue
}
var (
root cid.Cid
err error
)
if path, ok := BundleOverrides[av]; ok {
root, err = LoadBundleFromFile(ctx, bs, path)
} else if embedded, ok := GetEmbeddedBuiltinActorsBundle(av); ok {
root, err = LoadBundle(ctx, bs, bytes.NewReader(embedded))
} else {
err = fmt.Errorf("bundle for actors version v%d not found", av)
}
if err != nil {
return err
}
if root != manifestCid {
return fmt.Errorf("expected manifest for actors version %d does not match actual: %s != %s", av, manifestCid, root)
}
}
return nil
}