diff options
Diffstat (limited to 'scripts/automation')
4 files changed, 150 insertions, 262 deletions
diff --git a/scripts/automation/regression/functional_tests/scapy_server_test.py_disabled b/scripts/automation/regression/functional_tests/scapy_server_test.py_disabled deleted file mode 100755 index 5699438d..00000000 --- a/scripts/automation/regression/functional_tests/scapy_server_test.py_disabled +++ /dev/null @@ -1,222 +0,0 @@ -# scapy server unit test
-
-import sys,os
-from nose.plugins.skip import SkipTest
-if sys.version_info.major == 3:
- raise SkipTest("Python3 currently not supported")
-scapy_server_path = os.path.abspath(os.path.join(os.pardir, 'trex_control_plane', 'stl', 'services','scapy_server'))
-stl_pathname = os.path.abspath(os.path.join(os.pardir, os.pardir, 'trex_control_plane','stl'))
-sys.path.append(scapy_server_path)
-sys.path.append(stl_pathname)
-additional_stl_udp_pkts = os.path.abspath(os.path.join(os.pardir, os.pardir,'stl'))
-sys.path.append(additional_stl_udp_pkts)
-
-import trex_stl_lib
-from trex_stl_lib.api import *
-from copy import deepcopy
-
-import tempfile
-import hashlib
-from platform_cmd_link import *
-import functional_general_test
-from nose.tools import assert_equal
-from nose.tools import assert_not_equal
-from nose.tools import nottest
-from nose.plugins.attrib import attr
-import binascii
-from scapy_service import *
-from pprint import pprint
-import zmq
-import json
-import scapy_zmq_server
-import threading
-from scapy_zmq_client import Scapy_server_wrapper
-
-class scapy_service_tester(functional_general_test.CGeneralFunctional_Test):
- def setUp(self):
- self.s = Scapy_service()
-
- def tearDown(self):
- pass
-
- '''
- test for db and field update - checking check_update_test()
- '''
- def test_check_update(self):
- allData = self.s.get_all()
- allDataParsed = allData
- dbMD5 = allDataParsed['db_md5']
- fieldMD5 = allDataParsed['fields_md5']
- resT1 = self.s.check_update(dbMD5,fieldMD5)
- assert_equal(resT1,True)
- try:
- resT2 = False
- resT2 = self.s.check_update('falseMD5','falseMD5')
- except Exception as e:
- if e.message == "Fields DB is not up to date":
- resT2 = True
- else:
- raise
- assert_equal(resT2,True)
- try:
- resT3 = False
- resT3 = self.s.check_update(dbMD5,'falseMD5')
- except Exception as e:
- if e.message == "Fields DB is not up to date":
- resT3 = True
- else:
- raise
- assert_equal(resT3,True)
- try:
- resT4 = False
- resT4 = self.s.check_update('falseMD5',fieldMD5)
- except Exception as e:
- if e.message == "Protocol DB is not up to date":
- resT4 = True
- else:
- raise
- assert_equal(resT4,True)
-
-
- def test_check_updating_db(self):
- #assume i got old db
- try:
- result = self.s.check_update('falseMD5','falseMD5')
- except:
- newAllData = self.s.get_all()
- dbMD5 = newAllData['db_md5']
- fieldMD5 = newAllData['fields_md5']
- result = self.s.check_update(dbMD5,fieldMD5)
- assert_equal(result,True)
- else:
- raise Exception("scapy_server_test: check_updating_db failed")
-
-
- def _build_packet_test_method(self,original_pkt):
- test_pkt = original_pkt
- original_pkt = eval(original_pkt)
- test_res = self.s.build_pkt(test_pkt)
- test_pkt_buffer = test_res['buffer']
- resT1 = (test_pkt_buffer == binascii.b2a_base64(str(original_pkt)))
- assert_equal(resT1,True)
-
-
-#testing offsets of a packet
- def _get_all_offsets_test_method(self,original_pkt):
- test_pkt = original_pkt
- original_pkt = eval(original_pkt)
- original_pkt.build()
- tested_offsets_by_layers = self.s._get_all_pkt_offsets(test_pkt)
- layers = (test_pkt).split('/')
- offsets_by_layers = {}
- for layer in layers:
- fields_dict = {}
- layer_name = layer.partition('(')[0] #clear layer name to include only alpha-numeric
- layer_name = re.sub(r'\W+', '',layer_name)
- for f in original_pkt.fields_desc:
- size = f.get_size_bytes()
- name = f.name
- if f.name is 'load':
- size = len(original_pkt)
- layer_name = 'Raw'
- fields_dict[f.name]= [f.offset, size]
- fields_dict['global_offset'] = original_pkt.offset
- original_pkt = original_pkt.payload
- offsets_by_layers[layer_name] = fields_dict
- resT1 = (tested_offsets_by_layers == offsets_by_layers)
- assert_equal(resT1,True)
-
- def _offsets_and_buffer_test_method(self,mac_src,mac_dst,ip_src,ip_dst):
- pkt = Ether(src=mac_src,dst=mac_dst)/IP(src=ip_src,dst=ip_dst)/TCP()
- pkt_descriptor = "Ether(src='"+mac_src+"',dst='"+mac_dst+"')/IP(src='"+ip_src+"',dst='"+ip_dst+"')/TCP()"
- pkt_offsets = self.s._get_all_pkt_offsets(pkt_descriptor)
- pkt_buffer = str(pkt)
- #--------------------------Dest-MAC--------------------
- mac_start_index = pkt_offsets['Ether']['dst'][0]+pkt_offsets['Ether']['global_offset']
- mac_end_index = mac_start_index+pkt_offsets['Ether']['dst'][1]
- assert_equal(binascii.b2a_hex(pkt_buffer[mac_start_index:mac_end_index]),mac_dst.translate(None,':'))
- #--------------------------Src-MAC---------------------
- mac_start_index = pkt_offsets['Ether']['src'][0]+pkt_offsets['Ether']['global_offset']
- mac_end_index = mac_start_index+pkt_offsets['Ether']['src'][1]
- assert_equal(binascii.b2a_hex(pkt_buffer[mac_start_index:mac_end_index]),mac_src.translate(None,':'))
- #--------------------------Dest-IP---------------------
- ip_start_index = pkt_offsets['IP']['dst'][0]+pkt_offsets['IP']['global_offset']
- ip_end_index= ip_start_index+pkt_offsets['IP']['dst'][1]
- assert_equal(binascii.b2a_hex(pkt_buffer[ip_start_index:ip_end_index]),binascii.hexlify(socket.inet_aton(ip_dst)))
- #--------------------------Src-IP----------------------
- ip_start_index = pkt_offsets['IP']['src'][0]+pkt_offsets['IP']['global_offset']
- ip_end_index= ip_start_index+pkt_offsets['IP']['src'][1]
- assert_equal(binascii.b2a_hex(pkt_buffer[ip_start_index:ip_end_index]),binascii.hexlify(socket.inet_aton(ip_src)))
-
- def test_multi_packet(self):
- packets= [
- 'Ether()',
- 'Ether()/IP()',
- 'TCP()',
- 'UDP()',
- 'Ether()/IP()/TCP()/"test"',
- 'Ether()/IP()/UDP()',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")',
- 'Ether()/Dot1Q(vlan=12)/Dot1Q(vlan=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether()/Dot1Q(vlan=12)/IP(src="16.0.0.1",dst="48.0.0.1")/TCP(dport=12,sport=1025)',
- 'Ether()/Dot1Q(vlan=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether()/Dot1Q(vlan=12)/IPv6(src="::5")/TCP(dport=12,sport=1025)',
- 'Ether()/IP()/UDP()/IPv6(src="::5")/TCP(dport=12,sport=1025)',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether()/IP(dst="48.0.0.1")/TCP(dport=80,flags="S")',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether() / IP(src = "16.0.0.1", dst = "48.0.0.1") / UDP(dport = 12, sport = 1025)',
- 'Ether()/IP()/UDP()',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether()/IP(src="16.0.0.2",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- 'Ether()/IP(src="16.0.0.3",dst="48.0.0.1")/UDP(dport=12,sport=1025)',
- r'Ether()/IP()/IPv6()/IP(dst="48.0.0.1",options=IPOption("\x01\x01\x01\x00"))/UDP(dport=12,sport=1025)',
- r'Ether()/IP(dst="48.0.0.1",options=IPOption("\x01\x01\x01\x00"))/UDP(dport=12,sport=1025)',
- 'Ether()',
- 'Ether()/IP()/UDP(sport=1337,dport=4789)/VXLAN(vni=42)/Ether()/IP()/("x"*20)',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=3797,sport=3544)/IPv6(dst="2001:0:4137:9350:8000:f12a:b9c8:2815",src="2001:4860:0:2001::68")/UDP(dport=12,sport=1025)/ICMPv6Unknown()',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(sport=1025)/DNS()',
- 'Ether()/MPLS(label=17,cos=1,s=0,ttl=255)/MPLS(label=0,cos=1,s=1,ttl=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)/("x"*20)',
- 'Ether()/MPLS(label=17,cos=1,s=0,ttl=255)/MPLS(label=12,cos=1,s=1,ttl=12)/IP(src="16.0.0.1",dst="48.0.0.1")/UDP(dport=12,sport=1025)/("x"*20)',
- 'Ether()/IP(src="16.0.0.1",dst="48.0.0.1")/ICMP(type=3)',
- 'Ether()/IP()/GRE()/("x"*2)']
-
- for packet in packets:
- self._get_all_offsets_test_method(packet)
-
- for packet in packets:
- self._build_packet_test_method(packet)
-
-
-
- def test_offsets_and_buffer(self):
- self._offsets_and_buffer_test_method('ab:cd:ef:12:34:56','98:76:54:32:1a:bc','127.1.1.1','192.168.1.1')
- self._offsets_and_buffer_test_method('bb:bb:bb:bb:bb:bb','aa:aa:aa:aa:aa:aa','1.1.1.1','0.0.0.0')
-
-class scapy_server_thread(threading.Thread):
- def __init__(self,thread_id,server_port=5555):
- threading.Thread.__init__(self)
- self.thread_id = thread_id
- self.server_port = server_port
-
- def run(self):
- print('\nStarted scapy thread server')
- scapy_zmq_server.main(self.server_port)
- print('Thread server closed')
-
-# Scapy_server_wrapper is the CLIENT for the scapy server, it wraps the CLIENT: its default port is set to 5555, default server ip set to localhost
-class scapy_server_tester(scapy_service_tester):
- def setUp(self):
- self.thread1 = scapy_server_thread(thread_id=1)
- self.thread1.start()
- self.s = Scapy_server_wrapper(dest_scapy_port=5555,server_ip_address='localhost')
-
- def tearDown(self):
- self.s.call_method('shut_down',[])
- self.thread1.join()
-
-
-
-
diff --git a/scripts/automation/regression/functional_tests/stl_basic_tests.py b/scripts/automation/regression/functional_tests/stl_basic_tests.py index 2843d619..39f42e7d 100644 --- a/scripts/automation/regression/functional_tests/stl_basic_tests.py +++ b/scripts/automation/regression/functional_tests/stl_basic_tests.py @@ -222,7 +222,6 @@ class CStlBasic_Test(functional_general_test.CGeneralFunctional_Test): def test_stl_profiles (self): - p = [ ["udp_1pkt_1mac_override.py","-m 1 -l 50",True], ["syn_attack.py","-m 1 -l 50",True], diff --git a/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py b/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py index 11a9db42..ab99f1db 100755 --- a/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py +++ b/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py @@ -2,20 +2,20 @@ import os import sys stl_pathname = os.path.abspath(os.path.join(os.pardir, os.pardir)) -additional_stl_udp_pkts = os.path.abspath(os.path.join(os.pardir,os.pardir,os.pardir,os.pardir, os.pardir,'stl')) sys.path.append(stl_pathname) -sys.path.append(additional_stl_udp_pkts) -import trex_stl_lib + from trex_stl_lib.api import * -from copy import deepcopy -import sys import tempfile import hashlib import binascii from pprint import pprint -from scapy.layers.dns import * -from udp_1pkt_vxlan import VXLAN -from udp_1pkt_mpls import MPLS +#from scapy.layers.dns import DNS +#from scapy.contrib.mpls import MPLS + +#additional_stl_udp_pkts = os.path.abspath(os.path.join(os.pardir,os.pardir,os.pardir,os.pardir, os.pardir,'stl')) +#sys.path.append(additional_stl_udp_pkts) +#from udp_1pkt_vxlan import VXLAN +#sys.path.remove(additional_stl_udp_pkts) try: from cStringIO import StringIO @@ -26,8 +26,23 @@ except ImportError: class Scapy_service_api(): - def get_all(self): - """ get_all(self) + + def get_version_handler(self,client_v_major,client_v_minor): + """ get_version_handler(self,client_v_major,client_v_minor) + + Gives a handler to client to connect and use server api + + Parameters + ---------- + client_v_major - major number of api version on the client side + + Returns + ------- + Handler(string) to provide when using server api + """ + pass + def get_all(self,client_v_handler): + """ get_all(self,client_v_handler) Sends all the protocols and fields that Scapy Service supports. also sends the md5 of the Protocol DB and Fields DB used to check if the DB's are up to date @@ -46,8 +61,8 @@ class Scapy_service_api(): """ pass - def check_update(self,db_md5,field_md5): - """ check_update(self,db_md5,field_md5) + def check_update_of_dbs(self,client_v_handler,db_md5,field_md5): + """ check_update_of_dbs(self,client_v_handler,db_md5,field_md5) Checks if the Scapy Service running on the server has a newer version of the databases that the client has Parameters @@ -66,14 +81,14 @@ class Scapy_service_api(): pass - def build_pkt(self,pkt_descriptor): - """ build_pkt(self,pkt_descriptor) -> Dictionary (of Offsets,Show2 and Buffer) + def build_pkt(self,client_v_handler,pkt_model_descriptor): + """ build_pkt(self,client_v_handler,pkt_model_descriptor) -> Dictionary (of Offsets,Show2 and Buffer) Performs calculations on the given packet and returns results for that packet. Parameters ---------- - pkt_descriptor - A string describing a network packet, in Scapy Format + pkt_descriptor - An array of dictionaries describing a network packet Returns ------- @@ -89,7 +104,7 @@ class Scapy_service_api(): pass - def get_tree(self): + def get_tree(self,client_v_handler): """ get_tree(self) -> Dictionary describing an example of hierarchy in layers Scapy service holds a tree of layers that can be stacked to a recommended packet @@ -108,17 +123,43 @@ class Scapy_service_api(): None """ pass + + def reconstruct_pkt(self,client_v_handler,binary_pkt): + """ reconstruct_pkt(self,client_v_handler,binary_pkt) + Makes a Scapy valid packet from a binary string and returns all information returned in build_pkt + + Parameters + ---------- + Packet in binary, formatted in "base64" encoding + + Returns + ------- + All data provided in build_pkt: + show2 - detailed description of the packet + buffer - the packet presented in binary + offsets - the offset[in bytes] of each field in the packet + + """ + pass class ScapyException(Exception): pass class Scapy_service(Scapy_service_api): #---------------------------------------------------------------------------------------------------- - class scapyRegex: + class ScapyFieldDesc: def __init__(self,FieldName,regex='empty'): self.FieldName = FieldName self.regex = regex + #defualt values - should be changed when needed, or added to constructor + self.string_input ="" + self.string_input_mex_len = 1 + self.integer_input = 0 + self.integer_input_min = 0 + self.integer_input_max = 1 + self.input_array = [] + self.input_list_max_len = 1 def stringRegex(self): return self.regex @@ -129,10 +170,13 @@ class Scapy_service(Scapy_service_api): self.transport_protocols = {'TCP':self.Raw,'UDP':self.Raw} self.network_protocols = {'IP':self.transport_protocols ,'ARP':''} self.low_level_protocols = { 'Ether': self.network_protocols } - self.regexDB= {'MACField' : self.scapyRegex('MACField','^([0-9a-fA-F][0-9a-fA-F]:){5}([0-9a-fA-F][0-9a-fA-F])$'), - 'IPField' : self.scapyRegex('IPField','^(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9])\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9]|0)\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9]|0)\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[0-9])$')} + self.regexDB= {'MACField' : self.ScapyFieldDesc('MACField','^([0-9a-fA-F][0-9a-fA-F]:){5}([0-9a-fA-F][0-9a-fA-F])$'), + 'IPField' : self.ScapyFieldDesc('IPField','^(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9])\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9]|0)\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[1-9]|0)\.(25[0-5]|2[0-4][0-9]|[0-1]{1}[0-9]{2}|[1-9]{1}[0-9]{1}|[0-9])$')} self.all_protocols = self._build_lib() self.protocol_tree = {'ALL':{'Ether':{'ARP':{},'IP':{'TCP':{'RAW':'payload'},'UDP':{'RAW':'payload'}}}}} + self.version_major = '1' + self.version_minor = '01' + self.server_v_hashed = self._generate_version_hash(self.version_major,self.version_minor) def _all_protocol_structs(self): @@ -209,7 +253,7 @@ class Scapy_service(Scapy_service_api): if f in self.regexDB: fieldDict[f] = self.regexDB[f].stringRegex() else: - fieldDict[f] = self.scapyRegex(f).stringRegex() + fieldDict[f] = self.ScapyFieldDesc(f).stringRegex() return fieldDict def _show2_to_dict(self,pkt): @@ -267,12 +311,14 @@ class Scapy_service(Scapy_service_api): return res_md5 def get_version(self): - return {'built_by':'itraviv','version':'v1.01'} + return {'built_by':'itraviv','version':self.version_major+'.'+self.version_minor} def supported_methods(self,method_name='all'): if method_name=='all': methods = {} for f in dir(Scapy_service): + if f[0]=='_': + continue if inspect.ismethod(eval('Scapy_service.'+f)): param_list = inspect.getargspec(eval('Scapy_service.'+f))[0] del param_list[0] #deleting the parameter "self" that appears in every method @@ -284,24 +330,78 @@ class Scapy_service(Scapy_service_api): return True return False + def _generate_version_hash(self,v_major,v_minor): + v_for_hash = v_major+v_minor+v_major+v_minor + m = hashlib.md5() + m.update(v_for_hash) + v_handle = binascii.b2a_base64(m.digest()) + return unicode(v_handle,"utf-8") + + def _generate_invalid_version_error(self): + error_desc1 = "Provided version handler does not correspond to the server's version.\nUpdate client to latest version.\nServer version:"+self.version_major+"."+self.version_minor + return error_desc1 + + def _verify_version_handler(self,client_v_handler): + return (self.server_v_hashed == client_v_handler) + + def _parse_packet_dict(self,layer,scapy_layers,scapy_layer_names): + class_name = scapy_layer_names.index(layer['id']) + class_p = scapy_layers[class_name] # class pointer + kwargs = {} + if 'Fields' in layer: + for field in layer['Fields']: + key = field['id'] + value = field['value'] + if type(value) is list: + resolved_value = [] + for arg_class in value: + option_class_p = scapy.all.__dict__[arg_class["class"]] + option_kwargs = {} + for field in arg_class['Fields']: + option_kwargs[field['id']] = field['value'] + resolved_value.append(option_class_p(**option_kwargs)) + value = resolved_value + kwargs[key] = value + return class_p(**kwargs) + + def _packet_model_to_scapy_packet(self,data): + layers = Packet.__subclasses__() + layer_names = [ layer.__name__ for layer in layers] + base_layer = self._parse_packet_dict(data[0],layers,layer_names) + for i in range(1,len(data),1): + packet_layer = self._parse_packet_dict(data[i],layers,layer_names) + base_layer = base_layer/packet_layer + return base_layer + + #--------------------------------------------API implementation------------- - def get_tree(self): + def get_tree(self,client_v_handler): + if not (self._verify_version_handler(client_v_handler)): + raise ScapyException(self._generate_invalid_version_error()) return self.protocol_tree -# pkt_descriptor in string format - def build_pkt(self,pkt_descriptor): - pkt = eval(pkt_descriptor) + def get_version_handler(self,client_v_major,client_v_minor): + v_handle = self._generate_version_hash(client_v_major,client_v_minor) + return v_handle + +# pkt_descriptor in packet model format (dictionary) + def build_pkt(self,client_v_handler,pkt_model_descriptor): + if not (self._verify_version_handler(client_v_handler)): + raise ScapyException(self._generate_invalid_version_error()) + pkt = self._packet_model_to_scapy_packet(pkt_model_descriptor) show2data = self._show2_to_dict(pkt) bufferData = str(pkt) #pkt buffer bufferData = binascii.b2a_base64(bufferData) - pkt_offsets = self._get_all_pkt_offsets(pkt_descriptor) + pkt_offsets = self._get_all_pkt_offsets(pkt.command()) res = {} res['show2'] = show2data res['buffer'] = bufferData res['offsets'] = pkt_offsets return res - def get_all(self): + def get_all(self,client_v_handler): + if not (self._verify_version_handler(client_v_handler)): + raise ScapyException(self._generate_invalid_version_error()) fields=self._get_all_fields() db=self._get_all_db() fields_md5 = self._get_md5(fields) @@ -314,7 +414,9 @@ class Scapy_service(Scapy_service_api): return res #input in string encoded base64 - def check_update(self,db_md5,field_md5): + def check_update_of_dbs(self,client_v_handler,db_md5,field_md5): + if not (self._verify_version_handler(client_v_handler)): + raise ScapyException(self._generate_invalid_version_error()) fields=self._get_all_fields() db=self._get_all_db() current_db_md5 = self._get_md5(db) @@ -328,6 +430,16 @@ class Scapy_service(Scapy_service_api): else: raise ScapyException("Fields DB is not up to date") +#input of binary_pkt must be encoded in base64 + def reconstruct_pkt(self,client_v_handler,binary_pkt): + pkt_in_hex = binary_pkt.decode('base64') + scapy_pkt = Ether(pkt_in_hex) + scapy_pkt_cmd = scapy_pkt.command() + return self.build_pkt(client_v_handler,scapy_pkt_cmd) + + + + #--------------------------------------------------------------------------- diff --git a/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_zmq_server.py b/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_zmq_server.py index da22b697..82bd9fec 100755 --- a/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_zmq_server.py +++ b/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_zmq_server.py @@ -3,11 +3,10 @@ import time import sys
import os
-python2_zmq_path = os.path.abspath(os.path.join(os.pardir,os.pardir,os.pardir,os.pardir,
- os.pardir,'external_libs','pyzmq-14.5.0','python2','fedora18','64bit'))
stl_pathname = os.path.abspath(os.path.join(os.pardir, os.pardir))
-sys.path.append(stl_pathname)
-sys.path.append(python2_zmq_path)
+if stl_pathname not in sys.path:
+ sys.path.append(stl_pathname)
+from trex_stl_lib.api import *
import zmq
import inspect
from scapy_service import *
@@ -97,7 +96,7 @@ class Scapy_wrapper: return response
class Scapy_server():
- def __init__(self, port=4507):
+ def __init__(self, args,port=4507):
self.scapy_wrapper = Scapy_wrapper()
self.port = port
self.context = zmq.Context()
@@ -161,12 +160,12 @@ class Scapy_server(): #arg1 is port number for the server to listen to
-def main(arg1=4507):
- s = Scapy_server(arg1)
+def main(args,port):
+ s = Scapy_server(args,port)
s.activate()
if __name__=='__main__':
- if len(sys.argv)>1:
+
parser = ArgumentParser(description=' Runs Scapy Server ')
parser.add_argument('-s','--scapy-port',type=int, default = 4507, dest='scapy_port',
help='Select port to which Scapy Server will listen to.\n default is 4507.',action='store')
@@ -175,6 +174,6 @@ if __name__=='__main__': action='store_true',default = False)
args = parser.parse_args()
port = args.scapy_port
- sys.exit(main(port))
- else:
- sys.exit(main())
+ sys.exit(main(args,port))
+
+
|