upgrade.java

一套JAVA的CA证书签发系统.

/*************************************************************************
 *                                                                       *
 *  EJBCA: The OpenSource Certificate Authority                          *
 *                                                                       *
 *  This software is free software; you can redistribute it and/or       *
 *  modify it under the terms of the GNU Lesser General Public           *
 *  License as published by the Free Software Foundation; either         *
 *  version 2.1 of the License, or any later version.                    *
 *                                                                       *
 *  See terms of license at gnu.org.                                     *
 *                                                                       *
 *************************************************************************/
 
package se.anatom.ejbca.admin;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.rmi.RemoteException;

import javax.ejb.CreateException;
import javax.naming.Context;
import javax.naming.NamingException;

import org.apache.log4j.PropertyConfigurator;

import se.anatom.ejbca.upgrade.IUpgradeSessionRemote;
import se.anatom.ejbca.upgrade.IUpgradeSessionHome;

/**
 * Implements call to the upgrade function
 *
 * @version $Id: Upgrade.java,v 1.9 2004/05/14 08:25:32 anatom Exp $
 */
public class Upgrade extends BaseCommand {

    /**
     * 
     */
    public Upgrade() {
        super();
    }
    
    public boolean upgrade() {
        debug(">upgrade");
        
        boolean ret = false;
        
        String database = System.getProperty("ejbcaDB");
        debug("ejbcaDB="+database);
        String datasource = System.getProperty("ejbcaDS");
        debug("ejbcaDS="+datasource);
        String caname= System.getProperty("ejbcaCA");
        debug("ejbcaCA="+caname);
        String keystore = System.getProperty("ejbcaKS");
        debug("ejbcaKS="+keystore);
        String kspwd = System.getProperty("ejbcaKSPWD");
        debug("ejbcaKSPWD="+kspwd);
        String os = System.getProperty("ejbcaOS");
        debug("ejbcaOS="+os);
        String url = System.getProperty("ejbcaURL");
        debug("ejbcaURL="+url);
        
        // Check prerequisited
        if (!appServerRunning()) {
           error("The application server must be running.");
           return false;
        }
       // Upgrade the database
       try {
          IUpgradeSessionRemote upgradesession = getUpgradeSessionRemote();
          String[] args = new String[5];
          args[0] = database;
          args[1] = datasource;
          args[2] = caname;
          args[3] = keystore;
          args[4] = kspwd;
          ret = upgradesession.upgrade(administrator, args);
       } catch (Exception e) {
           error("Can't upgrade: ", e);
           ret = false;
       }
       
       if (ret) {
           // Fix the adminweb URL, must be done after database migration
           if(os.equalsIgnoreCase("windows")) {
               try {
                  String[] command = new String[4];
                  command[0] = "setup.cmd";
                  command[1] = "setbaseurl";
                  command[2] = url;
                  command[3] = "ejbca";
                  Process runcainit = Runtime.getRuntime().exec(command);
                  
                  BufferedReader br = new BufferedReader(new InputStreamReader(runcainit.getInputStream()));
                  Thread.sleep(1000);
                  String line = "";
                  while((line = br.readLine()) != null){
                      System.out.println(line);
                  }
                  if(runcainit.waitFor() != 0){                   
                      error("Error setting baseurl");
                      return false;
                  }               
              } catch (Exception e) {             
                  error("Error setting baseurl:", e);
                  return false;
              }   
          }
          if(os.equalsIgnoreCase("unix")) {           
              try {
                  String[] command = new String[4];
                  command[0] = "./setup.sh";
                  command[1] = "setbaseurl";
                  command[2] = url;
                  command[3] = "ejbca";
                  Process runcainit = Runtime.getRuntime().exec(command);
                                                  
                  BufferedReader br = new BufferedReader(new InputStreamReader(runcainit.getInputStream()));
                  Thread.sleep(1000);
                  String line = "";
                  while((line = br.readLine()) != null){
                      System.out.println(line);
                  }   
                  if(runcainit.waitFor() != 0){
                      error("Error setting baseurl");
                      return false;
                  }               
              } catch (Exception e) {
                  error("Error setting baseurl: ", e);
                  return false;
              }
          }           
       }

      debug("<upgrade");
      return ret;
    }

    protected IUpgradeSessionRemote getUpgradeSessionRemote() throws NamingException, CreateException, RemoteException {
        Context ctx = getInitialContext();
        IUpgradeSessionHome home = (IUpgradeSessionHome) javax.rmi.PortableRemoteObject.narrow(ctx.lookup("UpgradeSession"), IUpgradeSessionHome.class );            
        IUpgradeSessionRemote upgradesession = home.create();          
        return upgradesession;
     }
    
    /**
     * main Upgrade
     *
     * @param args command line arguments
     */
    public static void main(String[] args) {
        PropertyConfigurator.configure("log4j.properties");

        Upgrade upgrade = new Upgrade();
        try {
            boolean ret = upgrade.upgrade();
            if (!ret) {
                upgrade.error("Upgrade not performed, see server log for details.");
            } else {
            	upgrade.info("Upgrade completed.");   
            }
        } catch (Exception e) {
            //System.out.println(e.getMessage());
            upgrade.error("Error doing upgrade: ", e);
            System.exit(-1);
        }
    }

}