mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-06-28 15:28:30 +00:00
Make improvements
- Add rusage to redbean Lua API - Add more redbean documentation - Add pledge() to redbean Lua API - Polyfill OpenBSD pledge() for Linux - Increase PATH_MAX limit to 1024 characters - Untrack sibling processes after fork() on Windows
This commit is contained in:
Justine Tunney
parent
9a6bd304a5
commit
47b3274665
212 changed files with 2251 additions and 834 deletions
143
test/libc/runtime/fork_test.c
Normal file
143
test/libc/runtime/fork_test.c
Normal file
|
|
@ -0,0 +1,143 @@
|
|||
/*-*- 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/sigbits.h"
|
||||
#include "libc/calls/struct/sigaction.h"
|
||||
#include "libc/dce.h"
|
||||
#include "libc/log/check.h"
|
||||
#include "libc/macros.internal.h"
|
||||
#include "libc/runtime/runtime.h"
|
||||
#include "libc/sysv/consts/map.h"
|
||||
#include "libc/sysv/consts/msync.h"
|
||||
#include "libc/sysv/consts/prot.h"
|
||||
#include "libc/sysv/consts/sig.h"
|
||||
#include "libc/testlib/ezbench.h"
|
||||
#include "libc/testlib/testlib.h"
|
||||
|
||||
TEST(fork, testPipes) {
|
||||
int a, b;
|
||||
int ws, pid;
|
||||
int pipefds[2];
|
||||
ASSERT_NE(-1, pipe(pipefds));
|
||||
ASSERT_NE(-1, (pid = fork()));
|
||||
if (!pid) {
|
||||
a = 31337;
|
||||
close(pipefds[0]);
|
||||
write(pipefds[1], &a, sizeof(a));
|
||||
close(pipefds[1]);
|
||||
_exit(0);
|
||||
}
|
||||
EXPECT_NE(-1, close(pipefds[1]));
|
||||
EXPECT_EQ(sizeof(b), read(pipefds[0], &b, sizeof(b)));
|
||||
EXPECT_NE(-1, close(pipefds[0]));
|
||||
EXPECT_NE(-1, waitpid(pid, &ws, 0));
|
||||
EXPECT_EQ(31337, b);
|
||||
}
|
||||
|
||||
TEST(fork, testSharedMemory) {
|
||||
int ws, pid;
|
||||
int stackvar;
|
||||
int *sharedvar;
|
||||
int *privatevar;
|
||||
EXPECT_NE(MAP_FAILED,
|
||||
(sharedvar = mmap(NULL, FRAMESIZE, PROT_READ | PROT_WRITE,
|
||||
MAP_SHARED | MAP_ANONYMOUS, -1, 0)));
|
||||
EXPECT_NE(MAP_FAILED,
|
||||
(privatevar = mmap(NULL, FRAMESIZE, PROT_READ | PROT_WRITE,
|
||||
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0)));
|
||||
stackvar = 1;
|
||||
*sharedvar = 1;
|
||||
*privatevar = 1;
|
||||
EXPECT_NE(-1, (pid = fork()));
|
||||
if (!pid) {
|
||||
EXPECT_EQ(NULL, getenv("_FORK"));
|
||||
++stackvar;
|
||||
++*sharedvar;
|
||||
++*privatevar;
|
||||
msync((void *)ROUNDDOWN((intptr_t)&stackvar, FRAMESIZE), FRAMESIZE,
|
||||
MS_SYNC);
|
||||
EXPECT_NE(-1, msync(privatevar, FRAMESIZE, MS_SYNC));
|
||||
EXPECT_NE(-1, msync(sharedvar, FRAMESIZE, MS_SYNC));
|
||||
_exit(0);
|
||||
}
|
||||
EXPECT_NE(-1, waitpid(pid, &ws, 0));
|
||||
EXPECT_EQ(1, stackvar);
|
||||
EXPECT_EQ(2, *sharedvar);
|
||||
EXPECT_EQ(1, *privatevar);
|
||||
EXPECT_NE(-1, munmap(sharedvar, FRAMESIZE));
|
||||
EXPECT_NE(-1, munmap(privatevar, FRAMESIZE));
|
||||
}
|
||||
|
||||
static volatile bool gotsigusr1;
|
||||
static volatile bool gotsigusr2;
|
||||
|
||||
static void OnSigusr1(int sig) {
|
||||
gotsigusr1 = true;
|
||||
}
|
||||
|
||||
static void OnSigusr2(int sig) {
|
||||
gotsigusr2 = true;
|
||||
}
|
||||
|
||||
TEST(fork, childToChild) {
|
||||
if (IsWindows()) return; // :'(
|
||||
sigset_t mask, oldmask;
|
||||
int ws, parent, child1, child2;
|
||||
gotsigusr1 = false;
|
||||
gotsigusr2 = false;
|
||||
parent = getpid();
|
||||
signal(SIGUSR1, OnSigusr1);
|
||||
signal(SIGUSR2, OnSigusr2);
|
||||
sigemptyset(&mask);
|
||||
sigaddset(&mask, SIGUSR2);
|
||||
sigprocmask(SIG_BLOCK, &mask, &oldmask);
|
||||
ASSERT_NE(-1, (child1 = fork()));
|
||||
if (!child1) {
|
||||
kill(parent, SIGUSR2);
|
||||
sigsuspend(0);
|
||||
_Exit(!gotsigusr1);
|
||||
}
|
||||
sigdelset(&mask, SIGUSR2);
|
||||
sigsuspend(&mask);
|
||||
ASSERT_NE(-1, (child2 = fork()));
|
||||
if (!child2) {
|
||||
kill(child1, SIGUSR1);
|
||||
_Exit(0);
|
||||
}
|
||||
ASSERT_NE(-1, wait(&ws));
|
||||
EXPECT_TRUE(WIFEXITED(ws));
|
||||
EXPECT_EQ(0, WEXITSTATUS(ws));
|
||||
ASSERT_NE(-1, wait(&ws));
|
||||
EXPECT_TRUE(WIFEXITED(ws));
|
||||
EXPECT_EQ(0, WEXITSTATUS(ws));
|
||||
sigprocmask(SIG_SETMASK, &oldmask, 0);
|
||||
}
|
||||
|
||||
void ForkInSerial(void) {
|
||||
int pid, ws;
|
||||
ASSERT_NE(-1, (pid = fork()));
|
||||
if (!pid) _Exit(0);
|
||||
ASSERT_NE(-1, waitpid(pid, &ws, 0));
|
||||
ASSERT_TRUE(WIFEXITED(ws));
|
||||
ASSERT_EQ(0, WEXITSTATUS(ws));
|
||||
}
|
||||
|
||||
BENCH(fork, bench) {
|
||||
EZBENCH2("fork", donothing, ForkInSerial());
|
||||
}
|
||||
|
|
@ -22,7 +22,7 @@
|
|||
|
||||
TEST(GetDosArgv, empty) {
|
||||
size_t max = 4;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(0, GetDosArgv(u"", buf, size, argv, max));
|
||||
|
|
@ -33,7 +33,7 @@ TEST(GetDosArgv, empty) {
|
|||
|
||||
TEST(GetDosArgv, emptyish) {
|
||||
size_t max = 4;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(0, GetDosArgv(u" ", buf, size, argv, max));
|
||||
|
|
@ -44,7 +44,7 @@ TEST(GetDosArgv, emptyish) {
|
|||
|
||||
TEST(GetDosArgv, basicUsage) {
|
||||
size_t max = 4;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(3, GetDosArgv(u"a\t \"b c\" d ", buf, size, argv, max));
|
||||
|
|
@ -58,7 +58,7 @@ TEST(GetDosArgv, basicUsage) {
|
|||
|
||||
TEST(GetDosArgv, advancedUsage) {
|
||||
size_t max = 4;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(2, GetDosArgv(u"(╯°□°)╯︵ ┻━┻", buf, size, argv, max));
|
||||
|
|
@ -71,7 +71,7 @@ TEST(GetDosArgv, advancedUsage) {
|
|||
|
||||
TEST(GetDosArgv, testAegeanGothicSupplementaryPlanes) {
|
||||
size_t max = 4; /* these symbols are almost as old as dos */
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(2, GetDosArgv(u"𐄷𐄸𐄹𐄺𐄻𐄼 𐌰𐌱𐌲𐌳𐌴𐌵𐌶𐌷", buf, size, argv, max));
|
||||
|
|
@ -84,7 +84,7 @@ TEST(GetDosArgv, testAegeanGothicSupplementaryPlanes) {
|
|||
|
||||
TEST(GetDosArgv, realWorldUsage) {
|
||||
size_t max = 512;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(5, GetDosArgv(u"C:\\Users\\jtunn\\printargs.com oh yes yes yes",
|
||||
|
|
@ -145,7 +145,7 @@ TEST(GetDosArgv, quoteInMiddleOfArg_wontSplitArg) {
|
|||
|
||||
TEST(GetDosArgv, waqQuoting1) {
|
||||
size_t max = 4;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(2,
|
||||
|
|
@ -159,7 +159,7 @@ TEST(GetDosArgv, waqQuoting1) {
|
|||
|
||||
TEST(GetDosArgv, waqQuoting2) {
|
||||
size_t max = 4;
|
||||
size_t size = ARG_MAX;
|
||||
size_t size = ARG_MAX / 2;
|
||||
char *buf = malloc(size * sizeof(char));
|
||||
char **argv = malloc(max * sizeof(char *));
|
||||
EXPECT_EQ(2, GetDosArgv(u"\"a\\\"b c\" d", buf, size, argv, max));
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue