Remove accidental newlines from image files.

[safemode-ios.git] / MobileSafety.mm

diff --git a/MobileSafety.mm b/MobileSafety.mm
index 0118e0bb09d34dc739d9135703ec7a35320a8cb0..e31e3274c55eeefb3a5bd7baa6264be777a3b834 100644 (file)
--- a/MobileSafety.mm
+++ b/MobileSafety.mm
@@ -1,74 +1,150 @@
-/* Cydia Substrate - Meta-Library Insert for iPhoneOS
- * Copyright (C) 2008  Jay Freeman (saurik)
+/* Cydia Substrate - Powerful Code Insertion Platform
+ * Copyright (C) 2008-2010  Jay Freeman (saurik)

 */
 
 */
 

+/* GNU Lesser General Public License, Version 3 {{{ */

 /*
 /*

- *        Redistribution and use in source and binary
- * forms, with or without modification, are permitted
- * provided that the following conditions are met:
+ * Substrate is free software: you can redistribute it and/or modify it under
+ * the terms of the GNU Lesser General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or (at your
+ * option) any later version.

  *
  *

- * 1. Redistributions of source code must retain the
- *    above copyright notice, this list of conditions
- *    and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the
- *    above copyright notice, this list of conditions
- *    and the following disclaimer in the documentation
- *    and/or other materials provided with the
- *    distribution.
- * 3. The name of the author may not be used to endorse
- *    or promote products derived from this software
- *    without specific prior written permission.
+ * Substrate is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
+ * License for more details.

  *
  *

- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
- * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
- * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
- * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
- * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
- * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
- * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
- * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
- * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-*/
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Substrate.  If not, see <http://www.gnu.org/licenses/>.
+**/
+/* }}} */

 
 #import <CoreFoundation/CoreFoundation.h>
 #import <Foundation/Foundation.h>
 #import <CoreGraphics/CGGeometry.h>
 
 #import <CoreFoundation/CoreFoundation.h>
 #import <Foundation/Foundation.h>
 #import <CoreGraphics/CGGeometry.h>

+#import <UIKit/UIKit.h>
+
+#include "CydiaSubstrate.h"
+
+@class SBButtonBar;
+@class SBStatusBar;
+@class SBUIController;
+
+Class $SafeModeAlertItem;
+Class $SBAlertItemsController;

 
 

-#include <substrate.h>
-
-@protocol MobileSubstrate
-- (id) sharedInstance;
-- (void) activateAlertItem:(id)item;
-- (id) initWithTitle:(NSString *)title body:(NSString *)body;
-- (id) ms$initWithSize:(CGSize)size;
-- (void) ms$drawRect:(CGRect)rect;
-- (id) darkGrayColor;
-- (void) setBackgroundColor:(id)color;
-- (int) ms$maxIconColumns;
+@interface SBAlertItem : NSObject {
+}
+- (UIAlertView *) alertSheet;
+- (void) dismiss;

 @end
 
 @end
 

