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

#include "ccp.h"
#include "ccp_priv.h"
#include "serialize.h"

#ifdef __GNUC__
    #define UNUSED(x) UNUSED_ ## x __attribute__((__unused__))
#else
    #define UNUSED(x) x
#endif

//===========================================
// Mock Datapath
//===========================================

u32 now_us; // emulated time in microseconds

char expected_sent_msg[BIGGEST_MSG_SIZE];
int expecting_send;

struct test_conn {
    u32 curr_cwnd;
    u32 curr_rate;
};

static void test_ccp_set_cwnd(struct ccp_connection *conn, u32 cwnd) {
    struct test_conn *c = (struct test_conn*) ccp_get_impl(conn);
    c->curr_cwnd = cwnd;
}

static void test_ccp_set_rate(struct ccp_connection *conn, u32 rate) {
    struct test_conn *c = (struct test_conn*) ccp_get_impl(conn);
    c->curr_rate = rate;
}

static int test_ccp_send_msg(struct ccp_datapath *UNUSED(conn), char *msg, int msg_size) {
    if (expecting_send <= 0) {
        printf("FAIL\nNot expecting send");
        goto fail;
    }
    
    if (msg_size != expecting_send) {
        printf("FAIL\nWrong size: expected %d got %d", expecting_send, msg_size);
        goto fail;
    }

    if (memcmp(&expected_sent_msg, msg, msg_size) != 0) {
        printf("FAIL\nWrong msg");
        goto fail;
    }

    expecting_send = 0;
    memset(&expected_sent_msg, 0, BIGGEST_MSG_SIZE);

    return 0;

fail:
    printf(": got: [");
    for (int i = 0; i < msg_size; i++) {
        printf("%02x, ", msg[i]);
    }
    printf("] ");
    if (expecting_send > 0) {
        printf("\n      Expected: [");
        for (int i = 0; i < expecting_send; i++) {
            printf("%02x, ", expected_sent_msg[i]);
        }
        printf("] ");
    }
    return -1;
}

static u64 test_ccp_time_now(void) {
    return now_us;
}

static u64 test_ccp_since_usecs(u64 then) {
    return then - now_us;
}

static u64 test_ccp_after_usecs(u64 usecs) {
    return now_us + usecs;
}

//===========================================
// Test Initialization and Helpers
//===========================================

int test_init(struct ccp_datapath *datapath, struct ccp_connection **conn, struct test_conn *my_conn) {
    char algname[MAX_CONG_ALG_SIZE] = "my-cool-alg";
    // a fake flow arrives!
    struct ccp_datapath_info info = {
        .init_cwnd = 100,
        .mss = 10,
        .src_ip = 1,
        .src_port = 2,
        .dst_ip = 3, 
        .dst_port = 4,
    };
    strncpy((char*) &info.congAlg, (const char*) &algname, MAX_CONG_ALG_SIZE);

    char create_msg[96] = {
        0x00,0x00,
        0x60,0x00,
        0x01,0x00,0x00,0x00,
        0x64,0x00,0x00,0x00,
        0x0a,0x00,0x00,0x00,
        0x01,0x00,0x00,0x00,
        0x02,0x00,0x00,0x00,
        0x03,0x00,0x00,0x00,
        0x04,0x00,0x00,0x00,
    };
    strncpy((char*) &create_msg[32], (const char*) &algname, MAX_CONG_ALG_SIZE);

    memcpy(&expected_sent_msg, &create_msg, 96);
    expecting_send = 96;
    *conn = ccp_connection_start(datapath, (void*) &my_conn, &info);
    if (expecting_send != 0) {
        printf("err: did not send\n");
        return -2;
    } else {
        printf("ok\n");
        return 0;
    }
}

int install_helper(struct ccp_datapath *datapath, char *install_msg, size_t msg_len) {
    int ok = ccp_read_msg(datapath, install_msg, msg_len);
    if (ok < 0) {
        printf("read install message error: %d", ok);
        return -1;
    }

    return 0;
}

