path: root/src/main/java/de/pixart/messenger/crypto/PgpDecryptionService.java

package de.pixart.messenger.crypto;

import android.app.PendingIntent;
import android.content.Intent;
import android.util.Log;

import org.openintents.openpgp.OpenPgpMetadata;
import org.openintents.openpgp.util.OpenPgpApi;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URL;
import java.util.ArrayDeque;
import java.util.HashSet;
import java.util.List;

import de.pixart.messenger.Config;
import de.pixart.messenger.entities.Conversation;
import de.pixart.messenger.entities.DownloadableFile;
import de.pixart.messenger.entities.Message;
import de.pixart.messenger.http.HttpConnectionManager;
import de.pixart.messenger.services.XmppConnectionService;
import de.pixart.messenger.utils.MimeUtils;

public class PgpDecryptionService {

    protected final ArrayDeque<Message> messages = new ArrayDeque<>();
    protected final HashSet<Message> pendingNotifications = new HashSet<>();
    private final XmppConnectionService mXmppConnectionService;
    private OpenPgpApi openPgpApi = null;
    private Message currentMessage;
    private PendingIntent pendingIntent;
    private Intent userInteractionResult;


    public PgpDecryptionService(XmppConnectionService service) {
        this.mXmppConnectionService = service;
    }

    public synchronized boolean decrypt(final Message message, boolean notify) {
        messages.add(message);
        if (notify && pendingIntent == null) {
            pendingNotifications.add(message);
            continueDecryption();
            return false;
        } else {
            continueDecryption();
            return notify;
        }
    }

    public synchronized void decrypt(final List<Message> list) {
        for (Message message : list) {
            if (message.getEncryption() == Message.ENCRYPTION_PGP) {
                messages.add(message);
            }
        }
        continueDecryption();
    }

    public synchronized void discard(List<Message> discards) {
        this.messages.removeAll(discards);
        this.pendingNotifications.removeAll(discards);
    }

    public synchronized void discard(Message message) {
        this.messages.remove(message);
        this.pendingNotifications.remove(message);
    }

    public void giveUpCurrentDecryption() {
        Message message;
        synchronized (this) {
            if (currentMessage != null) {
                return;
            }
            message = messages.peekFirst();
            if (message == null) {
                return;
            }
            discard(message);
        }
        synchronized (message) {
            if (message.getEncryption() == Message.ENCRYPTION_PGP) {
                message.setEncryption(Message.ENCRYPTION_DECRYPTION_FAILED);
            }
        }
        mXmppConnectionService.updateMessage(message, false);
        continueDecryption(true);
    }

