1d7f687fcSJeff Roberson /*- 28a36da99SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 38a36da99SPedro F. Giffuni * 4d7f687fcSJeff Roberson * Copyright (c) 2008, Jeffrey Roberson <jeff@freebsd.org> 5d7f687fcSJeff Roberson * All rights reserved. 6d7f687fcSJeff Roberson * 73bc8c68dSJeff Roberson * Copyright (c) 2008 Nokia Corporation 83bc8c68dSJeff Roberson * All rights reserved. 93bc8c68dSJeff Roberson * 10d7f687fcSJeff Roberson * Redistribution and use in source and binary forms, with or without 11d7f687fcSJeff Roberson * modification, are permitted provided that the following conditions 12d7f687fcSJeff Roberson * are met: 13d7f687fcSJeff Roberson * 1. Redistributions of source code must retain the above copyright 14d7f687fcSJeff Roberson * notice unmodified, this list of conditions, and the following 15d7f687fcSJeff Roberson * disclaimer. 16d7f687fcSJeff Roberson * 2. Redistributions in binary form must reproduce the above copyright 17d7f687fcSJeff Roberson * notice, this list of conditions and the following disclaimer in the 18d7f687fcSJeff Roberson * documentation and/or other materials provided with the distribution. 19d7f687fcSJeff Roberson * 20d7f687fcSJeff Roberson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 21d7f687fcSJeff Roberson * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 22d7f687fcSJeff Roberson * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 23d7f687fcSJeff Roberson * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 24d7f687fcSJeff Roberson * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 25d7f687fcSJeff Roberson * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26d7f687fcSJeff Roberson * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27d7f687fcSJeff Roberson * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28d7f687fcSJeff Roberson * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 29d7f687fcSJeff Roberson * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30d7f687fcSJeff Roberson * 31d7f687fcSJeff Roberson */ 32d7f687fcSJeff Roberson 33d7f687fcSJeff Roberson #include <sys/cdefs.h> 34d7f687fcSJeff Roberson __FBSDID("$FreeBSD$"); 35d7f687fcSJeff Roberson 36dea0ed66SBjoern A. Zeeb #include "opt_ddb.h" 37dea0ed66SBjoern A. Zeeb 38d7f687fcSJeff Roberson #include <sys/param.h> 39d7f687fcSJeff Roberson #include <sys/systm.h> 40e5818a53SJeff Roberson #include <sys/sysctl.h> 41e5818a53SJeff Roberson #include <sys/ctype.h> 42d7f687fcSJeff Roberson #include <sys/sysproto.h> 430304c731SJamie Gritton #include <sys/jail.h> 44d7f687fcSJeff Roberson #include <sys/kernel.h> 45d7f687fcSJeff Roberson #include <sys/lock.h> 46d7f687fcSJeff Roberson #include <sys/malloc.h> 47d7f687fcSJeff Roberson #include <sys/mutex.h> 48d7f687fcSJeff Roberson #include <sys/priv.h> 49d7f687fcSJeff Roberson #include <sys/proc.h> 50d7f687fcSJeff Roberson #include <sys/refcount.h> 51d7f687fcSJeff Roberson #include <sys/sched.h> 52d7f687fcSJeff Roberson #include <sys/smp.h> 53d7f687fcSJeff Roberson #include <sys/syscallsubr.h> 54f299c47bSAllan Jude #include <sys/capsicum.h> 55d7f687fcSJeff Roberson #include <sys/cpuset.h> 563f289c3fSJeff Roberson #include <sys/domainset.h> 57d7f687fcSJeff Roberson #include <sys/sx.h> 58d7f687fcSJeff Roberson #include <sys/queue.h> 59e3709597SAttilio Rao #include <sys/libkern.h> 60d7f687fcSJeff Roberson #include <sys/limits.h> 61a03ee000SJeff Roberson #include <sys/bus.h> 62a03ee000SJeff Roberson #include <sys/interrupt.h> 633f289c3fSJeff Roberson #include <sys/vmmeter.h> 64d7f687fcSJeff Roberson 65d7f687fcSJeff Roberson #include <vm/uma.h> 66c0ae6688SJohn Baldwin #include <vm/vm.h> 673f289c3fSJeff Roberson #include <vm/vm_object.h> 6830c5525bSAndrew Gallatin #include <vm/vm_page.h> 6930c5525bSAndrew Gallatin #include <vm/vm_pageout.h> 70e5818a53SJeff Roberson #include <vm/vm_extern.h> 7130c5525bSAndrew Gallatin #include <vm/vm_param.h> 7230c5525bSAndrew Gallatin #include <vm/vm_phys.h> 7330c5525bSAndrew Gallatin #include <vm/vm_pagequeue.h> 74d7f687fcSJeff Roberson 75dea0ed66SBjoern A. Zeeb #ifdef DDB 76dea0ed66SBjoern A. Zeeb #include <ddb/ddb.h> 77dea0ed66SBjoern A. Zeeb #endif /* DDB */ 78dea0ed66SBjoern A. Zeeb 79d7f687fcSJeff Roberson /* 80d7f687fcSJeff Roberson * cpusets provide a mechanism for creating and manipulating sets of 81d7f687fcSJeff Roberson * processors for the purpose of constraining the scheduling of threads to 82d7f687fcSJeff Roberson * specific processors. 83d7f687fcSJeff Roberson * 84d7f687fcSJeff Roberson * Each process belongs to an identified set, by default this is set 1. Each 85d7f687fcSJeff Roberson * thread may further restrict the cpus it may run on to a subset of this 86d7f687fcSJeff Roberson * named set. This creates an anonymous set which other threads and processes 87d7f687fcSJeff Roberson * may not join by number. 88d7f687fcSJeff Roberson * 89d7f687fcSJeff Roberson * The named set is referred to herein as the 'base' set to avoid ambiguity. 90d7f687fcSJeff Roberson * This set is usually a child of a 'root' set while the anonymous set may 91d7f687fcSJeff Roberson * simply be referred to as a mask. In the syscall api these are referred to 92d7f687fcSJeff Roberson * as the ROOT, CPUSET, and MASK levels where CPUSET is called 'base' here. 93d7f687fcSJeff Roberson * 94d7f687fcSJeff Roberson * Threads inherit their set from their creator whether it be anonymous or 95d7f687fcSJeff Roberson * not. This means that anonymous sets are immutable because they may be 96d7f687fcSJeff Roberson * shared. To modify an anonymous set a new set is created with the desired 97d7f687fcSJeff Roberson * mask and the same parent as the existing anonymous set. This gives the 9893902625SJohn Baldwin * illusion of each thread having a private mask. 99d7f687fcSJeff Roberson * 100d7f687fcSJeff Roberson * Via the syscall apis a user may ask to retrieve or modify the root, base, 101d7f687fcSJeff Roberson * or mask that is discovered via a pid, tid, or setid. Modifying a set 102d7f687fcSJeff Roberson * modifies all numbered and anonymous child sets to comply with the new mask. 103d7f687fcSJeff Roberson * Modifying a pid or tid's mask applies only to that tid but must still 104d7f687fcSJeff Roberson * exist within the assigned parent set. 105d7f687fcSJeff Roberson * 10686855bf5SJohn Baldwin * A thread may not be assigned to a group separate from other threads in 107d7f687fcSJeff Roberson * the process. This is to remove ambiguity when the setid is queried with 108d7f687fcSJeff Roberson * a pid argument. There is no other technical limitation. 109d7f687fcSJeff Roberson * 110d7f687fcSJeff Roberson * This somewhat complex arrangement is intended to make it easy for 111d7f687fcSJeff Roberson * applications to query available processors and bind their threads to 112d7f687fcSJeff Roberson * specific processors while also allowing administrators to dynamically 113d7f687fcSJeff Roberson * reprovision by changing sets which apply to groups of processes. 114d7f687fcSJeff Roberson * 115d7f687fcSJeff Roberson * A simple application should not concern itself with sets at all and 116d7f687fcSJeff Roberson * rather apply masks to its own threads via CPU_WHICH_TID and a -1 id 11793902625SJohn Baldwin * meaning 'curthread'. It may query available cpus for that tid with a 118d7f687fcSJeff Roberson * getaffinity call using (CPU_LEVEL_CPUSET, CPU_WHICH_PID, -1, ...). 119d7f687fcSJeff Roberson */ 120e5818a53SJeff Roberson 121e5818a53SJeff Roberson LIST_HEAD(domainlist, domainset); 1229978bd99SMark Johnston struct domainset __read_mostly domainset_fixed[MAXMEMDOM]; 123662e7fa8SMark Johnston struct domainset __read_mostly domainset_prefer[MAXMEMDOM]; 124662e7fa8SMark Johnston struct domainset __read_mostly domainset_roundrobin; 125e5818a53SJeff Roberson 126d7f687fcSJeff Roberson static uma_zone_t cpuset_zone; 1273f289c3fSJeff Roberson static uma_zone_t domainset_zone; 128d7f687fcSJeff Roberson static struct mtx cpuset_lock; 129d7f687fcSJeff Roberson static struct setlist cpuset_ids; 1303f289c3fSJeff Roberson static struct domainlist cpuset_domains; 131d7f687fcSJeff Roberson static struct unrhdr *cpuset_unr; 132e5818a53SJeff Roberson static struct cpuset *cpuset_zero, *cpuset_default, *cpuset_kernel; 133e5818a53SJeff Roberson static struct domainset domainset0, domainset2; 134a03ee000SJeff Roberson 135444528c0SDavid Xu /* Return the size of cpuset_t at the kernel level */ 13660aa2c85SJonathan Anderson SYSCTL_INT(_kern_sched, OID_AUTO, cpusetsize, CTLFLAG_RD | CTLFLAG_CAPRD, 137f0188618SHans Petter Selasky SYSCTL_NULL_INT_PTR, sizeof(cpuset_t), "sizeof(cpuset_t)"); 138444528c0SDavid Xu 139a03ee000SJeff Roberson cpuset_t *cpuset_root; 140c0ae6688SJohn Baldwin cpuset_t cpuset_domain[MAXMEMDOM]; 141d7f687fcSJeff Roberson 1423f289c3fSJeff Roberson static int domainset_valid(const struct domainset *, const struct domainset *); 1433f289c3fSJeff Roberson 1443f289c3fSJeff Roberson /* 1453f289c3fSJeff Roberson * Find the first non-anonymous set starting from 'set'. 1463f289c3fSJeff Roberson */ 1473f289c3fSJeff Roberson static struct cpuset * 1483f289c3fSJeff Roberson cpuset_getbase(struct cpuset *set) 1493f289c3fSJeff Roberson { 1503f289c3fSJeff Roberson 1513f289c3fSJeff Roberson if (set->cs_id == CPUSET_INVALID) 1523f289c3fSJeff Roberson set = set->cs_parent; 1533f289c3fSJeff Roberson return (set); 1543f289c3fSJeff Roberson } 1553f289c3fSJeff Roberson 1563f289c3fSJeff Roberson /* 1573f289c3fSJeff Roberson * Walks up the tree from 'set' to find the root. 1583f289c3fSJeff Roberson */ 1593f289c3fSJeff Roberson static struct cpuset * 1603f289c3fSJeff Roberson cpuset_getroot(struct cpuset *set) 1613f289c3fSJeff Roberson { 1623f289c3fSJeff Roberson 1633f289c3fSJeff Roberson while ((set->cs_flags & CPU_SET_ROOT) == 0 && set->cs_parent != NULL) 1643f289c3fSJeff Roberson set = set->cs_parent; 1653f289c3fSJeff Roberson return (set); 1663f289c3fSJeff Roberson } 1673f289c3fSJeff Roberson 168d7f687fcSJeff Roberson /* 169d7f687fcSJeff Roberson * Acquire a reference to a cpuset, all pointers must be tracked with refs. 170d7f687fcSJeff Roberson */ 171d7f687fcSJeff Roberson struct cpuset * 172d7f687fcSJeff Roberson cpuset_ref(struct cpuset *set) 173d7f687fcSJeff Roberson { 174d7f687fcSJeff Roberson 175d7f687fcSJeff Roberson refcount_acquire(&set->cs_ref); 176d7f687fcSJeff Roberson return (set); 177d7f687fcSJeff Roberson } 178d7f687fcSJeff Roberson 179d7f687fcSJeff Roberson /* 1807a8f695aSBjoern A. Zeeb * Walks up the tree from 'set' to find the root. Returns the root 1817a8f695aSBjoern A. Zeeb * referenced. 1827a8f695aSBjoern A. Zeeb */ 1837a8f695aSBjoern A. Zeeb static struct cpuset * 1847a8f695aSBjoern A. Zeeb cpuset_refroot(struct cpuset *set) 1857a8f695aSBjoern A. Zeeb { 1867a8f695aSBjoern A. Zeeb 1873f289c3fSJeff Roberson return (cpuset_ref(cpuset_getroot(set))); 1887a8f695aSBjoern A. Zeeb } 1897a8f695aSBjoern A. Zeeb 1907a8f695aSBjoern A. Zeeb /* 1917a8f695aSBjoern A. Zeeb * Find the first non-anonymous set starting from 'set'. Returns this set 1927a8f695aSBjoern A. Zeeb * referenced. May return the passed in set with an extra ref if it is 1937a8f695aSBjoern A. Zeeb * not anonymous. 1947a8f695aSBjoern A. Zeeb */ 1957a8f695aSBjoern A. Zeeb static struct cpuset * 1967a8f695aSBjoern A. Zeeb cpuset_refbase(struct cpuset *set) 1977a8f695aSBjoern A. Zeeb { 1987a8f695aSBjoern A. Zeeb 1993f289c3fSJeff Roberson return (cpuset_ref(cpuset_getbase(set))); 2007a8f695aSBjoern A. Zeeb } 2017a8f695aSBjoern A. Zeeb 2027a8f695aSBjoern A. Zeeb /* 20393902625SJohn Baldwin * Release a reference in a context where it is safe to allocate. 204d7f687fcSJeff Roberson */ 205d7f687fcSJeff Roberson void 206d7f687fcSJeff Roberson cpuset_rel(struct cpuset *set) 207d7f687fcSJeff Roberson { 208d7f687fcSJeff Roberson cpusetid_t id; 209d7f687fcSJeff Roberson 210d7f687fcSJeff Roberson if (refcount_release(&set->cs_ref) == 0) 211d7f687fcSJeff Roberson return; 212d7f687fcSJeff Roberson mtx_lock_spin(&cpuset_lock); 213d7f687fcSJeff Roberson LIST_REMOVE(set, cs_siblings); 214d7f687fcSJeff Roberson id = set->cs_id; 215d7f687fcSJeff Roberson if (id != CPUSET_INVALID) 216d7f687fcSJeff Roberson LIST_REMOVE(set, cs_link); 217d7f687fcSJeff Roberson mtx_unlock_spin(&cpuset_lock); 218d7f687fcSJeff Roberson cpuset_rel(set->cs_parent); 219d7f687fcSJeff Roberson uma_zfree(cpuset_zone, set); 220d7f687fcSJeff Roberson if (id != CPUSET_INVALID) 221d7f687fcSJeff Roberson free_unr(cpuset_unr, id); 222d7f687fcSJeff Roberson } 223d7f687fcSJeff Roberson 224d7f687fcSJeff Roberson /* 225d7f687fcSJeff Roberson * Deferred release must be used when in a context that is not safe to 226d7f687fcSJeff Roberson * allocate/free. This places any unreferenced sets on the list 'head'. 227d7f687fcSJeff Roberson */ 228d7f687fcSJeff Roberson static void 229d7f687fcSJeff Roberson cpuset_rel_defer(struct setlist *head, struct cpuset *set) 230d7f687fcSJeff Roberson { 231d7f687fcSJeff Roberson 232d7f687fcSJeff Roberson if (refcount_release(&set->cs_ref) == 0) 233d7f687fcSJeff Roberson return; 234d7f687fcSJeff Roberson mtx_lock_spin(&cpuset_lock); 235d7f687fcSJeff Roberson LIST_REMOVE(set, cs_siblings); 236d7f687fcSJeff Roberson if (set->cs_id != CPUSET_INVALID) 237d7f687fcSJeff Roberson LIST_REMOVE(set, cs_link); 238d7f687fcSJeff Roberson LIST_INSERT_HEAD(head, set, cs_link); 239d7f687fcSJeff Roberson mtx_unlock_spin(&cpuset_lock); 240d7f687fcSJeff Roberson } 241d7f687fcSJeff Roberson 242d7f687fcSJeff Roberson /* 243d7f687fcSJeff Roberson * Complete a deferred release. Removes the set from the list provided to 244d7f687fcSJeff Roberson * cpuset_rel_defer. 245d7f687fcSJeff Roberson */ 246d7f687fcSJeff Roberson static void 247d7f687fcSJeff Roberson cpuset_rel_complete(struct cpuset *set) 248d7f687fcSJeff Roberson { 249d7f687fcSJeff Roberson LIST_REMOVE(set, cs_link); 250d7f687fcSJeff Roberson cpuset_rel(set->cs_parent); 251d7f687fcSJeff Roberson uma_zfree(cpuset_zone, set); 252d7f687fcSJeff Roberson } 253d7f687fcSJeff Roberson 254d7f687fcSJeff Roberson /* 255d7f687fcSJeff Roberson * Find a set based on an id. Returns it with a ref. 256d7f687fcSJeff Roberson */ 257d7f687fcSJeff Roberson static struct cpuset * 258413628a7SBjoern A. Zeeb cpuset_lookup(cpusetid_t setid, struct thread *td) 259d7f687fcSJeff Roberson { 260d7f687fcSJeff Roberson struct cpuset *set; 261d7f687fcSJeff Roberson 262d7f687fcSJeff Roberson if (setid == CPUSET_INVALID) 263d7f687fcSJeff Roberson return (NULL); 264d7f687fcSJeff Roberson mtx_lock_spin(&cpuset_lock); 265d7f687fcSJeff Roberson LIST_FOREACH(set, &cpuset_ids, cs_link) 266d7f687fcSJeff Roberson if (set->cs_id == setid) 267d7f687fcSJeff Roberson break; 268d7f687fcSJeff Roberson if (set) 269d7f687fcSJeff Roberson cpuset_ref(set); 270d7f687fcSJeff Roberson mtx_unlock_spin(&cpuset_lock); 271413628a7SBjoern A. Zeeb 272413628a7SBjoern A. Zeeb KASSERT(td != NULL, ("[%s:%d] td is NULL", __func__, __LINE__)); 273413628a7SBjoern A. Zeeb if (set != NULL && jailed(td->td_ucred)) { 2740304c731SJamie Gritton struct cpuset *jset, *tset; 275413628a7SBjoern A. Zeeb 2760304c731SJamie Gritton jset = td->td_ucred->cr_prison->pr_cpuset; 2770304c731SJamie Gritton for (tset = set; tset != NULL; tset = tset->cs_parent) 2780304c731SJamie Gritton if (tset == jset) 2790304c731SJamie Gritton break; 2800304c731SJamie Gritton if (tset == NULL) { 281413628a7SBjoern A. Zeeb cpuset_rel(set); 282413628a7SBjoern A. Zeeb set = NULL; 283413628a7SBjoern A. Zeeb } 284413628a7SBjoern A. Zeeb } 285413628a7SBjoern A. Zeeb 286d7f687fcSJeff Roberson return (set); 287d7f687fcSJeff Roberson } 288d7f687fcSJeff Roberson 289d7f687fcSJeff Roberson /* 29030b7c6f9SKyle Evans * Initialize a set in the space provided in 'set' with the provided parameters. 291d7f687fcSJeff Roberson * The set is returned with a single ref. May return EDEADLK if the set 292d7f687fcSJeff Roberson * will have no valid cpu based on restrictions from the parent. 293d7f687fcSJeff Roberson */ 294d7f687fcSJeff Roberson static int 29530b7c6f9SKyle Evans cpuset_init(struct cpuset *set, struct cpuset *parent, 2963f289c3fSJeff Roberson const cpuset_t *mask, struct domainset *domain, cpusetid_t id) 297d7f687fcSJeff Roberson { 298d7f687fcSJeff Roberson 2993f289c3fSJeff Roberson if (domain == NULL) 3003f289c3fSJeff Roberson domain = parent->cs_domain; 3013f289c3fSJeff Roberson if (mask == NULL) 3023f289c3fSJeff Roberson mask = &parent->cs_mask; 30373c40187SJeff Roberson if (!CPU_OVERLAP(&parent->cs_mask, mask)) 30473c40187SJeff Roberson return (EDEADLK); 3053f289c3fSJeff Roberson /* The domain must be prepared ahead of time. */ 3063f289c3fSJeff Roberson if (!domainset_valid(parent->cs_domain, domain)) 3073f289c3fSJeff Roberson return (EDEADLK); 308d7f687fcSJeff Roberson CPU_COPY(mask, &set->cs_mask); 309d7f687fcSJeff Roberson LIST_INIT(&set->cs_children); 310d7f687fcSJeff Roberson refcount_init(&set->cs_ref, 1); 311d7f687fcSJeff Roberson set->cs_flags = 0; 312d7f687fcSJeff Roberson mtx_lock_spin(&cpuset_lock); 3133f289c3fSJeff Roberson set->cs_domain = domain; 314e84c2db1SJohn Baldwin CPU_AND(&set->cs_mask, &parent->cs_mask); 315d7f687fcSJeff Roberson set->cs_id = id; 316d7f687fcSJeff Roberson set->cs_parent = cpuset_ref(parent); 317d7f687fcSJeff Roberson LIST_INSERT_HEAD(&parent->cs_children, set, cs_siblings); 318d7f687fcSJeff Roberson if (set->cs_id != CPUSET_INVALID) 319d7f687fcSJeff Roberson LIST_INSERT_HEAD(&cpuset_ids, set, cs_link); 320d7f687fcSJeff Roberson mtx_unlock_spin(&cpuset_lock); 321d7f687fcSJeff Roberson 32273c40187SJeff Roberson return (0); 323d7f687fcSJeff Roberson } 324d7f687fcSJeff Roberson 325d7f687fcSJeff Roberson /* 326d7f687fcSJeff Roberson * Create a new non-anonymous set with the requested parent and mask. May 327d7f687fcSJeff Roberson * return failures if the mask is invalid or a new number can not be 328d7f687fcSJeff Roberson * allocated. 32929d04ea8SKyle Evans * 33029d04ea8SKyle Evans * If *setp is not NULL, then it will be used as-is. The caller must take 33129d04ea8SKyle Evans * into account that *setp will be inserted at the head of cpuset_ids and 33229d04ea8SKyle Evans * plan any potentially conflicting cs_link usage accordingly. 333d7f687fcSJeff Roberson */ 334d7f687fcSJeff Roberson static int 335e84c2db1SJohn Baldwin cpuset_create(struct cpuset **setp, struct cpuset *parent, const cpuset_t *mask) 336d7f687fcSJeff Roberson { 337d7f687fcSJeff Roberson struct cpuset *set; 338d7f687fcSJeff Roberson cpusetid_t id; 339d7f687fcSJeff Roberson int error; 34029d04ea8SKyle Evans bool dofree; 341d7f687fcSJeff Roberson 342d7f687fcSJeff Roberson id = alloc_unr(cpuset_unr); 343d7f687fcSJeff Roberson if (id == -1) 344d7f687fcSJeff Roberson return (ENFILE); 34529d04ea8SKyle Evans dofree = (*setp == NULL); 34629d04ea8SKyle Evans if (*setp != NULL) 34729d04ea8SKyle Evans set = *setp; 34829d04ea8SKyle Evans else 3493f289c3fSJeff Roberson *setp = set = uma_zalloc(cpuset_zone, M_WAITOK | M_ZERO); 35030b7c6f9SKyle Evans error = cpuset_init(set, parent, mask, NULL, id); 351d7f687fcSJeff Roberson if (error == 0) 352d7f687fcSJeff Roberson return (0); 353d7f687fcSJeff Roberson free_unr(cpuset_unr, id); 35429d04ea8SKyle Evans if (dofree) 355d7f687fcSJeff Roberson uma_zfree(cpuset_zone, set); 356d7f687fcSJeff Roberson 357d7f687fcSJeff Roberson return (error); 358d7f687fcSJeff Roberson } 359d7f687fcSJeff Roberson 3603f289c3fSJeff Roberson static void 3613f289c3fSJeff Roberson cpuset_freelist_add(struct setlist *list, int count) 3623f289c3fSJeff Roberson { 3633f289c3fSJeff Roberson struct cpuset *set; 3643f289c3fSJeff Roberson int i; 3653f289c3fSJeff Roberson 3663f289c3fSJeff Roberson for (i = 0; i < count; i++) { 3673f289c3fSJeff Roberson set = uma_zalloc(cpuset_zone, M_ZERO | M_WAITOK); 3683f289c3fSJeff Roberson LIST_INSERT_HEAD(list, set, cs_link); 3693f289c3fSJeff Roberson } 3703f289c3fSJeff Roberson } 3713f289c3fSJeff Roberson 3723f289c3fSJeff Roberson static void 3733f289c3fSJeff Roberson cpuset_freelist_init(struct setlist *list, int count) 3743f289c3fSJeff Roberson { 3753f289c3fSJeff Roberson 3763f289c3fSJeff Roberson LIST_INIT(list); 3773f289c3fSJeff Roberson cpuset_freelist_add(list, count); 3783f289c3fSJeff Roberson } 3793f289c3fSJeff Roberson 3803f289c3fSJeff Roberson static void 3813f289c3fSJeff Roberson cpuset_freelist_free(struct setlist *list) 3823f289c3fSJeff Roberson { 3833f289c3fSJeff Roberson struct cpuset *set; 3843f289c3fSJeff Roberson 3853f289c3fSJeff Roberson while ((set = LIST_FIRST(list)) != NULL) { 3863f289c3fSJeff Roberson LIST_REMOVE(set, cs_link); 3873f289c3fSJeff Roberson uma_zfree(cpuset_zone, set); 3883f289c3fSJeff Roberson } 3893f289c3fSJeff Roberson } 3903f289c3fSJeff Roberson 3913f289c3fSJeff Roberson static void 3923f289c3fSJeff Roberson domainset_freelist_add(struct domainlist *list, int count) 3933f289c3fSJeff Roberson { 3943f289c3fSJeff Roberson struct domainset *set; 3953f289c3fSJeff Roberson int i; 3963f289c3fSJeff Roberson 3973f289c3fSJeff Roberson for (i = 0; i < count; i++) { 3983f289c3fSJeff Roberson set = uma_zalloc(domainset_zone, M_ZERO | M_WAITOK); 3993f289c3fSJeff Roberson LIST_INSERT_HEAD(list, set, ds_link); 4003f289c3fSJeff Roberson } 4013f289c3fSJeff Roberson } 4023f289c3fSJeff Roberson 4033f289c3fSJeff Roberson static void 4043f289c3fSJeff Roberson domainset_freelist_init(struct domainlist *list, int count) 4053f289c3fSJeff Roberson { 4063f289c3fSJeff Roberson 4073f289c3fSJeff Roberson LIST_INIT(list); 4083f289c3fSJeff Roberson domainset_freelist_add(list, count); 4093f289c3fSJeff Roberson } 4103f289c3fSJeff Roberson 4113f289c3fSJeff Roberson static void 4123f289c3fSJeff Roberson domainset_freelist_free(struct domainlist *list) 4133f289c3fSJeff Roberson { 4143f289c3fSJeff Roberson struct domainset *set; 4153f289c3fSJeff Roberson 4163f289c3fSJeff Roberson while ((set = LIST_FIRST(list)) != NULL) { 4173f289c3fSJeff Roberson LIST_REMOVE(set, ds_link); 4183f289c3fSJeff Roberson uma_zfree(domainset_zone, set); 4193f289c3fSJeff Roberson } 4203f289c3fSJeff Roberson } 4213f289c3fSJeff Roberson 4223f289c3fSJeff Roberson /* Copy a domainset preserving mask and policy. */ 4233f289c3fSJeff Roberson static void 4243f289c3fSJeff Roberson domainset_copy(const struct domainset *from, struct domainset *to) 4253f289c3fSJeff Roberson { 4263f289c3fSJeff Roberson 4273f289c3fSJeff Roberson DOMAINSET_COPY(&from->ds_mask, &to->ds_mask); 4283f289c3fSJeff Roberson to->ds_policy = from->ds_policy; 4293f289c3fSJeff Roberson to->ds_prefer = from->ds_prefer; 4303f289c3fSJeff Roberson } 4313f289c3fSJeff Roberson 4323f289c3fSJeff Roberson /* Return 1 if mask and policy are equal, otherwise 0. */ 4333f289c3fSJeff Roberson static int 4343f289c3fSJeff Roberson domainset_equal(const struct domainset *one, const struct domainset *two) 4353f289c3fSJeff Roberson { 4363f289c3fSJeff Roberson 4373f289c3fSJeff Roberson return (DOMAINSET_CMP(&one->ds_mask, &two->ds_mask) == 0 && 4383f289c3fSJeff Roberson one->ds_policy == two->ds_policy && 4393f289c3fSJeff Roberson one->ds_prefer == two->ds_prefer); 4403f289c3fSJeff Roberson } 4413f289c3fSJeff Roberson 4423f289c3fSJeff Roberson /* Return 1 if child is a valid subset of parent. */ 4433f289c3fSJeff Roberson static int 4443f289c3fSJeff Roberson domainset_valid(const struct domainset *parent, const struct domainset *child) 4453f289c3fSJeff Roberson { 4463f289c3fSJeff Roberson if (child->ds_policy != DOMAINSET_POLICY_PREFER) 4473f289c3fSJeff Roberson return (DOMAINSET_SUBSET(&parent->ds_mask, &child->ds_mask)); 4483f289c3fSJeff Roberson return (DOMAINSET_ISSET(child->ds_prefer, &parent->ds_mask)); 4493f289c3fSJeff Roberson } 4503f289c3fSJeff Roberson 4513f289c3fSJeff Roberson static int 4523f289c3fSJeff Roberson domainset_restrict(const struct domainset *parent, 4533f289c3fSJeff Roberson const struct domainset *child) 4543f289c3fSJeff Roberson { 4553f289c3fSJeff Roberson if (child->ds_policy != DOMAINSET_POLICY_PREFER) 4563f289c3fSJeff Roberson return (DOMAINSET_OVERLAP(&parent->ds_mask, &child->ds_mask)); 4573f289c3fSJeff Roberson return (DOMAINSET_ISSET(child->ds_prefer, &parent->ds_mask)); 4583f289c3fSJeff Roberson } 4593f289c3fSJeff Roberson 4603f289c3fSJeff Roberson /* 4613f289c3fSJeff Roberson * Lookup or create a domainset. The key is provided in ds_mask and 4623f289c3fSJeff Roberson * ds_policy. If the domainset does not yet exist the storage in 4633f289c3fSJeff Roberson * 'domain' is used to insert. Otherwise this storage is freed to the 4643f289c3fSJeff Roberson * domainset_zone and the existing domainset is returned. 4653f289c3fSJeff Roberson */ 4663f289c3fSJeff Roberson static struct domainset * 4673f289c3fSJeff Roberson _domainset_create(struct domainset *domain, struct domainlist *freelist) 4683f289c3fSJeff Roberson { 4693f289c3fSJeff Roberson struct domainset *ndomain; 47044e4def7SMark Johnston int i, j; 4713f289c3fSJeff Roberson 472b61f3142SMark Johnston KASSERT(domain->ds_cnt <= vm_ndomains, 473b61f3142SMark Johnston ("invalid domain count in domainset %p", domain)); 474b61f3142SMark Johnston KASSERT(domain->ds_policy != DOMAINSET_POLICY_PREFER || 475b61f3142SMark Johnston domain->ds_prefer < vm_ndomains, 476b61f3142SMark Johnston ("invalid preferred domain in domains %p", domain)); 477b61f3142SMark Johnston 4783f289c3fSJeff Roberson mtx_lock_spin(&cpuset_lock); 4793f289c3fSJeff Roberson LIST_FOREACH(ndomain, &cpuset_domains, ds_link) 4803f289c3fSJeff Roberson if (domainset_equal(ndomain, domain)) 4813f289c3fSJeff Roberson break; 4823f289c3fSJeff Roberson /* 4833f289c3fSJeff Roberson * If the domain does not yet exist we insert it and initialize 4843f289c3fSJeff Roberson * various iteration helpers which are not part of the key. 4853f289c3fSJeff Roberson */ 4863f289c3fSJeff Roberson if (ndomain == NULL) { 4873f289c3fSJeff Roberson LIST_INSERT_HEAD(&cpuset_domains, domain, ds_link); 4883f289c3fSJeff Roberson domain->ds_cnt = DOMAINSET_COUNT(&domain->ds_mask); 48944e4def7SMark Johnston for (i = 0, j = 0; i < DOMAINSET_FLS(&domain->ds_mask); i++) 490e5818a53SJeff Roberson if (DOMAINSET_ISSET(i, &domain->ds_mask)) 491e5818a53SJeff Roberson domain->ds_order[j++] = i; 4923f289c3fSJeff Roberson } 4933f289c3fSJeff Roberson mtx_unlock_spin(&cpuset_lock); 4943f289c3fSJeff Roberson if (ndomain == NULL) 4953f289c3fSJeff Roberson return (domain); 4963f289c3fSJeff Roberson if (freelist != NULL) 4973f289c3fSJeff Roberson LIST_INSERT_HEAD(freelist, domain, ds_link); 4983f289c3fSJeff Roberson else 4993f289c3fSJeff Roberson uma_zfree(domainset_zone, domain); 5003f289c3fSJeff Roberson return (ndomain); 5013f289c3fSJeff Roberson 5023f289c3fSJeff Roberson } 5033f289c3fSJeff Roberson 5043f289c3fSJeff Roberson /* 50530c5525bSAndrew Gallatin * Are any of the domains in the mask empty? If so, silently 506920239efSMark Johnston * remove them and update the domainset accordingly. If only empty 507920239efSMark Johnston * domains are present, we must return failure. 50830c5525bSAndrew Gallatin */ 50930c5525bSAndrew Gallatin static bool 51030c5525bSAndrew Gallatin domainset_empty_vm(struct domainset *domain) 51130c5525bSAndrew Gallatin { 5128e697504SMark Johnston domainset_t empty; 5138e697504SMark Johnston int i, j; 51430c5525bSAndrew Gallatin 5158e697504SMark Johnston DOMAINSET_ZERO(&empty); 5168e697504SMark Johnston for (i = 0; i < vm_ndomains; i++) 5178e697504SMark Johnston if (VM_DOMAIN_EMPTY(i)) 5188e697504SMark Johnston DOMAINSET_SET(i, &empty); 5198e697504SMark Johnston if (DOMAINSET_SUBSET(&empty, &domain->ds_mask)) 5208e697504SMark Johnston return (true); 5218e697504SMark Johnston 5228e697504SMark Johnston /* Remove empty domains from the set and recompute. */ 5239825eadfSRyan Libby DOMAINSET_ANDNOT(&domain->ds_mask, &empty); 524920239efSMark Johnston domain->ds_cnt = DOMAINSET_COUNT(&domain->ds_mask); 5258e697504SMark Johnston for (i = j = 0; i < DOMAINSET_FLS(&domain->ds_mask); i++) 526920239efSMark Johnston if (DOMAINSET_ISSET(i, &domain->ds_mask)) 527920239efSMark Johnston domain->ds_order[j++] = i; 5288e697504SMark Johnston 5298e697504SMark Johnston /* Convert a PREFER policy referencing an empty domain to RR. */ 5308e697504SMark Johnston if (domain->ds_policy == DOMAINSET_POLICY_PREFER && 5318e697504SMark Johnston DOMAINSET_ISSET(domain->ds_prefer, &empty)) { 532920239efSMark Johnston domain->ds_policy = DOMAINSET_POLICY_ROUNDROBIN; 533920239efSMark Johnston domain->ds_prefer = -1; 534920239efSMark Johnston } 53530c5525bSAndrew Gallatin 5368e697504SMark Johnston return (false); 53730c5525bSAndrew Gallatin } 53830c5525bSAndrew Gallatin 53930c5525bSAndrew Gallatin /* 5403f289c3fSJeff Roberson * Create or lookup a domainset based on the key held in 'domain'. 5413f289c3fSJeff Roberson */ 542e5818a53SJeff Roberson struct domainset * 5433f289c3fSJeff Roberson domainset_create(const struct domainset *domain) 5443f289c3fSJeff Roberson { 5453f289c3fSJeff Roberson struct domainset *ndomain; 5463f289c3fSJeff Roberson 547e5818a53SJeff Roberson /* 548e5818a53SJeff Roberson * Validate the policy. It must specify a useable policy number with 549e5818a53SJeff Roberson * only valid domains. Preferred must include the preferred domain 550e5818a53SJeff Roberson * in the mask. 551e5818a53SJeff Roberson */ 552e5818a53SJeff Roberson if (domain->ds_policy <= DOMAINSET_POLICY_INVALID || 553e5818a53SJeff Roberson domain->ds_policy > DOMAINSET_POLICY_MAX) 554e5818a53SJeff Roberson return (NULL); 555e5818a53SJeff Roberson if (domain->ds_policy == DOMAINSET_POLICY_PREFER && 556e5818a53SJeff Roberson !DOMAINSET_ISSET(domain->ds_prefer, &domain->ds_mask)) 557e5818a53SJeff Roberson return (NULL); 558e5818a53SJeff Roberson if (!DOMAINSET_SUBSET(&domainset0.ds_mask, &domain->ds_mask)) 559e5818a53SJeff Roberson return (NULL); 5603f289c3fSJeff Roberson ndomain = uma_zalloc(domainset_zone, M_WAITOK | M_ZERO); 5613f289c3fSJeff Roberson domainset_copy(domain, ndomain); 5623f289c3fSJeff Roberson return _domainset_create(ndomain, NULL); 5633f289c3fSJeff Roberson } 5643f289c3fSJeff Roberson 5653f289c3fSJeff Roberson /* 5663f289c3fSJeff Roberson * Update thread domainset pointers. 5673f289c3fSJeff Roberson */ 5683f289c3fSJeff Roberson static void 5693f289c3fSJeff Roberson domainset_notify(void) 5703f289c3fSJeff Roberson { 5713f289c3fSJeff Roberson struct thread *td; 5723f289c3fSJeff Roberson struct proc *p; 5733f289c3fSJeff Roberson 5743f289c3fSJeff Roberson sx_slock(&allproc_lock); 5753f289c3fSJeff Roberson FOREACH_PROC_IN_SYSTEM(p) { 5763f289c3fSJeff Roberson PROC_LOCK(p); 5773f289c3fSJeff Roberson if (p->p_state == PRS_NEW) { 5783f289c3fSJeff Roberson PROC_UNLOCK(p); 5793f289c3fSJeff Roberson continue; 5803f289c3fSJeff Roberson } 5813f289c3fSJeff Roberson FOREACH_THREAD_IN_PROC(p, td) { 5823f289c3fSJeff Roberson thread_lock(td); 5833f289c3fSJeff Roberson td->td_domain.dr_policy = td->td_cpuset->cs_domain; 5843f289c3fSJeff Roberson thread_unlock(td); 5853f289c3fSJeff Roberson } 5863f289c3fSJeff Roberson PROC_UNLOCK(p); 5873f289c3fSJeff Roberson } 5883f289c3fSJeff Roberson sx_sunlock(&allproc_lock); 589e5818a53SJeff Roberson kernel_object->domain.dr_policy = cpuset_kernel->cs_domain; 5903f289c3fSJeff Roberson } 5913f289c3fSJeff Roberson 5923f289c3fSJeff Roberson /* 5933f289c3fSJeff Roberson * Create a new set that is a subset of a parent. 5943f289c3fSJeff Roberson */ 5953f289c3fSJeff Roberson static struct domainset * 5963f289c3fSJeff Roberson domainset_shadow(const struct domainset *pdomain, 5973f289c3fSJeff Roberson const struct domainset *domain, struct domainlist *freelist) 5983f289c3fSJeff Roberson { 5993f289c3fSJeff Roberson struct domainset *ndomain; 6003f289c3fSJeff Roberson 6013f289c3fSJeff Roberson ndomain = LIST_FIRST(freelist); 6023f289c3fSJeff Roberson LIST_REMOVE(ndomain, ds_link); 6033f289c3fSJeff Roberson 6043f289c3fSJeff Roberson /* 6053f289c3fSJeff Roberson * Initialize the key from the request. 6063f289c3fSJeff Roberson */ 6073f289c3fSJeff Roberson domainset_copy(domain, ndomain); 6083f289c3fSJeff Roberson 6093f289c3fSJeff Roberson /* 6103f289c3fSJeff Roberson * Restrict the key by the parent. 6113f289c3fSJeff Roberson */ 6123f289c3fSJeff Roberson DOMAINSET_AND(&ndomain->ds_mask, &pdomain->ds_mask); 6133f289c3fSJeff Roberson 6143f289c3fSJeff Roberson return _domainset_create(ndomain, freelist); 6153f289c3fSJeff Roberson } 6163f289c3fSJeff Roberson 617d7f687fcSJeff Roberson /* 618d7f687fcSJeff Roberson * Recursively check for errors that would occur from applying mask to 619d7f687fcSJeff Roberson * the tree of sets starting at 'set'. Checks for sets that would become 620d7f687fcSJeff Roberson * empty as well as RDONLY flags. 621d7f687fcSJeff Roberson */ 622d7f687fcSJeff Roberson static int 623c9813d0aSJohn Baldwin cpuset_testupdate(struct cpuset *set, cpuset_t *mask, int check_mask) 624d7f687fcSJeff Roberson { 625d7f687fcSJeff Roberson struct cpuset *nset; 626d7f687fcSJeff Roberson cpuset_t newmask; 627d7f687fcSJeff Roberson int error; 628d7f687fcSJeff Roberson 629d7f687fcSJeff Roberson mtx_assert(&cpuset_lock, MA_OWNED); 630d7f687fcSJeff Roberson if (set->cs_flags & CPU_SET_RDONLY) 631d7f687fcSJeff Roberson return (EPERM); 632c9813d0aSJohn Baldwin if (check_mask) { 63373c40187SJeff Roberson if (!CPU_OVERLAP(&set->cs_mask, mask)) 63473c40187SJeff Roberson return (EDEADLK); 635d7f687fcSJeff Roberson CPU_COPY(&set->cs_mask, &newmask); 636d7f687fcSJeff Roberson CPU_AND(&newmask, mask); 637c9813d0aSJohn Baldwin } else 638c9813d0aSJohn Baldwin CPU_COPY(mask, &newmask); 63973c40187SJeff Roberson error = 0; 640d7f687fcSJeff Roberson LIST_FOREACH(nset, &set->cs_children, cs_siblings) 641c9813d0aSJohn Baldwin if ((error = cpuset_testupdate(nset, &newmask, 1)) != 0) 642d7f687fcSJeff Roberson break; 643d7f687fcSJeff Roberson return (error); 644d7f687fcSJeff Roberson } 645d7f687fcSJeff Roberson 646d7f687fcSJeff Roberson /* 647d7f687fcSJeff Roberson * Applies the mask 'mask' without checking for empty sets or permissions. 648d7f687fcSJeff Roberson */ 649d7f687fcSJeff Roberson static void 650d7f687fcSJeff Roberson cpuset_update(struct cpuset *set, cpuset_t *mask) 651d7f687fcSJeff Roberson { 652d7f687fcSJeff Roberson struct cpuset *nset; 653d7f687fcSJeff Roberson 654d7f687fcSJeff Roberson mtx_assert(&cpuset_lock, MA_OWNED); 655d7f687fcSJeff Roberson CPU_AND(&set->cs_mask, mask); 656d7f687fcSJeff Roberson LIST_FOREACH(nset, &set->cs_children, cs_siblings) 657d7f687fcSJeff Roberson cpuset_update(nset, &set->cs_mask); 658d7f687fcSJeff Roberson 659d7f687fcSJeff Roberson return; 660d7f687fcSJeff Roberson } 661d7f687fcSJeff Roberson 662d7f687fcSJeff Roberson /* 663d7f687fcSJeff Roberson * Modify the set 'set' to use a copy of the mask provided. Apply this new 664d7f687fcSJeff Roberson * mask to restrict all children in the tree. Checks for validity before 665d7f687fcSJeff Roberson * applying the changes. 666d7f687fcSJeff Roberson */ 667d7f687fcSJeff Roberson static int 668d7f687fcSJeff Roberson cpuset_modify(struct cpuset *set, cpuset_t *mask) 669d7f687fcSJeff Roberson { 67073c40187SJeff Roberson struct cpuset *root; 671d7f687fcSJeff Roberson int error; 672d7f687fcSJeff Roberson 673ba931c08SBjoern A. Zeeb error = priv_check(curthread, PRIV_SCHED_CPUSET); 674d7f687fcSJeff Roberson if (error) 675d7f687fcSJeff Roberson return (error); 67673c40187SJeff Roberson /* 6776aaa0b3cSBjoern A. Zeeb * In case we are called from within the jail 6786aaa0b3cSBjoern A. Zeeb * we do not allow modifying the dedicated root 6796aaa0b3cSBjoern A. Zeeb * cpuset of the jail but may still allow to 6806aaa0b3cSBjoern A. Zeeb * change child sets. 6816aaa0b3cSBjoern A. Zeeb */ 6826aaa0b3cSBjoern A. Zeeb if (jailed(curthread->td_ucred) && 6836aaa0b3cSBjoern A. Zeeb set->cs_flags & CPU_SET_ROOT) 6846aaa0b3cSBjoern A. Zeeb return (EPERM); 6856aaa0b3cSBjoern A. Zeeb /* 68673c40187SJeff Roberson * Verify that we have access to this set of 68773c40187SJeff Roberson * cpus. 68873c40187SJeff Roberson */ 6893f289c3fSJeff Roberson root = cpuset_getroot(set); 690d7f687fcSJeff Roberson mtx_lock_spin(&cpuset_lock); 6913f289c3fSJeff Roberson if (root && !CPU_SUBSET(&root->cs_mask, mask)) { 6923f289c3fSJeff Roberson error = EINVAL; 6933f289c3fSJeff Roberson goto out; 6943f289c3fSJeff Roberson } 695c9813d0aSJohn Baldwin error = cpuset_testupdate(set, mask, 0); 696d7f687fcSJeff Roberson if (error) 697d7f687fcSJeff Roberson goto out; 698d7f687fcSJeff Roberson CPU_COPY(mask, &set->cs_mask); 699c9813d0aSJohn Baldwin cpuset_update(set, mask); 700d7f687fcSJeff Roberson out: 701d7f687fcSJeff Roberson mtx_unlock_spin(&cpuset_lock); 702d7f687fcSJeff Roberson 703d7f687fcSJeff Roberson return (error); 704d7f687fcSJeff Roberson } 705d7f687fcSJeff Roberson 706d7f687fcSJeff Roberson /* 7073f289c3fSJeff Roberson * Recursively check for errors that would occur from applying mask to 7083f289c3fSJeff Roberson * the tree of sets starting at 'set'. Checks for sets that would become 7093f289c3fSJeff Roberson * empty as well as RDONLY flags. 7103f289c3fSJeff Roberson */ 7113f289c3fSJeff Roberson static int 7123f289c3fSJeff Roberson cpuset_testupdate_domain(struct cpuset *set, struct domainset *dset, 7133f289c3fSJeff Roberson struct domainset *orig, int *count, int check_mask) 7143f289c3fSJeff Roberson { 7153f289c3fSJeff Roberson struct cpuset *nset; 7163f289c3fSJeff Roberson struct domainset *domain; 7173f289c3fSJeff Roberson struct domainset newset; 7183f289c3fSJeff Roberson int error; 7193f289c3fSJeff Roberson 7203f289c3fSJeff Roberson mtx_assert(&cpuset_lock, MA_OWNED); 7213f289c3fSJeff Roberson if (set->cs_flags & CPU_SET_RDONLY) 7223f289c3fSJeff Roberson return (EPERM); 7233f289c3fSJeff Roberson domain = set->cs_domain; 7243f289c3fSJeff Roberson domainset_copy(domain, &newset); 7253f289c3fSJeff Roberson if (!domainset_equal(domain, orig)) { 7263f289c3fSJeff Roberson if (!domainset_restrict(domain, dset)) 7273f289c3fSJeff Roberson return (EDEADLK); 7283f289c3fSJeff Roberson DOMAINSET_AND(&newset.ds_mask, &dset->ds_mask); 7293f289c3fSJeff Roberson /* Count the number of domains that are changing. */ 7303f289c3fSJeff Roberson (*count)++; 7313f289c3fSJeff Roberson } 7323f289c3fSJeff Roberson error = 0; 7333f289c3fSJeff Roberson LIST_FOREACH(nset, &set->cs_children, cs_siblings) 7343f289c3fSJeff Roberson if ((error = cpuset_testupdate_domain(nset, &newset, domain, 7353f289c3fSJeff Roberson count, 1)) != 0) 7363f289c3fSJeff Roberson break; 7373f289c3fSJeff Roberson return (error); 7383f289c3fSJeff Roberson } 7393f289c3fSJeff Roberson 7403f289c3fSJeff Roberson /* 7413f289c3fSJeff Roberson * Applies the mask 'mask' without checking for empty sets or permissions. 7423f289c3fSJeff Roberson */ 7433f289c3fSJeff Roberson static void 7443f289c3fSJeff Roberson cpuset_update_domain(struct cpuset *set, struct domainset *domain, 7453f289c3fSJeff Roberson struct domainset *orig, struct domainlist *domains) 7463f289c3fSJeff Roberson { 7473f289c3fSJeff Roberson struct cpuset *nset; 7483f289c3fSJeff Roberson 7493f289c3fSJeff Roberson mtx_assert(&cpuset_lock, MA_OWNED); 7503f289c3fSJeff Roberson /* 7513f289c3fSJeff Roberson * If this domainset has changed from the parent we must calculate 7523f289c3fSJeff Roberson * a new set. Otherwise it simply inherits from the parent. When 7533f289c3fSJeff Roberson * we inherit from the parent we get a new mask and policy. If the 7543f289c3fSJeff Roberson * set is modified from the parent we keep the policy and only 7553f289c3fSJeff Roberson * update the mask. 7563f289c3fSJeff Roberson */ 7573f289c3fSJeff Roberson if (set->cs_domain != orig) { 7583f289c3fSJeff Roberson orig = set->cs_domain; 7593f289c3fSJeff Roberson set->cs_domain = domainset_shadow(domain, orig, domains); 7603f289c3fSJeff Roberson } else 7613f289c3fSJeff Roberson set->cs_domain = domain; 7623f289c3fSJeff Roberson LIST_FOREACH(nset, &set->cs_children, cs_siblings) 7633f289c3fSJeff Roberson cpuset_update_domain(nset, set->cs_domain, orig, domains); 7643f289c3fSJeff Roberson 7653f289c3fSJeff Roberson return; 7663f289c3fSJeff Roberson } 7673f289c3fSJeff Roberson 7683f289c3fSJeff Roberson /* 7693f289c3fSJeff Roberson * Modify the set 'set' to use a copy the domainset provided. Apply this new 7703f289c3fSJeff Roberson * mask to restrict all children in the tree. Checks for validity before 7713f289c3fSJeff Roberson * applying the changes. 7723f289c3fSJeff Roberson */ 7733f289c3fSJeff Roberson static int 7743f289c3fSJeff Roberson cpuset_modify_domain(struct cpuset *set, struct domainset *domain) 7753f289c3fSJeff Roberson { 7763f289c3fSJeff Roberson struct domainlist domains; 7773f289c3fSJeff Roberson struct domainset temp; 7783f289c3fSJeff Roberson struct domainset *dset; 7793f289c3fSJeff Roberson struct cpuset *root; 7803f289c3fSJeff Roberson int ndomains, needed; 7813f289c3fSJeff Roberson int error; 7823f289c3fSJeff Roberson 7833f289c3fSJeff Roberson error = priv_check(curthread, PRIV_SCHED_CPUSET); 7843f289c3fSJeff Roberson if (error) 7853f289c3fSJeff Roberson return (error); 7863f289c3fSJeff Roberson /* 7873f289c3fSJeff Roberson * In case we are called from within the jail 7883f289c3fSJeff Roberson * we do not allow modifying the dedicated root 7893f289c3fSJeff Roberson * cpuset of the jail but may still allow to 7903f289c3fSJeff Roberson * change child sets. 7913f289c3fSJeff Roberson */ 7923f289c3fSJeff Roberson if (jailed(curthread->td_ucred) && 7933f289c3fSJeff Roberson set->cs_flags & CPU_SET_ROOT) 7943f289c3fSJeff Roberson return (EPERM); 7953f289c3fSJeff Roberson domainset_freelist_init(&domains, 0); 7963f289c3fSJeff Roberson domain = domainset_create(domain); 7973f289c3fSJeff Roberson ndomains = needed = 0; 7983f289c3fSJeff Roberson do { 7993f289c3fSJeff Roberson if (ndomains < needed) { 8003f289c3fSJeff Roberson domainset_freelist_add(&domains, needed - ndomains); 8013f289c3fSJeff Roberson ndomains = needed; 8023f289c3fSJeff Roberson } 8033f289c3fSJeff Roberson root = cpuset_getroot(set); 8043f289c3fSJeff Roberson mtx_lock_spin(&cpuset_lock); 8053f289c3fSJeff Roberson dset = root->cs_domain; 8063f289c3fSJeff Roberson /* 8073f289c3fSJeff Roberson * Verify that we have access to this set of domains. 8083f289c3fSJeff Roberson */ 80945cdd437SMark Johnston if (!domainset_valid(dset, domain)) { 8103f289c3fSJeff Roberson error = EINVAL; 8113f289c3fSJeff Roberson goto out; 8123f289c3fSJeff Roberson } 8133f289c3fSJeff Roberson /* 8143f289c3fSJeff Roberson * If applying prefer we keep the current set as the fallback. 8153f289c3fSJeff Roberson */ 8163f289c3fSJeff Roberson if (domain->ds_policy == DOMAINSET_POLICY_PREFER) 8173f289c3fSJeff Roberson DOMAINSET_COPY(&set->cs_domain->ds_mask, 8183f289c3fSJeff Roberson &domain->ds_mask); 8193f289c3fSJeff Roberson /* 8203f289c3fSJeff Roberson * Determine whether we can apply this set of domains and 8213f289c3fSJeff Roberson * how many new domain structures it will require. 8223f289c3fSJeff Roberson */ 8233f289c3fSJeff Roberson domainset_copy(domain, &temp); 8243f289c3fSJeff Roberson needed = 0; 8253f289c3fSJeff Roberson error = cpuset_testupdate_domain(set, &temp, set->cs_domain, 8263f289c3fSJeff Roberson &needed, 0); 8273f289c3fSJeff Roberson if (error) 8283f289c3fSJeff Roberson goto out; 8293f289c3fSJeff Roberson } while (ndomains < needed); 8303f289c3fSJeff Roberson dset = set->cs_domain; 8313f289c3fSJeff Roberson cpuset_update_domain(set, domain, dset, &domains); 8323f289c3fSJeff Roberson out: 8333f289c3fSJeff Roberson mtx_unlock_spin(&cpuset_lock); 8343f289c3fSJeff Roberson domainset_freelist_free(&domains); 8353f289c3fSJeff Roberson if (error == 0) 8363f289c3fSJeff Roberson domainset_notify(); 8373f289c3fSJeff Roberson 8383f289c3fSJeff Roberson return (error); 8393f289c3fSJeff Roberson } 8403f289c3fSJeff Roberson 8413f289c3fSJeff Roberson /* 842d7f687fcSJeff Roberson * Resolve the 'which' parameter of several cpuset apis. 843d7f687fcSJeff Roberson * 844d7f687fcSJeff Roberson * For WHICH_PID and WHICH_TID return a locked proc and valid proc/tid. Also 845d7f687fcSJeff Roberson * checks for permission via p_cansched(). 846d7f687fcSJeff Roberson * 847d7f687fcSJeff Roberson * For WHICH_SET returns a valid set with a new reference. 848d7f687fcSJeff Roberson * 849d7f687fcSJeff Roberson * -1 may be supplied for any argument to mean the current proc/thread or 850d7f687fcSJeff Roberson * the base set of the current thread. May fail with ESRCH/EPERM. 851d7f687fcSJeff Roberson */ 8525bbb2169SAdrian Chadd int 853d7f687fcSJeff Roberson cpuset_which(cpuwhich_t which, id_t id, struct proc **pp, struct thread **tdp, 854d7f687fcSJeff Roberson struct cpuset **setp) 855d7f687fcSJeff Roberson { 856d7f687fcSJeff Roberson struct cpuset *set; 857d7f687fcSJeff Roberson struct thread *td; 858d7f687fcSJeff Roberson struct proc *p; 859d7f687fcSJeff Roberson int error; 860d7f687fcSJeff Roberson 861d7f687fcSJeff Roberson *pp = p = NULL; 862d7f687fcSJeff Roberson *tdp = td = NULL; 863d7f687fcSJeff Roberson *setp = set = NULL; 864d7f687fcSJeff Roberson switch (which) { 865d7f687fcSJeff Roberson case CPU_WHICH_PID: 866d7f687fcSJeff Roberson if (id == -1) { 867d7f687fcSJeff Roberson PROC_LOCK(curproc); 868d7f687fcSJeff Roberson p = curproc; 869d7f687fcSJeff Roberson break; 870d7f687fcSJeff Roberson } 871d7f687fcSJeff Roberson if ((p = pfind(id)) == NULL) 872d7f687fcSJeff Roberson return (ESRCH); 873d7f687fcSJeff Roberson break; 874d7f687fcSJeff Roberson case CPU_WHICH_TID: 875d7f687fcSJeff Roberson if (id == -1) { 876d7f687fcSJeff Roberson PROC_LOCK(curproc); 877d7f687fcSJeff Roberson p = curproc; 878d7f687fcSJeff Roberson td = curthread; 879d7f687fcSJeff Roberson break; 880d7f687fcSJeff Roberson } 88142fe684cSDavid Xu td = tdfind(id, -1); 882d7f687fcSJeff Roberson if (td == NULL) 883d7f687fcSJeff Roberson return (ESRCH); 88442fe684cSDavid Xu p = td->td_proc; 885d7f687fcSJeff Roberson break; 886d7f687fcSJeff Roberson case CPU_WHICH_CPUSET: 887d7f687fcSJeff Roberson if (id == -1) { 888d7f687fcSJeff Roberson thread_lock(curthread); 889a03ee000SJeff Roberson set = cpuset_refbase(curthread->td_cpuset); 890d7f687fcSJeff Roberson thread_unlock(curthread); 891d7f687fcSJeff Roberson } else 892413628a7SBjoern A. Zeeb set = cpuset_lookup(id, curthread); 893d7f687fcSJeff Roberson if (set) { 894d7f687fcSJeff Roberson *setp = set; 895d7f687fcSJeff Roberson return (0); 896d7f687fcSJeff Roberson } 897d7f687fcSJeff Roberson return (ESRCH); 898413628a7SBjoern A. Zeeb case CPU_WHICH_JAIL: 899413628a7SBjoern A. Zeeb { 900413628a7SBjoern A. Zeeb /* Find `set' for prison with given id. */ 901413628a7SBjoern A. Zeeb struct prison *pr; 902413628a7SBjoern A. Zeeb 903413628a7SBjoern A. Zeeb sx_slock(&allprison_lock); 9040304c731SJamie Gritton pr = prison_find_child(curthread->td_ucred->cr_prison, id); 905413628a7SBjoern A. Zeeb sx_sunlock(&allprison_lock); 906413628a7SBjoern A. Zeeb if (pr == NULL) 907413628a7SBjoern A. Zeeb return (ESRCH); 908413628a7SBjoern A. Zeeb cpuset_ref(pr->pr_cpuset); 9090304c731SJamie Gritton *setp = pr->pr_cpuset; 910413628a7SBjoern A. Zeeb mtx_unlock(&pr->pr_mtx); 911413628a7SBjoern A. Zeeb return (0); 912413628a7SBjoern A. Zeeb } 9139b33b154SJeff Roberson case CPU_WHICH_IRQ: 914c0ae6688SJohn Baldwin case CPU_WHICH_DOMAIN: 9159b33b154SJeff Roberson return (0); 916d7f687fcSJeff Roberson default: 917d7f687fcSJeff Roberson return (EINVAL); 918d7f687fcSJeff Roberson } 919d7f687fcSJeff Roberson error = p_cansched(curthread, p); 920d7f687fcSJeff Roberson if (error) { 921d7f687fcSJeff Roberson PROC_UNLOCK(p); 922d7f687fcSJeff Roberson return (error); 923d7f687fcSJeff Roberson } 924d7f687fcSJeff Roberson if (td == NULL) 925d7f687fcSJeff Roberson td = FIRST_THREAD_IN_PROC(p); 926d7f687fcSJeff Roberson *pp = p; 927d7f687fcSJeff Roberson *tdp = td; 928d7f687fcSJeff Roberson return (0); 929d7f687fcSJeff Roberson } 930d7f687fcSJeff Roberson 931d7f687fcSJeff Roberson static int 9323f289c3fSJeff Roberson cpuset_testshadow(struct cpuset *set, const cpuset_t *mask, 9333f289c3fSJeff Roberson const struct domainset *domain) 934d7f687fcSJeff Roberson { 935d7f687fcSJeff Roberson struct cpuset *parent; 9363f289c3fSJeff Roberson struct domainset *dset; 937d7f687fcSJeff Roberson 9383f289c3fSJeff Roberson parent = cpuset_getbase(set); 9393f289c3fSJeff Roberson /* 9403f289c3fSJeff Roberson * If we are restricting a cpu mask it must be a subset of the 9413f289c3fSJeff Roberson * parent or invalid CPUs have been specified. 9423f289c3fSJeff Roberson */ 9433f289c3fSJeff Roberson if (mask != NULL && !CPU_SUBSET(&parent->cs_mask, mask)) 9443f289c3fSJeff Roberson return (EINVAL); 9453f289c3fSJeff Roberson 9463f289c3fSJeff Roberson /* 9473f289c3fSJeff Roberson * If we are restricting a domain mask it must be a subset of the 9483f289c3fSJeff Roberson * parent or invalid domains have been specified. 9493f289c3fSJeff Roberson */ 9503f289c3fSJeff Roberson dset = parent->cs_domain; 9513f289c3fSJeff Roberson if (domain != NULL && !domainset_valid(dset, domain)) 9523f289c3fSJeff Roberson return (EINVAL); 9533f289c3fSJeff Roberson 9543f289c3fSJeff Roberson return (0); 955d7f687fcSJeff Roberson } 956d7f687fcSJeff Roberson 957d7f687fcSJeff Roberson /* 9583f289c3fSJeff Roberson * Create an anonymous set with the provided mask in the space provided by 9593f289c3fSJeff Roberson * 'nset'. If the passed in set is anonymous we use its parent otherwise 9603f289c3fSJeff Roberson * the new set is a child of 'set'. 9613f289c3fSJeff Roberson */ 9623f289c3fSJeff Roberson static int 9633f289c3fSJeff Roberson cpuset_shadow(struct cpuset *set, struct cpuset **nsetp, 9643f289c3fSJeff Roberson const cpuset_t *mask, const struct domainset *domain, 9653f289c3fSJeff Roberson struct setlist *cpusets, struct domainlist *domains) 9663f289c3fSJeff Roberson { 9673f289c3fSJeff Roberson struct cpuset *parent; 9683f289c3fSJeff Roberson struct cpuset *nset; 9693f289c3fSJeff Roberson struct domainset *dset; 9703f289c3fSJeff Roberson struct domainset *d; 9713f289c3fSJeff Roberson int error; 9723f289c3fSJeff Roberson 9733f289c3fSJeff Roberson error = cpuset_testshadow(set, mask, domain); 9743f289c3fSJeff Roberson if (error) 9753f289c3fSJeff Roberson return (error); 9763f289c3fSJeff Roberson 9773f289c3fSJeff Roberson parent = cpuset_getbase(set); 9783f289c3fSJeff Roberson dset = parent->cs_domain; 9793f289c3fSJeff Roberson if (mask == NULL) 9803f289c3fSJeff Roberson mask = &set->cs_mask; 9813f289c3fSJeff Roberson if (domain != NULL) 9823f289c3fSJeff Roberson d = domainset_shadow(dset, domain, domains); 9833f289c3fSJeff Roberson else 9843f289c3fSJeff Roberson d = set->cs_domain; 9853f289c3fSJeff Roberson nset = LIST_FIRST(cpusets); 98630b7c6f9SKyle Evans error = cpuset_init(nset, parent, mask, d, CPUSET_INVALID); 9873f289c3fSJeff Roberson if (error == 0) { 9883f289c3fSJeff Roberson LIST_REMOVE(nset, cs_link); 9893f289c3fSJeff Roberson *nsetp = nset; 9903f289c3fSJeff Roberson } 9913f289c3fSJeff Roberson return (error); 9923f289c3fSJeff Roberson } 9933f289c3fSJeff Roberson 9943f289c3fSJeff Roberson static struct cpuset * 9953f289c3fSJeff Roberson cpuset_update_thread(struct thread *td, struct cpuset *nset) 9963f289c3fSJeff Roberson { 9973f289c3fSJeff Roberson struct cpuset *tdset; 9983f289c3fSJeff Roberson 9993f289c3fSJeff Roberson tdset = td->td_cpuset; 10003f289c3fSJeff Roberson td->td_cpuset = nset; 10013f289c3fSJeff Roberson td->td_domain.dr_policy = nset->cs_domain; 10023f289c3fSJeff Roberson sched_affinity(td); 10033f289c3fSJeff Roberson 10043f289c3fSJeff Roberson return (tdset); 10053f289c3fSJeff Roberson } 10063f289c3fSJeff Roberson 10073f289c3fSJeff Roberson static int 10083f289c3fSJeff Roberson cpuset_setproc_test_maskthread(struct cpuset *tdset, cpuset_t *mask, 10093f289c3fSJeff Roberson struct domainset *domain) 10103f289c3fSJeff Roberson { 10113f289c3fSJeff Roberson struct cpuset *parent; 10123f289c3fSJeff Roberson 10133f289c3fSJeff Roberson parent = cpuset_getbase(tdset); 10143f289c3fSJeff Roberson if (mask == NULL) 10153f289c3fSJeff Roberson mask = &tdset->cs_mask; 10163f289c3fSJeff Roberson if (domain == NULL) 10173f289c3fSJeff Roberson domain = tdset->cs_domain; 10183f289c3fSJeff Roberson return cpuset_testshadow(parent, mask, domain); 10193f289c3fSJeff Roberson } 10203f289c3fSJeff Roberson 10213f289c3fSJeff Roberson static int 10223f289c3fSJeff Roberson cpuset_setproc_maskthread(struct cpuset *tdset, cpuset_t *mask, 10233f289c3fSJeff Roberson struct domainset *domain, struct cpuset **nsetp, 10243f289c3fSJeff Roberson struct setlist *freelist, struct domainlist *domainlist) 10253f289c3fSJeff Roberson { 10263f289c3fSJeff Roberson struct cpuset *parent; 10273f289c3fSJeff Roberson 10283f289c3fSJeff Roberson parent = cpuset_getbase(tdset); 10293f289c3fSJeff Roberson if (mask == NULL) 10303f289c3fSJeff Roberson mask = &tdset->cs_mask; 10313f289c3fSJeff Roberson if (domain == NULL) 10323f289c3fSJeff Roberson domain = tdset->cs_domain; 10333f289c3fSJeff Roberson return cpuset_shadow(parent, nsetp, mask, domain, freelist, 10343f289c3fSJeff Roberson domainlist); 10353f289c3fSJeff Roberson } 10363f289c3fSJeff Roberson 10373f289c3fSJeff Roberson static int 10383f289c3fSJeff Roberson cpuset_setproc_setthread_mask(struct cpuset *tdset, struct cpuset *set, 10393f289c3fSJeff Roberson cpuset_t *mask, struct domainset *domain) 10403f289c3fSJeff Roberson { 10413f289c3fSJeff Roberson struct cpuset *parent; 10423f289c3fSJeff Roberson 10433f289c3fSJeff Roberson parent = cpuset_getbase(tdset); 10443f289c3fSJeff Roberson 10453f289c3fSJeff Roberson /* 10463f289c3fSJeff Roberson * If the thread restricted its mask then apply that same 10473f289c3fSJeff Roberson * restriction to the new set, otherwise take it wholesale. 10483f289c3fSJeff Roberson */ 10493f289c3fSJeff Roberson if (CPU_CMP(&tdset->cs_mask, &parent->cs_mask) != 0) { 10503f289c3fSJeff Roberson CPU_COPY(&tdset->cs_mask, mask); 10513f289c3fSJeff Roberson CPU_AND(mask, &set->cs_mask); 10523f289c3fSJeff Roberson } else 10533f289c3fSJeff Roberson CPU_COPY(&set->cs_mask, mask); 10543f289c3fSJeff Roberson 10553f289c3fSJeff Roberson /* 10563f289c3fSJeff Roberson * If the thread restricted the domain then we apply the 10573f289c3fSJeff Roberson * restriction to the new set but retain the policy. 10583f289c3fSJeff Roberson */ 10593f289c3fSJeff Roberson if (tdset->cs_domain != parent->cs_domain) { 10603f289c3fSJeff Roberson domainset_copy(tdset->cs_domain, domain); 10613f289c3fSJeff Roberson DOMAINSET_AND(&domain->ds_mask, &set->cs_domain->ds_mask); 10623f289c3fSJeff Roberson } else 10633f289c3fSJeff Roberson domainset_copy(set->cs_domain, domain); 10643f289c3fSJeff Roberson 10653f289c3fSJeff Roberson if (CPU_EMPTY(mask) || DOMAINSET_EMPTY(&domain->ds_mask)) 10663f289c3fSJeff Roberson return (EDEADLK); 10673f289c3fSJeff Roberson 10683f289c3fSJeff Roberson return (0); 10693f289c3fSJeff Roberson } 10703f289c3fSJeff Roberson 10713f289c3fSJeff Roberson static int 10723f289c3fSJeff Roberson cpuset_setproc_test_setthread(struct cpuset *tdset, struct cpuset *set) 10733f289c3fSJeff Roberson { 10743f289c3fSJeff Roberson struct domainset domain; 10753f289c3fSJeff Roberson cpuset_t mask; 10763f289c3fSJeff Roberson 10773f289c3fSJeff Roberson if (tdset->cs_id != CPUSET_INVALID) 10783f289c3fSJeff Roberson return (0); 10793f289c3fSJeff Roberson return cpuset_setproc_setthread_mask(tdset, set, &mask, &domain); 10803f289c3fSJeff Roberson } 10813f289c3fSJeff Roberson 10823f289c3fSJeff Roberson static int 10833f289c3fSJeff Roberson cpuset_setproc_setthread(struct cpuset *tdset, struct cpuset *set, 10843f289c3fSJeff Roberson struct cpuset **nsetp, struct setlist *freelist, 10853f289c3fSJeff Roberson struct domainlist *domainlist) 10863f289c3fSJeff Roberson { 10873f289c3fSJeff Roberson struct domainset domain; 10883f289c3fSJeff Roberson cpuset_t mask; 10893f289c3fSJeff Roberson int error; 10903f289c3fSJeff Roberson 10913f289c3fSJeff Roberson /* 10923f289c3fSJeff Roberson * If we're replacing on a thread that has not constrained the 10933f289c3fSJeff Roberson * original set we can simply accept the new set. 10943f289c3fSJeff Roberson */ 10953f289c3fSJeff Roberson if (tdset->cs_id != CPUSET_INVALID) { 10963f289c3fSJeff Roberson *nsetp = cpuset_ref(set); 10973f289c3fSJeff Roberson return (0); 10983f289c3fSJeff Roberson } 10993f289c3fSJeff Roberson error = cpuset_setproc_setthread_mask(tdset, set, &mask, &domain); 11003f289c3fSJeff Roberson if (error) 11013f289c3fSJeff Roberson return (error); 11023f289c3fSJeff Roberson 1103dac521ebSKyle Evans return cpuset_shadow(set, nsetp, &mask, &domain, freelist, 11043f289c3fSJeff Roberson domainlist); 11053f289c3fSJeff Roberson } 11063f289c3fSJeff Roberson 1107*d431dea5SKyle Evans static int 1108*d431dea5SKyle Evans cpuset_setproc_newbase(struct thread *td, struct cpuset *set, 1109*d431dea5SKyle Evans struct cpuset *nroot, struct cpuset **nsetp, 1110*d431dea5SKyle Evans struct setlist *cpusets, struct domainlist *domainlist) 1111*d431dea5SKyle Evans { 1112*d431dea5SKyle Evans struct domainset ndomain; 1113*d431dea5SKyle Evans cpuset_t nmask; 1114*d431dea5SKyle Evans struct cpuset *pbase; 1115*d431dea5SKyle Evans int error; 1116*d431dea5SKyle Evans 1117*d431dea5SKyle Evans pbase = cpuset_getbase(td->td_cpuset); 1118*d431dea5SKyle Evans 1119*d431dea5SKyle Evans /* Copy process mask, then further apply the new root mask. */ 1120*d431dea5SKyle Evans CPU_COPY(&pbase->cs_mask, &nmask); 1121*d431dea5SKyle Evans CPU_AND(&nmask, &nroot->cs_mask); 1122*d431dea5SKyle Evans 1123*d431dea5SKyle Evans domainset_copy(pbase->cs_domain, &ndomain); 1124*d431dea5SKyle Evans DOMAINSET_AND(&ndomain.ds_mask, &set->cs_domain->ds_mask); 1125*d431dea5SKyle Evans 1126*d431dea5SKyle Evans /* Policy is too restrictive, will not work. */ 1127*d431dea5SKyle Evans if (CPU_EMPTY(&nmask) || DOMAINSET_EMPTY(&ndomain.ds_mask)) 1128*d431dea5SKyle Evans return (EDEADLK); 1129*d431dea5SKyle Evans 11303f289c3fSJeff Roberson /* 1131*d431dea5SKyle Evans * Remove pbase from the freelist in advance, it'll be pushed to 1132*d431dea5SKyle Evans * cpuset_ids on success. We assume here that cpuset_create() will not 1133*d431dea5SKyle Evans * touch pbase on failure, and we just enqueue it back to the freelist 1134*d431dea5SKyle Evans * to remain in a consistent state. 1135*d431dea5SKyle Evans */ 1136*d431dea5SKyle Evans pbase = LIST_FIRST(cpusets); 1137*d431dea5SKyle Evans LIST_REMOVE(pbase, cs_link); 1138*d431dea5SKyle Evans error = cpuset_create(&pbase, set, &nmask); 1139*d431dea5SKyle Evans if (error != 0) { 1140*d431dea5SKyle Evans LIST_INSERT_HEAD(cpusets, pbase, cs_link); 1141*d431dea5SKyle Evans return (error); 1142*d431dea5SKyle Evans } 1143*d431dea5SKyle Evans 1144*d431dea5SKyle Evans /* Duplicates some work from above... oh well. */ 1145*d431dea5SKyle Evans pbase->cs_domain = domainset_shadow(set->cs_domain, &ndomain, 1146*d431dea5SKyle Evans domainlist); 1147*d431dea5SKyle Evans *nsetp = pbase; 1148*d431dea5SKyle Evans return (0); 1149*d431dea5SKyle Evans } 1150*d431dea5SKyle Evans 1151*d431dea5SKyle Evans /* 1152*d431dea5SKyle Evans * Handle four cases for updating an entire process. 1153d7f687fcSJeff Roberson * 1154*d431dea5SKyle Evans * 1) Set is non-null and the process is not rebasing onto a new root. This 1155*d431dea5SKyle Evans * reparents all anonymous sets to the provided set and replaces all 1156*d431dea5SKyle Evans * non-anonymous td_cpusets with the provided set. 1157*d431dea5SKyle Evans * 2) Set is non-null and the process is rebasing onto a new root. This 1158*d431dea5SKyle Evans * creates a new base set if the process previously had its own base set, 1159*d431dea5SKyle Evans * then reparents all anonymous sets either to that set or the provided set 1160*d431dea5SKyle Evans * if one was not created. Non-anonymous sets are similarly replaced. 1161*d431dea5SKyle Evans * 3) Mask is non-null. This replaces or creates anonymous sets for every 11623f289c3fSJeff Roberson * thread with the existing base as a parent. 1163*d431dea5SKyle Evans * 4) domain is non-null. This creates anonymous sets for every thread 11643f289c3fSJeff Roberson * and replaces the domain set. 1165d7f687fcSJeff Roberson * 1166d7f687fcSJeff Roberson * This is overly complicated because we can't allocate while holding a 1167d7f687fcSJeff Roberson * spinlock and spinlocks must be held while changing and examining thread 1168d7f687fcSJeff Roberson * state. 1169d7f687fcSJeff Roberson */ 1170d7f687fcSJeff Roberson static int 11713f289c3fSJeff Roberson cpuset_setproc(pid_t pid, struct cpuset *set, cpuset_t *mask, 1172*d431dea5SKyle Evans struct domainset *domain, bool rebase) 1173d7f687fcSJeff Roberson { 1174d7f687fcSJeff Roberson struct setlist freelist; 1175d7f687fcSJeff Roberson struct setlist droplist; 11763f289c3fSJeff Roberson struct domainlist domainlist; 1177*d431dea5SKyle Evans struct cpuset *base, *nset, *nroot, *tdroot; 1178d7f687fcSJeff Roberson struct thread *td; 1179d7f687fcSJeff Roberson struct proc *p; 1180*d431dea5SKyle Evans int needed; 1181d7f687fcSJeff Roberson int nfree; 1182d7f687fcSJeff Roberson int error; 1183f299c47bSAllan Jude 1184d7f687fcSJeff Roberson /* 1185d7f687fcSJeff Roberson * The algorithm requires two passes due to locking considerations. 1186d7f687fcSJeff Roberson * 1187d7f687fcSJeff Roberson * 1) Lookup the process and acquire the locks in the required order. 1188d7f687fcSJeff Roberson * 2) If enough cpusets have not been allocated release the locks and 1189d7f687fcSJeff Roberson * allocate them. Loop. 1190d7f687fcSJeff Roberson */ 11913f289c3fSJeff Roberson cpuset_freelist_init(&freelist, 1); 11923f289c3fSJeff Roberson domainset_freelist_init(&domainlist, 1); 11933f289c3fSJeff Roberson nfree = 1; 1194d7f687fcSJeff Roberson LIST_INIT(&droplist); 1195d7f687fcSJeff Roberson nfree = 0; 1196*d431dea5SKyle Evans base = set; 1197*d431dea5SKyle Evans nroot = NULL; 1198*d431dea5SKyle Evans if (set != NULL) 1199*d431dea5SKyle Evans nroot = cpuset_getroot(set); 1200d7f687fcSJeff Roberson for (;;) { 1201d7f687fcSJeff Roberson error = cpuset_which(CPU_WHICH_PID, pid, &p, &td, &nset); 1202d7f687fcSJeff Roberson if (error) 1203d7f687fcSJeff Roberson goto out; 1204*d431dea5SKyle Evans tdroot = cpuset_getroot(td->td_cpuset); 1205*d431dea5SKyle Evans needed = p->p_numthreads; 1206*d431dea5SKyle Evans if (set != NULL && rebase && tdroot != nroot) 1207*d431dea5SKyle Evans needed++; 1208*d431dea5SKyle Evans if (nfree >= needed) 1209d7f687fcSJeff Roberson break; 1210d7f687fcSJeff Roberson PROC_UNLOCK(p); 1211*d431dea5SKyle Evans if (nfree < needed) { 1212*d431dea5SKyle Evans cpuset_freelist_add(&freelist, needed - nfree); 1213*d431dea5SKyle Evans domainset_freelist_add(&domainlist, needed - nfree); 1214*d431dea5SKyle Evans nfree = needed; 1215d7f687fcSJeff Roberson } 1216d7f687fcSJeff Roberson } 1217d7f687fcSJeff Roberson PROC_LOCK_ASSERT(p, MA_OWNED); 1218*d431dea5SKyle Evans 1219*d431dea5SKyle Evans /* 1220*d431dea5SKyle Evans * If we're changing roots and the root set is what has been specified 1221*d431dea5SKyle Evans * as the parent, then we'll check if the process was previously using 1222*d431dea5SKyle Evans * the root set and, if it wasn't, create a new base with the process's 1223*d431dea5SKyle Evans * mask applied to it. 1224*d431dea5SKyle Evans */ 1225*d431dea5SKyle Evans if (set != NULL && rebase && nroot != tdroot) { 1226*d431dea5SKyle Evans cpusetid_t base_id, root_id; 1227*d431dea5SKyle Evans 1228*d431dea5SKyle Evans root_id = td->td_ucred->cr_prison->pr_cpuset->cs_id; 1229*d431dea5SKyle Evans base_id = cpuset_getbase(td->td_cpuset)->cs_id; 1230*d431dea5SKyle Evans 1231*d431dea5SKyle Evans if (base_id != root_id) { 1232*d431dea5SKyle Evans error = cpuset_setproc_newbase(td, set, nroot, &base, 1233*d431dea5SKyle Evans &freelist, &domainlist); 1234*d431dea5SKyle Evans if (error != 0) 1235*d431dea5SKyle Evans goto unlock_out; 1236*d431dea5SKyle Evans } 1237*d431dea5SKyle Evans } 1238*d431dea5SKyle Evans 1239d7f687fcSJeff Roberson /* 1240d7f687fcSJeff Roberson * Now that the appropriate locks are held and we have enough cpusets, 124173c40187SJeff Roberson * make sure the operation will succeed before applying changes. The 124273c40187SJeff Roberson * proc lock prevents td_cpuset from changing between calls. 1243d7f687fcSJeff Roberson */ 1244d7f687fcSJeff Roberson error = 0; 1245d7f687fcSJeff Roberson FOREACH_THREAD_IN_PROC(p, td) { 124673c40187SJeff Roberson thread_lock(td); 12473f289c3fSJeff Roberson if (set != NULL) 12483f289c3fSJeff Roberson error = cpuset_setproc_test_setthread(td->td_cpuset, 1249*d431dea5SKyle Evans base); 12503f289c3fSJeff Roberson else 12513f289c3fSJeff Roberson error = cpuset_setproc_test_maskthread(td->td_cpuset, 12523f289c3fSJeff Roberson mask, domain); 125373c40187SJeff Roberson thread_unlock(td); 125473c40187SJeff Roberson if (error) 125573c40187SJeff Roberson goto unlock_out; 125673c40187SJeff Roberson } 125773c40187SJeff Roberson /* 125873c40187SJeff Roberson * Replace each thread's cpuset while using deferred release. We 1259374ae2a3SJeff Roberson * must do this because the thread lock must be held while operating 1260374ae2a3SJeff Roberson * on the thread and this limits the type of operations allowed. 126173c40187SJeff Roberson */ 126273c40187SJeff Roberson FOREACH_THREAD_IN_PROC(p, td) { 1263d7f687fcSJeff Roberson thread_lock(td); 12643f289c3fSJeff Roberson if (set != NULL) 1265*d431dea5SKyle Evans error = cpuset_setproc_setthread(td->td_cpuset, base, 12663f289c3fSJeff Roberson &nset, &freelist, &domainlist); 1267d7f687fcSJeff Roberson else 12683f289c3fSJeff Roberson error = cpuset_setproc_maskthread(td->td_cpuset, mask, 12693f289c3fSJeff Roberson domain, &nset, &freelist, &domainlist); 1270d7f687fcSJeff Roberson if (error) { 1271d7f687fcSJeff Roberson thread_unlock(td); 1272d7f687fcSJeff Roberson break; 1273d7f687fcSJeff Roberson } 12743f289c3fSJeff Roberson cpuset_rel_defer(&droplist, cpuset_update_thread(td, nset)); 1275d7f687fcSJeff Roberson thread_unlock(td); 1276d7f687fcSJeff Roberson } 127773c40187SJeff Roberson unlock_out: 1278d7f687fcSJeff Roberson PROC_UNLOCK(p); 1279d7f687fcSJeff Roberson out: 1280*d431dea5SKyle Evans if (base != NULL && base != set) 1281*d431dea5SKyle Evans cpuset_rel(base); 1282d7f687fcSJeff Roberson while ((nset = LIST_FIRST(&droplist)) != NULL) 1283d7f687fcSJeff Roberson cpuset_rel_complete(nset); 12843f289c3fSJeff Roberson cpuset_freelist_free(&freelist); 12853f289c3fSJeff Roberson domainset_freelist_free(&domainlist); 1286d7f687fcSJeff Roberson return (error); 1287d7f687fcSJeff Roberson } 1288d7f687fcSJeff Roberson 1289e5818a53SJeff Roberson static int 1290e5818a53SJeff Roberson bitset_strprint(char *buf, size_t bufsiz, const struct bitset *set, int setlen) 1291e5818a53SJeff Roberson { 1292e5818a53SJeff Roberson size_t bytes; 1293e5818a53SJeff Roberson int i, once; 1294e5818a53SJeff Roberson char *p; 1295e5818a53SJeff Roberson 1296e5818a53SJeff Roberson once = 0; 1297e5818a53SJeff Roberson p = buf; 1298e5818a53SJeff Roberson for (i = 0; i < __bitset_words(setlen); i++) { 1299e5818a53SJeff Roberson if (once != 0) { 1300e5818a53SJeff Roberson if (bufsiz < 1) 1301e5818a53SJeff Roberson return (0); 1302e5818a53SJeff Roberson *p = ','; 1303e5818a53SJeff Roberson p++; 1304e5818a53SJeff Roberson bufsiz--; 1305e5818a53SJeff Roberson } else 1306e5818a53SJeff Roberson once = 1; 1307e5818a53SJeff Roberson if (bufsiz < sizeof(__STRING(ULONG_MAX))) 1308e5818a53SJeff Roberson return (0); 1309e5818a53SJeff Roberson bytes = snprintf(p, bufsiz, "%lx", set->__bits[i]); 1310e5818a53SJeff Roberson p += bytes; 1311e5818a53SJeff Roberson bufsiz -= bytes; 1312e5818a53SJeff Roberson } 1313e5818a53SJeff Roberson return (p - buf); 1314e5818a53SJeff Roberson } 1315e5818a53SJeff Roberson 1316e5818a53SJeff Roberson static int 1317e5818a53SJeff Roberson bitset_strscan(struct bitset *set, int setlen, const char *buf) 1318e5818a53SJeff Roberson { 1319e5818a53SJeff Roberson int i, ret; 1320e5818a53SJeff Roberson const char *p; 1321e5818a53SJeff Roberson 1322e5818a53SJeff Roberson BIT_ZERO(setlen, set); 1323e5818a53SJeff Roberson p = buf; 1324e5818a53SJeff Roberson for (i = 0; i < __bitset_words(setlen); i++) { 1325e5818a53SJeff Roberson if (*p == ',') { 1326e5818a53SJeff Roberson p++; 1327e5818a53SJeff Roberson continue; 1328e5818a53SJeff Roberson } 1329e5818a53SJeff Roberson ret = sscanf(p, "%lx", &set->__bits[i]); 1330e5818a53SJeff Roberson if (ret == 0 || ret == -1) 1331e5818a53SJeff Roberson break; 1332e5818a53SJeff Roberson while (isxdigit(*p)) 1333e5818a53SJeff Roberson p++; 1334e5818a53SJeff Roberson } 1335e5818a53SJeff Roberson return (p - buf); 1336e5818a53SJeff Roberson } 1337e5818a53SJeff Roberson 1338d7f687fcSJeff Roberson /* 133971a19bdcSAttilio Rao * Return a string representing a valid layout for a cpuset_t object. 134071a19bdcSAttilio Rao * It expects an incoming buffer at least sized as CPUSETBUFSIZ. 134171a19bdcSAttilio Rao */ 134271a19bdcSAttilio Rao char * 134371a19bdcSAttilio Rao cpusetobj_strprint(char *buf, const cpuset_t *set) 134471a19bdcSAttilio Rao { 134571a19bdcSAttilio Rao 1346e5818a53SJeff Roberson bitset_strprint(buf, CPUSETBUFSIZ, (const struct bitset *)set, 1347e5818a53SJeff Roberson CPU_SETSIZE); 134871a19bdcSAttilio Rao return (buf); 134971a19bdcSAttilio Rao } 135071a19bdcSAttilio Rao 135171a19bdcSAttilio Rao /* 1352e3709597SAttilio Rao * Build a valid cpuset_t object from a string representation. 1353e3709597SAttilio Rao * It expects an incoming buffer at least sized as CPUSETBUFSIZ. 1354e3709597SAttilio Rao */ 1355e3709597SAttilio Rao int 1356e3709597SAttilio Rao cpusetobj_strscan(cpuset_t *set, const char *buf) 1357e3709597SAttilio Rao { 1358e5818a53SJeff Roberson char p; 1359e3709597SAttilio Rao 1360e3709597SAttilio Rao if (strlen(buf) > CPUSETBUFSIZ - 1) 1361e3709597SAttilio Rao return (-1); 1362e3709597SAttilio Rao 1363e5818a53SJeff Roberson p = buf[bitset_strscan((struct bitset *)set, CPU_SETSIZE, buf)]; 1364e5818a53SJeff Roberson if (p != '\0') 1365e3709597SAttilio Rao return (-1); 1366e3709597SAttilio Rao 1367e3709597SAttilio Rao return (0); 1368e3709597SAttilio Rao } 1369e3709597SAttilio Rao 1370e3709597SAttilio Rao /* 1371e5818a53SJeff Roberson * Handle a domainset specifier in the sysctl tree. A poiner to a pointer to 1372e5818a53SJeff Roberson * a domainset is in arg1. If the user specifies a valid domainset the 1373e5818a53SJeff Roberson * pointer is updated. 1374e5818a53SJeff Roberson * 1375e5818a53SJeff Roberson * Format is: 1376e5818a53SJeff Roberson * hex mask word 0,hex mask word 1,...:decimal policy:decimal preferred 1377e5818a53SJeff Roberson */ 1378e5818a53SJeff Roberson int 1379e5818a53SJeff Roberson sysctl_handle_domainset(SYSCTL_HANDLER_ARGS) 1380e5818a53SJeff Roberson { 1381e5818a53SJeff Roberson char buf[DOMAINSETBUFSIZ]; 1382e5818a53SJeff Roberson struct domainset *dset; 1383e5818a53SJeff Roberson struct domainset key; 1384e5818a53SJeff Roberson int policy, prefer, error; 1385e5818a53SJeff Roberson char *p; 1386e5818a53SJeff Roberson 1387e5818a53SJeff Roberson dset = *(struct domainset **)arg1; 1388e5818a53SJeff Roberson error = 0; 1389e5818a53SJeff Roberson 1390e5818a53SJeff Roberson if (dset != NULL) { 1391e5818a53SJeff Roberson p = buf + bitset_strprint(buf, DOMAINSETBUFSIZ, 1392e5818a53SJeff Roberson (const struct bitset *)&dset->ds_mask, DOMAINSET_SETSIZE); 1393e5818a53SJeff Roberson sprintf(p, ":%d:%d", dset->ds_policy, dset->ds_prefer); 1394e5818a53SJeff Roberson } else 1395e5818a53SJeff Roberson sprintf(buf, "<NULL>"); 1396e5818a53SJeff Roberson error = sysctl_handle_string(oidp, buf, sizeof(buf), req); 1397e5818a53SJeff Roberson if (error != 0 || req->newptr == NULL) 1398e5818a53SJeff Roberson return (error); 1399e5818a53SJeff Roberson 1400e5818a53SJeff Roberson /* 1401e5818a53SJeff Roberson * Read in and validate the string. 1402e5818a53SJeff Roberson */ 1403e5818a53SJeff Roberson memset(&key, 0, sizeof(key)); 1404e5818a53SJeff Roberson p = &buf[bitset_strscan((struct bitset *)&key.ds_mask, 1405e5818a53SJeff Roberson DOMAINSET_SETSIZE, buf)]; 1406e5818a53SJeff Roberson if (p == buf) 1407e5818a53SJeff Roberson return (EINVAL); 1408e5818a53SJeff Roberson if (sscanf(p, ":%d:%d", &policy, &prefer) != 2) 1409e5818a53SJeff Roberson return (EINVAL); 1410e5818a53SJeff Roberson key.ds_policy = policy; 1411e5818a53SJeff Roberson key.ds_prefer = prefer; 1412e5818a53SJeff Roberson 1413e5818a53SJeff Roberson /* Domainset_create() validates the policy.*/ 1414e5818a53SJeff Roberson dset = domainset_create(&key); 1415e5818a53SJeff Roberson if (dset == NULL) 1416e5818a53SJeff Roberson return (EINVAL); 1417e5818a53SJeff Roberson *(struct domainset **)arg1 = dset; 1418e5818a53SJeff Roberson 1419e5818a53SJeff Roberson return (error); 1420e5818a53SJeff Roberson } 1421e5818a53SJeff Roberson 1422e5818a53SJeff Roberson /* 14233f289c3fSJeff Roberson * Apply an anonymous mask or a domain to a single thread. 1424d7f687fcSJeff Roberson */ 14253f289c3fSJeff Roberson static int 14263f289c3fSJeff Roberson _cpuset_setthread(lwpid_t id, cpuset_t *mask, struct domainset *domain) 1427d7f687fcSJeff Roberson { 14283f289c3fSJeff Roberson struct setlist cpusets; 14293f289c3fSJeff Roberson struct domainlist domainlist; 1430d7f687fcSJeff Roberson struct cpuset *nset; 1431d7f687fcSJeff Roberson struct cpuset *set; 1432d7f687fcSJeff Roberson struct thread *td; 1433d7f687fcSJeff Roberson struct proc *p; 1434d7f687fcSJeff Roberson int error; 1435d7f687fcSJeff Roberson 14363f289c3fSJeff Roberson cpuset_freelist_init(&cpusets, 1); 14373f289c3fSJeff Roberson domainset_freelist_init(&domainlist, domain != NULL); 14388bd75bddSJeff Roberson error = cpuset_which(CPU_WHICH_TID, id, &p, &td, &set); 1439d7f687fcSJeff Roberson if (error) 1440d7f687fcSJeff Roberson goto out; 1441a03ee000SJeff Roberson set = NULL; 1442d7f687fcSJeff Roberson thread_lock(td); 14433f289c3fSJeff Roberson error = cpuset_shadow(td->td_cpuset, &nset, mask, domain, 14443f289c3fSJeff Roberson &cpusets, &domainlist); 14453f289c3fSJeff Roberson if (error == 0) 14463f289c3fSJeff Roberson set = cpuset_update_thread(td, nset); 1447d7f687fcSJeff Roberson thread_unlock(td); 1448d7f687fcSJeff Roberson PROC_UNLOCK(p); 1449a03ee000SJeff Roberson if (set) 1450a03ee000SJeff Roberson cpuset_rel(set); 1451d7f687fcSJeff Roberson out: 14523f289c3fSJeff Roberson cpuset_freelist_free(&cpusets); 14533f289c3fSJeff Roberson domainset_freelist_free(&domainlist); 1454d7f687fcSJeff Roberson return (error); 1455d7f687fcSJeff Roberson } 1456d7f687fcSJeff Roberson 1457d7f687fcSJeff Roberson /* 14583f289c3fSJeff Roberson * Apply an anonymous mask to a single thread. 14593f289c3fSJeff Roberson */ 14603f289c3fSJeff Roberson int 14613f289c3fSJeff Roberson cpuset_setthread(lwpid_t id, cpuset_t *mask) 14623f289c3fSJeff Roberson { 14633f289c3fSJeff Roberson 14643f289c3fSJeff Roberson return _cpuset_setthread(id, mask, NULL); 14653f289c3fSJeff Roberson } 14663f289c3fSJeff Roberson 14673f289c3fSJeff Roberson /* 146881198539SAlexander V. Chernikov * Apply new cpumask to the ithread. 146981198539SAlexander V. Chernikov */ 147081198539SAlexander V. Chernikov int 14717f7528fcSAdrian Chadd cpuset_setithread(lwpid_t id, int cpu) 147281198539SAlexander V. Chernikov { 147381198539SAlexander V. Chernikov cpuset_t mask; 147481198539SAlexander V. Chernikov 147581198539SAlexander V. Chernikov CPU_ZERO(&mask); 147681198539SAlexander V. Chernikov if (cpu == NOCPU) 147781198539SAlexander V. Chernikov CPU_COPY(cpuset_root, &mask); 147881198539SAlexander V. Chernikov else 147981198539SAlexander V. Chernikov CPU_SET(cpu, &mask); 1480e5818a53SJeff Roberson return _cpuset_setthread(id, &mask, NULL); 1481e5818a53SJeff Roberson } 1482c1d9ecf2SAlexander V. Chernikov 148381198539SAlexander V. Chernikov /* 1484662e7fa8SMark Johnston * Initialize static domainsets after NUMA information is available. This is 1485920239efSMark Johnston * called before memory allocators are initialized. 1486662e7fa8SMark Johnston */ 1487662e7fa8SMark Johnston void 1488662e7fa8SMark Johnston domainset_init(void) 1489662e7fa8SMark Johnston { 1490662e7fa8SMark Johnston struct domainset *dset; 1491662e7fa8SMark Johnston int i; 1492662e7fa8SMark Johnston 1493662e7fa8SMark Johnston dset = &domainset_roundrobin; 1494662e7fa8SMark Johnston DOMAINSET_COPY(&all_domains, &dset->ds_mask); 1495662e7fa8SMark Johnston dset->ds_policy = DOMAINSET_POLICY_ROUNDROBIN; 1496662e7fa8SMark Johnston dset->ds_prefer = -1; 1497662e7fa8SMark Johnston _domainset_create(dset, NULL); 1498662e7fa8SMark Johnston 1499662e7fa8SMark Johnston for (i = 0; i < vm_ndomains; i++) { 15009978bd99SMark Johnston dset = &domainset_fixed[i]; 15019978bd99SMark Johnston DOMAINSET_ZERO(&dset->ds_mask); 15029978bd99SMark Johnston DOMAINSET_SET(i, &dset->ds_mask); 15039978bd99SMark Johnston dset->ds_policy = DOMAINSET_POLICY_ROUNDROBIN; 15049978bd99SMark Johnston _domainset_create(dset, NULL); 15059978bd99SMark Johnston 1506662e7fa8SMark Johnston dset = &domainset_prefer[i]; 1507662e7fa8SMark Johnston DOMAINSET_COPY(&all_domains, &dset->ds_mask); 1508662e7fa8SMark Johnston dset->ds_policy = DOMAINSET_POLICY_PREFER; 1509662e7fa8SMark Johnston dset->ds_prefer = i; 1510662e7fa8SMark Johnston _domainset_create(dset, NULL); 1511662e7fa8SMark Johnston } 1512662e7fa8SMark Johnston } 1513662e7fa8SMark Johnston 1514662e7fa8SMark Johnston /* 1515e5818a53SJeff Roberson * Create the domainset for cpuset 0, 1 and cpuset 2. 151681198539SAlexander V. Chernikov */ 15173f289c3fSJeff Roberson void 15183f289c3fSJeff Roberson domainset_zero(void) 15193f289c3fSJeff Roberson { 1520920239efSMark Johnston struct domainset *dset, *tmp; 15213f289c3fSJeff Roberson 15223f289c3fSJeff Roberson mtx_init(&cpuset_lock, "cpuset", NULL, MTX_SPIN | MTX_RECURSE); 15233f289c3fSJeff Roberson 15243f289c3fSJeff Roberson dset = &domainset0; 1525662e7fa8SMark Johnston DOMAINSET_COPY(&all_domains, &dset->ds_mask); 1526e5818a53SJeff Roberson dset->ds_policy = DOMAINSET_POLICY_FIRSTTOUCH; 15273f289c3fSJeff Roberson dset->ds_prefer = -1; 15283f289c3fSJeff Roberson curthread->td_domain.dr_policy = _domainset_create(dset, NULL); 1529e5818a53SJeff Roberson 1530e5818a53SJeff Roberson domainset_copy(dset, &domainset2); 1531e5818a53SJeff Roberson domainset2.ds_policy = DOMAINSET_POLICY_INTERLEAVE; 1532e5818a53SJeff Roberson kernel_object->domain.dr_policy = _domainset_create(&domainset2, NULL); 1533662e7fa8SMark Johnston 1534662e7fa8SMark Johnston /* Remove empty domains from the global policies. */ 1535920239efSMark Johnston LIST_FOREACH_SAFE(dset, &cpuset_domains, ds_link, tmp) 1536920239efSMark Johnston if (domainset_empty_vm(dset)) 1537920239efSMark Johnston LIST_REMOVE(dset, ds_link); 15383f289c3fSJeff Roberson } 153981198539SAlexander V. Chernikov 154081198539SAlexander V. Chernikov /* 1541e5818a53SJeff Roberson * Creates system-wide cpusets and the cpuset for thread0 including three 1542c0ae6688SJohn Baldwin * sets: 1543d7f687fcSJeff Roberson * 1544d7f687fcSJeff Roberson * 0 - The root set which should represent all valid processors in the 154530b7c6f9SKyle Evans * system. This set is immutable. 1546d7f687fcSJeff Roberson * 1 - The default set which all processes are a member of until changed. 1547d7f687fcSJeff Roberson * This allows an administrator to move all threads off of given cpus to 1548d7f687fcSJeff Roberson * dedicate them to high priority tasks or save power etc. 1549e5818a53SJeff Roberson * 2 - The kernel set which allows restriction and policy to be applied only 1550e5818a53SJeff Roberson * to kernel threads and the kernel_object. 1551d7f687fcSJeff Roberson */ 1552d7f687fcSJeff Roberson struct cpuset * 1553d7f687fcSJeff Roberson cpuset_thread0(void) 1554d7f687fcSJeff Roberson { 1555d7f687fcSJeff Roberson struct cpuset *set; 155627a3c9d7SJeff Roberson int i; 1557a6c7423aSMatt Macy int error __unused; 1558d7f687fcSJeff Roberson 1559d7f687fcSJeff Roberson cpuset_zone = uma_zcreate("cpuset", sizeof(struct cpuset), NULL, NULL, 1560e5818a53SJeff Roberson NULL, NULL, UMA_ALIGN_CACHE, 0); 15613f289c3fSJeff Roberson domainset_zone = uma_zcreate("domainset", sizeof(struct domainset), 1562e5818a53SJeff Roberson NULL, NULL, NULL, NULL, UMA_ALIGN_CACHE, 0); 156381198539SAlexander V. Chernikov 1564d7f687fcSJeff Roberson /* 1565e5818a53SJeff Roberson * Create the root system set (0) for the whole machine. Doesn't use 1566d7f687fcSJeff Roberson * cpuset_create() due to NULL parent. 1567d7f687fcSJeff Roberson */ 1568d7f687fcSJeff Roberson set = uma_zalloc(cpuset_zone, M_WAITOK | M_ZERO); 156927a3c9d7SJeff Roberson CPU_COPY(&all_cpus, &set->cs_mask); 1570d7f687fcSJeff Roberson LIST_INIT(&set->cs_children); 1571d7f687fcSJeff Roberson LIST_INSERT_HEAD(&cpuset_ids, set, cs_link); 15721a7bb896SMateusz Guzik refcount_init(&set->cs_ref, 1); 157327a3c9d7SJeff Roberson set->cs_flags = CPU_SET_ROOT | CPU_SET_RDONLY; 15743f289c3fSJeff Roberson set->cs_domain = &domainset0; 1575d7f687fcSJeff Roberson cpuset_zero = set; 1576a03ee000SJeff Roberson cpuset_root = &set->cs_mask; 157781198539SAlexander V. Chernikov 1578d7f687fcSJeff Roberson /* 1579e5818a53SJeff Roberson * Now derive a default (1), modifiable set from that to give out. 1580d7f687fcSJeff Roberson */ 15813f289c3fSJeff Roberson set = uma_zalloc(cpuset_zone, M_WAITOK | M_ZERO); 158230b7c6f9SKyle Evans error = cpuset_init(set, cpuset_zero, NULL, NULL, 1); 1583d7f687fcSJeff Roberson KASSERT(error == 0, ("Error creating default set: %d\n", error)); 158481198539SAlexander V. Chernikov cpuset_default = set; 1585e5818a53SJeff Roberson /* 1586e5818a53SJeff Roberson * Create the kernel set (2). 1587e5818a53SJeff Roberson */ 1588e5818a53SJeff Roberson set = uma_zalloc(cpuset_zone, M_WAITOK | M_ZERO); 158930b7c6f9SKyle Evans error = cpuset_init(set, cpuset_zero, NULL, NULL, 2); 1590e5818a53SJeff Roberson KASSERT(error == 0, ("Error creating kernel set: %d\n", error)); 1591e5818a53SJeff Roberson set->cs_domain = &domainset2; 1592e5818a53SJeff Roberson cpuset_kernel = set; 159381198539SAlexander V. Chernikov 1594d7f687fcSJeff Roberson /* 1595d7f687fcSJeff Roberson * Initialize the unit allocator. 0 and 1 are allocated above. 1596d7f687fcSJeff Roberson */ 1597d57cd5ccSStephen J. Kiernan cpuset_unr = new_unrhdr(3, INT_MAX, NULL); 1598d7f687fcSJeff Roberson 159927a3c9d7SJeff Roberson /* 160027a3c9d7SJeff Roberson * If MD code has not initialized per-domain cpusets, place all 160127a3c9d7SJeff Roberson * CPUs in domain 0. 160227a3c9d7SJeff Roberson */ 160327a3c9d7SJeff Roberson for (i = 0; i < MAXMEMDOM; i++) 160427a3c9d7SJeff Roberson if (!CPU_EMPTY(&cpuset_domain[i])) 160527a3c9d7SJeff Roberson goto domains_set; 160627a3c9d7SJeff Roberson CPU_COPY(&all_cpus, &cpuset_domain[0]); 160727a3c9d7SJeff Roberson domains_set: 160827a3c9d7SJeff Roberson 1609e5818a53SJeff Roberson return (cpuset_default); 1610e5818a53SJeff Roberson } 1611e5818a53SJeff Roberson 1612e5818a53SJeff Roberson void 1613e5818a53SJeff Roberson cpuset_kernthread(struct thread *td) 1614e5818a53SJeff Roberson { 1615e5818a53SJeff Roberson struct cpuset *set; 1616e5818a53SJeff Roberson 1617e5818a53SJeff Roberson thread_lock(td); 1618e5818a53SJeff Roberson set = td->td_cpuset; 1619e5818a53SJeff Roberson td->td_cpuset = cpuset_ref(cpuset_kernel); 1620e5818a53SJeff Roberson thread_unlock(td); 1621e5818a53SJeff Roberson cpuset_rel(set); 1622d7f687fcSJeff Roberson } 1623d7f687fcSJeff Roberson 1624d7f687fcSJeff Roberson /* 1625413628a7SBjoern A. Zeeb * Create a cpuset, which would be cpuset_create() but 1626413628a7SBjoern A. Zeeb * mark the new 'set' as root. 1627413628a7SBjoern A. Zeeb * 162847479a8cSBjoern A. Zeeb * We are not going to reparent the td to it. Use cpuset_setproc_update_set() 162947479a8cSBjoern A. Zeeb * for that. 1630413628a7SBjoern A. Zeeb * 1631413628a7SBjoern A. Zeeb * In case of no error, returns the set in *setp locked with a reference. 1632413628a7SBjoern A. Zeeb */ 1633413628a7SBjoern A. Zeeb int 16340304c731SJamie Gritton cpuset_create_root(struct prison *pr, struct cpuset **setp) 1635413628a7SBjoern A. Zeeb { 1636413628a7SBjoern A. Zeeb struct cpuset *set; 1637413628a7SBjoern A. Zeeb int error; 1638413628a7SBjoern A. Zeeb 16390304c731SJamie Gritton KASSERT(pr != NULL, ("[%s:%d] invalid pr", __func__, __LINE__)); 1640413628a7SBjoern A. Zeeb KASSERT(setp != NULL, ("[%s:%d] invalid setp", __func__, __LINE__)); 1641413628a7SBjoern A. Zeeb 164229d04ea8SKyle Evans set = NULL; 164329d04ea8SKyle Evans error = cpuset_create(&set, pr->pr_cpuset, &pr->pr_cpuset->cs_mask); 1644413628a7SBjoern A. Zeeb if (error) 1645413628a7SBjoern A. Zeeb return (error); 1646413628a7SBjoern A. Zeeb 164729d04ea8SKyle Evans KASSERT(set != NULL, ("[%s:%d] cpuset_create returned invalid data", 1648413628a7SBjoern A. Zeeb __func__, __LINE__)); 1649413628a7SBjoern A. Zeeb 1650413628a7SBjoern A. Zeeb /* Mark the set as root. */ 1651413628a7SBjoern A. Zeeb set->cs_flags |= CPU_SET_ROOT; 165229d04ea8SKyle Evans *setp = set; 1653413628a7SBjoern A. Zeeb 1654413628a7SBjoern A. Zeeb return (0); 1655413628a7SBjoern A. Zeeb } 1656413628a7SBjoern A. Zeeb 1657413628a7SBjoern A. Zeeb int 1658413628a7SBjoern A. Zeeb cpuset_setproc_update_set(struct proc *p, struct cpuset *set) 1659413628a7SBjoern A. Zeeb { 1660413628a7SBjoern A. Zeeb int error; 1661413628a7SBjoern A. Zeeb 1662413628a7SBjoern A. Zeeb KASSERT(p != NULL, ("[%s:%d] invalid proc", __func__, __LINE__)); 1663413628a7SBjoern A. Zeeb KASSERT(set != NULL, ("[%s:%d] invalid set", __func__, __LINE__)); 1664413628a7SBjoern A. Zeeb 1665413628a7SBjoern A. Zeeb cpuset_ref(set); 1666*d431dea5SKyle Evans error = cpuset_setproc(p->p_pid, set, NULL, NULL, true); 1667413628a7SBjoern A. Zeeb if (error) 1668413628a7SBjoern A. Zeeb return (error); 1669413628a7SBjoern A. Zeeb cpuset_rel(set); 1670413628a7SBjoern A. Zeeb return (0); 1671413628a7SBjoern A. Zeeb } 1672413628a7SBjoern A. Zeeb 16739eb997cbSMark Johnston /* 16749eb997cbSMark Johnston * In Capability mode, the only accesses that are permitted are to the current 16759eb997cbSMark Johnston * thread and process' CPU and domain sets. 16769eb997cbSMark Johnston */ 16779eb997cbSMark Johnston static int 16789eb997cbSMark Johnston cpuset_check_capabilities(struct thread *td, cpulevel_t level, cpuwhich_t which, 16799eb997cbSMark Johnston id_t id) 16809eb997cbSMark Johnston { 16819eb997cbSMark Johnston if (IN_CAPABILITY_MODE(td)) { 16829eb997cbSMark Johnston if (level != CPU_LEVEL_WHICH) 16839eb997cbSMark Johnston return (ECAPMODE); 16849eb997cbSMark Johnston if (which != CPU_WHICH_TID && which != CPU_WHICH_PID) 16859eb997cbSMark Johnston return (ECAPMODE); 168669b565d7SMark Johnston if (id != -1 && 168769b565d7SMark Johnston !(which == CPU_WHICH_TID && id == td->td_tid) && 168869b565d7SMark Johnston !(which == CPU_WHICH_PID && id == td->td_proc->p_pid)) 16899eb997cbSMark Johnston return (ECAPMODE); 16909eb997cbSMark Johnston } 16919eb997cbSMark Johnston return (0); 16929eb997cbSMark Johnston } 16939eb997cbSMark Johnston 1694d7f687fcSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 1695d7f687fcSJeff Roberson struct cpuset_args { 1696d7f687fcSJeff Roberson cpusetid_t *setid; 1697d7f687fcSJeff Roberson }; 1698d7f687fcSJeff Roberson #endif 1699d7f687fcSJeff Roberson int 17008451d0ddSKip Macy sys_cpuset(struct thread *td, struct cpuset_args *uap) 1701d7f687fcSJeff Roberson { 1702d7f687fcSJeff Roberson struct cpuset *root; 1703d7f687fcSJeff Roberson struct cpuset *set; 1704d7f687fcSJeff Roberson int error; 1705d7f687fcSJeff Roberson 1706d7f687fcSJeff Roberson thread_lock(td); 1707a03ee000SJeff Roberson root = cpuset_refroot(td->td_cpuset); 1708d7f687fcSJeff Roberson thread_unlock(td); 170929d04ea8SKyle Evans set = NULL; 1710d7f687fcSJeff Roberson error = cpuset_create(&set, root, &root->cs_mask); 1711d7f687fcSJeff Roberson cpuset_rel(root); 1712d7f687fcSJeff Roberson if (error) 1713d7f687fcSJeff Roberson return (error); 1714d7f687fcSJeff Roberson error = copyout(&set->cs_id, uap->setid, sizeof(set->cs_id)); 1715a03ee000SJeff Roberson if (error == 0) 1716*d431dea5SKyle Evans error = cpuset_setproc(-1, set, NULL, NULL, false); 1717d7f687fcSJeff Roberson cpuset_rel(set); 1718d7f687fcSJeff Roberson return (error); 1719d7f687fcSJeff Roberson } 1720d7f687fcSJeff Roberson 1721d7f687fcSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 1722d7f687fcSJeff Roberson struct cpuset_setid_args { 1723d7f687fcSJeff Roberson cpuwhich_t which; 1724d7f687fcSJeff Roberson id_t id; 1725d7f687fcSJeff Roberson cpusetid_t setid; 1726d7f687fcSJeff Roberson }; 1727d7f687fcSJeff Roberson #endif 1728d7f687fcSJeff Roberson int 17298451d0ddSKip Macy sys_cpuset_setid(struct thread *td, struct cpuset_setid_args *uap) 1730d7f687fcSJeff Roberson { 1731ea2ebdc1SEdward Tomasz Napierala 1732ea2ebdc1SEdward Tomasz Napierala return (kern_cpuset_setid(td, uap->which, uap->id, uap->setid)); 1733ea2ebdc1SEdward Tomasz Napierala } 1734ea2ebdc1SEdward Tomasz Napierala 1735ea2ebdc1SEdward Tomasz Napierala int 1736ea2ebdc1SEdward Tomasz Napierala kern_cpuset_setid(struct thread *td, cpuwhich_t which, 1737ea2ebdc1SEdward Tomasz Napierala id_t id, cpusetid_t setid) 1738ea2ebdc1SEdward Tomasz Napierala { 1739d7f687fcSJeff Roberson struct cpuset *set; 1740d7f687fcSJeff Roberson int error; 1741d7f687fcSJeff Roberson 1742d7f687fcSJeff Roberson /* 1743d7f687fcSJeff Roberson * Presently we only support per-process sets. 1744d7f687fcSJeff Roberson */ 1745ea2ebdc1SEdward Tomasz Napierala if (which != CPU_WHICH_PID) 1746d7f687fcSJeff Roberson return (EINVAL); 1747ea2ebdc1SEdward Tomasz Napierala set = cpuset_lookup(setid, td); 1748d7f687fcSJeff Roberson if (set == NULL) 1749d7f687fcSJeff Roberson return (ESRCH); 1750*d431dea5SKyle Evans error = cpuset_setproc(id, set, NULL, NULL, false); 1751d7f687fcSJeff Roberson cpuset_rel(set); 1752d7f687fcSJeff Roberson return (error); 1753d7f687fcSJeff Roberson } 1754d7f687fcSJeff Roberson 1755d7f687fcSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 1756d7f687fcSJeff Roberson struct cpuset_getid_args { 1757d7f687fcSJeff Roberson cpulevel_t level; 1758d7f687fcSJeff Roberson cpuwhich_t which; 1759d7f687fcSJeff Roberson id_t id; 1760d7f687fcSJeff Roberson cpusetid_t *setid; 17612b69bb1fSKevin Lo }; 1762d7f687fcSJeff Roberson #endif 1763d7f687fcSJeff Roberson int 17648451d0ddSKip Macy sys_cpuset_getid(struct thread *td, struct cpuset_getid_args *uap) 1765d7f687fcSJeff Roberson { 1766ea2ebdc1SEdward Tomasz Napierala 1767ea2ebdc1SEdward Tomasz Napierala return (kern_cpuset_getid(td, uap->level, uap->which, uap->id, 1768ea2ebdc1SEdward Tomasz Napierala uap->setid)); 1769ea2ebdc1SEdward Tomasz Napierala } 1770ea2ebdc1SEdward Tomasz Napierala 1771ea2ebdc1SEdward Tomasz Napierala int 1772ea2ebdc1SEdward Tomasz Napierala kern_cpuset_getid(struct thread *td, cpulevel_t level, cpuwhich_t which, 1773ea2ebdc1SEdward Tomasz Napierala id_t id, cpusetid_t *setid) 1774ea2ebdc1SEdward Tomasz Napierala { 1775d7f687fcSJeff Roberson struct cpuset *nset; 1776d7f687fcSJeff Roberson struct cpuset *set; 1777d7f687fcSJeff Roberson struct thread *ttd; 1778d7f687fcSJeff Roberson struct proc *p; 1779ea2ebdc1SEdward Tomasz Napierala cpusetid_t tmpid; 1780d7f687fcSJeff Roberson int error; 1781d7f687fcSJeff Roberson 1782ea2ebdc1SEdward Tomasz Napierala if (level == CPU_LEVEL_WHICH && which != CPU_WHICH_CPUSET) 1783d7f687fcSJeff Roberson return (EINVAL); 1784ea2ebdc1SEdward Tomasz Napierala error = cpuset_which(which, id, &p, &ttd, &set); 1785d7f687fcSJeff Roberson if (error) 1786d7f687fcSJeff Roberson return (error); 1787ea2ebdc1SEdward Tomasz Napierala switch (which) { 1788d7f687fcSJeff Roberson case CPU_WHICH_TID: 1789d7f687fcSJeff Roberson case CPU_WHICH_PID: 1790d7f687fcSJeff Roberson thread_lock(ttd); 1791a03ee000SJeff Roberson set = cpuset_refbase(ttd->td_cpuset); 1792d7f687fcSJeff Roberson thread_unlock(ttd); 1793d7f687fcSJeff Roberson PROC_UNLOCK(p); 1794d7f687fcSJeff Roberson break; 1795d7f687fcSJeff Roberson case CPU_WHICH_CPUSET: 1796413628a7SBjoern A. Zeeb case CPU_WHICH_JAIL: 1797d7f687fcSJeff Roberson break; 17989b33b154SJeff Roberson case CPU_WHICH_IRQ: 1799c0ae6688SJohn Baldwin case CPU_WHICH_DOMAIN: 18009b33b154SJeff Roberson return (EINVAL); 1801d7f687fcSJeff Roberson } 1802ea2ebdc1SEdward Tomasz Napierala switch (level) { 1803d7f687fcSJeff Roberson case CPU_LEVEL_ROOT: 1804a03ee000SJeff Roberson nset = cpuset_refroot(set); 1805d7f687fcSJeff Roberson cpuset_rel(set); 1806d7f687fcSJeff Roberson set = nset; 1807d7f687fcSJeff Roberson break; 1808d7f687fcSJeff Roberson case CPU_LEVEL_CPUSET: 1809d7f687fcSJeff Roberson break; 1810d7f687fcSJeff Roberson case CPU_LEVEL_WHICH: 1811d7f687fcSJeff Roberson break; 1812d7f687fcSJeff Roberson } 1813ea2ebdc1SEdward Tomasz Napierala tmpid = set->cs_id; 1814d7f687fcSJeff Roberson cpuset_rel(set); 1815d7f687fcSJeff Roberson if (error == 0) 1816a1d0659cSJung-uk Kim error = copyout(&tmpid, setid, sizeof(tmpid)); 1817d7f687fcSJeff Roberson 1818d7f687fcSJeff Roberson return (error); 1819d7f687fcSJeff Roberson } 1820d7f687fcSJeff Roberson 1821d7f687fcSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 1822d7f687fcSJeff Roberson struct cpuset_getaffinity_args { 1823d7f687fcSJeff Roberson cpulevel_t level; 1824d7f687fcSJeff Roberson cpuwhich_t which; 18257f64829aSRuslan Ermilov id_t id; 18267f64829aSRuslan Ermilov size_t cpusetsize; 18277f64829aSRuslan Ermilov cpuset_t *mask; 1828d7f687fcSJeff Roberson }; 1829d7f687fcSJeff Roberson #endif 1830d7f687fcSJeff Roberson int 18318451d0ddSKip Macy sys_cpuset_getaffinity(struct thread *td, struct cpuset_getaffinity_args *uap) 1832d7f687fcSJeff Roberson { 183396ee4310SEdward Tomasz Napierala 183496ee4310SEdward Tomasz Napierala return (kern_cpuset_getaffinity(td, uap->level, uap->which, 183596ee4310SEdward Tomasz Napierala uap->id, uap->cpusetsize, uap->mask)); 183696ee4310SEdward Tomasz Napierala } 183796ee4310SEdward Tomasz Napierala 183896ee4310SEdward Tomasz Napierala int 183996ee4310SEdward Tomasz Napierala kern_cpuset_getaffinity(struct thread *td, cpulevel_t level, cpuwhich_t which, 184096ee4310SEdward Tomasz Napierala id_t id, size_t cpusetsize, cpuset_t *maskp) 184196ee4310SEdward Tomasz Napierala { 1842d7f687fcSJeff Roberson struct thread *ttd; 1843d7f687fcSJeff Roberson struct cpuset *nset; 1844d7f687fcSJeff Roberson struct cpuset *set; 1845d7f687fcSJeff Roberson struct proc *p; 1846d7f687fcSJeff Roberson cpuset_t *mask; 1847d7f687fcSJeff Roberson int error; 18487f64829aSRuslan Ermilov size_t size; 1849d7f687fcSJeff Roberson 185096ee4310SEdward Tomasz Napierala if (cpusetsize < sizeof(cpuset_t) || cpusetsize > CPU_MAXSIZE / NBBY) 1851d7f687fcSJeff Roberson return (ERANGE); 18529eb997cbSMark Johnston error = cpuset_check_capabilities(td, level, which, id); 18539eb997cbSMark Johnston if (error != 0) 18549eb997cbSMark Johnston return (error); 185596ee4310SEdward Tomasz Napierala size = cpusetsize; 1856d7f687fcSJeff Roberson mask = malloc(size, M_TEMP, M_WAITOK | M_ZERO); 185796ee4310SEdward Tomasz Napierala error = cpuset_which(which, id, &p, &ttd, &set); 1858d7f687fcSJeff Roberson if (error) 1859d7f687fcSJeff Roberson goto out; 186096ee4310SEdward Tomasz Napierala switch (level) { 1861d7f687fcSJeff Roberson case CPU_LEVEL_ROOT: 1862d7f687fcSJeff Roberson case CPU_LEVEL_CPUSET: 186396ee4310SEdward Tomasz Napierala switch (which) { 1864d7f687fcSJeff Roberson case CPU_WHICH_TID: 1865d7f687fcSJeff Roberson case CPU_WHICH_PID: 1866d7f687fcSJeff Roberson thread_lock(ttd); 1867d7f687fcSJeff Roberson set = cpuset_ref(ttd->td_cpuset); 1868d7f687fcSJeff Roberson thread_unlock(ttd); 1869d7f687fcSJeff Roberson break; 1870d7f687fcSJeff Roberson case CPU_WHICH_CPUSET: 1871413628a7SBjoern A. Zeeb case CPU_WHICH_JAIL: 1872d7f687fcSJeff Roberson break; 18739b33b154SJeff Roberson case CPU_WHICH_IRQ: 187429dfb631SConrad Meyer case CPU_WHICH_INTRHANDLER: 187529dfb631SConrad Meyer case CPU_WHICH_ITHREAD: 1876c0ae6688SJohn Baldwin case CPU_WHICH_DOMAIN: 18779b33b154SJeff Roberson error = EINVAL; 18789b33b154SJeff Roberson goto out; 1879d7f687fcSJeff Roberson } 188096ee4310SEdward Tomasz Napierala if (level == CPU_LEVEL_ROOT) 1881a03ee000SJeff Roberson nset = cpuset_refroot(set); 1882d7f687fcSJeff Roberson else 1883a03ee000SJeff Roberson nset = cpuset_refbase(set); 1884d7f687fcSJeff Roberson CPU_COPY(&nset->cs_mask, mask); 1885d7f687fcSJeff Roberson cpuset_rel(nset); 1886d7f687fcSJeff Roberson break; 1887d7f687fcSJeff Roberson case CPU_LEVEL_WHICH: 188896ee4310SEdward Tomasz Napierala switch (which) { 1889d7f687fcSJeff Roberson case CPU_WHICH_TID: 1890d7f687fcSJeff Roberson thread_lock(ttd); 1891d7f687fcSJeff Roberson CPU_COPY(&ttd->td_cpuset->cs_mask, mask); 1892d7f687fcSJeff Roberson thread_unlock(ttd); 1893d7f687fcSJeff Roberson break; 1894d7f687fcSJeff Roberson case CPU_WHICH_PID: 1895d7f687fcSJeff Roberson FOREACH_THREAD_IN_PROC(p, ttd) { 1896d7f687fcSJeff Roberson thread_lock(ttd); 1897d7f687fcSJeff Roberson CPU_OR(mask, &ttd->td_cpuset->cs_mask); 1898d7f687fcSJeff Roberson thread_unlock(ttd); 1899d7f687fcSJeff Roberson } 1900d7f687fcSJeff Roberson break; 1901d7f687fcSJeff Roberson case CPU_WHICH_CPUSET: 1902413628a7SBjoern A. Zeeb case CPU_WHICH_JAIL: 1903d7f687fcSJeff Roberson CPU_COPY(&set->cs_mask, mask); 1904d7f687fcSJeff Roberson break; 19059b33b154SJeff Roberson case CPU_WHICH_IRQ: 190629dfb631SConrad Meyer case CPU_WHICH_INTRHANDLER: 190729dfb631SConrad Meyer case CPU_WHICH_ITHREAD: 190829dfb631SConrad Meyer error = intr_getaffinity(id, which, mask); 19099b33b154SJeff Roberson break; 1910c0ae6688SJohn Baldwin case CPU_WHICH_DOMAIN: 191196ee4310SEdward Tomasz Napierala if (id < 0 || id >= MAXMEMDOM) 1912c0ae6688SJohn Baldwin error = ESRCH; 1913c0ae6688SJohn Baldwin else 191496ee4310SEdward Tomasz Napierala CPU_COPY(&cpuset_domain[id], mask); 1915c0ae6688SJohn Baldwin break; 1916d7f687fcSJeff Roberson } 1917d7f687fcSJeff Roberson break; 1918d7f687fcSJeff Roberson default: 1919d7f687fcSJeff Roberson error = EINVAL; 1920d7f687fcSJeff Roberson break; 1921d7f687fcSJeff Roberson } 1922d7f687fcSJeff Roberson if (set) 1923d7f687fcSJeff Roberson cpuset_rel(set); 1924d7f687fcSJeff Roberson if (p) 1925d7f687fcSJeff Roberson PROC_UNLOCK(p); 1926d7f687fcSJeff Roberson if (error == 0) 192796ee4310SEdward Tomasz Napierala error = copyout(mask, maskp, size); 1928d7f687fcSJeff Roberson out: 1929d7f687fcSJeff Roberson free(mask, M_TEMP); 1930d7f687fcSJeff Roberson return (error); 1931d7f687fcSJeff Roberson } 1932d7f687fcSJeff Roberson 1933d7f687fcSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 1934d7f687fcSJeff Roberson struct cpuset_setaffinity_args { 1935d7f687fcSJeff Roberson cpulevel_t level; 1936d7f687fcSJeff Roberson cpuwhich_t which; 19377f64829aSRuslan Ermilov id_t id; 19387f64829aSRuslan Ermilov size_t cpusetsize; 19397f64829aSRuslan Ermilov const cpuset_t *mask; 1940d7f687fcSJeff Roberson }; 1941d7f687fcSJeff Roberson #endif 1942d7f687fcSJeff Roberson int 19438451d0ddSKip Macy sys_cpuset_setaffinity(struct thread *td, struct cpuset_setaffinity_args *uap) 1944d7f687fcSJeff Roberson { 194596ee4310SEdward Tomasz Napierala 194696ee4310SEdward Tomasz Napierala return (kern_cpuset_setaffinity(td, uap->level, uap->which, 194796ee4310SEdward Tomasz Napierala uap->id, uap->cpusetsize, uap->mask)); 194896ee4310SEdward Tomasz Napierala } 194996ee4310SEdward Tomasz Napierala 195096ee4310SEdward Tomasz Napierala int 195196ee4310SEdward Tomasz Napierala kern_cpuset_setaffinity(struct thread *td, cpulevel_t level, cpuwhich_t which, 195296ee4310SEdward Tomasz Napierala id_t id, size_t cpusetsize, const cpuset_t *maskp) 195396ee4310SEdward Tomasz Napierala { 1954d7f687fcSJeff Roberson struct cpuset *nset; 1955d7f687fcSJeff Roberson struct cpuset *set; 1956d7f687fcSJeff Roberson struct thread *ttd; 1957d7f687fcSJeff Roberson struct proc *p; 1958d7f687fcSJeff Roberson cpuset_t *mask; 1959d7f687fcSJeff Roberson int error; 1960d7f687fcSJeff Roberson 196196ee4310SEdward Tomasz Napierala if (cpusetsize < sizeof(cpuset_t) || cpusetsize > CPU_MAXSIZE / NBBY) 1962d7f687fcSJeff Roberson return (ERANGE); 19639eb997cbSMark Johnston error = cpuset_check_capabilities(td, level, which, id); 19649eb997cbSMark Johnston if (error != 0) 19659eb997cbSMark Johnston return (error); 196696ee4310SEdward Tomasz Napierala mask = malloc(cpusetsize, M_TEMP, M_WAITOK | M_ZERO); 196796ee4310SEdward Tomasz Napierala error = copyin(maskp, mask, cpusetsize); 1968d7f687fcSJeff Roberson if (error) 1969d7f687fcSJeff Roberson goto out; 197073c40187SJeff Roberson /* 197173c40187SJeff Roberson * Verify that no high bits are set. 197273c40187SJeff Roberson */ 197396ee4310SEdward Tomasz Napierala if (cpusetsize > sizeof(cpuset_t)) { 197473c40187SJeff Roberson char *end; 197573c40187SJeff Roberson char *cp; 197673c40187SJeff Roberson 197773c40187SJeff Roberson end = cp = (char *)&mask->__bits; 197896ee4310SEdward Tomasz Napierala end += cpusetsize; 197973c40187SJeff Roberson cp += sizeof(cpuset_t); 198073c40187SJeff Roberson while (cp != end) 198173c40187SJeff Roberson if (*cp++ != 0) { 198273c40187SJeff Roberson error = EINVAL; 198373c40187SJeff Roberson goto out; 198473c40187SJeff Roberson } 198573c40187SJeff Roberson } 198696ee4310SEdward Tomasz Napierala switch (level) { 1987d7f687fcSJeff Roberson case CPU_LEVEL_ROOT: 1988d7f687fcSJeff Roberson case CPU_LEVEL_CPUSET: 198996ee4310SEdward Tomasz Napierala error = cpuset_which(which, id, &p, &ttd, &set); 1990d7f687fcSJeff Roberson if (error) 1991d7f687fcSJeff Roberson break; 199296ee4310SEdward Tomasz Napierala switch (which) { 1993d7f687fcSJeff Roberson case CPU_WHICH_TID: 1994d7f687fcSJeff Roberson case CPU_WHICH_PID: 1995d7f687fcSJeff Roberson thread_lock(ttd); 1996d7f687fcSJeff Roberson set = cpuset_ref(ttd->td_cpuset); 1997d7f687fcSJeff Roberson thread_unlock(ttd); 1998c6440f72SJeff Roberson PROC_UNLOCK(p); 1999d7f687fcSJeff Roberson break; 2000d7f687fcSJeff Roberson case CPU_WHICH_CPUSET: 2001413628a7SBjoern A. Zeeb case CPU_WHICH_JAIL: 2002d7f687fcSJeff Roberson break; 20039b33b154SJeff Roberson case CPU_WHICH_IRQ: 200429dfb631SConrad Meyer case CPU_WHICH_INTRHANDLER: 200529dfb631SConrad Meyer case CPU_WHICH_ITHREAD: 2006c0ae6688SJohn Baldwin case CPU_WHICH_DOMAIN: 20079b33b154SJeff Roberson error = EINVAL; 20089b33b154SJeff Roberson goto out; 2009d7f687fcSJeff Roberson } 201096ee4310SEdward Tomasz Napierala if (level == CPU_LEVEL_ROOT) 2011a03ee000SJeff Roberson nset = cpuset_refroot(set); 2012d7f687fcSJeff Roberson else 2013a03ee000SJeff Roberson nset = cpuset_refbase(set); 2014d7f687fcSJeff Roberson error = cpuset_modify(nset, mask); 2015d7f687fcSJeff Roberson cpuset_rel(nset); 2016d7f687fcSJeff Roberson cpuset_rel(set); 2017d7f687fcSJeff Roberson break; 2018d7f687fcSJeff Roberson case CPU_LEVEL_WHICH: 201996ee4310SEdward Tomasz Napierala switch (which) { 2020d7f687fcSJeff Roberson case CPU_WHICH_TID: 202196ee4310SEdward Tomasz Napierala error = cpuset_setthread(id, mask); 2022d7f687fcSJeff Roberson break; 2023d7f687fcSJeff Roberson case CPU_WHICH_PID: 2024*d431dea5SKyle Evans error = cpuset_setproc(id, NULL, mask, NULL, false); 2025d7f687fcSJeff Roberson break; 2026d7f687fcSJeff Roberson case CPU_WHICH_CPUSET: 2027413628a7SBjoern A. Zeeb case CPU_WHICH_JAIL: 202896ee4310SEdward Tomasz Napierala error = cpuset_which(which, id, &p, &ttd, &set); 2029d7f687fcSJeff Roberson if (error == 0) { 2030d7f687fcSJeff Roberson error = cpuset_modify(set, mask); 2031d7f687fcSJeff Roberson cpuset_rel(set); 2032d7f687fcSJeff Roberson } 2033d7f687fcSJeff Roberson break; 20349b33b154SJeff Roberson case CPU_WHICH_IRQ: 203529dfb631SConrad Meyer case CPU_WHICH_INTRHANDLER: 203629dfb631SConrad Meyer case CPU_WHICH_ITHREAD: 203729dfb631SConrad Meyer error = intr_setaffinity(id, which, mask); 20389b33b154SJeff Roberson break; 2039d7f687fcSJeff Roberson default: 2040d7f687fcSJeff Roberson error = EINVAL; 2041d7f687fcSJeff Roberson break; 2042d7f687fcSJeff Roberson } 2043d7f687fcSJeff Roberson break; 2044d7f687fcSJeff Roberson default: 2045d7f687fcSJeff Roberson error = EINVAL; 2046d7f687fcSJeff Roberson break; 2047d7f687fcSJeff Roberson } 2048d7f687fcSJeff Roberson out: 2049d7f687fcSJeff Roberson free(mask, M_TEMP); 2050d7f687fcSJeff Roberson return (error); 2051d7f687fcSJeff Roberson } 2052dea0ed66SBjoern A. Zeeb 20533f289c3fSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 20543f289c3fSJeff Roberson struct cpuset_getdomain_args { 20553f289c3fSJeff Roberson cpulevel_t level; 20563f289c3fSJeff Roberson cpuwhich_t which; 20573f289c3fSJeff Roberson id_t id; 20583f289c3fSJeff Roberson size_t domainsetsize; 20593f289c3fSJeff Roberson domainset_t *mask; 20603f289c3fSJeff Roberson int *policy; 20613f289c3fSJeff Roberson }; 20623f289c3fSJeff Roberson #endif 20633f289c3fSJeff Roberson int 20643f289c3fSJeff Roberson sys_cpuset_getdomain(struct thread *td, struct cpuset_getdomain_args *uap) 2065dea0ed66SBjoern A. Zeeb { 2066dea0ed66SBjoern A. Zeeb 20673f289c3fSJeff Roberson return (kern_cpuset_getdomain(td, uap->level, uap->which, 20683f289c3fSJeff Roberson uap->id, uap->domainsetsize, uap->mask, uap->policy)); 20693f289c3fSJeff Roberson } 20703f289c3fSJeff Roberson 20713f289c3fSJeff Roberson int 20723f289c3fSJeff Roberson kern_cpuset_getdomain(struct thread *td, cpulevel_t level, cpuwhich_t which, 20733f289c3fSJeff Roberson id_t id, size_t domainsetsize, domainset_t *maskp, int *policyp) 20743f289c3fSJeff Roberson { 20753f289c3fSJeff Roberson struct domainset outset; 20763f289c3fSJeff Roberson struct thread *ttd; 20773f289c3fSJeff Roberson struct cpuset *nset; 20783f289c3fSJeff Roberson struct cpuset *set; 20793f289c3fSJeff Roberson struct domainset *dset; 20803f289c3fSJeff Roberson struct proc *p; 20813f289c3fSJeff Roberson domainset_t *mask; 20823f289c3fSJeff Roberson int error; 20833f289c3fSJeff Roberson 20843f289c3fSJeff Roberson if (domainsetsize < sizeof(domainset_t) || 20853f289c3fSJeff Roberson domainsetsize > DOMAINSET_MAXSIZE / NBBY) 20863f289c3fSJeff Roberson return (ERANGE); 20879eb997cbSMark Johnston error = cpuset_check_capabilities(td, level, which, id); 20889eb997cbSMark Johnston if (error != 0) 20899eb997cbSMark Johnston return (error); 20903f289c3fSJeff Roberson mask = malloc(domainsetsize, M_TEMP, M_WAITOK | M_ZERO); 20913f289c3fSJeff Roberson bzero(&outset, sizeof(outset)); 20923f289c3fSJeff Roberson error = cpuset_which(which, id, &p, &ttd, &set); 20933f289c3fSJeff Roberson if (error) 20943f289c3fSJeff Roberson goto out; 20953f289c3fSJeff Roberson switch (level) { 20963f289c3fSJeff Roberson case CPU_LEVEL_ROOT: 20973f289c3fSJeff Roberson case CPU_LEVEL_CPUSET: 20983f289c3fSJeff Roberson switch (which) { 20993f289c3fSJeff Roberson case CPU_WHICH_TID: 21003f289c3fSJeff Roberson case CPU_WHICH_PID: 21013f289c3fSJeff Roberson thread_lock(ttd); 21023f289c3fSJeff Roberson set = cpuset_ref(ttd->td_cpuset); 21033f289c3fSJeff Roberson thread_unlock(ttd); 21043f289c3fSJeff Roberson break; 21053f289c3fSJeff Roberson case CPU_WHICH_CPUSET: 21063f289c3fSJeff Roberson case CPU_WHICH_JAIL: 21073f289c3fSJeff Roberson break; 21083f289c3fSJeff Roberson case CPU_WHICH_IRQ: 21093f289c3fSJeff Roberson case CPU_WHICH_INTRHANDLER: 21103f289c3fSJeff Roberson case CPU_WHICH_ITHREAD: 21113f289c3fSJeff Roberson case CPU_WHICH_DOMAIN: 21123f289c3fSJeff Roberson error = EINVAL; 21133f289c3fSJeff Roberson goto out; 21143f289c3fSJeff Roberson } 21153f289c3fSJeff Roberson if (level == CPU_LEVEL_ROOT) 21163f289c3fSJeff Roberson nset = cpuset_refroot(set); 21173f289c3fSJeff Roberson else 21183f289c3fSJeff Roberson nset = cpuset_refbase(set); 21193f289c3fSJeff Roberson domainset_copy(nset->cs_domain, &outset); 21203f289c3fSJeff Roberson cpuset_rel(nset); 21213f289c3fSJeff Roberson break; 21223f289c3fSJeff Roberson case CPU_LEVEL_WHICH: 21233f289c3fSJeff Roberson switch (which) { 21243f289c3fSJeff Roberson case CPU_WHICH_TID: 21253f289c3fSJeff Roberson thread_lock(ttd); 21263f289c3fSJeff Roberson domainset_copy(ttd->td_cpuset->cs_domain, &outset); 21273f289c3fSJeff Roberson thread_unlock(ttd); 21283f289c3fSJeff Roberson break; 21293f289c3fSJeff Roberson case CPU_WHICH_PID: 21303f289c3fSJeff Roberson FOREACH_THREAD_IN_PROC(p, ttd) { 21313f289c3fSJeff Roberson thread_lock(ttd); 21323f289c3fSJeff Roberson dset = ttd->td_cpuset->cs_domain; 21333f289c3fSJeff Roberson /* Show all domains in the proc. */ 21343f289c3fSJeff Roberson DOMAINSET_OR(&outset.ds_mask, &dset->ds_mask); 21353f289c3fSJeff Roberson /* Last policy wins. */ 21363f289c3fSJeff Roberson outset.ds_policy = dset->ds_policy; 21373f289c3fSJeff Roberson outset.ds_prefer = dset->ds_prefer; 21383f289c3fSJeff Roberson thread_unlock(ttd); 21393f289c3fSJeff Roberson } 21403f289c3fSJeff Roberson break; 21413f289c3fSJeff Roberson case CPU_WHICH_CPUSET: 21423f289c3fSJeff Roberson case CPU_WHICH_JAIL: 21433f289c3fSJeff Roberson domainset_copy(set->cs_domain, &outset); 21443f289c3fSJeff Roberson break; 21453f289c3fSJeff Roberson case CPU_WHICH_IRQ: 21463f289c3fSJeff Roberson case CPU_WHICH_INTRHANDLER: 21473f289c3fSJeff Roberson case CPU_WHICH_ITHREAD: 21483f289c3fSJeff Roberson case CPU_WHICH_DOMAIN: 21493f289c3fSJeff Roberson error = EINVAL; 21503f289c3fSJeff Roberson break; 21513f289c3fSJeff Roberson } 21523f289c3fSJeff Roberson break; 21533f289c3fSJeff Roberson default: 21543f289c3fSJeff Roberson error = EINVAL; 21553f289c3fSJeff Roberson break; 21563f289c3fSJeff Roberson } 21573f289c3fSJeff Roberson if (set) 21583f289c3fSJeff Roberson cpuset_rel(set); 21593f289c3fSJeff Roberson if (p) 21603f289c3fSJeff Roberson PROC_UNLOCK(p); 21613f289c3fSJeff Roberson /* 21623f289c3fSJeff Roberson * Translate prefer into a set containing only the preferred domain, 21633f289c3fSJeff Roberson * not the entire fallback set. 21643f289c3fSJeff Roberson */ 21653f289c3fSJeff Roberson if (outset.ds_policy == DOMAINSET_POLICY_PREFER) { 21663f289c3fSJeff Roberson DOMAINSET_ZERO(&outset.ds_mask); 21673f289c3fSJeff Roberson DOMAINSET_SET(outset.ds_prefer, &outset.ds_mask); 21683f289c3fSJeff Roberson } 21693f289c3fSJeff Roberson DOMAINSET_COPY(&outset.ds_mask, mask); 21703f289c3fSJeff Roberson if (error == 0) 21713f289c3fSJeff Roberson error = copyout(mask, maskp, domainsetsize); 21723f289c3fSJeff Roberson if (error == 0) 2173dd51fec3SBrooks Davis if (suword32(policyp, outset.ds_policy) != 0) 2174dd51fec3SBrooks Davis error = EFAULT; 21753f289c3fSJeff Roberson out: 21763f289c3fSJeff Roberson free(mask, M_TEMP); 21773f289c3fSJeff Roberson return (error); 21783f289c3fSJeff Roberson } 21793f289c3fSJeff Roberson 21803f289c3fSJeff Roberson #ifndef _SYS_SYSPROTO_H_ 21813f289c3fSJeff Roberson struct cpuset_setdomain_args { 21823f289c3fSJeff Roberson cpulevel_t level; 21833f289c3fSJeff Roberson cpuwhich_t which; 21843f289c3fSJeff Roberson id_t id; 21853f289c3fSJeff Roberson size_t domainsetsize; 21863f289c3fSJeff Roberson domainset_t *mask; 21873f289c3fSJeff Roberson int policy; 21883f289c3fSJeff Roberson }; 21893f289c3fSJeff Roberson #endif 21903f289c3fSJeff Roberson int 21913f289c3fSJeff Roberson sys_cpuset_setdomain(struct thread *td, struct cpuset_setdomain_args *uap) 21923f289c3fSJeff Roberson { 21933f289c3fSJeff Roberson 21943f289c3fSJeff Roberson return (kern_cpuset_setdomain(td, uap->level, uap->which, 21953f289c3fSJeff Roberson uap->id, uap->domainsetsize, uap->mask, uap->policy)); 21963f289c3fSJeff Roberson } 21973f289c3fSJeff Roberson 21983f289c3fSJeff Roberson int 21993f289c3fSJeff Roberson kern_cpuset_setdomain(struct thread *td, cpulevel_t level, cpuwhich_t which, 22003f289c3fSJeff Roberson id_t id, size_t domainsetsize, const domainset_t *maskp, int policy) 22013f289c3fSJeff Roberson { 22023f289c3fSJeff Roberson struct cpuset *nset; 22033f289c3fSJeff Roberson struct cpuset *set; 22043f289c3fSJeff Roberson struct thread *ttd; 22053f289c3fSJeff Roberson struct proc *p; 22063f289c3fSJeff Roberson struct domainset domain; 22073f289c3fSJeff Roberson domainset_t *mask; 22083f289c3fSJeff Roberson int error; 22093f289c3fSJeff Roberson 22103f289c3fSJeff Roberson if (domainsetsize < sizeof(domainset_t) || 22113f289c3fSJeff Roberson domainsetsize > DOMAINSET_MAXSIZE / NBBY) 22123f289c3fSJeff Roberson return (ERANGE); 221370d66bcfSEric van Gyzen if (policy <= DOMAINSET_POLICY_INVALID || 221470d66bcfSEric van Gyzen policy > DOMAINSET_POLICY_MAX) 221570d66bcfSEric van Gyzen return (EINVAL); 22169eb997cbSMark Johnston error = cpuset_check_capabilities(td, level, which, id); 22179eb997cbSMark Johnston if (error != 0) 22189eb997cbSMark Johnston return (error); 22193f289c3fSJeff Roberson memset(&domain, 0, sizeof(domain)); 22203f289c3fSJeff Roberson mask = malloc(domainsetsize, M_TEMP, M_WAITOK | M_ZERO); 22213f289c3fSJeff Roberson error = copyin(maskp, mask, domainsetsize); 22223f289c3fSJeff Roberson if (error) 22233f289c3fSJeff Roberson goto out; 22243f289c3fSJeff Roberson /* 22253f289c3fSJeff Roberson * Verify that no high bits are set. 22263f289c3fSJeff Roberson */ 22273f289c3fSJeff Roberson if (domainsetsize > sizeof(domainset_t)) { 22283f289c3fSJeff Roberson char *end; 22293f289c3fSJeff Roberson char *cp; 22303f289c3fSJeff Roberson 22313f289c3fSJeff Roberson end = cp = (char *)&mask->__bits; 22323f289c3fSJeff Roberson end += domainsetsize; 22333f289c3fSJeff Roberson cp += sizeof(domainset_t); 22343f289c3fSJeff Roberson while (cp != end) 22353f289c3fSJeff Roberson if (*cp++ != 0) { 22363f289c3fSJeff Roberson error = EINVAL; 22373f289c3fSJeff Roberson goto out; 22383f289c3fSJeff Roberson } 22393f289c3fSJeff Roberson } 22403f289c3fSJeff Roberson DOMAINSET_COPY(mask, &domain.ds_mask); 22413f289c3fSJeff Roberson domain.ds_policy = policy; 22423f289c3fSJeff Roberson 224363cdd18eSMark Johnston /* 224463cdd18eSMark Johnston * Sanitize the provided mask. 224563cdd18eSMark Johnston */ 224663cdd18eSMark Johnston if (!DOMAINSET_SUBSET(&all_domains, &domain.ds_mask)) { 224763cdd18eSMark Johnston error = EINVAL; 224863cdd18eSMark Johnston goto out; 224963cdd18eSMark Johnston } 225063cdd18eSMark Johnston 22513f289c3fSJeff Roberson /* Translate preferred policy into a mask and fallback. */ 22523f289c3fSJeff Roberson if (policy == DOMAINSET_POLICY_PREFER) { 22533f289c3fSJeff Roberson /* Only support a single preferred domain. */ 225470d66bcfSEric van Gyzen if (DOMAINSET_COUNT(&domain.ds_mask) != 1) { 225570d66bcfSEric van Gyzen error = EINVAL; 225670d66bcfSEric van Gyzen goto out; 225770d66bcfSEric van Gyzen } 22583f289c3fSJeff Roberson domain.ds_prefer = DOMAINSET_FFS(&domain.ds_mask) - 1; 22593f289c3fSJeff Roberson /* This will be constrained by domainset_shadow(). */ 226063cdd18eSMark Johnston DOMAINSET_COPY(&all_domains, &domain.ds_mask); 22613f289c3fSJeff Roberson } 22623f289c3fSJeff Roberson 226330c5525bSAndrew Gallatin /* 226430c5525bSAndrew Gallatin * When given an impossible policy, fall back to interleaving 226563cdd18eSMark Johnston * across all domains. 226630c5525bSAndrew Gallatin */ 226730c5525bSAndrew Gallatin if (domainset_empty_vm(&domain)) 226830c5525bSAndrew Gallatin domainset_copy(&domainset2, &domain); 226930c5525bSAndrew Gallatin 22703f289c3fSJeff Roberson switch (level) { 22713f289c3fSJeff Roberson case CPU_LEVEL_ROOT: 22723f289c3fSJeff Roberson case CPU_LEVEL_CPUSET: 22733f289c3fSJeff Roberson error = cpuset_which(which, id, &p, &ttd, &set); 22743f289c3fSJeff Roberson if (error) 22753f289c3fSJeff Roberson break; 22763f289c3fSJeff Roberson switch (which) { 22773f289c3fSJeff Roberson case CPU_WHICH_TID: 22783f289c3fSJeff Roberson case CPU_WHICH_PID: 22793f289c3fSJeff Roberson thread_lock(ttd); 22803f289c3fSJeff Roberson set = cpuset_ref(ttd->td_cpuset); 22813f289c3fSJeff Roberson thread_unlock(ttd); 22823f289c3fSJeff Roberson PROC_UNLOCK(p); 22833f289c3fSJeff Roberson break; 22843f289c3fSJeff Roberson case CPU_WHICH_CPUSET: 22853f289c3fSJeff Roberson case CPU_WHICH_JAIL: 22863f289c3fSJeff Roberson break; 22873f289c3fSJeff Roberson case CPU_WHICH_IRQ: 22883f289c3fSJeff Roberson case CPU_WHICH_INTRHANDLER: 22893f289c3fSJeff Roberson case CPU_WHICH_ITHREAD: 22903f289c3fSJeff Roberson case CPU_WHICH_DOMAIN: 22913f289c3fSJeff Roberson error = EINVAL; 22923f289c3fSJeff Roberson goto out; 22933f289c3fSJeff Roberson } 22943f289c3fSJeff Roberson if (level == CPU_LEVEL_ROOT) 22953f289c3fSJeff Roberson nset = cpuset_refroot(set); 22963f289c3fSJeff Roberson else 22973f289c3fSJeff Roberson nset = cpuset_refbase(set); 22983f289c3fSJeff Roberson error = cpuset_modify_domain(nset, &domain); 22993f289c3fSJeff Roberson cpuset_rel(nset); 23003f289c3fSJeff Roberson cpuset_rel(set); 23013f289c3fSJeff Roberson break; 23023f289c3fSJeff Roberson case CPU_LEVEL_WHICH: 23033f289c3fSJeff Roberson switch (which) { 23043f289c3fSJeff Roberson case CPU_WHICH_TID: 23053f289c3fSJeff Roberson error = _cpuset_setthread(id, NULL, &domain); 23063f289c3fSJeff Roberson break; 23073f289c3fSJeff Roberson case CPU_WHICH_PID: 2308*d431dea5SKyle Evans error = cpuset_setproc(id, NULL, NULL, &domain, false); 23093f289c3fSJeff Roberson break; 23103f289c3fSJeff Roberson case CPU_WHICH_CPUSET: 23113f289c3fSJeff Roberson case CPU_WHICH_JAIL: 23123f289c3fSJeff Roberson error = cpuset_which(which, id, &p, &ttd, &set); 23133f289c3fSJeff Roberson if (error == 0) { 23143f289c3fSJeff Roberson error = cpuset_modify_domain(set, &domain); 23153f289c3fSJeff Roberson cpuset_rel(set); 23163f289c3fSJeff Roberson } 23173f289c3fSJeff Roberson break; 23183f289c3fSJeff Roberson case CPU_WHICH_IRQ: 23193f289c3fSJeff Roberson case CPU_WHICH_INTRHANDLER: 23203f289c3fSJeff Roberson case CPU_WHICH_ITHREAD: 23213f289c3fSJeff Roberson default: 23223f289c3fSJeff Roberson error = EINVAL; 23233f289c3fSJeff Roberson break; 23243f289c3fSJeff Roberson } 23253f289c3fSJeff Roberson break; 23263f289c3fSJeff Roberson default: 23273f289c3fSJeff Roberson error = EINVAL; 23283f289c3fSJeff Roberson break; 23293f289c3fSJeff Roberson } 23303f289c3fSJeff Roberson out: 23313f289c3fSJeff Roberson free(mask, M_TEMP); 23323f289c3fSJeff Roberson return (error); 23333f289c3fSJeff Roberson } 23343f289c3fSJeff Roberson 23353f289c3fSJeff Roberson #ifdef DDB 2336e5818a53SJeff Roberson 23373f289c3fSJeff Roberson static void 23383f289c3fSJeff Roberson ddb_display_bitset(const struct bitset *set, int size) 23393f289c3fSJeff Roberson { 23403f289c3fSJeff Roberson int bit, once; 23413f289c3fSJeff Roberson 23423f289c3fSJeff Roberson for (once = 0, bit = 0; bit < size; bit++) { 23433f289c3fSJeff Roberson if (CPU_ISSET(bit, set)) { 2344dea0ed66SBjoern A. Zeeb if (once == 0) { 23453f289c3fSJeff Roberson db_printf("%d", bit); 2346dea0ed66SBjoern A. Zeeb once = 1; 2347dea0ed66SBjoern A. Zeeb } else 23483f289c3fSJeff Roberson db_printf(",%d", bit); 2349dea0ed66SBjoern A. Zeeb } 2350dea0ed66SBjoern A. Zeeb } 2351cd32bd7aSJohn Baldwin if (once == 0) 2352cd32bd7aSJohn Baldwin db_printf("<none>"); 2353cd32bd7aSJohn Baldwin } 2354cd32bd7aSJohn Baldwin 23553f289c3fSJeff Roberson void 23563f289c3fSJeff Roberson ddb_display_cpuset(const cpuset_t *set) 23573f289c3fSJeff Roberson { 23583f289c3fSJeff Roberson ddb_display_bitset((const struct bitset *)set, CPU_SETSIZE); 23593f289c3fSJeff Roberson } 23603f289c3fSJeff Roberson 23613f289c3fSJeff Roberson static void 23623f289c3fSJeff Roberson ddb_display_domainset(const domainset_t *set) 23633f289c3fSJeff Roberson { 23643f289c3fSJeff Roberson ddb_display_bitset((const struct bitset *)set, DOMAINSET_SETSIZE); 23653f289c3fSJeff Roberson } 23663f289c3fSJeff Roberson 2367cd32bd7aSJohn Baldwin DB_SHOW_COMMAND(cpusets, db_show_cpusets) 2368cd32bd7aSJohn Baldwin { 2369cd32bd7aSJohn Baldwin struct cpuset *set; 2370cd32bd7aSJohn Baldwin 2371cd32bd7aSJohn Baldwin LIST_FOREACH(set, &cpuset_ids, cs_link) { 2372cd32bd7aSJohn Baldwin db_printf("set=%p id=%-6u ref=%-6d flags=0x%04x parent id=%d\n", 23731a7bb896SMateusz Guzik set, set->cs_id, refcount_load(&set->cs_ref), set->cs_flags, 2374cd32bd7aSJohn Baldwin (set->cs_parent != NULL) ? set->cs_parent->cs_id : 0); 23753f289c3fSJeff Roberson db_printf(" cpu mask="); 2376cd32bd7aSJohn Baldwin ddb_display_cpuset(&set->cs_mask); 2377dea0ed66SBjoern A. Zeeb db_printf("\n"); 23783f289c3fSJeff Roberson db_printf(" domain policy %d prefer %d mask=", 23793f289c3fSJeff Roberson set->cs_domain->ds_policy, set->cs_domain->ds_prefer); 23803f289c3fSJeff Roberson ddb_display_domainset(&set->cs_domain->ds_mask); 23813f289c3fSJeff Roberson db_printf("\n"); 2382dea0ed66SBjoern A. Zeeb if (db_pager_quit) 2383dea0ed66SBjoern A. Zeeb break; 2384dea0ed66SBjoern A. Zeeb } 2385dea0ed66SBjoern A. Zeeb } 23863f289c3fSJeff Roberson 23873f289c3fSJeff Roberson DB_SHOW_COMMAND(domainsets, db_show_domainsets) 23883f289c3fSJeff Roberson { 23893f289c3fSJeff Roberson struct domainset *set; 23903f289c3fSJeff Roberson 23913f289c3fSJeff Roberson LIST_FOREACH(set, &cpuset_domains, ds_link) { 2392e5818a53SJeff Roberson db_printf("set=%p policy %d prefer %d cnt %d\n", 2393e5818a53SJeff Roberson set, set->ds_policy, set->ds_prefer, set->ds_cnt); 23943f289c3fSJeff Roberson db_printf(" mask ="); 23953f289c3fSJeff Roberson ddb_display_domainset(&set->ds_mask); 23963f289c3fSJeff Roberson db_printf("\n"); 23973f289c3fSJeff Roberson } 23983f289c3fSJeff Roberson } 2399dea0ed66SBjoern A. Zeeb #endif /* DDB */ 2400