Skip to content

Add multisig tests #220

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
vsubhuman merged 3 commits into from
Oct 31, 2021
Merged

Add multisig tests #220

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
6d904a8
add multisig tests
gclaramunt Aug 24, 2021
46f4c4b
Update tests for new tx builder
SebastienGllmt Oct 12, 2021
3399aba
Merge branch 'master' into multisig-test
vsubhuman Oct 31, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
54 changes: 54 additions & 0 deletions rust/src/address.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -979,4 +979,58 @@ mod tests {
let addr_net_3 = RewardAddress::new(NetworkInfo::mainnet().network_id(), &staking_cred).to_address();
assert_eq!(addr_net_3.to_bech32(None).unwrap(), "stake1uyevw2xnsc0pvn9t9r9c7qryfqfeerchgrlm3ea2nefr9hqxdekzz");
}

#[test]
fn bip32_24_base_multisig_hd_derivation() {
let spend = root_key_24()
.derive(harden(1854))
.derive(harden(1815))
.derive(harden(0))
.derive(0)
.derive(0)
.to_public();
let stake = root_key_24()
.derive(harden(1854))
.derive(harden(1815))
.derive(harden(0))
.derive(2)
.derive(0)
.to_public();
let spend_cred = StakeCredential::from_keyhash(&spend.to_raw_key().hash());
let stake_cred = StakeCredential::from_keyhash(&stake.to_raw_key().hash());
let addr_net_0 = BaseAddress::new(NetworkInfo::testnet().network_id(), &spend_cred, &stake_cred).to_address();
assert_eq!(addr_net_0.to_bech32(None).unwrap(), "addr_test1qz8fg2e9yn0ga6sav0760cxmx0antql96mfuhqgzcc5swugw2jqqlugnx9qjep9xvcx40z0zfyep55r2t3lav5smyjrs96cusg");
let addr_net_3 = BaseAddress::new(NetworkInfo::mainnet().network_id(), &spend_cred, &stake_cred).to_address();
assert_eq!(addr_net_3.to_bech32(None).unwrap(), "addr1qx8fg2e9yn0ga6sav0760cxmx0antql96mfuhqgzcc5swugw2jqqlugnx9qjep9xvcx40z0zfyep55r2t3lav5smyjrsxv9uuh");
}

#[test]
fn multisig_from_script() {

let spend = root_key_24()
.derive(harden(1852))
.derive(harden(1815))
.derive(harden(0))
.derive(0)
.derive(0)
.to_public();


let mut pubkey_native_scripts = NativeScripts::new();

let spending_hash = spend.to_raw_key().hash();
pubkey_native_scripts.add(&NativeScript::new_script_pubkey(&ScriptPubkey::new(&spending_hash)));
let oneof_native_script = NativeScript::new_script_n_of_k(&ScriptNOfK::new(1, &pubkey_native_scripts));

let script_hash = ScriptHash::from_bytes(
oneof_native_script.hash(ScriptHashNamespace::NativeScript).to_bytes()
).unwrap();

let spend_cred = StakeCredential::from_scripthash(&script_hash);
let stake_cred = StakeCredential::from_scripthash(&script_hash);
let addr_net_0 = BaseAddress::new(NetworkInfo::testnet().network_id(), &spend_cred, &stake_cred).to_address();
assert_eq!(addr_net_0.to_bech32(None).unwrap(), "addr_test1xr0de0mz3m9xmgtlmqqzu06s0uvfsczskdec8k7v4jhr7077mjlk9rk2dkshlkqq9cl4qlccnps9pvmns0duet9w8uls8flvxc");
let addr_net_3 = BaseAddress::new(NetworkInfo::mainnet().network_id(), &spend_cred, &stake_cred).to_address();
assert_eq!(addr_net_3.to_bech32(None).unwrap(), "addr1x80de0mz3m9xmgtlmqqzu06s0uvfsczskdec8k7v4jhr7077mjlk9rk2dkshlkqq9cl4qlccnps9pvmns0duet9w8ulsylzv28");
}
}
211 changes: 208 additions & 3 deletions rust/src/tx_builder.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,7 @@ impl TransactionBuilder {
});
self.input_types.bootstraps.insert(hash.to_bytes());
}

