Пример #1
0
 /**
  * Convert a single line into XML
  *
  * @param text
  */
 public void convertStringToJMRequest(String text, TestElement el) {
   ((HTTPSamplerBase) el).parseArguments(text);
 }
  /**
   * Compute sampler informations from Request Header
   *
   * @param sampler {@link HTTPSamplerBase}
   * @param request {@link HttpRequestHdr}
   * @throws Exception
   */
  protected void computeFromPostBody(HTTPSamplerBase sampler, HttpRequestHdr request)
      throws Exception {
    // If it was a HTTP GET request, then all parameters in the URL
    // has been handled by the sampler.setPath above, so we just need
    // to do parse the rest of the request if it is not a GET request
    if ((!HTTPConstants.CONNECT.equals(request.getMethod()))
        && (!HTTPConstants.GET.equals(request.getMethod()))) {
      // Check if it was a multipart http post request
      final String contentType = request.getContentType();
      MultipartUrlConfig urlConfig = request.getMultipartConfig(contentType);
      String contentEncoding = sampler.getContentEncoding();
      // Get the post data using the content encoding of the request
      String postData = null;
      if (log.isDebugEnabled()) {
        if (!StringUtils.isEmpty(contentEncoding)) {
          log.debug("Using encoding " + contentEncoding + " for request body");
        } else {
          log.debug("No encoding found, using JRE default encoding for request body");
        }
      }

      if (!StringUtils.isEmpty(contentEncoding)) {
        postData = new String(request.getRawPostData(), contentEncoding);
      } else {
        // Use default encoding
        postData = new String(request.getRawPostData(), PostWriter.ENCODING);
      }

      if (urlConfig != null) {
        urlConfig.parseArguments(postData);
        // Tell the sampler to do a multipart post
        sampler.setDoMultipartPost(true);
        // Remove the header for content-type and content-length, since
        // those values will most likely be incorrect when the sampler
        // performs the multipart request, because the boundary string
        // will change
        request.getHeaderManager().removeHeaderNamed(HttpRequestHdr.CONTENT_TYPE);
        request.getHeaderManager().removeHeaderNamed(HttpRequestHdr.CONTENT_LENGTH);

        // Set the form data
        sampler.setArguments(urlConfig.getArguments());
        // Set the file uploads
        sampler.setHTTPFiles(urlConfig.getHTTPFileArgs().asArray());
        // used when postData is pure xml (eg. an xml-rpc call) or for PUT
      } else if (postData.trim().startsWith("<?")
          || HTTPConstants.PUT.equals(sampler.getMethod())
          || isPotentialXml(postData)) {
        sampler.addNonEncodedArgument("", postData, "");
      } else if (contentType == null
          || (contentType.startsWith(HTTPConstants.APPLICATION_X_WWW_FORM_URLENCODED)
              && !isBinaryContent(contentType))) {
        // It is the most common post request, with parameter name and values
        // We also assume this if no content type is present, to be most backwards compatible,
        // but maybe we should only parse arguments if the content type is as expected
        sampler.parseArguments(postData.trim(), contentEncoding); // standard name=value postData
      } else if (postData.length() > 0) {
        if (isBinaryContent(contentType)) {
          try {
            File tempDir = new File(getBinaryDirectory());
            File out = File.createTempFile(request.getMethod(), getBinaryFileSuffix(), tempDir);
            FileUtils.writeByteArrayToFile(out, request.getRawPostData());
            HTTPFileArg[] files = {new HTTPFileArg(out.getPath(), "", contentType)};
            sampler.setHTTPFiles(files);
          } catch (IOException e) {
            log.warn("Could not create binary file: " + e);
          }
        } else {
          // Just put the whole postbody as the value of a parameter
          sampler.addNonEncodedArgument(
              "", postData, ""); // used when postData is pure xml (ex. an xml-rpc call)
        }
      }
    }
  }