tools/llvm-stub/llvm-stub.c

   1 /*===- llvm-stub.c - Stub executable to run llvm bytecode files -----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This tool is used by the gccld program to enable transparent execution of
  11 // bytecode files by the user.  Specifically, gccld outputs two files when asked
  12 // to compile a <program> file:
  13 //    1. It outputs the LLVM bytecode file to <program>.bc
  14 //    2. It outputs a stub executable that runs lli on <program>.bc
  15 //
  16 // This allows the end user to just say ./<program> and have the JIT executed
  17 // automatically.  On unix, the stub executable emitted is actually a bourne
  18 // shell script that does the forwarding.  Windows does not like #!/bin/sh
  19 // programs in .exe files, so we make it an actual program, defined here.
  20 //
  21 //===----------------------------------------------------------------------===*/
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26
  27 #include "llvm/Config/config.h"
  28
  29 #if defined(HAVE_UNISTD_H) && !defined(_MSC_VER)
  30 #include <unistd.h>
  31 #endif
  32
  33 #ifdef _WIN32
  34 #include <process.h>
  35 #include <io.h>
  36 #endif
  37
  38 int main(int argc, char** argv) {
  39   const char *Interp = getenv("LLVMINTERP");
  40   const char **Args;
  41   if (Interp == 0) Interp = "lli";
  42
  43   /* Set up the command line options to pass to the JIT. */
  44   Args = (const char**)malloc(sizeof(char*) * (argc+2));
  45   /* argv[0] is the JIT */
  46   Args[0] = Interp;
  47
  48 #ifdef __CYGWIN32__
  49   /* Cygwin strips the .exe suffix off of argv[0] to "help" us.  Put it back
  50    * on.
  51    */
  52   argv[0] = strcat(strcpy((char*)malloc(strlen(argv[0])+5), argv[0]), ".exe");
  53 #endif
  54
  55   /* argv[1] is argv[0] + ".bc". */
  56   Args[1] = strcat(strcpy((char*)malloc(strlen(argv[0])+4), argv[0]), ".bc");
  57
  58   /* The rest of the args are as before. */
  59   memcpy(Args+2, argv+1, sizeof(char*)*argc);
  60
  61   /* Run the JIT. */
  62   execvp(Interp, (char *const*)Args);
  63
  64   /* if _execv returns, the JIT could not be started. */
  65   fprintf(stderr, "Could not execute the LLVM JIT.  Either add 'lli' to your"
  66           " path, or set the\ninterpreter you want to use in the LLVMINTERP "
  67           "environment variable.\n");
  68   return 1;
  69 }