package morph import ( "bytes" "errors" "fmt" "strings" "text/tabwriter" "git.frostfs.info/TrueCloudLab/frostfs-contract/nns" morphUtil "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/util" 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 := morphUtil.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 := r.GetContractByID(1) if err != nil { return err } zone, _ := cmd.Flags().GetString(customZoneFlag) if zone != "" { return dumpCustomZoneHashes(cmd, cs.Hash, zone, c) } infos := []contractDumpInfo{{name: nnsContract, hash: cs.Hash}} irSize := 0 for ; irSize < lastGlagoliticLetter; irSize++ { ok, err := nnsIsAvailable(c, cs.Hash, getAlphabetNNSDomain(irSize)) if err != nil { return err } else if ok { break } } bw := io.NewBufBinWriter() if irSize != 0 { bw.Reset() for i := 0; i < irSize; i++ { emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly, 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 := 0; i < irSize; i++ { info := contractDumpInfo{name: fmt.Sprintf("alphabet %d", i)} if h, err := parseNNSResolveResult(alphaRes.Stack[i]); err == nil { info.hash = h } infos = append(infos, info) } } for _, ctrName := range contractList { bw.Reset() emit.AppCall(bw.BinWriter, cs.Hash, "resolve", callflag.ReadOnly, 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 := 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 morphUtil.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 := 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, nnsMaxTokens) for err == nil && len(items) != 0 { for i := range items { processItem(items[i]) } items, err = inv.TraverseIterator(sessionID, &iter, nnsMaxTokens) } 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 morphUtil.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]) } } }