Skip to content

Commit

Permalink
Merge pull request #12249 from nix-windows/stack-size
Browse files Browse the repository at this point in the history
Remove broken stack size logic from Windows
  • Loading branch information
Ericson2314 authored Feb 1, 2025
2 parents 39ef2e2 + 7c8c71f commit d949c8d
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 29 deletions.
31 changes: 2 additions & 29 deletions src/libutil/current-process.cc
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,6 @@
# include "namespaces.hh"
#endif

#ifndef _WIN32
# include <sys/resource.h>
#endif

namespace nix {

unsigned int getMaxCPU()
Expand Down Expand Up @@ -55,11 +51,11 @@ unsigned int getMaxCPU()
//////////////////////////////////////////////////////////////////////


#ifndef _WIN32
size_t savedStackSize = 0;

void setStackSize(size_t stackSize)
{
#ifndef _WIN32
struct rlimit limit;
if (getrlimit(RLIMIT_STACK, &limit) == 0 && limit.rlim_cur < stackSize) {
savedStackSize = limit.rlim_cur;
Expand All @@ -77,31 +73,8 @@ void setStackSize(size_t stackSize)
);
}
}
#else
ULONG_PTR stackLow, stackHigh;
GetCurrentThreadStackLimits(&stackLow, &stackHigh);
ULONG maxStackSize = stackHigh - stackLow;
ULONG currStackSize = 0;
// This retrieves the current promised stack size
SetThreadStackGuarantee(&currStackSize);
if (currStackSize < stackSize) {
savedStackSize = currStackSize;
ULONG newStackSize = std::min(static_cast<ULONG>(stackSize), maxStackSize);
if (SetThreadStackGuarantee(&newStackSize) == 0) {
logger->log(
lvlError,
HintFmt(
"Failed to increase stack size from %1% to %2% (maximum allowed stack size: %3%): %4%",
savedStackSize,
stackSize,
maxStackSize,
std::to_string(GetLastError())
).str()
);
}
}
#endif
}
#endif

void restoreProcessContext(bool restoreMounts)
{
Expand Down
3 changes: 3 additions & 0 deletions src/libutil/current-process.hh
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,13 @@ namespace nix {
*/
unsigned int getMaxCPU();

// It does not seem possible to dynamically change stack size on Windows.
#ifndef _WIN32
/**
* Change the stack size.
*/
void setStackSize(size_t stackSize);
#endif

/**
* Restore the original inherited Unix process context (such as signal
Expand Down
2 changes: 2 additions & 0 deletions src/nix/main.cc
Original file line number Diff line number Diff line change
Expand Up @@ -557,9 +557,11 @@ void mainWrapped(int argc, char * * argv)

int main(int argc, char * * argv)
{
#ifndef _WIN32
// Increase the default stack size for the evaluator and for
// libstdc++'s std::regex.
nix::setStackSize(64 * 1024 * 1024);
#endif

return nix::handleExceptions(argv[0], [&]() {
nix::mainWrapped(argc, argv);
Expand Down

0 comments on commit d949c8d

Please sign in to comment.