diff options
| -rw-r--r-- | arch/x86/boot/compressed/kaslr.c | 13 | 
1 files changed, 8 insertions, 5 deletions
| diff --git a/arch/x86/boot/compressed/kaslr.c b/arch/x86/boot/compressed/kaslr.c index d97647b5ffb7..531c9876f573 100644 --- a/arch/x86/boot/compressed/kaslr.c +++ b/arch/x86/boot/compressed/kaslr.c @@ -244,7 +244,7 @@ static void parse_gb_huge_pages(char *param, char *val)  } -static int handle_mem_memmap(void) +static int handle_mem_options(void)  {  	char *args = (char *)get_cmd_line_ptr();  	size_t len = strlen((char *)args); @@ -252,7 +252,8 @@ static int handle_mem_memmap(void)  	char *param, *val;  	u64 mem_size; -	if (!strstr(args, "memmap=") && !strstr(args, "mem=")) +	if (!strstr(args, "memmap=") && !strstr(args, "mem=") && +		!strstr(args, "hugepages"))  		return 0;  	tmp_cmdline = malloc(len + 1); @@ -277,6 +278,8 @@ static int handle_mem_memmap(void)  		if (!strcmp(param, "memmap")) {  			mem_avoid_memmap(val); +		} else if (strstr(param, "hugepages")) { +			parse_gb_huge_pages(param, val);  		} else if (!strcmp(param, "mem")) {  			char *p = val; @@ -416,7 +419,7 @@ static void mem_avoid_init(unsigned long input, unsigned long input_size,  	/* We don't need to set a mapping for setup_data. */  	/* Mark the memmap regions we need to avoid */ -	handle_mem_memmap(); +	handle_mem_options();  #ifdef CONFIG_X86_VERBOSE_BOOTUP  	/* Make sure video RAM can be used. */ @@ -629,7 +632,7 @@ static void process_mem_region(struct mem_vector *entry,  		/* If nothing overlaps, store the region and return. */  		if (!mem_avoid_overlap(®ion, &overlap)) { -			store_slot_info(®ion, image_size); +			process_gb_huge_pages(®ion, image_size);  			return;  		} @@ -639,7 +642,7 @@ static void process_mem_region(struct mem_vector *entry,  			beginning.start = region.start;  			beginning.size = overlap.start - region.start; -			store_slot_info(&beginning, image_size); +			process_gb_huge_pages(&beginning, image_size);  		}  		/* Return if overlap extends to or past end of region. */ | 
