/*
 *	\brief MARSIO Userspace ZeroCopy Driver Version 4
 *
 *	This is the user api header file of MARSIOv4 ZeroCopy Driver
 *
 *	\author	Qiuwen Lu<luqiuwen@iie.ac.cn>
 *			Institute of Information Engineering, Chinese Academy of Sciences
 *
 *	\date	2016-12-01
 */

#pragma once
#include <limits.h>

typedef enum
{
    /* 数据面线程数，没有缺省值 */
    MARSIO_OPT_THREAD_NUM,
    /* 数据面线程绑定掩码，没有缺省值，设置该掩码后，数据面线程数选项将被忽略。*/
    MARSIO_OPT_THREAD_MASK,
    /* 线程绑定模式，类型：uint32_t */
    MARSIO_OPT_THREAD_AFFINITY_MODE,

    /* 处理SIG信号，缺省值：类型：uint32_t，关闭（0）
     * 启用该选项后，将自动处理SIGINT、SIGTERM信号
     */
    MARSIO_OPT_EXIT_WHEN_ERR,

    /* 拓展数据面线程绑定掩码，适应大于64个核处理器的硬件平台
     * 设置该掩码后，数据面线程数选项将被忽略
     */
    MARSIO_OPT_THREAD_MASK_IN_CPUSET,

} marsio_opt_type_t;

typedef enum
{
    /* 发送后不释放数据包，由应用自行释放数据包 */
    MARSIO_SEND_OPT_NO_FREE = 1 << 0,
    /* 发送时计算发包哈希值，用于分流 */
    MARSIO_SEND_OPT_REHASH = 1 << 1,
    /* 快速报文路径 */
    MARSIO_SEND_OPT_FAST = 1 << 2,
    /* 报文追踪标记 */
    MARSIO_SEND_OPT_TRACE = 1 << 3,
    /* 控制报文标记 */
    MARSIO_SEND_OPT_CTRL = 1 << 4

} marsio_opt_send_t;

enum mr_sendpath_type
{
    /* 去往特定虚设备的发包路径 */
    MR_SENDPATH_VDEV,
    /* 路由查表确定发包路径 */
    MR_SENDPATH_ROUTE_NORMAL,
    /* 特定设备路由查表确定发包路径 */
    MR_SENDPATH_ROUTE_SPEC_DEV,
    /* MAX标记 */
    MR_SENDPATH_MAX
};

enum mr_sendpath_option
{
    /* 构建四层报文头 */
    MR_SENDPATH_OPT_BUILD_L4 = 0,
    /* 构建三层报文头 */
    MR_SENDPATH_OPT_BUILD_L3 = 1,
    /* 构建二层报文头 */
    MR_SENDPATH_OPT_BUILD_L2 = 2,
    /* 构建前Hook点回调 */
    MR_SENDPATH_OPT_HOOK_PREBUILD = 50,
    /* 构建后Hook点回调 */
    MR_SENDPATH_OPT_HOOK_POSTBUILD = 51,
};

enum mr_clone_options
{
    /* 拷贝区域 */
    MR_BUFF_CLONE_DATA = 1 << 0,
    MR_BUFF_CLONE_BUFF = 1 << 1,
    MR_BUFF_CLONE_CTRLZONE = 1 << 2,
};

enum mr_thread_affinity_mode
{
    /* 禁用线程亲和性设置 */
    MR_THREAD_AFFINITY_DISABLE = 0,
    /* 自动线程亲和性设置 */
    MR_THREAD_AFFINITY_AUTO = 1,
    /* 自定义线程亲和性设置 */
    MR_THREAD_AFFINITY_USER = 255
};

enum mr_timestamp_type
{
    /* 从网卡收取时或报文缓冲区申请时的时间戳 */
    MR_TIMESTAMP_RX_OR_ALLOC = 0,
};

enum mr_buff_metadata_type
{
    /* Rehash Index, uint32_t (4bytes) */
    MR_BUFF_REHASH_INDEX = 0,
    /* VLAN TCI, uint16_t (2bytes) */
    MR_BUFF_METADATA_VLAN_TCI = 1,
    /* dispatch ctx, > 28bytes */
    MR_BUFF_ROUTE_CTX = 2,
    /* session id, uint64_t (8bytes) */
    MR_BUFF_SESSION_ID = 3,
    /* dir, internal->external or external->internal, unsigned int (4bytes) */
    MR_BUFF_DIR = 4,
    /* payload offset, uint16_t (2bytes) */
    MR_BUFF_PAYLOAD_OFFSET = 5,
    /* link id, uint16_t (2bytes) */
    MR_BUFF_LINK_ID = 6,
    /* user 0, uint16_t (2bytes) */
    MR_BUFF_USER_0 = 254
};

