#include <stdlib.h>
#include <assert.h>
#include <string.h>

#include "packet_priv.h"
#include "marsio.h"

/******************************************************************************
 * set and get from struct packet
 ******************************************************************************/

void packet_set_origin(struct packet *pkt, enum packet_origin origin)
{
    pkt->origin = origin;
}

enum packet_origin packet_get_origin(const struct packet *pkt)
{
    return pkt->origin;
}

int8_t packet_get_layers(const struct packet *pkt)
{
    return pkt->layers_used;
}

const struct packet_layer *packet_get_layer(const struct packet *pkt, int8_t idx)
{
    if (idx < 0 || idx >= pkt->layers_used)
    {
        return NULL;
    }
    return &pkt->layers[idx];
}

const char *packet_get_data(const struct packet *pkt)
{
    return pkt->data_ptr;
}

uint16_t packet_get_len(const struct packet *pkt)
{
    return pkt->data_len;
}

const char *packet_get_payload(const struct packet *pkt)
{
    if (pkt->layers_used == 0)
    {
        return NULL;
    }
    return pkt->layers[pkt->layers_used - 1].pld_ptr;
}

uint16_t packet_get_payload_len(const struct packet *pkt)
{
    if (pkt->layers_used == 0)
    {
        return 0;
    }
    return pkt->layers[pkt->layers_used - 1].pld_len;
}

int packet_need_drop(const struct packet *pkt)
{
    return pkt->need_drop;
}

void packet_set_drop(struct packet *pkt)
{
    pkt->need_drop = 1;
}

void packet_set_io_ctx(struct packet *pkt, void *ctx)
{
    pkt->io_ctx = ctx;
}

void *packet_get_io_ctx(const struct packet *pkt)
{
    return pkt->io_ctx;
}

int packet_is_fragment(const struct packet *pkt)
{
    return (pkt->frag_layer) ? 1 : 0;
}

struct packet *packet_new(uint16_t pkt_len)
{
    struct packet *pkt = (struct packet *)calloc(1, sizeof(struct packet) + pkt_len);
    if (pkt == NULL)
    {
        return NULL;
    }
    pkt->data_len = pkt_len;
    pkt->data_ptr = (const char *)pkt + sizeof(struct packet);
    pkt->origin = PACKET_ORIGIN_USER;

    return pkt;
}

void packet_free(struct packet *pkt)
{
    if (pkt && pkt->origin == PACKET_ORIGIN_USER)
    {
        free((void *)pkt);
    }
}

struct packet *packet_dup(const struct packet *pkt)
{
    if (pkt == NULL)
    {
        return NULL;
    }

    struct packet *dup_pkt = packet_new(pkt->data_len);
    if (dup_pkt == NULL)
    {
        return NULL;
    }
    memcpy(dup_pkt, pkt, sizeof(struct packet));
    memcpy((char *)dup_pkt->data_ptr, pkt->data_ptr, pkt->data_len);
    dup_pkt->origin = PACKET_ORIGIN_USER;
    dup_pkt->io_ctx = dup_pkt;

    // update layers
    for (int8_t i = 0; i < pkt->layers_used; i++)
    {
        dup_pkt->layers[i].hdr_ptr = dup_pkt->data_ptr + pkt->layers[i].hdr_offset;
        dup_pkt->layers[i].pld_ptr = dup_pkt->data_ptr + pkt->layers[i].hdr_offset + pkt->layers[i].hdr_len;
    }

    // update frag_layer
    if (pkt->frag_layer)
    {
        dup_pkt->frag_layer = &dup_pkt->layers[pkt->frag_layer - pkt->layers];
    }

    return dup_pkt;
}

/******************************************************************************
 * set and get from mbuff
 ******************************************************************************/

void packet_set_ctrl(struct packet *pkt)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        marsio_buff_set_ctrlbuf(mbuff);
    }
}

