Ejemplo n.º 1
1
  private String readLine(ServletInputStream in) throws IOException {
    StringBuilder sbuf = new StringBuilder();
    int result;
    do {
      result = in.readLine(buf, 0, buf.length); // does +=
      if (result != -1) {
        sbuf.append(new String(buf, 0, result, encoding));
      }
    } while (result == buf.length); // loop only if the buffer was filled

    if (sbuf.length() == 0) {
      return null; // nothing read, must be at the end of stream
    }

    // Cut off the trailing \n or \r\n
    // It should always be \r\n but IE5 sometimes does just \n
    // Thanks to Luke Blaikie for helping make this work with \n
    int len = sbuf.length();
    if (len >= 2 && sbuf.charAt(len - 2) == '\r') {
      sbuf.setLength(len - 2); // cut \r\n
    } else if (len >= 1 && sbuf.charAt(len - 1) == '\n') {
      sbuf.setLength(len - 1); // cut \n
    }
    return sbuf.toString();
  }
Ejemplo n.º 2
0
  /*
   * protected method to get a job in JsonBean representation
   */
  @Override
  protected JsonBean getJob(HttpServletRequest request, HttpServletResponse response)
      throws XServletException, IOException, BaseEngineException {
    ServletInputStream is = request.getInputStream();
    byte[] b = new byte[101];
    while (is.readLine(b, 0, 100) != -1) {
      XLog.getLog(getClass()).warn("Printing :" + new String(b));
    }

    JsonBean jobBean = null;
    String jobId = getResourceName(request);
    if (jobId.endsWith("-B")) {
      jobBean = getBundleJob(request, response);
    } else {
      if (jobId.endsWith("-W")) {
        jobBean = getWorkflowJob(request, response);
      } else {
        if (jobId.contains("-W@")) {
          jobBean = getWorkflowAction(request, response);
        } else {
          if (jobId.contains("-C@")) {
            jobBean = getCoordinatorAction(request, response);
          } else {
            jobBean = getCoordinatorJob(request, response);
          }
        }
      }
    }

    return jobBean;
  }
  private Document getDocument(ServletInputStream inputStream, int contentLength) {
    int charCount = 0;
    byte[] byteBuffer = new byte[contentLength];

    while (true) {
      try {
        int readLength = inputStream.readLine(byteBuffer, charCount, 1024);
        // LogEvent.logFatal("IndicatorAggregationReportingServlet", String.valueOf(readLength), new
        // String(byteBuffer).trim());

        if (readLength == -1) {
          return DocumentHelper.parseText(new String(byteBuffer).trim());
        } else {
          charCount += readLength;
        }

      } catch (IOException e) {
        e.printStackTrace();
        return null;
      } catch (DocumentException de) {
        de.printStackTrace();
        return null;
      }
    }
  }
Ejemplo n.º 4
0
  /**
   * The method reads the next line from the servlet input stream into a byte array and returns a
   * String form of the array.Returns null if it reaches the end of the stream
   *
   * @return String The next line in the stream
   * @throws IOException - If there was a problem reading the stream or
   */
  private String readLine() throws IOException {
    int len = 0;
    String line = null;
    len = in.readLine(buff, 0, buff.length);
    if (len < 0) return null;
    line = new String(buff, 0, len, "ISO-8859-1");

    return line;
  }
Ejemplo n.º 5
0
 /** Convenience method to read HTTP header lines */
 private synchronized String getLine(ServletInputStream sis) throws IOException {
   byte b[] = new byte[1024];
   int read = sis.readLine(b, 0, b.length), index;
   String line = null;
   if (read != -1) {
     line = new String(b, 0, read);
     if ((index = line.indexOf('\n')) >= 0) line = line.substring(0, index - 1);
   }
   return line;
 }