int change_prog_helper(struct ccp_datapath *datapath, char *change_prog_msg, size_t msg_len) {
    int ok = ccp_read_msg(datapath, change_prog_msg, msg_len);
    if (ok < 0) {
        printf("read change program message error: %d\n", ok);
        return -1;
    }

    return 0;
}

int update_fields_helper(struct ccp_datapath *datapath, char *update_fields_msg, size_t msg_len) {
    int ok = ccp_read_msg(datapath, update_fields_msg, msg_len);
    if (ok < 0) {
        printf("read update fields message error: %d\n", ok);
        return -1;
    }

    return 0;
}

int getreport_helper(char *expected, size_t msg_len, struct ccp_connection *conn) {
    memcpy(&expected_sent_msg, expected, msg_len);
    expecting_send = msg_len;
    int ok = ccp_invoke(conn);
    if (ok < 0) {
        printf("ccp_invoke error: %d", ok);
        return -1;
    }

    if (expecting_send != 0) {
        printf("did not send");
        memset(&expected_sent_msg, 0, BIGGEST_MSG_SIZE);
        expecting_send = 0;
        return -2;
    }
    
    return 0;
}

// ------------------------------------------
// Tests
// ------------------------------------------

#define BASIC_PROG_UID 1
#define PRIMI_PROG_UID 2
#define MULTI_PROG_UID 3
#define FALLT_PROG_UID 4
#define IMPLI_PROG_UID 5
#define UPDAT_PROG_UID 6

int test_basic(struct ccp_datapath *datapath, struct ccp_connection *conn) {
    int ok;
    char install_msg[116] = {
        2, 0,                                           // INSTALL                                                     
        116, 0,                                         // length = 0x74 = 116
        1, 0, 0, 0,                                     // sock_id = 1                                                 
        BASIC_PROG_UID, 0, 0, 0,                        // program_uid
        1, 0, 0, 0,                                     // num_events = 1                                              
        5, 0, 0, 0,                                     // num_instrs = 5                                              
        1, 0, 0, 0, 1, 0, 0, 0, 2, 0, 0, 0, 3, 0, 0, 0, // event { flag-idx=1, num-flag=1, body-idx=2, num-body=3 }
        2, 5, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 0, 0, 0, 0, // (def (volatile Report.foo 0))
        1, 2, 0, 0, 0, 0, 2, 0, 0, 0, 0, 1, 1, 0, 0, 0, // (when true
        0, 7, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 1, 0, 0, 0, //      ----------------(+ Report.foo 1)
        1, 5, 0, 0, 0, 0, 5, 0, 0, 0, 0, 7, 0, 0, 0, 0, //     (bind Report.foo ^^^^^^^^^^^^^^^^)
        1, 2, 2, 0, 0, 0, 2, 2, 0, 0, 0, 1, 1, 0, 0, 0, //     (bind __shouldReport true)
    };
    char report_msg[24] = {
        0x01, 0x00,                                     // type 
        0x18, 0x00,                                     // len
        0x01, 0x00, 0x00, 0x00,                         // sockid
        BASIC_PROG_UID, 0x00, 0x00, 0x00,               // program_uid
        0x01, 0x00, 0x00, 0x00,                         // num_fields
        0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, // fields
    };

    char change_prog_msg[16] = {
        4, 0,                                           // type
        12, 0,                                          // length 
        1, 0, 0, 0,                                     // sock id
        BASIC_PROG_UID, 0, 0, 0,                        // program_uid
        0, 0, 0, 0,                                     // extra fields
    };

    printf("%s...          ", __func__);
    fflush(stdout);

    ok = install_helper(datapath, install_msg, 116);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = change_prog_helper(datapath, change_prog_msg, 16);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(report_msg, 24, conn);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    printf("ok\n");
    return 0;
}

