NFV/utils.h at master · aditya-nambiar/NFV · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#ifndef UTILS_H
#define UTILS_H

//(C++) Operations: Input/Output
#include <iostream>

//(C++) STL Operations: String, Vector, String stream, Unordered map
#include <string>
#include <vector>
#include <sstream>
#include <unordered_map>

//(C) Operations: Input/Output, String, Standard libraries(like atoi, malloc)
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

//(C) Operations: Socket programming
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <signal.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/select.h>

//(C) Operations: Multithreading
#include <pthread.h>

//(C) Operations: Raw socket programming with headers(IP, TCP, UDP)
#include <netinet/ip.h>
#include <netinet/tcp.h>
#include <netinet/udp.h>
#include <sys/ioctl.h>
#include <bits/ioctls.h>
#include <net/if.h>

//(C) Operations: TUN interface Programming
#include <linux/if_tun.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <errno.h>
#include <stdarg.h>

//(C) Operations: Date & Time
#include <ctime>

using namespace std;

#define BUFFER_SIZE 2048

extern int g_reuse;
extern int g_freeport;
extern int g_mme_port;
extern int g_hss_port;
extern int g_sgw1_port;
extern int g_sgw2_port;
extern int g_sgw3_port;
extern int g_pgw_port;
extern int g_public_sink_port;
extern int g_private_sink_port;
extern int g_pgw_server_for_sink_port;
extern const char *g_mme_addr;
extern const char *g_hss_addr;
extern const char *g_sgw1_addr;
extern const char *g_sgw2_addr;
extern const char *g_sgw3_addr;
extern const char *g_pgw_addr;
extern const char *g_public_sink_addr;
extern const char *g_private_sink_addr;
extern const char *g_pgw_server_for_sink_addr;
extern socklen_t g_addr_len;
extern timeval g_timeout;
extern long long g_stack_buf;

struct ClientDetails {
	int num;
	struct sockaddr_in client_sock_addr;

	ClientDetails();
	ClientDetails(const ClientDetails&);
	friend void swap(ClientDetails&, ClientDetails&);
	ClientDetails& operator=(ClientDetails);
	ClientDetails(ClientDetails&&);
	~ClientDetails();
};

void check_conn(int&);
void report_error(int);
void report_error(int, const char*);
void print(string);
void print(int);
void print(string, int);
void print_message(string);
void print_message(string, int);
void print_message(string, unsigned long long);
const char* to_char_array(unsigned long long);
char* allocate_str_mem(int);
uint8_t* allocate_uint8_mem(int);
struct ip* allocate_ip_mem(int);
struct tcphdr* allocate_tcp_mem(int);
void check_server_usage(int, char*[]);
void check_client_usage(int, char*[]);
void time_check(time_t&, double&, bool&);
void handle_exceptions();

#endif //UTILS_H