Ejemplo n.º 6
0
  /**
   * Reads the file content from the stream and writes it to the local file system
   *
   * @throws IOException - If there was a problem reading the stream
   */
  private void writeToFile() throws IOException {
    // Open an o/p stream
    File tmpFile = File.createTempFile(prefix, "upload", new File(dir));
    tmpFile.deleteOnExit();
    log.debug("Setting: " + uploadFileArg + " to: " + tmpFile.getPath());
    map.put(uploadFileArg, tmpFile.getPath());
    // FileOutputStream out = new FileOutputStream (dir+File.separator+fileName);
    FileOutputStream out = new FileOutputStream(tmpFile);

    // this flag checks if \r\n needs to be written to the file
    // the servlet output stream appends these characters at the end of the
    // last line of the content, which should be skipped
    // so in the loop, all \r\n but for the last line are written to the file
    boolean writeCR = false;
    int len = 0;
    String line = null;
    map.put(paramName, fileName);

    // for each line
    while ((len = in.readLine(buff, 0, buff.length)) > -1) {
      line = new String(buff, 0, len);

      // if end of content, break
      if (line.startsWith(boundary)) break;
      if (writeCR) {
        writeCR = false;
        out.write('\r');
        out.write('\n');
      }
      if (len > 2 && buff[len - 2] == '\r' && buff[len - 1] == '\n') {
        writeCR = true;
        out.write(buff, 0, len - 2);
      } else {
        out.write(buff, 0, len);
      }
    }
    out.close();
  }
Ejemplo n.º 7
0
 /**
  * Obtain the byte array for this part of the request.
  *
  * <p>It may be necessary to modify this code to handle additional encoding types such as Base64.
  *
  * @param input Object from which data is read
  * @param thisPage Object containing information for this request
  * @param encoding type of character encoding to be used
  * @return Byte array for this part of request
  * @throws IOException if io errors
  */
 protected byte[] readPart(ServletInputStream input, ThisPage thisPage, String encoding)
     throws IOException {
   HttpServletRequest req = thisPage.getRequest();
   String boundary = extractBoundary(req);
   byte buffer[] = new byte[4096];
   int bytesRead = -1;
   ByteArrayOutputStream working = null;
   boolean pendingRN = false;
   working = new ByteArrayOutputStream();
   /*
    * Read body of part
    */
   while (true) {
     bytesRead = input.readLine(buffer, 0, buffer.length);
     if (bytesRead < 0) {
       thisPage.addMessage("readPart method - Section of form not properly ended");
       thisPage.errorMessage();
       return null;
     } else if (bytesRead == 0) {
       thisPage.addMessage("readPart method - Read yielded 0 characters");
       thisPage.errorMessage();
       return null;
     } else if (bytesRead == 1 && buffer[0] == '\n') {
       if (pendingRN) {
         working.write('\r');
         working.write('\n');
         pendingRN = false;
       }
       working.write(buffer[0]);
     } else if (new String(buffer, 0, bytesRead).equals("--" + boundary + "\r\n")) {
       if (!pendingRN) {
         thisPage.addMessage("readPart method - Boundary reached without preceding end of line");
         thisPage.errorMessage();
         return null;
       }
       if (working.size() == 0) {
         return null;
       }
       return working.toByteArray();
     } else if (new String(buffer, 0, bytesRead).equals("--" + boundary + "--\r\n")) {
       thisPage.setEndOfPacket(true);
       if (!pendingRN) {
         thisPage.addMessage(
             "readPart method - Final boundary reached with preceding end of line");
         thisPage.errorMessage();
         return null;
       }
       if (working.size() == 0) {
         return null;
       }
       return working.toByteArray();
     } else if (buffer[bytesRead - 2] == '\r' && buffer[bytesRead - 1] == '\n') {
       if (pendingRN) {
         working.write('\r');
         working.write('\n');
       }
       if (bytesRead > 2) {
         working.write(buffer, 0, bytesRead - 2);
       }
       pendingRN = true;
     } else if (buffer[bytesRead - 1] == '\r') {
       if (pendingRN) {
         working.write('\r');
         working.write('\n');
         pendingRN = false;
       }
       if (bytesRead > 1) {
         working.write(buffer, 0, bytesRead - 1);
       }
       int nextChar = input.read();
       if (nextChar == '\n') {
         pendingRN = true;
       } else {
         working.write('\r');
         working.write(nextChar);
       }
     } else {
       if (pendingRN) {
         working.write('\r');
         working.write('\n');
         pendingRN = false;
       }
       working.write(buffer, 0, bytesRead);
     }
   }
 }
