diff options
Diffstat (limited to 'pki/base/common/src/com/netscape/cmscore/connector/Resender.java')
| -rw-r--r-- | pki/base/common/src/com/netscape/cmscore/connector/Resender.java | 249 |
1 files changed, 249 insertions, 0 deletions
diff --git a/pki/base/common/src/com/netscape/cmscore/connector/Resender.java b/pki/base/common/src/com/netscape/cmscore/connector/Resender.java new file mode 100644 index 000000000..78fddaf5c --- /dev/null +++ b/pki/base/common/src/com/netscape/cmscore/connector/Resender.java @@ -0,0 +1,249 @@ +// --- BEGIN COPYRIGHT BLOCK --- +// 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; version 2 of the License. +// +// 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., +// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. +// +// (C) 2007 Red Hat, Inc. +// All rights reserved. +// --- END COPYRIGHT BLOCK --- +package com.netscape.cmscore.connector; + + +import com.netscape.certsrv.base.*; +import com.netscape.certsrv.apps.*; +import com.netscape.certsrv.connector.*; +import com.netscape.certsrv.authority.*; +import com.netscape.certsrv.request.*; +import com.netscape.certsrv.authentication.*; +import com.netscape.certsrv.logging.*; +import com.netscape.cmscore.util.Debug; +import com.netscape.cmsutil.http.*; + +import java.util.Vector; +import java.util.Enumeration; +import java.io.*; + + +/** + * Resend requests at intervals to the server to check if it's been completed. + * Default interval is 5 minutes. + */ +public class Resender implements IResender { + public static final int SECOND = 1000; //milliseconds + public static final int MINUTE = 60 * SECOND; + public static final int HOUR = 60 * MINUTE; + public static final int DAY = 24 * HOUR; + + protected IAuthority mAuthority = null; + IRequestQueue mQueue = null; + protected IRemoteAuthority mDest = null; + + /* Vector of Request Id *Strings* */ + protected Vector mRequestIds = new Vector(); + + protected HttpConnection mConn = null; + + protected String mNickName = null; + + // default interval. + // XXX todo add another interval for requests unsent because server + // was down (versus being serviced in request queue) + protected int mInterval = 1 * MINUTE; + + public Resender(IAuthority authority, String nickName, IRemoteAuthority dest) { + mAuthority = authority; + mQueue = mAuthority.getRequestQueue(); + mDest = dest; + mNickName = nickName; + + //mConn = new HttpConnection(dest, + // new JssSSLSocketFactory(nickName)); + } + + public Resender( + IAuthority authority, String nickName, + IRemoteAuthority dest, int interval) { + mAuthority = authority; + mQueue = mAuthority.getRequestQueue(); + mDest = dest; + if (interval > 0) + mInterval = interval * SECOND; // interval specified in seconds. + + //mConn = new HttpConnection(dest, + // new JssSSLSocketFactory(nickName)); + } + + // must be done after a subsystem 'start' so queue is initialized. + private void initRequests() { + mQueue = mAuthority.getRequestQueue(); + // get all requests in mAuthority that are still pending. + IRequestList list = + mQueue.listRequestsByStatus(RequestStatus.SVC_PENDING); + + while (list != null && list.hasMoreElements()) { + RequestId rid = list.nextRequestId(); + + CMS.debug( + "added request Id " + rid + " in init to resend queue."); + // note these are added as strings + mRequestIds.addElement(rid.toString()); + } + } + + public void addRequest(IRequest r) { + synchronized (mRequestIds) { + // note the request ids are added as strings. + mRequestIds.addElement(r.getRequestId().toString()); + } + CMS.debug( + "added " + r.getRequestId() + " to resend queue"); + } + + public void run() { + + CMS.debug("Resender: In resender Thread run:"); + mConn = new HttpConnection(mDest, + new JssSSLSocketFactory(mNickName)); + initRequests(); + + do { + resend(); + try { + Thread.sleep(mInterval); + } catch (InterruptedException e) { + mAuthority.log(ILogger.LL_INFO, CMS.getLogMessage("CMSCORE_CONNECTOR_RESENDER_INTERRUPTED")); + continue; + } + } + while (true); + } + + private void resend() { + // clone a seperate list so mRequestIds can be modified + Vector rids = (Vector) mRequestIds.clone(); + Vector completedRids = new Vector(); + + // resend each request to CA to ping for status. + Enumeration enum1 = rids.elements(); + + while (enum1.hasMoreElements()) { + // request ids are added as strings. + String ridString = (String) enum1.nextElement(); + RequestId rid = new RequestId(ridString); + IRequest r = null; + + CMS.debug( + "resend processing request id " + rid); + + try { + r = mQueue.findRequest(rid); + } catch (EBaseException e) { + // XXX bad case. should we remove the rid now ? + mAuthority.log(ILogger.LL_WARN, CMS.getLogMessage("CMSCORE_CONNECTOR_REQUEST_NOT_FOUND", rid.toString())); + continue; + } + try { + if (r.getRequestStatus() != RequestStatus.SVC_PENDING) { + // request not pending anymore - aborted or cancelled. + completedRids.addElement(rid); + CMS.debug( + "request id " + rid + " no longer service pending"); + } else { + boolean completed = send(r); + + if (completed) { + completedRids.addElement(rid); + mAuthority.log(ILogger.LL_INFO, CMS.getLogMessage("CMSCORE_CONNECTOR_REQUEST_COMPLETED", rid.toString())); + } + } + } catch (IOException e) { + mAuthority.log(ILogger.LL_WARN, CMS.getLogMessage("CMSCORE_CONNECTOR_REQUEST_ERROR", rid.toString(), e.toString())); + } catch (EBaseException e) { + // if connection is down, don't send the remaining request + // as it will sure fail. + mAuthority.log(ILogger.LL_WARN, CMS.getLogMessage("CMSCORE_CONNECTOR_DOWN")); + if (e.toString().indexOf("connection not available") + >= 0) + break; + } + } + + // remove completed ones from list so they won't be resent. + Enumeration en = completedRids.elements(); + + synchronized (mRequestIds) { + while (en.hasMoreElements()) { + RequestId id = (RequestId) en.nextElement(); + + CMS.debug( + "Connector: Removed request " + id + " from re-send queue"); + mRequestIds.removeElement(id.toString()); + CMS.debug( + "Connector: mRequestIds now has " + + mRequestIds.size() + " elements."); + } + } + } + + // this is almost the same as connector's send. + private boolean send(IRequest r) + throws IOException, EBaseException { + IRequest reply = null; + + try { + HttpPKIMessage tomsg = new HttpPKIMessage(); + HttpPKIMessage replymsg = null; + + tomsg.fromRequest(r); + replymsg = (HttpPKIMessage) mConn.send(tomsg); + if(replymsg==null) + return false; + CMS.debug( + r.getRequestId() + " resent to CA"); + + RequestStatus replyStatus = + RequestStatus.fromString(replymsg.reqStatus); + int index = replymsg.reqId.lastIndexOf(':'); + RequestId replyRequestId = + new RequestId(replymsg.reqId.substring(index + 1)); + + if (Debug.ON) + Debug.trace("reply request id " + replyRequestId + + " for request " + r.getRequestId()); + + if (replyStatus != RequestStatus.COMPLETE) { + CMS.debug("resend " + + r.getRequestId() + " still not completed."); + return false; + } + + // request was completed. copy relevant contents. + replymsg.toRequest(r); + if (Debug.ON) + Debug.trace("resend request id was completed " + r.getRequestId()); + mQueue.markAsServiced(r); + mQueue.releaseRequest(r); + CMS.debug( + "resend released request " + r.getRequestId()); + return true; + } catch (EBaseException e) { + // same as not having sent it, so still want to resend. + mAuthority.log(ILogger.LL_FAILURE, CMS.getLogMessage("CMSCORE_CONNECTOR_RESEND_ERROR", r.getRequestId().toString(), e.toString())); + if (e.toString().indexOf("Connection refused by peer") > 0) + throw new EBaseException("connection not available"); + } + return false; + + } + +} + |