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
110
111
112
113
114
115
|
#ifndef UMP_APP_H
#define UMP_APP_H
/* Application layer data manipulating functions */
#include "ump-common.h"
#include "ump-pkt.h"
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
/* application layer data structure:
* offset 0, length 1 : IPv4 / IPv6
* offset 1, length 2 : source port
* offset 3, length 2 : destination port
* offset 5, length ? : src IP addr -> dest IP addr -> data */
#define UMP_APP(x) ((UmpApp*)(x))
#define UMP_APP_SIZE (UMP_PKT_SIZE - (((1 << 5) - 1) * 4))
typedef struct UmpAppStruct {
uint8_t data[UMP_APP_SIZE];
size_t real_data_len;
} UmpApp;
static inline uint8_t ump_app_get_is_ipv6 (UmpApp* app_data) {
uint8_t* ipv6 = UMP_COMMON_MOVE_POINTER (app_data, 0);
if (*ipv6) {
return true;
} else {
return false;
}
}
static inline void ump_app_set_is_ipv6 (UmpApp* app_data, bool is_ipv6) {
uint8_t* ipv6 = UMP_COMMON_MOVE_POINTER (app_data, 0);
if (is_ipv6) {
*ipv6 = 1;
} else {
*ipv6 = 0;
}
}
UMP_COMMON_DEFINE_BOTH (UmpApp, ump_app, src_port, 1, uint16_t);
UMP_COMMON_DEFINE_BOTH (UmpApp, ump_app, dest_port, 3, uint16_t);
static inline void* ump_app_get_src_ip_addr (UmpApp* app_data) {
return UMP_COMMON_MOVE_POINTER (app_data, 5);
}
static inline void ump_app_set_src_ip_addr (UmpApp* app_data, void* data) {
if (ump_app_get_is_ipv6 (app_data)) {
memcpy (ump_app_get_src_ip_addr (app_data), data,
UMP_COMMON_IPV6_ADDRLEN);
} else {
memcpy (ump_app_get_src_ip_addr (app_data), data,
UMP_COMMON_IPV4_ADDRLEN);
}
}
static inline void* ump_app_get_dest_ip_addr (UmpApp* app_data) {
if (ump_app_get_is_ipv6 (app_data)) {
return UMP_COMMON_MOVE_POINTER (app_data, 5 + UMP_COMMON_IPV6_ADDRLEN);
} else {
return UMP_COMMON_MOVE_POINTER (app_data, 5 + UMP_COMMON_IPV4_ADDRLEN);
}
}
static inline void ump_app_set_dest_ip_addr (UmpApp* app_data, void* data) {
if (ump_app_get_is_ipv6 (app_data)) {
memcpy (ump_app_get_dest_ip_addr (app_data), data,
UMP_COMMON_IPV6_ADDRLEN);
} else {
memcpy (ump_app_get_dest_ip_addr (app_data), data,
UMP_COMMON_IPV4_ADDRLEN);
}
}
static inline void* ump_app_get_real_data (UmpApp* app_data) {
if (ump_app_get_is_ipv6 (app_data)) {
return UMP_COMMON_MOVE_POINTER (
app_data, 5 + UMP_COMMON_IPV6_ADDRLEN * 2);
} else {
return UMP_COMMON_MOVE_POINTER (
app_data, 5 + UMP_COMMON_IPV4_ADDRLEN * 2);
}
}
static inline size_t ump_app_get_real_data_len (UmpApp* app_data) {
return app_data->real_data_len;
}
static inline size_t ump_app_get_real_data_max (UmpApp* app_data) {
return (uint8_t*)ump_app_get_real_data (app_data) - (uint8_t*)app_data;
}
static inline void ump_app_set_real_data (
UmpApp* app_data, const void* data, size_t len) {
uint8_t* real_data = ump_app_get_real_data (app_data);
memcpy (real_data, data, len);
app_data->real_data_len = len;
}
static inline bool ump_app_check_size (UmpApp* app_data, size_t size) {
if (ump_app_get_is_ipv6 (app_data)) {
return (5 + UMP_COMMON_IPV6_ADDRLEN * 2 >= size);
} else {
return (5 + UMP_COMMON_IPV4_ADDRLEN * 2 >= size);
}
}
int ump_app_fill_src_with_sockaddr (
UmpApp* app_data, const struct sockaddr* addr);
void ump_app_fill_sockaddr_with_dest (
UmpApp* app_data, struct sockaddr* addr, socklen_t* addr_len);
#endif /* UMP_TCP_H */
|
