netdev-dpdk: fix mbuf leaks
[cascardo/ovs.git] / utilities / ovs-pcap.in
1 #! @PYTHON@
2 #
3 # Copyright (c) 2010 Nicira, Inc.
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at:
8 #
9 #     http://www.apache.org/licenses/LICENSE-2.0
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16
17 from __future__ import print_function
18
19 import binascii
20 import getopt
21 import struct
22 import sys
23
24
25 class PcapException(Exception):
26     pass
27
28
29 class PcapReader(object):
30     def __init__(self, file_name):
31         self.file = open(file_name, "rb")
32         header = self.file.read(24)
33         if len(header) != 24:
34             raise PcapException("end of file reading pcap header")
35         magic, version, thiszone, sigfigs, snaplen, network = \
36             struct.unpack(">6I", header)
37         if magic == 0xa1b2c3d4:
38             self.header_format = ">4I"
39         elif magic == 0xd4c3b2a1:
40             self.header_format = "<4I"
41         else:
42             raise PcapException("bad magic %u reading pcap file "
43                         "(expected 0xa1b2c3d4 or 0xd4c3b2a1)" % magic)
44
45     def read(self):
46         header = self.file.read(16)
47         if len(header) == 0:
48             return None
49         elif len(header) != 16:
50             raise PcapException("end of file within pcap record header")
51
52         ts_sec, ts_usec, incl_len, orig_len = struct.unpack(self.header_format,
53                                                             header)
54         packet = self.file.read(incl_len)
55         if len(packet) != incl_len:
56             raise PcapException("end of file reading pcap packet data")
57         return packet
58 argv0 = sys.argv[0]
59
60
61 def usage():
62     print("""\
63 %(argv0)s: print pcap file packet data as hex
64 usage: %(argv0)s FILE
65 where FILE is a PCAP file.
66
67 The following options are also available:
68   -h, --help                  display this help message
69   -V, --version               display version information\
70 """ % {'argv0': argv0})
71     sys.exit(0)
72
73 if __name__ == "__main__":
74     try:
75         try:
76             options, args = getopt.gnu_getopt(sys.argv[1:], 'hV',
77                                               ['help', 'version'])
78         except getopt.GetoptException as geo:
79             sys.stderr.write("%s: %s\n" % (argv0, geo.msg))
80             sys.exit(1)
81
82         for key, value in options:
83             if key in ['-h', '--help']:
84                 usage()
85             elif key in ['-V', '--version']:
86                 print("ovs-pcap (Open vSwitch) @VERSION@")
87             else:
88                 sys.exit(0)
89
90         if len(args) != 1:
91             sys.stderr.write("%s: exactly 1 non-option argument required "
92                              "(use --help for help)\n" % argv0)
93             sys.exit(1)
94
95         reader = PcapReader(args[0])
96         while True:
97             packet = reader.read()
98             if packet is None:
99                 break
100
101             print(binascii.hexlify(packet))
102
103     except PcapException as e:
104         sys.stderr.write("%s: %s\n" % (argv0, e))
105         sys.exit(1)
106
107 # Local variables:
108 # mode: python
109 # End: