#pragma once

#include <marsio.h>
#include <tfe_stream.h>
#include <netinet/ether.h>
#include <MESA/stream.h>
#include <MESA/maat.h>
#include <pcap/pcap.h>

enum traffic_mirror_ethdev_type
{
    TRAFFIC_MIRROR_ETHDEV_AF_PACKET,
    TRAFFIC_MIRROR_ETHDEV_MARSIO
};

enum traffic_mirror_target_addr_type
{
    TRAFFIC_MIRROR_TARGET_BY_VLAN_ID,
    TRAFFIC_MIRROR_TARGET_BY_ETHER_ADDR,
};

enum
{
    TRAFFIC_MIRR_STAT_SUCC_BYTES,
    TRAFFIC_MIRR_STAT_SUCC_PKTS,

    TRAFFIC_MIRR_STAT_FAIL_BYTES,
    TRAFFIC_MIRR_STAT_FAIL_PKTS,

    TRAFFIC_MIRR_STAT_MAX,
};

struct traffic_mirror_stat
{
    struct fieldstat_easy *fs_handle;

    uint64_t stat_idx[TRAFFIC_MIRR_STAT_MAX];
    uint64_t stat_val[TRAFFIC_MIRR_STAT_MAX];

    struct event_base *evbase;
    struct event *ev;
};

struct traffic_mirror_instance
{
    void * logger;
    unsigned int enable;
    unsigned int nr_threads;
    unsigned int default_vlan_id;

    struct maat* maat_feather;

    /* DEFAULT MAC ADDRESS, IN VLAN MODE */
    char default_ether_addr_src;
    char default_ether_addr_dst;

    struct traffic_mirror_ethdev * ethdev;
    struct traffic_mirror_stat *stat;
};

struct policy_table_ex_data
{
    uuid_t rule_uuid;
    uuid_t profile_uuid;
    unsigned int atomic_refcnt;
    unsigned int enable;
    unsigned int is_profile_set;
};

struct profile_table_ex_data
{
    uuid_t profile_uuid;
    unsigned int atomic_refcnt;
    unsigned int nr_targets;

    /* Target VLANS */
    unsigned int * vlans;
    struct ether_addr * ether_addrs;
    int rewrite_mac;
    int rewrite_vlan;
};

struct traffic_mirror_ethdev_pcap
{
    pcap_t * pcap_device_handle;
    char sendbuf[TFE_THREAD_MAX][ETHER_MAX_LEN];
};

struct traffic_mirror_ethdev_mr4
{
    /* DEVICE INSTANCES */
    struct mr_instance * instance;
    struct mr_vdev * device;
    struct mr_sendpath * sendpath;

    /* THREAD SENDBUF */
    unsigned int nr_tx_threads;
    marsio_buff_t * sendbuf[TFE_THREAD_MAX];
};

struct traffic_mirror_ethdev
{
    /* PUBLIC */
    char str_device[TFE_SYMBOL_MAX];

    unsigned int mtu;
    unsigned int en_offload_vlan;
    unsigned int en_offload_ip_cksum;
    unsigned int en_offload_tcp_cksum;
    char local_ether_addr[6];

    enum traffic_mirror_ethdev_type type;
    union
    {
        struct traffic_mirror_ethdev_pcap * detail_pcap;
        struct traffic_mirror_ethdev_mr4 * detail_mr4;
    };

    /* FUNCTIONS */
    char * (*fn_send_prepare)(struct traffic_mirror_ethdev *, unsigned int tid);
    int (*fn_send_finish)(struct traffic_mirror_ethdev * ethdev, unsigned int, unsigned int);
    void (*fn_send_abort)(struct traffic_mirror_ethdev * ethdev, unsigned int tid);
    void (*fn_destroy)(struct traffic_mirror_ethdev * ethdev);
};

struct traffic_mirror_rebuild_target
{
    struct ether_addr ether_addr;
    unsigned int vlan_tci;
    int rewrite_as_target_mac;
    int rewrite_as_target_vlan;
};

extern struct traffic_mirror_instance *g_traffic_mirror_instance;
struct traffic_mirror_ethdev * traffic_mirror_ethdev_pcap_create(const char * str_ethdev, void * logger);
struct traffic_mirror_ethdev * traffic_mirror_ethdev_mr4_create(const char * str_ethdev, 
    const char * str_app_name, unsigned int nr_threads, void * logger);

void traffic_mirror_ethdev_destroy(struct traffic_mirror_ethdev * ethdev);
int traffic_mirror_ethdev_finish(struct traffic_mirror_ethdev * ethdev, unsigned int tid, unsigned int pktlen);

struct traffic_mirror_rebuild * traffic_mirror_rebuild_create(struct tfe_stream_addr * addr,
	struct ether_addr * s_ether_addr, struct ether_addr * d_ether_addr,
	struct traffic_mirror_rebuild_target * target, struct traffic_mirror_ethdev * ethdev);

void traffic_mirror_rebuild_destroy(struct traffic_mirror_rebuild * instance);
void traffic_mirror_rebuild_handshake(struct traffic_mirror_rebuild * instance, unsigned int tid);
void traffic_mirror_rebuild_data(struct traffic_mirror_rebuild * instance, unsigned int tid, const char * data,
    unsigned int datalen, enum tfe_conn_dir dir);
void traffic_mirror_rebuild_farewell(struct traffic_mirror_rebuild * instance, unsigned int tid);