1  /* Make sure that pthread_attr_getaffinity_np does not crash when the input
       2     cpuset size is smaller than that in the attribute structure.
       3  
       4     Copyright (C) 2013-2023 Free Software Foundation, Inc.
       5     This file is part of the GNU C Library.
       6  
       7     The GNU C Library is free software; you can redistribute it and/or
       8     modify it under the terms of the GNU Lesser General Public
       9     License as published by the Free Software Foundation; either
      10     version 2.1 of the License, or (at your option) any later version.
      11  
      12     The GNU C Library is distributed in the hope that it will be useful,
      13     but WITHOUT ANY WARRANTY; without even the implied warranty of
      14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      15     Lesser General Public License for more details.
      16  
      17     You should have received a copy of the GNU Lesser General Public
      18     License along with the GNU C Library; if not, see
      19     <https://www.gnu.org/licenses/>.  */
      20  
      21  #include <pthread.h>
      22  #include <stdio.h>
      23  #include <sched.h>
      24  #include <errno.h>
      25  #include <sys/param.h>
      26  
      27  
      28  #define RETURN_IF_FAIL(f, ...) \
      29    ({									      \
      30      int ret = f (__VA_ARGS__);						      \
      31      if (ret != 0)							      \
      32        {									      \
      33  	printf ("%s:%d: %s returned %d (errno = %d)\n", __FILE__, __LINE__,   \
      34  		#f, ret, errno);					      \
      35  	return ret;							      \
      36        }									      \
      37    })
      38  
      39  static int
      40  do_test (void)
      41  {
      42    for (int i = 0; i < 10; i++)
      43      {
      44        pthread_attr_t attr;
      45        cpu_set_t *cpuset = CPU_ALLOC (512);
      46        size_t cpusetsize = CPU_ALLOC_SIZE (512);
      47        CPU_ZERO_S (cpusetsize, cpuset);
      48  
      49        RETURN_IF_FAIL (pthread_attr_init, &attr);
      50        RETURN_IF_FAIL (pthread_attr_setaffinity_np, &attr, cpusetsize, cpuset);
      51        CPU_FREE (cpuset);
      52  
      53        cpuset = CPU_ALLOC (1);
      54        cpusetsize = CPU_ALLOC_SIZE (1);
      55        RETURN_IF_FAIL (pthread_attr_getaffinity_np, &attr, cpusetsize, cpuset);
      56        CPU_FREE (cpuset);
      57      }
      58    return 0;
      59  }
      60  
      61  
      62  #define TEST_FUNCTION do_test ()
      63  #include "../test-skeleton.c"