v0.1.7

added support for interactive tasking and better file tracking for tasks

Author: its-a-feature

mythic/graphql_queries.py

@@ -26,8 +26,8 @@
 )
 create_task = gql(
     """
-    mutation createTasking($callback_id: Int!, $command: String!, $params: String!, $token_id: Int, $tasking_location: String, $original_params: String, $parameter_group_name: String) {
-        createTask(callback_id: $callback_id, command: $command, params: $params, token_id: $token_id, tasking_location: $tasking_location, original_params: $original_params, parameter_group_name: $parameter_group_name) {
+    mutation createTasking($callback_id: Int!, $command: String!, $params: String!, $files: [String], $token_id: Int, $tasking_location: String, $original_params: String, $parameter_group_name: String, $is_interactive_task: Boolean, $interactive_task_type: Int, $parent_task_id: Int) {
+        createTask(callback_id: $callback_id, command: $command, params: $params, files: $files, token_id: $token_id, tasking_location: $tasking_location, original_params: $original_params, parameter_group_name: $parameter_group_name, is_interactive_task: $is_interactive_task, interactive_task_type: $interactive_task_type, parent_task_id: $parent_task_id) {
             status
             id
             display_id

mythic/mythic.py

@@ -576,13 +576,18 @@ async def issue_task(
         token_id: int = None,
         wait_for_complete: bool = False,
         custom_return_attributes: str = None,
+        file_ids: [str] = None,
         timeout: int = None,
+        is_interactive_task: bool = False,
+        interactive_task_type: int = None,
+        parent_task_id: int = None,
 ) -> dict:
     """
     Create a new task within Mythic for a specific callback.
     `return_on_status` indicates if this command should return immediately, mythic_classes.MythicStatus.Preprocessing, or wait for a certain status before returning.
         This can be helpful if you want ot make sure a task is completed before continuing
     If you have files that you need to upload and leverage as part of your tasking, use the `register_file` function to get back a file_id.
+        Then supply those file ids in the `file_ids` array AND as their appropriate
     If you return immediately from this task, you'll get a dictionary with a status, error, and id field for your new task.
     If you return on another status, you can use your own custom attributes or use the defaults outlined in graphql_queries.task_fragment.
     """
@@ -597,7 +602,11 @@ async def issue_task(
             "command": command_name,
             "params": parameter_string,
             "token_id": token_id,
+            "is_interactive_task": is_interactive_task,
+            "interactive_task_type": interactive_task_type,
+            "parent_task_id": parent_task_id,
             "tasking_location": "command_line" if isinstance(parameters, str) else "scripting",
+            "files": file_ids,
         },
     )
     if submission_status["createTask"]["status"] == "success":

mythic/mythic_classes.py

@@ -35,7 +35,7 @@ def __init__(
         self.http = "http://" if not ssl else "https://"
         self.ws = "ws://" if not ssl else "wss://"
         self.global_timeout = global_timeout if global_timeout is not None else -1
-        self.scripting_version = "0.1.6"
+        self.scripting_version = "0.1.7"
         self.current_operation_id = 0
         self.schema = schema
         self.log_level = log_level

setup.py

@@ -11,7 +11,7 @@
 # This call to setup() does all the work
 setup(
     name="mythic",
-    version="0.1.6",
+    version="0.1.7",
     description="Interact with Mythic C2 Framework Instances",
     long_description=README,
     long_description_content_type="text/markdown",