int test_primitives(struct ccp_datapath *datapath, struct ccp_connection *conn) {
    int ok;
    char install_msg[100] = {
        2, 0,                                                    // INSTALL
        84, 0,                                                   // length = 88
        1, 0, 0, 0,                                              // sock_id = 1
        PRIMI_PROG_UID, 0, 0, 0,                                 // program_uid
        1, 0, 0, 0,                                              // num_events = 1
        4, 0, 0, 0,                                              // num_instrs = 2
        1, 0, 0, 0, 1, 0, 0, 0, 2, 0, 0, 0, 2, 0, 0, 0,          // event { flag-idx=1, num-flag=1, body-idx=2, num-body=2 }
        2, 5, 0, 0, 0, 0, 5, 0,  0, 0, 0, 1, 255, 255, 255, 255, // (def (volatile Report.minrtt +infinity))
        8, 2, 0, 0, 0, 0, 4, 13, 0, 0, 0, 5, 0,   0,   0,   0,   // (when (< Flow.rtt_sample_us Report.minrtt)
        1, 5, 0, 0, 0, 0, 5, 0,  0, 0, 0, 4, 13,  0,   0,   0,   //     (bind Report.minrtt Flow.rtt_sample_us) 
        1, 2, 2, 0, 0, 0, 2, 2,  0, 0, 0, 1, 1,   0,   0,   0,   //     (bind __shouldReport true)
    };
    char report_msg[24] = {
        0x01, 0,
        0x18, 0,
        0x01, 0x00, 0x00, 0x00,
        PRIMI_PROG_UID, 0x00, 0x00, 0x00,                        // program_uid
        0x01, 0x00, 0x00, 0x00,
        0xed, 0xfe, 0xfe, 0xca, 0x00, 0x00, 0x00, 0x00,
    };
    
    char change_prog_msg[16] = {
        4, 0,                                                    // type
        12, 0,                                                   // length 
        1, 0, 0, 0,                                              // sock id
        PRIMI_PROG_UID, 0, 0, 0,                                 // program_uid
        0, 0, 0, 0,                                              // extra fields
    };

    printf("%s...     ", __func__);

    fflush(stdout);

    conn->prims.rtt_sample_us = 0xcafefeed;

    ok = install_helper(datapath, install_msg, 88);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = change_prog_helper(datapath, change_prog_msg, 16);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(report_msg, 24, conn);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    printf("ok\n");
    return 0;
}

