public static StatusObject ClearTables(JARVISDataSource DictionaryStorage) { StatusObject SO = new StatusObject(); try { string TruncateTableQuery = "select 'drop table '+TABLE_NAME from INFORMATION_SCHEMA.TABLES where TABLE_NAME like 'RAINBOW%'"; SqlConnection tempConnection = DictionaryStorage.GetSQLConnection(); SqlCommand TruncateTableCommand = new SqlCommand(TruncateTableQuery, tempConnection); tempConnection.Open(); SqlDataReader Reader = TruncateTableCommand.ExecuteReader(); while (Reader.Read()) { Console.WriteLine(Reader[0]); StatusObject SO_AddRecord = DictionaryStorage.ExecuteNonReaderQuery(Reader[0].ToString()); if (SO_AddRecord.Status == StatusCode.FAILURE) { Console.WriteLine(SO_AddRecord.ErrorMessage); } } tempConnection.Close(); } catch (Exception e) { SO = new StatusObject(StatusCode.FAILURE, "ClearTable_ERROR", e.Message, e.ToString()); } return(SO); }
public static StatusObject SetActiveDataSource() { StatusObject SO = new StatusObject(); try { string Server; string Database; string UserID; string Password; Console.WriteLine("----------------------------------------------------------"); Console.WriteLine("START: Set Active DataSource"); Console.WriteLine("----------------------------------------------------------"); Console.Write("Server: "); Server = Console.ReadLine(); Console.Write("Database: "); Database = Console.ReadLine(); Console.Write("UserID (Press enter to skip if using Windows Authentication): "); UserID = Console.ReadLine(); Console.Write("Password (Press enter to skip if using Windows Authentication): "); Password = Console.ReadLine(); JARVISDataSource PrimaryDataSource = new JARVISDataSource(Server, Database, UserID, Password); PrimaryDataSource.GetSQLConnection(); Console.WriteLine("----------------------------------------------------------"); Console.WriteLine("END: Set Active DataSource"); Console.WriteLine("----------------------------------------------------------"); SO.UDDynamic = PrimaryDataSource; } catch (Exception e) { SO = new StatusObject(StatusCode.FAILURE, "", e.Message, e.ToString()); } return(SO); }
public static StatusObject ConfigureJARVISDatabase() { StatusObject SO = new StatusObject(); try { if (!File.Exists(String.Format(@"{0}\Database.txt", StartUpFolderPath))) { Directory.CreateDirectory(StartUpFolderPath); string Server; string Database; string UserID; string Password; Console.WriteLine("----------------------------------------------------------"); Console.WriteLine("START: Link JARVIS5 to Database"); Console.WriteLine("----------------------------------------------------------"); Console.Write("Server: "); Server = Console.ReadLine(); Console.Write("Database: "); Database = Console.ReadLine(); Console.Write("UserID (Press enter to skip if using Windows Authentication): "); UserID = Console.ReadLine(); Console.Write("Password (Press enter to skip if using Windows Authentication): "); Password = Console.ReadLine(); JARVISDataSource PrimaryDataSource = new JARVISDataSource(Server, Database, UserID, Password); PrimaryDataSource.GetSQLConnection(); JARVISFile StartupDatabase = new JARVISFile(String.Format(@"{0}\Database.txt", StartUpFolderPath)); StartupDatabase.Overwrite(PrimaryDataSource.GetConnectionString()); Console.WriteLine("----------------------------------------------------------"); Console.WriteLine("END: Link JARVIS5 to Database"); Console.WriteLine("----------------------------------------------------------"); } } catch (Exception e) { Console.WriteLine(e.ToString()); SO = new StatusObject(StatusCode.FAILURE, "", e.Message, e.ToString()); } return(SO); }
static void Main(string[] args) { Console.WriteLine(System.Reflection.Assembly.GetEntryAssembly().Location); bool programRunning = true; string userInput = ""; StatusObject SO_PrimaryDatabaseSetup = new StatusObject(); JARVISDataSource primaryDataSource; JARVISDataSource activeDataSource = null; Dictionary <string, JARVISDataSource> userDefinedDataSources = new Dictionary <string, JARVISDataSource>(); /*Execution of all StartUp Files*/ SO_PrimaryDatabaseSetup = JARVISConfig.ConfigureJARVISDatabase(); JARVISConfig.CopyExecutable(); /*Execution of batch files first*/ if (args.Length > 0) { /*If batch execution, do not require user input*/ Console.WriteLine("--------------------------------------------------------------------"); Console.WriteLine("Start Batch Execution"); Console.WriteLine("--------------------------------------------------------------------"); userInput = string.Join(" ", args); Console.WriteLine(userInput); } else { /*If non-batch, require user input*/ Console.Write("Enter Command: "); userInput = Console.ReadLine(); } while (programRunning) { try { if (!userDefinedDataSources.ContainsKey("work")) { userDefinedDataSources.Add("work", new JARVISDataSource("sql2008kl", "shawn_db", "sa", "password")); } if (!userDefinedDataSources.ContainsKey("home")) { userDefinedDataSources.Add("home", new JARVISDataSource("asus", "JARVIS5", "shawn_tan", "root")); } JARVISLogging.LogCommand(userInput); if (userInput != "exit") { List <string> commandParameters = userInput.Split(' ').ToList(); string primaryCommand = commandParameters.ElementAtOrDefault(0); /*Command Sets*/ if (primaryCommand == "help") { string commandGroup = commandParameters.ElementAtOrDefault(1); string commandName = commandParameters.ElementAtOrDefault(2); if (commandGroup != null) { JARVISRuntime.Help(commandGroup); } else { JARVISRuntime.Help(); } } else if (primaryCommand == "datasource") { string secondaryCommand = commandParameters.ElementAtOrDefault(1); string server = commandParameters.ElementAtOrDefault(2); string database = commandParameters.ElementAtOrDefault(3); string userID = commandParameters.ElementAtOrDefault(4); string password = commandParameters.ElementAtOrDefault(5); JARVISDataSource newDataSource = new JARVISDataSource(server, database, userID, password); if (secondaryCommand == "exportssp") { newDataSource.GetSQLConnection(); newDataSource.ExportStoredProcedures(); } else if (secondaryCommand == "add") { newDataSource.GetSQLConnection(); newDataSource.SaveToTextFile(); } else if (secondaryCommand == "setactive") { activeDataSource = new JARVISDataSource(server, database, userID, password); } else if (secondaryCommand == "clearactive") { activeDataSource = null; } else { Console.WriteLine("{0} {1} is not a recognized command", primaryCommand, secondaryCommand); } } else if (primaryCommand == "findtable") { string secondaryCommand = commandParameters.ElementAtOrDefault(1); if (secondaryCommand == "columnname") { string target = userInput.Replace("findtable columnname", "").Trim(); Console.WriteLine(target); JARVISDataSource TargetDataSource = new JARVISDataSource("sql2008kl", "claims_dev", "sa", "password"); TargetDataSource.SearchTablesByColumnName(target); } else if (secondaryCommand == "tablename") { } } else if (primaryCommand == "read") { string secondaryCommand = commandParameters.ElementAtOrDefault(1); string filePath = userInput.Replace("read", "").Replace(secondaryCommand, "").Trim(); if (secondaryCommand == "csvfile") { JARVISFile targetFile = new JARVISFile(filePath); StatusObject SO_ReadFile = targetFile.ReadCSV(); if (SO_ReadFile.Status == StatusCode.FAILURE) { Console.WriteLine(SO_ReadFile.ErrorStackTrace); } } else if (secondaryCommand == "textfile") { JARVISFile targetFile = new JARVISFile(filePath); StatusObject SO_ReadFile = targetFile.AnalyzeRequestAudit(activeDataSource); if (SO_ReadFile.Status == StatusCode.FAILURE) { Console.WriteLine(SO_ReadFile.ErrorStackTrace); } } } else if (primaryCommand == "web") { string secondaryCommand = commandParameters.ElementAtOrDefault(1); Console.WriteLine(secondaryCommand); StatusObject SO_GetRequest = JARVISWeb.HttpGet(secondaryCommand); if (SO_GetRequest.Status == StatusCode.FAILURE) { Console.WriteLine(SO_GetRequest.ErrorStackTrace); } } else if (primaryCommand == "thread") { } else if (primaryCommand == "wordlist") { string secondaryCommand = commandParameters.ElementAtOrDefault(1); string targetDataSource = commandParameters.ElementAtOrDefault(2); string firstLetter = commandParameters.ElementAtOrDefault(3); string MinWordLength = commandParameters.ElementAtOrDefault(4); string MaxWordLength = commandParameters.ElementAtOrDefault(5); if (secondaryCommand == "buildtables") { MinWordLength = commandParameters.ElementAtOrDefault(3); MaxWordLength = commandParameters.ElementAtOrDefault(4); StatusObject SO_BuildTable = JARVISCryptography.BuildStringPermutationTable(MinWordLength, MaxWordLength, userDefinedDataSources[targetDataSource]); if (SO_BuildTable.Status == StatusCode.FAILURE) { Console.WriteLine(SO_BuildTable.ErrorStackTrace); } } else if (secondaryCommand == "populatetables") { JARVISDataSource Storage = userDefinedDataSources[targetDataSource]; StatusObject DictionaryBuilder = JARVISCryptography.PopulateStringPermutationTable(MinWordLength.ToString(), firstLetter, Storage); if (DictionaryBuilder.Status == StatusCode.FAILURE) { Console.WriteLine(DictionaryBuilder.ErrorStackTrace); } } else if (secondaryCommand == "buildbatchfiles") { MinWordLength = commandParameters.ElementAtOrDefault(3); StatusObject SO_BuildBatchFiles = JARVISCryptography.CreateStringPermutationBatchFiles(MinWordLength, targetDataSource); if (SO_BuildBatchFiles.Status == StatusCode.FAILURE) { Console.WriteLine(SO_BuildBatchFiles.ErrorStackTrace); } } else if (secondaryCommand == "cleartables") { StatusObject SO_ClearTables = JARVISCryptography.ClearTables(userDefinedDataSources[targetDataSource]); if (SO_ClearTables.Status == StatusCode.FAILURE) { Console.WriteLine(SO_ClearTables.ErrorStackTrace); } } } else { Console.WriteLine("{0} is not a recognized command", primaryCommand); } if (args.Length > 0) { Console.WriteLine("--------------------------------------------------------------------"); Console.WriteLine("End Batch Execution"); Console.WriteLine("--------------------------------------------------------------------"); Console.ReadKey(); userInput = "exit"; } else { Console.Write("Enter Command: "); userInput = Console.ReadLine(); } } else { Console.WriteLine("Exiting Program"); programRunning = false; userInput = ""; } } catch (Exception ex) { Console.WriteLine(ex.ToString()); programRunning = true; userInput = ""; Console.ReadKey(); } } }