[thirdparty/util-linux.git] / lib / cpuset.c

/*
 * SPDX-License-Identifier: LGPL-2.1-or-later
 *
 * Terminology:
 *
 *	cpuset	- (libc) cpu_set_t data structure represents set of CPUs
 *	cpumask	- string with hex mask (e.g. "0x00000001")
 *	cpulist - string with CPU ranges (e.g. "0-3,5,7,8")
 *
 * Based on code from taskset.c and Linux kernel.
 *
 * This file may be redistributed under the terms of the
 * GNU Lesser General Public License.
 *
 * Copyright (C) 2010 Karel Zak <kzak@redhat.com>
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sched.h>
#include <errno.h>
#include <string.h>
#include <ctype.h>
#ifdef HAVE_SYS_SYSCALL_H
#include <sys/syscall.h>
#endif

#include "cpuset.h"
#include "c.h"

static inline int val_to_char(int v)
{
	if (v >= 0 && v < 10)
		return '0' + v;
	if (v >= 10 && v < 16)
		return ('a' - 10) + v;
	return -1;
}

static inline int char_to_val(int c)
{
	int cl;

	if (c >= '0' && c <= '9')
		return c - '0';
	cl = tolower(c);
	if (cl >= 'a' && cl <= 'f')
		return cl + (10 - 'a');
	return -1;
}

static const char *nexttoken(const char *q,  int sep)
{
	if (q)
		q = strchr(q, sep);
	if (q)
		q++;
	return q;
}

/*
 * Number of bits in a CPU bitmask on current system
 */
int get_max_number_of_cpus(void)
{
#ifdef SYS_sched_getaffinity
	int n, cpus = 2048;
	size_t setsize;
	cpu_set_t *set = cpuset_alloc(cpus, &setsize, NULL);

	if (!set)
		return -1;	/* error */

	for (;;) {
		CPU_ZERO_S(setsize, set);

		/* the library version does not return size of cpumask_t */
		n = syscall(SYS_sched_getaffinity, 0, setsize, set);

		if (n < 0 && errno == EINVAL && cpus < 1024 * 1024) {
			cpuset_free(set);
			cpus *= 2;
			set = cpuset_alloc(cpus, &setsize, NULL);
			if (!set)
				return -1;	/* error */
			continue;
		}
		cpuset_free(set);
		return n * 8;
	}
#endif
	return -1;
}

/*
 * Allocates a new set for ncpus and returns size in bytes and size in bits
 */
cpu_set_t *cpuset_alloc(int ncpus, size_t *setsize, size_t *nbits)
{
	cpu_set_t *set = CPU_ALLOC(ncpus);

	if (!set)
		return NULL;
	if (setsize)
		*setsize = CPU_ALLOC_SIZE(ncpus);
	if (nbits)
		*nbits = cpuset_nbits(CPU_ALLOC_SIZE(ncpus));
	return set;
}

void cpuset_free(cpu_set_t *set)
{
	CPU_FREE(set);
}

#if !HAVE_DECL_CPU_ALLOC
/* Please, use CPU_COUNT_S() macro. This is fallback */
int __cpuset_count_s(size_t setsize, const cpu_set_t *set)
{
	int s = 0;
	const __cpu_mask *p = set->__bits;
	const __cpu_mask *end = &set->__bits[setsize / sizeof (__cpu_mask)];

	while (p < end) {
		__cpu_mask l = *p++;

		if (l == 0)
			continue;
# if LONG_BIT > 32
		l = (l & 0x5555555555555555ul) + ((l >> 1) & 0x5555555555555555ul);
		l = (l & 0x3333333333333333ul) + ((l >> 2) & 0x3333333333333333ul);
		l = (l & 0x0f0f0f0f0f0f0f0ful) + ((l >> 4) & 0x0f0f0f0f0f0f0f0ful);
		l = (l & 0x00ff00ff00ff00fful) + ((l >> 8) & 0x00ff00ff00ff00fful);
		l = (l & 0x0000ffff0000fffful) + ((l >> 16) & 0x0000ffff0000fffful);
		l = (l & 0x00000000fffffffful) + ((l >> 32) & 0x00000000fffffffful);
# else
		l = (l & 0x55555555ul) + ((l >> 1) & 0x55555555ul);
		l = (l & 0x33333333ul) + ((l >> 2) & 0x33333333ul);
		l = (l & 0x0f0f0f0ful) + ((l >> 4) & 0x0f0f0f0ful);
		l = (l & 0x00ff00fful) + ((l >> 8) & 0x00ff00fful);
		l = (l & 0x0000fffful) + ((l >> 16) & 0x0000fffful);
# endif
		s += l;
	}
	return s;
}
#endif

/*
 * Returns human readable representation of the cpuset. The output format is
 * a list of CPUs with ranges (for example, "0,1,3-9").
 */
char *cpulist_create(char *str, size_t len,
			cpu_set_t *set, size_t setsize)
{
	size_t i;
	char *ptr = str;
	int entry_made = 0;
	size_t max = cpuset_nbits(setsize);

	for (i = 0; i < max; i++) {
		if (CPU_ISSET_S(i, setsize, set)) {
			int rlen;
			size_t j, run = 0;
			entry_made = 1;
			for (j = i + 1; j < max; j++) {
				if (CPU_ISSET_S(j, setsize, set))
					run++;
				else
					break;
			}
			if (!run)
				rlen = snprintf(ptr, len, "%zu,", i);
			else if (run == 1) {
				rlen = snprintf(ptr, len, "%zu,%zu,", i, i + 1);
				i++;
			} else {
				rlen = snprintf(ptr, len, "%zu-%zu,", i, i + run);
				i += run;
			}
			if (rlen < 0 || (size_t) rlen >= len)
				return NULL;
			ptr += rlen;
			len -= rlen;
		}
	}
	ptr -= entry_made;
	*ptr = '\0';

	return str;
}

/*
 * Returns string with CPU mask.
 */
char *cpumask_create(char *str, size_t len,
			cpu_set_t *set, size_t setsize)
{
	char *ptr = str;
	char *ret = NULL;
	int cpu;

	for (cpu = cpuset_nbits(setsize) - 4; cpu >= 0; cpu -= 4) {
		char val = 0;

		if (len == (size_t) (ptr - str))
			break;

		if (CPU_ISSET_S(cpu, setsize, set))
			val |= 1;
		if (CPU_ISSET_S(cpu + 1, setsize, set))
			val |= 2;
		if (CPU_ISSET_S(cpu + 2, setsize, set))
			val |= 4;
		if (CPU_ISSET_S(cpu + 3, setsize, set))
			val |= 8;

		if (!ret && val)
			ret = ptr;
		*ptr++ = val_to_char(val);
	}
	*ptr = '\0';
	return ret ? ret : ptr - 1;
}

/*
 * Parses string with CPUs mask.
 */
int cpumask_parse(const char *str, cpu_set_t *set, size_t setsize)
{
	int len = strlen(str);
	const char *ptr = str + len - 1;
	int cpu = 0;

	/* skip 0x, it's all hex anyway */
	if (len > 1 && !memcmp(str, "0x", 2L))
		str += 2;

	CPU_ZERO_S(setsize, set);

	while (ptr >= str) {
		char val;

		/* cpu masks in /sys uses comma as a separator */
		if (*ptr == ',')
			ptr--;

		val = char_to_val(*ptr);
		if (val == (char) -1)
			return -1;
		if (val & 1)
			CPU_SET_S(cpu, setsize, set);
		if (val & 2)
			CPU_SET_S(cpu + 1, setsize, set);
		if (val & 4)
			CPU_SET_S(cpu + 2, setsize, set);
		if (val & 8)
			CPU_SET_S(cpu + 3, setsize, set);
		ptr--;
		cpu += 4;
	}

	return 0;
}

static int nextnumber(const char *str, char **end, unsigned int *result)
{
	errno = 0;
	if (str == NULL || *str == '\0' || !isdigit(*str))
		return -EINVAL;
	*result = (unsigned int) strtoul(str, end, 10);
	if (errno)
		return -errno;
	if (str == *end)
		return -EINVAL;
	return 0;
}

/*
 * Parses string with list of CPU ranges.
 * Returns 0 on success.
 * Returns 1 on error.
 * Returns 2 if fail is set and a cpu number passed in the list doesn't fit
 * into the cpu_set. If fail is not set cpu numbers that do not fit are
 * ignored and 0 is returned instead.
 */
