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
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Generate preferences for a "No Reassembly" profile.
# By Gerald Combs <gerald@wireshark.org>
#
# SPDX-License-Identifier: GPL-2.0-or-later
#
'''Generate preferences for a "No Reassembly" profile.'''
import argparse
import os.path
import re
import subprocess
import sys
def main():
if sys.version_info[0] < 3:
print("This requires Python 3")
sys.exit(2)
parser = argparse.ArgumentParser(description='No reassembly profile generator')
parser.add_argument('-p', '--program-path', default=os.path.curdir, help='Path to TShark.')
parser.add_argument('-v', '--verbose', action='store_const', const=True, default=False, help='Verbose output.')
args = parser.parse_args()
this_dir = os.path.dirname(__file__)
profile_path = os.path.join(this_dir, '..', 'profiles', 'No Reassembly', 'preferences')
tshark_path = os.path.join(args.program_path, 'tshark')
if not os.path.isfile(tshark_path):
print('tshark not found at {}\n'.format(tshark_path))
parser.print_usage()
sys.exit(1)
rd_pref_re = re.compile('^#\s*(.*(reassembl|desegment)\S*):\s*TRUE')
out_prefs = [
'# Generated by ' + os.path.basename(__file__), '',
'####### Protocols ########', '',
]
cp = subprocess.run([tshark_path, '-G', 'defaultprefs'], stdout=subprocess.PIPE, check=True, encoding='utf-8')
pref_lines = cp.stdout.splitlines()
for pref_line in pref_lines:
m = rd_pref_re.search(pref_line)
if m:
rd_pref = m.group(1) + ': FALSE'
if args.verbose is True:
print(rd_pref)
out_prefs.append(rd_pref)
if len(pref_lines) < 5000:
print("Too few preference lines.")
sys.exit(1)
if len(out_prefs) < 150:
print("Too few changed preferences.")
sys.exit(1)
with open(profile_path, 'w') as profile_f:
for pref_line in out_prefs:
profile_f.write(pref_line + '\n')
if __name__ == '__main__':
main()
|
