svn commit: r294345 - stable/10/sys/boot/efi/loader

[Ian Lepore]

Author: ian
Date: Tue Jan 19 21:39:21 2016
New Revision: 294345
URL: https://svnweb.freebsd.org/changeset/base/294345

Log:
  MFC r292584:
  
    Set env vars from values on the efi loader command line.
  
    Examine each cmdline arg and if it contains an '=' convert it to ascii and
    pass it to putenv(). This allows var=value settings to come in on the
    command line.
  
    This will allow overriding dhcp server-provided data in loader(8), as
    discussed in PR 202098
  
  PR:           202098
  Relnotes:	Yes (this goes together with r294343)

Modified:
  stable/10/sys/boot/efi/loader/main.c
Directory Properties:
  stable/10/   (props changed)

Modified: stable/10/sys/boot/efi/loader/main.c
==============================================================================
--- stable/10/sys/boot/efi/loader/main.c	Tue Jan 19 21:37:51 2016	(r294344)
+++ stable/10/sys/boot/efi/loader/main.c	Tue Jan 19 21:39:21 2016	(r294345)
@@ -63,10 +63,10 @@ EFI_GUID debugimg = DEBUG_IMAGE_INFO_TAB
 EFI_STATUS
 main(int argc, CHAR16 *argv[])
 {
-	char vendor[128];
+	char var[128];
 	EFI_LOADED_IMAGE *img;
 	EFI_GUID *guid;
-	int i;
+	int i, j, vargood;
 
 	/*
 	 * XXX Chicken-and-egg problem; we want to have console output
@@ -76,6 +76,29 @@ main(int argc, CHAR16 *argv[])
 	 */
 	cons_probe();
 
+	/*
+	 * Loop through the args, and for each one that contains an '=' that is
+	 * not the first character, add it to the environment.  This allows
+	 * loader and kernel env vars to be passed on the command line.  Convert
+	 * args from UCS-2 to ASCII (16 to 8 bit) as they are copied.
+	 */
+	for (i = 1; i < argc; i++) {
+		vargood = 0;
+		for (j = 0; argv[i][j] != 0; j++) {
+			if (j == sizeof(var)) {
+				vargood = 0;
+				break;
+			}
+			if (j > 0 && argv[i][j] == '=')
+				vargood = 1;
+			var[j] = (char)argv[i][j];
+		}
+		if (vargood) {
+			var[j] = 0;
+			putenv(var);
+		}
+	}
+
 	if (efi_copy_init()) {
 		printf("failed to allocate staging area\n");
 		return (EFI_BUFFER_TOO_SMALL);