int cpulist_parse(const char *str, cpu_set_t *set, size_t setsize, int fail)
{
	size_t max = cpuset_nbits(setsize);
	const char *p, *q;
	char *end = NULL;

	q = str;
	CPU_ZERO_S(setsize, set);

	while (p = q, q = nexttoken(q, ','), p) {
		unsigned int a;	/* beginning of range */
		unsigned int b;	/* end of range */
		unsigned int s;	/* stride */
		const char *c1, *c2;

		if (nextnumber(p, &end, &a) != 0)
			return 1;
		b = a;
		s = 1;
		p = end;

		c1 = nexttoken(p, '-');
		c2 = nexttoken(p, ',');

		if (c1 != NULL && (c2 == NULL || c1 < c2)) {
			if (nextnumber(c1, &end, &b) != 0)
				return 1;

			c1 = end && *end ? nexttoken(end, ':') : NULL;

			if (c1 != NULL && (c2 == NULL || c1 < c2)) {
				if (nextnumber(c1, &end, &s) != 0)
					return 1;
				if (s == 0)
					return 1;
			}
		}

		if (!(a <= b))
			return 1;
		while (a <= b) {
			if (fail && (a >= max))
				return 2;
			CPU_SET_S(a, setsize, set);
			a += s;
		}
	}

	if (end && *end)
		return 1;
	return 0;
}

#ifdef TEST_PROGRAM_CPUSET

#include <getopt.h>

int main(int argc, char *argv[])
{
	cpu_set_t *set;
	size_t setsize, buflen, nbits;
	char *buf, *mask = NULL, *range = NULL;
	int ncpus = 2048, rc, c;

	static const struct option longopts[] = {
	    { "ncpus", 1, NULL, 'n' },
	    { "mask",  1, NULL, 'm' },
	    { "range", 1, NULL, 'r' },
	    { NULL,    0, NULL, 0 }
	};

	while ((c = getopt_long(argc, argv, "n:m:r:", longopts, NULL)) != -1) {
		switch(c) {
		case 'n':
			ncpus = atoi(optarg);
			break;
		case 'm':
			mask = strdup(optarg);
			break;
		case 'r':
			range = strdup(optarg);
			break;
		default:
			goto usage_err;
		}
	}

	if (!mask && !range)
		goto usage_err;

	set = cpuset_alloc(ncpus, &setsize, &nbits);
	if (!set)
		err(EXIT_FAILURE, "failed to allocate cpu set");

	/*
	fprintf(stderr, "ncpus: %d, cpuset bits: %zd, cpuset bytes: %zd\n",
			ncpus, nbits, setsize);
	*/

	buflen = 7 * nbits;
	buf = malloc(buflen);
	if (!buf)
		err(EXIT_FAILURE, "failed to allocate cpu set buffer");

	if (mask)
		rc = cpumask_parse(mask, set, setsize);
	else
		rc = cpulist_parse(range, set, setsize, 0);

	if (rc)
		errx(EXIT_FAILURE, "failed to parse string: %s", mask ? : range);

	printf("%-15s = %15s ", mask ? : range,
				cpumask_create(buf, buflen, set, setsize));
	printf("[%s]\n", cpulist_create(buf, buflen, set, setsize));

	free(buf);
	free(mask);
	free(range);
	cpuset_free(set);

	return EXIT_SUCCESS;

usage_err:
	fprintf(stderr,
		"usage: %s [--ncpus <num>] --mask <mask> | --range <list>\n",
		program_invocation_short_name);
	exit(EXIT_FAILURE);
}
#endif
Commit	Line	Data
ff5a6d20	1	/*
79feaa60 KZ	2	* SPDX-License-Identifier: LGPL-2.1-or-later
79feaa60 KZ	3	*
ff5a6d20 KZ	4	* Terminology:
	5	*
	6	* cpuset - (libc) cpu_set_t data structure represents set of CPUs
	7	* cpumask - string with hex mask (e.g. "0x00000001")
	8	* cpulist - string with CPU ranges (e.g. "0-3,5,7,8")
	9	*
	10	* Based on code from taskset.c and Linux kernel.
	11	*
0f23ee0c KZ	12	* This file may be redistributed under the terms of the
	13	* GNU Lesser General Public License.
	14	*
ff5a6d20 KZ	15	* Copyright (C) 2010 Karel Zak <kzak@redhat.com>
	16	*/
	17
