net.tjado.passwdsafe.file.PasswdFileUri.java Source code

Java tutorial

Introduction

Here is the source code for net.tjado.passwdsafe.file.PasswdFileUri.java

Source

/*
 * Copyright () 2016 Jeff Harris <jefftharris@gmail.com>
 * All rights reserved. Use of the code is allowed under the
 * Artistic License 2.0 terms, as specified in the LICENSE file
 * distributed with this code, or available from
 * http://www.opensource.org/licenses/artistic-license-2.0.php
 */
package net.tjado.passwdsafe.file;

import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.io.InputStream;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.Locale;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.pwsafe.lib.exception.EndOfFileException;
import org.pwsafe.lib.exception.InvalidPassphraseException;
import org.pwsafe.lib.exception.UnsupportedFileVersionException;
import org.pwsafe.lib.file.Owner;
import org.pwsafe.lib.file.PwsFile;
import org.pwsafe.lib.file.PwsFileFactory;
import org.pwsafe.lib.file.PwsFileStorage;
import org.pwsafe.lib.file.PwsPassword;
import org.pwsafe.lib.file.PwsStorage;
import org.pwsafe.lib.file.PwsStreamStorage;

import android.content.ContentResolver;
import android.content.ContentUris;
import android.content.ContentValues;
import android.content.Context;
import android.content.SharedPreferences;
import android.database.Cursor;
import android.net.Uri;
import android.os.Environment;
import android.os.Parcel;
import android.os.Parcelable;
import android.provider.OpenableColumns;
import android.support.v4.os.EnvironmentCompat;
import android.util.Log;

import net.tjado.passwdsafe.Preferences;
import net.tjado.passwdsafe.R;
import net.tjado.passwdsafe.lib.ApiCompat;
import net.tjado.passwdsafe.lib.DocumentsContractCompat;
import net.tjado.passwdsafe.lib.PasswdSafeContract;
import net.tjado.passwdsafe.lib.PasswdSafeUtil;
import net.tjado.passwdsafe.lib.ProviderType;
import net.tjado.passwdsafe.pref.FileBackupPref;
import net.tjado.passwdsafe.util.Pair;

/**
 * The PasswdFileUri class encapsulates a URI to a password file
 */
public class PasswdFileUri implements Parcelable {
    private static final String TAG = "PasswdFileUri";

    private final Uri itsUri;
    private final Type itsType;
    private final File itsFile;
    private String itsTitle = null;
    private Pair<Boolean, Integer> itsWritableInfo;
    private boolean itsIsDeletable;
    private ProviderType itsSyncType = null;

    /** The type of URI */
    public enum Type {
        FILE, SYNC_PROVIDER, EMAIL, GENERIC_PROVIDER
    }

    /** Parcelable CREATOR instance */
    public static final Parcelable.Creator<PasswdFileUri> CREATOR = new Parcelable.Creator<PasswdFileUri>() {
        /* (non-Javadoc)
         * @see android.os.Parcelable.Creator#createFromParcel(android.os.Parcel)
         */
        @Override
        public PasswdFileUri createFromParcel(Parcel source) {
            return new PasswdFileUri(source);
        }

        /* (non-Javadoc)
         * @see android.os.Parcelable.Creator#newArray(int)
         */
        @Override
        public PasswdFileUri[] newArray(int size) {
            return new PasswdFileUri[size];
        }
    };

    /**
     * Creator for a PasswdFileUri that can work with an AsyncTask
     */
    public static class Creator {
        private final Uri itsFileUri;
        private final Context itsContext;
        private PasswdFileUri itsResolvedUri;
        private Throwable itsResolveEx;

        /**
         * Constructor
         */
        public Creator(Uri fileUri, Context ctx) {
            itsFileUri = fileUri;
            itsContext = ctx;
        }

        /**
         * Handle a pre-execute call in the main thread
         */
        public void onPreExecute() {
            switch (PasswdFileUri.getUriType(itsFileUri)) {
            case GENERIC_PROVIDER: {
                create();
                break;
            }
            case FILE:
            case SYNC_PROVIDER:
            case EMAIL: {
                break;
            }
            }
        }

        /**
         * Finish creating the PasswdFileUri, typically in a background thread
         */
        public PasswdFileUri finishCreate() {
            if ((itsResolvedUri == null) && (itsResolveEx == null)) {
                create();
            }
            return itsResolvedUri;
        }

        /**
         * Get an exception that occurred during the creation
         */
        public Throwable getResolveEx() {
            return itsResolveEx;
        }

