source/mir/util/HTTPParsedRequest.java

   1 /*
   2  * Copyright (C) 2001, 2002 The Mir-coders group
   3  *
   4  * This file is part of Mir.
   5  *
   6  * Mir is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * Mir is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with Mir; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  * In addition, as a special exception, The Mir-coders gives permission to link
  21  * the code of this program with  any library licensed under the Apache Software License,
  22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
  23  * (or with modified versions of the above that use the same license as the above),
  24  * and distribute linked combinations including the two.  You must obey the
  25  * GNU General Public License in all respects for all of the code used other than
  26  * the above mentioned libraries.  If you modify this file, you may extend this
  27  * exception to your version of the file, but you are not obligated to do so.
  28  * If you do not wish to do so, delete this exception statement from your version.
  29  */
  30 package mir.util;
  31
  32 import java.util.ArrayList;
  33 import java.util.Arrays;
  34 import java.util.Collection;
  35 import java.util.Collections;
  36 import java.util.Enumeration;
  37 import java.util.HashMap;
  38 import java.util.Iterator;
  39 import java.util.List;
  40 import java.util.Map;
  41
  42 import javax.servlet.http.HttpServletRequest;
  43
  44 import mir.log.LoggerWrapper;
  45
  46 import org.apache.commons.fileupload.DiskFileUpload;
  47 import org.apache.commons.fileupload.FileItem;
  48
  49 public class HTTPParsedRequest {
  50   static final String MULTIPART_FORMDATA_CONTENTTYPE = "multipart/form-data";
  51
  52   private HttpServletRequest request;
  53   private String encoding;
  54   private int maxUploadSize;
  55   private String tempDir;
  56
  57   private Map parameterStringValues;
  58   private Map parameterListValues;
  59   private List files;
  60
  61   private LoggerWrapper logger = new LoggerWrapper("Utility.HTTPParsedRequest");
  62
  63   public HTTPParsedRequest(HttpServletRequest aRequest, String anEncoding, int aMaxUploadSize, String aTempDir) throws UtilExc, UtilFailure {
  64     request = aRequest;
  65     encoding = anEncoding;
  66     maxUploadSize = aMaxUploadSize;
  67     tempDir = aTempDir;
  68
  69     parameterStringValues = new HashMap();
  70     parameterListValues = new HashMap();
  71     files = new ArrayList();
  72
  73     parseRequest(aRequest);
  74   }
  75
  76   public HTTPParsedRequest(HttpServletRequest aRequest, int aMaxUploadSize, String aTempDir) throws UtilExc, UtilFailure {
  77     this(aRequest, aRequest.getCharacterEncoding(), aMaxUploadSize, aTempDir);
  78   }
  79
  80   public HttpServletRequest getRequest() {
  81     return request;
  82   }
  83
  84   public String getParameter(String aName) {
  85     return (String) parameterStringValues.get(aName);
  86   }
  87
  88   public Collection getParameterNames() {
  89     return parameterStringValues.keySet();
  90   }
  91
  92   public String getHeader(String aName) {
  93     return request.getHeader(aName);
  94   }
  95
  96   public List getFiles() {
  97     return files;
  98   }
  99
 100   public List getParameterList(String aName) {
 101     if (parameterListValues.containsKey(aName))
 102       return (List) parameterListValues.get(aName);
 103     else
 104       return Collections.EMPTY_LIST;
 105   }
 106
 107   protected void parseRequest(HttpServletRequest aRequest) throws UtilExc, UtilFailure {
 108
 109     try {
 110       String contentType = aRequest.getContentType();
 111       List parts = StringRoutines.splitString(contentType, ";");
 112
 113       Enumeration e = aRequest.getParameterNames();
 114
 115       while (e.hasMoreElements()) {
 116         String name = (String) e.nextElement();
 117
 118         parameterStringValues.put(name, aRequest.getParameter(name));
 119         List listValue = new ArrayList(Arrays.asList(aRequest.getParameterValues(name)));
 120         parameterListValues.put(name, listValue);
 121       }
 122
 123       if (parts.size()>0 && ((String) parts.get(0)).trim().toLowerCase().equals(MULTIPART_FORMDATA_CONTENTTYPE)) {
 124         parseMultipartRequest();
 125       }
 126     }
 127     catch (Throwable t) {
 128       t.printStackTrace();
 129
 130       throw new UtilFailure(t);
 131     }
 132   }
 133
 134   protected void parseMultipartRequest() throws UtilExc, UtilFailure {
 135     try {
 136       DiskFileUpload upload = new DiskFileUpload();
 137
 138       upload.setSizeThreshold(100000);
 139       upload.setSizeMax(maxUploadSize);
 140       upload.setHeaderEncoding(encoding);
 141       upload.setRepositoryPath(tempDir);
 142
 143       List items = upload.parseRequest(request);
 144
 145       Iterator i = items.iterator();
 146       while (i.hasNext()) {
 147         FileItem item = (FileItem) i.next();
 148
 149         if (item.isFormField()) {
 150           if (!parameterStringValues.containsKey(item.getName())) {
 151             parameterStringValues.put(item.getFieldName(), item.getString(encoding));
 152           }
 153
 154           List listValue = (List) parameterListValues.get(item.getFieldName());
 155           if (listValue == null) {
 156             listValue = new ArrayList();
 157             parameterListValues.put(item.getFieldName(), listValue);
 158           }
 159           listValue.add(item.getString(encoding));
 160         }
 161         else {
 162           if (item.getSize()>0)
 163             files.add(item);
 164         }
 165       }
 166     }
 167     catch (Throwable t) {
 168       throw new UtilFailure(t);
 169     }
 170   }
 171 }