efcb71f8 KZ	18	#include <stdio.h>
	19	#include <stdlib.h>
	20	#include <unistd.h>
efcb71f8 KZ	21	#include <sched.h>
	22	#include <errno.h>
	23	#include <string.h>
	24	#include <ctype.h>
319563bf	25	#ifdef HAVE_SYS_SYSCALL_H
bae91ecf	26	#include <sys/syscall.h>
319563bf	27	#endif
efcb71f8 KZ	28
efcb71f8 KZ	29	#include "cpuset.h"
eb76ca98	30	#include "c.h"
efcb71f8 KZ	31
	32	static inline int val_to_char(int v)
	33	{
	34	if (v >= 0 && v < 10)
	35	return '0' + v;
35e82b29	36	if (v >= 10 && v < 16)
efcb71f8	37	return ('a' - 10) + v;
35e82b29	38	return -1;
efcb71f8 KZ	39	}
efcb71f8 KZ	40
efcb71f8 KZ	41	static inline int char_to_val(int c)
	42	{
	43	int cl;
	44
efcb71f8 KZ	45	if (c >= '0' && c <= '9')
efcb71f8 KZ	46	return c - '0';
35e82b29 SK	47	cl = tolower(c);
35e82b29 SK	48	if (cl >= 'a' && cl <= 'f')
efcb71f8	49	return cl + (10 - 'a');
35e82b29	50	return -1;
efcb71f8 KZ	51	}
	52
	53	static const char nexttoken(const char q, int sep)
	54	{
	55	if (q)
	56	q = strchr(q, sep);
	57	if (q)
	58	q++;
	59	return q;
	60	}
	61
bae91ecf KZ	62	/*
	63	* Number of bits in a CPU bitmask on current system
	64	*/
	65	int get_max_number_of_cpus(void)
	66	{
4797b1e5	67	#ifdef SYS_sched_getaffinity
bae91ecf KZ	68	int n, cpus = 2048;
	69	size_t setsize;
	70	cpu_set_t *set = cpuset_alloc(cpus, &setsize, NULL);
	71
	72	if (!set)
	73	return -1; /* error */
	74
	75	for (;;) {
	76	CPU_ZERO_S(setsize, set);
	77
	78	/* the library version does not return size of cpumask_t */
	79	n = syscall(SYS_sched_getaffinity, 0, setsize, set);
	80
	81	if (n < 0 && errno == EINVAL && cpus < 1024 * 1024) {
	82	cpuset_free(set);
	83	cpus *= 2;
	84	set = cpuset_alloc(cpus, &setsize, NULL);
	85	if (!set)
	86	return -1; /* error */
	87	continue;
	88	}
	89	cpuset_free(set);
	90	return n * 8;
	91	}
4797b1e5	92	#endif
bae91ecf KZ	93	return -1;
	94	}
	95
ff5a6d20 KZ	96	/*
	97	* Allocates a new set for ncpus and returns size in bytes and size in bits
	98	*/
	99	cpu_set_t cpuset_alloc(int ncpus, size_t setsize, size_t *nbits)
	100	{
	101	cpu_set_t *set = CPU_ALLOC(ncpus);
	102
	103	if (!set)
	104	return NULL;
	105	if (setsize)
	106	*setsize = CPU_ALLOC_SIZE(ncpus);
	107	if (nbits)
	108	*nbits = cpuset_nbits(CPU_ALLOC_SIZE(ncpus));
	109	return set;
	110	}
	111
	112	void cpuset_free(cpu_set_t *set)
	113	{
	114	CPU_FREE(set);
	115	}
	116
