^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 1) /*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 2) * Copyright © 2018 Alexey Dobriyan <adobriyan@gmail.com>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 3) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 4) * Permission to use, copy, modify, and distribute this software for any
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 5) * purpose with or without fee is hereby granted, provided that the above
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 6) * copyright notice and this permission notice appear in all copies.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 7) *
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 8) * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 9) * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 10) * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 11) * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 12) * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 13) * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 14) * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 15) */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 16) // Test that values in /proc/uptime increment monotonically
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 17) // while shifting across CPUs.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 18) #undef NDEBUG
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 19) #include <assert.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 20) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 21) #include <sys/syscall.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 22) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 23) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 24)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 25) #include <stdint.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 26) #include <sys/types.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 27) #include <sys/stat.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 28) #include <fcntl.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 29)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 30) #include "proc-uptime.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 31)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 32) static inline int sys_sched_getaffinity(pid_t pid, unsigned int len, unsigned long *m)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 33) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 34) return syscall(SYS_sched_getaffinity, pid, len, m);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 35) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 36)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 37) static inline int sys_sched_setaffinity(pid_t pid, unsigned int len, unsigned long *m)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 38) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 39) return syscall(SYS_sched_setaffinity, pid, len, m);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 40) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 41)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 42) int main(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 43) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 44) unsigned int len;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 45) unsigned long *m;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 46) unsigned int cpu;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 47) uint64_t u0, u1, i0, i1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 48) int fd;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 49)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 50) /* find out "nr_cpu_ids" */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 51) m = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 52) len = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 53) do {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 54) len += sizeof(unsigned long);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 55) free(m);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 56) m = malloc(len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 57) } while (sys_sched_getaffinity(0, len, m) == -EINVAL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 58)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 59) fd = open("/proc/uptime", O_RDONLY);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 60) assert(fd >= 0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 61)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 62) proc_uptime(fd, &u0, &i0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 63) for (cpu = 0; cpu < len * 8; cpu++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 64) memset(m, 0, len);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 65) m[cpu / (8 * sizeof(unsigned long))] |= 1UL << (cpu % (8 * sizeof(unsigned long)));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 66)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 67) /* CPU might not exist, ignore error */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 68) sys_sched_setaffinity(0, len, m);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 69)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 70) proc_uptime(fd, &u1, &i1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 71) assert(u1 >= u0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 72) assert(i1 >= i0);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 73) u0 = u1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 74) i0 = i1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 75) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 76)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 77) return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 78) }