/*
*
* Copyright (c) 2001-2003 The Apache Software Foundation. All rights
* reserved.
*
* "This product includes software developed by the
* Apache Software Foundation (http://www.apache.org/)."
*/
/*
Copyright 2006 OCLC Online Computer Library Center, Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package ORG.oclc.os.SRW;
import de.fuberlin.wiwiss.pubby.negotiation.MediaRangeSpec;
import javax.servlet.ServletConfig;
import javax.xml.transform.TransformerConfigurationException;
import org.apache.axis.transport.http.AxisHttpSession;
import org.apache.axis.transport.http.AxisServlet;
import org.apache.axis.transport.http.HTTPConstants;
import org.apache.axis.transport.http.HTTPTransport;
import org.apache.axis.transport.http.ServletEndpointContextImpl;
import org.apache.axis.AxisEngine;
import org.apache.axis.AxisFault;
import org.apache.axis.ConfigurationException;
import org.apache.axis.Constants;
import org.apache.axis.Message;
import org.apache.axis.MessageContext;
import org.apache.axis.components.logger.LogFactory;
import org.apache.axis.description.OperationDesc;
import org.apache.axis.description.ServiceDesc;
import org.apache.axis.handlers.soap.SOAPService;
import org.apache.axis.security.servlet.ServletSecurityProvider;
import org.apache.axis.utils.Admin;
import org.apache.axis.utils.JavaUtils;
import org.apache.axis.utils.Messages;
import org.apache.axis.utils.XMLUtils;
import org.apache.commons.logging.Log;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
//import javax.servlet.http.HttpUtils;
import javax.xml.soap.SOAPException;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.io.StringReader;
import java.io.StringWriter;
import java.net.ConnectException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.StringTokenizer;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;
import org.apache.axis.SOAPPart;
/**
*
* @author Doug Davis (dug@us.ibm.com)
* @author Steve Loughran
* xdoclet tags are not active yet; keep web.xml in sync
* web.servlet name="AxisServlet" display-name="Apache-Axis Servlet"
* web.servlet-mapping url-pattern="/servlet/AxisServlet"
* web.servlet-mapping url-pattern="*.jws"
* web.servlet-mapping url-pattern="/services/\*"
*/
public class SRWServlet extends AxisServlet {
protected static Log servletLog=LogFactory.getLog(SRWServlet.class.getName());
/**
* this log is for timing
*/
private static Log tlog =
LogFactory.getLog(Constants.TIME_LOG_CATEGORY);
/**
* a separate log for exceptions lets users route them
* differently from general low level debug info
*/
private static Log exceptionLog =
LogFactory.getLog(Constants.EXCEPTION_LOG_CATEGORY);
// public static final String INIT_PROPERTY_TRANSPORT_NAME =
// "transport.name";
//
// public static final String INIT_PROPERTY_USE_SECURITY =
// "use-servlet-security";
// public static final String INIT_PROPERTY_ENABLE_LIST =
// "axis.enableListQuery";
//
// public static final String INIT_PROPERTY_JWS_CLASS_DIR =
// "axis.jws.servletClassDir";
// These have default values.
private String portNumber, transportName;
private ServletSecurityProvider securityProvider = null;
/**
* cache of logging debug option; only evaluated at init time.
* So no dynamic switching of logging options with this servlet.
*/
private static boolean isDebug = false;
/**
* Should we enable the "?list" functionality on GETs? (off by
* default because deployment information is a potential security
* hole)
*/
private boolean enableList = false;
private HashMap<String, Transformer> transformers=new HashMap<String, Transformer>();
/**
* Cached path to JWS output directory
*/
private String jwsClassDir = null;
private boolean negotiatedContent;
// private boolean APP;
@Override
protected String getJWSClassDir() { return jwsClassDir; }
protected SRWServletInfo srwInfo=null;
/**
* create a new servlet instance
*/
public SRWServlet() {
if(servletLog instanceof org.apache.commons.logging.impl.SimpleLog)
((org.apache.commons.logging.impl.SimpleLog)servletLog).setLevel(3); // info
}
/**
* Initialization method.
*/
@Override
public void init() throws ServletException {
srwInfo=new SRWServletInfo();
ServletConfig config = getServletConfig();
srwInfo.init(config);
portNumber=config.getInitParameter("portNumber");
if(servletLog.isDebugEnabled())
servletLog.debug("portNumber="+portNumber);
if(portNumber==null)
portNumber=":80";
else
if(!portNumber.startsWith(":"))
portNumber=":"+portNumber;
super.init();
isDebug= servletLog.isDebugEnabled();
if(isDebug) servletLog.debug("In servlet init");
ServletContext context = config.getServletContext();
transportName = getOption(context,
INIT_PROPERTY_TRANSPORT_NAME,
HTTPTransport.DEFAULT_TRANSPORT_NAME);
if (JavaUtils.isTrueExplicitly(getOption(context, INIT_PROPERTY_USE_SECURITY, null))) {
securityProvider = new ServletSecurityProvider();
}
enableList =
JavaUtils.isTrueExplicitly(getOption(context, INIT_PROPERTY_ENABLE_LIST, null));
jwsClassDir = getOption(context, INIT_PROPERTY_JWS_CLASS_DIR, null);
/**
* There are DEFINATE problems here if
* getHomeDir and/or getDefaultJWSClassDir return null
* (as they could with WebLogic).
* This needs to be reexamined in the future, but this
* should fix any NPE's in the mean time.
*/
if (jwsClassDir != null) {
if (getHomeDir() != null) {
jwsClassDir = getHomeDir() + jwsClassDir;
}
} else {
jwsClassDir = getDefaultJWSClassDir();
}
}
@Override
protected void doDelete(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
MessageContext msgContext = createMessageContext(getEngine(), request, response);
if(!srwInfo.setSRWStuff(request, response, msgContext)) {
servletLog.error("srwInfo.setSRWStuff failed!");
return;
}
String uri=request.getRequestURI();
// APP=true;
if(servletLog.isDebugEnabled()) servletLog.debug("in doDelete: got APP request for "+uri);
SRWDatabase db=(SRWDatabase)msgContext.getProperty("db");
if(servletLog.isDebugEnabled()) servletLog.debug("deleting from database "+db);
if(servletLog.isDebugEnabled()) servletLog.debug("deleting from database "+db.dbname);
try {
if(db.delete(db.extractRecordIdFromUri(uri)))
response.setStatus(200);
else
response.setStatus(404);
}
catch(UnsupportedOperationException e) {
servletLog.error("delete not supported by database "+db, e);
response.setStatus(404);
}
finally {
SRWDatabase.putDb((String)msgContext.getProperty("dbname"), db);
}
}
@Override
protected void doPut(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
MessageContext msgContext = createMessageContext(getEngine(), request, response);
if(!srwInfo.setSRWStuff(request, response, msgContext)) {
servletLog.error("srwInfo.setSRWStuff failed!");
return;
}
// APP=true;
if(servletLog.isDebugEnabled()) servletLog.debug("in doPut: got APP request for "+request.getRequestURI());
SRWDatabase db=(SRWDatabase)msgContext.getProperty("db");
if(servletLog.isDebugEnabled()) servletLog.debug("adding to database "+db);
if(servletLog.isDebugEnabled()) servletLog.debug("adding to database "+db.dbname);
InputStream is=request.getInputStream();
int len, offset=0, totlen=request.getContentLength();
byte[] b=new byte[totlen];
while(totlen>0) {
len=is.read(b, offset, totlen);
totlen-=len;
offset+=len;
}
String uri=request.getRequestURI();
db.replace(db.extractRecordIdFromUri(uri), b);
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setPutResponseHeaders(new String(b, "UTF-8"), request, response);
response.setStatus(200);
SRWDatabase.putDb((String)msgContext.getProperty("dbname"), db);
return;
}
/**
* Process GET requests. This includes handoff of pseudo-SOAP requests
*
* @param request request in
* @param response request out
* @throws ServletException
* @throws IOException
*/
@Override
public void doGet(HttpServletRequest request, HttpServletResponse response)
throws ServletException, IOException {
servletLog.debug("Enter: doGet()");
try {
AxisEngine engine = getEngine();
ServletContext servletContext =
getServletConfig().getServletContext();
String pathInfo = request.getPathInfo();
String realpath = servletContext.getRealPath(request.getServletPath());
if (realpath == null) {
realpath = request.getServletPath();
}
boolean wsdlRequested = false;
boolean listRequested = false;
boolean hasParameters = request.getParameterNames().hasMoreElements();
//JWS pages are special; they are the servlet path and there
//is no pathinfo...we map the pathinfo to the servlet path to keep
//it happy
boolean isJWSPage = request.getRequestURI().endsWith(".jws");
if(isJWSPage) {
pathInfo= request.getServletPath();
}
// check first if we are doing WSDL or a list operation
String queryString = request.getQueryString();
boolean doExplain=false;
if (queryString != null) {
if (queryString.equalsIgnoreCase("wsdl")) {
wsdlRequested = true;
} else if (queryString.equalsIgnoreCase("list")) {
listRequested = true;
}
else {
String operation=request.getParameter("operation");
if(operation!=null) {
if(operation.equals("explain"))
doExplain=true;
}
else if(request.getParameter("query")==null &&
request.getParameter("scanClause")==null)
doExplain=true;
}
}
else
doExplain=true;
boolean hasNoPath = (pathInfo == null || pathInfo.equals(""));
if (!wsdlRequested && !listRequested && hasNoPath && srwInfo.defaultDatabase==null) {
// If the user requested the servlet (i.e. /axis/servlet/AxisServlet)
// with no service name, present the user with a list of deployed
// services to be helpful
// Don't do this if we are doing WSDL or list.
reportAvailableServices(response, request);
} else if (realpath != null || doExplain) {
// We have a pathname, so now we perform WSDL or list operations
// get message context w/ various properties set
MessageContext msgContext = createMessageContext(engine, request, response);
if(doExplain) {
srwInfo.handleExplain(request, response, msgContext);
return;
}
// NOTE: HttpUtils.getRequestURL has been deprecated.
// This line SHOULD be:
// String url = req.getRequestURL().toString()
// HOWEVER!!!! DON'T REPLACE IT! There's a bug in
// req.getRequestURL that is not in HttpUtils.getRequestURL
// req.getRequestURL returns "localhost" in the remote
// scenario rather than the actual host name.
//
// ? Still true? For which JVM's?
//String url = HttpUtils.getRequestURL(request).toString();
String url = request.getRequestURL().toString();
msgContext.setProperty(MessageContext.TRANS_URL, url);
if (wsdlRequested) {
// Do WSDL generation
msgContext.setTargetService("SRW");
processWsdlRequest(msgContext, response);
} else if (listRequested) {
// Do list, if it is enabled
processListRequest(response);
} else if (hasParameters) {
// If we have ?method=x¶m=y in the URL, make a stab
// at invoking the method with the parameters specified
// in the URL
processMethodRequest(msgContext, request, response);
SRWDatabase.putDb((String)msgContext.getProperty("dbname"),
(SRWDatabase) msgContext.getProperty("db"));
} else {
// See if we can locate the desired service. If we
// can't, return a 404 Not Found. Otherwise, just
// print the placeholder message.
String serviceName;
if (pathInfo.startsWith("/")) {
serviceName = pathInfo.substring(1);
} else {
serviceName = pathInfo;
}
SOAPService s = engine.getService(serviceName);
if (s == null) {
//no service: report it
if(isJWSPage) {
reportCantGetJWSService(request, response);
} else {
reportCantGetAxisService(request, response);
}
} else {
//print a snippet of service info.
reportServiceInfo(response, s, serviceName);
}
}
} else {
// We didn't have a real path in the request, so just
// print a message informing the user that they reached
// the servlet.
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
writer.println( "<html><h1>Axis HTTP Servlet</h1>" );
writer.println( Messages.getMessage("reachedServlet00"));
writer.println("<p>" +
Messages.getMessage("transportName00",
"<b>" + transportName + "</b>"));
writer.println("</html>");
writer.close();
}
} catch (AxisFault fault) {
servletLog.error(fault, fault);
reportTroubleInGet(fault, response);
} catch (Exception e) {
servletLog.error(e, e);
reportTroubleInGet(e, response);
} finally {
//writer.close();
servletLog.debug("Exit: doGet()");
}
}
/**
* when we get an exception or an axis fault in a GET, we handle
* it almost identically: we go 'something went wrong', set the response
* code to 500 and then dump info. But we dump different info for an axis fault
* or subclass thereof.
* @param exception what went wrong
* @param response current response
*/
private void reportTroubleInGet(Exception exception, HttpServletResponse response) throws IOException {
response.setContentType("text/html");
servletLog.error(exception, exception);
PrintWriter writer=response.getWriter();
response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
writer.println("<h2>" +
Messages.getMessage("error00") +
"</h2>");
writer.println("<p>" +
Messages.getMessage("somethingWrong00") +
"</p>");
if(exception instanceof AxisFault) {
AxisFault fault=(AxisFault)exception;
processAxisFault(fault);
writeFault(writer, fault);
} else {
logException(exception);
writer.println("<pre>Exception - " + exception + "<br>");
//dev systems only give fault dumps
if (isDevelopment()) {
writer.println(JavaUtils.stackToString(exception));
}
writer.println("</pre>");
}
writer.close();
}
/**
* routine called whenever an axis fault is caught; where they
* are logged and any other business. The method may modify the fault
* in the process
* @param fault what went wrong.
*/
@Override
protected void processAxisFault(AxisFault fault) {
//log the fault
Element runtimeException = fault.lookupFaultDetail(
Constants.QNAME_FAULTDETAIL_RUNTIMEEXCEPTION);
if (runtimeException != null) {
exceptionLog.info(Messages.getMessage("axisFault00"), fault);
//strip runtime details
fault.removeFaultDetail(Constants.QNAME_FAULTDETAIL_RUNTIMEEXCEPTION);
} else if (exceptionLog.isDebugEnabled()) {
exceptionLog.debug(Messages.getMessage("axisFault00"), fault);
}
//dev systems only give fault dumps
if (!isDevelopment()) {
//strip out the stack trace
fault.removeFaultDetail(Constants.QNAME_FAULTDETAIL_STACKTRACE);
}
}
/**
* log any exception to our output log, at our chosen level
* @param e what went wrong
*/
protected void logException(Exception e) {
exceptionLog.info(Messages.getMessage("exception00"), e);
}
/**
* this method writes a fault out to an HTML stream. This includes
* escaping the strings to defend against cross-site scripting attacks
* @param writer
* @param axisFault
*/
private void writeFault(PrintWriter writer, AxisFault axisFault) {
String localizedMessage = XMLUtils.xmlEncodeString(axisFault.getLocalizedMessage());
writer.println("<pre>Fault - " + localizedMessage + "<br>");
writer.println(axisFault.dumpToString());
writer.println("</pre>");
}
/**
* handle a ?wsdl request
* @param msgContext message context so far
* @param response response to write to
* @throws AxisFault when anything other than a Server.NoService fault is reported
* during WSDL generation
*/
protected void processWsdlRequest(MessageContext msgContext,
HttpServletResponse response) throws AxisFault, IOException {
AxisEngine engine = getEngine();
PrintWriter writer;
try {
engine.generateWSDL(msgContext);
Document doc = (Document) msgContext.getProperty("WSDL");
if (doc != null) {
response.setContentType("text/xml");
writer=response.getWriter();
XMLUtils.DocumentToWriter(doc, writer);
writer.close();
} else {
if (servletLog.isDebugEnabled()) {
servletLog.debug("processWsdlRequest: failed to create WSDL");
}
reportNoWSDL(response, "noWSDL02", null);
}
} catch (AxisFault axisFault) {
//the no-service fault is mapped to a no-wsdl error
if(axisFault.getFaultCode() .equals(Constants.QNAME_NO_SERVICE_FAULT_CODE)) {
//which we log
processAxisFault(axisFault);
//then report under a 404 error
response.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
reportNoWSDL(response, "noWSDL01", axisFault);
} else {
//all other faults get thrown
throw axisFault;
}
}
}
/**
* invoke an endpoint from a get request by building an XML request and
* handing it down. If anything goes wrong, we generate an XML formatted
* axis fault
* @param msgContext current message
* @param response to return data
* @param method method to invoke (may be null)
* @param args argument list in XML form
* @throws AxisFault iff something goes wrong when turning the response message
* into a SOAP string.
*/
protected void invokeEndpointFromGet(MessageContext msgContext,
HttpServletResponse response,
String method,
String args) throws AxisFault, IOException {
String body =
"<" + method + ">" + args + "</" + method + ">";
String msgtxt =
"<SOAP-ENV:Envelope" +
" xmlns:SOAP-ENV=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
"<SOAP-ENV:Body>" + body + "</SOAP-ENV:Body>" +
"</SOAP-ENV:Envelope>";
Message responseMsg=null;
try {
ByteArrayInputStream istream =
new ByteArrayInputStream(msgtxt.getBytes("ISO-8859-1"));
AxisEngine engine = getEngine();
Message msg = new Message(istream, false);
msgContext.setRequestMessage(msg);
engine.invoke(msgContext);
responseMsg = msgContext.getResponseMessage();
//turn off caching for GET requests
response.setHeader("Cache-Control", "no-cache");
response.setHeader("Pragma", "no-cache");
if (responseMsg == null) {
//tell everyone that something is wrong
throw new Exception(Messages.getMessage("noResponse01"));
}
} catch (AxisFault fault) {
processAxisFault(fault);
configureResponseFromAxisFault(response, fault);
if (responseMsg == null) {
responseMsg = new Message(fault);
}
} catch (Exception e) {
response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
responseMsg = convertExceptionToAxisFault(e,responseMsg);
}
//this call could throw an AxisFault. We delegate it up, because
//if we cant write the message there is not a lot we can do in pure SOAP terms.
response.setContentType("text/xml");
PrintWriter writer=response.getWriter();
writer.println(responseMsg.getSOAPPartAsString());
writer.close();
}
/**
* print a snippet of service info.
* @param response
* @param service
* @param serviceName where to put stuff
*/
protected void reportServiceInfo(HttpServletResponse response, SOAPService service, String serviceName) throws IOException {
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
writer.println("<h1>"
+ service.getName()
+"</h1>");
writer.println(
"<p>" +
Messages.getMessage("axisService00") +
"</p>");
writer.println(
"<i>" +
Messages.getMessage("perhaps00") +
"</i>");
writer.close();
}
/**
* respond to the ?list command.
* if enableList is set, we list the engine config. If it isnt, then an
* error is written out
* @param response
* @throws AxisFault, IOException
*/
protected void processListRequest(HttpServletResponse response) throws AxisFault, IOException {
AxisEngine engine = getEngine();
if (enableList) {
Document doc = Admin.listConfig(engine);
if (doc != null) {
response.setContentType("text/xml");
PrintWriter writer=response.getWriter();
XMLUtils.DocumentToWriter(doc, writer);
writer.close();
} else {
//error code is 404
response.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
writer.println("<h2>" +
Messages.getMessage("error00") +
"</h2>");
writer.println("<p>" +
Messages.getMessage("noDeploy00") +
"</p>");
writer.close();
}
} else {
// list not enable, return error
//error code is, what, 401
response.setStatus(HttpURLConnection.HTTP_FORBIDDEN);
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
writer.println("<h2>" +
Messages.getMessage("error00") +
"</h2>");
writer.println("<p><i>?list</i> " +
Messages.getMessage("disabled00") +
"</p>");
writer.close();
}
}
/**
* report that we have no WSDL
* @param res
* @param moreDetailCode optional name of a message to provide more detail
* @param axisFault optional fault string, for extra info at debug time only
*/
protected void reportNoWSDL(HttpServletResponse res,
String moreDetailCode, AxisFault axisFault) throws IOException {
res.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
res.setContentType("text/html");
PrintWriter writer=res.getWriter();
writer.println("<h2>" +
Messages.getMessage("error00") +
"</h2>");
writer.println("<p>" +
Messages.getMessage("noWSDL00") +
"</p>");
if(moreDetailCode!=null) {
writer.println("<p>"
+Messages.getMessage(moreDetailCode)
+"</p>");
}
if(axisFault!=null && isDevelopment()) {
//dev systems only give fault dumps
writeFault(writer, axisFault);
}
writer.close();
}
/**
* This method lists the available services; it is called when there is
* nothing to execute on a GET
* @param response
* @param request
* @throws ConfigurationException
* @throws AxisFault
*/
protected void reportAvailableServices(HttpServletResponse response,
HttpServletRequest request)
throws ConfigurationException, AxisFault, IOException {
AxisEngine engine = getEngine();
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
writer.println("<h2>And now... Some Services</h2>");
Iterator i;
try {
i = engine.getConfig().getDeployedServices();
} catch (ConfigurationException configException) {
//turn any internal configuration exceptions back into axis faults
//if that is what they are
if(configException.getContainedException() instanceof AxisFault) {
throw (AxisFault) configException.getContainedException();
} else {
throw configException;
}
}
String baseURL = getWebappBase(request)+"/services/";
writer.println("<ul>");
while (i.hasNext()) {
ServiceDesc sd = (ServiceDesc)i.next();
StringBuilder sb = new StringBuilder();
sb.append("<li>");
String name = sd.getName();
sb.append(name);
sb.append(" <a href=\"");
sb.append(baseURL);
sb.append(name);
sb.append("?wsdl\"><i>(wsdl)</i></a></li>");
writer.println(sb.toString());
ArrayList operations = sd.getOperations();
if (!operations.isEmpty()) {
writer.println("<ul>");
for (Iterator it = operations.iterator(); it.hasNext();) {
OperationDesc desc = (OperationDesc) it.next();
writer.println("<li>" + desc.getName());
}
writer.println("</ul>");
}
}
writer.println("</ul>");
writer.close();
}
/**
* generate the error response to indicate that there is apparently no endpoint there
* @param request the request that didnt have an edpoint
* @param response response we are generating
*/
protected void reportCantGetAxisService(HttpServletRequest request, HttpServletResponse response) throws IOException {
// no such service....
response.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
writer.println("<h2>" +
Messages.getMessage("error00") + "</h2>");
writer.println("<p>" +
Messages.getMessage("noService06") +
"</p>");
writer.close();
}
/**
* probe for a JWS page and report 'no service' if one is not found there
* @param request the request that didnt have an edpoint
* @param response response we are generating
*/
protected void reportCantGetJWSService(HttpServletRequest request, HttpServletResponse response) throws IOException {
//first look to see if there is a service
String realpath =
getServletConfig().getServletContext()
.getRealPath(request.getServletPath());
boolean foundJWSFile=(new File(realpath).exists()) &&
(realpath.endsWith(Constants.JWS_DEFAULT_FILE_EXTENSION));
response.setContentType("text/html");
PrintWriter writer=response.getWriter();
if(foundJWSFile) {
response.setStatus(HttpURLConnection.HTTP_OK);
writer.println(Messages.getMessage("foundJWS00") + "<p>");
String url = request.getRequestURI();
String urltext = Messages.getMessage("foundJWS01");
writer.println("<a href='"+url+"?wsdl'>"+urltext+"</a>");
} else {
response.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
writer.println(Messages.getMessage("noService06") );
}
writer.close();
}
/**
* Process a POST to the servlet by handing it off to the Axis Engine.
* Here is where SOAP messages are received
* @param req posted request
* @param res respose
* @throws ServletException trouble
* @throws IOException different trouble
*/
@Override
public void doPost(HttpServletRequest req, HttpServletResponse res)
throws ServletException, IOException
{
long t0=0, t1=0, t2=0, t3=0, t4=0;
String soapAction=null;
MessageContext msgContext=null;
if (isDebug)
servletLog.debug("Enter: doPost()");
if( tlog.isDebugEnabled() ) {
t0=System.currentTimeMillis();
}
Message responseMsg = null;
String contentType = req.getContentType().toLowerCase();
if(servletLog.isDebugEnabled()) servletLog.debug("in SRWServlet.doPost: contentType="+contentType);
if(contentType.indexOf("atom+xml")>0 || contentType.equals("application/x-www-form-urlencoded")) {
msgContext = createMessageContext(getEngine(), req, res);
if(!srwInfo.setSRWStuff(req, res, msgContext)) {
servletLog.error("srwInfo.setSRWStuff failed!");
return;
}
SRWDatabase db=(SRWDatabase)msgContext.getProperty("db");
try {
if(servletLog.isDebugEnabled()) servletLog.debug("adding to database "+db);
if(servletLog.isDebugEnabled()) servletLog.debug("adding to database "+db.dbname);
InputStream is=req.getInputStream();
int len, offset=0, totlen=req.getContentLength();
if(servletLog.isDebugEnabled()) servletLog.debug("adding "+totlen+" bytes to database");
byte[] b=new byte[totlen];
while(totlen>0) {
len=is.read(b, offset, totlen);
totlen-=len;
offset+=len;
}
if(servletLog.isDebugEnabled()) servletLog.debug("adding data to database: "+new String(b, "UTF-8"));
if(contentType.equals("application/x-www-form-urlencoded")) {
// String converted=convertFormDataToTrivialXML(new String(b, "UTF-8"));
// Transformer trans=null;
// trans=db.transformers.get(mediaType);
// if(servletLog.isDebugEnabled())
// servletLog.debug("trans="+trans);
// if(trans!=null) {
// StringWriter toRec = new StringWriter();
// StreamSource fromRec = new StreamSource(new StringReader(soapResponse));
// try {
// trans.transform(fromRec, new StreamResult(toRec));
// soapResponse=toRec.toString();
// if(soapResponse.substring(0, Math.min(100, soapResponse.length())).contains("<?xml-stylesheet")) {
// if(!mediaType.contains("html"))
// resp.setContentType("text/xml");
// }
// else
// resp.setContentType(mediaType);
// } catch (TransformerException e) {
// servletLog.error(e, e);
// resp.setContentType("text/xml");
// sos=resp.getOutputStream();
// srwInfo.writeXmlHeader(sos, msgContext, req,
// db.searchStyleSheet);
// if(db.httpHeaderSetter!=null)
// db.httpHeaderSetter.setGetResponseHeaders(db.searchRequest, db.response, soapResponse, req, resp);
// }
}
String recordID=db.add(b);
if(recordID!=null) {
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setPostResponseHeaders(new String(b, "UTF-8"), req, res);
res.setHeader("Content-Location", recordID+"/");
res.setStatus(201);
}
else {
log.error("db.add failed! returning status code 500");
res.setStatus(500);
}
}
catch(Exception e) {
servletLog.error(e, e);
throw new ServletException(e);
}
finally {
SRWDatabase.putDb((String)msgContext.getProperty("dbname"), db);
}
servletLog.debug("exit: doPost() after processing atompub request");
return;
}
try {
AxisEngine engine = getEngine();
if (engine == null) {
// !!! should return a SOAP fault...
ServletException se =
new ServletException(Messages.getMessage("noEngine00"));
servletLog.debug("No Engine!", se);
throw se;
}
res.setBufferSize(1024 * 8); // provide performance boost.
/** get message context w/ various properties set
*/
msgContext = createMessageContext(engine, req, res);
// ? OK to move this to 'getMessageContext',
// ? where it would also be picked up for 'doGet()' ?
if (securityProvider != null) {
if (isDebug) servletLog.debug("securityProvider:" + securityProvider);
msgContext.setProperty(MessageContext.SECURITY_PROVIDER, securityProvider);
}
/* Get request message
*/
Message requestMsg =
new Message(req.getInputStream(),
false,
req.getHeader(HTTPConstants.HEADER_CONTENT_TYPE),
req.getHeader(HTTPConstants.HEADER_CONTENT_LOCATION));
if(isDebug) servletLog.debug("Request Message:" + requestMsg);
/* Set the request(incoming) message field in the context */
/**********************************************************/
msgContext.setRequestMessage(requestMsg);
//String url = HttpUtils.getRequestURL(req).toString();
String url = req.getRequestURL().toString();
msgContext.setProperty(MessageContext.TRANS_URL, url);
try {
/**
* Save the SOAPAction header in the MessageContext bag.
* This will be used to tell the Axis Engine which service
* is being invoked. This will save us the trouble of
* having to parse the Request message - although we will
* need to double-check later on that the SOAPAction header
* does in fact match the URI in the body.
*/
// (is this last stmt true??? (I don't think so - Glen))
/********************************************************/
soapAction = getSoapAction(req);
if (soapAction != null) {
msgContext.setUseSOAPAction(true);
msgContext.setSOAPActionURI(soapAction);
}
else { // not a SOAP request
String content=new String(((SOAPPart)(requestMsg.getSOAPPart())).getAsBytes());
if(content.startsWith("<")) { // must be an ATOM request
return;
}
if(content.indexOf("query=")>=0 || content.indexOf("scanClause=")>=0) {
processMethodRequest(msgContext, new LateContentParsingHttpServletRequestWrapper(req, content), res);
return;
}
// nothing I recognize
AxisFault af = new AxisFault("Client.NoSOAPAction",
Messages.getMessage("noHeader00",
"SOAPAction"),
null, null);
exceptionLog.error(Messages.getMessage("genFault00"), af);
throw af;
}
// Create a Session wrapper for the HTTP session.
// These can/should be pooled at some point.
// (Sam is Watching! :-)
msgContext.setSession(new AxisHttpSession(req));
if( tlog.isDebugEnabled() ) {
t1=System.currentTimeMillis();
}
srwInfo.setSRWStuff(req, res, msgContext);
/* Invoke the Axis engine... */
/*****************************/
if(isDebug) servletLog.debug("Invoking Axis Engine.");
//here we run the message by the engine
engine.invoke(msgContext);
if(isDebug) servletLog.debug("Return from Axis Engine.");
if( tlog.isDebugEnabled() ) {
t2=System.currentTimeMillis();
}
responseMsg = msgContext.getResponseMessage();
if (responseMsg == null) {
//tell everyone that something is wrong
throw new Exception(Messages.getMessage("noResponse01"));
}
} catch (AxisFault fault) {
//log and sanitize
processAxisFault(fault);
configureResponseFromAxisFault(res,fault);
responseMsg = msgContext.getResponseMessage();
if (responseMsg == null) {
responseMsg = new Message(fault);
}
} catch (Exception e) {
//other exceptions are internal trouble
responseMsg = msgContext.getResponseMessage();
res.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
responseMsg = convertExceptionToAxisFault(e, responseMsg);
}
} catch (AxisFault fault) {
processAxisFault(fault);
configureResponseFromAxisFault(res, fault);
responseMsg = msgContext.getResponseMessage();
if (responseMsg == null) {
responseMsg = new Message(fault);
}
}
//determine content type from message response
contentType = responseMsg.getContentType(msgContext.getSOAPConstants());
if( tlog.isDebugEnabled() ) {
t3=System.currentTimeMillis();
}
/* Send response back along the wire... */
/***********************************/
if (responseMsg != null) {
sendResponse(getProtocolVersion(req), contentType,
res, responseMsg);
}
if (isDebug) {
servletLog.debug("Response sent.");
servletLog.debug("Exit: doPost()");
}
if( tlog.isDebugEnabled() ) {
t4=System.currentTimeMillis();
tlog.debug("axisServlet.doPost: " + soapAction +
" pre=" + (t1-t0) +
" invoke=" + (t2-t1) +
" post=" + (t3-t2) +
" send=" + (t4-t3) +
" " + msgContext.getTargetService() + "." +
((msgContext.getOperation( ) == null) ?
"" : msgContext.getOperation().getName()) );
}
}
/**
* Configure the servlet response status code and maybe other headers
* from the fault info.
* @param response response to configure
* @param fault what went wrong
*/
private void configureResponseFromAxisFault(HttpServletResponse response,
AxisFault fault) {
// then get the status code
// It's been suggested that a lack of SOAPAction
// should produce some other error code (in the 400s)...
int status = getHttpServletResponseStatus(fault);
if (status == HttpServletResponse.SC_UNAUTHORIZED) {
// unauth access results in authentication request
response.setHeader("WWW-Authenticate","Basic realm=\"AXIS\"");
}
response.setStatus(status);
}
/**
* turn any Exception into an AxisFault, log it, set the response
* status code according to what the specifications say and
* return a response message for posting. This will be the response
* message passed in if non-null; one generated from the fault otherwise.
*
* @param exception what went wrong
* @param responseMsg what response we have (if any)
* @return a response message to send to the user
*/
private Message convertExceptionToAxisFault(Exception exception,
Message responseMsg) {
logException(exception);
if (responseMsg == null) {
AxisFault fault=AxisFault.makeFault(exception);
processAxisFault(fault);
responseMsg = new Message(fault);
}
return responseMsg;
}
/**
* Extract information from AxisFault and map it to a HTTP Status code.
*
* @param af Axis Fault
* @return HTTP Status code.
*/
@Override
protected int getHttpServletResponseStatus(AxisFault af) {
return af.getFaultCode().getLocalPart().startsWith("Server.Unauth")
? HttpServletResponse.SC_UNAUTHORIZED
: HttpServletResponse.SC_INTERNAL_SERVER_ERROR;
// This will raise a 401 for both
// "Unauthenticated" & "Unauthorized"...
}
/**
* write a message to the response, set appropriate headers for content
* type..etc.
* @param clientVersion client protocol, one of the HTTPConstants strings
* @param res response
* @param responseMsg message to write
* @throws AxisFault
* @throws IOException if the response stream can not be written to
*/
private void sendResponse(final String clientVersion,
String contentType,
HttpServletResponse res, Message responseMsg)
throws AxisFault, IOException
{
if (responseMsg == null) {
res.setStatus(HttpServletResponse.SC_NO_CONTENT);
if(isDebug) servletLog.debug("NO AXIS MESSAGE TO RETURN!");
//String resp = Messages.getMessage("noData00");
//res.setContentLength((int) resp.getBytes().length);
//res.getWriter().print(resp);
} else {
if(isDebug) {
servletLog.debug("Returned Content-Type:" +
contentType);
// servletLog.debug("Returned Content-Length:" +
// responseMsg.getContentLength());
}
try {
res.setContentType(contentType);
/* My understand of Content-Length
* HTTP 1.0
* -Required for requests, but optional for responses.
* HTTP 1.1
* - Either Content-Length or HTTP Chunking is required.
* Most servlet engines will do chunking if content-length is not specified.
*
*
*/
//if(clientVersion == HTTPConstants.HEADER_PROTOCOL_V10) //do chunking if necessary.
// res.setContentLength(responseMsg.getContentLength());
responseMsg.writeTo(res.getOutputStream());
} catch (SOAPException e){
logException(e);
}
}
if (!res.isCommitted()) {
res.flushBuffer(); // Force it right now.
}
}
/**
* Place the Request message in the MessagContext object - notice
* that we just leave it as a 'ServletRequest' object and let the
* Message processing routine convert it - we don't do it since we
* don't know how it's going to be used - perhaps it might not
* even need to be parsed.
* @return a message context
*/
private MessageContext createMessageContext(AxisEngine engine,
HttpServletRequest req,
HttpServletResponse res)
{
MessageContext msgContext = new MessageContext(engine);
if(isDebug) {
servletLog.debug("MessageContext:" + msgContext);
servletLog.debug("HEADER_CONTENT_TYPE:" +
req.getHeader( HTTPConstants.HEADER_CONTENT_TYPE));
servletLog.debug("HEADER_CONTENT_LOCATION:" +
req.getHeader( HTTPConstants.HEADER_CONTENT_LOCATION));
servletLog.debug("Constants.MC_HOME_DIR:" + String.valueOf(getHomeDir()));
servletLog.debug("Constants.MC_RELATIVE_PATH:"+req.getServletPath());
servletLog.debug("HTTPConstants.MC_HTTP_SERVLETLOCATION:"+ String.valueOf(getWebInfPath()));
servletLog.debug("HTTPConstants.MC_HTTP_SERVLETPATHINFO:" +
req.getPathInfo() );
servletLog.debug("HTTPConstants.HEADER_AUTHORIZATION:" +
req.getHeader(HTTPConstants.HEADER_AUTHORIZATION));
servletLog.debug("Constants.MC_REMOTE_ADDR:"+req.getRemoteAddr());
servletLog.debug("configPath:" + String.valueOf(getWebInfPath()));
}
/* Set the Transport */
/*********************/
msgContext.setTransportName(transportName);
/* Save some HTTP specific info in the bag in case someone needs it */
/********************************************************************/
msgContext.setProperty(Constants.MC_JWS_CLASSDIR, jwsClassDir);
msgContext.setProperty(Constants.MC_HOME_DIR, getHomeDir());
msgContext.setProperty(Constants.MC_RELATIVE_PATH,
req.getServletPath());
msgContext.setProperty(HTTPConstants.MC_HTTP_SERVLET, this );
msgContext.setProperty(HTTPConstants.MC_HTTP_SERVLETREQUEST, req );
msgContext.setProperty(HTTPConstants.MC_HTTP_SERVLETRESPONSE, res );
msgContext.setProperty(HTTPConstants.MC_HTTP_SERVLETLOCATION,
getWebInfPath() );
msgContext.setProperty(HTTPConstants.MC_HTTP_SERVLETPATHINFO,
req.getPathInfo() );
msgContext.setProperty(HTTPConstants.HEADER_AUTHORIZATION,
req.getHeader(HTTPConstants.HEADER_AUTHORIZATION));
msgContext.setProperty(Constants.MC_REMOTE_ADDR, req.getRemoteAddr());
// Set up a javax.xml.rpc.server.ServletEndpointContext
ServletEndpointContextImpl sec = new ServletEndpointContextImpl();
msgContext.setProperty(Constants.MC_SERVLET_ENDPOINT_CONTEXT, sec);
/* Save the real path */
/**********************/
String realpath =
getServletConfig().getServletContext()
.getRealPath(req.getServletPath());
if (realpath != null) {
msgContext.setProperty(Constants.MC_REALPATH, realpath);
}
msgContext.setProperty(Constants.MC_CONFIGPATH, getWebInfPath());
// msgContext.setProperty("targetURL", req.getRequestURL().append('?').append(req.getQueryString()).toString());
// servletLog.debug("targetURL property set to "+msgContext.getProperty("targetURL"));
return msgContext;
}
/**
* Extract the SOAPAction header.
* if SOAPAction is null then we'll we be forced to scan the body for it.
* if SOAPAction is "" then use the URL
* @param req incoming request
* @return the action
* @throws AxisFault
*/
private String getSoapAction(HttpServletRequest req)
throws AxisFault
{
String soapAction =req.getHeader(HTTPConstants.HEADER_SOAP_ACTION);
if(isDebug) servletLog.debug("HEADER_SOAP_ACTION:" + soapAction);
/**
* Technically, if we don't find this header, we should probably fault.
* It's required in the SOAP HTTP binding.
*/
if (soapAction!=null && soapAction.length()==0)
soapAction = req.getContextPath(); // Is this right?
return soapAction;
}
/**
* Provided to allow overload of default JWSClassDir
* by derived class.
* @return directory for JWS files
*/
@Override
protected String getDefaultJWSClassDir() {
return (getWebInfPath() == null)
? null // ??? what is a good FINAL default for WebLogic?
: getWebInfPath() + File.separator + "jwsClasses";
}
protected String getMediaType(HttpServletRequest req, SRWDatabase db) {
String mediaType, mimeType=req.getParameter("http:accept");
negotiatedContent=false;
if(mimeType==null)
mimeType=req.getParameter("httpAccept");
if(mimeType==null) {
mimeType=req.getHeader("accept");
negotiatedContent=true;
}
mediaType=db.mediaTypes.get(mimeType);
if(mediaType==null) { // never seen this mimeType before
MediaRangeSpec mrs=db.conneg.getBestMatch(mimeType);
if(mrs!=null)
mediaType=mrs.getMediaType();
else
mediaType="";
db.mediaTypes.put(mimeType, mediaType);
}
servletLog.info("mimeType="+mimeType+" and "+"mediaType="+mediaType+" from IP="+req.getRemoteAddr());
return mediaType;
}
/**
* Return the HTTP protocol level 1.1 or 1.0
* by derived class.
* @return one of the HTTPConstants values
*/
protected String getProtocolVersion(HttpServletRequest req){
String ret= HTTPConstants.HEADER_PROTOCOL_V10;
String prot= req.getProtocol();
if(prot!= null){
int sindex= prot.indexOf('/');
if(-1 != sindex){
String ver= prot.substring(sindex+1);
if(HTTPConstants.HEADER_PROTOCOL_V11.equals(ver.trim())){
ret= HTTPConstants.HEADER_PROTOCOL_V11;
}
}
}
return ret;
}
// static Runtime rt=Runtime.getRuntime();
protected void processMethodRequest(
org.apache.axis.MessageContext msgContext, HttpServletRequest req,
HttpServletResponse resp)
throws org.apache.axis.AxisFault, IOException {
long startTime=System.currentTimeMillis();
servletLog.debug("enter processMethodRequest");
// servletLoginfo("at start: totalMemory="+rt.totalMemory()+", freeMemory="+rt.freeMemory());
servletLog.info(req.getQueryString());
if(!srwInfo.setSRWStuff(req, resp, msgContext)) {
servletLog.error("srwInfo.setSRWStuff failed!");
return;
}
SRWDatabase db=(SRWDatabase)msgContext.getProperty("db");
String mediaType=getMediaType(req, db);
if(servletLog.isDebugEnabled()) {
Enumeration enumer=req.getParameterNames();
String name;
while(enumer.hasMoreElements()) {
name=(String)enumer.nextElement();
servletLog.debug(name+"=\""+req.getParameter(name)+"\"");
}
}
String operation=req.getParameter("operation"),
query=req.getParameter("query"),
scanClause=req.getParameter("scanClause");
servletLog.debug("in processMethodRequest: operation="+operation);
if(query!=null) {
servletLog.debug("in processMethodRequest: query:\n"+Utilities.byteArrayToString(query.getBytes("UTF8")));
}
if(scanClause!=null)
servletLog.debug("in processMethodRequest: scanClause:\n"+Utilities.byteArrayToString(scanClause.getBytes("UTF8")));
if((operation!=null && operation.equals("searchRetrieve")) || query!=null) { // searchRetrieveRequest
int i;
StringBuilder sb=new StringBuilder();
sb.append("<soap:Envelope ")
.append("xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\" ")
.append("xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" ")
.append("xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">")
.append("<soap:Body xmlns:srw=\"http://www.loc.gov/zing/srw/\">")
.append("<srw:searchRetrieveRequest>")
.append("<srw:query>").append(encode(query)).append("</srw:query>");
// is this a request for a single record?
// if("APP".equals(req.getAttribute("service"))) {
// System.out.println("in SRWServlet.processMethodRequest: processing an APP request");
// // should we set a recordSchema parameter?
// if(req.getParameter("recordSchema")==null) { // if one wasn't provided
// String recordSchema=db.recordSchemas.get(mediaType);
// if(recordSchema!=null) {
// System.out.println("in SRWServlet.processMethodRequest: adding recordSchema="+recordSchema+" parm to request");
// req.setAttribute("recordSchema", recordSchema);
// }
// else
// System.out.println("in SRWServlet.processMethodRequest: mediaType "+mediaType+" has no associated recordSchema");
// }
// }
String extension, namespace, parm, t;
Enumeration parms=req.getParameterNames();
while(parms.hasMoreElements()) {
parm=(String)parms.nextElement();
if(parm.equals("sortKeys")) {
t=req.getParameter(parm);
if(t!=null)
sb.append("<srw:sortKeys>").append(t).append("</srw:sortKeys>");
}
else if(parm.equals("startRecord")) {
t=req.getParameter(parm);
if(t!=null) {
try {
i=Integer.parseInt(t);
if(i<1)
i=Integer.MAX_VALUE;
}
catch(NumberFormatException e){
i=Integer.MAX_VALUE;
}
sb.append("<srw:startRecord>").append(i).append("</srw:startRecord>");
}
}
else if(parm.equals("maximumRecords")) {
t=req.getParameter(parm);
if(t!=null) {
try {
i=Integer.parseInt(t);
if(i<0)
i=Integer.MAX_VALUE;
}
catch(NumberFormatException e){
i=Integer.MAX_VALUE;
}
sb.append("<srw:maximumRecords>").append(i).append("</srw:maximumRecords>");
}
}
else if(parm.equals("recordSchema")) {
t=req.getParameter(parm);
if(t==null) {
t=(String)req.getAttribute(parm);
System.out.println("in SRWServlet.processMethodRequest: hooray! adding recordSchema="+t);
}
if(t!=null)
sb.append("<srw:recordSchema>").append(t).append("</srw:recordSchema>");
}
else if(parm.equals("recordPacking")) {
t=req.getParameter(parm);
if(t!=null)
sb.append("<srw:recordPacking>").append(t).append("</srw:recordPacking>");
}
else if(parm.equals("resultSetTTL")) {
t=req.getParameter(parm);
if(t!=null) {
try {
i=Integer.parseInt(t);
if(i<0)
i=Integer.MAX_VALUE;
sb.append("<srw:resultSetTTL>").append(i).append("</srw:resultSetTTL>");
}
catch(NumberFormatException e){
servletLog.error("resultSetTTL="+t);
}
}
}
else if(parm.equals("version")) {
t=req.getParameter(parm);
if(t!=null)
sb.append("<srw:version>").append(t).append("</srw:version>");
}
}
boolean hasExtraRequestData=false;
parms=req.getParameterNames();
while(parms.hasMoreElements()) {
parm=(String)parms.nextElement();
servletLog.debug("parm="+parm);
extension=srwInfo.getExtension(parm);
servletLog.debug("extension="+extension);
if(extension!=null) {
if(!hasExtraRequestData)
sb.append("<srw:extraRequestData>");
namespace=srwInfo.getNamespace(parm);
sb.append(" <").append(extension).append(" xmlns=\"").append(namespace).append("\"");
t=req.getParameter(parm);
if(t!=null && t.length()>0)
sb.append(">").append(t).append("</").append(extension).append(">");
else
sb.append("/>");
hasExtraRequestData=true;
}
}
if(hasExtraRequestData) {
sb.append(" </srw:extraRequestData>");
servletLog.debug(sb.toString());
}
sb.append("</srw:searchRetrieveRequest></soap:Body></soap:Envelope>");
if(servletLog.isDebugEnabled()) {
servletLog.debug("request="+sb.toString());
servletLog.debug(Utilities.byteArrayToString(sb.toString().getBytes("UTF-8")));
}
msgContext.setProperty("sru", "");
AxisEngine engine=getEngine();
ByteArrayInputStream bais=new ByteArrayInputStream(sb.toString().getBytes("UTF-8"));
Message msg=new Message(bais, false);
msgContext.setRequestMessage(msg);
msgContext.setTargetService("SRW");
try {
engine.invoke(msgContext);
}
catch(Exception e) {
servletLog.error(sb.toString());
servletLog.error(e, e);
}
Message respMsg=msgContext.getResponseMessage();
if(respMsg!=null) {
javax.servlet.ServletOutputStream sos = null;
// code to strip SOAP stuff out. Hope this can go away some day
String soapResponse=respMsg.getSOAPPartAsString();
if("APP".equals(req.getAttribute("service"))||"APP".equals(req.getParameter("service"))) {
int start=soapResponse.indexOf("<recordData");
if(start>=0) {
String contentType=(String)req.getAttribute("ContentType");
if(contentType!=null)
resp.setContentType(contentType);
else
resp.setContentType("text/xml");
start=soapResponse.indexOf('>', start+1);
int stop=soapResponse.indexOf("</recordData>", start);
soapResponse=cleanup(soapResponse.substring(start+1, stop)
.toCharArray());
if(contentType!=null && contentType.toLowerCase().indexOf("html")>0) {
soapResponse=srwInfo.getXmlHeaders(req, db.searchStyleSheet)+Utilities.unXmlEncode(soapResponse);
}
if(negotiatedContent) {
String contentLocation=db.contentLocations.get(mediaType);
if(contentLocation!=null)
resp.setHeader("Content-Location", contentLocation);
negotiatedContent=false;
}
}
else { // this was an APP get, but we got nothing!
resp.setContentType("text/html");
sos=resp.getOutputStream();
sos.println("<html><head><title>Document not found</title></head><body><h2>404: Document not found</h2><hr/>OCLC SRW/SRU Server</body></html>");
sos.close();
resp.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
return;
}
}
else { // extract SRU response from inside the SRW response
int start=soapResponse.indexOf("<searchRetrieveResponse");
if(start>=0) {
int stop=soapResponse.indexOf("</searchRetrieveResponse>");
soapResponse="<searchRetrieveResponse xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:schemaLocation=\"http://www.loc.gov/zing/srw/ http://www.loc.gov/standards/sru/sru1-1archive/xml-files/srw-types.xsd\""+cleanup(soapResponse.substring(start+23, stop+25)
.toCharArray());
if(negotiatedContent)
resp.setHeader("Content-Location", req.getRequestURI()+"?"+req.getQueryString()+"&httpAccept="+mediaType);
}
else { // damn, no searchRetrieveResponse!
resp.setContentType("text/html");
sos=resp.getOutputStream();
sos.println("<html><head><title>Document not found</title></head><body><h2>404: Document not found (No searchRetrieveResponse)</h2><hr/>OCLC SRW/SRU Server</body></html>");
sos.close();
resp.setStatus(HttpURLConnection.HTTP_NOT_FOUND);
return;
}
}
// now, let's see if they wanted something else
if(soapResponse.length()>0 && !mediaType.equals("")) {
Transformer trans=null;
trans=db.transformers.get(mediaType);
if(servletLog.isDebugEnabled())
servletLog.debug("trans="+trans);
if(trans!=null) {
StringWriter toRec = new StringWriter();
StreamSource fromRec = new StreamSource(new StringReader(soapResponse));
try {
trans.transform(fromRec, new StreamResult(toRec));
soapResponse=toRec.toString();
if(soapResponse.substring(0, Math.min(100, soapResponse.length())).contains("<?xml-stylesheet")) {
if(!mediaType.contains("html"))
resp.setContentType("text/xml");
}
else
resp.setContentType(mediaType);
} catch (TransformerException e) {
servletLog.error(e, e);
resp.setContentType("text/xml");
sos=resp.getOutputStream();
srwInfo.writeXmlHeader(sos, msgContext, req,
db.searchStyleSheet);
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setGetResponseHeaders(db.searchRequest, db.response, soapResponse, req, resp);
}
}
else if(mediaType.contains("html")) {
// use the stylesheet that would have gone into the XML response as the transformer
String styleSheet=req.getParameter("stylesheet");
if(styleSheet==null)
styleSheet=db.searchStyleSheet;
String languages=req.getHeader("Accept-Language");
String realXsl=(String)SRWServletInfo.realXsls.get(languages+'/'+styleSheet);
if(servletLog.isDebugEnabled())
servletLog.debug("realXsl="+realXsl);
if(realXsl==null)
styleSheet=SRWServletInfo.findRealXsl(languages, req.getLocales(), styleSheet);
else
styleSheet=realXsl;
// if(!styleSheet.startsWith("/")) // relative URL
// styleSheet=req.getContextPath()+"/"+styleSheet;
if(servletLog.isDebugEnabled())
servletLog.debug("styleSheet="+styleSheet);
trans=transformers.get(styleSheet);
if(trans==null) { // gotta load it
// URL url=new URL("http://"+req.getServerName()+":"+req.getServerPort()+styleSheet);
URL url=new URL("http://localhost"+portNumber+styleSheet);
servletLog.info("stylesheet url: "+url);
HttpURLConnection conn=null;
try {
conn=(HttpURLConnection) url.openConnection();
servletLog.info("opened URL with response code "+conn.getResponseCode());
}
catch(ConnectException e) {
servletLog.error("ConnectException for stylesheet url: http://localhost"+portNumber+styleSheet);
}
if(conn.getResponseCode()!=200)
servletLog.error("response code="+conn.getResponseCode()+" getting "+url);
else {
StreamSource so=new StreamSource(conn.getInputStream());
so.setSystemId(url.toString());
try {
trans = TransformerFactory.newInstance().newTransformer(so);
transformers.put(mediaType, trans);
} catch (TransformerConfigurationException ex) {
servletLog.error("unable to build transformer "+url);
}
}
}
StringWriter toRec = new StringWriter();
StreamSource fromRec = new StreamSource(new StringReader(soapResponse));
try {
trans.transform(fromRec, new StreamResult(toRec));
soapResponse=toRec.toString();
if(soapResponse.substring(0, Math.min(100, soapResponse.length())).contains("<?xml-stylesheet")) {
resp.setContentType("text/xml");
}
else
resp.setContentType("text/html");
} catch (TransformerException e) {
servletLog.error(e, e);
resp.setContentType("text/xml");
sos=resp.getOutputStream();
srwInfo.writeXmlHeader(sos, msgContext, req,
db.searchStyleSheet);
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setGetResponseHeaders(db.searchRequest, db.response, soapResponse, req, resp);
}
}
else {
String xmlHeaders=srwInfo.getXmlHeaders(req, db.searchStyleSheet).toString();
if(trans==null && xmlHeaders.substring(0, Math.min(100, xmlHeaders.length())).contains("xml-stylesheet"))
resp.setContentType("text/xml");
else if(soapResponse.substring(0, Math.min(100, soapResponse.length())).contains("xml-stylesheet"))
resp.setContentType("text/xml");
else
resp.setContentType(mediaType);
sos=resp.getOutputStream();
if(trans==null)
srwInfo.writeXmlHeader(sos, msgContext, req,
db.searchStyleSheet);
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setGetResponseHeaders(db.searchRequest, db.response, soapResponse, req, resp);
}
}
else { //we don't care what they want, just return the SRU response
resp.setContentType("text/xml");
sos=resp.getOutputStream();
srwInfo.writeXmlHeader(sos, msgContext, req,
db.searchStyleSheet);
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setGetResponseHeaders(db.searchRequest, db.response, soapResponse, req, resp);
}
//writer.println(soapResponse);
//writer.close();
if(sos==null)
sos=resp.getOutputStream();
if(db.httpHeaderSetter!=null)
db.httpHeaderSetter.setGetResponseHeaders(db.searchRequest, db.response, soapResponse, req, resp);
try {
sos.write(soapResponse.getBytes("utf-8"));
sos.close();
}
catch (IOException e) {
// no need to worry about this, let it slide
}
}
else {
resp.setContentType("text/html");
PrintWriter writer=resp.getWriter();
writer.println("<p>"+Messages.getMessage("noResponse01")+"</p>");
servletLog.error("request generated no response!");
writer.close();
}
// servletLog.info("at exit: totalMemory="+rt.totalMemory()+", freeMemory="+rt.freeMemory());
servletLog.debug("exit processMethodRequest");
return;
}
if((operation!=null && operation.equals("scan")) || scanClause!=null) { // scanRequest
int i;
String t;
StringBuilder sb=new StringBuilder();
sb.append("<soap:Envelope ")
.append("xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\" ")
.append("xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" ")
.append("xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">")
.append("<soap:Body>")
.append("<srw:scanRequest ")
.append("xmlns:srw=\"http://www.loc.gov/zing/srw/\">")
.append("<srw:scanClause>");
if(scanClause!=null) {
sb.append(encode(scanClause));
}
else
sb.append("\"\"");
sb.append("</srw:scanClause>");
t=req.getParameter("responsePosition");
if(t!=null) {
try {
i=Integer.parseInt(t);
if(i<0)
i=Integer.MAX_VALUE;
}
catch(NumberFormatException e){
i=Integer.MAX_VALUE;
}
sb.append("<srw:responsePosition>").append(i).append("</srw:responsePosition>");
}
t=req.getParameter("maximumTerms");
if(t!=null) {
try {
i=Integer.parseInt(t);
if(i<1)
i=Integer.MAX_VALUE;
}
catch(NumberFormatException e){
i=Integer.MAX_VALUE;
}
sb.append("<srw:maximumTerms>").append(i).append("</srw:maximumTerms>");
}
sb.append("</srw:scanRequest></soap:Body></soap:Envelope>");
if(servletLog.isDebugEnabled())
servletLog.debug(sb.toString());
msgContext.setProperty("sru", "");
msgContext.setTargetService("SRW");
AxisEngine engine=getEngine();
if(servletLog.isDebugEnabled())
servletLog.debug("request="+sb.toString());
ByteArrayInputStream bais=new ByteArrayInputStream(sb.toString().getBytes("UTF-8"));
Message msg=new Message(bais, false);
msgContext.setRequestMessage(msg);
try {
engine.invoke(msgContext);
}
catch(Exception e) {
servletLog.error(e, e);
}
Message respMsg=msgContext.getResponseMessage();
if(respMsg!=null) {
resp.setContentType("text/xml");
PrintWriter writer=resp.getWriter();
// code to strip SOAP stuff out. Hope this can go away some day
String soapResponse=respMsg.getSOAPPartAsString();
int start=soapResponse.indexOf("<scanResponse");
if(start>=0) {
int stop=soapResponse.indexOf("</scanResponse>");
soapResponse=cleanup(soapResponse.substring(start, stop+15)
.toCharArray());
srwInfo.writeXmlHeader(writer, msgContext, req,
db.scanStyleSheet);
}
writer.println(soapResponse);
writer.close();
}
else {
resp.setContentType("text/html");
PrintWriter writer=resp.getWriter();
writer.println("<p>"+Messages.getMessage("noResponse01")+"</p>");
servletLog.error("request generated no response!");
writer.close();
}
if(servletLog.isDebugEnabled()) {
servletLog.debug("elapsed time: "+(System.currentTimeMillis()-startTime)+"ms");
servletLog.debug("exit processMethodRequest");
}
return;
}
servletLog.error("fell through processMethodRequest!");
servletLog.error("operation="+operation+", query="+query+", scanClause="+scanClause);
}
private String cleanup(char[] buf) {
boolean didOne=false, insideRecordData=false;
int i, j, len=buf.length;
for(i=0; i<len; i++) {
if(buf[i]==' ' && len-i>5) // might be " xsi:"
if(buf[i+1]=='x' && buf[i+2]=='s' && buf[i+3]=='i' &&
buf[i+4]==':') {
if(insideRecordData)
if(didOne)
continue;
else
didOne=true;
boolean foundQuote=false;
for(j=i+5; j<len; j++)
if(buf[j]=='"')
if(foundQuote)
break;
else
foundQuote=true;
if(j==len) // never found matching quotes, so ignore
continue;
// remove offending chars
//servletLog.info("i="+i+", j="+j+", len="+len);
System.arraycopy(buf, j+1, buf, i, len-j-1);
len-=(j-i)+1;
i--;
continue;
}
if(buf[i]=='<' && len-1>11) // might be "<recordData>"
if(buf[i+1]=='r' && buf[i+2]=='e' && buf[i+3]=='c' &&
buf[i+4]=='o' && buf[i+5]=='r' && buf[i+6]=='d' &&
buf[i+7]=='D' && buf[i+8]=='a' && buf[i+9]=='t' &&
buf[i+10]=='a') {
insideRecordData=true;
didOne=false;
}
if(buf[i]=='<' && len-1>12) // might be "</recordData>"
if(buf[i+1]=='/' && buf[i+2]=='r' && buf[i+3]=='e' &&
buf[i+4]=='c' && buf[i+5]=='o' && buf[i+6]=='r' &&
buf[i+7]=='d' && buf[i+8]=='D' && buf[i+9]=='a' &&
buf[i+10]=='t' && buf[i+11]=='a') {
insideRecordData=false;
}
}
return new String(buf, 0, len);
}
static String encode(String s) {
if(s==null)
return "";
StringBuilder sb=new StringBuilder();
char c, chars[]=s.toCharArray();
for(int i=0; i<chars.length; i++) {
c=chars[i];
if(c==' ' || c=='<' || c=='&' || c=='>' || c=='"' || c=='\'')
sb.append("&#").append(Integer.toString(c)).append(';');
else
sb.append(c);
}
return sb.toString();
}
private String convertFormDataToTrivialXML(String formData) {
int i;
String name, parm;
StringBuilder sb=new StringBuilder("<form>\n");
StringTokenizer st=new StringTokenizer(formData, "&");
while(st.hasMoreTokens()) {
parm=st.nextToken();
i=parm.indexOf('=');
if(i>0) {
name=parm.substring(0, i);
sb.append('<').append(name).append('>').append(parm.substring(i+1)).append("</").append(name).append(">\n");
}
}
sb.append("</form>\n");
return sb.toString();
}
}