mirror of
https://github.com/mfontanini/libtins
synced 2026-01-26 20:01:35 +01:00
139 lines
4.0 KiB
C++
139 lines
4.0 KiB
C++
/*
|
|
* libtins is a net packet wrapper library for crafting and
|
|
* interpreting sniffed packets.
|
|
*
|
|
* Copyright (C) 2011 Nasel
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#ifndef __TCP_H
|
|
#define __TCP_H
|
|
|
|
|
|
#include <vector>
|
|
#include <stdint.h>
|
|
#ifndef WIN32
|
|
#include <endian.h>
|
|
#endif
|
|
#include "pdu.h"
|
|
|
|
namespace Tins {
|
|
|
|
class TCP : public PDU {
|
|
public:
|
|
enum Flags {
|
|
FIN,
|
|
SYN,
|
|
RST,
|
|
PSH,
|
|
ACK,
|
|
URG,
|
|
ECE,
|
|
CWR
|
|
};
|
|
|
|
enum Options {
|
|
MSS = 2,
|
|
TSOPT = 8
|
|
};
|
|
|
|
TCP(uint16_t dport = 0, uint16_t sport = 0);
|
|
~TCP();
|
|
|
|
inline uint16_t dport() const { return _tcp.dport; }
|
|
inline uint16_t sport() const { return _tcp.sport; }
|
|
inline uint32_t seq() const { return _tcp.seq; }
|
|
inline uint32_t ack_seq() const { return _tcp.ack_seq; }
|
|
inline uint16_t window() const { return _tcp.window; }
|
|
inline uint16_t check() const { return _tcp.check; }
|
|
inline uint16_t urg_ptr() const { return _tcp.urg_ptr; }
|
|
|
|
void dport(uint16_t new_dport);
|
|
void sport(uint16_t new_sport);
|
|
void seq(uint32_t new_seq);
|
|
void ack_seq(uint32_t new_ack_seq);
|
|
void window(uint16_t new_window);
|
|
void check(uint16_t new_check);
|
|
void urg_ptr(uint16_t new_urg_ptr);
|
|
void payload(uint8_t *new_payload, uint32_t new_payload_size);
|
|
|
|
void set_mss(uint16_t value);
|
|
void set_timestamp(uint32_t value, uint32_t reply);
|
|
void set_flag(Flags tcp_flag, uint8_t value);
|
|
void add_option(Options tcp_option, uint8_t length = 0, uint8_t *data = 0);
|
|
|
|
/* Virtual methods */
|
|
uint32_t header_size() const;
|
|
void write_serialization(uint8_t *buffer, uint32_t total_sz);
|
|
private:
|
|
struct tcphdr {
|
|
uint16_t sport;
|
|
uint16_t dport;
|
|
uint32_t seq;
|
|
uint32_t ack_seq;
|
|
#if __BYTE_ORDER == __LITTLE_ENDIAN
|
|
uint16_t res1:4,
|
|
doff:4,
|
|
fin:1,
|
|
syn:1,
|
|
rst:1,
|
|
psh:1,
|
|
ack:1,
|
|
urg:1,
|
|
ece:1,
|
|
cwr:1;
|
|
#elif __BYTE_ORDER == __BIG_ENDIAN
|
|
uint16_t doff:4,
|
|
res1:4,
|
|
cwr:1,
|
|
ece:1,
|
|
urg:1,
|
|
ack:1,
|
|
psh:1,
|
|
rst:1,
|
|
syn:1,
|
|
fin:1;
|
|
#else
|
|
#error "Endian is not LE nor BE..."
|
|
#endif
|
|
uint16_t window;
|
|
uint16_t check;
|
|
uint16_t urg_ptr;
|
|
} __attribute__((packed));
|
|
|
|
struct TCPOption {
|
|
TCPOption(uint8_t okind, uint8_t olength, uint8_t *odata) :
|
|
kind(okind), length(olength), data(odata) { }
|
|
|
|
uint8_t *write(uint8_t *buffer);
|
|
|
|
uint8_t kind, length;
|
|
uint8_t *data;
|
|
};
|
|
|
|
static const uint16_t DEFAULT_WINDOW;
|
|
|
|
uint16_t do_checksum(uint8_t *start, uint8_t *end) const;
|
|
|
|
tcphdr _tcp;
|
|
std::vector<TCPOption> _options;
|
|
uint8_t *_payload;
|
|
uint32_t _payload_size, _options_size, _total_options_size;
|
|
};
|
|
};
|
|
|
|
#endif
|