/* 
 * Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com)
 * Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
 */

#include "test.h"
#include <pjsip.h>
#include <pjlib.h>

#define THIS_FILE   "transport_udp_test.c"

static pj_status_t multi_transport_test(pjsip_transport *tp[], unsigned num_tp)
{
    pj_status_t status;
    unsigned i = 0;
    pj_str_t s;
    pjsip_transport *udp_tp;
    pj_sockaddr_in rem_addr;    
    pjsip_tpselector tp_sel;

    for (;i<num_tp;++i)
    {
        pj_sockaddr_in addr;

        pj_sockaddr_in_init(&addr, NULL, (pj_uint16_t)(TEST_UDP_PORT+i));

        /* Start UDP transport. */
        status = pjsip_udp_transport_start( endpt, &addr, NULL, 1, &udp_tp);
        if (status != PJ_SUCCESS) {
            app_perror("   Error: unable to start UDP transport", status);
            return -110;
        }

        /* UDP transport must have initial reference counter set to 1. */
        if (pj_atomic_get(udp_tp->ref_cnt) != 1)
            return -120;

        /* Test basic transport attributes */
        status = generic_transport_test(udp_tp);
        if (status != PJ_SUCCESS)
            return status;

        tp[i] = udp_tp;
    }

    for (i = 0; i < num_tp; ++i) {
        udp_tp = tp[i];
        if (pj_atomic_get(udp_tp->ref_cnt) != 1)
            return -130;
    }

    /* Acquire transport test without selector. */
    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "1.1.1.1"), 80);
    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_UDP,
                                           &rem_addr, sizeof(rem_addr),
                                           NULL, &udp_tp);
    if (status != PJ_SUCCESS)
        return -140;

    for (i = 0; i < num_tp; ++i) {
        if (udp_tp == tp[i]) {
            break;
        }
    }
    if (i == num_tp)
        return -150;

    pjsip_transport_dec_ref(udp_tp);

    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
        return -160;

    /* Acquire transport test with selector. */
    pj_bzero(&tp_sel, sizeof(tp_sel));
    tp_sel.type = PJSIP_TPSELECTOR_TRANSPORT;
    tp_sel.u.transport = tp[num_tp-1];
    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "1.1.1.1"), 80);
    status = pjsip_endpt_acquire_transport(endpt, PJSIP_TRANSPORT_UDP,
                                           &rem_addr, sizeof(rem_addr),
                                           &tp_sel, &udp_tp);
    if (status != PJ_SUCCESS)
        return -170;

    if (udp_tp != tp[num_tp-1])
        return -180;

    pjsip_transport_dec_ref(udp_tp);

    if (pj_atomic_get(udp_tp->ref_cnt) != 1)
        return -190;

    return PJ_SUCCESS;
}

/*
 * UDP transport test.
 */
int transport_udp_test(void)
{
    enum { SEND_RECV_LOOP = 8 };
    enum { NUM_TP = 4 };
    pjsip_transport *tp[NUM_TP], *udp_tp;
    pj_sockaddr_in rem_addr;
    pj_str_t s;
    pj_status_t status;
    int rtt[SEND_RECV_LOOP], min_rtt;
    int i, pkt_lost;

    status = multi_transport_test(&tp[0], NUM_TP);
    if (status != PJ_SUCCESS)
        return status;

    /* Basic transport's send/receive loopback test. */
    pj_sockaddr_in_init(&rem_addr, pj_cstr(&s, "127.0.0.1"), TEST_UDP_PORT);
    for (i=0; i<SEND_RECV_LOOP; ++i) {
        status = transport_send_recv_test(PJSIP_TRANSPORT_UDP, tp[0], 
                                          "sip:alice@127.0.0.1:"TEST_UDP_PORT_STR,
                                          &rtt[i]);
        if (status != 0)
            return status;
    }

    min_rtt = 0xFFFFFFF;
    for (i=0; i<SEND_RECV_LOOP; ++i)
        if (rtt[i] < min_rtt) min_rtt = rtt[i];

    report_ival("udp-rtt-usec", min_rtt, "usec",
                "Best UDP transport round trip time, in microseconds "
                "(time from sending request until response is received. "
                "Tests were performed on local machine only)");


    /* Multi-threaded round-trip test. */
    status = transport_rt_test(PJSIP_TRANSPORT_UDP, tp[0], 
                               "sip:alice@127.0.0.1:"TEST_UDP_PORT_STR, 
                               &pkt_lost);
    if (status != 0)
        return status;

    if (pkt_lost != 0)
        PJ_LOG(3,(THIS_FILE, "   note: %d packet(s) was lost", pkt_lost));

    for (i = 0; i < NUM_TP; ++i) {
        udp_tp = tp[i];

        /* Check again that reference counter is 1. */
        if (pj_atomic_get(udp_tp->ref_cnt) != 1)
            return -80;

        /* Destroy this transport. */
        pjsip_transport_dec_ref(udp_tp);
        status = pjsip_transport_destroy(udp_tp);
        if (status != PJ_SUCCESS)
            return -90;
    }

    /* Flush events. */
    PJ_LOG(3,(THIS_FILE, "   Flushing events, 1 second..."));
    flush_events(1000);

    /* Done */
    return 0;
}
