static circuit_t * dummy_or_circuit_new(int n_p_cells, int n_n_cells) { or_circuit_t *circ = or_circuit_new(0, NULL); int i; cell_t cell; for (i=0; i < n_p_cells; ++i) { crypto_rand((void*)&cell, sizeof(cell)); cell_queue_append_packed_copy(TO_CIRCUIT(circ), &circ->p_chan_cells, 0, &cell, 1, 0); } for (i=0; i < n_n_cells; ++i) { crypto_rand((void*)&cell, sizeof(cell)); cell_queue_append_packed_copy(TO_CIRCUIT(circ), &TO_CIRCUIT(circ)->n_chan_cells, 1, &cell, 1, 0); } TO_CIRCUIT(circ)->purpose = CIRCUIT_PURPOSE_OR; return TO_CIRCUIT(circ); }
/* Set up a fake origin circuit with the specified number of cells, * Return a pointer to the newly-created dummy circuit */ circuit_t * dummy_origin_circuit_new(int n_cells) { origin_circuit_t *circ = origin_circuit_new(); int i; cell_t cell; for (i=0; i < n_cells; ++i) { crypto_rand((void*)&cell, sizeof(cell)); cell_queue_append_packed_copy(TO_CIRCUIT(circ), &TO_CIRCUIT(circ)->n_chan_cells, 1, &cell, 1, 0); } TO_CIRCUIT(circ)->purpose = CIRCUIT_PURPOSE_C_GENERAL; return TO_CIRCUIT(circ); }
static void test_cq_manip(void *arg) { packed_cell_t *pc1=NULL, *pc2=NULL, *pc3=NULL, *pc4=NULL, *pc_tmp=NULL; cell_queue_t cq; cell_t cell; (void) arg; cell_queue_init(&cq); tt_int_op(cq.n, OP_EQ, 0); pc1 = packed_cell_new(); pc2 = packed_cell_new(); pc3 = packed_cell_new(); pc4 = packed_cell_new(); tt_assert(pc1 && pc2 && pc3 && pc4); tt_ptr_op(NULL, OP_EQ, cell_queue_pop(&cq)); /* Add and remove a singleton. */ cell_queue_append(&cq, pc1); tt_int_op(cq.n, OP_EQ, 1); tt_ptr_op(pc1, OP_EQ, cell_queue_pop(&cq)); tt_int_op(cq.n, OP_EQ, 0); /* Add and remove four items */ cell_queue_append(&cq, pc4); cell_queue_append(&cq, pc3); cell_queue_append(&cq, pc2); cell_queue_append(&cq, pc1); tt_int_op(cq.n, OP_EQ, 4); tt_ptr_op(pc4, OP_EQ, cell_queue_pop(&cq)); tt_ptr_op(pc3, OP_EQ, cell_queue_pop(&cq)); tt_ptr_op(pc2, OP_EQ, cell_queue_pop(&cq)); tt_ptr_op(pc1, OP_EQ, cell_queue_pop(&cq)); tt_int_op(cq.n, OP_EQ, 0); tt_ptr_op(NULL, OP_EQ, cell_queue_pop(&cq)); /* Try a packed copy (wide, then narrow, which is a bit of a cheat, since a * real cell queue has only one type.) */ memset(&cell, 0, sizeof(cell)); cell.circ_id = 0x12345678; cell.command = 10; strlcpy((char*)cell.payload, "Lorax ipsum gruvvulus thneed amet, snergelly " "once-ler lerkim, sed do barbaloot tempor gluppitus ut labore et " "truffula magna aliqua.", sizeof(cell.payload)); cell_queue_append_packed_copy(NULL /*circ*/, &cq, 0 /*exitward*/, &cell, 1 /*wide*/, 0 /*stats*/); cell.circ_id = 0x2013; cell_queue_append_packed_copy(NULL /*circ*/, &cq, 0 /*exitward*/, &cell, 0 /*wide*/, 0 /*stats*/); tt_int_op(cq.n, OP_EQ, 2); pc_tmp = cell_queue_pop(&cq); tt_int_op(cq.n, OP_EQ, 1); tt_ptr_op(pc_tmp, OP_NE, NULL); tt_mem_op(pc_tmp->body, OP_EQ, "\x12\x34\x56\x78\x0a", 5); tt_mem_op(pc_tmp->body+5, OP_EQ, cell.payload, sizeof(cell.payload)); packed_cell_free(pc_tmp); pc_tmp = cell_queue_pop(&cq); tt_int_op(cq.n, OP_EQ, 0); tt_ptr_op(pc_tmp, OP_NE, NULL); tt_mem_op(pc_tmp->body, OP_EQ, "\x20\x13\x0a", 3); tt_mem_op(pc_tmp->body+3, OP_EQ, cell.payload, sizeof(cell.payload)); packed_cell_free(pc_tmp); pc_tmp = NULL; tt_ptr_op(NULL, OP_EQ, cell_queue_pop(&cq)); /* Now make sure cell_queue_clear works. */ cell_queue_append(&cq, pc2); cell_queue_append(&cq, pc1); tt_int_op(cq.n, OP_EQ, 2); cell_queue_clear(&cq); pc2 = pc1 = NULL; /* prevent double-free */ tt_int_op(cq.n, OP_EQ, 0); done: packed_cell_free(pc1); packed_cell_free(pc2); packed_cell_free(pc3); packed_cell_free(pc4); packed_cell_free(pc_tmp); cell_queue_clear(&cq); }