bump v0.1.1

added start/stop functions for c2 profiles and ability to save c2 instances

Author: its-a-feature

mythic/mythic.py

@@ -2172,6 +2172,72 @@ async def send_custom_webhook_message(
                                                                                       "webhook_type": webhook_type})
 
 
+# ####### C2 Functions #############
+
+async def start_stop_c2_profile(
+        mythic: mythic_classes.Mythic,
+        c2_profile_name: str,
+        action: str = "start"
+):
+    query = """
+    query getC2IdFromName($c2_name: String!) {
+      c2profile(where: {name: {_eq: $c2_name}}){
+        id
+      }
+    }
+    """
+    resp = await mythic_utilities.graphql_post(mythic=mythic, query=query, variables={
+        "c2_name": c2_profile_name,
+    })
+    if len(resp["c2profile"]) == 0:
+        raise Exception(f"Failed to find c2 profile {c2_profile_name}")
+    start_stop = """
+    mutation startStopC2Profile($id: Int!, $action: String!){
+        startStopProfile(id: $id, action: $action){
+            status
+            error
+            output
+        }
+    }
+    """
+    resp = await mythic_utilities.graphql_post(mythic=mythic, query=start_stop, variables={
+        "id": resp["c2profile"][0]["id"], "action": action,
+    })
+    return resp["startStopProfile"]
+
+
+async def create_saved_c2_instance(
+        mythic: mythic_classes.Mythic,
+        instance_name: str,
+        c2_profile_name: str,
+        c2_parameters: dict
+):
+    query = """
+    query getC2IdFromName($c2_name: String!) {
+      c2profile(where: {name: {_eq: $c2_name}}){
+        id
+      }
+    }
+    """
+    resp = await mythic_utilities.graphql_post(mythic=mythic, query=query, variables={
+        "c2_name": c2_profile_name,
+    })
+    if len(resp["c2profile"]) == 0:
+        raise Exception(f"Failed to find c2 profile {c2_profile_name}")
+    mutation = """
+    mutation createNewC2Instance($instance_name: String!, $c2_instance: String!, $c2profile_id: Int!){
+      create_c2_instance(c2_instance: $c2_instance, instance_name: $instance_name, c2profile_id: $c2profile_id){
+        status
+        error
+      }
+    }
+    """
+    resp = await mythic_utilities.graphql_post(mythic=mythic, query=mutation, variables={
+        "instance_name": instance_name, "c2profile_id": resp["c2profile"][0]["id"], "c2_instance": json.dumps(c2_parameters)
+    })
+    return resp["create_c2_instance"]
+
+
 # ####### Tag Functions ############
 
 

mythic/mythic_classes.py

@@ -26,7 +26,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.0"
+        self.scripting_version = "0.1.1"
         self.current_operation_id = 0
         self.schema = schema
 

mythic/mythic_utilities.py

@@ -70,7 +70,7 @@ async def get_ws_transport(mythic: mythic_classes.Mythic) -> WebsocketsTransport
     websocket_transport = WebsocketsTransport(
         url=f"{mythic.ws}{mythic.server_ip}:{mythic.server_port}/graphql/",
         headers=get_headers(mythic),
-        ssl=ssl_context,
+        ssl=ssl_context if mythic.ssl else False,
     )
     return websocket_transport
 

setup.py

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