Fetch per-numa (effectively per-kind) memory usage from the OS.

A simplified version, without assigning kinds, etc.

Author: kilobyte

Makefile.am

@@ -40,6 +40,7 @@ libmemkind_la_SOURCES = src/hbwmalloc.c \
                         src/ranking_queue.c \
                         src/slab_allocator.c \
                         src/ranking_controller.c \
+                        src/numus.c \
                         # end
 
 

src/numus.c

@@ -0,0 +1,81 @@
+#include <stdio.h>
+#include <stdbool.h>
+#include <string.h>
+#include <stdint.h>
+#include <stdlib.h>
+
+#define MAX_NODES 1024 /* kernel max */
+static struct {
+    size_t used;
+} numamem[MAX_NODES];
+
+static bool gather_numamem()
+{
+    FILE *f = fopen("/proc/self/numa_maps", "r");
+    if (!f)
+        return false;
+
+    char line[1024];
+    while (fgets(line, sizeof(line), f)) {
+        char *saveptr = 0;
+        char *err;
+        // We get page size of a mapping only at the end, thus we need to
+        // temporarily keep the counts.
+        int maxnode = 0;
+        short num[MAX_NODES];
+        size_t count[MAX_NODES];
+        uint64_t pagesize = 0;
+
+        char *tok = strtok_r(line, " ", &saveptr);
+        if (!tok) {
+            fprintf(stderr, "Empty line in proc/numa_maps\n");
+            fclose(f);
+            return false;
+        }
+        strtoul(tok, &err, 16);
+        if (*err) {
+            fprintf(stderr, "Invalid addr: %s\n", tok);
+            fclose(f);
+            return false;
+        }
+        //printf("Addr: %lx\n", val);
+
+        while ((tok = strtok_r(0, " ", &saveptr))) {
+            char *valp = strchr(tok, '=');
+            if (!valp) {
+                /* flags */
+                if (!strcmp(tok, "stack"))
+                    goto skip_mapping;
+                continue;
+            }
+            *valp++ = 0;
+
+            if (!strcmp(tok, "kernelpagesize_kB"))
+                pagesize = atoi(valp) * 1024ULL;
+            else if (tok[0] == 'N' && tok[1] >= '0' && tok[1] <= '9') {
+                // per-node counts are given as N3=15 which means 15 pages
+                // on node 3.
+                int node = atoi(tok + 1);
+                if (node >= MAX_NODES)
+                    continue;
+                if (maxnode + 1 >= MAX_NODES) // overflow (can't happen)
+                    continue;
+                num[maxnode] = node;
+                count[maxnode] = atol(valp);
+                maxnode++;
+            }
+        }
+
+        if (maxnode && !pagesize) {
+            fprintf(stderr, "Pages used but no page size\n");
+            fclose(f);
+            return false;
+        }
+        for (int i = 0; i < maxnode; i++)
+            numamem[num[i]].used += count[i] * pagesize;
+
+skip_mapping:
+    }
+    fclose(f);
+    return true;
+}