Expose the ping command to the CLI.

sensepost · Apr 28, 2020 · fee42b3 · fee42b3
1 parent a69fffc
commit fee42b3
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/objection/commands/frida_commands.py b/objection/commands/frida_commands.py
@@ -40,6 +40,21 @@ def frida_environment(args: list = None) -> None:
     ]))
 
 
+def ping(args: list = None) -> None:
+    """
+        Pings the agent.
+
+        :param args:
+        :return:
+    """
+
+    agent = state_connection.get_api()
+    if agent.ping():
+        click.secho('The agent responds ok!', fg='green')
+    else:
+        click.secho('The agent did not respond ok!', fg='red')
+
+
 def load_background(args: list = None) -> None:
     """
         Loads a Frida script and runs it in the background.

diff --git a/objection/console/commands.py b/objection/console/commands.py
@@ -69,6 +69,11 @@
         'exec': frida_commands.load_background
     },
 
+    'ping': {
+        'meta': 'Ping the injected agent',
+        'exec': frida_commands.ping
+    },
+
     # file manager commands
 
     'cd': {