From e2910a7cb431c42f7d943c0183626783a231a3b5 Mon Sep 17 00:00:00 2001
From: Evgeniy Stratonikov <evgeniy@nspcc.ru>
Date: Thu, 12 Aug 2021 16:39:46 +0300
Subject: [PATCH] vm/cli: add public key -> address conversion, fix #2121

Signed-off-by: Evgeniy Stratonikov <evgeniy@nspcc.ru>
---
 docs/cli.md            |  2 +-
 pkg/vm/cli/cli.go      |  8 ++++++++
 pkg/vm/cli/cli_test.go | 13 +++++++++++++
 3 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/docs/cli.md b/docs/cli.md
index 6ca7e521a..4a95a0e59 100644
--- a/docs/cli.md
+++ b/docs/cli.md
@@ -622,7 +622,7 @@ distribution.
 ## Conversion utility
 
 NeoGo provides conversion utility command to reverse data, convert script
-hashes to/from address, convert data to/from hexadecimal or base64
+hashes to/from address, convert public keys to hashes/addresses, convert data to/from hexadecimal or base64
 representation. All of this is done by a single `util convert` command like
 this:
 ```
diff --git a/pkg/vm/cli/cli.go b/pkg/vm/cli/cli.go
index 520c247b2..44fe6a90d 100644
--- a/pkg/vm/cli/cli.go
+++ b/pkg/vm/cli/cli.go
@@ -2,6 +2,7 @@ package cli
 
 import (
 	"bytes"
+	"crypto/elliptic"
 	"encoding/base64"
 	"encoding/hex"
 	"encoding/json"
@@ -17,6 +18,7 @@ import (
 	"github.com/abiosoft/ishell/v2"
 	"github.com/abiosoft/readline"
 	"github.com/nspcc-dev/neo-go/pkg/compiler"
+	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
 	"github.com/nspcc-dev/neo-go/pkg/encoding/address"
 	"github.com/nspcc-dev/neo-go/pkg/encoding/bigint"
 	"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
@@ -564,6 +566,12 @@ func Parse(args []string) (string, error) {
 			buf.WriteString(fmt.Sprintf("BE ScriptHash to Address\t%s\n", address.Uint160ToString(val)))
 			buf.WriteString(fmt.Sprintf("LE ScriptHash to Address\t%s\n", address.Uint160ToString(val.Reverse())))
 		}
+		if pub, err := keys.NewPublicKeyFromBytes(rawStr, elliptic.P256()); err == nil {
+			sh := pub.GetScriptHash()
+			buf.WriteString(fmt.Sprintf("Public key to BE ScriptHash\t%s\n", sh))
+			buf.WriteString(fmt.Sprintf("Public key to LE ScriptHash\t%s\n", sh.Reverse()))
+			buf.WriteString(fmt.Sprintf("Public key to Address\t%s\n", address.Uint160ToString(sh)))
+		}
 		buf.WriteString(fmt.Sprintf("Hex to String\t%s\n", fmt.Sprintf("%q", string(rawStr))))
 		buf.WriteString(fmt.Sprintf("Hex to Integer\t%s\n", bigint.FromBytes(rawStr)))
 		buf.WriteString(fmt.Sprintf("Swap Endianness\t%s\n", hex.EncodeToString(slice.CopyReverse(rawStr))))
diff --git a/pkg/vm/cli/cli_test.go b/pkg/vm/cli/cli_test.go
index 73ad1196c..6dba3b5b2 100644
--- a/pkg/vm/cli/cli_test.go
+++ b/pkg/vm/cli/cli_test.go
@@ -500,6 +500,19 @@ func TestParse(t *testing.T) {
 		e.checkNextLine(t, "String to Hex.*30303030303030303030303030303030303030303030303030303030303030303030343434333432")
 		e.checkNextLine(t, "String to Base64.*MDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDQ0NDM0Mg==")
 	})
+	t.Run("public key", func(t *testing.T) {
+		pub := "02b3622bf4017bdfe317c58aed5f4c753f206b7db896046fa7d774bbc4bf7f8dc2"
+		e := newTestVMCLI(t)
+		e.runProg(t, "parse "+pub)
+		e.checkNextLine(t, "Public key to BE ScriptHash.*ee9ea22c27e34bd0148fc4108e08f74e8f5048b2")
+		e.checkNextLine(t, "Public key to LE ScriptHash.*b248508f4ef7088e10c48f14d04be3272ca29eee")
+		e.checkNextLine(t, "Public key to Address.*Nhfg3TbpwogLvDGVvAvqyThbsHgoSUKwtn")
+		e.checkNextLine(t, "Hex to String")
+		e.checkNextLine(t, "Hex to Integer.*-7115107707948693452214836319400158580475150561081357074343221218306172781415678")
+		e.checkNextLine(t, "Swap Endianness.*c28d7fbfc4bb74d7a76f0496b87d6b203f754c5fed8ac517e3df7b01f42b62b302")
+		e.checkNextLine(t, "String to Hex.*303262333632326266343031376264666533313763353861656435663463373533663230366237646238393630343666613764373734626263346266376638646332")
+		e.checkNextLine(t, "String to Base64.*MDJiMzYyMmJmNDAxN2JkZmUzMTdjNThhZWQ1ZjRjNzUzZjIwNmI3ZGI4OTYwNDZmYTdkNzc0YmJjNGJmN2Y4ZGMy")
+	})
 }
 
 func TestPrintLogo(t *testing.T) {