1570 static int __init pcpu_verify_alloc_info(const struct pcpu_alloc_info *ai);
2380  * Allocate ai which is large enough for @nr_groups groups containing
2381  * @nr_units units.  The returned ai's groups[0].cpu_map points to the
2393 	struct pcpu_alloc_info *ai;  in pcpu_alloc_alloc_info()  local
2398 	base_size = ALIGN(struct_size(ai, groups, nr_groups),  in pcpu_alloc_alloc_info()
2399 			  __alignof__(ai->groups[0].cpu_map[0]));  in pcpu_alloc_alloc_info()
2400 	ai_size = base_size + nr_units * sizeof(ai->groups[0].cpu_map[0]);  in pcpu_alloc_alloc_info()
2405 	ai = ptr;  in pcpu_alloc_alloc_info()
2408 	ai->groups[0].cpu_map = ptr;  in pcpu_alloc_alloc_info()
2411 		ai->groups[0].cpu_map[unit] = NR_CPUS;  in pcpu_alloc_alloc_info()
2413 	ai->nr_groups = nr_groups;  in pcpu_alloc_alloc_info()
2414 	ai->__ai_size = PFN_ALIGN(ai_size);  in pcpu_alloc_alloc_info()
2416 	return ai;  in pcpu_alloc_alloc_info()
2421  * @ai: pcpu_alloc_info to free
2423  * Free @ai which was allocated by pcpu_alloc_alloc_info().
2425 void __init pcpu_free_alloc_info(struct pcpu_alloc_info *ai)  in pcpu_free_alloc_info()  argument
2427 	memblock_free(ai, ai->__ai_size);  in pcpu_free_alloc_info()
2433  * @ai: allocation info to dump
2435  * Print out information about @ai using loglevel @lvl.
2438 				 const struct pcpu_alloc_info *ai)  in pcpu_dump_alloc_info()  argument
2446 	v = ai->nr_groups;  in pcpu_dump_alloc_info()
2455 	upa = ai->alloc_size / ai->unit_size;  in pcpu_dump_alloc_info()
2460 	       lvl, ai->static_size, ai->reserved_size, ai->dyn_size,  in pcpu_dump_alloc_info()
2461 	       ai->unit_size, ai->alloc_size / ai->atom_size, ai->atom_size);  in pcpu_dump_alloc_info()
2463 	for (group = 0; group < ai->nr_groups; group++) {  in pcpu_dump_alloc_info()
2464 		const struct pcpu_group_info *gi = &ai->groups[group];  in pcpu_dump_alloc_info()
2489  * @ai: pcpu_alloc_info describing how to percpu area is shaped
2496  * @ai contains all information necessary to initialize the first
2499  * @ai->static_size is the size of static percpu area.
2501  * @ai->reserved_size, if non-zero, specifies the amount of bytes to
2509  * @ai->dyn_size determines the number of bytes available for dynamic
2510  * allocation in the first chunk.  The area between @ai->static_size +
2511  * @ai->reserved_size + @ai->dyn_size and @ai->unit_size is unused.
2513  * @ai->unit_size specifies unit size and must be aligned to PAGE_SIZE
2514  * and equal to or larger than @ai->static_size + @ai->reserved_size +
2515  * @ai->dyn_size.
2517  * @ai->atom_size is the allocation atom size and used as alignment
2520  * @ai->alloc_size is the allocation size and always multiple of
2521  * @ai->atom_size.  This is larger than @ai->atom_size if
2522  * @ai->unit_size is larger than @ai->atom_size.
2524  * @ai->nr_groups and @ai->groups describe virtual memory layout of
2527  * groupings.  If @ai->nr_groups is zero, a single group containing
2541 void __init pcpu_setup_first_chunk(const struct pcpu_alloc_info *ai,  in pcpu_setup_first_chunk()  argument
2544 	size_t size_sum = ai->static_size + ai->reserved_size + ai->dyn_size;  in pcpu_setup_first_chunk()
2560 		pcpu_dump_alloc_info(KERN_EMERG, ai);			\  in pcpu_setup_first_chunk()
2566 	PCPU_SETUP_BUG_ON(ai->nr_groups <= 0);  in pcpu_setup_first_chunk()
2568 	PCPU_SETUP_BUG_ON(!ai->static_size);  in pcpu_setup_first_chunk()
2573 	PCPU_SETUP_BUG_ON(ai->unit_size < size_sum);  in pcpu_setup_first_chunk()
2574 	PCPU_SETUP_BUG_ON(offset_in_page(ai->unit_size));  in pcpu_setup_first_chunk()
2575 	PCPU_SETUP_BUG_ON(ai->unit_size < PCPU_MIN_UNIT_SIZE);  in pcpu_setup_first_chunk()
2576 	PCPU_SETUP_BUG_ON(!IS_ALIGNED(ai->unit_size, PCPU_BITMAP_BLOCK_SIZE));  in pcpu_setup_first_chunk()
2577 	PCPU_SETUP_BUG_ON(ai->dyn_size < PERCPU_DYNAMIC_EARLY_SIZE);  in pcpu_setup_first_chunk()
2578 	PCPU_SETUP_BUG_ON(!IS_ALIGNED(ai->reserved_size, PCPU_MIN_ALLOC_SIZE));  in pcpu_setup_first_chunk()
2581 	PCPU_SETUP_BUG_ON(pcpu_verify_alloc_info(ai) < 0);  in pcpu_setup_first_chunk()
2584 	alloc_size = ai->nr_groups * sizeof(group_offsets[0]);  in pcpu_setup_first_chunk()
2587 	alloc_size = ai->nr_groups * sizeof(group_sizes[0]);  in pcpu_setup_first_chunk()
2602 	for (group = 0, unit = 0; group < ai->nr_groups; group++, unit += i) {  in pcpu_setup_first_chunk()
2603 		const struct pcpu_group_info *gi = &ai->groups[group];  in pcpu_setup_first_chunk()
2606 		group_sizes[group] = gi->nr_units * ai->unit_size;  in pcpu_setup_first_chunk()
2618 			unit_off[cpu] = gi->base_offset + i * ai->unit_size;  in pcpu_setup_first_chunk()
2636 	pcpu_dump_alloc_info(KERN_DEBUG, ai);  in pcpu_setup_first_chunk()
2638 	pcpu_nr_groups = ai->nr_groups;  in pcpu_setup_first_chunk()
2645 	pcpu_unit_pages = ai->unit_size >> PAGE_SHIFT;  in pcpu_setup_first_chunk()
2647 	pcpu_atom_size = ai->atom_size;  in pcpu_setup_first_chunk()
2651 	pcpu_stats_save_ai(ai);  in pcpu_setup_first_chunk()
2678 	static_size = ALIGN(ai->static_size, PCPU_MIN_ALLOC_SIZE);  in pcpu_setup_first_chunk()
2679 	dyn_size = ai->dyn_size - (static_size - ai->static_size);  in pcpu_setup_first_chunk()
2693 	if (ai->reserved_size)  in pcpu_setup_first_chunk()
2695 						ai->reserved_size);  in pcpu_setup_first_chunk()
2696 	tmp_addr = (unsigned long)base_addr + static_size + ai->reserved_size;  in pcpu_setup_first_chunk()
2796 	struct pcpu_alloc_info *ai;  in pcpu_build_alloc_info()  local
2886 	ai = pcpu_alloc_alloc_info(nr_groups, nr_units);  in pcpu_build_alloc_info()
2887 	if (!ai)  in pcpu_build_alloc_info()
2889 	cpu_map = ai->groups[0].cpu_map;  in pcpu_build_alloc_info()
2892 		ai->groups[group].cpu_map = cpu_map;  in pcpu_build_alloc_info()
2896 	ai->static_size = static_size;  in pcpu_build_alloc_info()
2897 	ai->reserved_size = reserved_size;  in pcpu_build_alloc_info()
2898 	ai->dyn_size = dyn_size;  in pcpu_build_alloc_info()
2899 	ai->unit_size = alloc_size / upa;  in pcpu_build_alloc_info()
2900 	ai->atom_size = atom_size;  in pcpu_build_alloc_info()
2901 	ai->alloc_size = alloc_size;  in pcpu_build_alloc_info()
2904 		struct pcpu_group_info *gi = &ai->groups[group];  in pcpu_build_alloc_info()
2911 		gi->base_offset = unit * ai->unit_size;  in pcpu_build_alloc_info()
2921 	return ai;  in pcpu_build_alloc_info()
3000 	struct pcpu_alloc_info *ai;  in pcpu_embed_first_chunk()  local
3005 	ai = pcpu_build_alloc_info(reserved_size, dyn_size, atom_size,  in pcpu_embed_first_chunk()
3007 	if (IS_ERR(ai))  in pcpu_embed_first_chunk()
3008 		return PTR_ERR(ai);  in pcpu_embed_first_chunk()
3010 	size_sum = ai->static_size + ai->reserved_size + ai->dyn_size;  in pcpu_embed_first_chunk()
3011 	areas_size = PFN_ALIGN(ai->nr_groups * sizeof(void *));  in pcpu_embed_first_chunk()
3021 	for (group = 0; group < ai->nr_groups; group++) {  in pcpu_embed_first_chunk()
3022 		struct pcpu_group_info *gi = &ai->groups[group];  in pcpu_embed_first_chunk()
3031 		ptr = pcpu_fc_alloc(cpu, gi->nr_units * ai->unit_size, atom_size, cpu_to_nd_fn);  in pcpu_embed_first_chunk()
3045 	max_distance += ai->unit_size * ai->groups[highest_group].nr_units;  in pcpu_embed_first_chunk()
3063 	for (group = 0; group < ai->nr_groups; group++) {  in pcpu_embed_first_chunk()
3064 		struct pcpu_group_info *gi = &ai->groups[group];  in pcpu_embed_first_chunk()
3067 		for (i = 0; i < gi->nr_units; i++, ptr += ai->unit_size) {  in pcpu_embed_first_chunk()
3070 				pcpu_fc_free(ptr, ai->unit_size);  in pcpu_embed_first_chunk()
3074 			memcpy(ptr, __per_cpu_load, ai->static_size);  in pcpu_embed_first_chunk()
3075 			pcpu_fc_free(ptr + size_sum, ai->unit_size - size_sum);  in pcpu_embed_first_chunk()
3080 	for (group = 0; group < ai->nr_groups; group++) {  in pcpu_embed_first_chunk()
3081 		ai->groups[group].base_offset = areas[group] - base;  in pcpu_embed_first_chunk()
3085 		PFN_DOWN(size_sum), ai->static_size, ai->reserved_size,  in pcpu_embed_first_chunk()
3086 		ai->dyn_size, ai->unit_size);  in pcpu_embed_first_chunk()
3088 	pcpu_setup_first_chunk(ai, base);  in pcpu_embed_first_chunk()
3092 	for (group = 0; group < ai->nr_groups; group++)  in pcpu_embed_first_chunk()
3095 				ai->groups[group].nr_units * ai->unit_size);  in pcpu_embed_first_chunk()
3097 	pcpu_free_alloc_info(ai);  in pcpu_embed_first_chunk()
3174 	struct pcpu_alloc_info *ai;  in pcpu_page_first_chunk()  local
3185 	ai = pcpu_build_alloc_info(reserved_size, 0, PAGE_SIZE, NULL);  in pcpu_page_first_chunk()
3186 	if (IS_ERR(ai))  in pcpu_page_first_chunk()
3187 		return PTR_ERR(ai);  in pcpu_page_first_chunk()
3188 	BUG_ON(ai->nr_groups != 1);  in pcpu_page_first_chunk()
3189 	upa = ai->alloc_size/ai->unit_size;  in pcpu_page_first_chunk()
3191 	if (WARN_ON(ai->groups[0].nr_units != nr_g0_units)) {  in pcpu_page_first_chunk()
3192 		pcpu_free_alloc_info(ai);  in pcpu_page_first_chunk()
3196 	unit_pages = ai->unit_size >> PAGE_SHIFT;  in pcpu_page_first_chunk()
3206 		unsigned int cpu = ai->groups[0].cpu_map[unit];  in pcpu_page_first_chunk()
3224 	vm.size = num_possible_cpus() * ai->unit_size;  in pcpu_page_first_chunk()
3229 			(unsigned long)vm.addr + unit * ai->unit_size;  in pcpu_page_first_chunk()
3240 		flush_cache_vmap_early(unit_addr, unit_addr + ai->unit_size);  in pcpu_page_first_chunk()
3243 		memcpy((void *)unit_addr, __per_cpu_load, ai->static_size);  in pcpu_page_first_chunk()
3248 		unit_pages, psize_str, ai->static_size,  in pcpu_page_first_chunk()
3249 		ai->reserved_size, ai->dyn_size);  in pcpu_page_first_chunk()
3251 	pcpu_setup_first_chunk(ai, vm.addr);  in pcpu_page_first_chunk()
3260 	pcpu_free_alloc_info(ai);  in pcpu_page_first_chunk()
3316 	struct pcpu_alloc_info *ai;  in setup_per_cpu_areas()  local
3319 	ai = pcpu_alloc_alloc_info(1, 1);  in setup_per_cpu_areas()
3321 	if (!ai || !fc)  in setup_per_cpu_areas()
3326 	ai->dyn_size = unit_size;  in setup_per_cpu_areas()
3327 	ai->unit_size = unit_size;  in setup_per_cpu_areas()
3328 	ai->atom_size = unit_size;  in setup_per_cpu_areas()
3329 	ai->alloc_size = unit_size;  in setup_per_cpu_areas()
3330 	ai->groups[0].nr_units = 1;  in setup_per_cpu_areas()
3331 	ai->groups[0].cpu_map[0] = 0;  in setup_per_cpu_areas()
3333 	pcpu_setup_first_chunk(ai, fc);  in setup_per_cpu_areas()
3334 	pcpu_free_alloc_info(ai);  in setup_per_cpu_areas()