forked from vjanelle/nprobe
-
Notifications
You must be signed in to change notification settings - Fork 0
/
engine.h
132 lines (121 loc) · 5.77 KB
/
engine.h
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
/*
* nProbe - a Netflow v5/v9/IPFIX probe for IPv4/v6
*
* Copyright (C) 2002-2010 Luca Deri <[email protected]>
*
* http://www.ntop.org/
*
* 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.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
/* ********************************** */
#define PCAP_LONG_SNAPLEN 1600
#define PCAP_DEFAULT_SNAPLEN 128
#define MAX_NUM_PLUGINS 24
#define CREATE_FLOW_CALLBACK 1
#define DELETE_FLOW_CALLBACK 2
#define PACKET_CALLBACK 3
extern V9V10TemplateElementId* getPluginTemplate(char* template_name);
extern int checkPluginExport(V9V10TemplateElementId *theTemplate, FlowDirection direction,
FlowHashBucket *theFlow, char *outBuffer,
uint* outBufferBegin, uint* outBufferMax);
extern int checkPluginPrint(V9V10TemplateElementId *theTemplate, FlowDirection direction,
FlowHashBucket *bkt, char *line_buffer, uint line_buffer_len);
typedef V9V10TemplateElementId* (*PluginConf)(void);
extern void discardBucket(FlowHashBucket *myBucket);
extern void* dequeueBucketToExport(void*);
typedef void (*PluginInitFctn)(int argc, char *argv[]);
typedef void (*PluginTermFctn)(void);
typedef void (*PluginFctn)(FlowHashBucket*, void*);
typedef void (*PluginPacketFctn)(u_char new_bucket, void *pluginData,
FlowHashBucket *bkt,
FlowDirection flow_direction,
u_short proto, u_char isFragment,
u_short numPkts, u_char tos,
u_short vlanId, struct eth_header *ehdr,
IpAddress *src, u_short sport,
IpAddress *dst, u_short dport,
u_int len, u_int8_t flags, u_int32_t tcpSeqNum,
u_int8_t icmpType, u_short numMplsLabels,
u_char mplsLabels[MAX_NUM_MPLS_LABELS][MPLS_LABEL_LEN],
const struct pcap_pkthdr *h, const u_char *p,
u_char *payload, int payloadLen);
typedef V9V10TemplateElementId* (*PluginGetPluginTemplateFctn)(char* template_name);
typedef int (*PluginCheckPluginExportFctn)(void*, V9V10TemplateElementId *theTemplate, FlowDirection direction,
FlowHashBucket *theFlow, char *outBuffer,
uint* outBufferBegin, uint* outBufferMax);
typedef int (*PluginCheckPluginPrintFctn)(void*, V9V10TemplateElementId *theTemplate, FlowDirection direction,
FlowHashBucket *theFlow, char *line_buffer, uint line_buffer_len);
typedef void (*PluginSetupFctn)(void);
typedef void (*PluginHelpFctn)(void);
typedef struct pluginInfo {
char *nprobe_revision, *name, *version, *descr, *author;
u_char always_enabled, enabled;
PluginInitFctn initFctn;
PluginTermFctn termFctn;
PluginConf pluginFlowConf;
PluginFctn deleteFlowFctn;
u_char call_packetFlowFctn_for_each_packet;
PluginPacketFctn packetFlowFctn;
PluginGetPluginTemplateFctn getPluginTemplateFctn;
PluginCheckPluginExportFctn checkPluginExportFctn;
PluginCheckPluginPrintFctn checkPluginPrintFctn;
PluginSetupFctn setupFctn;
PluginHelpFctn helpFctn;
} PluginInfo;
extern PluginInfo* PluginEntryFctn(void);
/* ********************************** */
extern char* _intoa(IpAddress addr, char* buf, u_short bufLen);
extern char* _intoaV4(unsigned int addr, char* buf, u_short bufLen);
extern char* formatTraffic(float numBits, int bits, char *buf);
extern char* formatPackets(float numPkts, char *buf);
extern u_char ttlPredictor(u_char x);
extern char* proto2name(u_int8_t proto);
extern void load_mappings(void);
extern void unload_mappings(void);
extern void setPayload(FlowHashBucket *bkt, const struct pcap_pkthdr *h,
u_char *payload, int payloadLen, FlowDirection direction);
extern void updateApplLatency(u_short proto, FlowHashBucket *bkt,
FlowDirection direction, struct timeval *stamp,
u_int8_t icmpType, u_int8_t icmpCode);
extern void updateTcpFlags(FlowHashBucket *bkt, FlowDirection direction,
struct timeval *stamp, u_int8_t flags);
extern int cmpIpAddress(IpAddress *src, IpAddress *dst);
extern void printICMPflags(u_int32_t flags, char *icmpBuf, int icmpBufLen);
extern void printFlow(FlowHashBucket *theFlow, FlowDirection direction);
extern int isFlowExpired(FlowHashBucket *myBucket, time_t theTime);
extern int isFlowExpiredSinceTooLong(FlowHashBucket *myBucket, time_t theTime);
extern void printBucket(FlowHashBucket *myBucket);
extern void walkHash(u_int32_t hash_idx, int flushHash);
extern void purgeBucket(FlowHashBucket *myBucket);
/* nprobe.c or nprobe_mod.c */
extern void queueBucketToExport(FlowHashBucket *myBucket);
/* plugin.c */
extern u_short num_plugins_enabled;
extern void initPlugins(int argc, char* argv[]);
extern void termPlugins(void);
extern void pluginCallback(u_char callbackType, FlowHashBucket* bucket,
FlowDirection direction,
u_short proto, u_char isFragment,
u_short numPkts, u_char tos,
u_short vlanId, struct eth_header *ehdr,
IpAddress *src, u_short sport,
IpAddress *dst, u_short dport,
u_int len, u_int8_t flags, u_int32_t tcpSeqNum,
u_int8_t icmpType, u_short numMplsLabels,
u_char mplsLabels[MAX_NUM_MPLS_LABELS][MPLS_LABEL_LEN],
const struct pcap_pkthdr *h, const u_char *p,
u_char *payload, int payloadLen);
extern void buildActivePluginsList(V9V10TemplateElementId *template_element_list[]);
extern void printMetadata(FILE *file);