updating streaming of filebrowser data

CHANGELOG.MD

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [3.3.0-rc12] - 2024-08-02
+
+### Changed
+
+- Updated the logic when downloading files to also update the timestamp on the mythictree entries so that the data is streamed to the UI properly
+
 ## [3.3.0-rc11] - 2024-08-01 
 
 ### Changed

MythicReactUI/CHANGELOG.MD

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.2.12] - 2024-08-02
+
+### Changed
+
+- Fixed the create credential dialog while issuing tasks using the old graphql mutation
+
 ## [0.2.11] - 2024-08-02
 
 ### Changed

MythicReactUI/src/components/pages/Callbacks/TaskParametersDialogRow.js

@@ -15,7 +15,7 @@ import DeleteIcon from '@mui/icons-material/Delete';
 import {useTheme} from '@mui/material/styles';
 import CancelIcon from '@mui/icons-material/Cancel';
 import {Typography} from '@mui/material';
-import {useMutation, gql } from '@apollo/client';
+import {useMutation, gql, useLazyQuery } from '@apollo/client';
 import { snackActions } from '../../utilities/Snackbar';
 import {CredentialTableNewCredentialDialog} from '../Search/CredentialTableNewCredentialDialog';
 import { MythicDialog } from '../../MythicComponents/MythicDialog';
@@ -60,13 +60,22 @@ fragment credentialData on credential{
 }
 `;
 const createCredentialMutation = gql`
