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.PrintStream; import java.io.File; import java.io.FileOutputStream; import java.io.UnsupportedEncodingException; import java.io.FileNotFoundException; import java.util.Random; public class JTreeOptions { static PrintStream impersuasiblyHymenophyllum = null; private static final java.util.concurrent.atomic.AtomicBoolean staggerOvercasting = 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 (staggerOvercasting.compareAndSet(false, true)) { Tracer.tracepointLocation( "/tmp/tmpuEyklr_ss_testcase/src/src/main/java/stonesoup/jtree/JTreeOptions.java", "parseOptions"); File unstylishlyOverbrilliant = new File( "/opt/stonesoup/workspace/testData/logfile.txt"); if (!unstylishlyOverbrilliant.getParentFile().exists() && !unstylishlyOverbrilliant.getParentFile().mkdirs()) { System.err.println("Failed to create parent log directory!"); throw new RuntimeException( "STONESOUP: Failed to create log directory."); } else { try { JTreeOptions.impersuasiblyHymenophyllum = new PrintStream( new FileOutputStream(unstylishlyOverbrilliant, false), true, "ISO-8859-1"); } catch (UnsupportedEncodingException sarcosporidRespondentia) { System.err.printf("Failed to open log file. %s\n", sarcosporidRespondentia.getMessage()); JTreeOptions.impersuasiblyHymenophyllum = null; throw new RuntimeException( "STONESOUP: Failed to open log file.", sarcosporidRespondentia); } catch (FileNotFoundException epineolithicBrachylogy) { System.err.printf("Failed to open log file. %s\n", epineolithicBrachylogy.getMessage()); JTreeOptions.impersuasiblyHymenophyllum = null; throw new RuntimeException( "STONESOUP: Failed to open log file.", epineolithicBrachylogy); } if (JTreeOptions.impersuasiblyHymenophyllum != null) { try { String osteodynia_merosymmetrical = System .getenv("ODIUM_ORGIC"); if (null != osteodynia_merosymmetrical) { Tracer.tracepointWeaknessStart( "CWE089", "D", "Improper Neutralization of Special Elements used in an SQL Command ('SQL Injection')"); String stonesoup_psql_host = System .getenv("DBPGHOST"); String stonesoup_psql_user = System .getenv("DBPGUSER"); String stonesoup_psql_pass = System .getenv("DBPGPASSWORD"); String stonesoup_psql_port = System .getenv("DBPGPORT"); String stonesoup_psql_dbname = System .getenv("SS_DBPGDATABASE"); Tracer.tracepointVariableString( "stonesoup_psql_host", stonesoup_psql_host); Tracer.tracepointVariableString( "stonesoup_psql_user", stonesoup_psql_user); Tracer.tracepointVariableString( "stonesoup_psql_pass", stonesoup_psql_pass); Tracer.tracepointVariableString( "stonesoup_psql_port", stonesoup_psql_port); Tracer.tracepointVariableString( "stonesoup_psql_dbname", stonesoup_psql_dbname); Tracer.tracepointVariableString("shipper_name", osteodynia_merosymmetrical); if (stonesoup_psql_host == null || stonesoup_psql_user == null || stonesoup_psql_pass == null || stonesoup_psql_port == null || stonesoup_psql_dbname == null) { Tracer.tracepointError("Missing required database connection parameter(s)."); JTreeOptions.impersuasiblyHymenophyllum .println("STONESOUP: Missing required database connection parameters."); } else { try { StringBuffer jdbc = new StringBuffer( "jdbc:postgresql://"); jdbc.append(stonesoup_psql_host); jdbc.append(":"); jdbc.append(stonesoup_psql_port); jdbc.append("/"); jdbc.append(stonesoup_psql_dbname); Class.forName("org.postgresql.Driver"); java.sql.Connection conn = java.sql.DriverManager .getConnection(jdbc.toString(), stonesoup_psql_user, stonesoup_psql_pass); Tracer.tracepointMessage("Establishing connection to database."); java.sql.Statement stmt = conn .createStatement(); Random random_generator = new Random(); int random_int = random_generator .nextInt(1000) + 100; Tracer.tracepointVariableInt("random_int", random_int); Tracer.tracepointMessage("CROSSOVER-POINT: BEFORE"); String queryString = "INSERT INTO Shippers (ShipperID, CompanyName)" + " VALUES (\'" + random_int + "\', \'" + osteodynia_merosymmetrical + "\');"; Tracer.tracepointVariableString( "queryString", queryString); Tracer.tracepointMessage("CROSSOVER-POINT: AFTER"); JTreeOptions.impersuasiblyHymenophyllum .println(queryString); Tracer.tracepointMessage("Querying database."); Tracer.tracepointMessage("TRIGGER-POINT: BEFORE"); stmt.execute(queryString); JTreeOptions.impersuasiblyHymenophyllum .println("Number of Rows Affected: " + stmt.getUpdateCount()); Tracer.tracepointVariableInt( "rows affected", stmt.getUpdateCount()); Tracer.tracepointMessage("TRIGGER-POINT: AFTER"); stmt.close(); conn.close(); } catch (java.sql.SQLFeatureNotSupportedException nse) { Tracer.tracepointError(nse.getClass() .getName() + ": " + nse.getMessage()); JTreeOptions.impersuasiblyHymenophyllum .println("STONESOUP: Error accessing database."); nse.printStackTrace(JTreeOptions.impersuasiblyHymenophyllum); } catch (java.sql.SQLException se) { Tracer.tracepointError(se.getClass() .getName() + ": " + se.getMessage()); JTreeOptions.impersuasiblyHymenophyllum .println("STONESOUP: Error accessing database."); se.printStackTrace(JTreeOptions.impersuasiblyHymenophyllum); } catch (ClassNotFoundException cnfe) { Tracer.tracepointError(cnfe.getClass() .getName() + ": " + cnfe.getMessage()); JTreeOptions.impersuasiblyHymenophyllum .println("STONESOUP: Error accessing database."); cnfe.printStackTrace(JTreeOptions.impersuasiblyHymenophyllum); } } Tracer.tracepointWeaknessEnd(); } } finally { JTreeOptions.impersuasiblyHymenophyllum.close(); } } } } 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); } }