diff --git a/src/conversations/java/eu/siacs/conversations/ui/WelcomeActivity.java b/src/conversations/java/eu/siacs/conversations/ui/WelcomeActivity.java index 4eb614d55..f71bc8f60 100644 --- a/src/conversations/java/eu/siacs/conversations/ui/WelcomeActivity.java +++ b/src/conversations/java/eu/siacs/conversations/ui/WelcomeActivity.java @@ -1,20 +1,49 @@ package eu.siacs.conversations.ui; +import android.Manifest; +import android.app.ProgressDialog; +import android.content.DialogInterface; import android.content.Intent; import android.content.pm.ActivityInfo; +import android.content.pm.PackageManager; +import android.database.sqlite.SQLiteDatabase; +import android.database.sqlite.SQLiteException; +import android.os.Build; import android.os.Bundle; import android.support.v7.app.ActionBar; +import android.support.v7.app.AlertDialog; import android.support.v7.app.AppCompatActivity; +import android.util.Log; +import android.view.LayoutInflater; +import android.view.View; import android.widget.Button; +import android.widget.EditText; +import android.widget.TextView; +import android.widget.Toast; +import java.io.File; +import java.io.FileInputStream; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.security.InvalidKeyException; +import java.security.NoSuchAlgorithmException; import java.util.List; +import javax.crypto.NoSuchPaddingException; + +import eu.siacs.conversations.Config; import eu.siacs.conversations.R; import eu.siacs.conversations.entities.Account; -import eu.siacs.conversations.utils.XmppUri; +import eu.siacs.conversations.persistance.DatabaseBackend; +import eu.siacs.conversations.persistance.FileBackend; +import eu.siacs.conversations.utils.EncryptDecryptFile; public class WelcomeActivity extends XmppActivity { + boolean importSuccessful = false; + @Override protected void refreshUiReal() { @@ -25,6 +54,8 @@ public class WelcomeActivity extends XmppActivity { } + private static final int REQUEST_READ_EXTERNAL_STORAGE = 0XD737; + @Override public void onStart() { super.onStart(); @@ -54,7 +85,19 @@ public class WelcomeActivity extends XmppActivity { ab.setDisplayShowHomeEnabled(false); ab.setDisplayHomeAsUpEnabled(false); } + //check if there is a backed up database -- + if (hasStoragePermission(REQUEST_READ_EXTERNAL_STORAGE)) { + backupAvailable(); + } + final Button importDatabase = findViewById(R.id.import_database); + final TextView importText = findViewById(R.id.import_text); final Button createAccount = findViewById(R.id.create_account); + if (backupAvailable()) { + importDatabase.setOnClickListener(v -> enterPasswordDialog()); + importDatabase.setVisibility(View.VISIBLE); + importText.setVisibility(View.VISIBLE); + } + createAccount.setOnClickListener(v -> { final Intent intent = new Intent(WelcomeActivity.this, MagicCreateActivity.class); intent.addFlags(Intent.FLAG_ACTIVITY_NO_ANIMATION); @@ -77,6 +120,216 @@ public class WelcomeActivity extends XmppActivity { } + public void enterPasswordDialog() { + LayoutInflater li = LayoutInflater.from(WelcomeActivity.this); + View promptsView = li.inflate(R.layout.backup_password, null); + final AlertDialog.Builder alertDialogBuilder = new AlertDialog.Builder(WelcomeActivity.this); + alertDialogBuilder.setView(promptsView); + final EditText userInput = promptsView + .findViewById(R.id.password); + alertDialogBuilder.setTitle(R.string.enter_password); + alertDialogBuilder.setMessage(R.string.enter_backup_password); + alertDialogBuilder + .setCancelable(false) + .setPositiveButton(R.string.ok, + new DialogInterface.OnClickListener() { + public void onClick(DialogInterface dialog, int id) { + final String password = userInput.getText().toString(); + final ProgressDialog pd = ProgressDialog.show(WelcomeActivity.this, getString(R.string.please_wait), getString(R.string.import_started), true); + if (!password.isEmpty()) { + new Thread(new Runnable() { + @Override + public void run() { + try { + checkDatabase(password); + } catch (IOException e) { + e.printStackTrace(); + } catch (Exception e) { + e.printStackTrace(); + } + pd.dismiss(); + } + }).start(); + } else { + AlertDialog.Builder builder = new AlertDialog.Builder(WelcomeActivity.this); + builder.setTitle(R.string.error); + builder.setMessage(R.string.password_should_not_be_empty); + builder.setNegativeButton(R.string.cancel, null); + builder.setPositiveButton(R.string.try_again, new DialogInterface.OnClickListener() { + @Override + public void onClick(DialogInterface dialog, int id) { + enterPasswordDialog(); + } + }); + builder.create().show(); + } + } + }) + .setNegativeButton(R.string.cancel, + new DialogInterface.OnClickListener() { + public void onClick(DialogInterface dialog, int id) { + Toast.makeText(WelcomeActivity.this, R.string.import_canceled, Toast.LENGTH_LONG).show(); + dialog.dismiss(); + } + } + ); + WelcomeActivity.this.runOnUiThread(new Runnable() { + public void run() { + // create alert dialog + AlertDialog alertDialog = alertDialogBuilder.create(); + // show it + alertDialog.show(); + } + }); + } + + private boolean backupAvailable() { + // Set the folder on the SDcard + File filePath = new File(FileBackend.getBackupDirectory() + "/database.db.crypt"); + Log.d(Config.LOGTAG, "DB Path: " + filePath.toString()); + if (filePath.exists()) { + Log.d(Config.LOGTAG, "DB Path existing"); + return true; + } else { + Log.d(Config.LOGTAG, "DB Path not existing"); + return false; + } + } + + private void checkDatabase(String decryptionKey) throws IOException { + // Set the folder on the SDcard + File directory = new File(FileBackend.getBackupDirectory()); + // Set the input file stream up: + FileInputStream inputFile = new FileInputStream(directory.getPath() + "/database.db.crypt"); + // Temp output for DB checks + File tempFile = new File(directory.getPath() + "/database.db.tmp"); + FileOutputStream outputTemp = new FileOutputStream(tempFile); + + try { + EncryptDecryptFile.decrypt(inputFile, outputTemp, decryptionKey); + } catch (NoSuchAlgorithmException e) { + Log.d(Config.LOGTAG, "Database importer: decryption failed with " + e); + e.printStackTrace(); + } catch (NoSuchPaddingException e) { + Log.d(Config.LOGTAG, "Database importer: decryption failed with " + e); + e.printStackTrace(); + } catch (InvalidKeyException e) { + Log.d(Config.LOGTAG, "Database importer: decryption failed (invalid key) with " + e); + e.printStackTrace(); + } catch (IOException e) { + Log.d(Config.LOGTAG, "Database importer: decryption failed (IO) with " + e); + e.printStackTrace(); + } catch (Exception e) { + Log.d(Config.LOGTAG, "Database importer: Error " + e); + e.printStackTrace(); + } + + SQLiteDatabase checkDB = null; + int DBVersion = DatabaseBackend.DATABASE_VERSION; + int BackupDBVersion = 0; + + try { + String dbPath = tempFile.toString(); + checkDB = SQLiteDatabase.openDatabase(dbPath, null, SQLiteDatabase.OPEN_READONLY); + BackupDBVersion = checkDB.getVersion(); + Log.d(Config.LOGTAG, "Backup found: " + checkDB + " Version: " + checkDB.getVersion()); + } catch (SQLiteException e) { + //database does't exist yet. + Log.d(Config.LOGTAG, "No backup found: " + checkDB); + } catch (Exception e) { + Log.d(Config.LOGTAG, "Error importing backup: " + e); + } + + if (checkDB != null) { + checkDB.close(); + } + if (checkDB != null) { + Log.d(Config.LOGTAG, "checkDB = " + checkDB.toString() + ", Backup DB = " + BackupDBVersion + ", DB = " + DBVersion); + } + if (checkDB != null && BackupDBVersion != 0 && BackupDBVersion <= DBVersion) { + try { + importDatabase(); + importSuccessful = true; + } catch (Exception e) { + importSuccessful = false; + e.printStackTrace(); + } finally { + if (tempFile.exists()) { + Log.d(Config.LOGTAG, "Delete temp file from " + tempFile.toString()); + tempFile.delete(); + } + if (importSuccessful) { + this.getPreferences().edit().putString("backup_password", decryptionKey).commit(); + restart(); + }else{ + WelcomeActivity.this.runOnUiThread(new Runnable() { + public void run() { + Toast.makeText(WelcomeActivity.this, R.string.import_failed, Toast.LENGTH_LONG).show(); + } + }); + } + } + } else if (checkDB != null && BackupDBVersion == 0) { + WelcomeActivity.this.runOnUiThread(new Runnable() { + public void run() { + Toast.makeText(WelcomeActivity.this, R.string.password_wrong, Toast.LENGTH_LONG).show(); + enterPasswordDialog(); + } + }); + } else { + WelcomeActivity.this.runOnUiThread(new Runnable() { + public void run() { + Toast.makeText(WelcomeActivity.this, R.string.import_failed, Toast.LENGTH_LONG).show(); + } + }); + } + if (tempFile.exists()) { + Log.d(Config.LOGTAG, "Delete temp file from " + tempFile.toString()); + tempFile.delete(); + } + } + + private void importDatabase() throws Exception { + // Set location for the db: + final OutputStream outputFile = new FileOutputStream(this.getDatabasePath(DatabaseBackend.DATABASE_NAME)); + // Set the folder on the SDcard + File directory = new File(FileBackend.getBackupDirectory()); + // Set the input file stream up: + final InputStream inputFile = new FileInputStream(directory.getPath() + "/database.db.tmp"); + //set temp file + File tempFile = new File(directory.getPath() + "/database.db.tmp"); + + // Transfer bytes from the input file to the output file + byte[] buffer = new byte[1024]; + int length; + while ((length = inputFile.read(buffer)) > 0) { + outputFile.write(buffer, 0, length); + } + } + + private void restart() { + //restart app + Log.d(Config.LOGTAG, "Restarting " + getBaseContext().getPackageManager().getLaunchIntentForPackage(getBaseContext().getPackageName())); + Intent intent = getBaseContext().getPackageManager().getLaunchIntentForPackage(getBaseContext().getPackageName()); + intent.addFlags(Intent.FLAG_ACTIVITY_CLEAR_TOP); + intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK); + startActivity(intent); + System.exit(0); + } + + public boolean hasStoragePermission(int requestCode) { + if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) { + if (checkSelfPermission(Manifest.permission.WRITE_EXTERNAL_STORAGE) != PackageManager.PERMISSION_GRANTED) { + requestPermissions(new String[]{Manifest.permission.READ_EXTERNAL_STORAGE, Manifest.permission.WRITE_EXTERNAL_STORAGE}, requestCode); + return false; + } else { + return true; + } + } else { + return true; + } + } + public void addInviteUri(Intent intent) { StartConversationActivity.addInviteUri(intent, getIntent()); } diff --git a/src/conversations/res/layout/welcome.xml b/src/conversations/res/layout/welcome.xml index e33c1abe8..e84113b1d 100644 --- a/src/conversations/res/layout/welcome.xml +++ b/src/conversations/res/layout/welcome.xml @@ -42,6 +42,23 @@ android:layout_marginTop="8dp" android:text="@string/welcome_text" android:textAppearance="@style/TextAppearance.Conversations.Body1"/> + +