270 lines
8.2 KiB
Python
Executable File
270 lines
8.2 KiB
Python
Executable File
#!/usr/bin/python3
|
|
"""install.py
|
|
|
|
- copy files from to_system into system directories
|
|
- set host name / network settings from aputools/servercfg file
|
|
"""
|
|
|
|
if bytes == str:
|
|
raise NotImplementedError('python2 not supported')
|
|
import os
|
|
import filecmp
|
|
import shutil
|
|
import time
|
|
from glob import glob
|
|
from ipaddress import IPv4Interface
|
|
from configparser import ConfigParser
|
|
from os.path import join, getmtime, exists, basename
|
|
|
|
os.chdir('/root/aputools/to_system')
|
|
|
|
DEL = '__to_delete__'
|
|
CFGPATH = '/root/aputools/servercfg/%s_%6.6x.cfg'
|
|
|
|
COMMENT = "; please refer to README.md for help"
|
|
|
|
TEMPLATE_CFG = """[NETWORK]
|
|
%s
|
|
enp1s0=192.168.127.1/24
|
|
enp2s0=192.168.2.1/24
|
|
enp3s0=192.168.3.1/24
|
|
enp4s0=dhcp
|
|
|
|
[ROUTER]
|
|
%s
|
|
3001=192.168.127.254:3001
|
|
""" % (COMMENT, COMMENT)
|
|
|
|
DHCP_HEADER = """
|
|
default-lease-time 600;
|
|
max-lease-time 7200;
|
|
authoritative;
|
|
"""
|
|
|
|
def write_when_new(filename, content, doit):
|
|
if not content.endswith('\n'):
|
|
content += '\n'
|
|
with open(filename) as fil:
|
|
old = fil.read()
|
|
if old == content:
|
|
return False
|
|
if doit:
|
|
with open(filename, 'w') as fil:
|
|
fil.write(content)
|
|
else:
|
|
print('===')
|
|
print(old)
|
|
print('---')
|
|
print(content)
|
|
print('===')
|
|
return True
|
|
|
|
|
|
def create_if(name, cfg, mac, dhcp_server_cfg):
|
|
result = dict(
|
|
TYPE='Ethernet',
|
|
NAME=name,
|
|
DEVICE=name,
|
|
BOOTPROTO='none',
|
|
ONBOOT='yes',
|
|
PROXY_METHOD='none',
|
|
BROWSER_ONLY='no',
|
|
IPV4_FAILURE_FATAL='yes',
|
|
IPV6INIT='no')
|
|
if cfg == 'off':
|
|
result['ONBOOT']='no'
|
|
elif cfg.startswith('wan') or cfg == 'dhcp':
|
|
result['BOOTPROTO']='dhcp'
|
|
# default: all <= 192.0.0.0
|
|
# others have to be added explicitly
|
|
dhcp_server_cfg.append((('0.0.0.0','128.0.0.0'), []))
|
|
dhcp_server_cfg.append((('128.0.0.0','192.0.0.0'), []))
|
|
for nw in cfg.split(',')[1:]:
|
|
nw = IPv4Interface(cfg).network
|
|
dhcp_server_cfg.append((nw.with_netmask.split('/'), []))
|
|
else:
|
|
cfgip = IPv4Interface(cfg)
|
|
network = cfgip.network
|
|
if network.prefixlen == 32:
|
|
otherip = IPv4Interface('%s/31' % cfgip.ip)
|
|
network = otherip.network
|
|
cfgip = network.network_address + (1 - int(otherip) % 2)
|
|
dhcp_server_cfg.append((network.with_netmask, [(str(otherip.ip), str(otherip.ip))]))
|
|
result['IPADDR'] = str(cfgip)
|
|
else: # subnet with multiple adresses -> static adresses only. dhcp range not yet implemented
|
|
result['IPADDR'] = str(cfgip.ip)
|
|
result['NETMASK'] = network.netmask
|
|
result['PREFIX'] = str(network.prefixlen)
|
|
return result
|
|
|
|
|
|
def walk(action):
|
|
for dirpath, _, files in os.walk('.'):
|
|
syspath = dirpath[1:] # remove leading '.'
|
|
action.dirpath = dirpath
|
|
action.syspath = syspath
|
|
if files:
|
|
match, mismatch, missing = filecmp.cmpfiles(dirpath, syspath, files)
|
|
if mismatch:
|
|
newer = [f for f in mismatch if getmtime(join(syspath, f)) > getmtime(join(dirpath, f))]
|
|
if newer:
|
|
action.newer(newer)
|
|
if len(newer) < len(mismatch):
|
|
newer = set(newer)
|
|
action.older([f for f in mismatch if f not in newer])
|
|
if missing:
|
|
if DEL in missing:
|
|
missing.remove(DEL)
|
|
with open(join(dirpath, DEL)) as fil:
|
|
to_delete = []
|
|
for f in fil:
|
|
f = f.strip()
|
|
if f and exists(join(syspath, f)):
|
|
if exists(join(dirpath, f)):
|
|
print('ERROR: %s in %s, but also in repo -> ignored' % (f, DEL))
|
|
else:
|
|
to_delete.append(f)
|
|
action.delete(to_delete)
|
|
if missing:
|
|
action.missing(missing)
|
|
|
|
|
|
class Show:
|
|
dirty = False
|
|
|
|
def show(self, title, files):
|
|
self.dirty = True
|
|
print('%s %s:\n %s' % (title, self.syspath, ' '.join(files)))
|
|
|
|
def newer(self, files):
|
|
self.show('get from', files)
|
|
|
|
def older(self, files):
|
|
self.show('replace in', files)
|
|
|
|
def missing(self, files):
|
|
self.show('install in', files)
|
|
|
|
def delete(self, files):
|
|
self.show('remove from', files)
|
|
|
|
|
|
class Do:
|
|
def newer(self, files):
|
|
for file in files:
|
|
shutil.copy(join(self.syspath, file), join(self.dirpath, file))
|
|
|
|
def older(self, files):
|
|
for file in files:
|
|
shutil.copy(join(self.dirpath, file), join(self.syspath, file))
|
|
|
|
def missing(self, files):
|
|
self.older(files)
|
|
|
|
def delete(self, files):
|
|
for file in files:
|
|
os.remove(join(self.syspath, file))
|
|
|
|
|
|
IFNAMES = ['enp%ds0' % i for i in range(1,5)]
|
|
|
|
|
|
def network(doit):
|
|
netaddr_dict = {}
|
|
for ifname in IFNAMES:
|
|
with open('/sys/class/net/%s/address' % ifname) as f:
|
|
netaddr_dict[ifname] = f.read().strip().lower()
|
|
netaddr = netaddr_dict[IFNAMES[0]]
|
|
apuid = int(''.join(netaddr.split(':')[-3:]), 16) & 0xfffffc
|
|
cfgfile = None
|
|
cfgfiles = []
|
|
for i in range(4):
|
|
# goodie: look for mac addresses of all 4 ports
|
|
cfgfiles += glob(CFGPATH % ('*', apuid + i))
|
|
with open('/etc/hostname') as f:
|
|
hostname = f.read().strip()
|
|
if not cfgfiles:
|
|
print('no cfg file found for %s' % hostname)
|
|
newname = input('enter host name: ')
|
|
if not newname:
|
|
print('no hostname given')
|
|
return False
|
|
cfgfile = CFGPATH % (newname, apuid)
|
|
with open(cfgfile, 'w') as f:
|
|
f.write(TEMPLATE_CFG)
|
|
elif len(cfgfiles) > 1:
|
|
print('ERROR: ambiguous cfg files: %s' % ', '.join(cfgfiles))
|
|
else:
|
|
cfgfile = cfgfiles[0]
|
|
if cfgfile != CFGPATH % (hostname, apuid):
|
|
if doit:
|
|
os.system('sh ../sethostname.sh')
|
|
else:
|
|
if cfgfile:
|
|
print('replace host name %r by %r' % (hostname, basename(cfgfile).rpartition('_')[0]))
|
|
show.dirty = True
|
|
if cfgfile is None:
|
|
return False
|
|
ifname = ''
|
|
parser = ConfigParser()
|
|
dh = []
|
|
try:
|
|
parser.read(cfgfile)
|
|
network = dict(parser['NETWORK'])
|
|
dhcp_server_cfg = []
|
|
for ifname in IFNAMES:
|
|
content = create_if(ifname, network.pop(ifname, 'off'), netaddr_dict[ifname], dhcp_server_cfg)
|
|
content = '\n'.join('%s=%s' % kv for kv in content.items())
|
|
todo = write_when_new('/etc/sysconfig/network-scripts/ifcfg-%s' % ifname, content, doit)
|
|
if todo:
|
|
print('change ', ifname)
|
|
show.dirty = True
|
|
if dh:
|
|
content = [DHCP_HEADER]
|
|
for subnet, rangelist in dhcp_server_cfg:
|
|
dh.append('subnet %s netmask %s {\n' % subnet)
|
|
dh.append(' option netmask %s;\n' % subnet[1])
|
|
for rng in rangelist:
|
|
dh.append(' range %s %s;\n' % rng)
|
|
dh.append('}\n')
|
|
content = ''.join(content)
|
|
todo = write_when_new('/etc/dhcp/dhcpd.conf', content, doit)
|
|
if todo:
|
|
print('change ', ifname)
|
|
show.dirty = True
|
|
content = []
|
|
dirty = False
|
|
for section in parser.sections():
|
|
if COMMENT not in parser[section]:
|
|
dirty = True
|
|
content.append('[%s]' % section)
|
|
content.append(COMMENT)
|
|
for key, value in parser[section].items():
|
|
content.append('%s=%s' % (key, value))
|
|
content.append('')
|
|
if dirty:
|
|
with open(cfgfile, 'w') as f:
|
|
f.write('\n'.join(content))
|
|
except Exception as e:
|
|
print('ERROR: can not handle %s %s: %r' % (hostname, ifname, e))
|
|
raise
|
|
return False
|
|
return True
|
|
|
|
|
|
print('---')
|
|
show = Show()
|
|
result = network(False)
|
|
walk(show)
|
|
|
|
if not result:
|
|
print('fix first above errors')
|
|
elif show.dirty:
|
|
print('---')
|
|
answer = input('do above? ')
|
|
if answer.lower().startswith('y'):
|
|
network(True)
|
|
walk(Do())
|
|
else:
|
|
print('nothing to do')
|