| .. | .. |
|---|
| 6 | 6 | #include <stdlib.h> |
|---|
| 7 | 7 | #include <string.h> |
|---|
| 8 | 8 | #include <errno.h> |
|---|
| 9 | +#include <bpf/libbpf.h> /* libbpf_num_possible_cpus */ |
|---|
| 9 | 10 | |
|---|
| 10 | 11 | static inline unsigned int bpf_num_possible_cpus(void) |
|---|
| 11 | 12 | { |
|---|
| 12 | | - static const char *fcpu = "/sys/devices/system/cpu/possible"; |
|---|
| 13 | | - unsigned int start, end, possible_cpus = 0; |
|---|
| 14 | | - char buff[128]; |
|---|
| 15 | | - FILE *fp; |
|---|
| 16 | | - int len, n, i, j = 0; |
|---|
| 13 | + int possible_cpus = libbpf_num_possible_cpus(); |
|---|
| 17 | 14 | |
|---|
| 18 | | - fp = fopen(fcpu, "r"); |
|---|
| 19 | | - if (!fp) { |
|---|
| 20 | | - printf("Failed to open %s: '%s'!\n", fcpu, strerror(errno)); |
|---|
| 15 | + if (possible_cpus < 0) { |
|---|
| 16 | + printf("Failed to get # of possible cpus: '%s'!\n", |
|---|
| 17 | + strerror(-possible_cpus)); |
|---|
| 21 | 18 | exit(1); |
|---|
| 22 | 19 | } |
|---|
| 23 | | - |
|---|
| 24 | | - if (!fgets(buff, sizeof(buff), fp)) { |
|---|
| 25 | | - printf("Failed to read %s!\n", fcpu); |
|---|
| 26 | | - exit(1); |
|---|
| 27 | | - } |
|---|
| 28 | | - |
|---|
| 29 | | - len = strlen(buff); |
|---|
| 30 | | - for (i = 0; i <= len; i++) { |
|---|
| 31 | | - if (buff[i] == ',' || buff[i] == '\0') { |
|---|
| 32 | | - buff[i] = '\0'; |
|---|
| 33 | | - n = sscanf(&buff[j], "%u-%u", &start, &end); |
|---|
| 34 | | - if (n <= 0) { |
|---|
| 35 | | - printf("Failed to retrieve # possible CPUs!\n"); |
|---|
| 36 | | - exit(1); |
|---|
| 37 | | - } else if (n == 1) { |
|---|
| 38 | | - end = start; |
|---|
| 39 | | - } |
|---|
| 40 | | - possible_cpus += end - start + 1; |
|---|
| 41 | | - j = i + 1; |
|---|
| 42 | | - } |
|---|
| 43 | | - } |
|---|
| 44 | | - |
|---|
| 45 | | - fclose(fp); |
|---|
| 46 | | - |
|---|
| 47 | 20 | return possible_cpus; |
|---|
| 48 | 21 | } |
|---|
| 49 | 22 | |
|---|
| .. | .. |
|---|
| 58 | 31 | # define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0])) |
|---|
| 59 | 32 | #endif |
|---|
| 60 | 33 | |
|---|
| 34 | +#ifndef sizeof_field |
|---|
| 35 | +#define sizeof_field(TYPE, MEMBER) sizeof((((TYPE *)0)->MEMBER)) |
|---|
| 36 | +#endif |
|---|
| 37 | + |
|---|
| 38 | +#ifndef offsetofend |
|---|
| 39 | +#define offsetofend(TYPE, MEMBER) \ |
|---|
| 40 | + (offsetof(TYPE, MEMBER) + sizeof_field(TYPE, MEMBER)) |
|---|
| 41 | +#endif |
|---|
| 42 | + |
|---|
| 61 | 43 | #endif /* __BPF_UTIL__ */ |
|---|