def create_config(self, vpn_acc: str, replacements: dict):
config_file = self._to_config_file(vpn_acc)
logger.log(self.log_lvl,
f'Create DHCP client VPN config[{config_file}]...')
FileHelper.copy(self.resource_dir.joinpath(self.DHCLIENT_CONFIG_TMPL),
config_file,
force=True)
FileHelper.replace_in_file(config_file, replacements, backup='')
FileHelper.chmod(config_file, mode=0o0644)
def add_hook(self, service_name: str, replacements: dict):
exit_hook_file = self._to_hook_file(service_name)
logger.log(self.log_lvl,
f'Create DHCP client VPN hook[{exit_hook_file}]...')
FileHelper.copy(self.resource_dir.joinpath(
self.DHCLIENT_EXIT_HOOK_TMPL),
exit_hook_file,
force=True)
FileHelper.replace_in_file(exit_hook_file, replacements, backup='')
FileHelper.chmod(exit_hook_file, mode=0o0744)
def change_host_name(hostname: str, log_lvl=logger.DEBUG):
prev_regex = re.escape('127.0.1.1') + r'\s+' + re.escape(
socket.gethostname()) + r'.*'
SystemHelper.exec_command(f'hostnamectl set-hostname {hostname}',
log_lvl=logger.down_lvl(log_lvl))
FileHelper.replace_in_file('/etc/hosts',
{prev_regex: f'127.0.1.1 {hostname}'},
regex=True)
SystemHelper.exec_command(f'hostnamectl', silent=True, log_lvl=log_lvl)
logger.sep(level=log_lvl, quantity=20)
SystemHelper.exec_command(f'cat /etc/hosts',
silent=True,
log_lvl=log_lvl)
logger.sep(level=log_lvl)
def create(self,
svc_opts: UnixServiceOpts,
replacements: dict,
auto_startup: bool = False):
service_fqn = self.to_service_fqn(svc_opts.service_dir,
svc_opts.service_name)
logger.info(
f'Add new service [{svc_opts.service_name}] in [{service_fqn}]...')
FileHelper.copy(self.resource_dir.joinpath(Systemd.SERVICE_FILE_TMPL),
service_fqn,
force=True)
FileHelper.replace_in_file(service_fqn, replacements, backup='')
FileHelper.chmod(service_fqn, mode=0o0644)
SystemHelper.exec_command("systemctl daemon-reload",
silent=True,
log_lvl=logger.INFO)
if auto_startup:
self.enable(svc_opts.service_name)
def setup(self, vpn_service: str, origin_resolv_conf: Path,
vpn_resolv_conf: Path, vpn_nameserver_hook_conf: Path):
if not self._available:
logger.error('[dnsmasq] is not yet installed or is corrupted')
sys.exit(ErrorCode.MISSING_REQUIREMENT)
logger.info('Setup DNS resolver[dnsmasq]...')
dnsmasq_vpn_cfg = self._dnsmasq_vpn_cfg(vpn_service)
runtime_resolv_cfg = self.adapt_dnsmasq(origin_resolv_conf,
vpn_service)
dnsmasq_opts = {
'{{DNS_RESOLVED_FILE}}':
self.__build_dnsmasq_conf('resolv-file', runtime_resolv_cfg),
'{{PORT}}':
self.__build_dnsmasq_conf('port',
self.dnsmasq_options().get('port',
None)),
'{{CACHE_SIZE}}':
self.__build_dnsmasq_conf(
'cache-size',
self.dnsmasq_options().get('cache_size', None))
}
logger.debug(
f'Add [dnsmasq] config for {vpn_service}[{dnsmasq_vpn_cfg}]...')
FileHelper.copy(self.resource_dir.joinpath(self.DNSMASQ_CONFIG_TMPL),
dnsmasq_vpn_cfg,
force=True)
FileHelper.replace_in_file(dnsmasq_vpn_cfg, dnsmasq_opts, backup='')
FileHelper.chmod(dnsmasq_vpn_cfg, mode=0o0644)
logger.debug(
f'Symlink [dnsmasq] VPN nameserver runtime configuration [{vpn_nameserver_hook_conf}]...'
)
FileHelper.create_symlink(vpn_nameserver_hook_conf,
self._dnsmasq_vpn_hook_cfg,
force=True)
logger.info(f'Generate System DNS config file from VPN service...')
FileHelper.write_file(vpn_resolv_conf,
self.__dnsmasq_resolv(vpn_service),
mode=0o0644)
FileHelper.create_symlink(vpn_resolv_conf,
DNSResolver.DNS_SYSTEM_FILE,
force=True)
self.service.enable(self.config.identity)
