[#1748] neofs-adm: Allow to dump hashes from a custom zone

Signed-off-by: Evgenii Stratonikov <evgeniy@morphbits.ru>

cmd/neofs-adm/internal/modules/morph/dump_hashes.go

@@ -3,14 +3,21 @@ package morph
 import (
 	"bytes"
 	"fmt"
+	"strings"
 	"text/tabwriter"
 
+	"github.com/nspcc-dev/neo-go/pkg/core/transaction"
 	"github.com/nspcc-dev/neo-go/pkg/io"
+	"github.com/nspcc-dev/neo-go/pkg/rpcclient"
+	"github.com/nspcc-dev/neo-go/pkg/rpcclient/actor"
+	"github.com/nspcc-dev/neo-go/pkg/rpcclient/unwrap"
 	"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/nspcc-dev/neo-go/pkg/wallet"
 	"github.com/nspcc-dev/neofs-contract/nns"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -35,6 +42,11 @@ func dumpContractHashes(cmd *cobra.Command, _ []string) error {
 		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
@@ -47,7 +59,6 @@ func dumpContractHashes(cmd *cobra.Command, _ []string) error {
 		}
 	}
 
-	buf := bytes.NewBuffer(nil)
 	bw := io.NewBufBinWriter()
 
 	if irSize != 0 {
@@ -91,37 +102,65 @@ func dumpContractHashes(cmd *cobra.Command, _ []string) error {
 		infos = append(infos, info)
 	}
 
-	bw.Reset()
+	fillContractVersion(cmd, c, infos)
-	for i := range infos {
+	printContractInfo(cmd, infos)
-		if infos[i].hash.Equals(util.Uint160{}) {
-			emit.Int(bw.BinWriter, 0)
-		} else {
-			emit.AppCall(bw.BinWriter, infos[i].hash, "version", callflag.NoneFlag)
-		}
-	}
 
-	res, err := c.InvokeScript(bw.Bytes(), nil)
+	return nil
+}
+
+func dumpCustomZoneHashes(cmd *cobra.Command, nnsHash util.Uint160, zone string, c Client) error {
+	const nnsMaxTokens = 100
+
+	// The actual signer is not important here.
+	account, err := wallet.NewAccount()
 	if err != nil {
-		return fmt.Errorf("can't fetch info from NNS: %w", err)
+		return fmt.Errorf("can't create a temporary account: %w", err)
 	}
 
-	if res.State == vmstate.Halt.String() {
+	signers := []actor.SignerAccount{{
-		for i := range res.Stack {
+		Signer: transaction.Signer{
-			infos[i].version = parseContractVersion(res.Stack[i])
+			Account: account.PrivateKey().GetScriptHash(),
+		},
+		Account: account,
+	}}
+	a, err := actor.New(c.(*rpcclient.Client), signers)
+	if err != nil {
+		return fmt.Errorf("can't get a list of NNS domains: %w", err)
+	}
+
+	arr, err := unwrap.Array(a.CallAndExpandIterator(nnsHash, "tokens", nnsMaxTokens))
+	if err != nil {
+		return fmt.Errorf("can't get a list of NNS domains: %w", err)
+	}
+
+	if !strings.HasPrefix(zone, ".") {
+		zone = "." + zone
+	}
+
+	var infos []contractDumpInfo
+	for i := range arr {
+		bs, err := arr[i].TryBytes()
+		if err != nil {
+			continue
 		}
-	}
 
-	tw := tabwriter.NewWriter(buf, 0, 2, 2, ' ', 0)
+		if !bytes.HasSuffix(bs, []byte(zone)) {
-	for _, info := range infos {
+			continue
-		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())
+		h, err := nnsResolveHash(c, nnsHash, string(bs))
+		if err != nil {
+			continue
+		}
+
+		infos = append(infos, contractDumpInfo{
+			hash: h,
+			name: strings.TrimSuffix(string(bs), zone),
+		})
+	}
+
+	fillContractVersion(cmd, c, infos)
+	printContractInfo(cmd, infos)
 
 	return nil
 }
@@ -138,3 +177,60 @@ func parseContractVersion(item stackitem.Item) string {
 	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 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])
+		}
+	}
+}

cmd/neofs-adm/internal/modules/morph/root.go

@@ -258,6 +258,7 @@ func init() {
 
 	RootCmd.AddCommand(dumpContractHashesCmd)
 	dumpContractHashesCmd.Flags().StringP(endpointFlag, "r", "", "N3 RPC node endpoint")
+	dumpContractHashesCmd.Flags().String(customZoneFlag, "", "Custom zone to search.")
 
 	RootCmd.AddCommand(dumpNetworkConfigCmd)
 	dumpNetworkConfigCmd.Flags().StringP(endpointFlag, "r", "", "N3 RPC node endpoint")