source: libpacketdump/libpacketdump.cc @ f01e6a8

4.0.1-hotfixescachetimestampsdevelopdpdk-ndagetsilivegetfragoffhelplibtrace4ndag_formatpfringrc-4.0.1rc-4.0.2rc-4.0.3rc-4.0.4ringdecrementfixringperformanceringtimestampfixes
Last change on this file since f01e6a8 was f01e6a8, checked in by Perry Lorier <perry@…>, 12 years ago

Be smarter about where you search for decoders.

  • Property mode set to 100644
File size: 6.3 KB
Line 
1#include <libtrace.h>
2#include <err.h>
3#include <time.h>
4#include "libpacketdump.h"
5#include "config.h"
6#include <stdio.h>
7#include <netdb.h>
8#include <stdlib.h>
9#include <getopt.h>
10
11#include <sys/socket.h>
12
13#ifdef HAVE_NETINET_ETHER
14#  include <netinet/ether.h>
15#endif
16
17
18#ifdef HAVE_INTTYPES_H
19#  include <inttypes.h>
20#else
21#  error "Can't find inttypes.h"
22#endif
23
24#ifdef HAVE_LIMITS_H
25#  include <limits.h>
26#endif
27
28#ifdef HAVE_SYS_LIMITS_H
29#  include <sys/limits.h>
30#endif
31
32#ifdef HAVE_SYS_TYPES_H
33#  include <sys/types.h>
34#endif
35#include <net/if.h>
36#include <netinet/in.h>
37#include <stdio.h>
38
39#include <net/if_arp.h>
40#ifdef HAVE_NETINET_IF_ETHER_H
41#  include <netinet/if_ether.h>
42#endif
43#include <dlfcn.h>
44#include <map>
45#include <string>
46#include <ctype.h>
47#include "libpacketdump.h"
48extern "C"{
49#include "grammar.h"
50}
51
52enum decode_style_t {
53    DECODE_NORMAL,
54    DECODE_PARSER
55};
56
57typedef void (*decode_norm_t)(uint16_t type,const char *packet,int len);
58typedef void (*decode_parser_t)(uint16_t type,const char *packet,int len, element_t* el);
59
60typedef union decode_funcs {
61    decode_norm_t decode_n;
62    decode_parser_t decode_p;
63} decode_funcs_t;
64
65typedef struct decoder {
66    enum decode_style_t style;
67    decode_funcs_t *func;
68    element_t *el; // make a union of structs with all args in it for all funcs?
69} decode_t;
70
71
72static std::map<std::string,std::map<uint16_t,decode_t> > decoders;
73
74#define WIDTH 16
75
76#ifndef DIRNAME
77#define DIRNAME "./"
78#warning "No DIRNAME set!"
79#endif
80
81void trace_dump_packet(struct libtrace_packet_t *packet)
82{
83        time_t sec = (time_t)trace_get_seconds(packet);
84        libtrace_linktype_t linktype;
85        uint32_t length;
86        const char *link=(char *)trace_get_packet_buffer(packet,&linktype,&length);
87
88        printf("\n%s",ctime(&sec));
89        printf(" Capture: Packet Length: %i/%i Direction Value: %i\n",
90                        (int)length,
91                        (int)trace_get_wire_length(packet),
92                        (int)trace_get_direction(packet));
93        if (!link) 
94                printf(" [No link layer available]\n");
95        else
96                decode_next(link,length, "link",
97                        linktype);
98}
99
100static void generic_decode(uint16_t type,const char *packet, int len) {
101        int i;
102        printf(" Unknown Protocol: %i",type);
103        for(i=0;i<len; /* Nothing */ ) {
104                int j;
105                printf("\n ");
106                for(j=0;j<WIDTH;j++) {
107                        if (i+j<len)
108                                printf(" %02x",(unsigned char)packet[i+j]);
109                        else
110                                printf("   ");
111                }
112                printf("    ");
113                for(j=0;j<WIDTH;j++) {
114                        if (i+j<len)
115                                if (isprint((unsigned char)packet[i+j]))
116                                        printf("%c",(unsigned char)packet[i+j]);
117                                else
118                                        printf(".");
119                        else
120                                printf("   ");
121                }
122                if (i+WIDTH>len)
123                        break;
124                else
125                        i+=WIDTH;
126        }
127        printf("\n");
128}
129
130static void *open_so_decoder(const char *name,int type)
131{
132        char path[1024];
133        void *hdl;
134        /* Only check LIBPKTDUMPDIR if we're not setuid.  Not bulletproof, but hopefully anyone who
135         * sets uid == euid will also clear the environment (eg sudo).
136         */
137        if (getuid() == geteuid() && getenv("LIBPKTDUMPDIR")) {
138                snprintf(path,sizeof(path),"%s/%s_%i.so",getenv("LIBPKTDUMPDIR"),name,type);
139                hdl = dlopen(path,RTLD_LAZY);
140                if (hdl)
141                        return hdl;
142        }
143        /* If the variable isn't set, *or* if we don't find anything, try the system location. */
144        snprintf(path,sizeof(path),DIRNAME "/%s_%i.so",name,type);
145        hdl = dlopen(path,RTLD_LAZY);
146        if (hdl)
147                return hdl;
148
149        return hdl;
150}
151
152static void *open_protocol_decoder(const char *name, int type)
153{
154        char path[1024];
155        void *hdl;
156        /* Only check LIBPKTDUMPDIR if we're not setuid.  Not bulletproof, but hopefully anyone who
157         * sets uid == euid will also clear the environment (eg sudo).
158         */
159        if (getuid() == geteuid() && getenv("LIBPKTDUMPDIR")) {
160                snprintf(path,sizeof(path),"%s/%s_%i.protocol",getenv("LIBPKTDUMPDIR"),name,type);
161                hdl = parse_protocol_file(path);
162                if (hdl)
163                        return hdl;
164        }
165        /* Try the system directory */
166        snprintf(path,sizeof(path),DIRNAME "/%s_%i.protocol",
167                name,type);
168        hdl = parse_protocol_file(path);
169
170        if (!hdl)
171                return hdl;
172
173        return hdl;
174}
175
176void decode_next(const char *packet,int len,const char *proto_name,int type)
177{
178        std::string sname(proto_name);
179
180        // if we haven't worked out how to decode this type yet, load the
181        // appropriate files to do so
182        if (decoders[sname].find(type)==decoders[sname].end()) {
183                void *hdl;
184                decode_funcs_t *func = new decode_funcs_t;
185                decode_t dec;
186
187                /* Try and find a .so to handle this protocol */
188                hdl = open_so_decoder(sname.c_str(),type);
189                if (hdl) {
190                        void *s=dlsym(hdl,"decode");
191                        if (s) {
192                                // use the shared library
193                                func->decode_n = (decode_norm_t)s;
194                                dec.style = DECODE_NORMAL;
195                                dec.el = NULL; 
196                        }
197                        else {
198                                dlclose(hdl);
199                                hdl = NULL;
200                        }
201                }
202
203                /* We didn't successfully open the .so, try finding a .protocol that we can use */
204                if (!hdl) {
205                        hdl = open_protocol_decoder(sname.c_str(),type);
206                        if (hdl) {
207                                // use the protocol file
208                                func->decode_p = decode_protocol_file;
209                                dec.style = DECODE_PARSER;
210                                dec.el = (element_t*)hdl;
211                        }
212                }
213
214                /* No matches found, fall back to the generic decoder. */
215                /* TODO: We should have a variety of fallback decoders based on the protocol. */
216                if(!hdl)
217                {
218                        // no protocol file either, use a generic one
219                        func->decode_n = generic_decode;
220                        dec.style = DECODE_NORMAL;
221                        dec.el = NULL;
222                } 
223
224                dec.func = func;
225                decoders[sname][type] = dec;
226        }
227
228        /* TODO: Instead of haxing this here, we should provide a series of generic_decode's
229         * and let the code above deal with it.
230         */
231        if (decoders[sname][type].func->decode_n == generic_decode) {
232                /* We can't decode a link, so lets skip that and see if libtrace
233                 * knows how to find us the ip header
234                 */
235
236                /* Also, don't try to skip if the linktype is not valid,
237                 * because libtrace will just assert fail and that's never
238                 * good */
239                if (sname=="link" && type != -1) {
240                        uint16_t newtype;
241                        uint32_t newlen=len;
242                        const char *network=(const char*)trace_get_payload_from_link((void*)packet,
243                                        (libtrace_linktype_t)type,
244                                        &newtype,&newlen);
245                        if (network) {
246                                printf("skipping unknown link header of type %i to network type %i\n",type,newtype);
247                                /* Should hex dump this too. */
248                                decode_next(network,newlen,"eth",newtype);
249                                return;
250                        }
251                }
252                else {
253                        printf("unknown protocol %s/%i\n",sname.c_str(),type);
254                }
255        }
256
257        // decode using the appropriate function
258        switch(decoders[sname][type].style)
259        {
260                case DECODE_NORMAL:
261                        decoders[sname][type].func->decode_n(type,packet,len);
262                        break;
263
264                case DECODE_PARSER:
265                        decoders[sname][type].func->decode_p(type,packet,len,
266                                decoders[sname][type].el);
267                        break;
268
269        };
270}
Note: See TracBrowser for help on using the repository browser.