Ejemplo n.º 8
0
  /**
   * Process the call to the servlet.
   *
   * <p>With regard to reading data from the request, a {@link javax.servlet.ServletInputStream}
   * object for binary data can be obtained by {@link HttpServletRequest#getInputStream()} or a
   * {@link java.io.BufferedReader} can be obtained for character data by using {@link
   * HttpServletRequest#getReader}.
   *
   * <p>With regard to writing data to the response a {@link javax.servlet.ServletOutputStream}
   * object for binary data can be obtained using {@link HttpServletResponse#getOutputStream()}
   * while a {@link java.io.PrintWriter} object can be obtained using {@link
   * HttpServletResponse#getWriter()}.
   *
   * <p>The {@link java.io.ByteArrayOutputStream} class can be used as a means of collecting the
   * bytes contained in the attached file.
   *
   * <p>It would be desirable to have tests for enctype and method.
   *
   * @param req Request object
   * @param res Response object
   * @throws IOException if io problems
   */
  public void service(HttpServletRequest req, HttpServletResponse res) throws IOException {
    ThisPage thisPage = new ThisPage(req, res, config);
    GenericPrinter output = thisPage.getPrinter();
    if (!req.getMethod().equalsIgnoreCase("post")) {
      output.println("<html><head>");
      output.println("<title>Must use POST method</title>");
      output.println("</head><body>");
      output.println("<h1>Must use POST method</h1>");
      output.println("<p>Request used " + req.getMethod() + " method</p>");
      output.println("<p>Must use POST method</p>");
      output.println("</body></html>");
      thisPage.sendContents();
      return;
    } else if (req.getHeader("content-type") == null) {
      output.println("<html><head>");
      output.println("<title>Missing content-type header</title>");
      output.println("</head><body>");
      output.println("<h1>Missing content-type header</h1>");
      output.println(
          "<p>Must use content-type header " + "to specify multipart/form-data encoding</p>");
      output.println("</body></html>");
      thisPage.sendContents();
      return;
    } else if (!req.getHeader("content-type").toLowerCase().startsWith("multipart/form-data")) {
      output.println("<html><head>");
      output.println("<title>Must use multipart/form-data encoding</title>");
      output.println("</head><body>");
      output.println("<h1>Must use multipart/form-data encoding</h1>");
      output.println("<p>content-type is " + req.getHeader("content-type") + " </p>");
      output.println("<p>Must use multipart/form-data encoding</p>");
      output.println("</body></html>");
      thisPage.sendContents();
      return;
    }
    String boundary = extractBoundary(req);
    if (boundary == null) {
      thisPage.addMessage("Unable to extract boundary value");
      thisPage.addMessage(req.getHeader("content-type"));
      thisPage.errorMessage();
      return;
    }
    int counter = 0;
    byte buffer[] = new byte[4096];
    byte extract[];
    int bytesRead = -1;
    ServletInputStream input = req.getInputStream();
    bytesRead = input.readLine(buffer, 0, buffer.length);
    if (!new String(buffer, 0, bytesRead, "ISO8859_1").startsWith("--" + boundary)) {
      thisPage.addMessage(
          "Should be separator "
              + "--"
              + boundary
              + " : "
              + " found "
              + new String(buffer, 0, bytesRead));
      thisPage.errorMessage();
      return;
    }
    while (true) {
      counter++;
      Contents part = new Contents();
      thisPage.addMessage("Starting part " + Integer.toString(counter) + " of form");
      while (true) {

        bytesRead = input.readLine(buffer, 0, buffer.length);
        if (bytesRead < 0) {
          thisPage.addMessage("Unexpected end of packet");
          thisPage.errorMessage();
          return;
        } else {
          String value = new String(buffer, 0, bytesRead, "ISO8859_1");
          if (value.endsWith("\r\n")) {
            value = stripEOL(value);
          }
          if (value.length() == 0) {
            extract = readPart(input, thisPage, part.getTransferEncoding());
            if (thisPage.getTerminateRequest()) {
              return;
            }
            part.setContents(extract);
            thisPage.addElement(part);
            break;
          } else {
            thisPage.addMessage("service method - Parsing line: " + value);
            part.parseLine(value);
          }
        }
      }
      if (thisPage.getEndOfPacket()) {
        break;
      }
    }
    if (thisPage.getTerminateRequest()) {
      return;
    }
    starter(thisPage);
    if (thisPage.getTerminateRequest()) {
      return;
    }
    processor(thisPage);
    if (thisPage.getTerminateRequest()) {
      return;
    }
    ender(thisPage);
    if (thisPage.getTerminateRequest()) {
      return;
    }
    if (thisPage.getRedirectAddress() != null) {
      thisPage
          .getResponse()
          .sendRedirect(thisPage.getResponse().encodeRedirectURL(thisPage.getRedirectAddress()));
    }
  }