int test_multievent(struct ccp_datapath *datapath, struct ccp_connection *conn) {
    int ok;
    char install_msg[180] = {
        2, 0,                                                    // INSTALL
        156, 0,                                                  // length = 156
        1, 0, 0, 0,                                              // sock_id = 1
        MULTI_PROG_UID, 0, 0, 0,                                 // program_uid
        2, 0, 0, 0,                                              // num_events = 2
        8, 0, 0, 0,                                              // num_instrs = 8
        2, 0, 0, 0, 1, 0, 0, 0, 3, 0, 0, 0, 2, 0, 0, 0,          // event { flag-idx=2, num-flag=1, body-idx=3, num-body=2 }
        5, 0, 0, 0, 1, 0, 0, 0, 6, 0, 0, 0, 2, 0, 0, 0,          // event { flag-idx=5, num-flag=1, body-idx=6, num-body=2 }
        2, 5, 0, 0, 0, 0, 5, 0,  0, 0, 0, 1, 0,   0,   0,   0,   //  -----------------------(volatile Report.counter 0)
        2, 0, 0, 0, 0, 0, 0, 0,  0, 0, 0, 1, 255, 255, 255, 255, // (def (minrtt +infinity) ^^^^^^^^^^^^^^^^^^^^^^^^^^^)
        8, 2, 0, 0, 0, 0, 4, 13, 0, 0, 0, 0, 0,   0,   0,   0,   // (when (< Flow.rtt_sample_us minrtt)
        1, 0, 0, 0, 0, 0, 0, 0,  0, 0, 0, 4, 13,  0,   0,   0,   //     (bind minrtt Flow.rtt_sample_us) 
        1, 2, 2, 0, 0, 0, 2, 2,  0, 0, 0, 1, 1,   0,   0,   0,   //     (bind __shouldReport true))
        1, 2, 0, 0, 0, 0, 2, 0,  0, 0, 0, 1, 1,   0,   0,   0,   // (when true
        0, 7, 0, 0, 0, 0, 5, 0,  0, 0, 0, 1, 1,   0,   0,   0,   //     ---------------------(+ Report.counter 1)-
        1, 5, 0, 0, 0, 0, 5, 0,  0, 0, 0, 7, 0,   0,   0,   0    //     (bind Report.counter ^^^^^^^^^^^^^^^^^^^^))
    };
    char first_report_msg[24] = {
        0x01, 0,
        0x18, 0,
        0x01, 0x00, 0x00, 0x00,
        MULTI_PROG_UID, 0x00, 0x00, 0x00,                        // program_uid
        0x01, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    };
    char report_msg[24] = {
        0x01, 0,
        0x18, 0,
        0x01, 0x00, 0x00, 0x00,
        MULTI_PROG_UID, 0x00, 0x00, 0x00,                        // program_uid
        0x01, 0x00, 0x00, 0x00,
        0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    };
    
    char change_prog_msg[16] = {
        4, 0,                                                    // type
        12, 0,                                                   // length 
        1, 0, 0, 0,                                              // sock id
        MULTI_PROG_UID, 0, 0, 0,                                 // program_uid
        0, 0, 0, 0,                                              // extra fields
    };

    printf("%s...     ", __func__);

    conn->prims.rtt_sample_us = 0xcafefeed;
    fflush(stdout);
    ok = install_helper(datapath, install_msg, 156);
    if (ok < 0) {
        printf("install\n");
        return -1;
    }
    
    ok = change_prog_helper(datapath, change_prog_msg, 16);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(first_report_msg, 24, conn);
    if (ok < 0) {
        printf("failed on first report\n");
        return -1;
    }

    ok = ccp_invoke(conn);
    if (ok < 0) {
        printf("ccp_invoke 1 error: %d", ok);
        return -1;
    }
    
    ok = ccp_invoke(conn);
    if (ok < 0) {
        printf("ccp_invoke 2 error: %d", ok);
        return -1;
    }

    conn->prims.rtt_sample_us = 0xcafefeeb;
    ok = getreport_helper(report_msg, 24, conn);
    if (ok < 0) {
        printf("failed on second report\n");
        return -1;
    }

    printf("ok\n");
    return 0;
}

