mirror of
https://github.com/mfontanini/libtins
synced 2026-01-29 13:04:28 +01:00
Added IP pdu class. It's compiling so far.
This commit is contained in:
70
include/ip.h
Normal file
70
include/ip.h
Normal file
@@ -0,0 +1,70 @@
|
||||
#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
|
||||
@@ -4,37 +4,40 @@
|
||||
|
||||
#include <stdint.h>
|
||||
|
||||
class PDU {
|
||||
public:
|
||||
PDU(uint32_t pdu_flag, PDU *next_pdu = 0);
|
||||
virtual ~PDU();
|
||||
namespace Tins {
|
||||
|
||||
/* This PDU's header size only. */
|
||||
virtual uint32_t header_size() const = 0;
|
||||
/* This PDU's trailer size only. Defaults to 0. */
|
||||
virtual uint32_t trailer_size() const { return 0; }
|
||||
|
||||
/* The size of the whole chain of PDUs, including this one. */
|
||||
uint32_t size() const;
|
||||
|
||||
inline const PDU *inner_pdu() const { return _inner_pdu; }
|
||||
|
||||
/* When setting a new inner_pdu, the instance takes
|
||||
* ownership of the object, therefore deleting it when
|
||||
* it's no longer required. */
|
||||
void inner_pdu(PDU *next_pdu);
|
||||
|
||||
/* Serializes the whole chain of PDU's, including this one. */
|
||||
uint8_t *serialize();
|
||||
|
||||
/* Serialize this PDU storing the result in buffer. */
|
||||
void serialize(uint8_t *buffer, uint32_t total_sz);
|
||||
protected:
|
||||
/* Each PDU's own implementation of serialization. */
|
||||
virtual void write_serialization(uint8_t *buffer, uint32_t total_sz) = 0;
|
||||
private:
|
||||
uint32_t _pdu_flag;
|
||||
PDU *_inner_pdu;
|
||||
class PDU {
|
||||
public:
|
||||
PDU(uint32_t flag, PDU *next_pdu = 0);
|
||||
virtual ~PDU();
|
||||
|
||||
/* This PDU's header size only. */
|
||||
virtual uint32_t header_size() const = 0;
|
||||
/* This PDU's trailer size only. Defaults to 0. */
|
||||
virtual uint32_t trailer_size() const { return 0; }
|
||||
/* The size of the whole chain of PDUs, including this one. */
|
||||
inline uint32_t size() const;
|
||||
inline uint32_t flag() const { return _flag; }
|
||||
inline const PDU *inner_pdu() const { return _inner_pdu; }
|
||||
|
||||
void flag(uint32_t new_flag);
|
||||
/* When setting a new inner_pdu, the instance takes
|
||||
* ownership of the object, therefore deleting it when
|
||||
* it's no longer required. */
|
||||
void inner_pdu(PDU *next_pdu);
|
||||
|
||||
/* Serializes the whole chain of PDU's, including this one. */
|
||||
uint8_t *serialize();
|
||||
|
||||
/* Serialize this PDU storing the result in buffer. */
|
||||
void serialize(uint8_t *buffer, uint32_t total_sz);
|
||||
protected:
|
||||
/* Each PDU's own implementation of serialization. */
|
||||
virtual void write_serialization(uint8_t *buffer, uint32_t total_sz) = 0;
|
||||
private:
|
||||
uint32_t _flag;
|
||||
PDU *_inner_pdu;
|
||||
};
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
16
include/utils.h
Normal file
16
include/utils.h
Normal file
@@ -0,0 +1,16 @@
|
||||
#ifndef __UTILS_H
|
||||
#define __UTILS_H
|
||||
|
||||
|
||||
#include <string>
|
||||
#include <stdint.h>
|
||||
|
||||
namespace Tins {
|
||||
/* Utils namespace. */
|
||||
namespace Utils {
|
||||
uint32_t ip_to_int(const std::string &ip);
|
||||
std::string ip_to_string(uint32_t ip);
|
||||
};
|
||||
};
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user