Mark DB current after backup upgrade process

Fixes signalapp#7596
tw-hx · Apr 2, 2018 · 2c82b36 · 2c82b36
1 parent c0da061
commit 2c82b36
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/org/thoughtcrime/securesms/database/DatabaseFactory.java b/src/org/thoughtcrime/securesms/database/DatabaseFactory.java
@@ -136,6 +136,7 @@ public static SQLiteDatabase getBackupDatabase(Context context) {
 
   public static void upgradeRestored(Context context, SQLiteDatabase database){
     getInstance(context).databaseHelper.onUpgrade(database, database.getVersion(), -1);
+    getInstance(context).databaseHelper.markCurrent(database);
   }
 
   private DatabaseFactory(@NonNull Context context) {

diff --git a/src/org/thoughtcrime/securesms/database/helpers/SQLCipherOpenHelper.java b/src/org/thoughtcrime/securesms/database/helpers/SQLCipherOpenHelper.java
@@ -185,6 +185,10 @@ public SQLiteDatabase getWritableDatabase() {
     return getWritableDatabase(databaseSecret.asString());
   }
 
+  public void markCurrent(SQLiteDatabase db) {
+    db.setVersion(DATABASE_VERSION);
+  }
+
   private void executeStatements(SQLiteDatabase db, String[] statements) {
     for (String statement : statements)
       db.execSQL(statement);