/*
 * Copyright 2001-2013 The Apache Software Foundation.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package org.apache.juddi.samples;

import java.security.InvalidAlgorithmParameterException;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.security.spec.KeySpec;
import javax.crypto.BadPaddingException;
import javax.crypto.Cipher;
import javax.crypto.IllegalBlockSizeException;
import javax.crypto.KeyGenerator;
import javax.crypto.NoSuchPaddingException;
import javax.crypto.SecretKey;
import javax.crypto.SecretKeyFactory;
import javax.crypto.spec.DESedeKeySpec;
import org.apache.commons.codec.binary.Base64;

/**
 * This call is a simple test class that shows how to encrypt stuff in Java
 * using 3DES
 *
 * @author <a href="mailto:alexoree@apache.org">Alex O'Ree</a>
 */
public class DES {

        private static final String UNICODE_FORMAT = "UTF8";
        public static final String DESEDE_ENCRYPTION_SCHEME = "DESede";
        static private KeySpec ks;
        static private SecretKeyFactory skf;
        static private Cipher cipher;
        static byte[] arrayBytes;
        static private String myEncryptionKey;
        static private String myEncryptionScheme;
        static SecretKey key;

        public DES() throws Exception {
        }

        public String encrypt(String clear) throws NoSuchPaddingException, NoSuchAlgorithmException, InvalidAlgorithmParameterException, InvalidKeyException, IllegalBlockSizeException, BadPaddingException {
                String encryptedString = null;
                try {
                        cipher.init(Cipher.ENCRYPT_MODE, key);
                        byte[] plainText = clear.getBytes(UNICODE_FORMAT);
                        byte[] encryptedText = cipher.doFinal(plainText);
                        encryptedString = new String(Base64.encodeBase64(encryptedText), "UTF-8");
                } catch (Exception e) {
                        e.printStackTrace();
                }
                return encryptedString;
        }

        public String decrypt(String str) throws NoSuchPaddingException, NoSuchAlgorithmException, InvalidAlgorithmParameterException, InvalidKeyException, IllegalBlockSizeException, BadPaddingException {
                String encryptedString = null;
                try {
                        cipher.init(Cipher.DECRYPT_MODE, key);
                        byte[] encryptedText = Base64.decodeBase64(str.getBytes("UTF-8"));
                        byte[] clear = cipher.doFinal(encryptedText);
                        encryptedString = new String(Base64.encodeBase64(clear), "UTF-8");
                } catch (Exception e) {
                        e.printStackTrace();
                }
                return encryptedString;
        }

        public static void main(String[] args) throws Exception {
                DES des = new DES();
                KeyGenerator kgen;
                try {
                        kgen = KeyGenerator.getInstance(DESEDE_ENCRYPTION_SCHEME);
                        kgen.init(168);
                        SecretKey skey = kgen.generateKey();
                        byte[] raw = skey.getEncoded();
                        myEncryptionKey = new String(Base64.encodeBase64(raw), "UTF-8");
                        myEncryptionScheme = DESEDE_ENCRYPTION_SCHEME;
                        arrayBytes = myEncryptionKey.getBytes(UNICODE_FORMAT);
                        ks = new DESedeKeySpec(arrayBytes);
                        skf = SecretKeyFactory.getInstance(myEncryptionScheme);
                        cipher = Cipher.getInstance(myEncryptionScheme);
                        key = skf.generateSecret(ks);

                        System.out.println(new String(Base64.encodeBase64(raw), "UTF-8"));
                        System.out.println(des.encrypt("password"));
                        System.out.println(des.decrypt(des.encrypt("password")));
                } catch (Exception ex) {
                        ex.printStackTrace();;
                }

        }
}