Improve Usage of SQLITE_OPEN_FILEPROTECTION_NONE in FIRMessagingRmqMa…

…nager.m (#12909)
firebase · May 7, 2024 · 96b6891 · 96b6891
1 parent ab94a58
commit 96b6891
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/FirebaseMessaging/Sources/FIRMessagingRmqManager.m b/FirebaseMessaging/Sources/FIRMessagingRmqManager.m
@@ -490,9 +490,11 @@ - (void)openDatabase {
     BOOL didOpenDatabase = YES;
     if (![fileManager fileExistsAtPath:path]) {
       // We've to separate between different versions here because of backwards compatbility issues.
-      int result = sqlite3_open_v2(
-          [path UTF8String], &self -> _database,
-          SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_FILEPROTECTION_NONE, NULL);
+      int flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE;
+#ifdef SQLITE_OPEN_FILEPROTECTION_NONE
+      flags |= SQLITE_OPEN_FILEPROTECTION_NONE;
+#endif
+      int result = sqlite3_open_v2([path UTF8String], &self -> _database, flags, NULL);
       if (result != SQLITE_OK) {
         NSString *errorString = FIRMessagingStringFromSQLiteResult(result);
         NSString *errorMessage = [NSString
@@ -509,9 +511,11 @@ - (void)openDatabase {
       [self createTableWithName:kTableS2DRmqIds command:kCreateTableS2DRmqIds];
     } else {
       // Calling sqlite3_open should create the database, since the file doesn't exist.
-      int result = sqlite3_open_v2(
-          [path UTF8String], &self -> _database,
-          SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE | SQLITE_OPEN_FILEPROTECTION_NONE, NULL);
+      int flags = SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE;
+#ifdef SQLITE_OPEN_FILEPROTECTION_NONE
+      flags |= SQLITE_OPEN_FILEPROTECTION_NONE;
+#endif
+      int result = sqlite3_open_v2([path UTF8String], &self -> _database, flags, NULL);
       if (result != SQLITE_OK) {
         NSString *errorString = FIRMessagingStringFromSQLiteResult(result);
         NSString *errorMessage =