Joshua Boniface
f198f62563
Completely restructure the daemon code to move the 4 discrete daemons into a single daemon that can be run on every hypervisor. Introduce the idea of a static list of "coordinator" nodes which are configured at install time to run Zookeeper and FRR in router mode, and which are allowed to take on client network management duties (gateway, DHCP, DNS, etc.) while also allowing them to run VMs (i.e. no dedicated "router" nodes required).
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# common.py - PVC daemon function library, common fuctions
|
|
# Part of the Parallel Virtual Cluster (PVC) system
|
|
#
|
|
# Copyright (C) 2018 Joshua M. Boniface <joshua@boniface.me>
|
|
#
|
|
# 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 <https://www.gnu.org/licenses/>.
|
|
#
|
|
###############################################################################
|
|
|
|
import subprocess
|
|
import threading
|
|
import signal
|
|
import os
|
|
import time
|
|
|
|
import pvcd.log as log
|
|
|
|
class OSDaemon(object):
|
|
def __init__(self, command, environment):
|
|
self.proc = subprocess.Popen(
|
|
command,
|
|
env=environment,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
|
|
def signal(self, sent_signal):
|
|
signal_map = {
|
|
'hup': signal.SIGHUP,
|
|
'int': signal.SIGINT,
|
|
'term': signal.SIGTERM
|
|
}
|
|
self.proc.send_signal(signal_map[sent_signal])
|
|
|
|
def run_os_daemon(command_string, environment=None):
|
|
command = command_string.split()
|
|
print(' '.join(command))
|
|
daemon = OSDaemon(command, environment)
|
|
return daemon
|
|
|
|
# Run a oneshot command, optionally without blocking
|
|
def run_os_command(command_string, background=False, environment=None):
|
|
command = command_string.split()
|
|
if background:
|
|
def runcmd():
|
|
subprocess.run(
|
|
command,
|
|
env=environment,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
thread = threading.Thread(target=runcmd, args=())
|
|
thread.start()
|
|
return 0, None, None
|
|
else:
|
|
command_output = subprocess.run(
|
|
command,
|
|
env=environment,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
return command_output.returncode, command_output.stdout.decode('ascii'), command_output.stderr.decode('ascii')
|
|
|
|
# Reload the firewall rules of the system
|
|
def reload_firewall_rules(rules_dir):
|
|
log.echo('Updating firewall rules', '', 'o')
|
|
rules_file = '{}/base.nft'.format(rules_dir)
|
|
retcode, stdout, stderr = run_os_command('/usr/sbin/nft -f {}'.format(rules_file))
|
|
if retcode != 0:
|
|
log.echo('Failed to reload rules: {}'.format(stderr), '', 'e')
|