123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142 |
- /* numa.c
- * Parse NUMA-related files.
- */
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #include <sys/types.h>
- #include <dirent.h>
- #include <limits.h>
- #include <ctype.h>
- #include <unistd.h>
- #include "lub/list.h"
- #include "cpumask.h"
- #include "numa.h"
- int numa_list_compare(const void *first, const void *second)
- {
- const numa_t *f = (const numa_t *)first;
- const numa_t *s = (const numa_t *)second;
- return (f->id - s->id);
- }
- static numa_t * numa_new(unsigned int id)
- {
- numa_t *new;
- if (!(new = malloc(sizeof(*new))))
- return NULL;
- new->id = id;
- cpus_init(new->cpumap);
- cpus_setall(new->cpumap);
- return new;
- }
- static void numa_free(numa_t *numa)
- {
- cpus_free(numa->cpumap);
- free(numa);
- }
- numa_t * numa_list_search(lub_list_t *numas, unsigned int id)
- {
- lub_list_node_t *node;
- numa_t search;
- search.id = id;
- node = lub_list_search(numas, &search);
- if (!node)
- return NULL;
- return (numa_t *)lub_list_node__get_data(node);
- }
- static numa_t * numa_list_add(lub_list_t *numas, numa_t *numa)
- {
- numa_t *old = numa_list_search(numas, numa->id);
- if (old) /* NUMA already exists. May be renew some fields later */
- return old;
- lub_list_add(numas, numa);
- return numa;
- }
- int numa_list_free(lub_list_t *numas)
- {
- lub_list_node_t *iter;
- while ((iter = lub_list__get_head(numas))) {
- numa_t *numa;
- numa = (numa_t *)lub_list_node__get_data(iter);
- numa_free(numa);
- lub_list_del(numas, iter);
- lub_list_node_free(iter);
- }
- lub_list_free(numas);
- return 0;
- }
- /* Show NUMA information */
- static void show_numa_info(numa_t *numa)
- {
- char buf[NR_CPUS + 1];
- cpumask_scnprintf(buf, sizeof(buf), numa->cpumap);
- buf[sizeof(buf) - 1] = '\0';
- printf("NUMA node %d cpumap %s\n", numa->id, buf);
- }
- /* Show NUMA list */
- int show_numas(lub_list_t *numas)
- {
- lub_list_node_t *iter;
- for (iter = lub_list_iterator_init(numas); iter;
- iter = lub_list_iterator_next(iter)) {
- numa_t *numa;
- numa = (numa_t *)lub_list_node__get_data(iter);
- show_numa_info(numa);
- }
- return 0;
- }
- /* Search for NUMA nodes */
- int scan_numas(lub_list_t *numas)
- {
- FILE *fd;
- char path[PATH_MAX];
- unsigned int id;
- numa_t *numa;
- char *str = NULL;
- size_t sz;
- cpumask_t cpumap;
- cpus_init(cpumap);
- for (id = 0; id < NR_NUMA_NODES; id++) {
- snprintf(path, sizeof(path),
- "%s/node%d", SYSFS_NUMA_PATH, id);
- path[sizeof(path) - 1] = '\0';
- if (access(path, F_OK))
- break;
- if (!(numa = numa_list_search(numas, id))) {
- numa = numa_new(id);
- numa_list_add(numas, numa);
- }
- /* Get NUMA node cpumap */
- snprintf(path, sizeof(path),
- "%s/node%d/cpumap", SYSFS_NUMA_PATH, id);
- path[sizeof(path) - 1] = '\0';
- if ((fd = fopen(path, "r"))) {
- if (getline(&str, &sz, fd) >= 0)
- cpumask_parse_user(str, strlen(str), cpumap);
- fclose(fd);
- cpus_and(numa->cpumap, numa->cpumap, cpumap);
- }
- }
- free(str);
- cpus_free(cpumap);
- return 0;
- }
|