1 /* Copyright (C) 2003-2022 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3 
4    The GNU C Library is free software; you can redistribute it and/or
5    modify it under the terms of the GNU Lesser General Public
6    License as published by the Free Software Foundation; either
7    version 2.1 of the License, or (at your option) any later version.
8 
9    The GNU C Library is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    Lesser General Public License for more details.
13 
14    You should have received a copy of the GNU Lesser General Public
15    License along with the GNU C Library; if not, see
16    <https://www.gnu.org/licenses/>.  */
17 
18 #include <errno.h>
19 #include <limits.h>
20 #include <stdlib.h>
21 #include <string.h>
22 #include <pthreadP.h>
23 #include <shlib-compat.h>
24 
25 
26 int
__pthread_attr_setaffinity_np(pthread_attr_t * attr,size_t cpusetsize,const cpu_set_t * cpuset)27 __pthread_attr_setaffinity_np (pthread_attr_t *attr, size_t cpusetsize,
28 			       const cpu_set_t *cpuset)
29 {
30   struct pthread_attr *iattr;
31 
32   iattr = (struct pthread_attr *) attr;
33 
34   if (cpuset == NULL || cpusetsize == 0)
35     {
36       if (iattr->extension != NULL)
37 	{
38 	  free (iattr->extension->cpuset);
39 	  iattr->extension->cpuset = NULL;
40 	  iattr->extension->cpusetsize = 0;
41 	}
42     }
43   else
44     {
45       int ret = __pthread_attr_extension (iattr);
46       if (ret != 0)
47 	return ret;
48 
49       if (iattr->extension->cpusetsize != cpusetsize)
50 	{
51 	  void *newp = realloc (iattr->extension->cpuset, cpusetsize);
52 	  if (newp == NULL)
53 	    return ENOMEM;
54 
55 	  iattr->extension->cpuset = newp;
56 	  iattr->extension->cpusetsize = cpusetsize;
57 	}
58 
59       memcpy (iattr->extension->cpuset, cpuset, cpusetsize);
60     }
61 
62   return 0;
63 }
64 libc_hidden_def (__pthread_attr_setaffinity_np)
65 versioned_symbol (libc, __pthread_attr_setaffinity_np,
66 		  pthread_attr_setaffinity_np, GLIBC_2_32);
67 
68 
69 #if SHLIB_COMPAT (libc, GLIBC_2_3_4, GLIBC_2_32)
70 /* Compat symbol with the old libc version.  */
71 strong_alias (__pthread_attr_setaffinity_np, __pthread_attr_setaffinity_alias)
72 compat_symbol (libc, __pthread_attr_setaffinity_alias,
73 	       pthread_attr_setaffinity_np, GLIBC_2_3_4);
74 #endif
75 
76 #if SHLIB_COMPAT (libc, GLIBC_2_3_3, GLIBC_2_3_4)
77 int
__pthread_attr_setaffinity_old(pthread_attr_t * attr,cpu_set_t * cpuset)78 __pthread_attr_setaffinity_old (pthread_attr_t *attr, cpu_set_t *cpuset)
79 {
80   /* The old interface by default assumed a 1024 processor bitmap.  */
81   return __pthread_attr_setaffinity_np (attr, 128, cpuset);
82 }
83 compat_symbol (libc, __pthread_attr_setaffinity_old,
84 	       pthread_attr_setaffinity_np, GLIBC_2_3_3);
85 #endif
86