Updated client to use new file endpoint (#610)

BC-SECURITY · May 17, 2023 · c352619 · c352619
1 parent 79a9a5c
commit c352619
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/empire/client/src/menus/InteractMenu.py b/empire/client/src/menus/InteractMenu.py
@@ -503,14 +503,14 @@ def execute_shortcut(self, command_name: str, params: List[str]):
                         try:
                             file_directory = post_body.get("options")["File"]
                             filename = file_directory.split("/")[-1]
-                            post_body.get("options")["File"] = filename
                             data = get_data_from_file(file_directory)
                         except Exception:
                             log.error("Invalid filename or file does not exist")
                             return
                         response = state.upload_file(filename, data)
                         if "id" in response.keys():
                             log.info("File uploaded to server successfully")
+                            post_body.get("options")["File"] = response["id"]
                         elif "detail" in response.keys():
                             if response["detail"].startswith("[!]"):
                                 msg = response["detail"]

diff --git a/empire/client/src/menus/UseModuleMenu.py b/empire/client/src/menus/UseModuleMenu.py
@@ -83,14 +83,14 @@ def execute(self):
                 try:
                     file_directory = self.record_options["File"]["value"]
                     filename = file_directory.split("/")[-1]
-                    self.record_options["File"]["value"] = filename
                     data = get_data_from_file(file_directory)
                 except Exception:
                     log.error("Invalid filename or file does not exist")
                     return
                 response = state.upload_file(filename, data)
                 if "id" in response.keys():
                     log.info("File uploaded to server successfully")
+                    self.record_options["File"]["value"] = response["id"]
 
                 elif "detail" in response.keys():
                     if response["detail"].startswith("[!]"):