spawn.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995
  1. /*!
  2. * \file lib/gis/spawn.c
  3. *
  4. * \brief GIS Library - Handles process spawning.
  5. *
  6. * (C) 2001-2014 by the GRASS Development Team
  7. *
  8. * This program is free software under the GNU General Public License
  9. * (>=v2). Read the file COPYING that comes with GRASS for details.
  10. *
  11. * \author Glynn Clements
  12. *
  13. * \date 2004-2006
  14. */
  15. #include <stdio.h>
  16. #include <stdlib.h>
  17. #include <string.h>
  18. #include <signal.h>
  19. #include <stdarg.h>
  20. #include <unistd.h>
  21. #include <fcntl.h>
  22. #include <errno.h>
  23. #include <sys/types.h>
  24. #ifndef __MINGW32__
  25. #include <sys/wait.h>
  26. #else
  27. #include <windows.h>
  28. #endif
  29. #include <grass/config.h>
  30. #include <grass/gis.h>
  31. #include <grass/glocale.h>
  32. #include <grass/spawn.h>
  33. /** \def MAX_ARGS Maximum number of arguments */
  34. /** \def MAX_BINDINGS Maximum number of bindings */
  35. /** \def MAX_SIGNALS Maximum number of signals */
  36. /** \def MAX_REDIRECTS Maximum number of redirects */
  37. #define MAX_ARGS 256
  38. #define MAX_BINDINGS 256
  39. #define MAX_SIGNALS 32
  40. #define MAX_REDIRECTS 32
  41. /**
  42. * \brief Spawns a new process.
  43. *
  44. * A more useful alternative to G_system(), which takes the
  45. * arguments of <b>command</b> as parameters.
  46. *
  47. * \param[in] command command to execute
  48. * \return -1 on error
  49. * \return process status on success
  50. */
  51. struct redirect
  52. {
  53. int dst_fd;
  54. int src_fd;
  55. const char *file;
  56. int mode;
  57. };
  58. struct signal
  59. {
  60. int which;
  61. int action;
  62. int signum;
  63. int valid;
  64. #ifndef __MINGW32__
  65. struct sigaction old_act;
  66. sigset_t old_mask;
  67. #endif
  68. };
  69. struct binding
  70. {
  71. const char *var;
  72. const char *val;
  73. };
  74. struct spawn
  75. {
  76. const char *args[MAX_ARGS];
  77. int num_args;
  78. struct redirect redirects[MAX_REDIRECTS];
  79. int num_redirects;
  80. struct signal signals[MAX_SIGNALS];
  81. int num_signals;
  82. struct binding bindings[MAX_BINDINGS];
  83. int num_bindings;
  84. int background;
  85. const char *directory;
  86. };
  87. static void parse_arglist(struct spawn *sp, va_list va);
  88. static void parse_argvec(struct spawn *sp, const char **va);
  89. #ifdef __MINGW32__
  90. struct buffer {
  91. char *str;
  92. size_t len;
  93. size_t size;
  94. };
  95. static const int INCREMENT = 50;
  96. static void clear(struct buffer *b)
  97. {
  98. b->len = 0;
  99. b->str[b->len] = '\0';
  100. }
  101. static void init(struct buffer *b)
  102. {
  103. b->str = G_malloc(1);
  104. b->size = 1;
  105. clear(b);
  106. }
  107. static char *release(struct buffer *b)
  108. {
  109. char *p = b->str;
  110. b->str = NULL;
  111. b->size = 0;
  112. b->len = 0;
  113. return p;
  114. }
  115. static void finish(struct buffer *b)
  116. {
  117. if (b->str)
  118. G_free(b->str);
  119. release(b);
  120. }
  121. static void ensure(struct buffer *b, size_t n)
  122. {
  123. if (b->size <= b->len + n + 1) {
  124. b->size = b->len + n + INCREMENT;
  125. b->str = G_realloc(b->str, b->size);
  126. }
  127. }
  128. static void append(struct buffer *b, const char *str)
  129. {
  130. size_t n = strlen(str);
  131. ensure(b, n);
  132. memcpy(&b->str[b->len], str, n);
  133. b->len += n;
  134. b->str[b->len] = '\0';
  135. }
  136. static void append_char(struct buffer *b, char c)
  137. {
  138. ensure(b, 1);
  139. b->str[b->len] = c;
  140. b->len++;
  141. b->str[b->len] = '\0';
  142. }
  143. static void escape_arg(struct buffer *result, const char *arg)
  144. {
  145. struct buffer buf;
  146. int quote, j;
  147. init(&buf);
  148. quote = arg[0] == '\0' || strchr(arg, ' ') || strchr(arg, '\t');
  149. if (quote)
  150. append_char(result, '\"');
  151. for (j = 0; arg[j]; j++) {
  152. int c = arg[j];
  153. int k;
  154. switch (c) {
  155. case '\\':
  156. append_char(&buf, '\\');
  157. break;
  158. case '\"':
  159. for (k = 0; k < buf.len; k++)
  160. append(result, "\\\\");
  161. clear(&buf);
  162. append(result, "\\\"");
  163. break;
  164. default:
  165. if (buf.len > 0) {
  166. append(result, buf.str);
  167. clear(&buf);
  168. }
  169. append_char(result, c);
  170. }
  171. }
  172. if (buf.len > 0)
  173. append(result, buf.str);
  174. if (quote) {
  175. append(result, buf.str);
  176. append_char(result, '\"');
  177. }
  178. finish(&buf);
  179. }
  180. static char *check_program(const char *pgm, const char *dir, const char *ext)
  181. {
  182. char pathname[GPATH_MAX];
  183. sprintf(pathname, "%s%s%s%s", dir, *dir ? "\\" : "", pgm, ext);
  184. return access(pathname, 0) == 0
  185. ? G_store(pathname)
  186. : NULL;
  187. }
  188. static char *find_program_ext(const char *pgm, const char *dir, char **pathext)
  189. {
  190. char *result;
  191. int i;
  192. if (result = check_program(pgm, dir, ""), result)
  193. return result;
  194. for (i = 0; pathext[i]; i++) {
  195. const char *ext = pathext[i];
  196. if (result = check_program(pgm, dir, ext), result)
  197. return result;
  198. }
  199. return NULL;
  200. }
  201. static char *find_program_dir_ext(const char *pgm, char **path, char **pathext)
  202. {
  203. char *result = NULL;
  204. int i;
  205. if (strchr(pgm, '\\') || strchr(pgm, '/')) {
  206. if (result = find_program_ext(pgm, "", pathext), result)
  207. return result;
  208. }
  209. else {
  210. if (result = find_program_ext(pgm, ".", pathext), result)
  211. return result;
  212. for (i = 0; path[i]; i++) {
  213. const char *dir = path[i];
  214. if (result = find_program_ext(pgm, dir, pathext), result)
  215. return result;
  216. }
  217. }
  218. return NULL;
  219. }
  220. static char *find_program(const char *pgm)
  221. {
  222. char **path = G_tokenize(getenv("PATH"), ";");
  223. char **pathext = G_tokenize(getenv("PATHEXT"), ";");
  224. char *result = find_program_dir_ext(pgm, path, pathext);
  225. G_free_tokens(path);
  226. G_free_tokens(pathext);
  227. return result;
  228. }
  229. static char *make_command_line(int shell, const char *cmd, const char **argv)
  230. {
  231. struct buffer result;
  232. int i;
  233. init(&result);
  234. if (shell) {
  235. const char *comspec = getenv("COMSPEC");
  236. append(&result, comspec ? comspec : "cmd.exe");
  237. append(&result, " /c \"");
  238. escape_arg(&result, cmd);
  239. }
  240. for (i = shell ? 1 : 0; argv[i]; i++) {
  241. if (result.len > 0)
  242. append_char(&result, ' ');
  243. escape_arg(&result, argv[i]);
  244. }
  245. append(&result, "\"");
  246. return release(&result);
  247. }
  248. static char *make_environment(const char **envp)
  249. {
  250. struct buffer result;
  251. int i;
  252. init(&result);
  253. for (i = 0; envp[i]; i++) {
  254. const char *env = envp[i];
  255. append(&result, env);
  256. append_char(&result, '\0');
  257. }
  258. return release(&result);
  259. }
  260. static HANDLE get_handle(int fd)
  261. {
  262. HANDLE h1, h2;
  263. if (fd < 0)
  264. return INVALID_HANDLE_VALUE;
  265. h1 = (HANDLE) _get_osfhandle(fd);
  266. if (!DuplicateHandle(GetCurrentProcess(), h1,
  267. GetCurrentProcess(), &h2,
  268. 0, TRUE, DUPLICATE_SAME_ACCESS))
  269. return INVALID_HANDLE_VALUE;
  270. return h2;
  271. }
  272. static int win_spawn(const char *cmd, const char **argv, const char **envp,
  273. const char *cwd, HANDLE handles[3], int background,
  274. int shell)
  275. {
  276. char *args = make_command_line(shell, cmd, argv);
  277. char *env = make_environment(envp);
  278. char *program = shell ? NULL : find_program(cmd);
  279. STARTUPINFO si;
  280. PROCESS_INFORMATION pi;
  281. BOOL result;
  282. DWORD exitcode;
  283. int i;
  284. if (!shell) {
  285. G_debug(3, "win_spawn: program = %s", program);
  286. if (!program) {
  287. G_free(args);
  288. G_free(env);
  289. return -1;
  290. }
  291. }
  292. G_debug(3, "win_spawn: args = %s", args);
  293. memset(&si, 0, sizeof(si));
  294. si.cb = sizeof(si);
  295. si.dwFlags |= STARTF_USESTDHANDLES;
  296. si.hStdInput = handles[0];
  297. si.hStdOutput = handles[1];
  298. si.hStdError = handles[2];
  299. result = CreateProcess(
  300. program, /* lpApplicationName */
  301. args, /* lpCommandLine */
  302. NULL, /* lpProcessAttributes */
  303. NULL, /* lpThreadAttributes */
  304. 1, /* bInheritHandles */
  305. 0, /* dwCreationFlags */
  306. env, /* lpEnvironment */
  307. cwd, /* lpCurrentDirectory */
  308. &si, /* lpStartupInfo */
  309. &pi /* lpProcessInformation */
  310. );
  311. G_free(args);
  312. G_free(env);
  313. G_free(program);
  314. if (!result) {
  315. G_warning(_("CreateProcess() failed: error = %d"), GetLastError());
  316. return -1;
  317. }
  318. CloseHandle(pi.hThread);
  319. for (i = 0; i < 3; i++)
  320. if (handles[i] != INVALID_HANDLE_VALUE)
  321. CloseHandle(handles[i]);
  322. if (!background) {
  323. WaitForSingleObject(pi.hProcess, INFINITE);
  324. if (!GetExitCodeProcess(pi.hProcess, &exitcode))
  325. return -1;
  326. CloseHandle(pi.hProcess);
  327. return (int) exitcode;
  328. }
  329. CloseHandle(pi.hProcess);
  330. return pi.dwProcessId;
  331. }
  332. static void do_redirects(struct redirect *redirects, int num_redirects, HANDLE handles[3])
  333. {
  334. int i;
  335. for (i = 0; i < 3; i++)
  336. handles[i] = get_handle(i);
  337. for (i = 0; i < num_redirects; i++) {
  338. struct redirect *r = &redirects[i];
  339. if (r->dst_fd < 0 || r->dst_fd > 2) {
  340. if (r->file || r->src_fd >= 0)
  341. G_warning(_("G_spawn: unable to redirect descriptor %d"), r->dst_fd);
  342. continue;
  343. }
  344. if (r->file) {
  345. r->src_fd = open(r->file, r->mode, 0666);
  346. if (r->src_fd < 0) {
  347. G_warning(_("G_spawn: unable to open file %s"), r->file);
  348. _exit(127);
  349. }
  350. handles[r->dst_fd] = get_handle(r->src_fd);
  351. close(r->src_fd);
  352. }
  353. else if (r->src_fd >= 0) {
  354. handles[r->dst_fd] = get_handle(r->src_fd);
  355. }
  356. else {
  357. if (r->dst_fd < 3) {
  358. CloseHandle(handles[r->dst_fd]);
  359. handles[r->dst_fd] = INVALID_HANDLE_VALUE;
  360. }
  361. close(r->dst_fd);
  362. }
  363. }
  364. }
  365. static void add_binding(const char **env, int *pnum, const struct binding *b)
  366. {
  367. char *str = G_malloc(strlen(b->var) + strlen(b->val) + 2);
  368. int n = *pnum;
  369. int i;
  370. sprintf(str, "%s=%s", b->var, b->val);
  371. for (i = 0; i < n; i++)
  372. if (G_strcasecmp(env[i], b->var) == 0) {
  373. env[i] = str;
  374. return;
  375. }
  376. env[n++] = str;
  377. *pnum = n;
  378. }
  379. static const char **do_bindings(const struct binding *bindings, int num_bindings)
  380. {
  381. const char **newenv;
  382. int i, n;
  383. for (i = 0; _environ[i]; i++)
  384. ;
  385. n = i;
  386. newenv = G_malloc((num_bindings + n + 1) * sizeof(char *));
  387. for (i = 0; i < n; i++)
  388. newenv[i] = _environ[i];
  389. for (i = 0; i < num_bindings; i++)
  390. add_binding(newenv, &n, &bindings[i]);
  391. newenv[num_bindings + n] = NULL;
  392. return newenv;
  393. }
  394. static int do_spawn(struct spawn *sp, const char *command)
  395. {
  396. HANDLE handles[3];
  397. const char **env;
  398. int status;
  399. do_redirects(sp->redirects, sp->num_redirects, handles);
  400. env = do_bindings(sp->bindings, sp->num_bindings);
  401. status = win_spawn(command, sp->args, env, sp->directory, handles, sp->background, 1);
  402. if (!sp->background && status < 0)
  403. G_warning(_("G_spawn: unable to execute command"));
  404. return status;
  405. }
  406. #else /* __MINGW32__ */
  407. static int undo_signals(const struct signal *signals, int num_signals, int which)
  408. {
  409. int error = 0;
  410. int i;
  411. for (i = num_signals - 1; i >= 0; i--) {
  412. const struct signal *s = &signals[i];
  413. if (s->which != which)
  414. continue;
  415. if (!s->valid)
  416. continue;
  417. switch (s->action) {
  418. case SSA_IGNORE:
  419. case SSA_DEFAULT:
  420. if (sigaction(s->signum, &s->old_act, NULL) < 0) {
  421. G_warning(_("G_spawn: unable to restore signal %d"),
  422. s->signum);
  423. error = 1;
  424. }
  425. break;
  426. case SSA_BLOCK:
  427. case SSA_UNBLOCK:
  428. if (sigprocmask(SIG_UNBLOCK, &s->old_mask, NULL) < 0) {
  429. G_warning(_("G_spawn: unable to restore signal %d"),
  430. s->signum);
  431. error = 1;
  432. }
  433. break;
  434. }
  435. }
  436. return !error;
  437. }
  438. static int do_signals(struct signal *signals, int num_signals, int which)
  439. {
  440. struct sigaction act;
  441. sigset_t mask;
  442. int error = 0;
  443. int i;
  444. sigemptyset(&act.sa_mask);
  445. act.sa_flags = SA_RESTART;
  446. for (i = 0; i < num_signals; i++) {
  447. struct signal *s = &signals[i];
  448. if (s->which != which)
  449. continue;
  450. switch (s->action) {
  451. case SSA_IGNORE:
  452. act.sa_handler = SIG_IGN;
  453. if (sigaction(s->signum, &act, &s->old_act) < 0) {
  454. G_warning(_("G_spawn: unable to reset signal %d"), s->signum);
  455. error = 1;
  456. }
  457. else
  458. s->valid = 1;
  459. break;
  460. case SSA_DEFAULT:
  461. act.sa_handler = SIG_DFL;
  462. if (sigaction(s->signum, &act, &s->old_act) < 0) {
  463. G_warning(_("G_spawn: unable to ignore signal %d"),
  464. s->signum);
  465. error = 1;
  466. }
  467. else
  468. s->valid = 1;
  469. break;
  470. case SSA_BLOCK:
  471. sigemptyset(&mask);
  472. sigaddset(&mask, s->signum);
  473. if (sigprocmask(SIG_BLOCK, &mask, &s->old_mask) < 0) {
  474. G_warning(_("G_spawn: unable to block signal %d"), s->signum);
  475. error = 1;
  476. }
  477. break;
  478. case SSA_UNBLOCK:
  479. sigemptyset(&mask);
  480. sigaddset(&mask, s->signum);
  481. if (sigprocmask(SIG_UNBLOCK, &mask, &s->old_mask) < 0) {
  482. G_warning(_("G_spawn: unable to unblock signal %d"),
  483. s->signum);
  484. error = 1;
  485. }
  486. else
  487. s->valid = 1;
  488. break;
  489. }
  490. }
  491. return !error;
  492. }
  493. static void do_redirects(struct redirect *redirects, int num_redirects)
  494. {
  495. int i;
  496. for (i = 0; i < num_redirects; i++) {
  497. struct redirect *r = &redirects[i];
  498. if (r->file) {
  499. r->src_fd = open(r->file, r->mode, 0666);
  500. if (r->src_fd < 0) {
  501. G_warning(_("G_spawn: unable to open file %s"), r->file);
  502. _exit(127);
  503. }
  504. if (dup2(r->src_fd, r->dst_fd) < 0) {
  505. G_warning(_("G_spawn: unable to duplicate descriptor %d to %d"),
  506. r->src_fd, r->dst_fd);
  507. _exit(127);
  508. }
  509. close(r->src_fd);
  510. }
  511. else if (r->src_fd >= 0) {
  512. if (dup2(r->src_fd, r->dst_fd) < 0) {
  513. G_warning(_("G_spawn: unable to duplicate descriptor %d to %d"),
  514. r->src_fd, r->dst_fd);
  515. _exit(127);
  516. }
  517. }
  518. else
  519. close(r->dst_fd);
  520. }
  521. }
  522. static void do_bindings(const struct binding *bindings, int num_bindings)
  523. {
  524. int i;
  525. for (i = 0; i < num_bindings; i++) {
  526. const struct binding *b = &bindings[i];
  527. char *str = G_malloc(strlen(b->var) + strlen(b->val) + 2);
  528. sprintf(str, "%s=%s", b->var, b->val);
  529. putenv(str);
  530. }
  531. }
  532. static int do_spawn(struct spawn *sp, const char *command)
  533. {
  534. int status = -1;
  535. pid_t pid;
  536. if (!do_signals(sp->signals, sp->num_signals, SST_PRE))
  537. return status;
  538. pid = fork();
  539. if (pid < 0) {
  540. G_warning(_("Unable to create a new process: %s"), strerror(errno));
  541. undo_signals(sp->signals, sp->num_signals, SST_PRE);
  542. return status;
  543. }
  544. if (pid == 0) {
  545. if (!undo_signals(sp->signals, sp->num_signals, SST_PRE))
  546. _exit(127);
  547. if (!do_signals(sp->signals, sp->num_signals, SST_CHILD))
  548. _exit(127);
  549. if (sp->directory)
  550. if (chdir(sp->directory) < 0) {
  551. G_warning(_("Unable to change directory to %s"), sp->directory);
  552. _exit(127);
  553. }
  554. do_redirects(sp->redirects, sp->num_redirects);
  555. do_bindings(sp->bindings, sp->num_bindings);
  556. execvp(command, (char **)sp->args);
  557. G_warning(_("Unable to execute command '%s': %s"), command, strerror(errno));
  558. _exit(127);
  559. }
  560. do_signals(sp->signals, sp->num_signals, SST_POST);
  561. if (sp->background)
  562. status = (int)pid;
  563. else {
  564. pid_t n;
  565. do
  566. n = waitpid(pid, &status, 0);
  567. while (n == (pid_t) - 1 && errno == EINTR);
  568. if (n != pid)
  569. status = -1;
  570. else {
  571. if (WIFEXITED(status))
  572. status = WEXITSTATUS(status);
  573. else if (WIFSIGNALED(status))
  574. status = WTERMSIG(status);
  575. else
  576. status = -0x100;
  577. }
  578. }
  579. undo_signals(sp->signals, sp->num_signals, SST_POST);
  580. undo_signals(sp->signals, sp->num_signals, SST_PRE);
  581. return status;
  582. }
  583. #endif /* __MINGW32__ */
  584. static void begin_spawn(struct spawn *sp)
  585. {
  586. sp->num_args = 0;
  587. sp->num_redirects = 0;
  588. sp->num_signals = 0;
  589. sp->num_bindings = 0;
  590. sp->background = 0;
  591. sp->directory = NULL;
  592. }
  593. #define NEXT_ARG(var, type) ((type) *(var)++)
  594. #define NEXT_ARG_INT(var) (int)((intptr_t) *(var)++)
  595. static void parse_argvec(struct spawn *sp, const char **va)
  596. {
  597. for (;;) {
  598. const char *arg = NEXT_ARG(va, const char *);
  599. const char *var, *val;
  600. if (!arg) {
  601. sp->args[sp->num_args++] = NULL;
  602. break;
  603. }
  604. else if (arg == SF_REDIRECT_FILE) {
  605. sp->redirects[sp->num_redirects].dst_fd = NEXT_ARG_INT(va);
  606. sp->redirects[sp->num_redirects].src_fd = -1;
  607. sp->redirects[sp->num_redirects].mode = NEXT_ARG_INT(va);
  608. sp->redirects[sp->num_redirects].file = NEXT_ARG(va, const char *);
  609. sp->num_redirects++;
  610. }
  611. else if (arg == SF_REDIRECT_DESCRIPTOR) {
  612. sp->redirects[sp->num_redirects].dst_fd = NEXT_ARG_INT(va);
  613. sp->redirects[sp->num_redirects].src_fd = NEXT_ARG_INT(va);
  614. sp->redirects[sp->num_redirects].file = NULL;
  615. sp->num_redirects++;
  616. }
  617. else if (arg == SF_CLOSE_DESCRIPTOR) {
  618. sp->redirects[sp->num_redirects].dst_fd = NEXT_ARG_INT(va);
  619. sp->redirects[sp->num_redirects].src_fd = -1;
  620. sp->redirects[sp->num_redirects].file = NULL;
  621. sp->num_redirects++;
  622. }
  623. else if (arg == SF_SIGNAL) {
  624. sp->signals[sp->num_signals].which = NEXT_ARG_INT(va);
  625. sp->signals[sp->num_signals].action = NEXT_ARG_INT(va);
  626. sp->signals[sp->num_signals].signum = NEXT_ARG_INT(va);
  627. sp->signals[sp->num_signals].valid = 0;
  628. sp->num_signals++;
  629. }
  630. else if (arg == SF_VARIABLE) {
  631. var = NEXT_ARG(va, const char *);
  632. val = getenv(var);
  633. sp->args[sp->num_args++] = val ? val : "";
  634. }
  635. else if (arg == SF_BINDING) {
  636. sp->bindings[sp->num_bindings].var = NEXT_ARG(va, const char *);
  637. sp->bindings[sp->num_bindings].val = NEXT_ARG(va, const char *);
  638. sp->num_bindings++;
  639. }
  640. else if (arg == SF_BACKGROUND) {
  641. sp->background = 1;
  642. }
  643. else if (arg == SF_DIRECTORY) {
  644. sp->directory = NEXT_ARG(va, const char *);
  645. }
  646. else if (arg == SF_ARGVEC) {
  647. parse_argvec(sp, NEXT_ARG(va, const char **));
  648. }
  649. else
  650. sp->args[sp->num_args++] = arg;
  651. }
  652. }
  653. static void parse_arglist(struct spawn *sp, va_list va)
  654. {
  655. for (;;) {
  656. const char *arg = va_arg(va, const char *);
  657. const char *var, *val;
  658. if (!arg) {
  659. sp->args[sp->num_args++] = NULL;
  660. break;
  661. }
  662. else if (arg == SF_REDIRECT_FILE) {
  663. sp->redirects[sp->num_redirects].dst_fd = va_arg(va, int);
  664. sp->redirects[sp->num_redirects].src_fd = -1;
  665. sp->redirects[sp->num_redirects].mode = va_arg(va, int);
  666. sp->redirects[sp->num_redirects].file = va_arg(va, const char *);
  667. sp->num_redirects++;
  668. }
  669. else if (arg == SF_REDIRECT_DESCRIPTOR) {
  670. sp->redirects[sp->num_redirects].dst_fd = va_arg(va, int);
  671. sp->redirects[sp->num_redirects].src_fd = va_arg(va, int);
  672. sp->redirects[sp->num_redirects].file = NULL;
  673. sp->num_redirects++;
  674. }
  675. else if (arg == SF_CLOSE_DESCRIPTOR) {
  676. sp->redirects[sp->num_redirects].dst_fd = va_arg(va, int);
  677. sp->redirects[sp->num_redirects].src_fd = -1;
  678. sp->redirects[sp->num_redirects].file = NULL;
  679. sp->num_redirects++;
  680. }
  681. else if (arg == SF_SIGNAL) {
  682. sp->signals[sp->num_signals].which = va_arg(va, int);
  683. sp->signals[sp->num_signals].action = va_arg(va, int);
  684. sp->signals[sp->num_signals].signum = va_arg(va, int);
  685. sp->signals[sp->num_signals].valid = 0;
  686. sp->num_signals++;
  687. }
  688. else if (arg == SF_VARIABLE) {
  689. var = va_arg(va, char *);
  690. val = getenv(var);
  691. sp->args[sp->num_args++] = val ? val : "";
  692. }
  693. else if (arg == SF_BINDING) {
  694. sp->bindings[sp->num_bindings].var = va_arg(va, const char *);
  695. sp->bindings[sp->num_bindings].val = va_arg(va, const char *);
  696. sp->num_bindings++;
  697. }
  698. else if (arg == SF_BACKGROUND) {
  699. sp->background = 1;
  700. }
  701. else if (arg == SF_DIRECTORY) {
  702. sp->directory = va_arg(va, const char *);
  703. }
  704. else if (arg == SF_ARGVEC) {
  705. parse_argvec(sp, va_arg(va, const char **));
  706. }
  707. else
  708. sp->args[sp->num_args++] = arg;
  709. }
  710. }
  711. /**
  712. * \brief Spawn new process based on <b>command</b>.
  713. *
  714. * This is a more advanced version of G_spawn().
  715. *
  716. * \param[in] command
  717. * \param[in] args arguments
  718. * \return -1 on error
  719. * \return process status on success
  720. */
  721. int G_vspawn_ex(const char *command, const char **args)
  722. {
  723. struct spawn sp;
  724. begin_spawn(&sp);
  725. parse_argvec(&sp, args);
  726. return do_spawn(&sp, command);
  727. }
  728. /**
  729. * \brief Spawn new process based on <b>command</b>.
  730. *
  731. * This is a more advanced version of G_spawn().
  732. *
  733. * \param[in] command
  734. * \return -1 on error
  735. * \return process status on success
  736. */
  737. int G_spawn_ex(const char *command, ...)
  738. {
  739. struct spawn sp;
  740. va_list va;
  741. begin_spawn(&sp);
  742. va_start(va, command);
  743. parse_arglist(&sp, va);
  744. va_end(va);
  745. return do_spawn(&sp, command);
  746. }
  747. /**
  748. * \brief Spawn new process based on <b>command</b>.
  749. *
  750. * \param[in] command
  751. * \return -1 on error
  752. * \return process status on success
  753. */
  754. int G_spawn(const char *command, ...)
  755. {
  756. const char *args[MAX_ARGS];
  757. int num_args = 0, i;
  758. va_list va;
  759. int status = -1;
  760. va_start(va, command);
  761. for (i = 0; ; i++) {
  762. const char *arg = va_arg(va, const char *);
  763. args[num_args++] = arg;
  764. if (!arg)
  765. break;
  766. }
  767. va_end(va);
  768. status = G_spawn_ex(
  769. command,
  770. #ifndef __MINGW32__
  771. SF_SIGNAL, SST_PRE, SSA_IGNORE, SIGINT,
  772. SF_SIGNAL, SST_PRE, SSA_IGNORE, SIGQUIT,
  773. SF_SIGNAL, SST_PRE, SSA_BLOCK, SIGCHLD,
  774. #endif
  775. SF_ARGVEC, args,
  776. NULL);
  777. return status;
  778. }
  779. int G_wait(int i_pid)
  780. {
  781. #ifdef __MINGW32__
  782. DWORD rights = PROCESS_QUERY_INFORMATION | SYNCHRONIZE;
  783. HANDLE hProcess = OpenProcess(rights, FALSE, (DWORD) i_pid);
  784. DWORD exitcode;
  785. if (!hProcess)
  786. return -1;
  787. WaitForSingleObject(hProcess, INFINITE);
  788. if (!GetExitCodeProcess(hProcess, &exitcode))
  789. exitcode = (DWORD) -1;
  790. CloseHandle(hProcess);
  791. return (int) exitcode;
  792. #else
  793. pid_t pid = (pid_t) i_pid;
  794. int status = -1;
  795. pid_t n;
  796. do
  797. n = waitpid(pid, &status, 0);
  798. while (n == (pid_t) - 1 && errno == EINTR);
  799. if (n != pid)
  800. return -1;
  801. else {
  802. if (WIFEXITED(status))
  803. return WEXITSTATUS(status);
  804. else if (WIFSIGNALED(status))
  805. return WTERMSIG(status);
  806. else
  807. return -0x100;
  808. }
  809. #endif
  810. }