1 #!/usr/bin/env python3
2
3 import os
4 import signal
5 from string import Template
6 import subprocess
7 import time
8 from TdcPlugin import TdcPlugin
9
10 from tdc_config import *
11
12 try:
13 from scapy.all import *
14 except ImportError:
15 print("Unable to import the scapy python module.")
16 print("\nIf not already installed, you may do so with:")
17 print("\t\tpip3 install scapy==2.4.2")
18 exit(1)
19
20 class SubPlugin(TdcPlugin):
21 def __init__(self):
22 self.sub_class = 'scapy/SubPlugin'
23 super().__init__()
24
25 def post_execute(self):
26 if 'scapy' not in self.args.caseinfo:
27 if self.args.verbose:
28 print('{}.post_execute: no scapy info in test case'.format(self.sub_class))
29 return
30
31 # Check for required fields
32 lscapyinfo = self.args.caseinfo['scapy']
33 if type(lscapyinfo) != list:
34 lscapyinfo = [ lscapyinfo, ]
35
36 for scapyinfo in lscapyinfo:
37 scapy_keys = ['iface', 'count', 'packet']
38 missing_keys = []
39 keyfail = False
40 for k in scapy_keys:
41 if k not in scapyinfo:
42 keyfail = True
43 missing_keys.append(k)
44 if keyfail:
45 print('{}: Scapy block present in the test, but is missing info:'
46 .format(self.sub_class))
47 print('{}'.format(missing_keys))
48
49 pkt = eval(scapyinfo['packet'])
50 if '$' in scapyinfo['iface']:
51 tpl = Template(scapyinfo['iface'])
52 scapyinfo['iface'] = tpl.safe_substitute(NAMES)
53 for count in range(scapyinfo['count']):
54 sendp(pkt, iface=scapyinfo['iface'])
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.