예제 #1
0
        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);
        }
예제 #2
0
        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();
                }
            }
        }