-static void MSAlert(id self, SEL sel) {
-    static bool loaded = false;
-    if (!loaded)
-        loaded = true;
-    else return;
+@interface SBAlertItemsController : NSObject {
+}
++ (SBAlertItemsController *) sharedInstance;
+- (void) activateAlertItem:(SBAlertItem *)item;
+@end
+
+@interface SBStatusBarTimeView : UIView {
+}
+- (id) textFont;
+@end
+
+@interface UIApplication (CydiaSubstrate)
+- (void) applicationOpenURL:(id)url;
+@end
+
+@interface UIAlertView (CydiaSubstrate)
+- (void) setForceHorizontalButtonsLayout:(BOOL)force;
+- (void) setBodyText:(NSString *)body;
+- (void) setNumberOfRows:(NSInteger)rows;
+@end
+
+void SafeModeAlertItem$alertSheet$buttonClicked$(id self, SEL sel, id sheet, int button) {
+    switch (button) {
+        case 1:
+        break;
+
+        case 2:
+            exit(0);
+        break;
+
+        case 3:
+            [[UIApplication sharedApplication] applicationOpenURL:[NSURL URLWithString:@"http://cydia.saurik.com/safemode/"]];
+        break;
+    }
+
+    [self dismiss];
+}
+
+void SafeModeAlertItem$configure$requirePasscodeForActions$(id self, SEL sel, BOOL configure, BOOL require) {
+    UIAlertView *sheet([self alertSheet]);
+    [sheet setDelegate:self];
+    [sheet setBodyText:@"We apologize for the inconvenience, but SpringBoard has just crashed.\n\nMobileSubstrate /did not/ cause this problem: it has protected you from it.\n\nYour device is now running in Safe Mode. All extensions that support this safety system are disabled.\n\nReboot (or restart SpringBoard) to return to the normal mode. To return to this dialog touch the status bar."];
+    [sheet addButtonWithTitle:@"OK"];
+    [sheet addButtonWithTitle:@"Restart"];
+    [sheet addButtonWithTitle:@"Help"];
+    [sheet setNumberOfRows:1];
+    if ([sheet respondsToSelector:@selector(setForceHorizontalButtonsLayout:)])
+        [sheet setForceHorizontalButtonsLayout:YES];
+}
+
+void SafeModeAlertItem$performUnlockAction(id self, SEL sel) {
+    [[$SBAlertItemsController sharedInstance] activateAlertItem:self];
+}

 
 

-    [[(id) objc_getClass("SBAlertItemsController") sharedInstance] activateAlertItem:
-        [[(id) objc_getClass("SBDismissOnlyAlertItem") alloc]
-            initWithTitle:@"Mobile Substrate Safe Mode"
-            body:@"We apologize for the inconvenience, but SpringBoard has just crashed.\n\nA recent software installation, upgrade, or removal might have been the cause of this.\n\nYour device is now running in Safe Mode. All extensions that support this safety system are disabled.\n\nReboot (or restart SpringBoard) to return to the normal mode."
-    ]];
+static void MSAlert() {
+    if ($SafeModeAlertItem == nil)
+        $SafeModeAlertItem = objc_lookUpClass("SafeModeAlertItem");
+    if ($SafeModeAlertItem == nil) {
+        $SafeModeAlertItem = objc_allocateClassPair(objc_getClass("SBAlertItem"), "SafeModeAlertItem", 0);
+        if ($SafeModeAlertItem == nil)
+            return;
+
+        class_addMethod($SafeModeAlertItem, @selector(alertSheet:buttonClicked:), (IMP) &SafeModeAlertItem$alertSheet$buttonClicked$, "v@:@i");
+        class_addMethod($SafeModeAlertItem, @selector(configure:requirePasscodeForActions:), (IMP) &SafeModeAlertItem$configure$requirePasscodeForActions$, "v@:cc");
+        class_addMethod($SafeModeAlertItem, @selector(performUnlockAction), (IMP) SafeModeAlertItem$performUnlockAction, "v@:");
+        objc_registerClassPair($SafeModeAlertItem);
+    }
+
+    if ($SBAlertItemsController != nil)
+        [[$SBAlertItemsController sharedInstance] activateAlertItem:[[[$SafeModeAlertItem alloc] init] autorelease]];
+}
+
+MSHook(void, SBStatusBar$touchesEnded$withEvent$, SBStatusBar *self, SEL sel, id touches, id event) {
+    MSAlert();
+    _SBStatusBar$touchesEnded$withEvent$(self, sel, touches, event);
+}
+
+MSHook(void, SBStatusBar$mouseDown$, SBStatusBar *self, SEL sel, void *event) {
+    MSAlert();
+    _SBStatusBar$mouseDown$(self, sel, event);

 }
 
 }
 

-static int SBButtonBar$maxIconColumns(id<MobileSubstrate> self, SEL sel) {
+MSHook(void, UIStatusBar$touchesBegan$withEvent$, id self, SEL sel, void *arg0, void *arg1) {
+    MSAlert();
+    _UIStatusBar$touchesBegan$withEvent$(self, sel, arg0, arg1);
+}
+
+MSHook(void, SBStatusBarDataManager$_updateTimeString, id self, SEL sel) {
+    if (char *_data = &MSHookIvar<char>(self, "_data")) {
+        char *timeString(_data + 20);
+        strcpy(timeString, "Exit Safe Mode");
+    }
+}
+
+static void SBIconController$showInfoAlertIfNeeded(id self, SEL sel) {
+    static bool loaded = false;
+    if (loaded)
+        return;
+    loaded = true;
+    MSAlert();
+}
+
+MSHook(int, SBButtonBar$maxIconColumns, SBButtonBar *self, SEL sel) {

     static int max;
     if (max == 0) {
     static int max;
     if (max == 0) {

-        max = [self ms$maxIconColumns];
+        max = _SBButtonBar$maxIconColumns(self, sel);

         if (NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults])
             if (NSDictionary *iconState = [defaults objectForKey:@"iconState"])
                 if (NSDictionary *buttonBar = [iconState objectForKey:@"buttonBar"])
         if (NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults])
             if (NSDictionary *iconState = [defaults objectForKey:@"iconState"])
                 if (NSDictionary *buttonBar = [iconState objectForKey:@"buttonBar"])


@@ -82,30 +158,61 @@ static int SBButtonBar$maxIconColumns(id<MobileSubstrate> self, SEL sel) {
     } return max;
 }
 
     } return max;
 }
 

-static id SBContentLayer$initWithSize$(id<MobileSubstrate> self, SEL sel, CGSize size) {
-    self = [self ms$initWithSize:size];
-    if (self == nil)
-        return nil;
-    [self setBackgroundColor:[(id) objc_getClass("UIColor") darkGrayColor]];
-    return self;
+MSHook(id, SBUIController$init, SBUIController *self, SEL sel) {
+    if ((self = _SBUIController$init(self, sel)) != nil) {
+        UIView *&_contentLayer(MSHookIvar<UIView *>(self, "_contentLayer"));
+        UIView *&_contentView(MSHookIvar<UIView *>(self, "_contentView"));
+
+        UIView *layer;
+        if (&_contentLayer != NULL)
+            layer = _contentLayer;
+        else if (&_contentView != NULL)
+            layer = _contentView;
+        else
+            layer = nil;
+
+        if (layer != nil)
+            [layer setBackgroundColor:[UIColor darkGrayColor]];
+    } return self;
+}
+
+#define Paper_ "/Library/MobileSubstrate/MobilePaper.png"
+
+MSHook(UIImage *, UIImage$defaultDesktopImage, UIImage *self, SEL sel) {
+    return [UIImage imageWithContentsOfFile:@Paper_];

 }
 
 }
 

