解决LC_CODE_SIGNATURE不存在时写入异常数据的问题

tobefuturer · Mar 23, 2018 · ea11c62 · ea11c62
1 parent b242c47
commit ea11c62
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/source/restore-symbol.m b/source/restore-symbol.m
@@ -177,13 +177,14 @@ void restore_symbol(NSString * inpath, NSString *outpath, NSString *jsonPath, bo
             }
         }
 
-        struct linkedit_data_command *command = (struct linkedit_data_command *)((char *)outData.mutableBytes + codesignature.commandOffset);
-        uint32_t tmp_offset =  command -> dataoff + increase_size_all_without_padding;
-        uint32_t final_offset = vm_addr_round(tmp_offset, 16);
-
-        string_table_padding = final_offset - tmp_offset;
-        command -> dataoff = final_offset;
-
+        if (codesignature) {
+            struct linkedit_data_command *command = (struct linkedit_data_command *)((char *)outData.mutableBytes + codesignature.commandOffset);
+            uint32_t tmp_offset =  command -> dataoff + increase_size_all_without_padding;
+            uint32_t final_offset = vm_addr_round(tmp_offset, 16);
+
+            string_table_padding = final_offset - tmp_offset;
+            command -> dataoff = final_offset;
+        }
     }