cosmopolitan/libc/calls/hefty/spawnve-sysv.c

/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :vi│
╞══════════════════════════════════════════════════════════════════════════════╡
│ Copyright 2020 Justine Alexandra Roberts Tunney                              │
│                                                                              │
│ This program is free software; you can redistribute it and/or modify         │
│ it under the terms of the GNU General Public License as published by         │
│ the Free Software Foundation; version 2 of the License.                      │
│                                                                              │
│ This program is distributed in the hope that it will be useful, but          │
│ WITHOUT ANY WARRANTY; without even the implied warranty of                   │
│ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU             │
│ General Public License for more details.                                     │
│                                                                              │
│ You should have received a copy of the GNU General Public License            │
│ along with this program; if not, write to the Free Software                  │
│ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA                │
│ 02110-1301 USA                                                               │
╚─────────────────────────────────────────────────────────────────────────────*/
#include "libc/assert.h"
#include "libc/calls/calls.h"
#include "libc/calls/hefty/internal.h"
#include "libc/calls/hefty/spawn.h"
#include "libc/calls/internal.h"
#include "libc/conv/conv.h"
#include "libc/dce.h"
#include "libc/mem/mem.h"
#include "libc/paths.h"
#include "libc/runtime/runtime.h"
#include "libc/str/str.h"
#include "libc/sysv/consts/at.h"
#include "libc/sysv/consts/fileno.h"
#include "libc/sysv/consts/o.h"

int spawnve$sysv(unsigned flags, int stdiofds[3], const char *program,
                 char *const argv[], char *const envp[]) {
  int rc, pid, fd;
  size_t i, j, len;
  int32_t tubes[3][2];
  char **argv2, MZqFpD[8];
  void *argv3;

  pid = 0;
  argv2 = NULL;
  argv3 = argv;

  /*
   * αcτµαlly pδrταblε εxεcµταblε w/ thompson shell script
   * morphology needs to be launched via command interpreter.
   */
  if (endswith(program, ".com") || endswith(program, ".exe")) {
    memset(MZqFpD, 0, sizeof(MZqFpD));
    fd = openat$sysv(AT_FDCWD, program, O_RDONLY, 0);
    read$sysv(fd, MZqFpD, sizeof(MZqFpD));
    close$sysv(fd);
    if (memcmp(MZqFpD, "MZqFpD", 6) == 0) {
      /*
       * If we got this:
       *
       *   spawn(/bin/echo, [echo, hi, there])
       *
       * It will become this:
       *
       *   spawn(/bin/sh, [sh, /bin/echo, hi, there])
       */
      len = 1;
      while (argv[len]) len++;
      if ((argv2 = malloc((2 + len + 1) * sizeof(char *)))) {
        i = 0, j = 1;
        argv2[i++] = "sh";
        argv2[i++] = program;
        while (j < len) argv2[i++] = argv[j++];
        argv2[i] = NULL;
        argv3 = argv2;
        program = "/bin/sh";
      }
    }
  }

  for (i = 0; i < 3; ++i) {
    if (stdiofds[i] == -1) {
      pid |= pipe$sysv(tubes[i]);
    }
  }

  if (pid != -1) {
    if ((pid = vfork()) == 0) {
      if (flags & SPAWN_DETACH) {
        if (setsid() == -1) abort();
        if ((rc = fork$sysv()) == -1) abort();
        if (rc > 0) _exit(0);
      }
      for (i = 0; i < 3; ++i) {
        if (stdiofds[i] == -1) {
          close$sysv(tubes[i][kIoMotion[i]]);
          fd = tubes[i][!kIoMotion[i]];
        } else {
          fd = stdiofds[i];
        }
        dup2$sysv(fd, i);
      }
      execve$sysv(program, argv3, envp);
      abort();
    }
  }

  for (i = 0; i < 3; ++i) {
    if (stdiofds[i] == -1) {
      close$sysv(tubes[i][!kIoMotion[i]]);
      fd = tubes[i][kIoMotion[i]];
      if (pid != -1) {
        stdiofds[i] = fd;
      } else {
        close$sysv(fd);
      }
    }
  }

  if (argv2) free(argv2);
  return pid;
}