-static void SBStatusBarTimeView$drawRect$(id<MobileSubstrate> self, SEL sel, CGRect rect) {
-    id &_time(MSHookIvar<id>(self, "_time"));
+MSHook(void, SBStatusBarTimeView$tile, SBStatusBarTimeView *self, SEL sel) {
+    NSString *&_time(MSHookIvar<NSString *>(self, "_time"));
+    CGRect &_textRect(MSHookIvar<CGRect>(self, "_textRect"));

     if (_time != nil)
     if (_time != nil)

-        [_time autorelease];
-    _time = [@"Safe Mode" retain];
-    return [self ms$drawRect:rect];
+        [_time release];
+    _time = [@"Exit Safe Mode" retain];
+    id font([self textFont]);
+    CGSize size([_time sizeWithFont:font]);
+    CGRect frame([self frame]);
+    _textRect.size = size;
+    _textRect.origin.x = (frame.size.width - size.width) / 2;
+    _textRect.origin.y = (frame.size.height - size.height) / 2;

 }
 
 #define Dylib_ "/Library/MobileSubstrate/MobileSubstrate.dylib"
 
 }
 
 #define Dylib_ "/Library/MobileSubstrate/MobileSubstrate.dylib"
 

-extern "C" void MSSafety() {
+MSInitialize {
+    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
+

     NSLog(@"MS:Warning: Entering Safe Mode");
 
     NSLog(@"MS:Warning: Entering Safe Mode");
 

-    MSHookMessage(objc_getClass("SBButtonBar"), @selector(maxIconColumns), (IMP) &SBButtonBar$maxIconColumns, "ms$");
-    MSHookMessage(objc_getClass("SBContentLayer"), @selector(initWithSize:), (IMP) &SBContentLayer$initWithSize$, "ms$");
-    MSHookMessage(objc_getClass("SBStatusBarTimeView"), @selector(drawRect:), (IMP) &SBStatusBarTimeView$drawRect$, "ms$");
+    _SBButtonBar$maxIconColumns = MSHookMessage(objc_getClass("SBButtonBar"), @selector(maxIconColumns), &$SBButtonBar$maxIconColumns);
+    _SBUIController$init = MSHookMessage(objc_getClass("SBUIController"), @selector(init), &$SBUIController$init);
+    _SBStatusBar$touchesEnded$withEvent$ = MSHookMessage(objc_getClass("SBStatusBar"), @selector(touchesEnded:withEvent:), &$SBStatusBar$touchesEnded$withEvent$);
+    _SBStatusBar$mouseDown$ = MSHookMessage(objc_getClass("SBStatusBar"), @selector(mouseDown:), &$SBStatusBar$mouseDown$);
+    _SBStatusBarDataManager$_updateTimeString = MSHookMessage(objc_getClass("SBStatusBarDataManager"), @selector(_updateTimeString), &$SBStatusBarDataManager$_updateTimeString);
+    _SBStatusBarTimeView$tile = MSHookMessage(objc_getClass("SBStatusBarTimeView"), @selector(tile), &$SBStatusBarTimeView$tile);
+
+    _UIStatusBar$touchesBegan$withEvent$ = MSHookMessage(objc_getClass("UIStatusBar"), @selector(touchesBegan:withEvent:), &$UIStatusBar$touchesBegan$withEvent$);
+
+    _UIImage$defaultDesktopImage = MSHookMessage(object_getClass(objc_getClass("UIImage")), @selector(defaultDesktopImage), &$UIImage$defaultDesktopImage);

 
     char *dil = getenv("DYLD_INSERT_LIBRARIES");
     if (dil == NULL)
 
     char *dil = getenv("DYLD_INSERT_LIBRARIES");
     if (dil == NULL)


@@ -125,9 +232,13 @@ extern "C" void MSSafety() {
         }
     }
 
         }
     }
 

+    $SBAlertItemsController = objc_getClass("SBAlertItemsController");
+

     if (Class _class = objc_getClass("SBIconController")) {
         SEL sel(@selector(showInfoAlertIfNeeded));
         if (Method method = class_getInstanceMethod(_class, sel))
     if (Class _class = objc_getClass("SBIconController")) {
         SEL sel(@selector(showInfoAlertIfNeeded));
         if (Method method = class_getInstanceMethod(_class, sel))

-            method_setImplementation(method, (IMP) &MSAlert);
+            method_setImplementation(method, (IMP) &SBIconController$showInfoAlertIfNeeded);

     }
     }

+
+    [pool release];

 }
 }