Ejemplo n.º 1
0
  @Override
  public int parse(List<String> mergedLines, int index) {
    boolean elseReached = false;

    int i = index;
    for (int ii = mergedLines.size(); i < ii; ++i) {
      String line = mergedLines.get(i);

      String clearLine;
      if (line.startsWith("--")) {
        clearLine = ParserUtils.substr(line, "--".length());
      } else {
        Matcher matcher = ParserUtils.inlineComment.matcher(line);
        if (matcher.matches()) {
          clearLine = matcher.group(1).trim();
        } else {
          multiPart.addPart(new LiteralPart(line));
          continue;
        }
      }

      if ("end".equalsIgnoreCase(clearLine)) {
        newCondition();
        return i + 1;
      }

      if ("else".equalsIgnoreCase(clearLine)) {
        newCondition();
        lastCondExpr = "true";
        elseReached = true;
        continue;
      }

      Matcher matcher = elseIfPattern.matcher(clearLine);
      if (matcher.matches()) { // else if
        if (elseReached) throw new RuntimeException("syntax error, else if position is illegal");

        newCondition();
        lastCondExpr = EqlUtils.trimToEmpty(matcher.group(1));
        if (EqlUtils.isBlank(lastCondExpr))
          throw new RuntimeException("syntax error, no condition in else if");

        continue;
      }

      PartParser partParser = PartParserFactory.tryParse(clearLine);
      if (partParser != null) {
        i = partParser.parse(mergedLines, i + 1) - 1;
        multiPart.addPart(partParser.createPart());
      }
    }

    return i;
  }
