1 files changed, 90 insertions, 1 deletions

diff --git a/drivers/base/node.c b/drivers/base/node.c
index cae346ef1b20..88eeed72b5d6 100644
--- a/drivers/base/node.c
+++ b/drivers/base/node.c
@@ -12,6 +12,7 @@
 #include <linux/topology.h>
 #include <linux/nodemask.h>
 #include <linux/cpu.h>
+#include <linux/device.h>
 
 static struct sysdev_class node_class = {
         set_kset_name("node"),
@@ -232,8 +233,96 @@ void unregister_one_node(int nid)
         unregister_node(&node_devices[nid]);
 }
 
+/*
+ * node states attributes
+ */
+
+static ssize_t print_nodes_state(enum node_states state, char *buf)
+{
+        int n;
+
+        n = nodelist_scnprintf(buf, PAGE_SIZE, node_states[state]);
+        if (n > 0 && PAGE_SIZE > n + 1) {
+                *(buf + n++) = '\n';
+                *(buf + n++) = '\0';
+        }
+        return n;
+}
+
+static ssize_t print_nodes_possible(struct sysdev_class *class, char *buf)
+{
+        return print_nodes_state(N_POSSIBLE, buf);
+}
+
+static ssize_t print_nodes_online(struct sysdev_class *class, char *buf)
+{
+        return print_nodes_state(N_ONLINE, buf);
+}
+
+static ssize_t print_nodes_has_normal_memory(struct sysdev_class *class,
+                                                char *buf)
+{
+        return print_nodes_state(N_NORMAL_MEMORY, buf);
+}
+
+static ssize_t print_nodes_has_cpu(struct sysdev_class *class, char *buf)
+{
+        return print_nodes_state(N_CPU, buf);
+}
+
+static SYSDEV_CLASS_ATTR(possible, 0444, print_nodes_possible, NULL);
+static SYSDEV_CLASS_ATTR(online, 0444, print_nodes_online, NULL);
+static SYSDEV_CLASS_ATTR(has_normal_memory, 0444, print_nodes_has_normal_memory,
+                                                                        NULL);
+static SYSDEV_CLASS_ATTR(has_cpu, 0444, print_nodes_has_cpu, NULL);
+
+#ifdef CONFIG_HIGHMEM
+static ssize_t print_nodes_has_high_memory(struct sysdev_class *class,
+                                                 char *buf)
+{
+        return print_nodes_state(N_HIGH_MEMORY, buf);
+}
+
+static SYSDEV_CLASS_ATTR(has_high_memory, 0444, print_nodes_has_high_memory,
+                                                                         NULL);
+#endif
+
+struct sysdev_class_attribute *node_state_attr[] = {
+        &attr_possible,
+        &attr_online,
+        &attr_has_normal_memory,
+#ifdef CONFIG_HIGHMEM
+        &attr_has_high_memory,
+#endif
+        &attr_has_cpu,
+};
+
+static int node_states_init(void)
+{
+        int i;
+        int err = 0;
+
+        for (i = 0;  i < NR_NODE_STATES; i++) {
+                int ret;
+                ret = sysdev_class_create_file(&node_class, node_state_attr[i]);
+                if (!err)
+                        err = ret;
+        }
+        return err;
+}
+
 static int __init register_node_type(void)
 {
-        return sysdev_class_register(&node_class);
+        int ret;
+
+        ret = sysdev_class_register(&node_class);
+        if (!ret)
+                ret = node_states_init();
+
+        /*
+         * Note:  we're not going to unregister the node class if we fail
+         * to register the node state class attribute files.
+         */
+        return ret;
 }
 postcore_initcall(register_node_type);