Example #1
0
    /** @see javax.servlet.ServletRequest#getParameter(java.lang.String) */
    public String getParameter(String name) {
      Object o = map.get(name);
      if (!(o instanceof byte[]) && LazyList.size(o) > 0) o = LazyList.get(o, 0);

      if (o instanceof byte[]) {
        try {
          String s = new String((byte[]) o, encoding);
          return s;
        } catch (Exception e) {
          e.printStackTrace();
        }
      } else if (o != null) return String.valueOf(o);
      return null;
    }
Example #2
0
 /** @see javax.servlet.ServletRequest#getParameterValues(java.lang.String) */
 public String[] getParameterValues(String name) {
   List l = map.getValues(name);
   if (l == null || l.size() == 0) return new String[0];
   String[] v = new String[l.size()];
   for (int i = 0; i < l.size(); i++) {
     Object o = l.get(i);
     if (o instanceof byte[]) {
       try {
         v[i] = new String((byte[]) o, encoding);
       } catch (Exception e) {
         e.printStackTrace();
       }
     } else if (o instanceof String) v[i] = (String) o;
   }
   return v;
 }
Example #3
0
  /**
   * @see javax.servlet.Filter#doFilter(javax.servlet.ServletRequest, javax.servlet.ServletResponse,
   *     javax.servlet.FilterChain)
   */
  public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
      throws IOException, ServletException {
    HttpServletRequest srequest = (HttpServletRequest) request;
    if (srequest.getContentType() == null
        || !srequest.getContentType().startsWith("multipart/form-data")) {
      chain.doFilter(request, response);
      return;
    }

    BufferedInputStream in = new BufferedInputStream(request.getInputStream());
    String content_type = srequest.getContentType();

    // TODO - handle encodings

    String boundary = "--" + value(content_type.substring(content_type.indexOf("boundary=")));
    byte[] byteBoundary = (boundary + "--").getBytes(StringUtil.__ISO_8859_1);

    MultiMap params = new MultiMap();
    for (Iterator i = request.getParameterMap().entrySet().iterator(); i.hasNext(); ) {
      Map.Entry entry = (Map.Entry) i.next();
      Object value = entry.getValue();
      if (value instanceof String[]) params.addValues(entry.getKey(), (String[]) value);
      else params.add(entry.getKey(), value);
    }

    try {
      // Get first boundary
      byte[] bytes = TypeUtil.readLine(in);
      String line = bytes == null ? null : new String(bytes, "UTF-8");
      if (line == null || !line.equals(boundary)) {
        throw new IOException("Missing initial multi part boundary");
      }

      // Read each part
      boolean lastPart = false;
      String content_disposition = null;
      while (!lastPart) {
        while (true) {
          bytes = TypeUtil.readLine(in);
          // If blank line, end of part headers
          if (bytes == null || bytes.length == 0) break;
          line = new String(bytes, "UTF-8");

          // place part header key and value in map
          int c = line.indexOf(':', 0);
          if (c > 0) {
            String key = line.substring(0, c).trim().toLowerCase();
            String value = line.substring(c + 1, line.length()).trim();
            if (key.equals("content-disposition")) content_disposition = value;
          }
        }
        // Extract content-disposition
        boolean form_data = false;
        if (content_disposition == null) {
          throw new IOException("Missing content-disposition");
        }

        StringTokenizer tok = new StringTokenizer(content_disposition, ";");
        String name = null;
        String filename = null;
        while (tok.hasMoreTokens()) {
          String t = tok.nextToken().trim();
          String tl = t.toLowerCase();
          if (t.startsWith("form-data")) form_data = true;
          else if (tl.startsWith("name=")) name = value(t);
          else if (tl.startsWith("filename=")) filename = value(t);
        }

        // Check disposition
        if (!form_data) {
          continue;
        }

        // It is valid for reset and submit buttons to have an empty name.
        // If no name is supplied, the browser skips sending the info for that field.
        // However, if you supply the empty string as the name, the browser sends the
        // field, with name as the empty string. So, only continue this loop if we
        // have not yet seen a name field.
        if (name == null) {
          continue;
        }

        OutputStream out = null;
        File file = null;
        try {
          if (filename != null && filename.length() > 0) {
            file = File.createTempFile("MultiPart", "", tempdir);
            out = new FileOutputStream(file);
            if (_fileOutputBuffer > 0) out = new BufferedOutputStream(out, _fileOutputBuffer);
            request.setAttribute(name, file);
            params.add(name, filename);

            if (_deleteFiles) {
              file.deleteOnExit();
              ArrayList files = (ArrayList) request.getAttribute(FILES);
              if (files == null) {
                files = new ArrayList();
                request.setAttribute(FILES, files);
              }
              files.add(file);
            }

          } else out = new ByteArrayOutputStream();

          int state = -2;
          int c;
          boolean cr = false;
          boolean lf = false;

          // loop for all lines`
          while (true) {
            int b = 0;
            while ((c = (state != -2) ? state : in.read()) != -1) {
              state = -2;
              // look for CR and/or LF
              if (c == 13 || c == 10) {
                if (c == 13) state = in.read();
                break;
              }
              // look for boundary
              if (b >= 0 && b < byteBoundary.length && c == byteBoundary[b]) b++;
              else {
                // this is not a boundary
                if (cr) out.write(13);
                if (lf) out.write(10);
                cr = lf = false;
                if (b > 0) out.write(byteBoundary, 0, b);
                b = -1;
                out.write(c);
              }
            }
            // check partial boundary
            if ((b > 0 && b < byteBoundary.length - 2) || (b == byteBoundary.length - 1)) {
              if (cr) out.write(13);
              if (lf) out.write(10);
              cr = lf = false;
              out.write(byteBoundary, 0, b);
              b = -1;
            }
            // boundary match
            if (b > 0 || c == -1) {
              if (b == byteBoundary.length) lastPart = true;
              if (state == 10) state = -2;
              break;
            }
            // handle CR LF
            if (cr) out.write(13);
            if (lf) out.write(10);
            cr = (c == 13);
            lf = (c == 10 || state == 10);
            if (state == 10) state = -2;
          }
        } finally {
          out.close();
        }

        if (file == null) {
          bytes = ((ByteArrayOutputStream) out).toByteArray();
          params.add(name, bytes);
        }
      }

      // handle request
      chain.doFilter(new Wrapper(srequest, params), response);
    } finally {
      deleteFiles(request);
    }
  }
Example #4
0
 /** @see javax.servlet.ServletRequest#getParameterNames() */
 public Enumeration getParameterNames() {
   return Collections.enumeration(map.keySet());
 }
Example #5
0
 /** @see javax.servlet.ServletRequest#getParameterMap() */
 public Map getParameterMap() {
   return Collections.unmodifiableMap(map.toStringArrayMap());
 }