Smack/smack-openpgp/src/main/java/org/jivesoftware/smackx/ox/OpenPgpMessage.java

/**
 *
 * Copyright 2017 Florian Schmaus.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jivesoftware.smackx.ox;

import java.io.IOException;
import java.nio.charset.Charset;
import java.util.HashSet;
import java.util.Set;

import org.jivesoftware.smack.util.Objects;
import org.jivesoftware.smackx.ox.element.CryptElement;
import org.jivesoftware.smackx.ox.element.OpenPgpContentElement;
import org.jivesoftware.smackx.ox.element.OpenPgpElement;
import org.jivesoftware.smackx.ox.element.SignElement;
import org.jivesoftware.smackx.ox.element.SigncryptElement;
import org.jivesoftware.smackx.ox.provider.OpenPgpContentElementProvider;

import org.xmlpull.v1.XmlPullParserException;

/**
 * This class embodies a decrypted {@link OpenPgpElement}.
 */
public class OpenPgpMessage {

    public enum State {
        /**
         * Represents a {@link SigncryptElement}.
         */
        signcrypt,
        /**
         * Represents a {@link SignElement}.
         */
        sign,
        /**
         * Represents a {@link CryptElement}.
         */
        crypt,
        ;
    }

    private final String element;
    private final State state;
    private final Metadata metadata;

    private OpenPgpContentElement openPgpContentElement;

    /**
     * Constructor.
     *
     * @param metadata Metadata about the encryption
     * @param content XML representation of the decrypted {@link OpenPgpContentElement}.
     */
    public OpenPgpMessage(String content, Metadata metadata) {
        this.metadata = Objects.requireNonNull(metadata);
        this.state = Objects.requireNonNull(metadata.getState());
        this.element = Objects.requireNonNull(content);
    }

    public OpenPgpMessage(byte[] bytes, Metadata metadata) {
        this(new String(Objects.requireNonNull(bytes), Charset.forName("UTF-8")), metadata);
    }

    /**
     * Return the decrypted {@link OpenPgpContentElement} of this message.
     * To determine, whether the element is a {@link SignElement}, {@link CryptElement} or {@link SigncryptElement},
     * please consult {@link #getState()}.
     *
     * @return {@link OpenPgpContentElement}
     * @throws XmlPullParserException if the parser encounters an error.
     * @throws IOException if the parser encounters an error.
     */
    public OpenPgpContentElement getOpenPgpContentElement() throws XmlPullParserException, IOException {
        ensureOpenPgpContentElementSet();

        return openPgpContentElement;
    }

    private void ensureOpenPgpContentElementSet() throws XmlPullParserException, IOException {
        if (openPgpContentElement != null)
            return;

        openPgpContentElement = OpenPgpContentElementProvider.parseOpenPgpContentElement(element);
        if (openPgpContentElement == null) {
            return;
        }

        // Determine the state of the content element.
        if (openPgpContentElement instanceof SigncryptElement) {
            if (state != State.signcrypt) {
                throw new IllegalStateException("OpenPgpContentElement was signed and encrypted, but is not a SigncryptElement.");
            }
        } else if (openPgpContentElement instanceof SignElement) {
            if (state != State.sign) {
                throw new IllegalStateException("OpenPgpContentElement was signed and unencrypted, but is not a SignElement.");
            }
        } else if (openPgpContentElement instanceof CryptElement) {
            if (state != State.crypt) {
                throw new IllegalStateException("OpenPgpContentElement was unsigned and encrypted, but is not a CryptElement.");
            }
        }
    }

    /**
     * Return the state of the message. This value determines, whether the message was a {@link SignElement},
     * {@link CryptElement} or {@link SigncryptElement}.
     *
     * @return state of the content element.
     * @throws IOException if the parser encounters an error.
     * @throws XmlPullParserException if the parser encounters and error.
     */
    public State getState() throws IOException, XmlPullParserException {
        ensureOpenPgpContentElementSet();
        return state;
    }

    public Metadata getMetadata() {
        return metadata;
    }

    public static class Metadata {

        private final Long encryptionKeyId;
        private final Set<Long> validSignatureIds;

        public Metadata(Long encryptionKeyId, Set<Long> validSignatureIds) {
            this.encryptionKeyId = encryptionKeyId;
            this.validSignatureIds = validSignatureIds;
        }

        public Long getEncryptionKeyId() {
            return encryptionKeyId;
        }

        public Set<Long> getValidSignatureIds() {
            return new HashSet<>(validSignatureIds);
        }

        public State getState() {
            if (validSignatureIds.size() != 0) {
                if (encryptionKeyId != null) {
                    return State.signcrypt;
                } else {
                    return State.sign;
                }
            } else {
                if (encryptionKeyId != null) {
                    return State.crypt;
                } else {
                    throw new IllegalStateException("OpenPGP message appears to be neither encrypted, " +
                            "nor signed.");
                }
            }
        }
    }
}