        /**
         * Create the PasswdFileUri
         */
        private void create() {
            try {
                itsResolvedUri = new PasswdFileUri(itsFileUri, itsContext);
            } catch (Throwable e) {
                itsResolveEx = e;
            }
        }
    }

    /** Constructor */
    private PasswdFileUri(Uri uri, Context ctx) {
        itsUri = uri;
        itsType = getUriType(uri);
        switch (itsType) {
        case FILE: {
            itsFile = new File(uri.getPath());
            resolveFileUri();
            break;
        }
        case GENERIC_PROVIDER: {
            itsFile = null;
            resolveGenericProviderUri(ctx);
            break;
        }
        case SYNC_PROVIDER: {
            itsFile = null;
            resolveSyncProviderUri(ctx);
            break;
        }
        case EMAIL:
            //noinspection UnnecessaryDefault
        default: {
            itsFile = null;
            itsWritableInfo = new Pair<>(false, null);
            itsIsDeletable = false;
            break;
        }
        }
    }

    /** Constructor from a File */
    private PasswdFileUri(File file) {
        itsUri = Uri.fromFile(file);
        itsType = Type.FILE;
        itsFile = file;
        resolveFileUri();
    }

    /** Constructor from parcelable data */
    private PasswdFileUri(Parcel source) {
        String str;
        itsUri = source.readParcelable(getClass().getClassLoader());
        itsType = Type.valueOf(source.readString());
        str = source.readString();
        //noinspection ConstantConditions
        itsFile = (str != null) ? new File(str) : null;
        itsTitle = source.readString();
        str = source.readString();
        //noinspection ConstantConditions
        itsSyncType = (str != null) ? ProviderType.valueOf(str) : null;
    }

    /** Load the password file */
    public PwsFile load(Owner<PwsPassword>.Param passwd, Context context)
            throws EndOfFileException, InvalidPassphraseException, IOException, UnsupportedFileVersionException {
        switch (itsType) {
        case FILE: {
            return PwsFileFactory.loadFile(itsFile.getAbsolutePath(), passwd);
        }
        case SYNC_PROVIDER: {
            ContentResolver cr = context.getContentResolver();
            InputStream is = cr.openInputStream(itsUri);
            String id = getIdentifier(context, false);
            PwsStorage storage = new PasswdFileSyncStorage(itsUri, id, is);
            return PwsFileFactory.loadFromStorage(storage, passwd);
        }
        case EMAIL:
        case GENERIC_PROVIDER: {
            ContentResolver cr = context.getContentResolver();
            InputStream is = cr.openInputStream(itsUri);
            String id = getIdentifier(context, false);
            PwsStorage storage;
            if (itsWritableInfo.first) {
                storage = new PasswdFileGenProviderStorage(itsUri, id, is);
            } else {
                storage = new PwsStreamStorage(id, is);
            }
            return PwsFileFactory.loadFromStorage(storage, passwd);
        }
        }
        return null;
    }

    /** Create a new file */
    public PwsFile createNew(Owner<PwsPassword>.Param passwd, Context context) throws IOException {
        switch (itsType) {
        case FILE: {
            PwsFile file = PwsFileFactory.newFile();
            file.setPassphrase(passwd);
            file.setStorage(new PwsFileStorage(itsFile.getAbsolutePath(), null));
            return file;
        }
        case SYNC_PROVIDER: {
            PwsFile file = PwsFileFactory.newFile();
            file.setPassphrase(passwd);
            String id = getIdentifier(context, false);
            file.setStorage(new PasswdFileSyncStorage(itsUri, id, null));
            return file;
        }
        case GENERIC_PROVIDER: {
            PwsFile file = PwsFileFactory.newFile();
            file.setPassphrase(passwd);
            String id = getIdentifier(context, false);
            file.setStorage(new PasswdFileGenProviderStorage(itsUri, id, null));
            return file;
        }
        case EMAIL: {
            throw new IOException("no file");
        }
        }
        return null;
    }

    /**
     * Validate a new file that is a child of the current URI. Return null if
     * successful; error string otherwise
     */
    public String validateNewChild(String fileName, Context ctx) {
        switch (itsType) {
        case FILE: {
            File f = new File(itsFile, fileName + ".psafe3");
            if (f.exists()) {
                return ctx.getString(R.string.file_exists);
            }
            return null;
        }
        case SYNC_PROVIDER: {
            return null;
        }
        case EMAIL:
        case GENERIC_PROVIDER: {
            break;
        }
        }
        return ctx.getString(R.string.new_file_not_supp_uri, toString());
    }

