From fd74b540012c9448a3f94d6af1b5656fb6e6bfc0 Mon Sep 17 00:00:00 2001
From: Alexey Min <alexey.min@gmail.com>
Date: Tue, 29 Mar 2022 15:36:08 +0300
Subject: [PATCH] config: allow using flash_lk2nd with heimdall too (MR 2174)

One of the main purposes of lk2nd is to have a replacement for
the stock bootloader "Odin mode" on many Samsung devices with
sane fastboot implementation, which supports many more features
like live booting and many others.

This is done by using heimdall to flash lk2nd to device's boot
partition. Allow this method to be used with this flash action.
---
 pmb/config/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pmb/config/__init__.py b/pmb/config/__init__.py
index 97fa673f..4bcf4b37 100644
--- a/pmb/config/__init__.py
+++ b/pmb/config/__init__.py
@@ -897,7 +897,10 @@ flashers = {
                     "--padding_size", "$FLASH_PAGESIZE",
                     "--output", "/vbmeta.img"],
                 ["heimdall", "flash", "--$PARTITION_VBMETA", "/vbmeta.img"],
-                ["rm", "-f", "/vbmeta.img"]]
+                ["rm", "-f", "/vbmeta.img"]],
+            "flash_lk2nd": [
+                ["heimdall_wait_for_device.sh"],
+                ["heimdall", "flash", "--$PARTITION_KERNEL", "$BOOT/lk2nd.img"]]
         },
     },
     "adb": {