diff --git a/Signal/src/util/NSTimer+OWS.h b/Signal/src/util/NSTimer+OWS.h index f51abfda8..e301b39b0 100644 --- a/Signal/src/util/NSTimer+OWS.h +++ b/Signal/src/util/NSTimer+OWS.h @@ -2,6 +2,8 @@ // Copyright (c) 2017 Open Whisper Systems. All rights reserved. // +NS_ASSUME_NONNULL_BEGIN + @interface NSTimer (OWS) // This method avoids the classic NSTimer retain cycle bug @@ -13,3 +15,5 @@ repeats:(BOOL)repeats; @end + +NS_ASSUME_NONNULL_END diff --git a/Signal/src/util/NSTimer+OWS.m b/Signal/src/util/NSTimer+OWS.m index 9b60cf8cf..6abc56790 100644 --- a/Signal/src/util/NSTimer+OWS.m +++ b/Signal/src/util/NSTimer+OWS.m @@ -5,6 +5,8 @@ #import "NSTimer+OWS.h" #import +NS_ASSUME_NONNULL_BEGIN + @interface NSTimerProxy : NSObject @property (nonatomic, weak) id target; @@ -32,12 +34,12 @@ static void *kNSTimer_OWS_Proxy = &kNSTimer_OWS_Proxy; @implementation NSTimer (OWS) -- (NSTimerProxy *)proxy +- (NSTimerProxy *)ows_proxy { return objc_getAssociatedObject(self, kNSTimer_OWS_Proxy); } -- (void)setProxy:(NSTimerProxy *)proxy +- (void)ows_setProxy:(NSTimerProxy *)proxy { OWSAssert(proxy); @@ -58,8 +60,10 @@ static void *kNSTimer_OWS_Proxy = &kNSTimer_OWS_Proxy; selector:@selector(timerFired:) userInfo:userInfo repeats:repeats]; - [timer setProxy:proxy]; + [timer ows_setProxy:proxy]; return timer; } @end + +NS_ASSUME_NONNULL_END