Index: StreamGenerator.java
===================================================================
RCS file: /home/cvspublic/xml-cocoon2/src/org/apache/cocoon/generation/StreamGenerator.java,v
retrieving revision 1.7
diff -u -r1.7 StreamGenerator.java
--- StreamGenerator.java 2001/10/11 07:28:21 1.7
+++ StreamGenerator.java 2001/10/22 04:48:03
@@ -13,8 +13,9 @@
import org.apache.cocoon.ProcessingException;
import org.apache.cocoon.ResourceNotFoundException;
import org.apache.cocoon.components.parser.Parser;
-import org.apache.cocoon.environment.Request;
+import org.apache.avalon.excalibur.pool.Poolable;
import org.apache.cocoon.environment.http.HttpEnvironment;
+import javax.servlet.http.HttpServletRequest;
import org.apache.cocoon.util.PostInputStream;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
@@ -45,7 +46,8 @@
* @author Kinga Dziembowski
* @version $Revision: 1.7 $ $Date: 2001/10/11 07:28:21 $
*/
-public class StreamGenerator extends ComposerGenerator {
+public class StreamGenerator extends ComposerGenerator implements Poolable
+{
public static final String CLASS = StreamGenerator.class.getName();
/** The parameter holding the name associated with the xml data **/
@@ -70,31 +72,27 @@
/**
* Generate XML data out of request InputStream.
*/
- public void generate() throws IOException, SAXException, ProcessingException {
+ public void generate() throws IOException, SAXException, ProcessingException
+ {
Parser parser = null;
String parameter = parameters.getParameter(StreamGenerator.FORM_NAME, null);
int len = 0;
-
- try {
- Request request = (Request)objectModel.get(Constants.REQUEST_OBJECT);
- if (request.getContentType().equals("application/x-www-form-urlencoded")) {
+ String contentType = null;
+ try
+ {
+ HttpServletRequest request = (HttpServletRequest) objectModel.get(HttpEnvironment.HTTP_REQUEST_OBJECT);
+ contentType = request.getContentType();
+ if (contentType.startsWith("application/x-www-form-urlencoded")) {
String sXml = request.getParameter(parameter);
inputSource = new InputSource(new StringReader(sXml));
- } else if (request.getContentType().equals("text/plain") ||
- request.getContentType().equals("text/xml") ||
- request.getContentType().equals("application/xml")) {
+ } else if (contentType.startsWith("text/plain") ||
+ contentType.startsWith("text/xml") ||
+ contentType.startsWith("application/xml")) {
len = request.getContentLength();
if (len > 0) {
- // we have hopefully an http request here
- javax.servlet.http.HttpServletRequest httpRequest =
- (javax.servlet.http.HttpServletRequest)objectModel.get(HttpEnvironment.HTTP_REQUEST_OBJECT);
- if (httpRequest != null) {
- PostInputStream anStream = new PostInputStream(httpRequest.getInputStream(), len);
+ PostInputStream anStream = new PostInputStream(request.getInputStream(), len);
inputSource = new InputSource(anStream);
- } else {
- throw new IOException("No http request object found");
- }
} else {
throw new IOException("getContentLen() == 0");
}
@@ -105,6 +103,11 @@
if (getLogger().isDebugEnabled()) {
getLogger().debug("processing stream ContentType= " + request.getContentType() + "ContentLen= " + len);
}
+ String charset = getCharacterEncoding(request, contentType) ;
+ if( charset != null)
+ {
+ this.inputSource.setEncoding(charset);
+ }
parser = (Parser)this.manager.lookup(Parser.ROLE);
parser.setContentHandler(super.contentHandler);
parser.setLexicalHandler(super.lexicalHandler);
@@ -124,5 +127,82 @@
}
}
}
+
+ /**
+ * Content type HTTP header can contains character encodinf info
+ * for ex. Content-Type: text/xml; charset=UTF-8
+ * If the servlet is following spec 2.3 and higher the servlet API can be used to retrieve character encoding part of
+ * Content-Type header. Some containers can choose to not unpack charset info - the spec is not strong about it.
+ * in any case this method can be used as a latest resource to retrieve the passed charset value.
+ * null is returned.
+ * It is very common mistake to send : Content-Type: text/xml; charset="UTF-8".
+ * Some containers are not filtering this mistake and the processing results in exception..
+ * The getCharacterEncoding() compensates for above mistake.
+ *
+ * @param contentType value associated with Content-Type HTTP header.
+ */
+ public String getCharacterEncoding(HttpServletRequest req, String contentType)
+ {
+ String charencoding = null;
+ String charset = "charset=";
+ if (contentType == null)
+ {
+ return (null);
+ }
+ int idx = contentType.indexOf(charset);
+ if (idx == -1)
+ {
+ return (null);
+ }
+ try
+ {
+ charencoding = req.getCharacterEncoding();
+
+ if ( charencoding != null)
+ {
+ getLogger().debug("charset from container: " + charencoding);
+ charencoding = charencoding.trim();
+ if ((charencoding.length() > 2) && (charencoding.startsWith("\""))&& (charencoding.endsWith("\"")))
+ {
+ charencoding = charencoding.substring(1, charencoding.length() - 1);
+ }
+ getLogger().debug("charset from container clean: " + charencoding);
+ return (charencoding);
+ }
+ else
+ {
+
+ return extractCharset( contentType, idx );
+ }
+ }
+ catch(Throwable e)
+ {
+ // We will be there if the container do not implement getCharacterEncoding() method
+ return extractCharset( contentType, idx );
+ }
+ }
+
+
+ protected String extractCharset(String contentType, int idx)
+ {
+ String charencoding = null;
+ String charset = "charset=";
+
+ getLogger().debug("charset from extractCharset");
+ charencoding = contentType.substring(idx + charset.length());
+ int idxEnd = charencoding.indexOf(";");
+ if (idxEnd != -1)
+ {
+ charencoding = charencoding.substring(0, idxEnd);
+ }
+ charencoding = charencoding.trim();
+ if ((charencoding.length() > 2) && (charencoding.startsWith("\""))&& (charencoding.endsWith("\"")))
+ {
+ charencoding = charencoding.substring(1, charencoding.length() - 1);
+ }
+ getLogger().debug("charset from extractCharset: " + charencoding);
+ return (charencoding.trim());
+
+ }
}