Clean up a use of std::distance.

[oota-llvm.git] / tools / llvm-stub / llvm-stub.c

diff --git a/tools/llvm-stub/llvm-stub.c b/tools/llvm-stub/llvm-stub.c
index 10b025633f8a3d9c2401688a47e9ec0972b48be7..18d20e7d0c9f8bea0be0d604864a38ea92f64f17 100644 (file)
--- a/tools/llvm-stub/llvm-stub.c
+++ b/tools/llvm-stub/llvm-stub.c
@@ -1,16 +1,16 @@
-/*===- 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
@@ -45,11 +45,16 @@ int main(int argc, char** argv) {
   /* 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". */