diff options
Diffstat (limited to 'resources')
| -rw-r--r-- | resources/libraries/robot/l2_traffic.robot | 86 | ||||
| -rwxr-xr-x | resources/traffic_scripts/send_ip_icmp.py | 81 |
2 files changed, 144 insertions, 23 deletions
diff --git a/resources/libraries/robot/l2_traffic.robot b/resources/libraries/robot/l2_traffic.robot index bb66d9035d..4ec0624c30 100644 --- a/resources/libraries/robot/l2_traffic.robot +++ b/resources/libraries/robot/l2_traffic.robot @@ -17,13 +17,33 @@ | Library | resources.libraries.python.TrafficScriptExecutor *** Keywords *** -| Send and receive ICMPv4 -| | [Documentation] | Send ICMPv4 echo request from source interface to destination interface. -| | [Arguments] | ${tg_node} | ${src_int} | ${dst_int} +| Send and receive ICMP Packet +| | [Documentation] | Send ICMPv4/ICMPv6 echo request from source interface to +| | ... | destination interface. +| | ... +| | ... | *Arguments:* +| | ... +| | ... | - {tg_node} - TG node. Type: dictionary +| | ... | - {src_int} - Source interface. Type: string +| | ... | - {dst_int} - Destination interface. Type: string +| | ... | - {src_ip} - Source IP address (Optional). Type: string +| | ... | - {dst_ip} - Destination IP address (Optional). Type: string +| | ... +| | ... | *Return:* +| | ... +| | ... | - No value returned +| | ... +| | ... | *Example:* +| | ... +| | ... | _NOTE:_ Default IP is IPv4 +| | ... +| | ... | \| Send and receive ICMP Packet \| ${nodes['TG']} \ +| | ... | \| ${tg_to_dut_if1} \| ${tg_to_dut_if2} \| +| | ... +| | [Arguments] | ${tg_node} | ${src_int} | ${dst_int} | +| | ... | ${src_ip}=192.168.100.1 | ${dst_ip}=192.168.100.2 | | ${src_mac}= | Get Interface Mac | ${tg_node} | ${src_int} | | ${dst_mac}= | Get Interface Mac | ${tg_node} | ${dst_int} -| | ${src_ip}= | Set Variable | 192.168.100.1 -| | ${dst_ip}= | Set Variable | 192.168.100.2 | | ${args}= | Traffic Script Gen Arg | ${dst_int} | ${src_int} | ${src_mac} | | | ... | ${dst_mac} | ${src_ip} | ${dst_ip} | | Run Traffic Script On Node | send_ip_icmp.py | ${tg_node} | ${args} @@ -32,6 +52,56 @@ | | [Documentation] | Send ICMPv4 echo request from both directions, | | ... | from interface1 to interface2 and | | ... | from interface2 to interface1. -| | [Arguments] | ${tg_node} | ${int1} | ${int2} -| | Send and receive ICMPv4 | ${tg_node} | ${int1} | ${int2} -| | Send and receive ICMPv4 | ${tg_node} | ${int2} | ${int1} +| | ... +| | ... | *Arguments:* +| | ... +| | ... | - {tg_node} - TG node. Type: dictionary +| | ... | - {src_int} - Source interface. Type: string +| | ... | - {dst_int} - Destination interface. Type: string +| | ... | - {src_ip} - Source IP address (Optional). Type: string +| | ... | - {dst_ip} - Destination IP address (Optional). Type: string +| | ... +| | ... | *Return:* +| | ... +| | ... | - No value returned +| | ... +| | ... | *Example:* +| | ... +| | ... | \| Send and receive ICMPv4 bidirectionally \| ${nodes['TG']} \ +| | ... | \| ${tg_to_dut_if1} \| ${tg_to_dut_if2} \| +| | ... +| | [Arguments] | ${tg_node} | ${int1} | ${int2} | ${src_ip}=192.168.100.1 | +| | ... | ${dst_ip}=192.168.100.2 +| | Send and receive ICMP Packet | ${tg_node} | ${int1} | ${int2} | +| | ... | ${src_ip} | ${dst_ip} +| | Send and receive ICMP Packet | ${tg_node} | ${int2} | ${int1} | +| | ... | ${dst_ip} | ${src_ip} + +| Send and receive ICMPv6 bidirectionally +| | [Documentation] | Send ICMPv6 echo request from both directions, +| | ... | from interface1 to interface2 and +| | ... | from interface2 to interface1. +| | ... +| | ... | *Arguments:* +| | ... +| | ... | - {tg_node} - TG node. Type: dictionary +| | ... | - {src_int} - Source interface. Type: string +| | ... | - {dst_int} - Destination interface. Type: string +| | ... | - {src_ip} - Source IP address (Optional). Type: string +| | ... | - {dst_ip} - Destination IP address (Optional). Type: string +| | ... +| | ... | *Return:* +| | ... +| | ... | - No value returned +| | ... +| | ... | *Example:* +| | ... +| | ... | \| Send and receive ICMPv6 bidirectionally \| ${nodes['TG']} \ +| | ... | \| ${tg_to_dut_if1} \| ${tg_to_dut_if2} \| +| | ... +| | [Arguments] | ${tg_node} | ${int1} | ${int2} | ${src_ip}=3ffe:63::1 | +| | ... | ${dst_ip}=3ffe:63::2 +| | Send and receive ICMP Packet | ${tg_node} | ${int1} | ${int2} | +| | ... | ${src_ip} | ${dst_ip} +| | Send and receive ICMP Packet | ${tg_node} | ${int2} | ${int1} | +| | ... | ${dst_ip} | ${src_ip}
\ No newline at end of file diff --git a/resources/traffic_scripts/send_ip_icmp.py b/resources/traffic_scripts/send_ip_icmp.py index 711bf3dd25..23e647f43c 100755 --- a/resources/traffic_scripts/send_ip_icmp.py +++ b/resources/traffic_scripts/send_ip_icmp.py @@ -12,18 +12,57 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Traffic script that sends an ip icmp packet -from one interface to the other""" +"""Traffic script that sends an IP ICMPv4/ICMPv6 packet +from one interface to the other one.""" import sys -from resources.libraries.python.PacketVerifier import RxQueue, TxQueue -from resources.libraries.python.TrafficScriptArg import TrafficScriptArg +import ipaddress + from scapy.layers.inet import ICMP, IP from scapy.all import Ether +from scapy.layers.inet6 import ICMPv6EchoRequest +from scapy.layers.inet6 import IPv6 + +from resources.libraries.python.PacketVerifier import RxQueue, TxQueue +from resources.libraries.python.TrafficScriptArg import TrafficScriptArg + + +def valid_ipv4(ip): + """Check if IP address has the correct IPv4 address format. + + :param ip: IP address. + :type ip: str + :return: True in case of correct IPv4 address format, + otherwise return false. + :rtype: bool + """ + try: + ipaddress.IPv4Address(unicode(ip)) + return True + except (AttributeError, ipaddress.AddressValueError): + return False + + +def valid_ipv6(ip): + """Check if IP address has the correct IPv6 address format. + + :param ip: IP address. + :type ip: str + :return: True in case of correct IPv6 address format, + otherwise return false. + :rtype: bool + """ + try: + ipaddress.IPv6Address(unicode(ip)) + return True + except (AttributeError, ipaddress.AddressValueError): + return False def main(): - """ Send IP icmp packet from one traffic generator interface to the other""" + """Send IP ICMPv4/ICMPv6 packet from one traffic generator interface to + the other one. + """ args = TrafficScriptArg(['src_mac', 'dst_mac', 'src_ip', 'dst_ip']) src_mac = args.get_arg('src_mac') @@ -37,11 +76,23 @@ def main(): txq = TxQueue(tx_if) sent_packets = [] - + ip_format = '' + icmp_format = '' # Create empty ip ICMP packet and add padding before sending - pkt_raw = Ether(src=src_mac, dst=dst_mac) / \ - IP(src=src_ip, dst=dst_ip) / \ - ICMP() + if valid_ipv4(src_ip) and valid_ipv4(dst_ip): + pkt_raw = (Ether(src=src_mac, dst=dst_mac) / + IP(src=src_ip, dst=dst_ip) / + ICMP()) + ip_format = 'IP' + icmp_format = 'ICMP' + elif valid_ipv6(src_ip) and valid_ipv6(dst_ip): + pkt_raw = (Ether(src=src_mac, dst=dst_mac) / + IPv6(src=src_ip, dst=dst_ip) / + ICMPv6EchoRequest()) + ip_format = 'IPv6' + icmp_format = 'ICMPv6EchoRequest' + else: + raise ValueError("IP(s) not in correct format") # Send created packet on one interface and receive on the other sent_packets.append(pkt_raw) @@ -53,13 +104,13 @@ def main(): if ether is None: raise RuntimeError('ICMP echo Rx timeout') - if not ether.haslayer(IP): - raise RuntimeError( - 'Not an IP packet received {0}'.format(ether.__repr__())) + if not ether.haslayer(ip_format): + raise RuntimeError('Not an IP packet received {0}' + .format(ether.__repr__())) - if not ether.haslayer(ICMP): - raise RuntimeError( - 'Not an ICMP packet received {0}'.format(ether.__repr__())) + if not ether.haslayer(icmp_format): + raise RuntimeError('Not an ICMP packet received {0}' + .format(ether.__repr__())) sys.exit(0) |