Python3: resources and libraries
[csit.git] / resources / traffic_scripts / dhcp / send_and_check_proxy_discover.py
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 (executable)
index d808971..0000000
+++ /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()