Scan for CPUs. Ignore HT

cpu.h

@@ -0,0 +1,22 @@
+#ifndef _cpu_h
+#define _cpu_h
+
+struct cpu_s {
+	unsigned int id; /* Logical processor ID */
+	unsigned int package_id;
+	unsigned int core_id;
+};
+typedef struct cpu_s cpu_t;
+
+#define SYSFS_CPU_PATH "/sys/devices/system/cpu"
+
+/* CPU IDs compare function */
+int cpu_list_compare(const void *first, const void *second);
+
+/* CPU list functions */
+int cpu_list_populate(lub_list_t *cpus);
+int cpu_list_free(lub_list_t *cpus);
+int scan_cpus(lub_list_t *cpus);
+int show_cpus(lub_list_t *cpus);
+
+#endif

cpu_parse.c

@@ -0,0 +1,176 @@
+/* cpu_parse.c
+ * Parse CPU-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 "cpu.h"
+
+#define STR(str) ( str ? str : "" )
+
+int cpu_list_compare(const void *first, const void *second)
+{
+	const cpu_t *f = (const cpu_t *)first;
+	const cpu_t *s = (const cpu_t *)second;
+	return (f->id - s->id);
+}
+
+static cpu_t * cpu_new(unsigned int id,
+	unsigned int package_id,
+	unsigned int core_id)
+{
+	cpu_t *new;
+
+	if (!(new = malloc(sizeof(*new))))
+		return NULL;
+	new->id = id;
+	new->package_id = package_id;
+	new->core_id = core_id;
+	return new;
+}
+
+static void cpu_free(cpu_t *cpu)
+{
+	free(cpu);
+}
+
+static cpu_t * cpu_list_search_ht(lub_list_t *cpus,
+	unsigned int package_id,
+	unsigned int core_id)
+{
+	lub_list_node_t *iter;
+
+	for (iter = lub_list_iterator_init(cpus); iter;
+		iter = lub_list_iterator_next(iter)) {
+		cpu_t *cpu;
+		cpu = (cpu_t *)lub_list_node__get_data(iter);
+		if (cpu->package_id != package_id)
+			continue;
+		if (cpu->core_id != core_id)
+			continue;
+		return cpu;
+	}
+
+	return NULL;
+}
+
+#if 0
+static cpu_t * cpu_list_search(lub_list_t *cpus, unsigned int id)
+{
+	lub_list_node_t *node;
+	cpu_t search;
+
+	search.id = id;
+	node = lub_list_search(cpus, &search);
+	if (!node)
+		return NULL;
+	return (cpu_t *)lub_list_node__get_data(node);
+}
+#endif
+
+static cpu_t * cpu_list_add(lub_list_t *cpus,
+	unsigned int id,
+	unsigned int package_id,
+	unsigned int core_id)
+{
+	lub_list_node_t *node;
+	cpu_t *new;
+	cpu_t search;
+
+	search.id = id;
+	node = lub_list_search(cpus, &search);
+	if (node) /* CPU already exists. May be renew some fields later */
+		return (cpu_t *)lub_list_node__get_data(node);
+	if (!(new = cpu_new(id, package_id, core_id)))
+		return NULL;
+	lub_list_add(cpus, new);
+
+	return new;
+}
+
+int cpu_list_free(lub_list_t *cpus)
+{
+	lub_list_node_t *iter;
+	while ((iter = lub_list__get_head(cpus))) {
+		cpu_t *cpu;
+		cpu = (cpu_t *)lub_list_node__get_data(iter);
+		cpu_free(cpu);
+		lub_list_del(cpus, iter);
+		lub_list_node_free(iter);
+	}
+	lub_list_free(cpus);
+	return 0;
+}
+
+/* Show CPU information */
+static void show_cpu_info(cpu_t *cpu)
+{
+	printf("CPU %d package %d core %d\n", cpu->id, cpu->package_id, cpu->core_id);
+}
+
+/* Show CPU list */
+int show_cpus(lub_list_t *cpus)
+{
+	lub_list_node_t *iter;
+	for (iter = lub_list_iterator_init(cpus); iter;
+		iter = lub_list_iterator_next(iter)) {
+		cpu_t *cpu;
+		cpu = (cpu_t *)lub_list_node__get_data(iter);
+		show_cpu_info(cpu);
+	}
+	return 0;
+}
+
+int scan_cpus(lub_list_t *cpus)
+{
+	FILE *fd;
+	char path[PATH_MAX];
+	unsigned int id;
+	unsigned int package_id;
+	unsigned int core_id;
+
+	for (id = 0; id < NR_CPUS; id++) {
+		sprintf(path, "%s/cpu%d", SYSFS_CPU_PATH, id);
+		if (access(path, F_OK))
+			break;
+
+		/* Try to get package_id */
+		sprintf(path, "%s/cpu%d/topology/physical_package_id",
+			SYSFS_CPU_PATH, id);
+		if (!(fd = fopen(path, "r")))
+			continue;
+		if (fscanf(fd, "%u", &package_id) < 0) {
+			fclose(fd);
+			continue;
+		}
+		fclose(fd);
+
+		/* Try to get core_id */
+		sprintf(path, "%s/cpu%d/topology/core_id",
+			SYSFS_CPU_PATH, id);
+		if (!(fd = fopen(path, "r")))
+			continue;
+		if (fscanf(fd, "%u", &core_id) < 0) {
+			fclose(fd);
+			continue;
+		}
+		fclose(fd);
+
+		/* Don't use second thread of Hyper Threading */
+		if (cpu_list_search_ht(cpus, package_id, core_id))
+			continue;
+
+		cpu_list_add(cpus, id, package_id, core_id);
+	}
+
+	return 0;
+}