| /* |
| * Copyright (c) 2007, 2015, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. |
| * |
| * This code 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 |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| import java.io.ByteArrayInputStream; |
| import java.io.IOException; |
| import java.security.GeneralSecurityException; |
| import javax.crypto.Cipher; |
| import javax.crypto.CipherInputStream; |
| |
| /** |
| * This is an abstract class for CipherInputStream/CipherOutputStream PBE |
| * functional tests. |
| */ |
| public abstract class CICO_PBE_Test { |
| /** |
| * Sample string for byte buffer. |
| */ |
| public static final String BYTE_ARR_BUFFER = "byteArrayBuffering"; |
| |
| /** |
| * Sample string for int buffer. |
| */ |
| public static final String INT_BYTE_BUFFER = "intByteBuffering"; |
| public static final String PASS_PHRASE = "Some password phrase!"; |
| |
| /** |
| * Text string size. |
| */ |
| public static final int TEXT_SIZE = 800; |
| |
| protected final byte[] plainText; |
| private final Cipher encryptCipher, decryptCipher; |
| |
| /** |
| * An CipherInputStream for reading cipher and plain text. |
| */ |
| private final CipherInputStream ciInput; |
| |
| /** |
| * Constructor by algorithm. |
| * @param pbeAlgo PBE algorithm to test. |
| * @throws GeneralSecurityException if any security error. |
| */ |
| public CICO_PBE_Test(PBEAlgorithm pbeAlgo) throws GeneralSecurityException { |
| // Do initialization of the plainText |
| plainText = TestUtilities.generateBytes(TEXT_SIZE); |
| // Do initialization of the ciphers |
| AbstractPBEWrapper pbeWrap = AbstractPBEWrapper.createWrapper(pbeAlgo, PASS_PHRASE); |
| encryptCipher = pbeWrap.initCipher(Cipher.ENCRYPT_MODE); |
| decryptCipher = pbeWrap.initCipher(Cipher.DECRYPT_MODE); |
| // init cipher input stream |
| ciInput = new CipherInputStream(new ByteArrayInputStream(plainText), |
| encryptCipher); |
| } |
| |
| protected byte[] getPlainText() { |
| return plainText; |
| } |
| |
| /** |
| * The body of the test. Should be defined in subclasses. |
| * @param type byteArrayBuffering or intByteBuffering |
| * @throws IOException I/O operation failed. |
| * @throws GeneralSecurityException all exceptions thrown. |
| */ |
| protected abstract void proceedTest(String type) |
| throws IOException, GeneralSecurityException; |
| |
| protected Cipher getEncryptCipher() { |
| return encryptCipher; |
| } |
| |
| public CipherInputStream getCiInput() { |
| return ciInput; |
| } |
| |
| public Cipher getDecryptCipher() { |
| return decryptCipher; |
| } |
| } |