cadi/core/src/test/java/org/onap/aaf/cadi/test/JU_AES.java

   1 /*******************************************************************************
   2  * ============LICENSE_START====================================================
   3  * * org.onap.aaf
   4  * * ===========================================================================
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  * * ===========================================================================
   7  * * Licensed under the Apache License, Version 2.0 (the "License");
   8  * * you may not use this file except in compliance with the License.
   9  * * You may obtain a copy of the License at
  10  * *
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0
  12  * *
  13  *  * Unless required by applicable law or agreed to in writing, software
  14  * * distributed under the License is distributed on an "AS IS" BASIS,
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * * See the License for the specific language governing permissions and
  17  * * limitations under the License.
  18  * * ============LICENSE_END====================================================
  19  * *
  20  * *
  21  ******************************************************************************/
  22 package org.onap.aaf.cadi.test;
  23
  24 import static org.hamcrest.CoreMatchers.*;
  25 import static org.junit.Assert.*;
  26 import org.junit.*;
  27
  28
  29 import java.io.ByteArrayInputStream;
  30 import java.io.ByteArrayOutputStream;
  31 import java.io.File;
  32 import java.io.IOException;
  33 import java.io.InputStream;
  34 import java.io.OutputStream;
  35 import java.io.PrintStream;
  36 import java.lang.reflect.Field;
  37 import java.nio.file.Files;
  38 import java.nio.file.Paths;
  39
  40 import javax.crypto.CipherInputStream;
  41 import javax.crypto.CipherOutputStream;
  42 import javax.crypto.SecretKey;
  43
  44 import org.onap.aaf.cadi.AES;
  45 import org.onap.aaf.cadi.CadiException;
  46 import org.onap.aaf.cadi.Symm;
  47
  48 public class JU_AES {
  49         private AES aes;
  50         private ByteArrayInputStream baisEncrypt;
  51         private ByteArrayInputStream baisDecrypt;
  52         private ByteArrayOutputStream baosEncrypt;
  53         private ByteArrayOutputStream baosDecrypt;
  54
  55         private ByteArrayOutputStream errStream;
  56
  57         @Before
  58         public void setup() throws Exception {
  59                 byte[] keyBytes = new byte[AES.AES_KEY_SIZE/8];
  60                 char[] codeset = Symm.base64.codeset;
  61                 int offset = (Math.abs(codeset[0]) + 47) % (codeset.length - keyBytes.length);
  62                 for(int i = 0; i < keyBytes.length; ++i) {
  63                         keyBytes[i] = (byte)codeset[i+offset];
  64                 }
  65                 aes = new AES(keyBytes, 0, keyBytes.length);
  66
  67                 errStream = new ByteArrayOutputStream();
  68                 System.setErr(new PrintStream(errStream));
  69         }
  70
  71         @After
  72         public void tearDown() {
  73                 System.setErr(System.err);
  74         }
  75
  76         @Test
  77         public void newKeyTest() throws Exception {
  78                 SecretKey secretKey = AES.newKey();
  79                 assertThat(secretKey.getAlgorithm(), is(AES.class.getSimpleName()));
  80         }
  81
  82         @Test
  83         public void encryptDecrpytFromBytes() throws Exception {
  84                 String orig = "I'm a password, really";
  85                 byte[] encrypted = aes.encrypt(orig.getBytes());
  86                 byte[] decrypted = aes.decrypt(encrypted);
  87                 assertThat(new String(decrypted), is(orig));
  88
  89                 Field aeskeySpec_field = AES.class.getDeclaredField("aeskeySpec");
  90                 aeskeySpec_field.setAccessible(true);
  91                 aeskeySpec_field.set(aes, null);
  92
  93                 try {
  94                         aes.encrypt(orig.getBytes());
  95                         fail("Should have thrown an exception");
  96                 } catch (CadiException e) {
  97                 }
  98                 try {
  99                         aes.decrypt(encrypted);
 100                         fail("Should have thrown an exception");
 101                 } catch (CadiException e) {
 102                 }
 103         }
 104
 105         @Test
 106         public void saveToFileTest() throws Exception {
 107                 String filePath = "test/output_key";
 108                 File keyfile = new File(filePath);
 109                 aes.save(keyfile);
 110                 assertTrue(Files.isReadable(Paths.get(filePath)));
 111                 assertFalse(Files.isWritable(Paths.get(filePath)));
 112                 assertFalse(Files.isExecutable(Paths.get(filePath)));
 113                 keyfile.delete();
 114         }
 115
 116         @Test
 117         public void encryptDecryptFromInputStream() throws Exception {
 118                 String orig = "I'm a password, really";
 119                 byte[] b64encrypted;
 120                 String output;
 121
 122                 CipherInputStream cisEncrypt;
 123                 CipherInputStream cisDecrypt;
 124
 125                 // Test CipherInputStream
 126                 baisEncrypt = new ByteArrayInputStream(orig.getBytes());
 127                 cisEncrypt = aes.inputStream(baisEncrypt, true);
 128                 baosEncrypt = new ByteArrayOutputStream();
 129                 transferFromInputStreamToOutputStream(cisEncrypt, baosEncrypt);
 130                 cisEncrypt.close();
 131
 132                 b64encrypted = baosEncrypt.toByteArray();
 133
 134                 baisDecrypt = new ByteArrayInputStream(b64encrypted);
 135                 cisDecrypt = aes.inputStream(baisDecrypt, false);
 136                 baosDecrypt = new ByteArrayOutputStream();
 137                 transferFromInputStreamToOutputStream(cisDecrypt, baosDecrypt);
 138                 cisDecrypt.close();
 139
 140                 output = new String(baosDecrypt.toByteArray());
 141                 assertThat(output, is(orig));
 142
 143                 Field aeskeySpec_field = AES.class.getDeclaredField("aeskeySpec");
 144                 aeskeySpec_field.setAccessible(true);
 145                 aeskeySpec_field.set(aes, null);
 146
 147                 assertNull(aes.inputStream(baisEncrypt, true));
 148                 assertThat(errStream.toString(), is("Error creating Aes CipherInputStream\n"));
 149         }
 150
 151         @Test
 152         public void encryptDecryptFromOutputStream() throws Exception {
 153                 String orig = "I'm a password, really";
 154                 byte[] b64encrypted;
 155                 String output;
 156
 157                 CipherOutputStream cosEncrypt;
 158                 CipherOutputStream cosDecrypt;
 159
 160                 // Test CipherOutputStream
 161                 baisEncrypt = new ByteArrayInputStream(orig.getBytes());
 162                 baosEncrypt = new ByteArrayOutputStream();
 163                 cosEncrypt = aes.outputStream(baosEncrypt, true);
 164                 transferFromInputStreamToOutputStream(baisEncrypt, cosEncrypt);
 165                 cosEncrypt.close();
 166
 167                 b64encrypted = baosEncrypt.toByteArray();
 168
 169                 baosDecrypt = new ByteArrayOutputStream();
 170                 cosDecrypt = aes.outputStream(baosDecrypt, false);
 171                 baisDecrypt = new ByteArrayInputStream(b64encrypted);
 172                 transferFromInputStreamToOutputStream(baisDecrypt, cosDecrypt);
 173                 cosDecrypt.close();
 174
 175                 output = new String(baosDecrypt.toByteArray());
 176                 assertThat(output, is(orig));
 177
 178                 Field aeskeySpec_field = AES.class.getDeclaredField("aeskeySpec");
 179                 aeskeySpec_field.setAccessible(true);
 180                 aeskeySpec_field.set(aes, null);
 181
 182                 assertNull(aes.outputStream(baosEncrypt, true));
 183                 assertThat(errStream.toString(), is("Error creating Aes CipherOutputStream\n"));
 184         }
 185
 186         public void transferFromInputStreamToOutputStream(InputStream is, OutputStream os) throws IOException {
 187                 byte[] buffer = new byte[200];
 188                 int len;
 189                 while ((len = is.read(buffer)) != -1) {
 190                     os.write(buffer, 0, len);
 191                 }
 192         }
 193
 194 }