diff options
Diffstat (limited to 'pki/base/console/src/com/netscape/admin/certsrv/task/CGITask.java')
-rw-r--r-- | pki/base/console/src/com/netscape/admin/certsrv/task/CGITask.java | 400 |
1 files changed, 400 insertions, 0 deletions
diff --git a/pki/base/console/src/com/netscape/admin/certsrv/task/CGITask.java b/pki/base/console/src/com/netscape/admin/certsrv/task/CGITask.java new file mode 100644 index 000000000..654477763 --- /dev/null +++ b/pki/base/console/src/com/netscape/admin/certsrv/task/CGITask.java @@ -0,0 +1,400 @@ +// --- 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.admin.certsrv.task; + +import com.netscape.admin.certsrv.*; +import java.util.*; +import java.io.*; +import java.net.URL; +import javax.swing.JFrame; +import com.netscape.management.client.TaskObject; +import com.netscape.management.client.IPage; +import com.netscape.management.client.console.ConsoleInfo; +import com.netscape.management.client.comm.*; +import com.netscape.management.client.util.*; + +/** + * Netscape Certificate Server 4.0 CGI base task + * + * @author Jack Pan-Chen + * @version $Revision: 14593 $, $Date: 2007-05-01 16:35:45 -0700 (Tue, 01 May 2007) $ + */ +public class CGITask extends CMSTaskObject + implements CommClient +{ + + /*========================================================== + * variables + *==========================================================*/ + private static final String PREFIX = "CGITASK"; + + protected boolean mFinished = false; + protected String mCmd = null; + protected String mAdminURL = null; + protected boolean mSuccess = false; + private String mReply = null; + protected String mSection = ""; + protected String mErrorMsg = ""; + protected String mWarnMsg = ""; + + private boolean mForceBasicAuth = false; + + /*========================================================== + * constructors + *==========================================================*/ + public CGITask() { + super(); + } + + /*========================================================== + * public methods + *==========================================================*/ + + /** + * Force the operation to complete with basic auth, instead + * of the default option, which is to first try a non + * authenticated request, then an authenticated one. + */ + + public void setForceBasicAuth(boolean value) { + mForceBasicAuth = value; + } + + /** + * Send an http request to the server and then popup a dialog if the + * operation is successful. + * + * @param viewInstance The calling page + */ + public boolean run(IPage viewInstance) { + if ( mCmd == null ) { + Debug.println( "Could not get execref for " + getDN() ); + return false; + } + + return run( viewInstance, mCmd ); + } + + /** + * Send an http request to the server. Return true if we're sure it + * succeeded, otherwise false. + * + * @param viewInstance The calling page + * @param cmd Command to execute + */ + boolean run(IPage viewInstance, String cmd) { + + // get the admin URL location first + mAdminURL = _consoleInfo.getAdminURL(); + if ( mAdminURL == null ) { + Debug.println( "Could not get adminURL for " + getDN() ); + return false; + } + + // Allow specifying e.g. "slapd-install" for instance + String instance = (String)_consoleInfo.get( cmd ); + + if ( instance == null ) + instance = (String)_consoleInfo.get( "ServerInstance" ); + String fullCmd = mAdminURL + instance + "/" + cmd; + + HttpManager h = new HttpManager(); + // tell the http manager to use UTF8 encoding + h.setResponseTimeout(60000); + h.setSendUTF8(true); + + try { + mSuccess = false; + mFinished = false; + + // _consoleInfo.get("arguments") is a hashtable of key/value pairs + // to use as the arguments to the CGI + Hashtable args = (Hashtable)_consoleInfo.get("arguments"); + ByteArrayInputStream data = null; + if (args != null && !args.isEmpty()) + data = encode(args); + Debug.println( "Posting " + fullCmd ); + // tell the http manager to notify us immediately of replies + // if we're using async mode + int flags = 0; + + if (mForceBasicAuth) { + flags |= CommManager.FORCE_BASIC_AUTH; + } + + if (data == null) + h.post(new URL(fullCmd), this, null, null, 0, + flags); + else + h.post(new URL(fullCmd), this, null, data, data.available(), + flags); + awaitSuccess(); + Debug.println( "Command executed: " + fullCmd ); + } catch (Exception e) { + if ( e instanceof java.net.ConnectException ) { + CMSAdminUtil.showMessageDialog(mResource, + PREFIX, "SERVERDOWN", CMSAdminUtil.ERROR_MESSAGE); + } + Debug.println( "Command " + fullCmd + " failed: " + e ); + } + return mSuccess; + } + + /** + * waiting for the http transaction to be finished. + */ + public synchronized void awaitSuccess() { + while (!mFinished) { + try {wait();} + catch (Exception e) { } + } + } + + /** + * http transaction finished, notify the process + */ + public synchronized void finish() { + mFinished = true; + notifyAll(); + } + + /** + * the operation is finished after we receive the http stream + */ + public void replyHandler(InputStream response, CommRecord cr) { +/* + try { + int nBytes = response.available(); + if ( nBytes > 0 ) { + // the response from the DS CGIs will typically be in + // UTF8 encoding + byte[] data = new byte[nBytes]; + nBytes = response.read( data ); + mReply = new String( data, 0, nBytes, "UTF8" ); + Debug.println( "CGITask.replyHandler: Response (" + nBytes + + " bytes) = " + mReply ); + int index = 0; + if ((mReply.indexOf("NMC_") != -1) && + ((index = mReply.indexOf(":")) != -1)) { + String sName = mReply.substring(0, index).trim(); + String sValue = mReply.substring(index+1).trim(); + if (sName.equalsIgnoreCase("NMC_Status")) { + int code = Integer.parseInt(sValue); + mSuccess = (code == 0); + } else if (sName.equalsIgnoreCase("NMC_ERRINFO")) { + mErrorMsg = sValue; + } else if (sName.equalsIgnoreCase("NMC_WARNINFO")) { + mWarnMsg = sValue; + } + } + } + } catch ( Exception e ) { + Debug.println( "CGITask.replyHandler: " + e.toString() ); + mSuccess = false; + } + finish(); +*/ + + try { + BufferedReader rspStream = + new BufferedReader(new InputStreamReader(response, "UTF8")); + String rspStr; + + Debug.println("replyHandler() - start"); + while ((rspStr = rspStream.readLine()) != null) + { + Debug.println("replyHandler() - read [" + rspStr + "]"); + // NMC_ messages are parsed, but not shown to the user + /* + if (_statusText != null && !rspStr.startsWith("NMC_")) { + _statusText.append(rspStr + "\n"); + Thread.yield(); // allow graphics repaints + } + */ + Debug.println("Start parsing"); + parse(rspStr); + } + } catch (Exception e) { + Debug.println("ConfigCert.Exception : " + e.toString()); + } + + Debug.println("ConfigCert.replyHandler: finished, mSuccess=" + + mSuccess); + + finish(); + } + + /** + * return the value for the given keyword in the reply + */ + private void parse(String s) { + String sName; + String sValue; + int iIndex; + + Debug.println("Parse input: " + s); + + if ((iIndex=s.indexOf(":")) != (-1)) + { + sName = s.substring(0, iIndex).trim(); + sValue = s.substring(iIndex+1).trim(); + Debug.println("Parse input: name=" + sName + " value=" + sValue); + //mWizardInfo.put(sName, sValue); +/* + if (mCgiResponse == null) + mCgiResponse = new Hashtable(); + mCgiResponse.put(sName, sValue); +*/ + if (sName.equalsIgnoreCase("NMC_Status")) + { + int code = Integer.parseInt(sValue); + mSuccess = (code == 0); + Debug.println("Parse input: code=" + code + " mSuccess=" + + mSuccess); + } else if (sName.equalsIgnoreCase("NMC_ERRINFO")) { + mErrorMsg = sValue; + } else if (sName.equalsIgnoreCase("NMC_WARNINFO")) { + mWarnMsg = sValue; + } + } + + Debug.println("Parse finished"); + } + + public String getErrorMessage() { + return mErrorMsg; + } + + /** + * this function will be called if error occurs + */ + public void errorHandler(Exception error, CommRecord cr) { + Debug.println("CGITask.errorHandler: " + error ); + + mSuccess = false; + finish(); + } + + + public String getDN() { + return _consoleInfo.getCurrentDN(); + } + + public String getReply() { + return mReply; + } + + /** + * Return the command, which should have been stored in the info. + */ + private String getCommand() { + String s = (String)_consoleInfo.get( "execref" ); + if ( s != null ) + return "bin/" + s; + return null; + } + + /** + * pass the username to the admin server + */ + public String username(Object authObject, CommRecord cr) { + Debug.println( "username = " + + (String)_consoleInfo.getAuthenticationDN()); + return _consoleInfo.getAuthenticationDN(); + } + + /** + * pass the user password to the admin server + */ + public String password(Object authObject, CommRecord cr) { + Debug.println( "password = " + + (String)_consoleInfo.get( "AdminUserPassword" ) ); + return (String)_consoleInfo.get( "AdminUserPassword" ); + } + +/* + protected void showDialog( JFrame frame, String msg, String item, + boolean error ) { + // display a message + if ( error ) { + DSUtil.showErrorDialog( frame, msg, item, "dirtask" ); + } else { + DSUtil.showInformationDialog( frame, msg, item, "dirtask" ); + } + } + + protected void showResultDialog( boolean success ) { + // popup a dialog + if ( success ) { + showDialog( new JFrame(), mSection+"-success", "", + false ); + } else { + showDialog( new JFrame(), mSection+"-failed", "", + true ); + } + } + + protected void showResultDialog( int errorCode, String arg ) { + // popup a dialog + String error = "error-" + Integer.toString( errorCode ) + "-msg"; + String title = mSection + "-failed-title"; + DSUtil.showErrorDialog( null, + title, + error, + arg, "dirtask" ); + } + + protected void showResultDialog( CGIThread thread ) { + CGIReportTask task = thread.getTask(); + if ( task.getStatus() != 0 ) { + showResultDialog( task.getStatus(), + (String)task.getResult("NMC_ErrInfo") ); + } else { + showResultDialog( task.getStatus() == 0 ); + } + } + */ + + /** + * Translates a hashtable into <code>x-www-form-urlencoded</code> format. + * Values are converted from Unicode to UTF8 before URL encoding. + * + * @param args <code>Hashtable</code> containing name/value pairs to be translated. + * @return a ByteArrayInputStream to the translated <code>Hashtable</code> contents. + */ + public static ByteArrayInputStream encode(Hashtable args) + { + if ((args == null) || (args.size() == 0)) + return (null); + + String p = ""; + Enumeration e = args.keys(); + + while (e.hasMoreElements()) + { + String name = (String)e.nextElement(); + String value = URLByteEncoder.encodeUTF8(args.get(name).toString()); + Debug.println("********** Encoding name --> "+name+" value --> "+value); + p += URLByteEncoder.encodeUTF8(name) + "=" + + value + (e.hasMoreElements()?"&":""); + } + + return new ByteArrayInputStream(p.getBytes()); + } + +} |