|
5 | 5 | #include <unistd.h>
|
6 | 6 | #include "whereami.h"
|
7 | 7 |
|
8 |
| -#define max_path_length 1024 |
| 8 | +#define MAX_PATH_LENGTH 1024 |
| 9 | + |
9 | 10 | char *get_install_path(const char * package_path)
|
10 | 11 | {
|
11 |
| - char fullpath[max_path_length - 1] = "\0"; |
12 |
| - int dirname_length = 0; |
13 |
| - int res = wai_getExecutablePath(fullpath, max_path_length, &dirname_length); |
14 |
| - if (res <= 0) |
15 |
| - return NULL; |
16 |
| - if (res <= dirname_length) |
| 12 | + char fullpath[MAX_PATH_LENGTH - 1] = "\0"; |
| 13 | + int fullpath_length = 0; |
| 14 | + int res = wai_getExecutablePath(fullpath, MAX_PATH_LENGTH, &fullpath_length); |
| 15 | + if (res <= 0 || res <= fullpath_length) |
17 | 16 | return NULL;
|
18 |
| - dirname_length -= strlen(package_path); |
19 |
| - char * path = (char*) malloc(dirname_length+1); |
20 |
| - memcpy(path, fullpath, dirname_length); |
21 |
| - path[dirname_length] = '\0'; |
| 17 | + // printf("fullpath: %s (%d)\n", fullpath, fullpath_length); |
| 18 | +#ifdef WIN32 |
| 19 | + for (size_t i = 0; i < fullpath_length; ++i) { |
| 20 | + if (fullpath[i] == '\\') { |
| 21 | + fullpath[i] = '/'; |
| 22 | + } |
| 23 | + } |
| 24 | +#endif |
| 25 | + // Remove package_path and everything after it from fullpath. |
| 26 | + size_t package_path_len = strlen(package_path); |
| 27 | + char *pos = strstr(fullpath, package_path); |
| 28 | + char * last_found = NULL; |
| 29 | + if (pos == NULL) { |
| 30 | + printf("Error: irace is installed at an unexpected path: %s\n", fullpath); |
| 31 | + exit(EXIT_FAILURE); |
| 32 | + } |
| 33 | + // Search the last occurrence in case fullpath is something like |
| 34 | + // /irace/bin/R/x86_64-pc-linux-gnu-library/4.1/irace/bin/irace . |
| 35 | + do { |
| 36 | + last_found = pos; |
| 37 | + pos = strstr(pos + package_path_len, package_path); |
| 38 | + } while (pos != NULL); |
| 39 | + *last_found = '\0'; |
| 40 | + fullpath_length = strlen(fullpath); |
| 41 | + char * path = (char*) malloc((fullpath_length + 1) * sizeof(char)); |
| 42 | + memcpy(path, fullpath, fullpath_length + 1); |
| 43 | + // printf("path: %s (%d)\n", fullpath, fullpath_length); |
22 | 44 | return path;
|
23 | 45 | }
|
| 46 | +#undef MAX_PATH_LENGTH |
24 | 47 |
|
25 | 48 | int exec_R(int argc, char *argv[], const char * user_code)
|
26 | 49 | {
|
| 50 | + const char * path = get_install_path("/irace/bin/"); |
| 51 | + if (path == NULL) |
| 52 | + path = ""; |
| 53 | + |
| 54 | +#define _libpath_str ".libPaths(c(r'{%s}', .libPaths()));%s" |
27 | 55 | #ifdef WIN32
|
28 |
| -# define EXE_EXT ".exe" |
| 56 | +// Windows execvp() does not pass command-line arguments correctly if they |
| 57 | +// contain space, tab, backslash, or double-quote characters. |
| 58 | +#define libpath_str "\"" _libpath_str "\"" |
29 | 59 | #else
|
30 |
| -# define EXE_EXT "" |
| 60 | +#define libpath_str _libpath_str |
31 | 61 | #endif
|
32 |
| - const char file[] = "R" EXE_EXT; |
33 |
| - const char * path = get_install_path("/irace/bin"); |
34 |
| - if (path == NULL) path = ""; |
35 |
| -#define libpath_str ".libPaths(c('%s', .libPaths()));%s" |
| 62 | + // -4 because of two times %s. |
36 | 63 | char * r_code = malloc(sizeof(char) * (strlen(libpath_str) - 4 + strlen(path) + strlen(user_code) + 1));
|
37 | 64 | sprintf(r_code, libpath_str, path, user_code);
|
38 |
| - //printf("%s\n", libpath); |
39 |
| - |
| 65 | +#undef _libpath_str |
| 66 | +#undef libpath_str |
| 67 | + // printf("%s\n", r_code); |
| 68 | + |
40 | 69 | char * const extra_argv[] = { "--vanilla", "--slave", "-e", r_code, "--args"};
|
41 | 70 | int extra_argc = sizeof(extra_argv) / sizeof(char *);
|
42 |
| - char **cmd_args = calloc(extra_argc + argc + 1, sizeof(char *)); |
| 71 | + char **cmd_args = malloc((extra_argc + argc + 1) * sizeof(char *)); |
43 | 72 | cmd_args[0] = argv[0];
|
44 | 73 | for (int i = 0; i < extra_argc; i++)
|
45 | 74 | cmd_args[i+1] = extra_argv[i];
|
46 | 75 | for (int i = 1; i < argc; i++)
|
47 | 76 | cmd_args[extra_argc + i] = argv[i];
|
48 |
| - return execvp(file, cmd_args); |
| 77 | + cmd_args[extra_argc + argc] = NULL; |
| 78 | + // for (int i = 0; i < extra_argc + argc; i++) |
| 79 | + // printf("%d: %s\n", i, cmd_args[i]); |
| 80 | + |
| 81 | +#ifdef WIN32 |
| 82 | +# define EXE_EXT ".exe" |
| 83 | +#else |
| 84 | +# define EXE_EXT "" |
| 85 | +#endif |
| 86 | + const char file[] = "R" EXE_EXT; |
| 87 | +#undef EXE_EXT |
| 88 | + if (execvp(file, cmd_args) == -1) { |
| 89 | + perror("Error executing R"); |
| 90 | + return EXIT_FAILURE; |
| 91 | + } |
| 92 | + return EXIT_SUCCESS; |
49 | 93 | }
|
0 commit comments