    /** Create a new children file URI */
    public PasswdFileUri createNewChild(String fileName, Context ctx) throws IOException {
        switch (itsType) {
        case FILE: {
            File file = new File(itsFile, fileName);
            return new PasswdFileUri(file);
        }
        case SYNC_PROVIDER: {
            ContentResolver cr = ctx.getContentResolver();
            ContentValues values = new ContentValues();
            values.put(PasswdSafeContract.Files.COL_TITLE, fileName);
            Uri childUri = cr.insert(itsUri, values);
            return new PasswdFileUri(childUri, ctx);
        }
        case EMAIL:
        case GENERIC_PROVIDER: {
            break;
        }
        }
        throw new IOException("Can't create child \"" + fileName + "\" for URI " + toString());
    }

    /** Delete a file */
    public void delete(Context context) throws IOException {
        switch (itsType) {
        case FILE: {
            if (!itsFile.delete()) {
                throw new IOException("Could not delete file: " + toString());
            }
            break;
        }
        case SYNC_PROVIDER: {
            ContentResolver cr = context.getContentResolver();
            int rc = cr.delete(itsUri, null, null);
            if (rc != 1) {
                throw new IOException("Could not delete file: " + toString());
            }
            break;
        }
        case GENERIC_PROVIDER: {
            ContentResolver cr = context.getContentResolver();
            if (!ApiCompat.documentsContractDeleteDocument(cr, itsUri)) {
                throw new IOException("Could not delete file: " + toString());
            }
            break;
        }
        case EMAIL: {
            throw new IOException("Delete not supported for " + toString());
        }
        }
    }

    /** Does the file exist at the URI */
    @SuppressWarnings("BooleanMethodIsAlwaysInverted")
    public boolean exists() {
        switch (itsType) {
        case FILE: {
            return (itsFile != null) && itsFile.exists();
        }
        case SYNC_PROVIDER: {
            return (itsSyncType != null);
        }
        case EMAIL:
        case GENERIC_PROVIDER: {
            return true;
        }
        }
        return false;
    }

    /** Is the file writable */
    public Pair<Boolean, Integer> isWritable() {
        return itsWritableInfo;
    }

    /** Is the file deletable */
    public boolean isDeletable() {
        return itsIsDeletable;
    }

    /** Get the URI of the file */
    public Uri getUri() {
        return itsUri;
    }

    /** Get the type of the URI */
    public Type getType() {
        return itsType;
    }

    /** Get the sync type of the URI */
    public ProviderType getSyncType() {
        return itsSyncType;
    }

    /** Get an identifier for the URI */
    public String getIdentifier(Context context, boolean shortId) {
        switch (itsType) {
        case FILE: {
            if (shortId) {
                return itsUri.getLastPathSegment();
            } else {
                return itsUri.getPath();
            }
        }
        case SYNC_PROVIDER: {
            if (itsSyncType != null) {
                return String.format("%s - %s", itsSyncType.getName(context), itsTitle);
            }
            return context.getString(R.string.unknown_sync_file);
        }
        case EMAIL: {
            return context.getString(R.string.email_attachment);
        }
        case GENERIC_PROVIDER: {
            if (itsTitle != null) {
                return itsTitle;
            }
            return context.getString(R.string.content_file);
        }
        }
        return "";
    }

    /* (non-Javadoc)
     * @see java.lang.Object#equals(java.lang.Object)
     */
    @Override
    public boolean equals(Object o) {
        if (!(o instanceof PasswdFileUri)) {
            return false;
        }
        PasswdFileUri uri = (PasswdFileUri) o;
        return itsUri.equals(uri.itsUri);
    }

    /** Convert the URI to a string */
    @Override
    public String toString() {
        return itsUri.toString();
    }

    /* (non-Javadoc)
     * @see android.os.Parcelable#describeContents()
     */
    @Override
    public int describeContents() {
        return 0;
    }

    /* (non-Javadoc)
     * @see android.os.Parcelable#writeToParcel(android.os.Parcel, int)
     */
    @Override
    public void writeToParcel(Parcel dest, int flags) {
        dest.writeParcelable(itsUri, flags);
        dest.writeString(itsType.name());
        dest.writeString((itsFile != null) ? itsFile.getAbsolutePath() : null);
        dest.writeString(itsTitle);
        dest.writeString((itsSyncType != null) ? itsSyncType.name() : null);
    }

    /** Get the URI type */
    private static Type getUriType(Uri uri) {
        if (uri.getScheme().equals(ContentResolver.SCHEME_FILE)) {
            return Type.FILE;
        }
        String auth = uri.getAuthority();
        if (PasswdSafeContract.AUTHORITY.equals(auth)) {
            return Type.SYNC_PROVIDER;
        } else if (auth.contains("mail")) {
            return Type.EMAIL;
        }
        return Type.GENERIC_PROVIDER;
    }

