/**
 * @file PacketInfo.hpp
 * @author Tobias
 * @brief class to provide general packet information
 * @date 2021-06-08
 *
 */

#pragma once

#include <rte_ether.h>
#include <rte_mbuf.h>

enum PacketType {
    NONE,
    IPv4ICMP,
    IPv4TCP,
    IPv4UDP,
    IPv6ICMP,
    IPv6TCP,
    IPv6UDP,
    IPv4,
    IPv6,
    ARP
};

class PacketInfo {
  public:
    inline PacketInfo() : _type(NONE), _eth_hdr(nullptr), _mbuf(nullptr) {}

    inline PacketInfo(rte_mbuf* const mbuf, rte_ether_hdr* const eth_hdr)
        : _type(NONE), _eth_hdr(eth_hdr), _mbuf(mbuf) {}

    inline ~PacketInfo() {
        //_mbuf = nullptr;
    }

    /**
     * @brief Get pointer to mbuf
     *
     * @return rte_mbuf*
     */
    inline rte_mbuf* const get_mbuf() { return _mbuf; }

    /**
     * @brief Get PacketInfos specialised type
     *
     * @return PacketType
     */
    inline PacketType const get_type() { return _type; }

    /**
     * @brief Get the source MAC-address
     *
     * @return rte_ether_addr
     */
    inline rte_ether_addr get_src_mac() { return _eth_hdr->s_addr; }

    /**
     * @brief Get the destination MAC-address
     *
     * @return rte_ether_addr
     */
    inline rte_ether_addr get_dst_mac() { return _eth_hdr->d_addr; }

  protected:
    inline PacketInfo(PacketType const type, rte_mbuf* const mbuf,
                      rte_ether_hdr* const eth_hdr)
        : _type(type), _eth_hdr(eth_hdr), _mbuf(mbuf) {}

    inline PacketInfo(PacketType const type, rte_mbuf* const mbuf)
        : _type(type), _eth_hdr(nullptr), _mbuf(mbuf) {}

    PacketType const _type;
    rte_ether_hdr* const _eth_hdr;

    inline rte_ether_hdr* const get_eth_hdr() { return _eth_hdr; }

    /**
     * @brief fill out standart ethernet header
     *
     * @param dst_mac MAC-address of destination
     * @param src_mac MAC-address of claimed source
     */
    inline void fill_eth_hdr(const rte_ether_addr& dst_mac,
                             const rte_ether_addr& src_mac) {
        _eth_hdr->d_addr = dst_mac;
        _eth_hdr->s_addr = src_mac;
    }

  private:
    rte_mbuf* const _mbuf;
};