Edit File: lscgctl
#!/usr/bin/python3 import argparse, json, logging, os, pwd, subprocess, sys from stat import * import common def command_requires_uid(command): return command in ['list', 'list-user', 'reset', 'reset-user', 'set', 'set-user'] def validate_environment(): if not os.path.exists('/sys/fs/cgroup/cgroup.controllers'): common.fatal_error("cgroups is not v2 on this machine") if os.getuid() != 0: common.fatal_error("this program must be run as root") common.ls_ok() if not os.path.isfile('/etc/systemd/system.control/user.slice.d/50-IOAccounting.conf'): logging.debug('Activate accounting') run_program(['systemctl', 'set-property', 'user.slice', 'IOAccounting=yes', 'MemoryAccounting=yes', 'TasksAccounting=yes']) systemctl_daemon_reload() def init_pgm(): common.init_logging() parser = argparse.ArgumentParser(prog="lscgctl", description='LiteSpeed cgroups Control Program') parser.add_argument("command", type=str, help="The cgroups command", choices=['list', 'list-all', 'list-user', 'reset', 'reset-all', 'reset-user', 'set', 'set-all', 'set-user', 'version']) parser.add_argument("uid", type=str, nargs='*', default=None, help="uid or user name for -user commands") parser.add_argument("--cpu", type=str, help="limit CPU access. Specify a percentage of a complete core, 100 is 1 complete core. Default is no limit (-1).") parser.add_argument("--io", type=str, help="limit I/O access. Specify in bytes/sec. Default is no limit (-1).") parser.add_argument("--iops", type=str, help="limit I/O access. Specify in IOs per second. Default is no limit (-1).") parser.add_argument('-l', '--log', type=int, help='set logging level, 10=Debug, 20=Info, 30=Warning, 40=Error. Default is Info') parser.add_argument("--mem", type=str, help="limit virtual memory access. Specify in bytes. Default is no limit (-1).") parser.add_argument('-q', '--quiet', action='store_true', help='turns off all logging and only outputs what is requested.') parser.add_argument("--tasks", type=str, help="limit number of tasks. Specify a maximum count of tasks that can be started.") args = parser.parse_args() if not args.quiet or args.log != None: if args.log != None: logging.getLogger().setLevel(args.log) else: logging.getLogger().setLevel(logging.INFO) logging.debug("Entering lscgctl") validate_environment() command = args.command if command_requires_uid(command): if args.uid == None: common.fatal_error("You must specify a user for the %s command" % command) users = common.get_users(args.uid) else: users = None if command == 'version': logging.info("Version: %s" % common.VERSION) if args.quiet: print(common.VERSION) sys.exit(0) if (command.startswith('set')) and (args.cpu == None and args.io == None and args.iops == None and args.mem == None and args.tasks == None): common.fatal_error("You specified command: %s and no qualifier to set with it" % command) return args, users def read_stat(file, pos): try: f = open(file, 'r') except Exception as err: common.fatal_error('Error opening %s: %s' % (file, err)) line = f.readline() pieces = line.split(' ') f.close() return pieces[pos] def run_program(args, fail_reason = ""): logging.debug('run: ' + str(args)) result = subprocess.run(args, capture_output=True, text=True) if fail_reason != "" and result.returncode == 0: common.fatal_error('Expected: ' + args[0] + ' to fail: ' + fail_reason) if fail_reason == "" and result.returncode != 0: common.fatal_error('Error in running: ' + args[0] + ' errors: ' + result.stdout + ' ' + result.stderr) return result.stdout def user_to_file(user): return 'user-%s.slice' % user.pw_uid def user_to_filename(user): return '/sys/fs/cgroup/user.slice/%s' % user_to_file(user) def systemctl_show(user): return run_program(['systemctl', 'show', user_to_file(user)]) def systemctl_daemon_reload(): logging.debug("Do daemon-reload") run_program(['systemctl', 'daemon-reload']) def get_systemctl_item(result, item): start = result.find(item) if start == -1: logging.debug('%s not found in systemctl output' % item) return "" eol = result[start:].find('\n') if eol == -1: logging.debug('new line not found in search after finding %s' % item) eol = len(result[start:]) equals = result[start:start+eol].find('=') if equals == -1 or equals > eol: logging.debug('equals not found in systemctl output search for %s: %s' % (item, result[start:start+eol])) return "" value = result[start+equals+1:start+eol] if value == 'infinity': return '' logging.debug('systemctl show for ' + item + ':' + value) return value def get_systemctl_dev_num(result, item): val = get_systemctl_item(result, item) space = val.find(' ') if space == -1: return '' logging.debug('result %s: %s, val: %s' % (item, val, val[space+1:])) return common.str_num_values(val[space+1:]) def get_systemctl_num(result, item): val = get_systemctl_item(result, item) logging.debug('result %s: %s' % (item, val)) return common.str_num_values(val) def get_systemctl_cpu(result): val = get_systemctl_item(result, 'CPUQuotaPerSecUSec') if val == 'infinity' or val == 'max' or val == '': return '' ms = val.find('ms') logging.debug('cpu val:' + val + ' ms: %d' % ms) if ms > 0: percent = int(val[:ms]) / 10 if percent == 0: return '0.%d' % int(val[:ms]) return '%d' % percent s = val.find('s') if s > 0: return '%d' % int((float(val[:s]) * 100)) # A bug in systemd cgroups return '%f' % float(val) / 10000 def list_user(user, dict): systemctl_results = systemctl_show(user) subdict = {} subdict['name'] = user.pw_name subdict['cpu'] = get_systemctl_cpu(systemctl_results) subdict['io'] = get_systemctl_dev_num(systemctl_results, 'IOReadBandwidthMax') subdict['iops'] = get_systemctl_dev_num(systemctl_results, 'IOReadIOPSMax') subdict['mem'] = get_systemctl_num(systemctl_results, "MemoryHigh") subdict['tasks'] = get_systemctl_num(systemctl_results, 'TasksMax') logging.debug("uid[" + str(user.pw_uid) + ']: ' + str(subdict)) dict[user.pw_uid] = subdict def command_list(users): dict = {} if users != None: for user in users: list_user(user, dict) else: users = pwd.getpwall() min_uid = common.get_min_uid() for listuser in users: if listuser.pw_uid >= min_uid and listuser.pw_name != 'nobody': list_user(listuser, dict) print(json.dumps(dict, indent=4)) return 0 def set_properties(user, properties): pgm_parms = ['systemctl', 'set-property', user_to_file(user)] pgm_parms.extend(properties) return run_program(pgm_parms) def set_device_parm(args, option, properties): devices = common.get_devices()[0] if option == common.OPTION_IO: title_read = 'IOReadBandwidthMax' title_write = 'IOWriteBandwidthMax' val = common.int_num_values(args.io) else: title_read = 'IOReadIOPSMax' title_write = 'IOWriteIOPSMax' val = common.int_num_values(args.iops) for device in devices: if val == -1: properties.append('%s=' % title_read) properties.append('%s=' % title_write) break else: properties.append('%s=%s %d' % (title_read, device, val)) properties.append('%s=%s %d' % (title_write, device, val)) return 0 def set_no_device_parm(args, option, properties): if option == common.OPTION_CPU: title = 'CPUQuota' val = common.int_num_values(args.cpu) format_set = '%s=%d%%' elif option == common.OPTION_MEM: title = 'MemoryHigh' val = common.int_num_values(args.mem) format_set = '%s=%d' else: title = 'TasksMax' val = common.int_num_values(args.tasks) format_set = '%s=%d' if val == -1: properties.append('%s=' % title) else: properties.append(format_set % (title, val)) def reset_user(args, user, reload): args.cpu = '-1' args.io = '-1' args.iops = '-1'; args.mem = '-1' args.tasks = '-1' return set_user(args, user, reload) def command_reset(args, users): if args.command == 'reset' or args.command == 'reset-user': for user in users: reset_user(args, user, True) else: users = pwd.getpwall() min_uid = common.get_min_uid() need_reload = False for listuser in users: if listuser.pw_uid >= min_uid: need_reload = True reset_user(args, listuser) if need_reload: systemctl_daemon_reload() if not args.quiet: logging.info("Reset successful") return 0 def set_user(args, users, reload): properties = [] if args.cpu != None: set_no_device_parm(args, common.OPTION_CPU, properties) if args.io != None: set_device_parm(args, common.OPTION_IO, properties) if args.iops != None: set_device_parm(args, common.OPTION_IOPS, properties) if args.mem != None: set_no_device_parm(args, common.OPTION_MEM, properties) if args.tasks != None: set_no_device_parm(args, common.OPTION_TASKS, properties) for user in users: set_properties(user, properties) if reload: systemctl_daemon_reload() if not args.quiet: return command_list(users) return 0 def command_set(args, users): if args.command == 'set' or args.command == 'set-user': set_user(args, users, True) else: users = pwd.getpwall() min_uid = common.get_min_uid() need_reload = False for listuser in users: if listuser.pw_uid >= min_uid: need_reload = True set_user(args, listuser, False) if need_reload: systemctl_daemon_reload() if not args.quiet: logging.info("Set successful") return 0 def do_pgm(args, users): logging.debug("Entering lscgctl, command: %s" % args.command) if 'reset' in args.command: ret = command_reset(args, users) elif 'list' in args.command: ret = command_list(users) elif 'set' in args.command: ret = command_set(args, users) else: common.fatal_error('Unexpected command: %s' % args.command) logging.debug("Exiting lscgctl") return ret def main(): args, users = init_pgm() return do_pgm(args, users) if __name__ == "__main__": main()