ee32c514 KZ	117	#if !HAVE_DECL_CPU_ALLOC
	118	/* Please, use CPU_COUNT_S() macro. This is fallback */
	119	int __cpuset_count_s(size_t setsize, const cpu_set_t *set)
	120	{
	121	int s = 0;
	122	const __cpu_mask *p = set->__bits;
	123	const __cpu_mask *end = &set->__bits[setsize / sizeof (__cpu_mask)];
	124
	125	while (p < end) {
	126	__cpu_mask l = *p++;
	127
	128	if (l == 0)
	129	continue;
	130	# if LONG_BIT > 32
	131	l = (l & 0x5555555555555555ul) + ((l >> 1) & 0x5555555555555555ul);
	132	l = (l & 0x3333333333333333ul) + ((l >> 2) & 0x3333333333333333ul);
	133	l = (l & 0x0f0f0f0f0f0f0f0ful) + ((l >> 4) & 0x0f0f0f0f0f0f0f0ful);
	134	l = (l & 0x00ff00ff00ff00fful) + ((l >> 8) & 0x00ff00ff00ff00fful);
	135	l = (l & 0x0000ffff0000fffful) + ((l >> 16) & 0x0000ffff0000fffful);
	136	l = (l & 0x00000000fffffffful) + ((l >> 32) & 0x00000000fffffffful);
	137	# else
	138	l = (l & 0x55555555ul) + ((l >> 1) & 0x55555555ul);
	139	l = (l & 0x33333333ul) + ((l >> 2) & 0x33333333ul);
	140	l = (l & 0x0f0f0f0ful) + ((l >> 4) & 0x0f0f0f0ful);
	141	l = (l & 0x00ff00fful) + ((l >> 8) & 0x00ff00fful);
	142	l = (l & 0x0000fffful) + ((l >> 16) & 0x0000fffful);
	143	# endif
	144	s += l;
	145	}
	146	return s;
	147	}
	148	#endif
	149
ff5a6d20 KZ	150	/*
	151	* Returns human readable representation of the cpuset. The output format is
	152	* a list of CPUs with ranges (for example, "0,1,3-9").
	153	*/
	154	char cpulist_create(char str, size_t len,
	155	cpu_set_t *set, size_t setsize)
efcb71f8	156	{
8a929ed7	157	size_t i;
efcb71f8 KZ	158	char *ptr = str;
efcb71f8 KZ	159	int entry_made = 0;
ff5a6d20	160	size_t max = cpuset_nbits(setsize);
efcb71f8	161
ff5a6d20 KZ	162	for (i = 0; i < max; i++) {
ff5a6d20 KZ	163	if (CPU_ISSET_S(i, setsize, set)) {
8a929ed7 KZ	164	int rlen;
8a929ed7 KZ	165	size_t j, run = 0;
efcb71f8	166	entry_made = 1;
ff5a6d20 KZ	167	for (j = i + 1; j < max; j++) {
ff5a6d20 KZ	168	if (CPU_ISSET_S(j, setsize, set))
efcb71f8 KZ	169	run++;
	170	else
	171	break;
	172	}
	173	if (!run)
e3ca1312	174	rlen = snprintf(ptr, len, "%zu,", i);
efcb71f8	175	else if (run == 1) {
e3ca1312	176	rlen = snprintf(ptr, len, "%zu,%zu,", i, i + 1);
efcb71f8 KZ	177	i++;
efcb71f8 KZ	178	} else {
e3ca1312	179	rlen = snprintf(ptr, len, "%zu-%zu,", i, i + run);
efcb71f8 KZ	180	i += run;
efcb71f8 KZ	181	}
06fa5817	182	if (rlen < 0 \|\| (size_t) rlen >= len)
ff5a6d20 KZ	183	return NULL;
ff5a6d20 KZ	184	ptr += rlen;
06fa5817	185	len -= rlen;
efcb71f8 KZ	186	}
	187	}
	188	ptr -= entry_made;
ff5a6d20	189	*ptr = '\0';
efcb71f8 KZ	190
	191	return str;
	192	}
	193
ff5a6d20 KZ	194	/*
	195	* Returns string with CPU mask.
	196	*/
	197	char cpumask_create(char str, size_t len,
	198	cpu_set_t *set, size_t setsize)
efcb71f8	199	{
efcb71f8	200	char *ptr = str;
ff5a6d20 KZ	201	char *ret = NULL;
ff5a6d20 KZ	202	int cpu;
efcb71f8	203
ff5a6d20	204	for (cpu = cpuset_nbits(setsize) - 4; cpu >= 0; cpu -= 4) {
efcb71f8	205	char val = 0;
ff5a6d20	206
8a929ed7	207	if (len == (size_t) (ptr - str))
ff5a6d20 KZ	208	break;
	209
	210	if (CPU_ISSET_S(cpu, setsize, set))
efcb71f8	211	val \|= 1;
ff5a6d20	212	if (CPU_ISSET_S(cpu + 1, setsize, set))
efcb71f8	213	val \|= 2;
ff5a6d20	214	if (CPU_ISSET_S(cpu + 2, setsize, set))
efcb71f8	215	val \|= 4;
ff5a6d20	216	if (CPU_ISSET_S(cpu + 3, setsize, set))
efcb71f8	217	val \|= 8;
ff5a6d20	218
efcb71f8 KZ	219	if (!ret && val)
	220	ret = ptr;
	221	*ptr++ = val_to_char(val);
	222	}
ff5a6d20	223	*ptr = '\0';
efcb71f8 KZ	224	return ret ? ret : ptr - 1;
	225	}
	226
