sys_pset.c revision 1.2.4.2
1 1.2.4.2 yamt /* $NetBSD: sys_pset.c,v 1.2.4.2 2008/01/21 09:46:25 yamt Exp $ */ 2 1.2.4.2 yamt 3 1.2.4.2 yamt /* 4 1.2.4.2 yamt * Copyright (c) 2008, Mindaugas Rasiukevicius <rmind at NetBSD org> 5 1.2.4.2 yamt * All rights reserved. 6 1.2.4.2 yamt * 7 1.2.4.2 yamt * Redistribution and use in source and binary forms, with or without 8 1.2.4.2 yamt * modification, are permitted provided that the following conditions 9 1.2.4.2 yamt * are met: 10 1.2.4.2 yamt * 1. Redistributions of source code must retain the above copyright 11 1.2.4.2 yamt * notice, this list of conditions and the following disclaimer. 12 1.2.4.2 yamt * 2. Redistributions in binary form must reproduce the above copyright 13 1.2.4.2 yamt * notice, this list of conditions and the following disclaimer in the 14 1.2.4.2 yamt * documentation and/or other materials provided with the distribution. 15 1.2.4.2 yamt * 16 1.2.4.2 yamt * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17 1.2.4.2 yamt * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 18 1.2.4.2 yamt * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 19 1.2.4.2 yamt * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 20 1.2.4.2 yamt * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 21 1.2.4.2 yamt * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 22 1.2.4.2 yamt * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 23 1.2.4.2 yamt * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 24 1.2.4.2 yamt * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 25 1.2.4.2 yamt * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 26 1.2.4.2 yamt * POSSIBILITY OF SUCH DAMAGE. 27 1.2.4.2 yamt */ 28 1.2.4.2 yamt 29 1.2.4.2 yamt /* 30 1.2.4.2 yamt * Implementation of the Processor Sets. 31 1.2.4.2 yamt * 32 1.2.4.2 yamt * Locking 33 1.2.4.2 yamt * The array of the processor-set structures and its members are protected 34 1.2.4.2 yamt * by the global psets_lock. Note that in scheduler, the very l_psid value 35 1.2.4.2 yamt * might be used without lock held. 36 1.2.4.2 yamt */ 37 1.2.4.2 yamt 38 1.2.4.2 yamt #include <sys/cdefs.h> 39 1.2.4.2 yamt __KERNEL_RCSID(0, "$NetBSD: sys_pset.c,v 1.2.4.2 2008/01/21 09:46:25 yamt Exp $"); 40 1.2.4.2 yamt 41 1.2.4.2 yamt #include <sys/param.h> 42 1.2.4.2 yamt 43 1.2.4.2 yamt #include <sys/cpu.h> 44 1.2.4.2 yamt #include <sys/kauth.h> 45 1.2.4.2 yamt #include <sys/kmem.h> 46 1.2.4.2 yamt #include <sys/lwp.h> 47 1.2.4.2 yamt #include <sys/mutex.h> 48 1.2.4.2 yamt #include <sys/proc.h> 49 1.2.4.2 yamt #include <sys/pset.h> 50 1.2.4.2 yamt #include <sys/sched.h> 51 1.2.4.2 yamt #include <sys/syscallargs.h> 52 1.2.4.2 yamt #include <sys/sysctl.h> 53 1.2.4.2 yamt #include <sys/systm.h> 54 1.2.4.2 yamt #include <sys/types.h> 55 1.2.4.2 yamt 56 1.2.4.2 yamt static pset_info_t ** psets; 57 1.2.4.2 yamt static kmutex_t psets_lock; 58 1.2.4.2 yamt static u_int psets_max; 59 1.2.4.2 yamt static u_int psets_count; 60 1.2.4.2 yamt 61 1.2.4.2 yamt static int psets_realloc(int); 62 1.2.4.2 yamt static int psid_validate(psetid_t, bool); 63 1.2.4.2 yamt static int kern_pset_create(psetid_t *); 64 1.2.4.2 yamt static int kern_pset_destroy(psetid_t); 65 1.2.4.2 yamt 66 1.2.4.2 yamt /* 67 1.2.4.2 yamt * Initialization of the processor-sets. 68 1.2.4.2 yamt */ 69 1.2.4.2 yamt void 70 1.2.4.2 yamt psets_init(void) 71 1.2.4.2 yamt { 72 1.2.4.2 yamt 73 1.2.4.2 yamt psets_max = max(MAXCPUS, 32); 74 1.2.4.2 yamt psets = kmem_zalloc(psets_max * sizeof(void *), KM_SLEEP); 75 1.2.4.2 yamt mutex_init(&psets_lock, MUTEX_DEFAULT, IPL_NONE); 76 1.2.4.2 yamt psets_count = 0; 77 1.2.4.2 yamt } 78 1.2.4.2 yamt 79 1.2.4.2 yamt /* 80 1.2.4.2 yamt * Reallocate the array of the processor-set structures. 81 1.2.4.2 yamt */ 82 1.2.4.2 yamt static int 83 1.2.4.2 yamt psets_realloc(int new_psets_max) 84 1.2.4.2 yamt { 85 1.2.4.2 yamt pset_info_t **new_psets, **old_psets; 86 1.2.4.2 yamt const u_int newsize = new_psets_max * sizeof(void *); 87 1.2.4.2 yamt u_int i, oldsize; 88 1.2.4.2 yamt 89 1.2.4.2 yamt if (new_psets_max < 1) 90 1.2.4.2 yamt return EINVAL; 91 1.2.4.2 yamt 92 1.2.4.2 yamt new_psets = kmem_zalloc(newsize, KM_SLEEP); 93 1.2.4.2 yamt mutex_enter(&psets_lock); 94 1.2.4.2 yamt old_psets = psets; 95 1.2.4.2 yamt oldsize = psets_max * sizeof(void *); 96 1.2.4.2 yamt 97 1.2.4.2 yamt /* Check if we can lower the size of the array */ 98 1.2.4.2 yamt if (new_psets_max < psets_max) { 99 1.2.4.2 yamt for (i = new_psets_max; i < psets_max; i++) { 100 1.2.4.2 yamt if (psets[i] == NULL) 101 1.2.4.2 yamt continue; 102 1.2.4.2 yamt mutex_exit(&psets_lock); 103 1.2.4.2 yamt kmem_free(new_psets, newsize); 104 1.2.4.2 yamt return EBUSY; 105 1.2.4.2 yamt } 106 1.2.4.2 yamt } 107 1.2.4.2 yamt 108 1.2.4.2 yamt /* Copy all pointers to the new array */ 109 1.2.4.2 yamt memcpy(new_psets, psets, newsize); 110 1.2.4.2 yamt psets_max = new_psets_max; 111 1.2.4.2 yamt psets = new_psets; 112 1.2.4.2 yamt mutex_exit(&psets_lock); 113 1.2.4.2 yamt 114 1.2.4.2 yamt kmem_free(old_psets, oldsize); 115 1.2.4.2 yamt return 0; 116 1.2.4.2 yamt } 117 1.2.4.2 yamt 118 1.2.4.2 yamt /* 119 1.2.4.2 yamt * Validate processor-set ID. 120 1.2.4.2 yamt */ 121 1.2.4.2 yamt static int 122 1.2.4.2 yamt psid_validate(psetid_t psid, bool chkps) 123 1.2.4.2 yamt { 124 1.2.4.2 yamt 125 1.2.4.2 yamt KASSERT(mutex_owned(&psets_lock)); 126 1.2.4.2 yamt 127 1.2.4.2 yamt if (chkps && (psid == PS_NONE || psid == PS_QUERY || psid == PS_MYID)) 128 1.2.4.2 yamt return 0; 129 1.2.4.2 yamt if (psid <= 0 || psid > psets_max) 130 1.2.4.2 yamt return EINVAL; 131 1.2.4.2 yamt if (psets[psid - 1] == NULL) 132 1.2.4.2 yamt return EINVAL; 133 1.2.4.2 yamt if (psets[psid - 1]->ps_flags & PSET_BUSY) 134 1.2.4.2 yamt return EBUSY; 135 1.2.4.2 yamt 136 1.2.4.2 yamt return 0; 137 1.2.4.2 yamt } 138 1.2.4.2 yamt 139 1.2.4.2 yamt /* 140 1.2.4.2 yamt * Create a processor-set. 141 1.2.4.2 yamt */ 142 1.2.4.2 yamt static int 143 1.2.4.2 yamt kern_pset_create(psetid_t *psid) 144 1.2.4.2 yamt { 145 1.2.4.2 yamt pset_info_t *pi; 146 1.2.4.2 yamt u_int i; 147 1.2.4.2 yamt 148 1.2.4.2 yamt if (psets_count == psets_max) 149 1.2.4.2 yamt return ENOMEM; 150 1.2.4.2 yamt 151 1.2.4.2 yamt pi = kmem_zalloc(sizeof(pset_info_t), KM_SLEEP); 152 1.2.4.2 yamt 153 1.2.4.2 yamt mutex_enter(&psets_lock); 154 1.2.4.2 yamt if (psets_count == psets_max) { 155 1.2.4.2 yamt mutex_exit(&psets_lock); 156 1.2.4.2 yamt kmem_free(pi, sizeof(pset_info_t)); 157 1.2.4.2 yamt return ENOMEM; 158 1.2.4.2 yamt } 159 1.2.4.2 yamt 160 1.2.4.2 yamt /* Find a free entry in the array */ 161 1.2.4.2 yamt for (i = 0; i < psets_max; i++) 162 1.2.4.2 yamt if (psets[i] == NULL) 163 1.2.4.2 yamt break; 164 1.2.4.2 yamt KASSERT(i != psets_max); 165 1.2.4.2 yamt 166 1.2.4.2 yamt psets[i] = pi; 167 1.2.4.2 yamt psets_count++; 168 1.2.4.2 yamt mutex_exit(&psets_lock); 169 1.2.4.2 yamt 170 1.2.4.2 yamt *psid = i + 1; 171 1.2.4.2 yamt return 0; 172 1.2.4.2 yamt } 173 1.2.4.2 yamt 174 1.2.4.2 yamt /* 175 1.2.4.2 yamt * Destroy a processor-set. 176 1.2.4.2 yamt */ 177 1.2.4.2 yamt static int 178 1.2.4.2 yamt kern_pset_destroy(psetid_t psid) 179 1.2.4.2 yamt { 180 1.2.4.2 yamt struct cpu_info *ci; 181 1.2.4.2 yamt pset_info_t *pi; 182 1.2.4.2 yamt struct lwp *l; 183 1.2.4.2 yamt CPU_INFO_ITERATOR cii; 184 1.2.4.2 yamt int error; 185 1.2.4.2 yamt 186 1.2.4.2 yamt mutex_enter(&psets_lock); 187 1.2.4.2 yamt if (psid == PS_MYID) { 188 1.2.4.2 yamt /* Use caller's processor-set ID */ 189 1.2.4.2 yamt psid = curlwp->l_psid; 190 1.2.4.2 yamt } 191 1.2.4.2 yamt error = psid_validate(psid, false); 192 1.2.4.2 yamt if (error) { 193 1.2.4.2 yamt mutex_exit(&psets_lock); 194 1.2.4.2 yamt return error; 195 1.2.4.2 yamt } 196 1.2.4.2 yamt 197 1.2.4.2 yamt /* Release the processor-set from all CPUs */ 198 1.2.4.2 yamt for (CPU_INFO_FOREACH(cii, ci)) { 199 1.2.4.2 yamt struct schedstate_percpu *spc; 200 1.2.4.2 yamt 201 1.2.4.2 yamt spc = &ci->ci_schedstate; 202 1.2.4.2 yamt if (spc->spc_psid != psid) 203 1.2.4.2 yamt continue; 204 1.2.4.2 yamt spc->spc_psid = PS_NONE; 205 1.2.4.2 yamt } 206 1.2.4.2 yamt /* Mark that processor-set is going to be destroyed */ 207 1.2.4.2 yamt pi = psets[psid - 1]; 208 1.2.4.2 yamt pi->ps_flags |= PSET_BUSY; 209 1.2.4.2 yamt mutex_exit(&psets_lock); 210 1.2.4.2 yamt 211 1.2.4.2 yamt /* Unmark the processor-set ID from each thread */ 212 1.2.4.2 yamt mutex_enter(&proclist_lock); 213 1.2.4.2 yamt LIST_FOREACH(l, &alllwp, l_list) { 214 1.2.4.2 yamt /* Safe to check and set without lock held */ 215 1.2.4.2 yamt if (l->l_psid != psid) 216 1.2.4.2 yamt continue; 217 1.2.4.2 yamt l->l_psid = PS_NONE; 218 1.2.4.2 yamt } 219 1.2.4.2 yamt mutex_exit(&proclist_lock); 220 1.2.4.2 yamt 221 1.2.4.2 yamt /* Destroy the processor-set */ 222 1.2.4.2 yamt mutex_enter(&psets_lock); 223 1.2.4.2 yamt psets[psid - 1] = NULL; 224 1.2.4.2 yamt psets_count--; 225 1.2.4.2 yamt mutex_exit(&psets_lock); 226 1.2.4.2 yamt 227 1.2.4.2 yamt kmem_free(pi, sizeof(pset_info_t)); 228 1.2.4.2 yamt return 0; 229 1.2.4.2 yamt } 230 1.2.4.2 yamt 231 1.2.4.2 yamt /* 232 1.2.4.2 yamt * General system calls for the processor-sets. 233 1.2.4.2 yamt */ 234 1.2.4.2 yamt 235 1.2.4.2 yamt int 236 1.2.4.2 yamt sys_pset_create(struct lwp *l, const struct sys_pset_create_args *uap, 237 1.2.4.2 yamt register_t *retval) 238 1.2.4.2 yamt { 239 1.2.4.2 yamt /* { 240 1.2.4.2 yamt syscallarg(psetid_t) *psid; 241 1.2.4.2 yamt } */ 242 1.2.4.2 yamt psetid_t psid; 243 1.2.4.2 yamt int error; 244 1.2.4.2 yamt 245 1.2.4.2 yamt /* Available only for super-user */ 246 1.2.4.2 yamt if (kauth_authorize_generic(l->l_cred, KAUTH_GENERIC_ISSUSER, NULL)) 247 1.2.4.2 yamt return EPERM; 248 1.2.4.2 yamt 249 1.2.4.2 yamt error = kern_pset_create(&psid); 250 1.2.4.2 yamt if (error) 251 1.2.4.2 yamt return error; 252 1.2.4.2 yamt 253 1.2.4.2 yamt error = copyout(&psid, SCARG(uap, psid), sizeof(psetid_t)); 254 1.2.4.2 yamt if (error) 255 1.2.4.2 yamt (void)kern_pset_destroy(psid); 256 1.2.4.2 yamt 257 1.2.4.2 yamt return error; 258 1.2.4.2 yamt } 259 1.2.4.2 yamt 260 1.2.4.2 yamt int 261 1.2.4.2 yamt sys_pset_destroy(struct lwp *l, const struct sys_pset_destroy_args *uap, 262 1.2.4.2 yamt register_t *retval) 263 1.2.4.2 yamt { 264 1.2.4.2 yamt /* { 265 1.2.4.2 yamt syscallarg(psetid_t) psid; 266 1.2.4.2 yamt } */ 267 1.2.4.2 yamt 268 1.2.4.2 yamt /* Available only for super-user */ 269 1.2.4.2 yamt if (kauth_authorize_generic(l->l_cred, KAUTH_GENERIC_ISSUSER, NULL)) 270 1.2.4.2 yamt return EPERM; 271 1.2.4.2 yamt 272 1.2.4.2 yamt return kern_pset_destroy(SCARG(uap, psid)); 273 1.2.4.2 yamt } 274 1.2.4.2 yamt 275 1.2.4.2 yamt int 276 1.2.4.2 yamt sys_pset_assign(struct lwp *l, const struct sys_pset_assign_args *uap, 277 1.2.4.2 yamt register_t *retval) 278 1.2.4.2 yamt { 279 1.2.4.2 yamt /* { 280 1.2.4.2 yamt syscallarg(psetid_t) psid; 281 1.2.4.2 yamt syscallarg(cpuid_t) cpuid; 282 1.2.4.2 yamt syscallarg(psetid_t) *opsid; 283 1.2.4.2 yamt } */ 284 1.2.4.2 yamt struct cpu_info *ci; 285 1.2.4.2 yamt struct schedstate_percpu *spc; 286 1.2.4.2 yamt psetid_t psid = SCARG(uap, psid), opsid = 0; 287 1.2.4.2 yamt CPU_INFO_ITERATOR cii; 288 1.2.4.2 yamt int error = 0; 289 1.2.4.2 yamt 290 1.2.4.2 yamt /* Available only for super-user, except the case of PS_QUERY */ 291 1.2.4.2 yamt if (kauth_authorize_generic(l->l_cred, KAUTH_GENERIC_ISSUSER, NULL) && 292 1.2.4.2 yamt psid != PS_QUERY) 293 1.2.4.2 yamt return EPERM; 294 1.2.4.2 yamt 295 1.2.4.2 yamt /* Find the target CPU */ 296 1.2.4.2 yamt for (CPU_INFO_FOREACH(cii, ci)) 297 1.2.4.2 yamt if (cpu_index(ci) == SCARG(uap, cpuid)) 298 1.2.4.2 yamt break; 299 1.2.4.2 yamt if (ci == NULL) 300 1.2.4.2 yamt return EINVAL; 301 1.2.4.2 yamt spc = &ci->ci_schedstate; 302 1.2.4.2 yamt 303 1.2.4.2 yamt mutex_enter(&psets_lock); 304 1.2.4.2 yamt error = psid_validate(psid, true); 305 1.2.4.2 yamt if (error) { 306 1.2.4.2 yamt mutex_exit(&psets_lock); 307 1.2.4.2 yamt return error; 308 1.2.4.2 yamt } 309 1.2.4.2 yamt opsid = spc->spc_psid; 310 1.2.4.2 yamt switch (psid) { 311 1.2.4.2 yamt case PS_QUERY: 312 1.2.4.2 yamt break; 313 1.2.4.2 yamt case PS_MYID: 314 1.2.4.2 yamt psid = curlwp->l_psid; 315 1.2.4.2 yamt default: 316 1.2.4.2 yamt spc->spc_psid = psid; 317 1.2.4.2 yamt } 318 1.2.4.2 yamt mutex_exit(&psets_lock); 319 1.2.4.2 yamt 320 1.2.4.2 yamt if (SCARG(uap, opsid) != NULL) 321 1.2.4.2 yamt error = copyout(&opsid, SCARG(uap, opsid), sizeof(psetid_t)); 322 1.2.4.2 yamt 323 1.2.4.2 yamt return error; 324 1.2.4.2 yamt } 325 1.2.4.2 yamt 326 1.2.4.2 yamt int 327 1.2.4.2 yamt sys__pset_bind(struct lwp *l, const struct sys__pset_bind_args *uap, 328 1.2.4.2 yamt register_t *retval) 329 1.2.4.2 yamt { 330 1.2.4.2 yamt /* { 331 1.2.4.2 yamt syscallarg(idtype_t) idtype; 332 1.2.4.2 yamt syscallarg(id_t) first_id; 333 1.2.4.2 yamt syscallarg(id_t) second_id; 334 1.2.4.2 yamt syscallarg(psetid_t) psid; 335 1.2.4.2 yamt syscallarg(psetid_t) *opsid; 336 1.2.4.2 yamt } */ 337 1.2.4.2 yamt struct cpu_info *ci; 338 1.2.4.2 yamt struct proc *p; 339 1.2.4.2 yamt struct lwp *t; 340 1.2.4.2 yamt id_t id1, id2; 341 1.2.4.2 yamt pid_t pid = 0; 342 1.2.4.2 yamt lwpid_t lid = 0; 343 1.2.4.2 yamt psetid_t psid, opsid; 344 1.2.4.2 yamt int error = 0, lcnt; 345 1.2.4.2 yamt 346 1.2.4.2 yamt psid = SCARG(uap, psid); 347 1.2.4.2 yamt 348 1.2.4.2 yamt /* Available only for super-user, except the case of PS_QUERY */ 349 1.2.4.2 yamt if (kauth_authorize_generic(l->l_cred, KAUTH_GENERIC_ISSUSER, NULL) && 350 1.2.4.2 yamt psid != PS_QUERY) 351 1.2.4.2 yamt return EPERM; 352 1.2.4.2 yamt 353 1.2.4.2 yamt mutex_enter(&psets_lock); 354 1.2.4.2 yamt error = psid_validate(psid, true); 355 1.2.4.2 yamt if (error) { 356 1.2.4.2 yamt mutex_exit(&psets_lock); 357 1.2.4.2 yamt return error; 358 1.2.4.2 yamt } 359 1.2.4.2 yamt if (psid == PS_MYID) 360 1.2.4.2 yamt psid = curlwp->l_psid; 361 1.2.4.2 yamt if (psid != PS_QUERY && psid != PS_NONE) 362 1.2.4.2 yamt psets[psid - 1]->ps_flags |= PSET_BUSY; 363 1.2.4.2 yamt mutex_exit(&psets_lock); 364 1.2.4.2 yamt 365 1.2.4.2 yamt /* 366 1.2.4.2 yamt * Get PID and LID from the ID. 367 1.2.4.2 yamt */ 368 1.2.4.2 yamt p = l->l_proc; 369 1.2.4.2 yamt id1 = SCARG(uap, first_id); 370 1.2.4.2 yamt id2 = SCARG(uap, second_id); 371 1.2.4.2 yamt 372 1.2.4.2 yamt switch (SCARG(uap, idtype)) { 373 1.2.4.2 yamt case P_PID: 374 1.2.4.2 yamt /* 375 1.2.4.2 yamt * Process: 376 1.2.4.2 yamt * First ID - PID; 377 1.2.4.2 yamt * Second ID - ignored; 378 1.2.4.2 yamt */ 379 1.2.4.2 yamt pid = (id1 == P_MYID) ? p->p_pid : id1; 380 1.2.4.2 yamt lid = 0; 381 1.2.4.2 yamt break; 382 1.2.4.2 yamt case P_LWPID: 383 1.2.4.2 yamt /* 384 1.2.4.2 yamt * Thread (LWP): 385 1.2.4.2 yamt * First ID - LID; 386 1.2.4.2 yamt * Second ID - PID; 387 1.2.4.2 yamt */ 388 1.2.4.2 yamt if (id1 == P_MYID) { 389 1.2.4.2 yamt pid = p->p_pid; 390 1.2.4.2 yamt lid = l->l_lid; 391 1.2.4.2 yamt break; 392 1.2.4.2 yamt } 393 1.2.4.2 yamt lid = id1; 394 1.2.4.2 yamt pid = (id2 == P_MYID) ? p->p_pid : id2; 395 1.2.4.2 yamt break; 396 1.2.4.2 yamt default: 397 1.2.4.2 yamt error = EINVAL; 398 1.2.4.2 yamt goto error; 399 1.2.4.2 yamt } 400 1.2.4.2 yamt 401 1.2.4.2 yamt /* Find the process */ 402 1.2.4.2 yamt p = p_find(pid, PFIND_UNLOCK_FAIL); 403 1.2.4.2 yamt if (p == NULL) { 404 1.2.4.2 yamt error = ESRCH; 405 1.2.4.2 yamt goto error; 406 1.2.4.2 yamt } 407 1.2.4.2 yamt mutex_enter(&p->p_smutex); 408 1.2.4.2 yamt mutex_exit(&proclist_lock); 409 1.2.4.2 yamt 410 1.2.4.2 yamt /* Disallow modification of the system processes */ 411 1.2.4.2 yamt if (p->p_flag & PK_SYSTEM) { 412 1.2.4.2 yamt mutex_exit(&p->p_smutex); 413 1.2.4.2 yamt error = EPERM; 414 1.2.4.2 yamt goto error; 415 1.2.4.2 yamt } 416 1.2.4.2 yamt 417 1.2.4.2 yamt /* Find the LWP(s) */ 418 1.2.4.2 yamt lcnt = 0; 419 1.2.4.2 yamt ci = NULL; 420 1.2.4.2 yamt LIST_FOREACH(t, &p->p_lwps, l_sibling) { 421 1.2.4.2 yamt if (lid && lid != t->l_lid) 422 1.2.4.2 yamt continue; 423 1.2.4.2 yamt /* 424 1.2.4.2 yamt * Bind the thread to the processor-set, 425 1.2.4.2 yamt * take some CPU and migrate. 426 1.2.4.2 yamt */ 427 1.2.4.2 yamt lwp_lock(t); 428 1.2.4.2 yamt opsid = t->l_psid; 429 1.2.4.2 yamt t->l_psid = psid; 430 1.2.4.2 yamt ci = sched_takecpu(l); 431 1.2.4.2 yamt /* Unlocks LWP */ 432 1.2.4.2 yamt lwp_migrate(t, ci); 433 1.2.4.2 yamt lcnt++; 434 1.2.4.2 yamt } 435 1.2.4.2 yamt mutex_exit(&p->p_smutex); 436 1.2.4.2 yamt if (lcnt == 0) { 437 1.2.4.2 yamt error = ESRCH; 438 1.2.4.2 yamt goto error; 439 1.2.4.2 yamt } 440 1.2.4.2 yamt *retval = lcnt; 441 1.2.4.2 yamt if (SCARG(uap, opsid)) 442 1.2.4.2 yamt error = copyout(&opsid, SCARG(uap, opsid), sizeof(psetid_t)); 443 1.2.4.2 yamt error: 444 1.2.4.2 yamt if (psid != PS_QUERY && psid != PS_NONE) { 445 1.2.4.2 yamt mutex_enter(&psets_lock); 446 1.2.4.2 yamt psets[psid - 1]->ps_flags &= ~PSET_BUSY; 447 1.2.4.2 yamt mutex_exit(&psets_lock); 448 1.2.4.2 yamt } 449 1.2.4.2 yamt return error; 450 1.2.4.2 yamt } 451 1.2.4.2 yamt 452 1.2.4.2 yamt /* 453 1.2.4.2 yamt * Sysctl nodes and initialization. 454 1.2.4.2 yamt */ 455 1.2.4.2 yamt 456 1.2.4.2 yamt static int 457 1.2.4.2 yamt sysctl_psets_max(SYSCTLFN_ARGS) 458 1.2.4.2 yamt { 459 1.2.4.2 yamt struct sysctlnode node; 460 1.2.4.2 yamt int error, newsize; 461 1.2.4.2 yamt 462 1.2.4.2 yamt node = *rnode; 463 1.2.4.2 yamt node.sysctl_data = &newsize; 464 1.2.4.2 yamt 465 1.2.4.2 yamt newsize = psets_max; 466 1.2.4.2 yamt error = sysctl_lookup(SYSCTLFN_CALL(&node)); 467 1.2.4.2 yamt if (error || newp == NULL) 468 1.2.4.2 yamt return error; 469 1.2.4.2 yamt 470 1.2.4.2 yamt if (newsize <= 0) 471 1.2.4.2 yamt return EINVAL; 472 1.2.4.2 yamt 473 1.2.4.2 yamt sysctl_unlock(); 474 1.2.4.2 yamt error = psets_realloc(newsize); 475 1.2.4.2 yamt sysctl_relock(); 476 1.2.4.2 yamt return error; 477 1.2.4.2 yamt } 478 1.2.4.2 yamt 479 1.2.4.2 yamt SYSCTL_SETUP(sysctl_pset_setup, "sysctl kern.pset subtree setup") 480 1.2.4.2 yamt { 481 1.2.4.2 yamt const struct sysctlnode *node = NULL; 482 1.2.4.2 yamt 483 1.2.4.2 yamt sysctl_createv(clog, 0, NULL, NULL, 484 1.2.4.2 yamt CTLFLAG_PERMANENT, 485 1.2.4.2 yamt CTLTYPE_NODE, "kern", NULL, 486 1.2.4.2 yamt NULL, 0, NULL, 0, 487 1.2.4.2 yamt CTL_KERN, CTL_EOL); 488 1.2.4.2 yamt sysctl_createv(clog, 0, NULL, &node, 489 1.2.4.2 yamt CTLFLAG_PERMANENT, 490 1.2.4.2 yamt CTLTYPE_NODE, "pset", 491 1.2.4.2 yamt SYSCTL_DESCR("Processor-set options"), 492 1.2.4.2 yamt NULL, 0, NULL, 0, 493 1.2.4.2 yamt CTL_KERN, CTL_CREATE, CTL_EOL); 494 1.2.4.2 yamt 495 1.2.4.2 yamt if (node == NULL) 496 1.2.4.2 yamt return; 497 1.2.4.2 yamt 498 1.2.4.2 yamt sysctl_createv(clog, 0, &node, NULL, 499 1.2.4.2 yamt CTLFLAG_PERMANENT | CTLFLAG_READWRITE, 500 1.2.4.2 yamt CTLTYPE_INT, "psets_max", 501 1.2.4.2 yamt SYSCTL_DESCR("Maximal count of the processor-sets"), 502 1.2.4.2 yamt sysctl_psets_max, 0, &psets_max, 0, 503 1.2.4.2 yamt CTL_CREATE, CTL_EOL); 504 1.2.4.2 yamt } 505
Indexes created Tue Sep 30 20:09:53 GMT 2025