Edit File: lsnsctl
#!/usr/bin/python3 import argparse, subprocess, sys, os, logging, json from stat import * import common def command_requires_uid(command): return command in ['disable-uid', 'enable-uid', 'set-min-uid', 'unmount'] def command_uses_uid(command): return command in ['disable-uid', 'enable-uid', 'list-disabled-uids', 'list-mount', 'set-min-uid', 'unmount'] def validate_environment(): if os.getuid() != 0: common.fatal_error("this program must be run as root") common.ls_ok() def init_pgm(): common.init_logging() parser = argparse.ArgumentParser(prog="lsnsctl", description='LiteSpeed Namespace Control Program') parser.add_argument("command", type=str, help="The namespace command", choices=['disable-uid', 'enable-uid', 'get-min-uid', 'list-disabled-uids', 'list-mount', 'set-min-uid', 'unmount', 'unmount-all', 'version']) parser.add_argument("--uid", type=str, help="uid or user name for set-min-uid and unmount commands") 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('-q', '--quiet', action='store_true', help='turns off all logging and only outputs what is requested.') 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) and args.uid == None: common.fatal_error("You must specify a --uid for the %s command" % command) if command_uses_uid(command) and args.uid != None: user, found = common.get_user(args.uid) else: user = None if command == 'version': logging.info("Version: %s" % common.VERSION) if args.quiet: print(common.VERSION) sys.exit(0) return args, user def command_get_min_uid(args): uid = common.get_min_uid() if args.quiet: print('%d' % uid) else: logging.info("Minimum UID: %d" % uid) def command_set_min_uid(args, user): fullfile = common.get_conf_file('lsns.conf') try: f = open(fullfile, 'w') except Exception as err: common.fatal_error('Error opening %s: %s' % (fullfile, err)) try: f.write('%d\n' % user.pw_uid) f.close() except Exception as err: common.fatal_error('Error writing %s: %s' % (fullfile, err)) command_get_min_uid(args) def get_mounted(): result = subprocess.run(['mount'], capture_output=True, text=True) if result.returncode != 0: common.fatal_error('Error in running: mount, errors: ' + result.stdout + ' ' + result.stderr) mounted = {} stdout_lines = result.stdout.split('\n') for line in stdout_lines: logging.debug('mount line: %s' % line) index = line.find("/var/lsns/") if index == -1: continue remainder = line[(index + 10):] logging.debug(' After prefix: %s' % remainder) components = remainder.split('/') if len(components) < 2: continue logging.debug(' UID: %s, vhost: %s' % (components[0], components[1])) if not components[0] in mounted: logging.debug(' New UID/vhost') mounted[components[0]] = [ components[1] ] elif components[1] in mounted[components[0]]: logging.debug(' Existing UID/vhost') else: logging.debug(' New vhost only') mounted[components[0]].append(components[1]) return mounted def command_list_mount(args, user): mounted = get_mounted() if user == None: print(json.dumps(mounted)) else: if str(user.pw_uid) in mounted: mount = { str(user.pw_uid): mounted[str(user.pw_uid)]} else: mount = {} print(json.dumps(mount)) def get_disabled_uids(missing_ok): try: f = open(common.get_disabled_uid_file(), "r") uids_raw = f.read() f.close() except Exception as err: if not missing_ok: common.fatal_error('Error opening UID file: %s' % err) else: return {} lines = uids_raw.split('\n') uids = {} for line in lines: if line in uids or line == '': continue uids[line] = True if not missing_ok and len(uids) == 0: common.fatal_error("No uids in disabled UIDs file") return uids def write_disabled_uids(uids): try: f = open(common.get_disabled_uid_file(), "w") for k in uids: f.write("%s\n" % k) f.close() except Exception as err: common.fatal_error('Error writing UID file: %s' % err) def command_disable_uid(args, user): uids = get_disabled_uids(True) if str(user.pw_uid) in uids: if args.quiet: sys.exit(0) logging.info('UID %s already disabled' % str(user.pw_uid)) sys.exit(0) uids[str(user.pw_uid)] = True write_disabled_uids(uids) common.restart_external([user], False) if not args.quiet: logging.info("Write %s to disabled uid list" % str(user.pw_uid)) def command_list_disabled_uids(args, user): uids = get_disabled_uids(True) uids = sorted(uids) print(json.dumps(uids)) def command_enable_uid(args, user): uids = get_disabled_uids(True) if not str(user.pw_uid) in uids: if args.quiet: sys.exit(0) logging.info('UID %s already enabled' % str(user.pw_uid)) sys.exit(0) del uids[str(user.pw_uid)] if len(uids) == 0: os.remove(common.get_disabled_uid_file()) if not args.quiet: logging.info('Deleting last disabled uid %s from file' % str(user.pw_uid)) else: write_disabled_uids(uids) if not args.quiet: logging.info("Deleting %s from disabled uid list", str(user.pw_uid)) common.restart_external([user], False) def command_unmount(args, user): unmount_args = [common.get_bin_file('unmount_ns')] if args.command == 'unmount-all': unmount_args.append('-a') else: unmount_args.append('-u') unmount_args.append(str(user.pw_uid)) result = subprocess.run(unmount_args, capture_output=True, text=True) if result.returncode != 0: common.fatal_error('Error in running: unmount, errors: ' + result.stdout + ' ' + result.stderr) if not args.quiet: print(result.stdout) print(result.stderr) if args.command == 'unmount-all': common.restart_external(None, True) else: common.restart_external([user], False) def do_pgm(args, user): logging.debug("Entering lsnsctl, command: %s" % args.command) if 'get-min-uid' == args.command: command_get_min_uid(args) elif 'list-mount' == args.command: command_list_mount(args, user) elif 'disable-uid' == args.command: command_disable_uid(args, user) elif 'enable-uid' == args.command: command_enable_uid(args, user) elif 'list-disabled-uids' == args.command: command_list_disabled_uids(args, user) elif 'set-min-uid' == args.command: command_set_min_uid(args, user) elif 'unmount' == args.command or 'unmount-all' == args.command: command_unmount(args, user) logging.debug("Exiting lsnsctl") def main(): args, user = init_pgm() return do_pgm(args, user) if __name__ == "__main__": main()