From 12937c9c78b56f52e1a15247704e39fca0491bf0 Mon Sep 17 00:00:00 2001
From: LA_DI_DA <11174151+0O0o0oOoO00@users.noreply.github.com>
Date: Thu, 20 Mar 2025 12:16:48 +0800
Subject: [PATCH] opt: make injector more automated

---
 blcrack/patchelf/patchelf.cc | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/blcrack/patchelf/patchelf.cc b/blcrack/patchelf/patchelf.cc
index cc7e3aa4a..ca8957b93 100644
--- a/blcrack/patchelf/patchelf.cc
+++ b/blcrack/patchelf/patchelf.cc
@@ -2737,21 +2737,21 @@ using ElfFile32 = ElfFile<Elf32_Ehdr, Elf32_Phdr, Elf32_Shdr, Elf32_Addr, Elf32_
 void do_local_patch(const std::string& game_lib_dir) {
     fs::path lib_dir(game_lib_dir);
     auto libtolua_file = lib_dir / "libtolua.so";
+    auto libtolua_backup_file = lib_dir / "libtolua.so.bak";
 
     ElfFile32 libtolua(readFile(libtolua_file));
 
     auto curr_needed = libtolua.getNeededLibs();
     for (const auto& lib : curr_needed) {
          if (lib == "libcracker.so") {
-             return;
+             goto ret;
          }
     }
-    auto libtolua_backup_file = lib_dir / "libtolua.so.bak";
-    if (!exists(libtolua_backup_file)) {
-        fs::copy(libtolua_file, libtolua_backup_file, fs::copy_options::overwrite_existing);
-    }
+    fs::copy(libtolua_file, libtolua_backup_file, fs::copy_options::overwrite_existing);
     libtolua.addNeeded({"libcracker.so"});
     writeFile(libtolua_file, libtolua.fileContents);
+
+    ret:
     fs::copy("libcracker.so", lib_dir / "libcracker.so", fs::copy_options::overwrite_existing);
 }
 
@@ -2760,19 +2760,19 @@ void do_local_depatch(const std::string& game_lib_dir) {
     auto libtolua_file = lib_dir / "libtolua.so";
     auto libtolua_backup_file = lib_dir / "libtolua.so.bak";
 
+    if (exists(libtolua_backup_file)) {
+        fs::copy(libtolua_backup_file, libtolua_file, fs::copy_options::overwrite_existing);
+        return;
+    }
+
     ElfFile32 libtolua(readFile(libtolua_file));
 
     auto curr_needed = libtolua.getNeededLibs();
     for (const auto& lib : curr_needed) {
         if (lib == "libcracker.so") {
-            if (exists(libtolua_backup_file)) {
-                fs::copy(libtolua_backup_file, libtolua_file, fs::copy_options::overwrite_existing);
-                return;
-            } else {
-                libtolua.removeNeeded({"libcracker.so"});
-                writeFile(libtolua_file, libtolua.fileContents);
-                return;
-            }
+            libtolua.removeNeeded({"libcracker.so"});
+            writeFile(libtolua_file, libtolua.fileContents);
+            return;
         }
     }
 }