1 /* $NetBSD: pthread_attr.c,v 1.10 2008/06/28 10:29:37 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.10 2008/06/28 10:29:37 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 return ENOMEM; 264 *param = p->ptap_sp; 265 return 0; 266 } 267 268 269 int 270 pthread_attr_setschedpolicy(pthread_attr_t *attr, int policy) 271 { 272 struct pthread_attr_private *p; 273 274 275 switch (policy) { 276 case SCHED_OTHER: 277 case SCHED_FIFO: 278 case SCHED_RR: 279 p = pthread__attr_init_private(attr); 280 if (p == NULL) 281 return ENOMEM; 282 p->ptap_policy = policy; 283 return 0; 284 default: 285 return ENOTSUP; 286 } 287 } 288 289 290 int 291 pthread_attr_getschedpolicy(const pthread_attr_t *attr, int *policy) 292 { 293 struct pthread_attr_private *p; 294 295 p = attr->pta_private; 296 if (p == NULL) { 297 *policy = SCHED_OTHER; 298 return 0; 299 } 300 *policy = p->ptap_policy; 301 return 0; 302 } 303 304 305 int 306 pthread_attr_getstack(const pthread_attr_t *attr, void **addr, size_t *size) 307 { 308 struct pthread_attr_private *p; 309 310 if ((p = attr->pta_private) == NULL) { 311 *addr = NULL; 312 *size = pthread__stacksize; 313 } else { 314 *addr = p->ptap_stackaddr; 315 *size = p->ptap_stacksize; 316 } 317 318 return 0; 319 } 320 321 322 int 323 pthread_attr_setstack(pthread_attr_t *attr, void *addr, size_t size) 324 { 325 struct pthread_attr_private *p; 326 327 p = pthread__attr_init_private(attr); 328 if (p == NULL) 329 return ENOMEM; 330 331 p->ptap_stackaddr = addr; 332 p->ptap_stacksize = size; 333 334 return 0; 335 } 336 337 338 int 339 pthread_attr_getstacksize(const pthread_attr_t *attr, size_t *size) 340 { 341 struct pthread_attr_private *p; 342 343 if ((p = attr->pta_private) == NULL) 344 *size = pthread__stacksize; 345 else 346 *size = p->ptap_stacksize; 347 348 return 0; 349 } 350 351 352 int 353 pthread_attr_setstacksize(pthread_attr_t *attr, size_t size) 354 { 355 struct pthread_attr_private *p; 356 357 if (size < sysconf(_SC_THREAD_STACK_MIN)) 358 return EINVAL; 359 360 p = pthread__attr_init_private(attr); 361 if (p == NULL) 362 return ENOMEM; 363 364 p->ptap_stacksize = size; 365 366 return 0; 367 } 368 369 370 int 371 pthread_attr_getstackaddr(const pthread_attr_t *attr, void **addr) 372 { 373 struct pthread_attr_private *p; 374 375 if ((p = attr->pta_private) == NULL) 376 *addr = NULL; 377 else 378 *addr = p->ptap_stackaddr; 379 380 return 0; 381 } 382 383 384 int 385 pthread_attr_setstackaddr(pthread_attr_t *attr, void *addr) 386 { 387 struct pthread_attr_private *p; 388 389 p = pthread__attr_init_private(attr); 390 if (p == NULL) 391 return ENOMEM; 392 393 p->ptap_stackaddr = addr; 394 395 return 0; 396 } 397 398 399 int 400 pthread_attr_getname_np(const pthread_attr_t *attr, char *name, size_t len, 401 void **argp) 402 { 403 struct pthread_attr_private *p; 404 405 if ((p = attr->pta_private) == NULL) { 406 name[0] = '\0'; 407 if (argp != NULL) 408 *argp = NULL; 409 } else { 410 strlcpy(name, p->ptap_name, len); 411 if (argp != NULL) 412 *argp = p->ptap_namearg; 413 } 414 415 return 0; 416 } 417 418 419 int 420 pthread_attr_setname_np(pthread_attr_t *attr, const char *name, void *arg) 421 { 422 struct pthread_attr_private *p; 423 int namelen; 424 425 p = pthread__attr_init_private(attr); 426 if (p == NULL) 427 return ENOMEM; 428 429 namelen = snprintf(p->ptap_name, PTHREAD_MAX_NAMELEN_NP, name, arg); 430 if (namelen >= PTHREAD_MAX_NAMELEN_NP) { 431 p->ptap_name[0] = '\0'; 432 return EINVAL; 433 } 434 p->ptap_namearg = arg; 435 436 return 0; 437 } 438 439 int 440 pthread_attr_setcreatesuspend_np(pthread_attr_t *attr) 441 { 442 attr->pta_flags |= PT_FLAG_SUSPENDED; 443 return 0; 444 } 445