def test_get_services(self):
"""Test get_services"""
self.comm.query_systemctl = self._mock_query_systemctl
self.assertEqual(common.get_services(self.query), self.golden)
def test_get_services_with_broken_systemctl(self):
"""Test get_services with broken systctl"""
self.comm.query_systemctl = self._mock_query_systemctl_broken
self.assertIn("Dummy Reason", common.get_services(self.query))
def main(argv):
"""Main program"""
parser = argparse.ArgumentParser()
parser.add_argument('--version', action='version',
version='%%(prog)s %s' % (__version__))
parser.add_argument("-m", "--machine-readable", action="store_true",
help="Output machine readable info")
parser.add_argument("-s", "--showlibs", action="store_true",
help="In human readable mode, show deleted libs")
parser.add_argument("-S", "--services", action="store_true",
help="Try to find systemd services for lib users")
parser.add_argument("-i", "--ignore-pattern", default=[],
metavar="GLOB", action='append',
help="Ignore deleted files matching %(metavar)s. "
"Can be specified multiple times.")
parser.add_argument("-I", "--ignore-literal", default=[],
metavar="LITERAL", action='append',
help="Ignore deleted files named %(metavar)s. "
"Can be specified multiple times.")
options = parser.parse_args(argv)
options.showitems = options.showlibs
NOLIBSPT.update(options.ignore_pattern)
NOLIBSNP.update(options.ignore_literal)
users = defaultdict(lambda: (set(), set()))
read_failure = False
for map_filename in glob.glob(common.LIBPROCFSPAT):
deletedlibs = set()
try:
pid = normpath(map_filename).split("/")[2]
except IndexError:
# This happens if the filenames look different
# than we expect (e.g. the user changed common.LIBPROCFSPAT)
pid = "unknown"
try:
mapsfile = open(map_filename)
deletedlibs = get_deleted_libs(mapsfile)
except IOError as exc:
read_failure = True
continue
mapsfile.close()
if deletedlibs:
argv = common.get_progargs(pid)
if not argv:
continue
users[argv][0].add(pid)
users[argv][1].update(deletedlibs)
if read_failure:
if os.geteuid() == 0:
sys.stderr.write(PERMWARNINGUID0)
else:
sys.stderr.write(PERMWARNING)
if len(users) > 0:
if options.machine_readable:
print(common.fmt_machine(users))
else:
print(common.fmt_human(users, options))
if options.services:
print()
print(common.get_services(users))