It will return memory size in specified range according to lmb.memory.region Try to share some code with lmb_free_memory_size() by passing get_free to __lmb_memory_size(). Signed-off-by: Yinghai Lu <yinghai@xxxxxxxxxx> --- arch/x86/include/asm/lmb.h | 1 + arch/x86/mm/lmb.c | 18 +++++++++++++++++- 2 files changed, 18 insertions(+), 1 deletions(-) diff --git a/arch/x86/include/asm/lmb.h b/arch/x86/include/asm/lmb.h index 4fb94b5..dd42ac1 100644 --- a/arch/x86/include/asm/lmb.h +++ b/arch/x86/include/asm/lmb.h @@ -17,5 +17,6 @@ void lmb_register_active_regions(int nid, unsigned long start_pfn, u64 lmb_hole_size(u64 start, u64 end); u64 lmb_find_area_node(int nid, u64 start, u64 end, u64 size, u64 align); u64 lmb_free_memory_size(u64 addr, u64 limit); +u64 lmb_memory_size(u64 addr, u64 limit); #endif diff --git a/arch/x86/mm/lmb.c b/arch/x86/mm/lmb.c index 3a16cdc..15fd3e4 100644 --- a/arch/x86/mm/lmb.c +++ b/arch/x86/mm/lmb.c @@ -226,7 +226,7 @@ void __init lmb_to_bootmem(u64 start, u64 end) } #endif -u64 __init lmb_free_memory_size(u64 addr, u64 limit) +static u64 __init __lmb_memory_size(u64 addr, u64 limit, bool get_free) { int i, count; struct range *range; @@ -256,6 +256,10 @@ u64 __init lmb_free_memory_size(u64 addr, u64 limit) } subtract_range(range, count, 0, addr); subtract_range(range, count, limit, -1ULL); + + /* Subtract lmb.reserved.region in range ? */ + if (!get_free) + goto sort_and_count_them; for (i = 0; i < lmb.reserved.cnt; i++) { struct lmb_region *r = &lmb.reserved.regions[i]; @@ -268,6 +272,8 @@ u64 __init lmb_free_memory_size(u64 addr, u64 limit) subtract_range(range, count, final_start, final_end); } + +sort_and_count_them: nr_range = clean_sort_range(range, count); free_size = 0; @@ -277,6 +283,16 @@ u64 __init lmb_free_memory_size(u64 addr, u64 limit) return free_size << PAGE_SHIFT; } +u64 __init lmb_free_memory_size(u64 addr, u64 limit) +{ + return __lmb_memory_size(addr, limit, true); +} + +u64 __init lmb_memory_size(u64 addr, u64 limit) +{ + return __lmb_memory_size(addr, limit, false); +} + void __init lmb_add_memory(u64 start, u64 end) { lmb_add(start, end - start); -- 1.6.4.2 -- To unsubscribe from this list: send the line "unsubscribe linux-arch" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html