cpumask.h 32 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067
  1. #ifndef __LINUX_CPUMASK_H
  2. #define __LINUX_CPUMASK_H
  3. /*
  4. * Cpumasks provide a bitmap suitable for representing the
  5. * set of CPU's in a system, one bit position per CPU number.
  6. *
  7. * The new cpumask_ ops take a "struct cpumask *"; the old ones
  8. * use cpumask_t.
  9. *
  10. * See detailed comments in the file linux/bitmap.h describing the
  11. * data type on which these cpumasks are based.
  12. *
  13. * For details of cpumask_scnprintf() and cpumask_parse_user(),
  14. * see bitmap_scnprintf() and bitmap_parse_user() in lib/bitmap.c.
  15. * For details of cpulist_scnprintf() and cpulist_parse(), see
  16. * bitmap_scnlistprintf() and bitmap_parselist(), also in bitmap.c.
  17. * For details of cpu_remap(), see bitmap_bitremap in lib/bitmap.c
  18. * For details of cpus_remap(), see bitmap_remap in lib/bitmap.c.
  19. * For details of cpus_onto(), see bitmap_onto in lib/bitmap.c.
  20. * For details of cpus_fold(), see bitmap_fold in lib/bitmap.c.
  21. *
  22. * . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .
  23. * Note: The alternate operations with the suffix "_nr" are used
  24. * to limit the range of the loop to nr_cpu_ids instead of
  25. * NR_CPUS when NR_CPUS > 64 for performance reasons.
  26. * If NR_CPUS is <= 64 then most assembler bitmask
  27. * operators execute faster with a constant range, so
  28. * the operator will continue to use NR_CPUS.
  29. *
  30. * Another consideration is that nr_cpu_ids is initialized
  31. * to NR_CPUS and isn't lowered until the possible cpus are
  32. * discovered (including any disabled cpus). So early uses
  33. * will span the entire range of NR_CPUS.
  34. * . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .
  35. *
  36. * The obsolescent cpumask operations are:
  37. *
  38. * void cpu_set(cpu, mask) turn on bit 'cpu' in mask
  39. * void cpu_clear(cpu, mask) turn off bit 'cpu' in mask
  40. * void cpus_setall(mask) set all bits
  41. * void cpus_clear(mask) clear all bits
  42. * int cpu_isset(cpu, mask) true iff bit 'cpu' set in mask
  43. * int cpu_test_and_set(cpu, mask) test and set bit 'cpu' in mask
  44. *
  45. * void cpus_and(dst, src1, src2) dst = src1 & src2 [intersection]
  46. * void cpus_or(dst, src1, src2) dst = src1 | src2 [union]
  47. * void cpus_xor(dst, src1, src2) dst = src1 ^ src2
  48. * void cpus_andnot(dst, src1, src2) dst = src1 & ~src2
  49. * void cpus_complement(dst, src) dst = ~src
  50. *
  51. * int cpus_equal(mask1, mask2) Does mask1 == mask2?
  52. * int cpus_intersects(mask1, mask2) Do mask1 and mask2 intersect?
  53. * int cpus_subset(mask1, mask2) Is mask1 a subset of mask2?
  54. * int cpus_empty(mask) Is mask empty (no bits sets)?
  55. * int cpus_full(mask) Is mask full (all bits sets)?
  56. * int cpus_weight(mask) Hamming weigh - number of set bits
  57. * int cpus_weight_nr(mask) Same using nr_cpu_ids instead of NR_CPUS
  58. *
  59. * void cpus_shift_right(dst, src, n) Shift right
  60. * void cpus_shift_left(dst, src, n) Shift left
  61. *
  62. * int first_cpu(mask) Number lowest set bit, or NR_CPUS
  63. * int next_cpu(cpu, mask) Next cpu past 'cpu', or NR_CPUS
  64. * int next_cpu_nr(cpu, mask) Next cpu past 'cpu', or nr_cpu_ids
  65. *
  66. * cpumask_t cpumask_of_cpu(cpu) Return cpumask with bit 'cpu' set
  67. * (can be used as an lvalue)
  68. * CPU_MASK_ALL Initializer - all bits set
  69. * CPU_MASK_NONE Initializer - no bits set
  70. * unsigned long *cpus_addr(mask) Array of unsigned long's in mask
  71. *
  72. * CPUMASK_ALLOC kmalloc's a structure that is a composite of many cpumask_t
  73. * variables, and CPUMASK_PTR provides pointers to each field.
  74. *
  75. * The structure should be defined something like this:
  76. * struct my_cpumasks {
  77. * cpumask_t mask1;
  78. * cpumask_t mask2;
  79. * };
  80. *
  81. * Usage is then:
  82. * CPUMASK_ALLOC(my_cpumasks);
  83. * CPUMASK_PTR(mask1, my_cpumasks);
  84. * CPUMASK_PTR(mask2, my_cpumasks);
  85. *
  86. * --- DO NOT reference cpumask_t pointers until this check ---
  87. * if (my_cpumasks == NULL)
  88. * "kmalloc failed"...
  89. *
  90. * References are now pointers to the cpumask_t variables (*mask1, ...)
  91. *
  92. *if NR_CPUS > BITS_PER_LONG
  93. * CPUMASK_ALLOC(m) Declares and allocates struct m *m =
  94. * kmalloc(sizeof(*m), GFP_KERNEL)
  95. * CPUMASK_FREE(m) Macro for kfree(m)
  96. *else
  97. * CPUMASK_ALLOC(m) Declares struct m _m, *m = &_m
  98. * CPUMASK_FREE(m) Nop
  99. *endif
  100. * CPUMASK_PTR(v, m) Declares cpumask_t *v = &(m->v)
  101. * ------------------------------------------------------------------------
  102. *
  103. * int cpumask_scnprintf(buf, len, mask) Format cpumask for printing
  104. * int cpumask_parse_user(ubuf, ulen, mask) Parse ascii string as cpumask
  105. * int cpulist_scnprintf(buf, len, mask) Format cpumask as list for printing
  106. * int cpulist_parse(buf, map) Parse ascii string as cpulist
  107. * int cpu_remap(oldbit, old, new) newbit = map(old, new)(oldbit)
  108. * void cpus_remap(dst, src, old, new) *dst = map(old, new)(src)
  109. * void cpus_onto(dst, orig, relmap) *dst = orig relative to relmap
  110. * void cpus_fold(dst, orig, sz) dst bits = orig bits mod sz
  111. *
  112. * for_each_cpu_mask(cpu, mask) for-loop cpu over mask using NR_CPUS
  113. * for_each_cpu_mask_nr(cpu, mask) for-loop cpu over mask using nr_cpu_ids
  114. *
  115. * int num_online_cpus() Number of online CPUs
  116. * int num_possible_cpus() Number of all possible CPUs
  117. * int num_present_cpus() Number of present CPUs
  118. *
  119. * int cpu_online(cpu) Is some cpu online?
  120. * int cpu_possible(cpu) Is some cpu possible?
  121. * int cpu_present(cpu) Is some cpu present (can schedule)?
  122. *
  123. * int any_online_cpu(mask) First online cpu in mask
  124. *
  125. * for_each_possible_cpu(cpu) for-loop cpu over cpu_possible_map
  126. * for_each_online_cpu(cpu) for-loop cpu over cpu_online_map
  127. * for_each_present_cpu(cpu) for-loop cpu over cpu_present_map
  128. *
  129. * Subtlety:
  130. * 1) The 'type-checked' form of cpu_isset() causes gcc (3.3.2, anyway)
  131. * to generate slightly worse code. Note for example the additional
  132. * 40 lines of assembly code compiling the "for each possible cpu"
  133. * loops buried in the disk_stat_read() macros calls when compiling
  134. * drivers/block/genhd.c (arch i386, CONFIG_SMP=y). So use a simple
  135. * one-line #define for cpu_isset(), instead of wrapping an inline
  136. * inside a macro, the way we do the other calls.
  137. */
  138. #include <linux/kernel.h>
  139. #include <linux/threads.h>
  140. #include <linux/bitmap.h>
  141. typedef struct cpumask { DECLARE_BITMAP(bits, NR_CPUS); } cpumask_t;
  142. extern cpumask_t _unused_cpumask_arg_;
  143. #define cpu_set(cpu, dst) __cpu_set((cpu), &(dst))
  144. static inline void __cpu_set(int cpu, volatile cpumask_t *dstp)
  145. {
  146. set_bit(cpu, dstp->bits);
  147. }
  148. #define cpu_clear(cpu, dst) __cpu_clear((cpu), &(dst))
  149. static inline void __cpu_clear(int cpu, volatile cpumask_t *dstp)
  150. {
  151. clear_bit(cpu, dstp->bits);
  152. }
  153. #define cpus_setall(dst) __cpus_setall(&(dst), NR_CPUS)
  154. static inline void __cpus_setall(cpumask_t *dstp, int nbits)
  155. {
  156. bitmap_fill(dstp->bits, nbits);
  157. }
  158. #define cpus_clear(dst) __cpus_clear(&(dst), NR_CPUS)
  159. static inline void __cpus_clear(cpumask_t *dstp, int nbits)
  160. {
  161. bitmap_zero(dstp->bits, nbits);
  162. }
  163. /* No static inline type checking - see Subtlety (1) above. */
  164. #define cpu_isset(cpu, cpumask) test_bit((cpu), (cpumask).bits)
  165. #define cpu_test_and_set(cpu, cpumask) __cpu_test_and_set((cpu), &(cpumask))
  166. static inline int __cpu_test_and_set(int cpu, cpumask_t *addr)
  167. {
  168. return test_and_set_bit(cpu, addr->bits);
  169. }
  170. #define cpus_and(dst, src1, src2) __cpus_and(&(dst), &(src1), &(src2), NR_CPUS)
  171. static inline void __cpus_and(cpumask_t *dstp, const cpumask_t *src1p,
  172. const cpumask_t *src2p, int nbits)
  173. {
  174. bitmap_and(dstp->bits, src1p->bits, src2p->bits, nbits);
  175. }
  176. #define cpus_or(dst, src1, src2) __cpus_or(&(dst), &(src1), &(src2), NR_CPUS)
  177. static inline void __cpus_or(cpumask_t *dstp, const cpumask_t *src1p,
  178. const cpumask_t *src2p, int nbits)
  179. {
  180. bitmap_or(dstp->bits, src1p->bits, src2p->bits, nbits);
  181. }
  182. #define cpus_xor(dst, src1, src2) __cpus_xor(&(dst), &(src1), &(src2), NR_CPUS)
  183. static inline void __cpus_xor(cpumask_t *dstp, const cpumask_t *src1p,
  184. const cpumask_t *src2p, int nbits)
  185. {
  186. bitmap_xor(dstp->bits, src1p->bits, src2p->bits, nbits);
  187. }
  188. #define cpus_andnot(dst, src1, src2) \
  189. __cpus_andnot(&(dst), &(src1), &(src2), NR_CPUS)
  190. static inline void __cpus_andnot(cpumask_t *dstp, const cpumask_t *src1p,
  191. const cpumask_t *src2p, int nbits)
  192. {
  193. bitmap_andnot(dstp->bits, src1p->bits, src2p->bits, nbits);
  194. }
  195. #define cpus_complement(dst, src) __cpus_complement(&(dst), &(src), NR_CPUS)
  196. static inline void __cpus_complement(cpumask_t *dstp,
  197. const cpumask_t *srcp, int nbits)
  198. {
  199. bitmap_complement(dstp->bits, srcp->bits, nbits);
  200. }
  201. #define cpus_equal(src1, src2) __cpus_equal(&(src1), &(src2), NR_CPUS)
  202. static inline int __cpus_equal(const cpumask_t *src1p,
  203. const cpumask_t *src2p, int nbits)
  204. {
  205. return bitmap_equal(src1p->bits, src2p->bits, nbits);
  206. }
  207. #define cpus_intersects(src1, src2) __cpus_intersects(&(src1), &(src2), NR_CPUS)
  208. static inline int __cpus_intersects(const cpumask_t *src1p,
  209. const cpumask_t *src2p, int nbits)
  210. {
  211. return bitmap_intersects(src1p->bits, src2p->bits, nbits);
  212. }
  213. #define cpus_subset(src1, src2) __cpus_subset(&(src1), &(src2), NR_CPUS)
  214. static inline int __cpus_subset(const cpumask_t *src1p,
  215. const cpumask_t *src2p, int nbits)
  216. {
  217. return bitmap_subset(src1p->bits, src2p->bits, nbits);
  218. }
  219. #define cpus_empty(src) __cpus_empty(&(src), NR_CPUS)
  220. static inline int __cpus_empty(const cpumask_t *srcp, int nbits)
  221. {
  222. return bitmap_empty(srcp->bits, nbits);
  223. }
  224. #define cpus_full(cpumask) __cpus_full(&(cpumask), NR_CPUS)
  225. static inline int __cpus_full(const cpumask_t *srcp, int nbits)
  226. {
  227. return bitmap_full(srcp->bits, nbits);
  228. }
  229. #define cpus_weight(cpumask) __cpus_weight(&(cpumask), NR_CPUS)
  230. static inline int __cpus_weight(const cpumask_t *srcp, int nbits)
  231. {
  232. return bitmap_weight(srcp->bits, nbits);
  233. }
  234. #define cpus_shift_right(dst, src, n) \
  235. __cpus_shift_right(&(dst), &(src), (n), NR_CPUS)
  236. static inline void __cpus_shift_right(cpumask_t *dstp,
  237. const cpumask_t *srcp, int n, int nbits)
  238. {
  239. bitmap_shift_right(dstp->bits, srcp->bits, n, nbits);
  240. }
  241. #define cpus_shift_left(dst, src, n) \
  242. __cpus_shift_left(&(dst), &(src), (n), NR_CPUS)
  243. static inline void __cpus_shift_left(cpumask_t *dstp,
  244. const cpumask_t *srcp, int n, int nbits)
  245. {
  246. bitmap_shift_left(dstp->bits, srcp->bits, n, nbits);
  247. }
  248. /**
  249. * to_cpumask - convert an NR_CPUS bitmap to a struct cpumask *
  250. * @bitmap: the bitmap
  251. *
  252. * There are a few places where cpumask_var_t isn't appropriate and
  253. * static cpumasks must be used (eg. very early boot), yet we don't
  254. * expose the definition of 'struct cpumask'.
  255. *
  256. * This does the conversion, and can be used as a constant initializer.
  257. */
  258. #define to_cpumask(bitmap) \
  259. ((struct cpumask *)(1 ? (bitmap) \
  260. : (void *)sizeof(__check_is_bitmap(bitmap))))
  261. static inline int __check_is_bitmap(const unsigned long *bitmap)
  262. {
  263. return 1;
  264. }
  265. /*
  266. * Special-case data structure for "single bit set only" constant CPU masks.
  267. *
  268. * We pre-generate all the 64 (or 32) possible bit positions, with enough
  269. * padding to the left and the right, and return the constant pointer
  270. * appropriately offset.
  271. */
  272. extern const unsigned long
  273. cpu_bit_bitmap[BITS_PER_LONG+1][BITS_TO_LONGS(NR_CPUS)];
  274. static inline const struct cpumask *get_cpu_mask(unsigned int cpu)
  275. {
  276. const unsigned long *p = cpu_bit_bitmap[1 + cpu % BITS_PER_LONG];
  277. p -= cpu / BITS_PER_LONG;
  278. return to_cpumask(p);
  279. }
  280. /*
  281. * In cases where we take the address of the cpumask immediately,
  282. * gcc optimizes it out (it's a constant) and there's no huge stack
  283. * variable created:
  284. */
  285. #define cpumask_of_cpu(cpu) (*get_cpu_mask(cpu))
  286. #define CPU_MASK_LAST_WORD BITMAP_LAST_WORD_MASK(NR_CPUS)
  287. #if NR_CPUS <= BITS_PER_LONG
  288. #define CPU_MASK_ALL \
  289. (cpumask_t) { { \
  290. [BITS_TO_LONGS(NR_CPUS)-1] = CPU_MASK_LAST_WORD \
  291. } }
  292. #define CPU_MASK_ALL_PTR (&CPU_MASK_ALL)
  293. #else
  294. #define CPU_MASK_ALL \
  295. (cpumask_t) { { \
  296. [0 ... BITS_TO_LONGS(NR_CPUS)-2] = ~0UL, \
  297. [BITS_TO_LONGS(NR_CPUS)-1] = CPU_MASK_LAST_WORD \
  298. } }
  299. /* cpu_mask_all is in init/main.c */
  300. extern cpumask_t cpu_mask_all;
  301. #define CPU_MASK_ALL_PTR (&cpu_mask_all)
  302. #endif
  303. #define CPU_MASK_NONE \
  304. (cpumask_t) { { \
  305. [0 ... BITS_TO_LONGS(NR_CPUS)-1] = 0UL \
  306. } }
  307. #define CPU_MASK_CPU0 \
  308. (cpumask_t) { { \
  309. [0] = 1UL \
  310. } }
  311. #define cpus_addr(src) ((src).bits)
  312. #if NR_CPUS > BITS_PER_LONG
  313. #define CPUMASK_ALLOC(m) struct m *m = kmalloc(sizeof(*m), GFP_KERNEL)
  314. #define CPUMASK_FREE(m) kfree(m)
  315. #else
  316. #define CPUMASK_ALLOC(m) struct m _m, *m = &_m
  317. #define CPUMASK_FREE(m)
  318. #endif
  319. #define CPUMASK_PTR(v, m) cpumask_t *v = &(m->v)
  320. #define cpu_remap(oldbit, old, new) \
  321. __cpu_remap((oldbit), &(old), &(new), NR_CPUS)
  322. static inline int __cpu_remap(int oldbit,
  323. const cpumask_t *oldp, const cpumask_t *newp, int nbits)
  324. {
  325. return bitmap_bitremap(oldbit, oldp->bits, newp->bits, nbits);
  326. }
  327. #define cpus_remap(dst, src, old, new) \
  328. __cpus_remap(&(dst), &(src), &(old), &(new), NR_CPUS)
  329. static inline void __cpus_remap(cpumask_t *dstp, const cpumask_t *srcp,
  330. const cpumask_t *oldp, const cpumask_t *newp, int nbits)
  331. {
  332. bitmap_remap(dstp->bits, srcp->bits, oldp->bits, newp->bits, nbits);
  333. }
  334. #define cpus_onto(dst, orig, relmap) \
  335. __cpus_onto(&(dst), &(orig), &(relmap), NR_CPUS)
  336. static inline void __cpus_onto(cpumask_t *dstp, const cpumask_t *origp,
  337. const cpumask_t *relmapp, int nbits)
  338. {
  339. bitmap_onto(dstp->bits, origp->bits, relmapp->bits, nbits);
  340. }
  341. #define cpus_fold(dst, orig, sz) \
  342. __cpus_fold(&(dst), &(orig), sz, NR_CPUS)
  343. static inline void __cpus_fold(cpumask_t *dstp, const cpumask_t *origp,
  344. int sz, int nbits)
  345. {
  346. bitmap_fold(dstp->bits, origp->bits, sz, nbits);
  347. }
  348. #if NR_CPUS == 1
  349. #define nr_cpu_ids 1
  350. #define first_cpu(src) ({ (void)(src); 0; })
  351. #define next_cpu(n, src) ({ (void)(src); 1; })
  352. #define any_online_cpu(mask) 0
  353. #define for_each_cpu_mask(cpu, mask) \
  354. for ((cpu) = 0; (cpu) < 1; (cpu)++, (void)mask)
  355. #else /* NR_CPUS > 1 */
  356. extern int nr_cpu_ids;
  357. int __first_cpu(const cpumask_t *srcp);
  358. int __next_cpu(int n, const cpumask_t *srcp);
  359. int __any_online_cpu(const cpumask_t *mask);
  360. #define first_cpu(src) __first_cpu(&(src))
  361. #define next_cpu(n, src) __next_cpu((n), &(src))
  362. #define any_online_cpu(mask) __any_online_cpu(&(mask))
  363. #define for_each_cpu_mask(cpu, mask) \
  364. for ((cpu) = -1; \
  365. (cpu) = next_cpu((cpu), (mask)), \
  366. (cpu) < NR_CPUS; )
  367. #endif
  368. #if NR_CPUS <= 64
  369. #define next_cpu_nr(n, src) next_cpu(n, src)
  370. #define cpus_weight_nr(cpumask) cpus_weight(cpumask)
  371. #define for_each_cpu_mask_nr(cpu, mask) for_each_cpu_mask(cpu, mask)
  372. #else /* NR_CPUS > 64 */
  373. int __next_cpu_nr(int n, const cpumask_t *srcp);
  374. #define next_cpu_nr(n, src) __next_cpu_nr((n), &(src))
  375. #define cpus_weight_nr(cpumask) __cpus_weight(&(cpumask), nr_cpu_ids)
  376. #define for_each_cpu_mask_nr(cpu, mask) \
  377. for ((cpu) = -1; \
  378. (cpu) = next_cpu_nr((cpu), (mask)), \
  379. (cpu) < nr_cpu_ids; )
  380. #endif /* NR_CPUS > 64 */
  381. /*
  382. * The following particular system cpumasks and operations manage
  383. * possible, present, active and online cpus.
  384. *
  385. * cpu_possible_mask- has bit 'cpu' set iff cpu is populatable
  386. * cpu_present_mask - has bit 'cpu' set iff cpu is populated
  387. * cpu_online_mask - has bit 'cpu' set iff cpu available to scheduler
  388. * cpu_active_mask - has bit 'cpu' set iff cpu available to migration
  389. *
  390. * If !CONFIG_HOTPLUG_CPU, present == possible, and active == online.
  391. *
  392. * The cpu_possible_mask is fixed at boot time, as the set of CPU id's
  393. * that it is possible might ever be plugged in at anytime during the
  394. * life of that system boot. The cpu_present_mask is dynamic(*),
  395. * representing which CPUs are currently plugged in. And
  396. * cpu_online_mask is the dynamic subset of cpu_present_mask,
  397. * indicating those CPUs available for scheduling.
  398. *
  399. * If HOTPLUG is enabled, then cpu_possible_mask is forced to have
  400. * all NR_CPUS bits set, otherwise it is just the set of CPUs that
  401. * ACPI reports present at boot.
  402. *
  403. * If HOTPLUG is enabled, then cpu_present_mask varies dynamically,
  404. * depending on what ACPI reports as currently plugged in, otherwise
  405. * cpu_present_mask is just a copy of cpu_possible_mask.
  406. *
  407. * (*) Well, cpu_present_mask is dynamic in the hotplug case. If not
  408. * hotplug, it's a copy of cpu_possible_mask, hence fixed at boot.
  409. *
  410. * Subtleties:
  411. * 1) UP arch's (NR_CPUS == 1, CONFIG_SMP not defined) hardcode
  412. * assumption that their single CPU is online. The UP
  413. * cpu_{online,possible,present}_masks are placebos. Changing them
  414. * will have no useful affect on the following num_*_cpus()
  415. * and cpu_*() macros in the UP case. This ugliness is a UP
  416. * optimization - don't waste any instructions or memory references
  417. * asking if you're online or how many CPUs there are if there is
  418. * only one CPU.
  419. */
  420. extern const struct cpumask *const cpu_possible_mask;
  421. extern const struct cpumask *const cpu_online_mask;
  422. extern const struct cpumask *const cpu_present_mask;
  423. extern const struct cpumask *const cpu_active_mask;
  424. /* These strip const, as traditionally they weren't const. */
  425. #define cpu_possible_map (*(cpumask_t *)cpu_possible_mask)
  426. #define cpu_online_map (*(cpumask_t *)cpu_online_mask)
  427. #define cpu_present_map (*(cpumask_t *)cpu_present_mask)
  428. #define cpu_active_map (*(cpumask_t *)cpu_active_mask)
  429. #if NR_CPUS > 1
  430. #define num_online_cpus() cpumask_weight(cpu_online_mask)
  431. #define num_possible_cpus() cpumask_weight(cpu_possible_mask)
  432. #define num_present_cpus() cpumask_weight(cpu_present_mask)
  433. #define cpu_online(cpu) cpumask_test_cpu((cpu), cpu_online_mask)
  434. #define cpu_possible(cpu) cpumask_test_cpu((cpu), cpu_possible_mask)
  435. #define cpu_present(cpu) cpumask_test_cpu((cpu), cpu_present_mask)
  436. #define cpu_active(cpu) cpumask_test_cpu((cpu), cpu_active_mask)
  437. #else
  438. #define num_online_cpus() 1
  439. #define num_possible_cpus() 1
  440. #define num_present_cpus() 1
  441. #define cpu_online(cpu) ((cpu) == 0)
  442. #define cpu_possible(cpu) ((cpu) == 0)
  443. #define cpu_present(cpu) ((cpu) == 0)
  444. #define cpu_active(cpu) ((cpu) == 0)
  445. #endif
  446. #define cpu_is_offline(cpu) unlikely(!cpu_online(cpu))
  447. /* These are the new versions of the cpumask operators: passed by pointer.
  448. * The older versions will be implemented in terms of these, then deleted. */
  449. #define cpumask_bits(maskp) ((maskp)->bits)
  450. #if NR_CPUS <= BITS_PER_LONG
  451. #define CPU_BITS_ALL \
  452. { \
  453. [BITS_TO_LONGS(NR_CPUS)-1] = CPU_MASK_LAST_WORD \
  454. }
  455. #else /* NR_CPUS > BITS_PER_LONG */
  456. #define CPU_BITS_ALL \
  457. { \
  458. [0 ... BITS_TO_LONGS(NR_CPUS)-2] = ~0UL, \
  459. [BITS_TO_LONGS(NR_CPUS)-1] = CPU_MASK_LAST_WORD \
  460. }
  461. #endif /* NR_CPUS > BITS_PER_LONG */
  462. #ifdef CONFIG_CPUMASK_OFFSTACK
  463. /* Assuming NR_CPUS is huge, a runtime limit is more efficient. Also,
  464. * not all bits may be allocated. */
  465. #define nr_cpumask_bits nr_cpu_ids
  466. #else
  467. #define nr_cpumask_bits NR_CPUS
  468. #endif
  469. /* verify cpu argument to cpumask_* operators */
  470. static inline unsigned int cpumask_check(unsigned int cpu)
  471. {
  472. #ifdef CONFIG_DEBUG_PER_CPU_MAPS
  473. WARN_ON_ONCE(cpu >= nr_cpumask_bits);
  474. #endif /* CONFIG_DEBUG_PER_CPU_MAPS */
  475. return cpu;
  476. }
  477. #if NR_CPUS == 1
  478. /* Uniprocessor. Assume all masks are "1". */
  479. static inline unsigned int cpumask_first(const struct cpumask *srcp)
  480. {
  481. return 0;
  482. }
  483. /* Valid inputs for n are -1 and 0. */
  484. static inline unsigned int cpumask_next(int n, const struct cpumask *srcp)
  485. {
  486. return n+1;
  487. }
  488. static inline unsigned int cpumask_next_zero(int n, const struct cpumask *srcp)
  489. {
  490. return n+1;
  491. }
  492. static inline unsigned int cpumask_next_and(int n,
  493. const struct cpumask *srcp,
  494. const struct cpumask *andp)
  495. {
  496. return n+1;
  497. }
  498. /* cpu must be a valid cpu, ie 0, so there's no other choice. */
  499. static inline unsigned int cpumask_any_but(const struct cpumask *mask,
  500. unsigned int cpu)
  501. {
  502. return 1;
  503. }
  504. #define for_each_cpu(cpu, mask) \
  505. for ((cpu) = 0; (cpu) < 1; (cpu)++, (void)mask)
  506. #define for_each_cpu_and(cpu, mask, and) \
  507. for ((cpu) = 0; (cpu) < 1; (cpu)++, (void)mask, (void)and)
  508. #else
  509. /**
  510. * cpumask_first - get the first cpu in a cpumask
  511. * @srcp: the cpumask pointer
  512. *
  513. * Returns >= nr_cpu_ids if no cpus set.
  514. */
  515. static inline unsigned int cpumask_first(const struct cpumask *srcp)
  516. {
  517. return find_first_bit(cpumask_bits(srcp), nr_cpumask_bits);
  518. }
  519. /**
  520. * cpumask_next - get the next cpu in a cpumask
  521. * @n: the cpu prior to the place to search (ie. return will be > @n)
  522. * @srcp: the cpumask pointer
  523. *
  524. * Returns >= nr_cpu_ids if no further cpus set.
  525. */
  526. static inline unsigned int cpumask_next(int n, const struct cpumask *srcp)
  527. {
  528. /* -1 is a legal arg here. */
  529. if (n != -1)
  530. cpumask_check(n);
  531. return find_next_bit(cpumask_bits(srcp), nr_cpumask_bits, n+1);
  532. }
  533. /**
  534. * cpumask_next_zero - get the next unset cpu in a cpumask
  535. * @n: the cpu prior to the place to search (ie. return will be > @n)
  536. * @srcp: the cpumask pointer
  537. *
  538. * Returns >= nr_cpu_ids if no further cpus unset.
  539. */
  540. static inline unsigned int cpumask_next_zero(int n, const struct cpumask *srcp)
  541. {
  542. /* -1 is a legal arg here. */
  543. if (n != -1)
  544. cpumask_check(n);
  545. return find_next_zero_bit(cpumask_bits(srcp), nr_cpumask_bits, n+1);
  546. }
  547. int cpumask_next_and(int n, const struct cpumask *, const struct cpumask *);
  548. int cpumask_any_but(const struct cpumask *mask, unsigned int cpu);
  549. /**
  550. * for_each_cpu - iterate over every cpu in a mask
  551. * @cpu: the (optionally unsigned) integer iterator
  552. * @mask: the cpumask pointer
  553. *
  554. * After the loop, cpu is >= nr_cpu_ids.
  555. */
  556. #define for_each_cpu(cpu, mask) \
  557. for ((cpu) = -1; \
  558. (cpu) = cpumask_next((cpu), (mask)), \
  559. (cpu) < nr_cpu_ids;)
  560. /**
  561. * for_each_cpu_and - iterate over every cpu in both masks
  562. * @cpu: the (optionally unsigned) integer iterator
  563. * @mask: the first cpumask pointer
  564. * @and: the second cpumask pointer
  565. *
  566. * This saves a temporary CPU mask in many places. It is equivalent to:
  567. * struct cpumask tmp;
  568. * cpumask_and(&tmp, &mask, &and);
  569. * for_each_cpu(cpu, &tmp)
  570. * ...
  571. *
  572. * After the loop, cpu is >= nr_cpu_ids.
  573. */
  574. #define for_each_cpu_and(cpu, mask, and) \
  575. for ((cpu) = -1; \
  576. (cpu) = cpumask_next_and((cpu), (mask), (and)), \
  577. (cpu) < nr_cpu_ids;)
  578. #endif /* SMP */
  579. #define CPU_BITS_NONE \
  580. { \
  581. [0 ... BITS_TO_LONGS(NR_CPUS)-1] = 0UL \
  582. }
  583. #define CPU_BITS_CPU0 \
  584. { \
  585. [0] = 1UL \
  586. }
  587. /**
  588. * cpumask_set_cpu - set a cpu in a cpumask
  589. * @cpu: cpu number (< nr_cpu_ids)
  590. * @dstp: the cpumask pointer
  591. */
  592. static inline void cpumask_set_cpu(unsigned int cpu, struct cpumask *dstp)
  593. {
  594. set_bit(cpumask_check(cpu), cpumask_bits(dstp));
  595. }
  596. /**
  597. * cpumask_clear_cpu - clear a cpu in a cpumask
  598. * @cpu: cpu number (< nr_cpu_ids)
  599. * @dstp: the cpumask pointer
  600. */
  601. static inline void cpumask_clear_cpu(int cpu, struct cpumask *dstp)
  602. {
  603. clear_bit(cpumask_check(cpu), cpumask_bits(dstp));
  604. }
  605. /**
  606. * cpumask_test_cpu - test for a cpu in a cpumask
  607. * @cpu: cpu number (< nr_cpu_ids)
  608. * @cpumask: the cpumask pointer
  609. *
  610. * No static inline type checking - see Subtlety (1) above.
  611. */
  612. #define cpumask_test_cpu(cpu, cpumask) \
  613. test_bit(cpumask_check(cpu), cpumask_bits((cpumask)))
  614. /**
  615. * cpumask_test_and_set_cpu - atomically test and set a cpu in a cpumask
  616. * @cpu: cpu number (< nr_cpu_ids)
  617. * @cpumask: the cpumask pointer
  618. *
  619. * test_and_set_bit wrapper for cpumasks.
  620. */
  621. static inline int cpumask_test_and_set_cpu(int cpu, struct cpumask *cpumask)
  622. {
  623. return test_and_set_bit(cpumask_check(cpu), cpumask_bits(cpumask));
  624. }
  625. /**
  626. * cpumask_setall - set all cpus (< nr_cpu_ids) in a cpumask
  627. * @dstp: the cpumask pointer
  628. */
  629. static inline void cpumask_setall(struct cpumask *dstp)
  630. {
  631. bitmap_fill(cpumask_bits(dstp), nr_cpumask_bits);
  632. }
  633. /**
  634. * cpumask_clear - clear all cpus (< nr_cpu_ids) in a cpumask
  635. * @dstp: the cpumask pointer
  636. */
  637. static inline void cpumask_clear(struct cpumask *dstp)
  638. {
  639. bitmap_zero(cpumask_bits(dstp), nr_cpumask_bits);
  640. }
  641. /**
  642. * cpumask_and - *dstp = *src1p & *src2p
  643. * @dstp: the cpumask result
  644. * @src1p: the first input
  645. * @src2p: the second input
  646. */
  647. static inline void cpumask_and(struct cpumask *dstp,
  648. const struct cpumask *src1p,
  649. const struct cpumask *src2p)
  650. {
  651. bitmap_and(cpumask_bits(dstp), cpumask_bits(src1p),
  652. cpumask_bits(src2p), nr_cpumask_bits);
  653. }
  654. /**
  655. * cpumask_or - *dstp = *src1p | *src2p
  656. * @dstp: the cpumask result
  657. * @src1p: the first input
  658. * @src2p: the second input
  659. */
  660. static inline void cpumask_or(struct cpumask *dstp, const struct cpumask *src1p,
  661. const struct cpumask *src2p)
  662. {
  663. bitmap_or(cpumask_bits(dstp), cpumask_bits(src1p),
  664. cpumask_bits(src2p), nr_cpumask_bits);
  665. }
  666. /**
  667. * cpumask_xor - *dstp = *src1p ^ *src2p
  668. * @dstp: the cpumask result
  669. * @src1p: the first input
  670. * @src2p: the second input
  671. */
  672. static inline void cpumask_xor(struct cpumask *dstp,
  673. const struct cpumask *src1p,
  674. const struct cpumask *src2p)
  675. {
  676. bitmap_xor(cpumask_bits(dstp), cpumask_bits(src1p),
  677. cpumask_bits(src2p), nr_cpumask_bits);
  678. }
  679. /**
  680. * cpumask_andnot - *dstp = *src1p & ~*src2p
  681. * @dstp: the cpumask result
  682. * @src1p: the first input
  683. * @src2p: the second input
  684. */
  685. static inline void cpumask_andnot(struct cpumask *dstp,
  686. const struct cpumask *src1p,
  687. const struct cpumask *src2p)
  688. {
  689. bitmap_andnot(cpumask_bits(dstp), cpumask_bits(src1p),
  690. cpumask_bits(src2p), nr_cpumask_bits);
  691. }
  692. /**
  693. * cpumask_complement - *dstp = ~*srcp
  694. * @dstp: the cpumask result
  695. * @srcp: the input to invert
  696. */
  697. static inline void cpumask_complement(struct cpumask *dstp,
  698. const struct cpumask *srcp)
  699. {
  700. bitmap_complement(cpumask_bits(dstp), cpumask_bits(srcp),
  701. nr_cpumask_bits);
  702. }
  703. /**
  704. * cpumask_equal - *src1p == *src2p
  705. * @src1p: the first input
  706. * @src2p: the second input
  707. */
  708. static inline bool cpumask_equal(const struct cpumask *src1p,
  709. const struct cpumask *src2p)
  710. {
  711. return bitmap_equal(cpumask_bits(src1p), cpumask_bits(src2p),
  712. nr_cpumask_bits);
  713. }
  714. /**
  715. * cpumask_intersects - (*src1p & *src2p) != 0
  716. * @src1p: the first input
  717. * @src2p: the second input
  718. */
  719. static inline bool cpumask_intersects(const struct cpumask *src1p,
  720. const struct cpumask *src2p)
  721. {
  722. return bitmap_intersects(cpumask_bits(src1p), cpumask_bits(src2p),
  723. nr_cpumask_bits);
  724. }
  725. /**
  726. * cpumask_subset - (*src1p & ~*src2p) == 0
  727. * @src1p: the first input
  728. * @src2p: the second input
  729. */
  730. static inline int cpumask_subset(const struct cpumask *src1p,
  731. const struct cpumask *src2p)
  732. {
  733. return bitmap_subset(cpumask_bits(src1p), cpumask_bits(src2p),
  734. nr_cpumask_bits);
  735. }
  736. /**
  737. * cpumask_empty - *srcp == 0
  738. * @srcp: the cpumask to that all cpus < nr_cpu_ids are clear.
  739. */
  740. static inline bool cpumask_empty(const struct cpumask *srcp)
  741. {
  742. return bitmap_empty(cpumask_bits(srcp), nr_cpumask_bits);
  743. }
  744. /**
  745. * cpumask_full - *srcp == 0xFFFFFFFF...
  746. * @srcp: the cpumask to that all cpus < nr_cpu_ids are set.
  747. */
  748. static inline bool cpumask_full(const struct cpumask *srcp)
  749. {
  750. return bitmap_full(cpumask_bits(srcp), nr_cpumask_bits);
  751. }
  752. /**
  753. * cpumask_weight - Count of bits in *srcp
  754. * @srcp: the cpumask to count bits (< nr_cpu_ids) in.
  755. */
  756. static inline unsigned int cpumask_weight(const struct cpumask *srcp)
  757. {
  758. return bitmap_weight(cpumask_bits(srcp), nr_cpumask_bits);
  759. }
  760. /**
  761. * cpumask_shift_right - *dstp = *srcp >> n
  762. * @dstp: the cpumask result
  763. * @srcp: the input to shift
  764. * @n: the number of bits to shift by
  765. */
  766. static inline void cpumask_shift_right(struct cpumask *dstp,
  767. const struct cpumask *srcp, int n)
  768. {
  769. bitmap_shift_right(cpumask_bits(dstp), cpumask_bits(srcp), n,
  770. nr_cpumask_bits);
  771. }
  772. /**
  773. * cpumask_shift_left - *dstp = *srcp << n
  774. * @dstp: the cpumask result
  775. * @srcp: the input to shift
  776. * @n: the number of bits to shift by
  777. */
  778. static inline void cpumask_shift_left(struct cpumask *dstp,
  779. const struct cpumask *srcp, int n)
  780. {
  781. bitmap_shift_left(cpumask_bits(dstp), cpumask_bits(srcp), n,
  782. nr_cpumask_bits);
  783. }
  784. /**
  785. * cpumask_copy - *dstp = *srcp
  786. * @dstp: the result
  787. * @srcp: the input cpumask
  788. */
  789. static inline void cpumask_copy(struct cpumask *dstp,
  790. const struct cpumask *srcp)
  791. {
  792. bitmap_copy(cpumask_bits(dstp), cpumask_bits(srcp), nr_cpumask_bits);
  793. }
  794. /**
  795. * cpumask_any - pick a "random" cpu from *srcp
  796. * @srcp: the input cpumask
  797. *
  798. * Returns >= nr_cpu_ids if no cpus set.
  799. */
  800. #define cpumask_any(srcp) cpumask_first(srcp)
  801. /**
  802. * cpumask_first_and - return the first cpu from *srcp1 & *srcp2
  803. * @src1p: the first input
  804. * @src2p: the second input
  805. *
  806. * Returns >= nr_cpu_ids if no cpus set in both. See also cpumask_next_and().
  807. */
  808. #define cpumask_first_and(src1p, src2p) cpumask_next_and(-1, (src1p), (src2p))
  809. /**
  810. * cpumask_any_and - pick a "random" cpu from *mask1 & *mask2
  811. * @mask1: the first input cpumask
  812. * @mask2: the second input cpumask
  813. *
  814. * Returns >= nr_cpu_ids if no cpus set.
  815. */
  816. #define cpumask_any_and(mask1, mask2) cpumask_first_and((mask1), (mask2))
  817. /**
  818. * cpumask_of - the cpumask containing just a given cpu
  819. * @cpu: the cpu (<= nr_cpu_ids)
  820. */
  821. #define cpumask_of(cpu) (get_cpu_mask(cpu))
  822. /**
  823. * cpumask_scnprintf - print a cpumask into a string as comma-separated hex
  824. * @buf: the buffer to sprintf into
  825. * @len: the length of the buffer
  826. * @srcp: the cpumask to print
  827. *
  828. * If len is zero, returns zero. Otherwise returns the length of the
  829. * (nul-terminated) @buf string.
  830. */
  831. static inline int cpumask_scnprintf(char *buf, int len,
  832. const struct cpumask *srcp)
  833. {
  834. return bitmap_scnprintf(buf, len, cpumask_bits(srcp), nr_cpumask_bits);
  835. }
  836. /**
  837. * cpumask_parse_user - extract a cpumask from a user string
  838. * @buf: the buffer to extract from
  839. * @len: the length of the buffer
  840. * @dstp: the cpumask to set.
  841. *
  842. * Returns -errno, or 0 for success.
  843. */
  844. static inline int cpumask_parse_user(const char __user *buf, int len,
  845. struct cpumask *dstp)
  846. {
  847. return bitmap_parse_user(buf, len, cpumask_bits(dstp), nr_cpumask_bits);
  848. }
  849. /**
  850. * cpulist_scnprintf - print a cpumask into a string as comma-separated list
  851. * @buf: the buffer to sprintf into
  852. * @len: the length of the buffer
  853. * @srcp: the cpumask to print
  854. *
  855. * If len is zero, returns zero. Otherwise returns the length of the
  856. * (nul-terminated) @buf string.
  857. */
  858. static inline int cpulist_scnprintf(char *buf, int len,
  859. const struct cpumask *srcp)
  860. {
  861. return bitmap_scnlistprintf(buf, len, cpumask_bits(srcp),
  862. nr_cpumask_bits);
  863. }
  864. /**
  865. * cpulist_parse_user - extract a cpumask from a user string of ranges
  866. * @buf: the buffer to extract from
  867. * @len: the length of the buffer
  868. * @dstp: the cpumask to set.
  869. *
  870. * Returns -errno, or 0 for success.
  871. */
  872. static inline int cpulist_parse(const char *buf, struct cpumask *dstp)
  873. {
  874. return bitmap_parselist(buf, cpumask_bits(dstp), nr_cpumask_bits);
  875. }
  876. /**
  877. * cpumask_size - size to allocate for a 'struct cpumask' in bytes
  878. *
  879. * This will eventually be a runtime variable, depending on nr_cpu_ids.
  880. */
  881. static inline size_t cpumask_size(void)
  882. {
  883. /* FIXME: Once all cpumask assignments are eliminated, this
  884. * can be nr_cpumask_bits */
  885. return BITS_TO_LONGS(NR_CPUS) * sizeof(long);
  886. }
  887. /*
  888. * cpumask_var_t: struct cpumask for stack usage.
  889. *
  890. * Oh, the wicked games we play! In order to make kernel coding a
  891. * little more difficult, we typedef cpumask_var_t to an array or a
  892. * pointer: doing &mask on an array is a noop, so it still works.
  893. *
  894. * ie.
  895. * cpumask_var_t tmpmask;
  896. * if (!alloc_cpumask_var(&tmpmask, GFP_KERNEL))
  897. * return -ENOMEM;
  898. *
  899. * ... use 'tmpmask' like a normal struct cpumask * ...
  900. *
  901. * free_cpumask_var(tmpmask);
  902. */
  903. #ifdef CONFIG_CPUMASK_OFFSTACK
  904. typedef struct cpumask *cpumask_var_t;
  905. bool alloc_cpumask_var_node(cpumask_var_t *mask, gfp_t flags, int node);
  906. bool alloc_cpumask_var(cpumask_var_t *mask, gfp_t flags);
  907. void alloc_bootmem_cpumask_var(cpumask_var_t *mask);
  908. void free_cpumask_var(cpumask_var_t mask);
  909. void free_bootmem_cpumask_var(cpumask_var_t mask);
  910. #else
  911. typedef struct cpumask cpumask_var_t[1];
  912. static inline bool alloc_cpumask_var(cpumask_var_t *mask, gfp_t flags)
  913. {
  914. return true;
  915. }
  916. static inline bool alloc_cpumask_var_node(cpumask_var_t *mask, gfp_t flags,
  917. int node)
  918. {
  919. return true;
  920. }
  921. static inline void alloc_bootmem_cpumask_var(cpumask_var_t *mask)
  922. {
  923. }
  924. static inline void free_cpumask_var(cpumask_var_t mask)
  925. {
  926. }
  927. static inline void free_bootmem_cpumask_var(cpumask_var_t mask)
  928. {
  929. }
  930. #endif /* CONFIG_CPUMASK_OFFSTACK */
  931. /* It's common to want to use cpu_all_mask in struct member initializers,
  932. * so it has to refer to an address rather than a pointer. */
  933. extern const DECLARE_BITMAP(cpu_all_bits, NR_CPUS);
  934. #define cpu_all_mask to_cpumask(cpu_all_bits)
  935. /* First bits of cpu_bit_bitmap are in fact unset. */
  936. #define cpu_none_mask to_cpumask(cpu_bit_bitmap[0])
  937. #define for_each_possible_cpu(cpu) for_each_cpu((cpu), cpu_possible_mask)
  938. #define for_each_online_cpu(cpu) for_each_cpu((cpu), cpu_online_mask)
  939. #define for_each_present_cpu(cpu) for_each_cpu((cpu), cpu_present_mask)
  940. /* Wrappers for arch boot code to manipulate normally-constant masks */
  941. void set_cpu_possible(unsigned int cpu, bool possible);
  942. void set_cpu_present(unsigned int cpu, bool present);
  943. void set_cpu_online(unsigned int cpu, bool online);
  944. void set_cpu_active(unsigned int cpu, bool active);
  945. void init_cpu_present(const struct cpumask *src);
  946. void init_cpu_possible(const struct cpumask *src);
  947. void init_cpu_online(const struct cpumask *src);
  948. #endif /* __LINUX_CPUMASK_H */