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.
86 lines
2.6 KiB
86 lines
2.6 KiB
#define NOMINMAX
|
|
#define _CRT_SECURE_NO_WARNINGS
|
|
|
|
#include <cli-parser.hpp>
|
|
#include <filesystem>
|
|
#include <llodctor/llodctor_pe.hpp>
|
|
#include <llosymbol_loader/llosymbol_loader_map.hpp>
|
|
|
|
int __cdecl main( int argc, const char *argv[] )
|
|
{
|
|
argparse::argument_parser_t cli_parser( "llo stage-one", "llo stage one demo" );
|
|
cli_parser.add_argument()
|
|
.names( { "-i", "--input" } )
|
|
.description( "input file, must be a supported file format..." )
|
|
.required( true );
|
|
|
|
cli_parser.add_argument()
|
|
.names( { "-s", "--symbols" } )
|
|
.description( "path to symbols file, must be a supported file format... pdb, map, etc..." );
|
|
|
|
cli_parser.enable_help();
|
|
auto err = cli_parser.parse( argc, argv );
|
|
|
|
if ( err )
|
|
{
|
|
std::cout << err << std::endl;
|
|
cli_parser.print_help();
|
|
return -1;
|
|
}
|
|
|
|
auto file_name{ cli_parser.get< std::string >( "input" ) };
|
|
|
|
if ( !std::filesystem::exists( file_name ) )
|
|
{
|
|
std::printf( "[!] input file does not exist...\n" );
|
|
return -1;
|
|
}
|
|
|
|
std::vector< std::uint8_t > image;
|
|
|
|
if ( !llo::utils::open_binary_file( file_name, image ) )
|
|
{
|
|
std::printf( "[!] failed to read file off disk...\n" );
|
|
return -1;
|
|
}
|
|
|
|
std::shared_ptr< llo::s1::symbol_loader_base_t > symbol_loader = nullptr;
|
|
|
|
if ( cli_parser.exists( "symbols" ) )
|
|
{
|
|
std::filesystem::path symbols_file{ cli_parser.get< std::string >( "symbols" ) };
|
|
if ( symbols_file.extension().compare( ".map" ) )
|
|
{
|
|
symbol_loader = std::make_shared< llo::s1::symbol_loader_map_t >( symbols_file.string() );
|
|
}
|
|
else if ( symbols_file.extension().compare( ".pdb" ) )
|
|
{
|
|
// TODO:
|
|
// symbol_loader = std::make_shared< llo::s1::symbol_loader_pdb_t >( symbols_file );
|
|
}
|
|
}
|
|
|
|
std::shared_ptr< llo::s1::dctor_base_t > file_dctor = nullptr;
|
|
std::filesystem::path file_path{ file_name };
|
|
|
|
if ( file_path.extension().compare( ".exe" ) || file_path.extension().compare( ".dll" ) ||
|
|
file_path.extension().compare( ".sys" ) )
|
|
{
|
|
file_dctor = std::make_shared< llo::s1::dctor_pe_t >( image, symbol_loader.get() );
|
|
}
|
|
else if ( file_path.extension().compare( ".o" ) )
|
|
{
|
|
// TODO: elf file format...
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
std::printf( "[!] unknown file extension: %s\n", file_path.extension().c_str() );
|
|
return -1;
|
|
}
|
|
|
|
llo::lloiff_t iff( file_name );
|
|
file_dctor->generate( iff );
|
|
|
|
std::printf( "> number of sections = %d\n", iff.get_sections().size() );
|
|
} |