-${credentialFragment}
-mutation createCredential($comment: String!, $account: String!, $realm: String!, $type: String!, $credential: bytea!) {
-    insert_credential_one(object: {account: $account, credential_raw: $credential, comment: $comment, realm: $realm, type: $type}) {
-      ...credentialData
+mutation createCredential($comment: String!, $account: String!, $realm: String!, $type: String!, $credential: String!) {
+    createCredential(account: $account, credential: $credential, comment: $comment, realm: $realm, credential_type: $type) {
+      id
+      status
+      error
     }
   }
 `;
+const getCredentialQuery = gql`
+${credentialFragment}
+query getCredential($id: Int!){
+    credential_by_pk(id: $id){
+        ...credentialData
+    }
+}
+`;
 
 export function TaskParametersDialogRow(props){
     const [value, setValue] = React.useState('');
@@ -157,13 +166,25 @@ export function TaskParametersDialogRow(props){
             console.log(data);
             setBackdropOpen(false);
         }
+    });
+    const [getCredential] = useLazyQuery(getCredentialQuery, {
+        onCompleted: (data) => {
+            updateToLatestCredential.current = true;
+            props.addedCredential(data.credential_by_pk);
+        },
+        onError: (data) => {
+            console.log(data);
+        }
     })
     const [createCredential] = useMutation(createCredentialMutation, {
         fetchPolicy: "no-cache",
         onCompleted: (data) => {
             snackActions.success("Successfully created new credential");
-            updateToLatestCredential.current = true;
-            props.addedCredential(data.insert_credential_one);
+            if(data.createCredential.status === "success"){
+                getCredential({variables: {id: data.createCredential.id}});
+            } else {
+                snackActions.error(data.createCredential.error);
+            }
         },
         onError: (data) => {
             snackActions.error("Failed to create credential");

MythicReactUI/src/index.js

@@ -14,7 +14,7 @@ import {snackActions} from './components/utilities/Snackbar';
 import jwt_decode from 'jwt-decode';
 import {meState} from './cache';
 
-export const mythicUIVersion = "0.2.11";
+export const mythicUIVersion = "0.2.12";
 
 let fetchingNewToken = false;
 

VERSION

@@ -1 +1 @@
-3.3.0-rc11
+3.3.0-rc12

mythic-docker/src/VERSION

@@ -1 +1 @@
-3.3.0-rc11
+3.3.0-rc12

mythic-docker/src/rabbitmq/util_agent_message_actions_post_response.go

@@ -1354,7 +1354,7 @@ func associateFileMetaWithMythicTree(pathData utils.AnalyzedPath, fileMeta datab
 	fileMetaData := map[string]interface{}{
 		"access_time": time.Now().Unix(),
 		"modify_time": time.Now().Unix(),
-		"size":        fileMeta.ChunkSize * fileMeta.TotalChunks,
+		"size":        fileMeta.Size,
 		"permissions": map[string]interface{}{},
 	}
 	newTree.Metadata = GetMythicJSONTextFromStruct(fileMetaData)
@@ -1368,9 +1368,15 @@ func associateFileMetaWithMythicTree(pathData utils.AnalyzedPath, fileMeta datab
 	// now that we know the mythictree entry exists, associate this filemeta with it
 	fileMeta.MythicTreeID.Valid = true
 	fileMeta.MythicTreeID.Int64 = int64(newTree.ID)
-	if _, err := database.DB.NamedExec(`UPDATE filemeta SET mythictree_id=:mythictree_id WHERE id=:id`, fileMeta); err != nil {
+	_, err := database.DB.NamedExec(`UPDATE filemeta SET mythictree_id=:mythictree_id WHERE id=:id`, fileMeta)
+	if err != nil {
 		logging.LogError(err, "Failed to associate filemeta with mythictree ")
 		go SendAllOperationsMessage(fmt.Sprintf("Failed to associate file with file browser: %s\n", fileMeta.AgentFileID), task.OperationID, "", database.MESSAGE_LEVEL_WARNING)
+		return
+	}
+	_, err = database.DB.Exec(`UPDATE mythictree SET "timestamp"=now() WHERE id=$1`, newTree.ID)
+	if err != nil {
+		logging.LogError(err, "failed to update timestamp on mythictree to indicate new file association happened")
 	}
 }
 func addFilePermissions(fileBrowser *agentMessagePostResponseFileBrowser) map[string]interface{} {
@@ -2000,8 +2006,14 @@ func addFileMetaToMythicTree(task databaseStructs.Task, newFile databaseStructs.
 	} else if err == nil {
 		newFile.MythicTreeID.Int64 = int64(fileBrowser.ID)
 		newFile.MythicTreeID.Valid = true
-		if _, err := database.DB.NamedExec(`UPDATE filemeta SET mythictree_id=:mythictree_id WHERE id=:id`, newFile); err != nil {
+		_, err = database.DB.NamedExec(`UPDATE filemeta SET mythictree_id=:mythictree_id WHERE id=:id`, newFile)
+		if err != nil {
 			logging.LogError(err, "Failed to update file meta with mythic tree id")
+			return
+		}
+		_, err = database.DB.Exec(`UPDATE mythictree SET "timestamp"=now() WHERE id=$1`, fileBrowser.ID)
+		if err != nil {
+			logging.LogError(err, "failed to update timestamp on mythictree to indicate new file association happened")
 		}
 
 	} else {

mythic-react-docker/mythic/public/asset-manifest.json

@@ -1,15 +1,15 @@
 {
   "files": {
     "main.css": "/new/static/css/main.7e143bf2.css",
-    "main.js": "/new/static/js/main.220682dd.js",
+    "main.js": "/new/static/js/main.e6eb2feb.js",
     "static/media/mythic-red.png": "/new/static/media/mythic-red.203468a4e5240d239aa0.png",
     "static/media/mythic_red_small.svg": "/new/static/media/mythic_red_small.793b41cc7135cdede246661ec232976b.svg",
     "index.html": "/new/index.html",
     "main.7e143bf2.css.map": "/new/static/css/main.7e143bf2.css.map",
-    "main.220682dd.js.map": "/new/static/js/main.220682dd.js.map"
+    "main.e6eb2feb.js.map": "/new/static/js/main.e6eb2feb.js.map"
   },
   "entrypoints": [
     "static/css/main.7e143bf2.css",
-    "static/js/main.220682dd.js"
+    "static/js/main.e6eb2feb.js"
   ]
 }

mythic-react-docker/mythic/public/index.html

@@ -1 +1 @@
-<!doctype html><html lang="en"><head><meta charset="utf-8"/><link rel="icon" href="/new/favicon.ico"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><link rel="apple-touch-icon" href="/new/logo192.png"/><link rel="manifest" href="/new/manifest.json"/><title>Mythic</title><script defer="defer" src="/new/static/js/main.220682dd.js"></script><link href="/new/static/css/main.7e143bf2.css" rel="stylesheet"></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>
+<!doctype html><html lang="en"><head><meta charset="utf-8"/><link rel="icon" href="/new/favicon.ico"/><meta name="viewport" content="width=device-width,initial-scale=1"/><meta name="theme-color" content="#000000"/><link rel="apple-touch-icon" href="/new/logo192.png"/><link rel="manifest" href="/new/manifest.json"/><title>Mythic</title><script defer="defer" src="/new/static/js/main.e6eb2feb.js"></script><link href="/new/static/css/main.7e143bf2.css" rel="stylesheet"></head><body><noscript>You need to enable JavaScript to run this app.</noscript><div id="root"></div></body></html>