ff5a6d20	227	/*
68845fed	228	* Parses string with CPUs mask.
ff5a6d20 KZ	229	*/
ff5a6d20 KZ	230	int cpumask_parse(const char str, cpu_set_t set, size_t setsize)
efcb71f8 KZ	231	{
	232	int len = strlen(str);
	233	const char *ptr = str + len - 1;
ff5a6d20	234	int cpu = 0;
efcb71f8 KZ	235
	236	/* skip 0x, it's all hex anyway */
	237	if (len > 1 && !memcmp(str, "0x", 2L))
	238	str += 2;
	239
ff5a6d20 KZ	240	CPU_ZERO_S(setsize, set);
ff5a6d20 KZ	241
efcb71f8	242	while (ptr >= str) {
bae91ecf KZ	243	char val;
	244
	245	/* cpu masks in /sys uses comma as a separator */
	246	if (*ptr == ',')
	247	ptr--;
	248
	249	val = char_to_val(*ptr);
efcb71f8 KZ	250	if (val == (char) -1)
	251	return -1;
	252	if (val & 1)
ff5a6d20	253	CPU_SET_S(cpu, setsize, set);
efcb71f8	254	if (val & 2)
ff5a6d20	255	CPU_SET_S(cpu + 1, setsize, set);
efcb71f8	256	if (val & 4)
ff5a6d20	257	CPU_SET_S(cpu + 2, setsize, set);
efcb71f8	258	if (val & 8)
ff5a6d20	259	CPU_SET_S(cpu + 3, setsize, set);
efcb71f8	260	ptr--;
ff5a6d20	261	cpu += 4;
efcb71f8 KZ	262	}
	263
	264	return 0;
	265	}
	266
65bcbf10 KZ	267	static int nextnumber(const char str, char end, unsigned int result)
	268	{
	269	errno = 0;
	270	if (str == NULL \|\| str == '\0' \|\| !isdigit(str))
	271	return -EINVAL;
	272	*result = (unsigned int) strtoul(str, end, 10);
	273	if (errno)
	274	return -errno;
	275	if (str == *end)
	276	return -EINVAL;
	277	return 0;
	278	}
	279
