crlcreatethread.java

一个免费的CA,基于EJB平台的,老师叫我们测试,现把之共享出来让大家参考

/*************************************************************************
 *                                                                       *
 *  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 org.ejbca.appserver.jboss;

import java.util.Calendar;
import java.util.Date;

import javax.ejb.EJBException;
import javax.naming.Context;
import javax.naming.InitialContext;

import org.apache.log4j.Logger;
import org.ejbca.core.model.log.Admin;

import org.ejbca.core.ejb.ca.crl.ICreateCRLSessionHome;
import org.ejbca.core.ejb.ca.crl.ICreateCRLSessionRemote;



/** 
 * A thread managing the automatic creation of CRLs.
*/
public class CRLCreateThread extends Thread 
{ 
    private static Logger log = Logger.getLogger(CRLCreateThread.class);

    /** Constants used internally. */
  private static final int CHECK_DAILY     = 1;
  private static final int CHECK_HOURLY = 2;
  private static final int CHECK_30MIN    = 3;
  private static final int CHECK_15MIN    = 4;
  private static final int CHECK_1MIN      = 5;

  /** We may create new CRLs if the current one expires within this the confiured overlap time + the pollTime */
  private long m_pollTime = 1*60*1000; // Default polltime is 1 minute
  
  private boolean run = false;
  private int check = 0; 
  private ICreateCRLSessionRemote createcrlsession = null;
  private Admin administrator = new Admin(Admin.TYPE_CACOMMANDLINE_USER);
  
   public CRLCreateThread(String polltime){
   	    super();
   	    
    	if(polltime.equalsIgnoreCase(CRLCreateService.POLLTIME_DAILY)){
     		check = CHECK_DAILY;
            m_pollTime = 24*60*60*1000;
     	}
   	    if(polltime.equalsIgnoreCase(CRLCreateService.POLLTIME_HOURLY)){
   		  check = CHECK_HOURLY;
          m_pollTime = 60*60*1000;
   	    }
   	    if(polltime.equalsIgnoreCase(CRLCreateService.POLLTIME_30MIN)){
   		  check = CHECK_30MIN;
          m_pollTime = 30*60*1000;
   	    }
   	    if(polltime.equalsIgnoreCase(CRLCreateService.POLLTIME_15MIN)){
   		  check = CHECK_15MIN;
          m_pollTime = 15*60*1000;
   	    }   
   	    if(polltime.equalsIgnoreCase(CRLCreateService.POLLTIME_1MIN)){
   		  check = CHECK_1MIN;
          m_pollTime = 60*1000;
   	    }
   	    
   	    try{
   	        Context context = new InitialContext();
   	        ICreateCRLSessionHome home = (ICreateCRLSessionHome) javax.rmi.PortableRemoteObject.narrow(context.lookup(
   	        "CreateCRLSession"), ICreateCRLSessionHome.class);
   	        this.createcrlsession = home.create(); 
   	    }catch(Exception e){
   	        throw new EJBException(e);
   	    }   	      
   	    
   }


   public void run() 
   {
       run=true;
       while(run){
           try{       	  	       	   
               sleep(getTimeToNextPoll());       	    
               try{      
                   if(run)	
                       // Create a new CRL if the old one expires within our polltime + an overlap time, so a CRL is always created
                       // at least the configured (in CA configuration) overlap time minutes before the old one expires.
                       // In this way we will in the worst case get the overlap time as the time 
                       // when applications can get the new CRL before the old one expires.
                       this.createcrlsession.createCRLs(administrator, m_pollTime);
                   log.debug("CRLCreateThread: createCRLs");
               }catch(Exception e){
                   log.error("Error generating CRLs: ", e);
               }       	    
           }catch( InterruptedException e){}
       }        
   }

  public void stopThread()
  {
  	this.run = false;
    this.check = 0;  	
  }
  
  
  /**
   * Method calculating the time in milliseconds to the next Poll  
   */  
  private long getTimeToNextPoll(){      
      long nexttime = 0;
      Calendar nextcalendar = Calendar.getInstance();

      switch(check){
           case CHECK_DAILY :
               nextcalendar.add(Calendar.DATE,1);
               nextcalendar.set(Calendar.HOUR_OF_DAY,0);
               nextcalendar.set(Calendar.MINUTE, 0);
               nextcalendar.set(Calendar.SECOND, 0);
               nexttime = nextcalendar.getTimeInMillis();               
               break;
           case CHECK_HOURLY :
               nextcalendar.add(Calendar.HOUR_OF_DAY,1);               
               nextcalendar.set(Calendar.MINUTE, 0);
               nextcalendar.set(Calendar.SECOND, 0);
               nexttime = nextcalendar.getTimeInMillis();
               break;
           case CHECK_30MIN :               
               nextcalendar.add(Calendar.MINUTE,(30 - (nextcalendar.get(Calendar.MINUTE) % 30)));               
               nextcalendar.set(Calendar.SECOND, 0);
               nexttime = nextcalendar.getTimeInMillis();
               break;
           case CHECK_15MIN :
               nextcalendar.add(Calendar.MINUTE,(15 - (nextcalendar.get(Calendar.MINUTE) % 15)));               
               nextcalendar.set(Calendar.SECOND, 0);
               nexttime = nextcalendar.getTimeInMillis(); 
               break;
           case CHECK_1MIN :
               nextcalendar.add(Calendar.MINUTE,1);               
               nextcalendar.set(Calendar.SECOND, 0);
               nexttime = nextcalendar.getTimeInMillis(); 
               break;
           default : 
               log.error("Invalid Polltime set for CRLCreateService! Using 1 minute.");
               nextcalendar.add(Calendar.MINUTE,1);               
               nextcalendar.set(Calendar.SECOND, 0);
               nexttime = nextcalendar.getTimeInMillis(); 
               break;
               
      }            

      return nexttime - (new Date()).getTime();
  }

 
    
}