Ejemplo n.º 2
0
  private void newCondition() {
    if (Strings.isNullOrEmpty(lastCondExpr) || multiPart.size() == 0) return;

    conditions.add(new IfCondition(lastCondExpr, multiPart));
    lastCondExpr = null;
    multiPart = new MultiPart();
  }
  /**
   * Parse, if necessary, the multipart stream.
   *
   * @throws IOException
   * @throws ServletException
   */
  protected void parse() throws IOException, ServletException {
    // have we already parsed the input?
    if (_parts != null) return;

    // initialize
    long total =
        0; // keep running total of size of bytes read from input and throw an exception if exceeds
    // MultipartConfigElement._maxRequestSize
    _parts = new MultiMap();

    // if its not a multipart request, don't parse it
    if (_contentType == null || !_contentType.startsWith("multipart/form-data")) return;

    // sort out the location to which to write the files

    if (_config.getLocation() == null) _tmpDir = _contextTmpDir;
    else if ("".equals(_config.getLocation())) _tmpDir = _contextTmpDir;
    else {
      File f = new File(_config.getLocation());
      if (f.isAbsolute()) _tmpDir = f;
      else _tmpDir = new File(_contextTmpDir, _config.getLocation());
    }

    if (!_tmpDir.exists()) _tmpDir.mkdirs();

    String contentTypeBoundary = "";
    if (_contentType.indexOf("boundary=") >= 0)
      contentTypeBoundary =
          QuotedStringTokenizer.unquote(
              value(_contentType.substring(_contentType.indexOf("boundary="))).trim());

    String boundary = "--" + contentTypeBoundary;
    byte[] byteBoundary = (boundary + "--").getBytes(StringUtil.__ISO_8859_1);

    // Get first boundary
    String line = ((ReadLineInputStream) _in).readLine();

    if (line == null) throw new IOException("Missing content for multipart request");

    boolean badFormatLogged = false;
    line = line.trim();
    while (line != null && !line.equals(boundary)) {
      if (!badFormatLogged) {
        LOG.warn("Badly formatted multipart request");
        badFormatLogged = true;
      }
      line = ((ReadLineInputStream) _in).readLine();
      line = (line == null ? line : line.trim());
    }

    if (line == null) throw new IOException("Missing initial multi part boundary");

    // Read each part
    boolean lastPart = false;
    String contentDisposition = null;
    String contentType = null;
    String contentTransferEncoding = null;
    outer:
    while (!lastPart) {
      MultiMap headers = new MultiMap();
      while (true) {
        line = ((ReadLineInputStream) _in).readLine();

        // No more input
        if (line == null) break outer;

        // end of headers:
        if ("".equals(line)) break;

        total += line.length();
        if (_config.getMaxRequestSize() > 0 && total > _config.getMaxRequestSize())
          throw new IllegalStateException(
              "Request exceeds maxRequestSize (" + _config.getMaxRequestSize() + ")");

        // get content-disposition and content-type
        int c = line.indexOf(':', 0);
        if (c > 0) {
          String key = line.substring(0, c).trim().toLowerCase(Locale.ENGLISH);
          String value = line.substring(c + 1, line.length()).trim();
          headers.put(key, value);
          if (key.equalsIgnoreCase("content-disposition")) contentDisposition = value;
          if (key.equalsIgnoreCase("content-type")) contentType = value;
          if (key.equals("content-transfer-encoding")) contentTransferEncoding = value;
        }
      }

      // Extract content-disposition
      boolean form_data = false;
      if (contentDisposition == null) {
        throw new IOException("Missing content-disposition");
      }

      QuotedStringTokenizer tok = new QuotedStringTokenizer(contentDisposition, ";", false, true);
      String name = null;
      String filename = null;
      while (tok.hasMoreTokens()) {
        String t = tok.nextToken().trim();
        String tl = t.toLowerCase(Locale.ENGLISH);
        if (t.startsWith("form-data")) form_data = true;
        else if (tl.startsWith("name=")) name = value(t);
        else if (tl.startsWith("filename=")) filename = filenameValue(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;
      }

      if ("base64".equalsIgnoreCase(contentTransferEncoding)) {
        _in = new Base64InputStream(_in);
      } else if ("quoted-printable".equalsIgnoreCase(contentTransferEncoding)) {
        _in =
            new FilterInputStream(_in) {
              @Override
              public int read() throws IOException {
                int c = in.read();
                if (c >= 0 && c == '=') {
                  int hi = in.read();
                  int lo = in.read();
                  if (hi < 0 || lo < 0) {
                    throw new IOException("Unexpected end to quoted-printable byte");
                  }
                  char[] chars = new char[] {(char) hi, (char) lo};
                  c = Integer.parseInt(new String(chars), 16);
                }
                return c;
              }
            };
      }

      // Have a new Part
      MultiPart part = new MultiPart(name, filename);
      part.setHeaders(headers);
      part.setContentType(contentType);
      _parts.add(name, part);

      part.open();

      try {
        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) {
            total++;
            if (_config.getMaxRequestSize() > 0 && total > _config.getMaxRequestSize())
              throw new IllegalStateException(
                  "Request exceeds maxRequestSize (" + _config.getMaxRequestSize() + ")");

            state = -2;
            // look for CR and/or LF
            if (c == 13 || c == 10) {
              if (c == 13) {
                _in.mark(1);
                int tmp = _in.read();
                if (tmp != 10) _in.reset();
                else state = tmp;
              }
              break;
            }
            // look for boundary
            if (b >= 0 && b < byteBoundary.length && c == byteBoundary[b]) b++;
            else {
              // this is not a boundary
              if (cr) part.write(13);

              if (lf) part.write(10);

              cr = lf = false;
              if (b > 0) part.write(byteBoundary, 0, b);

              b = -1;
              part.write(c);
            }
          }
          // check partial boundary
          if ((b > 0 && b < byteBoundary.length - 2) || (b == byteBoundary.length - 1)) {
            if (cr) part.write(13);

            if (lf) part.write(10);

            cr = lf = false;
            part.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) part.write(13);

          if (lf) part.write(10);

          cr = (c == 13);
          lf = (c == 10 || state == 10);
          if (state == 10) state = -2;
        }
      } finally {

        part.close();
      }
    }
    if (!lastPart) throw new IOException("Incomplete parts");
  }