test_circuitmux.c 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. /* Copyright (c) 2013-2017, The Tor Project, Inc. */
  2. /* See LICENSE for licensing information */
  3. #define TOR_CHANNEL_INTERNAL_
  4. #define CIRCUITMUX_PRIVATE
  5. #define RELAY_PRIVATE
  6. #include "or.h"
  7. #include "channel.h"
  8. #include "circuitmux.h"
  9. #include "relay.h"
  10. #include "scheduler.h"
  11. #include "test.h"
  12. /* XXXX duplicated function from test_circuitlist.c */
  13. static channel_t *
  14. new_fake_channel(void)
  15. {
  16. channel_t *chan = tor_malloc_zero(sizeof(channel_t));
  17. channel_init(chan);
  18. return chan;
  19. }
  20. static int
  21. has_queued_writes(channel_t *c)
  22. {
  23. (void) c;
  24. return 1;
  25. }
  26. /** Test destroy cell queue with no interference from other queues. */
  27. static void
  28. test_cmux_destroy_cell_queue(void *arg)
  29. {
  30. circuitmux_t *cmux = NULL;
  31. channel_t *ch = NULL;
  32. circuit_t *circ = NULL;
  33. destroy_cell_queue_t *cq = NULL;
  34. packed_cell_t *pc = NULL;
  35. destroy_cell_t *dc = NULL;
  36. scheduler_init();
  37. (void) arg;
  38. cmux = circuitmux_alloc();
  39. tt_assert(cmux);
  40. ch = new_fake_channel();
  41. ch->has_queued_writes = has_queued_writes;
  42. ch->wide_circ_ids = 1;
  43. circ = circuitmux_get_first_active_circuit(cmux, &cq);
  44. tt_ptr_op(circ, OP_EQ, NULL);
  45. tt_ptr_op(cq, OP_EQ, NULL);
  46. circuitmux_append_destroy_cell(ch, cmux, 100, 10);
  47. circuitmux_append_destroy_cell(ch, cmux, 190, 6);
  48. circuitmux_append_destroy_cell(ch, cmux, 30, 1);
  49. tt_int_op(circuitmux_num_cells(cmux), OP_EQ, 3);
  50. circ = circuitmux_get_first_active_circuit(cmux, &cq);
  51. tt_ptr_op(circ, OP_EQ, NULL);
  52. tt_assert(cq);
  53. tt_int_op(cq->n, OP_EQ, 3);
  54. dc = destroy_cell_queue_pop(cq);
  55. tt_assert(dc);
  56. tt_uint_op(dc->circid, OP_EQ, 100);
  57. tt_int_op(circuitmux_num_cells(cmux), OP_EQ, 2);
  58. done:
  59. circuitmux_free(cmux);
  60. channel_free(ch);
  61. packed_cell_free(pc);
  62. tor_free(dc);
  63. }
  64. struct testcase_t circuitmux_tests[] = {
  65. { "destroy_cell_queue", test_cmux_destroy_cell_queue, TT_FORK, NULL, NULL },
  66. END_OF_TESTCASES
  67. };