118 lines
3.0 KiB
C++
118 lines
3.0 KiB
C++
#include "TinsNetworkInterfaceCard.h"
|
|
#include <functional>
|
|
#include <mutex>
|
|
#include <tins/ethernetII.h>
|
|
#include <tins/network_interface.h>
|
|
#include <tins/packet_sender.h>
|
|
#include <tins/sniffer.h>
|
|
|
|
ByteVectorHash TinsNetworkInterfaceCard::byteVectorHash;
|
|
|
|
TinsNetworkInterfaceCard::TinsNetworkInterfaceCard() : TinsNetworkInterfaceCard(Tins::NetworkInterface::default_interface())
|
|
{
|
|
}
|
|
|
|
TinsNetworkInterfaceCard::TinsNetworkInterfaceCard(const std::string& name) : TinsNetworkInterfaceCard(name.c_str())
|
|
{
|
|
|
|
}
|
|
|
|
TinsNetworkInterfaceCard::TinsNetworkInterfaceCard(const char* name) : TinsNetworkInterfaceCard(Tins::NetworkInterface(name))
|
|
{
|
|
|
|
}
|
|
|
|
TinsNetworkInterfaceCard::TinsNetworkInterfaceCard(const Tins::IPv4Address & ip) : TinsNetworkInterfaceCard(Tins::NetworkInterface(ip))
|
|
{
|
|
|
|
}
|
|
|
|
TinsNetworkInterfaceCard::TinsNetworkInterfaceCard(const Tins::IPv6Address & ipv6) : TinsNetworkInterfaceCard(Tins::NetworkInterface(ipv6))
|
|
{
|
|
|
|
}
|
|
|
|
TinsNetworkInterfaceCard::TinsNetworkInterfaceCard(const Tins::NetworkInterface & networkInterface)
|
|
{
|
|
snifferConfig = std::make_unique<Tins::SnifferConfiguration>();
|
|
snifferConfig->set_promisc_mode(true);
|
|
snifferConfig->set_immediate_mode(true);
|
|
sniffer = std::make_unique<Tins::Sniffer>(networkInterface.name());
|
|
packetSender = std::make_unique<Tins::PacketSender>(networkInterface);
|
|
sendPduHashList = std::make_unique<HashList>();
|
|
hashListMutex = std::make_unique<std::mutex>();
|
|
}
|
|
|
|
TinsNetworkInterfaceCard::~TinsNetworkInterfaceCard()
|
|
{
|
|
}
|
|
|
|
bool TinsNetworkInterfaceCard::handle(const Tins::PDU &pdu, IPacketHandler *callBackHandler)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
void TinsNetworkInterfaceCard::sendPacket(const Tins::PDU &pdu)
|
|
{
|
|
Tins::PDU * clonePdu = pdu.clone();
|
|
if (clonePdu == nullptr){
|
|
return;
|
|
}
|
|
|
|
Tins::PDU & clonePduRef = *clonePdu;
|
|
Tins::PDU * phyLayerLessPdu = getPhyLessPduPtr(clonePduRef);
|
|
|
|
// layer3 only
|
|
if (phyLayerLessPdu != nullptr)
|
|
{
|
|
|
|
packetSender->send(clonePduRef);
|
|
}
|
|
|
|
delete clonePdu;
|
|
}
|
|
|
|
void TinsNetworkInterfaceCard::addPduToHashList(IN Tins::PDU &pdu)
|
|
{
|
|
std::lock_guard<std::mutex> lockGuard(*hashListMutex);
|
|
const ByteVector byteVector = pdu.serialize();
|
|
const std::size_t byteVectorHashValue = byteVectorHash(byteVector);
|
|
sendPduHashList->push_back(byteVectorHashValue);
|
|
}
|
|
|
|
bool TinsNetworkInterfaceCard::searchAndRemoveHashListItem(IN const std::size_t removedHash)
|
|
{
|
|
std::lock_guard<std::mutex> lockGuard(*hashListMutex);
|
|
const HashList::const_iterator start = sendPduHashList->cbegin();
|
|
const HashList::const_iterator end = sendPduHashList->cend();
|
|
const HashList::const_iterator findIt = std::find(start, end, removedHash);
|
|
if (findIt == end)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
sendPduHashList->erase(findIt);
|
|
return true;
|
|
}
|
|
|
|
void TinsNetworkInterfaceCard::startListen()
|
|
{
|
|
|
|
}
|
|
|
|
void TinsNetworkInterfaceCard::stopListen()
|
|
{
|
|
|
|
}
|
|
|
|
Tins::PDU * TinsNetworkInterfaceCard::getPhyLessPduPtr(IN Tins::PDU & pdu) const
|
|
{
|
|
const Tins::EthernetII * ethPdu = pdu.find_pdu<Tins::EthernetII>();
|
|
if (ethPdu == nullptr)
|
|
{
|
|
return &pdu;
|
|
}
|
|
|
|
return getPhyLessPduPtr(*(ethPdu->inner_pdu()));
|
|
}
|