blob: 6bc192074eedef9e6b7786895ddabf680fd5ea05 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#include <stddef.h>
#include "demo.h"
static void bad_input(void)
{
print("bad input");
}
int strcmp(const char *s1, const char *s2)
{
while (*s1 && *s1 == *s2)
s1++, s2++;
return (int)*(const unsigned char *)s1 - (int)*(const unsigned char *)s2;
}
char *strtok_input(char **tokens)
{
char *start = *tokens;
while (*start && *start == ' ')
++start;
if (!*start) {
*tokens = start;
return NULL;
}
char *end = start + 1;
while (*end && *end != ' ')
++end;
*tokens = *end ? end + 1 : end;
*end = '\0';
return start;
}
int parse_cpu_mask(char **tokens, unsigned *mask)
{
*mask = 0;
char *token;
while ((token = strtok_input(tokens))) {
if (token[0] != 'c' || token[1] != 'p' || token[2] != 'u'
|| !('0' <= token[3] && token[3] < '0' + NUM_CPUS) || token[4]) {
bad_input();
return -1;
}
*mask |= 1 << (token[3] - '0');
}
if (!*mask) {
print("must specify at least one cpu");
return -1;
}
return 0;
}
|
