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