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
|
/*
pmacct (Promiscuous mode IP Accounting package)
pmacct is Copyright (C) 2003-2014 by Paolo Lucente
*/
/*
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#include <sys/poll.h>
/* defines */
#define NUM_MEMORY_POOLS 16
#define MEMORY_POOL_SIZE 8192
#define MAX_HOSTS 32771
#define MAX_QUERIES 4096
/* Structures */
struct acc {
struct pkt_primitives primitives;
pm_counter_t bytes_counter;
pm_counter_t packet_counter;
pm_counter_t flow_counter;
u_int8_t flow_type;
u_int32_t tcp_flags;
unsigned int signature;
u_int8_t reset_flag;
struct timeval rstamp; /* classifiers: reset timestamp */
struct cache_bgp_primitives *cbgp;
struct pkt_nat_primitives *pnat;
struct pkt_mpls_primitives *pmpls;
char *pcust;
struct pkt_vlen_hdr_primitives *pvlen;
struct acc *next;
};
struct bucket_desc {
unsigned int num;
unsigned short int howmany;
};
struct memory_pool_desc {
int id;
unsigned char *base_ptr;
unsigned char *ptr;
int space_left;
int len;
struct memory_pool_desc *next;
};
struct query_header {
int type; /* type of query */
pm_cfgreg_t what_to_count; /* aggregation */
pm_cfgreg_t what_to_count_2; /* aggregation */
unsigned int num; /* number of queries */
unsigned int ip_sz; /* IP addresses size (in bytes) */
unsigned int cnt_sz; /* counters size (in bytes) */
struct extra_primitives extras; /* offsets for non-standard aggregation primitives structures */
int datasize; /* total length of aggregation primitives structures */
char passwd[12]; /* OBSOLETED: password */
};
struct query_entry {
pm_cfgreg_t what_to_count; /* aggregation */
pm_cfgreg_t what_to_count_2; /* aggregation */
struct pkt_primitives data; /* actual data */
struct pkt_bgp_primitives pbgp; /* extended BGP data */
struct pkt_nat_primitives pnat; /* extended NAT + timestamp data */
struct pkt_mpls_primitives pmpls; /* extended MPLS data */
char *pcust; /* custom-defined data */
struct pkt_vlen_hdr_primitives *pvlen;/* variable-length data */
};
struct reply_buffer {
unsigned char buf[LARGEBUFLEN];
unsigned char *ptr;
int len;
int packed;
};
struct stripped_class {
pm_class_t id;
char protocol[MAX_PROTOCOL_LEN];
};
struct stripped_pkt_len_distrib {
char str[MAX_PKT_LEN_DISTRIB_LEN];
};
struct imt_custom_primitive_entry {
/* compiled from map */
u_char name[MAX_CUSTOM_PRIMITIVE_NAMELEN];
u_int16_t field_type;
u_int16_t len;
u_int8_t semantics;
/* compiled internally */
u_int16_t off;
pm_cfgreg_t type;
};
struct imt_custom_primitives {
struct imt_custom_primitive_entry primitive[MAX_CUSTOM_PRIMITIVES];
int len;
int num;
};
/* prototypes */
#if (!defined __ACCT_C)
#define EXT extern
#else
#define EXT
#endif
EXT void insert_accounting_structure(struct primitives_ptrs *);
EXT struct acc *search_accounting_structure(struct primitives_ptrs *);
EXT int compare_accounting_structure(struct acc *, struct primitives_ptrs *);
#undef EXT
#if (!defined __MEMORY_C)
#define EXT extern
#else
#define EXT
#endif
EXT void init_memory_pool_table();
EXT void clear_memory_pool_table();
EXT struct memory_pool_desc *request_memory_pool(int);
#undef EXT
#if (!defined __SERVER_C)
#define EXT extern
#else
#define EXT
#endif
EXT void set_reset_flag(struct acc *);
EXT void reset_counters(struct acc *);
EXT int build_query_server(char *);
EXT void process_query_data(int, unsigned char *, int, struct extra_primitives *, int, int);
EXT void mask_elem(struct pkt_primitives *, struct pkt_bgp_primitives *, struct pkt_nat_primitives *,
struct pkt_mpls_primitives *, struct acc *, u_int64_t, u_int64_t,
struct extra_primitives *);
EXT void enQueue_elem(int, struct reply_buffer *, void *, int, int);
EXT void Accumulate_Counters(struct pkt_data *, struct acc *);
EXT int test_zero_elem(struct acc *);
#undef EXT
#if (!defined __IMT_PLUGIN_C)
#define EXT extern
#else
#define EXT
#endif
EXT void sum_host_insert(struct primitives_ptrs *);
EXT void sum_port_insert(struct primitives_ptrs *);
EXT void sum_as_insert(struct primitives_ptrs *);
#if defined HAVE_L2
EXT void sum_mac_insert(struct primitives_ptrs *);
#endif
EXT void exit_now(int);
EXT void free_extra_allocs();
#undef EXT
/* global vars */
#if (!defined __IMT_PLUGIN_C && !defined __PMACCT_CLIENT_C)
#define EXT extern
#else
#define EXT
#endif
EXT void (*insert_func)(struct primitives_ptrs *); /* pointer to INSERT function */
EXT unsigned char *mpd; /* memory pool descriptors table */
EXT unsigned char *a; /* accounting in-memory table */
EXT struct memory_pool_desc *current_pool; /* pointer to currently used memory pool */
EXT struct acc **lru_elem_ptr; /* pointer to Last Recently Used (lru) element in a bucket */
EXT int no_more_space;
EXT struct timeval cycle_stamp; /* timestamp for the current cycle */
EXT struct timeval table_reset_stamp; /* global table reset timestamp */
#undef EXT
|
