diff options
Diffstat (limited to 'resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py')
| -rwxr-xr-x | resources/traffic_scripts/dhcp/send_dhcp_v6_messages.py | 54 |
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." |