oscar.oscarLab.ca.all.util.KeyPairGen.java Source code

Introduction

Here is the source code for oscar.oscarLab.ca.all.util.KeyPairGen.java

Source

/**
 * Copyright (c) 2001-2002. Department of Family Medicine, McMaster University. All Rights Reserved.
 * This software is published under the GPL GNU General Public License.
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version. 
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * This software was written for the
 * Department of Family Medicine
 * McMaster University
 * Hamilton
 * Ontario, Canada
 */

package oscar.oscarLab.ca.all.util;

import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.NoSuchAlgorithmException;

import org.apache.commons.codec.binary.Base64;
import org.apache.log4j.Logger;
import org.oscarehr.common.dao.OscarKeyDao;
import org.oscarehr.common.dao.PublicKeyDao;
import org.oscarehr.common.model.OscarKey;
import org.oscarehr.common.model.PublicKey;
import org.oscarehr.util.MiscUtils;
import org.oscarehr.util.SpringUtils;

public final class KeyPairGen {

    private static Logger logger = MiscUtils.getLogger();

    /**
     * Creates a new instance of KeyPairGen
     */
    private KeyPairGen() {
        // utility class needs no instantiation
    }

    /**
     *  Create a key pair for the specified service and store it in the database
     */
    public static String createKeys(String name, String type) {
        byte[] pubKey;
        byte[] privKey;
        Base64 base64 = new Base64();

        // Generate an RSA key
        try {
            KeyPairGenerator keyGen = KeyPairGenerator.getInstance("RSA");
            keyGen.initialize(1024);
            KeyPair key = keyGen.generateKeyPair();

            pubKey = base64.encode(key.getPublic().getEncoded());
            privKey = base64.encode(key.getPrivate().getEncoded());

            if (name.equals("oscar")) {
                // the primary key is name, therefore this statement will only
                // be able to run once and the oscar key pair will not be overwritten

                OscarKeyDao oscarKeyDao = (OscarKeyDao) SpringUtils.getBean("oscarKeyDao");
                OscarKey oscarKey = new OscarKey();
                oscarKey.setName("oscar");
                oscarKey.setPublicKey(new String(pubKey, MiscUtils.ENCODING));
                oscarKey.setPrivateKey(new String(privKey, MiscUtils.ENCODING));

                oscarKeyDao.persist(oscarKey);

                // no keys need to be returned so return "success" instead to 
                // indicate the operation completed successfully
                return ("success");
            } else {

                PublicKeyDao publicKeyDao = (PublicKeyDao) SpringUtils.getBean("publicKeyDao");
                PublicKey publicKeyObject = new PublicKey();
                publicKeyObject.setService(name);
                publicKeyObject.setType(type);
                publicKeyObject.setBase64EncodedPublicKey(new String(pubKey, MiscUtils.ENCODING));
                publicKeyObject.setBase64EncodedPrivateKey(new String(privKey, MiscUtils.ENCODING));

                publicKeyDao.persist(publicKeyObject);

                return (publicKeyObject.getBase64EncodedPrivateKey());
            }

        } catch (NoSuchAlgorithmException e) {
            logger.error("Could not create RSA key", e);
            return null;
        } catch (Exception e) {
            logger.error("Could not save keys", e);
            return null;
        }

    }

    /**
     *  Retrieve oscars public key and return it as a string
     */
    public static String getPublic() {
        try {
            OscarKeyDao oscarKeyDao = (OscarKeyDao) SpringUtils.getBean("oscarKeyDao");
            OscarKey oscarKey = oscarKeyDao.find("oscar");

            return (oscarKey.getPublicKey());
        } catch (Exception e) {
            logger.error("Could not get public key", e);
            return null;
        }

    }

    /**
     *  Check if the specified service name has already been used to create a key
     *  pair, return 'true' if it has and 'false' otherwise.
     */
    public static boolean checkName(String name) {
        if (name.equals("oscar")) {
            OscarKeyDao oscarKeyDao = (OscarKeyDao) SpringUtils.getBean("oscarKeyDao");
            OscarKey oscarKey = oscarKeyDao.find("oscar");
            return (oscarKey != null);
        } else {
            PublicKeyDao publicKeyDao = (PublicKeyDao) SpringUtils.getBean("publicKeyDao");
            PublicKey publicKeyObject = publicKeyDao.find(name);
            return (publicKeyObject != null);
        }
    }
}