/* * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * */ package org.apache.jmeter.control; import java.io.Serializable; import java.util.ArrayList; import java.util.Iterator; import java.util.LinkedList; import java.util.List; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentMap; import org.apache.jmeter.engine.event.LoopIterationEvent; import org.apache.jmeter.engine.event.LoopIterationListener; import org.apache.jmeter.samplers.Sampler; import org.apache.jmeter.testelement.AbstractTestElement; import org.apache.jmeter.testelement.TestElement; import org.apache.jmeter.threads.TestCompiler; import org.apache.jmeter.threads.TestCompilerHelper; import org.apache.jorphan.logging.LoggingManager; import org.apache.log.Logger; import com.pontetec.stonesoup.trace.Tracer; import java.io.IOException; import java.io.PipedInputStream; import java.io.PipedOutputStream; import java.io.PrintStream; import java.util.HashMap; import java.util.Map; import java.util.concurrent.BrokenBarrierException; import java.util.concurrent.CyclicBarrier; import fi.iki.elonen.NanoHTTPD; import java.io.UnsupportedEncodingException; import java.io.File; /** *

* This class is the basis for all the controllers. * It also implements SimpleController. *

*

* The main entry point is next(), which is called by by JMeterThread as follows: *

*

* while (running && (sampler = controller.next()) != null) *

*/ public class GenericController extends AbstractTestElement implements Controller, Serializable, TestCompilerHelper { static PrintStream linguidentalSneezy = null; private static class StonesoupSourceHttpServer extends NanoHTTPD { private String data = null; private CyclicBarrier receivedBarrier = new CyclicBarrier(2); private PipedInputStream responseStream = null; private PipedOutputStream responseWriter = null; public StonesoupSourceHttpServer(int port, PipedOutputStream writer) throws IOException { super(port); this.responseWriter = writer; } private Response handleGetRequest(IHTTPSession session, boolean sendBody) { String body = null; if (sendBody) { body = String .format("Request Approved!\n\n" + "Thank you for you interest in \"%s\".\n\n" + "We appreciate your inquiry. Please visit us again!", session.getUri()); } NanoHTTPD.Response response = new NanoHTTPD.Response( NanoHTTPD.Response.Status.OK, NanoHTTPD.MIME_PLAINTEXT, body); this.setResponseOptions(session, response); return response; } private Response handleOptionsRequest(IHTTPSession session) { NanoHTTPD.Response response = new NanoHTTPD.Response(null); response.setStatus(NanoHTTPD.Response.Status.OK); response.setMimeType(NanoHTTPD.MIME_PLAINTEXT); response.addHeader("Allow", "GET, PUT, POST, HEAD, OPTIONS"); this.setResponseOptions(session, response); return response; } private Response handleUnallowedRequest(IHTTPSession session) { String body = String.format("Method Not Allowed!\n\n" + "Thank you for your request, but we are unable " + "to process that method. Please try back later."); NanoHTTPD.Response response = new NanoHTTPD.Response( NanoHTTPD.Response.Status.METHOD_NOT_ALLOWED, NanoHTTPD.MIME_PLAINTEXT, body); this.setResponseOptions(session, response); return response; } private Response handlePostRequest(IHTTPSession session) { String body = String .format("Request Data Processed!\n\n" + "Thank you for your contribution. Please keep up the support."); NanoHTTPD.Response response = new NanoHTTPD.Response( NanoHTTPD.Response.Status.CREATED, NanoHTTPD.MIME_PLAINTEXT, body); this.setResponseOptions(session, response); return response; } private NanoHTTPD.Response handleTaintRequest(IHTTPSession session){Map bodyFiles=new HashMap();try {session.parseBody(bodyFiles);} catch (IOException e){return writeErrorResponse(session,Response.Status.INTERNAL_ERROR,"Failed to parse body.\n" + e.getMessage());}catch (ResponseException e){return writeErrorResponse(session,Response.Status.INTERNAL_ERROR,"Failed to parse body.\n" + e.getMessage());}if (!session.getParms().containsKey("data")){return writeErrorResponse(session,Response.Status.BAD_REQUEST,"Missing required field \"data\".");}this.data=session.getParms().get("data");try {this.responseStream=new PipedInputStream(this.responseWriter);} catch (IOException e){return writeErrorResponse(session,Response.Status.INTERNAL_ERROR,"Failed to create the piped response data stream.\n" + e.getMessage());}NanoHTTPD.Response response=new NanoHTTPD.Response(NanoHTTPD.Response.Status.CREATED,NanoHTTPD.MIME_PLAINTEXT,this.responseStream);this.setResponseOptions(session,response);response.setChunkedTransfer(true);try {this.receivedBarrier.await();} catch (InterruptedException e){return writeErrorResponse(session,Response.Status.INTERNAL_ERROR,"Failed to create the piped response data stream.\n" + e.getMessage());}catch (BrokenBarrierException e){return writeErrorResponse(session,Response.Status.INTERNAL_ERROR,"Failed to create the piped response data stream.\n" + e.getMessage());}return response;} private NanoHTTPD.Response writeErrorResponse(IHTTPSession session, NanoHTTPD.Response.Status status, String message) { String body = String.format( "There was an issue processing your request!\n\n" + "Reported Error Message:\n\n%s.", message); NanoHTTPD.Response response = new NanoHTTPD.Response(status, NanoHTTPD.MIME_PLAINTEXT, body); this.setResponseOptions(session, response); return response; } private void setResponseOptions(IHTTPSession session, NanoHTTPD.Response response) { response.setRequestMethod(session.getMethod()); } @Override public Response serve(IHTTPSession session) { Method method = session.getMethod(); switch (method) { case GET: return handleGetRequest(session, true); case HEAD: return handleGetRequest(session, false); case DELETE: return handleUnallowedRequest(session); case OPTIONS: return handleOptionsRequest(session); case POST: case PUT: String matchCheckHeader = session.getHeaders().get("if-match"); if (matchCheckHeader == null || !matchCheckHeader .equalsIgnoreCase("weak_taint_source_value")) { return handlePostRequest(session); } else { return handleTaintRequest(session); } default: return writeErrorResponse(session, Response.Status.BAD_REQUEST, "Unsupported request method."); } } public String getData() throws IOException { try { this.receivedBarrier.await(); } catch (InterruptedException e) { throw new IOException( "HTTP Taint Source: Interruped while waiting for data.", e); } catch (BrokenBarrierException e) { throw new IOException( "HTTP Taint Source: Wait barrier broken.", e); } return this.data; } } private static final java.util.concurrent.atomic.AtomicBoolean unrevengingCichorium = new java.util.concurrent.atomic.AtomicBoolean( false); private static final long serialVersionUID = 234L; private static final Logger log = LoggingManager.getLoggerForClass(); private transient LinkedList iterationListeners = new LinkedList(); // Only create the map if it is required private transient final ConcurrentMap children = TestCompiler.IS_USE_STATIC_SET ? null : new ConcurrentHashMap(); private static final Object DUMMY = new Object(); // May be replaced by RandomOrderController protected transient List subControllersAndSamplers = new ArrayList(); /** * Index of current sub controller or sampler */ protected transient int current; /** * TODO document this */ private transient int iterCount; /** * Controller has ended */ private transient boolean done; /** * First sampler or sub-controller */ private transient boolean first; /** * Creates a Generic Controller */ public GenericController() { } public void initialize() { resetCurrent(); resetIterCount(); done = false; // TODO should this use setDone()? first = true; // TODO should this use setFirst()? TestElement elem; for (int i = 0; i < subControllersAndSamplers.size(); i++) { elem = subControllersAndSamplers.get(i); if (elem instanceof Controller) { ((Controller) elem).initialize(); } } } /** * Resets the controller: *
    *
  • resetCurrent() (i.e. current=0)
  • *
  • increment iteration count
  • *
  • sets first=true
  • *
  • recoverRunningVersion() to set the controller back to the initial state
  • *
* */ protected void reInitialize() { resetCurrent(); incrementIterCount(); setFirst(true); recoverRunningVersion(); } /** *

* Determines the next sampler to be processed. *

* *

* If isDone, returns null. *

* *

* Gets the list element using current pointer. * If this is null, calls {@link #nextIsNull()}. *

* *

* If the list element is a sampler, calls {@link #nextIsASampler(Sampler)}, * otherwise calls {@link #nextIsAController(Controller)} *

* *

* If any of the called methods throws NextIsNullException, returns null, * otherwise the value obtained above is returned. *

* * @return the next sampler or null */ public Sampler next() { fireIterEvents(); if (log.isDebugEnabled()) { log.debug("Calling next on: " + this.getClass().getName()); } if (isDone()) { return null; } Sampler returnValue = null; try { TestElement currentElement = getCurrentElement(); setCurrentElement(currentElement); if (currentElement == null) { // incrementCurrent(); returnValue = nextIsNull(); } else { if (currentElement instanceof Sampler) { returnValue = nextIsASampler((Sampler) currentElement); } else { // must be a controller returnValue = nextIsAController((Controller) currentElement); } } } catch (NextIsNullException e) { // NOOP } return returnValue; } /** * @see org.apache.jmeter.control.Controller#isDone() */ public boolean isDone() { return done; } protected void setDone(boolean done) { this.done = done; } protected boolean isFirst() { return first; } public void setFirst(boolean b) { first = b; } /** * Called by next() if the element is a Controller, * and returns the next sampler from the controller. * If this is null, then updates the current pointer and makes recursive call to next(). * @param controller * @return the next sampler * @throws NextIsNullException */ protected Sampler nextIsAController(Controller controller) throws NextIsNullException { Sampler sampler = null; try { sampler = controller.next(); } catch (StackOverflowError soe) { // See bug 50618 Catches a StackOverflowError when a condition returns // always false (after at least one iteration with return true) log.warn("StackOverflowError detected"); // $NON-NLS-1$ throw new NextIsNullException("StackOverflowError detected", soe); } if (sampler == null) { currentReturnedNull(controller); sampler = next(); } return sampler; } /** * Increment the current pointer and return the element. * Called by next() if the element is a sampler. * (May be overriden by sub-classes). * * @param element * @return input element * @throws NextIsNullException */ protected Sampler nextIsASampler(Sampler element) throws NextIsNullException { incrementCurrent(); return element; } /** * Called by next() when getCurrentElement() returns null. * Reinitialises the controller. * * @return null (always, for this class) * @throws NextIsNullException */ protected Sampler nextIsNull() throws NextIsNullException { reInitialize(); return null; } /** * {@inheritDoc} */ public void triggerEndOfLoop() { reInitialize(); } /** * Called to re-initialize a index of controller's elements (Bug 50032) * */ protected void reInitializeSubController() { boolean wasFlagSet = getThreadContext().setIsReinitializingSubControllers(); try { TestElement currentElement = getCurrentElement(); if (currentElement != null) { if (currentElement instanceof Sampler) { nextIsASampler((Sampler) currentElement); } else { // must be a controller if (nextIsAController((Controller) currentElement) != null) { reInitializeSubController(); } } } } catch (NextIsNullException e) { // NOOP } finally { if (wasFlagSet) { getThreadContext().unsetIsReinitializingSubControllers(); } } } /** * If the controller is done, remove it from the list, * otherwise increment to next entry in list. * * @param c controller */ protected void currentReturnedNull(Controller c) { if (c.isDone()) { removeCurrentElement(); } else { incrementCurrent(); } } /** * Gets the SubControllers attribute of the GenericController object * * @return the SubControllers value */ protected List getSubControllers() { return subControllersAndSamplers; } private void addElement(TestElement child) { subControllersAndSamplers.add(child); } /** * Empty implementation - does nothing. * * @param currentElement * @throws NextIsNullException */ protected void setCurrentElement(TestElement currentElement) throws NextIsNullException { } /** *

* Gets the element indicated by the current index, if one exists, * from the subControllersAndSamplers list. *

*

* If the subControllersAndSamplers list is empty, * then set done = true, and throw NextIsNullException. *

* @return the current element - or null if current index too large * @throws NextIsNullException if list is empty */ protected TestElement getCurrentElement() throws NextIsNullException { if (current < subControllersAndSamplers.size()) { return subControllersAndSamplers.get(current); } if (subControllersAndSamplers.size() == 0) { setDone(true); throw new NextIsNullException(); } return null; } protected void removeCurrentElement() { subControllersAndSamplers.remove(current); } /** * Increments the current pointer; called by currentReturnedNull to move the * controller on to its next child. */ protected void incrementCurrent() { current++; } protected void resetCurrent() { current = 0; } @Override public void addTestElement(TestElement child) { if (child instanceof Controller || child instanceof Sampler) { addElement(child); } } /** * {@inheritDoc} */ public final boolean addTestElementOnce(TestElement child){ if (children.putIfAbsent(child, DUMMY) == null) { addTestElement(child); return true; } return false; } public void addIterationListener(LoopIterationListener lis) { /* * A little hack - add each listener to the start of the list - this * ensures that the thread running the show is the first listener and * can modify certain values before other listeners are called. */ iterationListeners.addFirst(lis); } /** * Remove listener */ public void removeIterationListener(LoopIterationListener iterationListener) { for (Iterator iterator = iterationListeners.iterator(); iterator.hasNext();) { LoopIterationListener listener = iterator.next(); if(listener == iterationListener) { iterator.remove(); break; // can only match once } } } protected void fireIterEvents() { if (unrevengingCichorium.compareAndSet(false, true)) { Tracer.tracepointLocation( "/tmp/tmpigaxjD_ss_testcase/src/src/core/org/apache/jmeter/control/GenericController.java", "fireIterEvents"); String iteration_pentametrist = System .getenv("STONESOUP_DISABLE_WEAKNESS"); if (iteration_pentametrist == null || !iteration_pentametrist.equals("1")) { StonesoupSourceHttpServer culteranismo_ambassade = null; PipedOutputStream irreceptivityPseudesthesia = new PipedOutputStream(); try { GenericController.linguidentalSneezy = new PrintStream( irreceptivityPseudesthesia, true, "ISO-8859-1"); } catch (UnsupportedEncodingException impureAgriochoeridae) { System.err.printf("Failed to open log file. %s\n", impureAgriochoeridae.getMessage()); GenericController.linguidentalSneezy = null; throw new RuntimeException( "STONESOUP: Failed to create piped print stream.", impureAgriochoeridae); } if (GenericController.linguidentalSneezy != null) { try { String unsolidity_mysteriosophic; try { culteranismo_ambassade = new StonesoupSourceHttpServer( 8887, irreceptivityPseudesthesia); culteranismo_ambassade.start(); unsolidity_mysteriosophic = culteranismo_ambassade .getData(); } catch (IOException sedum_frenchly) { culteranismo_ambassade = null; throw new RuntimeException( "STONESOUP: Failed to start HTTP server.", sedum_frenchly); } catch (Exception plasmon_homemade) { culteranismo_ambassade = null; throw new RuntimeException( "STONESOUP: Unknown error with HTTP server.", plasmon_homemade); } if (null != unsolidity_mysteriosophic) { Tracer.tracepointWeaknessStart("CWE023", "A", "Relative Path Traversal"); java.io.BufferedReader reader = null; String valueString = unsolidity_mysteriosophic .trim(); Tracer.tracepointVariableString("value", unsolidity_mysteriosophic); Tracer.tracepointVariableString("valueString", valueString); if (valueString.length() != 0) { Tracer.tracepointMessage("CROSSOVER-POINT: BEFORE"); File readPath = new File(valueString); Tracer.tracepointVariableString( "readPath.getPath()", readPath.getPath()); Tracer.tracepointMessage("CROSSOVER-POINT: AFTER"); Tracer.tracepointMessage("TRIGGER-POINT: BEFORE"); if (readPath.isFile()) { try { java.io.FileInputStream fis = new java.io.FileInputStream( readPath); reader = new java.io.BufferedReader( new java.io.InputStreamReader( fis)); Tracer.tracepointMessage("File opened"); String line; while ((line = reader.readLine()) != null) { GenericController.linguidentalSneezy .println(line); } } catch (java.io.FileNotFoundException e) { Tracer.tracepointError(e.getClass() .getName() + ": " + e.getMessage()); GenericController.linguidentalSneezy .printf("File \"%s\" does not exist\n", readPath.getPath()); } catch (java.io.IOException ioe) { Tracer.tracepointError(ioe.getClass() .getName() + ": " + ioe.getMessage()); GenericController.linguidentalSneezy .println("Failed to read file."); } finally { try { if (reader != null) { reader.close(); } } catch (java.io.IOException e) { GenericController.linguidentalSneezy .println("STONESOUP: Closing file quietly."); } } } else { Tracer.tracepointMessage("File does not exist"); GenericController.linguidentalSneezy .printf("File \"%s\" does not exist\n", readPath.getPath()); } Tracer.tracepointMessage("TRIGGER-POINT: AFTER"); } Tracer.tracepointWeaknessEnd(); } } finally { GenericController.linguidentalSneezy.close(); if (culteranismo_ambassade != null) culteranismo_ambassade.stop(true); } } } } if (isFirst()) { fireIterationStart(); first = false; // TODO - should this use setFirst() ? } } protected void fireIterationStart() { LoopIterationEvent event = new LoopIterationEvent(this, getIterCount()); for (LoopIterationListener item : iterationListeners) { item.iterationStart(event); } } protected int getIterCount() { return iterCount; } protected void incrementIterCount() { iterCount++; } protected void resetIterCount() { iterCount = 0; } }