Java tutorial
/* * Copyright (C) 2015 Dominik Schrmann <dominik@dominikschuermann.de> * * 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.sufficientlysecure.keychain.ui; import java.util.ArrayList; import java.util.Iterator; import android.app.Activity; import android.content.ContentResolver; import android.content.Intent; import android.database.Cursor; import android.net.Uri; import android.os.AsyncTask; import android.os.Bundle; import android.support.annotation.Nullable; import android.support.v4.app.Fragment; import android.support.v4.app.FragmentActivity; import android.util.Pair; import android.view.LayoutInflater; import android.view.View; import android.view.ViewGroup; import org.sufficientlysecure.keychain.R; import org.sufficientlysecure.keychain.pgp.CanonicalizedSecretKey.SecretKeyType; import org.sufficientlysecure.keychain.provider.KeychainContract; import org.sufficientlysecure.keychain.provider.KeychainContract.KeyRings; import org.sufficientlysecure.keychain.provider.KeychainContract.Keys; import org.sufficientlysecure.keychain.service.input.RequiredInputParcel; import org.sufficientlysecure.keychain.ui.util.Notify; import org.sufficientlysecure.keychain.util.FileHelper; public class BackupRestoreFragment extends Fragment { // masterKeyId & subKeyId for multi-key export private Iterator<Pair<Long, Long>> mIdsForRepeatAskPassphrase; private static final int REQUEST_REPEAT_PASSPHRASE = 0x00007002; private static final int REQUEST_CODE_INPUT = 0x00007003; @Override public View onCreateView(LayoutInflater inflater, @Nullable ViewGroup container, @Nullable Bundle savedInstanceState) { View view = inflater.inflate(R.layout.backup_restore_fragment, container, false); View backupAll = view.findViewById(R.id.backup_all); View backupPublicKeys = view.findViewById(R.id.backup_public_keys); final View restore = view.findViewById(R.id.restore); backupAll.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View v) { exportToFile(true); } }); backupPublicKeys.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View v) { exportToFile(false); } }); restore.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View v) { restore(); } }); return view; } private void exportToFile(boolean includeSecretKeys) { FragmentActivity activity = getActivity(); if (activity == null) { return; } if (!includeSecretKeys) { startBackup(false); return; } new AsyncTask<ContentResolver, Void, ArrayList<Pair<Long, Long>>>() { @Override protected ArrayList<Pair<Long, Long>> doInBackground(ContentResolver... resolver) { ArrayList<Pair<Long, Long>> askPassphraseIds = new ArrayList<>(); Cursor cursor = resolver[0].query(KeyRings.buildUnifiedKeyRingsUri(), new String[] { KeyRings.MASTER_KEY_ID, KeyRings.HAS_SECRET, }, KeyRings.HAS_SECRET + " != 0", null, null); try { if (cursor != null) { while (cursor.moveToNext()) { SecretKeyType secretKeyType = SecretKeyType.fromNum(cursor.getInt(1)); switch (secretKeyType) { // all of these make no sense to ask case PASSPHRASE_EMPTY: case DIVERT_TO_CARD: case UNAVAILABLE: continue; case GNU_DUMMY: { Long masterKeyId = cursor.getLong(0); Long subKeyId = getFirstSubKeyWithPassphrase(masterKeyId, resolver[0]); if (subKeyId != null) { askPassphraseIds.add(new Pair<>(masterKeyId, subKeyId)); } continue; } default: { long masterKeyId = cursor.getLong(0); askPassphraseIds.add(new Pair<>(masterKeyId, masterKeyId)); } } } } } finally { if (cursor != null) { cursor.close(); } } return askPassphraseIds; } private Long getFirstSubKeyWithPassphrase(long masterKeyId, ContentResolver resolver) { Cursor cursor = resolver.query(KeychainContract.Keys.buildKeysUri(masterKeyId), new String[] { Keys.KEY_ID, Keys.HAS_SECRET, }, Keys.HAS_SECRET + " != 0", null, null); try { if (cursor != null) { while (cursor.moveToNext()) { SecretKeyType secretKeyType = SecretKeyType.fromNum(cursor.getInt(1)); switch (secretKeyType) { case PASSPHRASE_EMPTY: case DIVERT_TO_CARD: case UNAVAILABLE: return null; case GNU_DUMMY: continue; default: { return cursor.getLong(0); } } } } } finally { if (cursor != null) { cursor.close(); } } return null; } @Override protected void onPostExecute(ArrayList<Pair<Long, Long>> askPassphraseIds) { super.onPostExecute(askPassphraseIds); FragmentActivity activity = getActivity(); if (activity == null) { return; } mIdsForRepeatAskPassphrase = askPassphraseIds.iterator(); if (mIdsForRepeatAskPassphrase.hasNext()) { startPassphraseActivity(); return; } startBackup(true); } }.execute(activity.getContentResolver()); } private void startPassphraseActivity() { Activity activity = getActivity(); if (activity == null) { return; } Intent intent = new Intent(activity, PassphraseDialogActivity.class); Pair<Long, Long> keyPair = mIdsForRepeatAskPassphrase.next(); long masterKeyId = keyPair.first; long subKeyId = keyPair.second; RequiredInputParcel requiredInput = RequiredInputParcel.createRequiredDecryptPassphrase(masterKeyId, subKeyId); requiredInput.mSkipCaching = true; intent.putExtra(PassphraseDialogActivity.EXTRA_REQUIRED_INPUT, requiredInput); startActivityForResult(intent, REQUEST_REPEAT_PASSPHRASE); } @Override public void onActivityResult(int requestCode, int resultCode, Intent data) { switch (requestCode) { case REQUEST_REPEAT_PASSPHRASE: { if (resultCode != Activity.RESULT_OK) { return; } if (mIdsForRepeatAskPassphrase.hasNext()) { startPassphraseActivity(); return; } startBackup(true); break; } case REQUEST_CODE_INPUT: { if (resultCode != Activity.RESULT_OK || data == null) { return; } Uri uri = data.getData(); if (uri == null) { Notify.create(getActivity(), R.string.no_file_selected, Notify.Style.ERROR).show(); return; } Intent intent = new Intent(getActivity(), DecryptActivity.class); intent.setAction(Intent.ACTION_VIEW); intent.setData(uri); startActivity(intent); break; } default: { super.onActivityResult(requestCode, resultCode, data); } } } private void startBackup(boolean exportSecret) { Intent intent = new Intent(getActivity(), BackupActivity.class); intent.putExtra(BackupActivity.EXTRA_SECRET, exportSecret); startActivity(intent); } private void restore() { FileHelper.openDocument(this, null, "*/*", false, REQUEST_CODE_INPUT); } }