rawsock.h

Go to the documentation of this file.

#ifndef RAWSOCK_H_INCLUDED
#define RAWSOCK_H_INCLUDED

#include <net/ethernet.h>
#include <linux/udp.h>  
#include <linux/ip.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdbool.h>
#include <string.h>

#ifdef __ANDROID__
    #define udphdr __kernel_udphdr
#endif

#define MAC_FILE_PATH_SIZE 23 
#define MAC_ADDR_SIZE 6 
// Errors
#define ERR_WLAN_NOIF 0 
#define ERR_WLAN_SOCK -1 
#define ERR_WLAN_GETIFADDRS -2 
#define ERR_WLAN_INDEX -3 
#define ERR_WLAN_GETSRCMAC -4 
#define ERR_WLAN_GETIFINDEX -5 
#define ERR_WLAN_GETSRCIP -6 
#define ERR_IPHEAD_SOCK -10 
#define ERR_IPHEAD_NOSRCADDR -11 
#define ERR_VIFPRINTER_SOCK -20 
#define ERR_VIFPRINTER_GETIFADDRS -21 
// wlanLookup() modes
#define WLANLOOKUP_WLAN 0 
#define WLANLOOKUP_NONWLAN 1 
// Names
#define MAC_NULL 0x00 
#define MAC_BROADCAST 0x01 
#define MAC_UNICAST 0x02 
#define MAC_MULTICAST 0x03 
#define MAC_ZERO 0x04 
// Useful constants
// Additional EtherTypes
#define ETHERTYPE_GEONET 0x8947 
#define ETHERTYPE_WSMP 0x88DC 
// Special wlanLookup index values
#define WLANLOOKUP_LOOPBACK -1 
// IP constants
#define BASIC_IHL 5 
#define IPV4 4 
#define BASIC_UDP_TTL 64 
// UDP constant
#define UDPHEADERLEN 8 
// Useful masks
#define FLAG_NOFRAG_MASK (1<<6) 
#define FLAG_RESERVED_MASK (1<<7) 
#define FLAG_MOREFRAG_MASK (1<<5) 
// Checksum protocols, to be used inside the validateEthCsum() function 
//  (0x00->0x7F should be simple types, 0x80->0xFF should be combined types)
#define CSUM_IP 0x00 
#define CSUM_UDP 0x01 
#define CSUM_UDPIP 0x80 
// Useful macros for printing MAC addresses inside the printf() familty of functions
#define PRI_MAC "%02x:%02x:%02x:%02x:%02x:%02x" 
#define MAC_PRINTER(mac_array) mac_array[0], mac_array[1], mac_array[2], mac_array[3], mac_array[4], mac_array[5] 
// Useful macros for reading MAC addresses inside the scanf() familty of functions
#define SCN_MAC "%x:%x:%x:%x:%x:%x%*c" 
#define MAC_SCANNER(mac_array) &mac_array[0], &mac_array[1], &mac_array[2], &mac_array[3], &mac_array[4], &mac_array[5] 
// Size definitions (macros)
#define UDP_PACKET_SIZE(data) sizeof(struct udphdr)+sizeof(data)  
#define IP_UDP_PACKET_SIZE(data) sizeof(struct iphdr)+sizeof(struct udphdr)+sizeof(data) 
#define ETH_IP_UDP_PACKET_SIZE(data) sizeof(struct ether_header)+sizeof(struct iphdr)+sizeof(struct udphdr)+sizeof(data) 
#define UDP_PACKET_SIZE_S(size) sizeof(struct udphdr)+size 
#define IP_UDP_PACKET_SIZE_S(size) sizeof(struct iphdr)+sizeof(struct udphdr)+size 
#define ETH_IP_UDP_PACKET_SIZE_S(size) sizeof(struct ether_header)+sizeof(struct iphdr)+sizeof(struct udphdr)+size 
#ifndef BYTE_TYPE
#define BYTE_TYPE
typedef uint8_t byte_t; 
#endif

typedef uint8_t * macaddr_t; 
typedef unsigned short ethertype_t; 
typedef int rawsockerr_t; 
typedef unsigned char csumt_t; 
typedef __sum16 csum16_t; 
struct ipaddrs {
    in_addr_t src; 
    in_addr_t dst; 
};

typedef enum {
    UNSET_P,    
    UDP,        
    AMQP_0_9,   
    AMQP_1_0    
} protocol_t;

// General utilities
rawsockerr_t wlanLookup(char *devname, int *ifindex, macaddr_t mac, struct in_addr *srcIP, int index, int mode);
rawsockerr_t vifPrinter(FILE *stream);
macaddr_t prepareMacAddrT();
unsigned int macAddrTypeGet(macaddr_t mac);
void freeMacAddrT(macaddr_t mac);
void rs_printerror(FILE *stream,rawsockerr_t code);
void display_packet(const char *text,byte_t *packet,unsigned int len);
void display_packetc(const char *text,byte_t *packet,unsigned int len);
uint64_t hton64 (uint64_t hostu64); // Like 'htonl()' but for 64-bits unsigned integers
uint64_t ntoh64 (uint64_t netu64); // Like 'ntohl()' but for 64-bits unsigned integers

// Ethernet level functions
void etherheadPopulateB(struct ether_header *etherHeader, macaddr_t mac, ethertype_t type);
void etherheadPopulate(struct ether_header *etherHeader, macaddr_t macsrc, macaddr_t macdst, ethertype_t type);
size_t etherEncapsulate(byte_t *packet,struct ether_header *header,byte_t *sdu,size_t sdusize);
void getSrcMAC(struct ether_header *etherHeader, macaddr_t macsrc);

// IP level functions
rawsockerr_t IP4headPopulateB(struct iphdr *IPhead, char *devname,unsigned char tos,unsigned short frag_offset, unsigned char ttl, unsigned char protocol,unsigned int flags,struct ipaddrs *addrs);
rawsockerr_t IP4headPopulateS(struct iphdr *IPhead, char *devname, struct in_addr destIP, unsigned char tos,unsigned short frag_offset, unsigned char ttl, unsigned char protocol,unsigned int flags,struct ipaddrs *addrs);
rawsockerr_t IP4headPopulate(struct iphdr *IPhead, char *devname, char *destIP, unsigned char tos,unsigned short frag_offset, unsigned char ttl, unsigned char protocol,unsigned int flags,struct ipaddrs *addrs);
void IP4headAddID(struct iphdr *IPhead, unsigned short id);
void IP4headAddTotLen(struct iphdr *IPhead, unsigned short len);
size_t IP4Encapsulate(byte_t *packet,struct iphdr *header,byte_t *sdu,size_t sdusize);

// UDP level functions
void UDPheadPopulate(struct udphdr *UDPhead, unsigned short sourceport, unsigned short destport);
size_t UDPencapsulate(byte_t *packet,struct udphdr *header,byte_t *data,size_t payloadsize,struct ipaddrs addrs);

// Receiving device functions
byte_t *UDPgetpacketpointers(byte_t *pktbuf,struct ether_header **etherHeader, struct iphdr **IPheader,struct udphdr **UDPheader);
unsigned short UDPgetpayloadsize(struct udphdr *UDPheader);
bool validateEthCsum(byte_t *packet, csum16_t csum, csum16_t *combinedcsum, csumt_t type, void *args);

// Test functions, to inject errors inside packets - should never be used under normal circumstances
void test_injectIPCsumError(byte_t *IPpacket);
void test_injectUDPCsumError(byte_t *UDPpacket);

#endif