com.eduworks.cruncher.security.CruncherAesEncrypt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ew.levr.security Show documentation
The newest version!
package com.eduworks.cruncher.security;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidAlgorithmParameterException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.Map;

import javax.crypto.Cipher;
import javax.crypto.CipherOutputStream;
import javax.crypto.NoSuchPaddingException;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;

import org.apache.commons.codec.binary.Base64;
import org.json.JSONException;
import org.json.JSONObject;

import com.eduworks.resolver.Context;
import com.eduworks.resolver.Cruncher;

public class CruncherAesEncrypt extends Cruncher
{
	@Override
	public Object resolve(Context c, Map parameters, Map dataStreams) throws JSONException
	{
		CruncherRsaGenerate.checkProvider();
		Object obj = getObj(c, parameters, dataStreams);
		if (obj == null)
			return null;
		String payload = obj.toString();
		String secret = getAsString("secret", c, parameters, dataStreams);
		String iv = getAsString("iv", c, parameters, dataStreams);

		try
		{
			IvParameterSpec ivParameter = new IvParameterSpec(Base64.decodeBase64(iv));
			SecretKeySpec aesKey = new SecretKeySpec(Base64.decodeBase64(secret), "AES");

			// Encrypt cipher
			Cipher encryptCipher = Cipher.getInstance("AES/CTR/PKCS5Padding");
			encryptCipher.init(Cipher.ENCRYPT_MODE, aesKey, ivParameter);

			// Encrypt
			ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
			CipherOutputStream cipherOutputStream = new CipherOutputStream(outputStream, encryptCipher);
			cipherOutputStream.write(payload.getBytes());
			cipherOutputStream.flush();
			cipherOutputStream.close();
			byte[] encryptedBytes = outputStream.toByteArray();

			return new String(Base64.encodeBase64(encryptedBytes));
		} catch (InvalidKeyException e)
		{
			e.printStackTrace();
		} catch (IOException e)
		{
			e.printStackTrace();
		} catch (NoSuchAlgorithmException e)
		{
			e.printStackTrace();
		} catch (NoSuchPaddingException e)
		{
			e.printStackTrace();
		} catch (InvalidAlgorithmParameterException e)
		{
			e.printStackTrace();
		}
		return null;
	}

	@Override
	public String getDescription()
	{
		return "Encrypts an AES string using a secret and IV.";
	}

	@Override
	public String getReturn()
	{
		return "String";
	}

	@Override
	public String getAttribution()
	{
		return ATTRIB_NONE;
	}

	@Override
	public JSONObject getParameters() throws JSONException
	{
		return jo("obj", "String", "iv", "String", "secret", "String");
	}
}