com.scudata.expression.fn.string.DESEncrypt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
There is a newer version: 20241126
package com.scudata.expression.fn.string;

import javax.crypto.*;
import javax.crypto.spec.*;

import java.io.UnsupportedEncodingException;

import com.scudata.common.Logger;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.expression.IParam;
import com.scudata.expression.fn.CharFunction;
import com.scudata.resources.EngineMessage;

/**
 * des(input, key, iv; charset) DES????/????
 * parament input	???룬blob????String
 * parament key		???룬??????ͳһ
 * parament iv		??ʼ??????????????ͳһ
 * parament charset	?ַ????룬???input/key/ivΪ?ַ???ʱʹ?ã?Ĭ??Ϊutf8
 * ѡ??@d	???ܣ?Ĭ??ִ?м???
 * ѡ??@c	CBC(Cipher Block Chaining)ģʽ??Ĭ??ʹ??ECB(Electronic Codebook)ģʽ	
 * ѡ??@n	???ģʽNoPadding??ȱʡPKCS5Padding
 * ???أ?inputΪ?ַ???ʱ?????ַ?????Ϊ??????????ʱ???ض?????????
 * @author bd
 *
 */
public class DESEncrypt extends CharFunction {	
    // DES??Կ?㷨
    private static final String KEY_ALGORITHM = "DES";
 
    // ????/?????㷨/????ģʽ/??䷽ʽ
    private static final String CIPHER_ALGORITHM_E = "DES/ECB/PKCS5Padding";
    private static final String CIPHER_ALGORITHM_C = "DES/CBC/PKCS5Padding";
    private static final String CIPHER_ALGORITHM_EN = "DES/ECB/NoPadding";
    private static final String CIPHER_ALGORITHM_CN = "DES/CBC/NoPadding";
    
    protected static final byte[] DEF_IV = DESEncrypt.encode("1a3b5c7d", "utf-8");
     
