fix: contstuction payloads (#190)

* add:system tests

* add: go.mod

* update: rename send_test

* update: own go mod

* block tests

* add: accounts, mempool and network tests

* some construction endpoints

* add: system-test workflow

* fix: lint

* update: workflows

* fix: lint

* fix: system tests

* fix: makefile rosetta-cli

* fix: payloads get signers

* add: payloads systemtest

* lint :)

* changelog + rabbit

* fix: package name

Author: JulianToledano

CHANGELOG.md

@@ -45,6 +45,10 @@ Ref: https://keepachangelog.com/en/1.0.0/
 * [180](https://github.com/cosmos/rosetta/pull/180) Update to cosmos-sdk v0.52.0-rc.1.
 * [180](https://github.com/cosmos/rosetta/pull/180) Added bech32 prefix flag.
 
+### Bug Fixes
+
+* [#190](https://github.com/cosmos/rosetta/pull/190) Fixed construction payloads to properly handle transaction signers.
+
 ## [v0.50.11](https://github.com/cosmos/rosetta/releases/tag/v0.50.11) 2024-12-19
 
 ### Improvements

client_online.go

@@ -97,7 +97,7 @@ func NewClient(cfg *Config) (*Client, error) {
 		bank:                nil,
 		tmRPC:               nil,
 		version:             fmt.Sprintf("%s/%s", info.AppName, v),
-		converter:           NewConverter(cfg.Codec, cfg.InterfaceRegistry, txConfig),
+		converter:           NewConverter(cfg.Codec, cfg.InterfaceRegistry, txConfig, address.NewBech32Codec(cfg.Bech32Prefix)),
 	}, nil
 }
 

converter.go

@@ -16,6 +16,7 @@ import (
 	secp "github.com/decred/dcrd/dcrec/secp256k1/v4"
 
 	signingv1beta1 "cosmossdk.io/api/cosmos/tx/signing/v1beta1"
+	"cosmossdk.io/core/address"
 	sdkmath "cosmossdk.io/math"
 	banktypes "cosmossdk.io/x/bank/types"
 
@@ -107,9 +108,10 @@ type converter struct {
 	bytesToSign     func(tx authsigning.Tx, signerData authsigning.SignerData) (b []byte, err error)
 	ir              codectypes.InterfaceRegistry
 	cdc             *codec.ProtoCodec
+	ac              address.Codec
 }
 
-func NewConverter(cdc *codec.ProtoCodec, ir codectypes.InterfaceRegistry, cfg sdkclient.TxConfig) Converter {
+func NewConverter(cdc *codec.ProtoCodec, ir codectypes.InterfaceRegistry, cfg sdkclient.TxConfig, ac address.Codec) Converter {
 	return converter{
 		newTxBuilder:    cfg.NewTxBuilder,
 		txBuilderFromTx: cfg.WrapTxBuilder,
@@ -131,6 +133,7 @@ func NewConverter(cdc *codec.ProtoCodec, ir codectypes.InterfaceRegistry, cfg sd
 		},
 		ir:  ir,
 		cdc: cdc,
+		ac:  ac,
 	}
 }
 
@@ -677,7 +680,7 @@ func (c converter) SigningComponents(tx authsigning.Tx, metadata *ConstructionMe
 		return nil, nil, crgerrs.WrapError(crgerrs.ErrConverter, fmt.Sprintf("getting signers v2 from tx %s", err.Error()))
 	}
 
-	signers, err := tx.GetSignaturesV2()
+	signers, err := tx.GetSigners()
 	if err != nil {
 		return nil, nil, crgerrs.WrapError(crgerrs.ErrConverter, fmt.Sprintf("getting signers v2 from tx %s", err.Error()))
 	}
@@ -707,18 +710,23 @@ func (c converter) SigningComponents(tx authsigning.Tx, metadata *ConstructionMe
 		// by checking if the signer at index i matches the pubkey at index
 		pubKey, err := c.ToSDK().PubKey(rosPubKeys[0])
 		if err != nil {
-			return nil, nil, crgerrs.WrapError(crgerrs.ErrConverter, fmt.Sprintf("while setting signatures %s", err.Error()))
+			return nil, nil, crgerrs.WrapError(crgerrs.ErrConverter, fmt.Sprintf("while checking pubkey %s", err.Error()))
 		}
-		if !bytes.Equal(pubKey.Address().Bytes(), signer.PubKey.Address()) {
+		if !bytes.Equal(pubKey.Address().Bytes(), signer) {
 			return nil, nil, crgerrs.WrapError(
 				crgerrs.ErrBadArgument,
 				fmt.Sprintf("public key at index %d does not match the expected transaction signer: %X <-> %X", i, rosPubKeys[i].Bytes, signer),
 			)
 		}
 
+		addr, err := c.ac.BytesToString(signer)
+		if err != nil {
+			return nil, nil, crgerrs.WrapError(crgerrs.ErrConverter, fmt.Sprintf("while converting to bech32 address: %s", err.Error()))
+		}
+
 		// set the signer data
 		signerData := authsigning.SignerData{
-			Address:       string(signer.PubKey.Address()),
+			Address:       addr,
 			ChainID:       metadata.ChainID,
 			AccountNumber: metadata.SignersData[i].AccountNumber,
 			Sequence:      metadata.SignersData[i].Sequence,
@@ -731,11 +739,8 @@ func (c converter) SigningComponents(tx authsigning.Tx, metadata *ConstructionMe
 			return nil, nil, crgerrs.WrapError(crgerrs.ErrUnknown, fmt.Sprintf("unable to sign tx: %s", err.Error()))
 		}
 
-		// set payload
-		signerAddress := sdk.AccAddress(signer.PubKey.Address()).String()
-
 		payloadsToSign[i] = &rosettatypes.SigningPayload{
-			AccountIdentifier: &rosettatypes.AccountIdentifier{Address: signerAddress},
+			AccountIdentifier: &rosettatypes.AccountIdentifier{Address: addr},
 			Bytes:             signBytes,
 			SignatureType:     rosettatypes.Ecdsa,
 		}

converter_test.go

@@ -44,7 +44,7 @@ func (s *ConverterTestSuite) SetupTest() {
 	// instantiate converter
 	cdc, ir := rosetta.MakeCodec()
 	txConfig := authtx.NewTxConfig(cdc, address.NewBech32Codec("cosmos"), address.NewBech32Codec("cosmosvaloper"), authtx.DefaultSignModes)
-	s.c = rosetta.NewConverter(cdc, ir, txConfig)
+	s.c = rosetta.NewConverter(cdc, ir, txConfig, address.NewBech32Codec("cosmos"))
 	// add utils
 	s.ir = ir
 	s.cdc = cdc

tests/systemtests/accounts_test.go

@@ -1,6 +1,6 @@
 //go:build system_test
 
-package systemtests
+package rossettaSystemTests
 
 import (
 	"testing"

tests/systemtests/block_test.go

@@ -1,6 +1,6 @@
 //go:build system_test
 
-package systemtests
+package rossettaSystemTests
 
 import (
 	"testing"

tests/systemtests/construction_test.go

@@ -1,10 +1,11 @@
 //go:build system_test
 
-package systemtests
+package rossettaSystemTests
 
 import (
 	"encoding/base64"
 	"encoding/hex"
+	"fmt"
 	"strings"
 	"testing"
 
@@ -25,12 +26,13 @@ func TestDerive(t *testing.T) {
 	rosettaRest := newRestClient(rosetta)
 
 	pubKey := secp256k1.GenPrivKey().PubKey()
-	addr, err := address.NewBech32Codec("cosmos").BytesToString(pubKey.Address().Bytes())
-	assert.NoError(t, err)
-
 	hexPk := strings.Split(pubKey.String(), "{")[1]
+
 	res, err := rosettaRest.constructionDerive(hexPk[:len(hexPk)-1])
 	assert.NoError(t, err)
+
+	addr, err := address.NewBech32Codec("cosmos").BytesToString(pubKey.Address().Bytes())
+	assert.NoError(t, err)
 	assert.Equal(t, addr, gjson.GetBytes(res, "address").String())
 }
 
@@ -82,3 +84,30 @@ func TestMetadata(t *testing.T) {
 	assert.Equal(t, gjson.GetBytes(res, "metadata.gas_price").String(), "123uatom")
 	assert.Greater(t, gjson.GetBytes(res, "suggested_fee.0.value").Int(), int64(0))
 }
+
+func TestPayloads(t *testing.T) {
+	sut.ResetChain(t)
+	sut.StartChain(t)
+
+	rosetta.restart(t)
+	rosettaRest := newRestClient(rosetta)
+
+	cli := systemtests.NewCLIWrapper(t, sut, verbose)
+	addr := cli.GetKeyAddr("node0")
+	bz, err := base64.StdEncoding.DecodeString(cli.GetPubKeyByCustomField(addr, "address"))
+	assert.NoError(t, err)
+
+	pk := secp256k1.PubKey{Key: bz}
+	hexPk := strings.Split(pk.String(), "{")[1]
+	hexPk = hexPk[:len(hexPk)-1]
+
+	op := operation{
+		msgType:  "/cosmos.bank.v1beta1.MsgSend",
+		metadata: fmt.Sprintf(`{"from_address": "%s", "to_address": "%s", "amount":[{"amount":"123", "denom":"stake"}]}`, addr, cli.AddKey("to_address")),
+	}
+
+	res, err := rosettaRest.constructionPayloads(`"signer_data":[{"account_number":1, "sequence": 0}]`, hexPk, op)
+	assert.NoError(t, err)
+	assert.NotEmpty(t, gjson.GetBytes(res, "unsigned_transaction"))
+	assert.Equal(t, gjson.GetBytes(res, "payloads.0.address").String(), addr)
+}

tests/systemtests/go.mod

@@ -1,4 +1,4 @@
-module rossetaSystemTests
+module rossettaSystemTests
 
 go 1.23.1
 

tests/systemtests/main_test.go

@@ -1,6 +1,6 @@
 //go:build system_test
 
-package systemtests
+package rossettaSystemTests
 
 import (
 	"testing"

tests/systemtests/mempool_test.go

@@ -1,6 +1,6 @@
 //go:build system_test
 
-package systemtests
+package rossettaSystemTests
 
 import (
 	"testing"