GArray *bp_block_merkle_tree(const struct bp_block *block)
{
if (!block->vtx || !block->vtx->len)
return NULL;
GArray *arr = g_array_new(FALSE, TRUE, sizeof(bu256_t));
unsigned int i;
for (i = 0; i < block->vtx->len; i++) {
struct bp_tx *tx;
tx = g_ptr_array_index(block->vtx, i);
bp_tx_calc_sha256(tx);
g_array_append_val(arr, tx->sha256);
}
unsigned int j = 0, nSize;
for (nSize = block->vtx->len; nSize > 1; nSize = (nSize + 1) / 2) {
for (i = 0; i < nSize; i += 2) {
unsigned int i2 = MIN(i+1, nSize-1);
bu256_t hash;
bu_Hash_((unsigned char *) &hash,
&g_array_index(arr, bu256_t, j+i), sizeof(bu256_t),
&g_array_index(arr, bu256_t, j+i2),sizeof(bu256_t));
g_array_append_val(arr, hash);
}
j += nSize;
}
return arr;
}
static void runtest(const char *json_base_fn, const char *ser_in_fn,
const char *block_in_hash, const char *tx_in_hash)
{
/* read wallet data */
char *json_fn = test_filename(json_base_fn);
json_t *wallet = read_json(json_fn);
assert(wallet != NULL);
/* read block data containing incoming payment */
char *fn = test_filename(ser_in_fn);
void *data;
size_t data_len;
assert(bu_read_file(fn, &data, &data_len, 1 * 1024 * 1024) == true);
struct bp_block block_in;
bp_block_init(&block_in);
struct const_buffer buf = { data, data_len };
assert(deser_bp_block(&block_in, &buf) == true);
bp_block_calc_sha256(&block_in);
/* verify block-in data matches expected block-in hash */
bu256_t check_hash;
assert(hex_bu256(&check_hash, block_in_hash) == true);
assert(bu256_equal(&block_in.sha256, &check_hash) == true);
/* load key that has received an incoming payment */
struct bp_key key;
assert(bp_key_init(&key) == true);
load_json_key(wallet, &key);
/* load key into keyset */
struct bp_keyset ks;
bpks_init(&ks);
assert(bpks_add(&ks, &key) == true);
/* find key matches in block */
parr *matches;
matches = bp_block_match(&block_in, &ks);
assert(matches != NULL);
assert(matches->len == 1);
struct bp_block_match *match = parr_idx(matches, 0);
assert(match->n == 1); /* match 2nd tx, index 1 */
/* get matching transaction */
struct bp_tx *tx = parr_idx(block_in.vtx, match->n);
bp_tx_calc_sha256(tx);
/* verify txid matches expected */
char tx_hexstr[BU256_STRSZ];
bu256_hex(tx_hexstr, &tx->sha256);
assert(strcmp(tx_hexstr, tx_in_hash) == 0);
/* verify mask matches 2nd txout (1 << 1) */
BIGNUM tmp_mask;
BN_init(&tmp_mask);
BN_one(&tmp_mask);
BN_lshift(&tmp_mask, &tmp_mask, 1);
assert(BN_cmp(&tmp_mask, &match->mask) == 0);
/* build merkle tree, tx's branch */
parr *mtree = bp_block_merkle_tree(&block_in);
assert(mtree != NULL);
parr *mbranch = bp_block_merkle_branch(&block_in, mtree, match->n);
assert(mbranch != NULL);
/* verify merkle branch for tx matches expected */
bu256_t mrk_check;
bp_check_merkle_branch(&mrk_check, &tx->sha256, mbranch, match->n);
assert(bu256_equal(&mrk_check, &block_in.hashMerkleRoot) == true);
/* release resources */
parr_free(mtree, true);
parr_free(mbranch, true);
BN_clear_free(&tmp_mask);
parr_free(matches, true);
bpks_free(&ks);
bp_key_free(&key);
bp_block_free(&block_in);
json_decref(wallet);
free(data);
free(fn);
free(json_fn);
}
static void runtest(const char *json_fn_base, const char *ser_fn_base)
{
char *fn = test_filename(json_fn_base);
json_t *meta = read_json(fn);
assert(json_is_object(meta));
char *ser_fn = test_filename(ser_fn_base);
void *data = NULL;
size_t data_len = 0;
bool rc = bu_read_file(ser_fn, &data, &data_len, 100 * 1024 * 1024);
assert(rc);
const char *hashstr = json_string_value(json_object_get(meta, "hash"));
assert(hashstr != NULL);
unsigned int size = json_integer_value(json_object_get(meta, "size"));
assert(data_len == size);
struct bp_tx tx;
bp_tx_init(&tx);
struct const_buffer buf = { data, data_len };
rc = deser_bp_tx(&tx, &buf);
assert(rc);
cstring *gs = cstr_new_sz(10000);
ser_bp_tx(gs, &tx);
if (gs->len != data_len) {
fprintf(stderr, "gs->len %ld, data_len %lu\n",
(long)gs->len, data_len);
assert(gs->len == data_len);
}
assert(memcmp(gs->str, data, data_len) == 0);
bp_tx_calc_sha256(&tx);
char hexstr[BU256_STRSZ];
bu256_hex(hexstr, &tx.sha256);
if (strcmp(hexstr, hashstr)) {
fprintf(stderr, "tx: wanted hash %s,\n got hash %s\n",
hashstr, hexstr);
assert(!strcmp(hexstr, hashstr));
}
assert(tx.vin->len == 1);
assert(tx.vout->len == 2);
struct bp_tx tx_copy;
bp_tx_init(&tx_copy);
bp_tx_copy(&tx_copy, &tx);
bp_tx_calc_sha256(&tx_copy);
assert(bu256_equal(&tx_copy.sha256, &tx.sha256) == true);
bp_tx_free(&tx);
bp_tx_free(&tx_copy);
cstr_free(gs, true);
free(data);
free(fn);
free(ser_fn);
json_decref(meta);
}
static void test_tx_valid(bool is_valid, struct bp_hashtab *input_map,
cstring *tx_ser, bool enforce_p2sh)
{
struct bp_tx tx;
bp_tx_init(&tx);
struct const_buffer buf = { tx_ser->str, tx_ser->len };
assert(deser_bp_tx(&tx, &buf) == true);
if (is_valid) {
/* checking for valid tx; !bp_tx_valid implies test fail */
assert(bp_tx_valid(&tx) == true);
} else {
/* checking for invalid tx; bp_tx_valid==false implies test
* succeeded; no more work to do; bp_tx_valid==true
* implies the test will detect the invalid condition
* further down in the code
*/
if (bp_tx_valid(&tx) == false)
goto out;
}
bp_tx_calc_sha256(&tx);
unsigned int i;
for (i = 0; i < tx.vin->len; i++) {
struct bp_txin *txin;
txin = parr_idx(tx.vin, i);
assert(txin != NULL);
cstring *scriptPubKey = bp_hashtab_get(input_map,
&txin->prevout);
if (scriptPubKey == NULL) {
if (!is_valid) {
/* if testing tx_invalid.json, missing input
* is invalid, and therefore correct
*/
continue;
}
char tx_hexstr[BU256_STRSZ], hexstr[BU256_STRSZ];
bu256_hex(tx_hexstr, &tx.sha256);
bu256_hex(hexstr, &txin->prevout.hash);
dump_comments();
fprintf(stderr,
"tx-valid: TX %s\n"
"tx-valid: prevout (%s, %u) not found\n",
tx_hexstr, hexstr, txin->prevout.n);
assert(scriptPubKey != NULL);
}
bool rc = bp_script_verify(txin->scriptSig, scriptPubKey,
&tx, i,
enforce_p2sh ? SCRIPT_VERIFY_P2SH :
SCRIPT_VERIFY_NONE, 0);
if (rc != is_valid) {
char tx_hexstr[BU256_STRSZ];
bu256_hex(tx_hexstr, &tx.sha256);
dump_comments();
fprintf(stderr,
"tx-valid: TX %s\n"
"tx-valid: txin %u script verification failed\n",
tx_hexstr, i);
assert(rc == is_valid);
}
}
out:
bp_tx_free(&tx);
}
static void output_json_txid(json_t *obj)
{
bp_tx_calc_sha256(&tx);
bp_json_set_new_int256(obj, "txid", &tx.sha256);
}