Ejemplo n.º 9
0
 public Hashtable processData(
     ServletInputStream is, String boundary, String saveInDir, int clength)
     throws IllegalArgumentException, IOException {
   if (is == null) throw new IllegalArgumentException("InputStream");
   if (boundary == null || boundary.trim().length() < 1)
     throw new IllegalArgumentException("\"" + boundary + "\" is an illegal boundary indicator");
   boundary = "--" + boundary;
   StringTokenizer stLine = null, stFields = null;
   FileInfo fileInfo = null;
   Hashtable dataTable = new Hashtable(5);
   String line = null, field = null, paramName = null;
   boolean saveFiles = (saveInDir != null && saveInDir.trim().length() > 0);
   boolean isFile = false;
   if (saveFiles) { // Create the required directory (including parent dirs)
     File f = new File(saveInDir);
     f.mkdirs();
   }
   line = getLine(is);
   if (line == null || !line.startsWith(boundary))
     throw new IOException("Boundary not found; boundary = " + boundary + ", line = " + line);
   while (line != null) {
     if (line == null || !line.startsWith(boundary)) return dataTable;
     line = getLine(is);
     if (line == null) return dataTable;
     stLine = new StringTokenizer(line, ";\r\n");
     if (stLine.countTokens() < 2) throw new IllegalArgumentException("Bad data in second line");
     line = stLine.nextToken().toLowerCase();
     if (line.indexOf("form-data") < 0)
       throw new IllegalArgumentException("Bad data in second line");
     stFields = new StringTokenizer(stLine.nextToken(), "=\"");
     if (stFields.countTokens() < 2)
       throw new IllegalArgumentException("Bad data in second line");
     fileInfo = new FileInfo();
     stFields.nextToken();
     paramName = stFields.nextToken();
     isFile = false;
     if (stLine.hasMoreTokens()) {
       field = stLine.nextToken();
       stFields = new StringTokenizer(field, "=\"");
       if (stFields.countTokens() > 1) {
         if (stFields.nextToken().trim().equalsIgnoreCase("filename")) {
           fileInfo.name = paramName;
           String value = stFields.nextToken();
           if (value != null && value.trim().length() > 0) {
             fileInfo.clientFileName = value;
             isFile = true;
           } else {
             line = getLine(is); // Skip "Content-Type:" line
             line = getLine(is); // Skip blank line
             line = getLine(is); // Skip blank line
             line = getLine(is); // Position to boundary line
             continue;
           }
         }
       } else if (field.toLowerCase().indexOf("filename") >= 0) {
         line = getLine(is); // Skip "Content-Type:" line
         line = getLine(is); // Skip blank line
         line = getLine(is); // Skip blank line
         line = getLine(is); // Position to boundary line
         continue;
       }
     }
     boolean skipBlankLine = true;
     if (isFile) {
       line = getLine(is);
       if (line == null) return dataTable;
       if (line.trim().length() < 1) skipBlankLine = false;
       else {
         stLine = new StringTokenizer(line, ": ");
         if (stLine.countTokens() < 2)
           throw new IllegalArgumentException("Bad data in third line");
         stLine.nextToken(); // Content-Type
         fileInfo.fileContentType = stLine.nextToken();
       }
     }
     if (skipBlankLine) {
       line = getLine(is);
       if (line == null) return dataTable;
     }
     if (!isFile) {
       line = getLine(is);
       if (line == null) return dataTable;
       dataTable.put(paramName, line);
       // If parameter is dir, change saveInDir to dir
       if (paramName.equals("dir")) saveInDir = line;
       line = getLine(is);
       continue;
     }
     try {
       UplInfo uplInfo = new UplInfo(clength);
       UploadMonitor.set(fileInfo.clientFileName, uplInfo);
       OutputStream os = null;
       String path = null;
       if (saveFiles)
         os = new FileOutputStream(path = getFileName(saveInDir, fileInfo.clientFileName));
       else os = new ByteArrayOutputStream(ONE_MB);
       boolean readingContent = true;
       byte previousLine[] = new byte[2 * ONE_MB];
       byte temp[] = null;
       byte currentLine[] = new byte[2 * ONE_MB];
       int read, read3;
       if ((read = is.readLine(previousLine, 0, previousLine.length)) == -1) {
         line = null;
         break;
       }
       while (readingContent) {
         if ((read3 = is.readLine(currentLine, 0, currentLine.length)) == -1) {
           line = null;
           uplInfo.aborted = true;
           break;
         }
         if (compareBoundary(boundary, currentLine)) {
           os.write(previousLine, 0, read - 2);
           line = new String(currentLine, 0, read3);
           break;
         } else {
           os.write(previousLine, 0, read);
           uplInfo.currSize += read;
           temp = currentLine;
           currentLine = previousLine;
           previousLine = temp;
           read = read3;
         } // end else
       } // end while
       os.flush();
       os.close();
       if (!saveFiles) {
         ByteArrayOutputStream baos = (ByteArrayOutputStream) os;
         fileInfo.setFileContents(baos.toByteArray());
       } else fileInfo.file = new File(path);
       dataTable.put(paramName, fileInfo);
       uplInfo.currSize = uplInfo.totalSize;
     } // end try
     catch (IOException e) {
       throw e;
     }
   }
   return dataTable;
 }
