#include <stdio.h>
#include <stdlib.h>

#include "test.h"
#include "ikcp.c"
#include "UdpSocket.cpp"

UdpSocket* pls1 = NULL;
UdpSocket* pls2 = NULL;

void printbuff ( uint8_t* buf, int len )
{
    printf ( "____________________\n" );

    for ( int i = 0 ; i < len; ++i )
    {
        if ( i % 0x10 == 0 && i != 0 )
            printf ( "\n" );

        printf ( "0x%02x ", buf[i] );
    }

    printf ( "\n____________________\n" );
}

int udp_output_zt ( const char* buf, int len, ikcpcb* kcp, void* user )
{
    ( void ) kcp;
    UdpSocket* p = reinterpret_cast<UdpSocket*> ( user );
    p->Send ( reinterpret_cast<const uint8_t*> ( buf ), len );
    return 0;
}

// 测试用例
void test()
{
    ikcpcb* kcp1 = ikcp_create ( 0x11223344, ( void* ) pls1 );
    ikcpcb* kcp2 = ikcp_create ( 0x11223344, ( void* ) pls2 );

    kcp1->output = udp_output_zt;
    kcp2->output = udp_output_zt;

    IUINT32 current = iclock();
    IUINT32 slap = current + 20;
    IUINT32 index = 0;
    IUINT32 next = 0;
    int count = 0;

    ikcp_wndsize ( kcp1, 128, 128 );
    ikcp_wndsize ( kcp2, 128, 128 );

    ikcp_nodelay ( kcp1, 1, 10, 2, 1 );
    ikcp_nodelay ( kcp2, 1, 10, 2, 1 );
    kcp1->rx_minrto = 10;
    kcp1->fastresend = 1;

    char buffer[2000];
    int hr;

    while ( 1 )
    {
        isleep ( 1 );
        current = iclock();
        ikcp_update ( kcp1, iclock() );
        ikcp_update ( kcp2, iclock() );

        for ( ; current >= slap; slap += 20 )
        {
            ( ( IUINT32* ) buffer ) [0] = index++;
            ( ( IUINT32* ) buffer ) [1] = current;

            ikcp_send ( kcp1, buffer, 8 );
        }

        while ( 1 )
        {
            hr = pls2->Recv ( reinterpret_cast<uint8_t*> ( buffer ), 2000 );

            if ( hr < 0 )
                break;


            ikcp_input ( kcp2, buffer, hr );
        }

        while ( 1 )
        {
            hr = pls1->Recv ( reinterpret_cast<uint8_t*> ( buffer ), 2000 );

            if ( hr < 0 )
                break;

            printbuff ( reinterpret_cast<uint8_t*> ( buffer ), hr );

            ikcp_input ( kcp1, buffer, hr );
        }

        while ( 1 )
        {
            hr = ikcp_recv ( kcp2, buffer, 10 );

            if ( hr < 0 )
                break;

            ikcp_send ( kcp2, buffer, hr );
        }

        while ( 1 )
        {
            hr = ikcp_recv ( kcp1, buffer, 10 );

            if ( hr < 0 )
                break;

            IUINT32 sn = * ( IUINT32* ) ( buffer + 0 );
            IUINT32 ts = * ( IUINT32* ) ( buffer + 4 );

            if ( sn != next )
            {
                printf ( "ERROR sn %d<->%d\n", ( int ) count, ( int ) next );
                return;
            }

            next++;
            count++;

        }

        if ( next > 2 )
            break;
    }

    ikcp_release ( kcp1 );
    ikcp_release ( kcp2 );
}

int main ( int argc, char* argv[] )
{
    ( void ) argc;
    ( void ) argv;
    pls1 = new UdpSocket ( 32345, "127.0.0.1", 32346 );
    pls2 = new UdpSocket ( 32346, "127.0.0.1", 32345 );

    test();
    return 0;
}