263 lines
11 KiB
Java
263 lines
11 KiB
Java
/*
|
|
* Copyright (C) 2018 Open Whisper Systems
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package org.thoughtcrime.securesms.database;
|
|
|
|
import android.content.Context;
|
|
import androidx.annotation.NonNull;
|
|
|
|
import net.sqlcipher.database.SQLiteDatabase;
|
|
|
|
import org.thoughtcrime.securesms.DatabaseUpgradeActivity;
|
|
import org.thoughtcrime.securesms.crypto.AttachmentSecret;
|
|
import org.thoughtcrime.securesms.crypto.AttachmentSecretProvider;
|
|
import org.thoughtcrime.securesms.crypto.DatabaseSecret;
|
|
import org.thoughtcrime.securesms.crypto.DatabaseSecretProvider;
|
|
import org.thoughtcrime.securesms.crypto.MasterSecret;
|
|
import org.thoughtcrime.securesms.database.helpers.ClassicOpenHelper;
|
|
import org.thoughtcrime.securesms.database.helpers.SQLCipherMigrationHelper;
|
|
import org.thoughtcrime.securesms.database.helpers.SQLCipherOpenHelper;
|
|
import org.thoughtcrime.securesms.loki.database.LokiAPIDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.LokiBackupFilesDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.LokiMessageDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.LokiPreKeyBundleDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.LokiPreKeyRecordDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.LokiThreadDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.LokiUserDatabase;
|
|
import org.thoughtcrime.securesms.loki.database.SharedSenderKeysDatabase;
|
|
import org.thoughtcrime.securesms.util.TextSecurePreferences;
|
|
|
|
public class DatabaseFactory {
|
|
|
|
private static final Object lock = new Object();
|
|
|
|
private static DatabaseFactory instance;
|
|
|
|
private final SQLCipherOpenHelper databaseHelper;
|
|
private final SmsDatabase sms;
|
|
private final MmsDatabase mms;
|
|
private final AttachmentDatabase attachments;
|
|
private final MediaDatabase media;
|
|
private final ThreadDatabase thread;
|
|
private final MmsSmsDatabase mmsSmsDatabase;
|
|
private final IdentityDatabase identityDatabase;
|
|
private final DraftDatabase draftDatabase;
|
|
private final PushDatabase pushDatabase;
|
|
private final GroupDatabase groupDatabase;
|
|
private final RecipientDatabase recipientDatabase;
|
|
private final GroupReceiptDatabase groupReceiptDatabase;
|
|
private final OneTimePreKeyDatabase preKeyDatabase;
|
|
private final SignedPreKeyDatabase signedPreKeyDatabase;
|
|
private final SessionDatabase sessionDatabase;
|
|
private final SearchDatabase searchDatabase;
|
|
private final JobDatabase jobDatabase;
|
|
private final StickerDatabase stickerDatabase;
|
|
|
|
// Loki
|
|
private final LokiAPIDatabase lokiAPIDatabase;
|
|
private final LokiPreKeyRecordDatabase lokiContactPreKeyDatabase;
|
|
private final LokiPreKeyBundleDatabase lokiPreKeyBundleDatabase;
|
|
private final LokiMessageDatabase lokiMessageDatabase;
|
|
private final LokiThreadDatabase lokiThreadDatabase;
|
|
private final LokiUserDatabase lokiUserDatabase;
|
|
private final LokiBackupFilesDatabase lokiBackupFilesDatabase;
|
|
private final SharedSenderKeysDatabase sskDatabase;
|
|
|
|
public static DatabaseFactory getInstance(Context context) {
|
|
synchronized (lock) {
|
|
if (instance == null)
|
|
instance = new DatabaseFactory(context.getApplicationContext());
|
|
|
|
return instance;
|
|
}
|
|
}
|
|
|
|
public static MmsSmsDatabase getMmsSmsDatabase(Context context) {
|
|
return getInstance(context).mmsSmsDatabase;
|
|
}
|
|
|
|
public static ThreadDatabase getThreadDatabase(Context context) {
|
|
return getInstance(context).thread;
|
|
}
|
|
|
|
public static SmsDatabase getSmsDatabase(Context context) {
|
|
return getInstance(context).sms;
|
|
}
|
|
|
|
public static MmsDatabase getMmsDatabase(Context context) {
|
|
return getInstance(context).mms;
|
|
}
|
|
|
|
public static AttachmentDatabase getAttachmentDatabase(Context context) {
|
|
return getInstance(context).attachments;
|
|
}
|
|
|
|
public static MediaDatabase getMediaDatabase(Context context) {
|
|
return getInstance(context).media;
|
|
}
|
|
|
|
public static IdentityDatabase getIdentityDatabase(Context context) {
|
|
return getInstance(context).identityDatabase;
|
|
}
|
|
|
|
public static DraftDatabase getDraftDatabase(Context context) {
|
|
return getInstance(context).draftDatabase;
|
|
}
|
|
|
|
public static PushDatabase getPushDatabase(Context context) {
|
|
return getInstance(context).pushDatabase;
|
|
}
|
|
|
|
public static GroupDatabase getGroupDatabase(Context context) {
|
|
return getInstance(context).groupDatabase;
|
|
}
|
|
|
|
public static RecipientDatabase getRecipientDatabase(Context context) {
|
|
return getInstance(context).recipientDatabase;
|
|
}
|
|
|
|
public static GroupReceiptDatabase getGroupReceiptDatabase(Context context) {
|
|
return getInstance(context).groupReceiptDatabase;
|
|
}
|
|
|
|
public static OneTimePreKeyDatabase getPreKeyDatabase(Context context) {
|
|
return getInstance(context).preKeyDatabase;
|
|
}
|
|
|
|
public static SignedPreKeyDatabase getSignedPreKeyDatabase(Context context) {
|
|
return getInstance(context).signedPreKeyDatabase;
|
|
}
|
|
|
|
public static SessionDatabase getSessionDatabase(Context context) {
|
|
return getInstance(context).sessionDatabase;
|
|
}
|
|
|
|
public static SearchDatabase getSearchDatabase(Context context) {
|
|
return getInstance(context).searchDatabase;
|
|
}
|
|
|
|
public static JobDatabase getJobDatabase(Context context) {
|
|
return getInstance(context).jobDatabase;
|
|
}
|
|
|
|
public static StickerDatabase getStickerDatabase(Context context) {
|
|
return getInstance(context).stickerDatabase;
|
|
}
|
|
|
|
public static SQLiteDatabase getBackupDatabase(Context context) {
|
|
return getInstance(context).databaseHelper.getReadableDatabase();
|
|
}
|
|
|
|
// region Loki
|
|
public static LokiAPIDatabase getLokiAPIDatabase(Context context) {
|
|
return getInstance(context).lokiAPIDatabase;
|
|
}
|
|
|
|
public static LokiPreKeyRecordDatabase getLokiPreKeyRecordDatabase(Context context) {
|
|
return getInstance(context).lokiContactPreKeyDatabase;
|
|
}
|
|
|
|
public static LokiPreKeyBundleDatabase getLokiPreKeyBundleDatabase(Context context) {
|
|
return getInstance(context).lokiPreKeyBundleDatabase;
|
|
}
|
|
|
|
public static LokiMessageDatabase getLokiMessageDatabase(Context context) {
|
|
return getInstance(context).lokiMessageDatabase;
|
|
}
|
|
|
|
public static LokiThreadDatabase getLokiThreadDatabase(Context context) {
|
|
return getInstance(context).lokiThreadDatabase;
|
|
}
|
|
|
|
public static LokiUserDatabase getLokiUserDatabase(Context context) {
|
|
return getInstance(context).lokiUserDatabase;
|
|
}
|
|
|
|
public static LokiBackupFilesDatabase getLokiBackupFilesDatabase(Context context) {
|
|
return getInstance(context).lokiBackupFilesDatabase;
|
|
}
|
|
|
|
public static SharedSenderKeysDatabase getSSKDatabase(Context context) {
|
|
return getInstance(context).sskDatabase;
|
|
}
|
|
// endregion
|
|
|
|
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) {
|
|
SQLiteDatabase.loadLibs(context);
|
|
|
|
DatabaseSecret databaseSecret = new DatabaseSecretProvider(context).getOrCreateDatabaseSecret();
|
|
AttachmentSecret attachmentSecret = AttachmentSecretProvider.getInstance(context).getOrCreateAttachmentSecret();
|
|
|
|
this.databaseHelper = new SQLCipherOpenHelper(context, databaseSecret);
|
|
this.sms = new SmsDatabase(context, databaseHelper);
|
|
this.mms = new MmsDatabase(context, databaseHelper);
|
|
this.attachments = new AttachmentDatabase(context, databaseHelper, attachmentSecret);
|
|
this.media = new MediaDatabase(context, databaseHelper);
|
|
this.thread = new ThreadDatabase(context, databaseHelper);
|
|
this.mmsSmsDatabase = new MmsSmsDatabase(context, databaseHelper);
|
|
this.identityDatabase = new IdentityDatabase(context, databaseHelper);
|
|
this.draftDatabase = new DraftDatabase(context, databaseHelper);
|
|
this.pushDatabase = new PushDatabase(context, databaseHelper);
|
|
this.groupDatabase = new GroupDatabase(context, databaseHelper);
|
|
this.recipientDatabase = new RecipientDatabase(context, databaseHelper);
|
|
this.groupReceiptDatabase = new GroupReceiptDatabase(context, databaseHelper);
|
|
this.preKeyDatabase = new OneTimePreKeyDatabase(context, databaseHelper);
|
|
this.signedPreKeyDatabase = new SignedPreKeyDatabase(context, databaseHelper);
|
|
this.sessionDatabase = new SessionDatabase(context, databaseHelper);
|
|
this.searchDatabase = new SearchDatabase(context, databaseHelper);
|
|
this.jobDatabase = new JobDatabase(context, databaseHelper);
|
|
this.stickerDatabase = new StickerDatabase(context, databaseHelper, attachmentSecret);
|
|
this.lokiAPIDatabase = new LokiAPIDatabase(context, databaseHelper);
|
|
this.lokiContactPreKeyDatabase = new LokiPreKeyRecordDatabase(context, databaseHelper);
|
|
this.lokiPreKeyBundleDatabase = new LokiPreKeyBundleDatabase(context, databaseHelper);
|
|
this.lokiMessageDatabase = new LokiMessageDatabase(context, databaseHelper);
|
|
this.lokiThreadDatabase = new LokiThreadDatabase(context, databaseHelper);
|
|
this.lokiUserDatabase = new LokiUserDatabase(context, databaseHelper);
|
|
this.lokiBackupFilesDatabase = new LokiBackupFilesDatabase(context, databaseHelper);
|
|
this.sskDatabase = new SharedSenderKeysDatabase(context, databaseHelper);
|
|
}
|
|
|
|
public void onApplicationLevelUpgrade(@NonNull Context context, @NonNull MasterSecret masterSecret,
|
|
int fromVersion, DatabaseUpgradeActivity.DatabaseUpgradeListener listener)
|
|
{
|
|
databaseHelper.getWritableDatabase();
|
|
|
|
ClassicOpenHelper legacyOpenHelper = null;
|
|
|
|
if (fromVersion < DatabaseUpgradeActivity.ASYMMETRIC_MASTER_SECRET_FIX_VERSION) {
|
|
legacyOpenHelper = new ClassicOpenHelper(context);
|
|
legacyOpenHelper.onApplicationLevelUpgrade(context, masterSecret, fromVersion, listener);
|
|
}
|
|
|
|
if (fromVersion < DatabaseUpgradeActivity.SQLCIPHER && TextSecurePreferences.getNeedsSqlCipherMigration(context)) {
|
|
if (legacyOpenHelper == null) {
|
|
legacyOpenHelper = new ClassicOpenHelper(context);
|
|
}
|
|
|
|
SQLCipherMigrationHelper.migrateCiphertext(context, masterSecret,
|
|
legacyOpenHelper.getWritableDatabase(),
|
|
databaseHelper.getWritableDatabase(),
|
|
listener);
|
|
}
|
|
}
|
|
|
|
}
|