    /** Resolve fields for a file URI */
    private void resolveFileUri() {
        boolean writable;
        Integer extraMsgId = null;
        do {
            if ((itsFile == null) || !itsFile.canWrite()) {
                writable = false;
                break;
            }
            // Check mount state on kitkat or higher
            if (ApiCompat.SDK_VERSION < ApiCompat.SDK_KITKAT) {
                writable = true;
                break;
            }
            writable = !EnvironmentCompat.getStorageState(itsFile).equals(Environment.MEDIA_MOUNTED_READ_ONLY);
            if (!writable) {
                extraMsgId = R.string.read_only_media;
            }
        } while (false);
        itsWritableInfo = new Pair<>(writable, extraMsgId);
        itsIsDeletable = writable;
    }

    /** Resolve fields for a generic provider URI */
    private void resolveGenericProviderUri(Context context) {
        ContentResolver cr = context.getContentResolver();
        boolean writable = false;
        boolean deletable = false;
        itsTitle = "(unknown)";
        Cursor cursor = cr.query(itsUri, null, null, null, null);
        try {
            if ((cursor != null) && cursor.moveToFirst()) {
                int colidx = cursor.getColumnIndex(OpenableColumns.DISPLAY_NAME);
                if (colidx != -1) {
                    itsTitle = cursor.getString(colidx);
                }

                int flags = 0;
                colidx = cursor.getColumnIndex(DocumentsContractCompat.COLUMN_FLAGS);
                if (colidx != -1) {
                    flags = cursor.getInt(colidx);
                }

                PasswdSafeUtil.dbginfo(TAG, "file %s, %x", itsTitle, flags);
                writable = (flags & DocumentsContractCompat.FLAG_SUPPORTS_WRITE) != 0;
                deletable = (flags & DocumentsContractCompat.FLAG_SUPPORTS_DELETE) != 0;
            }
        } finally {
            if (cursor != null) {
                cursor.close();
            }
        }
        itsWritableInfo = new Pair<>(writable, null);
        itsIsDeletable = deletable;
    }

    /** Resolve fields for a sync provider URI */
    private void resolveSyncProviderUri(Context context) {
        itsWritableInfo = new Pair<>(true, null);
        itsIsDeletable = true;
        if (itsSyncType != null) {
            return;
        }

        long providerId = -1;
        boolean isFile = false;
        switch (PasswdSafeContract.MATCHER.match(itsUri)) {
        case PasswdSafeContract.MATCH_PROVIDER:
        case PasswdSafeContract.MATCH_PROVIDER_FILES: {
            providerId = Long.valueOf(itsUri.getPathSegments().get(1));
            break;
        }
        case PasswdSafeContract.MATCH_PROVIDER_FILE: {
            providerId = Long.valueOf(itsUri.getPathSegments().get(1));
            isFile = true;
            break;
        }
        }

        if (providerId != -1) {
            ContentResolver cr = context.getContentResolver();
            resolveSyncProvider(providerId, cr);
            if (isFile) {
                resolveSyncFile(cr);
            }
        }
    }

    /** Resolve sync provider information */
    private void resolveSyncProvider(long providerId, ContentResolver cr) {
        Uri providerUri = ContentUris.withAppendedId(PasswdSafeContract.Providers.CONTENT_URI, providerId);
        Cursor providerCursor = cr.query(providerUri, PasswdSafeContract.Providers.PROJECTION, null, null, null);
        try {
            if ((providerCursor != null) && providerCursor.moveToFirst()) {
                String typeStr = providerCursor.getString(PasswdSafeContract.Providers.PROJECTION_IDX_TYPE);
                try {
                    itsSyncType = ProviderType.valueOf(typeStr);
                    itsTitle = providerCursor.getString(PasswdSafeContract.Providers.PROJECTION_IDX_ACCT);
                } catch (IllegalArgumentException e) {
                    Log.e(TAG, "Unknown provider type: " + typeStr);
                }
            }
        } finally {
            if (providerCursor != null) {
                providerCursor.close();
            }
        }
    }

    /** Resolve sync file information */
    private void resolveSyncFile(ContentResolver cr) {
        Cursor fileCursor = cr.query(itsUri, PasswdSafeContract.Files.PROJECTION, null, null, null);
        try {
            if ((fileCursor != null) && fileCursor.moveToFirst()) {
                itsTitle = fileCursor.getString(PasswdSafeContract.Files.PROJECTION_IDX_TITLE);
            }
        } finally {
            if (fileCursor != null) {
                fileCursor.close();
            }
        }
    }

