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 the com.oreilly.servlet library, any library
  22  * licensed under the Apache Software License, The Sun (tm) Java Advanced
  23  * Imaging library (JAI), The Sun JIMI library (or with modified versions of
  24  * the above that use the same license as the above), and distribute linked
  25  * combinations including the two.  You must obey the GNU General Public
  26  * License in all respects for all of the code used other than the above
  27  * mentioned libraries.  If you modify this file, you may extend this exception
  28  * to your version of the file, but you are not obligated to do so.  If you do
  29  * not wish to do so, delete this exception statement from your version.
  30  */
  31
  32 package mir.util;
  33
  34 import java.util.Arrays;
  35 import java.util.Enumeration;
  36 import java.util.HashMap;
  37 import java.util.Iterator;
  38 import java.util.List;
  39 import java.util.Map;
  40 import java.util.Vector;
  41 import javax.servlet.http.HttpServletRequest;
  42
  43 import org.apache.commons.fileupload.FileItem;
  44 import org.apache.commons.fileupload.FileUpload;
  45 import mir.log.LoggerWrapper;
  46
  47 public class HTTPParsedRequest {
  48   static final String MULTIPART_FORMDATA_CONTENTTYPE = "multipart/form-data";
  49
  50   private HttpServletRequest request;
  51   private String encoding;
  52   private int maxUploadSize;
  53   private String tempDir;
  54
  55   private Map stringValues;
  56   private Map listValues;
  57   private List files;
  58
  59   private LoggerWrapper logger;
  60
  61   public HTTPParsedRequest(HttpServletRequest aRequest, String anEncoding, int aMaxUploadSize, String aTempDir) throws UtilExc, UtilFailure {
  62     logger = new LoggerWrapper("Utility,HTTPParsedRequest");
  63
  64     request = aRequest;
  65     encoding = anEncoding;
  66     maxUploadSize = aMaxUploadSize;
  67     tempDir = aTempDir;
  68
  69     stringValues = new HashMap();
  70     listValues = new HashMap();
  71     files = new Vector();
  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) stringValues.get(aName);
  86   }
  87
  88   public List getFiles() {
  89     return files;
  90   }
  91
  92   public List getParameterList(String aName) {
  93     if (listValues.containsKey(aName))
  94       return (List) listValues.get(aName);
  95     else
  96       return new Vector();
  97   }
  98
  99   protected void parseRequest(HttpServletRequest aRequest) throws UtilExc, UtilFailure {
 100
 101     try {
 102       String contentType = aRequest.getContentType();
 103       List parts = StringRoutines.splitString(contentType, ";");
 104
 105       Enumeration e = aRequest.getParameterNames();
 106
 107       while (e.hasMoreElements()) {
 108         String name = (String) e.nextElement();
 109
 110         stringValues.put(name, aRequest.getParameter(name));
 111         List listValue = new Vector(Arrays.asList(aRequest.getParameterValues(name)));
 112         listValues.put(name, listValue);
 113       }
 114
 115       if (parts.size()>0 && ((String) parts.get(0)).trim().toLowerCase().equals(MULTIPART_FORMDATA_CONTENTTYPE)) {
 116         parseMultipartRequest();
 117       }
 118     }
 119     catch (Throwable t) {
 120       t.printStackTrace();
 121
 122       throw new UtilFailure(t);
 123     }
 124   }
 125
 126   protected void parseMultipartRequest() throws UtilExc, UtilFailure {
 127     try {
 128       FileUpload upload = new FileUpload();
 129
 130       upload.setSizeMax(maxUploadSize);
 131       upload.setSizeThreshold(4096);
 132       upload.setRepositoryPath(tempDir);
 133
 134       List items = upload.parseRequest(request);
 135
 136       Iterator i = items.iterator();
 137       while (i.hasNext()) {
 138         FileItem item = (FileItem) i.next();
 139
 140         if (item.isFormField()) {
 141           if (!stringValues.containsKey(item.getName())) {
 142             stringValues.put(item.getFieldName(), item.getString(encoding));
 143           }
 144
 145           List listValue = (List) listValues.get(item.getFieldName());
 146           if (listValue == null) {
 147             listValue = new Vector();
 148             listValues.put(item.getFieldName(), listValue);
 149           }
 150           listValue.add(item.getString(encoding));
 151         }
 152         else {
 153           if (item.getSize()>0)
 154             files.add(item);
 155         }
 156       }
 157     }
 158     catch (Throwable t) {
 159       throw new UtilFailure(t);
 160     }
 161   }
 162 }