From d4b2ae7ddbaae8660ddaff710bf4a5459ff0657f Mon Sep 17 00:00:00 2001 From: Yaroslav Brustinov Date: Thu, 28 Jan 2016 18:28:45 +0200 Subject: HLTAPI updates, aggregated results tests count print --- .../trex_control_plane/client/trex_hltapi.py | 568 ++++++++++++++------- 1 file changed, 386 insertions(+), 182 deletions(-) (limited to 'scripts/automation/trex_control_plane/client/trex_hltapi.py') diff --git a/scripts/automation/trex_control_plane/client/trex_hltapi.py b/scripts/automation/trex_control_plane/client/trex_hltapi.py index c25c73cb..ad0a4d3a 100755 --- a/scripts/automation/trex_control_plane/client/trex_hltapi.py +++ b/scripts/automation/trex_control_plane/client/trex_hltapi.py @@ -1,188 +1,320 @@ #!/router/bin/python +''' +Supported function/arguments: +connect() + device + port_list + username + reset + break_locks + +cleanup_session() + maintain_lock + port_list + port_handle + +traffic_config() + mode ( create | modify | remove | reset ) + port_handle + transmit_mode ( continuous | multi_burst | single_burst ) + rate_pps + stream_id + bidirectional + l2_encap + mac_src + mac_dst + l3_protocol + ip_src_addr + ip_dst_addr + l3_length + l4_protocol + +traffic_control() + action ( run | stop ) + port_handle + +''' + + + + + + import trex_root_path from client_utils.packet_builder import CTRexPktBuilder from trex_stateless_client import STLClient from common.trex_streams import * -from client_utils.general_utils import id_count_gen +from client_utils.general_utils import id_count_gen, get_integer import dpkt +import socket +from misc_methods import print_r +import traceback +from functools import partial + +class HLT_ERR(dict): + def __init__(self, log = 'Unknown error', **kwargs): + dict.__init__(self, {'status': 0}) + if type(log) is dict: + dict.update(self, log) + else: + self['log'] = log + dict.update(self, kwargs) + +class HLT_OK(dict): + def __init__(self, init_dict = {}, **kwargs): + dict.__init__(self, {'status': 1, 'log': None}) + dict.update(self, init_dict) + dict.update(self, kwargs) class CTRexHltApi(object): - def __init__(self): + def __init__(self, verbose = 1): self.trex_client = None self.connected = False + self.verbose = verbose # self._stream_db = CStreamList() - self._port_data = {} + #self._port_data = {} + self._stream_id_gen = id_count_gen() # common to all ports + self._hlt_streams_history = {} # streams per stream_id in format of hlt arguments # ----- session functions ----- # + # device: ip or hostname, will take all available ports # sync = RPC, async = ZMQ - def connect(self, device, port_list, username, sync_port = 4501, async_port = 4500, reset=False, break_locks=False): - ret_dict = {"status": 0} - self.trex_client = STLClient(username, device, sync_port, async_port) + def connect(self, device, port_list, username='', reset=False, break_locks=False, sync_port=4501, async_port=4500, virtual = False): - rc = self.trex_client.connect() - if rc.bad(): + try: + device = socket.gethostbyname(device) # work with ip + except: # give it another try + try: + device = socket.gethostbyname(device) + except Exception as e: + return HLT_ERR('[ERR] Could not translate hostname "%s" to IP: %s' % (device, e)) + try: + self.trex_client = STLClient(username, device, sync_port, async_port, verbose_level = self.verbose) + except Exception as e: self.trex_client = None - ret_dict.update({"log": rc.err()}) - return ret_dict - # arrived here, connection successfully created with server - # next, try acquiring ports of TRex - port_list = self.parse_port_list(port_list) - response = self.trex_client.acquire(port_list, force=break_locks) - res_ok, log = CTRexHltApi.process_response(port_list, response) - if not res_ok: - self.trex_client.disconnect() + return HLT_ERR('[ERR] Could not init stateless client %s: %s' % (device, e)) + try: + self.trex_client.connect() + except Exception as e: self.trex_client = None - ret_dict.update({"log": log}) - # TODO: should revert taken ports?! - return ret_dict + return HLT_ERR('[ERR] Could not connect to device %s: %s' % (device, e)) + + # connection successfully created with server, try acquiring ports of TRex + try: + port_list = self.parse_port_list(port_list) + self.trex_client.acquire(ports = port_list, force = break_locks) + except Exception as e: + self.trex_client = None + return HLT_ERR('[ERR] Could not acquire ports %s: %s' % (port_list, e)) + + # since only supporting single TRex at the moment, 1:1 map + port_handle = self.trex_client.get_acquired_ports() + # arrived here, all desired ports were successfully acquired - print log if reset: # remove all port traffic configuration from TRex - response = self.trex_client.remove_all_streams(port_list) - res_ok, log = CTRexHltApi.process_response(port_list, response) - if not res_ok: - self.trex_client.disconnect() + try: + self.trex_client.reset(ports = port_list) + except Exception as e: self.trex_client = None - ret_dict.update({"log": log}) - return ret_dict - print log - port_handle = {device: {port: port # since only supporting single TRex at the moment, 1:1 map - for port in port_list} - } - ret_dict.update({"status": 1, - "log": None, - "port_handle": port_handle, - "offline": 0}) # ports are online + return HLT_ERR('Error in reset traffic: %s' % e) + self.connected = True - self._port_data_init(port_list) - return ret_dict + return HLT_OK(port_handle = port_handle) - def cleanup_session(self, port_list, maintain_lock=False): - ret_dict = {"status": 0} + def cleanup_session(self, maintain_lock = False, **kwargs): if not maintain_lock: # release taken ports - if port_list == "all": - port_list = self.trex_client.get_acquired_ports() - else: - port_list = self.parse_port_list(port_list) - response = self.trex_client.release(port_list) - res_ok, log = CTRexHltApi.process_response(port_list, response) - if not res_ok: - ret_dict.update({"log": log}) - return ret_dict - ret_dict.update({"status": 1, - "log": None}) - self.trex_client.disconnect() + port_list = kwargs.get('port_list', kwargs.get('port_handle', 'all')) + try: + if port_list == 'all': + port_list = self.trex_client.get_acquired_ports() + else: + port_list = self.parse_port_list(port_list) + except Exception as e: + return HLT_ERR('Unable to determine which ports to release: %s' % e) + try: + self.trex_client.release(port_list) + except Exception as e: + return HLT_ERR('Unable to release ports %s: %s' % (port_list, e)) + try: + self.trex_client.disconnect() + except Exception as e: + return HLT_ERR('Error disconnecting: %s' % e) self.trex_client = None self.connected = False - return ret_dict + return HLT_OK() - def interface_config(self, port_handle, mode="config"): - ALLOWED_MODES = ["config", "modify", "destroy"] + def interface_config(self, port_handle, mode='config'): + ALLOWED_MODES = ['config', 'modify', 'destroy'] if mode not in ALLOWED_MODES: - raise ValueError("mode must be one of the following values: {modes}".format(modes=ALLOWED_MODES)) + return HLT_ERR('Mode must be one of the following values: %s' % ALLOWED_MODES) # pass this function for now... - return {"status": 1, "log": None} + return HLT_OK() # ----- traffic functions ----- # - def traffic_config(self, mode, port_list, - l2_encap="ethernet_ii", mac_src="00:00:01:00:00:01", mac_dst="00:00:00:00:00:00", - l3_protocol="ipv4", ip_src_addr="0.0.0.0", ip_dst_addr="192.0.0.1", l3_length=110, - transmit_mode="continuous", rate_pps=100, - **kwargs): - if type(port_list) is not list(): - port_list = [port_list] - ALLOWED_MODES = ["create", "modify", "remove", "enable", "disable", "reset"] + def traffic_config(self, mode, port_handle, transmit_mode = 'continuous', rate_pps = 1, **kwargs): + stream_id = kwargs.get('stream_id') + if type(stream_id) in (list, set): + for each_stream_id in stream_id: + traffic_config(self, mode, port_handle, transmit_mode = 'continuous', rate_pps = 1, stream_id = each_stream_id, **kwargs): + + if type(port_handle) is not list: + port_handle = [port_handle] + ALLOWED_MODES = ['create', 'modify', 'remove', 'enable', 'disable', 'reset'] if mode not in ALLOWED_MODES: - raise ValueError("mode must be one of the following values: {modes}".format(modes=ALLOWED_MODES)) - if mode == "create": + return HLT_ERR('Mode must be one of the following values: %s' % ALLOWED_MODES) + + if mode == 'reset': + try: + self.trex_client.remove_all_streams(port_handle) + return HLT_OK() + except Exception as e: + return HLT_ERR('Could not reset streams at ports %s: %s' % (port_handle, e)) + + if mode == 'remove': + if stream_id is None: + return HLT_ERR('Please specify stream_id to remove.') + if type(stream_id) is str and stream_id == 'all': + try: + self.trex_client.remove_all_streams(port_handle) + except Exception as e: + return HLT_ERR('Could not remove all streams at ports %s: %s' % (port_handle, e)) + else: + try: + self._remove_stream(stream_id, port_handle) + except Exception as e: + return HLT_ERR('Could not remove streams with specified by %s, error: %s' % (stream_id, log)) + return HLT_OK() + + if mode == 'enable': + stream_id = kwargs.get('stream_id') + if stream_id is None: + return HLT_ERR('Please specify stream_id to enable.') + if stream_id not in self._hlt_streams_history: + return HLT_ERR('This stream_id (%s) was not used before, please create new.' % stream_id) + self._hlt_streams_history[stream_id].update(kwargs) # <- the modification + + + if mode == 'modify': # we remove stream and create new one with same stream_id + stream_id = kwargs.get('stream_id') + if stream_id is None: + return HLT_ERR('Please specify stream_id to modify.') + if stream_id not in self._hlt_streams_history: + return HLT_ERR('This stream_id (%s) was not used before, please create new.' % stream_id) + self._hlt_streams_history[stream_id].update(kwargs) # <- the modification + kwargs = self._hlt_streams_history[stream_id] + for port_id in port_handle: + if stream_id not in self.trex_client.get_stream_id_list(port_id): + return HLT_ERR('Port %s does not have stream_id %s.' % (port_id, stream_id)) + try: + self.trex_client.remove_stream(stream_id, port_handle) + except Exception as e: + return HLT_ERR('Could not remove streams specified by %s, error: %s' % (stream_id, log)) + + if mode == 'create' or mode == 'modify': # create a new stream with desired attributes, starting by creating packet + + if 'bidirectional' in kwargs: # two streams with opposite src and dst MAC + del kwargs['bidirectional'] + bidirect_err = 'When using bidirectional flag, ' + if len(port_handle) != 2: + return HLT_ERR(bidirect_err + 'number of ports should be exactly 2') + if mac_src not in kwargs or mac_dst not in kwargs: + return HLT_ERR(bidirect_err + 'mac_src and mac_dst should be specified') + try: + res1 = self.traffic_config(mode, port_handle[0], transmit_mode, rate_pps, **kwargs) + res2 = self.traffic_config(mode, port_handle[1], transmit_mode, rate_pps, mac_src = kwargs['mac_dst'], mac_dst = kwargs['mac_src'], **kwargs) + except Exception as e: + return HLT_ERR('Could not generate bidirectional traffic: %s' % e) + return HLT_OK(stream_id = [res1['stream_id'], res2['stream_id']]) + try: - packet = CTRexHltApi.generate_stream(l2_encap, mac_src, mac_dst, - l3_protocol, ip_src_addr, ip_dst_addr, l3_length) + packet = CTRexHltApi._generate_stream(**kwargs) + except Exception as e: + return HLT_ERR('Could not generate stream: %s' % e) # set transmission attributes - tx_mode = CTxMode(transmit_mode, rate_pps, **kwargs) + #try: + # tx_mode = CTxMode(type = transmit_mode, pps = rate_pps, **kwargs) + #except Exception as e: + # return HLT_ERR('Could not init CTxMode: %s' % e) + + try: # set rx_stats rx_stats = CRxStats() # defaults with disabled + except Exception as e: + return HLT_ERR('Could not init CTxMode: %s' % e) + + try: # join the generated data into stream - stream_obj = CStream() - stream_obj_params = {"enabled": True, - "self_start": True, - "next_stream_id": -1, - "isg": 0.0, - "mode": tx_mode, - "rx_stats": rx_stats, - "packet": packet} # vm is excluded from this list since CTRexPktBuilder obj is passed - stream_obj.load_data(**stream_obj_params) + stream_obj = STLStream(packet = packet, + pps = rate_pps, + enabled = False, + self_start = True, + isg = 0.0, + rx_stats = rx_stats, + next_stream_id = -1 + ) + # using CStream + #stream_obj_params = {'enabled': False, + # 'self_start': True, + # 'next_stream_id': -1, + # 'isg': 0.0, + # 'mode': tx_mode, + # 'rx_stats': rx_stats, + # 'packet': packet} # vm is excluded from this list since CTRexPktBuilder obj is passed + #stream_obj.load_data(**stream_obj_params) except Exception as e: # some exception happened during the stream creation - return {"status": 0, "log": str(e)} - # try adding the stream per port, until free stream_id is found - for port_id in port_list: - port_data = self._port_data.get(port_id) - id_candidate = None - # TODO: change this to better implementation - while True: - id_candidate = port_data["stream_id_gen"].next() - response = self.trex_client.add_stream(stream_id=id_candidate, - stream_obj=stream_obj.dump(), - port_id_list=port_id) - res_ok, log = CTRexHltApi.process_response(port_id, response) - if res_ok: - # found non-taken stream_id on server - # save it for modifying needs - port_data["streams"].update({id_candidate: stream_obj}) - break - else: - print log - # proceed to another iteration to use another id - print 'need another iteration?' - continue - return {"status": 1, - "stream_id": id_candidate, - "log": None} + return HLT_ERR(e) - else: - raise NotImplementedError("mode '{0}' is not supported yet on TRex".format(mode)) + if stream_id is None: + stream_id = self._get_next_stream_id() + # try adding the stream per ports + #print 'one' + #print stream_obj.dump() + + try: + self.trex_client.add_streams(streams=stream_obj.dump(), + ports=port_handle) + self._hlt_streams_history[stream_id] = kwargs + except Exception as e: + return HLT_ERR('Could not add stream to ports: %s' % e) + return HLT_OK(stream_id = stream_id) + + return HLT_ERR('Got to the end of traffic_config, mode not implemented or forgot "return" function somewhere.') def traffic_control(self, action, port_handle, **kwargs): - ALLOWED_ACTIONS = ["clear_stats", "run", "stop", "sync_run"] + ALLOWED_ACTIONS = ['clear_stats', 'run', 'stop', 'sync_run'] if action not in ALLOWED_ACTIONS: - raise ValueError("action must be one of the following values: {actions}".format(actions=ALLOWED_ACTIONS)) - # ret_dict = {"status": 0, "stopped": 1} - port_list = self.parse_port_list(port_handle) - if type(port_list) is not list(): - port_list = [port_list] - if action == "run": - if not set(kwargs.keys()) >= set(['mul', 'duration']): - raise ValueError("For 'run' action should be specified mul and duration arguments") - response = self.trex_client.start_traffic(kwargs['mul'], kwargs['duration'], port_id_list=port_list) - res_ok, log = CTRexHltApi.process_response(port_list, response) - if res_ok: - return {"status": 1, - "stopped": 0, - "log": None} - else: - print log - elif action == "stop": - response = self.trex_client.stop_traffic(port_id_list=port_list) - res_ok, log = CTRexHltApi.process_response(port_list, response) - if res_ok: - return {"status": 1, - "stopped": 1, - "log": None} + return HLT_ERR('Action must be one of the following values: {actions}'.format(actions=ALLOWED_ACTIONS)) + if type(port_handle) is not list(): + port_handle = [port_handle] + + if action == 'run': + + try: + self.trex_client.start(duration = kwargs.get('duration', -1), ports = port_handle) + except Exception as e: + return HLT_ERR('Could start traffic: %s' % e) + return HLT_OK(stopped = 0) + + elif action == 'stop': + + try: + self.trex_client.stop_traffic(port_id_list=port_handle) + except Exception as e: + return HLT_ERR('Could start traffic: %s' % e) + return HLT_OK(stopped = 1) else: - raise NotImplementedError("action '{0}' is not supported yet on TRex".format(action)) + return HLT_ERR("Action '{0}' is not supported yet on TRex".format(action)) # if we arrived here, this means that operation FAILED! - return {"status": 0, - "log": log} - + return HLT_ERR("Probably action '%s' is not implemented" % action) def traffic_stats(self, port_handle, mode): ALLOWED_MODES = ["aggregate", "streams", "all"] @@ -200,53 +332,87 @@ class CTRexHltApi(object): rx_stats = CRxStats() # defaults with disabled # join the generated data into stream stream_obj = CStream() - stream_obj_params = {"enabled": True, - "self_start": True, - "next_stream_id": -1, - "isg": 0.0, - "mode": tx_mode, - "rx_stats": rx_stats, - "packet": packet} # vm is excluded from this list since CTRexPktBuilder obj is passed + stream_obj_params = {'enabled': True, + 'self_start': True, + 'next_stream_id': -1, + 'isg': 0.0, + 'mode': tx_mode, + 'rx_stats': rx_stats, + 'packet': packet} # vm is excluded from this list since CTRexPktBuilder obj is passed stream_obj.load_data(**stream_obj_params) except Exception as e: # some exception happened during the stream creation - return {"status": 0, "log": str(e)} + return {'status': 0, 'log': str(e)} # try adding the stream, until free stream_id is found - port_data = self._port_data.get(port_handle) + #port_data = self._port_data.get(port_handle) id_candidate = None # TODO: change this to better implementation while True: - id_candidate = port_data["stream_id_gen"].next() - response = self.trex_client.add_stream(stream_id=id_candidate, - stream_obj=stream_obj, - port_id=port_handle) + id_candidate = port_data['stream_id_gen'].next() + response = self.trex_client.add_streams(stream_id=id_candidate, + stream_obj=stream_obj, + port_id=port_handle) res_ok, log = CTRexHltApi.process_response(port_handle, response) if res_ok: # found non-taken stream_id on server # save it for modifying needs - port_data["streams"].update({id_candidate: stream_obj}) + port_data['streams'].update({id_candidate: stream_obj}) break else: # proceed to another iteration to use another id continue - return {"status": 1, - "stream_id": id_candidate, - "log": None} + return {'status': 1, + 'stream_id': id_candidate, + 'log': None} else: raise NotImplementedError("mode '{0}' is not supported yet on TRex".format(mode)) def get_aggregate_port_stats(self, port_handle): - return self.traffic_stats(port_handle, mode="aggregate") + return self.traffic_stats(port_handle, mode='aggregate') def get_stream_stats(self, port_handle): return self.traffic_stats(port_handle, mode="streams") + def get_port_streams(self, port_id): + return self.trex_client.ports[port_id].get_all_streams() + # ----- internal functions ----- # - def _port_data_init(self, port_list): - for port in port_list: - self._port_data[port] = {"stream_id_gen": id_count_gen(), - "streams": {}} + def _get_next_stream_id(self): + return self._stream_id_gen.next() + # remove streams from given port(s). + # stream_id can be: + # * int - exact stream_id value + # * list - list of stream_id values or strings (see below) + # * string - exact stream_id value, mix of ranges/list separated by comma: 2, 4-13 + def _remove_stream(self, stream_id, port_handle): + if get_integer(stream_id) is not None: # exact value of int or str + self.trex_client.remove_stream(get_integer(stream_id), port_handle) # actual remove + if type(stream_id) in (list, set): # list of values/strings + for each_stream_id in stream_id: + self._remove_stream(each_stream_id, port_handle) # recurse + return + if type(stream_id) is str: # range or list in string + if stream_id.find(',') != -1: + for each_stream_id_element in stream_id.split(','): + self._remove_stream(each_stream_id_element, port_handle) # recurse + return + if stream_id.find('-') != -1: + stream_id_min, stream_id_max = stream_id.split('-', 1) + stream_id_min = get_integer(stream_id_min) + stream_id_max = get_integer(stream_id_max) + if stream_id_min is None: + raise Exception('_remove_stream: wrong range param %s' % stream_id_min) + if stream_id_max is None: + raise Exception('_remove_stream: wrong range param %s' % stream_id_max) + if stream_id_max < stream_id_min: + raise Exception('_remove_stream: right range param is smaller than left one: %s-%s' % (stream_id_min, stream_id_max)) + for each_stream_id in xrange(stream_id_min, stream_id_max + 1): + self._remove_stream(each_stream_id, port_handle) # recurse + return + raise Exception('_remove_stream: wrong param %s' % stream_id) + + # obsolete @staticmethod def process_response(port_list, response): log = response.data() if response.good() else response.err() @@ -268,51 +434,89 @@ class CTRexHltApi(object): @staticmethod def join_batch_response(responses): if type(responses) is list(): - return "\n". join([str(response) for response in responses]) + return '\n'. join([str(response) for response in responses]) return responses @staticmethod - def generate_stream(l2_encap, mac_src, mac_dst, - l3_protocol, ip_src_addr, ip_dst_addr, l3_length): - ALLOWED_L3_PROTOCOL = {"ipv4": dpkt.ethernet.ETH_TYPE_IP, - "ipv6": dpkt.ethernet.ETH_TYPE_IP6, - "arp": dpkt.ethernet.ETH_TYPE_ARP} - ALLOWED_L4_PROTOCOL = {"tcp": dpkt.ip.IP_PROTO_TCP, - "udp": dpkt.ip.IP_PROTO_UDP, - "icmp": dpkt.ip.IP_PROTO_ICMP, - "icmpv6": dpkt.ip.IP_PROTO_ICMP6, - "igmp": dpkt.ip.IP_PROTO_IGMP, - "rtp": dpkt.ip.IP_PROTO_IRTP, - "isis": dpkt.ip.IP_PROTO_ISIS, - "ospf": dpkt.ip.IP_PROTO_OSPF} + def _generate_stream(l2_encap = 'ethernet_ii', mac_src = '00:00:01:00:00:01', mac_dst = '00:00:00:00:00:00', + l3_protocol = 'ipv4', ip_src_addr = '0.0.0.0', ip_dst_addr = '192.0.0.1', l3_length = 110, + l4_protocol = 'tcp', + **kwards): + ALLOWED_L3_PROTOCOL = {'ipv4': dpkt.ethernet.ETH_TYPE_IP, + 'ipv6': dpkt.ethernet.ETH_TYPE_IP6, + 'arp': dpkt.ethernet.ETH_TYPE_ARP} + ALLOWED_L4_PROTOCOL = {'tcp': dpkt.ip.IP_PROTO_TCP, + 'udp': dpkt.ip.IP_PROTO_UDP, + 'icmp': dpkt.ip.IP_PROTO_ICMP, + 'icmpv6': dpkt.ip.IP_PROTO_ICMP6, + 'igmp': dpkt.ip.IP_PROTO_IGMP, + 'rtp': dpkt.ip.IP_PROTO_IRTP, + 'isis': dpkt.ip.IP_PROTO_ISIS, + 'ospf': dpkt.ip.IP_PROTO_OSPF} pkt_bld = CTRexPktBuilder() - if l2_encap == "ethernet_ii": - pkt_bld.add_pkt_layer("l2", dpkt.ethernet.Ethernet()) + if l2_encap == 'ethernet_ii': + #('dst', '6s', ''), + #('src', '6s', ''), + #('type', 'H', ETH_TYPE_IP) + pkt_bld.add_pkt_layer('l2', dpkt.ethernet.Ethernet()) # set Ethernet layer attributes - pkt_bld.set_eth_layer_addr("l2", "src", mac_src) - pkt_bld.set_eth_layer_addr("l2", "dst", mac_dst) + pkt_bld.set_eth_layer_addr('l2', 'src', mac_src) + pkt_bld.set_eth_layer_addr('l2', 'dst', mac_dst) else: raise NotImplementedError("l2_encap does not support the desired encapsulation '{0}'".format(l2_encap)) # set l3 on l2 if l3_protocol not in ALLOWED_L3_PROTOCOL: - raise ValueError("l3_protocol must be one of the following: {0}".format(ALLOWED_L3_PROTOCOL)) - pkt_bld.set_layer_attr("l2", "type", ALLOWED_L3_PROTOCOL[l3_protocol]) + raise ValueError('l3_protocol must be one of the following: {0}'.format(ALLOWED_L3_PROTOCOL)) + pkt_bld.set_layer_attr('l2', 'type', ALLOWED_L3_PROTOCOL[l3_protocol]) # set l3 attributes - if l3_protocol == "ipv4": - pkt_bld.add_pkt_layer("l3", dpkt.ip.IP()) - pkt_bld.set_ip_layer_addr("l3", "src", ip_src_addr) - pkt_bld.set_ip_layer_addr("l3", "dst", ip_dst_addr) - pkt_bld.set_layer_attr("l3", "len", l3_length) + if l3_protocol == 'ipv4': + #('v_hl', 'B', (4 << 4) | (20 >> 2)), + #('tos', 'B', 0), + #('len', 'H', 20), + #('id', 'H', 0), + #('off', 'H', 0), + #('ttl', 'B', 64), + #('p', 'B', 0), + #('sum', 'H', 0), + #('src', '4s', '\x00' * 4), + #('dst', '4s', '\x00' * 4) + pkt_bld.add_pkt_layer('l3', dpkt.ip.IP()) + pkt_bld.set_ip_layer_addr('l3', 'src', ip_src_addr) + pkt_bld.set_ip_layer_addr('l3', 'dst', ip_dst_addr) + pkt_bld.set_layer_attr('l3', 'len', l3_length) else: raise NotImplementedError("l3_protocol '{0}' is not supported by TRex yet.".format(l3_protocol)) - pkt_bld.dump_pkt_to_pcap("stream_test.pcap") + # set l4 on l3 + if l4_protocol not in ALLOWED_L4_PROTOCOL: + raise ValueError('l4_protocol must be one of the following: {0}'.format(ALLOWED_L3_PROTOCOL)) + pkt_bld.set_layer_attr('l3', 'p', ALLOWED_L4_PROTOCOL[l4_protocol]) + + # set l4 attributes + if l4_protocol == 'tcp': + pkt_bld.add_pkt_layer('l4', dpkt.tcp.TCP()) + #('sport', 'H', 0xdead), + #('dport', 'H', 0), + #('seq', 'I', 0xdeadbeefL), + #('ack', 'I', 0), + #('off_x2', 'B', ((5 << 4) | 0)), + #('flags', 'B', TH_SYN), + #('win', 'H', TCP_WIN_MAX), + #('sum', 'H', 0), + #('urp', 'H', 0) + #pkt_bld.set_ip_layer_addr('l4', 'sport', ip_src_addr) + #pkt_bld.set_ip_layer_addr('l4', 'dport', ip_dst_addr) + else: + raise NotImplementedError("l4_protocol '{0}' is not supported by TRex yet.".format(l3_protocol)) + + pkt_bld.set_pkt_payload('Hello, World' + '!'*58) + pkt_bld.dump_pkt_to_pcap('stream_test.pcap') return pkt_bld -if __name__ == "__main__": +if __name__ == '__main__': pass -- cgit 1.2.3-korg