#!/usr/bin/env python3 # Daemon.py - PVC hypervisor network daemon # Part of the Parallel Virtual Cluster (PVC) system # # Copyright (C) 2018 Joshua M. Boniface # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . # ############################################################################### import kazoo.client import sys import os import signal import socket import psutil import configparser import time import daemon_lib.ansiiprint as ansiiprint import daemon_lib.zkhandler as zkhandler import pvcnd.VXNetworkInstance as VXNetworkInstance print(ansiiprint.bold() + "pvcnd - Parallel Virtual Cluster network daemon" + ansiiprint.end()) # Get the config file variable from the environment try: pvcnd_config_file = os.environ['PVCND_CONFIG_FILE'] except: print('ERROR: The "PVCND_CONFIG_FILE" environment variable must be set before starting pvcnd.') exit(1) myhostname = socket.gethostname() myshorthostname = myhostname.split('.', 1)[0] mydomainname = ''.join(myhostname.split('.', 1)[1:]) # Config values dictionary config_values = [ 'zookeeper', 'vni_dev', 'vni_dev_ip', ] def readConfig(pvcnd_config_file, myhostname): print('Loading configuration from file {}'.format(pvcnd_config_file)) o_config = configparser.ConfigParser() o_config.read(pvcnd_config_file) config = {} try: entries = o_config[myhostname] except: try: entries = o_config['default'] except: print('ERROR: Config file is not valid!') exit(1) for entry in config_values: try: config[entry] = entries[entry] except: try: config[entry] = o_config['default'][entry] except: print('ERROR: Config file missing required value "{}" for this host!'.format(entry)) exit(1) return config config = readConfig(pvcnd_config_file, myhostname) zk_conn = kazoo.client.KazooClient(hosts=config['zookeeper']) try: print('Connecting to Zookeeper instance at {}'.format(config['zookeeper'])) zk_conn.start() except: print('ERROR: Failed to connect to Zookeeper!') exit(1) # Handle zookeeper failures gracefully def zk_listener(state): global zk_conn if state == kazoo.client.KazooState.SUSPENDED: ansiiprint.echo('Connection to Zookeeper list; retrying', '', 'e') while True: _zk_conn = kazoo.client.KazooClient(hosts=config['zookeeper']) try: _zk_conn.start() zk_conn = _zk_conn break except: time.sleep(1) elif state == kazoo.client.KazooState.CONNECTED: anisiprint.echo('Connection to Zookeeper started', '', 'o') else: pass zk_conn.add_listener(zk_listener) # Cleanup function def cleanup(signum, frame): ansiiprint.echo('Terminating daemon', '', 'e') # Close the Zookeeper connection try: zk_conn.stop() zk_conn.close() except: pass # Exit exit(0) # Handle signals with cleanup signal.signal(signal.SIGTERM, cleanup) signal.signal(signal.SIGINT, cleanup) signal.signal(signal.SIGQUIT, cleanup) # What this daemon does: # 1. Configure public networks dynamically on startup (e.g. bonding, vlans, etc.) from config # * no /etc/network/interfaces config for these - just mgmt interface via DHCP! # 2. Watch ZK /networks # 3. Provision required network interfaces when a network is added # a. create vxlan interface targeting local dev from config # b. create bridge interface # c. add vxlan to bridge # d. set interfaces up # 4. Remove network interfaces when network disapears # Zookeeper schema: # networks/ # / # ipnet e.g. 10.101.0.0/24 # gateway e.g. 10.101.0.1 [1] # routers e.g. 10.101.0.2,10.101.0.3 [2] # dhcp e.g. YES [3] # reservations/ # / # address e.g. 10.101.0.30 # mac e.g. ff:ff:fe:ab:cd:ef # fwrules/ # / # description e.g. Allow HTTP from any to this net # src e.g. any # dest e.g. this # port e.g. 80 # Notes: # [1] becomes a VIP between the pair of routers in multi-router envs # [2] becomes real addrs on the pair of routers in multi-router envs # [2] should match gateway in single-router envs for consistency # [3] enables or disables a DHCP subnet definition for the network # Prepare underlying interface if config['vni_dev_ip'] == 'dhcp': vni_dev = config['vni_dev'] ansiiprint.echo('Configuring VNI parent device {} with DHCP IP'.format(vni_dev), '', 'o') os.system( 'ip link set {0} up'.format( vni_dev ) ) os.system( 'dhclient {0}'.format( vni_dev ) ) else: vni_dev = config['vni_dev'] vni_dev_ip = config['vni_dev_ip'] ansiiprint.echo('Configuring VNI parent device {} with IP {}'.format(vni_dev, vni_dev_ip), '', 'o') os.system( 'ip link set {0} up'.format( vni_dev ) ) os.system( 'ip address add {0} dev {1}'.format( vni_dev_ip, vni_dev ) ) # Prepare VNI list t_vni = dict() vni_list = [] @zk_conn.ChildrenWatch('/networks') def updatenetworks(new_vni_list): global vni_list print(ansiiprint.blue() + 'Network list: ' + ansiiprint.end() + '{}'.format(' '.join(new_vni_list))) # Add new VNIs for vni in new_vni_list: if vni not in vni_list: vni_list.append(vni) t_vni[vni] = VXNetworkInstance.VXNetworkInstance(vni, zk_conn, config) t_vni[vni].createNetwork() # Remove deleted VNIs for vni in vni_list: if vni not in new_vni_list: vni_list.remove(vni) t_vni[vni].removeNetwork() # Tick loop while True: try: time.sleep(0.1) except: break