X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Ftraffic_scripts%2Fdhcp%2Fsend_and_check_proxy_discover.py;fp=resources%2Ftraffic_scripts%2Fdhcp%2Fsend_and_check_proxy_discover.py;h=0000000000000000000000000000000000000000;hp=d8089713fd10643f92d178bbe024cf4b04ed7e94;hb=d68951ac245150eeefa6e0f4156e4c1b5c9e9325;hpb=ed0258a440cfad7023d643f717ab78ac568dc59b diff --git a/resources/traffic_scripts/dhcp/send_and_check_proxy_discover.py b/resources/traffic_scripts/dhcp/send_and_check_proxy_discover.py deleted file mode 100755 index d8089713fd..0000000000 --- a/resources/traffic_scripts/dhcp/send_and_check_proxy_discover.py +++ /dev/null @@ -1,79 +0,0 @@ -#!/usr/bin/env python -# Copyright (c) 2016 Cisco and/or its affiliates. -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at: -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -"""Traffic script that sends DHCP DISCOVER packet - and check if is received on interface.""" - -import sys - -from scapy.all import Ether -from scapy.layers.inet import IP, UDP -from scapy.layers.inet import UDP_SERVICES -from scapy.layers.dhcp import DHCP, BOOTP - -from resources.libraries.python.PacketVerifier import RxQueue, TxQueue -from resources.libraries.python.TrafficScriptArg import TrafficScriptArg - - -def is_discover(pkt): - """If DHCP message type option is set to dhcp discover return True, - else return False. False is returned also if exception occurs.""" - dhcp_discover = 1 - try: - dhcp_options = pkt['BOOTP']['DHCP options'].options - message_type = filter(lambda x: x[0] == 'message-type', - dhcp_options) - message_type = message_type[0][1] - return message_type == dhcp_discover - except: - return False - - -def main(): - """Send DHCP DISCOVER packet.""" - - args = TrafficScriptArg(['tx_src_ip', 'tx_dst_ip']) - - tx_if = args.get_arg('tx_if') - rx_if = args.get_arg('rx_if') - - rxq = RxQueue(rx_if) - txq = TxQueue(tx_if) - - tx_src_ip = args.get_arg('tx_src_ip') - tx_dst_ip = args.get_arg('tx_dst_ip') - - sent_packets = [] - - dhcp_discover = Ether(dst="ff:ff:ff:ff:ff:ff") / \ - IP(src=tx_src_ip, dst=tx_dst_ip) / \ - UDP(sport=UDP_SERVICES.bootpc, dport=UDP_SERVICES.bootps) / \ - BOOTP(op=1,) / \ - DHCP(options=[("message-type", "discover"), - "end"]) - - sent_packets.append(dhcp_discover) - txq.send(dhcp_discover) - - for _ in range(10): - dhcp_discover = rxq.recv(2) - if is_discover(dhcp_discover): - break - else: - raise RuntimeError("DHCP DISCOVER Rx timeout") - - sys.exit(0) - -if __name__ == "__main__": - main()