nixos/nix
1
1
Fork 0
mirror of https://github.com/NixOS/nix.git synced 2025-11-09 20:16:03 +01:00
Code Activity

Restore stack size in child processes

Browse source 
Fixes #4673.
This commit is contained in:
Eelco Dolstra 2021-04-07 13:40:13 +02:00
parent 9b9e703df4
commit 906adadacd
3 changed files with 32 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

27
src/libutil/util.cc
View file

@ -32,6 +32,7 @@
#ifdef __linux__
#include <sys/prctl.h>
#include <sys/resource.h>
#endif
@ -1618,11 +1619,37 @@ static void restoreSignals()
throw SysError("restoring signals");
}
#if __linux__
rlim_t savedStackSize = 0;
#endif
void setStackSize(size_t stackSize)
{
#if __linux__
struct rlimit limit;
if (getrlimit(RLIMIT_STACK, &limit) == 0 && limit.rlim_cur < stackSize) {
savedStackSize = limit.rlim_cur;
limit.rlim_cur = stackSize;
setrlimit(RLIMIT_STACK, &limit);
}
#endif
}
void restoreProcessContext()
{
restoreSignals();
restoreAffinity();
#if __linux__
if (savedStackSize) {
struct rlimit limit;
if (getrlimit(RLIMIT_STACK, &limit) == 0) {
limit.rlim_cur = savedStackSize;
setrlimit(RLIMIT_STACK, &limit);
}
}
#endif
}
/* RAII helper to automatically deregister a callback. */