Ejemplo n.º 10
0
    long readPart(String encoding, String boundary, ServletInputStream sis, ServletAccessDescr sad)
        throws IOException {
      long currentPos = 0;
      byte[] buff = new byte[BUFF_SIZE * 1024];
      int boundaryLen = boundary.length();
      if (boundaryLen >= (buff.length - 2))
        throw new IOException(
            "Boundary length exceeds allowed:" + boundaryLen + "/" + (buff.length - 2));
      String hl;
      int ss;
      // parse part headers
      do {
        int len = sis.readLine(buff, 0, buff.length);
        if (len < 0) throw new IOException("Trying reading parts after EOS");
        currentPos += len;
        if (buff[len - 1] != '\n') throw new IOException("Part header is over " + buff.length);
        hl = new String(buff, 0, len - 2, "US-ASCII");
        // System.err.println("h:" + hl);
        if (hl.length() == 0) break;
        ss = hl.indexOf(boundary);
        if (ss >= 0) {
          if (hl.indexOf("--", boundaryLen + ss) == boundaryLen + ss) return -currentPos; //
          continue; // 1st part starts with it
        }

        hl = new String(buff, 0, len - 2, encoding);
        ss = hl.indexOf(':');
        if (ss < 0)
          throw new IOException("Illegal multipart header, no value separator ':' in " + hl);
        String header = hl.substring(0, ss).toLowerCase();
        String value = hl.substring(ss + 1).trim();
        headers.put(header, value);
        if (header.equals("Content-Disposition".toLowerCase())) {
          // TODO add real parser as for cookies
          if (name != null)
            throw new IOException("Multiple header 'Content-Disposition' for the same part: " + hl);
          int ni = value.toLowerCase().indexOf("name=\"");
          if (ni < 0) throw new IOException("Part name is missed in 'Content-Disposition'" + hl);
          int eq = value.indexOf('"', ni + "name=\"".length());
          // TODO name can be encoded so right approach is decoding it here
          name = value.substring(ni + "name=\"".length(), eq);

          ni = value.toLowerCase().indexOf("filename=\"");
          if (ni >= 0) {
            eq = value.indexOf('"', ni + "filename=\"".length());
            filename = value.substring(ni + "filename=\"".length(), eq);
          }
        }
        // TODO Analyze content-type multipart/mixed defining new boundary, so they need to be
        // stacked
      } while (true);
      // TODO for content type multipart/mixed need another loop for another boundary
      // read part content
      fileSize = 0;
      boolean writingInMem = sad.multipartThreshold > 0;
      OutputStream wos = cos;
      int state = no_rn;

      try {
        read_part:
        do {
          int len = sis.readLine(buff, 0, buff.length);
          if (len < 0) throw new IOException("Unexpected end of multipart data at " + currentPos);
          // System.err.println("" + len + "==" + new String(buff, 0, len));

          switch (state) {
            case no_rn:
              if (buff[len - 1] == '\n') {
                if (len > 1)
                  if (buff[len - 2] == '\r') {
                    wos.write(buff, 0, len - 2);
                    fileSize += len - 2;
                    state = last_rn;
                    break;
                  }
              } else if (buff[len - 1] == '\r') {
                wos.write(buff, 0, len - 1);
                fileSize += len - 1;
                state = last_r;
                break;
              }
              wos.write(buff, 0, len);
              fileSize += len;
              break;
            case last_r:
              if (buff[len - 1] == '\n') {
                if (len == 1) {
                  state = last_rn;
                  break;
                }

                if (buff[len - 2] == '\r') {
                  wos.write('\r');
                  fileSize++;
                  wos.write(buff, 0, len - 2);
                  fileSize += len - 2;
                  state = last_rn;
                  break;
                }
              } else if (buff[len - 1] == '\r') {
                wos.write('\r');
                fileSize++;
                wos.write(buff, 0, len - 1);
                fileSize += len - 1;
                state = last_r;
                break;
              }
              wos.write('\r');
              fileSize++;
              wos.write(buff, 0, len);
              fileSize += len;
              state = no_rn;
              break;
            case last_rn:
              if (buff[len - 1] == '\n') {
                if (len > 1)
                  if (buff[len - 2] == '\r') {
                    hl = new String(buff, 0, len - 2, "US-ASCII");
                    ss = hl.indexOf(boundary);
                    if (ss >= 0) {
                      if (hl.indexOf("--", ss + boundaryLen) == (ss + boundaryLen))
                        return -currentPos - fileSize;
                      else break read_part;
                    } else {
                      wos.write('\r');
                      fileSize++;
                      wos.write('\n');
                      fileSize++;
                      wos.write(buff, 0, len - 2);
                      fileSize += len - 2;
                      break;
                    }
                  }
              } else if (buff[len - 1]
                  == '\r') { // can't be boundary limiter since buffer is bigger than boundary

              }
              wos.write('\r');
              fileSize++;
              wos.write('\n');
              fileSize++;
              wos.write(buff, 0, len);
              fileSize += len;
              state = no_rn;
              break;
            default:
              throw new IllegalStateException();
          }
          if (sad.multipartMaxFile > 0 && sad.multipartMaxFile <= fileSize)
            throw new IOException("File size exceeds limit of " + sad.multipartMaxFile);
          if (writingInMem && sad.multipartThreshold < fileSize) {
            if (sad.multipartLocation == null || sad.multipartLocation.getName().length() == 0)
              partFile = File.createTempFile(name, "multipart");
            else partFile = File.createTempFile(name, "multipart", sad.multipartLocation);
            partFile.deleteOnExit();
            wos = new FileOutputStream(partFile);
            wos.write(cos.toByteArray());
            cos.close();
            cos = null;
            writingInMem = false;
          }
        } while (true); // TODO add condition
      } finally {
        if (wos != null) { // can b e null ?
          try {
            wos.close();
          } catch (Exception e) {
          }
        }
      }
      return currentPos + fileSize;
    }