Увеличьте размер стека в Linux с помощью setrlimit

Question

Читая информацию о том, как увеличить размер стека для приложения C++, скомпилированного с помощью gnu, во время компиляции я понял, что это можно сделать с помощью setrlimit в начале программы. Тем не менее я не смог найти ни одного удачного примера того, как его использовать и в какой части программы применять его, чтобы получить размер стека 64M для программы на C++, кто-нибудь может мне помочь?

Thanlks

Answer 1

Обычно вы устанавливаете размер стека на ранней стадии, например, в начале main() перед вызовом любых других функций. Обычно логика будет такой:

• вызов getrlimit получить текущий размер стека
• если текущий размер <требуемый размер стека, то
  • вызов setrlimit увеличить размер стека до необходимого размера

В C это может быть закодировано примерно так:

#include <sys/resource.h>
#include <stdio.h>

int main (int argc, char **argv)
{
    const rlim_t kStackSize = 64L * 1024L * 1024L;   // min stack size = 64 Mb
    struct rlimit rl;
    int result;

    result = getrlimit(RLIMIT_STACK, &rl);
    if (result == 0)
    {
        if (rl.rlim_cur < kStackSize)
        {
            rl.rlim_cur = kStackSize;
            result = setrlimit(RLIMIT_STACK, &rl);
            if (result != 0)
            {
                fprintf(stderr, "setrlimit returned result = %d\n", result);
            }
        }
    }

    // ...

    return 0;
}

Answer 2

Посмотрите, не ограничивает ли это максимум времени выполнения:

[wally@zf conf]$  ulimit -all
core file size          (blocks, -c) 0
data seg size           (kbytes, -d) unlimited
scheduling priority             (-e) 0
file size               (blocks, -f) unlimited
pending signals                 (-i) 16114
max locked memory       (kbytes, -l) 32
max memory size         (kbytes, -m) unlimited
open files                      (-n) 1024
pipe size            (512 bytes, -p) 8
POSIX message queues     (bytes, -q) 819200
real-time priority              (-r) 0
stack size              (kbytes, -s) 10240
cpu time               (seconds, -t) unlimited
max user processes              (-u) 16114
virtual memory          (kbytes, -v) unlimited
file locks                      (-x) unlimited

Обратите внимание, что размер стека по умолчанию ограничен 10 МБ. Итак, чтобы увеличить его до 64 МиБ:

[wally@zf conf]$ ulimit -s 64M
-bash: ulimit: 64M: invalid number
[wally@zf conf]$ ulimit -s 65536
[wally@zf conf]$ ulimit -all
core file size          (blocks, -c) 0
data seg size           (kbytes, -d) unlimited
scheduling priority             (-e) 0
file size               (blocks, -f) unlimited
pending signals                 (-i) 16114
max locked memory       (kbytes, -l) 32
max memory size         (kbytes, -m) unlimited
open files                      (-n) 1024
pipe size            (512 bytes, -p) 8
POSIX message queues     (bytes, -q) 819200
real-time priority              (-r) 0
stack size              (kbytes, -s) 65536
cpu time               (seconds, -t) unlimited
max user processes              (-u) 16114
virtual memory          (kbytes, -v) unlimited
file locks                      (-x) unlimited

Answer 3

Чтобы выйти за пределы жесткого ограничения в setrlimit (в OSX это только 64 МБ по умолчанию), создайте новый поток, используя pthreads с выбранным размером стека. Вот фрагмент кода C:

    // Call function f with a 256MB stack.
    static int bigstack(void *(*f)(void *), void* userdata) {

      pthread_t thread;
      pthread_attr_t attr;

      // allocate a 256MB region for the stack.
      size_t stacksize = 256*1024*1024;
      pthread_attr_init(&attr);
      pthread_attr_setstacksize(&attr, stacksize);

      int rc = pthread_create(&thread, &attr, f, userdata);
      if (rc){
        printf("ERROR: return code from pthread_create() is %d\n", rc);
        return 0;
      }
      pthread_join(thread, NULL);
      return 1;

    }