def __init__(self, profile_name = None, config = None):
self._dbus_exporter = None
storage_provider = storage.PickleProvider()
storage_factory = storage.Factory(storage_provider)
monitors_repository = monitors.Repository()
hardware_inventory = hardware.Inventory()
device_matcher = hardware.DeviceMatcher()
device_matcher_udev = hardware.DeviceMatcherUdev()
plugin_instance_factory = plugins.instance.Factory()
self.variables = profiles.variables.Variables()
self.config = GlobalConfig() if config is None else config
if self.config.get_bool(consts.CFG_DYNAMIC_TUNING):
log.info("dynamic tuning is enabled (can be overriden in plugins)")
else:
log.info("dynamic tuning is globally disabled")
plugins_repository = plugins.Repository(monitors_repository, storage_factory, hardware_inventory,\
device_matcher, device_matcher_udev, plugin_instance_factory, self.config, self.variables)
def_instance_priority = int(self.config.get(consts.CFG_DEFAULT_INSTANCE_PRIORITY, consts.CFG_DEF_DEFAULT_INSTANCE_PRIORITY))
unit_manager = units.Manager(plugins_repository, monitors_repository, def_instance_priority)
profile_factory = profiles.Factory()
profile_merger = profiles.Merger()
profile_locator = profiles.Locator(consts.LOAD_DIRECTORIES)
profile_loader = profiles.Loader(profile_locator, profile_factory, profile_merger, self.config, self.variables)
self._daemon = daemon.Daemon(unit_manager, profile_loader, profile_name, self.config, self)
self._controller = controller.Controller(self._daemon, self.config)
self._init_signals()
self._pid_file = None
def _load_global_config(self, file_name=consts.GLOBAL_CONFIG_FILE):
"""
Loads global configuration file.
"""
try:
config_parser = ConfigParser(delimiters=('='), inline_comment_prefixes=('#'))
config_parser.optionxform = str
with open(file_name) as f:
config_parser.read_string("[" + consts.MAGIC_HEADER_NAME + "]\n" + f.read(), file_name)
config, functions = GlobalConfig.get_global_config_spec()
for option in config_parser.options(consts.MAGIC_HEADER_NAME):
if option in config:
try:
func = getattr(config_parser, functions[option])
config[option] = func(consts.MAGIC_HEADER_NAME, option)
except Error:
raise TunedException("Global TuneD configuration file '%s' is not valid."
% file_name)
else:
config[option] = config_parser.get(consts.MAGIC_HEADER_NAME, option, raw=True)
except IOError as e:
raise TunedException("Global TuneD configuration file '%s' not found."
% file_name)
except Error as e:
raise TunedException("Error parsing global TuneD configuration file '%s'."
% file_name)
return config
def __init__(self, profile_name=None):
storage_provider = storage.PickleProvider()
storage_factory = storage.Factory(storage_provider)
monitors_repository = monitors.Repository()
hardware_inventory = hardware.Inventory()
device_matcher = hardware.DeviceMatcher()
plugin_instance_factory = plugins.instance.Factory()
self.variables = profiles.variables.Variables()
self.config = GlobalConfig()
if self.config.get_bool(consts.CFG_DYNAMIC_TUNING):
log.info("dynamic tuning is enabled (can be overriden in plugins)")
else:
log.info("dynamic tuning is globally disabled")
plugins_repository = plugins.Repository(monitors_repository, storage_factory, hardware_inventory, device_matcher, plugin_instance_factory, self.config, self.variables)
unit_manager = units.Manager(plugins_repository, monitors_repository)
profile_factory = profiles.Factory()
profile_merger = profiles.Merger()
profile_locator = profiles.Locator(consts.LOAD_DIRECTORIES)
profile_loader = profiles.Loader(profile_locator, profile_factory, profile_merger, self.variables)
self._daemon = daemon.Daemon(unit_manager, profile_loader, profile_name, self.config, self)
self._controller = controller.Controller(self._daemon, self.config)
self._dbus_exporter = None
self._init_signals()
self._pid_file = None
def __init__(self, plugins_repository, monitors_repository, def_instance_priority, hardware_inventory, config = None): super(Manager, self).__init__() self._plugins_repository = plugins_repository self._monitors_repository = monitors_repository self._def_instance_priority = def_instance_priority self._hardware_inventory = hardware_inventory self._instances = [] self._plugins = [] self._config = config or GlobalConfig() self._cmd = commands()
def __init__(self, profile_name = None, config = None):
# os.uname()[2] is for the python-2.7 compatibility, it's the release string
# like e.g. '5.15.13-100.fc34.x86_64'
log.info("TuneD: %s, kernel: %s" % (tuned.version.TUNED_VERSION_STR, os.uname()[2]))
self._dbus_exporter = None
storage_provider = storage.PickleProvider()
storage_factory = storage.Factory(storage_provider)
self.config = GlobalConfig() if config is None else config
if self.config.get_bool(consts.CFG_DYNAMIC_TUNING):
log.info("dynamic tuning is enabled (can be overridden in plugins)")
else:
log.info("dynamic tuning is globally disabled")
monitors_repository = monitors.Repository()
udev_buffer_size = self.config.get_size("udev_buffer_size", consts.CFG_DEF_UDEV_BUFFER_SIZE)
hardware_inventory = hardware.Inventory(buffer_size=udev_buffer_size)
device_matcher = hardware.DeviceMatcher()
device_matcher_udev = hardware.DeviceMatcherUdev()
plugin_instance_factory = plugins.instance.Factory()
self.variables = profiles.variables.Variables()
plugins_repository = plugins.Repository(monitors_repository, storage_factory, hardware_inventory,\
device_matcher, device_matcher_udev, plugin_instance_factory, self.config, self.variables)
def_instance_priority = int(self.config.get(consts.CFG_DEFAULT_INSTANCE_PRIORITY, consts.CFG_DEF_DEFAULT_INSTANCE_PRIORITY))
unit_manager = units.Manager(
plugins_repository, monitors_repository,
def_instance_priority, hardware_inventory, self.config)
profile_factory = profiles.Factory()
profile_merger = profiles.Merger()
profile_locator = profiles.Locator(consts.LOAD_DIRECTORIES)
profile_loader = profiles.Loader(profile_locator, profile_factory, profile_merger, self.config, self.variables)
self._daemon = daemon.Daemon(unit_manager, profile_loader, profile_name, self.config, self)
self._controller = controller.Controller(self._daemon, self.config)
self._init_signals()
self._pid_file = None
import tuned.version as ver
from tuned.utils.global_config import GlobalConfig
def check_positive(value):
try:
val = int(value)
except ValueError:
val = -1
if val <= 0:
raise argparse.ArgumentTypeError("%s has to be >= 0" % value)
return val
if __name__ == "__main__":
config = GlobalConfig()
parser = argparse.ArgumentParser(description="Manage tuned daemon.")
parser.add_argument('--version',
"-v",
action="version",
version="%%(prog)s %s.%s.%s" %
(ver.TUNED_VERSION_MAJOR, ver.TUNED_VERSION_MINOR,
ver.TUNED_VERSION_PATCH))
parser.add_argument("--debug",
"-d",
action="store_true",
help="show debug messages")
parser.add_argument(
"--async",
"-a",
action="store_true",
type=str,
metavar="name",
help="tuning profile to be activated")
parser.add_argument('--version',
"-v",
action="version",
version="%%(prog)s %s.%s.%s" %
(ver.TUNED_VERSION_MAJOR, ver.TUNED_VERSION_MINOR,
ver.TUNED_VERSION_PATCH))
args = parser.parse_args(sys.argv[1:])
if os.geteuid() != 0:
error("Superuser permissions are required to run the daemon.")
sys.exit(1)
config = GlobalConfig()
log = tuned.logs.get()
if args.debug:
log.setLevel("DEBUG")
try:
maxBytes = config.get_size("log_file_max_size",
consts.LOG_FILE_MAXBYTES)
backupCount = config.get("log_file_count", consts.LOG_FILE_COUNT)
if args.daemon:
if args.log is None:
args.log = consts.LOG_FILE
log.switch_to_file(args.log, maxBytes, backupCount)
else:
if args.log is not None:
log.switch_to_file(args.log, maxBytes, backupCount)
class Application(object):
def __init__(self, profile_name = None, config = None):
self._dbus_exporter = None
storage_provider = storage.PickleProvider()
storage_factory = storage.Factory(storage_provider)
self.config = GlobalConfig() if config is None else config
if self.config.get_bool(consts.CFG_DYNAMIC_TUNING):
log.info("dynamic tuning is enabled (can be overridden in plugins)")
else:
log.info("dynamic tuning is globally disabled")
monitors_repository = monitors.Repository()
udev_buffer_size = self.config.get_size("udev_buffer_size", consts.CFG_DEF_UDEV_BUFFER_SIZE)
hardware_inventory = hardware.Inventory(buffer_size=udev_buffer_size)
device_matcher = hardware.DeviceMatcher()
device_matcher_udev = hardware.DeviceMatcherUdev()
plugin_instance_factory = plugins.instance.Factory()
self.variables = profiles.variables.Variables()
plugins_repository = plugins.Repository(monitors_repository, storage_factory, hardware_inventory,\
device_matcher, device_matcher_udev, plugin_instance_factory, self.config, self.variables)
def_instance_priority = int(self.config.get(consts.CFG_DEFAULT_INSTANCE_PRIORITY, consts.CFG_DEF_DEFAULT_INSTANCE_PRIORITY))
unit_manager = units.Manager(
plugins_repository, monitors_repository,
def_instance_priority, hardware_inventory, self.config)
profile_factory = profiles.Factory()
profile_merger = profiles.Merger()
profile_locator = profiles.Locator(consts.LOAD_DIRECTORIES)
profile_loader = profiles.Loader(profile_locator, profile_factory, profile_merger, self.config, self.variables)
self._daemon = daemon.Daemon(unit_manager, profile_loader, profile_name, self.config, self)
self._controller = controller.Controller(self._daemon, self.config)
self._init_signals()
self._pid_file = None
def _handle_signal(self, signal_number, handler):
def handler_wrapper(_signal_number, _frame):
if signal_number == _signal_number:
handler()
signal.signal(signal_number, handler_wrapper)
def _init_signals(self):
self._handle_signal(signal.SIGHUP, self._controller.reload)
self._handle_signal(signal.SIGINT, self._controller.terminate)
self._handle_signal(signal.SIGTERM, self._controller.terminate)
def attach_to_dbus(self, bus_name, object_name, interface_name):
if self._dbus_exporter is not None:
raise TunedException("DBus interface is already initialized.")
self._dbus_exporter = exports.dbus.DBusExporter(bus_name, interface_name, object_name)
exports.register_exporter(self._dbus_exporter)
exports.register_object(self._controller)
def _daemonize_parent(self, parent_in_fd, child_out_fd):
"""
Wait till the child signalizes that the initialization is complete by writing
some uninteresting data into the pipe.
"""
os.close(child_out_fd)
(read_ready, drop, drop) = select.select([parent_in_fd], [], [], consts.DAEMONIZE_PARENT_TIMEOUT)
if len(read_ready) != 1:
os.close(parent_in_fd)
raise TunedException("Cannot daemonize, timeout when waiting for the child process.")
response = os.read(parent_in_fd, 8)
os.close(parent_in_fd)
if len(response) == 0:
raise TunedException("Cannot daemonize, no response from child process received.")
try:
val = struct.unpack("?", response)[0]
except struct.error:
raise TunedException("Cannot daemonize, invalid response from child process received.")
if val != True:
raise TunedException("Cannot daemonize, child process reports failure.")
def write_pid_file(self, pid_file = consts.PID_FILE):
self._pid_file = pid_file
self._delete_pid_file()
try:
dir_name = os.path.dirname(self._pid_file)
if not os.path.exists(dir_name):
os.makedirs(dir_name)
with os.fdopen(os.open(self._pid_file, os.O_CREAT|os.O_TRUNC|os.O_WRONLY , 0o644), "w") as f:
f.write("%d" % os.getpid())
except (OSError,IOError) as error:
log.critical("cannot write the PID to %s: %s" % (self._pid_file, str(error)))
def _delete_pid_file(self):
if os.path.exists(self._pid_file):
try:
os.unlink(self._pid_file)
except OSError as error:
log.warning("cannot remove existing PID file %s, %s" % (self._pid_file, str(error)))
def _daemonize_child(self, pid_file, parent_in_fd, child_out_fd):
"""
Finishes daemonizing process, writes a PID file and signalizes to the parent
that the initialization is complete.
"""
os.close(parent_in_fd)
os.chdir("/")
os.setsid()
os.umask(0)
try:
pid = os.fork()
if pid > 0:
sys.exit(0)
except OSError as error:
log.critical("cannot daemonize, fork() error: %s" % str(error))
val = struct.pack("?", False)
os.write(child_out_fd, val)
os.close(child_out_fd)
raise TunedException("Cannot daemonize, second fork() failed.")
fd = open("/dev/null", "w+")
os.dup2(fd.fileno(), sys.stdin.fileno())
os.dup2(fd.fileno(), sys.stdout.fileno())
os.dup2(fd.fileno(), sys.stderr.fileno())
self.write_pid_file(pid_file)
log.debug("successfully daemonized")
val = struct.pack("?", True)
os.write(child_out_fd, val)
os.close(child_out_fd)
def daemonize(self, pid_file = consts.PID_FILE):
"""
Daemonizes the application. In case of failure, TunedException is raised
in the parent process. If the operation is successfull, the main process
is terminated and only child process returns from this method.
"""
parent_child_fds = os.pipe()
try:
child_pid = os.fork()
except OSError as error:
os.close(parent_child_fds[0])
os.close(parent_child_fds[1])
raise TunedException("Cannot daemonize, fork() failed.")
try:
if child_pid > 0:
self._daemonize_parent(*parent_child_fds)
sys.exit(0)
else:
self._daemonize_child(pid_file, *parent_child_fds)
except:
# pass exceptions only into parent process
if child_pid > 0:
raise
else:
sys.exit(1)
@property
def daemon(self):
return self._daemon
@property
def controller(self):
return self._controller
def run(self, daemon):
# override global config if ran from command line with daemon option (-d)
if daemon:
self.config.set(consts.CFG_DAEMON, True)
if not self.config.get_bool(consts.CFG_DAEMON, consts.CFG_DEF_DAEMON):
log.warn("Using one shot no deamon mode, most of the functionality will be not available, it can be changed in global config")
result = self._controller.run()
if self.config.get_bool(consts.CFG_DAEMON, consts.CFG_DEF_DAEMON):
exports.stop()
if self._pid_file is not None:
self._delete_pid_file()
return result
class Application(object):
def __init__(self, profile_name=None):
storage_provider = storage.PickleProvider()
storage_factory = storage.Factory(storage_provider)
monitors_repository = monitors.Repository()
hardware_inventory = hardware.Inventory()
device_matcher = hardware.DeviceMatcher()
plugin_instance_factory = plugins.instance.Factory()
self.variables = profiles.variables.Variables()
self.config = GlobalConfig()
if self.config.get_bool(consts.CFG_DYNAMIC_TUNING):
log.info("dynamic tuning is enabled (can be overriden in plugins)")
else:
log.info("dynamic tuning is globally disabled")
plugins_repository = plugins.Repository(monitors_repository, storage_factory, hardware_inventory, device_matcher, plugin_instance_factory, self.config, self.variables)
unit_manager = units.Manager(plugins_repository, monitors_repository)
profile_factory = profiles.Factory()
profile_merger = profiles.Merger()
profile_locator = profiles.Locator(consts.LOAD_DIRECTORIES)
profile_loader = profiles.Loader(profile_locator, profile_factory, profile_merger, self.variables)
self._daemon = daemon.Daemon(unit_manager, profile_loader, profile_name, self.config, self)
self._controller = controller.Controller(self._daemon, self.config)
self._dbus_exporter = None
self._init_signals()
self._pid_file = None
def _handle_signal(self, signal_number, handler):
def handler_wrapper(_signal_number, _frame):
if signal_number == _signal_number:
handler()
signal.signal(signal_number, handler_wrapper)
def _init_signals(self):
self._handle_signal(signal.SIGHUP, self._controller.reload)
self._handle_signal(signal.SIGINT, self._controller.terminate)
self._handle_signal(signal.SIGTERM, self._controller.terminate)
def attach_to_dbus(self, bus_name, object_name, interface_name):
if self._dbus_exporter is not None:
raise TunedException("DBus interface is already initialized.")
self._dbus_exporter = exports.dbus.DBusExporter(bus_name, interface_name, object_name)
exports.register_exporter(self._dbus_exporter)
exports.register_object(self._controller)
def _daemonize_parent(self, parent_in_fd, child_out_fd):
"""
Wait till the child signalizes that the initialization is complete by writing
some uninteresting data into the pipe.
"""
os.close(child_out_fd)
(read_ready, drop, drop) = select.select([parent_in_fd], [], [], consts.DAEMONIZE_PARENT_TIMEOUT)
if len(read_ready) != 1:
os.close(parent_in_fd)
raise TunedException("Cannot daemonize, timeout when waiting for the child process.")
response = os.read(parent_in_fd, 8)
os.close(parent_in_fd)
if len(response) == 0:
raise TunedException("Cannot daemonize, no response from child process received.")
if response != ("%c" % True):
raise TunedException("Cannot daemonize, child process reports failure.")
def write_pid_file(self, pid_file = consts.PID_FILE):
self._pid_file = pid_file
self._delete_pid_file()
try:
dir_name = os.path.dirname(self._pid_file)
if not os.path.exists(dir_name):
os.makedirs(dir_name)
fd = os.open(self._pid_file, os.O_CREAT|os.O_TRUNC|os.O_WRONLY , 0644)
os.write(fd, "%d" % os.getpid())
os.close(fd)
except (OSError,IOError) as error:
log.critical("cannot write the PID to %s: %s" % (self._pid_file, str(error)))
def _delete_pid_file(self):
if os.path.exists(self._pid_file):
try:
os.unlink(self._pid_file)
except OSError as error:
log.warning("cannot remove existing PID file %s, %s" % (self._pid_file, str(error)))
def _daemonize_child(self, pid_file, parent_in_fd, child_out_fd):
"""
Finishes daemonizing process, writes a PID file and signalizes to the parent
that the initialization is complete.
"""
os.close(parent_in_fd)
os.chdir("/")
os.setsid()
os.umask(0)
try:
pid = os.fork()
if pid > 0:
sys.exit(0)
except OSError as error:
log.critical("cannot daemonize, fork() error: %s" % str(error))
os.write(child_out_fd, "%c" % False)
os.close(child_out_fd)
raise TunedException("Cannot daemonize, second fork() failed.")
si = file("/dev/null", "r")
so = file("/dev/null", "a+")
se = file("/dev/null", "a+", 0)
os.dup2(si.fileno(), sys.stdin.fileno())
os.dup2(so.fileno(), sys.stdout.fileno())
os.dup2(se.fileno(), sys.stderr.fileno())
self.write_pid_file(pid_file)
log.debug("successfully daemonized")
os.write(child_out_fd, "%c" % True)
os.close(child_out_fd)
def daemonize(self, pid_file = consts.PID_FILE):
"""
Daemonizes the application. In case of failure, TunedException is raised
in the parent process. If the operation is successfull, the main process
is terminated and only child process returns from this method.
"""
parent_child_fds = os.pipe()
try:
child_pid = os.fork()
except OSError as error:
os.close(parent_child_fds[0])
os.close(parent_child_fds[1])
raise TunedException("Cannot daemonize, fork() failed.")
try:
if child_pid > 0:
self._daemonize_parent(*parent_child_fds)
sys.exit(0)
else:
self._daemonize_child(pid_file, *parent_child_fds)
except:
# pass exceptions only into parent process
if child_pid > 0:
raise
else:
sys.exit(1)
@property
def daemon(self):
return self._daemon
@property
def controller(self):
return self._controller
def run(self, daemon):
# override global config if ran from command line with daemon option (-d)
if daemon:
self.config.set(consts.CFG_DAEMON, True)
if self.config.get_bool(consts.CFG_DAEMON, consts.CFG_DEF_DAEMON):
exports.start()
else:
log.warn("Using one shot no deamon mode, most of the functionality will be not available, it can be changed in global config")
result = self._controller.run()
if self.config.get_bool(consts.CFG_DAEMON, consts.CFG_DEF_DAEMON):
exports.stop()
if self._pid_file is not None:
self._delete_pid_file()
return result
type=str,
metavar="name",
help="tuning profile to be activated")
parser.add_argument('--version',
"-v",
action="version",
version="%%(prog)s %s.%s.%s" %
(ver.TUNED_VERSION_MAJOR, ver.TUNED_VERSION_MINOR,
ver.TUNED_VERSION_PATCH))
args = parser.parse_args(sys.argv[1:])
if os.geteuid() != 0:
error("Superuser permissions are required to run the daemon.")
sys.exit(1)
config = GlobalConfig()
log = tuned.logs.get()
if args.debug:
log.setLevel("DEBUG")
try:
if args.daemon:
if args.log is None:
args.log = consts.LOG_FILE
log.switch_to_file(args.log)
else:
if args.log is not None:
log.switch_to_file(args.log)
app = tuned.daemon.Application(args.profile, config)
