mirror/ELFIO
1
0
Fork 0
mirror of https://github.com/serge1/ELFIO.git synced 2024-12-27 12:17:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for writing RPL/RPX

Browse Source 
== DETAILS
Honor the compression flag and write compressed section data.
This commit is contained in:
Nathan Strong 2022-10-25 19:18:51 -07:00 committed by Serge Lamikhov-Center
parent 582d929d9d
commit ac542cbb8b
1 changed files with 37 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
elfio/elfio_section.hpp
View File

@ -318,11 +318,47 @@ template <class T> class section_impl : public section
void save_data( std::ostream& stream, std::streampos data_offset ) const void save_data( std::ostream& stream, std::streampos data_offset ) const
{ {
adjust_stream_size( stream, data_offset ); adjust_stream_size( stream, data_offset );
stream.write( get_data(), get_size() );
if(get_flags() & SHF_RPX_DEFLATE) {
Elf_Xword compressed_size = get_size();
auto compressed_ptr = compress_data(compressed_size);
stream.write( compressed_ptr.get(), compressed_size);
} else {
stream.write( get_data(), get_size() );
}
} }
//------------------------------------------------------------------------------ //------------------------------------------------------------------------------
private: private:
std::unique_ptr<char[]> compress_data(Elf_Xword &size) const {
auto compressed = std::unique_ptr<char[]>(new char[size]);
int z_result = 0;
z_stream s = { 0 };
s.zalloc = Z_NULL;
s.zfree = Z_NULL;
s.opaque = Z_NULL;
if(Z_OK != (z_result = deflateInit(&s, Z_DEFAULT_COMPRESSION))) {
std::cerr << "failed to init zlib for compression: " << z_result << std::endl;
return nullptr;
}
s.avail_in = size;
s.next_in = (Bytef *)data.get();
s.avail_out = size;
s.next_out = (Bytef *)compressed.get();
z_result = deflate(&s, Z_FINISH);
if(z_result != Z_OK && z_result != Z_STREAM_END) {
std::cerr << "deflate failed: " << z_result << std::endl;
deflateEnd(&s);
return nullptr;
}
size = size - s.avail_out;
deflateEnd(&s);
return compressed;
}
T header = { 0 }; T header = { 0 };
Elf_Half index = 0; Elf_Half index = 0;
std::string name; std::string name;