/*
 * File:   nattest.cpp
 * Author: dev
 *
 * Created on 19.08.2015, 21:39:53
 */

#include "nattest.h"
#include "../src/map/NatRange.h"
#include <iostream>
#include <thread>
CPPUNIT_TEST_SUITE_REGISTRATION(nattest);

nattest::nattest() {
    Tins::NetworkInterface net1("vboxnet0");
    Tins::NetworkInterface net2("vboxnet1");
    otonat::NatRange range1(net1, "10.0.0.0", "255.255.240.0");
    otonat::NatRange range2(net2, "172.27.0.0", "255.255.0.0");
    otonat::NatMap::NatRangeList list;
    list.push_back(range1);
    list.push_back(range2);
    natMap = otonat::NatMap(list);
}

nattest::~nattest() {
}

void nattest::setUp() {
}

void nattest::tearDown() {
}

void nattest::testIpCalcEth0() {
    Tins::IPv4Address expetedIp = deviceIpEth0;
    Tins::IPv4Address resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth0, eth0Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);

    expetedIp = Tins::IPv4Address("10.168.23.42");
    resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth1, eth0Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);

    expetedIp = Tins::IPv4Address("10.27.123.4");
    resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth2, eth0Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);
}

void nattest::testIpCalcEth1() {
    Tins::IPv4Address expetedIp = Tins::IPv4Address("192.168.23.40");
    Tins::IPv4Address resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth0, eth1Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);

    expetedIp = deviceIpEth1;
    resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth1, eth1Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);

    expetedIp = Tins::IPv4Address("192.168.23.4");
    resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth2, eth1Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);
}

void nattest::testIpCalcEth2() {
    Tins::IPv4Address expetedIp = Tins::IPv4Address("172.16.3.40");
    Tins::IPv4Address resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth0, eth2Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);

    expetedIp = Tins::IPv4Address("172.24.23.42");
    resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth1, eth2Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);

    expetedIp = deviceIpEth2;
    resultIp = otonat::NatRange::mapIPv4Address(deviceIpEth2, eth2Info);
    CPPUNIT_ASSERT_EQUAL(expetedIp, resultIp);
}

