コード例 #1
0
  /**
   * Parse a single part
   *
   * @param ts
   * @throws java.io.IOException
   * @throws org.apache.cocoon.servlet.multipart.MultipartException
   */
  private void parsePart(DSpaceTokenStream ts) throws IOException, MultipartException {

    Hashtable headers = new Hashtable();
    headers = readHeaders(ts);
    try {
      if (headers.containsKey("filename")) {
        if (!"".equals(headers.get("filename"))) {
          parseFilePart(ts, headers);
        } else {
          // IE6 sends an empty part with filename="" for
          // empty upload fields. Just parse away the part
          byte[] buf = new byte[32];
          while (ts.getState() == DSpaceTokenStream.STATE_READING) ts.read(buf);
        }
      } else if (((String) headers.get("content-disposition")).toLowerCase().equals("form-data")) {
        parseInlinePart(ts, headers);
      }

      // FIXME: multipart/mixed parts are untested.
      else if (((String) headers.get("content-disposition")).toLowerCase().indexOf("multipart")
          > -1) {
        parseMultiPart(
            new DSpaceTokenStream(ts, MAX_BOUNDARY_SIZE), "--" + (String) headers.get("boundary"));
        ts.read(); // read past boundary
      } else {
        throw new MultipartException("Unknown part type");
      }
    } catch (IOException e) {
      throw new MultipartException("Malformed stream: " + e.getMessage());
    } catch (NullPointerException e) {
      e.printStackTrace();
      throw new MultipartException("Malformed header");
    }
  }
コード例 #2
0
  /**
   * Read string until newline or end of stream
   *
   * @param in
   * @throws java.io.IOException
   */
  private String readln(DSpaceTokenStream in) throws IOException {

    ByteArrayOutputStream bos = new ByteArrayOutputStream();

    int b = in.read();

    while ((b != -1) && (b != '\r')) {
      bos.write(b);
      b = in.read();
    }

    if (b == '\r') {
      in.read(); // read '\n'
    }

    return new String(bos.toByteArray(), this.characterEncoding);
  }
コード例 #3
0
  /**
   * Parse a multipart block
   *
   * @param ts
   * @param boundary
   * @throws java.io.IOException
   * @throws org.apache.cocoon.servlet.multipart.MultipartException
   */
  private void parseMultiPart(DSpaceTokenStream ts, String boundary)
      throws IOException, MultipartException {

    ts.setBoundary(boundary.getBytes());
    ts.read(); // read first boundary away
    ts.setBoundary(("\r\n" + boundary).getBytes());

    while (ts.getState() == DSpaceTokenStream.STATE_NEXTPART) {
      ts.nextPart();
      parsePart(ts);
    }

    if (ts.getState() != DSpaceTokenStream.STATE_ENDMULTIPART) { // sanity check
      throw new MultipartException("Malformed stream");
    }
  }
コード例 #4
0
  /**
   * Parse an inline part
   *
   * @param in
   * @param headers
   * @throws java.io.IOException
   */
  private void parseInlinePart(DSpaceTokenStream in, Hashtable headers) throws IOException {

    // Buffer incoming bytes for proper string decoding (there can be multibyte chars)
    ByteArrayOutputStream bos = new ByteArrayOutputStream();

    while (in.getState() == DSpaceTokenStream.STATE_READING) {
      int c = in.read();
      if (c != -1) bos.write(c);
    }

    String field = (String) headers.get("name");
    Vector v = (Vector) this.parts.get(field);

    if (v == null) {
      v = new Vector();
      this.parts.put(field, v);
    }

    v.add(new String(bos.toByteArray(), this.characterEncoding));
  }
コード例 #5
0
  /**
   * Parse a file part
   *
   * @param in
   * @param headers
   * @throws java.io.IOException
   * @throws org.apache.cocoon.servlet.multipart.MultipartException
   */
  private void parseFilePart(DSpaceTokenStream in, Hashtable headers)
      throws IOException, MultipartException {

    byte[] buf = new byte[FILE_BUFFER_SIZE];
    OutputStream out;
    File file = null;

    if (oversized) {
      out = new NullOutputStream();
    } else if (!saveUploadedFilesToDisk) {
      out = new ByteArrayOutputStream();
    } else {
      String fileName = (String) headers.get("filename");
      if (File.separatorChar == '\\') fileName = fileName.replace('/', '\\');
      else fileName = fileName.replace('\\', '/');

      String filePath = uploadDirectory.getPath() + File.separator;
      fileName = new File(fileName).getName();
      file = new File(filePath + fileName);

      if (!allowOverwrite && !file.createNewFile()) {
        if (silentlyRename) {
          int c = 0;
          do {
            file = new File(filePath + c++ + "_" + fileName);
          } while (!file.createNewFile());
        } else {
          throw new MultipartException(
              "Duplicate file '" + file.getName() + "' in '" + file.getParent() + "'");
        }
      }

      out = new FileOutputStream(file);
    }

    if (hasSession) { // upload widget support
      this.uploadStatus.put("finished", Boolean.FALSE);
      this.uploadStatus.put("started", Boolean.TRUE);
      this.uploadStatus.put("widget", headers.get("name"));
      this.uploadStatus.put("filename", headers.get("filename"));
    }

    int length = 0; // Track length for OversizedPart
    try {
      int read = 0;
      while (in.getState() == DSpaceTokenStream.STATE_READING) {
        // read data
        read = in.read(buf);
        length += read;
        out.write(buf, 0, read);

        if (this.hasSession) {
          this.uploadStatus.put(
              "sent", new Integer(((Integer) this.uploadStatus.get("sent")).intValue() + read));
        }
      }
      if (this.hasSession) { // upload widget support
        this.uploadStatus.put(
            "uploadsdone",
            new Integer(((Integer) this.uploadStatus.get("uploadsdone")).intValue() + 1));
        this.uploadStatus.put("error", Boolean.FALSE);
      }
    } catch (IOException ioe) {
      // don't let incomplete file uploads pile up in the upload dir.
      // this usually happens with aborted form submits containing very large files.
      out.close();
      out = null;
      if (file != null) file.delete();
      if (this.hasSession) { // upload widget support
        this.uploadStatus.put("error", Boolean.TRUE);
      }
      throw ioe;
    } finally {
      if (out != null) out.close();
    }

    String name = (String) headers.get("name");
    if (oversized) {
      this.parts.put(
          name, new RejectedPart(headers, length, this.contentLength, this.maxUploadSize));
    } else if (file == null) {
      byte[] bytes = ((ByteArrayOutputStream) out).toByteArray();
      this.parts.put(name, new PartInMemory(headers, bytes));
    } else {
      this.parts.put(name, new PartOnDisk(headers, file));
    }
  }