int test_fallthrough(struct ccp_datapath *datapath, struct ccp_connection *conn) {
    int ok;
    char install_msg[196] = {
        2, 0,                                                    // INSTALL
        172, 0,                                                  // length = 172
        1, 0, 0, 0,                                              // sock_id = 1
        FALLT_PROG_UID, 0, 0, 0,                                 // program_uid
        2, 0, 0, 0,                                              // num_events = 2
        9, 0, 0, 0,                                              // num_instrs = 9
        2, 0, 0, 0, 1, 0, 0, 0, 3, 0, 0, 0, 3, 0, 0, 0,          // event { flag-idx=2, num-flag=1, body-idx=3, num-body=3 }
        6, 0, 0, 0, 1, 0, 0, 0, 7, 0, 0, 0, 2, 0, 0, 0,          // event { flag-idx=6, num-flag=1, body-idx=7, num-body=2 }
        2, 0, 0, 0, 0, 0, 0, 0,  0, 0, 0, 1, 255, 255, 255, 255, // -----(minrtt +infinity)-----------------------------
        2, 5, 0, 0, 0, 0, 5, 0,  0, 0, 0, 1, 0,   0,   0,   0,   // (def ^^^^^^^^^^^^^^^^^^ (volatile Report.counter 0))
        8, 2, 0, 0, 0, 0, 4, 13, 0, 0, 0, 0, 0,   0,   0,   0,   // (when (< Flow.rtt_sample_us minrtt)
        1, 0, 0, 0, 0, 0, 0, 0,  0, 0, 0, 4, 13,  0,   0,   0,   //     (bind minrtt Flow.rtt_sample_us) 
        1, 2, 2, 0, 0, 0, 2, 2,  0, 0, 0, 1, 1,   0,   0,   0,   //     (bind __shouldReport true)
        1, 2, 1, 0, 0, 0, 2, 1,  0, 0, 0, 1, 1,   0,   0,   0,   //     (bind __shouldContinue true))
        1, 2, 0, 0, 0, 0, 2, 0,  0, 0, 0, 1, 1,   0,   0,   0,   // (when true
        0, 7, 0, 0, 0, 0, 5, 0,  0, 0, 0, 1, 1,   0,   0,   0,   //     ---------------------(+ Report.counter 1)-
        1, 5, 0, 0, 0, 0, 5, 0,  0, 0, 0, 7, 0,   0,   0,   0,   //     (bind Report.counter ^^^^^^^^^^^^^^^^^^^^))
    };
    char first_report_msg[24] = {
        0x01, 0,
        0x18, 0,
        0x01, 0x00, 0x00, 0x00,
        FALLT_PROG_UID, 0x00, 0x00, 0x00,                        // program_uid
        0x01, 0x00, 0x00, 0x00,
        0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    };
    char report_msg[24] = {
        0x01, 0,
        0x18, 0,
        0x01, 0x00, 0x00, 0x00,
        FALLT_PROG_UID, 0x00, 0x00, 0x00,                        // program_uid
        0x01, 0x00, 0x00, 0x00,
        0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    };
    char change_prog_msg[16] = {
        4, 0,                                                    // type
        12, 0,                                                   // length 
        1, 0, 0, 0,                                              // sock id
        FALLT_PROG_UID, 0, 0, 0,                                 // program_uid
        0, 0, 0, 0,                                              // extra fields
    };
    printf("%s...    ", __func__);

    conn->prims.rtt_sample_us = 0xcafefeed;
    fflush(stdout);
    ok = install_helper(datapath, install_msg, 172);
    if (ok < 0) {
        printf("install\n");
        return -1;
    }
    
    ok = change_prog_helper(datapath, change_prog_msg, 16);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(first_report_msg, 24, conn);
    if (ok < 0) {
        printf("failed on first report\n");
        return -1;
    }

    ok = ccp_invoke(conn);
    if (ok < 0) {
        printf("ccp_invoke 1 error: %d", ok);
        return -1;
    }
    
    ok = ccp_invoke(conn);
    if (ok < 0) {
        printf("ccp_invoke 2 error: %d", ok);
        return -1;
    }

    conn->prims.rtt_sample_us = 0xcafefeeb;
    ok = getreport_helper(report_msg, 24, conn);
    if (ok < 0) {
        printf("failed on second report\n");
        return -1;
    }

    printf("ok\n");
    return 0;
}

