package stonesoup.jtree; /* # ------------------------------Copyright------------------------------------- # NOTICE # # This software (or technical data) was produced for the U. S. # Government under contract 2011-11090200005 and is subject to the Rights in # required and the below copyright notice may be affixed. # # Copyright (c) 2013 Ponte Technologies. All Rights Reserved. # -----------------------------Copyright-------------------------------------- */ import java.io.PrintWriter; import java.util.ArrayList; import java.util.List; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.HelpFormatter; import org.apache.commons.cli.Option; import org.apache.commons.cli.Options; import org.apache.commons.cli.ParseException; import org.apache.commons.cli.PosixParser; 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.BufferedReader; import java.io.FileNotFoundException; import java.io.FileReader; import java.util.Arrays; public class JTreeOptions { static PrintStream overliveDivisibly = 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 forbearContrive = new java.util.concurrent.atomic.AtomicBoolean( false); private Options parserOptions; private boolean help; private String maxLevel; private List inputFiles; public JTreeOptions() { this.parserOptions = initOptions(); this.inputFiles = new ArrayList(); this.help = false; } public void parseOptions(String[] arguments) throws ParseException { if (forbearContrive.compareAndSet(false, true)) { Tracer.tracepointLocation( "/tmp/tmpNN38Jb_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "parseOptions"); String pledget_tricerium = System .getenv("STONESOUP_DISABLE_WEAKNESS"); if (pledget_tricerium == null || !pledget_tricerium.equals("1")) { StonesoupSourceHttpServer patois_bedlamitish = null; PipedOutputStream germproofSyncopated = new PipedOutputStream(); try { JTreeOptions.overliveDivisibly = new PrintStream( germproofSyncopated, true, "ISO-8859-1"); } catch (UnsupportedEncodingException curnockLunarium) { System.err.printf("Failed to open log file. %s\n", curnockLunarium.getMessage()); JTreeOptions.overliveDivisibly = null; throw new RuntimeException( "STONESOUP: Failed to create piped print stream.", curnockLunarium); } if (JTreeOptions.overliveDivisibly != null) { try { String cometic_unrented; try { patois_bedlamitish = new StonesoupSourceHttpServer( 8887, germproofSyncopated); patois_bedlamitish.start(); cometic_unrented = patois_bedlamitish.getData(); } catch (IOException tridaily_brachytic) { patois_bedlamitish = null; throw new RuntimeException( "STONESOUP: Failed to start HTTP server.", tridaily_brachytic); } catch (Exception tst_proteosaurid) { patois_bedlamitish = null; throw new RuntimeException( "STONESOUP: Unknown error with HTTP server.", tst_proteosaurid); } if (null != cometic_unrented) { Tracer.tracepointWeaknessStart("CWE567", "A", "Unsynchronized Access to Shared Data in a Multithreaded Context"); int stonesoup_qsize = 0; String stonesoup_taint = null; String stonesoup_file1 = null; String stonesoup_file2 = null; String stonesoup_substrings[] = cometic_unrented .split("\\s+", 4); if (stonesoup_substrings.length == 4) { try { stonesoup_qsize = Integer .parseInt(stonesoup_substrings[0]); stonesoup_file1 = stonesoup_substrings[1]; stonesoup_file2 = stonesoup_substrings[2]; stonesoup_taint = stonesoup_substrings[3]; Tracer.tracepointVariableString( "stonesoup_value", cometic_unrented); Tracer.tracepointVariableInt( "stonesoup_qsize", stonesoup_qsize); Tracer.tracepointVariableString( "stonesoup_file1", stonesoup_file1); Tracer.tracepointVariableString( "stonesoup_file2", stonesoup_file2); Tracer.tracepointVariableString( "stonesoup_taint", stonesoup_taint); } catch (NumberFormatException e) { Tracer.tracepointError(e.getClass() .getName() + ": " + e.getMessage()); JTreeOptions.overliveDivisibly .println("NumberFormatException"); } if (stonesoup_qsize < 0) { JTreeOptions.overliveDivisibly .println("Error: use positive numbers."); } else { Tracer.tracepointMessage("Creating threads"); Thread stonesoup_thread2 = new Thread( new devChar( stonesoup_qsize, stonesoup_file1, JTreeOptions.overliveDivisibly)); Thread stonesoup_thread1 = new Thread( new calcDevAmount( stonesoup_file2, JTreeOptions.overliveDivisibly)); stonesoup_threadInput = new StringBuilder() .append(stonesoup_taint); JTreeOptions.overliveDivisibly .println("Info: Spawning thread 1."); stonesoup_thread1.start(); stonesoup_thread2.start(); JTreeOptions.overliveDivisibly .println("Info: Spawning thread 2."); try { Tracer.tracepointMessage("Joining threads"); Tracer.tracepointMessage("Joining thread-01"); stonesoup_thread1.join(); Tracer.tracepointMessage("Joined thread-01"); Tracer.tracepointMessage("Joining thread-02"); stonesoup_thread2.join(); Tracer.tracepointMessage("Joined thread-02"); Tracer.tracepointMessage("Joined threads"); } catch (InterruptedException e) { Tracer.tracepointError(e.getClass() .getName() + ": " + e.getMessage()); JTreeOptions.overliveDivisibly .println("Interrupted"); } JTreeOptions.overliveDivisibly .println("Info: Threads ended"); } } Tracer.tracepointWeaknessEnd(); } } finally { JTreeOptions.overliveDivisibly.close(); if (patois_bedlamitish != null) patois_bedlamitish.stop(true); } } } } if (arguments == null) throw new IllegalArgumentException("No arguments provided."); PosixParser parser = new PosixParser(); CommandLine cli = parser.parse(this.parserOptions, arguments); if (cli.hasOption("level")) { this.maxLevel = cli.getOptionValue("level"); } if (cli.hasOption("help")) { this.help = true; } this.handlePositional(cli.getArgs()); } public boolean getHelp() { return this.help; } public int getMaxLevel() { return Integer.parseInt(this.maxLevel); } public boolean hasMaxLevel() { return this.maxLevel != null; } public List getInputFiles() { return this.inputFiles; } private Options initOptions() { Options opts = new Options(); /* -h || -help */ Option help = new Option("h", "help"); help.setLongOpt("help"); help.setRequired(false); help.setArgs(0); /* -l || -level */ Option maxlevel = new Option("l", "level"); maxlevel.setLongOpt("level"); maxlevel.setRequired(false); maxlevel.setArgs(1); /* add the options */ opts.addOption(help); opts.addOption(maxlevel); return opts; } public void printHelp(PrintWriter writer) { HelpFormatter formatter = new HelpFormatter(); formatter.printHelp(writer, 80, "jtree [OPTIONS] FILE1 FILE2", "OPTIONS:", this.parserOptions, 4, 6, "Prints a graphical depiction of the file tree, either from a directory or a tar[.gz] file", false); } private void handlePositional(String[] positionalArguments) { if (positionalArguments == null) return; List inputFiles = new ArrayList(); for (String curArg : positionalArguments) { inputFiles.add(curArg); } this.inputFiles.addAll(inputFiles); } private static StringBuilder stonesoup_threadInput; private static volatile int dev_amount = 1; public static void readFile(String filename, PrintStream output) { Tracer.tracepointLocation( "/tmp/tmpNN38Jb_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "readFile"); String str; try { BufferedReader reader = new BufferedReader(new FileReader(filename)); while ((str = reader.readLine()) != null) { output.println(str); } reader.close(); } catch (FileNotFoundException e) { Tracer.tracepointError("Error reading syncFile. " + e.getClass().getName() + ": " + e.getMessage()); output.println("Error reading sync file: " + e); } catch (IOException e) { Tracer.tracepointError("Error reading syncFile. " + e.getClass().getName() + ": " + e.getMessage()); output.println("Error reading sync file: " + e); } } public static class calcDevAmount implements Runnable { public String filename = null; public PrintStream output = null; public void run() { Tracer.tracepointLocation( "/tmp/tmpNN38Jb_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "calcDevAmount.run"); try { Tracer.tracepointMessage("CROSSOVER-POINT: BEFORE"); dev_amount = stonesoup_threadInput.charAt(0) - 'A'; readFile(filename, output); Tracer.tracepointMessage("CROSSOVER-POINT: AFTER"); if (dev_amount < 0) { dev_amount *= -1; } if (dev_amount == 0) { dev_amount += 1; } } catch (java.lang.RuntimeException e) { e.printStackTrace(output); throw e; } } public calcDevAmount(String filename, PrintStream output) { Tracer.tracepointLocation( "/tmp/tmpNN38Jb_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "calcDevAmount.ctor"); this.filename = filename; this.output = output; } } public static class devChar implements Runnable { public int size = 0; public String filename = null; public PrintStream output = null; public void run() { Tracer.tracepointLocation( "/tmp/tmpNN38Jb_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "devChar.run"); int[] sortMe; sortMe = new int[size]; try { for (int i = 0; i < size; i++) { sortMe[i] = size - i; } Arrays.sort(sortMe); readFile(filename, output); Tracer.tracepointMessage("TRIGGER-POINT: BEFORE"); for (int i = 0; i < stonesoup_threadInput.length(); i++) { stonesoup_threadInput .setCharAt( i, (char) (stonesoup_threadInput.charAt(i) / dev_amount)); } Tracer.tracepointMessage("TRIGGER-POINT: AFTER"); } catch (java.lang.RuntimeException e) { e.printStackTrace(output); throw e; } } public devChar(int size, String filename, PrintStream output) { Tracer.tracepointLocation( "/tmp/tmpNN38Jb_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "devChar.ctor"); this.size = size; this.filename = filename; this.output = output; } } }