add multi sign support(fix #13)

chainlibpy/amino/transaction.py

@@ -0,0 +1,96 @@
+import base64
+import hashlib
+import json
+from typing import List
+
+import ecdsa
+
+from chainlibpy.amino import StdFee, StdSignDoc, SyncMode
+from chainlibpy.amino.message import Msg
+from chainlibpy.amino.tx import Pubkey, Signature, StdTx
+from chainlibpy.wallet import Wallet
+
+
+class Transaction:
+ """A Cosmos transaction.
+
+ After initialization, one or more token transfers can be added by
+ calling the `add_transfer()` method. Finally, call `get_pushable()`
+ to get a signed transaction that can be pushed to the `POST /txs`
+ endpoint of the Cosmos REST API.
+ """
+
+ def __init__(
+ self,
+ wallet: Wallet,
+ account_num: int,
+ sequence: int,
+ fee: StdFee = StdFee.default(),
+ memo: str = "",
+ chain_id: str = "crypto-org-chain-mainnet-1",
+ sync_mode: SyncMode = "sync",
+ timeout_height: int = 0,
+ multi_sign_address: str = None,
+ ) -> None:
+ self._multi_sign_address = multi_sign_address
+ self._wallet = wallet
+ self._account_num = str(account_num)
+ self._sequence = str(sequence)
+ self._fee = fee
+ self._memo = memo
+ self._chain_id = chain_id
+ self._sync_mode = sync_mode
+ self._msgs: List[Msg] = []
+ self._timeout_height = str(timeout_height)
+
+ def add_msg(self, msg: Msg):
+ self._msgs.append(msg)
+
+ def get_pushable(self) -> dict:
+ """get the request post to the /txs."""
+ std_tx = StdTx(self._msgs, self._fee, self._memo, self._timeout_height, [self.signature])
+
+ pushable_tx = {
+ "tx": std_tx.to_dict(),
+ "mode": self._sync_mode,
+ }
+ return pushable_tx
+
+ @property
+ def signature(self) -> Signature:
+ pubkey = self._wallet.public_key
+ base64_pubkey = base64.b64encode(pubkey).decode("utf-8")
+ pubkey = Pubkey(value=base64_pubkey)
+ raw_signature = self.sign()
+ sig_str = base64.b64encode(raw_signature).decode("utf-8")
+ signature = Signature(sig_str, pubkey, self._account_num, self._sequence)
+ return signature
+
+ def sign(self) -> bytes:
+ sign_doc = self._get_sign_doc()
+ message_str = json.dumps(
+ sign_doc.to_dict(), separators=(",", ":"), sort_keys=True
+ )
+ message_bytes = message_str.encode("utf-8")
+
+ privkey = ecdsa.SigningKey.from_string(
+ self._wallet.private_key, curve=ecdsa.SECP256k1
+ )
+ signature_compact = privkey.sign_deterministic(
+ message_bytes,
+ hashfunc=hashlib.sha256,
+ sigencode=ecdsa.util.sigencode_string_canonize,
+ )
+ return signature_compact
+
+ def _get_sign_doc(self) -> StdSignDoc:
+ sign_doc = StdSignDoc(
+ account_number=self._account_num,
+ sequence=self._sequence,
+ chain_id=self._chain_id,
+ memo=self._memo,
+ fee=self._fee.to_dict(),
+ msgs=self._msgs,
+ timeout_height=self._timeout_height,
+ )
+ return sign_doc

chainlibpy/generated/cosmos/crypto/multisig/keys_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

chainlibpy/grpc_client.py

@@ -59,6 +59,15 @@ class NetworkConfig:
  exponent=8,
  derivation_path="m/44'/1'/0'/0/0",
  ),
+ "devnet": NetworkConfig(
+ grpc_endpoint="0.0.0.0:26653",
+ chain_id="chain_id",
+ address_prefix="cro",
+ coin_denom="cro",
+ coin_base_denom="basecro",
+ derivation_path="m/44'/394'/0'/0/0",
+ exponent=8,
+ ),
 }
 
 