	protected Object doQuery(Object[] objs) {
		try {
			if (objs==null || objs.length<2){
				MessageManager mm = EngineMessage.get();
				throw new RQException("des" + mm.getMessage("function.invalidParam"));
			}
			Object[] params = objs;
			String charset = "utf-8";
			if (param.getType() == IParam.Semicolon) {
				Object obj = objs[0];
				if (obj instanceof Object[]) {
					params = (Object[])obj;
				}
				else {
					params = new Object[1];
					params[0] = obj;
				}
				//des(...;charset)?????
				if (objs[1] instanceof String) {
					charset = (String) objs[1];
				}
				else {
					MessageManager mm = EngineMessage.get();
					throw new RQException("des(...;charset), charset is invalid." + mm.getMessage("function.invalidParam"));
				}
			}
			if (params == null || params.length < 2){
				MessageManager mm = EngineMessage.get();
				throw new RQException("des" + mm.getMessage("function.invalidParam"));
			}
			//boolean string = params[0] instanceof String;
			byte[] inputb = DESEncrypt.encode(params[0], charset);
			byte[] keyb = DESEncrypt.encode(params[1], charset);
			byte[] ivb = null;
			if (params.length > 2) {
				ivb = DESEncrypt.encode(params[2], charset);
			}
			boolean dec = false;
			boolean modeC = false;
			boolean nopad = false;
			boolean autopad = false;
			boolean decStr = false;
			if (option != null) {
				if (option.indexOf('d')>-1) {
					dec = true;
				}
				if (option.indexOf('c')>-1) {
					modeC = true;
				}
				if (option.indexOf('n')>-1) {
					nopad = true;
				}
				if (option.indexOf('a')>-1) {
					autopad = true;
				}
				if (option.indexOf('s')>-1) {
					decStr = true;
				}
			}
			if (nopad) {
				if (inputb.length % 8 != 0 ) {
					if (autopad) {
						MessageManager mm = EngineMessage.get();
						Logger.info("des(input, key, iv; charset): " + mm.getMessage("encrypt.autopadding", "input", 8));
						inputb = AESEncrypt.autoPadding(inputb, 8);
					}
					else {
						MessageManager mm = EngineMessage.get();
						throw new RQException("des(input, key, iv; charset): " + mm.getMessage("encrypt.nopadding", "input", 8));
					}
				}
			}
			if (keyb.length < 8) {
				if (autopad) {
					keyb = AESEncrypt.padding(keyb, 8);
				}
				else {
					MessageManager mm = EngineMessage.get();
					throw new RQException("des(input, key, iv; charset): " + mm.getMessage("encrypt.lenless", "key", 8));
				}
			}
			Cipher ci = null;
			if (modeC) {
				if (ivb == null) {
					MessageManager mm = EngineMessage.get();
					Logger.info("des(input, key, iv; charset): " + mm.getMessage("encrypt.CBCnoiv"));
					ivb = DEF_IV;
				}
				else if (ivb.length != 8) {
					if (autopad) {
						ivb = AESEncrypt.padding(keyb, 8);
					}
					else {
						MessageManager mm = EngineMessage.get();
						throw new RQException("des(input, key, iv; charset): " + mm.getMessage("encrypt.wronglen", "iv", 8));
					}
				}
				if (nopad) {
					ci = Cipher.getInstance(DESEncrypt.CIPHER_ALGORITHM_CN);
				} else {
					ci = Cipher.getInstance(DESEncrypt.CIPHER_ALGORITHM_C);
				}
			} else {
				if (nopad) {
					ci = Cipher.getInstance(DESEncrypt.CIPHER_ALGORITHM_EN);
				} else {
					ci = Cipher.getInstance(DESEncrypt.CIPHER_ALGORITHM_E);
				}
			}
			IvParameterSpec iv = null;
			if (ivb != null) {
				iv = new IvParameterSpec(ivb);
			}
			DESKeySpec dks = new DESKeySpec(keyb);
			SecretKeyFactory keyFactory = SecretKeyFactory.getInstance(KEY_ALGORITHM);  
	        //??????Կ  
	        SecretKey key = keyFactory.generateSecret(dks);
			byte[] result = null;
			if (dec) {
				if (!modeC) {
					ci.init(Cipher.DECRYPT_MODE, key);
					result = ci.doFinal(inputb);
				}
				else {
					ci.init(Cipher.DECRYPT_MODE, key, iv);
					result = ci.doFinal(inputb);
				}
				if (decStr) {
					return DESEncrypt.encode(result, charset);
				}
			} else {
				if (!modeC) {
					ci.init(Cipher.ENCRYPT_MODE, key);
					result = ci.doFinal(inputb);
				}
				else {
					ci.init(Cipher.ENCRYPT_MODE, key, iv);
					result = ci.doFinal(inputb);
				}
			}
			/*
			if (string) {			
				if (charset.equalsIgnoreCase("unicodeescape")  ||
						charset.equalsIgnoreCase("unicode escape") ||
						charset.equalsIgnoreCase("unicode-escape") ){
					return CharEncode.utf_unescape(new String(result));
				}else{
					return new String(result, charset);
				}
			}
			*/
			return result;
		} catch (Exception e) {
			Logger.error(e.getMessage());
		}
		
		return null;
	}
	
	protected static String encode(byte[] bytes, String charset) throws UnsupportedEncodingException {		
		if (charset.equalsIgnoreCase("unicodeescape")  ||
				charset.equalsIgnoreCase("unicode escape") ||
				charset.equalsIgnoreCase("unicode-escape") ){
			return CharEncode.utf_unescape(new String(bytes));
		}else{
			return new String(bytes, charset);
		}
	}
	
	protected static byte[] encode(Object obj, String encodeName) {
		if (obj == null) return null;
		if(obj instanceof byte[] ){
			return (byte[])obj;
		}else { //encode
			String str = null;
			if(obj instanceof String ){
				str = (String)obj;
			}
			else {
				str = obj.toString();
			}
			if (str != null) {
				if (encodeName.equalsIgnoreCase("unicodeescape")  ||
						encodeName.equalsIgnoreCase("unicode escape") ||
						encodeName.equalsIgnoreCase("unicode-escape") ){
					str = CharEncode.utf_escape(str);
					return str.getBytes(); 
				} else {
					try {
						return str.getBytes(encodeName);
					} catch (UnsupportedEncodingException e) {
						e.printStackTrace();
						Logger.error(e.getMessage());
						return null;
					}
				}
			}
			else {
				MessageManager mm = EngineMessage.get();
				throw new RQException("encode error: " + mm.getMessage("function.paramTypeError"));
			}
		}
	}
}