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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
| | /* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright Red Hat
* Author: David Gibson <david@gibson.dropbear.id.au>
*
* Tracking for logical "flows" of packets.
*/
#ifndef FLOW_H
#define FLOW_H
#define FLOW_TIMER_INTERVAL 1000 /* ms */
/**
* enum flow_type - Different types of packet flows we track
*/
enum flow_type {
/* Represents an invalid or unused flow */
FLOW_TYPE_NONE = 0,
/* A TCP connection between a socket and tap interface */
FLOW_TCP,
/* A TCP connection between a host socket and ns socket */
FLOW_TCP_SPLICE,
FLOW_NUM_TYPES,
};
extern const char *flow_type_str[];
#define FLOW_TYPE(f) \
((f)->type < FLOW_NUM_TYPES ? flow_type_str[(f)->type] : "?")
extern const uint8_t flow_proto[];
#define FLOW_PROTO(f) \
((f)->type < FLOW_NUM_TYPES ? flow_proto[(f)->type] : 0)
/**
* struct flowside - Common information for one side of a flow
* @eaddr: Endpoint address (remote address from passt's PoV)
* @faddr: Forwarding address (local address from passt's PoV)
* @eport: Endpoint port
* @fport: Forwarding port
* @pif: pif ID on which this side of the flow exists
*/
struct flowside {
union inany_addr faddr;
union inany_addr eaddr;
in_port_t fport;
in_port_t eport;
uint8_t pif;
};
static_assert(_Alignof(struct flowside) == _Alignof(uint32_t),
"Unexpected alignment for struct flowside");
/** flowside_from_af - Initialize flowside from addresses
* @fside: flowside to initialize
* @pif: pif if of this flowside
* @af: Address family (AF_INET or AF_INET6)
* @faddr: Forwarding address (pointer to in_addr or in6_addr)
* @fport: Forwarding port
* @eaddr: Endpoint address (pointer to in_addr or in6_addr)
* @eport: Endpoint port
*/
static inline void flowside_from_af(struct flowside *fside, uint8_t pif, int af,
const void *faddr, in_port_t fport,
const void *eaddr, in_port_t eport)
{
fside->pif = pif;
inany_from_af(&fside->faddr, af, faddr);
inany_from_af(&fside->eaddr, af, eaddr);
fside->fport = fport;
fside->eport = eport;
}
int flowside_from_sock(struct flowside *fside, uint8_t pif, int s,
const void *fsa, const void *esa);
/** flowside_complete - Check if flowside is fully initialized
* @fside: flowside to check
*/
static inline bool flowside_complete(const struct flowside *fside)
{
return fside->pif != PIF_NONE &&
!IN6_IS_ADDR_UNSPECIFIED(&fside->faddr) &&
!IN6_IS_ADDR_UNSPECIFIED(&fside->eaddr) &&
fside->fport != 0 && fside->eport != 0;
}
#define SIDES 2
/**
* flowside_eq() - Check if two flowsides are equal
* @left, @right: Flowsides to compare
*
* Return: true if equal, false otherwise
*/
static inline bool flowside_eq(const struct flowside *left,
const struct flowside *right)
{
return left->pif == right->pif &&
inany_equals(&left->eaddr, &right->eaddr) &&
left->eport == right->eport &&
inany_equals(&left->faddr, &right->faddr) &&
left->fport == right->fport;
}
/**
* struct flow_common - Common fields for packet flows
* @side[]: Information for each side of the flow
* @type: Type of packet flow
*/
struct flow_common {
struct flowside side[SIDES];
uint8_t type;
};
#define FLOW_INDEX_BITS 17 /* 128k - 1 */
#define FLOW_MAX MAX_FROM_BITS(FLOW_INDEX_BITS)
#define FLOW_TABLE_PRESSURE 30 /* % of FLOW_MAX */
#define FLOW_FILE_PRESSURE 30 /* % of c->nofile */
/** flow_complete - Check if common parts of flow are fully initialized
* @flow: flow to check
*/
static inline bool flow_complete(const struct flow_common *f)
{
return f->type != FLOW_TYPE_NONE && f->type < FLOW_NUM_TYPES &&
flowside_complete(&f->side[0]) &&
flowside_complete(&f->side[1]);
}
void flow_new_dbg(const struct flow_common *f, unsigned side);
#define FLOW_NEW_DBG(flow, side) (flow_new_dbg(&(flow)->f, (side)))
void flow_fwd_dbg(const struct flow_common *f, unsigned side);
#define FLOW_FWD_DBG(flow, side) (flow_fwd_dbg(&(flow)->f, (side)))
/**
* struct flow_sidx - ID for one side of a specific flow
* @side: Side referenced (0 or 1)
* @flow: Index of flow referenced
*/
typedef struct flow_sidx {
unsigned side :1;
unsigned flow :FLOW_INDEX_BITS;
} flow_sidx_t;
static_assert(sizeof(flow_sidx_t) <= sizeof(uint32_t),
"flow_sidx_t must fit within 32 bits");
#define FLOW_SIDX_NONE ((flow_sidx_t){ .flow = FLOW_MAX })
/**
* flow_sidx_eq() - Test if two sidx values are equal
* @a, @b: sidx values
*
* Return: true iff @a and @b refer to the same side of the same flow
*/
static inline bool flow_sidx_eq(flow_sidx_t a, flow_sidx_t b)
{
return (a.flow == b.flow) && (a.side == b.side);
}
uint64_t flow_hash_insert(const struct ctx *c, flow_sidx_t sidx);
void flow_hash_remove(const struct ctx *c, flow_sidx_t sidx);
flow_sidx_t flow_hash_lookup(const struct ctx *c, uint8_t proto, uint8_t pif,
int af, const void *eaddr, const void *faddr,
in_port_t eport, in_port_t fport);
union flow;
void flow_init(void);
void flow_defer_handler(const struct ctx *c, const struct timespec *now);
void flow_log_(const struct flow_common *f, int pri, const char *fmt, ...)
__attribute__((format(printf, 3, 4)));
#define flow_log(f_, pri, ...) flow_log_(&(f_)->f, (pri), __VA_ARGS__)
#define flow_dbg(f, ...) flow_log((f), LOG_DEBUG, __VA_ARGS__)
#define flow_err(f, ...) flow_log((f), LOG_ERR, __VA_ARGS__)
#define flow_trace(f, ...) \
do { \
if (log_trace) \
flow_dbg((f), __VA_ARGS__); \
} while (0)
void flow_init(void);
#endif /* FLOW_H */
|