#ifdef __cplusplus
extern "C"
{
#endif

#include <netinet/in.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>

#ifndef MARSIO_SOCKET_ID_ANY
#define MARSIO_SOCKET_ID_ANY -1
#endif

#ifndef MARSIO_LCORE_ID_ANY
#define MARSIO_LCORE_ID_ANY -1
#endif

typedef uint32_t device_id_t;
typedef uint32_t port_id_t;
typedef uint32_t queue_id_t;
typedef uint32_t thread_id_t;
typedef void marsio_buff_t;

struct mr_sendpath;
struct mr_instance;
struct mr_vdev;

struct mr_instance * marsio_create();
struct mr_instance * marsio_current();

typedef int (*fn_sendpath_hook_t)(struct mr_sendpath * sendpath, marsio_buff_t * mbuf[], unsigned int nr_mbuf,
                                  void * arg);

int marsio_option_get(struct mr_instance * instance, int opt_type, void * out_opt, size_t out_opt_buffer);
int marsio_option_set(struct mr_instance * instance, marsio_opt_type_t opt_type, void * opt, size_t sz_opt);
int marsio_init(struct mr_instance * instance, const char * appsym);
int marsio_thread_init(struct mr_instance * instance);
int marsio_destory(struct mr_instance * instance);

int marsio_poll_register_eventfd(struct mr_instance * instance, int eventfd, unsigned int tid);
int marsio_poll_wait(struct mr_instance * instance, struct mr_vdev * vdevs[], unsigned int nr_vdevs, unsigned int tid,
                     int timeout);

struct mr_vdev * marsio_open_device(struct mr_instance * instance, const char * devsym, unsigned int nr_rxstream,
                                    unsigned int nr_txstream);

void marsio_close_device(struct mr_vdev * vdev);

void marsio_get_device_ether_addr(struct mr_vdev * vdev, void * str_ether_addr, uint8_t size);

struct mr_vdev * marsio_device_lookup(struct mr_instance * instance, const char * devsym);

int marsio_recv_burst(struct mr_vdev * vdev, queue_id_t qid, marsio_buff_t * mbufs[], int nr_mbufs);
int marsio_recv_all_burst(struct mr_instance * instance, queue_id_t qid, marsio_buff_t * mbufs[], int nr_mbufs);
int marsio_send_burst(struct mr_sendpath * sendpath, queue_id_t qid, marsio_buff_t * mbufs[], int nr_mbufs);
int marsio_send_burst_with_options(struct mr_sendpath * sendpath, queue_id_t sid, marsio_buff_t * mbufs[], int nr_mbufs,
                                   uint16_t options);
void marsio_send_burst_flush(struct mr_sendpath * sendpath, queue_id_t sid);

int marsio_udp_header_construct(marsio_buff_t * buff, uint16_t s_port, uint16_t d_port);
int marsio_ipv4_header_construct(marsio_buff_t * buff, uint32_t s_ip, uint32_t d_ip, uint8_t proto);

struct mr_sendpath * marsio_sendpath_create_by_droute(struct mr_vdev * dest_device, struct in_addr addr);
struct mr_sendpath * marsio_sendpath_create_by_route(struct mr_instance * instance, struct in_addr addr);
struct mr_sendpath * marsio_sendpath_create_by_vdev(struct mr_vdev * dest_device);

struct mr_sendpath * marsio_sendpath_create(struct mr_instance * instance, int type, ...);
int marsio_sendpath_option_set(struct mr_instance * instance, struct mr_sendpath * sendpath, int opt, ...);
int marsio_sendpath_option_get(struct mr_instance * instance, struct mr_sendpath * sendpath, int opt, ...);
void marsio_sendpath_destory(struct mr_sendpath * sendpath);

// simply get control zone data ptr.
void * marsio_buff_ctrlzone(marsio_buff_t * m, uint8_t id);

// a safe way to read control zone data.
void * marsio_buff_ctrlzone_data(marsio_buff_t * m, uint8_t id, uint8_t * size);

// a safe way to set control zone data.
void marsio_buff_ctrlzone_set(marsio_buff_t * m, uint8_t id, void * ptr_data, uint8_t size);

// a safe way to reset control zone data.
void marsio_buff_ctrlzone_reset(marsio_buff_t * m);

void marsio_buff_reset(marsio_buff_t * m);

marsio_buff_t * marsio_buff_getnext_seg(marsio_buff_t * m);
marsio_buff_t * marsio_buff_getnext_pkt(marsio_buff_t * m);

void marsio_buff_append_pkt(marsio_buff_t * head, marsio_buff_t * next);
void marsio_buff_append_seg(marsio_buff_t * head, marsio_buff_t * next);
void marsio_buff_chain_pkt(marsio_buff_t * pkt, marsio_buff_t * next);

uint16_t marsio_buff_headroom(const marsio_buff_t * m);
uint16_t marsio_buff_tailroom(const marsio_buff_t * m);

char * marsio_buff_mtod(marsio_buff_t * m);

uint32_t marsio_buff_buflen(marsio_buff_t * m);

uint32_t marsio_buff_datalen(marsio_buff_t * m);

char * marsio_buff_prepend(marsio_buff_t * m, uint16_t len);

char * marsio_buff_append(marsio_buff_t * m, uint16_t len);

char * marsio_buff_adj(marsio_buff_t * m, uint16_t len);

int marsio_buff_trim(marsio_buff_t * m, uint16_t len);

char * marsio_buff_offset_set(marsio_buff_t * m, off_t offset, int is_relative);

off_t marsio_buff_offset_get(marsio_buff_t * m);

/* 支持写时复制的报文修改裁剪函数 */
marsio_buff_t * marsio_buff_prepend_cw(struct mr_instance * instance, marsio_buff_t * m, uint16_t len, void ** ptr_out);

marsio_buff_t * marsio_buff_append_cw(struct mr_instance * instance, marsio_buff_t * m, uint16_t len, void ** ptr_out);

marsio_buff_t * marsio_buff_adj_cw(struct mr_instance * instance, marsio_buff_t * m, uint16_t len, void ** ptr_out);

marsio_buff_t * marsio_buff_trim_cw(struct mr_instance * instance, marsio_buff_t * m, uint16_t len, void ** ptr_out);

uint16_t marsio_buff_headroom(const marsio_buff_t * m);

uint16_t marsio_buff_tailroom(const marsio_buff_t * m);

uint32_t marsio_get_pkt_type(marsio_buff_t * m);

void marsio_pktmbuf_dump(FILE * f, const marsio_buff_t * m, unsigned dump_len);

marsio_buff_t * marsio_buff_clone_deep(struct mr_instance * instance, marsio_buff_t * md, int socket_id, int thread_id);

marsio_buff_t * marsio_buff_clone_with_options(struct mr_instance * instance, marsio_buff_t * md, int socket_id,
                                               int thread_id, uint16_t options);

int marsio_buff_malloc_device(struct mr_vdev * vdev, marsio_buff_t * marsio_buff[], unsigned int nr_mbufs,
                              int socket_id, int thread_id);

int marsio_buff_malloc_global(struct mr_instance * instance, marsio_buff_t * marsio_buff[], unsigned int nr_mbufs,
                              int socket_id, int thread_id);

void marsio_buff_free(struct mr_instance * instance, marsio_buff_t * marsio_buff[], unsigned int nr_mbufs,
                      int socket_id, int thread_id);

void marsio_buff_free_v2(struct mr_instance * instance, marsio_buff_t * buff[], unsigned int nr_buffs);

void marsio_buff_do_rehash(struct mr_instance * mr_instance, marsio_buff_t * m);

int marsio_buff_is_ctrlbuf(marsio_buff_t * m);
void marsio_buff_set_ctrlbuf(marsio_buff_t * m);

marsio_buff_t * marsio_buff_malloc_smartoffload(struct mr_vdev * vdev, const char * pkt, unsigned int pkt_len);

uint64_t marsio_buff_get_timestamp(marsio_buff_t * m);
void marsio_buff_set_timestamp(marsio_buff_t * m, uint64_t timestamp);
int marsio_buff_get_timestamp_ex(marsio_buff_t * m, enum mr_timestamp_type ts_type, struct timespec * ts);

void marsio_buff_set_rehash_index(marsio_buff_t * m, uint32_t hash);
uint32_t marsio_buff_get_rehash_index(marsio_buff_t * m);

int marsio_buff_get_metadata(marsio_buff_t * m, enum mr_buff_metadata_type type, void * data, unsigned int sz_data);
int marsio_buff_set_metadata(marsio_buff_t * m, enum mr_buff_metadata_type type, void * data, unsigned int sz_data);
int marsio_buff_unset_metadata(marsio_buff_t * m, enum mr_buff_metadata_type type);

/* sid */
typedef uint16_t sid_t;
#define MR_SID_LIST_MAXLEN 8

int marsio_buff_get_sid_list(marsio_buff_t * m, sid_t * out_slist, uint8_t sz_out_slist);
int marsio_buff_set_sid_list(marsio_buff_t * m, sid_t * slist, uint8_t sz_slist);
int marsio_buff_append_sid_list(marsio_buff_t * m, sid_t * slist, uint8_t sz_slist);
int marsio_buff_prepend_sid_list(marsio_buff_t * m, sid_t * slist, uint8_t sz_slist);
int marsio_buff_get_current_sid(marsio_buff_t * m, sid_t * sid);

/******************************** data path trace**************************************/
#define DP_TRACE_MEASUREMENT_TYPE_TRACE (1 << 0)
#define DP_TRACE_MEASUREMENT_TYPE_TELEMETRY (1 << 1)

int marsio_dp_trace_measurements_can_emit(struct mr_instance * instance, const marsio_buff_t * mbuf,
                                          uint8_t measurement_type);

int marsio_dp_trace_measurement_emit_str(struct mr_instance * instance, marsio_buff_t * mbuf, uint8_t measurement_type,
                                         const char * module, const char * str);

int marsio_dp_trace_measurement_emit_fmt(struct mr_instance * instance, marsio_buff_t * mbuf, uint8_t measurement_type,
                                         const char * module, const char * format, ...);

/////////////////////// only for data path trace telemetry ////////////
#ifndef MR_BPF_EXPRESSION_MAX
#define MR_BPF_EXPRESSION_MAX 256
#endif

#define DP_TRACE_TRAFFIC_LINK_ID_ARRAY_SIZE_MAX 8

#define DP_TRACE_JOB_NUM_MAX 16

#define DP_TRACE_RING_NUM 4

typedef uint16_t job_bitmap_t;
typedef void pcapng_file_t;

struct dp_trace_buffer_telemetry
{
    job_bitmap_t jobs_id;
    unsigned int snaplen;
    char * buffer;
    uint16_t buffer_len;
    uint16_t buffer_used;

    uint16_t traffic_link_id;
    char inner_src_addr_str[INET6_ADDRSTRLEN];
    char inner_dst_addr_str[INET6_ADDRSTRLEN];
    int32_t inner_src_port;
    int32_t inner_dst_port;

    uint8_t egress_action;
};

struct dp_trace_record_header
{
    uint8_t measurement_type;
    char appsym[16];
    char module[16];
    struct timespec ts;
    uint16_t recode_len;
};

struct dp_trace_job_desc
{
    bool enable;
    uint8_t measurement_type;
    int8_t rule_index;
    unsigned int pkt_cnt_max; // The final number of captured packets
    unsigned int sampling;
    unsigned int snaplen;
    uint8_t traffic_link_id_cnt;
    uint16_t traffic_link_ids[DP_TRACE_TRAFFIC_LINK_ID_ARRAY_SIZE_MAX];
    char bpf_expr[MR_BPF_EXPRESSION_MAX];
};

int marsio_dp_trace_job_id_uesd_get(struct mr_instance * instance, job_bitmap_t * jobs_id);

job_bitmap_t marsio_dp_trace_job_add(struct mr_instance * instance, const struct dp_trace_job_desc * desc);

int marsio_dp_trace_job_del(struct mr_instance * instance, job_bitmap_t job_id);

int marsio_dp_trace_mbuf_recv_burst(struct mr_instance * instance, queue_id_t qid, marsio_buff_t * mbufs[],
                                    int nr_mbufs);

void marsio_dp_trace_mbuf_free(struct mr_instance * instance, marsio_buff_t * mbufs[], int nr_mbufs);

int marsio_dp_trace_buffer_info_get(const marsio_buff_t * mbuf, struct dp_trace_buffer_telemetry * info);

int marsio_dp_trace_mbuf_refcnt_update(const marsio_buff_t * mbuf, int16_t value);

/////////////////////// only for data path trace telemetry ////////////

/***********************************************************************************/

#ifdef __cplusplus
}
#endif