mirror of
https://github.com/wiiu-env/EnvironmentLoader.git
synced 2024-11-27 16:24:16 +01:00
347 lines
12 KiB
C++
347 lines
12 KiB
C++
/*
|
|
Copyright (C) 2001-2015 by Serge Lamikhov-Center
|
|
|
|
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.
|
|
*/
|
|
|
|
#ifndef ELFIO_SECTION_HPP
|
|
#define ELFIO_SECTION_HPP
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
#include "utils/logger.h"
|
|
#include <zlib.h>
|
|
|
|
namespace ELFIO {
|
|
|
|
class section
|
|
{
|
|
friend class elfio;
|
|
public:
|
|
virtual ~section() {};
|
|
|
|
ELFIO_GET_ACCESS_DECL ( Elf_Half, index );
|
|
ELFIO_GET_SET_ACCESS_DECL( std::string, name );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Word, type );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Xword, flags );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Word, info );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Word, link );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Xword, addr_align );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Xword, entry_size );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf64_Addr, address );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Xword, size );
|
|
ELFIO_GET_SET_ACCESS_DECL( Elf_Word, name_string_offset );
|
|
ELFIO_GET_ACCESS_DECL ( Elf64_Off, offset );
|
|
size_t stream_size;
|
|
size_t get_stream_size() const
|
|
{
|
|
return stream_size;
|
|
}
|
|
|
|
void set_stream_size(size_t value)
|
|
{
|
|
stream_size = value;
|
|
}
|
|
|
|
virtual const char* get_data() const = 0;
|
|
virtual void set_data( const char* pData, Elf_Word size ) = 0;
|
|
virtual void set_data( const std::string& data ) = 0;
|
|
virtual void append_data( const char* pData, Elf_Word size ) = 0;
|
|
virtual void append_data( const std::string& data ) = 0;
|
|
|
|
protected:
|
|
ELFIO_SET_ACCESS_DECL( Elf64_Off, offset );
|
|
ELFIO_SET_ACCESS_DECL( Elf_Half, index );
|
|
|
|
virtual void load( std::istream& stream,
|
|
std::streampos header_offset ) = 0;
|
|
virtual void save( std::ostream& stream,
|
|
std::streampos header_offset,
|
|
std::streampos data_offset ) = 0;
|
|
virtual bool is_address_initialized() const = 0;
|
|
};
|
|
|
|
|
|
template< class T >
|
|
class section_impl : public section
|
|
{
|
|
public:
|
|
//------------------------------------------------------------------------------
|
|
section_impl( const endianess_convertor* convertor_ ) : convertor( convertor_ )
|
|
{
|
|
std::fill_n( reinterpret_cast<char*>( &header ), sizeof( header ), '\0' );
|
|
is_address_set = false;
|
|
data = 0;
|
|
data_size = 0;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
~section_impl()
|
|
{
|
|
delete [] data;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
// Section info functions
|
|
ELFIO_GET_SET_ACCESS( Elf_Word, type, header.sh_type );
|
|
ELFIO_GET_SET_ACCESS( Elf_Xword, flags, header.sh_flags );
|
|
ELFIO_GET_SET_ACCESS( Elf_Xword, size, header.sh_size );
|
|
ELFIO_GET_SET_ACCESS( Elf_Word, link, header.sh_link );
|
|
ELFIO_GET_SET_ACCESS( Elf_Word, info, header.sh_info );
|
|
ELFIO_GET_SET_ACCESS( Elf_Xword, addr_align, header.sh_addralign );
|
|
ELFIO_GET_SET_ACCESS( Elf_Xword, entry_size, header.sh_entsize );
|
|
ELFIO_GET_SET_ACCESS( Elf_Word, name_string_offset, header.sh_name );
|
|
ELFIO_GET_ACCESS ( Elf64_Addr, address, header.sh_addr );
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
Elf_Half
|
|
get_index() const
|
|
{
|
|
return index;
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
std::string
|
|
get_name() const
|
|
{
|
|
return name;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
set_name( std::string name_ )
|
|
{
|
|
name = name_;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
set_address( Elf64_Addr value )
|
|
{
|
|
header.sh_addr = value;
|
|
header.sh_addr = (*convertor)( header.sh_addr );
|
|
is_address_set = true;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
bool
|
|
is_address_initialized() const
|
|
{
|
|
return is_address_set;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
const char*
|
|
get_data() const
|
|
{
|
|
return data;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
set_data( const char* raw_data, Elf_Word size )
|
|
{
|
|
if ( get_type() != SHT_NOBITS ) {
|
|
delete [] data;
|
|
data = new char[size];
|
|
if ( 0 != data && 0 != raw_data ) {
|
|
data_size = size;
|
|
std::copy( raw_data, raw_data + size, data );
|
|
}
|
|
}
|
|
|
|
set_size( size );
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
set_data( const std::string& str_data )
|
|
{
|
|
return set_data( str_data.c_str(), (Elf_Word)str_data.size() );
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
append_data( const char* raw_data, Elf_Word size )
|
|
{
|
|
if ( get_type() != SHT_NOBITS ) {
|
|
if ( get_size() + size < data_size ) {
|
|
std::copy( raw_data, raw_data + size, data + get_size() );
|
|
}
|
|
else {
|
|
data_size = 2*( data_size + size);
|
|
char* new_data;
|
|
new_data = new char[data_size];
|
|
|
|
if ( 0 != new_data ) {
|
|
std::copy( data, data + get_size(), new_data );
|
|
std::copy( raw_data, raw_data + size, new_data + get_size() );
|
|
delete [] data;
|
|
data = new_data;
|
|
}
|
|
}
|
|
set_size( get_size() + size );
|
|
}
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
append_data( const std::string& str_data )
|
|
{
|
|
return append_data( str_data.c_str(), (Elf_Word)str_data.size() );
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
protected:
|
|
//------------------------------------------------------------------------------
|
|
ELFIO_GET_SET_ACCESS( Elf64_Off, offset, header.sh_offset );
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
set_index( Elf_Half value )
|
|
{
|
|
index = value;
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
load( std::istream& stream,
|
|
std::streampos header_offset )
|
|
{
|
|
std::fill_n( reinterpret_cast<char*>( &header ), sizeof( header ), '\0' );
|
|
|
|
stream.seekg ( 0, stream.end );
|
|
set_stream_size ( stream.tellg() );
|
|
|
|
stream.seekg( header_offset );
|
|
stream.read( reinterpret_cast<char*>( &header ), sizeof( header ) );
|
|
|
|
|
|
Elf_Xword size = get_size();
|
|
if ( 0 == data && SHT_NULL != get_type() && SHT_NOBITS != get_type() && size < get_stream_size()) {
|
|
|
|
data = new char[size + 1];
|
|
|
|
if ( ( 0 != size ) && ( 0 != data ) ) {
|
|
stream.seekg( (*convertor)( header.sh_offset ) );
|
|
if (get_flags() & 0x08000000){
|
|
uint32_t uncompressed_size = size;
|
|
stream.read( (char *) &uncompressed_size, 4);
|
|
stream.read( data, size - 4);
|
|
|
|
char* uncompressedData = new char[uncompressed_size + 1];
|
|
|
|
int ret = 0;
|
|
z_stream s;
|
|
memset(&s, 0, sizeof(s));
|
|
|
|
s.zalloc = Z_NULL;
|
|
s.zfree = Z_NULL;
|
|
s.opaque = Z_NULL;
|
|
|
|
ret = inflateInit_(&s, ZLIB_VERSION, sizeof(s));
|
|
if (ret != Z_OK)
|
|
return;
|
|
|
|
s.avail_in = size - 4;
|
|
s.next_in = (Bytef *)data;
|
|
|
|
s.avail_out = uncompressed_size;
|
|
s.next_out = (Bytef *)&uncompressedData[0];
|
|
|
|
ret = inflate(&s, Z_FINISH);
|
|
if (ret != Z_OK && ret != Z_STREAM_END){
|
|
DEBUG_FUNCTION_LINE("NOOOO");
|
|
}
|
|
|
|
inflateEnd(&s);
|
|
|
|
delete [] data;
|
|
data = uncompressedData;
|
|
data_size = uncompressed_size;
|
|
set_size(uncompressed_size);
|
|
data[data_size] = 0; // Ensure data is ended with 0 to avoid oob read
|
|
|
|
}else{
|
|
stream.read( data, size );
|
|
data[size] = 0; // Ensure data is ended with 0 to avoid oob read
|
|
data_size = size;
|
|
}
|
|
}else{
|
|
set_size(0);
|
|
DEBUG_FUNCTION_LINE("Failed to allocate memory.");
|
|
}
|
|
}
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
save( std::ostream& stream,
|
|
std::streampos header_offset,
|
|
std::streampos data_offset )
|
|
{
|
|
if ( 0 != get_index() ) {
|
|
header.sh_offset = data_offset;
|
|
header.sh_offset = (*convertor)( header.sh_offset );
|
|
}
|
|
|
|
save_header( stream, header_offset );
|
|
if ( get_type() != SHT_NOBITS && get_type() != SHT_NULL &&
|
|
get_size() != 0 && data != 0 ) {
|
|
save_data( stream, data_offset );
|
|
}
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
private:
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
save_header( std::ostream& stream,
|
|
std::streampos header_offset ) const
|
|
{
|
|
stream.seekp( header_offset );
|
|
stream.write( reinterpret_cast<const char*>( &header ), sizeof( header ) );
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void
|
|
save_data( std::ostream& stream,
|
|
std::streampos data_offset ) const
|
|
{
|
|
stream.seekp( data_offset );
|
|
stream.write( get_data(), get_size() );
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
private:
|
|
T header;
|
|
Elf_Half index;
|
|
std::string name;
|
|
char* data;
|
|
Elf_Word data_size;
|
|
const endianess_convertor* convertor;
|
|
bool is_address_set;
|
|
};
|
|
|
|
} // namespace ELFIO
|
|
|
|
#endif // ELFIO_SECTION_HPP
|