Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/* license: https://mit-license.org
* ==============================================================================
* The MIT License (MIT)
*
* Copyright (c) 2019 Albert Moky
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
* ==============================================================================
*/
package chat.dim.plugins;
import javax.crypto.BadPaddingException;
import javax.crypto.Cipher;
import javax.crypto.IllegalBlockSizeException;
import javax.crypto.NoSuchPaddingException;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.security.*;
import java.util.Map;
import java.util.Random;
import chat.dim.format.Base64;
import chat.dim.impl.SymmetricKeyImpl;
/**
* AES Key
*
* keyInfo format: {
* algorithm: "AES",
* keySize : 32, // optional
* data : "{BASE64_ENCODE}}" // password data
* iv : "{BASE64_ENCODE}", // initialization vector
* }
*/
public final class AESKey extends SymmetricKeyImpl {
private final Cipher cipher;
private final SecretKeySpec keySpec;
private final IvParameterSpec ivSpec;
public AESKey(Map dictionary) throws NoSuchPaddingException, NoSuchAlgorithmException, NoSuchProviderException {
super(dictionary);
// TODO: check algorithm parameters
// 1. check mode = 'CBC'
// 2. check padding = 'PKCS7Padding'
Cipher aes;
try {
aes = Cipher.getInstance("AES/CBC/PKCS7Padding", "BC");
} catch (NoSuchAlgorithmException e) {
//e.printStackTrace();
aes = Cipher.getInstance("AES/CBC/PKCS7Padding");
}
cipher = aes;
keySpec = new SecretKeySpec(getData(), "AES");
ivSpec = new IvParameterSpec(getInitVector());
}
private int getKeySize() {
// TODO: get from key data
Object size = dictionary.get("keySize");
if (size == null) {
return 32;
} else {
return (int) size;
}
}
private int getBlockSize() {
// TODO: get from iv data
Object size = dictionary.get("blockSize");
if (size == null) {
return cipher.getBlockSize();
} else {
return (int) size;
}
}
private static byte[] randomData(int size) {
Random random = new Random();
byte[] buffer = new byte[size];
random.nextBytes(buffer);
return buffer;
}
private byte[] zeroData(int size) {
return new byte[size];
}
private byte[] getInitVector() {
Object iv = dictionary.get("iv");
if (iv == null) {
iv = dictionary.get("I");
}
if (iv != null) {
return Base64.decode((String) iv);
}
// zero iv
int blockSize = getBlockSize();
byte[] zeros = zeroData(blockSize);
dictionary.put("iv", Base64.encode(zeros));
return zeros;
}
@Override
public byte[] getData() {
Object data = dictionary.get("data");
if (data == null) {
data = dictionary.get("D");
}
if (data != null) {
return Base64.decode((String) data);
}
//
// key data empty? generate new key info
//
// random key data
int keySize = getKeySize();
byte[] pw = randomData(keySize);
dictionary.put("data", Base64.encode(pw));
// random initialization vector
int blockSize = getBlockSize();
byte[] iv = randomData(blockSize);
dictionary.put("iv", Base64.encode(iv));
// other parameters
//dictionary.put("mode", "CBC");
//dictionary.put("padding", "PKCS7");
return pw;
}
@Override
public byte[] encrypt(byte[] plaintext) {
try {
cipher.init(Cipher.ENCRYPT_MODE, keySpec, ivSpec);
return cipher.doFinal(plaintext);
} catch (InvalidKeyException | InvalidAlgorithmParameterException |
IllegalBlockSizeException | BadPaddingException e) {
e.printStackTrace();
return null;
}
}
@Override
public byte[] decrypt(byte[] ciphertext) {
try {
cipher.init(Cipher.DECRYPT_MODE, keySpec, ivSpec);
return cipher.doFinal(ciphertext);
} catch (InvalidKeyException | InvalidAlgorithmParameterException |
IllegalBlockSizeException | BadPaddingException e) {
e.printStackTrace();
return null;
}
}
static {
Security.addProvider(new org.bouncycastle.jce.provider.BouncyCastleProvider());
}
}
