Skip to content

Commit

Permalink
feat: add more spending condition tests
Browse files Browse the repository at this point in the history 
From stacks-blockchain, src/chainstate/stacks/auth.rs:

- tx_stacks_spending_condition_p2sh
- tx_stacks_spending_condition_p2wsh (after adding P2WSH and P2WPKH support)
- tx_stacks_auth
  • Loading branch information
@fariedt @reedrosenbluth
fariedt authored and reedrosenbluth committed Oct 20, 2021
1 parent da31bf1 commit 930f34d
Showing 1 changed file with 222 additions and 12 deletions.
234 changes: 222 additions & 12 deletions packages/transactions/tests/authorization.test.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,12 @@ import {
emptyMessageSignature,
serializeSpendingCondition,
SingleSigSpendingCondition,
SponsoredAuthorization,
StandardAuthorization,
} from '../src/authorization';
import { BufferArray } from '../src/utils';

import {AddressHashMode, PubKeyEncoding} from '../src/constants';
import {AddressHashMode, AuthType, PubKeyEncoding} from '../src/constants';

import {BufferReader} from '../src/bufferReader';
import {createStacksPrivateKey, createStacksPublicKey, signWithKey,} from '../src/keys';
Expand Down Expand Up @@ -63,7 +66,7 @@ test('Single sig spending condition uncompressed', () => {
// address hash mode
AddressHashMode.SerializeP2PKH,
// signer
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11,
// nonce
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7b,
Expand All @@ -72,9 +75,9 @@ test('Single sig spending condition uncompressed', () => {
// key encoding,
PubKeyEncoding.Uncompressed,
// signature
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff
]
Expand Down Expand Up @@ -110,22 +113,22 @@ test('Multi sig spending condition uncompressed', () => {
// address hash mode
AddressHashMode.SerializeP2SH,
// signer
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11,
// nonce
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7b,
// fee rate
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0xc8,
// length prefixed list of fields
0x00, 0x00, 0x00, 0x03,
0x02, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0x02, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff,
0x02, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0x02, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff,
0x00, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02, 0x02,
Expand All @@ -137,3 +140,210 @@ test('Multi sig spending condition uncompressed', () => {

expect(serializedSpendingCondition.toString('hex')).toEqual(spendingConditionBytes.toString('hex'));
});

// auth.rs: tx_stacks_spending_condition_p2sh() (compressed multisig)
test('Multi sig P2SH spending condition compressed', () => {
const addressHashMode = AddressHashMode.SerializeP2SH;
const nonce = new BigNum(456);
const fee = new BigNum(567);
const pubKey = '03ef2340518b5867b23598a9cf74611f8b98064f7d55cdb8c107c67b5efcbc5c77';
const pubKeys = [pubKey, pubKey, pubKey];

const spendingCondition = createMultiSigSpendingCondition(
addressHashMode,
2,
pubKeys,
nonce,
fee
);
spendingCondition.signer = '11'.repeat(20);

const signature1 = createMessageSignature('ff'.repeat(65));
const signature2 = createMessageSignature('fe'.repeat(65));
const fields = [signature1, signature2, createStacksPublicKey(pubKeys[0])];
spendingCondition.fields = fields.map(sig => createTransactionAuthField(PubKeyEncoding.Compressed, sig));

const serializedSpendingCondition = serializeSpendingCondition(spendingCondition);

// prettier-ignore
const spendingConditionBytesHex = [
// address hash mode
AddressHashMode.SerializeP2SH,
// signer
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11,
// nonce
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0xc8,
// fee rate
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x37,
// length prefixed list of fields
0x00, 0x00, 0x00, 0x03,
0x02, // AuthFieldType.SignatureCompressed,
// [...Buffer.from(signature1.data, 'hex')],
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff,
0x02, // AuthFieldType.SignatureCompressed,
// [...Buffer.from(signature2.data, 'hex')],
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0x00, // PubKeyEncoding.Compressed,
// [...fields[2].data as Buffer],
0x03, 0xef, 0x23, 0x40, 0x51, 0x8b, 0x58, 0x67, 0xb2, 0x35, 0x98, 0xa9, 0xcf, 0x74, 0x61,
0x1f, 0x8b, 0x98, 0x06, 0x4f, 0x7d, 0x55, 0xcd, 0xb8, 0xc1, 0x07, 0xc6, 0x7b, 0x5e, 0xfc,
0xbc, 0x5c, 0x77,
// required signatures
0x00, 0x02
].flat();

const spendingConditionBytes = Buffer.from(spendingConditionBytesHex);

expect(serializedSpendingCondition).toEqual(spendingConditionBytes);
});


// auth.rs: tx_stacks_spending_condition_p2wsh()
test('Multi sig P2WSH spending condition compressed', () => {
const addressHashMode = AddressHashMode.SerializeP2WSH;
const nonce = new BigNum(456);
const fee = new BigNum(567);
const pubKey = '03ef2340518b5867b23598a9cf74611f8b98064f7d55cdb8c107c67b5efcbc5c77';
const pubKeys = [pubKey, pubKey, pubKey];

const spendingCondition = createMultiSigSpendingCondition(
addressHashMode,
2,
pubKeys,
nonce,
fee
);
spendingCondition.signer = '11'.repeat(20);

const signature1 = createMessageSignature('ff'.repeat(65));
const signature2 = createMessageSignature('fe'.repeat(65));
const fields = [signature1, signature2, createStacksPublicKey(pubKeys[0])];
spendingCondition.fields = fields.map(sig => createTransactionAuthField(PubKeyEncoding.Compressed, sig));

const serializedSpendingCondition = serializeSpendingCondition(spendingCondition);

// prettier-ignore
const spendingConditionBytesHex = [
// address hash mode
AddressHashMode.SerializeP2WSH,
// signer
0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
0x11, 0x11, 0x11, 0x11, 0x11,
// nonce
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0xc8,
// fee rate
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0x37,
// length prefixed list of fields
0x00, 0x00, 0x00, 0x03,
0x02, // AuthFieldType.SignatureCompressed,
// [...Buffer.from(signature1.data, 'hex')],
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff,
0x02, // AuthFieldType.SignatureCompressed,
// [...Buffer.from(signature2.data, 'hex')],
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0xfe, 0xfe, 0xfe, 0xfe, 0xfe,
0x00, // PubKeyEncoding.Compressed,
// [...fields[2].data as Buffer],
0x03, 0xef, 0x23, 0x40, 0x51, 0x8b, 0x58, 0x67, 0xb2, 0x35, 0x98, 0xa9, 0xcf, 0x74, 0x61,
0x1f, 0x8b, 0x98, 0x06, 0x4f, 0x7d, 0x55, 0xcd, 0xb8, 0xc1, 0x07, 0xc6, 0x7b, 0x5e, 0xfc,
0xbc, 0x5c, 0x77,
// required signatures
0x00, 0x02
].flat();

const spendingConditionBytes = Buffer.from(spendingConditionBytesHex);

expect(serializedSpendingCondition).toEqual(spendingConditionBytes);
});

// auth.rs: tx_stacks_auth()
test('Spending conditions', () => {
const bn123 = new BigNum(123);
const bn345 = new BigNum(345);
const bn456 = new BigNum(456);
const bn567 = new BigNum(567);
const signer = '11'.repeat(20);
const signatureFF = createMessageSignature('ff'.repeat(65));
const signatureFE = createMessageSignature('fe'.repeat(65));
const pubKeyUncompressed = '04ef2340518b5867b23598a9cf74611f8b98064f7d55cdb8c107c67b5efcbc5c771f112f919b00a6c6c5f51f7c63e1762fe9fac9b66ec75a053db7f51f4a52712b';
const pubKeyCompressed = '03ef2340518b5867b23598a9cf74611f8b98064f7d55cdb8c107c67b5efcbc5c77';

const sp1 = createSingleSigSpendingCondition(AddressHashMode.SerializeP2PKH, '', bn123, bn567);
sp1.signer = signer;
sp1.keyEncoding = PubKeyEncoding.Uncompressed;
sp1.signature = signatureFF;

const sp2 = createSingleSigSpendingCondition(AddressHashMode.SerializeP2PKH, '', bn345, bn567);
sp2.signer = signer;
sp2.keyEncoding = PubKeyEncoding.Compressed;
sp2.signature = signatureFF;

const sp3 = createMultiSigSpendingCondition(AddressHashMode.SerializeP2SH,
2,
[pubKeyUncompressed, pubKeyUncompressed, pubKeyUncompressed],
bn123,
bn567);
sp3.signer = signer;
sp3.fields = [signatureFF, signatureFE, createStacksPublicKey(pubKeyUncompressed)].map(sig => createTransactionAuthField(PubKeyEncoding.Uncompressed, sig));

const sp4 = createMultiSigSpendingCondition(AddressHashMode.SerializeP2SH,
2,
[pubKeyCompressed, pubKeyCompressed, pubKeyCompressed],
bn123,
bn567);
sp4.signer = signer;
sp4.fields = [signatureFF, signatureFE, createStacksPublicKey(pubKeyCompressed)].map(sig => createTransactionAuthField(PubKeyEncoding.Compressed, sig));

const sp5 = createSingleSigSpendingCondition(AddressHashMode.SerializeP2WPKH, '', bn345, bn567);
sp5.signer = signer;
sp5.keyEncoding = PubKeyEncoding.Compressed;
sp5.signature = signatureFE;

const sp6 = createMultiSigSpendingCondition(AddressHashMode.SerializeP2WSH,
2,
[pubKeyCompressed, pubKeyCompressed, pubKeyCompressed],
bn456,
bn567);
sp6.signer = signer;
sp6.fields = [signatureFF, signatureFE, createStacksPublicKey(pubKeyCompressed)].map(sig => createTransactionAuthField(PubKeyEncoding.Compressed, sig));

const spendingConditions = [sp1, sp2, sp3, sp4, sp5, sp6];

for (let i = 0; i < spendingConditions.length; i++) {
const serialized1 = serializeSpendingCondition(spendingConditions[i]);
const serialized2 = serializeSpendingCondition(spendingConditions[(i + 1) % spendingConditions.length]);

const standard = new StandardAuthorization(spendingConditions[i]);
const standardArray = new BufferArray();
standardArray.appendByte(AuthType.Standard);
standardArray.push(serialized1);
const standardBytes = Buffer.from(standardArray.concatBuffer());

const sponsored = new SponsoredAuthorization(spendingConditions[i], spendingConditions[(i + 1) % spendingConditions.length]);
const sponsoredArray = new BufferArray();
sponsoredArray.appendByte(AuthType.Sponsored);
sponsoredArray.push(serialized1);
sponsoredArray.push(serialized2);
const sponsoredBytes = Buffer.from(sponsoredArray.concatBuffer());

expect(standard.serialize()).toEqual(standardBytes);
expect(sponsored.serialize()).toEqual(sponsoredBytes);
}

});

0 comments on commit 930f34d

Please sign in to comment.