/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.myfaces.tobago.internal.webapp;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.myfaces.tobago.internal.component.AbstractUIPage;
import javax.faces.FacesException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.File;
import java.io.UnsupportedEncodingException;
import java.util.Arrays;
import java.util.Collections;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
public class TobagoMultipartFormdataRequest extends HttpServletRequestWrapper {
private static final Logger LOG = LoggerFactory.getLogger(TobagoMultipartFormdataRequest.class);
public static final long ONE_KB = 1024;
public static final long ONE_MB = ONE_KB * ONE_KB;
public static final long ONE_GB = ONE_KB * ONE_MB;
private Map<String, String[]> parameters;
private Map<String, FileItem> fileItems;
public TobagoMultipartFormdataRequest(HttpServletRequest request) {
this(request, System.getProperty("java.io.tmpdir"), ONE_MB);
}
public TobagoMultipartFormdataRequest(HttpServletRequest request, String repositoryPath, long maxSize) {
super(request);
init(request, repositoryPath, maxSize);
}
private void init(HttpServletRequest request, String repositoryPath, long maxSize) {
if (!ServletFileUpload.isMultipartContent(request)) {
String errorText = "contentType is not multipart/form-data but '" + request.getContentType() + "'";
LOG.error(errorText);
throw new FacesException(errorText);
} else {
parameters = new HashMap<String, String[]>();
fileItems = new HashMap<String, FileItem>();
DiskFileItemFactory factory = new DiskFileItemFactory();
factory.setRepository(new File(repositoryPath));
ServletFileUpload upload = new ServletFileUpload(factory);
upload.setSizeMax(maxSize);
if (upload.getHeaderEncoding() != null) {
// TODO: enable configuration of 'accept-charset'
upload.setHeaderEncoding(AbstractUIPage.FORM_ACCEPT_CHARSET);
}
List<FileItem> itemList;
try {
itemList = (List<FileItem>) upload.parseRequest(request);
} catch (FileUploadException e) {
//LOG.error(e);
throw new FacesException(e);
}
if (LOG.isDebugEnabled()) {
LOG.debug("parametercount = " + itemList.size() + " + " + request.getParameterMap().size());
}
for (FileItem item : itemList) {
String key = item.getFieldName();
if (LOG.isDebugEnabled()) {
String value = item.getString();
if (value.length() > 100) {
value = value.substring(0, 100) + " [...]";
}
LOG.debug("Parameter: '" + key + "'='" + value + "' isFormField=" + item.isFormField()
+ " contentType='" + item.getContentType() + "'");
}
if (item.isFormField()) {
String newValue;
try {
// TODO: enable configuration of 'accept-charset'
newValue = item.getString(AbstractUIPage.FORM_ACCEPT_CHARSET);
} catch (UnsupportedEncodingException e) {
LOG.error("Caught: " + e.getMessage(), e);
newValue = item.getString();
}
addParameter(key, newValue);
} else {
fileItems.put(key, item);
}
}
// merging the GET parameters:
Enumeration e = request.getParameterNames();
while(e.hasMoreElements()) {
final String name = (String) e.nextElement();
final String[] newValues = request.getParameterValues(name);
if (LOG.isDebugEnabled()) {
LOG.debug("Parameter: '" + name + "'='" + Arrays.toString(newValues) + "' (GET)");
}
for (String newValue : newValues) {
addParameter(name, newValue);
}
}
}
}
private void addParameter(String key, String newValue) {
final String[] inStock = parameters.get(key);
final String[] values;
if (inStock == null) {
values = new String[]{newValue};
} else {
values = new String[inStock.length + 1];
System.arraycopy(inStock, 0, values, 0, inStock.length);
values[inStock.length] = newValue;
}
parameters.put(key, values);
}
public FileItem getFileItem(String key) {
if (fileItems != null) {
return fileItems.get(key);
}
return null;
}
public String getParameter(String key) {
String parameter = null;
String[] values = (String[]) parameters.get(key);
if (values != null) {
parameter = values[0];
}
return parameter;
}
public Enumeration getParameterNames() {
return Collections.enumeration(parameters.keySet());
}
public String[] getParameterValues(String key) {
return (String[]) parameters.get(key);
}
public Map getParameterMap() {
return parameters;
}
public static long getMaxSize(String param) {
if (param != null) {
String number = param.toLowerCase(Locale.ENGLISH);
long factor = 1;
if (number.endsWith("g")) {
factor = ONE_GB;
number = number.substring(0, number.length() - 1);
} else if (number.endsWith("m")) {
factor = ONE_MB;
number = number.substring(0, number.length() - 1);
} else if (number.endsWith("k")) {
factor = ONE_KB;
number = number.substring(0, number.length() - 1);
}
try {
return Long.parseLong(number.trim()) * factor;
} catch (NumberFormatException e) {
LOG.error("Given max file size for "
+ TobagoMultipartFormdataRequest.class.getName() + " " + param + " couldn't parsed to a number");
}
}
return ONE_MB;
}
}