#!/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')