You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.9 KiB
106 lines
3.9 KiB
#include <Windows.h>
|
|
#include <cli-parser.hpp>
|
|
#include <vmp_rtn.hpp>
|
|
#include <vmprofiler.hpp>
|
|
#include <xtils.hpp>
|
|
|
|
using namespace llvm;
|
|
|
|
int main( int argc, const char *argv[] )
|
|
{
|
|
argparse::argument_parser_t parser( "vmdevirt", "virtual instruction pseudo code generator" );
|
|
parser.add_argument().name( "--vmp2file" ).required( true ).description( "path to .vmp2 file..." );
|
|
|
|
parser.enable_help();
|
|
auto err = parser.parse( argc, argv );
|
|
|
|
if ( err )
|
|
{
|
|
std::cout << err << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
if ( parser.exists( "help" ) )
|
|
{
|
|
parser.print_help();
|
|
return 0;
|
|
}
|
|
|
|
std::vector< std::uint8_t > vmp2file;
|
|
const auto umtils = xtils::um_t::get_instance();
|
|
|
|
if ( !umtils->open_binary_file( parser.get< std::string >( "vmp2file" ), vmp2file ) )
|
|
{
|
|
std::printf( "[!] failed to open vmp2 file...\n" );
|
|
return -1;
|
|
}
|
|
|
|
const auto file_header = reinterpret_cast< vmp2::v3::file_header * >( vmp2file.data() );
|
|
|
|
if ( file_header->version != vmp2::version_t::v3 )
|
|
{
|
|
std::printf( "[!] invalid vmp2 file version... this build uses v3...\n" );
|
|
return -1;
|
|
}
|
|
|
|
auto first_block = reinterpret_cast< vmp2::v3::code_block_t * >( reinterpret_cast< std::uintptr_t >( file_header ) +
|
|
file_header->code_block_offset );
|
|
|
|
std::vector< vm::instrs::code_block_t > vmp_code_blocks;
|
|
// convert code blocks back to vm::instrs::code_block_t form...
|
|
for ( auto [ code_block, code_block_num ] = std::tuple{ first_block, 0u };
|
|
code_block_num < file_header->code_block_count;
|
|
code_block = reinterpret_cast< vmp2::v3::code_block_t * >( reinterpret_cast< std::uintptr_t >( code_block ) +
|
|
code_block->next_block_offset ),
|
|
++code_block_num )
|
|
{
|
|
vm::instrs::code_block_t _code_block{ code_block->vip_begin, code_block->jcc };
|
|
std::for_each( code_block->vinstr, code_block->vinstr + code_block->vinstr_count,
|
|
[ & ]( const vm::instrs::virt_instr_t &vinstr ) { _code_block.vinstrs.push_back( vinstr ); } );
|
|
|
|
vmp_code_blocks.push_back( _code_block );
|
|
}
|
|
|
|
LLVMContext llvm_ctx;
|
|
Module llvm_module( "", llvm_ctx );
|
|
|
|
vm::ctx_t vm_ctx( reinterpret_cast< std::uintptr_t >( file_header ) + file_header->module_offset,
|
|
file_header->image_base, file_header->module_size, file_header->vm_entry_rva );
|
|
|
|
if ( !vm_ctx.init() )
|
|
{
|
|
std::printf( "> failed to init vm::ctx_t... this can be for many reason... make sure you are using the correct "
|
|
"vmemu version for this project...\n" );
|
|
|
|
return false;
|
|
}
|
|
|
|
vm::vmp_rtn_t vmp_rtn( &llvm_ctx, &llvm_module, &vm_ctx, first_block->vip_begin, vmp_code_blocks );
|
|
auto func = vmp_rtn.lift();
|
|
llvm_module.print( llvm::outs(), nullptr );
|
|
|
|
llvm::LLVMInitializeX86TargetInfo();
|
|
llvm::LLVMInitializeX86Target();
|
|
llvm::LLVMInitializeX86TargetMC();
|
|
llvm::LLVMInitializeX86AsmParser();
|
|
llvm::LLVMInitializeX86AsmPrinter();
|
|
|
|
TargetOptions opt;
|
|
llvm::SmallVector< char, 128 > buff;
|
|
llvm::raw_svector_ostream dest( buff );
|
|
legacy::PassManager pass;
|
|
|
|
auto TargetTriple = sys::getDefaultTargetTriple();
|
|
llvm_module.setTargetTriple( TargetTriple );
|
|
|
|
std::string Error;
|
|
auto Target = TargetRegistry::lookupTarget( TargetTriple, Error );
|
|
|
|
auto reloc_model = Optional< Reloc::Model >();
|
|
auto target_machine = Target->createTargetMachine( TargetTriple, "generic", "", opt, reloc_model );
|
|
llvm_module.setDataLayout( target_machine->createDataLayout() );
|
|
target_machine->addPassesToEmitFile( pass, dest, nullptr, CGFT_ObjectFile );
|
|
pass.run( llvm_module );
|
|
|
|
std::printf( "> obj size = %d\n", buff.size() );
|
|
} |