kern/cpumap: add cpumap_all function

2 files changed, 24 insertions, 1 deletions

diff --git a/kern/cpumap.c b/kern/cpumap.c
index a76450b7..68a84ab9 100644
--- a/kern/cpumap.c
+++ b/kern/cpumap.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2013 Richard Braun.
+ * Copyright (c) 2013-2014 Richard Braun.
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -19,15 +19,33 @@
 #include <kern/cpumap.h>
 #include <kern/error.h>
 #include <kern/kmem.h>
+#include <kern/param.h>
 #include <kern/stddef.h>
+#include <machine/cpu.h>
+
+static struct cpumap cpumap_active_cpus __read_mostly;
 
 static struct kmem_cache cpumap_cache;
 
 void
 cpumap_setup(void)
 {
+    unsigned int i, nr_cpus;
+
     kmem_cache_init(&cpumap_cache, "cpumap", sizeof(struct cpumap),
                     0, NULL, NULL, NULL, 0);
+
+    cpumap_zero(&cpumap_active_cpus);
+    nr_cpus = cpu_count();
+
+    for (i = 0; i < nr_cpus; i++)
+        cpumap_set(&cpumap_active_cpus, i);
+}
+
+const struct cpumap *
+cpumap_all(void)
+{
+    return &cpumap_active_cpus;
 }
 
 int
diff --git a/kern/cpumap.h b/kern/cpumap.h
index 9cbdfa15..0e57c30d 100644
--- a/kern/cpumap.h
+++ b/kern/cpumap.h
@@ -140,6 +140,11 @@ cpumap_find_first_zero(const struct cpumap *cpumap)
 void cpumap_setup(void);
 
 /*
+ * Return a cpumap representing all active processors.
+ */
+const struct cpumap * cpumap_all(void);
+
+/*
  * Allocate a CPU map.
  *
  * The new map is uninitialized.