diff options
| -rw-r--r-- | tools/driver/main.cc | 47 |
1 files changed, 46 insertions, 1 deletions
diff --git a/tools/driver/main.cc b/tools/driver/main.cc index a4616bd..55d666f 100644 --- a/tools/driver/main.cc +++ b/tools/driver/main.cc @@ -7,6 +7,10 @@ #include "llvm/ADT/SmallString.h" #include "llvm/Support/FileSystem.h" #include "llvm/Support/FileUtilities.h" +#ifdef CPSC_RUN_PATH +#include "llvm/Support/MemoryBuffer.h" +#include "llvm/Support/Regex.h" +#endif #include "llvm/Support/Path.h" #include "llvm/Support/Program.h" #include "llvm/Support/raw_ostream.h" @@ -20,6 +24,46 @@ int compilationTerminated() { return 1; } +#ifdef CPSC_RUN_PATH +bool checkLliVersion(llvm::StringRef LliPath) { + llvm::SmallString<256> VersionPath; + if (auto EC = llvm::sys::fs::createTemporaryFile("cpsc-lli-version", "txt", VersionPath)) { + llvm::errs() << "cpsc: fatal error: cannot check lli version: " << EC.message() << "\n"; + return false; + } + llvm::FileRemover TemporaryVersion(VersionPath); + std::string ExecutionError; + int Result = llvm::sys::ExecuteAndWait(LliPath, {LliPath, "--version"}, + std::nullopt, {llvm::StringRef(""), llvm::StringRef(VersionPath), llvm::StringRef("")}, + 5, 0, &ExecutionError); + if (Result != 0) { + llvm::errs() << "cpsc: fatal error: cannot check lli version at '" << LliPath + << "' (status " << Result << ")"; + if (!ExecutionError.empty()) llvm::errs() << ": " << ExecutionError; + llvm::errs() << "\n"; + return false; + } + auto Version = llvm::MemoryBuffer::getFile(VersionPath); + if (!Version) { + llvm::errs() << "cpsc: fatal error: cannot read lli version: " + << Version.getError().message() << "\n"; + return false; + } + llvm::SmallVector<llvm::StringRef, 2> Matches; + if (!llvm::Regex("LLVM version ([0-9]+)\\.").match((*Version)->getBuffer(), &Matches)) { + llvm::errs() << "cpsc: fatal error: cannot identify LLVM version of lli at '" + << LliPath << "'; LLVM 23 is required\n"; + return false; + } + if (Matches[1] != "23") { + llvm::errs() << "cpsc: fatal error: -r requires lli from LLVM 23; found LLVM " + << Matches[1] << " at '" << LliPath << "'\n"; + return false; + } + return true; +} +#endif + } // namespace int main(int argc, char **argv) { @@ -73,10 +117,11 @@ int main(int argc, char **argv) { #ifdef CPSC_RUN_PATH auto Lli = llvm::sys::findProgramByName("lli"); if (!Lli) { - llvm::errs() << "cpsc: fatal error: -r requires lli in PATH: " << Lli.getError().message() << "\n"; + llvm::errs() << "cpsc: fatal error: -r requires lli from LLVM 23 in PATH: " << Lli.getError().message() << "\n"; return compilationTerminated(); } LliPath = *Lli; + if (!checkLliVersion(LliPath)) return compilationTerminated(); #elif defined(CPSC_RUN_NONE) llvm::errs() << "cpsc: fatal error: -r is disabled in this build (CPSC_RUN_MODE=NONE)\n"; return compilationTerminated(); |