UIAlertViewCrashFix.h

//
//  UIAlertViewCrashFix.h
//
//  Created by Josh Hudnall on 4/28/15.
//
//

#import <Foundation/Foundation.h>

@interface UIAlertController (CrashFix)

@end


@interface UIViewController (CrashFix)

@end

UIAlertViewCrashFix.m

//
//  UIAlertViewCrashFix.m
//
//  Created by Josh Hudnall on 4/28/15.
//
//

#import <objc/runtime.h>
#import "UIAlertView+CrashFix.h"

@implementation UIAlertController (CrashFix)

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];
        Method original, swizzle;
        
        // Swizzle
        original = class_getInstanceMethod(class, @selector(shouldAutorotate));
        swizzle = class_getInstanceMethod(class, @selector(emf_shouldAutorotate));
        method_exchangeImplementations(original, swizzle);

        // Swizzle
        original = class_getInstanceMethod(class, @selector(supportedInterfaceOrientations));
        swizzle = class_getInstanceMethod(class, @selector(emf_supportedInterfaceOrientations));
        method_exchangeImplementations(original, swizzle);

        // Swizzle
        original = class_getInstanceMethod(class, @selector(preferredInterfaceOrientationForPresentation));
        swizzle = class_getInstanceMethod(class, @selector(emf_preferredInterfaceOrientationForPresentation));
        method_exchangeImplementations(original, swizzle);
    });
}

- (BOOL)emf_shouldAutorotate {
    return IS_IPAD;
}

- (NSUInteger)emf_supportedInterfaceOrientations {
    return (IS_IPAD) ? UIInterfaceOrientationMaskAll : UIInterfaceOrientationPortrait;
}

- (UIInterfaceOrientation)emf_preferredInterfaceOrientationForPresentation {
    UIDevice* device = [UIDevice currentDevice];
    if (device.orientation == UIInterfaceOrientationPortraitUpsideDown) {
        return UIInterfaceOrientationPortraitUpsideDown;
    }
    return UIInterfaceOrientationPortrait;
}

@end


@implementation UIViewController (CrashFix)

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];
        Method original, swizzle;
        
        // Swizzle
        original = class_getInstanceMethod(class, @selector(shouldAutorotate));
        swizzle = class_getInstanceMethod(class, @selector(emf_shouldAutorotate));
        method_exchangeImplementations(original, swizzle);

        // Swizzle
        original = class_getInstanceMethod(class, @selector(supportedInterfaceOrientations));
        swizzle = class_getInstanceMethod(class, @selector(emf_supportedInterfaceOrientations));
        method_exchangeImplementations(original, swizzle);

        // Swizzle
        original = class_getInstanceMethod(class, @selector(preferredInterfaceOrientationForPresentation));
        swizzle = class_getInstanceMethod(class, @selector(emf_preferredInterfaceOrientationForPresentation));
        method_exchangeImplementations(original, swizzle);
    });
}

- (BOOL)emf_shouldAutorotate {
    return IS_IPAD;
}

- (NSUInteger)emf_supportedInterfaceOrientations {
    return (IS_IPAD) ? UIInterfaceOrientationMaskAll : UIInterfaceOrientationPortrait;
}

- (UIInterfaceOrientation)emf_preferredInterfaceOrientationForPresentation {
    UIDevice* device = [UIDevice currentDevice];
    if (device.orientation == UIInterfaceOrientationPortraitUpsideDown) {
        return UIInterfaceOrientationPortraitUpsideDown;
    }
    return UIInterfaceOrientationPortrait;
}

@end