Rework network library for new zkhandler

This commit is contained in:
Joshua Boniface 2021-05-29 20:53:42 -04:00
parent b186a75b4e
commit 62cb72b62f
1 changed files with 155 additions and 170 deletions

View File

@ -21,28 +21,24 @@
import re import re
from kazoo.exceptions import NoNodeError
import daemon_lib.zkhandler as zkhandler
# #
# Cluster search functions # Cluster search functions
# #
def getClusterNetworkList(zk_conn): def getClusterNetworkList(zkhandler):
# Get a list of VNIs by listing the children of /networks # Get a list of VNIs by listing the children of /networks
vni_list = zkhandler.listchildren(zk_conn, '/networks') vni_list = zkhandler.children('/networks')
description_list = [] description_list = []
# For each VNI, get the corresponding description from the data # For each VNI, get the corresponding description from the data
for vni in vni_list: for vni in vni_list:
description_list.append(zkhandler.readdata(zk_conn, '/networks/{}'.format(vni))) description_list.append(zkhandler.read('/networks/{}'.format(vni)))
return vni_list, description_list return vni_list, description_list
def searchClusterByVNI(zk_conn, vni): def searchClusterByVNI(zkhandler, vni):
try: try:
# Get the lists # Get the lists
vni_list, description_list = getClusterNetworkList(zk_conn) vni_list, description_list = getClusterNetworkList(zkhandler)
# We're looking for UUID, so find that element ID # We're looking for UUID, so find that element ID
index = vni_list.index(vni) index = vni_list.index(vni)
# Get the name_list element at that index # Get the name_list element at that index
@ -54,10 +50,10 @@ def searchClusterByVNI(zk_conn, vni):
return description return description
def searchClusterByDescription(zk_conn, description): def searchClusterByDescription(zkhandler, description):
try: try:
# Get the lists # Get the lists
vni_list, description_list = getClusterNetworkList(zk_conn) vni_list, description_list = getClusterNetworkList(zkhandler)
# We're looking for name, so find that element ID # We're looking for name, so find that element ID
index = description_list.index(description) index = description_list.index(description)
# Get the uuid_list element at that index # Get the uuid_list element at that index
@ -69,43 +65,43 @@ def searchClusterByDescription(zk_conn, description):
return vni return vni
def getNetworkVNI(zk_conn, network): def getNetworkVNI(zkhandler, network):
# Validate and obtain alternate passed value # Validate and obtain alternate passed value
if network.isdigit(): if network.isdigit():
net_description = searchClusterByVNI(zk_conn, network) net_description = searchClusterByVNI(zkhandler, network)
net_vni = searchClusterByDescription(zk_conn, net_description) net_vni = searchClusterByDescription(zkhandler, net_description)
else: else:
net_vni = searchClusterByDescription(zk_conn, network) net_vni = searchClusterByDescription(zkhandler, network)
net_description = searchClusterByVNI(zk_conn, net_vni) net_description = searchClusterByVNI(zkhandler, net_vni)
return net_vni return net_vni
def getNetworkDescription(zk_conn, network): def getNetworkDescription(zkhandler, network):
# Validate and obtain alternate passed value # Validate and obtain alternate passed value
if network.isdigit(): if network.isdigit():
net_description = searchClusterByVNI(zk_conn, network) net_description = searchClusterByVNI(zkhandler, network)
net_vni = searchClusterByDescription(zk_conn, net_description) net_vni = searchClusterByDescription(zkhandler, net_description)
else: else:
net_vni = searchClusterByDescription(zk_conn, network) net_vni = searchClusterByDescription(zkhandler, network)
net_description = searchClusterByVNI(zk_conn, net_vni) net_description = searchClusterByVNI(zkhandler, net_vni)
return net_description return net_description
def getNetworkDHCPLeases(zk_conn, vni): def getNetworkDHCPLeases(zkhandler, vni):
# Get a list of DHCP leases by listing the children of /networks/<vni>/dhcp4_leases # Get a list of DHCP leases by listing the children of /networks/<vni>/dhcp4_leases
dhcp4_leases = zkhandler.listchildren(zk_conn, '/networks/{}/dhcp4_leases'.format(vni)) dhcp4_leases = zkhandler.children('/networks/{}/dhcp4_leases'.format(vni))
return sorted(dhcp4_leases) return sorted(dhcp4_leases)
def getNetworkDHCPReservations(zk_conn, vni): def getNetworkDHCPReservations(zkhandler, vni):
# Get a list of DHCP reservations by listing the children of /networks/<vni>/dhcp4_reservations # Get a list of DHCP reservations by listing the children of /networks/<vni>/dhcp4_reservations
dhcp4_reservations = zkhandler.listchildren(zk_conn, '/networks/{}/dhcp4_reservations'.format(vni)) dhcp4_reservations = zkhandler.children('/networks/{}/dhcp4_reservations'.format(vni))
return sorted(dhcp4_reservations) return sorted(dhcp4_reservations)
def getNetworkACLs(zk_conn, vni, _direction): def getNetworkACLs(zkhandler, vni, _direction):
# Get the (sorted) list of active ACLs # Get the (sorted) list of active ACLs
if _direction == 'both': if _direction == 'both':
directions = ['in', 'out'] directions = ['in', 'out']
@ -114,32 +110,32 @@ def getNetworkACLs(zk_conn, vni, _direction):
full_acl_list = [] full_acl_list = []
for direction in directions: for direction in directions:
unordered_acl_list = zkhandler.listchildren(zk_conn, '/networks/{}/firewall_rules/{}'.format(vni, direction)) unordered_acl_list = zkhandler.children('/networks/{}/firewall_rules/{}'.format(vni, direction))
ordered_acls = dict() ordered_acls = dict()
for acl in unordered_acl_list: for acl in unordered_acl_list:
order = zkhandler.readdata(zk_conn, '/networks/{}/firewall_rules/{}/{}/order'.format(vni, direction, acl)) order = zkhandler.read('/networks/{}/firewall_rules/{}/{}/order'.format(vni, direction, acl))
ordered_acls[order] = acl ordered_acls[order] = acl
for order in sorted(ordered_acls.keys()): for order in sorted(ordered_acls.keys()):
rule = zkhandler.readdata(zk_conn, '/networks/{}/firewall_rules/{}/{}/rule'.format(vni, direction, acl)) rule = zkhandler.read('/networks/{}/firewall_rules/{}/{}/rule'.format(vni, direction, acl))
full_acl_list.append({'direction': direction, 'order': int(order), 'description': ordered_acls[order], 'rule': rule}) full_acl_list.append({'direction': direction, 'order': int(order), 'description': ordered_acls[order], 'rule': rule})
return full_acl_list return full_acl_list
def getNetworkInformation(zk_conn, vni): def getNetworkInformation(zkhandler, vni):
description = zkhandler.readdata(zk_conn, '/networks/{}'.format(vni)) description = zkhandler.read('/networks/{}'.format(vni))
nettype = zkhandler.readdata(zk_conn, '/networks/{}/nettype'.format(vni)) nettype = zkhandler.read('/networks/{}/nettype'.format(vni))
domain = zkhandler.readdata(zk_conn, '/networks/{}/domain'.format(vni)) domain = zkhandler.read('/networks/{}/domain'.format(vni))
name_servers = zkhandler.readdata(zk_conn, '/networks/{}/name_servers'.format(vni)) name_servers = zkhandler.read('/networks/{}/name_servers'.format(vni))
ip6_network = zkhandler.readdata(zk_conn, '/networks/{}/ip6_network'.format(vni)) ip6_network = zkhandler.read('/networks/{}/ip6_network'.format(vni))
ip6_gateway = zkhandler.readdata(zk_conn, '/networks/{}/ip6_gateway'.format(vni)) ip6_gateway = zkhandler.read('/networks/{}/ip6_gateway'.format(vni))
dhcp6_flag = zkhandler.readdata(zk_conn, '/networks/{}/dhcp6_flag'.format(vni)) dhcp6_flag = zkhandler.read('/networks/{}/dhcp6_flag'.format(vni))
ip4_network = zkhandler.readdata(zk_conn, '/networks/{}/ip4_network'.format(vni)) ip4_network = zkhandler.read('/networks/{}/ip4_network'.format(vni))
ip4_gateway = zkhandler.readdata(zk_conn, '/networks/{}/ip4_gateway'.format(vni)) ip4_gateway = zkhandler.read('/networks/{}/ip4_gateway'.format(vni))
dhcp4_flag = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_flag'.format(vni)) dhcp4_flag = zkhandler.read('/networks/{}/dhcp4_flag'.format(vni))
dhcp4_start = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_start'.format(vni)) dhcp4_start = zkhandler.read('/networks/{}/dhcp4_start'.format(vni))
dhcp4_end = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_end'.format(vni)) dhcp4_end = zkhandler.read('/networks/{}/dhcp4_end'.format(vni))
# Construct a data structure to represent the data # Construct a data structure to represent the data
network_information = { network_information = {
@ -164,19 +160,19 @@ def getNetworkInformation(zk_conn, vni):
return network_information return network_information
def getDHCPLeaseInformation(zk_conn, vni, mac_address): def getDHCPLeaseInformation(zkhandler, vni, mac_address):
# Check whether this is a dynamic or static lease # Check whether this is a dynamic or static lease
try: if zkhandler.exists('/networks/{}/dhcp4_leases/{}'.format(vni, mac_address)):
zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_leases/{}'.format(vni, mac_address))
type_key = 'dhcp4_leases' type_key = 'dhcp4_leases'
except NoNodeError: elif zkhandler.exists('/networks/{}/dhcp4_reservations/{}'.format(vni, mac_address)):
zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_reservations/{}'.format(vni, mac_address))
type_key = 'dhcp4_reservations' type_key = 'dhcp4_reservations'
else:
return {}
hostname = zkhandler.readdata(zk_conn, '/networks/{}/{}/{}/hostname'.format(vni, type_key, mac_address)) hostname = zkhandler.read('/networks/{}/{}/{}/hostname'.format(vni, type_key, mac_address))
ip4_address = zkhandler.readdata(zk_conn, '/networks/{}/{}/{}/ipaddr'.format(vni, type_key, mac_address)) ip4_address = zkhandler.read('/networks/{}/{}/{}/ipaddr'.format(vni, type_key, mac_address))
if type_key == 'dhcp4_leases': if type_key == 'dhcp4_leases':
timestamp = zkhandler.readdata(zk_conn, '/networks/{}/{}/{}/expiry'.format(vni, type_key, mac_address)) timestamp = zkhandler.read('/networks/{}/{}/{}/expiry'.format(vni, type_key, mac_address))
else: else:
timestamp = 'static' timestamp = 'static'
@ -190,9 +186,9 @@ def getDHCPLeaseInformation(zk_conn, vni, mac_address):
return lease_information return lease_information
def getACLInformation(zk_conn, vni, direction, description): def getACLInformation(zkhandler, vni, direction, description):
order = zkhandler.readdata(zk_conn, '/networks/{}/firewall_rules/{}/{}/order'.format(vni, direction, description)) order = zkhandler.read('/networks/{}/firewall_rules/{}/{}/order'.format(vni, direction, description))
rule = zkhandler.readdata(zk_conn, '/networks/{}/firewall_rules/{}/{}/rule'.format(vni, direction, description)) rule = zkhandler.read('/networks/{}/firewall_rules/{}/{}/rule'.format(vni, direction, description))
# Construct a data structure to represent the data # Construct a data structure to represent the data
acl_information = { acl_information = {
@ -235,7 +231,7 @@ def isValidIP(ipaddr):
# #
# Direct functions # Direct functions
# #
def add_network(zk_conn, vni, description, nettype, def add_network(zkhandler, vni, description, nettype,
domain, name_servers, ip4_network, ip4_gateway, ip6_network, ip6_gateway, domain, name_servers, ip4_network, ip4_gateway, ip6_network, ip6_gateway,
dhcp4_flag, dhcp4_start, dhcp4_end): dhcp4_flag, dhcp4_start, dhcp4_end):
# Ensure start and end DHCP ranges are set if the flag is set # Ensure start and end DHCP ranges are set if the flag is set
@ -243,10 +239,10 @@ def add_network(zk_conn, vni, description, nettype,
return False, 'ERROR: DHCPv4 start and end addresses are required for a DHCPv4-enabled network.' return False, 'ERROR: DHCPv4 start and end addresses are required for a DHCPv4-enabled network.'
# Check if a network with this VNI or description already exists # Check if a network with this VNI or description already exists
if zkhandler.exists(zk_conn, '/networks/{}'.format(vni)): if zkhandler.exists('/networks/{}'.format(vni)):
return False, 'ERROR: A network with VNI "{}" already exists!'.format(vni) return False, 'ERROR: A network with VNI "{}" already exists!'.format(vni)
for network in zkhandler.listchildren(zk_conn, '/networks'): for network in zkhandler.children('/networks'):
network_description = zkhandler.readdata(zk_conn, '/networks/{}'.format(network)) network_description = zkhandler.read('/networks/{}'.format(network))
if network_description == description: if network_description == description:
return False, 'ERROR: A network with description "{}" already exists!'.format(description) return False, 'ERROR: A network with description "{}" already exists!'.format(description)
@ -263,87 +259,87 @@ def add_network(zk_conn, vni, description, nettype,
domain = '{}.local'.format(description) domain = '{}.local'.format(description)
# Add the new network to Zookeeper # Add the new network to Zookeeper
zkhandler.writedata(zk_conn, { zkhandler.write([
'/networks/{}'.format(vni): description, ('/networks/{}'.format(vni), description),
'/networks/{}/nettype'.format(vni): nettype, ('/networks/{}/nettype'.format(vni), nettype),
'/networks/{}/domain'.format(vni): domain, ('/networks/{}/domain'.format(vni), domain),
'/networks/{}/name_servers'.format(vni): name_servers, ('/networks/{}/name_servers'.format(vni), name_servers),
'/networks/{}/ip6_network'.format(vni): ip6_network, ('/networks/{}/ip6_network'.format(vni), ip6_network),
'/networks/{}/ip6_gateway'.format(vni): ip6_gateway, ('/networks/{}/ip6_gateway'.format(vni), ip6_gateway),
'/networks/{}/dhcp6_flag'.format(vni): dhcp6_flag, ('/networks/{}/dhcp6_flag'.format(vni), dhcp6_flag),
'/networks/{}/ip4_network'.format(vni): ip4_network, ('/networks/{}/ip4_network'.format(vni), ip4_network),
'/networks/{}/ip4_gateway'.format(vni): ip4_gateway, ('/networks/{}/ip4_gateway'.format(vni), ip4_gateway),
'/networks/{}/dhcp4_flag'.format(vni): dhcp4_flag, ('/networks/{}/dhcp4_flag'.format(vni), dhcp4_flag),
'/networks/{}/dhcp4_start'.format(vni): dhcp4_start, ('/networks/{}/dhcp4_start'.format(vni), dhcp4_start),
'/networks/{}/dhcp4_end'.format(vni): dhcp4_end, ('/networks/{}/dhcp4_end'.format(vni), dhcp4_end),
'/networks/{}/dhcp4_leases'.format(vni): '', ('/networks/{}/dhcp4_leases'.format(vni), ''),
'/networks/{}/dhcp4_reservations'.format(vni): '', ('/networks/{}/dhcp4_reservations'.format(vni), ''),
'/networks/{}/firewall_rules'.format(vni): '', ('/networks/{}/firewall_rules'.format(vni), ''),
'/networks/{}/firewall_rules/in'.format(vni): '', ('/networks/{}/firewall_rules/in'.format(vni), ''),
'/networks/{}/firewall_rules/out'.format(vni): '' ('/networks/{}/firewall_rules/out'.format(vni), '')
}) ])
return True, 'Network "{}" added successfully!'.format(description) return True, 'Network "{}" added successfully!'.format(description)
def modify_network(zk_conn, vni, description=None, domain=None, name_servers=None, def modify_network(zkhandler, vni, description=None, domain=None, name_servers=None,
ip4_network=None, ip4_gateway=None, ip6_network=None, ip6_gateway=None, ip4_network=None, ip4_gateway=None, ip6_network=None, ip6_gateway=None,
dhcp4_flag=None, dhcp4_start=None, dhcp4_end=None): dhcp4_flag=None, dhcp4_start=None, dhcp4_end=None):
# Add the modified parameters to Zookeeper # Add the modified parameters to Zookeeper
zk_data = dict() update_data = list()
if description is not None: if description is not None:
zk_data.update({'/networks/{}'.format(vni): description}) update_data.append(('/networks/{}'.format(vni), description))
if domain is not None: if domain is not None:
zk_data.update({'/networks/{}/domain'.format(vni): domain}) update_data.append(('/networks/{}/domain'.format(vni), domain))
if name_servers is not None: if name_servers is not None:
zk_data.update({'/networks/{}/name_servers'.format(vni): name_servers}) update_data.append(('/networks/{}/name_servers'.format(vni), name_servers))
if ip4_network is not None: if ip4_network is not None:
zk_data.update({'/networks/{}/ip4_network'.format(vni): ip4_network}) update_data.append(('/networks/{}/ip4_network'.format(vni), ip4_network))
if ip4_gateway is not None: if ip4_gateway is not None:
zk_data.update({'/networks/{}/ip4_gateway'.format(vni): ip4_gateway}) update_data.append(('/networks/{}/ip4_gateway'.format(vni), ip4_gateway))
if ip6_network is not None: if ip6_network is not None:
zk_data.update({'/networks/{}/ip6_network'.format(vni): ip6_network}) update_data.append(('/networks/{}/ip6_network'.format(vni), ip6_network))
if ip6_network: if ip6_network:
zk_data.update({'/networks/{}/dhcp6_flag'.format(vni): 'True'}) update_data.append(('/networks/{}/dhcp6_flag'.format(vni), 'True'))
else: else:
zk_data.update({'/networks/{}/dhcp6_flag'.format(vni): 'False'}) update_data.append(('/networks/{}/dhcp6_flag'.format(vni), 'False'))
if ip6_gateway is not None: if ip6_gateway is not None:
zk_data.update({'/networks/{}/ip6_gateway'.format(vni): ip6_gateway}) update_data.append(('/networks/{}/ip6_gateway'.format(vni), ip6_gateway))
else: else:
# If we're changing the network, but don't also specify the gateway, # If we're changing the network, but don't also specify the gateway,
# generate a new one automatically # generate a new one automatically
if ip6_network: if ip6_network:
ip6_netpart, ip6_maskpart = ip6_network.split('/') ip6_netpart, ip6_maskpart = ip6_network.split('/')
ip6_gateway = '{}1'.format(ip6_netpart) ip6_gateway = '{}1'.format(ip6_netpart)
zk_data.update({'/networks/{}/ip6_gateway'.format(vni): ip6_gateway}) update_data.append(('/networks/{}/ip6_gateway'.format(vni), ip6_gateway))
if dhcp4_flag is not None: if dhcp4_flag is not None:
zk_data.update({'/networks/{}/dhcp4_flag'.format(vni): dhcp4_flag}) update_data.append(('/networks/{}/dhcp4_flag'.format(vni), dhcp4_flag))
if dhcp4_start is not None: if dhcp4_start is not None:
zk_data.update({'/networks/{}/dhcp4_start'.format(vni): dhcp4_start}) update_data.append(('/networks/{}/dhcp4_start'.format(vni), dhcp4_start))
if dhcp4_end is not None: if dhcp4_end is not None:
zk_data.update({'/networks/{}/dhcp4_end'.format(vni): dhcp4_end}) update_data.append(('/networks/{}/dhcp4_end'.format(vni), dhcp4_end))
zkhandler.writedata(zk_conn, zk_data) zkhandler.write(update_data)
return True, 'Network "{}" modified successfully!'.format(vni) return True, 'Network "{}" modified successfully!'.format(vni)
def remove_network(zk_conn, network): def remove_network(zkhandler, network):
# Validate and obtain alternate passed value # Validate and obtain alternate passed value
vni = getNetworkVNI(zk_conn, network) vni = getNetworkVNI(zkhandler, network)
description = getNetworkDescription(zk_conn, network) description = getNetworkDescription(zkhandler, network)
if not vni: if not vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
# Delete the configuration # Delete the configuration
zkhandler.deletekey(zk_conn, '/networks/{}'.format(vni)) zkhandler.delete('/networks/{}'.format(vni))
return True, 'Network "{}" removed successfully!'.format(description) return True, 'Network "{}" removed successfully!'.format(description)
def add_dhcp_reservation(zk_conn, network, ipaddress, macaddress, hostname): def add_dhcp_reservation(zkhandler, network, ipaddress, macaddress, hostname):
# Validate and obtain standard passed value # Validate and obtain standard passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
@ -356,45 +352,45 @@ def add_dhcp_reservation(zk_conn, network, ipaddress, macaddress, hostname):
if not isValidIP(ipaddress): if not isValidIP(ipaddress):
return False, 'ERROR: IP address "{}" is not valid!'.format(macaddress) return False, 'ERROR: IP address "{}" is not valid!'.format(macaddress)
if zkhandler.exists(zk_conn, '/networks/{}/dhcp4_reservations/{}'.format(net_vni, macaddress)): if zkhandler.exists('/networks/{}/dhcp4_reservations/{}'.format(net_vni, macaddress)):
return False, 'ERROR: A reservation with MAC "{}" already exists!'.format(macaddress) return False, 'ERROR: A reservation with MAC "{}" already exists!'.format(macaddress)
# Add the new static lease to ZK # Add the new static lease to ZK
try: try:
zkhandler.writedata(zk_conn, { zkhandler.write([
'/networks/{}/dhcp4_reservations/{}'.format(net_vni, macaddress): 'static', ('/networks/{}/dhcp4_reservations/{}'.format(net_vni, macaddress), 'static'),
'/networks/{}/dhcp4_reservations/{}/hostname'.format(net_vni, macaddress): hostname, ('/networks/{}/dhcp4_reservations/{}/hostname'.format(net_vni, macaddress), hostname),
'/networks/{}/dhcp4_reservations/{}/ipaddr'.format(net_vni, macaddress): ipaddress ('/networks/{}/dhcp4_reservations/{}/ipaddr'.format(net_vni, macaddress), ipaddress)
}) ])
except Exception as e: except Exception as e:
return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e) return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e)
return True, 'DHCP reservation "{}" added successfully!'.format(macaddress) return True, 'DHCP reservation "{}" added successfully!'.format(macaddress)
def remove_dhcp_reservation(zk_conn, network, reservation): def remove_dhcp_reservation(zkhandler, network, reservation):
# Validate and obtain standard passed value # Validate and obtain standard passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
match_description = '' match_description = ''
# Check if the reservation matches a static reservation description, a mac, or an IP address currently in the database # Check if the reservation matches a static reservation description, a mac, or an IP address currently in the database
dhcp4_reservations_list = getNetworkDHCPReservations(zk_conn, net_vni) dhcp4_reservations_list = getNetworkDHCPReservations(zkhandler, net_vni)
for macaddr in dhcp4_reservations_list: for macaddr in dhcp4_reservations_list:
hostname = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_reservations/{}/hostname'.format(net_vni, macaddr)) hostname = zkhandler.read('/networks/{}/dhcp4_reservations/{}/hostname'.format(net_vni, macaddr))
ipaddress = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_reservations/{}/ipaddr'.format(net_vni, macaddr)) ipaddress = zkhandler.read('/networks/{}/dhcp4_reservations/{}/ipaddr'.format(net_vni, macaddr))
if reservation == macaddr or reservation == hostname or reservation == ipaddress: if reservation == macaddr or reservation == hostname or reservation == ipaddress:
match_description = macaddr match_description = macaddr
lease_type_zk = 'reservations' lease_type_zk = 'reservations'
lease_type_human = 'static reservation' lease_type_human = 'static reservation'
# Check if the reservation matches a dynamic reservation description, a mac, or an IP address currently in the database # Check if the reservation matches a dynamic reservation description, a mac, or an IP address currently in the database
dhcp4_leases_list = getNetworkDHCPLeases(zk_conn, net_vni) dhcp4_leases_list = getNetworkDHCPLeases(zkhandler, net_vni)
for macaddr in dhcp4_leases_list: for macaddr in dhcp4_leases_list:
hostname = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_leases/{}/hostname'.format(net_vni, macaddr)) hostname = zkhandler.read('/networks/{}/dhcp4_leases/{}/hostname'.format(net_vni, macaddr))
ipaddress = zkhandler.readdata(zk_conn, '/networks/{}/dhcp4_leases/{}/ipaddr'.format(net_vni, macaddr)) ipaddress = zkhandler.read('/networks/{}/dhcp4_leases/{}/ipaddr'.format(net_vni, macaddr))
if reservation == macaddr or reservation == hostname or reservation == ipaddress: if reservation == macaddr or reservation == hostname or reservation == ipaddress:
match_description = macaddr match_description = macaddr
lease_type_zk = 'leases' lease_type_zk = 'leases'
@ -404,23 +400,20 @@ def remove_dhcp_reservation(zk_conn, network, reservation):
return False, 'ERROR: No DHCP reservation or lease exists matching "{}"!'.format(reservation) return False, 'ERROR: No DHCP reservation or lease exists matching "{}"!'.format(reservation)
# Remove the entry from zookeeper # Remove the entry from zookeeper
try: zkhandler.delete('/networks/{}/dhcp4_{}/{}'.format(net_vni, lease_type_zk, match_description))
zkhandler.deletekey(zk_conn, '/networks/{}/dhcp4_{}/{}'.format(net_vni, lease_type_zk, match_description))
except Exception:
return False, 'ERROR: Failed to write to Zookeeper!'
return True, 'DHCP {} "{}" removed successfully!'.format(lease_type_human, match_description) return True, 'DHCP {} "{}" removed successfully!'.format(lease_type_human, match_description)
def add_acl(zk_conn, network, direction, description, rule, order): def add_acl(zkhandler, network, direction, description, rule, order):
# Validate and obtain standard passed value # Validate and obtain standard passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
# Check if the ACL matches a description currently in the database # Check if the ACL matches a description currently in the database
match_description = '' match_description = ''
full_acl_list = getNetworkACLs(zk_conn, net_vni, 'both') full_acl_list = getNetworkACLs(zkhandler, net_vni, 'both')
for acl in full_acl_list: for acl in full_acl_list:
if acl['description'] == description: if acl['description'] == description:
match_description = acl['description'] match_description = acl['description']
@ -435,7 +428,7 @@ def add_acl(zk_conn, network, direction, description, rule, order):
direction = "out" direction = "out"
# Handle reordering # Handle reordering
full_acl_list = getNetworkACLs(zk_conn, net_vni, direction) full_acl_list = getNetworkACLs(zkhandler, net_vni, direction)
acl_list_length = len(full_acl_list) acl_list_length = len(full_acl_list)
# Set order to len # Set order to len
if not order or int(order) > acl_list_length: if not order or int(order) > acl_list_length:
@ -448,44 +441,37 @@ def add_acl(zk_conn, network, direction, description, rule, order):
full_acl_list.insert(order, {'direction': direction, 'description': description, 'rule': rule}) full_acl_list.insert(order, {'direction': direction, 'description': description, 'rule': rule})
# Update the existing ordering # Update the existing ordering
updated_orders = dict()
for idx, acl in enumerate(full_acl_list): for idx, acl in enumerate(full_acl_list):
if acl['description'] == description: if acl['description'] == description:
continue continue
updated_orders[ if idx == acl['order']:
'/networks/{}/firewall_rules/{}/{}/order'.format(net_vni, direction, acl['description']) continue
] = idx else:
zkhandler.write([
if updated_orders: ('/networks/{}/firewall_rules/{}/{}/order'.format(net_vni, direction, acl['description']), idx)
try: ])
zkhandler.writedata(zk_conn, updated_orders)
except Exception as e:
return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e)
# Add the new rule # Add the new rule
try: zkhandler.write([
zkhandler.writedata(zk_conn, { ('/networks/{}/firewall_rules/{}/{}'.format(net_vni, direction, description), ''),
'/networks/{}/firewall_rules/{}/{}'.format(net_vni, direction, description): '', ('/networks/{}/firewall_rules/{}/{}/order'.format(net_vni, direction, description), order),
'/networks/{}/firewall_rules/{}/{}/order'.format(net_vni, direction, description): order, ('/networks/{}/firewall_rules/{}/{}/rule'.format(net_vni, direction, description), rule)
'/networks/{}/firewall_rules/{}/{}/rule'.format(net_vni, direction, description): rule ])
})
except Exception as e:
return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e)
return True, 'Firewall rule "{}" added successfully!'.format(description) return True, 'Firewall rule "{}" added successfully!'.format(description)
def remove_acl(zk_conn, network, description): def remove_acl(zkhandler, network, description):
# Validate and obtain standard passed value # Validate and obtain standard passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
match_description = '' match_description = ''
# Check if the ACL matches a description currently in the database # Check if the ACL matches a description currently in the database
acl_list = getNetworkACLs(zk_conn, net_vni, 'both') acl_list = getNetworkACLs(zkhandler, net_vni, 'both')
for acl in acl_list: for acl in acl_list:
if acl['description'] == description: if acl['description'] == description:
match_description = acl['description'] match_description = acl['description']
@ -496,76 +482,75 @@ def remove_acl(zk_conn, network, description):
# Remove the entry from zookeeper # Remove the entry from zookeeper
try: try:
zkhandler.deletekey(zk_conn, '/networks/{}/firewall_rules/{}/{}'.format(net_vni, match_direction, match_description)) zkhandler.delete('/networks/{}/firewall_rules/{}/{}'.format(net_vni, match_direction, match_description))
except Exception as e: except Exception as e:
return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e) return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e)
# Update the existing ordering # Update the existing ordering
updated_acl_list = getNetworkACLs(zk_conn, net_vni, match_direction) updated_acl_list = getNetworkACLs(zkhandler, net_vni, match_direction)
updated_orders = dict()
for idx, acl in enumerate(updated_acl_list): for idx, acl in enumerate(updated_acl_list):
updated_orders[ if acl['description'] == description:
'/networks/{}/firewall_rules/{}/{}/order'.format(net_vni, match_direction, acl['description']) continue
] = idx
if updated_orders: if idx == acl['order']:
try: continue
zkhandler.writedata(zk_conn, updated_orders) else:
except Exception as e: zkhandler.write([
return False, 'ERROR: Failed to write to Zookeeper! Exception: "{}".'.format(e) ('/networks/{}/firewall_rules/{}/{}/order'.format(net_vni, match_direction, acl['description']), idx)
])
return True, 'Firewall rule "{}" removed successfully!'.format(match_description) return True, 'Firewall rule "{}" removed successfully!'.format(match_description)
def get_info(zk_conn, network): def get_info(zkhandler, network):
# Validate and obtain alternate passed value # Validate and obtain alternate passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
network_information = getNetworkInformation(zk_conn, network) network_information = getNetworkInformation(zkhandler, network)
if not network_information: if not network_information:
return False, 'ERROR: Could not get information about network "{}"'.format(network) return False, 'ERROR: Could not get information about network "{}"'.format(network)
return True, network_information return True, network_information
def get_list(zk_conn, limit, is_fuzzy=True): def get_list(zkhandler, limit, is_fuzzy=True):
net_list = [] net_list = []
full_net_list = zkhandler.listchildren(zk_conn, '/networks') full_net_list = zkhandler.children('/networks')
for net in full_net_list: for net in full_net_list:
description = zkhandler.readdata(zk_conn, '/networks/{}'.format(net)) description = zkhandler.read('/networks/{}'.format(net))
if limit: if limit:
try: try:
if not is_fuzzy: if not is_fuzzy:
limit = '^' + limit + '$' limit = '^' + limit + '$'
if re.match(limit, net): if re.match(limit, net):
net_list.append(getNetworkInformation(zk_conn, net)) net_list.append(getNetworkInformation(zkhandler, net))
if re.match(limit, description): if re.match(limit, description):
net_list.append(getNetworkInformation(zk_conn, net)) net_list.append(getNetworkInformation(zkhandler, net))
except Exception as e: except Exception as e:
return False, 'Regex Error: {}'.format(e) return False, 'Regex Error: {}'.format(e)
else: else:
net_list.append(getNetworkInformation(zk_conn, net)) net_list.append(getNetworkInformation(zkhandler, net))
return True, net_list return True, net_list
def get_list_dhcp(zk_conn, network, limit, only_static=False, is_fuzzy=True): def get_list_dhcp(zkhandler, network, limit, only_static=False, is_fuzzy=True):
# Validate and obtain alternate passed value # Validate and obtain alternate passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
dhcp_list = [] dhcp_list = []
if only_static: if only_static:
full_dhcp_list = getNetworkDHCPReservations(zk_conn, net_vni) full_dhcp_list = getNetworkDHCPReservations(zkhandler, net_vni)
else: else:
full_dhcp_list = getNetworkDHCPReservations(zk_conn, net_vni) full_dhcp_list = getNetworkDHCPReservations(zkhandler, net_vni)
full_dhcp_list += getNetworkDHCPLeases(zk_conn, net_vni) full_dhcp_list += getNetworkDHCPLeases(zkhandler, net_vni)
if limit: if limit:
try: try:
@ -591,14 +576,14 @@ def get_list_dhcp(zk_conn, network, limit, only_static=False, is_fuzzy=True):
valid_lease = True valid_lease = True
if valid_lease: if valid_lease:
dhcp_list.append(getDHCPLeaseInformation(zk_conn, net_vni, lease)) dhcp_list.append(getDHCPLeaseInformation(zkhandler, net_vni, lease))
return True, dhcp_list return True, dhcp_list
def get_list_acl(zk_conn, network, limit, direction, is_fuzzy=True): def get_list_acl(zkhandler, network, limit, direction, is_fuzzy=True):
# Validate and obtain alternate passed value # Validate and obtain alternate passed value
net_vni = getNetworkVNI(zk_conn, network) net_vni = getNetworkVNI(zkhandler, network)
if not net_vni: if not net_vni:
return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network) return False, 'ERROR: Could not find network "{}" in the cluster!'.format(network)
@ -611,7 +596,7 @@ def get_list_acl(zk_conn, network, limit, direction, is_fuzzy=True):
direction = "out" direction = "out"
acl_list = [] acl_list = []
full_acl_list = getNetworkACLs(zk_conn, net_vni, direction) full_acl_list = getNetworkACLs(zkhandler, net_vni, direction)
if limit: if limit:
try: try: