Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S965972AbeAOMla (ORCPT + 1 other); Mon, 15 Jan 2018 07:41:30 -0500 Received: from mail.cn.fujitsu.com ([183.91.158.132]:30897 "EHLO heian.cn.fujitsu.com" rhost-flags-OK-OK-OK-FAIL) by vger.kernel.org with ESMTP id S965884AbeAOMlY (ORCPT ); Mon, 15 Jan 2018 07:41:24 -0500 X-IronPort-AV: E=Sophos;i="5.43,368,1503331200"; d="scan'208";a="35325139" From: Chao Fan To: , , , , , , , CC: , , Chao Fan Subject: [PATCH v6 4/5] kaslr: calculate the memory region in kaslr_mem Date: Mon, 15 Jan 2018 20:40:15 +0800 Message-ID: <20180115124016.17683-5-fanc.fnst@cn.fujitsu.com> X-Mailer: git-send-email 2.14.3 In-Reply-To: <20180115124016.17683-1-fanc.fnst@cn.fujitsu.com> References: <20180115124016.17683-1-fanc.fnst@cn.fujitsu.com> MIME-Version: 1.0 Content-Type: text/plain X-yoursite-MailScanner-ID: 45E3048AEA0D.A8EE4 X-yoursite-MailScanner: Found to be clean X-yoursite-MailScanner-From: fanc.fnst@cn.fujitsu.com Sender: linux-kernel-owner@vger.kernel.org List-ID: X-Mailing-List: linux-kernel@vger.kernel.org Return-Path: If there is no kaslr_mem= region specified, use region directely. Otherwise, calculate the intersecting between efi or e820 memmap entry and kaslr_mem. Rename process_mem_region to slots_count to match slots_fetch_random, and rename new function sa process_mem_region. Signed-off-by: Chao Fan --- arch/x86/boot/compressed/kaslr.c | 64 +++++++++++++++++++++++++++++++++------- 1 file changed, 53 insertions(+), 11 deletions(-) diff --git a/arch/x86/boot/compressed/kaslr.c b/arch/x86/boot/compressed/kaslr.c index 5615f26364f9..fc531fa1f10c 100644 --- a/arch/x86/boot/compressed/kaslr.c +++ b/arch/x86/boot/compressed/kaslr.c @@ -558,9 +558,9 @@ static unsigned long slots_fetch_random(void) return 0; } -static void process_mem_region(struct mem_vector *entry, - unsigned long minimum, - unsigned long image_size) +static void slots_count(struct mem_vector *entry, + unsigned long minimum, + unsigned long image_size) { struct mem_vector region, overlap; struct slot_area slot_area; @@ -637,6 +637,52 @@ static void process_mem_region(struct mem_vector *entry, } } +static bool process_mem_region(struct mem_vector region, + unsigned long long minimum, + unsigned long long image_size) +{ + /* + * If kaslr_mem= specified, walk all the regions, and + * filter the intersection to slots_count. + */ + if (num_usable_region > 0) { + int i; + + for (i = 0; i < num_usable_region; i++) { + struct mem_vector entry; + unsigned long long start, end, entry_end, region_end; + + start = mem_usable[i].start; + end = start + mem_usable[i].size; + region_end = region.start + region.size; + + entry.start = clamp(region.start, start, end); + entry_end = clamp(region_end, start, end); + + if (entry.start < entry_end) { + entry.size = entry_end - entry.start; + slots_count(&entry, minimum, image_size); + } + + if (slot_area_index == MAX_SLOT_AREA) { + debug_putstr("Aborted memmap scan (slot_areas full)!\n"); + return 1; + } + } + return 0; + } + + /* + * If no kaslr_mem stored, use region directly + */ + slots_count(®ion, minimum, image_size); + if (slot_area_index == MAX_SLOT_AREA) { + debug_putstr("Aborted memmap scan (slot_areas full)!\n"); + return 1; + } + return 0; +} + #ifdef CONFIG_EFI /* * Returns true if mirror region found (and must have been processed @@ -709,11 +755,9 @@ process_efi_entries(unsigned long minimum, unsigned long image_size) region.start = md->phys_addr; region.size = md->num_pages << EFI_PAGE_SHIFT; - process_mem_region(®ion, minimum, image_size); - if (slot_area_index == MAX_SLOT_AREA) { - debug_putstr("Aborted EFI scan (slot_areas full)!\n"); + + if (process_mem_region(region, minimum, image_size)) break; - } } return true; } @@ -740,11 +784,9 @@ static void process_e820_entries(unsigned long minimum, continue; region.start = entry->addr; region.size = entry->size; - process_mem_region(®ion, minimum, image_size); - if (slot_area_index == MAX_SLOT_AREA) { - debug_putstr("Aborted e820 scan (slot_areas full)!\n"); + + if (process_mem_region(region, minimum, image_size)) break; - } } } -- 2.14.3