int packet_is_ctrl(const struct packet *pkt)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        return marsio_buff_is_ctrlbuf(mbuff);
    }
    else
    {
        return 0;
    }
}

// 1: E2I, 0: I2E
void packet_set_direction(struct packet *pkt, int dir)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_set_metadata(mbuff, MR_BUFF_DIR, &dir, sizeof(dir)) != 0)
        {
            PACKET_LOG_ERROR("failed to set direction");
        }
    }
}

// 1: E2I, 0: I2E
int packet_get_direction(const struct packet *pkt)
{
    int direction = 0;
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_get_metadata(mbuff, MR_BUFF_DIR, &direction, sizeof(direction)) <= 0)
        {
            PACKET_LOG_ERROR("failed to get direction");
        }
    }

    return direction;
}

void packet_set_session_id(struct packet *pkt, uint64_t sess_id)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_set_metadata(mbuff, MR_BUFF_SESSION_ID, &sess_id, sizeof(sess_id)) != 0)
        {
            PACKET_LOG_ERROR("failed to set session id");
        }
    }
}

uint64_t packet_get_session_id(const struct packet *pkt)
{
    uint64_t sess_id = 0;
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_get_metadata(mbuff, MR_BUFF_SESSION_ID, &sess_id, sizeof(sess_id)) <= 0)
        {
            PACKET_LOG_ERROR("failed to get session id");
        }
    }

    return sess_id;
}

void packet_set_domain(struct packet *pkt, uint64_t domain)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
// TODO
#if 0
        if (marsio_buff_set_metadata(mbuff, MR_BUFF_DOMAIN, &domain, sizeof(domain)) != 0)
        {
            PACKET_LOG_ERROR("failed to set domain");
        }
#endif
    }
}

uint64_t packet_get_domain(const struct packet *pkt)
{
    uint64_t domain = 0;
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
// TODO
#if 0
        if (marsio_buff_get_metadata(mbuff, MR_BUFF_DOMAIN, &domain, sizeof(domain)) <= 0)
        {
            PACKET_LOG_ERROR("failed to get domain");
        }
#endif
    }

    return domain;
}

void packet_set_route_ctx(struct packet *pkt, const char *route, int len)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_set_metadata(mbuff, MR_BUFF_ROUTE_CTX, (void *)route, len) != 0)
        {
            PACKET_LOG_ERROR("failed to set route ctx");
        }
    }
}

// return len of route ctx
int packet_get_route_ctx(const struct packet *pkt, char *buff, int size)
{
    int len = 0;
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        len = marsio_buff_get_metadata(mbuff, MR_BUFF_ROUTE_CTX, buff, size);
        if (len <= 0)
        {
            PACKET_LOG_ERROR("failed to get route ctx");
        }
    }

    return len;
}

void packet_set_sid_list(struct packet *pkt, uint16_t *sid, int num)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_set_sid_list(mbuff, sid, num) != 0)
        {
            PACKET_LOG_ERROR("failed to set sid list");
        }
    }
}

// return number of sid
int packet_get_sid_list(const struct packet *pkt, uint16_t *sid, int num)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        return marsio_buff_get_sid_list(mbuff, sid, num);
    }

    return 0;
}

void packet_prepend_sid_list(struct packet *pkt, uint16_t *sid, int num)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_prepend_sid_list(mbuff, sid, num) != 0)
        {
            PACKET_LOG_ERROR("failed to prepend sid list");
        }
    }
}

void packet_append_sid_list(struct packet *pkt, uint16_t *sid, int num)
{
    if (packet_get_origin(pkt) == PACKET_ORIGIN_MARSIO)
    {
        marsio_buff_t *mbuff = (marsio_buff_t *)packet_get_io_ctx(pkt);
        assert(mbuff != NULL);
        if (marsio_buff_append_sid_list(mbuff, sid, num) != 0)
        {
            PACKET_LOG_ERROR("failed to append sid list");
        }
    }
}