1 /* $NetBSD: pthread_attr.c,v 1.11 2008/10/10 09:13:20 ad Exp $ */ 2 3 /*- 4 * Copyright (c) 2001, 2002, 2003, 2008 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Nathan J. Williams. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 */ 31 32 #include <sys/cdefs.h> 33 __RCSID("$NetBSD: pthread_attr.c,v 1.11 2008/10/10 09:13:20 ad Exp $"); 34 35 #include <errno.h> 36 #include <stdio.h> 37 #include <stdlib.h> 38 #include <string.h> 39 #include <unistd.h> 40 41 #include "pthread.h" 42 #include "pthread_int.h" 43 44 static struct pthread_attr_private *pthread__attr_init_private( 45 pthread_attr_t *); 46 47 static struct pthread_attr_private * 48 pthread__attr_init_private(pthread_attr_t *attr) 49 { 50 struct pthread_attr_private *p; 51 52 if ((p = attr->pta_private) != NULL) 53 return p; 54 55 p = malloc(sizeof(*p)); 56 if (p != NULL) { 57 memset(p, 0, sizeof(*p)); 58 attr->pta_private = p; 59 p->ptap_policy = SCHED_OTHER; 60 } 61 return p; 62 } 63 64 65 int 66 pthread_attr_init(pthread_attr_t *attr) 67 { 68 69 attr->pta_magic = PT_ATTR_MAGIC; 70 attr->pta_flags = 0; 71 attr->pta_private = NULL; 72 73 return 0; 74 } 75 76 77 int 78 pthread_attr_destroy(pthread_attr_t *attr) 79 { 80 struct pthread_attr_private *p; 81 82 if ((p = attr->pta_private) != NULL) 83 free(p); 84 85 return 0; 86 } 87 88 89 int 90 pthread_attr_get_np(pthread_t thread, pthread_attr_t *attr) 91 { 92 struct pthread_attr_private *p; 93 94 p = pthread__attr_init_private(attr); 95 if (p == NULL) 96 return ENOMEM; 97 98 attr->pta_flags = thread->pt_flags & 99 (PT_FLAG_DETACHED | PT_FLAG_SCOPE_SYSTEM | PT_FLAG_EXPLICIT_SCHED); 100 101 p->ptap_namearg = thread->pt_name; 102 p->ptap_stackaddr = thread->pt_stack.ss_sp; 103 p->ptap_stacksize = thread->pt_stack.ss_size; 104 p->ptap_guardsize = (size_t)sysconf(_SC_PAGESIZE); 105 return pthread_getschedparam(thread, &p->ptap_policy, &p->ptap_sp); 106 } 107 108 109 int 110 pthread_attr_getdetachstate(const pthread_attr_t *attr, int *detachstate) 111 { 112 113 if (attr->pta_flags & PT_FLAG_DETACHED) 114 *detachstate = PTHREAD_CREATE_DETACHED; 115 else 116 *detachstate = PTHREAD_CREATE_JOINABLE; 117 118 return 0; 119 } 120 121 122 int 123 pthread_attr_setdetachstate(pthread_attr_t *attr, int detachstate) 124 { 125 126 switch (detachstate) { 127 case PTHREAD_CREATE_JOINABLE: 128 attr->pta_flags &= ~PT_FLAG_DETACHED; 129 break; 130 case PTHREAD_CREATE_DETACHED: 131 attr->pta_flags |= PT_FLAG_DETACHED; 132 break; 133 default: 134 return EINVAL; 135 } 136 137 return 0; 138 } 139 140 141 int 142 pthread_attr_getguardsize(const pthread_attr_t *attr, size_t *guard) 143 { 144 struct pthread_attr_private *p; 145 146 if ((p = attr->pta_private) == NULL) 147 *guard = (size_t)sysconf(_SC_PAGESIZE); 148 else 149 *guard = p->ptap_guardsize; 150 151 return 0; 152 } 153 154 155 int 156 pthread_attr_setguardsize(pthread_attr_t *attr, size_t guard) 157 { 158 struct pthread_attr_private *p; 159 160 p = pthread__attr_init_private(attr); 161 if (p == NULL) 162 return ENOMEM; 163 164 p->ptap_guardsize = guard; 165 166 return 0; 167 } 168 169 170 int 171 pthread_attr_getinheritsched(const pthread_attr_t *attr, int *inherit) 172 { 173 174 if (attr->pta_flags & PT_FLAG_EXPLICIT_SCHED) 175 *inherit = PTHREAD_EXPLICIT_SCHED; 176 else 177 *inherit = PTHREAD_INHERIT_SCHED; 178 179 return 0; 180 } 181 182 183 int 184 pthread_attr_setinheritsched(pthread_attr_t *attr, int inherit) 185 { 186 187 switch (inherit) { 188 case PTHREAD_INHERIT_SCHED: 189 attr->pta_flags &= ~PT_FLAG_EXPLICIT_SCHED; 190 break; 191 case PTHREAD_EXPLICIT_SCHED: 192 attr->pta_flags |= PT_FLAG_EXPLICIT_SCHED; 193 break; 194 default: 195 return EINVAL; 196 } 197 198 return 0; 199 } 200 201 202 int 203 pthread_attr_getscope(const pthread_attr_t *attr, int *scope) 204 { 205 206 if (attr->pta_flags & PT_FLAG_SCOPE_SYSTEM) 207 *scope = PTHREAD_SCOPE_SYSTEM; 208 else 209 *scope = PTHREAD_SCOPE_PROCESS; 210 211 return 0; 212 } 213 214 215 int 216 pthread_attr_setscope(pthread_attr_t *attr, int scope) 217 { 218 219 switch (scope) { 220 case PTHREAD_SCOPE_PROCESS: 221 attr->pta_flags &= ~PT_FLAG_SCOPE_SYSTEM; 222 break; 223 case PTHREAD_SCOPE_SYSTEM: 224 attr->pta_flags |= PT_FLAG_SCOPE_SYSTEM; 225 break; 226 default: 227 return EINVAL; 228 } 229 230 return 0; 231 } 232 233 234 int 235 pthread_attr_setschedparam(pthread_attr_t *attr, 236 const struct sched_param *param) 237 { 238 struct pthread_attr_private *p; 239 int error; 240 241 if (param == NULL) 242 return EINVAL; 243 p = pthread__attr_init_private(attr); 244 if (p == NULL) 245 return ENOMEM; 246 error = pthread__checkpri(param->sched_priority); 247 if (error == 0) 248 p->ptap_sp = *param; 249 return error; 250 } 251 252 253 int 254 pthread_attr_getschedparam(const pthread_attr_t *attr, 255 struct sched_param *param) 256 { 257 struct pthread_attr_private *p; 258 259 if (param == NULL) 260 return EINVAL; 261 p = attr->pta_private; 262 if (p == NULL) 263 memset(param, 0, sizeof(*param)); 264 else 265 *param = p->ptap_sp; 266 return 0; 267 } 268 269 270 int 271 pthread_attr_setschedpolicy(pthread_attr_t *attr, int policy) 272 { 273 struct pthread_attr_private *p; 274 275 276 switch (policy) { 277 case SCHED_OTHER: 278 case SCHED_FIFO: 279 case SCHED_RR: 280 p = pthread__attr_init_private(attr); 281 if (p == NULL) 282 return ENOMEM; 283 p->ptap_policy = policy; 284 return 0; 285 default: 286 return ENOTSUP; 287 } 288 } 289 290 291 int 292 pthread_attr_getschedpolicy(const pthread_attr_t *attr, int *policy) 293 { 294 struct pthread_attr_private *p; 295 296 p = attr->pta_private; 297 if (p == NULL) { 298 *policy = SCHED_OTHER; 299 return 0; 300 } 301 *policy = p->ptap_policy; 302 return 0; 303 } 304 305 306 int 307 pthread_attr_getstack(const pthread_attr_t *attr, void **addr, size_t *size) 308 { 309 struct pthread_attr_private *p; 310 311 if ((p = attr->pta_private) == NULL) { 312 *addr = NULL; 313 *size = pthread__stacksize; 314 } else { 315 *addr = p->ptap_stackaddr; 316 *size = p->ptap_stacksize; 317 } 318 319 return 0; 320 } 321 322 323 int 324 pthread_attr_setstack(pthread_attr_t *attr, void *addr, size_t size) 325 { 326 struct pthread_attr_private *p; 327 328 p = pthread__attr_init_private(attr); 329 if (p == NULL) 330 return ENOMEM; 331 332 p->ptap_stackaddr = addr; 333 p->ptap_stacksize = size; 334 335 return 0; 336 } 337 338 339 int 340 pthread_attr_getstacksize(const pthread_attr_t *attr, size_t *size) 341 { 342 struct pthread_attr_private *p; 343 344 if ((p = attr->pta_private) == NULL) 345 *size = pthread__stacksize; 346 else 347 *size = p->ptap_stacksize; 348 349 return 0; 350 } 351 352 353 int 354 pthread_attr_setstacksize(pthread_attr_t *attr, size_t size) 355 { 356 struct pthread_attr_private *p; 357 358 if (size < sysconf(_SC_THREAD_STACK_MIN)) 359 return EINVAL; 360 361 p = pthread__attr_init_private(attr); 362 if (p == NULL) 363 return ENOMEM; 364 365 p->ptap_stacksize = size; 366 367 return 0; 368 } 369 370 371 int 372 pthread_attr_getstackaddr(const pthread_attr_t *attr, void **addr) 373 { 374 struct pthread_attr_private *p; 375 376 if ((p = attr->pta_private) == NULL) 377 *addr = NULL; 378 else 379 *addr = p->ptap_stackaddr; 380 381 return 0; 382 } 383 384 385 int 386 pthread_attr_setstackaddr(pthread_attr_t *attr, void *addr) 387 { 388 struct pthread_attr_private *p; 389 390 p = pthread__attr_init_private(attr); 391 if (p == NULL) 392 return ENOMEM; 393 394 p->ptap_stackaddr = addr; 395 396 return 0; 397 } 398 399 400 int 401 pthread_attr_getname_np(const pthread_attr_t *attr, char *name, size_t len, 402 void **argp) 403 { 404 struct pthread_attr_private *p; 405 406 if ((p = attr->pta_private) == NULL) { 407 name[0] = '\0'; 408 if (argp != NULL) 409 *argp = NULL; 410 } else { 411 strlcpy(name, p->ptap_name, len); 412 if (argp != NULL) 413 *argp = p->ptap_namearg; 414 } 415 416 return 0; 417 } 418 419 420 int 421 pthread_attr_setname_np(pthread_attr_t *attr, const char *name, void *arg) 422 { 423 struct pthread_attr_private *p; 424 int namelen; 425 426 p = pthread__attr_init_private(attr); 427 if (p == NULL) 428 return ENOMEM; 429 430 namelen = snprintf(p->ptap_name, PTHREAD_MAX_NAMELEN_NP, name, arg); 431 if (namelen >= PTHREAD_MAX_NAMELEN_NP) { 432 p->ptap_name[0] = '\0'; 433 return EINVAL; 434 } 435 p->ptap_namearg = arg; 436 437 return 0; 438 } 439 440 int 441 pthread_attr_setcreatesuspend_np(pthread_attr_t *attr) 442 { 443 attr->pta_flags |= PT_FLAG_SUSPENDED; 444 return 0; 445 } 446