Files
Michael Fabian 'Xaymar' Dirks c4947bd12a compiler: Move and modernize
compiler is blitzcc, what I previously called compiler is now compiler_lib
2019-01-19 18:28:07 +01:00

338 lines
7.9 KiB
C++

#pragma warning(disable : 4786)
#include "libs.hpp"
#include <fstream>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <set>
#include <string>
#include <vector>
#include <assem_x86/assem_x86.hpp>
#include <bbruntime_dll.hpp>
#include <codegen_x86/codegen_x86.hpp>
#include <config.hpp>
#include <environ.hpp>
#include <ex.hpp>
#include <linker.hpp>
#include <parser.hpp>
#include <stdutil.hpp>
static void showInfo()
{
const int major = (VERSION & 0xffff) / 100, minor = (VERSION & 0xffff) % 100;
std::cout << "BlitzCC V" << major << "." << minor << std::endl;
std::cout << "(C)opyright 2000-2003 Blitz Research Ltd" << std::endl;
}
static void showUsage()
{
std::cout << "Usage: blitzcc [-h|-q|+q|-c|-d|-k|+k|-v|-o exefile] [sourcefile.bb]" << std::endl;
}
static void showHelp()
{
showUsage();
std::cout << "-h : show this help" << std::endl;
std::cout << "-q : quiet mode" << std::endl;
std::cout << "+q : very quiet mode" << std::endl;
std::cout << "-c : compile only" << std::endl;
std::cout << "-d : debug compile" << std::endl;
std::cout << "-k : dump keywords" << std::endl;
std::cout << "+k : dump keywords and syntax" << std::endl;
std::cout << "-v : version info" << std::endl;
std::cout << "-o exefile : generate executable" << std::endl;
}
static void err(const std::string& t)
{
throw std::runtime_error(t);
}
static void usageErr()
{
err("Usage error");
}
static std::string quickHelp(const std::string& kw)
{
Environ* e = runtimeEnviron;
Decl* d = e->funcDecls->findDecl(tolower(kw));
if (!d || d->type->funcType() == 0)
return "No quick help available for " + kw;
std::string t = kw;
FuncType* f = d->type->funcType();
if (f->returnType == Type::float_type)
t += '#';
else if (f->returnType == Type::string_type)
t += '$';
t += " ";
if (f->returnType != Type::void_type)
t += "( ";
for (int k = 0; k < f->params->size(); ++k) {
std::string s;
if (k)
s += ',';
Decl* p = f->params->decls[k];
s += p->name;
if (p->type == Type::float_type)
s += '#';
else if (p->type == Type::string_type)
s += '$';
else if (p->type == Type::void_type)
s += '*';
if (p->defType)
s = '[' + s + ']';
t += s;
}
if (f->returnType != Type::void_type) {
t += f->params->size() ? " )" : ")";
}
return t;
}
static void dumpKeys(bool lang, bool mod, bool help)
{
if (lang) {
std::map<std::string, int>::iterator it;
std::map<std::string, int>& keywords = Toker::getKeywords();
for (it = keywords.begin(); it != keywords.end(); ++it) {
if (it->first.find(' ') != std::string::npos)
continue;
std::cout << it->first << std::endl;
}
}
if (!mod)
return;
for (int k = 0; k < keyWords.size(); ++k) {
std::string t = keyWords[k];
if (t[0] == '_')
continue;
if (!isalpha(t[0]))
t = t.substr(1);
for (int n = 0; n < t.size(); ++n) {
if (!isalnum(t[n]) && t[n] != '_') {
t = t.substr(0, n);
break;
}
}
if (help)
t = quickHelp(t);
std::cout << t << std::endl;
}
}
static std::string verstr(int ver)
{
return itoa((ver & 65535) / 100) + "." + itoa((ver & 65535) % 100);
}
static void versInfo()
{
std::cout << "Compiler version:" << verstr(bcc_ver) << std::endl;
std::cout << "Runtime version:" << verstr(run_ver) << std::endl;
std::cout << "Debugger version:" << verstr(dbg_ver) << std::endl;
std::cout << "Linker version:" << verstr(lnk_ver) << std::endl;
}
int main(int argc, char* argv[])
{
std::shared_ptr<Module> module;
std::shared_ptr<Environ> v_environ;
std::shared_ptr<ProgNode> prog;
try {
std::string in_file, out_file, args;
bool debug = false, quiet = false, veryquiet = false, compileonly = false;
bool dumpkeys = false, dumphelp = false, showhelp = false, dumpasm = false;
bool versinfo = false;
for (int k = 1; k < argc; ++k) {
std::string t = argv[k];
t = tolower(t);
if (t == "-h") {
showhelp = true;
} else if (t == "-a") {
dumpasm = true;
} else if (t == "-q") {
quiet = true;
} else if (t == "+q") {
quiet = veryquiet = true;
} else if (t == "-c") {
compileonly = true;
} else if (t == "-d") {
debug = true;
} else if (t == "-k") {
dumpkeys = true;
} else if (t == "+k") {
dumpkeys = dumphelp = true;
} else if (t == "-v") {
versinfo = true;
} else if (t == "-o") {
if (out_file.size() || k == argc - 1)
usageErr();
out_file = argv[++k];
} else {
if (in_file.size() || t[0] == '-' || t[0] == '+')
usageErr();
in_file = argv[k];
for (++k; k < argc; ++k) {
std::string t = argv[k];
if (t.find(' ') != std::string::npos)
t = '\"' + t + '\"';
if (args.size())
args += ' ';
args += t;
}
}
}
if (out_file.size() && !in_file.size())
usageErr();
if (const char* er = openLibs())
err(er);
if (const char* er = linkLibs())
err(er);
if (showhelp)
showHelp();
if (dumpkeys)
dumpKeys(true, true, dumphelp);
if (versinfo)
versInfo();
if (!in_file.size()) {
return 0;
}
if (in_file[0] == '\"') {
if ((in_file.size() < 3) || (in_file[in_file.size() - 1] != '\"')) {
usageErr();
}
in_file = in_file.substr(1, in_file.size() - 2);
}
std::ifstream in(in_file.c_str());
if (!in)
err("Unable to open input file");
if (!quiet) {
showInfo();
std::cout << "Compiling \"" << in_file << "\"" << std::endl;
}
int n = in_file.rfind('/');
if (n == std::string::npos)
n = in_file.rfind('\\');
if (n != std::string::npos) {
if (!n || in_file[n - 1] == ':')
++n;
SetCurrentDirectory(in_file.substr(0, n).c_str());
}
try {
//parse
if (!veryquiet)
std::cout << "Parsing..." << std::endl;
Toker toker(in);
Parser parser(toker);
prog = parser.parse(in_file);
//semant
if (!veryquiet)
std::cout << "Generating..." << std::endl;
v_environ = prog->semant(runtimeEnviron);
//translate
if (!veryquiet)
std::cout << "Translating..." << std::endl;
qstreambuf qbuf;
std::iostream asmcode(&qbuf);
Codegen_x86 codegen(asmcode, debug);
prog->translate(&codegen, userFuncs);
if (dumpasm) {
std::cout << std::endl << std::string(qbuf.data(), qbuf.size()) << std::endl;
}
//assemble
if (!veryquiet)
std::cout << "Assembling..." << std::endl;
module = linkerLib->createModule();
Assem_x86 assem(asmcode, module);
assem.assemble();
} catch (Ex& x) {
std::string file = '\"' + x.file + '\"';
int row = ((x.pos >> 16) & 65535) + 1, col = (x.pos & 65535) + 1;
std::cout << file << ":" << row << ":" << col << ":" << row << ":" << col << ":" << x.ex << std::endl;
throw std::runtime_error(file);
}
delete prog;
if (out_file.size()) {
if (!veryquiet)
std::cout << "Creating executable \"" << out_file << "\"..." << std::endl;
if (!module->createExe(out_file.c_str(), (home + "/bin/runtime.dll").c_str())) {
err("Error creating executable");
}
} else if (!compileonly) {
void* entry = module->link(runtimeModule);
if (!entry)
return 0;
HMODULE dbgHandle = 0;
/*std::shared_ptr<Debugger> debugger;
if (debug) {
dbgHandle = LoadLibrary((home + "\\debugger.dll").c_str());
if (dbgHandle) {
typedef Debugger*(_cdecl * GetDebugger)(Module*, Environ*);
GetDebugger gd = (GetDebugger)GetProcAddress(dbgHandle, "debuggerGetDebugger");
if (gd)
debugger = gd(module, v_environ);
}
if (!debugger)
err("Error launching debugger");
}*/
if (!veryquiet)
std::cout << "Executing..." << std::endl;
runtimeLib->execute((void (*)())entry, args.c_str(), nullptr);
if (dbgHandle)
FreeLibrary(dbgHandle);
}
delete module;
delete v_environ;
closeLibs();
} catch (std::exception& e) {
std::cout << "Unexpected exception: " << e.what() << std::endl;
} catch (...) {
std::cout << "Unexpected error." << std::endl;
}
std::cin.get();
return 0;
}