serverRoots setters nil check (#4)

google · Aug 3, 2016 · fb13db8 · fb13db8
1 parent baa2043
commit fb13db8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Source/MOLAuthenticatingURLSession.m b/Source/MOLAuthenticatingURLSession.m
@@ -63,6 +63,10 @@ - (void)setUserAgent:(NSString *)userAgent {
 #pragma mark Server Roots properties
 
 - (void)setServerRootsPemFile:(NSString *)serverRootsPemFile {
+  if (!serverRootsPemFile) {
+    _serverRootsPemFile = nil;
+    return;
+  }
   NSError *error;
   NSData *rootsData = [NSData dataWithContentsOfFile:serverRootsPemFile
                                              options:0
@@ -75,6 +79,10 @@ - (void)setServerRootsPemFile:(NSString *)serverRootsPemFile {
 }
 
 - (void)setServerRootsPemData:(NSData *)serverRootsPemData {
+  if (!serverRootsPemData) {
+    _serverRootsPemData = nil;
+    return;
+  }
   NSString *pemStrings = [[NSString alloc] initWithData:serverRootsPemData
                                                encoding:NSASCIIStringEncoding];
   NSArray *certs = [MOLCertificate certificatesFromPEM:pemStrings];