pub fn add_input(&mut self, address: &Address, input: &TransactionInput, amount: &Value) {
match &BaseAddress::from_address(address) {
Some(addr) => {
Expand Down Expand Up @@ -1591,7 +1591,7 @@ mod tests {
),
&Value::new(&to_bignum(2_400_000))
);

tx_builder.set_ttl(1);

let change_addr = ByronAddress::from_base58("Ae2tdPwUPEZGUEsuMAhvDcy94LKsZxDjCbgaiBBMgYpR8sKf96xJmit7Eho").unwrap();
Expand Down Expand Up @@ -1635,7 +1635,7 @@ mod tests {
),
&input_value
);

tx_builder.set_ttl(1);

let change_addr = ByronAddress::from_base58("Ae2tdPwUPEZGUEsuMAhvDcy94LKsZxDjCbgaiBBMgYpR8sKf96xJmit7Eho").unwrap();
Expand Down Expand Up @@ -1891,4 +1891,209 @@ mod tests {

assert!(tx_builder.add_change_if_needed(&change_addr).is_err())
}

fn build_tx_pay_to_multisig() {
let linear_fee = LinearFee::new(&to_bignum(10), &to_bignum(2));
let mut tx_builder =
TransactionBuilder::new(&linear_fee, &to_bignum(1), &to_bignum(1), &to_bignum(1), MAX_VALUE_SIZE, MAX_TX_SIZE);
let spend = root_key_15()
.derive(harden(1854))
.derive(harden(1815))
.derive(harden(0))
.derive(0)
.derive(0)
.to_public();

let stake = root_key_15()
.derive(harden(1852))
.derive(harden(1815))
.derive(harden(0))
.derive(2)
.derive(0)
.to_public();

let spend_cred = StakeCredential::from_keyhash(&spend.to_raw_key().hash());
let stake_cred = StakeCredential::from_keyhash(&stake.to_raw_key().hash());

let addr_net_0 = BaseAddress::new(NetworkInfo::testnet().network_id(), &spend_cred, &stake_cred).to_address();

tx_builder.add_key_input(
&spend.to_raw_key().hash(),
&TransactionInput::new(&genesis_id(), 0),
&Value::new(&to_bignum(1_000_000))
);
tx_builder.add_output(&TransactionOutput::new(
&addr_net_0,
&Value::new(&to_bignum(999_000 ))
)).unwrap();
tx_builder.set_ttl(1000);
tx_builder.set_fee(&to_bignum(1_000));

assert_eq!(tx_builder.outputs.len(),1);
assert_eq!(
tx_builder.get_explicit_input().unwrap().checked_add(&tx_builder.get_implicit_input().unwrap()).unwrap(),
tx_builder.get_explicit_output().unwrap().checked_add(&Value::new(&tx_builder.get_fee_if_set().unwrap())).unwrap()
);


let _final_tx = tx_builder.build().unwrap();
let _deser_t = TransactionBody::from_bytes(_final_tx.to_bytes()).unwrap();

assert_eq!(_deser_t.to_bytes(), _final_tx.to_bytes());
}

fn build_full_tx(body: &TransactionBody,
witness_set: &TransactionWitnessSet,
auxiliary_data: Option<AuxiliaryData>) -> Transaction {
return Transaction::new(
body,
witness_set,
auxiliary_data
);
}

#[test]
fn build_tx_multisig_spend_1on1_unsigned() {
let linear_fee = LinearFee::new(&to_bignum(10), &to_bignum(2));
let mut tx_builder =
TransactionBuilder::new(&linear_fee, &to_bignum(1), &to_bignum(1), &to_bignum(1), MAX_VALUE_SIZE, MAX_TX_SIZE);

let spend = root_key_15()//multisig
.derive(harden(1854))
.derive(harden(1815))
.derive(harden(0))
.derive(0)
.derive(0)
.to_public();
let stake = root_key_15()//multisig
.derive(harden(1854))
.derive(harden(1815))
.derive(harden(0))
.derive(2)
.derive(0)
.to_public();

let change_key = root_key_15()
.derive(harden(1852))
.derive(harden(1815))
.derive(harden(0))
.derive(1)
.derive(0)
.to_public();

let spend_cred = StakeCredential::from_keyhash(&spend.to_raw_key().hash());
let stake_cred = StakeCredential::from_keyhash(&stake.to_raw_key().hash());
let change_cred = StakeCredential::from_keyhash(&change_key.to_raw_key().hash());
let addr_multisig = BaseAddress::new(NetworkInfo::testnet().network_id(), &spend_cred, &stake_cred).to_address();
let addr_output = BaseAddress::new(NetworkInfo::testnet().network_id(), &change_cred, &stake_cred).to_address();

tx_builder.add_input(
&addr_multisig,
&TransactionInput::new(&genesis_id(), 0),
&Value::new(&to_bignum(1_000_000))
);

tx_builder.add_output(&TransactionOutput::new(
&addr_output,
&Value::new(&to_bignum(999_000 ))
)).unwrap();
tx_builder.set_ttl(1000);
tx_builder.set_fee(&to_bignum(1_000));

let mut auxiliary_data = AuxiliaryData::new();
let mut pubkey_native_scripts = NativeScripts::new();
let mut oneof_native_scripts = NativeScripts::new();

let spending_hash = spend.to_raw_key().hash();
pubkey_native_scripts.add(&NativeScript::new_script_pubkey(&ScriptPubkey::new(&spending_hash)));
oneof_native_scripts.add(&NativeScript::new_script_n_of_k(&ScriptNOfK::new(1, &pubkey_native_scripts)));
auxiliary_data.set_native_scripts(&oneof_native_scripts);
tx_builder.set_auxiliary_data(&auxiliary_data);


let body = tx_builder.build().unwrap();

assert_eq!(tx_builder.outputs.len(), 1);
assert_eq!(
tx_builder.get_explicit_input().unwrap().checked_add(&tx_builder.get_implicit_input().unwrap()).unwrap(),
tx_builder.get_explicit_output().unwrap().checked_add(&Value::new(&tx_builder.get_fee_if_set().unwrap())).unwrap()
);


let _final_tx = tx_builder.build().unwrap();
let _deser_t = TransactionBody::from_bytes(_final_tx.to_bytes()).unwrap();

assert_eq!(_deser_t.to_bytes(), _final_tx.to_bytes());
assert_eq!(_deser_t.auxiliary_data_hash.unwrap(), utils::hash_auxiliary_data(&auxiliary_data));
}

#[test]
fn build_tx_multisig_1on1_signed() {
let linear_fee = LinearFee::new(&to_bignum(10), &to_bignum(2));
let mut tx_builder =
TransactionBuilder::new(&linear_fee, &to_bignum(1), &to_bignum(1), &to_bignum(1), MAX_VALUE_SIZE, MAX_TX_SIZE);
let spend = root_key_15()
.derive(harden(1854))//multisig
.derive(harden(1815))
.derive(harden(0))
.derive(0)
.derive(0)
.to_public();
let stake = root_key_15()
.derive(harden(1854))//multisig
.derive(harden(1815))
.derive(harden(0))
.derive(2)
.derive(0)
.to_public();

let spend_cred = StakeCredential::from_keyhash(&spend.to_raw_key().hash());
let stake_cred = StakeCredential::from_keyhash(&stake.to_raw_key().hash());
let addr_net_0 = BaseAddress::new(NetworkInfo::testnet().network_id(), &spend_cred, &stake_cred).to_address();
tx_builder.add_key_input(
&spend.to_raw_key().hash(),
&TransactionInput::new(&genesis_id(), 0),
&Value::new(&to_bignum(1_000_000))
);
tx_builder.add_output(&TransactionOutput::new(
&addr_net_0,
&Value::new(&to_bignum(999_000 ))
)).unwrap();
tx_builder.set_ttl(1000);
tx_builder.set_fee(&to_bignum(1_000));

let mut auxiliary_data = AuxiliaryData::new();
let mut pubkey_native_scripts = NativeScripts::new();
let mut oneof_native_scripts = NativeScripts::new();

let spending_hash = spend.to_raw_key().hash();
pubkey_native_scripts.add(&NativeScript::new_script_pubkey(&ScriptPubkey::new(&spending_hash)));
oneof_native_scripts.add(&NativeScript::new_script_n_of_k(&ScriptNOfK::new(1, &pubkey_native_scripts)));
auxiliary_data.set_native_scripts(&oneof_native_scripts);
tx_builder.set_auxiliary_data(&auxiliary_data);


let body = tx_builder.build().unwrap();

assert_eq!(tx_builder.outputs.len(), 1);
assert_eq!(
tx_builder.get_explicit_input().unwrap().checked_add(&tx_builder.get_implicit_input().unwrap()).unwrap(),
tx_builder.get_explicit_output().unwrap().checked_add(&Value::new(&tx_builder.get_fee_if_set().unwrap())).unwrap()
);

let mut witness_set = TransactionWitnessSet::new();
let mut vkw = Vkeywitnesses::new();
vkw.add(&make_vkey_witness(
&hash_transaction(&body),
&PrivateKey::from_normal_bytes(
&hex::decode("c660e50315d76a53d80732efda7630cae8885dfb85c46378684b3c6103e1284a").unwrap()
).unwrap()
));
witness_set.set_vkeys(&vkw);

let _final_tx = build_full_tx(&body, &witness_set, None);
let _deser_t = Transaction::from_bytes(_final_tx.to_bytes()).unwrap();
assert_eq!(_deser_t.to_bytes(), _final_tx.to_bytes());
assert_eq!(_deser_t.body().auxiliary_data_hash.unwrap(), utils::hash_auxiliary_data(&auxiliary_data));
}
}