aboutsummaryrefslogtreecommitdiffstats
path: root/resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py
diff options
context:
space:
mode:
Diffstat (limited to 'resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py')
-rwxr-xr-xresources/traffic_scripts/dhcp/send_dhcp_v6_messages.py54
1 files changed, 42 insertions, 12 deletions
diff --git a/resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py b/resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py
index 50dec47bb8..09742565ee 100755
--- a/resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py
+++ b/resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py
@@ -30,6 +30,32 @@ def _check_udp_checksum(pkt):
return new['UDP'].chksum == pkt['UDP'].chksum
+def _get_dhcpv6_msgtype(msg_index):
+ """Return DHCPv6 message type string.
+
+ :param msg_index: Index of message type.
+ :return: Message type.
+ :type msg_index: int
+ :rtype msg_str: str
+ """
+ dhcp6_messages = {
+ 1: "SOLICIT",
+ 2: "ADVERTISE",
+ 3: "REQUEST",
+ 4: "CONFIRM",
+ 5: "RENEW",
+ 6: "REBIND",
+ 7: "REPLY",
+ 8: "RELEASE",
+ 9: "DECLINE",
+ 10: "RECONFIGURE",
+ 11: "INFORMATION-REQUEST",
+ 12: "RELAY-FORW",
+ 13: "RELAY-REPL"
+ }
+ return dhcp6_messages[msg_index]
+
+
def dhcpv6_solicit(tx_if, rx_if, dhcp_multicast_ip, link_local_ip, proxy_ip,
server_ip, server_mac, client_duid, client_mac):
"""Send and check DHCPv6 SOLICIT proxy packet.
@@ -91,11 +117,12 @@ def dhcpv6_solicit(tx_if, rx_if, dhcp_multicast_ip, link_local_ip, proxy_ip,
ether['IPv6'].dst, server_ip))
print "Destination IP address: OK."
- if ether['IPv6']['UDP']\
- ['DHCPv6 Relay Forward Message (Relay Agent/Server Message)']:
- print "Relay Agent/Server Message: OK."
- else:
- raise RuntimeError("Relay Agent/Server Message error.")
+ msgtype = _get_dhcpv6_msgtype(ether['IPv6']['UDP']
+ ['DHCPv6 Relay Forward Message (Relay Agent/Server Message)'].msgtype)
+ if msgtype != 'RELAY-FORW':
+ raise RuntimeError("Message type error: {} != RELAY-FORW".format(
+ msgtype))
+ print "Message type: OK."
linkaddr = ether['IPv6']['UDP']\
['DHCPv6 Relay Forward Message (Relay Agent/Server Message)'].linkaddr
@@ -168,7 +195,8 @@ def dhcpv6_advertise(rx_if, tx_if, link_local_ip, proxy_ip,
raise RuntimeError("Checksum error!")
print "Checksum: OK."
- msgtype = ether['IPv6']['UDP']['DHCPv6 Advertise Message'].msgtype
+ msgtype = _get_dhcpv6_msgtype(ether['IPv6']['UDP']
+ ['DHCPv6 Advertise Message'].msgtype)
if msgtype != 'ADVERTISE':
raise RuntimeError("Message type error: {} != ADVERTISE".format(
msgtype))
@@ -227,11 +255,12 @@ def dhcpv6_request(tx_if, rx_if, dhcp_multicast_ip, link_local_ip, proxy_ip,
ether['IPv6'].dst, server_ip))
print "Destination IP address: OK."
- if ether['IPv6']['UDP']\
- ['DHCPv6 Relay Forward Message (Relay Agent/Server Message)']:
- print "Relay Agent/Server Message: OK."
- else:
- raise RuntimeError("Relay Agent/Server Message error.")
+ msgtype = _get_dhcpv6_msgtype(ether['IPv6']['UDP']
+ ['DHCPv6 Relay Forward Message (Relay Agent/Server Message)'].msgtype)
+ if msgtype != 'RELAY-FORW':
+ raise RuntimeError("Message type error: {} != RELAY-FORW".format(
+ msgtype))
+ print "Message type: OK."
linkaddr = ether['IPv6']['UDP']\
['DHCPv6 Relay Forward Message (Relay Agent/Server Message)'].linkaddr
@@ -293,7 +322,8 @@ def dhcpv6_reply(rx_if, tx_if, link_local_ip, proxy_ip, server_ip, server_mac,
raise RuntimeError("Checksum error!")
print "Checksum: OK."
- msgtype = ether['IPv6']['UDP']['DHCPv6 Reply Message'].msgtype
+ msgtype = _get_dhcpv6_msgtype(ether['IPv6']['UDP']
+ ['DHCPv6 Reply Message'].msgtype)
if msgtype != 'REPLY':
raise RuntimeError("Message type error: {} != REPLY".format(msgtype))
print "Message type: OK."