Solve a corner case of dealloc hooking on JS derived class objects using MobileSubstrate.

[cycript.git] / Application.mm

diff --git a/Application.mm b/Application.mm
index 46b24ac64b0a870b31c6b7d62e55ea43df5fc0a5..e910f66f6e221beea067f8d1722a6eec1fa434ce 100644 (file)
--- a/Application.mm
+++ b/Application.mm
@@ -244,7 +244,11 @@ int main(int argc, const char *argv[]) {
 
         if (size >= 2 && start[0] == '#' && start[1] == '!') {
             start += 2;
-            while (start != end && *start++ != '\n');
+
+            if (void *line = memchr(start, '\n', end - start))
+                start = reinterpret_cast<char *>(line);
+            else
+                start = end;
         }
 
         driver.data_ = start;