|
Packit |
d740b4 |
/* Utility functions for reading sysfs values */
|
|
Packit |
d740b4 |
#define _GNU_SOURCE 1
|
|
Packit |
d740b4 |
#include <stdio.h>
|
|
Packit |
d740b4 |
#include <sys/fcntl.h>
|
|
Packit |
d740b4 |
#include <stdlib.h>
|
|
Packit |
d740b4 |
#include <unistd.h>
|
|
Packit |
d740b4 |
#include <stdarg.h>
|
|
Packit |
d740b4 |
#include <ctype.h>
|
|
Packit |
d740b4 |
#include "numa.h"
|
|
Packit |
d740b4 |
#include "numaint.h"
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
#define SYSFS_BLOCK 4096
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
hidden char *sysfs_read(char *name)
|
|
Packit |
d740b4 |
{
|
|
Packit |
d740b4 |
char *buf;
|
|
Packit |
d740b4 |
int n;
|
|
Packit |
d740b4 |
int fd;
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
fd = open(name, O_RDONLY);
|
|
Packit |
d740b4 |
buf = malloc(SYSFS_BLOCK);
|
|
Packit |
d740b4 |
if (!buf)
|
|
Packit |
d740b4 |
return NULL;
|
|
Packit |
d740b4 |
n = read(fd, buf, SYSFS_BLOCK - 1);
|
|
Packit |
d740b4 |
close(fd);
|
|
Packit |
d740b4 |
if (n <= 0) {
|
|
Packit |
d740b4 |
free(buf);
|
|
Packit |
d740b4 |
return NULL;
|
|
Packit |
d740b4 |
}
|
|
Packit |
d740b4 |
buf[n] = 0;
|
|
Packit |
d740b4 |
return buf;
|
|
Packit |
d740b4 |
}
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
hidden int sysfs_node_read(struct bitmask *mask, char *fmt, ...)
|
|
Packit |
d740b4 |
{
|
|
Packit |
d740b4 |
int n;
|
|
Packit |
d740b4 |
va_list ap;
|
|
Packit |
d740b4 |
char *p, *fn, *m, *end;
|
|
Packit |
d740b4 |
int num;
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
va_start(ap, fmt);
|
|
Packit |
d740b4 |
n = vasprintf(&fn, fmt, ap);
|
|
Packit |
d740b4 |
va_end(ap);
|
|
Packit |
d740b4 |
if (n < 0)
|
|
Packit |
d740b4 |
return -1;
|
|
Packit |
d740b4 |
p = sysfs_read(fn);
|
|
Packit |
d740b4 |
free(fn);
|
|
Packit |
d740b4 |
if (!p)
|
|
Packit |
d740b4 |
return -1;
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
m = p;
|
|
Packit |
d740b4 |
do {
|
|
Packit |
d740b4 |
num = strtol(m, &end, 0);
|
|
Packit |
d740b4 |
if (m == end)
|
|
Packit |
d740b4 |
return -1;
|
|
Packit |
d740b4 |
if (num < 0)
|
|
Packit |
d740b4 |
return -2;
|
|
Packit |
d740b4 |
if (num >= numa_num_task_nodes())
|
|
Packit |
d740b4 |
return -1;
|
|
Packit |
d740b4 |
numa_bitmask_setbit(mask, num);
|
|
Packit |
d740b4 |
|
|
Packit |
d740b4 |
/* Continuation not supported by kernel yet. */
|
|
Packit |
d740b4 |
m = end;
|
|
Packit |
d740b4 |
while (isspace(*m) || *m == ',')
|
|
Packit |
d740b4 |
m++;
|
|
Packit |
d740b4 |
} while (isdigit(*m));
|
|
Packit |
d740b4 |
free(p);
|
|
Packit |
d740b4 |
return 0;
|
|
Packit |
d740b4 |
}
|