68 lines
2.2 KiB
C
68 lines
2.2 KiB
C
#ifndef _FOUR_TUPLE_H
|
|
#define _FOUR_TUPLE_H
|
|
|
|
#ifdef __cpluscplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
|
|
#include <netinet/in.h>
|
|
|
|
enum addr_type
|
|
{
|
|
ADDR_TYPE_V4,
|
|
ADDR_TYPE_V6,
|
|
};
|
|
|
|
struct addr_v4
|
|
{
|
|
struct in_addr src_addr; /* network order */
|
|
struct in_addr dst_addr; /* network order */
|
|
};
|
|
|
|
struct addr_v6
|
|
{
|
|
struct in6_addr src_addr; /* network order */
|
|
struct in6_addr dst_addr; /* network order */
|
|
};
|
|
|
|
struct four_tuple
|
|
{
|
|
enum addr_type addr_type;
|
|
in_port_t src_port; /* network order */
|
|
in_port_t dst_port; /* network order */
|
|
union
|
|
{
|
|
struct addr_v4 addr_v4;
|
|
struct addr_v6 addr_v6;
|
|
};
|
|
};
|
|
|
|
#define INIT_ADDR_V4(name, src_addr_str, src_port_num, dst_addr_str, dst_port_num) \
|
|
struct four_tuple name; \
|
|
memset(&name, 0, sizeof(name)); \
|
|
(name).addr_type = ADDR_TYPE_V4; \
|
|
(name).src_port = htons((src_port_num)); \
|
|
(name).dst_port = htons((dst_port_num)); \
|
|
inet_pton(AF_INET, (src_addr_str), &(name).addr_v4.src_addr); \
|
|
inet_pton(AF_INET, (dst_addr_str), &(name).addr_v4.dst_addr);
|
|
|
|
#define INIT_ADDR_V6(name, src_addr_str, src_port_num, dst_addr_str, dst_port_num) \
|
|
struct four_tuple name; \
|
|
memset(&name, 0, sizeof(name)); \
|
|
(name).addr_type = ADDR_TYPE_V6; \
|
|
(name).src_port = htons((src_port_num)); \
|
|
(name).dst_port = htons((dst_port_num)); \
|
|
inet_pton(AF_INET6, (src_addr_str), &(name).addr_v6.src_addr); \
|
|
inet_pton(AF_INET6, (dst_addr_str), &(name).addr_v6.dst_addr);
|
|
|
|
void four_tuple_copy(struct four_tuple *dst, struct four_tuple *src);
|
|
char *four_tuple_to_string(const struct four_tuple *addr);
|
|
void four_tuple_reverse(const struct four_tuple *orin, struct four_tuple *out);
|
|
|
|
#ifdef __cpluscplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|