blob: bd97299cf43b6bd733fda04c038074035e78ae60 [file] [log] [blame] [raw]
// Copyright (c) 2015, Matt Godbolt
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// * Redistributions of source code must retain the above copyright notice,
// this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
const temp = require('temp'),
path = require('path'),
fs = require('fs-extra'),
LlvmIrParser = require('./llvm-ir'),
AsmParser = require('./asm-parser'),
utils = require('./utils'),
_ = require('underscore'),
packager = require('./packager').Packager,
exec = require('./exec'),
logger = require('./logger').logger,
compilerOptInfo = require("compiler-opt-info"),
argumentParsers = require("./compilers/argument-parsers"),
CompilerArguments = require("./compiler-arguments"),
cfg = require('./cfg'),
languages = require('./languages').list,
toolchainUtils = require('./toolchain-utils');
class BaseCompiler {
constructor(compilerInfo, env) {
// Information about our compiler
this.compiler = compilerInfo;
this.lang = languages[compilerInfo.lang];
if (!this.lang) {
throw new Error(`Missing language info for ${compilerInfo.lang}`);
}
this.compileFilename = `example${this.lang.extensions[0]}`;
this.env = env;
// Partial application of compilerProps with the proper language id applied to it
this.compilerProps = _.partial(this.env.compilerProps, this.lang.id);
this.compiler.supportsIntel = !!this.compiler.intelAsm;
this.alwaysResetLdPath = this.env.ceProps('alwaysResetLdPath');
if (!this.compiler.options) this.compiler.options = "";
if (!this.compiler.optArg) this.compiler.optArg = "";
if (!this.compiler.supportsOptOutput) this.compiler.supportsOptOutput = false;
if (!this.compiler.disabledFilters)
this.compiler.disabledFilters = [];
else if (typeof this.compiler.disabledFilters === "string")
this.compiler.disabledFilters = this.compiler.disabledFilters.split(',');
this.asm = new AsmParser(this.compilerProps);
this.llvmIr = new LlvmIrParser(this.compilerProps);
this.toolchainPath = toolchainUtils.getToolchainPath(this.compiler.exe, this.compiler.options);
this.possibleArguments = new CompilerArguments(this.compiler.id);
this.possibleTools = _.values(compilerInfo.tools);
const demanglerExe = this.compiler.demangler;
if (demanglerExe) {
this.demanglerClass = require(this.compiler.demanglerClassFile).Demangler;
}
this.outputFilebase = "output";
this.mtime = null;
this.packager = new packager();
if (compilerInfo.buildenvsetup && compilerInfo.buildenvsetup.id) {
const buildenvsetupclass = require(`./buildenvsetup/buildenvsetup-${compilerInfo.buildenvsetup.id}`);
this.buildenvsetup = new buildenvsetupclass(compilerInfo, env, async (compiler, args, options) => {
return this.execCompilerCached(compiler, args, options);
});
} else {
this.buildenvsetup = null;
}
}
newTempDir() {
return new Promise((resolve, reject) => {
temp.mkdir({prefix: 'compiler-explorer-compiler', dir: process.env.tmpDir}, (err, dirPath) => {
if (err)
reject(`Unable to open temp file: ${err}`);
else
resolve(dirPath);
});
});
}
optOutputRequested(options) {
return options.some(x => x === "-fsave-optimization-record");
}
getRemote() {
if (this.compiler.exe === null && this.compiler.remote)
return this.compiler.remote;
return false;
}
exec(compiler, args, options) {
// Here only so can be overridden by compiler implementations.
return exec.execute(compiler, args, options);
}
getCompilerCacheKey(compiler, args, options) {
return {mtime: this.mtime, compiler, args, options};
}
async execCompilerCached(compiler, args, options) {
const key = this.getCompilerCacheKey(compiler, args, options);
let result = await this.env.compilerCacheGet(key);
if (!result) {
result = await exec.execute(compiler, args, options);
if (result.okToCache)
this.env.compilerCachePut(key, result);
}
return result;
}
getDefaultExecOptions() {
return {
timeoutMs: this.env.ceProps("compileTimeoutMs", 7500),
maxErrorOutput: this.env.ceProps("max-error-output", 5000),
env: this.env.getEnv(this.compiler.needsMulti),
wrapper: this.compilerProps("compiler-wrapper")
};
}
async runCompiler(compiler, options, inputFilename, execOptions) {
if (!execOptions) {
execOptions = this.getDefaultExecOptions();
}
if (!execOptions.customCwd) {
execOptions.customCwd = path.dirname(inputFilename);
}
const result = await this.exec(compiler, options, execOptions);
result.inputFilename = inputFilename;
const transformedInput = result.filenameTransform(inputFilename);
result.stdout = utils.parseOutput(result.stdout, transformedInput);
result.stderr = utils.parseOutput(result.stderr, transformedInput);
return result;
}
supportsObjdump() {
return this.compiler.objdumper !== "";
}
async objdump(outputFilename, result, maxSize, intelAsm, demangle) {
const args = ["-d", outputFilename, "-l", "--insn-width=16"];
if (demangle) args.push("-C");
if (intelAsm) args.push("-M", "intel");
const execOptions = {maxOutput: maxSize, customCwd: path.dirname(outputFilename)};
const objResult = await this.exec(this.compiler.objdumper, args, execOptions);
result.asm = objResult.stdout;
if (objResult.code !== 0) {
result.asm = `<No output: objdump returned ${objResult.code}>`;
}
return result;
}
async execBinary(executable, maxSize, executeParameters) {
// We might want to save this in the compilation environment once execution is made available
const timeoutMs = this.env.ceProps('binaryExecTimeoutMs', 2000);
try {
// TODO make config
const execResult = await exec.sandbox(executable, executeParameters.args, {
maxOutput: maxSize,
timeoutMs: timeoutMs,
ldPath: _.union(this.compiler.ldPath, executeParameters.ldPath).join(":"),
input: executeParameters.stdin
});
execResult.stdout = utils.parseOutput(execResult.stdout);
execResult.stderr = utils.parseOutput(execResult.stderr);
return execResult;
} catch (err) {
// TODO: is this the best way? Perhaps failures in sandbox shouldn't reject
// with "results", but instead should play on?
return {
stdout: err.stdout ? utils.parseOutput(err.stdout) : [],
stderr: err.stderr ? utils.parseOutput(err.stderr) : [],
code: err.code !== undefined ? err.code : -1
};
}
}
filename(fn) {
return fn;
}
optionsForFilter(filters, outputFilename) {
let options = ['-g', '-o', this.filename(outputFilename)];
if (this.compiler.intelAsm && filters.intel && !filters.binary) {
options = options.concat(this.compiler.intelAsm.split(" "));
}
if (!filters.binary) options = options.concat('-S');
return options;
}
findLibVersion(selectedLib) {
const foundLib = _.find(this.compiler.libs, (o, libId) => libId === selectedLib.id);
if (!foundLib) return false;
return _.find(foundLib.versions, (o, versionId) => (
versionId === selectedLib.version ||
(o.alias && o.alias.includes(selectedLib.version))));
}
findAutodetectStaticLibLink(linkname) {
const foundLib = _.findKey(this.compiler.libs, (lib) => {
return (lib.versions.autodetect && lib.versions.autodetect.staticliblink &&
lib.versions.autodetect.staticliblink.includes(linkname));
});
if (!foundLib) return false;
return {
id: foundLib,
version: "autodetect"
};
}
getSortedStaticLibraries(libraries) {
const dictionary = {};
const links = _.uniq(_.flatten(_.map(libraries, (selectedLib) => {
const foundVersion = this.findLibVersion(selectedLib);
if (!foundVersion) return false;
return _.map(foundVersion.staticliblink, (lib) => {
if (lib) {
dictionary[lib] = foundVersion;
return [lib, foundVersion.dependencies];
} else {
return false;
}
});
})));
let sortedlinks = [];
_.each(links, (libToInsertName) => {
const libToInsertObj = dictionary[libToInsertName];
let idxToInsert = sortedlinks.length;
for (let idx = 0; idx < sortedlinks.length; idx++) {
const libCompareName = sortedlinks[idx];
const libCompareObj = dictionary[libCompareName];
if (libToInsertObj && libCompareObj &&
_.intersection(libToInsertObj.dependencies, libCompareObj.staticliblink).length > 0) {
idxToInsert = idx;
break;
} else if (libToInsertObj &&
libToInsertObj.dependencies.includes(libCompareName)) {
idxToInsert = idx;
break;
} else if (libCompareObj &&
libCompareObj.dependencies.includes(libToInsertName)) {
continue;
} else if (libToInsertObj &&
libToInsertObj.staticliblink.includes(libToInsertName) &&
libToInsertObj.staticliblink.includes(libCompareName)) {
if (libToInsertObj.staticliblink.indexOf(libToInsertName) >
libToInsertObj.staticliblink.indexOf(libCompareName)) {
continue;
} else {
idxToInsert = idx;
}
break;
} else if (libCompareObj && libCompareObj.staticliblink.includes(libToInsertName) &&
libCompareObj.staticliblink.includes(libCompareName)) {
if (libCompareObj.staticliblink.indexOf(libToInsertName) >
libCompareObj.staticliblink.indexOf(libCompareName)) {
continue;
} else {
idxToInsert = idx;
}
break;
}
}
if (idxToInsert < sortedlinks.length) {
sortedlinks.splice(idxToInsert, 0, libToInsertName);
} else {
sortedlinks.push(libToInsertName);
}
});
return sortedlinks;
}
getStaticLibraryLinks(libraries) {
const linkFlag = this.compiler.linkFlag || "-l";
return _.map(this.getSortedStaticLibraries(libraries), (lib) => {
if (lib) {
return linkFlag + lib;
} else {
return false;
}
});
}
getSharedLibraryLinks(libraries) {
const linkFlag = this.compiler.linkFlag || "-l";
return _.flatten(_.map(libraries, (selectedLib) => {
const foundVersion = this.findLibVersion(selectedLib);
if (!foundVersion) return false;
return _.map(foundVersion.liblink, (lib) => {
if (lib) {
return linkFlag + lib;
} else {
return false;
}
});
}));
}
getSharedLibraryPaths(libraries) {
return _.flatten(_.map(libraries, (selectedLib) => {
const foundVersion = this.findLibVersion(selectedLib);
if (!foundVersion) return false;
return foundVersion.libpath;
}));
}
getSharedLibraryPathsAsArguments(libraries) {
const pathFlag = this.compiler.rpathFlag || "-Wl,-rpath,";
const libPathFlag = this.compiler.libpathFlag || "-L";
let toolchainLibraryPaths = [];
if (this.toolchainPath) {
toolchainLibraryPaths = [
path.join(this.toolchainPath, '/lib64'),
path.join(this.toolchainPath, '/lib32')
];
}
return _.union(
[libPathFlag + '.'],
[pathFlag + '.'],
this.compiler.ldPath.map(path => pathFlag + path),
toolchainLibraryPaths.map(path => pathFlag + path),
this.getSharedLibraryPaths(libraries).map(path => pathFlag + path),
this.getSharedLibraryPaths(libraries).map(path => libPathFlag + path));
}
getSharedLibraryPathsAsLdLibraryPaths(/*libraries*/) {
if (this.alwaysResetLdPath) {
return [];
} else {
return process.env.LD_LIBRARY_PATH ? process.env.LD_LIBRARY_PATH : [];
}
}
getIncludeArguments(libraries) {
const includeFlag = this.compiler.includeFlag || "-I";
return _.flatten(_.map(libraries, (selectedLib) => {
const foundVersion = this.findLibVersion(selectedLib);
if (!foundVersion) return false;
return _.map(foundVersion.path, (path) => includeFlag + path);
}));
}
prepareArguments(userOptions, filters, backendOptions, inputFilename, outputFilename, libraries) {
let options = this.optionsForFilter(filters, outputFilename, userOptions);
backendOptions = backendOptions || {};
if (this.compiler.options) {
options = options.concat(this.compiler.options.split(" "));
}
if (this.compiler.supportsOptOutput && backendOptions.produceOptInfo) {
options = options.concat(this.compiler.optArg);
}
const libIncludes = this.getIncludeArguments(libraries);
let libLinks = [];
let libPaths = [];
let staticLibLinks = [];
if (filters.binary) {
libLinks = this.getSharedLibraryLinks(libraries);
libPaths = this.getSharedLibraryPathsAsArguments(libraries);
staticLibLinks = this.getStaticLibraryLinks(libraries);
}
userOptions = this.filterUserOptions(userOptions) || [];
return options.concat(libIncludes, libPaths, libLinks, userOptions,
[this.filename(inputFilename)], staticLibLinks);
}
filterUserOptions(userOptions) {
return userOptions;
}
async generateAST(inputFilename, options) {
// These options make Clang produce an AST dump
const newOptions = _.filter(options, option => option !== '-fcolor-diagnostics')
.concat(["-Xclang", "-ast-dump", "-fsyntax-only"]);
const execOptions = this.getDefaultExecOptions();
// A higher max output is needed for when the user includes headers
execOptions.maxOutput = 1024 * 1024 * 1024;
return this.processAstOutput(
await this.runCompiler(this.compiler.exe, newOptions, this.filename(inputFilename), execOptions));
}
async generateIR(inputFilename, options, filters) {
// These options make Clang produce an IR
const newOptions = _.filter(options, option => option !== '-fcolor-diagnostics')
.concat(this.compiler.irArg);
const execOptions = this.getDefaultExecOptions();
// A higher max output is needed for when the user includes headers
execOptions.maxOutput = 1024 * 1024 * 1024;
const output = await this.runCompiler(this.compiler.exe, newOptions, this.filename(inputFilename), execOptions);
if (output.code !== 0) {
return [{text: "Failed to run compiler to get IR code"}];
}
const ir = await this.processIrOutput(output, filters);
return ir.asm;
}
async processIrOutput(output, filters) {
const irPath = this.getIrOutputFilename(output.inputFilename);
if (await fs.pathExists(irPath)) {
const output = await fs.readFile(irPath, 'utf-8');
// uses same filters as main compiler
return this.llvmIr.process(output, filters);
}
return {
asm: [{text: 'Internal error; unable to open output path'}],
labelDefinitions: {}
};
}
getIrOutputFilename(inputFilename) {
return inputFilename.replace(path.extname(inputFilename), '.ll');
}
getOutputFilename(dirPath, outputFilebase) {
// NB keep lower case as ldc compiler `tolower`s the output name
return path.join(dirPath, `${outputFilebase}.s`);
}
getExecutableFilename(dirPath, outputFilebase) {
return this.getOutputFilename(dirPath, outputFilebase);
}
async generateGccDump(inputFilename, options, gccDumpOptions) {
// Maybe we should not force any RTL dump and let user hand-pick what he needs
const addOpts = [];
/* if not defined, consider it true */
if (gccDumpOptions.treeDump !== false) {
addOpts.push("-fdump-tree-all");
}
if (gccDumpOptions.rtlDump !== false) {
addOpts.push("-fdump-rtl-all");
}
const newOptions = options.concat(addOpts);
const execOptions = this.getDefaultExecOptions();
// A higher max output is needed for when the user includes headers
execOptions.maxOutput = 1024 * 1024 * 1024;
return this.processGccDumpOutput(
gccDumpOptions,
await this.runCompiler(this.compiler.exe, newOptions, this.filename(inputFilename), execOptions));
}
async checkOutputFileAndDoPostProcess(asmResult, outputFilename, filters) {
try {
const stat = await fs.stat(outputFilename);
asmResult.asmSize = stat.size;
} catch (e) {
// Ignore errors
}
return this.postProcess(asmResult, outputFilename, filters);
}
runToolsOfType(tools, type, compilationInfo) {
let tooling = [];
if (tools) {
tools.forEach((tool) => {
const matches = this.possibleTools.filter(possibleTool => {
return possibleTool.getId() === tool.id &&
possibleTool.getType() === type;
});
if (matches[0]) {
const toolPromise = matches[0].runTool(compilationInfo,
compilationInfo.inputFilename, tool.args, tool.stdin);
tooling.push(toolPromise);
}
});
}
return tooling;
}
buildExecutable(compiler, options, inputFilename, execOptions) {
// default implementation, but should be overridden by compilers
return this.runCompiler(compiler, options, inputFilename, execOptions);
}
async getRequiredLibraryVersions(libraries) {
const libraryDetails = {};
_.each(libraries, (selectedLib) => {
const foundVersion = this.findLibVersion(selectedLib);
if (foundVersion) libraryDetails[selectedLib.id] = foundVersion;
});
return libraryDetails;
}
async setupBuildEnvironment(key, dirPath) {
if (this.buildenvsetup) {
const libraryDetails = await this.getRequiredLibraryVersions(key.libraries);
return this.buildenvsetup.setup(key, dirPath, libraryDetails);
} else {
return Promise.resolve();
}
}
async buildExecutableInFolder(key, dirPath) {
const buildEnvironment = this.setupBuildEnvironment(key, dirPath);
const inputFilename = path.join(dirPath, this.compileFilename);
const writerOfSource = fs.writeFile(inputFilename, key.source);
const outputFilename = this.getExecutableFilename(dirPath, this.outputFilebase);
const buildFilters = Object.assign({}, key.filters);
buildFilters.binary = true;
buildFilters.execute = true;
const compilerArguments = _.compact(
this.prepareArguments(key.options, buildFilters, key.backendOptions,
inputFilename, outputFilename, key.libraries)
);
const execOptions = this.getDefaultExecOptions();
execOptions.ldPath = this.getSharedLibraryPathsAsLdLibraryPaths(key.libraries);
await writerOfSource;
await buildEnvironment;
const result = await this.buildExecutable(key.compiler.exe, compilerArguments, inputFilename,
execOptions);
result.executableFilename = outputFilename;
result.compilationOptions = compilerArguments;
return result;
}
async getOrBuildExecutable(key) {
const dirPath = await this.newTempDir();
const compilationResultFilename = "compilation-result.json";
try {
const outputFilename = await this.env.executableGet(key, dirPath);
if (outputFilename) {
logger.debug(`Using cached package ${outputFilename}`);
await this.packager.unpack(outputFilename, dirPath);
const buildResults = JSON.parse(await fs.readFile(path.join(dirPath, compilationResultFilename)));
return Object.assign({}, buildResults, {
code: 0,
inputFilename: path.join(dirPath, this.compileFilename),
dirPath: dirPath,
executableFilename: this.getExecutableFilename(dirPath, this.outputFilebase)
});
}
logger.debug("Tried to get executable from cache, but got a cache miss");
} catch (err) {
logger.error("Tried to get executable from cache, but got an error:", {err});
}
const compilationResult = await this.buildExecutableInFolder(key, dirPath);
if (compilationResult.code !== 0) {
return compilationResult;
}
const packDir = await this.newTempDir();
const packagedFile = path.join(packDir, "package.tgz");
try {
await fs.writeFile(path.join(dirPath, compilationResultFilename), JSON.stringify(compilationResult));
await this.packager.package(dirPath, packagedFile);
await this.env.executablePut(key, packagedFile);
} catch (err) {
logger.error("Caught an error trying to put to cache: ", {err});
} finally {
fs.remove(packDir);
}
return compilationResult;
}
runExecutable(executable, executeParameters) {
const maxExecOutputSize = this.env.ceProps("max-executable-output-size", 32 * 1024);
return this.execBinary(executable, maxExecOutputSize, executeParameters);
}
async handleExecution(key, executeParameters) {
const buildResult = await this.getOrBuildExecutable(key);
if (buildResult.code !== 0) {
return {
code: -1,
didExecute: false,
buildResult,
stderr: [],
stdout: []
};
} else {
if (!fs.existsSync(buildResult.executableFilename)) {
const verboseResult = {
code: -1,
didExecute: false,
buildResult,
stderr: [],
stdout: []
};
verboseResult.buildResult.stderr.push({text:"Compiler did not produce an executable"});
return verboseResult;
}
}
executeParameters.ldPath = this.getSharedLibraryPathsAsLdLibraryPaths(key.libraries);
const result = await this.runExecutable(buildResult.executableFilename, executeParameters);
result.didExecute = true;
result.buildResult = buildResult;
return result;
}
getCacheKey(source, options, backendOptions, filters, tools, libraries) {
return {compiler: this.compiler, source, options, backendOptions, filters, tools, libraries};
}
getCompilationInfo(key, result) {
const compilationinfo = Object.assign({}, key, result);
compilationinfo.outputFilename = this.getOutputFilename(result.dirPath, this.outputFilebase);
compilationinfo.executableFilename = this.getExecutableFilename(result.dirPath, this.outputFilebase);
return compilationinfo;
}
tryAutodetectLibraries(libsAndOptions) {
const linkFlag = this.compiler.linkFlag || "-l";
const detectedLibs = [];
const foundlibOptions = [];
_.each(libsAndOptions.options, (option) => {
if (option.indexOf(linkFlag) === 0) {
const libVersion = this.findAutodetectStaticLibLink(option.substr(linkFlag.length).trim());
if (libVersion) {
foundlibOptions.push(option);
detectedLibs.push(libVersion);
}
}
});
if (detectedLibs.length > 0) {
libsAndOptions.options = _.filter(libsAndOptions.options, (option) => !foundlibOptions.includes(option));
libsAndOptions.libraries = _.union(libsAndOptions.libraries, detectedLibs);
return true;
} else {
return false;
}
}
async doCompilation(inputFilename, dirPath, key, options, filters, backendOptions, libraries, tools) {
let buildEnvironment = Promise.resolve();
if (filters.binary) {
buildEnvironment = this.setupBuildEnvironment(key, dirPath);
}
const inputFilenameSafe = this.filename(inputFilename);
const outputFilename = this.getOutputFilename(dirPath, this.outputFilebase);
options = _.compact(
this.prepareArguments(options, filters, backendOptions, inputFilename, outputFilename, libraries)
);
const execOptions = this.getDefaultExecOptions();
execOptions.ldPath = this.getSharedLibraryPathsAsLdLibraryPaths(key.libraries);
const makeAst = backendOptions.produceAst && this.compiler.supportsAstView;
const makeIr = backendOptions.produceIr && this.compiler.supportsIrView;
const makeGccDump = backendOptions.produceGccDump && backendOptions.produceGccDump.opened
&& this.compiler.supportsGccDump;
await buildEnvironment;
const [asmResult, astResult, gccDumpResult, irResult, toolsResult] = await Promise.all([
this.runCompiler(this.compiler.exe, options, inputFilenameSafe, execOptions),
(makeAst ? this.generateAST(inputFilename, options) : ""),
(makeGccDump ? this.generateGccDump(inputFilename, options, backendOptions.produceGccDump) : ""),
(makeIr ? this.generateIR(inputFilename, options, filters) : ""),
Promise.all(this.runToolsOfType(tools, "independent", this.getCompilationInfo(key, {
inputFilename,
dirPath,
outputFilename
})))
]);
asmResult.dirPath = dirPath;
asmResult.compilationOptions = options;
// Here before the check to ensure dump reports even on failure cases
if (this.compiler.supportsGccDump && gccDumpResult) {
asmResult.gccDumpOutput = gccDumpResult;
}
asmResult.tools = toolsResult;
if (asmResult.code !== 0) {
asmResult.asm = "<Compilation failed>";
return [asmResult];
}
if (this.compiler.supportsOptOutput && this.optOutputRequested(options)) {
const optPath = path.join(dirPath, `${this.outputFilebase}.opt.yaml`);
if (await fs.pathExists(optPath)) {
asmResult.hasOptOutput = true;
asmResult.optPath = optPath;
}
}
if (astResult) {
asmResult.hasAstOutput = true;
asmResult.astOutput = astResult;
}
if (irResult) {
asmResult.hasIrOutput = true;
asmResult.irOutput = irResult;
}
return this.checkOutputFileAndDoPostProcess(asmResult, outputFilename, filters);
}
async compile(source, options, backendOptions, filters, bypassCache, tools, executionParameters, libraries) {
const optionsError = this.checkOptions(options);
if (optionsError) throw optionsError;
const sourceError = this.checkSource(source);
if (sourceError) throw sourceError;
const libsAndOptions = {libraries, options};
if (this.tryAutodetectLibraries(libsAndOptions)) {
libraries = libsAndOptions.libraries;
options = libsAndOptions.options;
}
// Don't run binary for unsupported compilers, even if we're asked.
if (filters.binary && !this.compiler.supportsBinary) {
delete filters.binary;
}
const executeParameters = {
args: executionParameters.args || [],
stdin: executionParameters.stdin || ""
};
const key = this.getCacheKey(source, options, backendOptions, filters, tools, libraries);
const doExecute = filters.execute;
filters = Object.assign({}, filters);
filters.execute = false;
if (!bypassCache) {
const result = await this.env.cacheGet(key);
if (result) {
if (doExecute) {
result.execResult = await this.env.enqueue(async () => {
return this.handleExecution(key, executeParameters);
});
}
return result;
}
}
return this.env.enqueue(async () => {
source = this.preProcess(source);
if (filters.binary && !source.match(this.compilerProps("stubRe"))) {
source += "\n" + this.compilerProps("stubText") + "\n";
}
if (backendOptions.executorRequest) {
return this.handleExecution(key, executeParameters);
}
const dirPath = await this.newTempDir();
const inputFilename = path.join(dirPath, this.compileFilename);
await fs.writeFile(inputFilename, source);
// TODO make const when I can
let [result, optOutput] = await this.doCompilation(
inputFilename, dirPath, key, options, filters, backendOptions, libraries, tools);
// Start the execution as soon as we can, but only await it at the end.
const execPromise = doExecute ? this.handleExecution(key, executeParameters) : null;
if (result.hasOptOutput) {
delete result.optPath;
result.optOutput = optOutput;
}
result.tools = _.union(result.tools, await Promise.all(this.runToolsOfType(tools, "postcompilation",
this.getCompilationInfo(key, result))));
if (result.dirPath) {
fs.remove(result.dirPath);
result.dirPath = undefined;
}
if (!backendOptions.skipAsm) {
if (result.okToCache) {
const res = this.processAsm(result, filters, options);
result.asm = res.asm;
result.labelDefinitions = res.labelDefinitions;
} else {
result.asm = [{text: result.asm}];
}
// TODO rephrase this so we don't need to reassign
result = filters.demangle ? await this.postProcessAsm(result, filters) : result;
if (this.compiler.supportsCfg && backendOptions.produceCfg) {
result.cfg = cfg.generateStructure(this.compiler.compilerType,
this.compiler.version, result.asm);
}
}
result.popularArguments = this.possibleArguments.getPopularArguments(options);
if (result.okToCache) {
this.env.cachePut(key, result).then(() => {
// purely here to ensure the cache put happens (if it's not "then"-ned it might not start. as it
// happens it does cos internally to cachePut it "catch"-es. but I don't want to rely on this.
});
}
if (doExecute) {
result.execResult = await execPromise;
}
return result;
});
}
processAsm(result, filters, options) {
if ((options && options.includes('-emit-llvm')) || this.llvmIr.isLlvmIr(result.asm)) {
return this.llvmIr.process(result.asm, filters);
}
return this.asm.process(result.asm, filters);
}
async postProcessAsm(result) {
if (!result.okToCache || !this.demanglerClass || !result.asm) return result;
const demangler = new this.demanglerClass(this.compiler.demangler, this);
return demangler.process(result);
}
async processOptOutput(optPath) {
const output = [];
const optStream = fs.createReadStream(optPath, {encoding: "utf-8"})
.pipe(new compilerOptInfo.LLVMOptTransformer());
for await (const opt of optStream) {
if (opt.DebugLoc && opt.DebugLoc.File && opt.DebugLoc.File.indexOf(this.compileFilename) > -1) {
output.push(opt);
}
}
if (this.compiler.demangler) {
const result = JSON.stringify(output, null, 4);
try {
const demangleResult = await this.exec(
this.compiler.demangler, ["-n", "-p"], {input: result});
return JSON.parse(demangleResult.stdout);
} catch (exception) {
// swallow exception and return non-demangled output
logger.warn(`Caught exception ${exception} during opt demangle parsing`);
}
}
return output;
}
couldSupportASTDump(version) {
const versionRegex = /version (\d+.\d+)/;
const versionMatch = versionRegex.exec(version);
if (versionMatch) {
const versionNum = parseFloat(versionMatch[1]);
return version.toLowerCase().indexOf("clang") > -1 && versionNum >= 3.3;
}
return false;
}
isCfgCompiler(compilerVersion) {
return compilerVersion.includes("clang") ||
compilerVersion.match(/^([\w-]*-)?g((\+\+)|(cc)|(dc))/g) !== null;
}
processAstOutput(output) {
output = output.stdout;
output = output.map(x => x.text);
// Top level decls start with |- or `-
const topLevelRegex = /^([|`])-/;
// Refers to the user's source file rather than a system header
const sourceRegex = /<source>/g;
// Refers to whatever the most recent file specified was
const lineRegex = /<line:/;
let mostRecentIsSource = false;
// Remove all AST nodes which aren't directly from the user's source code
for (let i = 0; i < output.length; ++i) {
if (output[i].match(topLevelRegex)) {
if (output[i].match(lineRegex) && mostRecentIsSource) {
// do nothing
} else if (!output[i].match(sourceRegex)) {
// This is a system header or implicit definition,
// remove everything up to the next top level decl
// Top level decls with invalid sloc as the file don't change the most recent file
let slocRegex = /<<invalid sloc>>/;
if (!output[i].match(slocRegex)) {
mostRecentIsSource = false;
}
let spliceMax = i + 1;
while (output[spliceMax] && !output[spliceMax].match(topLevelRegex)) {
spliceMax++;
}
output.splice(i, spliceMax - i);
--i;
} else {
mostRecentIsSource = true;
}
}
}
output = output.join('\n');
// Filter out the symbol addresses
const addressRegex = /^([^A-Za-z]*[A-Za-z]+) 0x[a-z0-9]+/mg;
output = output.replace(addressRegex, '$1');
// Filter out <invalid sloc> and <<invalid sloc>>
let slocRegex = / ?<?<invalid sloc>>?/g;
output = output.replace(slocRegex, '');
// Unify file references
output = output.replace(sourceRegex, 'line');
return output;
}
async processGccDumpOutput(opts, result) {
const rootDir = path.dirname(result.inputFilename);
const allFiles = await fs.readdir(rootDir);
if (opts.treeDump === false && opts.rtlDump === false) {
return {
all: [],
selectedPass: "",
currentPassOutput: 'Nothing selected for dump:\nselect at least one of Tree/RTL filter',
syntaxHighlight: false
};
}
const output = {
all: [],
selectedPass: opts.pass,
currentPassOutput: '<No pass selected>',
syntaxHighlight: false
};
let passFound = false;
// Phase letter is one of {i, l, r, t}
// {outpufilename}.{extension}.{passNumber}{phaseLetter}.{phaseName}
const dumpFilenameRegex = /^.+?\..+?\.(\d+?[ilrt]\..+)$/;
for (let filename of allFiles) {
const match = dumpFilenameRegex.exec(filename);
if (match) {
const pass = match[1];
output.all.push(pass);
const filePath = path.join(rootDir, filename);
if (opts.pass === pass && (await fs.stat(filePath)).isFile()) {
passFound = true;
output.currentPassOutput = await fs.readFile(filePath, 'utf-8');
if (output.currentPassOutput.match(/^\s*$/)) {
output.currentPassOutput = 'File for selected pass is empty.';
} else {
output.syntaxHighlight = true;
}
}
}
}
if (opts.pass && !passFound) {
output.currentPassOutput = `Pass '${opts.pass}' was requested
but is not valid anymore with current filters.
Please select another pass or change filters.`;
}
return output;
}
async execPostProcess(result, postProcesses, outputFilename, maxSize) {
const postCommand = `cat "${outputFilename}" | ${postProcesses.join(" | ")}`;
return this.handlePostProcessResult(
result,
await this.exec("bash", ["-c", postCommand], {maxOutput: maxSize}));
}
preProcess(source) {
return source;
}
async postProcess(result, outputFilename, filters) {
const postProcess = _.compact(this.compiler.postProcess);
const maxSize = this.env.ceProps("max-asm-size", 64 * 1024 * 1024);
const optPromise = result.hasOptOutput ? this.processOptOutput(result.optPath) : "";
const asmPromise = (filters.binary && this.supportsObjdump())
? this.objdump(outputFilename, result, maxSize, filters.intel, filters.demangle)
: (async () => {
if (result.asmSize === undefined) {
result.asm = "<No output file>";
return result;
}
if (result.asmSize >= maxSize) {
result.asm = `<No output: generated assembly was too large (${result.asmSize} > ${maxSize} bytes)>`;
return result;
}
if (postProcess.length) {
return this.execPostProcess(result, postProcess, outputFilename, maxSize);
} else {
const contents = await fs.readFile(outputFilename);
result.asm = contents.toString();
return result;
}
})();
return Promise.all([asmPromise, optPromise]);
}
handlePostProcessResult(result, postResult) {
result.asm = postResult.stdout;
if (postResult.code !== 0) {
result.asm = `<Error during post processing: ${postResult.code}>`;
logger.error("Error during post-processing: ", result);
}
return result;
}
checkOptions(options) {
const error = this.env.findBadOptions(options);
if (error.length > 0) return `Bad options: ${error.join(", ")}`;
return null;
}
// This check for arbitrary user-controlled preprocessor inclusions
// can be circumvented in more than one way. The goal here is to respond
// to simple attempts with a clear diagnostic; the service still needs to
// assume that malicious actors can make the compiler open arbitrary files.
checkSource(source) {
const re = /^\s*#\s*i(nclude|mport)(_next)?\s+["<](\/|.*\.\.)[">]/;
const failed = [];
utils.splitLines(source).forEach((line, index) => {
if (line.match(re)) {
failed.push(`<stdin>:${index + 1}:1: no absolute or relative includes please`);
}
});
if (failed.length > 0) return failed.join("\n");
return null;
}
getArgumentParser() {
let exe = this.compiler.exe.toLowerCase();
if (exe.indexOf("clang") >= 0) { // check this first as "clang++" matches "g++"
return argumentParsers.Clang;
} else if (exe.indexOf("g++") >= 0 || exe.indexOf("gcc") >= 0) {
return argumentParsers.GCC;
}
//there is a lot of code around that makes this assumption.
//probably not the best thing to do :D
return argumentParsers.GCC;
}
getVersion() {
logger.info(`Gathering ${this.compiler.id} version information on ${this.compiler.exe}`);
const execOptions = this.getDefaultExecOptions();
const versionFlag = this.compiler.versionFlag || '--version';
execOptions.timeoutMs = 0; // No timeout for --version. A sort of workaround for slow EFS/NFS on the prod site
return this.execCompilerCached(this.compiler.exe, [versionFlag], execOptions);
}
async initialise(mtime) {
this.mtime = mtime;
if (this.getRemote()) return this;
const compiler = this.compiler.exe;
const versionRe = new RegExp(this.compiler.versionRe || '.*', 'i');
const result = await this.env.enqueue(async () => {
try {
return await this.getVersion();
} catch (err) {
logger.error(`Unable to get version for compiler '${compiler}' - ${err}`);
return null;
}
});
if (!result) return null;
if (result.code !== 0) {
logger.warn(`Compiler '${compiler}' - non-zero result ${result.code}`);
}
let version = "";
_.each(utils.splitLines(result.stdout + result.stderr), line => {
if (version) return;
const match = line.match(versionRe);
if (match) version = match[0];
});
if (!version) {
logger.error(`Unable to find compiler version for '${compiler}' with re ${versionRe}:`, result);
return null;
}
logger.debug(`${compiler} is version '${version}'`);
this.compiler.version = version;
this.compiler.supportsCfg = this.isCfgCompiler(version);
this.compiler.supportsAstView = this.couldSupportASTDump(version);
return this.getArgumentParser().parse(this);
}
getInfo() {
return this.compiler;
}
getDefaultFilters() {
// TODO; propagate to UI?
return {
intel: true,
commentOnly: true,
directives: true,
labels: true,
optOutput: false
};
}
}
module.exports = BaseCompiler;