mirror of
https://github.com/polserver/polserver
synced 2026-08-13 08:23:08 -04:00
63326cd44c
* Add format/ CompiledScriptSerializer * Add CompiledScript output, write with CompiledScriptSerializer
111 lines
4 KiB
C++
111 lines
4 KiB
C++
#include "CompiledScriptSerializer.h"
|
|
|
|
#include <cstring>
|
|
#include <fstream>
|
|
|
|
#include "../clib/clib.h"
|
|
#include "../clib/logfacility.h"
|
|
|
|
#include "compiler/representation/CompiledScript.h"
|
|
#include "compiler/representation/ExportedFunction.h"
|
|
#include "compiler/representation/ModuleDescriptor.h"
|
|
#include "compiler/representation/ModuleFunctionDescriptor.h"
|
|
#include "StoredToken.h"
|
|
#include "filefmt.h"
|
|
|
|
namespace Pol
|
|
{
|
|
namespace Bscript
|
|
{
|
|
namespace Compiler
|
|
{
|
|
CompiledScriptSerializer::CompiledScriptSerializer( const CompiledScript& compiled_script )
|
|
: compiled_script( compiled_script )
|
|
{
|
|
}
|
|
|
|
void CompiledScriptSerializer::write( const std::string& pathname ) const
|
|
{
|
|
std::ofstream ofs( pathname, std::ofstream::binary );
|
|
|
|
BSCRIPT_FILE_HDR hdr{};
|
|
hdr.magic2[0] = BSCRIPT_FILE_MAGIC0;
|
|
hdr.magic2[1] = BSCRIPT_FILE_MAGIC1;
|
|
hdr.version = ESCRIPT_FILE_VER_CURRENT; // auto-set to latest version (see filefmt.h)
|
|
hdr.globals = static_cast<unsigned short>( compiled_script.global_variable_names.size() );
|
|
ofs.write( reinterpret_cast<const char*>( &hdr ), sizeof hdr );
|
|
|
|
BSCRIPT_SECTION_HDR sechdr{};
|
|
|
|
if ( compiled_script.program_info )
|
|
{
|
|
BSCRIPT_PROGDEF_HDR progdef_hdr{};
|
|
memset( &progdef_hdr, 0, sizeof progdef_hdr );
|
|
sechdr.type = BSCRIPT_SECTION_PROGDEF;
|
|
sechdr.length = sizeof progdef_hdr;
|
|
ofs.write( reinterpret_cast<const char*>( &sechdr ), sizeof sechdr );
|
|
progdef_hdr.expectedArgs = compiled_script.program_info->parameter_count;
|
|
ofs.write( reinterpret_cast<const char*>( &progdef_hdr ), sizeof progdef_hdr );
|
|
}
|
|
|
|
for ( auto& module : compiled_script.module_descriptors )
|
|
{
|
|
sechdr.type = BSCRIPT_SECTION_MODULE;
|
|
sechdr.length = 0;
|
|
ofs.write( reinterpret_cast<const char*>( &sechdr ), sizeof sechdr );
|
|
|
|
BSCRIPT_MODULE_HDR modhdr{};
|
|
memset( &modhdr, 0, sizeof modhdr );
|
|
Clib::stracpy( modhdr.modulename, module.name.c_str(), sizeof modhdr.modulename );
|
|
modhdr.nfuncs = static_cast<unsigned int>( module.functions.size() );
|
|
ofs.write( reinterpret_cast<const char*>( &modhdr ), sizeof modhdr );
|
|
for ( auto& module_func : module.functions )
|
|
{
|
|
BSCRIPT_MODULE_FUNCTION func{};
|
|
memset( &func, 0, sizeof func );
|
|
|
|
Clib::stracpy( func.funcname, module_func.name.c_str(), sizeof func.funcname );
|
|
func.nargs = static_cast<unsigned char>( module_func.parameter_count );
|
|
|
|
ofs.write( reinterpret_cast<const char*>( &func ), sizeof func );
|
|
}
|
|
}
|
|
|
|
INFO_PRINT << "code section: offset " << ofs.tellp() << "\n";
|
|
u32 code_len = compiled_script.code.size() * sizeof( StoredToken );
|
|
sechdr.type = BSCRIPT_SECTION_CODE;
|
|
sechdr.length = sizeof code_len + code_len;
|
|
ofs.write( reinterpret_cast<const char*>( &sechdr ), sizeof sechdr );
|
|
ofs.write( reinterpret_cast<const char*>( &code_len ), sizeof code_len );
|
|
ofs.write( reinterpret_cast<const char*>( compiled_script.code.data() ), code_len );
|
|
|
|
INFO_PRINT << "data section: offset " << ofs.tellp() << "\n";
|
|
|
|
const std::vector<uint8_t>& data = compiled_script.data;
|
|
u32 data_len = data.size();
|
|
sechdr.type = BSCRIPT_SECTION_SYMBOLS;
|
|
sechdr.length = sizeof data_len + data_len;
|
|
ofs.write( reinterpret_cast<const char*>( &sechdr ), sizeof sechdr );
|
|
ofs.write( reinterpret_cast<const char*>( &data_len ), sizeof data_len );
|
|
ofs.write( reinterpret_cast<const char*>( data.data() ), data_len );
|
|
|
|
if ( !compiled_script.exported_functions.empty() )
|
|
{
|
|
BSCRIPT_EXPORTED_FUNCTION bef{};
|
|
sechdr.type = BSCRIPT_SECTION_EXPORTED_FUNCTIONS;
|
|
sechdr.length =
|
|
static_cast<unsigned int>( compiled_script.exported_functions.size() * sizeof bef );
|
|
ofs.write( reinterpret_cast<const char*>( &sechdr ), sizeof sechdr );
|
|
for ( auto& elem : compiled_script.exported_functions )
|
|
{
|
|
Clib::stracpy( bef.funcname, elem.name.c_str(), sizeof bef.funcname );
|
|
bef.nargs = elem.parameter_count;
|
|
bef.PC = elem.entrypoint_program_counter;
|
|
ofs.write( reinterpret_cast<const char*>( &bef ), sizeof bef );
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace Compiler
|
|
} // namespace Bscript
|
|
} // namespace Pol
|