^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) #include <linux/nodemask.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) #include <linux/module.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) #include <linux/random.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) int __next_node_in(int node, const nodemask_t *srcp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) int ret = __next_node(node, srcp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) if (ret == MAX_NUMNODES)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) ret = __first_node(srcp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) return ret;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) EXPORT_SYMBOL(__next_node_in);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) #ifdef CONFIG_NUMA
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) * Return the bit number of a random bit set in the nodemask.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) * (returns NUMA_NO_NODE if nodemask is empty)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) int node_random(const nodemask_t *maskp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) int w, bit = NUMA_NO_NODE;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) w = nodes_weight(*maskp);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) if (w)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) bit = bitmap_ord_to_pos(maskp->bits,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) get_random_int() % w, MAX_NUMNODES);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29) return bit;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31) #endif