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
189
190
191
192
193
194
195
196
197
198
199
200
|
#ifndef _STREAM_MANAGE_H_
#define _STREAM_MANAGE_H_
#include "stream_internal.h"
#include "stream_register.h"
#include "timeout.h"
#include "token_bucket.h"
#ifdef __cplusplus
extern "C" {
#endif
extern time_t g_CurrentTime;
extern int tcp_default_unorder;
extern unsigned short link_default_nopkt_time;
extern int tcp_creatlink_model;
extern int tcp_flood_detect_model;
extern int tcp_dataflood_pktlen;
extern int tcp_support_all;
#ifdef __cplusplus
}
#endif
#define TCP_CTEAT_LINK_BYSYN 0x01
#define TCP_CTEAT_LINK_BYDATA 0x02
#define TCP_FLOOD_DETECT_OFF 0x0
#define TCP_SYNFLOOD_DETECT_ON 0x01
#define TCP_DATAFLOOD_DETECT_ON 0x02
#define DROP 0x00
#define PASS 0x01
#define RETAIN 0x02
#define SMART_OFFLOAD_FLAG_UNKNOWN 0x0
#define SMART_OFFLOAD_FLAG_UDP 0x01
#define SMART_OFFLOAD_FLAG_TCP 0x01
#define SMART_OFFLOAD_FLAG_TCPALL 0x02
#define UNORDER_MAX_NUM (tcp_default_unorder)
//�ײ����ӹ�����״̬����
#define TCP_SYN_STATE 0
#define TCP_DATA_STATE 1
#define TCP_NOUSE_STATE 2
#define MAX_TCP_STATE 3
#define UDP_ONE_STATE 0
#define UDP_TWO_STATE 1
#define UDP_MORE_STATE 2
#define MAX_UDP_STATE 3
#define STREAM_LINK_REUSE_BYSYN 0
#define STREAM_LINK_JUST_EST 1
#define STREAM_LINK_DATA 2
#define STREAM_LINK_CLOSE 3
#define STREAM_LINK_RESET 4
#define STREAM_LINK_TIMEOUT 5
#define STREAM_LINK_LRU_OUT 6
#define STREAM_FAKE_CLOSE_BY_DEPRIVE 7
#define STREAM_CLOSE_BY_DUMPFILE_END 8 /* 2021-05-20 lijia add, for pcap dumpfile mode */
#define SINGLE_ACK_RECEVED_NO 0
#define SINGLE_ACK_RECEVED 1
struct stream_status{ /* 2014-03-31 lijia add */
unsigned int tcp_num;
unsigned int syn_cnt;
unsigned int ack_cnt;
};
struct streamindex
{
struct timeout timeout;
struct streamindex *phashprev;
struct streamindex *phashnext;
struct streaminfo_private stream;
};
struct stream_list
{
int cnt;
int max_cnt;
struct timeouts *streamindex_timer;
long last_update_timer_ms;
long interval_to_next_timeout_ms;
timeout_error_t timer_error;
struct token_bucket *timeout_ratelimiter;
};
struct stream_index_list_item
{
struct streamindex index;
STAILQ_ENTRY(stream_index_list_item) entries;
};
struct global_stream
{
struct streamindex **tcp_stream_table;
unsigned short *tcp_stream_talbe_hash_count;
struct stream_list tcpList[MAX_TCP_STATE];
struct streamindex **udp_stream_table;
unsigned short *udp_stream_talbe_hash_count;
struct stream_list udpList[MAX_UDP_STATE];
STAILQ_HEAD(STREAM_INDEX_LIST,stream_index_list_item) freeList;
struct stream_index_list_item *__freeList_real_head;
int freeList_cnt;
int freeList_max_cnt;
struct timeouts *user_define_timer;
long user_define_timer_cnt;
long interval_to_next_timeout_ms;
timeout_error_t timer_error;
struct token_bucket *tcp_timeout_ratelimiter;
struct token_bucket *udp_timeout_ratelimiter;
struct token_bucket *udp_opening_ratelimiter;
struct token_bucket *tcp_opening_ratelimiter;
};
typedef struct _stStreamFunInfo
{
struct timeout timeout;
const struct streaminfo *pstream;
int set_timer_s;
int entry_id;
char pad[7];
char appState;
enum fun_type_t funtype;/* 2014-12-31 lqy add */
char (*pfun)(void);
void *pAppInfo;
struct _stStreamFunInfo *next;
}StreamFunInfo;
#ifdef __cplusplus
extern "C" {
#endif
extern struct global_stream **G_MESA_GLOBAL_STREAM;
//extern struct thread_node_new *threads_pool_new;
struct streaminfo *findstream(struct streaminfo *ptmp);
void streamleavlist(struct streamindex *pindex,struct stream_list *plist);
void streamaddlist(struct streamindex *pindex,struct stream_list *plist);
struct streamindex *findstreamindex(struct streamindex *pindex, const raw_pkt_t *raw_pkt);
int lrustream(struct streamindex *pindex);
void hash_add_stream(struct streamindex *pindex);
void hash_del_stream(struct streamindex *pindex);
int stream_process_tcp(struct streaminfo *a_tcp, const void *, const void *,const raw_pkt_t * raw_pkt,void **apme,unsigned char *popstate);
int stream_process_tcp_allpkt(struct streaminfo *a_tcp,const void *, const void *,const raw_pkt_t * raw_pkt,void **apme,unsigned char *popstate);
//int stream_process_tcp_takeover(struct streaminfo *a_tcp,const raw_pkt_t * raw_pkt);
int stream_process_tcp_takeover(struct streaminfo *a_tcp,const void *iphdr);
int stream_process_udp(struct streaminfo *a_udp,const void *, const void *, const raw_pkt_t * raw_pkt,void **apme,unsigned char *popstate);
int stream_process_ipv4(struct streaminfo *pfstream,const struct ip *a_packet,int thread_num,unsigned char routedir,const raw_pkt_t *raw_pkt);
int stream_process_ipv6(struct streaminfo *pfstream,const struct ip6_hdr *a_packet,int thread_num,unsigned char routedir,const raw_pkt_t *raw_pkt);
int stream_process_ipv4_frag(struct streaminfo *pfstream,const struct ip *a_packet,int thread_num,unsigned char routedir,const raw_pkt_t *raw_pkt);
int stream_process_ipv6_frag(struct streaminfo *pfstream,const struct ip6_hdr *a_packet,int thread_num,unsigned char routedir,const raw_pkt_t *raw_pkt);
int stream_process_polling(int thread_seq);
int stream_process_arp(struct streaminfo *pfstream, const struct mesa_arp_hdr * this_layer_hdr, int thread_num,unsigned char routedir, const raw_pkt_t *raw_pkt);
//void freeall_stream(int threadcount);
void free_thread_stream(int thread_seq);
int dealipv4tcppkt(struct streamindex *pfstream,const struct mesa_ip4_hdr * a_packet,int thread_num,unsigned char routedir, const raw_pkt_t *raw_pkt,int);
struct streamindex *malloc_and_copy_streamindex(int threadnum, struct streamindex *pindex_stack);
void free_streamindex(int threadnum,struct streamindex *pindex);
int is_proxy_stream(const struct streaminfo *pstream);
int mkaddrhash(char *data,int datalen);
void wy_init_hash();
void udp_free_stream(struct streamindex *pstream);
int tcp_free_stream(struct streamindex *pstream, const void*, const void *,const raw_pkt_t *raw_pkt);
void update_opposite_addr_info(struct streaminfo_private *top_stream_pr, struct streaminfo_private *pstream_pr,
struct streaminfo_private *p_stack, unsigned char cur_dir);
int polling_stream_timeout(int tid);
int checkstreamorder(const struct streaminfo_private *top_stream_pr, struct streaminfo_private *pheap_stream_pr,
struct streaminfo_private *pstack_stream_pr, int *heap_streaminfo_skip_layer_num, int *stack_streaminfo_skip_layer_num);
struct streaminfo_private *copy_stream_info_to_heap_single_layer(int mem_used_type, const struct streaminfo_private *stack_stream_pr, int reverse);
unsigned int sapp_jhash_4words(unsigned int a, unsigned int b, unsigned int c, unsigned int initval, unsigned int *out_val);
void stream_get_scratch_frag_stat(struct streaminfo *pstream, int *frag_cnt, int *frag_len);
#ifdef __cplusplus
}
#endif
#endif
|
