56 lines
2.0 KiB
C++
56 lines
2.0 KiB
C++
#ifndef NATMAP_H
|
|
#define NATMAP_H
|
|
|
|
#include <vector>
|
|
#include <map>
|
|
#include <queue>
|
|
#include <tins/tins.h>
|
|
#include "NatRange.h"
|
|
namespace otonat {
|
|
|
|
class NatMap {
|
|
public:
|
|
typedef std::vector<NatRange> NatRangeList;
|
|
typedef std::queue<const Tins::PDU *> PduQueue;
|
|
typedef std::pair<Tins::IPv4Address, Tins::IPv4Address> IPv4AddressEntry;
|
|
typedef std::map<Tins::IPv4Address, Tins::HWAddress < 6 >> IpAdressMacMap;
|
|
typedef std::map<Tins::IPv4Address, Tins::IPv4Address> IpAdressMap;
|
|
|
|
NatMap() {
|
|
}
|
|
|
|
NatMap(NatRangeList rangeList);
|
|
virtual ~NatMap();
|
|
NatMap(const NatMap& other);
|
|
NatMap& operator=(const NatMap& other);
|
|
NatRangeList ranges;
|
|
IpAdressMacMap arpMap;
|
|
IpAdressMap transMap;
|
|
IpAdressMap reqIpMap;
|
|
PduQueue incommingPduQueue;
|
|
PduQueue outgoingPduQueue;
|
|
void handlePdu(const Tins::PDU * pdu);
|
|
|
|
protected:
|
|
|
|
private:
|
|
bool handleIp(Tins::IP * ip, const Tins::PDU * originPDU);
|
|
bool handleArp(Tins::ARP * arp);
|
|
bool handleArpReq(Tins::ARP * arp);
|
|
bool handleArpReply(Tins::ARP * arp);
|
|
Tins::IPv4Address InsertOrUdpateTranslateIpAddress(const Tins::IPv4Address & originIp, const NatRange & range);
|
|
Tins::IPv4Address InsertOrUdpateTranslateIpAddress(const Tins::IPv4Address & originIp, const Tins::IPv4Address & transIp, NatRangeList & rangeList);
|
|
void TranslateIpPacket(Tins::IP * ip, const Tins::IPv4Address & transIp);
|
|
Tins::IPv4Address zeroIp;
|
|
bool isForMeOrFromMeIp(const Tins::IP * ip);
|
|
bool isForMeOrFromMeArp(const Tins::ARP * arp);
|
|
static bool isForMeOrFromMeIp(const Tins::IP * ip, const NatRangeList & rangeList);
|
|
bool isIpInMyRanges(const Tins::IPv4Address & ipAddr);
|
|
static bool isIpInMyRanges(const Tins::IPv4Address & ipAddr, const NatRangeList & rangeList);
|
|
void SendTranslatedArpRequest(const Tins::ARP * arp);
|
|
};
|
|
}
|
|
|
|
|
|
#endif // NATMAP_H
|