localejbcahardtokenbatchjobsessionbean.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.hardtoken;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;

import javax.ejb.CreateException;
import javax.ejb.EJBException;
import javax.naming.NamingException;
import javax.sql.DataSource;

import se.anatom.ejbca.BaseSessionBean;
import se.anatom.ejbca.SecConst;
import se.anatom.ejbca.log.Admin;
import se.anatom.ejbca.log.ILogSessionLocal;
import se.anatom.ejbca.log.ILogSessionLocalHome;
import se.anatom.ejbca.log.LogEntry;
import se.anatom.ejbca.ra.UserAdminData;
import se.anatom.ejbca.ra.UserDataLocal;
import se.anatom.ejbca.ra.UserDataLocalHome;

/**
 * Remote interface for bean used by hardtoken batchprograms to retrieve users to generate from EJBCA RA.
 *
 * @version $Id: LocalEjbcaHardTokenBatchJobSessionBean.java,v 1.0 2003/01/23 06:26:14 14:01:39 herrvendil Exp $
 */
public class LocalEjbcaHardTokenBatchJobSessionBean extends BaseSessionBean  {

    /** Columns in the database used in select */
    private final String USERDATA_COL = "username, subjectDN, subjectAltName, subjectEmail, status, type, clearpassword, timeCreated, timeModified, endEntityprofileId, certificateProfileId, tokenType, hardTokenIssuerId, cAId";

    /** Var holding JNDI name of datasource */
    private String dataSource = "";

    /** The home interface of  User Admin entity bean */
    private UserDataLocalHome useradminsession = null;

    /** The local interface of  hard token session bean */
    private IHardTokenSessionLocal hardtokensession = null;

    /** The remote interface of  log session bean */
    private ILogSessionLocal logsession = null;
    

    
    /**
     * Default create for SessionBean without any creation Arguments.
     * @throws CreateException if bean instance can't be created
     */

    public void ejbCreate() throws CreateException {
        debug(">ejbCreate()");
      try{
        dataSource = (String)lookup("java:comp/env/DataSource", java.lang.String.class);
        debug("DataSource=" + dataSource);

        useradminsession = (UserDataLocalHome) lookup("java:comp/env/ejb/UserDataLocal", UserDataLocalHome.class);

        debug("<ejbCreate()");
      }catch(Exception e){
         throw new EJBException(e);
      }

    }


    /** Gets connection to Datasource used for manual SQL searches
     * @return Connection
     */
    private Connection getConnection() throws SQLException, NamingException {
        DataSource ds = (DataSource)getInitialContext().lookup(dataSource);
        return ds.getConnection();
    } //getConnection

    /** Gets connection to hard token session bean
     * @return IHardTokenSessionLocal
     */
    private IHardTokenSessionLocal getHardTokenSession() {
        if(hardtokensession == null){
          try{
            IHardTokenSessionLocalHome hardtokensessionhome = (IHardTokenSessionLocalHome) lookup("java:comp/env/ejb/HardTokenSessionLocal",IHardTokenSessionLocalHome.class);
            hardtokensession = hardtokensessionhome.create();
          }catch(Exception e){
             throw new EJBException(e);
          }
        }
        return hardtokensession;
    } //getHardTokenSession

    /** Gets connection to log session bean
     * @return Connection
     */
    private ILogSessionLocal getLogSession() {
        if(logsession == null){
          try{
            ILogSessionLocalHome logsessionhome = (ILogSessionLocalHome) lookup("java:comp/env/ejb/LogSessionLocal",ILogSessionLocalHome.class);
            logsession = logsessionhome.create();
          }catch(Exception e){
             throw new EJBException(e);
          }
        }
        return logsession;
    } //getLogSession



    /**
     * Returns the next user scheduled for batch generation for the given issuer.
     *
     * @param admin the administrator performing the actions
     * @param issuercert the certificate of the hard token issuer.
     *
     * @return The next user to generate or NULL if there are no users i queue.
     * @throws EJBException if a communication or other error occurs.
     */


    public UserAdminData getNextHardTokenToGenerate(Admin admin, String alias) throws UnavailableTokenException{
      debug(">getNextHardTokenToGenerate()");
      debug("alias " + alias);
      UserAdminData returnval=null;
      int issuerid = getHardTokenSession().getHardTokenIssuerId(admin, alias);
  
      debug("issuerid " + issuerid);      
      
      if(issuerid != IHardTokenSessionLocal.NO_ISSUER){
        Connection con = null;
        ResultSet rs = null;
        PreparedStatement ps = null;
        
        try{
           // Construct SQL query.
        	debug("HERE");
            con = getConnection();            
            ps = con.prepareStatement("select " + USERDATA_COL + " from UserData where hardTokenIssuerId=? and tokenType>? and (status=? or status=?)" );
            ps.setInt(1,issuerid);
            ps.setInt(2,SecConst.TOKEN_SOFT);
            ps.setInt(3,UserDataLocal.STATUS_NEW);
            ps.setInt(4,UserDataLocal.STATUS_KEYRECOVERY);
            
            // Execute query.
            rs = ps.executeQuery();      
                                                
            // Assemble result.
            
           if(rs.next()){
              returnval = new UserAdminData(rs.getString(1), rs.getString(2), rs.getInt(14), rs.getString(3), rs.getString(4), rs.getInt(5), rs.getInt(6)
                                               , rs.getInt(10), rs.getInt(11)
                                               , new java.util.Date(rs.getLong(8)), new java.util.Date(rs.getLong(9))
                                               ,  rs.getInt(12), rs.getInt(13));
              returnval.setPassword(rs.getString(7));
              debug("found user" + returnval.getUsername());
            }
            if(returnval !=null){
              getHardTokenSession().getIsHardTokenProfileAvailableToIssuer(admin, issuerid, returnval);
              getLogSession().log(admin, returnval.getCAId(), LogEntry.MODULE_HARDTOKEN, new java.util.Date(),returnval.getUsername(), null, LogEntry.EVENT_INFO_HARDTOKEN_USERDATASENT,"Userdata sent for token generation to issuer with alias :" + alias);
            }
        }catch(Exception e){
          getLogSession().log(admin, admin.getCAId(), LogEntry.MODULE_HARDTOKEN, new java.util.Date(),null, null, LogEntry.EVENT_ERROR_HARDTOKEN_USERDATASENT,"Error when retrieving next token for issuer with alias: " + alias);
          throw new EJBException(e);
        } finally {
           try{
             if(rs != null) rs.close();
             if(ps != null) ps.close();
             if(con!= null) con.close();
           }catch(SQLException se){
               error("Fel vid upprensning: ", se);
           }
        }
      }
      
      debug("<getNextHardTokenToGenerate()");
      return returnval;
    }// getNextHardTokenToGenerate

    /**
     * Returns a Collection of users scheduled for batch generation for the given issuer.
     * A maximum of MAX_RETURNED_QUEUE_SIZE users will be returned by call.
     *
     * @param admin the administrator performing the actions
     * @param issuercert the certificate of the hard token issuer.
     *
     * @return A Collection of users to generate or NULL if there are no users i queue.
     * @throws EJBException if a communication or other error occurs.
     */

    public Collection getNextHardTokensToGenerate(Admin admin, String alias) throws UnavailableTokenException{
      debug(">getNextHardTokensToGenerate()");