@@ -160,4 +169,4 @@ def broadcast_transaction(
  else:
  raise TypeError("Unexcepted mode, should be [sync, async, block]")
 
- self.tx_client.BroadcastTx(BroadcastTxRequest(tx_bytes=tx_byte, mode=_mode))
+ return self.tx_client.BroadcastTx(BroadcastTxRequest(tx_bytes=tx_byte, mode=_mode))

chainlibpy/multisign/bitarray.py

@@ -0,0 +1,74 @@
+import base64
+
+from chainlibpy.amino.basic import BasicObj
+from chainlibpy.generated.cosmos.crypto.multisig.v1beta1.multisig_pb2 import (
+ CompactBitArray as ProtoCompactBitArray,
+)
+from chainlibpy.multisign.bits import Bits
+
+
+class CompactBitArray(BasicObj):
+ MaxInt32 = 2147483647
+
+ def __init__(self, bits: int):
+ self.extra_bits_stored = 0
+ self.elems = bytearray()
+ if bits <= 0:
+ raise Exception(f"invalid bits {bits}")
+ n_elems = (bits + 7) // 8
+ if n_elems <= 0 or n_elems > self.MaxInt32:
+ raise Exception(f"invalid bits {bits}")
+ self.extra_bits_stored = bits % 8
+ self.elems = bytearray([0] * n_elems)
+
+ def __repr__(self):
+ elems = base64.b64encode(self.elems).decode('utf-8')
+ return f"extra_bits_stored:{self.extra_bits_stored}, elems:{elems}"
+
+ def bit_array(self) -> ProtoCompactBitArray:
+ return ProtoCompactBitArray(
+ extra_bits_stored=self.extra_bits_stored,
+ elems=bytes(self.elems)
+ )
+
+ def count(self) -> int:
+ """returns the number of bits in the bitarray."""
+ if self.extra_bits_stored == 0:
+ return len(self.elems) * 8
+ return (len(self.elems) - 1) * 8 + int(self.extra_bits_stored)
+
+ def get_index(self, index: int) -> bool:
+ """returns the bit at index i within the bit array.
+
+ The behavior is undefined if i >= self.count()
+ """
+ if index < 0 or index >= self.count():
+ return False
+ return (self.elems[index >> 3] & (1 << (7 - (index % 8)))) > 0
+
+ def set_index(self, i: int, v: bool) -> bool:
+ """set_index sets the bit at index i within the bit array.
+
+ Returns true if and only if the operation succeeded. The
+ behavior is undefined if i >= self.count()
+ """
+ if i < 0 or i >= self.count():
+ return False
+
+ if v:
+ self.elems[i >> 3] |= (1 << (7 - (i % 8)))
+ else:
+ self.elems[i >> 3] &= ~(1 << (7 - (i % 8)))
+ return True
+
+ def num_true_bits_before(self, index: int) -> int:
+ ones_count = 0
+ max_ = self.count()
+ index = min(index, max_)
+ elem = 0
+ while True:
+ if elem*8+7 >= index:
+ ones_count += Bits.ones_count8(self.elems[elem]) >> (7 - (index % 8) + 1)
+ return ones_count
+ ones_count += Bits.ones_count8(self.elems[elem])
+ elem += 1

chainlibpy/multisign/bits.py

@@ -0,0 +1,27 @@
+class Bits(object):
+ pop8tab = [
+ 0x00, 0x01, 0x01, 0x02, 0x01, 0x02, 0x02, 0x03, 0x01, 0x02, 0x02, 0x03, 0x02, 0x03, 0x03, 0x04,
+ 0x01, 0x02, 0x02, 0x03, 0x02, 0x03, 0x03, 0x04, 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05,
+ 0x01, 0x02, 0x02, 0x03, 0x02, 0x03, 0x03, 0x04, 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05,
+ 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05, 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06,
+ 0x01, 0x02, 0x02, 0x03, 0x02, 0x03, 0x03, 0x04, 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05,
+ 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05, 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06,
+ 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05, 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06,
+ 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06, 0x04, 0x05, 0x05, 0x06, 0x05, 0x06, 0x06, 0x07,
+ 0x01, 0x02, 0x02, 0x03, 0x02, 0x03, 0x03, 0x04, 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05,
+ 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05, 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06,
+ 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05, 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06,
+ 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06, 0x04, 0x05, 0x05, 0x06, 0x05, 0x06, 0x06, 0x07,
+ 0x02, 0x03, 0x03, 0x04, 0x03, 0x04, 0x04, 0x05, 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06,
+ 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06, 0x04, 0x05, 0x05, 0x06, 0x05, 0x06, 0x06, 0x07,
+ 0x03, 0x04, 0x04, 0x05, 0x04, 0x05, 0x05, 0x06, 0x04, 0x05, 0x05, 0x06, 0x05, 0x06, 0x06, 0x07,
+ 0x04, 0x05, 0x05, 0x06, 0x05, 0x06, 0x06, 0x07, 0x05, 0x06, 0x06, 0x07, 0x06, 0x07, 0x07, 0x08
+ ]
+
+ @classmethod
+ def ones_count8(cls, x: int) -> int:
+ """returns the number of one bits ("population count") in x.
+
+ from: go/match/bits/bits.go
+ """
+ return int(cls.pop8tab[x])