void nattest::testTranslateArpIp() {
    natMap.reqIpMap.clear();
    natMap.transMap.clear();
    Tins::EthernetII arp1 = Tins::ARP::make_arp_request("172.17.0.20", "172.16.3.55", "00:00:00:00:00:02");
    Tins::EthernetII arp2 = Tins::ARP::make_arp_request("172.27.0.20", "172.16.3.55", "00:00:00:00:00:02");
    Tins::EthernetII arp3 = Tins::ARP::make_arp_reply("10.0.3.55", "10.0.0.20", "00:00:00:00:00:02", "00:00:00:00:00:01");
    Tins::EthernetII arp4 = Tins::ARP::make_arp_reply("10.128.3.55", "10.0.0.20", "00:00:00:00:00:03", "00:00:00:00:00:01");
    Tins::EthernetII ethW = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:02") / Tins::IP("172.17.0.20", "172.16.3.55") / Tins::TCP();
    Tins::EthernetII eth = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:02") / Tins::IP("172.27.0.20", "172.16.3.55") / Tins::TCP();
    Tins::EthernetII ethAck = Tins::EthernetII("00:00:00:00:00:02", "00:00:00:00:00:01") / Tins::IP("10.0.3.55", "10.0.0.20") / Tins::TCP();
    Tins::EthernetII eth2 = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:03") / Tins::IP("172.27.0.20", "172.17.3.55") / Tins::TCP();
    Tins::EthernetII eth2Ack = Tins::EthernetII("00:00:00:00:00:03", "00:00:00:00:00:01") / Tins::IP("10.0.3.55", "10.0.0.20") / Tins::TCP();
    Tins::EthernetII eth3 = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:04") / Tins::IP("172.27.0.20", "172.18.1.40") / Tins::TCP();
    Tins::EthernetII eth3Ack = Tins::EthernetII("00:00:00:00:00:04", "00:00:00:00:00:01") / Tins::IP("10.0.1.40", "10.0.0.20") / Tins::TCP();
    Tins::EthernetII eth4 = Tins::EthernetII("00:00:00:00:00:02", "00:00:00:00:00:05") / Tins::IP("10.0.3.55", "10.0.1.41") / Tins::TCP();
    Tins::EthernetII eth4Ack = Tins::EthernetII("00:00:00:00:00:05", "00:00:00:00:00:02") / Tins::IP("172.27.1.41", "172.17.3.55") / Tins::TCP();

    natMap.handlePdu(arp1.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(arp2.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultArp1 = natMap.outgoingPduQueue.front();
    checkArp(resultArp1->rfind_pdu<Tins::ARP>(), Tins::ARP::REQUEST, "00:00:00:00:00:00", "00:00:00:00:00:02", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();

    natMap.handlePdu(arp3.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultArp2 = natMap.outgoingPduQueue.front();
    checkArp(resultArp2->rfind_pdu<Tins::ARP>(), Tins::ARP::REPLY, "00:00:00:00:00:02", "00:00:00:00:00:01", "172.16.3.55", "172.27.0.20");
    natMap.outgoingPduQueue.pop();

    natMap.handlePdu(arp4.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(ethW.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result = natMap.outgoingPduQueue.front();
    checkEth(result->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:02", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(ethAck.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultAck = natMap.outgoingPduQueue.front();
    checkEth(resultAck->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:02", "00:00:00:00:00:01", "172.16.3.55", "172.27.0.20");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth2.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result2 = natMap.outgoingPduQueue.front();
    checkEth(result2->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:03", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth2Ack.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result2Ack = natMap.outgoingPduQueue.front();
    checkEth(result2Ack->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:03", "00:00:00:00:00:01", "172.17.3.55", "172.27.0.20");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth3.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result3 = natMap.outgoingPduQueue.front();
    checkEth(result3->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:04", "10.0.0.20", "10.0.1.40");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth3Ack.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result3Ack = natMap.outgoingPduQueue.front();
    checkEth(result3Ack->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:04", "00:00:00:00:00:01", "172.18.1.40", "172.27.0.20");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth4.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result4 = natMap.outgoingPduQueue.front();
    checkEth(result4->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:02", "00:00:00:00:00:05", "172.17.3.55", "172.27.1.41");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth4Ack.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result4Ack = natMap.outgoingPduQueue.front();
    checkEth(result4Ack->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:05", "00:00:00:00:00:02", "10.0.1.41", "10.0.3.55");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    CPPUNIT_ASSERT(natMap.transMap.size() == 9);
    CPPUNIT_ASSERT(natMap.reqIpMap.empty());
    /*for (auto& entry : natMap.transMap){
        std::cout << std::endl << "ip_dst: " << entry.first << std::endl;
        std::cout << "ip_src: " << entry.second << std::endl;
    }*/
}

void nattest::testTranslateArp() {
    natMap.reqIpMap.clear();
    natMap.transMap.clear();
    Tins::EthernetII arp1 = Tins::ARP::make_arp_request("172.17.0.20", "172.16.3.55", "00:00:00:00:00:02");
    Tins::EthernetII arp2 = Tins::ARP::make_arp_request("172.27.0.20", "172.16.3.55", "00:00:00:00:00:02");
    Tins::EthernetII arp3 = Tins::ARP::make_arp_reply("10.0.3.55", "10.0.0.20", "00:00:00:00:00:02", "00:00:00:00:00:01");
    Tins::EthernetII arp4 = Tins::ARP::make_arp_reply("10.128.3.55", "10.0.0.20", "00:00:00:00:00:03", "00:00:00:00:00:01");

    natMap.handlePdu(arp1.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(arp2.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultArp1 = natMap.outgoingPduQueue.front();
    checkArp(resultArp1->rfind_pdu<Tins::ARP>(), Tins::ARP::REQUEST, "00:00:00:00:00:00", "00:00:00:00:00:02", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();

    natMap.handlePdu(arp3.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultArp2 = natMap.outgoingPduQueue.front();
    checkArp(resultArp2->rfind_pdu<Tins::ARP>(), Tins::ARP::REPLY, "00:00:00:00:00:02", "00:00:00:00:00:01", "172.16.3.55", "172.27.0.20");
    natMap.outgoingPduQueue.pop();

    natMap.handlePdu(arp4.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());
}

void nattest::testTranslateIp() {
    natMap.reqIpMap.clear();
    natMap.transMap.clear();
    Tins::EthernetII ethW = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:02") / Tins::IP("172.17.0.20", "172.16.3.55") / Tins::TCP();
    Tins::EthernetII eth = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:02") / Tins::IP("172.27.0.20", "172.16.3.55") / Tins::TCP();
    Tins::EthernetII ethAck = Tins::EthernetII("00:00:00:00:00:02", "00:00:00:00:00:01") / Tins::IP("10.0.3.55", "10.0.0.20") / Tins::TCP();
    Tins::EthernetII eth2 = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:03") / Tins::IP("172.27.0.20", "172.17.3.55") / Tins::TCP();
    Tins::EthernetII eth2Ack = Tins::EthernetII("00:00:00:00:00:03", "00:00:00:00:00:01") / Tins::IP("10.0.3.55", "10.0.0.20") / Tins::TCP();
    Tins::EthernetII eth3 = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:04") / Tins::IP("172.27.0.20", "172.18.1.40") / Tins::TCP();
    Tins::EthernetII eth3Ack = Tins::EthernetII("00:00:00:00:00:04", "00:00:00:00:00:01") / Tins::IP("10.0.1.40", "10.0.0.20") / Tins::TCP();
    Tins::EthernetII eth4 = Tins::EthernetII("00:00:00:00:00:02", "00:00:00:00:00:05") / Tins::IP("10.0.3.55", "10.0.1.41") / Tins::TCP();
    Tins::EthernetII eth4Ack = Tins::EthernetII("00:00:00:00:00:05", "00:00:00:00:00:02") / Tins::IP("172.27.1.41", "172.17.3.55") / Tins::TCP();

    natMap.handlePdu(eth.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultArp1 = natMap.outgoingPduQueue.front();
    checkArp(resultArp1->rfind_pdu<Tins::ARP>(), Tins::ARP::REQUEST, "00:00:00:00:00:00", "00:00:00:00:00:02", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();

    natMap.transMap.insert(otonat::NatMap::IPv4AddressEntry(Tins::IPv4Address("172.27.0.20"), Tins::IPv4Address("10.0.0.20")));

    natMap.handlePdu(ethW.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result = natMap.outgoingPduQueue.front();
    checkEth(result->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:02", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(ethAck.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * resultAck = natMap.outgoingPduQueue.front();
    checkEth(resultAck->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:02", "00:00:00:00:00:01", "172.16.3.55", "172.27.0.20");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth2.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result2 = natMap.outgoingPduQueue.front();
    checkEth(result2->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:03", "10.0.0.20", "10.0.3.55");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth2Ack.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result2Ack = natMap.outgoingPduQueue.front();
    checkEth(result2Ack->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:03", "00:00:00:00:00:01", "172.17.3.55", "172.27.0.20");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth3.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result3 = natMap.outgoingPduQueue.front();
    checkEth(result3->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:04", "10.0.0.20", "10.0.1.40");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth3Ack.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result3Ack = natMap.outgoingPduQueue.front();
    checkEth(result3Ack->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:04", "00:00:00:00:00:01", "172.18.1.40", "172.27.0.20");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth4.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result4 = natMap.outgoingPduQueue.front();
    checkEth(result4->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:02", "00:00:00:00:00:05", "172.17.3.55", "172.27.1.41");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(eth4Ack.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.size() == 1);
    const Tins::PDU * result4Ack = natMap.outgoingPduQueue.front();
    checkEth(result4Ack->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:05", "00:00:00:00:00:02", "10.0.1.41", "10.0.3.55");
    natMap.outgoingPduQueue.pop();
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());
}

void nattest::testForMeFromMe() {
    Tins::EthernetII forMe = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:03") / Tins::IP("172.16.0.1", "172.17.3.55") / Tins::TCP();
    Tins::EthernetII FromMe = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:03") / Tins::IP("172.27.0.20", "172.16.0.1") / Tins::TCP();
    natMap.transMap.insert(otonat::NatMap::IPv4AddressEntry(Tins::IPv4Address("172.27.0.20"), Tins::IPv4Address("10.0.0.20")));

    natMap.handlePdu(forMe.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());

    natMap.handlePdu(FromMe.clone());
    CPPUNIT_ASSERT(natMap.outgoingPduQueue.empty());
}

void nattest::testNatInterfaces() {
    CPPUNIT_ASSERT(!natMap.ranges.empty());
}

void nattest::testQueues() {
    Tins::EthernetII forMe = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:03") / Tins::IP("172.16.0.1", "172.17.3.55") / Tins::TCP();
    Tins::EthernetII FromMe = Tins::EthernetII("00:00:00:00:00:01", "00:00:00:00:00:03") / Tins::IP("172.27.0.20", "172.16.0.1") / Tins::TCP();
    natMap.pushPduToIncommingPduQueue(forMe.clone());
    natMap.pushPduToOutgoingPduQueue(FromMe.clone());
    const Tins::PDU * result1 = natMap.popPduOutgoingPduQueue();
    checkEth(result1->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:03", "172.27.0.20", "172.16.0.1");
    const Tins::PDU * result2 = natMap.popPduIncommingPduQueue();
    checkEth(result2->rfind_pdu<Tins::EthernetII>(), "00:00:00:00:00:01", "00:00:00:00:00:03", "172.16.0.1", "172.17.3.55");
    delete result1;
    delete result2;
}

void nattest::testThreadQueues() {
    std::thread threads[10];
    for (int i = 0; i < 10; ++i)
    {
        threads[i] = std::thread(&nattest::testQueues, this);
    }
    
    for (auto& th : threads) 
    {
        th.join();
    }
}

void nattest::printIp(const Tins::IP & ip) {
    std::cout << std::endl << "### Ip-Packet ###" << std::endl;
    std::cout << "ip_dst: " << ip.dst_addr() << std::endl;
    std::cout << "ip_src: " << ip.src_addr() << std::endl;
    std::cout << "#################" << std::endl;
}

void nattest::printEth(const Tins::EthernetII & eth) {
    std::cout << std::endl << "+++ Ethernet-Frame +++" << std::endl;
    std::cout << "frame_dst: " << eth.dst_addr() << std::endl;
    std::cout << "frame_src: " << eth.src_addr();
    const Tins::IP * ip = eth.find_pdu<Tins::IP>();
    if (ip != nullptr) {
        printIp(*ip);
    }

    const Tins::ARP * arp = eth.find_pdu<Tins::ARP>();
    if (arp != nullptr) {
        printArp(*arp);
    } else {
        std::cout << std::endl;
    }

    std::cout << "++++++++++++++++++++++" << std::endl;
}

void nattest::printArp(const Tins::ARP & arp) {
    std::cout << std::endl << "@@@ ARP-Frame @@@" << std::endl;
    switch (arp.opcode()) {
        case Tins::ARP::REQUEST:
            std::cout << "type: Request" << std::endl;
            break;
        case Tins::ARP::REPLY:
            std::cout << "type:  Reply" << std::endl;
            break;
    }

    std::cout << "target_hw_addr: " << arp.target_hw_addr() << std::endl;
    std::cout << "target_ip_addr: " << arp.target_ip_addr() << std::endl;
    std::cout << "sender_hw_addr: " << arp.sender_hw_addr() << std::endl;
    std::cout << "sender_ip_addr: " << arp.sender_ip_addr() << std::endl;
    std::cout << "@@@@@@@@@@@@@@@@@" << std::endl;
}

void nattest::checkAndPrintIp(const Tins::IP & ip, const Tins::IPv4Address & targetDstIp, const Tins::IPv4Address & targetSrcIp) {
    printIp(ip);
    checkIp(ip, targetDstIp, targetSrcIp);
}

void nattest::checkAndPrintEth(const Tins::EthernetII & eth, const Tins::HWAddress<6> & targetDstMac, const Tins::HWAddress<6> & targetSrcMac, const Tins::IPv4Address & targetDstIp, const Tins::IPv4Address & targetSrcIp) {
    printEth(eth);
    checkEth(eth, targetDstMac, targetSrcMac, targetDstIp, targetSrcIp);
}

void nattest::checkAndPrintArp(const Tins::ARP& arp, uint16_t targetArpType, const Tins::HWAddress<6>& targetTargetMac, const Tins::HWAddress<6>& targetSenderMac, const Tins::IPv4Address& targetTaregtIp, const Tins::IPv4Address& targetSenderIp) {
    printArp(arp);
    checkArp(arp, targetArpType, targetTargetMac, targetSenderMac, targetTaregtIp, targetSenderIp);
}

void nattest::checkIp(const Tins::IP & ip, const Tins::IPv4Address & targetDstIp, const Tins::IPv4Address & targetSrcIp) {
    const Tins::IPv4Address & packetSrcIp = ip.src_addr();
    const Tins::IPv4Address & packetDstIp = ip.dst_addr();
    CPPUNIT_ASSERT_EQUAL(targetSrcIp, packetSrcIp);
    CPPUNIT_ASSERT_EQUAL(targetDstIp, packetDstIp);
}

void nattest::checkEth(const Tins::EthernetII & eth, const Tins::HWAddress<6> & targetDstMac, const Tins::HWAddress<6> & targetSrcMac, const Tins::IPv4Address & targetDstIp, const Tins::IPv4Address & targetSrcIp) {
    const Tins::HWAddress<6> & frameSrcMac = eth.src_addr();
    const Tins::HWAddress<6> & frameDstMac = eth.dst_addr();
    CPPUNIT_ASSERT_EQUAL(targetSrcMac, frameSrcMac);
    CPPUNIT_ASSERT_EQUAL(targetDstMac, frameDstMac);
    const Tins::IP * ip = eth.find_pdu<Tins::IP>();
    if (ip != nullptr) {
        checkIp(*ip, targetDstIp, targetSrcIp);
    }
}

void nattest::checkArp(const Tins::ARP & arp, uint16_t targetArpType, const Tins::HWAddress<6> & targetTargetMac, const Tins::HWAddress<6> & targetSenderMac, const Tins::IPv4Address & targetTaregtIp, const Tins::IPv4Address & targetSenderIp) {
    const Tins::HWAddress<6> & senderMac = arp.sender_hw_addr();
    const Tins::HWAddress<6> & targetMac = arp.target_hw_addr();
    const Tins::IPv4Address & senderIp = arp.sender_ip_addr();
    const Tins::IPv4Address & targetIp = arp.target_ip_addr();
    CPPUNIT_ASSERT_EQUAL(targetArpType, arp.opcode());
    CPPUNIT_ASSERT_EQUAL(targetSenderMac, senderMac);
    CPPUNIT_ASSERT_EQUAL(targetTargetMac, targetMac);
    CPPUNIT_ASSERT_EQUAL(targetSenderIp, senderIp);
    CPPUNIT_ASSERT_EQUAL(targetTaregtIp, targetIp);
}