forked from TrueCloudLab/frostfs-node
Ekaterina Lebedeva
a685fcdc96
Since Go 1.22 a "for" statement with a "range" clause is able to iterate through integer values from zero to an upper limit. gopatch script: @@ var i, e expression @@ -for i := 0; i <= e - 1; i++ { +for i := range e { ... } @@ var i, e expression @@ -for i := 0; i <= e; i++ { +for i := range e + 1 { ... } @@ var i, e expression @@ -for i := 0; i < e; i++ { +for i := range e { ... } Signed-off-by: Ekaterina Lebedeva <ekaterina.lebedeva@yadro.com>
266 lines
6.9 KiB
Go
266 lines
6.9 KiB
Go
package contract
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"text/tabwriter"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-contract/nns"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/commonflags"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/constants"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/helper"
|
|
morphClient "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/invoker"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/management"
|
|
"github.com/nspcc-dev/neo-go/pkg/rpcclient/unwrap"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/vmstate"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const lastGlagoliticLetter = 41
|
|
|
|
type contractDumpInfo struct {
|
|
hash util.Uint160
|
|
name string
|
|
version string
|
|
}
|
|
|
|
func dumpContractHashes(cmd *cobra.Command, _ []string) error {
|
|
c, err := helper.GetN3Client(viper.GetViper())
|
|
if err != nil {
|
|
return fmt.Errorf("can't create N3 client: %w", err)
|
|
}
|
|
|
|
r := management.NewReader(invoker.New(c, nil))
|
|
cs, err := helper.GetContractByID(r, 1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
zone, _ := cmd.Flags().GetString(commonflags.CustomZoneFlag)
|
|
if zone != "" {
|
|
return dumpCustomZoneHashes(cmd, cs.Hash, zone, c)
|
|
}
|
|
|
|
infos := []contractDumpInfo{{name: constants.NNSContract, hash: cs.Hash}}
|
|
|
|
irSize := 0
|
|
for ; irSize < lastGlagoliticLetter; irSize++ {
|
|
ok, err := helper.NNSIsAvailable(c, cs.Hash, helper.GetAlphabetNNSDomain(irSize))
|
|
if err != nil {
|
|
return err
|
|
} else if ok {
|
|
break
|
|
}
|
|
}
|
|
|
|
bw := io.NewBufBinWriter()
|
|
|
|
if irSize != 0 {
|
|
bw.Reset()
|
|
for i := range irSize {
|
|
emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly,
|
|
helper.GetAlphabetNNSDomain(i),
|
|
int64(nns.TXT))
|
|
}
|
|
|
|
alphaRes, err := c.InvokeScript(bw.Bytes(), nil)
|
|
if err != nil {
|
|
return fmt.Errorf("can't fetch info from NNS: %w", err)
|
|
}
|
|
|
|
for i := range irSize {
|
|
info := contractDumpInfo{name: fmt.Sprintf("alphabet %d", i)}
|
|
if h, err := helper.ParseNNSResolveResult(alphaRes.Stack[i]); err == nil {
|
|
info.hash = h
|
|
}
|
|
infos = append(infos, info)
|
|
}
|
|
}
|
|
|
|
for _, ctrName := range constants.ContractList {
|
|
bw.Reset()
|
|
emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly,
|
|
helper.DomainOf(ctrName), int64(nns.TXT))
|
|
|
|
res, err := c.InvokeScript(bw.Bytes(), nil)
|
|
if err != nil {
|
|
return fmt.Errorf("can't fetch info from NNS: %w", err)
|
|
}
|
|
|
|
info := contractDumpInfo{name: ctrName}
|
|
if len(res.Stack) != 0 {
|
|
if h, err := helper.ParseNNSResolveResult(res.Stack[0]); err == nil {
|
|
info.hash = h
|
|
}
|
|
}
|
|
infos = append(infos, info)
|
|
}
|
|
|
|
fillContractVersion(cmd, c, infos)
|
|
printContractInfo(cmd, infos)
|
|
|
|
return nil
|
|
}
|
|
|
|
func dumpCustomZoneHashes(cmd *cobra.Command, nnsHash util.Uint160, zone string, c helper.Client) error {
|
|
const nnsMaxTokens = 100
|
|
|
|
inv := invoker.New(c, nil)
|
|
|
|
if !strings.HasPrefix(zone, ".") {
|
|
zone = "." + zone
|
|
}
|
|
|
|
var infos []contractDumpInfo
|
|
processItem := func(item stackitem.Item) {
|
|
bs, err := item.TryBytes()
|
|
if err != nil {
|
|
cmd.PrintErrf("Invalid NNS record: %v\n", err)
|
|
return
|
|
}
|
|
|
|
if !bytes.HasSuffix(bs, []byte(zone)) || bytes.HasPrefix(bs, []byte(morphClient.NNSGroupKeyName)) {
|
|
// Related https://github.com/nspcc-dev/neofs-contract/issues/316.
|
|
return
|
|
}
|
|
|
|
h, err := helper.NNSResolveHash(inv, nnsHash, string(bs))
|
|
if err != nil {
|
|
cmd.PrintErrf("Could not resolve name %s: %v\n", string(bs), err)
|
|
return
|
|
}
|
|
|
|
infos = append(infos, contractDumpInfo{
|
|
hash: h,
|
|
name: strings.TrimSuffix(string(bs), zone),
|
|
})
|
|
}
|
|
|
|
script, err := smartcontract.CreateCallAndPrefetchIteratorScript(nnsHash, "tokens", nnsMaxTokens)
|
|
if err != nil {
|
|
return fmt.Errorf("create prefetch script: %w", err)
|
|
}
|
|
|
|
arr, sessionID, iter, err := unwrap.ArrayAndSessionIterator(inv.Run(script))
|
|
if err != nil {
|
|
if errors.Is(err, unwrap.ErrNoSessionID) {
|
|
items, err := unwrap.Array(inv.CallAndExpandIterator(nnsHash, "tokens", nnsMaxTokens))
|
|
if err != nil {
|
|
return fmt.Errorf("can't get a list of NNS domains: %w", err)
|
|
}
|
|
if len(items) == nnsMaxTokens {
|
|
cmd.PrintErrln("Provided RPC endpoint doesn't support sessions, some hashes might be lost.")
|
|
}
|
|
for i := range items {
|
|
processItem(items[i])
|
|
}
|
|
} else {
|
|
return err
|
|
}
|
|
} else {
|
|
for i := range arr {
|
|
processItem(arr[i])
|
|
}
|
|
|
|
defer func() {
|
|
_ = inv.TerminateSession(sessionID)
|
|
}()
|
|
|
|
items, err := inv.TraverseIterator(sessionID, &iter, 0)
|
|
for err == nil && len(items) != 0 {
|
|
for i := range items {
|
|
processItem(items[i])
|
|
}
|
|
items, err = inv.TraverseIterator(sessionID, &iter, 0)
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("error during NNS domains iteration: %w", err)
|
|
}
|
|
}
|
|
|
|
fillContractVersion(cmd, c, infos)
|
|
printContractInfo(cmd, infos)
|
|
|
|
return nil
|
|
}
|
|
|
|
func parseContractVersion(item stackitem.Item) string {
|
|
bi, err := item.TryInteger()
|
|
if err != nil || bi.Sign() == 0 || !bi.IsInt64() {
|
|
return "unknown"
|
|
}
|
|
|
|
v := bi.Int64()
|
|
major := v / 1_000_000
|
|
minor := (v % 1_000_000) / 1000
|
|
patch := v % 1_000
|
|
return fmt.Sprintf("v%d.%d.%d", major, minor, patch)
|
|
}
|
|
|
|
func printContractInfo(cmd *cobra.Command, infos []contractDumpInfo) {
|
|
if len(infos) == 0 {
|
|
return
|
|
}
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
tw := tabwriter.NewWriter(buf, 0, 2, 2, ' ', 0)
|
|
for _, info := range infos {
|
|
if info.version == "" {
|
|
info.version = "unknown"
|
|
}
|
|
_, _ = tw.Write([]byte(fmt.Sprintf("%s\t(%s):\t%s\n",
|
|
info.name, info.version, info.hash.StringLE())))
|
|
}
|
|
_ = tw.Flush()
|
|
|
|
cmd.Print(buf.String())
|
|
}
|
|
|
|
func fillContractVersion(cmd *cobra.Command, c helper.Client, infos []contractDumpInfo) {
|
|
bw := io.NewBufBinWriter()
|
|
sub := io.NewBufBinWriter()
|
|
for i := range infos {
|
|
if infos[i].hash.Equals(util.Uint160{}) {
|
|
emit.Int(bw.BinWriter, 0)
|
|
} else {
|
|
sub.Reset()
|
|
emit.AppCall(sub.BinWriter, infos[i].hash, "version", callflag.NoneFlag)
|
|
if sub.Err != nil {
|
|
panic(fmt.Errorf("BUG: can't create version script: %w", bw.Err))
|
|
}
|
|
|
|
script := sub.Bytes()
|
|
emit.Instruction(bw.BinWriter, opcode.TRY, []byte{byte(3 + len(script) + 2), 0})
|
|
bw.BinWriter.WriteBytes(script)
|
|
emit.Instruction(bw.BinWriter, opcode.ENDTRY, []byte{2 + 1})
|
|
emit.Opcodes(bw.BinWriter, opcode.PUSH0)
|
|
}
|
|
}
|
|
emit.Opcodes(bw.BinWriter, opcode.NOP) // for the last ENDTRY target
|
|
if bw.Err != nil {
|
|
panic(fmt.Errorf("BUG: can't create version script: %w", bw.Err))
|
|
}
|
|
|
|
res, err := c.InvokeScript(bw.Bytes(), nil)
|
|
if err != nil {
|
|
cmd.Printf("Can't fetch version from NNS: %v\n", err)
|
|
return
|
|
}
|
|
|
|
if res.State == vmstate.Halt.String() {
|
|
for i := range res.Stack {
|
|
infos[i].version = parseContractVersion(res.Stack[i])
|
|
}
|
|
}
|
|
}
|