solana multisig

apps/solana/account.go

@@ -0,0 +1,50 @@
+package solana
+
+import (
+	"encoding/hex"
+	"fmt"
+
+	solana "github.com/gagliardetto/solana-go"
+	"github.com/mr-tron/base58"
+)
+
+func MPK(public string) solana.PublicKey {
+	key, err := PublicKeyFromString(public)
+	if err != nil {
+		panic(err)
+	}
+
+	return key
+}
+
+func PublicKeyFromString(public string) (solana.PublicKey, error) {
+	b, err := base58.Decode(public)
+	if err != nil || len(b) != solana.PublicKeyLength {
+		b, err = hex.DecodeString(public)
+	}
+
+	if len(b) != solana.PublicKeyLength {
+		return solana.PublicKey{}, fmt.Errorf("invalid public key length %d", len(b))
+	}
+
+	return solana.PublicKeyFromBytes(b), nil
+}
+
+func VerifyHolderKey(public string) error {
+	_, err := PublicKeyFromString(public)
+	return err
+}
+
+func VerifyMessageSignature(public string, msg, signature []byte) error {
+	pub, err := PublicKeyFromString(public)
+	if err != nil {
+		return err
+	}
+
+	sig := solana.SignatureFromBytes(signature)
+	if pub.Verify(msg, sig) {
+		return nil
+	}
+
+	return fmt.Errorf("solana.VerifyMessageSignature(%s, %x, %x)", public, msg, signature)
+}

apps/solana/account_test.go

@@ -0,0 +1,29 @@
+package solana
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestVerifyPublic(t *testing.T) {
+	public := "fb17b60698d36d45bc624c8e210b4c845233c99a7ae312a27e883a8aa8444b9b"
+	err := VerifyHolderKey(public)
+	assert.Nil(t, err)
+
+	key1, err := solana.NewRandomPrivateKey()
+	assert.Nil(t, err)
+	pub1 := key1.PublicKey()
+	t.Log(key1.String(), pub1.String())
+
+	key2, err := solana.NewRandomPrivateKey()
+	assert.Nil(t, err)
+	pub2 := key2.PublicKey()
+	t.Log(key2.String(), pub2.String())
+
+	pri3, err := solana.NewRandomPrivateKey()
+	assert.Nil(t, err)
+	pub3 := pri3.PublicKey()
+	t.Log(pri3.String(), pub3.String())
+}

apps/solana/address.go

@@ -0,0 +1,85 @@
+package solana
+
+import (
+	"encoding/binary"
+
+	"github.com/MixinNetwork/safe/apps/solana/squads_mpl"
+	solana "github.com/gagliardetto/solana-go"
+)
+
+const (
+	DefaultAuthorityIndex uint32 = 1
+)
+
+func GetAuthorityPDA(ms solana.PublicKey, index uint32) solana.PublicKey {
+	indexBytes := make([]byte, 4)
+	binary.LittleEndian.PutUint32(indexBytes, index)
+
+	address, _, err := solana.FindProgramAddress(
+		[][]byte{
+			[]byte("squad"),
+			ms.Bytes(),
+			indexBytes,
+			[]byte("authority"),
+		},
+		squads_mpl.ProgramID,
+	)
+	if err != nil {
+		panic(err)
+	}
+	return address
+}
+
+func GetDefaultAuthorityPDA(ms solana.PublicKey) solana.PublicKey {
+	return GetAuthorityPDA(ms, DefaultAuthorityIndex)
+}
+
+func GetMultisigPDA(createKey solana.PublicKey) solana.PublicKey {
+	address, _, err := solana.FindProgramAddress(
+		[][]byte{
+			[]byte("squad"),
+			createKey.Bytes(),
+			[]byte("multisig"),
+		},
+		squads_mpl.ProgramID,
+	)
+	if err != nil {
+		panic(err)
+	}
+	return address
+}
+
+func GetTransactionPDA(ms solana.PublicKey, nonce uint32) solana.PublicKey {
+	nonceBytes := make([]byte, 4)
+	binary.LittleEndian.PutUint32(nonceBytes, nonce)
+
+	address, _, err := solana.FindProgramAddress(
+		[][]byte{
+			[]byte("squad"),
+			ms.Bytes(),
+			nonceBytes,
+			[]byte("transaction"),
+		},
+		squads_mpl.ProgramID,
+	)
+	if err != nil {
+		panic(err)
+	}
+	return address
+}
+
+func GetInstructionPDA(txPda solana.PublicKey, index uint8) solana.PublicKey {
+	address, _, err := solana.FindProgramAddress(
+		[][]byte{
+			[]byte("squad"),
+			txPda.Bytes(),
+			{index},
+			[]byte("instruction"),
+		},
+		squads_mpl.ProgramID,
+	)
+	if err != nil {
+		panic(err)
+	}
+	return address
+}