xref: /openbmc/linux/drivers/base/node.c (revision 113094f7)
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 #endif
431 			,
432 		       nid, K(node_page_state(pgdat, NR_FILE_DIRTY)),
433 		       nid, K(node_page_state(pgdat, NR_WRITEBACK)),
434 		       nid, K(node_page_state(pgdat, NR_FILE_PAGES)),
435 		       nid, K(node_page_state(pgdat, NR_FILE_MAPPED)),
436 		       nid, K(node_page_state(pgdat, NR_ANON_MAPPED)),
437 		       nid, K(i.sharedram),
438 		       nid, sum_zone_node_page_state(nid, NR_KERNEL_STACK_KB),
439 		       nid, K(sum_zone_node_page_state(nid, NR_PAGETABLE)),
440 		       nid, K(node_page_state(pgdat, NR_UNSTABLE_NFS)),
441 		       nid, K(sum_zone_node_page_state(nid, NR_BOUNCE)),
442 		       nid, K(node_page_state(pgdat, NR_WRITEBACK_TEMP)),
443 		       nid, K(sreclaimable +
444 			      node_page_state(pgdat, NR_KERNEL_MISC_RECLAIMABLE)),
445 		       nid, K(sreclaimable + sunreclaimable),
446 		       nid, K(sreclaimable),
447 		       nid, K(sunreclaimable)
448 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
449 		       ,
450 		       nid, K(node_page_state(pgdat, NR_ANON_THPS) *
451 				       HPAGE_PMD_NR),
452 		       nid, K(node_page_state(pgdat, NR_SHMEM_THPS) *
453 				       HPAGE_PMD_NR),
454 		       nid, K(node_page_state(pgdat, NR_SHMEM_PMDMAPPED) *
455 				       HPAGE_PMD_NR)
456 #endif
457 		       );
458 	n += hugetlb_report_node_meminfo(nid, buf + n);
459 	return n;
460 }
461 
462 #undef K
463 static DEVICE_ATTR(meminfo, S_IRUGO, node_read_meminfo, NULL);
464 
465 static ssize_t node_read_numastat(struct device *dev,
466 				struct device_attribute *attr, char *buf)
467 {
468 	return sprintf(buf,
469 		       "numa_hit %lu\n"
470 		       "numa_miss %lu\n"
471 		       "numa_foreign %lu\n"
472 		       "interleave_hit %lu\n"
473 		       "local_node %lu\n"
474 		       "other_node %lu\n",
475 		       sum_zone_numa_state(dev->id, NUMA_HIT),
476 		       sum_zone_numa_state(dev->id, NUMA_MISS),
477 		       sum_zone_numa_state(dev->id, NUMA_FOREIGN),
478 		       sum_zone_numa_state(dev->id, NUMA_INTERLEAVE_HIT),
479 		       sum_zone_numa_state(dev->id, NUMA_LOCAL),
480 		       sum_zone_numa_state(dev->id, NUMA_OTHER));
481 }
482 static DEVICE_ATTR(numastat, S_IRUGO, node_read_numastat, NULL);
483 
484 static ssize_t node_read_vmstat(struct device *dev,
485 				struct device_attribute *attr, char *buf)
486 {
487 	int nid = dev->id;
488 	struct pglist_data *pgdat = NODE_DATA(nid);
489 	int i;
490 	int n = 0;
491 
492 	for (i = 0; i < NR_VM_ZONE_STAT_ITEMS; i++)
493 		n += sprintf(buf+n, "%s %lu\n", vmstat_text[i],
494 			     sum_zone_node_page_state(nid, i));
495 
496 #ifdef CONFIG_NUMA
497 	for (i = 0; i < NR_VM_NUMA_STAT_ITEMS; i++)
498 		n += sprintf(buf+n, "%s %lu\n",
499 			     vmstat_text[i + NR_VM_ZONE_STAT_ITEMS],
500 			     sum_zone_numa_state(nid, i));
501 #endif
502 
503 	for (i = 0; i < NR_VM_NODE_STAT_ITEMS; i++)
504 		n += sprintf(buf+n, "%s %lu\n",
505 			     vmstat_text[i + NR_VM_ZONE_STAT_ITEMS +
506 			     NR_VM_NUMA_STAT_ITEMS],
507 			     node_page_state(pgdat, i));
508 
509 	return n;
510 }
511 static DEVICE_ATTR(vmstat, S_IRUGO, node_read_vmstat, NULL);
512 
513 static ssize_t node_read_distance(struct device *dev,
514 			struct device_attribute *attr, char *buf)
515 {
516 	int nid = dev->id;
517 	int len = 0;
518 	int i;
519 
520 	/*
521 	 * buf is currently PAGE_SIZE in length and each node needs 4 chars
522 	 * at the most (distance + space or newline).
523 	 */
524 	BUILD_BUG_ON(MAX_NUMNODES * 4 > PAGE_SIZE);
525 
526 	for_each_online_node(i)
527 		len += sprintf(buf + len, "%s%d", i ? " " : "", node_distance(nid, i));
528 
529 	len += sprintf(buf + len, "\n");
530 	return len;
531 }
532 static DEVICE_ATTR(distance, S_IRUGO, node_read_distance, NULL);
533 
534 static struct attribute *node_dev_attrs[] = {
535 	&dev_attr_cpumap.attr,
536 	&dev_attr_cpulist.attr,
537 	&dev_attr_meminfo.attr,
538 	&dev_attr_numastat.attr,
539 	&dev_attr_distance.attr,
540 	&dev_attr_vmstat.attr,
541 	NULL
542 };
543 ATTRIBUTE_GROUPS(node_dev);
544 
545 #ifdef CONFIG_HUGETLBFS
546 /*
547  * hugetlbfs per node attributes registration interface:
548  * When/if hugetlb[fs] subsystem initializes [sometime after this module],
549  * it will register its per node attributes for all online nodes with
550  * memory.  It will also call register_hugetlbfs_with_node(), below, to
551  * register its attribute registration functions with this node driver.
552  * Once these hooks have been initialized, the node driver will call into
553  * the hugetlb module to [un]register attributes for hot-plugged nodes.
554  */
555 static node_registration_func_t __hugetlb_register_node;
556 static node_registration_func_t __hugetlb_unregister_node;
557 
558 static inline bool hugetlb_register_node(struct node *node)
559 {
560 	if (__hugetlb_register_node &&
561 			node_state(node->dev.id, N_MEMORY)) {
562 		__hugetlb_register_node(node);
563 		return true;
564 	}
565 	return false;
566 }
567 
568 static inline void hugetlb_unregister_node(struct node *node)
569 {
570 	if (__hugetlb_unregister_node)
571 		__hugetlb_unregister_node(node);
572 }
573 
574 void register_hugetlbfs_with_node(node_registration_func_t doregister,
575 				  node_registration_func_t unregister)
576 {
577 	__hugetlb_register_node   = doregister;
578 	__hugetlb_unregister_node = unregister;
579 }
580 #else
581 static inline void hugetlb_register_node(struct node *node) {}
582 
583 static inline void hugetlb_unregister_node(struct node *node) {}
584 #endif
585 
586 static void node_device_release(struct device *dev)
587 {
588 	struct node *node = to_node(dev);
589 
590 #if defined(CONFIG_MEMORY_HOTPLUG_SPARSE) && defined(CONFIG_HUGETLBFS)
591 	/*
592 	 * We schedule the work only when a memory section is
593 	 * onlined/offlined on this node. When we come here,
594 	 * all the memory on this node has been offlined,
595 	 * so we won't enqueue new work to this work.
596 	 *
597 	 * The work is using node->node_work, so we should
598 	 * flush work before freeing the memory.
599 	 */
600 	flush_work(&node->node_work);
601 #endif
602 	kfree(node);
603 }
604 
605 /*
606  * register_node - Setup a sysfs device for a node.
607  * @num - Node number to use when creating the device.
608  *
609  * Initialize and register the node device.
610  */
611 static int register_node(struct node *node, int num)
612 {
613 	int error;
614 
615 	node->dev.id = num;
616 	node->dev.bus = &node_subsys;
617 	node->dev.release = node_device_release;
618 	node->dev.groups = node_dev_groups;
619 	error = device_register(&node->dev);
620 
621 	if (error)
622 		put_device(&node->dev);
623 	else {
624 		hugetlb_register_node(node);
625 
626 		compaction_register_node(node);
627 	}
628 	return error;
629 }
630 
631 /**
632  * unregister_node - unregister a node device
633  * @node: node going away
634  *
635  * Unregisters a node device @node.  All the devices on the node must be
636  * unregistered before calling this function.
637  */
638 void unregister_node(struct node *node)
639 {
640 	hugetlb_unregister_node(node);		/* no-op, if memoryless node */
641 	node_remove_accesses(node);
642 	node_remove_caches(node);
643 	device_unregister(&node->dev);
644 }
645 
646 struct node *node_devices[MAX_NUMNODES];
647 
648 /*
649  * register cpu under node
650  */
651 int register_cpu_under_node(unsigned int cpu, unsigned int nid)
652 {
653 	int ret;
654 	struct device *obj;
655 
656 	if (!node_online(nid))
657 		return 0;
658 
659 	obj = get_cpu_device(cpu);
660 	if (!obj)
661 		return 0;
662 
663 	ret = sysfs_create_link(&node_devices[nid]->dev.kobj,
664 				&obj->kobj,
665 				kobject_name(&obj->kobj));
666 	if (ret)
667 		return ret;
668 
669 	return sysfs_create_link(&obj->kobj,
670 				 &node_devices[nid]->dev.kobj,
671 				 kobject_name(&node_devices[nid]->dev.kobj));
672 }
673 
674 /**
675  * register_memory_node_under_compute_node - link memory node to its compute
676  *					     node for a given access class.
677  * @mem_nid:	Memory node number
678  * @cpu_nid:	Cpu  node number
679  * @access:	Access class to register
680  *
681  * Description:
682  * 	For use with platforms that may have separate memory and compute nodes.
683  * 	This function will export node relationships linking which memory
684  * 	initiator nodes can access memory targets at a given ranked access
685  * 	class.
686  */
687 int register_memory_node_under_compute_node(unsigned int mem_nid,
688 					    unsigned int cpu_nid,
689 					    unsigned access)
690 {
691 	struct node *init_node, *targ_node;
692 	struct node_access_nodes *initiator, *target;
693 	int ret;
694 
695 	if (!node_online(cpu_nid) || !node_online(mem_nid))
696 		return -ENODEV;
697 
698 	init_node = node_devices[cpu_nid];
699 	targ_node = node_devices[mem_nid];
700 	initiator = node_init_node_access(init_node, access);
701 	target = node_init_node_access(targ_node, access);
702 	if (!initiator || !target)
703 		return -ENOMEM;
704 
705 	ret = sysfs_add_link_to_group(&initiator->dev.kobj, "targets",
706 				      &targ_node->dev.kobj,
707 				      dev_name(&targ_node->dev));
708 	if (ret)
709 		return ret;
710 
711 	ret = sysfs_add_link_to_group(&target->dev.kobj, "initiators",
712 				      &init_node->dev.kobj,
713 				      dev_name(&init_node->dev));
714 	if (ret)
715 		goto err;
716 
717 	return 0;
718  err:
719 	sysfs_remove_link_from_group(&initiator->dev.kobj, "targets",
720 				     dev_name(&targ_node->dev));
721 	return ret;
722 }
723 
724 int unregister_cpu_under_node(unsigned int cpu, unsigned int nid)
725 {
726 	struct device *obj;
727 
728 	if (!node_online(nid))
729 		return 0;
730 
731 	obj = get_cpu_device(cpu);
732 	if (!obj)
733 		return 0;
734 
735 	sysfs_remove_link(&node_devices[nid]->dev.kobj,
736 			  kobject_name(&obj->kobj));
737 	sysfs_remove_link(&obj->kobj,
738 			  kobject_name(&node_devices[nid]->dev.kobj));
739 
740 	return 0;
741 }
742 
743 #ifdef CONFIG_MEMORY_HOTPLUG_SPARSE
744 static int __ref get_nid_for_pfn(unsigned long pfn)
745 {
746 	if (!pfn_valid_within(pfn))
747 		return -1;
748 #ifdef CONFIG_DEFERRED_STRUCT_PAGE_INIT
749 	if (system_state < SYSTEM_RUNNING)
750 		return early_pfn_to_nid(pfn);
751 #endif
752 	return pfn_to_nid(pfn);
753 }
754 
755 /* register memory section under specified node if it spans that node */
756 int register_mem_sect_under_node(struct memory_block *mem_blk, void *arg)
757 {
758 	int ret, nid = *(int *)arg;
759 	unsigned long pfn, sect_start_pfn, sect_end_pfn;
760 
761 	mem_blk->nid = nid;
762 
763 	sect_start_pfn = section_nr_to_pfn(mem_blk->start_section_nr);
764 	sect_end_pfn = section_nr_to_pfn(mem_blk->end_section_nr);
765 	sect_end_pfn += PAGES_PER_SECTION - 1;
766 	for (pfn = sect_start_pfn; pfn <= sect_end_pfn; pfn++) {
767 		int page_nid;
768 
769 		/*
770 		 * memory block could have several absent sections from start.
771 		 * skip pfn range from absent section
772 		 */
773 		if (!pfn_present(pfn)) {
774 			pfn = round_down(pfn + PAGES_PER_SECTION,
775 					 PAGES_PER_SECTION) - 1;
776 			continue;
777 		}
778 
779 		/*
780 		 * We need to check if page belongs to nid only for the boot
781 		 * case, during hotplug we know that all pages in the memory
782 		 * block belong to the same node.
783 		 */
784 		if (system_state == SYSTEM_BOOTING) {
785 			page_nid = get_nid_for_pfn(pfn);
786 			if (page_nid < 0)
787 				continue;
788 			if (page_nid != nid)
789 				continue;
790 		}
791 		ret = sysfs_create_link_nowarn(&node_devices[nid]->dev.kobj,
792 					&mem_blk->dev.kobj,
793 					kobject_name(&mem_blk->dev.kobj));
794 		if (ret)
795 			return ret;
796 
797 		return sysfs_create_link_nowarn(&mem_blk->dev.kobj,
798 				&node_devices[nid]->dev.kobj,
799 				kobject_name(&node_devices[nid]->dev.kobj));
800 	}
801 	/* mem section does not span the specified node */
802 	return 0;
803 }
804 
805 /* unregister memory section under all nodes that it spans */
806 int unregister_mem_sect_under_nodes(struct memory_block *mem_blk,
807 				    unsigned long phys_index)
808 {
809 	NODEMASK_ALLOC(nodemask_t, unlinked_nodes, GFP_KERNEL);
810 	unsigned long pfn, sect_start_pfn, sect_end_pfn;
811 
812 	if (!mem_blk) {
813 		NODEMASK_FREE(unlinked_nodes);
814 		return -EFAULT;
815 	}
816 	if (!unlinked_nodes)
817 		return -ENOMEM;
818 	nodes_clear(*unlinked_nodes);
819 
820 	sect_start_pfn = section_nr_to_pfn(phys_index);
821 	sect_end_pfn = sect_start_pfn + PAGES_PER_SECTION - 1;
822 	for (pfn = sect_start_pfn; pfn <= sect_end_pfn; pfn++) {
823 		int nid;
824 
825 		nid = get_nid_for_pfn(pfn);
826 		if (nid < 0)
827 			continue;
828 		if (!node_online(nid))
829 			continue;
830 		if (node_test_and_set(nid, *unlinked_nodes))
831 			continue;
832 		sysfs_remove_link(&node_devices[nid]->dev.kobj,
833 			 kobject_name(&mem_blk->dev.kobj));
834 		sysfs_remove_link(&mem_blk->dev.kobj,
835 			 kobject_name(&node_devices[nid]->dev.kobj));
836 	}
837 	NODEMASK_FREE(unlinked_nodes);
838 	return 0;
839 }
840 
841 int link_mem_sections(int nid, unsigned long start_pfn, unsigned long end_pfn)
842 {
843 	return walk_memory_range(start_pfn, end_pfn, (void *)&nid,
844 					register_mem_sect_under_node);
845 }
846 
847 #ifdef CONFIG_HUGETLBFS
848 /*
849  * Handle per node hstate attribute [un]registration on transistions
850  * to/from memoryless state.
851  */
852 static void node_hugetlb_work(struct work_struct *work)
853 {
854 	struct node *node = container_of(work, struct node, node_work);
855 
856 	/*
857 	 * We only get here when a node transitions to/from memoryless state.
858 	 * We can detect which transition occurred by examining whether the
859 	 * node has memory now.  hugetlb_register_node() already check this
860 	 * so we try to register the attributes.  If that fails, then the
861 	 * node has transitioned to memoryless, try to unregister the
862 	 * attributes.
863 	 */
864 	if (!hugetlb_register_node(node))
865 		hugetlb_unregister_node(node);
866 }
867 
868 static void init_node_hugetlb_work(int nid)
869 {
870 	INIT_WORK(&node_devices[nid]->node_work, node_hugetlb_work);
871 }
872 
873 static int node_memory_callback(struct notifier_block *self,
874 				unsigned long action, void *arg)
875 {
876 	struct memory_notify *mnb = arg;
877 	int nid = mnb->status_change_nid;
878 
879 	switch (action) {
880 	case MEM_ONLINE:
881 	case MEM_OFFLINE:
882 		/*
883 		 * offload per node hstate [un]registration to a work thread
884 		 * when transitioning to/from memoryless state.
885 		 */
886 		if (nid != NUMA_NO_NODE)
887 			schedule_work(&node_devices[nid]->node_work);
888 		break;
889 
890 	case MEM_GOING_ONLINE:
891 	case MEM_GOING_OFFLINE:
892 	case MEM_CANCEL_ONLINE:
893 	case MEM_CANCEL_OFFLINE:
894 	default:
895 		break;
896 	}
897 
898 	return NOTIFY_OK;
899 }
900 #endif	/* CONFIG_HUGETLBFS */
901 #endif /* CONFIG_MEMORY_HOTPLUG_SPARSE */
902 
903 #if !defined(CONFIG_MEMORY_HOTPLUG_SPARSE) || \
904     !defined(CONFIG_HUGETLBFS)
905 static inline int node_memory_callback(struct notifier_block *self,
906 				unsigned long action, void *arg)
907 {
908 	return NOTIFY_OK;
909 }
910 
911 static void init_node_hugetlb_work(int nid) { }
912 
913 #endif
914 
915 int __register_one_node(int nid)
916 {
917 	int error;
918 	int cpu;
919 
920 	node_devices[nid] = kzalloc(sizeof(struct node), GFP_KERNEL);
921 	if (!node_devices[nid])
922 		return -ENOMEM;
923 
924 	error = register_node(node_devices[nid], nid);
925 
926 	/* link cpu under this node */
927 	for_each_present_cpu(cpu) {
928 		if (cpu_to_node(cpu) == nid)
929 			register_cpu_under_node(cpu, nid);
930 	}
931 
932 	INIT_LIST_HEAD(&node_devices[nid]->access_list);
933 	/* initialize work queue for memory hot plug */
934 	init_node_hugetlb_work(nid);
935 	node_init_caches(nid);
936 
937 	return error;
938 }
939 
940 void unregister_one_node(int nid)
941 {
942 	if (!node_devices[nid])
943 		return;
944 
945 	unregister_node(node_devices[nid]);
946 	node_devices[nid] = NULL;
947 }
948 
949 /*
950  * node states attributes
951  */
952 
953 static ssize_t print_nodes_state(enum node_states state, char *buf)
954 {
955 	int n;
956 
957 	n = scnprintf(buf, PAGE_SIZE - 1, "%*pbl",
958 		      nodemask_pr_args(&node_states[state]));
959 	buf[n++] = '\n';
960 	buf[n] = '\0';
961 	return n;
962 }
963 
964 struct node_attr {
965 	struct device_attribute attr;
966 	enum node_states state;
967 };
968 
969 static ssize_t show_node_state(struct device *dev,
970 			       struct device_attribute *attr, char *buf)
971 {
972 	struct node_attr *na = container_of(attr, struct node_attr, attr);
973 	return print_nodes_state(na->state, buf);
974 }
975 
976 #define _NODE_ATTR(name, state) \
977 	{ __ATTR(name, 0444, show_node_state, NULL), state }
978 
979 static struct node_attr node_state_attr[] = {
980 	[N_POSSIBLE] = _NODE_ATTR(possible, N_POSSIBLE),
981 	[N_ONLINE] = _NODE_ATTR(online, N_ONLINE),
982 	[N_NORMAL_MEMORY] = _NODE_ATTR(has_normal_memory, N_NORMAL_MEMORY),
983 #ifdef CONFIG_HIGHMEM
984 	[N_HIGH_MEMORY] = _NODE_ATTR(has_high_memory, N_HIGH_MEMORY),
985 #endif
986 	[N_MEMORY] = _NODE_ATTR(has_memory, N_MEMORY),
987 	[N_CPU] = _NODE_ATTR(has_cpu, N_CPU),
988 };
989 
990 static struct attribute *node_state_attrs[] = {
991 	&node_state_attr[N_POSSIBLE].attr.attr,
992 	&node_state_attr[N_ONLINE].attr.attr,
993 	&node_state_attr[N_NORMAL_MEMORY].attr.attr,
994 #ifdef CONFIG_HIGHMEM
995 	&node_state_attr[N_HIGH_MEMORY].attr.attr,
996 #endif
997 	&node_state_attr[N_MEMORY].attr.attr,
998 	&node_state_attr[N_CPU].attr.attr,
999 	NULL
1000 };
1001 
1002 static struct attribute_group memory_root_attr_group = {
1003 	.attrs = node_state_attrs,
1004 };
1005 
1006 static const struct attribute_group *cpu_root_attr_groups[] = {
1007 	&memory_root_attr_group,
1008 	NULL,
1009 };
1010 
1011 #define NODE_CALLBACK_PRI	2	/* lower than SLAB */
1012 static int __init register_node_type(void)
1013 {
1014 	int ret;
1015 
1016  	BUILD_BUG_ON(ARRAY_SIZE(node_state_attr) != NR_NODE_STATES);
1017  	BUILD_BUG_ON(ARRAY_SIZE(node_state_attrs)-1 != NR_NODE_STATES);
1018 
1019 	ret = subsys_system_register(&node_subsys, cpu_root_attr_groups);
1020 	if (!ret) {
1021 		static struct notifier_block node_memory_callback_nb = {
1022 			.notifier_call = node_memory_callback,
1023 			.priority = NODE_CALLBACK_PRI,
1024 		};
1025 		register_hotmemory_notifier(&node_memory_callback_nb);
1026 	}
1027 
1028 	/*
1029 	 * Note:  we're not going to unregister the node class if we fail
1030 	 * to register the node state class attribute files.
1031 	 */
1032 	return ret;
1033 }
1034 postcore_initcall(register_node_type);
1035