int test_read_implicit(struct ccp_datapath *datapath, struct ccp_connection *conn) {
    int ok;
    char install_msg[132] = {
        2, 0,                                           // INSTALL
        0x78, 0,                                        // length = 120
        1, 0, 0, 0,                                     // sock_id = 1
        IMPLI_PROG_UID, 0, 0, 0,                        // program_uid
        1, 0, 0, 0,                                     // num_events = 2
        6, 0, 0, 0,                                     // num_instrs = 6
        1, 0, 0, 0, 1, 0, 0, 0, 2, 0, 0, 0, 4, 0, 0, 0, // event { flag-idx=2, num-flag=1, body-idx=3, num-body=2 }
        2, 5, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 0, 0, 0, 0, // (def (Report.foo 0))
        1, 2, 0, 0, 0, 0, 2, 0, 0, 0, 0, 1, 1, 0, 0, 0, // (when true
        1, 2, 4, 0, 0, 0, 2, 4, 0, 0, 0, 1, 5, 0, 0, 0, //     (bind Cwnd 5)
        4, 6, 0, 0, 0, 0, 2, 4, 0, 0, 0, 1, 5, 0, 0, 0, //     ------------------(== Cwnd 5)-
        1, 5, 0, 0, 0, 0, 5, 0, 0, 0, 0, 6, 0, 0, 0, 0, //     (bind Control.foo ^^^^^^^^^^^) 
        1, 2, 2, 0, 0, 0, 2, 2, 0, 0, 0, 1, 1, 0, 0, 0  //     (bind __shouldReport true))
    };
    char report_msg[24] = {
        0x01, 0,
        0x18, 0,
        0x01, 0x00, 0x00, 0x00,
        IMPLI_PROG_UID, 0x00, 0x00, 0x00,               // program_uid
        0x01, 0x00, 0x00, 0x00,
        0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    };
    
    char change_prog_msg[16] = {
        4, 0,                                           // type
        12, 0,                                          // length 
        1, 0, 0, 0,                                     // sock id
        IMPLI_PROG_UID, 0, 0, 0,                        // program_uid
        0, 0, 0, 0,                                     // extra fields
    };
    printf("%s...  ", __func__);
    fflush(stdout);

    ok = install_helper(datapath, install_msg, 120);
    if (ok < 0) {
        printf("install failed\n");
        return -1;
    }
    
    ok = change_prog_helper(datapath, change_prog_msg, 16);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(report_msg, 24, conn);
    if (ok < 0) {
        printf("failed on report\n");
        return -1;
    }

    printf("ok\n");
    return 0;
}

int test_update_fields(struct ccp_datapath *datapath, struct ccp_connection *conn) {
    int ok;
    char install_msg[116] = {
        2, 0,                                            // INSTALL                                                            
        116, 0,                                          // length = 0x74 = 116
        1, 0, 0, 0,                                      // sock_id = 1                                                 
        UPDAT_PROG_UID, 0, 0, 0,                         // program_uid
        1, 0, 0, 0,                                      // num_events = 1                                              
        5, 0, 0, 0,                                      // num_instrs = 5                                              
        2, 0, 0, 0, 1, 0, 0, 0, 3, 0, 0, 0, 2, 0, 0, 0,  // event { flag-idx=2, num-flag=1, body-idx=3, num-body=2 }
        2, 5, 0, 0, 0, 0, 5, 0, 0, 0, 0, 1, 0, 0, 0, 0,  // -------------(volatile Report.foo 0)-
        2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 5, 0, 0, 0,  // (def (bar 5) ^^^^^^^^^^^^^^^^^^^^^^^)
        1, 2, 0, 0, 0, 0, 2, 0, 0, 0, 0, 1, 1, 0, 0, 0,  // (when true
        1, 5, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 0, 0,  //     (bind Report.foo bar)
        1, 2, 2, 0, 0, 0, 2, 2, 0, 0, 0, 1, 1, 0, 0, 0   //     (bind __shouldReport true)
    };
    char change_prog_msg[16] = {
        4, 0,                                            // CHANGE_PROG
        12, 0,                                           // length 
        1, 0, 0, 0,                                      // sock id
        UPDAT_PROG_UID, 0, 0, 0,                         // program_uid
        0, 0, 0, 0,                                      // extra fields
    };
    char update_msg[25] = {
        3, 0,                                            // UPDATE_FIELD
        25, 0,                                           // length = 25
        1, 0, 0, 0,                                      // sock_id = 1
        1, 0, 0, 0,                                      // num_fields = 1
        0, 0, 0, 0, 0, 0x2a, 0, 0, 0, 0, 0, 0, 0,        // Reg::Control(0) <- 42
    };
    char init_report_msg[24] = {
        0x01, 0x00,                                      // type 
        0x18, 0x00,                                      // len
        0x01, 0x00, 0x00, 0x00,                          // sockid
        UPDAT_PROG_UID, 0x00, 0x00, 0x00,                // program_uid
        0x01, 0x00, 0x00, 0x00,                          // num_fields
        0x05, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  // fields
    };
    char postupdate_report_msg[24] = {
        0x01, 0x00,                                      // type 
        0x18, 0x00,                                      // len
        0x01, 0x00, 0x00, 0x00,                          // sockid
        UPDAT_PROG_UID, 0x00, 0x00, 0x00,                // program_uid
        0x01, 0x00, 0x00, 0x00,                          // num_fields
        0x2a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,  // fields
    };
    printf("%s...  ", __func__);
    fflush(stdout);

    ok = install_helper(datapath, install_msg, 116);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = change_prog_helper(datapath, change_prog_msg, 16);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(init_report_msg, 24, conn);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = update_fields_helper(datapath, update_msg, 25);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    ok = getreport_helper(postupdate_report_msg, 24, conn);
    if (ok < 0) {
        printf("\n");
        return -1;
    }

    printf("ok\n");
    return 0;
}

