mirror of
https://github.com/mfontanini/libtins
synced 2026-01-25 11:41:35 +01:00
71 lines
2.2 KiB
C++
71 lines
2.2 KiB
C++
#ifndef __IP_H
|
|
#define __IP_H
|
|
|
|
#ifndef WIN32
|
|
#include <endian.h>
|
|
#endif
|
|
#include <string>
|
|
#include <stdint.h>
|
|
#include "pdu.h"
|
|
|
|
namespace Tins {
|
|
|
|
class IP : public PDU {
|
|
public:
|
|
IP(const std::string &ip_dst = "", const std::string &ip_src = "");
|
|
IP(uint32_t ip_dst = 0, uint32_t ip_src = 0);
|
|
|
|
inline uint8_t tos() const { return _ip.tos; }
|
|
inline uint16_t tot_len() const { return _ip.tot_len; }
|
|
inline uint16_t id() const { return _ip.id; }
|
|
inline uint16_t frag_off() const { return _ip.frag_off; }
|
|
inline uint8_t ttl() const { return _ip.ttl; }
|
|
inline uint8_t protocol() const { return _ip.protocol; }
|
|
inline uint16_t check() const { return _ip.check; }
|
|
inline uint32_t source_address() const { return _ip.saddr; }
|
|
inline uint32_t dest_address() const { return _ip.daddr; }
|
|
|
|
void tos(uint8_t new_tos);
|
|
void tot_len(uint16_t new_tot_len);
|
|
void id(uint16_t new_id);
|
|
void frag_off(uint16_t new_frag_off);
|
|
void ttl(uint8_t new_ttl);
|
|
void protocol(uint8_t new_protocol);
|
|
void check(uint16_t new_check);
|
|
void source_address(const std::string &ip);
|
|
void source_address(uint32_t ip);
|
|
void dest_address(const std::string &ip);
|
|
void dest_address(uint32_t ip);
|
|
|
|
/* Virtual methods */
|
|
uint32_t header_size();
|
|
void write_serialization(uint8_t *buffer, uint32_t total_sz);
|
|
private:
|
|
struct iphdr {
|
|
#if __BYTE_ORDER == __LITTLE_ENDIAN
|
|
unsigned int ihl:4;
|
|
unsigned int version:4;
|
|
#elif __BYTE_ORDER == __BIG_ENDIAN
|
|
unsigned int version:4;
|
|
unsigned int ihl:4;
|
|
#else
|
|
# error "Endian is not LE nor BE..."
|
|
#endif
|
|
uint8_t tos;
|
|
uint16_t tot_len;
|
|
uint16_t id;
|
|
uint16_t frag_off;
|
|
uint8_t ttl;
|
|
uint8_t protocol;
|
|
uint16_t check;
|
|
uint32_t saddr;
|
|
uint32_t daddr;
|
|
/*The options start here. */
|
|
} __attribute__((packed));
|
|
|
|
iphdr _ip;
|
|
};
|
|
};
|
|
|
|
#endif
|