diff --git a/test/sipsess.c b/test/sipsess.c index c0051fb0b..ae7613207 100644 --- a/test/sipsess.c +++ b/test/sipsess.c @@ -40,12 +40,25 @@ enum connect_action { }; +enum answer_action { + ANSW_NONE = 0, + ANSW_CANCEL = 1, +}; + + +enum offer_action { + OFFER_NONE = 0, + OFFER_ANSW = 1, +}; + + struct test { struct sip *sip; struct sipsess_sock *sock; struct sipsess *a; struct sipsess *b; struct tmr ans_tmr; + struct tmr ack_tmr; bool estab_a; bool estab_b; bool answr_a; @@ -54,15 +67,20 @@ struct test { bool progr_b; bool offer_a; bool offer_b; + bool ack_a; + bool ack_b; enum rel100_mode rel100_a; enum rel100_mode rel100_b; enum sdp_neg_state sdp_state; enum rel100_state rel100_state_a; enum rel100_state rel100_state_b; enum connect_action conn_action; + enum offer_action offer_action; + enum answer_action answ_action; prack_func *prack_action; int progr_ret_code; int answ_ret_code; + int ack_cnt; bool upd_a; bool upd_b; struct mbuf *desc; @@ -119,6 +137,34 @@ static void exit_handler(void *arg) } +static void check_ack(void *arg) +{ + struct test *test = arg; + bool ack_a = sipsess_ack_pending(test->a); + bool ack_b = sipsess_ack_pending(test->b); + + test->ack_a |= ack_a; + test->ack_b |= ack_b; + + if (ack_a || ack_b) + tmr_start(&test->ack_tmr, 1, check_ack, test); + else + stop_test(); +} + + +static void wait_for_ack(struct test *test) +{ + test->ack_a = sipsess_ack_pending(test->a); + test->ack_b = sipsess_ack_pending(test->b); + + if (!test->ack_a && !test->ack_b) + return; + + tmr_start(&test->ack_tmr, 1, check_ack, test); +} + + static void send_answer_b(void *arg) { struct test *test = arg; @@ -131,21 +177,37 @@ static void send_answer_b(void *arg) } -static void send_update_a(void *arg) +static int make_sdp(struct mbuf **mbp, const char *sdp) { - struct test *test = arg; struct mbuf *desc; int err; - desc = mbuf_alloc(sizeof(sdp_a)); - if (!desc) { - err = ENOMEM; - goto out; - } - err = mbuf_write_str(desc, sdp_a); + desc = mbuf_alloc(strlen(sdp) + 1); + if (!desc) + return ENOMEM; + + err = mbuf_write_str(desc, sdp); TEST_ERR(err); mbuf_set_pos(desc, 0); +out: + if (err) + mem_deref(desc); + else + *mbp = desc; + + return err; +} + + +static void send_update_a(void *arg) +{ + struct test *test = arg; + struct mbuf *desc; + int err; + + err = make_sdp(&desc, sdp_a); + TEST_ERR(err); err = sipsess_modify(test->a, desc); TEST_ERR(err); @@ -164,16 +226,9 @@ static void send_update_b(void *arg) struct mbuf *desc; int err; - desc = mbuf_alloc(sizeof(sdp_b)); - if (!desc) { - err = ENOMEM; - goto out; - } - err = mbuf_write_str(desc, sdp_b); + err = make_sdp(&desc, sdp_b); TEST_ERR(err); - mbuf_set_pos(desc, 0); - err = sipsess_modify(test->b, desc); TEST_ERR(err); @@ -210,17 +265,9 @@ static int desc_handler_a(struct mbuf **descp, const struct sa *src, (void)dst; (void)arg; - desc = mbuf_alloc(sizeof(sdp_a)); - if (!desc) { - err = ENOMEM; - goto out; - } - - err = mbuf_write_str(desc, sdp_a); - if (err) - goto out; + err = make_sdp(&desc, sdp_a); + TEST_ERR(err); - mbuf_set_pos(desc, 0); *descp = desc; out: @@ -251,8 +298,8 @@ static int offer_handler_b(struct mbuf **descp, const struct sip_msg *msg, void *arg) { struct test *test = arg; - (void)descp; (void)msg; + int err = 0; if (test->sdp_state == INITIAL || test->sdp_state == EARLY_CONFIRMED) test->sdp_state = OFFER_RECEIVED; @@ -262,7 +309,13 @@ static int offer_handler_b(struct mbuf **descp, const struct sip_msg *msg, test->offer_b = true; - return 0; + if (test->offer_action == OFFER_ANSW) { + err = make_sdp(descp, sdp_b); + TEST_ERR(err); + } + +out: + return err; } @@ -288,6 +341,9 @@ static int answer_handler_a(const struct sip_msg *msg, void *arg) tmr_start(&test->ans_tmr, 0, send_answer_b, test); } + if (test->answ_action == ANSW_CANCEL) + wait_for_ack(test); + return 0; } @@ -393,19 +449,9 @@ static void conn_handler(const struct sip_msg *msg, void *arg) if (sip_msg_hdr_has_value(msg, SIP_HDR_REQUIRE, "100rel")) test->rel100_state_b |= REL100_REQUIRE; - desc = mbuf_alloc(sizeof(sdp_b)); - if (!desc) { - abort_test(test, ENOMEM); - return; - } - - err = mbuf_write_str(desc, sdp_b); - if (err) { - abort_test(test, err); - return; - } + err = make_sdp(&desc, sdp_b); + TEST_ERR(err); - mbuf_set_pos(desc, 0); test->desc = desc; if (test->conn_action & CONN_PROGRESS @@ -561,12 +607,10 @@ int test_sipsess(void) estab_handler_a, NULL, NULL, close_handler, &test, NULL); mem_deref(callid); - if (err) - goto out; + TEST_ERR(err); err = re_main_timeout(200); - if (err) - goto out; + TEST_ERR(err); if (test.err) { err = test.err; @@ -580,6 +624,29 @@ int test_sipsess(void) ASSERT_TRUE(test.answr_a); ASSERT_TRUE(!test.offer_b); + /* test re-invite with wait for ACK */ + test.sdp_state = INITIAL; + test.answ_action = ANSW_CANCEL; + test.offer_action = OFFER_ANSW; + err = make_sdp(&test.desc, sdp_a); + TEST_ERR(err); + err = sipsess_modify(test.a, test.desc); + TEST_ERR(err); + test.desc = mem_deref(test.desc); + + err = re_main_timeout(200); + TEST_ERR(err); + + if (test.err) { + err = test.err; + goto out; + } + + ASSERT_TRUE(test.ack_b); + ASSERT_TRUE(!sipsess_ack_pending(test.a)); + ASSERT_TRUE(!sipsess_ack_pending(test.b)); + ASSERT_TRUE(test.sdp_state == ANSWER_RECEIVED); + out: test.a = mem_deref(test.a); test.b = mem_deref(test.b);