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.Arrays;
  33 import java.util.Enumeration;
  34 import java.util.HashMap;
  35 import java.util.Iterator;
  36 import java.util.List;
  37 import java.util.Map;
  38 import java.util.Vector;
  39
  40 import javax.servlet.http.HttpServletRequest;
  41
  42 import mir.log.LoggerWrapper;
  43
  44 import org.apache.commons.fileupload.FileItem;
  45 import org.apache.commons.fileupload.FileUpload;
  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 String getHeader(String aName) {
  89     return request.getHeader(aName);
  90   }
  91
  92   public List getFiles() {
  93     return files;
  94   }
  95
  96   public List getParameterList(String aName) {
  97     if (listValues.containsKey(aName))
  98       return (List) listValues.get(aName);
  99     else
 100       return new Vector();
 101   }
 102
 103   protected void parseRequest(HttpServletRequest aRequest) throws UtilExc, UtilFailure {
 104
 105     try {
 106       String contentType = aRequest.getContentType();
 107       List parts = StringRoutines.splitString(contentType, ";");
 108
 109       Enumeration e = aRequest.getParameterNames();
 110
 111       while (e.hasMoreElements()) {
 112         String name = (String) e.nextElement();
 113
 114         stringValues.put(name, aRequest.getParameter(name));
 115         List listValue = new Vector(Arrays.asList(aRequest.getParameterValues(name)));
 116         listValues.put(name, listValue);
 117       }
 118
 119       if (parts.size()>0 && ((String) parts.get(0)).trim().toLowerCase().equals(MULTIPART_FORMDATA_CONTENTTYPE)) {
 120         parseMultipartRequest();
 121       }
 122     }
 123     catch (Throwable t) {
 124       t.printStackTrace();
 125
 126       throw new UtilFailure(t);
 127     }
 128   }
 129
 130   protected void parseMultipartRequest() throws UtilExc, UtilFailure {
 131     try {
 132       FileUpload upload = new FileUpload();
 133
 134       upload.setSizeMax(maxUploadSize);
 135       upload.setSizeThreshold(4096);
 136       upload.setRepositoryPath(tempDir);
 137
 138       List items = upload.parseRequest(request);
 139
 140       Iterator i = items.iterator();
 141       while (i.hasNext()) {
 142         FileItem item = (FileItem) i.next();
 143
 144         if (item.isFormField()) {
 145           if (!stringValues.containsKey(item.getName())) {
 146             stringValues.put(item.getFieldName(), item.getString(encoding));
 147           }
 148
 149           List listValue = (List) listValues.get(item.getFieldName());
 150           if (listValue == null) {
 151             listValue = new Vector();
 152             listValues.put(item.getFieldName(), listValue);
 153           }
 154           listValue.add(item.getString(encoding));
 155         }
 156         else {
 157           if (item.getSize()>0)
 158             files.add(item);
 159         }
 160       }
 161     }
 162     catch (Throwable t) {
 163       throw new UtilFailure(t);
 164     }
 165   }
 166 }