1  /* Tests for sched_getaffinity with large buffers.
       2     Copyright (C) 2021-2023 Free Software Foundation, Inc.
       3     This file is part of the GNU C Library.
       4  
       5     The GNU C Library is free software; you can redistribute it and/or
       6     modify it under the terms of the GNU Lesser General Public
       7     License as published by the Free Software Foundation; either
       8     version 2.1 of the License, or (at your option) any later version.
       9  
      10     The GNU C Library is distributed in the hope that it will be useful,
      11     but WITHOUT ANY WARRANTY; without even the implied warranty of
      12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      13     Lesser General Public License for more details.
      14  
      15     You should have received a copy of the GNU Lesser General Public
      16     License along with the GNU C Library; if not, see
      17     <https://www.gnu.org/licenses/>.  */
      18  
      19  #include <array_length.h>
      20  #include <sched.h>
      21  #include <support/check.h>
      22  
      23  /* NB: this test may fail on system with more than 32k cpus.  */
      24  
      25  static int
      26  do_test (void)
      27  {
      28    /* The values are larger than the default cpu_set_t.  */
      29    const int bufsize[] = { 1<<11, 1<<12, 1<<13, 1<<14, 1<<15, 1<<16, 1<<17 };
      30    int cpucount[array_length (bufsize)];
      31  
      32    for (int i = 0; i < array_length (bufsize); i++)
      33      {
      34        cpu_set_t *cpuset = CPU_ALLOC (bufsize[i]);
      35        TEST_VERIFY (cpuset != NULL);
      36        size_t size = CPU_ALLOC_SIZE (bufsize[i]);
      37        TEST_COMPARE (sched_getaffinity (0, size, cpuset), 0);
      38        cpucount[i] = CPU_COUNT_S (size, cpuset);
      39        CPU_FREE (cpuset);
      40      }
      41  
      42    for (int i = 0; i < array_length (cpucount) - 1; i++)
      43      TEST_COMPARE (cpucount[i], cpucount[i + 1]);
      44  
      45    return 0;
      46  }
      47  
      48  #include <support/test-driver.c>