void addGlobalMapping(const GlobalValue *GV, void *Addr) { addGlobalMapping(getMangledName(GV), (uintptr_t)Addr); }
bool JITRotation::initializeWithFile(const char* filename) { // note: string literals in rotation scripts are actually compiled as their FNV1A hash instead const char header[] = "class Actor;\n" "uint64 AuraCount(const Actor* actor, uint64 identifier, const Actor* source);\n" "double GlobalCooldownRemaining(const Actor* actor);\n" "double CooldownRemaining(const Actor* actor, uint64 identifier);\n" "double AuraTimeRemaining(const Actor* actor, uint64 identifier, const Actor* source);\n" "Actor* Pet(Actor* owner);\n" "uint64 TP(const Actor* actor);\n" "uint64 MP(const Actor* actor);\n" "uint64 MaximumMP(const Actor* actor);\n" "double GlobalCooldown(const Actor* actor);\n" "double Time(const Actor* actor);\n" "void RemoveAura(Actor* actor, uint64 identifier, const Actor* source);\n" "uint8 IsReady(const Actor* actor, uint64 identifier);\n" "void Command(Actor* actor, uint64 identifier);\n" "void StopAutoAttack(Actor* actor);\n" "__end __hidden const uint64 NextAction(Actor* self, const Actor* target) {\n" ; const char footer[] = "\n}"; FILE* f = fopen(filename, "r"); if (!f) { printf("Unable to open file %s\n", filename); return false; } fseek(f, 0, SEEK_END); auto fsize = ftell(f); rewind(f); auto sourceLength = sizeof(header) + fsize + sizeof(footer); auto source = (char*)malloc(sourceLength); if (!source) { printf("Unable to allocate memory for rotation source\n"); fclose(f); return false; } memcpy(source, header, sizeof(header)); if (fread(source + sizeof(header), fsize, 1, f) != 1) { printf("Unable to read file %s\n", filename); fclose(f); return false; } memcpy(source + sizeof(header) + fsize, footer, sizeof(footer)); fclose(f); // PREPROCESS Preprocessor pp; if (!pp.process_file(filename, source, sourceLength)) { printf("Couldn't preprocess file.\n"); free(source); return false; } free(source); // PARSE Parser p; ASTSequence* ast = p.generate_ast(pp.tokens()); if (p.errors().size() > 0) { delete ast; for (const ParseError& e : p.errors()) { printf("Error: %s\n", e.message.c_str()); e.token->print_pointer(); } return false; } if (!ast) { printf("Couldn't generate AST.\n"); return false; } // GENERATE CODE LLVMCodeGenerator cg; if (!cg.build_ir(ast)) { printf("Couldn't build IR.\n"); delete ast; return false; } delete ast; // COMPILE LLVMInitializeNativeTarget(); auto module = cg.release_module(); std::string error; auto engine = llvm::EngineBuilder(module).setErrorStr(&error).setEngineKind(llvm::EngineKind::JIT).create(); if (!engine) { printf("Couldn't build execution engine: %s\n", error.c_str()); return false; } engine->addGlobalMapping(module->getFunction("^AuraCount"), (void*)&JITRotation::ActorAuraCount); engine->addGlobalMapping(module->getFunction("^GlobalCooldownRemaining"), (void*)&JITRotation::ActorGlobalCooldownRemaining); engine->addGlobalMapping(module->getFunction("^CooldownRemaining"), (void*)&JITRotation::ActorCooldownRemaining); engine->addGlobalMapping(module->getFunction("^AuraTimeRemaining"), (void*)&JITRotation::ActorAuraTimeRemaining); engine->addGlobalMapping(module->getFunction("^Pet"), (void*)&JITRotation::ActorPet); engine->addGlobalMapping(module->getFunction("^TP"), (void*)&JITRotation::ActorTP); engine->addGlobalMapping(module->getFunction("^MP"), (void*)&JITRotation::ActorMP); engine->addGlobalMapping(module->getFunction("^MaximumMP"), (void*)&JITRotation::ActorMaximumMP); engine->addGlobalMapping(module->getFunction("^GlobalCooldown"), (void*)&JITRotation::ActorGlobalCooldown); engine->addGlobalMapping(module->getFunction("^Time"), (void*)&JITRotation::ActorTime); engine->addGlobalMapping(module->getFunction("^RemoveAura"), (void*)&JITRotation::ActorRemoveAura); engine->addGlobalMapping(module->getFunction("^IsReady"), (void*)&JITRotation::ActionIsReady); engine->addGlobalMapping(module->getFunction("^Command"), (void*)&JITRotation::ActorCommand); engine->addGlobalMapping(module->getFunction("^StopAutoAttack"), (void*)&JITRotation::ActorStopAutoAttack); _jitNextAction = decltype(_jitNextAction)((std::intptr_t)engine->getPointerToFunction(module->getFunction("^NextAction"))); return _jitNextAction; }