mirror of
https://code.semirocket.science/wrapsix
synced 2024-09-20 07:11:06 +03:00
220 lines
8.2 KiB
C
220 lines
8.2 KiB
C
#ifndef WRAPPER_H
|
|
#define WRAPPER_H
|
|
|
|
#include <pcap.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <netinet/ip6.h> /* ip6_hdr */
|
|
|
|
/* Default snap length (maximum bytes per packet to capture) */
|
|
#define SNAP_LEN BUFSIZ
|
|
|
|
/* Ethernet headers are always exactly 14 bytes */
|
|
#define SIZE_ETHERNET 14
|
|
|
|
/* IPv6 headers are always exactly 40 bytes */
|
|
#define SIZE_IP6 40
|
|
|
|
/* MAC address structure */
|
|
struct s_mac_addr {
|
|
unsigned char a;
|
|
unsigned char b;
|
|
unsigned char c;
|
|
unsigned char d;
|
|
unsigned char e;
|
|
unsigned char f;
|
|
};
|
|
|
|
/* Ethernet header structure */
|
|
struct s_ethernet {
|
|
struct s_mac_addr dest; /* 48 b; destination host (MAC) address */
|
|
struct s_mac_addr src; /* 48 b; source host (MAC) address */
|
|
unsigned short type; /* 16 b; IP/ARP/RARP/... */
|
|
};
|
|
|
|
/* ARP structure */
|
|
struct s_arp {
|
|
unsigned short hw; /* 16 b; hardware type [0x0001] */
|
|
unsigned short proto; /* 16 b; protocol type [0x0800] */
|
|
unsigned char hw_len; /* 8 b; length of hardware address in bytes [0x06] */
|
|
unsigned char proto_len; /* 8 b; length of protocol address in bytes [0x04] */
|
|
unsigned short opcode; /* 16 b; operation code: [0x0001] or [0x0002] */
|
|
struct s_mac_addr mac_src; /* 48 b; sender hardware address */
|
|
struct in_addr ip_src; /* 32 b; sender protocol address */
|
|
struct s_mac_addr mac_dest; /* 48 b; target hardware address */
|
|
struct in_addr ip_dest; /* 32 b; target protocol address */
|
|
}__attribute__((__packed__));
|
|
|
|
/* IPv4 header structure */
|
|
struct s_ip4 {
|
|
unsigned char ver_ihl;
|
|
unsigned char tos; /* 8 b; type of service */
|
|
unsigned short pckt_len; /* 16 b; total length of the packet (IP header + payload) */
|
|
unsigned short id; /* 16 b; id of the packet - for purpose of fragmentation */
|
|
unsigned short flags_offset; /* 16 b; 3 b - flags, 13 b - fragment offset in bytes */
|
|
unsigned char ttl; /* 8 b; time to live */
|
|
unsigned char proto; /* 8 b; protocol in the payload */
|
|
unsigned short checksum; /* 16 b */
|
|
struct in_addr ip_src; /* 32 b; source address */
|
|
struct in_addr ip_dest; /* 32 b; destination address */
|
|
};
|
|
|
|
/* pseudo IPv4 header for checksum */
|
|
struct s_ip4_pseudo {
|
|
struct in_addr ip_src; /* 32 b; source address */
|
|
struct in_addr ip_dest; /* 32 b; destination address */
|
|
unsigned char zeros; /* 8 b */
|
|
unsigned char proto; /* 8 b; protocol in the payload */
|
|
unsigned short len; /* 16 b; payload length */
|
|
};
|
|
|
|
/* IPv6 header structure */
|
|
struct s_ip6 {
|
|
unsigned char ver; /* 8 b; version */
|
|
unsigned char traffic_class; /* 8 b; traffic class */
|
|
unsigned short flow_label; /* 16 b; flow label (qos) */
|
|
unsigned short len; /* 16 b; payload length */
|
|
unsigned char next_header; /* 8 b; next header */
|
|
unsigned char hop_limit; /* 8 b; hop limit (replaces ttl) */
|
|
struct in6_addr ip_src; /* 128 b; source address */
|
|
struct in6_addr ip_dest; /* 128 b; destination address */
|
|
};
|
|
|
|
/* pseudo IPv6 header for checksum */
|
|
struct s_ip6_pseudo {
|
|
struct in6_addr ip_src; /* 128 b; source address */
|
|
struct in6_addr ip_dest; /* 128 b; destination address */
|
|
unsigned int len; /* 32 b; payload length */
|
|
unsigned int zeros:24; /* 24 b; reserved */
|
|
unsigned char next_header; /* 8 b; next header */
|
|
};
|
|
|
|
/* IPv6 fragment header */
|
|
struct s_ip6_fragment {
|
|
unsigned char next_header; /* 8 b; next header */
|
|
unsigned char zeros; /* 8 b; reserved */
|
|
unsigned short offset_flag; /* 16 b; 13 b - data offset, 2 b - reserved, 1 b - flag */
|
|
unsigned int id; /* 32 b; identification */
|
|
};
|
|
|
|
/* TCP structure */
|
|
struct s_tcp {
|
|
unsigned short port_src; /* 16 b; source port */
|
|
unsigned short port_dest; /* 16 b; destination port */
|
|
unsigned int seq; /* 32 b; sequence number */
|
|
unsigned int ack; /* 32 b; acknowledgement number */
|
|
unsigned char dataoff_res; /* 8 b; 4 b - data offset, 4 b - reserved (zeros) */
|
|
unsigned char flags; /* 8 b; 8 flags */
|
|
unsigned short windows_size; /* 16 b; size of the receive window */
|
|
unsigned short checksum; /* 16 b */
|
|
unsigned short urgent_pointer; /* 16 b; indicates last urgent data byte */
|
|
};
|
|
|
|
/* UDP structure */
|
|
struct s_udp {
|
|
unsigned short port_src; /* 16 b; source port */
|
|
unsigned short port_dest; /* 16 b; destination port */
|
|
unsigned short len; /* 16 b; header & data length */
|
|
unsigned short checksum; /* 16 b */
|
|
};
|
|
|
|
/* ICMP header structure */
|
|
struct s_icmp {
|
|
unsigned char type; /* 8 b; ICMP type */
|
|
unsigned char code; /* 8 b; further specification of ICMP type */
|
|
unsigned short checksum; /* 16 b */
|
|
};
|
|
|
|
/* ICMP - ping structure */
|
|
struct s_icmp_ping {
|
|
unsigned short id; /* 16 b; ID value for ECHO REPLY */
|
|
unsigned short seq; /* 16 b; sequence value for ECHO REPLY */
|
|
};
|
|
|
|
/* ICMPv6 - NDP option structure */
|
|
struct s_icmp_ndp_option {
|
|
unsigned char type; /* 8 b; type of the option */
|
|
unsigned char len; /* 8 b; length of the option (including this header!) */
|
|
};
|
|
|
|
/* ICMPv6 - NDP NS structure */
|
|
struct s_icmp_ndp_ns {
|
|
unsigned int zeros; /* 32 b; reserved section */
|
|
struct in6_addr target; /* 128 b; target IP address */
|
|
};
|
|
|
|
/* ICMPv6 - NDP NA structure */
|
|
struct s_icmp_ndp_na {
|
|
unsigned char flags; /* 8 b; 3 flags */
|
|
unsigned int zeros:24; /* 24 b; reserved section */
|
|
struct in6_addr target; /* 128 b; target IP address */
|
|
unsigned char o_type; /* 8 b; option - type */
|
|
unsigned char o_len; /* 8 b; option - length */
|
|
struct s_mac_addr o_tlla; /* 48 b; option - target link-layer address */
|
|
};
|
|
/* INNAF = ICMPv6 NDP NA Flag */
|
|
#define INNAF_R 0x80 /* router flag */
|
|
#define INNAF_S 0x40 /* solicited flag */
|
|
#define INNAF_O 0x20 /* override flag */
|
|
|
|
/* Missing ethertypes */
|
|
#define ETHERTYPE_IPV6 0x86dd
|
|
|
|
/* ICMP types */
|
|
#define ICMP4_ECHO_REQUEST 0x8
|
|
#define ICMP4_ECHO_REPLY 0x0
|
|
|
|
/* ICMPv6 types */
|
|
#define ICMP6_ECHO_REQUEST 0x80
|
|
#define ICMP6_ECHO_REPLY 0x81
|
|
#define ICMP6_NDP_RS 0x85
|
|
#define ICMP6_NDP_RA 0x86
|
|
#define ICMP6_NDP_NS 0x87
|
|
#define ICMP6_NDP_NA 0x88
|
|
#define ICMP6_NDP_RM 0x89
|
|
|
|
/* Prototypes */
|
|
int get_mac_addr(const char *dev, struct s_mac_addr *addr);
|
|
int get_ip_addr(const char *dev, struct in_addr *addr);
|
|
int get_dev_index(const char *dev);
|
|
|
|
void process_packet(u_char *args, const struct pcap_pkthdr *header, const u_char *packet);
|
|
|
|
void process_packet4(const struct s_ethernet *eth, const unsigned char *packet);
|
|
void process_tcp4(const struct s_ethernet *eth_hdr, struct s_ip4 *ip_hdr, const unsigned char *payload, unsigned short data_size);
|
|
void process_udp4(const struct s_ethernet *eth_hdr, struct s_ip4 *ip_hdr, const unsigned char *payload, unsigned short data_size);
|
|
void process_icmp4(const struct s_ethernet *eth_hdr, struct s_ip4 *ip_hdr, const unsigned char *payload, unsigned short data_size);
|
|
void process_arp(const struct s_ethernet *eth_hdr, const unsigned char *arp_packet);
|
|
|
|
void process_packet6(const struct s_ethernet *eth, const unsigned char *packet);
|
|
void process_tcp6(const struct s_ethernet *eth_hdr, struct s_ip6 *ip_hdr, const unsigned char *payload);
|
|
void process_udp6(const struct s_ethernet *eth, struct s_ip6 *ip, const unsigned char *payload);
|
|
void process_icmp6(const struct s_ethernet *eth, struct s_ip6 *ip, const unsigned char *payload);
|
|
void process_ndp(const struct s_ethernet *eth_hdr, struct s_ip6 *ip_hdr, unsigned char *icmp_data);
|
|
|
|
void send_there(struct in_addr ip4_addr, unsigned char ttl, unsigned int type, unsigned char *payload, unsigned int paylen);
|
|
void send_raw(unsigned char *packet, int packet_size);
|
|
void send_raw_ipv4(struct in_addr ip4_addr, unsigned char *packet, int packet_size);
|
|
|
|
unsigned short checksum(const void *_buf, int len);
|
|
unsigned short checksum_ipv4(struct in_addr ip_src, struct in_addr ip_dest, unsigned short length, unsigned char proto, unsigned char *data);
|
|
unsigned short checksum_ipv6(struct in6_addr ip_src, struct in6_addr ip_dest, unsigned short paylen, unsigned char proto, unsigned char *data);
|
|
|
|
/* Variables */
|
|
extern struct s_mac_addr *mac; /* MAC address of the device */
|
|
extern char *dev; /* capture device name */
|
|
extern int dev_index; /* capture device index */
|
|
extern struct in_addr *dev_ip; /* IP address associated with the device */
|
|
extern struct in6_addr ip6addr_wrapsix; /* IPv6 prefix of WrapSix addresses */
|
|
extern struct in_addr ip4addr_wrapsix; /* IPv4 address for WrapSix */
|
|
|
|
#endif
|