# Copyright 2019-2021 VyOS maintainers and contributors <maintainers@vyos.io> # # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public # License as published by the Free Software Foundation; either # version 2.1 of the License, or (at your option) any later version. # # This library 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 # Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public # License along with this library. If not, see <http://www.gnu.org/licenses/>. import os from inspect import signature from inspect import _empty from vyos.ifconfig.section import Section from vyos.util import popen from vyos.util import cmd from vyos.util import read_file from vyos.util import write_file from vyos import debug class Control(Section): _command_get = {} _command_set = {} _signature = {} def __init__(self, **kargs): # some commands (such as operation comands - show interfaces, etc.) # need to query the interface statistics. If the interface # code is used and the debugging is enabled, the screen output # will include both the command but also the debugging for that command # to prevent this, debugging can be explicitely disabled # if debug is not explicitely disabled the the config, enable it self.debug = '' if kargs.get('debug', True) and debug.enabled('ifconfig'): self.debug = 'ifconfig' def _debug_msg (self, message): return debug.message(message, self.debug) def _popen(self, command): return popen(command, self.debug) def _cmd(self, command): return cmd(command, self.debug) def _get_command(self, config, name): """ Using the defined names, set data write to sysfs. """ cmd = self._command_get[name]['shellcmd'].format(**config) return self._command_get[name].get('format', lambda _: _)(self._cmd(cmd)) def _values(self, name, validate, value): """ looks at the validation function "validate" for the interface sysfs or command and returns a dict with the right options to call it """ if name not in self._signature: self._signature[name] = signature(validate) values = {} for k in self._signature[name].parameters: default = self._signature[name].parameters[k].default if default is not _empty: continue if k == 'self': values[k] = self elif k == 'ifname': values[k] = self.ifname else: values[k] = value return values def _set_command(self, config, name, value): """ Using the defined names, set data write to sysfs. """ # the code can pass int as int value = str(value) validate = self._command_set[name].get('validate', None) if validate: try: validate(**self._values(name, validate, value)) except Exception as e: raise e.__class__(f'Could not set {name}. {e}') convert = self._command_set[name].get('convert', None) if convert: value = convert(value) possible = self._command_set[name].get('possible', None) if possible and not possible(config['ifname'], value): return False config = {**config, **{'value': value}} cmd = self._command_set[name]['shellcmd'].format(**config) return self._command_set[name].get('format', lambda _: _)(self._cmd(cmd)) _sysfs_get = {} _sysfs_set = {} def _read_sysfs(self, filename): """ Provide a single primitive w/ error checking for reading from sysfs. """ value = None if os.path.exists(filename): value = read_file(filename) self._debug_msg("read '{}' < '{}'".format(value, filename)) return value def _write_sysfs(self, filename, value): """ Provide a single primitive w/ error checking for writing to sysfs. """ if os.path.isfile(filename): write_file(filename, str(value)) self._debug_msg("write '{}' > '{}'".format(value, filename)) return True return False def _get_sysfs(self, config, name): """ Using the defined names, get data write from sysfs. """ filename = self._sysfs_get[name]['location'].format(**config) if not filename: return None return self._read_sysfs(filename) def _set_sysfs(self, config, name, value): """ Using the defined names, set data write to sysfs. """ # the code can pass int as int value = str(value) validate = self._sysfs_set[name].get('validate', None) if validate: try: validate(**self._values(name, validate, value)) except Exception as e: raise e.__class__(f'Could not set {name}. {e}') config = {**config, **{'value': value}} convert = self._sysfs_set[name].get('convert', None) if convert: value = convert(value) commited = self._write_sysfs( self._sysfs_set[name]['location'].format(**config), value) if not commited: errmsg = self._sysfs_set.get('errormsg', '') if errmsg: raise TypeError(errmsg.format(**config)) return commited def get_interface(self, name): if name in self._sysfs_get: return self._get_sysfs(self.config, name) if name in self._command_get: return self._get_command(self.config, name) raise KeyError(f'{name} is not a attribute of the interface we can get') def set_interface(self, name, value): if name in self._sysfs_set: return self._set_sysfs(self.config, name, value) if name in self._command_set: return self._set_command(self.config, name, value) raise KeyError(f'{name} is not a attribute of the interface we can set')