km fixup

Signed-off-by: Daniel Maslowski <[email protected]>
linuxboot · Sep 22, 2024 · 546b561 · 546b561
1 parent 044a9ee
commit 546b561
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/cmds/km/main.go b/cmds/km/main.go
@@ -31,10 +31,9 @@ func main() {
 		log.Fatalf("cannot read input file: %v", err)
 	}
 
-	KEYMMagic := []byte("__KEYM__")
-	offset := bytes.Index(data, KEYMMagic)
-	if offset == -1 {
-		log.Fatalf("no %v magic (%x) found", string(KEYMMagic), KEYMMagic)
+	keymMagic := []byte("__KEYM__")
+	if !bytes.Contains(data, keymMagic) {
+		log.Fatalf("no %v magic (%x) found", string(keymMagic), keymMagic)
 	}
 
 	m := cbntkey.Manifest{}