Compare commits
33 Commits
Author | SHA1 | Date | |
---|---|---|---|
fd720d2321 | |||
818b7535eb | |||
3db1515ecf | |||
4ec30a8ec8 | |||
01a64f321d | |||
16089ef4b2 | |||
474106c519 | |||
4d64fd2190 | |||
8dc7b17251 | |||
27c4f4115b | |||
fa7446151e | |||
16ab0983a7 | |||
c803572e24 | |||
308a36b4ba | |||
f72c4f0567 | |||
5d2b146b08 | |||
2a63f1ff52 | |||
199eecda97 | |||
10026e9a04 | |||
b0814f257a | |||
6f792dfb16 | |||
46c486e56a | |||
346d4403df | |||
11342dbb78 | |||
be25749aa4 | |||
6098f57b03 | |||
8249a305df | |||
eb49f25fa4 | |||
fbb9b6d6da | |||
603010e3cc | |||
eca52455c3 | |||
9697b35de4 | |||
853481bd9a |
@ -1,2 +0,0 @@
|
||||
/build
|
||||
/node_modules
|
36
.eslintrc
36
.eslintrc
@ -1,36 +0,0 @@
|
||||
{
|
||||
"root": true,
|
||||
"parser": "@typescript-eslint/parser",
|
||||
"plugins": [
|
||||
"@typescript-eslint"
|
||||
],
|
||||
"extends": [
|
||||
"eslint:recommended",
|
||||
"plugin:@typescript-eslint/eslint-recommended",
|
||||
"plugin:@typescript-eslint/recommended"
|
||||
],
|
||||
"rules": {
|
||||
"no-constant-condition": [
|
||||
"error",
|
||||
{
|
||||
"checkLoops": false
|
||||
}
|
||||
],
|
||||
"quotes": [
|
||||
"error",
|
||||
"single",
|
||||
{
|
||||
"allowTemplateLiterals": true
|
||||
}
|
||||
],
|
||||
"semi": "error",
|
||||
"indent": [
|
||||
"error",
|
||||
4,
|
||||
{
|
||||
"SwitchCase": 1
|
||||
}
|
||||
],
|
||||
"eqeqeq": "error"
|
||||
}
|
||||
}
|
24
.gitea/workflows/build.yml
Normal file
24
.gitea/workflows/build.yml
Normal file
@ -0,0 +1,24 @@
|
||||
name: 'CI'
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- master
|
||||
|
||||
# Jobs
|
||||
jobs:
|
||||
# Build Project
|
||||
build:
|
||||
name: Build
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: Checkout Repository
|
||||
uses: actions/checkout@v4
|
||||
# Dependencies
|
||||
- name: Install Dependencies
|
||||
run: npm ci
|
||||
# Test
|
||||
- name: Test
|
||||
run: npm test
|
||||
# Lint
|
||||
- name: Lint
|
||||
run: npm run lint
|
117
data/function.h
Normal file
117
data/function.h
Normal file
@ -0,0 +1,117 @@
|
||||
#include <functional>
|
||||
#include <utility>
|
||||
|
||||
// Information Interface
|
||||
template <typename Ret, typename... Args>
|
||||
class __FunctionInfo {
|
||||
typedef Ret (*type)(Args...);
|
||||
public:
|
||||
[[nodiscard]] virtual bool can_overwrite() const = 0;
|
||||
[[nodiscard]] virtual type get() const = 0;
|
||||
[[nodiscard]] virtual type *get_addr() const = 0;
|
||||
virtual void update(type new_func) = 0;
|
||||
virtual ~__FunctionInfo() = default;
|
||||
};
|
||||
|
||||
// Thunks
|
||||
typedef void *(*thunk_enabler_t)(void *target, void *thunk);
|
||||
extern thunk_enabler_t thunk_enabler;
|
||||
|
||||
// Function
|
||||
template <unsigned int, typename T>
|
||||
class __Function;
|
||||
template <unsigned int discriminator, typename Ret, typename... Args>
|
||||
class __Function<discriminator, Ret(Args...)> final {
|
||||
// Prevent Copying
|
||||
__PREVENT_COPY(__Function);
|
||||
__PREVENT_DESTRUCTION(__Function);
|
||||
|
||||
// Instance
|
||||
static __Function *instance;
|
||||
|
||||
// Current Function
|
||||
typedef __FunctionInfo<Ret, Args...> *func_t;
|
||||
const func_t func;
|
||||
|
||||
public:
|
||||
// Types
|
||||
typedef Ret (*ptr_type)(Args...);
|
||||
typedef std::function<Ret(Args...)> type;
|
||||
typedef std::function<Ret(const type &, Args...)> overwrite_type;
|
||||
|
||||
// State
|
||||
const bool enabled;
|
||||
const char *const name;
|
||||
|
||||
// Backup Of Original Function Pointer
|
||||
const ptr_type backup;
|
||||
|
||||
#ifdef {{ BUILDING_SYMBOLS_GUARD }}
|
||||
// Constructor
|
||||
__Function(const char *const name_, const func_t func_):
|
||||
func(func_),
|
||||
enabled(func->can_overwrite()),
|
||||
name(name_),
|
||||
backup(func->get())
|
||||
{
|
||||
instance = this;
|
||||
}
|
||||
#else
|
||||
// Prevent Construction
|
||||
__PREVENT_JUST_CONSTRUCTION(__Function);
|
||||
#endif
|
||||
|
||||
// Overwrite Function
|
||||
[[nodiscard]] bool overwrite(const overwrite_type &target) {
|
||||
// Check If Enabled
|
||||
if (!enabled) {
|
||||
return false;
|
||||
}
|
||||
// Enable Thunk
|
||||
enable_thunk();
|
||||
// Overwrite
|
||||
type original = get_thunk_target();
|
||||
thunk_target = [original, target](Args... args) {
|
||||
return target(original, std::forward<Args>(args)...);
|
||||
};
|
||||
return true;
|
||||
}
|
||||
|
||||
// Getters
|
||||
[[nodiscard]] ptr_type get(const bool result_will_be_stored) {
|
||||
if (!enabled) {
|
||||
return nullptr;
|
||||
} else {
|
||||
if (result_will_be_stored) {
|
||||
enable_thunk();
|
||||
}
|
||||
return func->get();
|
||||
}
|
||||
}
|
||||
[[nodiscard]] ptr_type *get_vtable_addr() const {
|
||||
return func->get_addr();
|
||||
}
|
||||
|
||||
private:
|
||||
// Thunk
|
||||
[[nodiscard]] type get_thunk_target() const {
|
||||
if (thunk_target) {
|
||||
return thunk_target;
|
||||
} else {
|
||||
return backup;
|
||||
}
|
||||
}
|
||||
static Ret thunk(Args... args) {
|
||||
return instance->get_thunk_target()(std::forward<Args>(args)...);
|
||||
}
|
||||
// Enable Thunk
|
||||
type thunk_target;
|
||||
bool thunk_enabled = false;
|
||||
void enable_thunk() {
|
||||
if (!thunk_enabled && enabled) {
|
||||
ptr_type real_thunk = (ptr_type) thunk_enabler((void *) backup, (void *) thunk);
|
||||
func->update(real_thunk);
|
||||
thunk_enabled = true;
|
||||
}
|
||||
}
|
||||
};
|
16
data/internal.h
Normal file
16
data/internal.h
Normal file
@ -0,0 +1,16 @@
|
||||
#define __PREVENT_DESTRUCTION(self) \
|
||||
~self() = delete
|
||||
#define __PREVENT_JUST_CONSTRUCTION(self) \
|
||||
self() = delete
|
||||
#define __PREVENT_CONSTRUCTION(self) \
|
||||
__PREVENT_JUST_CONSTRUCTION(self); \
|
||||
__PREVENT_DESTRUCTION(self)
|
||||
#define __PREVENT_COPY(self) \
|
||||
self(const self &) = delete; \
|
||||
self &operator=(const self &) = delete
|
||||
template <typename T>
|
||||
static constexpr decltype(sizeof(T)) __real_sizeof() {
|
||||
return sizeof(std::conditional_t<std::is_reference_v<T>, void *, T>);
|
||||
}
|
||||
template <typename T>
|
||||
using __make_reference = T &;
|
69
data/out.cpp
Normal file
69
data/out.cpp
Normal file
@ -0,0 +1,69 @@
|
||||
#define {{ BUILDING_SYMBOLS_GUARD }}
|
||||
#include "{{ headerPath }}"
|
||||
|
||||
// Global Instance
|
||||
template <unsigned int id, typename Ret, typename... Args>
|
||||
__Function<id, Ret(Args...)> *__Function<id, Ret(Args...)>::instance;
|
||||
|
||||
// Normal Function Information
|
||||
template <typename Ret, typename... Args>
|
||||
class __NormalFunctionInfo final : public __FunctionInfo<Ret, Args...> {
|
||||
typedef Ret (*type)(Args...);
|
||||
type func;
|
||||
public:
|
||||
// Constructor
|
||||
explicit __NormalFunctionInfo(const type func_):
|
||||
func(func_) {}
|
||||
// Functions
|
||||
[[nodiscard]] bool can_overwrite() const override {
|
||||
return true;
|
||||
}
|
||||
[[nodiscard]] type get() const override {
|
||||
return func;
|
||||
}
|
||||
[[nodiscard]] type *get_addr() const override {
|
||||
return nullptr;
|
||||
}
|
||||
void update(const type new_func) override {
|
||||
func = new_func;
|
||||
}
|
||||
};
|
||||
|
||||
// Virtual Function Information
|
||||
template <typename Ret, typename... Args>
|
||||
class __VirtualFunctionInfo final : public __FunctionInfo<Ret, Args...> {
|
||||
typedef Ret (*type)(Args...);
|
||||
type *const addr;
|
||||
void *const parent;
|
||||
public:
|
||||
// Constructor
|
||||
__VirtualFunctionInfo(type *const addr_, void *const parent_):
|
||||
addr(addr_),
|
||||
parent(parent_) {}
|
||||
// Functions
|
||||
[[nodiscard]] bool can_overwrite() const override {
|
||||
// If this function's address matches its parent's,
|
||||
// then it was just inherited and does not actually exist.
|
||||
// Overwriting this function would also overwrite its parent
|
||||
// which would cause undesired behavior.
|
||||
return get() != parent;
|
||||
}
|
||||
[[nodiscard]] type get() const override {
|
||||
return *get_addr();
|
||||
}
|
||||
[[nodiscard]] type *get_addr() const override {
|
||||
return addr;
|
||||
}
|
||||
void update(const type new_func) override {
|
||||
// Address Should Have Already Been Updated
|
||||
if (get() != new_func) {
|
||||
__builtin_trap();
|
||||
}
|
||||
}
|
||||
};
|
||||
#undef super
|
||||
|
||||
// Disable Warnings
|
||||
#pragma GCC diagnostic ignored "-Winvalid-offsetof"
|
||||
|
||||
{{ main }}
|
42
data/out.h
Normal file
42
data/out.h
Normal file
@ -0,0 +1,42 @@
|
||||
#pragma once
|
||||
|
||||
// Check Architecture
|
||||
#ifndef __arm__
|
||||
#error "Symbols Are ARM-Only"
|
||||
#endif
|
||||
|
||||
// Type Traits
|
||||
#include <type_traits>
|
||||
|
||||
// Internal Macros
|
||||
{{ include internal.h }}
|
||||
|
||||
// Function Object
|
||||
{{ include function.h }}
|
||||
|
||||
// Headers
|
||||
#include <cstddef>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <cstring>
|
||||
|
||||
// Shortcuts
|
||||
typedef unsigned char uchar;
|
||||
typedef unsigned short ushort;
|
||||
typedef unsigned int uint;
|
||||
|
||||
// Forward Declarations
|
||||
{{ forwardDeclarations }}
|
||||
|
||||
// Extra Headers
|
||||
{{ extraHeaders }}
|
||||
|
||||
// Warnings
|
||||
#pragma GCC diagnostic push
|
||||
#pragma GCC diagnostic ignored "-Wshadow"
|
||||
|
||||
{{ main }}
|
||||
|
||||
// Cleanup Warnings
|
||||
#pragma GCC diagnostic pop
|
84
eslint.config.mjs
Normal file
84
eslint.config.mjs
Normal file
@ -0,0 +1,84 @@
|
||||
// @ts-check
|
||||
import eslint from '@eslint/js';
|
||||
import tseslint from 'typescript-eslint';
|
||||
import stylistic from '@stylistic/eslint-plugin';
|
||||
import * as path from 'node:path';
|
||||
import * as url from 'node:url';
|
||||
|
||||
export default tseslint.config(
|
||||
eslint.configs.recommended,
|
||||
tseslint.configs.strictTypeChecked,
|
||||
tseslint.configs.stylisticTypeChecked,
|
||||
stylistic.configs.recommended,
|
||||
{
|
||||
languageOptions: {
|
||||
parserOptions: {
|
||||
projectService: {
|
||||
allowDefaultProject: [
|
||||
'eslint.config.mjs'
|
||||
]
|
||||
},
|
||||
tsconfigRootDir: path.dirname(url.fileURLToPath(import.meta.url))
|
||||
}
|
||||
},
|
||||
rules: {
|
||||
'@typescript-eslint/no-unnecessary-condition': [
|
||||
'error',
|
||||
{
|
||||
allowConstantLoopConditions: 'only-allowed-literals'
|
||||
}
|
||||
],
|
||||
'@stylistic/quotes': [
|
||||
'error',
|
||||
'single',
|
||||
{
|
||||
allowTemplateLiterals: 'avoidEscape'
|
||||
}
|
||||
],
|
||||
'@stylistic/indent': [
|
||||
'error',
|
||||
4,
|
||||
{
|
||||
SwitchCase: 1
|
||||
}
|
||||
],
|
||||
'@stylistic/comma-dangle': [
|
||||
'error',
|
||||
'never'
|
||||
],
|
||||
'@stylistic/quote-props': [
|
||||
'error',
|
||||
'as-needed'
|
||||
],
|
||||
'@stylistic/semi': [
|
||||
'error',
|
||||
'always'
|
||||
],
|
||||
'@stylistic/eol-last': [
|
||||
'error',
|
||||
'never'
|
||||
],
|
||||
'@stylistic/lines-between-class-members': 'off',
|
||||
'@stylistic/object-curly-spacing': 'off',
|
||||
'@stylistic/brace-style': [
|
||||
'error',
|
||||
'1tbs'
|
||||
],
|
||||
eqeqeq: 'error'
|
||||
}
|
||||
},
|
||||
{
|
||||
ignores: [
|
||||
'build/',
|
||||
'node_modules/'
|
||||
]
|
||||
},
|
||||
{
|
||||
files: [
|
||||
'src/test/**/*'
|
||||
],
|
||||
rules: {
|
||||
'@typescript-eslint/no-floating-promises': 'off'
|
||||
}
|
||||
}
|
||||
);
|
981
package-lock.json
generated
981
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
20
package.json
20
package.json
@ -4,18 +4,20 @@
|
||||
"description": "",
|
||||
"main": "build/index.js",
|
||||
"scripts": {
|
||||
"start": "tsc && node build/index.js",
|
||||
"lint": "eslint . --ext .ts"
|
||||
"build": "rm -rf build && tsc",
|
||||
"start": "npm run build && node build/index.js",
|
||||
"lint": "eslint .",
|
||||
"test": "npm run build && node --test build/test"
|
||||
},
|
||||
"author": "TheBrokenRail",
|
||||
"license": "MIT",
|
||||
"devDependencies": {
|
||||
"@tsconfig/node-lts": "^18.12.5",
|
||||
"@tsconfig/strictest": "^2.0.2",
|
||||
"@types/node": "^20.10.6",
|
||||
"@typescript-eslint/eslint-plugin": "^6.17.0",
|
||||
"@typescript-eslint/parser": "^6.17.0",
|
||||
"eslint": "^8.56.0",
|
||||
"typescript": "^5.3.3"
|
||||
"@stylistic/eslint-plugin": "^4.0.1",
|
||||
"@tsconfig/node18": "^18.2.4",
|
||||
"@tsconfig/strictest": "^2.0.5",
|
||||
"@types/node": "^22.13.4",
|
||||
"eslint": "^9.20.1",
|
||||
"typescript": "^5.7.3",
|
||||
"typescript-eslint": "^8.24.0"
|
||||
}
|
||||
}
|
||||
|
194
src/common.ts
194
src/common.ts
@ -1,38 +1,71 @@
|
||||
import * as fs from 'node:fs';
|
||||
import * as path from 'node:path';
|
||||
|
||||
// Constants
|
||||
export const INDENT = ' ';
|
||||
export const POINTER_SIZE = 4;
|
||||
export const RTTI_SIZE = POINTER_SIZE;
|
||||
export const EXTENSION = '.def';
|
||||
export const STRUCTURE_FILES: {[id: string]: string} = {};
|
||||
export const STRUCTURE_FILES: Record<string, string> = {};
|
||||
export const COMMENT = '//';
|
||||
export const INTERNAL = '__';
|
||||
export const BUILDING_SYMBOLS_GUARD = 'BUILDING_SYMBOLS_LIB';
|
||||
// Read Definition File
|
||||
export function readDefinition(name: string) {
|
||||
if (!STRUCTURE_FILES[name]) {
|
||||
throw new Error(`Missing Definition File: ${name}${EXTENSION}`);
|
||||
}
|
||||
return fs.readFileSync(STRUCTURE_FILES[name]!, {encoding: 'utf8'});
|
||||
return fs.readFileSync(STRUCTURE_FILES[name], {encoding: 'utf8'});
|
||||
}
|
||||
export function syntaxError(message?: string) {
|
||||
throw new Error('Syntax Error' + (message ? `: ${message}` : ''));
|
||||
// Error Handling
|
||||
export function extendErrorMessage(e: unknown, message: string) {
|
||||
let extra: string | null = null;
|
||||
if (typeof e === 'string') {
|
||||
extra = e;
|
||||
} else if (e instanceof Error) {
|
||||
extra = e.message;
|
||||
}
|
||||
if (extra) {
|
||||
message += ': ' + extra;
|
||||
}
|
||||
return message;
|
||||
}
|
||||
export function syntaxError(message?: string): never {
|
||||
throw new Error(extendErrorMessage(message, 'Syntax Error'));
|
||||
}
|
||||
// Convert 'int x' Into {type: 'int', name: 'x'}
|
||||
const POINTER_TOKENS = ['*', '&'];
|
||||
export function parseTypeAndName(piece: string) {
|
||||
piece = piece.trim();
|
||||
// Split On Last Space
|
||||
const index = piece.lastIndexOf(' ');
|
||||
if (index === -1) {
|
||||
syntaxError('Unable To Find Name/Type Divider');
|
||||
}
|
||||
let name = piece.substring(index + 1)!;
|
||||
let type = piece.substring(0, index)!;
|
||||
let name = piece.substring(index + 1);
|
||||
let type = piece.substring(0, index);
|
||||
// Move Asterisks From Name To Type
|
||||
while (name.startsWith('*')) {
|
||||
while (POINTER_TOKENS.some(x => name.startsWith(x))) {
|
||||
const x = name.substring(0, 1);
|
||||
name = name.substring(1);
|
||||
if (!type.endsWith('*')) {
|
||||
if (!POINTER_TOKENS.some(x => type.endsWith(x))) {
|
||||
type += ' ';
|
||||
}
|
||||
type += '*';
|
||||
type += x;
|
||||
}
|
||||
// Trim
|
||||
name = name.trim();
|
||||
if (name.length === 0) {
|
||||
syntaxError();
|
||||
}
|
||||
type = type.trim();
|
||||
if (type.length === 0) {
|
||||
syntaxError();
|
||||
}
|
||||
// Return
|
||||
return {type, name};
|
||||
}
|
||||
// Convert To Uppercase-Snake-Case
|
||||
export function toUpperSnakeCase(str: string) {
|
||||
let wasUpper = false;
|
||||
let nextIsUpper = false;
|
||||
@ -55,69 +88,134 @@ export function toUpperSnakeCase(str: string) {
|
||||
}
|
||||
return out;
|
||||
}
|
||||
// Convert 'int' To 'int ' But Leave 'int *' As Is
|
||||
export function formatType(type: string) {
|
||||
if (!type.endsWith('*')) {
|
||||
type = type.trim();
|
||||
if (!POINTER_TOKENS.some(x => type.endsWith(x))) {
|
||||
type += ' ';
|
||||
}
|
||||
return type;
|
||||
}
|
||||
export const COMMENT = '//';
|
||||
// Convert Number To Hexadecimal
|
||||
export function toHex(x: number) {
|
||||
return '0x' + x.toString(16);
|
||||
}
|
||||
// Generate C++ Size Assertion
|
||||
export function assertSize(name: string, size: number) {
|
||||
let out = '';
|
||||
// Define Size Macro
|
||||
const macro = toUpperSnakeCase(name) + '_SIZE';
|
||||
out += `#define ${macro} ${toHex(size)}\n`;
|
||||
// Check Size
|
||||
out += `static_assert(sizeof(${name}) == ${macro}, "Invalid Size");\n`;
|
||||
// Return
|
||||
return out;
|
||||
return `static_assert(sizeof(${name}) == ${toHex(size)}, "Invalid Size");\n`;
|
||||
}
|
||||
// Parse Integer With Error Checking
|
||||
export function safeParseInt(str: string) {
|
||||
str = str.trim();
|
||||
const x = parseInt(str);
|
||||
if (isNaN(x)) {
|
||||
throw new Error('Invalid Integer: ' + str);
|
||||
}
|
||||
return x;
|
||||
}
|
||||
// Generate "Self" Argument For Functions
|
||||
export function getSelfArg(type: string) {
|
||||
return `${type} *self`;
|
||||
}
|
||||
export function getArgNames(args: string) {
|
||||
// Remove Parentheses
|
||||
args = args.substring(1, args.length - 1);
|
||||
if (args.length === 0) {
|
||||
return [];
|
||||
}
|
||||
// Split
|
||||
const argsList = args.split(',');
|
||||
// Parse
|
||||
const out = [];
|
||||
for (let arg of argsList) {
|
||||
arg = arg.trim();
|
||||
// Remove Type
|
||||
const nameStart = Math.max(arg.lastIndexOf(' '), arg.lastIndexOf('*')) + 1;
|
||||
arg = arg.substring(nameStart);
|
||||
// Collect
|
||||
out.push(arg);
|
||||
}
|
||||
// Return
|
||||
return out;
|
||||
}
|
||||
// Prepend Argument To Function
|
||||
export function prependArg(args: string, arg: string) {
|
||||
if (args !== '()') {
|
||||
arg += ', ';
|
||||
}
|
||||
return '(' + arg + args.substring(1);
|
||||
}
|
||||
export function removeFirstArg(args: string) {
|
||||
let index = args.indexOf(',');
|
||||
if (index === -1) {
|
||||
index = args.indexOf(')');
|
||||
} else {
|
||||
index++;
|
||||
}
|
||||
return '(' + args.substring(index).trim();
|
||||
// Get Data Directory
|
||||
export function getDataDir() {
|
||||
return path.join(__dirname, '..', 'data');
|
||||
}
|
||||
// Format File From Data Directory
|
||||
export function formatFile(file: string, options: Record<string, string>, includeOtherFiles = true) {
|
||||
const newOptions = Object.assign({}, options);
|
||||
const dataDir = getDataDir();
|
||||
// Include Other Files
|
||||
if (includeOtherFiles) {
|
||||
const otherFiles = fs.readdirSync(dataDir);
|
||||
for (const otherFile of otherFiles) {
|
||||
newOptions[`include ${otherFile}`] = formatFile(otherFile, options, false);
|
||||
}
|
||||
}
|
||||
// Format
|
||||
file = path.join(dataDir, file);
|
||||
let data = fs.readFileSync(file, 'utf8');
|
||||
for (const key in newOptions) {
|
||||
let value = newOptions[key];
|
||||
if (value) {
|
||||
value = value.trim();
|
||||
data = data.replace(`{{ ${key} }}`, value);
|
||||
}
|
||||
}
|
||||
return data.trim() + '\n';
|
||||
}
|
||||
// Generate Code That Will Disable C++ Construction
|
||||
export function preventConstruction(self: string) {
|
||||
let out = '';
|
||||
out += `${INDENT}${INTERNAL}PREVENT_CONSTRUCTION(${self});\n`;
|
||||
out += `${INDENT}${INTERNAL}PREVENT_COPY(${self});\n`;
|
||||
return out;
|
||||
}
|
||||
// Remove Array Information From Variable Name
|
||||
export function extractArrayInfo(name: string) {
|
||||
let arrayInfo = '';
|
||||
const arrayInfoIndex = name.indexOf('[');
|
||||
if (arrayInfoIndex !== -1) {
|
||||
arrayInfo = name.substring(arrayInfoIndex);
|
||||
name = name.substring(0, arrayInfoIndex);
|
||||
}
|
||||
return {
|
||||
name,
|
||||
arrayInfo
|
||||
};
|
||||
}
|
||||
// Forward Function Arguments Into Call
|
||||
export function forwardArguments(args: string, extra: string[]) {
|
||||
const separator = ',';
|
||||
const openingParen = '(';
|
||||
const opening = ['<', openingParen, '['];
|
||||
const closingParen = ')';
|
||||
const closing = ['>', closingParen, ']'];
|
||||
// Remove Enclosing Parenthesis
|
||||
args = args.trim();
|
||||
if (!args.startsWith(openingParen) || !args.endsWith(closingParen)) {
|
||||
syntaxError();
|
||||
}
|
||||
args = args.substring(openingParen.length, args.length - closingParen.length);
|
||||
// Parse
|
||||
const parts: string[] = [];
|
||||
let start = 0;
|
||||
const stack = [];
|
||||
for (let i = 0; i < args.length; i++) {
|
||||
const c = args.charAt(i);
|
||||
if (stack.length === 0 && c === separator) {
|
||||
parts.push(args.substring(start, i));
|
||||
start = i + 1;
|
||||
} else if (opening.includes(c)) {
|
||||
stack.push(c);
|
||||
} else {
|
||||
const i = closing.indexOf(c);
|
||||
if (i !== -1) {
|
||||
const x = stack.pop();
|
||||
if (x !== opening[i]) {
|
||||
syntaxError();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if (stack.length !== 0) {
|
||||
syntaxError();
|
||||
}
|
||||
if (start < args.length) {
|
||||
parts.push(args.substring(start));
|
||||
}
|
||||
// Handle
|
||||
const out: string[] = [];
|
||||
for (const part of parts) {
|
||||
out.push(extractArrayInfo(parseTypeAndName(part.trim()).name).name);
|
||||
}
|
||||
out.unshift(...extra);
|
||||
return openingParen + out.join(separator + ' ') + closingParen;
|
||||
}
|
173
src/index.ts
173
src/index.ts
@ -1,45 +1,56 @@
|
||||
import * as fs from 'node:fs';
|
||||
import { STRUCTURE_FILES, EXTENSION, INDENT } from './common';
|
||||
import * as path from 'node:path';
|
||||
import { STRUCTURE_FILES, EXTENSION, formatFile, getDataDir, extendErrorMessage, BUILDING_SYMBOLS_GUARD } from './common';
|
||||
import { getStructure } from './map';
|
||||
import { Struct } from './struct';
|
||||
|
||||
// Arguments
|
||||
if (process.argv.length < 5) {
|
||||
console.log('USAGE: npm start -- <Source Output File> <Header Output File> <Input Files...>');
|
||||
// Handle Errors
|
||||
process.on('uncaughtException', function (e) {
|
||||
console.error('Error: ' + e.message);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
// Arguments
|
||||
function nextArgument() {
|
||||
const out = process.argv.shift();
|
||||
if (!out) {
|
||||
throw new Error('Not Enough Arguments!');
|
||||
}
|
||||
return out;
|
||||
}
|
||||
process.argv.shift();
|
||||
process.argv.shift();
|
||||
nextArgument();
|
||||
nextArgument();
|
||||
// Output Files
|
||||
function invalidFileType(file: string) {
|
||||
throw new Error(`Invalid File Type: ${file}`);
|
||||
}
|
||||
const sourceOutput = process.argv.shift()!;
|
||||
if (!sourceOutput.endsWith('.cpp')) {
|
||||
invalidFileType(sourceOutput);
|
||||
}
|
||||
const headerOutput = process.argv.shift()!;
|
||||
const sourceOutput = nextArgument();
|
||||
fs.rmSync(sourceOutput, {force: true, recursive: true});
|
||||
fs.mkdirSync(sourceOutput, {recursive: true});
|
||||
const headerOutput = nextArgument();
|
||||
if (!headerOutput.endsWith('.h')) {
|
||||
invalidFileType(headerOutput);
|
||||
}
|
||||
const extraHeaders: string[] = [];
|
||||
// Input Files
|
||||
const extraHeaderFiles: string[] = [];
|
||||
while (process.argv.length > 0) {
|
||||
const file = process.argv.shift()!;
|
||||
if (file.endsWith(EXTENSION)) {
|
||||
const filePath = nextArgument();
|
||||
const file = path.parse(filePath);
|
||||
if (file.ext === EXTENSION) {
|
||||
// Get Class Name
|
||||
const parts = file.split('/');
|
||||
let name = parts[parts.length - 1]!;
|
||||
name = name.substring(0, name.length - EXTENSION.length);
|
||||
const fullName = file.base;
|
||||
const name = file.name;
|
||||
// Store
|
||||
if (name in STRUCTURE_FILES) {
|
||||
throw new Error(`Multiple Definition Files Provided: ${name}${EXTENSION}`);
|
||||
if (STRUCTURE_FILES[name]) {
|
||||
throw new Error(`Multiple Definition Files Provided: ${fullName}`);
|
||||
}
|
||||
STRUCTURE_FILES[name] = file;
|
||||
} else if (file.endsWith('.h')) {
|
||||
STRUCTURE_FILES[name] = filePath;
|
||||
} else if (file.ext === '.h') {
|
||||
// Extra Headers
|
||||
extraHeaders.push(file);
|
||||
extraHeaderFiles.push(filePath);
|
||||
} else {
|
||||
// Invalid File Type
|
||||
invalidFileType(file);
|
||||
invalidFileType(filePath);
|
||||
}
|
||||
}
|
||||
|
||||
@ -52,9 +63,9 @@ function loadSymbols() {
|
||||
}
|
||||
// Sort
|
||||
structureObjects.sort((a, b) => {
|
||||
if (a.getName() > b.getName()) {
|
||||
if (a.name > b.name) {
|
||||
return 1;
|
||||
} else if (a.getName() < b.getName()) {
|
||||
} else if (a.name < b.name) {
|
||||
return -1;
|
||||
} else {
|
||||
return 0;
|
||||
@ -67,7 +78,7 @@ function loadSymbols() {
|
||||
// Sort Structures By Dependency
|
||||
function dependencySort(structureObjects: Struct[]) {
|
||||
let loops = 0;
|
||||
const MAX_LOOPS = 100;
|
||||
const MAX_LOOPS = 1000;
|
||||
while (true) {
|
||||
if (loops > MAX_LOOPS) {
|
||||
throw new Error('Unable To Sort Dependencies');
|
||||
@ -112,105 +123,75 @@ function makeHeaderPart() {
|
||||
|
||||
// Generate Code
|
||||
let structures = '';
|
||||
let isFirst = true;
|
||||
for (const structure of structureObjects) {
|
||||
const name = structure.getName();
|
||||
if (isFirst) {
|
||||
isFirst = false;
|
||||
} else {
|
||||
structures += '\n';
|
||||
}
|
||||
const name = structure.name;
|
||||
structures += `// ${name}\n`;
|
||||
try {
|
||||
structures += structure.generate();
|
||||
} catch (e) {
|
||||
console.log(`Error Generating Header: ${name}: ${e instanceof Error ? e.stack : e}`);
|
||||
process.exit(1);
|
||||
throw new Error(extendErrorMessage(e, 'Generating Header: ' + name));
|
||||
}
|
||||
structures += '\n';
|
||||
}
|
||||
|
||||
// Return
|
||||
let result = '';
|
||||
result += '// Init\n';
|
||||
result += 'void init_symbols();\n\n';
|
||||
result += structures;
|
||||
return result;
|
||||
return structures;
|
||||
}
|
||||
|
||||
// Create Main Header
|
||||
function makeMainHeader(output: string) {
|
||||
let result = '';
|
||||
result += '#pragma once\n';
|
||||
result += '\n// Check Architecture\n';
|
||||
result += '#ifndef __arm__\n';
|
||||
result += '#error "Symbols Are ARM-Only"\n';
|
||||
result += '#endif\n';
|
||||
result += '\n// Shortcuts\n';
|
||||
result += 'typedef unsigned char uchar;\n';
|
||||
result += 'typedef unsigned short ushort;\n';
|
||||
result += 'typedef unsigned int uint;\n';
|
||||
result += '\n// Forward Declarations\n';
|
||||
// Forward Declarations
|
||||
let forwardDeclarations = '';
|
||||
for (const name in STRUCTURE_FILES) {
|
||||
result += `typedef struct ${name} ${name};\n`;
|
||||
forwardDeclarations += `typedef struct ${name} ${name};\n`;
|
||||
}
|
||||
result += '\n// Extra Headers\n';
|
||||
for (const file of extraHeaders) {
|
||||
result += fs.readFileSync(file, {encoding: 'utf8'});
|
||||
forwardDeclarations = forwardDeclarations.trim();
|
||||
// Extra Headers
|
||||
let extraHeaders = '';
|
||||
for (const file of extraHeaderFiles) {
|
||||
extraHeaders += fs.readFileSync(file, {encoding: 'utf8'});
|
||||
}
|
||||
result += '\n// Headers\n';
|
||||
result += '#include <cstddef>\n';
|
||||
result += '#include <string>\n';
|
||||
result += '#include <vector>\n';
|
||||
result += '#include <map>\n';
|
||||
result += '\n// Warnings\n';
|
||||
result += '#pragma GCC diagnostic push\n';
|
||||
result += '#pragma GCC diagnostic ignored "-Winvalid-offsetof"\n';
|
||||
result += '#pragma GCC diagnostic ignored "-Wshadow"\n\n';
|
||||
result += makeHeaderPart();
|
||||
result += '\n// Cleanup Warnings\n';
|
||||
result += '#pragma GCC diagnostic pop\n';
|
||||
extraHeaders = extraHeaders.trim();
|
||||
// Main
|
||||
const main = makeHeaderPart().trim();
|
||||
// Write
|
||||
const result = formatFile('out.h', {BUILDING_SYMBOLS_GUARD, forwardDeclarations, extraHeaders, main, data: getDataDir()});
|
||||
fs.writeFileSync(output, result);
|
||||
}
|
||||
makeMainHeader(headerOutput);
|
||||
|
||||
// Generate Compiled Code
|
||||
function makeCompiledCode(output: string) {
|
||||
function makeCompiledCode(outputDir: string) {
|
||||
// Load Symbols
|
||||
const structureObjects = loadSymbols();
|
||||
|
||||
// Generate
|
||||
let declarations = '';
|
||||
let init = '';
|
||||
let isFirst = true;
|
||||
let first = true;
|
||||
for (const structure of structureObjects) {
|
||||
const name = structure.getName();
|
||||
if (isFirst) {
|
||||
isFirst = false;
|
||||
} else {
|
||||
declarations += '\n';
|
||||
init += '\n';
|
||||
}
|
||||
declarations += `// ${name}\n`;
|
||||
init += `${INDENT}// ${name}\n`;
|
||||
try {
|
||||
const code = structure.generateCode();
|
||||
declarations += code.functions;
|
||||
init += code.init;
|
||||
} catch (e) {
|
||||
console.log(`Error Generating Code: ${name}: ${e instanceof Error ? e.stack : e}`);
|
||||
process.exit(1);
|
||||
}
|
||||
// Thunks
|
||||
let declarations = '';
|
||||
if (first) {
|
||||
first = false;
|
||||
declarations += '// Thunk Enabler\n';
|
||||
declarations += 'thunk_enabler_t thunk_enabler;\n\n';
|
||||
}
|
||||
|
||||
// Structure
|
||||
const name = structure.name;
|
||||
declarations += `// ${name}\n`;
|
||||
try {
|
||||
declarations += structure.generateCode().trim();
|
||||
} catch (e) {
|
||||
throw new Error(extendErrorMessage(e, 'Generating Code: ' + name));
|
||||
}
|
||||
declarations += '\n';
|
||||
|
||||
// Write
|
||||
let result = '';
|
||||
result += `#include "${fs.realpathSync(headerOutput)}"\n`;
|
||||
result += '\n#include <cstring>\n';
|
||||
result += '\n// Init\n';
|
||||
result += 'void init_symbols() {\n';
|
||||
result += init;
|
||||
result += '}\n\n';
|
||||
result += declarations;
|
||||
const headerPath = fs.realpathSync(headerOutput);
|
||||
const main = declarations.trim();
|
||||
const result = formatFile('out.cpp', {BUILDING_SYMBOLS_GUARD, headerPath, main, data: getDataDir()});
|
||||
const output = path.join(outputDir, name + '.cpp');
|
||||
fs.writeFileSync(output, result);
|
||||
}
|
||||
}
|
||||
makeCompiledCode(sourceOutput);
|
@ -1,52 +1,53 @@
|
||||
import { COMMENT, EXTENSION, getSelfArg, parseTypeAndName, prependArg, readDefinition, safeParseInt, syntaxError } from './common';
|
||||
import { COMMENT, extendErrorMessage, EXTENSION, parseTypeAndName, readDefinition, safeParseInt, syntaxError } from './common';
|
||||
import { Method } from './method';
|
||||
import { Property, StaticProperty } from './property';
|
||||
import { Struct } from './struct';
|
||||
|
||||
// Error Handling
|
||||
export class ErrorOnLine {
|
||||
readonly error: unknown;
|
||||
readonly file: string;
|
||||
readonly line: number;
|
||||
constructor (error: unknown, file: string, line: number) {
|
||||
this.error = error;
|
||||
this.file = file;
|
||||
this.line = line;
|
||||
class ErrorOnLine extends Error {
|
||||
constructor(e: unknown, file: string, line: number) {
|
||||
// Get Message
|
||||
let message: string;
|
||||
if (e instanceof ErrorOnLine) {
|
||||
message = e.message;
|
||||
} else {
|
||||
message = extendErrorMessage(e, file + ':' + line.toString());
|
||||
}
|
||||
// Call Constructor
|
||||
super(message);
|
||||
Object.setPrototypeOf(this, ErrorOnLine.prototype);
|
||||
}
|
||||
}
|
||||
|
||||
// Parse Property
|
||||
function parseProperty(args: string) {
|
||||
const parts = args.split(' = ');
|
||||
if (parts.length !== 2) {
|
||||
export const OFFSET_SEPARATOR = ' = ';
|
||||
export function parseProperty(args: string) {
|
||||
const [a, b] = args.split(OFFSET_SEPARATOR);
|
||||
if (!a || !b) {
|
||||
syntaxError('Invalid Piece Count');
|
||||
}
|
||||
const {type, name} = parseTypeAndName(parts[0]!);
|
||||
const offset = safeParseInt(parts[1]!);
|
||||
const {type, name} = parseTypeAndName(a);
|
||||
const offset = safeParseInt(b);
|
||||
return {type, name, offset};
|
||||
}
|
||||
|
||||
// Parse Method
|
||||
function parseMethod(args: string, self: string, insertSelfArg: boolean, isInherited: boolean) {
|
||||
export function parseMethod(args: string, self: string, insertSelfArg: boolean, isInherited: boolean) {
|
||||
const argsStart = args.indexOf('(');
|
||||
if (argsStart === -1) {
|
||||
syntaxError('Cannot Find Arguments');
|
||||
}
|
||||
const start = args.substring(0, argsStart).trim();
|
||||
const {type, name} = parseTypeAndName(start);
|
||||
const end = args.substring(argsStart).trim().split(' = ');
|
||||
if (end.length !== 2) {
|
||||
const end = args.substring(argsStart).trim().split(OFFSET_SEPARATOR);
|
||||
if (!end[0] || !end[1]) {
|
||||
syntaxError('Invalid Piece Count');
|
||||
}
|
||||
let methodArgs = end[0]!;
|
||||
const methodArgs = end[0].trim();
|
||||
if (!methodArgs.startsWith('(') || !methodArgs.endsWith(')')) {
|
||||
syntaxError('Invalid Method Arguments');
|
||||
}
|
||||
if (insertSelfArg) {
|
||||
const selfArg = getSelfArg(self);
|
||||
methodArgs = prependArg(methodArgs, selfArg);
|
||||
}
|
||||
const address = safeParseInt(end[1]!);
|
||||
const address = safeParseInt(end[1]);
|
||||
return new Method(self, name, type, methodArgs, address, isInherited, !insertSelfArg);
|
||||
}
|
||||
|
||||
@ -116,29 +117,29 @@ export function load(target: Struct, name: string, isExtended: boolean) {
|
||||
case 'vtable-size': {
|
||||
// Set VTable Size
|
||||
if (!isExtended) {
|
||||
target.setVTableSize(safeParseInt(args));
|
||||
target.getVTable().setSize(safeParseInt(args));
|
||||
}
|
||||
break;
|
||||
}
|
||||
case 'vtable': {
|
||||
// Set VTable Address
|
||||
target.ensureVTable();
|
||||
const vtable = target.getVTable();
|
||||
if (!isExtended && args.length > 0) {
|
||||
target.setVTableAddress(safeParseInt(args));
|
||||
vtable.setAddress(safeParseInt(args));
|
||||
}
|
||||
break;
|
||||
}
|
||||
case 'property': {
|
||||
// Add Property
|
||||
const info = parseProperty(args);
|
||||
target.addProperty(new Property(info.offset, info.type, info.name, target.getName()));
|
||||
target.addProperty(new Property(info.offset, info.type, info.name, target.name));
|
||||
break;
|
||||
}
|
||||
case 'static-property': {
|
||||
// Add Static Property
|
||||
if (!isExtended) {
|
||||
const info = parseProperty(args);
|
||||
target.addStaticProperty(new StaticProperty(info.offset, info.type, info.name, target.getName()));
|
||||
target.addStaticProperty(new StaticProperty(info.offset, info.type, info.name, target.name));
|
||||
}
|
||||
break;
|
||||
}
|
||||
@ -150,14 +151,14 @@ export function load(target: Struct, name: string, isExtended: boolean) {
|
||||
}
|
||||
case 'virtual-method': {
|
||||
// Add Virtual Method
|
||||
const method = parseMethod(args, target.getName(), true, isExtended);
|
||||
const method = parseMethod(args, target.name, true, isExtended);
|
||||
target.addMethod(method, true);
|
||||
break;
|
||||
}
|
||||
case 'static-method': {
|
||||
// Add Static Method
|
||||
if (!isExtended) {
|
||||
const method = parseMethod(args, target.getName(), false, false);
|
||||
const method = parseMethod(args, target.name, false, false);
|
||||
target.addMethod(method, false);
|
||||
}
|
||||
break;
|
||||
@ -165,23 +166,30 @@ export function load(target: Struct, name: string, isExtended: boolean) {
|
||||
case 'constructor': {
|
||||
// Constructor
|
||||
if (!isExtended) {
|
||||
let data = `${target.getName()} *constructor`;
|
||||
let data = `${target.name} *constructor`;
|
||||
if (args.startsWith('(')) {
|
||||
// No Custom Name
|
||||
data += ' ';
|
||||
} else {
|
||||
// Use Custom Name
|
||||
data += '_';
|
||||
}
|
||||
data += args;
|
||||
const method = parseMethod(data, target.getName(), true, false);
|
||||
const method = parseMethod(data, target.name, true, false);
|
||||
target.addMethod(method, false);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case 'vtable-destructor-offset': {
|
||||
// Set VTable Destructor Offset
|
||||
target.setVTableDestructorOffset(safeParseInt(args));
|
||||
target.getVTable().setDestructorOffset(safeParseInt(args));
|
||||
break;
|
||||
}
|
||||
case 'mark-as-simple': {
|
||||
// Mark As Simple
|
||||
if (isExtended) {
|
||||
throw new Error('Cannot Extend Simple Structure');
|
||||
}
|
||||
target.markAsSimple();
|
||||
break;
|
||||
}
|
||||
default: {
|
||||
@ -189,9 +197,6 @@ export function load(target: Struct, name: string, isExtended: boolean) {
|
||||
}
|
||||
}
|
||||
} catch (e) {
|
||||
if (e instanceof ErrorOnLine) {
|
||||
throw e;
|
||||
}
|
||||
// Find Line Number
|
||||
let lineNumber = 1;
|
||||
for (let i = 0; i <= startOfCommand; i++) {
|
||||
|
18
src/map.ts
18
src/map.ts
@ -1,14 +1,15 @@
|
||||
import { Struct } from './struct';
|
||||
import { ErrorOnLine, load } from './loader';
|
||||
import { load } from './loader';
|
||||
import { extendErrorMessage } from './common';
|
||||
|
||||
// Store Loaded Structures
|
||||
const structures: {[id: string]: Struct} = {};
|
||||
const structures: Record<string, Struct> = {};
|
||||
|
||||
// Get Or Load Structure
|
||||
export function getStructure(name: string) {
|
||||
if (name in structures) {
|
||||
if (structures[name]) {
|
||||
// Already Loaded
|
||||
return structures[name]!;
|
||||
return structures[name];
|
||||
} else {
|
||||
// Load Structure
|
||||
try {
|
||||
@ -20,14 +21,7 @@ export function getStructure(name: string) {
|
||||
// Return
|
||||
return target;
|
||||
} catch (e) {
|
||||
let error = e;
|
||||
let extra = '';
|
||||
if (e instanceof ErrorOnLine) {
|
||||
extra = `${e.file}:${e.line}: `;
|
||||
error = e.error;
|
||||
}
|
||||
console.log(`Error Loading ${name}: ${extra}${error instanceof Error ? error.stack : error}`);
|
||||
process.exit(1);
|
||||
throw new Error(extendErrorMessage(e, 'Loading ' + name));
|
||||
}
|
||||
}
|
||||
}
|
@ -1,73 +1,86 @@
|
||||
import { INDENT, formatType, getArgNames } from './common';
|
||||
import { INDENT, INTERNAL, formatType, getSelfArg, prependArg, toHex } from './common';
|
||||
|
||||
// A Template Parameter So Each Template Instantiation Is Unique
|
||||
let nextDiscriminator = 0;
|
||||
|
||||
// An Individual Method
|
||||
export class Method {
|
||||
readonly #discriminator: number;
|
||||
readonly self: string;
|
||||
readonly shortName: string;
|
||||
readonly returnType: string;
|
||||
readonly args: string;
|
||||
readonly #args: string;
|
||||
readonly address: number;
|
||||
readonly isInherited: boolean;
|
||||
readonly hasVarargs: boolean;
|
||||
readonly isStatic: boolean;
|
||||
|
||||
// Constructor
|
||||
constructor(self: string, name: string, returnType: string, args: string, address: number, isInherited: boolean, isStatic: boolean) {
|
||||
this.#discriminator = nextDiscriminator++;
|
||||
this.self = self;
|
||||
this.shortName = name;
|
||||
this.returnType = returnType;
|
||||
this.args = args;
|
||||
this.#args = args;
|
||||
this.address = address;
|
||||
this.isInherited = isInherited;
|
||||
this.hasVarargs = this.args.includes('...');
|
||||
this.isStatic = isStatic;
|
||||
}
|
||||
|
||||
// Get Type
|
||||
getNameWithCustomSelf(self: string) {
|
||||
return `${self}_${this.shortName}`;
|
||||
}
|
||||
getName() {
|
||||
return this.getNameWithCustomSelf(this.self);
|
||||
// Getters
|
||||
getName(separator = '_') {
|
||||
return this.self + separator + this.shortName;
|
||||
}
|
||||
getType() {
|
||||
return `${this.getName()}_t`;
|
||||
return this.getName() + '_t';
|
||||
}
|
||||
#getRawType() {
|
||||
return INTERNAL + 'raw_' + this.getType();
|
||||
}
|
||||
getProperty() {
|
||||
return `${INDENT}${this.#getRawType()} *${this.shortName};\n`;
|
||||
}
|
||||
#getFullType() {
|
||||
return `${INTERNAL}Function<${this.#discriminator.toString()}, ${this.#getRawType()}>`;
|
||||
}
|
||||
doesReturnValue() {
|
||||
return this.returnType.trim() !== 'void';
|
||||
}
|
||||
getArgs(includeSelf = true) {
|
||||
let args = this.#args.trim();
|
||||
if (includeSelf && !this.isStatic) {
|
||||
const selfArg = getSelfArg(this.self);
|
||||
args = prependArg(args, selfArg);
|
||||
}
|
||||
return args;
|
||||
}
|
||||
|
||||
// Generate Type Definition
|
||||
generateTypedef() {
|
||||
// Typedefs
|
||||
generateTypedefs() {
|
||||
let out = '';
|
||||
|
||||
// Normal Definition
|
||||
const returnType = formatType(this.returnType);
|
||||
out += `typedef ${returnType}(*${this.getType()})${this.args};\n`;
|
||||
|
||||
// Fancy Overwrite Does Not Support Varargs
|
||||
if (!this.hasVarargs) {
|
||||
// Overwrite Helper
|
||||
out += `#define __overwrite_helper_for_${this.getName()}(method, original) \\\n`;
|
||||
out += `${INDENT}[]${this.args} { \\\n`;
|
||||
out += `${INDENT}${INDENT}${returnType.trim() === 'void' ? '' : 'return '}method(${['original'].concat(getArgNames(this.args)).join(', ')}); \\\n`;
|
||||
out += `${INDENT}}\n`;
|
||||
}
|
||||
|
||||
// Return
|
||||
out += `typedef ${formatType(this.returnType)}${this.#getRawType()}${this.getArgs()};\n`;
|
||||
out += `typedef const std::function<${this.#getRawType()}> &${this.getType()};\n`;
|
||||
return out;
|
||||
}
|
||||
|
||||
// Generate Variable Definition
|
||||
generateDefinition(nameSuffix?: string) {
|
||||
return `${this.getType()} ${this.getName()}${nameSuffix !== undefined ? nameSuffix : ''};\n`;
|
||||
// Overwrite Helper
|
||||
#getVirtualCall(self: string = this.self) {
|
||||
return `${self}_vtable::base->${this.shortName}`;
|
||||
}
|
||||
|
||||
// Generate "New Method" Test
|
||||
generateNewMethodTest(parent: string | null, prefix: string, suffix: string) {
|
||||
let out = `#define __is_new_method_${this.getName()}() (`;
|
||||
if (!this.isInherited) {
|
||||
out += 'true';
|
||||
generate(code: boolean, isVirtual: boolean, parentSelf?: string) {
|
||||
let out = '';
|
||||
out += 'extern ';
|
||||
out += `${this.#getFullType()} *const ${this.getName()}`;
|
||||
if (code) {
|
||||
out += ` = new ${this.#getFullType()}(${JSON.stringify(this.getName('::'))}, `;
|
||||
if (isVirtual) {
|
||||
const parentMethod = parentSelf ? this.#getVirtualCall(parentSelf) : 'nullptr';
|
||||
out += `new ${INTERNAL}VirtualFunctionInfo(&${this.#getVirtualCall()}, (void *) ${parentMethod})`;
|
||||
} else {
|
||||
out += `((void *) ${prefix}${this.getName()}${suffix}) != ((void *) ${prefix}${this.getNameWithCustomSelf(parent!)}${suffix})`;
|
||||
out += `new ${INTERNAL}NormalFunctionInfo((${this.#getRawType()} *) ${toHex(this.address)})`;
|
||||
}
|
||||
out += ')\n';
|
||||
out += ')';
|
||||
}
|
||||
out += ';\n';
|
||||
return out;
|
||||
}
|
||||
}
|
@ -1,5 +1,6 @@
|
||||
import { formatType } from './common';
|
||||
import { extractArrayInfo, formatType, INTERNAL } from './common';
|
||||
|
||||
// A Single Property
|
||||
export class Property {
|
||||
readonly offset: number;
|
||||
readonly #type: string;
|
||||
@ -15,46 +16,36 @@ export class Property {
|
||||
}
|
||||
|
||||
// Getters
|
||||
type() {
|
||||
return `__type_${this.fullName()}`;
|
||||
}
|
||||
typedef() {
|
||||
let arrayInfo = '';
|
||||
const arrayInfoIndex = this.#name.indexOf('[');
|
||||
if (arrayInfoIndex !== -1) {
|
||||
arrayInfo = this.#name.substring(arrayInfoIndex);
|
||||
}
|
||||
return `typedef ${formatType(this.#type)}${this.type()}${arrayInfo};\n`;
|
||||
parts() {
|
||||
const info = extractArrayInfo(this.#name);
|
||||
return {
|
||||
type: formatType(this.#type),
|
||||
name: info.name,
|
||||
arrayInfo: info.arrayInfo
|
||||
};
|
||||
}
|
||||
name() {
|
||||
let name = this.#name;
|
||||
const arrayInfoIndex = this.#name.indexOf('[');
|
||||
if (arrayInfoIndex !== -1) {
|
||||
name = name.substring(0, arrayInfoIndex);
|
||||
return this.parts().name;
|
||||
}
|
||||
return name;
|
||||
#fullName(separator: string) {
|
||||
return this.#self + separator + this.name();
|
||||
}
|
||||
fullName() {
|
||||
return `${this.#self}_${this.name()}`;
|
||||
return this.#fullName('_');
|
||||
}
|
||||
prettyName() {
|
||||
return this.#fullName('::');
|
||||
}
|
||||
rawType() {
|
||||
return this.#type;
|
||||
}
|
||||
}
|
||||
|
||||
// A Single Static Property
|
||||
export class StaticProperty extends Property {
|
||||
// Constructor
|
||||
constructor(address: number, type: string, name: string, self: string) {
|
||||
super(address, type, name, self);
|
||||
}
|
||||
|
||||
// Generate Variable Definition
|
||||
globalPointerDefinition() {
|
||||
return `${this.type()} *${this.fullName()}_pointer;\n`;
|
||||
}
|
||||
|
||||
// Generate Macro
|
||||
macro() {
|
||||
return `#define ${this.fullName()} (*${this.fullName()}_pointer)\n`;
|
||||
// Reference
|
||||
referenceDefinition(addSelf: boolean) {
|
||||
const parts = this.parts();
|
||||
return `${INTERNAL}make_reference<${parts.type.trim()}${parts.arrayInfo}> ${addSelf ? this.prettyName() : this.name()}`;
|
||||
}
|
||||
}
|
241
src/struct.ts
241
src/struct.ts
@ -1,10 +1,11 @@
|
||||
import { INDENT, STRUCTURE_FILES, toHex, assertSize, formatType, getArgNames, removeFirstArg } from './common';
|
||||
import { INDENT, STRUCTURE_FILES, toHex, assertSize, INTERNAL, preventConstruction, BUILDING_SYMBOLS_GUARD, formatType, forwardArguments } from './common';
|
||||
import { Method } from './method';
|
||||
import { Property, StaticProperty } from './property';
|
||||
import { VTable } from './vtable';
|
||||
|
||||
// A Structure/Class
|
||||
export class Struct {
|
||||
readonly #name: string;
|
||||
readonly name: string;
|
||||
#vtable: VTable | null;
|
||||
readonly #methods: Method[];
|
||||
readonly #properties: Property[];
|
||||
@ -12,10 +13,11 @@ export class Struct {
|
||||
readonly #dependencies: string[];
|
||||
readonly #staticProperties: StaticProperty[];
|
||||
#directParent: string | null;
|
||||
#isSimple: boolean;
|
||||
|
||||
// Constructor
|
||||
constructor(name: string) {
|
||||
this.#name = name;
|
||||
this.name = name;
|
||||
this.#methods = [];
|
||||
this.#properties = [];
|
||||
this.#vtable = null;
|
||||
@ -23,6 +25,7 @@ export class Struct {
|
||||
this.#dependencies = [];
|
||||
this.#staticProperties = [];
|
||||
this.#directParent = null;
|
||||
this.#isSimple = false;
|
||||
}
|
||||
|
||||
// Dependencies
|
||||
@ -34,39 +37,29 @@ export class Struct {
|
||||
}
|
||||
|
||||
// Ensure VTable Exists
|
||||
ensureVTable() {
|
||||
getVTable() {
|
||||
if (this.#vtable === null) {
|
||||
this.#vtable = new VTable(this.#name);
|
||||
this.#vtable = new VTable(this.name);
|
||||
this.addProperty(this.#vtable.property);
|
||||
}
|
||||
}
|
||||
|
||||
// Set VTable Destructor Offset
|
||||
setVTableDestructorOffset(offset: number) {
|
||||
this.ensureVTable();
|
||||
this.#vtable!.setDestructorOffset(offset);
|
||||
return this.#vtable;
|
||||
}
|
||||
|
||||
// Setters
|
||||
setSize(size: number) {
|
||||
this.#size = size;
|
||||
}
|
||||
// Getters
|
||||
getName() {
|
||||
return this.#name;
|
||||
}
|
||||
|
||||
// Add Method
|
||||
addMethod(method: Method, isVirtual: boolean) {
|
||||
if (method.returnType !== this.#name && method.returnType in STRUCTURE_FILES) {
|
||||
if (method.returnType !== this.name && method.returnType in STRUCTURE_FILES) {
|
||||
this.#addDependency(method.returnType);
|
||||
}
|
||||
if (isVirtual) {
|
||||
if (method.self !== this.#name) {
|
||||
if (method.self !== this.name) {
|
||||
throw new Error();
|
||||
}
|
||||
this.ensureVTable();
|
||||
this.#vtable!.add(method);
|
||||
this.getVTable().add(method);
|
||||
} else {
|
||||
if (method.isInherited) {
|
||||
this.#addDependency(method.self);
|
||||
@ -79,7 +72,7 @@ export class Struct {
|
||||
this.#properties.push(property);
|
||||
// Add Dependency If Needed
|
||||
const type = property.rawType();
|
||||
if (type in STRUCTURE_FILES) {
|
||||
if (STRUCTURE_FILES[type]) {
|
||||
this.#addDependency(type);
|
||||
}
|
||||
}
|
||||
@ -88,14 +81,21 @@ export class Struct {
|
||||
this.#staticProperties.push(property);
|
||||
}
|
||||
|
||||
// Configure VTable
|
||||
setVTableSize(size: number) {
|
||||
this.ensureVTable();
|
||||
this.#vtable!.setSize(size);
|
||||
// "Simple" Structures
|
||||
markAsSimple() {
|
||||
this.#isSimple = true;
|
||||
}
|
||||
#checkSimple() {
|
||||
const checks: [string, boolean][] = [
|
||||
['Cannot Inherit', this.#directParent !== null],
|
||||
['Must Have A Defined Size', this.#size === null],
|
||||
['Cannot Have A VTable', this.#vtable !== null]
|
||||
];
|
||||
for (const check of checks) {
|
||||
if (check[1]) {
|
||||
throw new Error('Simple Structures ' + check[0]);
|
||||
}
|
||||
}
|
||||
setVTableAddress(address: number) {
|
||||
this.ensureVTable();
|
||||
this.#vtable!.setAddress(address);
|
||||
}
|
||||
|
||||
// Generate Properties
|
||||
@ -114,10 +114,9 @@ export class Struct {
|
||||
|
||||
// Add Properties
|
||||
let out = '';
|
||||
for (let i = 0; i < sortedProperties.length; i++) {
|
||||
const property = sortedProperties[i]!;
|
||||
const lastProperty = sortedProperties[i - 1];
|
||||
|
||||
let lastProperty: Property | null = null;
|
||||
let i = 0;
|
||||
for (const property of sortedProperties) {
|
||||
// Padding
|
||||
let offsetFromLastProperty = property.offset;
|
||||
if (lastProperty) {
|
||||
@ -125,64 +124,77 @@ export class Struct {
|
||||
}
|
||||
let paddingSize = toHex(offsetFromLastProperty);
|
||||
if (lastProperty) {
|
||||
paddingSize += ` - sizeof(${lastProperty.type()})`;
|
||||
const parts = lastProperty.parts();
|
||||
paddingSize += ` - ${INTERNAL}real_sizeof<${parts.type.trim()}${parts.arrayInfo}>()`;
|
||||
}
|
||||
if (!this.#isSimple) {
|
||||
out += `${INDENT}uchar ${INTERNAL}padding${i.toString()}[${paddingSize}];\n`;
|
||||
}
|
||||
out += `${INDENT}uchar __padding${i}[${paddingSize}];\n`;
|
||||
|
||||
// The Actual Property
|
||||
if (property !== sizeProperty) {
|
||||
out += `${INDENT}${property.type()} ${property.name()};\n`;
|
||||
const parts = property.parts();
|
||||
out += `${INDENT}${parts.type}${parts.name}${parts.arrayInfo};\n`;
|
||||
}
|
||||
|
||||
// Advance
|
||||
lastProperty = property;
|
||||
i++;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
// Generate C++ Method Shortcuts
|
||||
#generateMethod(method: Method, isVirtual: boolean) {
|
||||
let out = '';
|
||||
out += INDENT;
|
||||
if (method.isStatic) {
|
||||
out += 'static ';
|
||||
}
|
||||
const args = method.getArgs(false);
|
||||
out += formatType(method.returnType) + method.shortName + args + ' {\n';
|
||||
out += INDENT + INDENT;
|
||||
if (method.doesReturnValue()) {
|
||||
out += 'return ';
|
||||
}
|
||||
if (isVirtual) {
|
||||
out += `this->vtable->${method.shortName}`;
|
||||
} else {
|
||||
out += `${method.getName()}->get(false)`;
|
||||
}
|
||||
const extra = [];
|
||||
if (!method.isStatic) {
|
||||
extra.push(`(${method.self} *) this`);
|
||||
}
|
||||
out += forwardArguments(args, extra) + ';\n';
|
||||
out += `${INDENT}}\n`;
|
||||
return out;
|
||||
}
|
||||
#generateMethods() {
|
||||
let out = '';
|
||||
out += `#ifndef ${BUILDING_SYMBOLS_GUARD}\n`;
|
||||
// Normal Methods
|
||||
const getArgsOuter = (method: Method) => {
|
||||
let out = method.args;
|
||||
if (!method.isStatic) {
|
||||
// Remove "self"
|
||||
out = removeFirstArg(out);
|
||||
}
|
||||
return out;
|
||||
};
|
||||
const getArgsInner = (method: Method) => {
|
||||
const list = getArgNames(method.args);
|
||||
if (!method.isStatic) {
|
||||
// Replace "self" With "this"
|
||||
list[0] = `(${method.self} *) this`;
|
||||
}
|
||||
return list.join(', ');
|
||||
};
|
||||
for (const method of this.#methods) {
|
||||
if (!method.hasVarargs) {
|
||||
const returnType = method.returnType;
|
||||
const shortName = method.shortName;
|
||||
const fullName = method.getName();
|
||||
const args = getArgsOuter(method);
|
||||
out += `${INDENT}inline ${formatType(returnType)}${shortName}${args} { \\\n`;
|
||||
out += `${INDENT}${INDENT}${returnType.trim() === 'void' ? '' : 'return '}${fullName}(${getArgsInner(method)});\n`;
|
||||
out += `${INDENT}}\n`;
|
||||
}
|
||||
out += this.#generateMethod(method, false);
|
||||
}
|
||||
// Virtual Methods
|
||||
if (this.#vtable !== null) {
|
||||
const virtualMethods = this.#vtable.getMethods();
|
||||
for (const method of virtualMethods) {
|
||||
if (method && !method.hasVarargs) {
|
||||
const returnType = method.returnType;
|
||||
const shortName = method.shortName;
|
||||
const args = getArgsOuter(method);
|
||||
out += `${INDENT}inline ${formatType(returnType)}${shortName}${args} { \\\n`;
|
||||
out += `${INDENT}${INDENT}${returnType.trim() === 'void' ? '' : 'return '}this->vtable->${shortName}(${getArgsInner(method)});\n`;
|
||||
if (method) {
|
||||
out += this.#generateMethod(method, true);
|
||||
}
|
||||
}
|
||||
}
|
||||
// Allocation Method
|
||||
if (this.#size !== null) {
|
||||
// THIS DOES NOT CONSTRUCT THE OBJECT
|
||||
out += `${INDENT}static ${this.name} *allocate() {\n`;
|
||||
out += `${INDENT}${INDENT}return (${this.name} *) ::operator new(sizeof(${this.name}));\n`;
|
||||
out += `${INDENT}}\n`;
|
||||
}
|
||||
}
|
||||
}
|
||||
// Return
|
||||
out += '#endif\n';
|
||||
return out;
|
||||
}
|
||||
|
||||
@ -190,55 +202,43 @@ export class Struct {
|
||||
generate() {
|
||||
let out = '';
|
||||
|
||||
// Static Properties
|
||||
for (const property of this.#staticProperties) {
|
||||
out += property.typedef();
|
||||
out += `extern ${property.globalPointerDefinition()}`;
|
||||
out += property.macro();
|
||||
}
|
||||
|
||||
// Methods
|
||||
for (const method of this.#methods) {
|
||||
if (!method.isInherited) {
|
||||
out += method.generateTypedef();
|
||||
out += `extern ${method.generateDefinition()}`;
|
||||
out += method.generateNewMethodTest(this.#directParent, '', '');
|
||||
}
|
||||
// Check "Simple" Status
|
||||
if (this.#isSimple) {
|
||||
this.#checkSimple();
|
||||
}
|
||||
|
||||
// VTable
|
||||
if (this.#vtable !== null) {
|
||||
out += this.#vtable.generate(this.#directParent);
|
||||
out += this.#vtable.generate();
|
||||
}
|
||||
|
||||
// Property Typedefs
|
||||
for (const property of this.#properties) {
|
||||
out += property.typedef();
|
||||
// Method Wrappers
|
||||
let typedefs = '';
|
||||
let methodsOut = '';
|
||||
for (const method of this.#methods) {
|
||||
if (!method.isInherited) {
|
||||
typedefs += method.generateTypedefs();
|
||||
methodsOut += method.generate(false, false);
|
||||
}
|
||||
}
|
||||
out += typedefs;
|
||||
out += `#ifndef ${BUILDING_SYMBOLS_GUARD}\n`;
|
||||
out += methodsOut;
|
||||
out += '#endif\n';
|
||||
|
||||
// Structure
|
||||
out += `struct ${this.#name} {\n`;
|
||||
out += `struct ${this.name} final {\n`;
|
||||
out += this.#generateProperties();
|
||||
out += this.#generateMethods();
|
||||
out += `};\n`;
|
||||
|
||||
// Sanity Check Offsets
|
||||
for (let i = 0; i < this.#properties.length; i++) {
|
||||
const property = this.#properties[i]!;
|
||||
const name = property.name();
|
||||
const offset = property.offset;
|
||||
out += `static_assert(offsetof(${this.#name}, ${name}) == ${toHex(offset)}, "Invalid Offset");\n`;
|
||||
for (const property of this.#staticProperties) {
|
||||
// Static Property References
|
||||
out += `${INDENT}static ${property.referenceDefinition(false)};\n`;
|
||||
}
|
||||
|
||||
// Sanity Check Size
|
||||
if (this.#size !== null) {
|
||||
out += assertSize(this.#name, this.#size);
|
||||
}
|
||||
|
||||
// Allocation Function
|
||||
if (this.#size !== null) {
|
||||
out += `${this.#name} *alloc_${this.#name}();\n`;
|
||||
if (!this.#isSimple) {
|
||||
// Disable Construction Of Complex Structures
|
||||
out += preventConstruction(this.name);
|
||||
}
|
||||
out += '};\n';
|
||||
|
||||
// Return
|
||||
return out;
|
||||
@ -246,39 +246,40 @@ export class Struct {
|
||||
|
||||
// Generate Compiled Code
|
||||
generateCode() {
|
||||
let declarations = '';
|
||||
let init = '';
|
||||
let out = '';
|
||||
|
||||
// Static Properties
|
||||
for (const property of this.#staticProperties) {
|
||||
init += `${INDENT}${property.fullName()}_pointer = (${property.type()} *) ${toHex(property.offset)};\n`;
|
||||
declarations += property.globalPointerDefinition();
|
||||
out += `${property.referenceDefinition(true)} = *(std::remove_reference_t<decltype(${property.prettyName()})> *) ${toHex(property.offset)};\n`;
|
||||
}
|
||||
|
||||
// Methods
|
||||
for (const method of this.#methods) {
|
||||
if (!method.isInherited) {
|
||||
init += `${INDENT}${method.getName()} = (${method.getType()}) ${toHex(method.address)};\n`;
|
||||
declarations += method.generateDefinition();
|
||||
out += method.generate(true, false);
|
||||
}
|
||||
}
|
||||
|
||||
// Sanity Check Offsets
|
||||
for (const property of this.#properties) {
|
||||
const name = property.name();
|
||||
const offset = property.offset;
|
||||
out += `static_assert(offsetof(${this.name}, ${name}) == ${toHex(offset)}, "Invalid Offset");\n`;
|
||||
}
|
||||
|
||||
// Sanity Check Size
|
||||
if (this.#size !== null) {
|
||||
out += assertSize(this.name, this.#size);
|
||||
}
|
||||
|
||||
// VTable
|
||||
if (this.#vtable !== null) {
|
||||
const vtable = this.#vtable.generateCode();
|
||||
declarations += vtable.declarations;
|
||||
init += vtable.init;
|
||||
}
|
||||
|
||||
// Allocation Function
|
||||
if (this.#size !== null) {
|
||||
declarations += `${this.#name} *alloc_${this.#name}() {\n`;
|
||||
declarations += `${INDENT}return new ${this.#name};\n`;
|
||||
declarations += '}\n';
|
||||
const vtable = this.#vtable.generateCode(this.#directParent);
|
||||
out += vtable;
|
||||
}
|
||||
|
||||
// Return
|
||||
return {functions: declarations, init};
|
||||
return out;
|
||||
}
|
||||
|
||||
// Set Direct Parent (Used For "New Method" Testing)
|
||||
|
44
src/test/formatting.ts
Normal file
44
src/test/formatting.ts
Normal file
@ -0,0 +1,44 @@
|
||||
import { describe, it } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import { formatType, toHex, toUpperSnakeCase } from '../common';
|
||||
|
||||
describe('Upper-Snake-Case', () => {
|
||||
const tests: [string, string, string][] = [
|
||||
['One Word', 'Hello', 'HELLO'],
|
||||
['Two Words', 'HelloWorld', 'HELLO_WORLD'],
|
||||
['Empty', '', ''],
|
||||
['All Uppercase', 'HELLO', 'HELLO'],
|
||||
['All Lowercase', 'hello', 'HELLO']
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test[0], () => {
|
||||
assert.strictEqual(toUpperSnakeCase(test[1]), test[2]);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Formatting Types For Concatenation', () => {
|
||||
const tests: [string, string, string][] = [
|
||||
['Value', 'int', 'int '],
|
||||
['Pointer', 'int *', 'int *'],
|
||||
['Reference', 'int &', 'int &']
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test[0], () => {
|
||||
assert.strictEqual(formatType(test[1]), test[2]);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Hexadecimal', () => {
|
||||
const tests: [string, number, string][] = [
|
||||
['Zero', 0, '0x0'],
|
||||
['Small Number', 10, '0xa'],
|
||||
['Large Number', 100, '0x64']
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test[0], () => {
|
||||
assert.strictEqual(toHex(test[1]), test[2]);
|
||||
});
|
||||
}
|
||||
});
|
166
src/test/parsing.ts
Normal file
166
src/test/parsing.ts
Normal file
@ -0,0 +1,166 @@
|
||||
import { describe, it } from 'node:test';
|
||||
import assert from 'node:assert/strict';
|
||||
import { extractArrayInfo, forwardArguments, parseTypeAndName, prependArg, safeParseInt } from '../common';
|
||||
import { parseMethod, parseProperty } from '../loader';
|
||||
|
||||
describe('Parsing Variable Declarations', () => {
|
||||
const tests: { name: string, input: string, out: { type: string, name: string } }[] = [
|
||||
{name: 'Simple', input: 'int x', out: {type: 'int', name: 'x'}},
|
||||
{name: 'Pointer', input: 'int *y', out: {type: 'int *', name: 'y'}},
|
||||
{name: 'Extra Space', input: ' float a ', out: {type: 'float', name: 'a'}},
|
||||
{name: 'Ugly Pointer', input: 'int* z', out: {type: 'int*', name: 'z'}},
|
||||
{name: 'Double Pointer', input: 'int **a', out: {type: 'int **', name: 'a'}},
|
||||
{name: 'Ugly Double Pointer', input: 'int* *b', out: {type: 'int**', name: 'b'}},
|
||||
{name: 'Reference', input: 'int &c', out: {type: 'int &', name: 'c'}},
|
||||
{name: 'Reference-To-Pointer', input: 'int *&d', out: {type: 'int *&', name: 'd'}}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
const obj = parseTypeAndName(test.input);
|
||||
assert.strictEqual(obj.type, test.out.type);
|
||||
assert.strictEqual(obj.name, test.out.name);
|
||||
});
|
||||
}
|
||||
it('Invalid', () => {
|
||||
assert.throws(() => {
|
||||
parseTypeAndName('abc');
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Prepending Arguments To Functions', () => {
|
||||
const tests: { name: string, input: string, arg: string, out: string }[] = [
|
||||
{name: 'Empty', input: '()', arg: 'int x', out: '(int x)'},
|
||||
{name: 'Non-Empty', input: '(int x)', arg: 'int y', out: '(int y, int x)'}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
assert.strictEqual(prependArg(test.input, test.arg), test.out);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Parsing Integers', () => {
|
||||
const tests: { name: string, input: string, out?: number }[] = [
|
||||
{name: 'Zero', input: '0', out: 0},
|
||||
{name: 'Negative', input: '-5', out: -5},
|
||||
{name: 'Positive', input: '5', out: 5},
|
||||
{name: 'Extra Space', input: ' 5 ', out: 5},
|
||||
{name: 'Hexadecimal', input: '0x10', out: 16},
|
||||
{name: 'Empty', input: ''},
|
||||
{name: 'Text', input: 'abc'}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
if (test.out !== undefined) {
|
||||
assert.strictEqual(safeParseInt(test.input), test.out);
|
||||
} else {
|
||||
assert.throws(() => {
|
||||
safeParseInt(test.input);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Parsing Properties', () => {
|
||||
const tests: { name: string, input: string, out?: { type: string, name: string, offset: number } }[] = [
|
||||
{name: 'Basic', input: 'int x = 0x0', out: {type: 'int', name: 'x', offset: 0}},
|
||||
{name: 'Extra Space', input: 'int z = 0x0', out: {type: 'int', name: 'z', offset: 0}},
|
||||
{name: 'Pointer', input: 'float *y = 0x4', out: {type: 'float *', name: 'y', offset: 4}},
|
||||
{name: 'Empty', input: ''},
|
||||
{name: 'Malformed #1', input: 'int x'},
|
||||
{name: 'Malformed #2', input: 'int x=2'},
|
||||
{name: 'Malformed #3', input: 'int x = '}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
if (test.out) {
|
||||
const property = parseProperty(test.input);
|
||||
assert.strictEqual(property.type, test.out.type);
|
||||
assert.strictEqual(property.name, test.out.name);
|
||||
assert.strictEqual(property.offset, test.out.offset);
|
||||
} else {
|
||||
assert.throws(() => {
|
||||
parseProperty(test.input);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Parsing Methods', () => {
|
||||
const tests: { name: string, input: string, self: string, isStatic: boolean, out?: { name: string, returnType: string, args: string, address: number } }[] = [
|
||||
{name: 'Basic', input: 'void func() = 0x10', self: 'Test', isStatic: false, out: {name: 'func', returnType: 'void', args: '(Test *self)', address: 16}},
|
||||
{name: 'Advanced', input: 'int bar(int x, float y, std::vector<float> *arr) = 0x20', self: 'Foo', isStatic: false, out: {name: 'bar', returnType: 'int', args: '(Foo *self, int x, float y, std::vector<float> *arr)', address: 32}},
|
||||
{name: 'Extra Space', input: 'int bar (int x) = 0x20', self: 'Foo', isStatic: false, out: {name: 'bar', returnType: 'int', args: '(Foo *self, int x)', address: 32}},
|
||||
{name: 'Static', input: 'int thing(float x) = 0x30', self: 'Crazy', isStatic: true, out: {name: 'thing', returnType: 'int', args: '(float x)', address: 48}},
|
||||
{name: 'Empty', input: '', self: 'Test', isStatic: false},
|
||||
{name: 'Malformed #1', input: 'int broken', self: 'Test', isStatic: false},
|
||||
{name: 'Malformed #2', input: 'int broken = 0x10', self: 'Test', isStatic: false},
|
||||
{name: 'Malformed #3', input: 'int broken()', self: 'Test', isStatic: false},
|
||||
{name: 'Malformed #4', input: 'int broken()=0x0', self: 'Test', isStatic: false},
|
||||
{name: 'Malformed #5', input: 'int broken() = ', self: 'Test', isStatic: false},
|
||||
{name: 'Malformed #6', input: 'abc', self: 'Test', isStatic: false}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
if (test.out) {
|
||||
const method = parseMethod(test.input, test.self, !test.isStatic, false);
|
||||
assert.strictEqual(method.shortName, test.out.name);
|
||||
assert.strictEqual(method.returnType, test.out.returnType);
|
||||
assert.strictEqual(method.getArgs(), test.out.args);
|
||||
assert.strictEqual(method.address, test.out.address);
|
||||
} else {
|
||||
assert.throws(() => {
|
||||
parseMethod(test.input, test.self, !test.isStatic, false);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Extracting Array Information', () => {
|
||||
const tests: { name: string, input: string, out: { name: string, arrayInfo: string } }[] = [
|
||||
{name: 'Basic', input: 'x', out: {name: 'x', arrayInfo: ''}},
|
||||
{name: 'Array', input: 'x[10]', out: {name: 'x', arrayInfo: '[10]'}},
|
||||
{name: 'Multi-Dimensional Array', input: 'x[10][10]', out: {name: 'x', arrayInfo: '[10][10]'}}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
const out = extractArrayInfo(test.input);
|
||||
assert.strictEqual(out.name, test.out.name);
|
||||
assert.strictEqual(out.arrayInfo, test.out.arrayInfo);
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe('Forwarding Arguments', () => {
|
||||
const tests: { name: string, input: string, extra?: string[], out?: string }[] = [
|
||||
{name: 'No Arguments', input: '()', out: '()'},
|
||||
{name: 'One Argument', input: '(int x)', out: '(x)'},
|
||||
{name: 'Two Argument', input: '(int x, float y)', out: '(x, y)'},
|
||||
{name: 'Arrays', input: '(int one[10], float two[20])', out: '(one, two)'},
|
||||
{name: 'Generics #1', input: '(Obj<float> x, Obj2<int, int[10]> *y)', out: '(x, y)'},
|
||||
{name: 'Generics #2', input: '(function<int(int, int)> &callback, void *data)', out: '(callback, data)'},
|
||||
{name: 'Extra Arguments', input: '(int x)', extra: ['this'], out: '(this, x)'},
|
||||
{name: 'Malformed #1', input: '(int)'},
|
||||
{name: 'Malformed #2', input: '('},
|
||||
{name: 'Malformed #3', input: ')'},
|
||||
{name: 'Malformed #4', input: '(Obj<float x, int y)'},
|
||||
{name: 'Malformed #5', input: '(int x, float y[5)'},
|
||||
{name: 'Malformed #6', input: '(Obj<5] x)'}
|
||||
];
|
||||
for (const test of tests) {
|
||||
it(test.name, () => {
|
||||
if (test.out) {
|
||||
const out = forwardArguments(test.input, test.extra ?? []);
|
||||
assert.strictEqual(out, test.out);
|
||||
} else {
|
||||
assert.throws(() => {
|
||||
forwardArguments(test.input, test.extra ?? []);
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
138
src/vtable.ts
138
src/vtable.ts
@ -1,14 +1,17 @@
|
||||
import { INDENT, POINTER_SIZE, RTTI_SIZE, assertSize, getSelfArg, toHex } from './common';
|
||||
import { BUILDING_SYMBOLS_GUARD, INDENT, INTERNAL, POINTER_SIZE, assertSize, preventConstruction, toHex } from './common';
|
||||
import { Method } from './method';
|
||||
import { Property } from './property';
|
||||
|
||||
// A VTable
|
||||
const structuresWithVTableAddress: string[] = [];
|
||||
export class VTable {
|
||||
readonly #self: string;
|
||||
#address: number | null;
|
||||
#size: number | null;
|
||||
readonly #methods: Method[];
|
||||
readonly #methods: (Method | undefined)[];
|
||||
readonly property: Property;
|
||||
#destructorOffset: number;
|
||||
readonly #destructors: Method[];
|
||||
|
||||
// Constructor
|
||||
constructor(self: string) {
|
||||
@ -17,6 +20,7 @@ export class VTable {
|
||||
this.#size = null;
|
||||
this.#methods = [];
|
||||
this.#destructorOffset = 0;
|
||||
this.#destructors = [];
|
||||
// Create Property
|
||||
this.property = new Property(0, this.#getName() + ' *', 'vtable', this.#self);
|
||||
}
|
||||
@ -24,6 +28,7 @@ export class VTable {
|
||||
// Setters
|
||||
setAddress(address: number) {
|
||||
this.#address = address;
|
||||
structuresWithVTableAddress.push(this.#self);
|
||||
}
|
||||
setSize(size: number) {
|
||||
this.#size = size;
|
||||
@ -33,7 +38,7 @@ export class VTable {
|
||||
}
|
||||
|
||||
// Add To VTable
|
||||
#add(target: Method[], method: Method) {
|
||||
#add(target: (Method | undefined)[], method: Method) {
|
||||
// Check Offset
|
||||
const offset = method.address;
|
||||
if ((offset % POINTER_SIZE) !== 0) {
|
||||
@ -77,77 +82,93 @@ export class VTable {
|
||||
out.push(...this.#methods);
|
||||
// Add Destructors (https://stackoverflow.com/a/17960941)
|
||||
const destructor_return = `${this.#self} *`;
|
||||
const destructor_args = `(${getSelfArg(this.#self)})`;
|
||||
this.#add(out, new Method(this.#self, 'destructor_complete', destructor_return, destructor_args, 0x0 + this.#destructorOffset, false, false));
|
||||
this.#add(out, new Method(this.#self, 'destructor_deleting', destructor_return, destructor_args, 0x4 + this.#destructorOffset, false, false));
|
||||
const destructor_args = '()';
|
||||
if (this.#destructors.length === 0) {
|
||||
this.#destructors.push(
|
||||
new Method(this.#self, 'destructor_complete', destructor_return, destructor_args, 0x0 + this.#destructorOffset, false, false),
|
||||
new Method(this.#self, 'destructor_deleting', destructor_return, destructor_args, 0x4 + this.#destructorOffset, false, false)
|
||||
);
|
||||
}
|
||||
for (const destructor of this.#destructors) {
|
||||
this.#add(out, destructor);
|
||||
}
|
||||
// Return
|
||||
return out;
|
||||
}
|
||||
|
||||
// Get Parent Method
|
||||
#getParentSelf(method: Method, parent: string | null) {
|
||||
if (method.isInherited && parent) {
|
||||
// Parent Exists
|
||||
let out: string;
|
||||
if (structuresWithVTableAddress.includes(parent)) {
|
||||
out = parent;
|
||||
} else {
|
||||
// Unable To Determine
|
||||
out = this.#self;
|
||||
}
|
||||
return out;
|
||||
} else {
|
||||
// Method Has No Parent
|
||||
return undefined;
|
||||
}
|
||||
}
|
||||
|
||||
// Generate Header Code
|
||||
generate(directParent: string | null) {
|
||||
canGenerateWrappers() {
|
||||
return this.#address !== null;
|
||||
}
|
||||
generate() {
|
||||
let out = '';
|
||||
|
||||
// Check
|
||||
this.#check();
|
||||
|
||||
// Method Prototypes
|
||||
// Wrappers
|
||||
const methods = this.getMethods();
|
||||
for (let i = 0; i < methods.length; i++) {
|
||||
const info = methods[i];
|
||||
let typedefs = '';
|
||||
let methodsOut = '';
|
||||
for (const info of methods) {
|
||||
if (info) {
|
||||
out += info.generateTypedef();
|
||||
typedefs += info.generateTypedefs();
|
||||
if (this.canGenerateWrappers()) {
|
||||
methodsOut += info.generate(false, true);
|
||||
}
|
||||
}
|
||||
}
|
||||
out += typedefs;
|
||||
out += `#ifndef ${BUILDING_SYMBOLS_GUARD}\n`;
|
||||
out += methodsOut;
|
||||
out += '#endif\n';
|
||||
|
||||
// Structure
|
||||
out += `typedef struct ${this.#getName()} ${this.#getName()};\n`;
|
||||
out += `struct ${this.#getName()} {\n`;
|
||||
out += `struct ${this.#getName()} final {\n`;
|
||||
for (let i = 0; i < methods.length; i++) {
|
||||
let name = `unknown${i}`;
|
||||
let type = 'void *';
|
||||
const info = methods[i];
|
||||
if (info) {
|
||||
name = info.shortName;
|
||||
type = info.getType() + ' ';
|
||||
out += info.getProperty();
|
||||
} else {
|
||||
out += `${INDENT}void *${INTERNAL}unknown${i.toString()};\n`;
|
||||
}
|
||||
out += `${INDENT}${type}${name};\n`;
|
||||
}
|
||||
out += `};\n`;
|
||||
|
||||
// Sanity Check Size
|
||||
if (this.#size !== null) {
|
||||
out += assertSize(this.#getName(), this.#size);
|
||||
if (this.#size === null) {
|
||||
// Prevent Construction
|
||||
out += preventConstruction(this.#getName());
|
||||
}
|
||||
|
||||
// Pointers
|
||||
if (this.#address !== null) {
|
||||
// Base
|
||||
out += `extern ${this.#getName()} *${this.#getName()}_base;\n`;
|
||||
// Methods
|
||||
for (let i = 0; i < methods.length; i++) {
|
||||
const info = methods[i];
|
||||
if (info) {
|
||||
const type = `${info.getType()} *`;
|
||||
out += `extern ${type}${info.getName()}_vtable_addr;\n`;
|
||||
out += info.generateNewMethodTest(directParent, '*', '_vtable_addr');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Duplication Method
|
||||
if (this.#size !== null) {
|
||||
out += `${this.#getName()} *dup_${this.#getName()}(${this.#getName()} *vtable);\n`;
|
||||
out += `${INDENT}static ${this.#getName()} *base;\n`;
|
||||
}
|
||||
out += '};\n';
|
||||
|
||||
// Return
|
||||
return out;
|
||||
}
|
||||
|
||||
// Generate Compiled Code
|
||||
generateCode() {
|
||||
let declarations = '';
|
||||
let init = '';
|
||||
generateCode(directParent: string | null) {
|
||||
let out = '';
|
||||
|
||||
// Check
|
||||
this.#check();
|
||||
@ -155,38 +176,25 @@ export class VTable {
|
||||
// Pointers
|
||||
if (this.#address !== null) {
|
||||
// Base
|
||||
init += `${INDENT}${this.#getName()}_base = (${this.#getName()} *) ${toHex(this.#address)};\n`;
|
||||
declarations += `${this.#getName()} *${this.#getName()}_base;\n`;
|
||||
// Methods
|
||||
out += `${this.#getName()} *${this.#getName()}::base = (${this.#getName()} *) ${toHex(this.#address)};\n`;
|
||||
}
|
||||
|
||||
// Method Wrappers
|
||||
if (this.canGenerateWrappers()) {
|
||||
const methods = this.getMethods();
|
||||
for (let i = 0; i < methods.length; i++) {
|
||||
const info = methods[i];
|
||||
for (const info of methods) {
|
||||
if (info) {
|
||||
const vtableAddress = this.#address + (i * POINTER_SIZE);
|
||||
const type = `${info.getType()} *`;
|
||||
init += `${INDENT}${info.getName()}_vtable_addr = (${type}) ${toHex(vtableAddress)};\n`;
|
||||
declarations += `${type}${info.getName()}_vtable_addr;\n`;
|
||||
out += info.generate(true, true, this.#getParentSelf(info, directParent));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Duplication Method
|
||||
// Sanity Check Size
|
||||
if (this.#size !== null) {
|
||||
declarations += `${this.#getName()} *dup_${this.#getName()}(${this.#getName()} *vtable) {\n`;
|
||||
declarations += `${INDENT}uchar *real_vtable = (uchar *) vtable;\n`;
|
||||
declarations += `${INDENT}real_vtable -= ${RTTI_SIZE};\n`;
|
||||
declarations += `${INDENT}size_t real_vtable_size = sizeof(${this.#getName()}) + ${RTTI_SIZE};\n`;
|
||||
declarations += `${INDENT}uchar *new_vtable = (uchar *) ::operator new(real_vtable_size);\n`;
|
||||
declarations += `${INDENT}if (new_vtable == NULL) {\n`;
|
||||
declarations += `${INDENT}${INDENT}return NULL;\n`;
|
||||
declarations += `${INDENT}}\n`;
|
||||
declarations += `${INDENT}memcpy((void *) new_vtable, (void *) real_vtable, real_vtable_size);\n`;
|
||||
declarations += `${INDENT}new_vtable += ${RTTI_SIZE};\n`;
|
||||
declarations += `${INDENT}return (${this.#getName()} *) new_vtable;\n`;
|
||||
declarations += '}\n';
|
||||
out += assertSize(this.#getName(), this.#size);
|
||||
}
|
||||
|
||||
// Return
|
||||
return {declarations, init};
|
||||
return out;
|
||||
}
|
||||
}
|
@ -13,8 +13,10 @@
|
||||
<item>static-method</item>
|
||||
<item>constructor</item>
|
||||
<item>vtable-destructor-offset</item>
|
||||
<item>mark-as-simple</item>
|
||||
</list>
|
||||
<list name="types">
|
||||
<item>const</item>
|
||||
<item>char</item>
|
||||
<item>uchar</item>
|
||||
<item>short</item>
|
@ -2,16 +2,20 @@ syntax def "\.def$"
|
||||
comment "//"
|
||||
|
||||
# Commands
|
||||
color magenta "\<(extends|size|vtable(-size|-destructor-offset)?|property|static-property|((static|virtual)-)?method|constructor)\>"
|
||||
color magenta "\<(extends|size|vtable(-size|-destructor-offset)?|property|static-property|((static|virtual)-)?method|constructor|mark-as-simple)\>"
|
||||
|
||||
# Types
|
||||
color green "\<(char|uchar|short|ushort|int|uint|float|bool|void|std::(string|vector|map))\>"
|
||||
color green "\<((u?(char|short|int))|float|bool|void|std::(string|vector|map))\>"
|
||||
|
||||
# Numbers
|
||||
color yellow "0x[a-f0-9]+"
|
||||
# Non-hex numbers
|
||||
color red " [0-9][a-f0-9]+;"
|
||||
|
||||
# Comments
|
||||
color brightblue "//.*"
|
||||
|
||||
# Whitespace.
|
||||
color normal "[[:space:]]+"
|
||||
# Trailing whitespace.
|
||||
color ,green "[[:space:]]+$"
|
@ -0,0 +1,24 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||
<plist version="1.0">
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>Comments</string>
|
||||
<key>scope</key>
|
||||
<string>source.toml</string>
|
||||
<key>settings</key>
|
||||
<dict>
|
||||
<key>shellVariables</key>
|
||||
<array>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>TM_COMMENT_START</string>
|
||||
<key>value</key>
|
||||
<string>//</string>
|
||||
</dict>
|
||||
</array>
|
||||
</dict>
|
||||
<key>uuid</key>
|
||||
<string>C5C885D7-2733-4632-B709-B5B9DD518F90</string>
|
||||
</dict>
|
||||
</plist>
|
@ -0,0 +1,55 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||
<plist version="1.0">
|
||||
<dict>
|
||||
<key>fileTypes</key>
|
||||
<array>
|
||||
<string>def</string>
|
||||
</array>
|
||||
<key>scopeName</key>
|
||||
<string>source.symbol-processor</string>
|
||||
<key>name</key>
|
||||
<string>Symbol Processor Definition</string>
|
||||
<key>patterns</key>
|
||||
<array>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>comment.line.double-slash.symbol-processor</string>
|
||||
<key>match</key>
|
||||
<string>//.*$</string>
|
||||
</dict>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>constant.numeric.symbol-processor.hex</string>
|
||||
<key>match</key>
|
||||
<string>\b0[xX][0-9a-fA-F]+</string>
|
||||
</dict>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>constant.numeric.symbol-processor.decimal</string>
|
||||
<key>match</key>
|
||||
<string>\b[0-9]+</string>
|
||||
</dict>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>keyword.control.symbol-processor</string>
|
||||
<key>match</key>
|
||||
<string>\b(extends|size|vtable-size|vtable-destructor-offset|vtable|property|static-property|method|virtual-method|static-method|constructor|mark-as-simple)\b</string>
|
||||
</dict>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>storage.type.symbol-processor</string>
|
||||
<key>match</key>
|
||||
<string>\b(const|char|uchar|short|ushort|int|uint|float|bool|void|std::string|std::vector|std::map|unsigned|long)\b</string>
|
||||
</dict>
|
||||
<dict>
|
||||
<key>name</key>
|
||||
<string>keyword.operator.symbol-processor</string>
|
||||
<key>match</key>
|
||||
<string>(=|;)</string>
|
||||
</dict>
|
||||
</array>
|
||||
<key>uuid</key>
|
||||
<string>D44198D4-5AEB-40E5-B4E4-0E11C69FFA42</string>
|
||||
</dict>
|
||||
</plist>
|
@ -0,0 +1,16 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||
<plist version="1.0">
|
||||
<dict>
|
||||
<key>contactEmailRot13</key>
|
||||
<string>connor24nolan@live.com</string>
|
||||
<key>contactName</key>
|
||||
<string>TheBrokenRail</string>
|
||||
<key>description</key>
|
||||
<string>Symbol Processor Definition File</string>
|
||||
<key>name</key>
|
||||
<string>Symbol Processor</string>
|
||||
<key>uuid</key>
|
||||
<string>8209EEB8-4193-4E63-BDBB-0407E47ADF50</string>
|
||||
</dict>
|
||||
</plist>
|
@ -1,11 +1,11 @@
|
||||
{
|
||||
"extends": [
|
||||
"@tsconfig/strictest/tsconfig",
|
||||
"@tsconfig/node-lts/tsconfig"
|
||||
"@tsconfig/node18/tsconfig"
|
||||
],
|
||||
"compilerOptions": {
|
||||
"rootDir": "./src",
|
||||
"outDir": "./build"
|
||||
"outDir": "build",
|
||||
"rootDir": "src"
|
||||
},
|
||||
"include": [
|
||||
"src"
|
||||
|
Loading…
x
Reference in New Issue
Block a user