tools/llee/ExecveHandler.c

   1 /*===-- ExecveHandler.c - Replaces execve() to run LLVM 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 file implements a replacement execve() to spawn off LLVM programs to run
  11  * transparently, without needing to be (JIT-)compiled manually by the user.
  12  *
  13 \*===----------------------------------------------------------------------===*/
  14
  15 #include "SysUtils.h"
  16 #include "llvm/Config/unistd.h"
  17 #include <errno.h>
  18 #include <stdlib.h>
  19 #include <fcntl.h>
  20 #include <stdio.h>
  21 #include <string.h>
  22
  23 /*
  24  * This is the expected header for all valid LLVM bytecode files.
  25  * The first four characters must be exactly this.
  26  */
  27 static const char llvmHeader[] = "llvm";
  28
  29 /*
  30  * This replacement execve() function first checks the file to be executed
  31  * to see if it is a valid LLVM bytecode file, and then either invokes our
  32  * execution environment or passes it on to the system execve() call.
  33  */
  34 int execve(const char *filename, char *const argv[], char *const envp[])
  35 {
  36   /* Open the file, test to see if first four characters are "llvm" */
  37   size_t headerSize = strlen(llvmHeader);
  38   char header[headerSize];
  39   char* realFilename = 0;
  40   /*
  41    * If the program is specified with a relative or absolute path,
  42    * then just use the path and filename as is, otherwise search for it.
  43    */
  44   if (filename[0] != '.' && filename[0] != '/')
  45     realFilename = FindExecutable(filename);
  46   else
  47     realFilename = (char*) filename;
  48   if (!realFilename) {
  49     fprintf(stderr, "Cannot find path to `%s', exiting.\n", filename);
  50     return -1;
  51   }
  52   errno = 0;
  53   int file = open(realFilename, O_RDONLY);
  54   /* Check validity of `file' */
  55   if (errno) return EIO;
  56   /* Read the header from the file */
  57   ssize_t bytesRead = read(file, header, headerSize);
  58   close(file);
  59   if (bytesRead != (ssize_t)headerSize) return EIO;
  60   if (!memcmp(llvmHeader, header, headerSize)) {
  61     /*
  62      * This is a bytecode file, so execute the JIT with the program and
  63      * parameters.
  64      */
  65     unsigned argvSize, idx;
  66     for (argvSize = 0, idx = 0; argv[idx] && argv[idx][0]; ++idx)
  67       ++argvSize;
  68     char **LLIargs = (char**) malloc(sizeof(char*) * (argvSize+2));
  69     char *LLIpath = FindExecutable("lli");
  70     if (!LLIpath) {
  71       fprintf(stderr, "Cannot find path to `lli', exiting.\n");
  72       return -1;
  73     }
  74     LLIargs[0] = LLIpath;
  75     LLIargs[1] = realFilename;
  76     for (idx = 1; idx != argvSize; ++idx)
  77       LLIargs[idx+1] = argv[idx];
  78     LLIargs[argvSize + 1] = '\0';
  79     return executeProgram(LLIpath, LLIargs, envp);
  80   }
  81   return executeProgram(filename, argv, envp);
  82 }