    protected synchronized void decryptNext() {
        if (pendingIntent == null
                && getOpenPgpApi() != null
                && (currentMessage = messages.poll()) != null) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    executeApi(currentMessage);
                    decryptNext();
                }
            }).start();
        }
    }

    public synchronized void continueDecryption(boolean resetPending) {
        if (resetPending) {
            this.pendingIntent = null;
        }
        continueDecryption();
    }

    public synchronized void continueDecryption(Intent userInteractionResult) {
        this.pendingIntent = null;
        this.userInteractionResult = userInteractionResult;
        continueDecryption();
    }

    public synchronized void continueDecryption() {
        if (currentMessage == null) {
            decryptNext();
        }
    }

    private synchronized OpenPgpApi getOpenPgpApi() {
        if (openPgpApi == null) {
            this.openPgpApi = mXmppConnectionService.getOpenPgpApi();
        }
        return this.openPgpApi;
    }

    private void executeApi(Message message) {
        boolean skipNotificationPush = false;
        synchronized (message) {
            Intent params = userInteractionResult != null ? userInteractionResult : new Intent();
            params.setAction(OpenPgpApi.ACTION_DECRYPT_VERIFY);
            if (message.getType() == Message.TYPE_TEXT) {
                InputStream is = new ByteArrayInputStream(message.getBody().getBytes());
                final OutputStream os = new ByteArrayOutputStream();
                Intent result = getOpenPgpApi().executeApi(params, is, os);
                switch (result.getIntExtra(OpenPgpApi.RESULT_CODE, OpenPgpApi.RESULT_CODE_ERROR)) {
                    case OpenPgpApi.RESULT_CODE_SUCCESS:
                        try {
                            os.flush();
                            final String body = os.toString();
                            if (body == null) {
                                throw new IOException("body was null");
                            }
                            message.setBody(body);
                            message.setEncryption(Message.ENCRYPTION_DECRYPTED);
                            final HttpConnectionManager manager = mXmppConnectionService.getHttpConnectionManager();
                            if (message.trusted()
                                    && message.treatAsDownloadable()
                                    && manager.getAutoAcceptFileSize() > 0) {
                                manager.createNewDownloadConnection(message);
                            }
                        } catch (IOException e) {
                            message.setEncryption(Message.ENCRYPTION_DECRYPTION_FAILED);
                        }
                        mXmppConnectionService.updateMessage(message);
                        break;
                    case OpenPgpApi.RESULT_CODE_USER_INTERACTION_REQUIRED:
                        synchronized (PgpDecryptionService.this) {
                            PendingIntent pendingIntent = result.getParcelableExtra(OpenPgpApi.RESULT_INTENT);
                            messages.addFirst(message);
                            currentMessage = null;
                            storePendingIntent(pendingIntent);
                        }
                        break;
                    case OpenPgpApi.RESULT_CODE_ERROR:
                        message.setEncryption(Message.ENCRYPTION_DECRYPTION_FAILED);
                        mXmppConnectionService.updateMessage(message);
                        break;
                }
            } else if (message.isFileOrImage()) {
                try {
                    final DownloadableFile inputFile = mXmppConnectionService.getFileBackend().getFile(message, false);
                    final DownloadableFile outputFile = mXmppConnectionService.getFileBackend().getFile(message, true);
                    if (outputFile.getParentFile().mkdirs()) {
                        Log.d(Config.LOGTAG, "created parent directories for " + outputFile.getAbsolutePath());
                    }
                    outputFile.createNewFile();
                    InputStream is = new FileInputStream(inputFile);
                    OutputStream os = new FileOutputStream(outputFile);
                    Intent result = getOpenPgpApi().executeApi(params, is, os);
                    switch (result.getIntExtra(OpenPgpApi.RESULT_CODE, OpenPgpApi.RESULT_CODE_ERROR)) {
                        case OpenPgpApi.RESULT_CODE_SUCCESS:
                            OpenPgpMetadata openPgpMetadata = result.getParcelableExtra(OpenPgpApi.RESULT_METADATA);
                            String originalFilename = openPgpMetadata.getFilename();
                            String originalExtension = originalFilename == null ? null : MimeUtils.extractRelevantExtension(originalFilename);
                            if (originalExtension != null && MimeUtils.extractRelevantExtension(outputFile.getName()) == null) {
                                Log.d(Config.LOGTAG, "detected original filename during pgp decryption");
                                String mime = MimeUtils.guessMimeTypeFromExtension(originalExtension);
                                String path = outputFile.getName() + "." + originalExtension;
                                DownloadableFile fixedFile = mXmppConnectionService.getFileBackend().getFileForPath(path, mime);
                                if (fixedFile.getParentFile().mkdirs()) {
                                    Log.d(Config.LOGTAG, "created parent directories for " + fixedFile.getAbsolutePath());
                                }
                                if (outputFile.renameTo(fixedFile)) {
                                    Log.d(Config.LOGTAG, "renamed " + outputFile.getAbsolutePath() + " to " + fixedFile.getAbsolutePath());
                                    message.setRelativeFilePath(path);
                                }
                            }
                            URL url = message.getFileParams().url;
                            mXmppConnectionService.getFileBackend().updateFileParams(message, url);
                            message.setEncryption(Message.ENCRYPTION_DECRYPTED);
                            mXmppConnectionService.updateMessage(message);
                            if (!inputFile.delete()) {
                                Log.w(Config.LOGTAG, "unable to delete pgp encrypted source file " + inputFile.getAbsolutePath());
                            }
                            skipNotificationPush = true;
                            mXmppConnectionService.getFileBackend().updateMediaScanner(outputFile, () -> notifyIfPending(message));
                            break;
                        case OpenPgpApi.RESULT_CODE_USER_INTERACTION_REQUIRED:
                            synchronized (PgpDecryptionService.this) {
                                PendingIntent pendingIntent = result.getParcelableExtra(OpenPgpApi.RESULT_INTENT);
                                messages.addFirst(message);
                                currentMessage = null;
                                storePendingIntent(pendingIntent);
                            }
                            break;
                        case OpenPgpApi.RESULT_CODE_ERROR:
                            message.setEncryption(Message.ENCRYPTION_DECRYPTION_FAILED);
                            mXmppConnectionService.updateMessage(message);
                            break;
                    }
                } catch (final IOException e) {
                    message.setEncryption(Message.ENCRYPTION_DECRYPTION_FAILED);
                    mXmppConnectionService.updateMessage(message);
                }
            }
        }
        if (!skipNotificationPush) {
            notifyIfPending(message);
        }
    }

    private synchronized void notifyIfPending(Message message) {
        if (pendingNotifications.remove(message)) {
            mXmppConnectionService.getNotificationService().push(message);
        }
    }

    private void storePendingIntent(PendingIntent pendingIntent) {
        this.pendingIntent = pendingIntent;
        mXmppConnectionService.updateConversationUi();
    }

    public synchronized boolean hasPendingIntent(Conversation conversation) {
        if (pendingIntent == null) {
            return false;
        } else {
            for (Message message : messages) {
                if (message.getConversation() == conversation) {
                    return true;
                }
            }
            return false;
        }
    }

    public PendingIntent getPendingIntent() {
        return pendingIntent;
    }

    public boolean isConnected() {
        return getOpenPgpApi() != null;
    }
}