1 files changed, 114 insertions, 0 deletions
diff --git a/scripts/internal/print_access_denied.py b/scripts/internal/print_access_denied.py
new file mode 100644
index 00000000..8381928d
--- /dev/null
+++ b/scripts/internal/print_access_denied.py
@@ -0,0 +1,114 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2009, Giampaolo Rodola'. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""
+Helper script iterates over all processes and .
+It prints how many AccessDenied exceptions are raised in total and
+for what Process method.
+
+$ make print-access-denied
+username             0     0.0%   SUCCESS
+cpu_num              0     0.0%   SUCCESS
+num_ctx_switches     0     0.0%   SUCCESS
+pid                  0     0.0%   SUCCESS
+cmdline              0     0.0%   SUCCESS
+create_time          0     0.0%   SUCCESS
+ionice               0     0.0%   SUCCESS
+cpu_percent          0     0.0%   SUCCESS
+terminal             0     0.0%   SUCCESS
+ppid                 0     0.0%   SUCCESS
+nice                 0     0.0%   SUCCESS
+status               0     0.0%   SUCCESS
+cpu_times            0     0.0%   SUCCESS
+memory_info          0     0.0%   SUCCESS
+threads              0     0.0%   SUCCESS
+uids                 0     0.0%   SUCCESS
+num_threads          0     0.0%   SUCCESS
+name                 0     0.0%   SUCCESS
+gids                 0     0.0%   SUCCESS
+cpu_affinity         0     0.0%   SUCCESS
+memory_percent       0     0.0%   SUCCESS
+memory_full_info     70   20.8%   ACCESS DENIED
+memory_maps          70   20.8%   ACCESS DENIED
+exe                  174  51.8%   ACCESS DENIED
+connections          237  70.5%   ACCESS DENIED
+num_fds              237  70.5%   ACCESS DENIED
+cwd                  237  70.5%   ACCESS DENIED
+io_counters          237  70.5%   ACCESS DENIED
+open_files           237  70.5%   ACCESS DENIED
+environ              237  70.5%   ACCESS DENIED
+--------------------------
+total:                1736 (336 total processes)
+"""
+
+from __future__ import print_function, division
+from collections import defaultdict
+import sys
+
+import psutil
+
+
+def term_supports_colors(file=sys.stdout):
+    try:
+        import curses
+        assert file.isatty()
+        curses.setupterm()
+        assert curses.tigetnum("colors") > 0
+    except Exception:
+        return False
+    else:
+        return True
+
+
+COLORS = term_supports_colors()
+
+
+def hilite(s, ok=True, bold=False):
+    """Return an highlighted version of 'string'."""
+    if not COLORS:
+        return s
+    attr = []
+    if ok is None:  # no color
+        pass
+    elif ok:   # green
+        attr.append('32')
+    else:   # red
+        attr.append('31')
+    if bold:
+        attr.append('1')
+    return '\x1b[%sm%s\x1b[0m' % (';'.join(attr), s)
+
+
+def main():
+    tot_procs = 0
+    tot_ads = 0
+    signaler = object()
+    d = defaultdict(int)
+    for p in psutil.process_iter(attrs=[], ad_value=signaler):
+        tot_procs += 1
+        for methname, value in p.info.items():
+            if value is signaler:
+                tot_ads += 1
+                d[methname] += 1
+            else:
+                d[methname] += 0
+
+    for methname, ads in sorted(d.items(), key=lambda x: x[1]):
+        perc = (ads / tot_procs) * 100
+        s = "%-20s %-3s %5.1f%%   " % (methname, ads, perc)
+        if not ads:
+            s += "SUCCESS"
+            s = hilite(s, ok=True)
+        else:
+            s += "ACCESS DENIED"
+            s = hilite(s, ok=False)
+        print(s)
+    print("--------------------------")
+    print("total: %19s (%s total processes)" % (tot_ads, tot_procs))
+
+
+if __name__ == '__main__':
+    main()