diff options
Diffstat (limited to 'tools/ramscrgen')
-rw-r--r-- | tools/ramscrgen/.gitignore | 2 | ||||
-rw-r--r-- | tools/ramscrgen/elf.cpp | 544 | ||||
-rw-r--r-- | tools/ramscrgen/main.cpp | 374 | ||||
-rw-r--r-- | tools/ramscrgen/sym_file.cpp | 984 |
4 files changed, 952 insertions, 952 deletions
diff --git a/tools/ramscrgen/.gitignore b/tools/ramscrgen/.gitignore index f17f6e7..4d0ae82 100644 --- a/tools/ramscrgen/.gitignore +++ b/tools/ramscrgen/.gitignore @@ -1 +1 @@ -ramscrgen
+ramscrgen diff --git a/tools/ramscrgen/elf.cpp b/tools/ramscrgen/elf.cpp index 7cef9f9..7e78704 100644 --- a/tools/ramscrgen/elf.cpp +++ b/tools/ramscrgen/elf.cpp @@ -1,272 +1,272 @@ -#include <cstdio>
-#include <cstring>
-#include <cstdint>
-#include <map>
-#include <vector>
-#include <string>
-#include "ramscrgen.h"
-#include "elf.h"
-
-#define SHN_COMMON 0xFFF2
-
-static std::string s_elfPath;
-static std::string s_archiveFilePath;
-static std::string s_archiveObjectPath;
-
-static FILE *s_file;
-
-static std::uint32_t s_sectionHeaderOffset;
-static int s_sectionHeaderEntrySize;
-static int s_sectionCount;
-static int s_shstrtabIndex;
-
-static std::uint32_t s_symtabOffset;
-static std::uint32_t s_strtabOffset;
-
-static std::uint32_t s_symbolCount;
-static std::uint32_t s_elfFileOffset;
-
-struct Symbol
-{
- std::uint32_t nameOffset;
- std::uint32_t size;
-};
-
-static void Seek(long offset)
-{
- if (std::fseek(s_file, s_elfFileOffset + offset, SEEK_SET) != 0)
- FATAL_ERROR("error: failed to seek to %ld in \"%s\"", offset, s_elfPath.c_str());
-}
-
-static void Skip(long offset)
-{
- if (std::fseek(s_file, offset, SEEK_CUR) != 0)
- FATAL_ERROR("error: failed to skip %ld bytes in \"%s\"", offset, s_elfPath.c_str());
-}
-
-static std::uint32_t ReadInt8()
-{
- int c = std::fgetc(s_file);
-
- if (c < 0)
- FATAL_ERROR("error: unexpected EOF when reading ELF file \"%s\"\n", s_elfPath.c_str());
-
- return c;
-}
-
-static std::uint32_t ReadInt16()
-{
- std::uint32_t val = 0;
- val |= ReadInt8();
- val |= ReadInt8() << 8;
- return val;
-}
-
-static std::uint32_t ReadInt32()
-{
- std::uint32_t val = 0;
- val |= ReadInt8();
- val |= ReadInt8() << 8;
- val |= ReadInt8() << 16;
- val |= ReadInt8() << 24;
- return val;
-}
-
-static std::string ReadString()
-{
- std::string s;
- char c;
-
- while ((c = ReadInt8()) != 0)
- s += c;
-
- return s;
-}
-
-static void VerifyElfIdent()
-{
- char expectedMagic[4] = { 0x7F, 'E', 'L', 'F' };
- char magic[4];
-
- if (std::fread(magic, 4, 1, s_file) != 1)
- FATAL_ERROR("error: failed to read ELF magic from \"%s\"\n", s_elfPath.c_str());
-
- if (std::memcmp(magic, expectedMagic, 4) != 0)
- FATAL_ERROR("error: ELF magic did not match in \"%s\"\n", s_elfPath.c_str());
-
- if (std::fgetc(s_file) != 1)
- FATAL_ERROR("error: \"%s\" not 32-bit ELF\n", s_elfPath.c_str());
-
- if (std::fgetc(s_file) != 1)
- FATAL_ERROR("error: \"%s\" not little-endian ELF\n", s_elfPath.c_str());
-}
-
-static void VerifyAr()
-{
- char expectedMagic[8] = {'!', '<', 'a', 'r', 'c', 'h', '>', '\n'};
- char magic[8];
-
- if (std::fread(magic, 8, 1, s_file) != 1)
- FATAL_ERROR("error: failed to read AR magic from \"%s\"\n", s_archiveFilePath.c_str());
-
- if (std::memcmp(magic, expectedMagic, 8) != 0)
- FATAL_ERROR("error: AR magic did not match in \"%s\"\n", s_archiveFilePath.c_str());
-}
-
-static void ReadElfHeader()
-{
- Seek(0x20);
- s_sectionHeaderOffset = ReadInt32();
- Seek(0x2E);
- s_sectionHeaderEntrySize = ReadInt16();
- s_sectionCount = ReadInt16();
- s_shstrtabIndex = ReadInt16();
-}
-
-static void FindArObj()
-{
- char file_ident[17] = {0};
- char filesize_s[11] = {0};
- char expectedEndMagic[2] = { 0x60, 0x0a };
- char end_magic[2];
- std::size_t filesize;
-
- Seek(8);
- while (!std::feof(s_file)) {
- if (std::fread(file_ident, 16, 1, s_file) != 1)
- FATAL_ERROR("error: failed to read file ident in \"%s\"\n", s_archiveFilePath.c_str());
- Skip(32);
- if (std::fread(filesize_s, 10, 1, s_file) != 1)
- FATAL_ERROR("error: failed to read filesize in \"%s\"\n", s_archiveFilePath.c_str());
- if (std::fread(end_magic, 2, 1, s_file) != 1)
- FATAL_ERROR("error: failed to read end sentinel in \"%s\"\n", s_archiveFilePath.c_str());
- if (std::memcmp(end_magic, expectedEndMagic, 2) != 0)
- FATAL_ERROR("error: corrupted archive header in \"%s\" at \"%s\"\n", s_archiveFilePath.c_str(), file_ident);
-
- char * ptr = std::strchr(file_ident, '/');
- if (ptr != nullptr)
- *ptr = 0;
- filesize = std::strtoul(filesize_s, nullptr, 10);
- if (std::strncmp(s_archiveObjectPath.c_str(), file_ident, 16) == 0) {
- s_elfFileOffset = std::ftell(s_file);
- return;
- }
- Skip(filesize);
- }
-
- FATAL_ERROR("error: could not find object \"%s\" in archive \"%s\"\n", s_archiveObjectPath.c_str(), s_archiveFilePath.c_str());
-}
-
-static std::string GetSectionName(std::uint32_t shstrtabOffset, int index)
-{
- Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * index);
- std::uint32_t nameOffset = ReadInt32();
- Seek(shstrtabOffset + nameOffset);
- return ReadString();
-}
-
-static void FindTableOffsets()
-{
- s_symtabOffset = 0;
- s_strtabOffset = 0;
-
- Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * s_shstrtabIndex + 0x10);
- std::uint32_t shstrtabOffset = ReadInt32();
-
- for (int i = 0; i < s_sectionCount; i++)
- {
- std::string name = GetSectionName(shstrtabOffset, i);
-
- if (name == ".symtab")
- {
- if (s_symtabOffset)
- FATAL_ERROR("error: mutiple .symtab sections found in \"%s\"\n", s_elfPath.c_str());
- Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * i + 0x10);
- s_symtabOffset = ReadInt32();
- std::uint32_t size = ReadInt32();
- s_symbolCount = size / 16;
- }
- else if (name == ".strtab")
- {
- if (s_strtabOffset)
- FATAL_ERROR("error: mutiple .strtab sections found in \"%s\"\n", s_elfPath.c_str());
- Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * i + 0x10);
- s_strtabOffset = ReadInt32();
- }
- }
-
- if (!s_symtabOffset)
- FATAL_ERROR("error: couldn't find .symtab section in \"%s\"\n", s_elfPath.c_str());
-
- if (!s_strtabOffset)
- FATAL_ERROR("error: couldn't find .strtab section in \"%s\"\n", s_elfPath.c_str());
-}
-
-static std::map<std::string, std::uint32_t> GetCommonSymbols_Shared()
-{
- VerifyElfIdent();
- ReadElfHeader();
- FindTableOffsets();
-
- std::map<std::string, std::uint32_t> commonSymbols;
-
- std::vector<Symbol> commonSymbolVec;
-
- Seek(s_symtabOffset);
-
- for (std::uint32_t i = 0; i < s_symbolCount; i++)
- {
- Symbol sym;
- sym.nameOffset = ReadInt32();
- Skip(4);
- sym.size = ReadInt32();
- Skip(2);
- std::uint16_t sectionIndex = ReadInt16();
- if (sectionIndex == SHN_COMMON)
- commonSymbolVec.push_back(sym);
- }
-
- for (const Symbol& sym : commonSymbolVec)
- {
- Seek(s_strtabOffset + sym.nameOffset);
- std::string name = ReadString();
- commonSymbols[name] = sym.size;
- }
-
- return commonSymbols;
-}
-
-std::map<std::string, std::uint32_t> GetCommonSymbolsFromLib(std::string sourcePath, std::string libpath)
-{
- std::size_t colonPos = libpath.find(':');
- if (colonPos == std::string::npos)
- FATAL_ERROR("error: missing colon separator in libfile \"%s\"\n", s_elfPath.c_str());
-
- s_archiveObjectPath = libpath.substr(colonPos + 1);
- s_archiveFilePath = sourcePath + "/" + libpath.substr(1, colonPos - 1);
- s_elfPath = sourcePath + "/" + libpath.substr(1);
-
- s_file = std::fopen(s_archiveFilePath.c_str(), "rb");
-
- if (s_file == NULL)
- FATAL_ERROR("error: failed to open \"%s\" for reading\n", s_archiveFilePath.c_str());
-
- VerifyAr();
- FindArObj();
- return GetCommonSymbols_Shared();
-}
-
-std::map<std::string, std::uint32_t> GetCommonSymbols(std::string sourcePath, std::string path)
-{
- s_elfFileOffset = 0;
- if (path[0] == '*')
- return GetCommonSymbolsFromLib(sourcePath, path);
-
- s_elfPath = sourcePath + "/" + path;
- s_file = std::fopen(s_elfPath.c_str(), "rb");
-
- if (s_file == NULL)
- FATAL_ERROR("error: failed to open \"%s\" for reading\n", path.c_str());
-
- return GetCommonSymbols_Shared();
-}
+#include <cstdio> +#include <cstring> +#include <cstdint> +#include <map> +#include <vector> +#include <string> +#include "ramscrgen.h" +#include "elf.h" + +#define SHN_COMMON 0xFFF2 + +static std::string s_elfPath; +static std::string s_archiveFilePath; +static std::string s_archiveObjectPath; + +static FILE *s_file; + +static std::uint32_t s_sectionHeaderOffset; +static int s_sectionHeaderEntrySize; +static int s_sectionCount; +static int s_shstrtabIndex; + +static std::uint32_t s_symtabOffset; +static std::uint32_t s_strtabOffset; + +static std::uint32_t s_symbolCount; +static std::uint32_t s_elfFileOffset; + +struct Symbol +{ + std::uint32_t nameOffset; + std::uint32_t size; +}; + +static void Seek(long offset) +{ + if (std::fseek(s_file, s_elfFileOffset + offset, SEEK_SET) != 0) + FATAL_ERROR("error: failed to seek to %ld in \"%s\"", offset, s_elfPath.c_str()); +} + +static void Skip(long offset) +{ + if (std::fseek(s_file, offset, SEEK_CUR) != 0) + FATAL_ERROR("error: failed to skip %ld bytes in \"%s\"", offset, s_elfPath.c_str()); +} + +static std::uint32_t ReadInt8() +{ + int c = std::fgetc(s_file); + + if (c < 0) + FATAL_ERROR("error: unexpected EOF when reading ELF file \"%s\"\n", s_elfPath.c_str()); + + return c; +} + +static std::uint32_t ReadInt16() +{ + std::uint32_t val = 0; + val |= ReadInt8(); + val |= ReadInt8() << 8; + return val; +} + +static std::uint32_t ReadInt32() +{ + std::uint32_t val = 0; + val |= ReadInt8(); + val |= ReadInt8() << 8; + val |= ReadInt8() << 16; + val |= ReadInt8() << 24; + return val; +} + +static std::string ReadString() +{ + std::string s; + char c; + + while ((c = ReadInt8()) != 0) + s += c; + + return s; +} + +static void VerifyElfIdent() +{ + char expectedMagic[4] = { 0x7F, 'E', 'L', 'F' }; + char magic[4]; + + if (std::fread(magic, 4, 1, s_file) != 1) + FATAL_ERROR("error: failed to read ELF magic from \"%s\"\n", s_elfPath.c_str()); + + if (std::memcmp(magic, expectedMagic, 4) != 0) + FATAL_ERROR("error: ELF magic did not match in \"%s\"\n", s_elfPath.c_str()); + + if (std::fgetc(s_file) != 1) + FATAL_ERROR("error: \"%s\" not 32-bit ELF\n", s_elfPath.c_str()); + + if (std::fgetc(s_file) != 1) + FATAL_ERROR("error: \"%s\" not little-endian ELF\n", s_elfPath.c_str()); +} + +static void VerifyAr() +{ + char expectedMagic[8] = {'!', '<', 'a', 'r', 'c', 'h', '>', '\n'}; + char magic[8]; + + if (std::fread(magic, 8, 1, s_file) != 1) + FATAL_ERROR("error: failed to read AR magic from \"%s\"\n", s_archiveFilePath.c_str()); + + if (std::memcmp(magic, expectedMagic, 8) != 0) + FATAL_ERROR("error: AR magic did not match in \"%s\"\n", s_archiveFilePath.c_str()); +} + +static void ReadElfHeader() +{ + Seek(0x20); + s_sectionHeaderOffset = ReadInt32(); + Seek(0x2E); + s_sectionHeaderEntrySize = ReadInt16(); + s_sectionCount = ReadInt16(); + s_shstrtabIndex = ReadInt16(); +} + +static void FindArObj() +{ + char file_ident[17] = {0}; + char filesize_s[11] = {0}; + char expectedEndMagic[2] = { 0x60, 0x0a }; + char end_magic[2]; + std::size_t filesize; + + Seek(8); + while (!std::feof(s_file)) { + if (std::fread(file_ident, 16, 1, s_file) != 1) + FATAL_ERROR("error: failed to read file ident in \"%s\"\n", s_archiveFilePath.c_str()); + Skip(32); + if (std::fread(filesize_s, 10, 1, s_file) != 1) + FATAL_ERROR("error: failed to read filesize in \"%s\"\n", s_archiveFilePath.c_str()); + if (std::fread(end_magic, 2, 1, s_file) != 1) + FATAL_ERROR("error: failed to read end sentinel in \"%s\"\n", s_archiveFilePath.c_str()); + if (std::memcmp(end_magic, expectedEndMagic, 2) != 0) + FATAL_ERROR("error: corrupted archive header in \"%s\" at \"%s\"\n", s_archiveFilePath.c_str(), file_ident); + + char * ptr = std::strchr(file_ident, '/'); + if (ptr != nullptr) + *ptr = 0; + filesize = std::strtoul(filesize_s, nullptr, 10); + if (std::strncmp(s_archiveObjectPath.c_str(), file_ident, 16) == 0) { + s_elfFileOffset = std::ftell(s_file); + return; + } + Skip(filesize); + } + + FATAL_ERROR("error: could not find object \"%s\" in archive \"%s\"\n", s_archiveObjectPath.c_str(), s_archiveFilePath.c_str()); +} + +static std::string GetSectionName(std::uint32_t shstrtabOffset, int index) +{ + Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * index); + std::uint32_t nameOffset = ReadInt32(); + Seek(shstrtabOffset + nameOffset); + return ReadString(); +} + +static void FindTableOffsets() +{ + s_symtabOffset = 0; + s_strtabOffset = 0; + + Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * s_shstrtabIndex + 0x10); + std::uint32_t shstrtabOffset = ReadInt32(); + + for (int i = 0; i < s_sectionCount; i++) + { + std::string name = GetSectionName(shstrtabOffset, i); + + if (name == ".symtab") + { + if (s_symtabOffset) + FATAL_ERROR("error: mutiple .symtab sections found in \"%s\"\n", s_elfPath.c_str()); + Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * i + 0x10); + s_symtabOffset = ReadInt32(); + std::uint32_t size = ReadInt32(); + s_symbolCount = size / 16; + } + else if (name == ".strtab") + { + if (s_strtabOffset) + FATAL_ERROR("error: mutiple .strtab sections found in \"%s\"\n", s_elfPath.c_str()); + Seek(s_sectionHeaderOffset + s_sectionHeaderEntrySize * i + 0x10); + s_strtabOffset = ReadInt32(); + } + } + + if (!s_symtabOffset) + FATAL_ERROR("error: couldn't find .symtab section in \"%s\"\n", s_elfPath.c_str()); + + if (!s_strtabOffset) + FATAL_ERROR("error: couldn't find .strtab section in \"%s\"\n", s_elfPath.c_str()); +} + +static std::map<std::string, std::uint32_t> GetCommonSymbols_Shared() +{ + VerifyElfIdent(); + ReadElfHeader(); + FindTableOffsets(); + + std::map<std::string, std::uint32_t> commonSymbols; + + std::vector<Symbol> commonSymbolVec; + + Seek(s_symtabOffset); + + for (std::uint32_t i = 0; i < s_symbolCount; i++) + { + Symbol sym; + sym.nameOffset = ReadInt32(); + Skip(4); + sym.size = ReadInt32(); + Skip(2); + std::uint16_t sectionIndex = ReadInt16(); + if (sectionIndex == SHN_COMMON) + commonSymbolVec.push_back(sym); + } + + for (const Symbol& sym : commonSymbolVec) + { + Seek(s_strtabOffset + sym.nameOffset); + std::string name = ReadString(); + commonSymbols[name] = sym.size; + } + + return commonSymbols; +} + +std::map<std::string, std::uint32_t> GetCommonSymbolsFromLib(std::string sourcePath, std::string libpath) +{ + std::size_t colonPos = libpath.find(':'); + if (colonPos == std::string::npos) + FATAL_ERROR("error: missing colon separator in libfile \"%s\"\n", s_elfPath.c_str()); + + s_archiveObjectPath = libpath.substr(colonPos + 1); + s_archiveFilePath = sourcePath + "/" + libpath.substr(1, colonPos - 1); + s_elfPath = sourcePath + "/" + libpath.substr(1); + + s_file = std::fopen(s_archiveFilePath.c_str(), "rb"); + + if (s_file == NULL) + FATAL_ERROR("error: failed to open \"%s\" for reading\n", s_archiveFilePath.c_str()); + + VerifyAr(); + FindArObj(); + return GetCommonSymbols_Shared(); +} + +std::map<std::string, std::uint32_t> GetCommonSymbols(std::string sourcePath, std::string path) +{ + s_elfFileOffset = 0; + if (path[0] == '*') + return GetCommonSymbolsFromLib(sourcePath, path); + + s_elfPath = sourcePath + "/" + path; + s_file = std::fopen(s_elfPath.c_str(), "rb"); + + if (s_file == NULL) + FATAL_ERROR("error: failed to open \"%s\" for reading\n", path.c_str()); + + return GetCommonSymbols_Shared(); +} diff --git a/tools/ramscrgen/main.cpp b/tools/ramscrgen/main.cpp index faeba46..5e5894f 100644 --- a/tools/ramscrgen/main.cpp +++ b/tools/ramscrgen/main.cpp @@ -1,187 +1,187 @@ -// Copyright(c) 2016 YamaArashi
-//
-// 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 <cstdio>
-#include <cstring>
-#include <string>
-#include "ramscrgen.h"
-#include "sym_file.h"
-#include "elf.h"
-
-void HandleCommonInclude(std::string filename, std::string sourcePath, std::string symOrderPath, std::string lang)
-{
- auto commonSymbols = GetCommonSymbols(sourcePath, filename);
- std::size_t dotIndex;
-
- if (filename[0] == '*') {
- dotIndex = filename.find_last_of(':');
- filename = filename.substr(dotIndex + 1);
- }
-
- dotIndex = filename.find_last_of('.');
-
- if (dotIndex == std::string::npos)
- FATAL_ERROR("error: \"%s\" doesn't have a file extension\n", filename.c_str());
-
- std::string symOrderFilename = filename.substr(0, dotIndex + 1) + "txt";
-
- SymFile symFile(symOrderPath + "/" + symOrderFilename);
-
- while (!symFile.IsAtEnd())
- {
- symFile.HandleLangConditional(lang);
-
- std::string label = symFile.GetLabel(false);
-
- if (label.length() == 0)
- {
- unsigned long length;
- if (symFile.ReadInteger(length))
- {
- if (length & 3)
- symFile.RaiseWarning("gap length %d is not multiple of 4", length);
- printf(". += 0x%lX;\n", length);
- }
- }
- else
- {
- if (commonSymbols.count(label) == 0)
- symFile.RaiseError("no common symbol named \"%s\"", label.c_str());
- unsigned long size = commonSymbols[label];
- int alignment = 4;
- if (size > 4)
- alignment = 8;
- if (size > 8)
- alignment = 16;
- printf(". = ALIGN(%d);\n", alignment);
- printf("%s = .;\n", label.c_str());
- printf(". += 0x%lX;\n", size);
- }
-
- symFile.ExpectEmptyRestOfLine();
- }
-}
-
-void ConvertSymFile(std::string filename, std::string sectionName, std::string lang, bool common, std::string sourcePath, std::string commonSymPath, std::string libSourcePath)
-{
- SymFile symFile(filename);
-
- while (!symFile.IsAtEnd())
- {
- symFile.HandleLangConditional(lang);
-
- Directive directive = symFile.GetDirective();
-
- switch (directive)
- {
- case Directive::Include:
- {
- std::string incFilename = symFile.ReadPath();
- symFile.ExpectEmptyRestOfLine();
- printf(". = ALIGN(4);\n");
- if (common)
- HandleCommonInclude(incFilename, incFilename[0] == '*' ? libSourcePath : sourcePath, commonSymPath, lang);
- else
- printf("%s(%s);\n", incFilename.c_str(), sectionName.c_str());
- break;
- }
- case Directive::Space:
- {
- unsigned long length;
- if (!symFile.ReadInteger(length))
- symFile.RaiseError("expected integer after .space directive");
- symFile.ExpectEmptyRestOfLine();
- printf(". += 0x%lX;\n", length);
- break;
- }
- case Directive::Align:
- {
- unsigned long amount;
- if (!symFile.ReadInteger(amount))
- symFile.RaiseError("expected integer after .align directive");
- if (amount > 4)
- symFile.RaiseError("max alignment amount is 4");
- amount = 1UL << amount;
- symFile.ExpectEmptyRestOfLine();
- printf(". = ALIGN(%lu);\n", amount);
- break;
- }
- case Directive::Unknown:
- {
- std::string label = symFile.GetLabel();
-
- if (label.length() != 0)
- {
- printf("%s = .;\n", label.c_str());
- }
-
- symFile.ExpectEmptyRestOfLine();
-
- break;
- }
- }
- }
-}
-
-int main(int argc, char **argv)
-{
- if (argc < 4)
- {
- fprintf(stderr, "Usage: %s SECTION_NAME SYM_FILE LANG [-c SRC_PATH,COMMON_SYM_PATH]", argv[0]);
- return 1;
- }
-
- bool common = false;
- std::string sectionName = std::string(argv[1]);
- std::string symFileName = std::string(argv[2]);
- std::string lang = std::string(argv[3]);
- std::string sourcePath;
- std::string commonSymPath;
- std::string libSourcePath;
-
- if (argc > 4)
- {
- if (std::strcmp(argv[4], "-c") != 0)
- FATAL_ERROR("error: unrecognized argument \"%s\"\n", argv[4]);
-
- if (argc < 6)
- FATAL_ERROR("error: missing SRC_PATH,COMMON_SYM_PATH after \"-c\"\n");
-
- common = true;
- std::string paths = std::string(argv[5]);
- std::size_t commaPos = paths.find(',');
-
- if (commaPos == std::string::npos)
- FATAL_ERROR("error: missing comma in argument after \"-c\"\n");
-
- sourcePath = paths.substr(0, commaPos);
- commonSymPath = paths.substr(commaPos + 1);
- commaPos = commonSymPath.find(',');
- if (commaPos == std::string::npos) {
- libSourcePath = "tools/agbcc/lib";
- } else {
- libSourcePath = commonSymPath.substr(commaPos + 1);
- commonSymPath = commonSymPath.substr(0, commaPos);
- }
- }
-
- ConvertSymFile(symFileName, sectionName, lang, common, sourcePath, commonSymPath, libSourcePath);
- return 0;
-}
+// Copyright(c) 2016 YamaArashi +// +// 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 <cstdio> +#include <cstring> +#include <string> +#include "ramscrgen.h" +#include "sym_file.h" +#include "elf.h" + +void HandleCommonInclude(std::string filename, std::string sourcePath, std::string symOrderPath, std::string lang) +{ + auto commonSymbols = GetCommonSymbols(sourcePath, filename); + std::size_t dotIndex; + + if (filename[0] == '*') { + dotIndex = filename.find_last_of(':'); + filename = filename.substr(dotIndex + 1); + } + + dotIndex = filename.find_last_of('.'); + + if (dotIndex == std::string::npos) + FATAL_ERROR("error: \"%s\" doesn't have a file extension\n", filename.c_str()); + + std::string symOrderFilename = filename.substr(0, dotIndex + 1) + "txt"; + + SymFile symFile(symOrderPath + "/" + symOrderFilename); + + while (!symFile.IsAtEnd()) + { + symFile.HandleLangConditional(lang); + + std::string label = symFile.GetLabel(false); + + if (label.length() == 0) + { + unsigned long length; + if (symFile.ReadInteger(length)) + { + if (length & 3) + symFile.RaiseWarning("gap length %d is not multiple of 4", length); + printf(". += 0x%lX;\n", length); + } + } + else + { + if (commonSymbols.count(label) == 0) + symFile.RaiseError("no common symbol named \"%s\"", label.c_str()); + unsigned long size = commonSymbols[label]; + int alignment = 4; + if (size > 4) + alignment = 8; + if (size > 8) + alignment = 16; + printf(". = ALIGN(%d);\n", alignment); + printf("%s = .;\n", label.c_str()); + printf(". += 0x%lX;\n", size); + } + + symFile.ExpectEmptyRestOfLine(); + } +} + +void ConvertSymFile(std::string filename, std::string sectionName, std::string lang, bool common, std::string sourcePath, std::string commonSymPath, std::string libSourcePath) +{ + SymFile symFile(filename); + + while (!symFile.IsAtEnd()) + { + symFile.HandleLangConditional(lang); + + Directive directive = symFile.GetDirective(); + + switch (directive) + { + case Directive::Include: + { + std::string incFilename = symFile.ReadPath(); + symFile.ExpectEmptyRestOfLine(); + printf(". = ALIGN(4);\n"); + if (common) + HandleCommonInclude(incFilename, incFilename[0] == '*' ? libSourcePath : sourcePath, commonSymPath, lang); + else + printf("%s(%s);\n", incFilename.c_str(), sectionName.c_str()); + break; + } + case Directive::Space: + { + unsigned long length; + if (!symFile.ReadInteger(length)) + symFile.RaiseError("expected integer after .space directive"); + symFile.ExpectEmptyRestOfLine(); + printf(". += 0x%lX;\n", length); + break; + } + case Directive::Align: + { + unsigned long amount; + if (!symFile.ReadInteger(amount)) + symFile.RaiseError("expected integer after .align directive"); + if (amount > 4) + symFile.RaiseError("max alignment amount is 4"); + amount = 1UL << amount; + symFile.ExpectEmptyRestOfLine(); + printf(". = ALIGN(%lu);\n", amount); + break; + } + case Directive::Unknown: + { + std::string label = symFile.GetLabel(); + + if (label.length() != 0) + { + printf("%s = .;\n", label.c_str()); + } + + symFile.ExpectEmptyRestOfLine(); + + break; + } + } + } +} + +int main(int argc, char **argv) +{ + if (argc < 4) + { + fprintf(stderr, "Usage: %s SECTION_NAME SYM_FILE LANG [-c SRC_PATH,COMMON_SYM_PATH]", argv[0]); + return 1; + } + + bool common = false; + std::string sectionName = std::string(argv[1]); + std::string symFileName = std::string(argv[2]); + std::string lang = std::string(argv[3]); + std::string sourcePath; + std::string commonSymPath; + std::string libSourcePath; + + if (argc > 4) + { + if (std::strcmp(argv[4], "-c") != 0) + FATAL_ERROR("error: unrecognized argument \"%s\"\n", argv[4]); + + if (argc < 6) + FATAL_ERROR("error: missing SRC_PATH,COMMON_SYM_PATH after \"-c\"\n"); + + common = true; + std::string paths = std::string(argv[5]); + std::size_t commaPos = paths.find(','); + + if (commaPos == std::string::npos) + FATAL_ERROR("error: missing comma in argument after \"-c\"\n"); + + sourcePath = paths.substr(0, commaPos); + commonSymPath = paths.substr(commaPos + 1); + commaPos = commonSymPath.find(','); + if (commaPos == std::string::npos) { + libSourcePath = "tools/agbcc/lib"; + } else { + libSourcePath = commonSymPath.substr(commaPos + 1); + commonSymPath = commonSymPath.substr(0, commaPos); + } + } + + ConvertSymFile(symFileName, sectionName, lang, common, sourcePath, commonSymPath, libSourcePath); + return 0; +} diff --git a/tools/ramscrgen/sym_file.cpp b/tools/ramscrgen/sym_file.cpp index 1272ef1..5379bd9 100644 --- a/tools/ramscrgen/sym_file.cpp +++ b/tools/ramscrgen/sym_file.cpp @@ -1,492 +1,492 @@ -// Copyright(c) 2016 YamaArashi
-//
-// 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 <cstdio>
-#include <cstdarg>
-#include <climits>
-#include "ramscrgen.h"
-#include "sym_file.h"
-#include "char_util.h"
-
-SymFile::SymFile(std::string filename) : m_filename(filename)
-{
- FILE *fp = std::fopen(filename.c_str(), "rb");
-
- if (fp == NULL)
- FATAL_ERROR("Failed to open \"%s\" for reading.\n", filename.c_str());
-
- std::fseek(fp, 0, SEEK_END);
-
- m_size = std::ftell(fp);
-
- if (m_size < 0)
- FATAL_ERROR("File size of \"%s\" is less than zero.\n", filename.c_str());
-
- m_buffer = new char[m_size + 1];
-
- std::rewind(fp);
-
- if (std::fread(m_buffer, m_size, 1, fp) != 1)
- FATAL_ERROR("Failed to read \"%s\".\n", filename.c_str());
-
- m_buffer[m_size] = 0;
-
- std::fclose(fp);
-
- m_pos = 0;
- m_lineNum = 1;
- m_lineStart = 0;
- m_inLangConditional = false;
-
- RemoveComments();
-}
-
-SymFile::SymFile(SymFile&& other) : m_filename(std::move(other.m_filename))
-{
- m_buffer = other.m_buffer;
- m_pos = other.m_pos;
- m_size = other.m_size;
- m_lineNum = other.m_lineNum;
- m_lineStart = other.m_lineStart;
-
- other.m_buffer = nullptr;
-}
-
-SymFile::~SymFile()
-{
- delete[] m_buffer;
-}
-
-// Removes comments to simplify further processing.
-// It stops upon encountering a null character,
-// which may or may not be the end of file marker.
-// If it's not, the error will be caught later.
-void SymFile::RemoveComments()
-{
- long pos = 0;
- char stringChar = 0;
-
- for (;;)
- {
- if (m_buffer[pos] == 0)
- return;
-
- if (stringChar != 0)
- {
- if (m_buffer[pos] == '\\' && m_buffer[pos + 1] == stringChar)
- {
- pos += 2;
- }
- else
- {
- if (m_buffer[pos] == stringChar)
- stringChar = 0;
- pos++;
- }
- }
- else if (m_buffer[pos] == '@' && (pos == 0 || m_buffer[pos - 1] != '\\'))
- {
- while (m_buffer[pos] != '\n' && m_buffer[pos] != 0)
- m_buffer[pos++] = ' ';
- }
- else if (m_buffer[pos] == '/' && m_buffer[pos + 1] == '*')
- {
- m_buffer[pos++] = ' ';
- m_buffer[pos++] = ' ';
-
- char commentStringChar = 0;
-
- for (;;)
- {
- if (m_buffer[pos] == 0)
- return;
-
- if (commentStringChar != 0)
- {
- if (m_buffer[pos] == '\\' && m_buffer[pos + 1] == commentStringChar)
- {
- m_buffer[pos++] = ' ';
- m_buffer[pos++] = ' ';
- }
- else
- {
- if (m_buffer[pos] == commentStringChar)
- commentStringChar = 0;
- if (m_buffer[pos] != '\n')
- m_buffer[pos] = ' ';
- pos++;
- }
- }
- else
- {
- if (m_buffer[pos] == '*' && m_buffer[pos + 1] == '/')
- {
- m_buffer[pos++] = ' ';
- m_buffer[pos++] = ' ';
- break;
- }
- else
- {
- if (m_buffer[pos] == '"' || m_buffer[pos] == '\'')
- commentStringChar = m_buffer[pos];
- if (m_buffer[pos] != '\n')
- m_buffer[pos] = ' ';
- pos++;
- }
- }
- }
- }
- else
- {
- if (m_buffer[pos] == '"' || m_buffer[pos] == '\'')
- stringChar = m_buffer[pos];
- pos++;
- }
- }
-}
-
-// Checks if we're at a particular directive and if so, consumes it.
-// Returns whether the directive was found.
-bool SymFile::CheckForDirective(std::string name)
-{
- long i;
- long length = static_cast<long>(name.length());
-
- for (i = 0; i < length && m_pos + i < m_size; i++)
- if (name[i] != m_buffer[m_pos + i])
- return false;
-
- if (i < length)
- return false;
-
- m_pos += length;
-
- return true;
-}
-
-// Checks if we're at a known directive and if so, consumes it.
-// Returns which directive was found.
-Directive SymFile::GetDirective()
-{
- SkipWhitespace();
-
- if (CheckForDirective(".include"))
- return Directive::Include;
- else if (CheckForDirective(".space"))
- return Directive::Space;
- else if (CheckForDirective(".align"))
- return Directive::Align;
- else
- return Directive::Unknown;
-}
-
-// Checks if we're at label.
-// Returns the name if so and an empty string if not.
-std::string SymFile::GetLabel(bool requireColon)
-{
- long start = m_pos;
- long pos = m_pos;
-
- if (IsIdentifierStartingChar(m_buffer[pos]))
- {
- pos++;
-
- while (IsIdentifierChar(m_buffer[pos]))
- pos++;
- }
-
- if (requireColon)
- {
- if (m_buffer[pos] == ':')
- {
- if (pos != start)
- m_pos = pos + 1;
- }
- else
- {
- pos = start;
- }
- }
- else
- {
- m_pos = pos;
- }
-
- return std::string(&m_buffer[start], pos - start);
-}
-
-// Skips tabs and spaces.
-void SymFile::SkipWhitespace()
-{
- while (m_buffer[m_pos] == '\t' || m_buffer[m_pos] == ' ')
- m_pos++;
-}
-
-// Reads include path.
-std::string SymFile::ReadPath()
-{
- SkipWhitespace();
-
- if (m_buffer[m_pos] != '"')
- RaiseError("expected file path");
-
- m_pos++;
-
- int length = 0;
- long startPos = m_pos;
-
- while (m_buffer[m_pos] != '"')
- {
- unsigned char c = m_buffer[m_pos++];
-
- if (c == 0)
- {
- if (m_pos >= m_size)
- RaiseError("unexpected EOF in include string");
- else
- RaiseError("unexpected null character in include string");
- }
-
- if (!IsAsciiPrintable(c))
- RaiseError("unexpected character '\\x%02X' in include string", c);
-
- // Don't bother allowing any escape sequences.
- if (c == '\\')
- {
- c = m_buffer[m_pos];
- RaiseError("unexpected escape '\\%c' in include string", c);
- }
-
- length++;
-
- if (length > kMaxPath)
- RaiseError("path is too long");
- }
-
- m_pos++; // Go past the right quote.
-
- return std::string(&m_buffer[startPos], length);
-}
-
-// If we're at a comma, consumes it.
-// Returns whether a comma was found.
-bool SymFile::ConsumeComma()
-{
- if (m_buffer[m_pos] == ',')
- {
- m_pos++;
- return true;
- }
-
- return false;
-}
-
-// Converts digit character to numerical value.
-static int ConvertDigit(char c, int radix)
-{
- int digit;
-
- if (c >= '0' && c <= '9')
- digit = c - '0';
- else if (c >= 'A' && c <= 'F')
- digit = 10 + c - 'A';
- else if (c >= 'a' && c <= 'f')
- digit = 10 + c - 'a';
- else
- return -1;
-
- return (digit < radix) ? digit : -1;
-}
-
-// Reads an integer.
-bool SymFile::ReadInteger(unsigned long& n)
-{
- SkipWhitespace();
-
- if (!IsAsciiDigit(m_buffer[m_pos]))
- return false;
-
- int startPos = m_pos;
- int radix = 10;
-
- if (m_buffer[m_pos] == '0' && m_buffer[m_pos + 1] == 'x')
- {
- radix = 16;
- m_pos += 2;
- }
-
- unsigned long cutoff = ULONG_MAX / radix;
- unsigned long cutoffRemainder = ULONG_MAX % radix;
- int digit;
-
- n = 0;
-
- while ((digit = ConvertDigit(m_buffer[m_pos], radix)) != -1)
- {
- if (n < cutoff || (n == cutoff && (unsigned long)digit <= cutoffRemainder))
- {
- n = n * radix + digit;
- }
- else
- {
- m_pos++;
-
- while (ConvertDigit(m_buffer[m_pos], radix) != -1)
- m_pos++;
-
- RaiseError("integer is too large (%s)", std::string(&m_buffer[startPos], m_pos - startPos).c_str());
- }
-
- m_pos++;
- }
-
- return true;
-}
-
-// Asserts that the rest of the line is empty and moves to the next one.
-void SymFile::ExpectEmptyRestOfLine()
-{
- SkipWhitespace();
-
- if (m_buffer[m_pos] == 0)
- {
- if (m_pos >= m_size)
- RaiseWarning("file doesn't end with newline");
- else
- RaiseError("unexpected null character");
- }
- else if (m_buffer[m_pos] == '\n')
- {
- m_pos++;
- m_lineStart = m_pos;
- m_lineNum++;
- }
- else if (m_buffer[m_pos] == '\r')
- {
- RaiseError("only Unix-style LF newlines are supported");
- }
- else
- {
- RaiseError("junk at end of line");
- }
-}
-
-
-void SymFile::SkipLine()
-{
- while (m_buffer[m_pos] != 0 && m_buffer[m_pos] != '\n')
- m_pos++;
-
- if (m_buffer[m_pos] == '\n')
- m_pos++;
-}
-
-// Checks if we're at the end of the file.
-bool SymFile::IsAtEnd()
-{
- return (m_pos >= m_size);
-}
-
-void SymFile::HandleLangConditional(std::string lang)
-{
- if (m_buffer[m_pos] != '#')
- return;
-
- m_pos++;
-
- if (CheckForDirective("begin"))
- {
- if (m_inLangConditional)
- RaiseError("already inside language conditional");
-
- SkipWhitespace();
-
- std::string label = GetLabel(false);
-
- if (label.length() == 0)
- RaiseError("no language name after #begin");
-
- ExpectEmptyRestOfLine();
-
- if (lang == label)
- {
- m_inLangConditional = true;
- }
- else
- {
- while (!IsAtEnd() && m_buffer[m_pos] != '#')
- SkipLine();
-
- if (m_buffer[m_pos] != '#')
- RaiseError("unterminated language conditional");
-
- m_pos++;
-
- if (!CheckForDirective("end"))
- RaiseError("expected #end");
-
- ExpectEmptyRestOfLine();
- }
- }
- else if (CheckForDirective("end"))
- {
- if (!m_inLangConditional)
- RaiseError("not inside language conditional");
-
- m_inLangConditional = false;
-
- ExpectEmptyRestOfLine();
- }
- else
- {
- RaiseError("unknown # directive");
- }
-}
-
-// Reports a diagnostic message.
-void SymFile::ReportDiagnostic(const char* type, const char* format, std::va_list args)
-{
- const int bufferSize = 1024;
- char buffer[bufferSize];
- std::vsnprintf(buffer, bufferSize, format, args);
- std::fprintf(stderr, "%s:%ld: %s: %s\n", m_filename.c_str(), m_lineNum, type, buffer);
-}
-
-#define DO_REPORT(type) \
-do \
-{ \
- std::va_list args; \
- va_start(args, format); \
- ReportDiagnostic(type, format, args); \
- va_end(args); \
-} while (0)
-
-// Reports an error diagnostic and terminates the program.
-void SymFile::RaiseError(const char* format, ...)
-{
- DO_REPORT("error");
- std::exit(1);
-}
-
-// Reports a warning diagnostic.
-void SymFile::RaiseWarning(const char* format, ...)
-{
- DO_REPORT("warning");
-}
+// Copyright(c) 2016 YamaArashi +// +// 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 <cstdio> +#include <cstdarg> +#include <climits> +#include "ramscrgen.h" +#include "sym_file.h" +#include "char_util.h" + +SymFile::SymFile(std::string filename) : m_filename(filename) +{ + FILE *fp = std::fopen(filename.c_str(), "rb"); + + if (fp == NULL) + FATAL_ERROR("Failed to open \"%s\" for reading.\n", filename.c_str()); + + std::fseek(fp, 0, SEEK_END); + + m_size = std::ftell(fp); + + if (m_size < 0) + FATAL_ERROR("File size of \"%s\" is less than zero.\n", filename.c_str()); + + m_buffer = new char[m_size + 1]; + + std::rewind(fp); + + if (std::fread(m_buffer, m_size, 1, fp) != 1) + FATAL_ERROR("Failed to read \"%s\".\n", filename.c_str()); + + m_buffer[m_size] = 0; + + std::fclose(fp); + + m_pos = 0; + m_lineNum = 1; + m_lineStart = 0; + m_inLangConditional = false; + + RemoveComments(); +} + +SymFile::SymFile(SymFile&& other) : m_filename(std::move(other.m_filename)) +{ + m_buffer = other.m_buffer; + m_pos = other.m_pos; + m_size = other.m_size; + m_lineNum = other.m_lineNum; + m_lineStart = other.m_lineStart; + + other.m_buffer = nullptr; +} + +SymFile::~SymFile() +{ + delete[] m_buffer; +} + +// Removes comments to simplify further processing. +// It stops upon encountering a null character, +// which may or may not be the end of file marker. +// If it's not, the error will be caught later. +void SymFile::RemoveComments() +{ + long pos = 0; + char stringChar = 0; + + for (;;) + { + if (m_buffer[pos] == 0) + return; + + if (stringChar != 0) + { + if (m_buffer[pos] == '\\' && m_buffer[pos + 1] == stringChar) + { + pos += 2; + } + else + { + if (m_buffer[pos] == stringChar) + stringChar = 0; + pos++; + } + } + else if (m_buffer[pos] == '@' && (pos == 0 || m_buffer[pos - 1] != '\\')) + { + while (m_buffer[pos] != '\n' && m_buffer[pos] != 0) + m_buffer[pos++] = ' '; + } + else if (m_buffer[pos] == '/' && m_buffer[pos + 1] == '*') + { + m_buffer[pos++] = ' '; + m_buffer[pos++] = ' '; + + char commentStringChar = 0; + + for (;;) + { + if (m_buffer[pos] == 0) + return; + + if (commentStringChar != 0) + { + if (m_buffer[pos] == '\\' && m_buffer[pos + 1] == commentStringChar) + { + m_buffer[pos++] = ' '; + m_buffer[pos++] = ' '; + } + else + { + if (m_buffer[pos] == commentStringChar) + commentStringChar = 0; + if (m_buffer[pos] != '\n') + m_buffer[pos] = ' '; + pos++; + } + } + else + { + if (m_buffer[pos] == '*' && m_buffer[pos + 1] == '/') + { + m_buffer[pos++] = ' '; + m_buffer[pos++] = ' '; + break; + } + else + { + if (m_buffer[pos] == '"' || m_buffer[pos] == '\'') + commentStringChar = m_buffer[pos]; + if (m_buffer[pos] != '\n') + m_buffer[pos] = ' '; + pos++; + } + } + } + } + else + { + if (m_buffer[pos] == '"' || m_buffer[pos] == '\'') + stringChar = m_buffer[pos]; + pos++; + } + } +} + +// Checks if we're at a particular directive and if so, consumes it. +// Returns whether the directive was found. +bool SymFile::CheckForDirective(std::string name) +{ + long i; + long length = static_cast<long>(name.length()); + + for (i = 0; i < length && m_pos + i < m_size; i++) + if (name[i] != m_buffer[m_pos + i]) + return false; + + if (i < length) + return false; + + m_pos += length; + + return true; +} + +// Checks if we're at a known directive and if so, consumes it. +// Returns which directive was found. +Directive SymFile::GetDirective() +{ + SkipWhitespace(); + + if (CheckForDirective(".include")) + return Directive::Include; + else if (CheckForDirective(".space")) + return Directive::Space; + else if (CheckForDirective(".align")) + return Directive::Align; + else + return Directive::Unknown; +} + +// Checks if we're at label. +// Returns the name if so and an empty string if not. +std::string SymFile::GetLabel(bool requireColon) +{ + long start = m_pos; + long pos = m_pos; + + if (IsIdentifierStartingChar(m_buffer[pos])) + { + pos++; + + while (IsIdentifierChar(m_buffer[pos])) + pos++; + } + + if (requireColon) + { + if (m_buffer[pos] == ':') + { + if (pos != start) + m_pos = pos + 1; + } + else + { + pos = start; + } + } + else + { + m_pos = pos; + } + + return std::string(&m_buffer[start], pos - start); +} + +// Skips tabs and spaces. +void SymFile::SkipWhitespace() +{ + while (m_buffer[m_pos] == '\t' || m_buffer[m_pos] == ' ') + m_pos++; +} + +// Reads include path. +std::string SymFile::ReadPath() +{ + SkipWhitespace(); + + if (m_buffer[m_pos] != '"') + RaiseError("expected file path"); + + m_pos++; + + int length = 0; + long startPos = m_pos; + + while (m_buffer[m_pos] != '"') + { + unsigned char c = m_buffer[m_pos++]; + + if (c == 0) + { + if (m_pos >= m_size) + RaiseError("unexpected EOF in include string"); + else + RaiseError("unexpected null character in include string"); + } + + if (!IsAsciiPrintable(c)) + RaiseError("unexpected character '\\x%02X' in include string", c); + + // Don't bother allowing any escape sequences. + if (c == '\\') + { + c = m_buffer[m_pos]; + RaiseError("unexpected escape '\\%c' in include string", c); + } + + length++; + + if (length > kMaxPath) + RaiseError("path is too long"); + } + + m_pos++; // Go past the right quote. + + return std::string(&m_buffer[startPos], length); +} + +// If we're at a comma, consumes it. +// Returns whether a comma was found. +bool SymFile::ConsumeComma() +{ + if (m_buffer[m_pos] == ',') + { + m_pos++; + return true; + } + + return false; +} + +// Converts digit character to numerical value. +static int ConvertDigit(char c, int radix) +{ + int digit; + + if (c >= '0' && c <= '9') + digit = c - '0'; + else if (c >= 'A' && c <= 'F') + digit = 10 + c - 'A'; + else if (c >= 'a' && c <= 'f') + digit = 10 + c - 'a'; + else + return -1; + + return (digit < radix) ? digit : -1; +} + +// Reads an integer. +bool SymFile::ReadInteger(unsigned long& n) +{ + SkipWhitespace(); + + if (!IsAsciiDigit(m_buffer[m_pos])) + return false; + + int startPos = m_pos; + int radix = 10; + + if (m_buffer[m_pos] == '0' && m_buffer[m_pos + 1] == 'x') + { + radix = 16; + m_pos += 2; + } + + unsigned long cutoff = ULONG_MAX / radix; + unsigned long cutoffRemainder = ULONG_MAX % radix; + int digit; + + n = 0; + + while ((digit = ConvertDigit(m_buffer[m_pos], radix)) != -1) + { + if (n < cutoff || (n == cutoff && (unsigned long)digit <= cutoffRemainder)) + { + n = n * radix + digit; + } + else + { + m_pos++; + + while (ConvertDigit(m_buffer[m_pos], radix) != -1) + m_pos++; + + RaiseError("integer is too large (%s)", std::string(&m_buffer[startPos], m_pos - startPos).c_str()); + } + + m_pos++; + } + + return true; +} + +// Asserts that the rest of the line is empty and moves to the next one. +void SymFile::ExpectEmptyRestOfLine() +{ + SkipWhitespace(); + + if (m_buffer[m_pos] == 0) + { + if (m_pos >= m_size) + RaiseWarning("file doesn't end with newline"); + else + RaiseError("unexpected null character"); + } + else if (m_buffer[m_pos] == '\n') + { + m_pos++; + m_lineStart = m_pos; + m_lineNum++; + } + else if (m_buffer[m_pos] == '\r') + { + RaiseError("only Unix-style LF newlines are supported"); + } + else + { + RaiseError("junk at end of line"); + } +} + + +void SymFile::SkipLine() +{ + while (m_buffer[m_pos] != 0 && m_buffer[m_pos] != '\n') + m_pos++; + + if (m_buffer[m_pos] == '\n') + m_pos++; +} + +// Checks if we're at the end of the file. +bool SymFile::IsAtEnd() +{ + return (m_pos >= m_size); +} + +void SymFile::HandleLangConditional(std::string lang) +{ + if (m_buffer[m_pos] != '#') + return; + + m_pos++; + + if (CheckForDirective("begin")) + { + if (m_inLangConditional) + RaiseError("already inside language conditional"); + + SkipWhitespace(); + + std::string label = GetLabel(false); + + if (label.length() == 0) + RaiseError("no language name after #begin"); + + ExpectEmptyRestOfLine(); + + if (lang == label) + { + m_inLangConditional = true; + } + else + { + while (!IsAtEnd() && m_buffer[m_pos] != '#') + SkipLine(); + + if (m_buffer[m_pos] != '#') + RaiseError("unterminated language conditional"); + + m_pos++; + + if (!CheckForDirective("end")) + RaiseError("expected #end"); + + ExpectEmptyRestOfLine(); + } + } + else if (CheckForDirective("end")) + { + if (!m_inLangConditional) + RaiseError("not inside language conditional"); + + m_inLangConditional = false; + + ExpectEmptyRestOfLine(); + } + else + { + RaiseError("unknown # directive"); + } +} + +// Reports a diagnostic message. +void SymFile::ReportDiagnostic(const char* type, const char* format, std::va_list args) +{ + const int bufferSize = 1024; + char buffer[bufferSize]; + std::vsnprintf(buffer, bufferSize, format, args); + std::fprintf(stderr, "%s:%ld: %s: %s\n", m_filename.c_str(), m_lineNum, type, buffer); +} + +#define DO_REPORT(type) \ +do \ +{ \ + std::va_list args; \ + va_start(args, format); \ + ReportDiagnostic(type, format, args); \ + va_end(args); \ +} while (0) + +// Reports an error diagnostic and terminates the program. +void SymFile::RaiseError(const char* format, ...) +{ + DO_REPORT("error"); + std::exit(1); +} + +// Reports a warning diagnostic. +void SymFile::RaiseWarning(const char* format, ...) +{ + DO_REPORT("warning"); +} |