#include "config.h" #define ZLIB_EVEN_IF_EXPANDS 1 #include "../queries.c" #include #include #include #include #ifdef NDEBUG #error "assert required for tests" #endif /* AUTOGENERATED MOCKS START */ /* Generated stub for daemon_conn_read_next */ struct io_plan *daemon_conn_read_next(struct io_conn *conn UNNEEDED, struct daemon_conn *dc UNNEEDED) { fprintf(stderr, "daemon_conn_read_next called!\n"); abort(); } /* Generated stub for daemon_conn_wake */ void daemon_conn_wake(struct daemon_conn *dc UNNEEDED) { fprintf(stderr, "daemon_conn_wake called!\n"); abort(); } /* Generated stub for decode_channel_update_timestamps */ struct channel_update_timestamps *decode_channel_update_timestamps(const tal_t *ctx UNNEEDED, const struct tlv_reply_channel_range_tlvs_timestamps_tlv *timestamps_tlv UNNEEDED) { fprintf(stderr, "decode_channel_update_timestamps called!\n"); abort(); } /* Generated stub for decode_scid_query_flags */ bigsize_t *decode_scid_query_flags(const tal_t *ctx UNNEEDED, const struct tlv_query_short_channel_ids_tlvs_query_flags *qf UNNEEDED) { fprintf(stderr, "decode_scid_query_flags called!\n"); abort(); } /* Generated stub for decode_short_ids */ struct short_channel_id *decode_short_ids(const tal_t *ctx UNNEEDED, const u8 *encoded UNNEEDED) { fprintf(stderr, "decode_short_ids called!\n"); abort(); } /* Generated stub for fromwire_gossip_dev_set_max_scids_encode_size */ bool fromwire_gossip_dev_set_max_scids_encode_size(const void *p UNNEEDED, u32 *max UNNEEDED) { fprintf(stderr, "fromwire_gossip_dev_set_max_scids_encode_size called!\n"); abort(); } /* Generated stub for get_cupdate_parts */ void get_cupdate_parts(const u8 *channel_update UNNEEDED, const u8 *parts[2] UNNEEDED, size_t sizes[2]) { fprintf(stderr, "get_cupdate_parts called!\n"); abort(); } /* Generated stub for get_node */ struct node *get_node(struct routing_state *rstate UNNEEDED, const struct node_id *id UNNEEDED) { fprintf(stderr, "get_node called!\n"); abort(); } /* Generated stub for gossip_store_get */ const u8 *gossip_store_get(const tal_t *ctx UNNEEDED, struct gossip_store *gs UNNEEDED, u64 offset UNNEEDED) { fprintf(stderr, "gossip_store_get called!\n"); abort(); } /* Generated stub for master_badmsg */ void master_badmsg(u32 type_expected UNNEEDED, const u8 *msg) { fprintf(stderr, "master_badmsg called!\n"); abort(); } /* Generated stub for peer_supplied_good_gossip */ void peer_supplied_good_gossip(struct peer *peer UNNEEDED, size_t amount UNNEEDED) { fprintf(stderr, "peer_supplied_good_gossip called!\n"); abort(); } /* Generated stub for queue_peer_from_store */ void queue_peer_from_store(struct peer *peer UNNEEDED, const struct broadcastable *bcast UNNEEDED) { fprintf(stderr, "queue_peer_from_store called!\n"); abort(); } /* Generated stub for queue_peer_msg */ void queue_peer_msg(struct peer *peer UNNEEDED, const u8 *msg TAKES UNNEEDED) { fprintf(stderr, "queue_peer_msg called!\n"); abort(); } /* Generated stub for towire_errorfmt */ u8 *towire_errorfmt(const tal_t *ctx UNNEEDED, const struct channel_id *channel UNNEEDED, const char *fmt UNNEEDED, ...) { fprintf(stderr, "towire_errorfmt called!\n"); abort(); } /* AUTOGENERATED MOCKS END */ /* Generated stub for status_fmt */ void status_fmt(enum log_level level UNNEEDED, const char *fmt UNNEEDED, ...) { } static const char *test_vectors[] = { "{\n" " \"msg\" : {\n" " \"type\" : \"QueryChannelRange\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"firstBlockNum\" : 100000,\n" " \"numberOfBlocks\" : 1500,\n" " \"extensions\" : [ ]\n" " },\n" " \"hex\" : \"01070f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206000186a0000005dc\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"QueryChannelRange\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"firstBlockNum\" : 35000,\n" " \"numberOfBlocks\" : 100,\n" " \"extensions\" : [ \"WANT_TIMESTAMPS | WANT_CHECKSUMS\" ]\n" " },\n" " \"hex\" : \"01070f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206000088b800000064010103\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"ReplyChannelRange\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"firstBlockNum\" : 756230,\n" " \"numberOfBlocks\" : 1500,\n" " \"complete\" : 1,\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"UNCOMPRESSED\",\n" " \"array\" : [ \"0x0x142\", \"0x0x15465\", \"0x69x42692\" ]\n" " }\n" " },\n" " \"hex\" : \"01080f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206000b8a06000005dc01001900000000000000008e0000000000003c69000000000045a6c4\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"ReplyChannelRange\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"firstBlockNum\" : 1600,\n" " \"numberOfBlocks\" : 110,\n" " \"complete\" : 1,\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"array\" : [ \"0x0x142\", \"0x0x15465\", \"0x4x3318\" ]\n" " }\n" " },\n" " \"hex\" : \"01080f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206000006400000006e01001601789c636000833e08659309a65878be010010a9023a\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"ReplyChannelRange\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"firstBlockNum\" : 122334,\n" " \"numberOfBlocks\" : 1500,\n" " \"complete\" : 1,\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"UNCOMPRESSED\",\n" " \"array\" : [ \"0x0x12355\", \"0x7x30934\", \"0x70x57793\" ]\n" " },\n" " \"timestamps\" : {\n" " \"encoding\" : \"UNCOMPRESSED\",\n" " \"timestamps\" : [ {\n" " \"timestamp1\" : 164545,\n" " \"timestamp2\" : 948165\n" " }, {\n" " \"timestamp1\" : 489645,\n" " \"timestamp2\" : 4786864\n" " }, {\n" " \"timestamp1\" : 46456,\n" " \"timestamp2\" : 9788415\n" " } ]\n" " },\n" " \"checksums\" : {\n" " \"checksums\" : [ {\n" " \"checksum1\" : 1111,\n" " \"checksum2\" : 2222\n" " }, {\n" " \"checksum1\" : 3333,\n" " \"checksum2\" : 4444\n" " }, {\n" " \"checksum1\" : 5555,\n" " \"checksum2\" : 6666\n" " } ]\n" " }\n" " },\n" " \"hex\" : \"01080f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e22060001ddde000005dc01001900000000000000304300000000000778d6000000000046e1c1011900000282c1000e77c5000778ad00490ab00000b57800955bff031800000457000008ae00000d050000115c000015b300001a0a\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"ReplyChannelRange\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"firstBlockNum\" : 122334,\n" " \"numberOfBlocks\" : 1500,\n" " \"complete\" : 1,\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"array\" : [ \"0x0x12355\", \"0x7x30934\", \"0x70x57793\" ]\n" " },\n" " \"timestamps\" : {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"timestamps\" : [ {\n" " \"timestamp1\" : 164545,\n" " \"timestamp2\" : 948165\n" " }, {\n" " \"timestamp1\" : 489645,\n" " \"timestamp2\" : 4786864\n" " }, {\n" " \"timestamp1\" : 46456,\n" " \"timestamp2\" : 9788415\n" " } ]\n" " },\n" " \"checksums\" : {\n" " \"checksums\" : [ {\n" " \"checksum1\" : 1111,\n" " \"checksum2\" : 2222\n" " }, {\n" " \"checksum1\" : 3333,\n" " \"checksum2\" : 4444\n" " }, {\n" " \"checksum1\" : 5555,\n" " \"checksum2\" : 6666\n" " } ]\n" " }\n" " },\n" " \"hex\" : \"01080f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e22060001ddde000005dc01001801789c63600001036730c55e710d4cbb3d3c080017c303b1012201789c63606a3ac8c0577e9481bd622d8327d7060686ad150c53a3ff0300554707db031800000457000008ae00000d050000115c000015b300001a0a\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"QueryShortChannelIds\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"UNCOMPRESSED\",\n" " \"array\" : [ \"0x0x142\", \"0x0x15465\", \"0x69x42692\" ]\n" " },\n" " \"extensions\" : [ ]\n" " },\n" " \"hex\" : \"01050f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206001900000000000000008e0000000000003c69000000000045a6c4\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"QueryShortChannelIds\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"array\" : [ \"0x0x4564\", \"0x2x47550\", \"0x69x42692\" ]\n" " },\n" " \"extensions\" : [ ]\n" " },\n" " \"hex\" : \"01050f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206001801789c63600001c12b608a69e73e30edbaec0800203b040e\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"QueryShortChannelIds\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"UNCOMPRESSED\",\n" " \"array\" : [ \"0x0x12232\", \"0x0x15556\", \"0x69x42692\" ]\n" " },\n" " \"extensions\" : [ {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"array\" : [ 1, 2, 4 ]\n" " } ]\n" " },\n" " \"hex\" : \"01050f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e22060019000000000000002fc80000000000003cc4000000000045a6c4010c01789c6364620100000e0008\"\n" "}\n", "{\n" " \"msg\" : {\n" " \"type\" : \"QueryShortChannelIds\",\n" " \"chainHash\" : \"0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206\",\n" " \"shortChannelIds\" : {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"array\" : [ \"0x0x14200\", \"0x0x46645\", \"0x69x42692\" ]\n" " },\n" " \"extensions\" : [ {\n" " \"encoding\" : \"COMPRESSED_ZLIB\",\n" " \"array\" : [ 1, 2, 4 ]\n" " } ]\n" " },\n" " \"hex\" : \"01050f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206001801789c63600001f30a30c5b0cd144cb92e3b020017c6034a010c01789c6364620100000e0008\"\n" "}\n", }; static void get_chainhash(const char *test_vector, const jsmntok_t *obj, struct bitcoin_blkid *chain_hash) { const jsmntok_t *tok = json_get_member(test_vector, obj, "chainHash"); size_t hexlen = tok->end - tok->start; assert(hex_decode(test_vector + tok->start, hexlen, chain_hash, sizeof(*chain_hash))); } static u8 *get_scid_array(const tal_t *ctx, const char *test_vector, const jsmntok_t *obj) { const jsmntok_t *scids, *arr, *encoding, *t; size_t i; u8 *encoded; scids = json_get_member(test_vector, obj, "shortChannelIds"); arr = json_get_member(test_vector, scids, "array"); encoded = encoding_start(ctx); encoding = json_get_member(test_vector, scids, "encoding"); json_for_each_arr(i, t, arr) { struct short_channel_id scid; assert(json_to_short_channel_id(test_vector, t, &scid)); encoding_add_short_channel_id(&encoded, &scid); } if (json_tok_streq(test_vector, encoding, "UNCOMPRESSED")) { encoding_end_no_compress(&encoded, 1); encoded[0] = ARR_UNCOMPRESSED; } else { assert(json_tok_streq(test_vector, encoding, "COMPRESSED_ZLIB")); assert(encoding_end_zlib(&encoded, 1)); encoded[0] = ARR_ZLIB; } return encoded; } static u8 *test_query_channel_range(const char *test_vector, const jsmntok_t *obj) { struct bitcoin_blkid chain_hash; u32 firstBlockNum, numberOfBlocks; const jsmntok_t *opt, *t; struct tlv_query_channel_range_tlvs *tlvs = tlv_query_channel_range_tlvs_new(NULL); u8 *msg; size_t i; get_chainhash(test_vector, obj, &chain_hash); assert(json_to_number(test_vector, json_get_member(test_vector, obj, "firstBlockNum"), &firstBlockNum)); assert(json_to_number(test_vector, json_get_member(test_vector, obj, "numberOfBlocks"), &numberOfBlocks)); opt = json_get_member(test_vector, obj, "extensions"); json_for_each_arr(i, t, opt) { assert(json_tok_streq(test_vector, t, "WANT_TIMESTAMPS | WANT_CHECKSUMS")); tlvs->query_option = tal(tlvs, struct tlv_query_channel_range_tlvs_query_option); tlvs->query_option->query_option_flags = QUERY_ADD_TIMESTAMPS | QUERY_ADD_CHECKSUMS; } msg = towire_query_channel_range(NULL, &chain_hash, firstBlockNum, numberOfBlocks, tlvs); tal_free(tlvs); return msg; } static u8 *test_reply_channel_range(const char *test_vector, const jsmntok_t *obj) { struct bitcoin_blkid chain_hash; u32 firstBlockNum, numberOfBlocks, complete; const jsmntok_t *opt, *t; size_t i; u8 *msg; u8 *encoded_scids; u8 *ctx = tal(NULL, u8); struct tlv_reply_channel_range_tlvs *tlvs = tlv_reply_channel_range_tlvs_new(ctx); get_chainhash(test_vector, obj, &chain_hash); assert(json_to_number(test_vector, json_get_member(test_vector, obj, "firstBlockNum"), &firstBlockNum)); assert(json_to_number(test_vector, json_get_member(test_vector, obj, "numberOfBlocks"), &numberOfBlocks)); assert(json_to_number(test_vector, json_get_member(test_vector, obj, "complete"), &complete)); encoded_scids = get_scid_array(ctx, test_vector, obj); opt = json_get_member(test_vector, obj, "timestamps"); if (opt) { const jsmntok_t *encodingtok, *tstok; tlvs->timestamps_tlv = tal(tlvs, struct tlv_reply_channel_range_tlvs_timestamps_tlv); tlvs->timestamps_tlv->encoded_timestamps = encoding_start(tlvs->timestamps_tlv); encodingtok = json_get_member(test_vector, opt, "encoding"); tstok = json_get_member(test_vector, opt, "timestamps"); json_for_each_arr(i, t, tstok) { struct channel_update_timestamps ts; assert(json_to_number(test_vector, json_get_member(test_vector, t, "timestamp1"), &ts.timestamp_node_id_1)); assert(json_to_number(test_vector, json_get_member(test_vector, t, "timestamp2"), &ts.timestamp_node_id_2)); encoding_add_timestamps(&tlvs->timestamps_tlv->encoded_timestamps, &ts); } if (json_tok_streq(test_vector, encodingtok, "UNCOMPRESSED")) { encoding_end_no_compress(&tlvs->timestamps_tlv->encoded_timestamps, 0); tlvs->timestamps_tlv->encoding_type = ARR_UNCOMPRESSED; } else { assert(json_tok_streq(test_vector, encodingtok, "COMPRESSED_ZLIB")); assert(encoding_end_zlib(&tlvs->timestamps_tlv->encoded_timestamps, 0)); tlvs->timestamps_tlv->encoding_type = ARR_ZLIB; } } opt = json_get_member(test_vector, obj, "checksums"); if (opt) { const jsmntok_t *cstok; tlvs->checksums_tlv = tal(tlvs, struct tlv_reply_channel_range_tlvs_checksums_tlv); tlvs->checksums_tlv->checksums = tal_arr(tlvs->checksums_tlv, struct channel_update_checksums, 0); cstok = json_get_member(test_vector, opt, "checksums"); json_for_each_arr(i, t, cstok) { struct channel_update_checksums cs; assert(json_to_number(test_vector, json_get_member(test_vector, t, "checksum1"), &cs.checksum_node_id_1)); assert(json_to_number(test_vector, json_get_member(test_vector, t, "checksum2"), &cs.checksum_node_id_2)); tal_arr_expand(&tlvs->checksums_tlv->checksums, cs); } } msg = towire_reply_channel_range( NULL, &chain_hash, firstBlockNum, numberOfBlocks, complete, encoded_scids, tlvs); tal_free(ctx); return msg; } static struct tlv_query_short_channel_ids_tlvs_query_flags * get_query_flags_array(const tal_t *ctx, const char *test_vector, const jsmntok_t *opt) { const jsmntok_t *encoding, *arr, *t; struct tlv_query_short_channel_ids_tlvs_query_flags *tlv = tal(ctx, struct tlv_query_short_channel_ids_tlvs_query_flags); size_t i; arr = json_get_member(test_vector, opt, "array"); tlv->encoded_query_flags = encoding_start(tlv); encoding = json_get_member(test_vector, opt, "encoding"); json_for_each_arr(i, t, arr) { bigsize_t f; assert(json_to_u64(test_vector, t, &f)); encoding_add_query_flag(&tlv->encoded_query_flags, f); } if (json_tok_streq(test_vector, encoding, "UNCOMPRESSED")) { encoding_end_no_compress(&tlv->encoded_query_flags, 0); tlv->encoding_type = ARR_UNCOMPRESSED; } else { assert(json_tok_streq(test_vector, encoding, "COMPRESSED_ZLIB")); assert(encoding_end_zlib(&tlv->encoded_query_flags, 0)); tlv->encoding_type = ARR_ZLIB; } return tlv; } static u8 *test_query_short_channel_ids(const char *test_vector, const jsmntok_t *obj) { struct bitcoin_blkid chain_hash; const jsmntok_t *opt; u8 *encoded, *msg; struct tlv_query_short_channel_ids_tlvs *tlvs; get_chainhash(test_vector, obj, &chain_hash); encoded = get_scid_array(NULL, test_vector, obj); opt = json_get_member(test_vector, obj, "extensions"); if (opt && opt->size != 0) { /* Only handle a single */ assert(opt->size == 1); tlvs = tlv_query_short_channel_ids_tlvs_new(encoded); tlvs->query_flags = get_query_flags_array(tlvs, test_vector, opt + 1); } else tlvs = NULL; msg = towire_query_short_channel_ids(NULL, &chain_hash, encoded, tlvs); tal_free(encoded); return msg; } int main(void) { jsmntok_t *toks = tal_arr(NULL, jsmntok_t, 1000); setup_locale(); for (size_t i = 0; i < ARRAY_SIZE(test_vectors); i++) { jsmn_parser parser; u8 *m; const char *hex_m; const jsmntok_t *msg, *type, *hex; jsmn_init(&parser); assert(jsmn_parse(&parser, test_vectors[i], strlen(test_vectors[i]), toks, tal_count(toks)) > 0); msg = json_get_member(test_vectors[i], toks, "msg"); hex = json_get_member(test_vectors[i], toks, "hex"); type = json_get_member(test_vectors[i], msg, "type"); if (json_tok_streq(test_vectors[i], type, "QueryChannelRange")) m = test_query_channel_range(test_vectors[i], msg); else if (json_tok_streq(test_vectors[i], type, "ReplyChannelRange")) m = test_reply_channel_range(test_vectors[i], msg); else if (json_tok_streq(test_vectors[i], type, "QueryShortChannelIds")) m = test_query_short_channel_ids(test_vectors[i], msg); else abort(); hex_m = tal_hex(m, m); assert(json_tok_streq(test_vectors[i], hex, hex_m)); tal_free(m); } tal_free(toks); return 0; }