mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-03-03 15:38:22 +00:00
a359de7893
This changes *NSYNC to allocate waiters on the stack so our locks don't need to depend on dynamic memory. This make our runtiem simpler, and it also fixes bugs with thread cancellation support.
111 lines
5.3 KiB
C
111 lines
5.3 KiB
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 2021 Justine Alexandra Roberts Tunney │
|
||
│ │
|
||
│ Permission to use, copy, modify, and/or distribute this software for │
|
||
│ any purpose with or without fee is hereby granted, provided that the │
|
||
│ above copyright notice and this permission notice appear in all copies. │
|
||
│ │
|
||
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL │
|
||
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED │
|
||
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE │
|
||
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL │
|
||
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR │
|
||
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER │
|
||
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR │
|
||
│ PERFORMANCE OF THIS SOFTWARE. │
|
||
╚─────────────────────────────────────────────────────────────────────────────*/
|
||
#include "libc/calls/ntspawn.h"
|
||
#include "libc/assert.h"
|
||
#include "libc/calls/struct/sigaction.internal.h"
|
||
#include "libc/calls/syscall_support-nt.internal.h"
|
||
#include "libc/errno.h"
|
||
#include "libc/intrin/pushpop.internal.h"
|
||
#include "libc/macros.internal.h"
|
||
#include "libc/nt/enum/filemapflags.h"
|
||
#include "libc/nt/enum/pageflags.h"
|
||
#include "libc/nt/enum/processcreationflags.h"
|
||
#include "libc/nt/errors.h"
|
||
#include "libc/nt/memory.h"
|
||
#include "libc/nt/process.h"
|
||
#include "libc/nt/runtime.h"
|
||
#include "libc/nt/struct/processinformation.h"
|
||
#include "libc/nt/struct/securityattributes.h"
|
||
#include "libc/nt/struct/startupinfo.h"
|
||
#include "libc/sysv/errfuns.h"
|
||
#ifdef __x86_64__
|
||
|
||
struct SpawnBlock {
|
||
union {
|
||
struct {
|
||
char16_t cmdline[ARG_MAX / 2];
|
||
char16_t envvars[ARG_MAX / 2];
|
||
char buf[ARG_MAX];
|
||
};
|
||
char __pad[ROUNDUP(ARG_MAX / 2 * 3 * sizeof(char16_t), FRAMESIZE)];
|
||
};
|
||
};
|
||
|
||
/**
|
||
* Spawns process on Windows NT.
|
||
*
|
||
* This function delegates to CreateProcess() with UTF-8 → UTF-16
|
||
* translation and argv escaping. Please note this will NOT escape
|
||
* command interpreter syntax.
|
||
*
|
||
* @param prog won't be PATH searched
|
||
* @param argv specifies prog arguments
|
||
* @param envp[𝟶,m-2] specifies "foo=bar" environment variables, which
|
||
* don't need to be passed in sorted order; however, this function
|
||
* goes faster the closer they are to sorted
|
||
* @param envp[m-1] is NULL
|
||
* @param extravar is added to envp to avoid setenv() in caller
|
||
* @param bInheritHandles means handles already marked inheritable will
|
||
* be inherited; which, assuming the System V wrapper functions are
|
||
* being used, should mean (1) all files and sockets that weren't
|
||
* opened with O_CLOEXEC; and (2) all memory mappings
|
||
* @param opt_out_lpProcessInformation can be used to return process and
|
||
* thread IDs to parent, as well as open handles that need close()
|
||
* @return 0 on success, or -1 w/ errno
|
||
* @see spawnve() which abstracts this function
|
||
*/
|
||
textwindows int ntspawn(
|
||
const char *prog, char *const argv[], char *const envp[],
|
||
const char *extravar,
|
||
const struct NtSecurityAttributes *opt_lpProcessAttributes,
|
||
const struct NtSecurityAttributes *opt_lpThreadAttributes,
|
||
bool32 bInheritHandles, uint32_t dwCreationFlags,
|
||
const char16_t *opt_lpCurrentDirectory,
|
||
const struct NtStartupInfo *lpStartupInfo,
|
||
struct NtProcessInformation *opt_out_lpProcessInformation) {
|
||
int rc;
|
||
int64_t handle;
|
||
struct SpawnBlock *block;
|
||
char16_t prog16[PATH_MAX];
|
||
rc = -1;
|
||
block = NULL;
|
||
_init_sigchld();
|
||
if (__mkntpath(prog, prog16) == -1) return -1;
|
||
if ((handle = CreateFileMapping(-1, 0, pushpop(kNtPageReadwrite), 0,
|
||
sizeof(*block), 0)) &&
|
||
(block = MapViewOfFileEx(handle, kNtFileMapRead | kNtFileMapWrite, 0, 0,
|
||
sizeof(*block), 0)) &&
|
||
mkntcmdline(block->cmdline, argv) != -1 &&
|
||
mkntenvblock(block->envvars, envp, extravar, block->buf) != -1 &&
|
||
CreateProcess(prog16, block->cmdline, opt_lpProcessAttributes,
|
||
opt_lpThreadAttributes, bInheritHandles,
|
||
dwCreationFlags | kNtCreateUnicodeEnvironment |
|
||
kNtInheritParentAffinity,
|
||
block->envvars, opt_lpCurrentDirectory, lpStartupInfo,
|
||
opt_out_lpProcessInformation)) {
|
||
rc = 0;
|
||
} else if (GetLastError() == kNtErrorSharingViolation) {
|
||
etxtbsy();
|
||
}
|
||
if (block) UnmapViewOfFile(block);
|
||
if (handle) CloseHandle(handle);
|
||
return __fix_enotdir(rc, prog16);
|
||
}
|
||
|
||
#endif /* __x86_64__ */
|