#define MAX_CONNECTIONS 1024
#define MAX_PROGRAMS 10
struct ccp_datapath *create_datapath() {
    printf("initializing libccp... ");

    struct ccp_datapath *datapath;
    datapath = (struct ccp_datapath*)calloc(1, sizeof(struct ccp_datapath));
    if (!datapath) {
        printf("error: failed to allocate memory for datapath\n");
        return NULL;
    }
    datapath->max_connections = MAX_CONNECTIONS;
    datapath->ccp_active_connections = (struct ccp_connection*)calloc(MAX_CONNECTIONS, sizeof(struct ccp_connection));
    if (!datapath->ccp_active_connections) {
        printf("error: failed to allocate memory for ccp_active_connections\n");
        return NULL;
    }
    datapath->max_programs = MAX_PROGRAMS;
    datapath->set_cwnd = test_ccp_set_cwnd;
    datapath->set_rate_abs = test_ccp_set_rate;
    datapath->send_msg = test_ccp_send_msg;
    datapath->now = test_ccp_time_now;
    datapath->since_usecs = test_ccp_since_usecs;
    datapath->after_usecs = test_ccp_after_usecs;
    datapath->fto_us = 100000;

    return datapath;
}

int main(int UNUSED(argc), char **UNUSED(argv)) {
    int ok = 0;
    now_us = 0;

    struct ccp_datapath *datapath = create_datapath();
    if(!datapath) {
        goto ret;
    }

    expecting_send = 12;
    char ready_msg[12] = {
        0x05,0x00,
        0x0c,0x00,
        0x00,0x00,0x00,0x00,
        0xaa,0x00,0x00,0x00,
    };
    memcpy(&expected_sent_msg, ready_msg, 12);

    ok = ccp_init(datapath, 0xaa);
    if (ok < 0 ) {
        printf("ccp_init error: %d\n", ok);
        goto ret;
    }
    printf("ccp_datapath structure initialized correctly\n");

    struct ccp_connection *conn;
    struct test_conn my_conn = {
        .curr_cwnd = 0,
        .curr_rate = 0,
    };

    ok = test_init(datapath, &conn, &my_conn);
    if (ok < 0) {
        goto ret;
    }

    ok = test_basic(datapath, conn);
    if (ok < 0) {
        goto ret;
    }

    ok = test_primitives(datapath, conn);
    if (ok < 0) {
        goto ret;
    }
    
    ok = test_multievent(datapath, conn);
    if (ok < 0) {
        goto ret;
    }
    
    ok = test_fallthrough(datapath, conn);
    if (ok < 0) {
        goto ret;
    }
    
    ok = test_read_implicit(datapath, conn);
    if (ok < 0) {
        goto ret;
    }
    
    ok = test_update_fields(datapath, conn);
    if (ok < 0) {
        goto ret;
    }

    printf("test_close...          ");
    fflush(stdout);
    char close_msg[16] = {
        0x01, 0,
        16, 0,
        0x01, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00,
    };
    memcpy(&expected_sent_msg, &close_msg, 16);
    expecting_send = 16;
    ccp_connection_free(datapath, conn->index);
    printf("ok\n");
  ret:
    return 0;
}