From 954e4d5df13b14aee9c1112db67a23729d12acfd Mon Sep 17 00:00:00 2001
From: Chen Liang <chliang@google.com>
Date: Mon, 9 Jul 2018 14:33:41 -0700
Subject: avoid calling [UIApplication sharedApplication] in app extensions
 (#1503)

---
 Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m')

diff --git a/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m b/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m
index c5ad337..7cea178 100644
--- a/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m
+++ b/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m
@@ -21,6 +21,7 @@
 
 #import "FIRMessagingConstants.h"
 #import "FIRMessagingLogger.h"
+#import "FIRMessagingUtilities.h"
 #import "FIRMessaging_Private.h"
 
 static const BOOL kDefaultAutoRegisterEnabledValue = YES;
@@ -98,7 +99,11 @@ static NSString *kReceiveDataMessageSelectorString = @"messaging:didReceiveMessa
     return;
   }
 
-  NSObject<UIApplicationDelegate> *appDelegate = [[UIApplication sharedApplication] delegate];
+  UIApplication *application = FIRMessagingUIApplication();
+  if (!application) {
+    return;
+  }
+  NSObject<UIApplicationDelegate> *appDelegate = [application delegate];
   [self swizzleAppDelegateMethods:appDelegate];
 
   // Add KVO listener on [UNUserNotificationCenter currentNotificationCenter]'s delegate property
-- 
cgit v1.2.3