diff options
| author | Amlal <amlal@nekernel.org> | 2025-05-01 08:35:38 +0200 |
|---|---|---|
| committer | Amlal <amlal@nekernel.org> | 2025-05-01 08:35:38 +0200 |
| commit | 35ac989886dcab70ff4ba6bc6a1359d1290560ad (patch) | |
| tree | 4ba89bacb3209e8d5763bf9759776c453a1aa413 | |
| parent | 6af5722b2b29731b3d96abfa7c2a512f5984312b (diff) | |
ld64: Make linker more robust, and extend PEF to version 4.
Signed-off-by: Amlal <amlal@nekernel.org>
| -rw-r--r-- | dev/LibCompiler/Detail/ClUtils.h | 8 | ||||
| -rw-r--r-- | dev/LibCompiler/PEF.h | 10 | ||||
| -rw-r--r-- | dev/LibCompiler/src/Assembler32x0.cc | 1 | ||||
| -rw-r--r-- | dev/LibCompiler/src/Assembler64x0.cc | 2 | ||||
| -rw-r--r-- | dev/LibCompiler/src/AssemblerAMD64.cc | 2 | ||||
| -rw-r--r-- | dev/LibCompiler/src/AssemblerARM64.cc | 2 | ||||
| -rw-r--r-- | dev/LibCompiler/src/AssemblerPowerPC.cc | 2 | ||||
| -rw-r--r-- | dev/LibCompiler/src/CCompiler64x0.cc | 5 | ||||
| -rw-r--r-- | dev/LibCompiler/src/CCompilerARM64.cc | 5 | ||||
| -rw-r--r-- | dev/LibCompiler/src/CCompilerPower64.cc | 5 | ||||
| -rw-r--r-- | dev/LibCompiler/src/CPlusPlusCompilerAMD64.cc | 10 | ||||
| -rw-r--r-- | dev/LibCompiler/src/CPlusPlusCompilerPreProcessor.cc | 9 | ||||
| -rw-r--r-- | dev/LibCompiler/src/DynamicLinkerPEF.cc | 137 |
13 files changed, 120 insertions, 78 deletions
diff --git a/dev/LibCompiler/Detail/ClUtils.h b/dev/LibCompiler/Detail/ClUtils.h index 1f31be7..224e4d3 100644 --- a/dev/LibCompiler/Detail/ClUtils.h +++ b/dev/LibCompiler/Detail/ClUtils.h @@ -9,6 +9,7 @@ #include <LibCompiler/AssemblyInterface.h> #include <LibCompiler/ErrorID.h> #include <LibCompiler/Parser.h> +#include <Vendor/Dialogs.h> #define kZero64Section ".zero64" #define kCode64Section ".code64" @@ -47,4 +48,11 @@ inline void print_warning(std::string reason, std::string file) noexcept { kStdOut << kYellow << "drv: " << kWhite << reason << kBlank << std::endl; } + +/// @internal +inline void segfault_handler(std::int32_t _) { + pfd::notify("NeKernel Compiler Driver System", + "Driver just crashed, please report this to the developers."); + std::exit(LIBCOMPILER_EXEC_ERROR); +} } // namespace Detail
\ No newline at end of file diff --git a/dev/LibCompiler/PEF.h b/dev/LibCompiler/PEF.h index f32d830..6e8d288 100644 --- a/dev/LibCompiler/PEF.h +++ b/dev/LibCompiler/PEF.h @@ -31,7 +31,7 @@ #define kPefMagicLen (5) -#define kPefVersion (3) +#define kPefVersion (4) #define kPefNameLen (255) #define kPefBaseOrigin (0x40000000) @@ -53,7 +53,7 @@ enum { }; enum { - kPefSubArchAMD, + kPefSubArchAMD = 200, kPefSubArchIntel, kPefSubArchARM, kPefSubArchGeneric, @@ -61,7 +61,7 @@ enum { }; enum { - kPefKindExec = 1, /* .o */ + kPefKindExec = 1, /* .exec */ kPefKindDylib = 2, /* .dylib */ kPefKindObject = 4, /* .obj */ kPefKindDebug = 5, /* .dbg */ @@ -72,7 +72,7 @@ enum { /* PEF container */ typedef struct PEFContainer final { CharType Magic[kPefMagicLen]; - UInt32 Linker; + UInt32 Linker; /* Linker used to link executable */ UInt32 Version; UInt32 Kind; UInt32 Abi; @@ -81,6 +81,7 @@ typedef struct PEFContainer final { UIntPtr Start; /* Origin of code */ SizeType HdrSz; /* Size of header */ SizeType Count; /* container header count */ + UInt32 Checksum; /* Whole binary checksum */ } PACKED PEFContainer, *PEFContainerPtr; /* First PEFCommandHeader starts after PEFContainer */ @@ -95,6 +96,7 @@ typedef struct PEFCommandHeader final { UInt32 Flags; /* container flags */ UInt16 Kind; /* container kind */ UIntPtr Offset; /* file offset */ + UIntPtr VMAddress; /* VM offset */ SizeType Size; /* file size */ } PACKED PEFCommandHeader, *PEFCommandHeaderPtr; diff --git a/dev/LibCompiler/src/Assembler32x0.cc b/dev/LibCompiler/src/Assembler32x0.cc index ac24946..a5a435c 100644 --- a/dev/LibCompiler/src/Assembler32x0.cc +++ b/dev/LibCompiler/src/Assembler32x0.cc @@ -45,5 +45,6 @@ ///////////////////////////////////////////////////////////////////////////////////////// LIBCOMPILER_MODULE(NEAssemblerMain32000) { + return 0; } diff --git a/dev/LibCompiler/src/Assembler64x0.cc b/dev/LibCompiler/src/Assembler64x0.cc index 6d4d31a..4707be2 100644 --- a/dev/LibCompiler/src/Assembler64x0.cc +++ b/dev/LibCompiler/src/Assembler64x0.cc @@ -67,6 +67,8 @@ static bool asm_read_attributes(std::string line); ///////////////////////////////////////////////////////////////////////////////////////// LIBCOMPILER_MODULE(AssemblerMain64x0) { + ::signal(SIGSEGV, Detail::segfault_handler); + for (size_t i = 1; i < argc; ++i) { if (argv[i][0] == '-') { if (strcmp(argv[i], "--ver") == 0 || strcmp(argv[i], "--v") == 0) { diff --git a/dev/LibCompiler/src/AssemblerAMD64.cc b/dev/LibCompiler/src/AssemblerAMD64.cc index f8772b8..72e8015 100644 --- a/dev/LibCompiler/src/AssemblerAMD64.cc +++ b/dev/LibCompiler/src/AssemblerAMD64.cc @@ -88,6 +88,8 @@ static bool asm_read_attributes(std::string line); LIBCOMPILER_MODULE(AssemblerMainAMD64) { //////////////// CPU OPCODES BEGIN //////////////// + ::signal(SIGSEGV, Detail::segfault_handler); + std::string opcodes_jump[kJumpLimit] = {"ja", "jae", "jb", "jbe", "jc", "je", "jg", "jge", "jl", "jle", "jna", "jnae", "jnb", "jnbe", "jnc", "jne", "jng", "jnge", "jnl", "jnle", "jno", "jnp", "jns", "jnz", diff --git a/dev/LibCompiler/src/AssemblerARM64.cc b/dev/LibCompiler/src/AssemblerARM64.cc index eb3e915..ab2c89e 100644 --- a/dev/LibCompiler/src/AssemblerARM64.cc +++ b/dev/LibCompiler/src/AssemblerARM64.cc @@ -74,6 +74,8 @@ static bool asm_read_attributes(std::string line); ///////////////////////////////////////////////////////////////////////////////////////// LIBCOMPILER_MODULE(AssemblerMainARM64) { + ::signal(SIGSEGV, Detail::segfault_handler); + for (size_t i = 1; i < argc; ++i) { if (argv[i][0] == '-') { if (strcmp(argv[i], "--ver") == 0 || strcmp(argv[i], "--v") == 0) { diff --git a/dev/LibCompiler/src/AssemblerPowerPC.cc b/dev/LibCompiler/src/AssemblerPowerPC.cc index ffe5c6a..bd44e6e 100644 --- a/dev/LibCompiler/src/AssemblerPowerPC.cc +++ b/dev/LibCompiler/src/AssemblerPowerPC.cc @@ -74,6 +74,8 @@ static bool asm_read_attributes(std::string line); ///////////////////////////////////////////////////////////////////////////////////////// LIBCOMPILER_MODULE(AssemblerMainPower64) { + ::signal(SIGSEGV, Detail::segfault_handler); + for (size_t i = 1; i < argc; ++i) { if (argv[i][0] == '-') { if (strcmp(argv[i], "--ver") == 0 || strcmp(argv[i], "--v") == 0) { diff --git a/dev/LibCompiler/src/CCompiler64x0.cc b/dev/LibCompiler/src/CCompiler64x0.cc index 6fa05fb..c43abfe 100644 --- a/dev/LibCompiler/src/CCompiler64x0.cc +++ b/dev/LibCompiler/src/CCompiler64x0.cc @@ -13,6 +13,7 @@ #include <LibCompiler/Backend/64x0.h> #include <LibCompiler/Parser.h> #include <LibCompiler/UUID.h> +#include <LibCompiler/Detail/ClUtils.h> #include <cstdio> #include <fstream> #include <iostream> @@ -86,9 +87,7 @@ struct CompilerState final { } // namespace Detail static Detail::CompilerState kState; -static SizeType kErrorLimit = 100; static std::string kIfFunction = ""; -static Int32 kAcceptableErrors = 0; namespace Detail { /// @brief prints an error into stdout. @@ -1201,6 +1200,8 @@ static void cc_print_help() { #define kExt ".c" LIBCOMPILER_MODULE(NeOSCompilerCLang64x0) { + ::signal(SIGSEGV, Detail::segfault_handler); + kCompilerTypes.push_back({.fName = "void", .fValue = "void"}); kCompilerTypes.push_back({.fName = "char", .fValue = "byte"}); kCompilerTypes.push_back({.fName = "short", .fValue = "hword"}); diff --git a/dev/LibCompiler/src/CCompilerARM64.cc b/dev/LibCompiler/src/CCompilerARM64.cc index b2f36e9..26867f1 100644 --- a/dev/LibCompiler/src/CCompilerARM64.cc +++ b/dev/LibCompiler/src/CCompilerARM64.cc @@ -13,6 +13,7 @@ #include <LibCompiler/Backend/Aarch64.h> #include <LibCompiler/Parser.h> #include <LibCompiler/UUID.h> +#include <LibCompiler/Detail/ClUtils.h> #include <cstdio> #include <fstream> #include <iostream> @@ -86,9 +87,7 @@ struct CompilerState final { } // namespace Detail static Detail::CompilerState kState; -static SizeType kErrorLimit = 100; static std::string kIfFunction = ""; -static Int32 kAcceptableErrors = 0; namespace Detail { /// @brief prints an error into stdout. @@ -1201,6 +1200,8 @@ static void cc_print_help() { #define kCExtension ".c" LIBCOMPILER_MODULE(NeOSCompilerCLangARM64) { + ::signal(SIGSEGV, Detail::segfault_handler); + kCompilerTypes.push_back({.fName = "void", .fValue = "void"}); kCompilerTypes.push_back({.fName = "char", .fValue = "byte"}); kCompilerTypes.push_back({.fName = "short", .fValue = "hword"}); diff --git a/dev/LibCompiler/src/CCompilerPower64.cc b/dev/LibCompiler/src/CCompilerPower64.cc index ee9c5a4..f10aa79 100644 --- a/dev/LibCompiler/src/CCompilerPower64.cc +++ b/dev/LibCompiler/src/CCompilerPower64.cc @@ -10,6 +10,7 @@ #include <LibCompiler/Backend/PowerPC.h> #include <LibCompiler/Parser.h> #include <LibCompiler/UUID.h> +#include <LibCompiler/Detail/ClUtils.h> #include <cstdio> #include <fstream> #include <iostream> @@ -77,9 +78,7 @@ struct CompilerState final { } // namespace Detail static Detail::CompilerState kState; -static SizeType kErrorLimit = 100; static std::string kIfFunction = ""; -static Int32 kAcceptableErrors = 0; namespace Detail { /// @brief prints an error into stdout. @@ -1220,6 +1219,8 @@ static void cc_print_help() { #define kExt ".c" LIBCOMPILER_MODULE(NeOSCompilerCLangPowerPC) { + ::signal(SIGSEGV, Detail::segfault_handler); + kCompilerTypes.push_back({.fName = "void", .fValue = "void"}); kCompilerTypes.push_back({.fName = "char", .fValue = "byte"}); kCompilerTypes.push_back({.fName = "short", .fValue = "hword"}); diff --git a/dev/LibCompiler/src/CPlusPlusCompilerAMD64.cc b/dev/LibCompiler/src/CPlusPlusCompilerAMD64.cc index 8597d41..b4725a4 100644 --- a/dev/LibCompiler/src/CPlusPlusCompilerAMD64.cc +++ b/dev/LibCompiler/src/CPlusPlusCompilerAMD64.cc @@ -20,7 +20,6 @@ #include <LibCompiler/Detail/ClUtils.h> #include <LibCompiler/Parser.h> #include <LibCompiler/UUID.h> -#include <Vendor/Dialogs.h> #include <cstdio> @@ -51,13 +50,6 @@ ///////////////////////////////////// /// @internal -static void cxxdrv_chaos_handler(std::int32_t _) { - pfd::notify("NeKernel C++ Compiler Driver", - "CxxDrv just crashed, please report this to the developers."); - std::exit(LIBCOMPILER_EXEC_ERROR); -} - -/// @internal namespace Detail { std::filesystem::path expand_home(const std::filesystem::path& p) { if (!p.empty() && p.string()[0] == '~') { @@ -939,7 +931,7 @@ LIBCOMPILER_MODULE(CompilerCPlusPlusAMD64) { kFactory.Mount(new AssemblyCPlusPlusInterface()); kCompilerFrontend = new CompilerFrontendCPlusPlus(); - ::signal(SIGSEGV, cxxdrv_chaos_handler); + ::signal(SIGSEGV, Detail::segfault_handler); for (auto index = 1UL; index < argc; ++index) { if (argv[index][0] == '-') { diff --git a/dev/LibCompiler/src/CPlusPlusCompilerPreProcessor.cc b/dev/LibCompiler/src/CPlusPlusCompilerPreProcessor.cc index d2194a8..5d035da 100644 --- a/dev/LibCompiler/src/CPlusPlusCompilerPreProcessor.cc +++ b/dev/LibCompiler/src/CPlusPlusCompilerPreProcessor.cc @@ -751,6 +751,13 @@ LIBCOMPILER_MODULE(CPlusPlusPreprocessorMain) { kMacros.push_back(macro_unreachable); + Detail::bpp_macro macro_unused; + + macro_unreachable.fName = "__unused"; + macro_unreachable.fValue = "__libcompiler_unused"; + + kMacros.push_back(macro_unused); + Detail::bpp_macro macro_0; macro_0.fName = "__false"; @@ -784,7 +791,7 @@ LIBCOMPILER_MODULE(CPlusPlusPreprocessorMain) { kMacros.push_back(macro_size_t); macro_size_t.fName = "__UINTPTR_TYPE__"; - macro_size_t.fValue = "unsigned int"; + macro_size_t.fValue = "unsigned long long int"; kMacros.push_back(macro_size_t); diff --git a/dev/LibCompiler/src/DynamicLinkerPEF.cc b/dev/LibCompiler/src/DynamicLinkerPEF.cc index fb2403f..0def365 100644 --- a/dev/LibCompiler/src/DynamicLinkerPEF.cc +++ b/dev/LibCompiler/src/DynamicLinkerPEF.cc @@ -37,6 +37,12 @@ //! Format header. #include <format> +//! LibCompiler utils. +#include <LibCompiler/Detail/ClUtils.h> + +//! I/O stream from std c++ +#include <iostream> + #define kLinkerVersionStr \ "NeKernel 64-Bit Linker (Preferred Executable Format) {}, (c) Amlal El Mahrouss " \ "2024-2025 " \ @@ -48,18 +54,18 @@ #define kPefNoCpu (0U) #define kPefNoSubCpu (0U) -#define kStdOut (std::cout << "\e[0;31m" << "ld64: " << "\e[0;97m") - #define kLinkerDefaultOrigin kPefBaseOrigin #define kLinkerId (0x5046FF) #define kLinkerAbiContainer "__PEFContainer:ABI:" #define kPrintF printf -#define kLinkerSplash() kStdOut << std::format(kLinkerVersionStr, kDistVersion) +#define kLinkerSplash() kOutCon << std::format(kLinkerVersionStr, kDistVersion) /// @brief PEF stack size symbol. #define kLinkerStackSizeSymbol "__PEFSizeOfReserveStack" +#define kOutCon (std::cout << "\e[0;31m" << "ld64: " << "\e[0;97m") + namespace Detail { struct DynamicLinkerBlob final { std::vector<CharType> mBlob{}; // PEF code/bss/data blob. @@ -68,19 +74,19 @@ struct DynamicLinkerBlob final { } // namespace Detail enum { + kABITypeNull = 0, kABITypeStart = 0x1010, /* Invalid ABI start of ABI list. */ kABITypeNE = 0x5046, /* PF (NeKernel's PEF ABI) */ kABITypeInvalid = 0xFFFF, }; -static LibCompiler::String kOutput = "a.out"; +static LibCompiler::String kOutput = "a" kPefExt; static Int32 kAbi = kABITypeNE; static Int32 kSubArch = kPefNoSubCpu; static Int32 kArch = LibCompiler::kPefArchInvalid; static Bool kFatBinaryEnable = false; static Bool kStartFound = false; static Bool kDuplicateSymbols = false; -static Bool kVerbose = false; /* ld64 is to be found, mld is to be found at runtime. */ static const CharType* kLdDefineSymbol = ":UndefinedSymbol:"; @@ -90,14 +96,13 @@ static const CharType* kLdDynamicSym = ":RuntimeSymbol:"; static std::vector<LibCompiler::String> kObjectList; static std::vector<Detail::DynamicLinkerBlob> kObjectBytes; -static uintptr_t kMIBCount = 8; -static uintptr_t kByteCount = 1024; - /// @brief NE 64-bit Linker. /// @note This linker is made for PEF executable, thus NE based OSes. LIBCOMPILER_MODULE(DynamicLinker64PEF) { bool is_executable = true; + ::signal(SIGSEGV, Detail::segfault_handler); + /** * @brief parse flags and trigger options. */ @@ -105,18 +110,18 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { if (StringCompare(argv[linker_arg], "-help") == 0) { kLinkerSplash(); - kStdOut << "-version: Show linker version.\n"; - kStdOut << "-help: Show linker help.\n"; - kStdOut << "-ld-verbose: Enable linker trace.\n"; - kStdOut << "-dylib: Output as a Dynamic PEF.\n"; - kStdOut << "-fat: Output as a FAT PEF.\n"; - kStdOut << "-32k: Output as a 32x0 PEF.\n"; - kStdOut << "-64k: Output as a 64x0 PEF.\n"; - kStdOut << "-amd64: Output as a AMD64 PEF.\n"; - kStdOut << "-rv64: Output as a RISC-V PEF.\n"; - kStdOut << "-power64: Output as a POWER PEF.\n"; - kStdOut << "-arm64: Output as a ARM64 PEF.\n"; - kStdOut << "-output: Select the output file name.\n"; + kOutCon << "-version: Show linker version.\n"; + kOutCon << "-help: Show linker help.\n"; + kOutCon << "-ld-verbose: Enable linker trace.\n"; + kOutCon << "-dylib: Output as a Dynamic PEF.\n"; + kOutCon << "-fat: Output as a FAT PEF.\n"; + kOutCon << "-32k: Output as a 32x0 PEF.\n"; + kOutCon << "-64k: Output as a 64x0 PEF.\n"; + kOutCon << "-amd64: Output as a AMD64 PEF.\n"; + kOutCon << "-rv64: Output as a RISC-V PEF.\n"; + kOutCon << "-power64: Output as a POWER PEF.\n"; + kOutCon << "-arm64: Output as a ARM64 PEF.\n"; + kOutCon << "-output: Select the output file name.\n"; return EXIT_SUCCESS; } else if (StringCompare(argv[linker_arg], "-version") == 0) { @@ -176,7 +181,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { continue; } else { if (argv[linker_arg][0] == '-') { - kStdOut << "unknown flag: " << argv[linker_arg] << "\n"; + kOutCon << "unknown flag: " << argv[linker_arg] << "\n"; return EXIT_FAILURE; } @@ -187,10 +192,10 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { } if (kOutput.empty()) { - kStdOut << "no output filename set." << std::endl; + kOutCon << "no output filename set." << std::endl; return LIBCOMPILER_EXEC_ERROR; } else if (kObjectList.empty()) { - kStdOut << "no input files." << std::endl; + kOutCon << "no input files." << std::endl; return LIBCOMPILER_EXEC_ERROR; } else { namespace FS = std::filesystem; @@ -200,7 +205,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { if (!FS::exists(obj)) { // if filesystem doesn't find file // -> throw error. - kStdOut << "no such file: " << obj << std::endl; + kOutCon << "no such file: " << obj << std::endl; return LIBCOMPILER_EXEC_ERROR; } } @@ -208,7 +213,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { // PEF expects a valid target architecture when outputing a binary. if (kArch == 0) { - kStdOut << "no target architecture set, can't continue." << std::endl; + kOutCon << "no target architecture set, can't continue." << std::endl; return LIBCOMPILER_EXEC_ERROR; } @@ -230,12 +235,13 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { // specify the start address, can be 0x10000 pef_container.Start = kLinkerDefaultOrigin; pef_container.HdrSz = sizeof(LibCompiler::PEFContainer); + pef_container.Checksum = 0UL; std::ofstream output_fc(kOutput, std::ofstream::binary); if (output_fc.bad()) { if (kVerbose) { - kStdOut << "error: " << strerror(errno) << "\n"; + kOutCon << "error: " << strerror(errno) << "\n"; } return LIBCOMPILER_FILE_NOT_FOUND; @@ -259,12 +265,12 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { if (ae_header.fMagic[0] == kAEMag0 && ae_header.fMagic[1] == kAEMag1 && ae_header.fSize == sizeof(LibCompiler::AEHeader)) { if (ae_header.fArch != kArch) { - if (kVerbose) kStdOut << "info: is this a FAT binary? : "; + if (kVerbose) kOutCon << "Info: is this a FAT binary? : "; if (!kFatBinaryEnable) { - if (kVerbose) kStdOut << "No.\n"; + if (kVerbose) kOutCon << "No.\n"; - kStdOut << "error: object " << objectFile + kOutCon << "Error: object " << objectFile << " is a different kind of architecture and output isn't " "treated as a FAT binary." << std::endl; @@ -272,7 +278,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { return LIBCOMPILER_FAT_ERROR; } else { if (kVerbose) { - kStdOut << "Architecture matches what we expect.\n"; + kOutCon << "Architecture matches what we expect.\n"; } } } @@ -281,20 +287,22 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { archs |= ae_header.fArch; std::size_t cnt = ae_header.fCount; - if (kVerbose) kStdOut << "object header found, record count: " << cnt << "\n"; + if (kVerbose) kOutCon << "Object header found, record count: " << cnt << "\n"; pef_container.Count = cnt; char_type* raw_ae_records = new char_type[cnt * sizeof(LibCompiler::AERecordHeader)]; if (!raw_ae_records) { - if (kVerbose) kStdOut << "allocation failure for records of n: " << cnt << "\n"; + if (kVerbose) kOutCon << "Allocation failure for records of count: " << cnt << "\n"; } memset(raw_ae_records, 0, cnt * sizeof(LibCompiler::AERecordHeader)); auto* ae_records = reader_protocol.Read(raw_ae_records, cnt); + auto org = kLinkerDefaultOrigin; + for (size_t ae_record_index = 0; ae_record_index < cnt; ++ae_record_index) { LibCompiler::PEFCommandHeader command_header{0}; std::size_t offset_of_obj = ae_records[ae_record_index].fOffset; @@ -327,12 +335,15 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { command_header.Kind = ae_records[ae_record_index].fKind; command_header.Size = ae_records[ae_record_index].fSize; command_header.Cpu = ae_header.fArch; + command_header.VMAddress = org; /// TODO: command_header.SubCpu = ae_header.fSubArch; + org += command_header.Size; + if (kVerbose) { - kStdOut << "Record: " << ae_records[ae_record_index].fName << " is marked.\n"; + kOutCon << "Record: " << ae_records[ae_record_index].fName << " is marked.\n"; - kStdOut << "Record offset: " << command_header.Offset << "\n"; + kOutCon << "Offset: " << command_header.Offset << "\n"; } command_headers.emplace_back(command_header); @@ -344,8 +355,6 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { std::vector<char> bytes; bytes.resize(ae_header.fCodeSize); - // TODO: Port this to NeFS. - reader_protocol.FP.seekg(std::streamsize(ae_header.fStartCode)); reader_protocol.FP.read(bytes.data(), std::streamsize(ae_header.fCodeSize)); @@ -358,7 +367,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { continue; } - kStdOut << "Not an object container: " << objectFile << std::endl; + kOutCon << "Not an container: " << objectFile << std::endl; // don't continue, it is a fatal error. return LIBCOMPILER_EXEC_ERROR; } @@ -368,7 +377,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { output_fc << pef_container; if (kVerbose) { - kStdOut << "Wrote container header.\n"; + kOutCon << "Wrote container to: " << output_fc.tellp() << ".\n"; } output_fc.seekp(std::streamsize(pef_container.HdrSz)); @@ -382,7 +391,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { // check if this symbol needs to be resolved. if (LibCompiler::String(command_hdr.Name).find(kLdDefineSymbol) != LibCompiler::String::npos && LibCompiler::String(command_hdr.Name).find(kLdDynamicSym) == LibCompiler::String::npos) { - if (kVerbose) kStdOut << "Found undefined symbol: " << command_hdr.Name << "\n"; + if (kVerbose) kOutCon << "Found undefined symbol: " << command_hdr.Name << "\n"; if (auto it = std::find(not_found.begin(), not_found.end(), LibCompiler::String(command_hdr.Name)); @@ -426,7 +435,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { not_found.erase(it); - if (kVerbose) kStdOut << "found symbol: " << command_hdr.Name << "\n"; + if (kVerbose) kOutCon << "Found symbol: " << command_hdr.Name << "\n"; break; } @@ -441,11 +450,11 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { if (!kStartFound && is_executable) { if (kVerbose) - kStdOut << "undefined entrypoint: " << kPefStart + kOutCon << "Undefined entrypoint: " << kPefStart << ", you may have forget to link " - "against your compiler's runtime library.\n"; + "against the C++ runtime library.\n"; - kStdOut << "undefined entrypoint " << kPefStart << " for executable: " << kOutput << "\n"; + kOutCon << "Undefined entrypoint " << kPefStart << " for executable: " << kOutput << "\n"; } // step 4: write all PEF commands. @@ -481,11 +490,11 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { } case LibCompiler::kPefArch32000: case LibCompiler::kPefArch64000: { - abi += " ZWS"; + abi += "_NEP"; break; } default: { - abi += " IDK"; + abi += "_IDK"; break; } } @@ -538,10 +547,22 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { std::vector<LibCompiler::String> dupl_symbols; std::vector<LibCompiler::String> resolve_symbols; - constexpr Int32 cPaddingOffset = 16; + constexpr Int32 kPaddingOffset = 16; size_t previous_offset = - (command_headers.size() * sizeof(LibCompiler::PEFCommandHeader)) + cPaddingOffset; + (command_headers.size() * sizeof(LibCompiler::PEFCommandHeader)) + kPaddingOffset; + + LibCompiler::PEFCommandHeader end_exec_hdr; + + end_exec_hdr.Offset = output_fc.tellp(); + end_exec_hdr.Flags = LibCompiler::kPefLinkerID; + end_exec_hdr.Kind = LibCompiler::kPefZero; + + MemoryCopy(end_exec_hdr.Name, "Container:Exec:END", strlen("Container:Exec:END")); + + end_exec_hdr.Size = strlen(end_exec_hdr.Name); + + command_headers.push_back(end_exec_hdr); // Finally write down the command headers. // And check for any duplications @@ -581,9 +602,9 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { } if (kVerbose) { - kStdOut << "Command header name: " << name << "\n"; - kStdOut << "Real address of command header content: " - << command_headers[commandHeaderIndex].Offset << "\n"; + kOutCon << "Command name: " << name << "\n"; + kOutCon << "VMAddress of command content: " << command_headers[commandHeaderIndex].Offset + << "\n"; } output_fc << command_headers[commandHeaderIndex]; @@ -597,7 +618,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { LibCompiler::String(command_headers[sub_command_header_index].Name).find(kLdDynamicSym) == LibCompiler::String::npos) { if (kVerbose) { - kStdOut << "ignore :UndefinedSymbol: command header...\n"; + kOutCon << "Ignoring :UndefinedSymbol: headers...\n"; } // ignore :UndefinedSymbol: headers, they do not contain code. @@ -612,7 +633,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { dupl_symbols.emplace_back(command_hdr.Name); } - if (kVerbose) kStdOut << "found duplicate symbol: " << command_hdr.Name << "\n"; + if (kVerbose) kOutCon << "Found duplicate symbols of: " << command_hdr.Name << "\n"; kDuplicateSymbols = true; } @@ -621,7 +642,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { if (!dupl_symbols.empty()) { for (auto& symbol : dupl_symbols) { - kStdOut << "Multiple symbols of: " << symbol << " detected, cannot continue.\n"; + kOutCon << "Multiple symbols of: " << symbol << " detected, cannot continue.\n"; } return LIBCOMPILER_EXEC_ERROR; @@ -634,7 +655,7 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { } if (kVerbose) { - kStdOut << "wrote contents of: " << kOutput << "\n"; + kOutCon << "Wrote contents of: " << kOutput << "\n"; } // step 3: check if we have those symbols @@ -651,16 +672,16 @@ LIBCOMPILER_MODULE(DynamicLinker64PEF) { if (!unreferenced_symbols.empty()) { for (auto& unreferenced_symbol : unreferenced_symbols) { - kStdOut << "undefined symbol " << unreferenced_symbol << "\n"; + kOutCon << "Undefined symbol " << unreferenced_symbol << "\n"; } return LIBCOMPILER_EXEC_ERROR; } - if (!kStartFound || kDuplicateSymbols && std::filesystem::exists(kOutput) || - !unreferenced_symbols.empty()) { + if ((!kStartFound || kDuplicateSymbols) && + (std::filesystem::exists(kOutput) || !unreferenced_symbols.empty())) { if (kVerbose) { - kStdOut << "file: " << kOutput << ", is corrupt, removing file...\n"; + kOutCon << "File: " << kOutput << ", is corrupt, removing file...\n"; } return LIBCOMPILER_EXEC_ERROR; |
