[Add] Multi-Sig Support in Wallets

src/Neo/Wallets/Wallet.cs

@@ -136,6 +136,26 @@ protected Wallet(string path, ProtocolSettings settings)
             Path = path;
         }
 
+        public WalletAccount CreateMultiSigAccount(params ECPoint[] publicKeys) =>
+            CreateMultiSigAccount(publicKeys.Length, publicKeys);
+
+        public WalletAccount CreateMultiSigAccount(int m, params ECPoint[] publicKeys)
+        {
+            ArgumentOutOfRangeException.ThrowIfEqual(publicKeys.Length, 0, nameof(publicKeys));
+            ArgumentOutOfRangeException.ThrowIfGreaterThan(m, publicKeys.Length, nameof(publicKeys));
+            ArgumentOutOfRangeException.ThrowIfLessThan(m, 1, nameof(m));
+            ArgumentOutOfRangeException.ThrowIfGreaterThan(m, 1024, nameof(m));
+
+            var contract = Contract.CreateMultiSigContract(m, publicKeys);
+            var account = GetAccounts().FirstOrDefault(
+                f =>
+                    f.HasKey &&
+                    f.Lock == false &&
+                    publicKeys.Contains(f.GetKey().PublicKey));
+
+            return CreateAccount(contract, account?.GetKey());
+        }
+
         /// <summary>
         /// Creates a standard account for the wallet.
         /// </summary>
@@ -237,6 +257,12 @@ public WalletAccount CreateAccount(Contract contract, byte[] privateKey)
             return result;
         }
 
+        public IEnumerable<WalletAccount> GetMultiSigAccounts() =>
+            GetAccounts()
+                .Where(static w =>
+                    w.Lock == false &&
+                    IsMultiSigContract(w.Contract.Script));
+
         /// <summary>
         /// Gets the account with the specified public key.
         /// </summary>

tests/Neo.UnitTests/Wallets/UT_Wallet.cs

@@ -13,6 +13,7 @@
 using Neo.Cryptography;
 using Neo.Cryptography.ECC;
 using Neo.Extensions;
+using Neo.Extensions.Factories;
 using Neo.Network.P2P;
 using Neo.Network.P2P.Payloads;
 using Neo.Sign;
@@ -22,7 +23,9 @@
 using Neo.Wallets;
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Numerics;
+using Helper = Neo.SmartContract.Helper;
 
 namespace Neo.UnitTests.Wallets
 {
@@ -511,5 +514,28 @@ public void TestContainsKeyPair()
             contains = wallet.ContainsSignable(pair.PublicKey);
             Assert.IsFalse(contains); // locked
         }
+
+        [TestMethod]
+        public void TestMultiSigAccount()
+        {
+            var expectedWallet = new MyWallet();
+            var expectedPrivateKey = RandomNumberFactory.NextBytes(32, cryptography: true);
+
+            var expectedWalletAccount = expectedWallet.CreateAccount(expectedPrivateKey);
+            var expectedAccountKey = expectedWalletAccount.GetKey();
+            var actualMultiSigAccount = expectedWallet.CreateMultiSigAccount([expectedAccountKey.PublicKey]);
+
+            Assert.IsNotNull(actualMultiSigAccount);
+            Assert.AreNotEqual(expectedWalletAccount.ScriptHash, actualMultiSigAccount.ScriptHash);
+            Assert.AreEqual(expectedAccountKey.PublicKey, actualMultiSigAccount.GetKey().PublicKey);
+            Assert.IsTrue(Helper.IsMultiSigContract(actualMultiSigAccount.Contract.Script));
+            Assert.IsTrue(expectedWallet.GetMultiSigAccounts().Contains(actualMultiSigAccount));
+
+            var notExpectedAccountKeys = new ECPoint[1025];
+            Assert.ThrowsExactly<ArgumentOutOfRangeException>(() => expectedWallet.CreateMultiSigAccount());
+            Assert.ThrowsExactly<ArgumentOutOfRangeException>(() => expectedWallet.CreateMultiSigAccount(2, [expectedAccountKey.PublicKey]));
+            Assert.ThrowsExactly<ArgumentOutOfRangeException>(() => expectedWallet.CreateMultiSigAccount(0, [expectedAccountKey.PublicKey]));
+            Assert.ThrowsExactly<ArgumentOutOfRangeException>(() => expectedWallet.CreateMultiSigAccount(1025, notExpectedAccountKeys));
+        }
     }
 }