/* * Copyright 2018 New Vector Ltd * * 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.matrix.olm; import android.util.Log; import java.util.Arrays; public class OlmPkDecryption { private static final String LOG_TAG = "OlmPkDecryption"; /** Session Id returned by JNI. * This value uniquely identifies the native session instance. **/ private transient long mNativeId; public OlmPkDecryption() throws OlmException { try { mNativeId = createNewPkDecryptionJni(); } catch (Exception e) { throw new OlmException(OlmException.EXCEPTION_CODE_PK_DECRYPTION_CREATION, e.getMessage()); } } private native long createNewPkDecryptionJni(); private native void releasePkDecryptionJni(); public void releaseDecryption() { if (0 != mNativeId) { releasePkDecryptionJni(); } mNativeId = 0; } public boolean isReleased() { return (0 == mNativeId); } public String generateKey() throws OlmException { try { byte[] key = generateKeyJni(); return new String(key, "UTF-8"); } catch (Exception e) { Log.e(LOG_TAG, "## setRecipientKey(): failed " + e.getMessage()); throw new OlmException(OlmException.EXCEPTION_CODE_PK_DECRYPTION_GENERATE_KEY, e.getMessage()); } } private native byte[] generateKeyJni(); public String decrypt(OlmPkMessage aMessage) throws OlmException { if (null == aMessage) { return null; } try { byte[] plaintextBuffer = decryptJni(aMessage); String plaintext = new String(plaintextBuffer, "UTF-8"); Arrays.fill(plaintextBuffer, (byte) 0); return plaintext; } catch (Exception e) { Log.e(LOG_TAG, "## pkDecrypt(): failed " + e.getMessage()); throw new OlmException(OlmException.EXCEPTION_CODE_PK_DECRYPTION_DECRYPT, e.getMessage()); } } private native byte[] decryptJni(OlmPkMessage aMessage); }