    /** A PwsStorage save helper for files */
    public static class SaveHelper implements PwsStorage.SaveHelper {
        private final Context itsContext;

        public SaveHelper(Context context) {
            itsContext = context;
        }

        /** Get the save context */
        public Context getContext() {
            return itsContext;
        }

        /* (non-Javadoc)
         * @see org.pwsafe.lib.file.PwsStorage.SaveHelper#getSaveFileName(java.io.File, boolean)
         */
        @Override
        public String getSaveFileName(File file, boolean isV3) {
            String name = file.getName();
            Pattern pat = Pattern.compile("^(.*)_\\d{8}_\\d{6}\\.ibak$");
            Matcher match = pat.matcher(name);
            //noinspection ConstantConditions
            if ((match != null) && match.matches()) {
                name = match.group(1);
                if (isV3) {
                    name += ".psafe3";
                } else {
                    name += ".dat";
                }
            }
            return name;
        }

        /* (non-Javadoc)
         * @see org.pwsafe.lib.file.PwsStorage.SaveHelper#createBackupFile(java.io.File, java.io.File)
         */
        @Override
        public void createBackupFile(File fromFile, File toFile) throws IOException {
            SharedPreferences prefs = Preferences.getSharedPrefs(itsContext);
            FileBackupPref backupPref = Preferences.getFileBackupPref(prefs);

            File dir = toFile.getParentFile();
            String fileName = toFile.getName();
            int dotpos = fileName.lastIndexOf('.');
            if (dotpos != -1) {
                fileName = fileName.substring(0, dotpos);
            }

            final Pattern pat = Pattern.compile("^" + Pattern.quote(fileName) + "_\\d{8}_\\d{6}\\.ibak$");
            File[] backupFiles = dir.listFiles(new FileFilter() {
                public boolean accept(File f) {
                    return f.isFile() && pat.matcher(f.getName()).matches();
                }
            });
            if (backupFiles != null) {
                Arrays.sort(backupFiles);

                int numBackups = backupPref.getNumBackups();
                if (numBackups > 0) {
                    --numBackups;
                }
                for (int i = 0, numFiles = backupFiles.length; numFiles > numBackups; ++i, --numFiles) {
                    if (!backupFiles[i].equals(fromFile)) {
                        if (!backupFiles[i].delete()) {
                            Log.e(TAG, "Error removing backup: " + backupFiles[i]);
                        }
                    }
                }
            }

            if (backupPref != FileBackupPref.BACKUP_NONE) {
                SimpleDateFormat bakTime = new SimpleDateFormat("yyyyMMdd_HHmmss", Locale.US);
                String bakName = fileName + "_" + bakTime.format(new Date()) + ".ibak";
                File bakFile = new File(dir, bakName);
                if (!toFile.renameTo(bakFile)) {
                    throw new IOException("Can not create backup file: " + bakFile);
                }
            }
        }
    }

    /** A PwsStreamStorage implementation for sync providers */
    private static class PasswdFileSyncStorage extends PwsStreamStorage {
        private final Uri itsUri;

        /** Constructor */
        public PasswdFileSyncStorage(Uri uri, String id, InputStream stream) {
            super(id, stream);
            itsUri = uri;
        }

        /** Save the file contents */
        @Override
        public boolean save(byte[] data, boolean isV3) {
            File file = null;
            try {
                PasswdFileUri.SaveHelper helper = (PasswdFileUri.SaveHelper) getSaveHelper();
                Context ctx = helper.getContext();
                file = File.createTempFile("passwd", ".tmp", ctx.getCacheDir());
                PwsFileStorage.writeFile(file, data);

                Uri fileUri = PasswdClientProvider.addFile(file);
                ContentResolver cr = ctx.getContentResolver();
                ContentValues values = new ContentValues();
                values.put(PasswdSafeContract.Files.COL_FILE, fileUri.toString());
                cr.update(itsUri, values, null, null);

                PasswdSafeUtil.dbginfo(TAG, "SyncStorage update %s with %s", itsUri, file);
                return true;
            } catch (Exception e) {
                Log.e(TAG, "Error saving " + itsUri, e);
                return false;
            } finally {
                if (file != null) {
                    PasswdClientProvider.removeFile(file);
                    if (!file.delete()) {
                        Log.e(TAG, "Error deleting " + file);
                    }
                }
            }
        }
    }
}