summaryrefslogtreecommitdiff
path: root/hypodermic/main.py
diff options
context:
space:
mode:
authorjakob <jakob@memeware.net>2017-08-29 17:34:30 -0400
committerjakob <jakob@memeware.net>2017-08-29 17:34:30 -0400
commitede6bb6dd1002ff259381bd5620a792bd7dcb8b6 (patch)
tree402fee4da486eeb78bcafd5622a4807a7912ce09 /hypodermic/main.py
parent749c1d3e310a3df5d6f2ce10135129aa3f6142b7 (diff)
Fixed library symbolic link for development tree.
Diffstat (limited to 'hypodermic/main.py')
-rw-r--r--hypodermic/main.py19
1 files changed, 17 insertions, 2 deletions
diff --git a/hypodermic/main.py b/hypodermic/main.py
index b0d709b..0f77de5 100644
--- a/hypodermic/main.py
+++ b/hypodermic/main.py
@@ -20,9 +20,7 @@
import argparse
import sys
import textwrap
-import pprint #
-from hypodermic.memory import maps
from hypodermic.ptrace import Process
@@ -43,6 +41,13 @@ class CustomHelp(argparse.HelpFormatter):
return both.add_usage(usage, actions, groups, prefix)
+def _alert(quiet: bool):
+ def alert(*args):
+ if not quiet:
+ print(*args)
+ return alert
+
+
def main():
parser = argparse.ArgumentParser(
add_help=False,
@@ -50,6 +55,13 @@ def main():
usage="%(prog)s [-a pid] [-c path] [options]",
description="Don't share needles, brah!"
)
+ parser._optionals.title = "Optional Arguments"
+
+ parser.add_argument(
+ "-q",
+ "--quiet",
+ help="Suppress everything but critical output"
+ )
doc = parser.add_argument_group("Documentation")
doc.add_argument(
@@ -82,12 +94,15 @@ def main():
)
args = parser.parse_args()
+ alert = _alert(args.quiet)
if args.attach is None and args.create is None:
print("No action specified. Quitting!")
sys.exit(1)
if args.create:
+ alert("Creating process at path {}".format(args.create))
p = Process(path=args.create)
else:
+ alert("Attaching to process with pid {}".format(args.attach))
p = Process(pid=args.attach)