mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-01-31 19:43:32 +00:00
ec2cb88058
- Document more compiler flags - Expose new __print_maps() api - Better overflow checking in mmap() - Improve the shell example somewhat - Fix minor runtime bugs regarding stacks - Make kill() on fork()+execve()'d children work - Support CLONE_CHILD_CLEARTID for proper joining - Fix recent possible deadlock regression with --ftrace
109 lines
4.8 KiB
C
109 lines
4.8 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/calls.h"
|
|
#include "libc/calls/getconsolectrlevent.internal.h"
|
|
#include "libc/calls/internal.h"
|
|
#include "libc/dce.h"
|
|
#include "libc/intrin/kprintf.h"
|
|
#include "libc/macros.internal.h"
|
|
#include "libc/nt/console.h"
|
|
#include "libc/nt/enum/ctrlevent.h"
|
|
#include "libc/nt/enum/processaccess.h"
|
|
#include "libc/nt/enum/th32cs.h"
|
|
#include "libc/nt/errors.h"
|
|
#include "libc/nt/process.h"
|
|
#include "libc/nt/runtime.h"
|
|
#include "libc/nt/struct/processentry32.h"
|
|
#include "libc/sysv/errfuns.h"
|
|
|
|
textwindows int sys_kill_nt(int pid, int sig) {
|
|
bool32 ok;
|
|
int64_t h;
|
|
int event, ntpid;
|
|
|
|
// is killing everything except init really worth supporting?
|
|
if (pid == -1) return einval();
|
|
|
|
// XXX: NT doesn't really have process groups. For instance the
|
|
// CreateProcess() flag for starting a process group actually
|
|
// just does an "ignore ctrl-c" internally.
|
|
pid = ABS(pid);
|
|
|
|
// If we're targeting current process group then just call raise().
|
|
if (!pid || pid == getpid()) {
|
|
return raise(sig);
|
|
}
|
|
|
|
// GenerateConsoleCtrlEvent() will always signal groups and there's
|
|
// nothing we can do about it, unless we have a GUI GetMessage loop
|
|
// and alternatively create a centralized signal daemon like cygwin
|
|
if ((event = GetConsoleCtrlEvent(sig)) != -1) {
|
|
// we're killing with SIGINT or SIGQUIT which are the only two
|
|
// signals we can really use, since TerminateProcess() makes
|
|
// everything else effectively a SIGKILL ;_;
|
|
if (__isfdkind(pid, kFdProcess)) {
|
|
ntpid = GetProcessId(g_fds.p[pid].handle);
|
|
} else if (!__isfdopen(pid)) {
|
|
ntpid = pid; // XXX: suboptimal
|
|
} else {
|
|
return esrch();
|
|
}
|
|
if (GenerateConsoleCtrlEvent(event, ntpid)) {
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// is this a cosmo pid that was returned by fork?
|
|
if (__isfdkind(pid, kFdProcess)) {
|
|
// since windows can't execve we need to kill the grandchildren
|
|
// TODO(jart): should we just kill the whole tree too? there's
|
|
// no obvious way to tell if it's the execve shell
|
|
int64_t hSnap, hProc, hChildProc;
|
|
struct NtProcessEntry32 pe = {.dwSize = sizeof(struct NtProcessEntry32)};
|
|
ntpid = GetProcessId(g_fds.p[pid].handle);
|
|
hSnap = CreateToolhelp32Snapshot(kNtTh32csSnapprocess, 0);
|
|
if (Process32First(hSnap, &pe)) {
|
|
do {
|
|
if (pe.th32ParentProcessID == ntpid) {
|
|
if ((h = OpenProcess(kNtProcessTerminate, false, pe.th32ProcessID))) {
|
|
TerminateProcess(h, 128 + sig);
|
|
CloseHandle(h);
|
|
}
|
|
}
|
|
} while (Process32Next(hSnap, &pe));
|
|
}
|
|
ok = TerminateProcess(g_fds.p[pid].handle, 128 + sig);
|
|
if (!ok && GetLastError() == kNtErrorAccessDenied) ok = true;
|
|
return 0;
|
|
}
|
|
|
|
// XXX: Is this a raw new technology pid? Because that's messy.
|
|
if ((h = OpenProcess(kNtProcessTerminate, false, pid))) {
|
|
ok = TerminateProcess(h, 128 + sig);
|
|
if (!ok && GetLastError() == kNtErrorAccessDenied) {
|
|
ok = true; // cargo culting other codebases here
|
|
}
|
|
CloseHandle(h);
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|