xref: /openbmc/linux/drivers/base/node.c (revision f8e17c17)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Basic Node interface support
4  */
5 
6 #include <linux/module.h>
7 #include <linux/init.h>
8 #include <linux/mm.h>
9 #include <linux/memory.h>
10 #include <linux/vmstat.h>
11 #include <linux/notifier.h>
12 #include <linux/node.h>
13 #include <linux/hugetlb.h>
14 #include <linux/compaction.h>
15 #include <linux/cpumask.h>
16 #include <linux/topology.h>
17 #include <linux/nodemask.h>
18 #include <linux/cpu.h>
19 #include <linux/device.h>
20 #include <linux/pm_runtime.h>
21 #include <linux/swap.h>
22 #include <linux/slab.h>
23 
24 static struct bus_type node_subsys = {
25 	.name = "node",
26 	.dev_name = "node",
27 };
28 
29 
30 static ssize_t node_read_cpumap(struct device *dev, bool list, char *buf)
31 {
32 	ssize_t n;
33 	cpumask_var_t mask;
34 	struct node *node_dev = to_node(dev);
35 
36 	/* 2008/04/07: buf currently PAGE_SIZE, need 9 chars per 32 bits. */
37 	BUILD_BUG_ON((NR_CPUS/32 * 9) > (PAGE_SIZE-1));
38 
39 	if (!alloc_cpumask_var(&mask, GFP_KERNEL))
40 		return 0;
41 
42 	cpumask_and(mask, cpumask_of_node(node_dev->dev.id), cpu_online_mask);
43 	n = cpumap_print_to_pagebuf(list, buf, mask);
44 	free_cpumask_var(mask);
45 
46 	return n;
47 }
48 
49 static inline ssize_t node_read_cpumask(struct device *dev,
50 				struct device_attribute *attr, char *buf)
51 {
52 	return node_read_cpumap(dev, false, buf);
53 }
54 static inline ssize_t node_read_cpulist(struct device *dev,
55 				struct device_attribute *attr, char *buf)
56 {
57 	return node_read_cpumap(dev, true, buf);
58 }
59 
60 static DEVICE_ATTR(cpumap,  S_IRUGO, node_read_cpumask, NULL);
61 static DEVICE_ATTR(cpulist, S_IRUGO, node_read_cpulist, NULL);
62 
63 /**
64  * struct node_access_nodes - Access class device to hold user visible
65  * 			      relationships to other nodes.
66  * @dev:	Device for this memory access class
67  * @list_node:	List element in the node's access list
68  * @access:	The access class rank
69  * @hmem_attrs: Heterogeneous memory performance attributes
70  */
71 struct node_access_nodes {
72 	struct device		dev;
73 	struct list_head	list_node;
74 	unsigned		access;
75 #ifdef CONFIG_HMEM_REPORTING
76 	struct node_hmem_attrs	hmem_attrs;
77 #endif
78 };
79 #define to_access_nodes(dev) container_of(dev, struct node_access_nodes, dev)
80 
81 static struct attribute *node_init_access_node_attrs[] = {
82 	NULL,
83 };
84 
85 static struct attribute *node_targ_access_node_attrs[] = {
86 	NULL,
87 };
88 
89 static const struct attribute_group initiators = {
90 	.name	= "initiators",
91 	.attrs	= node_init_access_node_attrs,
92 };
93 
94 static const struct attribute_group targets = {
95 	.name	= "targets",
96 	.attrs	= node_targ_access_node_attrs,
97 };
98 
99 static const struct attribute_group *node_access_node_groups[] = {
100 	&initiators,
101 	&targets,
102 	NULL,
103 };
104 
105 static void node_remove_accesses(struct node *node)
106 {
107 	struct node_access_nodes *c, *cnext;
108 
109 	list_for_each_entry_safe(c, cnext, &node->access_list, list_node) {
110 		list_del(&c->list_node);
111 		device_unregister(&c->dev);
112 	}
113 }
114 
115 static void node_access_release(struct device *dev)
116 {
117 	kfree(to_access_nodes(dev));
118 }
119 
120 static struct node_access_nodes *node_init_node_access(struct node *node,
121 						       unsigned access)
122 {
123 	struct node_access_nodes *access_node;
124 	struct device *dev;
125 
126 	list_for_each_entry(access_node, &node->access_list, list_node)
127 		if (access_node->access == access)
128 			return access_node;
129 
130 	access_node = kzalloc(sizeof(*access_node), GFP_KERNEL);
131 	if (!access_node)
132 		return NULL;
133 
134 	access_node->access = access;
135 	dev = &access_node->dev;
136 	dev->parent = &node->dev;
137 	dev->release = node_access_release;
138 	dev->groups = node_access_node_groups;
139 	if (dev_set_name(dev, "access%u", access))
140 		goto free;
141 
142 	if (device_register(dev))
143 		goto free_name;
144 
145 	pm_runtime_no_callbacks(dev);
146 	list_add_tail(&access_node->list_node, &node->access_list);
147 	return access_node;
148 free_name:
149 	kfree_const(dev->kobj.name);
150 free:
151 	kfree(access_node);
152 	return NULL;
153 }
154 
155 #ifdef CONFIG_HMEM_REPORTING
156 #define ACCESS_ATTR(name) 						   \
157 static ssize_t name##_show(struct device *dev,				   \
158 			   struct device_attribute *attr,		   \
159 			   char *buf)					   \
160 {									   \
161 	return sprintf(buf, "%u\n", to_access_nodes(dev)->hmem_attrs.name); \
162 }									   \
163 static DEVICE_ATTR_RO(name);
164 
165 ACCESS_ATTR(read_bandwidth)
166 ACCESS_ATTR(read_latency)
167 ACCESS_ATTR(write_bandwidth)
168 ACCESS_ATTR(write_latency)
169 
170 static struct attribute *access_attrs[] = {
171 	&dev_attr_read_bandwidth.attr,
172 	&dev_attr_read_latency.attr,
173 	&dev_attr_write_bandwidth.attr,
174 	&dev_attr_write_latency.attr,
175 	NULL,
176 };
177 
178 /**
179  * node_set_perf_attrs - Set the performance values for given access class
180  * @nid: Node identifier to be set
181  * @hmem_attrs: Heterogeneous memory performance attributes
182  * @access: The access class the for the given attributes
183  */
184 void node_set_perf_attrs(unsigned int nid, struct node_hmem_attrs *hmem_attrs,
185 			 unsigned access)
186 {
187 	struct node_access_nodes *c;
188 	struct node *node;
189 	int i;
190 
191 	if (WARN_ON_ONCE(!node_online(nid)))
192 		return;
193 
194 	node = node_devices[nid];
195 	c = node_init_node_access(node, access);
196 	if (!c)
197 		return;
198 
199 	c->hmem_attrs = *hmem_attrs;
200 	for (i = 0; access_attrs[i] != NULL; i++) {
201 		if (sysfs_add_file_to_group(&c->dev.kobj, access_attrs[i],
202 					    "initiators")) {
203 			pr_info("failed to add performance attribute to node %d\n",
204 				nid);
205 			break;
206 		}
207 	}
208 }
209 
210 /**
211  * struct node_cache_info - Internal tracking for memory node caches
212  * @dev:	Device represeting the cache level
213  * @node:	List element for tracking in the node
214  * @cache_attrs:Attributes for this cache level
215  */
216 struct node_cache_info {
217 	struct device dev;
218 	struct list_head node;
219 	struct node_cache_attrs cache_attrs;
220 };
221 #define to_cache_info(device) container_of(device, struct node_cache_info, dev)
222 
223 #define CACHE_ATTR(name, fmt) 						\
224 static ssize_t name##_show(struct device *dev,				\
225 			   struct device_attribute *attr,		\
226 			   char *buf)					\
227 {									\
228 	return sprintf(buf, fmt "\n", to_cache_info(dev)->cache_attrs.name);\
229 }									\
230 DEVICE_ATTR_RO(name);
231 
232 CACHE_ATTR(size, "%llu")
233 CACHE_ATTR(line_size, "%u")
234 CACHE_ATTR(indexing, "%u")
235 CACHE_ATTR(write_policy, "%u")
236 
237 static struct attribute *cache_attrs[] = {
238 	&dev_attr_indexing.attr,
239 	&dev_attr_size.attr,
240 	&dev_attr_line_size.attr,
241 	&dev_attr_write_policy.attr,
242 	NULL,
243 };
244 ATTRIBUTE_GROUPS(cache);
245 
246 static void node_cache_release(struct device *dev)
247 {
248 	kfree(dev);
249 }
250 
251 static void node_cacheinfo_release(struct device *dev)
252 {
253 	struct node_cache_info *info = to_cache_info(dev);
254 	kfree(info);
255 }
256 
257 static void node_init_cache_dev(struct node *node)
258 {
259 	struct device *dev;
260 
261 	dev = kzalloc(sizeof(*dev), GFP_KERNEL);
262 	if (!dev)
263 		return;
264 
265 	dev->parent = &node->dev;
266 	dev->release = node_cache_release;
267 	if (dev_set_name(dev, "memory_side_cache"))
268 		goto free_dev;
269 
270 	if (device_register(dev))
271 		goto free_name;
272 
273 	pm_runtime_no_callbacks(dev);
274 	node->cache_dev = dev;
275 	return;
276 free_name:
277 	kfree_const(dev->kobj.name);
278 free_dev:
279 	kfree(dev);
280 }
281 
282 /**
283  * node_add_cache() - add cache attribute to a memory node
284  * @nid: Node identifier that has new cache attributes
285  * @cache_attrs: Attributes for the cache being added
286  */
287 void node_add_cache(unsigned int nid, struct node_cache_attrs *cache_attrs)
288 {
289 	struct node_cache_info *info;
290 	struct device *dev;
291 	struct node *node;
292 
293 	if (!node_online(nid) || !node_devices[nid])
294 		return;
295 
296 	node = node_devices[nid];
297 	list_for_each_entry(info, &node->cache_attrs, node) {
298 		if (info->cache_attrs.level == cache_attrs->level) {
299 			dev_warn(&node->dev,
300 				"attempt to add duplicate cache level:%d\n",
301 				cache_attrs->level);
302 			return;
303 		}
304 	}
305 
306 	if (!node->cache_dev)
307 		node_init_cache_dev(node);
308 	if (!node->cache_dev)
309 		return;
310 
311 	info = kzalloc(sizeof(*info), GFP_KERNEL);
312 	if (!info)
313 		return;
314 
315 	dev = &info->dev;
316 	dev->parent = node->cache_dev;
317 	dev->release = node_cacheinfo_release;
318 	dev->groups = cache_groups;
319 	if (dev_set_name(dev, "index%d", cache_attrs->level))
320 		goto free_cache;
321 
322 	info->cache_attrs = *cache_attrs;
323 	if (device_register(dev)) {
324 		dev_warn(&node->dev, "failed to add cache level:%d\n",
325 			 cache_attrs->level);
326 		goto free_name;
327 	}
328 	pm_runtime_no_callbacks(dev);
329 	list_add_tail(&info->node, &node->cache_attrs);
330 	return;
331 free_name:
332 	kfree_const(dev->kobj.name);
333 free_cache:
334 	kfree(info);
335 }
336 
337 static void node_remove_caches(struct node *node)
338 {
339 	struct node_cache_info *info, *next;
340 
341 	if (!node->cache_dev)
342 		return;
343 
344 	list_for_each_entry_safe(info, next, &node->cache_attrs, node) {
345 		list_del(&info->node);
346 		device_unregister(&info->dev);
347 	}
348 	device_unregister(node->cache_dev);
349 }
350 
351 static void node_init_caches(unsigned int nid)
352 {
353 	INIT_LIST_HEAD(&node_devices[nid]->cache_attrs);
354 }
355 #else
356 static void node_init_caches(unsigned int nid) { }
357 static void node_remove_caches(struct node *node) { }
358 #endif
359 
360 #define K(x) ((x) << (PAGE_SHIFT - 10))
361 static ssize_t node_read_meminfo(struct device *dev,
362 			struct device_attribute *attr, char *buf)
363 {
364 	int n;
365 	int nid = dev->id;
366 	struct pglist_data *pgdat = NODE_DATA(nid);
367 	struct sysinfo i;
368 	unsigned long sreclaimable, sunreclaimable;
369 
370 	si_meminfo_node(&i, nid);
371 	sreclaimable = node_page_state(pgdat, NR_SLAB_RECLAIMABLE);
372 	sunreclaimable = node_page_state(pgdat, NR_SLAB_UNRECLAIMABLE);
373 	n = sprintf(buf,
374 		       "Node %d MemTotal:       %8lu kB\n"
375 		       "Node %d MemFree:        %8lu kB\n"
376 		       "Node %d MemUsed:        %8lu kB\n"
377 		       "Node %d Active:         %8lu kB\n"
378 		       "Node %d Inactive:       %8lu kB\n"
379 		       "Node %d Active(anon):   %8lu kB\n"
380 		       "Node %d Inactive(anon): %8lu kB\n"
381 		       "Node %d Active(file):   %8lu kB\n"
382 		       "Node %d Inactive(file): %8lu kB\n"
383 		       "Node %d Unevictable:    %8lu kB\n"
384 		       "Node %d Mlocked:        %8lu kB\n",
385 		       nid, K(i.totalram),
386 		       nid, K(i.freeram),
387 		       nid, K(i.totalram - i.freeram),
388 		       nid, K(node_page_state(pgdat, NR_ACTIVE_ANON) +
389 				node_page_state(pgdat, NR_ACTIVE_FILE)),
390 		       nid, K(node_page_state(pgdat, NR_INACTIVE_ANON) +
391 				node_page_state(pgdat, NR_INACTIVE_FILE)),
392 		       nid, K(node_page_state(pgdat, NR_ACTIVE_ANON)),
393 		       nid, K(node_page_state(pgdat, NR_INACTIVE_ANON)),
394 		       nid, K(node_page_state(pgdat, NR_ACTIVE_FILE)),
395 		       nid, K(node_page_state(pgdat, NR_INACTIVE_FILE)),
396 		       nid, K(node_page_state(pgdat, NR_UNEVICTABLE)),
397 		       nid, K(sum_zone_node_page_state(nid, NR_MLOCK)));
398 
399 #ifdef CONFIG_HIGHMEM
400 	n += sprintf(buf + n,
401 		       "Node %d HighTotal:      %8lu kB\n"
402 		       "Node %d HighFree:       %8lu kB\n"
403 		       "Node %d LowTotal:       %8lu kB\n"
404 		       "Node %d LowFree:        %8lu kB\n",
405 		       nid, K(i.totalhigh),
406 		       nid, K(i.freehigh),
407 		       nid, K(i.totalram - i.totalhigh),
408 		       nid, K(i.freeram - i.freehigh));
409 #endif
410 	n += sprintf(buf + n,
411 		       "Node %d Dirty:          %8lu kB\n"
412 		       "Node %d Writeback:      %8lu kB\n"
413 		       "Node %d FilePages:      %8lu kB\n"
414 		       "Node %d Mapped:         %8lu kB\n"
415 		       "Node %d AnonPages:      %8lu kB\n"
416 		       "Node %d Shmem:          %8lu kB\n"
417 		       "Node %d KernelStack:    %8lu kB\n"
418 		       "Node %d PageTables:     %8lu kB\n"
419 		       "Node %d NFS_Unstable:   %8lu kB\n"
420 		       "Node %d Bounce:         %8lu kB\n"
421 		       "Node %d WritebackTmp:   %8lu kB\n"
422 		       "Node %d KReclaimable:   %8lu kB\n"
423 		       "Node %d Slab:           %8lu kB\n"
424 		       "Node %d SReclaimable:   %8lu kB\n"
425 		       "Node %d SUnreclaim:     %8lu kB\n"
426 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
427 		       "Node %d AnonHugePages:  %8lu kB\n"
428 		       "Node %d ShmemHugePages: %8lu kB\n"
429 		       "Node %d ShmemPmdMapped: %8lu kB\n"
430 		       "Node %d FileHugePages: %8lu kB\n"
431 		       "Node %d FilePmdMapped: %8lu kB\n"
432 #endif
433 			,
434 		       nid, K(node_page_state(pgdat, NR_FILE_DIRTY)),
435 		       nid, K(node_page_state(pgdat, NR_WRITEBACK)),
436 		       nid, K(node_page_state(pgdat, NR_FILE_PAGES)),
437 		       nid, K(node_page_state(pgdat, NR_FILE_MAPPED)),
438 		       nid, K(node_page_state(pgdat, NR_ANON_MAPPED)),
439 		       nid, K(i.sharedram),
440 		       nid, sum_zone_node_page_state(nid, NR_KERNEL_STACK_KB),
441 		       nid, K(sum_zone_node_page_state(nid, NR_PAGETABLE)),
442 		       nid, K(node_page_state(pgdat, NR_UNSTABLE_NFS)),
443 		       nid, K(sum_zone_node_page_state(nid, NR_BOUNCE)),
444 		       nid, K(node_page_state(pgdat, NR_WRITEBACK_TEMP)),
445 		       nid, K(sreclaimable +
446 			      node_page_state(pgdat, NR_KERNEL_MISC_RECLAIMABLE)),
447 		       nid, K(sreclaimable + sunreclaimable),
448 		       nid, K(sreclaimable),
449 		       nid, K(sunreclaimable)
450 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
451 		       ,
452 		       nid, K(node_page_state(pgdat, NR_ANON_THPS) *
453 				       HPAGE_PMD_NR),
454 		       nid, K(node_page_state(pgdat, NR_SHMEM_THPS) *
455 				       HPAGE_PMD_NR),
456 		       nid, K(node_page_state(pgdat, NR_SHMEM_PMDMAPPED) *
457 				       HPAGE_PMD_NR),
458 		       nid, K(node_page_state(pgdat, NR_FILE_THPS) *
459 				       HPAGE_PMD_NR),
460 		       nid, K(node_page_state(pgdat, NR_FILE_PMDMAPPED) *
461 				       HPAGE_PMD_NR)
462 #endif
463 		       );
464 	n += hugetlb_report_node_meminfo(nid, buf + n);
465 	return n;
466 }
467 
468 #undef K
469 static DEVICE_ATTR(meminfo, S_IRUGO, node_read_meminfo, NULL);
470 
471 static ssize_t node_read_numastat(struct device *dev,
472 				struct device_attribute *attr, char *buf)
473 {
474 	return sprintf(buf,
475 		       "numa_hit %lu\n"
476 		       "numa_miss %lu\n"
477 		       "numa_foreign %lu\n"
478 		       "interleave_hit %lu\n"
479 		       "local_node %lu\n"
480 		       "other_node %lu\n",
481 		       sum_zone_numa_state(dev->id, NUMA_HIT),
482 		       sum_zone_numa_state(dev->id, NUMA_MISS),
483 		       sum_zone_numa_state(dev->id, NUMA_FOREIGN),
484 		       sum_zone_numa_state(dev->id, NUMA_INTERLEAVE_HIT),
485 		       sum_zone_numa_state(dev->id, NUMA_LOCAL),
486 		       sum_zone_numa_state(dev->id, NUMA_OTHER));
487 }
488 static DEVICE_ATTR(numastat, S_IRUGO, node_read_numastat, NULL);
489 
490 static ssize_t node_read_vmstat(struct device *dev,
491 				struct device_attribute *attr, char *buf)
492 {
493 	int nid = dev->id;
494 	struct pglist_data *pgdat = NODE_DATA(nid);
495 	int i;
496 	int n = 0;
497 
498 	for (i = 0; i < NR_VM_ZONE_STAT_ITEMS; i++)
499 		n += sprintf(buf+n, "%s %lu\n", zone_stat_name(i),
500 			     sum_zone_node_page_state(nid, i));
501 
502 #ifdef CONFIG_NUMA
503 	for (i = 0; i < NR_VM_NUMA_STAT_ITEMS; i++)
504 		n += sprintf(buf+n, "%s %lu\n", numa_stat_name(i),
505 			     sum_zone_numa_state(nid, i));
506 #endif
507 
508 	for (i = 0; i < NR_VM_NODE_STAT_ITEMS; i++)
509 		n += sprintf(buf+n, "%s %lu\n", node_stat_name(i),
510 			     node_page_state(pgdat, i));
511 
512 	return n;
513 }
514 static DEVICE_ATTR(vmstat, S_IRUGO, node_read_vmstat, NULL);
515 
516 static ssize_t node_read_distance(struct device *dev,
517 			struct device_attribute *attr, char *buf)
518 {
519 	int nid = dev->id;
520 	int len = 0;
521 	int i;
522 
523 	/*
524 	 * buf is currently PAGE_SIZE in length and each node needs 4 chars
525 	 * at the most (distance + space or newline).
526 	 */
527 	BUILD_BUG_ON(MAX_NUMNODES * 4 > PAGE_SIZE);
528 
529 	for_each_online_node(i)
530 		len += sprintf(buf + len, "%s%d", i ? " " : "", node_distance(nid, i));
531 
532 	len += sprintf(buf + len, "\n");
533 	return len;
534 }
535 static DEVICE_ATTR(distance, S_IRUGO, node_read_distance, NULL);
536 
537 static struct attribute *node_dev_attrs[] = {
538 	&dev_attr_cpumap.attr,
539 	&dev_attr_cpulist.attr,
540 	&dev_attr_meminfo.attr,
541 	&dev_attr_numastat.attr,
542 	&dev_attr_distance.attr,
543 	&dev_attr_vmstat.attr,
544 	NULL
545 };
546 ATTRIBUTE_GROUPS(node_dev);
547 
548 #ifdef CONFIG_HUGETLBFS
549 /*
550  * hugetlbfs per node attributes registration interface:
551  * When/if hugetlb[fs] subsystem initializes [sometime after this module],
552  * it will register its per node attributes for all online nodes with
553  * memory.  It will also call register_hugetlbfs_with_node(), below, to
554  * register its attribute registration functions with this node driver.
555  * Once these hooks have been initialized, the node driver will call into
556  * the hugetlb module to [un]register attributes for hot-plugged nodes.
557  */
558 static node_registration_func_t __hugetlb_register_node;
559 static node_registration_func_t __hugetlb_unregister_node;
560 
561 static inline bool hugetlb_register_node(struct node *node)
562 {
563 	if (__hugetlb_register_node &&
564 			node_state(node->dev.id, N_MEMORY)) {
565 		__hugetlb_register_node(node);
566 		return true;
567 	}
568 	return false;
569 }
570 
571 static inline void hugetlb_unregister_node(struct node *node)
572 {
573 	if (__hugetlb_unregister_node)
574 		__hugetlb_unregister_node(node);
575 }
576 
577 void register_hugetlbfs_with_node(node_registration_func_t doregister,
578 				  node_registration_func_t unregister)
579 {
580 	__hugetlb_register_node   = doregister;
581 	__hugetlb_unregister_node = unregister;
582 }
583 #else
584 static inline void hugetlb_register_node(struct node *node) {}
585 
586 static inline void hugetlb_unregister_node(struct node *node) {}
587 #endif
588 
589 static void node_device_release(struct device *dev)
590 {
591 	struct node *node = to_node(dev);
592 
593 #if defined(CONFIG_MEMORY_HOTPLUG_SPARSE) && defined(CONFIG_HUGETLBFS)
594 	/*
595 	 * We schedule the work only when a memory section is
596 	 * onlined/offlined on this node. When we come here,
597 	 * all the memory on this node has been offlined,
598 	 * so we won't enqueue new work to this work.
599 	 *
600 	 * The work is using node->node_work, so we should
601 	 * flush work before freeing the memory.
602 	 */
603 	flush_work(&node->node_work);
604 #endif
605 	kfree(node);
606 }
607 
608 /*
609  * register_node - Setup a sysfs device for a node.
610  * @num - Node number to use when creating the device.
611  *
612  * Initialize and register the node device.
613  */
614 static int register_node(struct node *node, int num)
615 {
616 	int error;
617 
618 	node->dev.id = num;
619 	node->dev.bus = &node_subsys;
620 	node->dev.release = node_device_release;
621 	node->dev.groups = node_dev_groups;
622 	error = device_register(&node->dev);
623 
624 	if (error)
625 		put_device(&node->dev);
626 	else {
627 		hugetlb_register_node(node);
628 
629 		compaction_register_node(node);
630 	}
631 	return error;
632 }
633 
634 /**
635  * unregister_node - unregister a node device
636  * @node: node going away
637  *
638  * Unregisters a node device @node.  All the devices on the node must be
639  * unregistered before calling this function.
640  */
641 void unregister_node(struct node *node)
642 {
643 	hugetlb_unregister_node(node);		/* no-op, if memoryless node */
644 	node_remove_accesses(node);
645 	node_remove_caches(node);
646 	device_unregister(&node->dev);
647 }
648 
649 struct node *node_devices[MAX_NUMNODES];
650 
651 /*
652  * register cpu under node
653  */
654 int register_cpu_under_node(unsigned int cpu, unsigned int nid)
655 {
656 	int ret;
657 	struct device *obj;
658 
659 	if (!node_online(nid))
660 		return 0;
661 
662 	obj = get_cpu_device(cpu);
663 	if (!obj)
664 		return 0;
665 
666 	ret = sysfs_create_link(&node_devices[nid]->dev.kobj,
667 				&obj->kobj,
668 				kobject_name(&obj->kobj));
669 	if (ret)
670 		return ret;
671 
672 	return sysfs_create_link(&obj->kobj,
673 				 &node_devices[nid]->dev.kobj,
674 				 kobject_name(&node_devices[nid]->dev.kobj));
675 }
676 
677 /**
678  * register_memory_node_under_compute_node - link memory node to its compute
679  *					     node for a given access class.
680  * @mem_nid:	Memory node number
681  * @cpu_nid:	Cpu  node number
682  * @access:	Access class to register
683  *
684  * Description:
685  * 	For use with platforms that may have separate memory and compute nodes.
686  * 	This function will export node relationships linking which memory
687  * 	initiator nodes can access memory targets at a given ranked access
688  * 	class.
689  */
690 int register_memory_node_under_compute_node(unsigned int mem_nid,
691 					    unsigned int cpu_nid,
692 					    unsigned access)
693 {
694 	struct node *init_node, *targ_node;
695 	struct node_access_nodes *initiator, *target;
696 	int ret;
697 
698 	if (!node_online(cpu_nid) || !node_online(mem_nid))
699 		return -ENODEV;
700 
701 	init_node = node_devices[cpu_nid];
702 	targ_node = node_devices[mem_nid];
703 	initiator = node_init_node_access(init_node, access);
704 	target = node_init_node_access(targ_node, access);
705 	if (!initiator || !target)
706 		return -ENOMEM;
707 
708 	ret = sysfs_add_link_to_group(&initiator->dev.kobj, "targets",
709 				      &targ_node->dev.kobj,
710 				      dev_name(&targ_node->dev));
711 	if (ret)
712 		return ret;
713 
714 	ret = sysfs_add_link_to_group(&target->dev.kobj, "initiators",
715 				      &init_node->dev.kobj,
716 				      dev_name(&init_node->dev));
717 	if (ret)
718 		goto err;
719 
720 	return 0;
721  err:
722 	sysfs_remove_link_from_group(&initiator->dev.kobj, "targets",
723 				     dev_name(&targ_node->dev));
724 	return ret;
725 }
726 
727 int unregister_cpu_under_node(unsigned int cpu, unsigned int nid)
728 {
729 	struct device *obj;
730 
731 	if (!node_online(nid))
732 		return 0;
733 
734 	obj = get_cpu_device(cpu);
735 	if (!obj)
736 		return 0;
737 
738 	sysfs_remove_link(&node_devices[nid]->dev.kobj,
739 			  kobject_name(&obj->kobj));
740 	sysfs_remove_link(&obj->kobj,
741 			  kobject_name(&node_devices[nid]->dev.kobj));
742 
743 	return 0;
744 }
745 
746 #ifdef CONFIG_MEMORY_HOTPLUG_SPARSE
747 static int __ref get_nid_for_pfn(unsigned long pfn)
748 {
749 	if (!pfn_valid_within(pfn))
750 		return -1;
751 #ifdef CONFIG_DEFERRED_STRUCT_PAGE_INIT
752 	if (system_state < SYSTEM_RUNNING)
753 		return early_pfn_to_nid(pfn);
754 #endif
755 	return pfn_to_nid(pfn);
756 }
757 
758 /* register memory section under specified node if it spans that node */
759 static int register_mem_sect_under_node(struct memory_block *mem_blk,
760 					 void *arg)
761 {
762 	unsigned long memory_block_pfns = memory_block_size_bytes() / PAGE_SIZE;
763 	unsigned long start_pfn = section_nr_to_pfn(mem_blk->start_section_nr);
764 	unsigned long end_pfn = start_pfn + memory_block_pfns - 1;
765 	int ret, nid = *(int *)arg;
766 	unsigned long pfn;
767 
768 	for (pfn = start_pfn; pfn <= end_pfn; pfn++) {
769 		int page_nid;
770 
771 		/*
772 		 * memory block could have several absent sections from start.
773 		 * skip pfn range from absent section
774 		 */
775 		if (!pfn_present(pfn)) {
776 			pfn = round_down(pfn + PAGES_PER_SECTION,
777 					 PAGES_PER_SECTION) - 1;
778 			continue;
779 		}
780 
781 		/*
782 		 * We need to check if page belongs to nid only for the boot
783 		 * case, during hotplug we know that all pages in the memory
784 		 * block belong to the same node.
785 		 */
786 		if (system_state == SYSTEM_BOOTING) {
787 			page_nid = get_nid_for_pfn(pfn);
788 			if (page_nid < 0)
789 				continue;
790 			if (page_nid != nid)
791 				continue;
792 		}
793 
794 		/*
795 		 * If this memory block spans multiple nodes, we only indicate
796 		 * the last processed node.
797 		 */
798 		mem_blk->nid = nid;
799 
800 		ret = sysfs_create_link_nowarn(&node_devices[nid]->dev.kobj,
801 					&mem_blk->dev.kobj,
802 					kobject_name(&mem_blk->dev.kobj));
803 		if (ret)
804 			return ret;
805 
806 		return sysfs_create_link_nowarn(&mem_blk->dev.kobj,
807 				&node_devices[nid]->dev.kobj,
808 				kobject_name(&node_devices[nid]->dev.kobj));
809 	}
810 	/* mem section does not span the specified node */
811 	return 0;
812 }
813 
814 /*
815  * Unregister a memory block device under the node it spans. Memory blocks
816  * with multiple nodes cannot be offlined and therefore also never be removed.
817  */
818 void unregister_memory_block_under_nodes(struct memory_block *mem_blk)
819 {
820 	if (mem_blk->nid == NUMA_NO_NODE)
821 		return;
822 
823 	sysfs_remove_link(&node_devices[mem_blk->nid]->dev.kobj,
824 			  kobject_name(&mem_blk->dev.kobj));
825 	sysfs_remove_link(&mem_blk->dev.kobj,
826 			  kobject_name(&node_devices[mem_blk->nid]->dev.kobj));
827 }
828 
829 int link_mem_sections(int nid, unsigned long start_pfn, unsigned long end_pfn)
830 {
831 	return walk_memory_blocks(PFN_PHYS(start_pfn),
832 				  PFN_PHYS(end_pfn - start_pfn), (void *)&nid,
833 				  register_mem_sect_under_node);
834 }
835 
836 #ifdef CONFIG_HUGETLBFS
837 /*
838  * Handle per node hstate attribute [un]registration on transistions
839  * to/from memoryless state.
840  */
841 static void node_hugetlb_work(struct work_struct *work)
842 {
843 	struct node *node = container_of(work, struct node, node_work);
844 
845 	/*
846 	 * We only get here when a node transitions to/from memoryless state.
847 	 * We can detect which transition occurred by examining whether the
848 	 * node has memory now.  hugetlb_register_node() already check this
849 	 * so we try to register the attributes.  If that fails, then the
850 	 * node has transitioned to memoryless, try to unregister the
851 	 * attributes.
852 	 */
853 	if (!hugetlb_register_node(node))
854 		hugetlb_unregister_node(node);
855 }
856 
857 static void init_node_hugetlb_work(int nid)
858 {
859 	INIT_WORK(&node_devices[nid]->node_work, node_hugetlb_work);
860 }
861 
862 static int node_memory_callback(struct notifier_block *self,
863 				unsigned long action, void *arg)
864 {
865 	struct memory_notify *mnb = arg;
866 	int nid = mnb->status_change_nid;
867 
868 	switch (action) {
869 	case MEM_ONLINE:
870 	case MEM_OFFLINE:
871 		/*
872 		 * offload per node hstate [un]registration to a work thread
873 		 * when transitioning to/from memoryless state.
874 		 */
875 		if (nid != NUMA_NO_NODE)
876 			schedule_work(&node_devices[nid]->node_work);
877 		break;
878 
879 	case MEM_GOING_ONLINE:
880 	case MEM_GOING_OFFLINE:
881 	case MEM_CANCEL_ONLINE:
882 	case MEM_CANCEL_OFFLINE:
883 	default:
884 		break;
885 	}
886 
887 	return NOTIFY_OK;
888 }
889 #endif	/* CONFIG_HUGETLBFS */
890 #endif /* CONFIG_MEMORY_HOTPLUG_SPARSE */
891 
892 #if !defined(CONFIG_MEMORY_HOTPLUG_SPARSE) || \
893     !defined(CONFIG_HUGETLBFS)
894 static inline int node_memory_callback(struct notifier_block *self,
895 				unsigned long action, void *arg)
896 {
897 	return NOTIFY_OK;
898 }
899 
900 static void init_node_hugetlb_work(int nid) { }
901 
902 #endif
903 
904 int __register_one_node(int nid)
905 {
906 	int error;
907 	int cpu;
908 
909 	node_devices[nid] = kzalloc(sizeof(struct node), GFP_KERNEL);
910 	if (!node_devices[nid])
911 		return -ENOMEM;
912 
913 	error = register_node(node_devices[nid], nid);
914 
915 	/* link cpu under this node */
916 	for_each_present_cpu(cpu) {
917 		if (cpu_to_node(cpu) == nid)
918 			register_cpu_under_node(cpu, nid);
919 	}
920 
921 	INIT_LIST_HEAD(&node_devices[nid]->access_list);
922 	/* initialize work queue for memory hot plug */
923 	init_node_hugetlb_work(nid);
924 	node_init_caches(nid);
925 
926 	return error;
927 }
928 
929 void unregister_one_node(int nid)
930 {
931 	if (!node_devices[nid])
932 		return;
933 
934 	unregister_node(node_devices[nid]);
935 	node_devices[nid] = NULL;
936 }
937 
938 /*
939  * node states attributes
940  */
941 
942 static ssize_t print_nodes_state(enum node_states state, char *buf)
943 {
944 	int n;
945 
946 	n = scnprintf(buf, PAGE_SIZE - 1, "%*pbl",
947 		      nodemask_pr_args(&node_states[state]));
948 	buf[n++] = '\n';
949 	buf[n] = '\0';
950 	return n;
951 }
952 
953 struct node_attr {
954 	struct device_attribute attr;
955 	enum node_states state;
956 };
957 
958 static ssize_t show_node_state(struct device *dev,
959 			       struct device_attribute *attr, char *buf)
960 {
961 	struct node_attr *na = container_of(attr, struct node_attr, attr);
962 	return print_nodes_state(na->state, buf);
963 }
964 
965 #define _NODE_ATTR(name, state) \
966 	{ __ATTR(name, 0444, show_node_state, NULL), state }
967 
968 static struct node_attr node_state_attr[] = {
969 	[N_POSSIBLE] = _NODE_ATTR(possible, N_POSSIBLE),
970 	[N_ONLINE] = _NODE_ATTR(online, N_ONLINE),
971 	[N_NORMAL_MEMORY] = _NODE_ATTR(has_normal_memory, N_NORMAL_MEMORY),
972 #ifdef CONFIG_HIGHMEM
973 	[N_HIGH_MEMORY] = _NODE_ATTR(has_high_memory, N_HIGH_MEMORY),
974 #endif
975 	[N_MEMORY] = _NODE_ATTR(has_memory, N_MEMORY),
976 	[N_CPU] = _NODE_ATTR(has_cpu, N_CPU),
977 };
978 
979 static struct attribute *node_state_attrs[] = {
980 	&node_state_attr[N_POSSIBLE].attr.attr,
981 	&node_state_attr[N_ONLINE].attr.attr,
982 	&node_state_attr[N_NORMAL_MEMORY].attr.attr,
983 #ifdef CONFIG_HIGHMEM
984 	&node_state_attr[N_HIGH_MEMORY].attr.attr,
985 #endif
986 	&node_state_attr[N_MEMORY].attr.attr,
987 	&node_state_attr[N_CPU].attr.attr,
988 	NULL
989 };
990 
991 static struct attribute_group memory_root_attr_group = {
992 	.attrs = node_state_attrs,
993 };
994 
995 static const struct attribute_group *cpu_root_attr_groups[] = {
996 	&memory_root_attr_group,
997 	NULL,
998 };
999 
1000 #define NODE_CALLBACK_PRI	2	/* lower than SLAB */
1001 static int __init register_node_type(void)
1002 {
1003 	int ret;
1004 
1005  	BUILD_BUG_ON(ARRAY_SIZE(node_state_attr) != NR_NODE_STATES);
1006  	BUILD_BUG_ON(ARRAY_SIZE(node_state_attrs)-1 != NR_NODE_STATES);
1007 
1008 	ret = subsys_system_register(&node_subsys, cpu_root_attr_groups);
1009 	if (!ret) {
1010 		static struct notifier_block node_memory_callback_nb = {
1011 			.notifier_call = node_memory_callback,
1012 			.priority = NODE_CALLBACK_PRI,
1013 		};
1014 		register_hotmemory_notifier(&node_memory_callback_nb);
1015 	}
1016 
1017 	/*
1018 	 * Note:  we're not going to unregister the node class if we fail
1019 	 * to register the node state class attribute files.
1020 	 */
1021 	return ret;
1022 }
1023 postcore_initcall(register_node_type);
1024