-/*===- llvm-stub.c - Stub executable to run llvm bytecode files -----------===//
-//
+/*===- llvm-stub.c - Stub executable to run llvm bitcode files ------------===//
+//
// The LLVM Compiler Infrastructure
//
-// This file was developed by the LLVM research group and is distributed under
-// the University of Illinois Open Source License. See LICENSE.TXT for details.
-//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
//===----------------------------------------------------------------------===//
//
// This tool is used by the gccld program to enable transparent execution of
-// bytecode files by the user. Specifically, gccld outputs two files when asked
+// bitcode files by the user. Specifically, gccld outputs two files when asked
// to compile a <program> file:
-// 1. It outputs the LLVM bytecode file to <program>.bc
+// 1. It outputs the LLVM bitcode file to <program>.bc
// 2. It outputs a stub executable that runs lli on <program>.bc
//
// This allows the end user to just say ./<program> and have the JIT executed
// automatically. On unix, the stub executable emitted is actually a bourne
-// shell script that does the forwarding. Windows doesn't not like #!/bin/sh
+// shell script that does the forwarding. Windows does not like #!/bin/sh
// programs in .exe files, so we make it an actual program, defined here.
//
//===----------------------------------------------------------------------===*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
-#include "Config/unistd.h" /* provides definition of execve */
+
+#include "llvm/Config/config.h"
+
+#if defined(HAVE_UNISTD_H) && !defined(_MSC_VER)
+#include <unistd.h>
+#endif
+
+#ifdef _WIN32
+#include <process.h>
+#include <io.h>
+#endif
int main(int argc, char** argv) {
const char *Interp = getenv("LLVMINTERP");
/* argv[0] is the JIT */
Args[0] = Interp;
-#ifdef __CYGWIN32__
- /* Cygwin strips the .exe suffix off of argv[0] to "help" us. Put it back
- * on.
- */
- argv[0] = strcat(strcpy((char*)malloc(strlen(argv[0])+5), argv[0]), ".exe");
+#ifdef LLVM_ON_WIN32
+ {
+ int len = strlen(argv[0]);
+ if (len < 4 || strcmp(argv[0] + len - 4, ".exe") != 0) {
+ /* .exe suffix is stripped off of argv[0] if the executable was run on the
+ * command line without one. Put it back on.
+ */
+ argv[0] = strcat(strcpy((char*)malloc(len + 5), argv[0]), ".exe");
+ }
+ }
#endif
/* argv[1] is argv[0] + ".bc". */
Args[1] = strcat(strcpy((char*)malloc(strlen(argv[0])+4), argv[0]), ".bc");
/* The rest of the args are as before. */
- memcpy(Args+2, argv+1, sizeof(char*)*argc);
+ memcpy((char **)Args+2, argv+1, sizeof(char*)*argc);
/* Run the JIT. */
- execvp(Interp, (char *const*)Args);
-
+#ifndef _WIN32
+ execvp(Interp, (char **)Args); /* POSIX execvp takes a char *const[]. */
+#else
+ execvp(Interp, Args); /* windows execvp takes a const char *const *. */
+#endif
/* if _execv returns, the JIT could not be started. */
fprintf(stderr, "Could not execute the LLVM JIT. Either add 'lli' to your"
" path, or set the\ninterpreter you want to use in the LLVMINTERP "