ff5a6d20	280	/*
68845fed	281	* Parses string with list of CPU ranges.
59fb133a HC	282	* Returns 0 on success.
	283	* Returns 1 on error.
	284	* Returns 2 if fail is set and a cpu number passed in the list doesn't fit
	285	* into the cpu_set. If fail is not set cpu numbers that do not fit are
	286	* ignored and 0 is returned instead.
ff5a6d20	287	*/
b16f615a	288	int cpulist_parse(const char str, cpu_set_t set, size_t setsize, int fail)
efcb71f8	289	{
b16f615a	290	size_t max = cpuset_nbits(setsize);
efcb71f8	291	const char p, q;
65bcbf10	292	char *end = NULL;
ff5a6d20	293
f27ce071	294	q = str;
ff5a6d20	295	CPU_ZERO_S(setsize, set);
efcb71f8 KZ	296
	297	while (p = q, q = nexttoken(q, ','), p) {
	298	unsigned int a; /* beginning of range */
	299	unsigned int b; /* end of range */
	300	unsigned int s; /* stride */
	301	const char c1, c2;
	302
65bcbf10	303	if (nextnumber(p, &end, &a) != 0)
efcb71f8 KZ	304	return 1;
	305	b = a;
	306	s = 1;
65bcbf10	307	p = end;
efcb71f8 KZ	308
	309	c1 = nexttoken(p, '-');
	310	c2 = nexttoken(p, ',');
65bcbf10	311
efcb71f8	312	if (c1 != NULL && (c2 == NULL \|\| c1 < c2)) {
65bcbf10	313	if (nextnumber(c1, &end, &b) != 0)
efcb71f8	314	return 1;
65bcbf10 KZ	315
	316	c1 = end && *end ? nexttoken(end, ':') : NULL;
	317
289127f5	318	if (c1 != NULL && (c2 == NULL \|\| c1 < c2)) {
65bcbf10	319	if (nextnumber(c1, &end, &s) != 0)
efcb71f8	320	return 1;
72232a26 HC	321	if (s == 0)
72232a26 HC	322	return 1;
efcb71f8 KZ	323	}
	324	}
	325
	326	if (!(a <= b))
	327	return 1;
	328	while (a <= b) {
b16f615a	329	if (fail && (a >= max))
59fb133a	330	return 2;
ff5a6d20	331	CPU_SET_S(a, setsize, set);
efcb71f8 KZ	332	a += s;
	333	}
	334	}
	335
65bcbf10	336	if (end && *end)
f27ce071	337	return 1;
efcb71f8 KZ	338	return 0;
efcb71f8 KZ	339	}
125b6a91	340
e8f7acb0	341	#ifdef TEST_PROGRAM_CPUSET
125b6a91	342
125b6a91 KZ	343	#include <getopt.h>
	344
	345	int main(int argc, char *argv[])
	346	{
ff5a6d20 KZ	347	cpu_set_t *set;
ff5a6d20 KZ	348	size_t setsize, buflen, nbits;
125b6a91 KZ	349	char buf, mask = NULL, *range = NULL;
	350	int ncpus = 2048, rc, c;
	351
6c7d5ae9	352	static const struct option longopts[] = {
71f08e97 SK	353	{ "ncpus", 1, NULL, 'n' },
	354	{ "mask", 1, NULL, 'm' },
	355	{ "range", 1, NULL, 'r' },
	356	{ NULL, 0, NULL, 0 }
125b6a91 KZ	357	};
	358
	359	while ((c = getopt_long(argc, argv, "n:m:r:", longopts, NULL)) != -1) {
	360	switch(c) {
	361	case 'n':
	362	ncpus = atoi(optarg);
	363	break;
	364	case 'm':
	365	mask = strdup(optarg);
	366	break;
	367	case 'r':
	368	range = strdup(optarg);
	369	break;
	370	default:
	371	goto usage_err;
	372	}
	373	}
	374
	375	if (!mask && !range)
	376	goto usage_err;
	377
ff5a6d20	378	set = cpuset_alloc(ncpus, &setsize, &nbits);
125b6a91 KZ	379	if (!set)
	380	err(EXIT_FAILURE, "failed to allocate cpu set");
	381
ff5a6d20 KZ	382	/*
	383	fprintf(stderr, "ncpus: %d, cpuset bits: %zd, cpuset bytes: %zd\n",
	384	ncpus, nbits, setsize);
	385	*/
	386
	387	buflen = 7 * nbits;
	388	buf = malloc(buflen);
125b6a91 KZ	389	if (!buf)
	390	err(EXIT_FAILURE, "failed to allocate cpu set buffer");
	391
	392	if (mask)
ff5a6d20	393	rc = cpumask_parse(mask, set, setsize);
125b6a91	394	else
b16f615a	395	rc = cpulist_parse(range, set, setsize, 0);
125b6a91 KZ	396
	397	if (rc)
	398	errx(EXIT_FAILURE, "failed to parse string: %s", mask ? : range);
	399
ff5a6d20 KZ	400	printf("%-15s = %15s ", mask ? : range,
	401	cpumask_create(buf, buflen, set, setsize));
	402	printf("[%s]\n", cpulist_create(buf, buflen, set, setsize));
125b6a91 KZ	403
125b6a91 KZ	404	free(buf);
41cd1503	405	free(mask);
9da2972c	406	free(range);
ff5a6d20	407	cpuset_free(set);
125b6a91 KZ	408
	409	return EXIT_SUCCESS;
	410
	411	usage_err:
	412	fprintf(stderr,
65bcbf10	413	"usage: %s [--ncpus <num>] --mask <mask> \| --range <list>\n",
125b6a91 KZ	414	program_invocation_short_name);
	415	exit(EXIT_FAILURE);
	416	}
	417	#endif