git: 9front

Download patch

ref: 80ccbe0b30bdc378f088a5477cead1df7851f097
parent: 4d402cb0032f93484194ae0f370b7053bc9ef05c
author: cinap_lenrek <cinap_lenrek@felloff.net>
date: Fri Dec 2 15:08:36 EST 2022

9boot: delay GetMemoryMap until right before ExitBootServices (thanks michael forney!)

Michael Forney reported the following:

The UEFI specification suggests calling GetMemoryMap immediately
before ExitBootServices to ensure that the loader has the current
memory map. The firmware is able to enforce this since we have to
pass a "MapKey" from the last GetMemoryMap to ExitBootServices.

The T14 AMD gen 1 firmware is quite strict about this, and even a call
to OutputString will invalidate the MapKey. This causes
ExitBootServices to fail, and we enter 9front with boot services still
running. This causes all sorts of problems including leaving the
IOMMU enabled, breaking 9front's PCI drivers.

---

Michael sent a patch that moves memconf() to unload(), but i prefer
to make this explicit by calling memconf() before and keep unload() as
just that, not touching the configuration.

--- a/sys/src/boot/efi/efi.c
+++ b/sys/src/boot/efi/efi.c
@@ -42,7 +42,7 @@
 	eficall(ST->BootServices->ExitBootServices, IH, MK);
 }
 
-static void
+void
 memconf(char **cfg)
 {
 	static uchar memtype[EfiMaxMemoryType] = {
@@ -73,6 +73,10 @@
 	if(eficall(ST->BootServices->GetMemoryMap, &mapsize, mapbuf, &MK, &entsize, &entvers))
 		return;
 
+	/* only called to get MK for ExitBootServices() */
+	if(cfg == nil)
+		return;
+
 	s = *cfg;
 	for(p = mapbuf; mapsize >= entsize; p += entsize, mapsize -= entsize){
 		t = (EFI_MEMORY_DESCRIPTOR*)p;
@@ -93,7 +97,7 @@
 	*s = '\0';
 	if(s > *cfg){
 		s[-1] = '\n';
-		print(*cfg);
+		/* print(*cfg); -- no printing allowed, can change MK */
 		*cfg = s;
 	}
 }
@@ -276,7 +280,7 @@
 void
 eficonfig(char **cfg)
 {
-	memconf(cfg);
+	/* memconf(cfg); -- must be called right before unload() */
 	acpiconf(cfg);
 	screenconf(cfg);
 }
--- a/sys/src/boot/efi/fns.h
+++ b/sys/src/boot/efi/fns.h
@@ -37,3 +37,4 @@
 
 uintptr eficall(void *proc, ...);
 void eficonfig(char **cfg);
+void memconf(char**);
--- a/sys/src/boot/efi/sub.c
+++ b/sys/src/boot/efi/sub.c
@@ -154,50 +154,53 @@
 #define BOOTARGS	((char*)(CONFADDR+BOOTLINELEN))
 #define	BOOTARGSLEN	(4096-0x200-BOOTLINELEN)
 
-char *confend;
+static char *confend;
 
 static char*
-getconf(char *s, char *buf)
+findconf(char *s)
 {
 	char *p, *e;
-	int n;
-
-	n = strlen(s);
+	int n = strlen(s);
 	for(p = BOOTARGS; p < confend; p = e+1){
-		for(e = p+1; e < confend; e++)
-			if(*e == '\n')
-				break;
-		if(memcmp(p, s, n) == 0){
-			p += n;
-			n = e - p;
-			buf[n] = 0;
-			memmove(buf, p, n);
-			return buf;
+		for(e = p; e < confend && *e != '\n'; e++){
+			if(*e == '\0')
+				return nil;
 		}
+		if(e - p >= n && memcmp(p, s, n) == 0)
+			return p;
 	}
 	return nil;
 }
 
+static char*
+getconf(char *s, char *buf)
+{
+	char *p, *e;
+
+	if((p = findconf(s)) == nil)
+		return nil;
+	p += strlen(s);
+	for(e = p; *e != '\n'; e++)
+		;
+	memmove(buf, p, e - p);
+	buf[e - p] = '\0';
+	return buf;
+}
+
 static int
 delconf(char *s)
 {
 	char *p, *e;
 
-	for(p = BOOTARGS; p < confend; p = e){
-		for(e = p+1; e < confend; e++){
-			if(*e == '\n'){
-				e++;
-				break;
-			}
-		}
-		if(memcmp(p, s, strlen(s)) == 0){
-			memmove(p, e, confend - e);
-			confend -= e - p;
-			*confend = 0;
-			return 1;
-		}
-	}
-	return 0;
+	if((p = findconf(s)) == nil)
+		return 0;
+	for(e = p; *e != '\n'; e++)
+		;
+	e++;
+	memmove(p, e, confend - e);
+	confend -= e - p;
+	*confend = '\0';
+	return 1;
 }
 
 char*
@@ -364,6 +367,8 @@
 
 	close(f);
 	print("boot\n");
+
+	memconf(findconf("*e820=")?nil:&confend);
 	unload();
 
 	jump(e);
--