From bf959371c4505bfe27b0682611c035d64b90efd3 Mon Sep 17 00:00:00 2001 From: lihan Date: Tue, 27 Feb 2018 12:56:36 +0800 Subject: [PATCH] fixup set rlimit stack failed condition refs #6119 refs #6215 --- lib/base/application.cpp | 47 ++++++++++++++++++++-------------------- 1 file changed, 24 insertions(+), 23 deletions(-) diff --git a/lib/base/application.cpp b/lib/base/application.cpp index 70b2c40e9..b6e09bfce 100644 --- a/lib/base/application.cpp +++ b/lib/base/application.cpp @@ -256,32 +256,33 @@ void Application::SetResourceLimits(void) else rl.rlim_cur = rl.rlim_max; - if (setrlimit(RLIMIT_STACK, &rl) < 0) + if (setrlimit(RLIMIT_STACK, &rl) < 0) { Log(LogNotice, "Application", "Could not adjust resource limit for stack size (RLIMIT_STACK)"); - else if (set_stack_rlimit) { - char **new_argv = static_cast(malloc(sizeof(char *) * (argc + 2))); + if (set_stack_rlimit) { + char **new_argv = static_cast(malloc(sizeof(char *) * (argc + 2))); - if (!new_argv) { - perror("malloc"); - Exit(EXIT_FAILURE); + if (!new_argv) { + perror("malloc"); + Exit(EXIT_FAILURE); + } + + new_argv[0] = argv[0]; + new_argv[1] = strdup("--no-stack-rlimit"); + + if (!new_argv[1]) { + perror("strdup"); + exit(1); + } + + for (int i = 1; i < argc; i++) + new_argv[i + 1] = argv[i]; + + new_argv[argc + 1] = NULL; + + (void) execvp(new_argv[0], new_argv); + perror("execvp"); + _exit(EXIT_FAILURE); } - - new_argv[0] = argv[0]; - new_argv[1] = strdup("--no-stack-rlimit"); - - if (!new_argv[1]) { - perror("strdup"); - exit(1); - } - - for (int i = 1; i < argc; i++) - new_argv[i + 1] = argv[i]; - - new_argv[argc + 1] = NULL; - - (void) execvp(new_argv[0], new_argv); - perror("execvp"); - _exit(EXIT_FAILURE); } # else /* RLIMIT_STACK */ Log(LogNotice, "Application", "System does not support adjusting the resource limit for stack size (RLIMIT_STACK)");