mirror of
https://github.com/godotengine/godot.git
synced 2024-12-09 10:09:20 +08:00
1681 lines
52 KiB
C++
1681 lines
52 KiB
C++
/*************************************************************************/
|
|
/* Copyright (c) 2015 dx, http://kaimi.ru */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person */
|
|
/* obtaining a copy of this software and associated documentation */
|
|
/* files (the "Software"), to deal in the Software without */
|
|
/* restriction, including without limitation the rights to use, */
|
|
/* copy, modify, merge, publish, distribute, sublicense, and/or */
|
|
/* sell copies of the Software, and to permit persons to whom the */
|
|
/* Software is furnished to do so, subject to the following conditions: */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
#include <string>
|
|
#include <vector>
|
|
#include <istream>
|
|
#include <ostream>
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
#include <set>
|
|
#include <string.h>
|
|
#include "pe_exception.h"
|
|
#include "pe_base.h"
|
|
|
|
namespace pe_bliss
|
|
{
|
|
using namespace pe_win;
|
|
|
|
//Constructor
|
|
pe_base::pe_base(std::istream& file, const pe_properties& props, bool read_debug_raw_data)
|
|
{
|
|
props_ = props.duplicate().release();
|
|
|
|
//Save istream state
|
|
std::ios_base::iostate state = file.exceptions();
|
|
std::streamoff old_offset = file.tellg();
|
|
|
|
try
|
|
{
|
|
file.exceptions(std::ios::goodbit);
|
|
//Read DOS header, PE headers and section data
|
|
read_dos_header(file);
|
|
read_pe(file, read_debug_raw_data);
|
|
}
|
|
catch(const std::exception&)
|
|
{
|
|
//If something went wrong, restore istream state
|
|
file.seekg(old_offset);
|
|
file.exceptions(state);
|
|
file.clear();
|
|
//Rethrow
|
|
throw;
|
|
}
|
|
|
|
//Restore istream state
|
|
file.seekg(old_offset);
|
|
file.exceptions(state);
|
|
file.clear();
|
|
}
|
|
|
|
pe_base::pe_base(const pe_properties& props, uint32_t section_alignment, bool dll, uint16_t subsystem)
|
|
{
|
|
props_ = props.duplicate().release();
|
|
props_->create_pe(section_alignment, subsystem);
|
|
|
|
has_overlay_ = false;
|
|
memset(&dos_header_, 0, sizeof(dos_header_));
|
|
|
|
dos_header_.e_magic = 0x5A4D; //"MZ"
|
|
//Magic numbers from MSVC++ build
|
|
dos_header_.e_maxalloc = 0xFFFF;
|
|
dos_header_.e_cblp = 0x90;
|
|
dos_header_.e_cp = 3;
|
|
dos_header_.e_cparhdr = 4;
|
|
dos_header_.e_sp = 0xB8;
|
|
dos_header_.e_lfarlc = 64;
|
|
|
|
set_characteristics(image_file_executable_image | image_file_relocs_stripped);
|
|
|
|
if(get_pe_type() == pe_type_32)
|
|
set_characteristics_flags(image_file_32bit_machine);
|
|
|
|
if(dll)
|
|
set_characteristics_flags(image_file_dll);
|
|
|
|
set_subsystem_version(5, 1); //WinXP
|
|
set_os_version(5, 1); //WinXP
|
|
}
|
|
|
|
pe_base::pe_base(const pe_base& pe)
|
|
:dos_header_(pe.dos_header_),
|
|
rich_overlay_(pe.rich_overlay_),
|
|
sections_(pe.sections_),
|
|
has_overlay_(pe.has_overlay_),
|
|
full_headers_data_(pe.full_headers_data_),
|
|
debug_data_(pe.debug_data_),
|
|
props_(0)
|
|
{
|
|
props_ = pe.props_->duplicate().release();
|
|
}
|
|
|
|
pe_base& pe_base::operator=(const pe_base& pe)
|
|
{
|
|
dos_header_ = pe.dos_header_;
|
|
rich_overlay_ = pe.rich_overlay_;
|
|
sections_ = pe.sections_;
|
|
has_overlay_ = pe.has_overlay_;
|
|
full_headers_data_ = pe.full_headers_data_;
|
|
debug_data_ = pe.debug_data_;
|
|
delete props_;
|
|
props_ = 0;
|
|
props_ = pe.props_->duplicate().release();
|
|
|
|
return *this;
|
|
}
|
|
|
|
pe_base::~pe_base()
|
|
{
|
|
delete props_;
|
|
}
|
|
|
|
//Returns dos header
|
|
const image_dos_header& pe_base::get_dos_header() const
|
|
{
|
|
return dos_header_;
|
|
}
|
|
|
|
//Returns dos header
|
|
image_dos_header& pe_base::get_dos_header()
|
|
{
|
|
return dos_header_;
|
|
}
|
|
|
|
//Returns PE headers start position (e_lfanew)
|
|
int32_t pe_base::get_pe_header_start() const
|
|
{
|
|
return dos_header_.e_lfanew;
|
|
}
|
|
|
|
//Strips MSVC stub overlay
|
|
void pe_base::strip_stub_overlay()
|
|
{
|
|
rich_overlay_.clear();
|
|
}
|
|
|
|
//Fills MSVC stub overlay with character c
|
|
void pe_base::fill_stub_overlay(char c)
|
|
{
|
|
if(rich_overlay_.length())
|
|
rich_overlay_.assign(rich_overlay_.length(), c);
|
|
}
|
|
|
|
//Sets stub MSVS overlay
|
|
void pe_base::set_stub_overlay(const std::string& data)
|
|
{
|
|
rich_overlay_ = data;
|
|
}
|
|
|
|
//Returns stub overlay
|
|
const std::string& pe_base::get_stub_overlay() const
|
|
{
|
|
return rich_overlay_;
|
|
}
|
|
|
|
//Realigns all sections
|
|
void pe_base::realign_all_sections()
|
|
{
|
|
for(unsigned int i = 0; i < sections_.size(); i++)
|
|
realign_section(i);
|
|
}
|
|
|
|
//Returns number of sections from PE header
|
|
uint16_t pe_base::get_number_of_sections() const
|
|
{
|
|
return props_->get_number_of_sections();
|
|
}
|
|
|
|
//Updates number of sections in PE header
|
|
uint16_t pe_base::update_number_of_sections()
|
|
{
|
|
uint16_t new_number = static_cast<uint16_t>(sections_.size());
|
|
props_->set_number_of_sections(new_number);
|
|
return new_number;
|
|
}
|
|
|
|
//Returns section alignment
|
|
uint32_t pe_base::get_section_alignment() const
|
|
{
|
|
return props_->get_section_alignment();
|
|
}
|
|
|
|
//Returns image sections list
|
|
section_list& pe_base::get_image_sections()
|
|
{
|
|
return sections_;
|
|
}
|
|
|
|
//Returns image sections list
|
|
const section_list& pe_base::get_image_sections() const
|
|
{
|
|
return sections_;
|
|
}
|
|
|
|
//Realigns section by index
|
|
void pe_base::realign_section(uint32_t index)
|
|
{
|
|
//Check index
|
|
if(sections_.size() <= index)
|
|
throw pe_exception("Section not found", pe_exception::section_not_found);
|
|
|
|
//Get section iterator
|
|
section_list::iterator it = sections_.begin() + index;
|
|
section& s = *it;
|
|
|
|
//Calculate, how many null bytes we have in the end of raw section data
|
|
std::size_t strip = 0;
|
|
for(std::size_t i = (*it).get_raw_data().length(); i >= 1; --i)
|
|
{
|
|
if(s.get_raw_data()[i - 1] == 0)
|
|
strip++;
|
|
else
|
|
break;
|
|
}
|
|
|
|
if(it == sections_.end() - 1) //If we're realigning the last section
|
|
{
|
|
//We can strip ending null bytes
|
|
s.set_size_of_raw_data(static_cast<uint32_t>(s.get_raw_data().length() - strip));
|
|
s.get_raw_data().resize(s.get_raw_data().length() - strip, 0);
|
|
}
|
|
else
|
|
{
|
|
//Else just set size of raw data
|
|
uint32_t raw_size_aligned = s.get_aligned_raw_size(get_file_alignment());
|
|
s.set_size_of_raw_data(raw_size_aligned);
|
|
s.get_raw_data().resize(raw_size_aligned, 0);
|
|
}
|
|
}
|
|
|
|
//Returns file alignment
|
|
uint32_t pe_base::get_file_alignment() const
|
|
{
|
|
return props_->get_file_alignment();
|
|
}
|
|
|
|
//Sets file alignment
|
|
void pe_base::set_file_alignment(uint32_t alignment)
|
|
{
|
|
//Check alignment
|
|
if(alignment < minimum_file_alignment)
|
|
throw pe_exception("File alignment can't be less than 512", pe_exception::incorrect_file_alignment);
|
|
|
|
if(!pe_utils::is_power_of_2(alignment))
|
|
throw pe_exception("File alignment must be a power of 2", pe_exception::incorrect_file_alignment);
|
|
|
|
if(alignment > get_section_alignment())
|
|
throw pe_exception("File alignment must be <= section alignment", pe_exception::incorrect_file_alignment);
|
|
|
|
//Set file alignment without any additional checks
|
|
set_file_alignment_unchecked(alignment);
|
|
}
|
|
|
|
//Returns size of image
|
|
uint32_t pe_base::get_size_of_image() const
|
|
{
|
|
return props_->get_size_of_image();
|
|
}
|
|
|
|
//Returns image entry point
|
|
uint32_t pe_base::get_ep() const
|
|
{
|
|
return props_->get_ep();
|
|
}
|
|
|
|
//Sets image entry point (just a value of PE header)
|
|
void pe_base::set_ep(uint32_t new_ep)
|
|
{
|
|
props_->set_ep(new_ep);
|
|
}
|
|
|
|
//Returns number of RVA and sizes (number of DATA_DIRECTORY entries)
|
|
uint32_t pe_base::get_number_of_rvas_and_sizes() const
|
|
{
|
|
return props_->get_number_of_rvas_and_sizes();
|
|
}
|
|
|
|
//Sets number of RVA and sizes (number of DATA_DIRECTORY entries)
|
|
void pe_base::set_number_of_rvas_and_sizes(uint32_t number)
|
|
{
|
|
props_->set_number_of_rvas_and_sizes(number);
|
|
}
|
|
|
|
//Returns PE characteristics
|
|
uint16_t pe_base::get_characteristics() const
|
|
{
|
|
return props_->get_characteristics();
|
|
}
|
|
|
|
//Sets PE characteristics (a value inside header)
|
|
void pe_base::set_characteristics(uint16_t ch)
|
|
{
|
|
props_->set_characteristics(ch);
|
|
}
|
|
|
|
//Returns section from RVA
|
|
section& pe_base::section_from_rva(uint32_t rva)
|
|
{
|
|
//Search for section
|
|
for(section_list::iterator i = sections_.begin(); i != sections_.end(); ++i)
|
|
{
|
|
section& s = *i;
|
|
//Return section if found
|
|
if(rva >= s.get_virtual_address() && rva < s.get_virtual_address() + s.get_aligned_virtual_size(get_section_alignment()))
|
|
return s;
|
|
}
|
|
|
|
throw pe_exception("No section found by presented address", pe_exception::no_section_found);
|
|
}
|
|
|
|
//Returns section from RVA
|
|
const section& pe_base::section_from_rva(uint32_t rva) const
|
|
{
|
|
//Search for section
|
|
for(section_list::const_iterator i = sections_.begin(); i != sections_.end(); ++i)
|
|
{
|
|
const section& s = *i;
|
|
//Return section if found
|
|
if(rva >= s.get_virtual_address() && rva < s.get_virtual_address() + s.get_aligned_virtual_size(get_section_alignment()))
|
|
return s;
|
|
}
|
|
|
|
throw pe_exception("No section found by presented address", pe_exception::no_section_found);
|
|
}
|
|
|
|
//Returns section from directory ID
|
|
section& pe_base::section_from_directory(uint32_t directory_id)
|
|
{
|
|
return section_from_rva(get_directory_rva(directory_id));
|
|
}
|
|
|
|
//Returns section from directory ID
|
|
const section& pe_base::section_from_directory(uint32_t directory_id) const
|
|
{
|
|
return section_from_rva(get_directory_rva(directory_id));
|
|
}
|
|
|
|
//Sets section virtual size (actual for the last one of this PE or for unbound section)
|
|
void pe_base::set_section_virtual_size(section& s, uint32_t vsize)
|
|
{
|
|
//Check if we're changing virtual size of the last section
|
|
//Of course, we can change virtual size of section that's not bound to this PE file
|
|
if(sections_.empty() || std::find_if(sections_.begin(), sections_.end() - 1, section_ptr_finder(s)) != sections_.end() - 1)
|
|
throw pe_exception("Can't change virtual size of any section, except last one", pe_exception::error_changing_section_virtual_size);
|
|
|
|
//If we're setting virtual size to zero
|
|
if(vsize == 0)
|
|
{
|
|
//Check if section is empty
|
|
if(s.empty())
|
|
throw pe_exception("Cannot set virtual size of empty section to zero", pe_exception::error_changing_section_virtual_size);
|
|
|
|
//Set virtual size equal to aligned size of raw data
|
|
s.set_virtual_size(s.get_size_of_raw_data());
|
|
}
|
|
else
|
|
{
|
|
s.set_virtual_size(vsize);
|
|
}
|
|
|
|
//Update image size if we're changing virtual size for the last section of this PE
|
|
if(!sections_.empty() || &s == &(*(sections_.end() - 1)))
|
|
update_image_size();
|
|
}
|
|
|
|
//Expands section raw or virtual size to hold data from specified RVA with specified size
|
|
//Section must be free (not bound to any image)
|
|
//or the last section of this image
|
|
bool pe_base::expand_section(section& s, uint32_t needed_rva, uint32_t needed_size, section_expand_type expand)
|
|
{
|
|
//Check if we're changing the last section
|
|
//Of course, we can change the section that's not bound to this PE file
|
|
if(sections_.empty() || std::find_if(sections_.begin(), sections_.end() - 1, section_ptr_finder(s)) != sections_.end() - 1)
|
|
throw pe_exception("Can't expand any section, except last one", pe_exception::error_expanding_section);
|
|
|
|
//Check if we should expand our section
|
|
if(expand == expand_section_raw && section_data_length_from_rva(s, needed_rva, section_data_raw) < needed_size)
|
|
{
|
|
//Expand section raw data
|
|
s.get_raw_data().resize(needed_rva - s.get_virtual_address() + needed_size);
|
|
recalculate_section_sizes(s, false);
|
|
return true;
|
|
}
|
|
else if(expand == expand_section_virtual && section_data_length_from_rva(s, needed_rva, section_data_virtual) < needed_size)
|
|
{
|
|
//Expand section virtual data
|
|
set_section_virtual_size(s, needed_rva - s.get_virtual_address() + needed_size);
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
//Updates image virtual size
|
|
void pe_base::update_image_size()
|
|
{
|
|
//Write virtual size of image to headers
|
|
if(!sections_.empty())
|
|
set_size_of_image(sections_.back().get_virtual_address() + sections_.back().get_aligned_virtual_size(get_section_alignment()));
|
|
else
|
|
set_size_of_image(get_size_of_headers());
|
|
}
|
|
|
|
//Returns checksum of PE file from header
|
|
uint32_t pe_base::get_checksum() const
|
|
{
|
|
return props_->get_checksum();
|
|
}
|
|
|
|
//Sets checksum of PE file
|
|
void pe_base::set_checksum(uint32_t checksum)
|
|
{
|
|
props_->set_checksum(checksum);
|
|
}
|
|
|
|
//Returns timestamp of PE file from header
|
|
uint32_t pe_base::get_time_date_stamp() const
|
|
{
|
|
return props_->get_time_date_stamp();
|
|
}
|
|
|
|
//Sets timestamp of PE file
|
|
void pe_base::set_time_date_stamp(uint32_t timestamp)
|
|
{
|
|
props_->set_time_date_stamp(timestamp);
|
|
}
|
|
|
|
//Returns Machine field value of PE file from header
|
|
uint16_t pe_base::get_machine() const
|
|
{
|
|
return props_->get_machine();
|
|
}
|
|
|
|
//Sets Machine field value of PE file
|
|
void pe_base::set_machine(uint16_t machine)
|
|
{
|
|
props_->set_machine(machine);
|
|
}
|
|
|
|
//Prepares section before attaching it
|
|
void pe_base::prepare_section(section& s)
|
|
{
|
|
//Calculate its size of raw data
|
|
s.set_size_of_raw_data(static_cast<uint32_t>(pe_utils::align_up(s.get_raw_data().length(), get_file_alignment())));
|
|
|
|
//Check section virtual and raw size
|
|
if(!s.get_size_of_raw_data() && !s.get_virtual_size())
|
|
throw pe_exception("Virtual and Physical sizes of section can't be 0 at the same time", pe_exception::zero_section_sizes);
|
|
|
|
//If section virtual size is zero
|
|
if(!s.get_virtual_size())
|
|
{
|
|
s.set_virtual_size(s.get_size_of_raw_data());
|
|
}
|
|
else
|
|
{
|
|
//Else calculate its virtual size
|
|
s.set_virtual_size(
|
|
std::max<uint32_t>(pe_utils::align_up(s.get_size_of_raw_data(), get_file_alignment()),
|
|
pe_utils::align_up(s.get_virtual_size(), get_section_alignment())));
|
|
}
|
|
}
|
|
|
|
//Adds section to image
|
|
section& pe_base::add_section(section s)
|
|
{
|
|
if(sections_.size() >= maximum_number_of_sections)
|
|
throw pe_exception("Maximum number of sections has been reached", pe_exception::no_more_sections_can_be_added);
|
|
|
|
//Prepare section before adding it
|
|
prepare_section(s);
|
|
|
|
//Calculate section virtual address
|
|
if(!sections_.empty())
|
|
{
|
|
s.set_virtual_address(pe_utils::align_up(sections_.back().get_virtual_address() + sections_.back().get_aligned_virtual_size(get_section_alignment()), get_section_alignment()));
|
|
|
|
//We should align last section raw size, if it wasn't aligned
|
|
section& last = sections_.back();
|
|
last.set_size_of_raw_data(static_cast<uint32_t>(pe_utils::align_up(last.get_raw_data().length(), get_file_alignment())));
|
|
}
|
|
else
|
|
{
|
|
s.set_virtual_address(
|
|
s.get_virtual_address() == 0
|
|
? pe_utils::align_up(get_size_of_headers(), get_section_alignment())
|
|
: pe_utils::align_up(s.get_virtual_address(), get_section_alignment()));
|
|
}
|
|
|
|
//Add section to the end of section list
|
|
sections_.push_back(s);
|
|
//Set number of sections in PE header
|
|
set_number_of_sections(static_cast<uint16_t>(sections_.size()));
|
|
//Recalculate virtual size of image
|
|
set_size_of_image(get_size_of_image() + s.get_aligned_virtual_size(get_section_alignment()));
|
|
//Return last section
|
|
return sections_.back();
|
|
}
|
|
|
|
//Returns true if sectios "s" is already attached to this PE file
|
|
bool pe_base::section_attached(const section& s) const
|
|
{
|
|
return sections_.end() != std::find_if(sections_.begin(), sections_.end(), section_ptr_finder(s));
|
|
}
|
|
|
|
//Returns true if directory exists
|
|
bool pe_base::directory_exists(uint32_t id) const
|
|
{
|
|
return props_->directory_exists(id);
|
|
}
|
|
|
|
//Removes directory
|
|
void pe_base::remove_directory(uint32_t id)
|
|
{
|
|
props_->remove_directory(id);
|
|
}
|
|
|
|
//Returns directory RVA
|
|
uint32_t pe_base::get_directory_rva(uint32_t id) const
|
|
{
|
|
return props_->get_directory_rva(id);
|
|
}
|
|
|
|
//Returns directory size
|
|
uint32_t pe_base::get_directory_size(uint32_t id) const
|
|
{
|
|
return props_->get_directory_size(id);
|
|
}
|
|
|
|
//Sets directory RVA (just a value of PE header, no moving occurs)
|
|
void pe_base::set_directory_rva(uint32_t id, uint32_t rva)
|
|
{
|
|
return props_->set_directory_rva(id, rva);
|
|
}
|
|
|
|
//Sets directory size (just a value of PE header, no moving occurs)
|
|
void pe_base::set_directory_size(uint32_t id, uint32_t size)
|
|
{
|
|
return props_->set_directory_size(id, size);
|
|
}
|
|
|
|
//Strips only zero DATA_DIRECTORY entries to count = min_count
|
|
//Returns resulting number of data directories
|
|
//strip_iat_directory - if true, even not empty IAT directory will be stripped
|
|
uint32_t pe_base::strip_data_directories(uint32_t min_count, bool strip_iat_directory)
|
|
{
|
|
return props_->strip_data_directories(min_count, strip_iat_directory);
|
|
}
|
|
|
|
//Returns true if image has import directory
|
|
bool pe_base::has_imports() const
|
|
{
|
|
return directory_exists(image_directory_entry_import);
|
|
}
|
|
|
|
//Returns true if image has export directory
|
|
bool pe_base::has_exports() const
|
|
{
|
|
return directory_exists(image_directory_entry_export);
|
|
}
|
|
|
|
//Returns true if image has resource directory
|
|
bool pe_base::has_resources() const
|
|
{
|
|
return directory_exists(image_directory_entry_resource);
|
|
}
|
|
|
|
//Returns true if image has security directory
|
|
bool pe_base::has_security() const
|
|
{
|
|
return directory_exists(image_directory_entry_security);
|
|
}
|
|
|
|
//Returns true if image has relocations
|
|
bool pe_base::has_reloc() const
|
|
{
|
|
return directory_exists(image_directory_entry_basereloc) && !(get_characteristics() & image_file_relocs_stripped);
|
|
}
|
|
|
|
//Returns true if image has TLS directory
|
|
bool pe_base::has_tls() const
|
|
{
|
|
return directory_exists(image_directory_entry_tls);
|
|
}
|
|
|
|
//Returns true if image has config directory
|
|
bool pe_base::has_config() const
|
|
{
|
|
return directory_exists(image_directory_entry_load_config);
|
|
}
|
|
|
|
//Returns true if image has bound import directory
|
|
bool pe_base::has_bound_import() const
|
|
{
|
|
return directory_exists(image_directory_entry_bound_import);
|
|
}
|
|
|
|
//Returns true if image has delay import directory
|
|
bool pe_base::has_delay_import() const
|
|
{
|
|
return directory_exists(image_directory_entry_delay_import);
|
|
}
|
|
|
|
//Returns true if image has COM directory
|
|
bool pe_base::is_dotnet() const
|
|
{
|
|
return directory_exists(image_directory_entry_com_descriptor);
|
|
}
|
|
|
|
//Returns true if image has exception directory
|
|
bool pe_base::has_exception_directory() const
|
|
{
|
|
return directory_exists(image_directory_entry_exception);
|
|
}
|
|
|
|
//Returns true if image has debug directory
|
|
bool pe_base::has_debug() const
|
|
{
|
|
return directory_exists(image_directory_entry_debug);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from RVA inside section "s" (checks bounds)
|
|
char* pe_base::section_data_from_rva(section& s, uint32_t rva)
|
|
{
|
|
//Check if RVA is inside section "s"
|
|
if(rva >= s.get_virtual_address() && rva < s.get_virtual_address() + s.get_aligned_virtual_size(get_section_alignment()))
|
|
{
|
|
if(s.get_raw_data().empty())
|
|
throw pe_exception("Section raw data is empty and cannot be changed", pe_exception::section_is_empty);
|
|
|
|
return &s.get_raw_data()[rva - s.get_virtual_address()];
|
|
}
|
|
|
|
throw pe_exception("RVA not found inside section", pe_exception::rva_not_exists);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from RVA inside section "s" (checks bounds)
|
|
const char* pe_base::section_data_from_rva(const section& s, uint32_t rva, section_data_type datatype) const
|
|
{
|
|
//Check if RVA is inside section "s"
|
|
if(rva >= s.get_virtual_address() && rva < s.get_virtual_address() + s.get_aligned_virtual_size(get_section_alignment()))
|
|
return (datatype == section_data_raw ? s.get_raw_data().data() : s.get_virtual_data(get_section_alignment()).c_str()) + rva - s.get_virtual_address();
|
|
|
|
throw pe_exception("RVA not found inside section", pe_exception::rva_not_exists);
|
|
}
|
|
|
|
//Returns section TOTAL RAW/VIRTUAL data length from RVA inside section
|
|
uint32_t pe_base::section_data_length_from_rva(uint32_t rva, section_data_type datatype, bool include_headers) const
|
|
{
|
|
//if RVA is inside of headers and we're searching them too...
|
|
if(include_headers && rva < full_headers_data_.length())
|
|
return static_cast<unsigned long>(full_headers_data_.length());
|
|
|
|
const section& s = section_from_rva(rva);
|
|
return static_cast<unsigned long>(datatype == section_data_raw ? s.get_raw_data().length() /* instead of SizeOfRawData */ : s.get_aligned_virtual_size(get_section_alignment()));
|
|
}
|
|
|
|
//Returns section TOTAL RAW/VIRTUAL data length from VA inside section for PE32
|
|
uint32_t pe_base::section_data_length_from_va(uint32_t va, section_data_type datatype, bool include_headers) const
|
|
{
|
|
return section_data_length_from_rva(va_to_rva(va), datatype, include_headers);
|
|
}
|
|
|
|
//Returns section TOTAL RAW/VIRTUAL data length from VA inside section for PE32/PE64
|
|
uint32_t pe_base::section_data_length_from_va(uint64_t va, section_data_type datatype, bool include_headers) const
|
|
{
|
|
return section_data_length_from_rva(va_to_rva(va), datatype, include_headers);
|
|
}
|
|
|
|
//Returns section remaining RAW/VIRTUAL data length from RVA "rva_inside" to the end of section containing RVA "rva"
|
|
uint32_t pe_base::section_data_length_from_rva(uint32_t rva, uint32_t rva_inside, section_data_type datatype, bool include_headers) const
|
|
{
|
|
//if RVAs are inside of headers and we're searching them too...
|
|
if(include_headers && rva < full_headers_data_.length() && rva_inside < full_headers_data_.length())
|
|
return static_cast<unsigned long>(full_headers_data_.length() - rva_inside);
|
|
|
|
const section& s = section_from_rva(rva);
|
|
if(rva_inside < s.get_virtual_address())
|
|
throw pe_exception("RVA not found inside section", pe_exception::rva_not_exists);
|
|
|
|
//Calculate remaining length of section data from "rva" address
|
|
long length = static_cast<long>(datatype == section_data_raw ? s.get_raw_data().length() /* instead of SizeOfRawData */ : s.get_aligned_virtual_size(get_section_alignment()))
|
|
+ s.get_virtual_address() - rva_inside;
|
|
|
|
if(length < 0)
|
|
return 0;
|
|
|
|
return static_cast<unsigned long>(length);
|
|
}
|
|
|
|
//Returns section remaining RAW/VIRTUAL data length from VA "va_inside" to the end of section containing VA "va" for PE32
|
|
uint32_t pe_base::section_data_length_from_va(uint32_t va, uint32_t va_inside, section_data_type datatype, bool include_headers) const
|
|
{
|
|
return section_data_length_from_rva(va_to_rva(va), va_to_rva(va_inside), datatype, include_headers);
|
|
}
|
|
|
|
//Returns section remaining RAW/VIRTUAL data length from VA "va_inside" to the end of section containing VA "va" for PE32/PE64
|
|
uint32_t pe_base::section_data_length_from_va(uint64_t va, uint64_t va_inside, section_data_type datatype, bool include_headers) const
|
|
{
|
|
return section_data_length_from_rva(va_to_rva(va), va_to_rva(va_inside), datatype, include_headers);
|
|
}
|
|
|
|
//Returns section remaining RAW/VIRTUAL data length from RVA to the end of section "s" (checks bounds)
|
|
uint32_t pe_base::section_data_length_from_rva(const section& s, uint32_t rva_inside, section_data_type datatype) const
|
|
{
|
|
//Check rva_inside
|
|
if(rva_inside >= s.get_virtual_address() && rva_inside < s.get_virtual_address() + s.get_aligned_virtual_size(get_section_alignment()))
|
|
{
|
|
//Calculate remaining length of section data from "rva" address
|
|
int32_t length = static_cast<int32_t>(datatype == section_data_raw ? s.get_raw_data().length() /* instead of SizeOfRawData */ : s.get_aligned_virtual_size(get_section_alignment()))
|
|
+ s.get_virtual_address() - rva_inside;
|
|
|
|
if(length < 0)
|
|
return 0;
|
|
|
|
return static_cast<uint32_t>(length);
|
|
}
|
|
|
|
throw pe_exception("RVA not found inside section", pe_exception::rva_not_exists);
|
|
}
|
|
|
|
//Returns section remaining RAW/VIRTUAL data length from VA to the end of section "s" for PE32 (checks bounds)
|
|
uint32_t pe_base::section_data_length_from_va(const section& s, uint32_t va_inside, section_data_type datatype) const
|
|
{
|
|
return section_data_length_from_rva(s, va_to_rva(va_inside), datatype);
|
|
}
|
|
|
|
//Returns section remaining RAW/VIRTUAL data length from VA to the end of section "s" for PE32/PE64 (checks bounds)
|
|
uint32_t pe_base::section_data_length_from_va(const section& s, uint64_t va_inside, section_data_type datatype) const
|
|
{
|
|
return section_data_length_from_rva(s, va_to_rva(va_inside), datatype);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from RVA inside section
|
|
char* pe_base::section_data_from_rva(uint32_t rva, bool include_headers)
|
|
{
|
|
//if RVA is inside of headers and we're searching them too...
|
|
if(include_headers && rva < full_headers_data_.length())
|
|
return &full_headers_data_[rva];
|
|
|
|
section& s = section_from_rva(rva);
|
|
|
|
if(s.get_raw_data().empty())
|
|
throw pe_exception("Section raw data is empty and cannot be changed", pe_exception::section_is_empty);
|
|
|
|
return &s.get_raw_data()[rva - s.get_virtual_address()];
|
|
}
|
|
|
|
//Returns corresponding section data pointer from RVA inside section
|
|
const char* pe_base::section_data_from_rva(uint32_t rva, section_data_type datatype, bool include_headers) const
|
|
{
|
|
//if RVA is inside of headers and we're searching them too...
|
|
if(include_headers && rva < full_headers_data_.length())
|
|
return &full_headers_data_[rva];
|
|
|
|
const section& s = section_from_rva(rva);
|
|
return (datatype == section_data_raw ? s.get_raw_data().data() : s.get_virtual_data(get_section_alignment()).c_str()) + rva - s.get_virtual_address();
|
|
}
|
|
|
|
//Reads DOS headers from istream
|
|
void pe_base::read_dos_header(std::istream& file, image_dos_header& header)
|
|
{
|
|
//Check istream flags
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("PE file stream is bad or closed.", pe_exception::bad_pe_file);
|
|
|
|
//Read DOS header and check istream
|
|
file.read(reinterpret_cast<char*>(&header), sizeof(image_dos_header));
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Unable to read IMAGE_DOS_HEADER", pe_exception::bad_dos_header);
|
|
|
|
//Check DOS header magic
|
|
if(header.e_magic != 0x5a4d) //"MZ"
|
|
throw pe_exception("IMAGE_DOS_HEADER signature is incorrect", pe_exception::bad_dos_header);
|
|
}
|
|
|
|
//Reads DOS headers from istream
|
|
void pe_base::read_dos_header(std::istream& file)
|
|
{
|
|
read_dos_header(file, dos_header_);
|
|
}
|
|
|
|
//Reads PE image from istream
|
|
void pe_base::read_pe(std::istream& file, bool read_debug_raw_data)
|
|
{
|
|
//Get istream size
|
|
std::streamoff filesize = pe_utils::get_file_size(file);
|
|
|
|
//Check if PE header is DWORD-aligned
|
|
if((dos_header_.e_lfanew % sizeof(uint32_t)) != 0)
|
|
throw pe_exception("PE header is not DWORD-aligned", pe_exception::bad_dos_header);
|
|
|
|
//Seek to NT headers
|
|
file.seekg(dos_header_.e_lfanew);
|
|
if(file.bad() || file.fail())
|
|
throw pe_exception("Cannot reach IMAGE_NT_HEADERS", pe_exception::image_nt_headers_not_found);
|
|
|
|
//Read NT headers
|
|
file.read(get_nt_headers_ptr(), get_sizeof_nt_header() - sizeof(image_data_directory) * image_numberof_directory_entries);
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading IMAGE_NT_HEADERS", pe_exception::error_reading_image_nt_headers);
|
|
|
|
//Check PE signature
|
|
if(get_pe_signature() != 0x4550) //"PE"
|
|
throw pe_exception("Incorrect PE signature", pe_exception::pe_signature_incorrect);
|
|
|
|
//Check number of directories
|
|
if(get_number_of_rvas_and_sizes() > image_numberof_directory_entries)
|
|
set_number_of_rvas_and_sizes(image_numberof_directory_entries);
|
|
|
|
if(get_number_of_rvas_and_sizes() > 0)
|
|
{
|
|
//Read data directory headers, if any
|
|
file.read(get_nt_headers_ptr() + (get_sizeof_nt_header() - sizeof(image_data_directory) * image_numberof_directory_entries), sizeof(image_data_directory) * get_number_of_rvas_and_sizes());
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading DATA_DIRECTORY headers", pe_exception::error_reading_data_directories);
|
|
}
|
|
|
|
//Check section number
|
|
//Images with zero section number accepted
|
|
if(get_number_of_sections() > maximum_number_of_sections)
|
|
throw pe_exception("Incorrect number of sections", pe_exception::section_number_incorrect);
|
|
|
|
//Check PE magic
|
|
if(get_magic() != get_needed_magic())
|
|
throw pe_exception("Incorrect PE signature", pe_exception::pe_signature_incorrect);
|
|
|
|
//Check section alignment
|
|
if(!pe_utils::is_power_of_2(get_section_alignment()))
|
|
throw pe_exception("Incorrect section alignment", pe_exception::incorrect_section_alignment);
|
|
|
|
//Check file alignment
|
|
if(!pe_utils::is_power_of_2(get_file_alignment()))
|
|
throw pe_exception("Incorrect file alignment", pe_exception::incorrect_file_alignment);
|
|
|
|
if(get_file_alignment() != get_section_alignment() && (get_file_alignment() < minimum_file_alignment || get_file_alignment() > get_section_alignment()))
|
|
throw pe_exception("Incorrect file alignment", pe_exception::incorrect_file_alignment);
|
|
|
|
//Check size of image
|
|
if(pe_utils::align_up(get_size_of_image(), get_section_alignment()) == 0)
|
|
throw pe_exception("Incorrect size of image", pe_exception::incorrect_size_of_image);
|
|
|
|
//Read rich data overlay / DOS stub (if any)
|
|
if(static_cast<uint32_t>(dos_header_.e_lfanew) > sizeof(image_dos_header))
|
|
{
|
|
rich_overlay_.resize(dos_header_.e_lfanew - sizeof(image_dos_header));
|
|
file.seekg(sizeof(image_dos_header));
|
|
file.read(&rich_overlay_[0], dos_header_.e_lfanew - sizeof(image_dos_header));
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading 'Rich' & 'DOS stub' overlay", pe_exception::error_reading_overlay);
|
|
}
|
|
|
|
//Calculate first section raw position
|
|
//Sum is safe here
|
|
uint32_t first_section = dos_header_.e_lfanew + get_size_of_optional_header() + sizeof(image_file_header) + sizeof(uint32_t) /* Signature */;
|
|
|
|
if(get_number_of_sections() > 0)
|
|
{
|
|
//Go to first section
|
|
file.seekg(first_section);
|
|
if(file.bad() || file.fail())
|
|
throw pe_exception("Cannot reach section headers", pe_exception::image_section_headers_not_found);
|
|
}
|
|
|
|
uint32_t last_raw_size = 0;
|
|
|
|
//Read all sections
|
|
for(int i = 0; i < get_number_of_sections(); i++)
|
|
{
|
|
section s;
|
|
//Read section header
|
|
file.read(reinterpret_cast<char*>(&s.get_raw_header()), sizeof(image_section_header));
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading section header", pe_exception::error_reading_section_header);
|
|
|
|
//Save next section header position
|
|
std::streamoff next_sect = file.tellg();
|
|
|
|
//Check section virtual and raw sizes
|
|
if(!s.get_size_of_raw_data() && !s.get_virtual_size())
|
|
throw pe_exception("Virtual and Physical sizes of section can't be 0 at the same time", pe_exception::zero_section_sizes);
|
|
|
|
//Check for adequate values of section fields
|
|
if(!pe_utils::is_sum_safe(s.get_virtual_address(), s.get_virtual_size()) || s.get_virtual_size() > pe_utils::two_gb
|
|
|| !pe_utils::is_sum_safe(s.get_pointer_to_raw_data(), s.get_size_of_raw_data()) || s.get_size_of_raw_data() > pe_utils::two_gb)
|
|
throw pe_exception("Incorrect section address or size", pe_exception::section_incorrect_addr_or_size);
|
|
|
|
if(s.get_size_of_raw_data() != 0)
|
|
{
|
|
//If section has raw data
|
|
|
|
//If section raw data size is greater than virtual, fix it
|
|
last_raw_size = s.get_size_of_raw_data();
|
|
if(pe_utils::align_up(s.get_size_of_raw_data(), get_file_alignment()) > pe_utils::align_up(s.get_virtual_size(), get_section_alignment()))
|
|
s.set_size_of_raw_data(s.get_virtual_size());
|
|
|
|
//Check virtual and raw section sizes and addresses
|
|
if(s.get_virtual_address() + pe_utils::align_up(s.get_virtual_size(), get_section_alignment()) > pe_utils::align_up(get_size_of_image(), get_section_alignment())
|
|
||
|
|
pe_utils::align_down(s.get_pointer_to_raw_data(), get_file_alignment()) + s.get_size_of_raw_data() > static_cast<uint32_t>(filesize))
|
|
throw pe_exception("Incorrect section address or size", pe_exception::section_incorrect_addr_or_size);
|
|
|
|
//Seek to section raw data
|
|
file.seekg(pe_utils::align_down(s.get_pointer_to_raw_data(), get_file_alignment()));
|
|
if(file.bad() || file.fail())
|
|
throw pe_exception("Cannot reach section data", pe_exception::image_section_data_not_found);
|
|
|
|
//Read section raw data
|
|
s.get_raw_data().resize(s.get_size_of_raw_data());
|
|
file.read(&s.get_raw_data()[0], s.get_size_of_raw_data());
|
|
if(file.bad() || file.fail())
|
|
throw pe_exception("Error reading section data", pe_exception::image_section_data_not_found);
|
|
}
|
|
|
|
//Check virtual address and size of section
|
|
if(s.get_virtual_address() + s.get_aligned_virtual_size(get_section_alignment()) > pe_utils::align_up(get_size_of_image(), get_section_alignment()))
|
|
throw pe_exception("Incorrect section address or size", pe_exception::section_incorrect_addr_or_size);
|
|
|
|
//Save section
|
|
sections_.push_back(s);
|
|
|
|
//Seek to the next section header
|
|
file.seekg(next_sect);
|
|
}
|
|
|
|
//Check size of headers: SizeOfHeaders can't be larger than first section VA
|
|
if(!sections_.empty() && get_size_of_headers() > sections_.front().get_virtual_address())
|
|
throw pe_exception("Incorrect size of headers", pe_exception::incorrect_size_of_headers);
|
|
|
|
//If image has more than two sections
|
|
if(sections_.size() >= 2)
|
|
{
|
|
//Check sections virtual sizes
|
|
for(section_list::const_iterator i = sections_.begin() + 1; i != sections_.end(); ++i)
|
|
{
|
|
if((*i).get_virtual_address() != (*(i - 1)).get_virtual_address() + (*(i - 1)).get_aligned_virtual_size(get_section_alignment()))
|
|
throw pe_exception("Section table is incorrect", pe_exception::image_section_table_incorrect);
|
|
}
|
|
}
|
|
|
|
//Check if image has overlay in the end of file
|
|
has_overlay_ = !sections_.empty() && filesize > static_cast<std::streamoff>(sections_.back().get_pointer_to_raw_data() + last_raw_size);
|
|
|
|
{
|
|
//Additionally, read data from the beginning of istream to size of headers
|
|
file.seekg(0);
|
|
uint32_t size_of_headers = std::min<uint32_t>(get_size_of_headers(), static_cast<uint32_t>(filesize));
|
|
|
|
if(!sections_.empty())
|
|
{
|
|
for(section_list::const_iterator i = sections_.begin(); i != sections_.end(); ++i)
|
|
{
|
|
if(!(*i).empty())
|
|
{
|
|
size_of_headers = std::min<uint32_t>(get_size_of_headers(), (*i).get_pointer_to_raw_data());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
full_headers_data_.resize(size_of_headers);
|
|
file.read(&full_headers_data_[0], size_of_headers);
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading file", pe_exception::error_reading_file);
|
|
}
|
|
|
|
//Moreover, if there's debug directory, read its raw data for some debug info types
|
|
while(read_debug_raw_data && has_debug())
|
|
{
|
|
try
|
|
{
|
|
//Check the length in bytes of the section containing debug directory
|
|
if(section_data_length_from_rva(get_directory_rva(image_directory_entry_debug), get_directory_rva(image_directory_entry_debug), section_data_virtual, true) < sizeof(image_debug_directory))
|
|
break;
|
|
|
|
unsigned long current_pos = get_directory_rva(image_directory_entry_debug);
|
|
|
|
//First IMAGE_DEBUG_DIRECTORY table
|
|
image_debug_directory directory = section_data_from_rva<image_debug_directory>(current_pos, section_data_virtual, true);
|
|
|
|
//Iterate over all IMAGE_DEBUG_DIRECTORY directories
|
|
while(directory.PointerToRawData
|
|
&& current_pos < get_directory_rva(image_directory_entry_debug) + get_directory_size(image_directory_entry_debug))
|
|
{
|
|
//If we have something to read
|
|
if((directory.Type == image_debug_type_codeview
|
|
|| directory.Type == image_debug_type_misc
|
|
|| directory.Type == image_debug_type_coff)
|
|
&& directory.SizeOfData)
|
|
{
|
|
std::string data;
|
|
data.resize(directory.SizeOfData);
|
|
file.seekg(directory.PointerToRawData);
|
|
file.read(&data[0], directory.SizeOfData);
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading file", pe_exception::error_reading_file);
|
|
|
|
debug_data_.insert(std::make_pair(directory.PointerToRawData, data));
|
|
}
|
|
|
|
//Go to next debug entry
|
|
current_pos += sizeof(image_debug_directory);
|
|
directory = section_data_from_rva<image_debug_directory>(current_pos, section_data_virtual, true);
|
|
}
|
|
|
|
break;
|
|
}
|
|
catch(const pe_exception&)
|
|
{
|
|
//Don't throw any exception here, if debug info is corrupted or incorrect
|
|
break;
|
|
}
|
|
catch(const std::bad_alloc&)
|
|
{
|
|
//Don't throw any exception here, if debug info is corrupted or incorrect
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
//Returns PE type of this image
|
|
pe_type pe_base::get_pe_type() const
|
|
{
|
|
return props_->get_pe_type();
|
|
}
|
|
|
|
//Returns PE type (PE or PE+) from pe_type enumeration (minimal correctness checks)
|
|
pe_type pe_base::get_pe_type(std::istream& file)
|
|
{
|
|
//Save state of the istream
|
|
std::ios_base::iostate state = file.exceptions();
|
|
std::streamoff old_offset = file.tellg();
|
|
image_nt_headers32 nt_headers;
|
|
image_dos_header header;
|
|
|
|
try
|
|
{
|
|
//Read dos header
|
|
file.exceptions(std::ios::goodbit);
|
|
read_dos_header(file, header);
|
|
|
|
//Seek to the NT headers start
|
|
file.seekg(header.e_lfanew);
|
|
if(file.bad() || file.fail())
|
|
throw pe_exception("Cannot reach IMAGE_NT_HEADERS", pe_exception::image_nt_headers_not_found);
|
|
|
|
//Read NT headers (we're using 32-bit version, because there's no significant differencies between 32 and 64 bit version structures)
|
|
file.read(reinterpret_cast<char*>(&nt_headers), sizeof(image_nt_headers32) - sizeof(image_data_directory) * image_numberof_directory_entries);
|
|
if(file.bad() || file.eof())
|
|
throw pe_exception("Error reading IMAGE_NT_HEADERS", pe_exception::error_reading_image_nt_headers);
|
|
|
|
//Check NT headers signature
|
|
if(nt_headers.Signature != 0x4550) //"PE"
|
|
throw pe_exception("Incorrect PE signature", pe_exception::pe_signature_incorrect);
|
|
|
|
//Check NT headers magic
|
|
if(nt_headers.OptionalHeader.Magic != image_nt_optional_hdr32_magic && nt_headers.OptionalHeader.Magic != image_nt_optional_hdr64_magic)
|
|
throw pe_exception("Incorrect PE signature", pe_exception::pe_signature_incorrect);
|
|
}
|
|
catch(const std::exception&)
|
|
{
|
|
//If something went wrong, restore istream state
|
|
file.exceptions(state);
|
|
file.seekg(old_offset);
|
|
file.clear();
|
|
//Retrhow exception
|
|
throw;
|
|
}
|
|
|
|
//Restore stream state
|
|
file.exceptions(state);
|
|
file.seekg(old_offset);
|
|
file.clear();
|
|
|
|
//Determine PE type and return it
|
|
return nt_headers.OptionalHeader.Magic == image_nt_optional_hdr64_magic ? pe_type_64 : pe_type_32;
|
|
}
|
|
|
|
//Returns true if image has overlay data at the end of file
|
|
bool pe_base::has_overlay() const
|
|
{
|
|
return has_overlay_;
|
|
}
|
|
|
|
//Clears PE characteristics flag
|
|
void pe_base::clear_characteristics_flags(uint16_t flags)
|
|
{
|
|
set_characteristics(get_characteristics() & ~flags);
|
|
}
|
|
|
|
//Sets PE characteristics flag
|
|
void pe_base::set_characteristics_flags(uint16_t flags)
|
|
{
|
|
set_characteristics(get_characteristics() | flags);
|
|
}
|
|
|
|
//Returns true if PE characteristics flag set
|
|
bool pe_base::check_characteristics_flag(uint16_t flag) const
|
|
{
|
|
return (get_characteristics() & flag) ? true : false;
|
|
}
|
|
|
|
//Returns subsystem value
|
|
uint16_t pe_base::get_subsystem() const
|
|
{
|
|
return props_->get_subsystem();
|
|
}
|
|
|
|
//Sets subsystem value
|
|
void pe_base::set_subsystem(uint16_t subsystem)
|
|
{
|
|
props_->set_subsystem(subsystem);
|
|
}
|
|
|
|
//Returns true if image has console subsystem
|
|
bool pe_base::is_console() const
|
|
{
|
|
return get_subsystem() == image_subsystem_windows_cui;
|
|
}
|
|
|
|
//Returns true if image has Windows GUI subsystem
|
|
bool pe_base::is_gui() const
|
|
{
|
|
return get_subsystem() == image_subsystem_windows_gui;
|
|
}
|
|
|
|
//Sets required operation system version
|
|
void pe_base::set_os_version(uint16_t major, uint16_t minor)
|
|
{
|
|
props_->set_os_version(major, minor);
|
|
}
|
|
|
|
//Returns required operation system version (minor word)
|
|
uint16_t pe_base::get_minor_os_version() const
|
|
{
|
|
return props_->get_minor_os_version();
|
|
}
|
|
|
|
//Returns required operation system version (major word)
|
|
uint16_t pe_base::get_major_os_version() const
|
|
{
|
|
return props_->get_major_os_version();
|
|
}
|
|
|
|
//Sets required subsystem version
|
|
void pe_base::set_subsystem_version(uint16_t major, uint16_t minor)
|
|
{
|
|
props_->set_subsystem_version(major, minor);
|
|
}
|
|
|
|
//Returns required subsystem version (minor word)
|
|
uint16_t pe_base::get_minor_subsystem_version() const
|
|
{
|
|
return props_->get_minor_subsystem_version();
|
|
}
|
|
|
|
//Returns required subsystem version (major word)
|
|
uint16_t pe_base::get_major_subsystem_version() const
|
|
{
|
|
return props_->get_major_subsystem_version();
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section "s" for PE32 (checks bounds)
|
|
char* pe_base::section_data_from_va(section& s, uint32_t va) //Always returns raw data
|
|
{
|
|
return section_data_from_rva(s, va_to_rva(va));
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section "s" for PE32 (checks bounds)
|
|
const char* pe_base::section_data_from_va(const section& s, uint32_t va, section_data_type datatype) const
|
|
{
|
|
return section_data_from_rva(s, va_to_rva(va), datatype);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section for PE32
|
|
char* pe_base::section_data_from_va(uint32_t va, bool include_headers) //Always returns raw data
|
|
{
|
|
return section_data_from_rva(va_to_rva(va), include_headers);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section for PE32
|
|
const char* pe_base::section_data_from_va(uint32_t va, section_data_type datatype, bool include_headers) const
|
|
{
|
|
return section_data_from_rva(va_to_rva(va), datatype, include_headers);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section "s" for PE32/PE64 (checks bounds)
|
|
char* pe_base::section_data_from_va(section& s, uint64_t va) //Always returns raw data
|
|
{
|
|
return section_data_from_rva(s, va_to_rva(va));
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section "s" for PE32/PE64 (checks bounds)
|
|
const char* pe_base::section_data_from_va(const section& s, uint64_t va, section_data_type datatype) const
|
|
{
|
|
return section_data_from_rva(s, va_to_rva(va), datatype);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section for PE32/PE64
|
|
char* pe_base::section_data_from_va(uint64_t va, bool include_headers) //Always returns raw data
|
|
{
|
|
return section_data_from_rva(va_to_rva(va), include_headers);
|
|
}
|
|
|
|
//Returns corresponding section data pointer from VA inside section for PE32/PE64
|
|
const char* pe_base::section_data_from_va(uint64_t va, section_data_type datatype, bool include_headers) const
|
|
{
|
|
return section_data_from_rva(va_to_rva(va), datatype, include_headers);
|
|
}
|
|
|
|
//Returns section from VA inside it for PE32
|
|
section& pe_base::section_from_va(uint32_t va)
|
|
{
|
|
return section_from_rva(va_to_rva(va));
|
|
}
|
|
|
|
//Returns section from VA inside it for PE32/PE64
|
|
section& pe_base::section_from_va(uint64_t va)
|
|
{
|
|
return section_from_rva(va_to_rva(va));
|
|
}
|
|
|
|
//Returns section from RVA inside it for PE32
|
|
const section& pe_base::section_from_va(uint32_t va) const
|
|
{
|
|
return section_from_rva(va_to_rva(va));
|
|
}
|
|
|
|
//Returns section from RVA inside it for PE32/PE64
|
|
const section& pe_base::section_from_va(uint64_t va) const
|
|
{
|
|
return section_from_rva(va_to_rva(va));
|
|
}
|
|
|
|
uint32_t pe_base::va_to_rva(uint32_t va, bool bound_check) const
|
|
{
|
|
return props_->va_to_rva(va, bound_check);
|
|
}
|
|
|
|
uint32_t pe_base::va_to_rva(uint64_t va, bool bound_check) const
|
|
{
|
|
return props_->va_to_rva(va, bound_check);
|
|
}
|
|
|
|
uint32_t pe_base::rva_to_va_32(uint32_t rva) const
|
|
{
|
|
return props_->rva_to_va_32(rva);
|
|
}
|
|
|
|
uint64_t pe_base::rva_to_va_64(uint32_t rva) const
|
|
{
|
|
return props_->rva_to_va_64(rva);
|
|
}
|
|
|
|
//Relative Virtual Address (RVA) to Virtual Address (VA) convertion for PE32
|
|
void pe_base::rva_to_va(uint32_t rva, uint32_t& va) const
|
|
{
|
|
va = rva_to_va_32(rva);
|
|
}
|
|
|
|
//Relative Virtual Address (RVA) to Virtual Address (VA) convertions for PE32/PE64
|
|
void pe_base::rva_to_va(uint32_t rva, uint64_t& va) const
|
|
{
|
|
va = rva_to_va_64(rva);
|
|
}
|
|
|
|
//Returns section from file offset (4gb max)
|
|
section& pe_base::section_from_file_offset(uint32_t offset)
|
|
{
|
|
return *file_offset_to_section(offset);
|
|
}
|
|
|
|
//Returns section from file offset (4gb max)
|
|
const section& pe_base::section_from_file_offset(uint32_t offset) const
|
|
{
|
|
return *file_offset_to_section(offset);
|
|
}
|
|
|
|
//Returns section and offset (raw data only) from its start from RVA
|
|
const std::pair<uint32_t, const section*> pe_base::section_and_offset_from_rva(uint32_t rva) const
|
|
{
|
|
const section& s = section_from_rva(rva);
|
|
return std::make_pair(rva - s.get_virtual_address(), &s);
|
|
}
|
|
|
|
//Returns DLL Characteristics
|
|
uint16_t pe_base::get_dll_characteristics() const
|
|
{
|
|
return props_->get_dll_characteristics();
|
|
}
|
|
|
|
//Sets DLL Characteristics
|
|
void pe_base::set_dll_characteristics(uint16_t characteristics)
|
|
{
|
|
props_->set_dll_characteristics(characteristics);
|
|
}
|
|
|
|
//Returns size of headers
|
|
uint32_t pe_base::get_size_of_headers() const
|
|
{
|
|
return props_->get_size_of_headers();
|
|
}
|
|
|
|
//Returns size of optional header
|
|
uint16_t pe_base::get_size_of_optional_header() const
|
|
{
|
|
return props_->get_size_of_optional_header();
|
|
}
|
|
|
|
//Returns PE signature
|
|
uint32_t pe_base::get_pe_signature() const
|
|
{
|
|
return props_->get_pe_signature();
|
|
}
|
|
|
|
//Returns magic value
|
|
uint32_t pe_base::get_magic() const
|
|
{
|
|
return props_->get_magic();
|
|
}
|
|
|
|
//Returns image base for PE32
|
|
void pe_base::get_image_base(uint32_t& base) const
|
|
{
|
|
base = get_image_base_32();
|
|
}
|
|
|
|
//Returns image base for PE32 and PE64 respectively
|
|
uint32_t pe_base::get_image_base_32() const
|
|
{
|
|
return props_->get_image_base_32();
|
|
}
|
|
|
|
//Sets image base for PE32 and PE64 respectively
|
|
uint64_t pe_base::get_image_base_64() const
|
|
{
|
|
return props_->get_image_base_64();
|
|
}
|
|
|
|
//RVA to RAW file offset convertion (4gb max)
|
|
uint32_t pe_base::rva_to_file_offset(uint32_t rva) const
|
|
{
|
|
//Maybe, RVA is inside PE headers
|
|
if(rva < get_size_of_headers())
|
|
return rva;
|
|
|
|
const section& s = section_from_rva(rva);
|
|
return s.get_pointer_to_raw_data() + rva - s.get_virtual_address();
|
|
}
|
|
|
|
//RAW file offset to RVA convertion (4gb max)
|
|
uint32_t pe_base::file_offset_to_rva(uint32_t offset) const
|
|
{
|
|
//Maybe, offset is inside PE headers
|
|
if(offset < get_size_of_headers())
|
|
return offset;
|
|
|
|
const section_list::const_iterator it = file_offset_to_section(offset);
|
|
return offset - (*it).get_pointer_to_raw_data() + (*it).get_virtual_address();
|
|
}
|
|
|
|
//RAW file offset to section convertion helper (4gb max)
|
|
section_list::const_iterator pe_base::file_offset_to_section(uint32_t offset) const
|
|
{
|
|
section_list::const_iterator it = std::find_if(sections_.begin(), sections_.end(), section_by_raw_offset(offset));
|
|
if(it == sections_.end())
|
|
throw pe_exception("No section found by presented file offset", pe_exception::no_section_found);
|
|
|
|
return it;
|
|
}
|
|
|
|
//RAW file offset to section convertion helper (4gb max)
|
|
section_list::iterator pe_base::file_offset_to_section(uint32_t offset)
|
|
{
|
|
section_list::iterator it = std::find_if(sections_.begin(), sections_.end(), section_by_raw_offset(offset));
|
|
if(it == sections_.end())
|
|
throw pe_exception("No section found by presented file offset", pe_exception::no_section_found);
|
|
|
|
return it;
|
|
}
|
|
|
|
//RVA from section raw data offset
|
|
uint32_t pe_base::rva_from_section_offset(const section& s, uint32_t raw_offset_from_section_start)
|
|
{
|
|
return s.get_virtual_address() + raw_offset_from_section_start;
|
|
}
|
|
|
|
//Returns image base for PE32/PE64
|
|
void pe_base::get_image_base(uint64_t& base) const
|
|
{
|
|
base = get_image_base_64();
|
|
}
|
|
|
|
//Sets new image base
|
|
void pe_base::set_image_base(uint32_t base)
|
|
{
|
|
props_->set_image_base(base);
|
|
}
|
|
|
|
void pe_base::set_image_base_64(uint64_t base)
|
|
{
|
|
props_->set_image_base_64(base);
|
|
}
|
|
|
|
//Sets heap size commit for PE32 and PE64 respectively
|
|
void pe_base::set_heap_size_commit(uint32_t size)
|
|
{
|
|
props_->set_heap_size_commit(size);
|
|
}
|
|
|
|
void pe_base::set_heap_size_commit(uint64_t size)
|
|
{
|
|
props_->set_heap_size_commit(size);
|
|
}
|
|
|
|
//Sets heap size reserve for PE32 and PE64 respectively
|
|
void pe_base::set_heap_size_reserve(uint32_t size)
|
|
{
|
|
props_->set_heap_size_reserve(size);
|
|
}
|
|
|
|
void pe_base::set_heap_size_reserve(uint64_t size)
|
|
{
|
|
props_->set_heap_size_reserve(size);
|
|
}
|
|
|
|
//Sets stack size commit for PE32 and PE64 respectively
|
|
void pe_base::set_stack_size_commit(uint32_t size)
|
|
{
|
|
props_->set_stack_size_commit(size);
|
|
}
|
|
|
|
void pe_base::set_stack_size_commit(uint64_t size)
|
|
{
|
|
props_->set_stack_size_commit(size);
|
|
}
|
|
|
|
//Sets stack size reserve for PE32 and PE64 respectively
|
|
void pe_base::set_stack_size_reserve(uint32_t size)
|
|
{
|
|
props_->set_stack_size_reserve(size);
|
|
}
|
|
|
|
void pe_base::set_stack_size_reserve(uint64_t size)
|
|
{
|
|
props_->set_stack_size_reserve(size);
|
|
}
|
|
|
|
//Returns heap size commit for PE32 and PE64 respectively
|
|
uint32_t pe_base::get_heap_size_commit_32() const
|
|
{
|
|
return props_->get_heap_size_commit_32();
|
|
}
|
|
|
|
uint64_t pe_base::get_heap_size_commit_64() const
|
|
{
|
|
return props_->get_heap_size_commit_64();
|
|
}
|
|
|
|
//Returns heap size reserve for PE32 and PE64 respectively
|
|
uint32_t pe_base::get_heap_size_reserve_32() const
|
|
{
|
|
return props_->get_heap_size_reserve_32();
|
|
}
|
|
|
|
uint64_t pe_base::get_heap_size_reserve_64() const
|
|
{
|
|
return props_->get_heap_size_reserve_64();
|
|
}
|
|
|
|
//Returns stack size commit for PE32 and PE64 respectively
|
|
uint32_t pe_base::get_stack_size_commit_32() const
|
|
{
|
|
return props_->get_stack_size_commit_32();
|
|
}
|
|
|
|
uint64_t pe_base::get_stack_size_commit_64() const
|
|
{
|
|
return props_->get_stack_size_commit_64();
|
|
}
|
|
|
|
//Returns stack size reserve for PE32 and PE64 respectively
|
|
uint32_t pe_base::get_stack_size_reserve_32() const
|
|
{
|
|
return props_->get_stack_size_reserve_32();
|
|
}
|
|
|
|
uint64_t pe_base::get_stack_size_reserve_64() const
|
|
{
|
|
return props_->get_stack_size_reserve_64();
|
|
}
|
|
|
|
//Returns heap size commit for PE32
|
|
void pe_base::get_heap_size_commit(uint32_t& size) const
|
|
{
|
|
size = get_heap_size_commit_32();
|
|
}
|
|
|
|
//Returns heap size commit for PE32/PE64
|
|
void pe_base::get_heap_size_commit(uint64_t& size) const
|
|
{
|
|
size = get_heap_size_commit_64();
|
|
}
|
|
|
|
//Returns heap size reserve for PE32
|
|
void pe_base::get_heap_size_reserve(uint32_t& size) const
|
|
{
|
|
size = get_heap_size_reserve_32();
|
|
}
|
|
|
|
//Returns heap size reserve for PE32/PE64
|
|
void pe_base::get_heap_size_reserve(uint64_t& size) const
|
|
{
|
|
size = get_heap_size_reserve_64();
|
|
}
|
|
|
|
//Returns stack size commit for PE32
|
|
void pe_base::get_stack_size_commit(uint32_t& size) const
|
|
{
|
|
size = get_stack_size_commit_32();
|
|
}
|
|
|
|
//Returns stack size commit for PE32/PE64
|
|
void pe_base::get_stack_size_commit(uint64_t& size) const
|
|
{
|
|
size = get_stack_size_commit_64();
|
|
}
|
|
|
|
//Returns stack size reserve for PE32
|
|
void pe_base::get_stack_size_reserve(uint32_t& size) const
|
|
{
|
|
size = get_stack_size_reserve_32();
|
|
}
|
|
|
|
//Returns stack size reserve for PE32/PE64
|
|
void pe_base::get_stack_size_reserve(uint64_t& size) const
|
|
{
|
|
size = get_stack_size_reserve_64();
|
|
}
|
|
|
|
//Realigns file (changes file alignment)
|
|
void pe_base::realign_file(uint32_t new_file_alignment)
|
|
{
|
|
//Checks alignment for correctness
|
|
set_file_alignment(new_file_alignment);
|
|
realign_all_sections();
|
|
}
|
|
|
|
//Helper function to recalculate RAW and virtual section sizes and strip it, if necessary
|
|
void pe_base::recalculate_section_sizes(section& s, bool auto_strip)
|
|
{
|
|
prepare_section(s); //Recalculate section raw addresses
|
|
|
|
//Strip RAW size of section, if it is the last one
|
|
//For all others it must be file-aligned and calculated by prepare_section() call
|
|
if(auto_strip && !(sections_.empty() || &s == &*(sections_.end() - 1)))
|
|
{
|
|
//Strip ending raw data nullbytes to optimize size
|
|
std::string& raw_data = s.get_raw_data();
|
|
if(!raw_data.empty())
|
|
{
|
|
std::string::size_type i = raw_data.length();
|
|
for(; i != 1; --i)
|
|
{
|
|
if(raw_data[i - 1] != 0)
|
|
break;
|
|
}
|
|
|
|
raw_data.resize(i);
|
|
}
|
|
|
|
s.set_size_of_raw_data(static_cast<uint32_t>(raw_data.length()));
|
|
}
|
|
|
|
//Can occur only for last section
|
|
if(pe_utils::align_up(s.get_virtual_size(), get_section_alignment()) < pe_utils::align_up(s.get_size_of_raw_data(), get_file_alignment()))
|
|
set_section_virtual_size(s, pe_utils::align_up(s.get_size_of_raw_data(), get_section_alignment())); //Recalculate section virtual size
|
|
}
|
|
|
|
//Returns data from the beginning of image
|
|
//Size = SizeOfHeaders
|
|
const std::string& pe_base::get_full_headers_data() const
|
|
{
|
|
return full_headers_data_;
|
|
}
|
|
|
|
const pe_base::debug_data_list& pe_base::get_raw_debug_data_list() const
|
|
{
|
|
return debug_data_;
|
|
}
|
|
|
|
//Sets number of sections
|
|
void pe_base::set_number_of_sections(uint16_t number)
|
|
{
|
|
props_->set_number_of_sections(number);
|
|
}
|
|
|
|
//Sets size of image
|
|
void pe_base::set_size_of_image(uint32_t size)
|
|
{
|
|
props_->set_size_of_image(size);
|
|
}
|
|
|
|
//Sets size of headers
|
|
void pe_base::set_size_of_headers(uint32_t size)
|
|
{
|
|
props_->set_size_of_headers(size);
|
|
}
|
|
|
|
//Sets size of optional headers
|
|
void pe_base::set_size_of_optional_header(uint16_t size)
|
|
{
|
|
props_->set_size_of_optional_header(size);
|
|
}
|
|
|
|
//Returns nt headers data pointer
|
|
char* pe_base::get_nt_headers_ptr()
|
|
{
|
|
return props_->get_nt_headers_ptr();
|
|
}
|
|
|
|
//Returns nt headers data pointer
|
|
const char* pe_base::get_nt_headers_ptr() const
|
|
{
|
|
return props_->get_nt_headers_ptr();
|
|
}
|
|
|
|
//Returns sizeof() nt headers
|
|
uint32_t pe_base::get_sizeof_nt_header() const
|
|
{
|
|
return props_->get_sizeof_nt_header();
|
|
}
|
|
|
|
//Returns sizeof() optional headers
|
|
uint32_t pe_base::get_sizeof_opt_headers() const
|
|
{
|
|
return props_->get_sizeof_opt_headers();
|
|
}
|
|
|
|
//Sets file alignment (no checks)
|
|
void pe_base::set_file_alignment_unchecked(uint32_t alignment)
|
|
{
|
|
props_->set_file_alignment_unchecked(alignment);
|
|
}
|
|
|
|
//Sets base of code
|
|
void pe_base::set_base_of_code(uint32_t base)
|
|
{
|
|
props_->set_base_of_code(base);
|
|
}
|
|
|
|
//Returns base of code
|
|
uint32_t pe_base::get_base_of_code() const
|
|
{
|
|
return props_->get_base_of_code();
|
|
}
|
|
|
|
//Returns needed magic of image
|
|
uint32_t pe_base::get_needed_magic() const
|
|
{
|
|
return props_->get_needed_magic();
|
|
}
|
|
}
|