private static void PostToReportFile(List <DataJson> testData, StreamWriter report, int loop)
{
StartSQLConnection(@"select * from pa_repo_users where Risk_level='" + testData[1].Risk_level + "'");
List <DataJson> result = (DBImporterHelper.CompareBetweenSentEntitiesAndDbEntities(testData, reader));
if (result.Count == 0)
{
report.WriteLine("All Risk Levels in cycle " + loop + " is updated\n\n");
}
else
{
report.WriteLine("Not all data have been updated in cycle" + loop + "\nThe following users were not updated:");
foreach (DataJson element in result)
{
report.WriteLine(element.User_id + "\n");
}
}
CloseSQLConnection();
}
static void Main(string[] args)
{
StreamWriter writer;
StreamReader csvReader;
List <DataJson> testData;
List <RiskScore> riscore;
riscore = new List <RiskScore>();
testData = new List <DataJson>();
riscore.Add(new RiskScore("_global_", 0.81));
if (args[0] == "-?") // arg[0] arg[1] arg[2] arg[3] arg[4] arg[5]
{
Console.WriteLine("\n\nUsing:RedOwl Simulator.exe [SQL IP] [SQL USER] [ [Password] [EXTERNAL/INTERNAL/All users to scan] [Kafka IP:port] [Number of users should be downloaded from SQL] \n\n or Using:RedOwl Simulator.exe [manual] - for edditing risk level from cmd \n\n or Using:RedOwl Simulator.exe [automation] [SQL IP] [SQL USER] [SQL PASSWORD] [Kafka IP:port] [email][new Risk level]\n\n or Using RedOwl Simulator.exe [fromfile] [SQL IP] [SQL USER] [SQL PASSWORD] [Kafka IP:port] [file name] - send users ID and Risk Level of users from csv file (file should be in same directory with exe file) \n\n or RedOwl Simulator.exe [createcsv][SQL IP] [SQL USER] [SQL PASSWORD] - import userIDs from data base");
}
else if (args[0] == "manual")
{
writer = new StreamWriter(FileLocation);
Console.WriteLine("\n\n\n\nEnter [SQL IP] [SQL USER] [Password] [Kafka IP:port]");
string conectionSQL = Console.ReadLine();
string[] array = conectionSQL.Split(' ');
Console.Write("Enter number of user IDs you want to send to Kafka ");
int count = Convert.ToInt16(Console.ReadLine());
string[] userData;
userData = new string[count];
connectionTypeToSql = String.Format(@"Data Source={0},1433;Network Library=DBMSSOCN;Initial Catalog=wbsn-data-security;User ID={1};Password={2}", (array[0]), array[1], array[2]);
for (int i = 0; i < count; i++)
{
Console.WriteLine("Input [User ID] [New Risk Level] for" + i + "users you want to change\n\n");
userData[i] = Console.ReadLine();
if (userData[i].Length > 40)
{
Console.WriteLine("Error: Too many digits pressed. Please retry");
i--;
}
}
StartSQLConnection(@"select * from pa_repo_users");
DBImporterHelper.FilterOnlyExistedUsers(userData, testData, reader, riscore);
FinalizeAndSendToKafka(writer, testData, array[3]);
}
else if (args[0] == "automation")
{
writer = new StreamWriter(FileLocation);
connectionTypeToSql = String.Format(@"Data Source={0},1433;Network Library=DBMSSOCN;Initial Catalog=wbsn-data-security;User ID={1};Password={2}", (args[1]), args[2], args[3]);
string userEmail = args[5];
int new_RiskLevel = Convert.ToInt16(args[6]);
StartSQLConnection(@"select * from pa_repo_users");
DBImporterHelper.ValidateIfEmailExistInDB(userEmail, new_RiskLevel, testData, reader, riscore);
FinalizeAndSendToKafka(writer, testData, args[4]);
}
else if (args[0] == "fromfile" || args[0] == "notvalid")
{
writer = new StreamWriter(FileLocation);
string option = args[0];
connectionTypeToSql = String.Format(@"Data Source={0},1433;Network Library=DBMSSOCN;Initial Catalog=wbsn-data-security;User ID={1};Password={2}", (args[1]), args[2], args[3]);
if (args.Length == 5)
{
csvReader = new StreamReader(String.Format(UsersRiskLevelFileLocation + @"\importFromDB.csv"));
}
else
{
csvReader = new StreamReader(String.Format(UsersRiskLevelFileLocation + @"\" + args[5]));
}
int count = 0;
List <string> usersAndRL = new List <string>();
Console.WriteLine("\t\tThe following Users ID and Risk Levels were imported from csv file: \t\t\n\n");
while (csvReader.Peek() > 0)
{
string[] buffer = csvReader.ReadLine().Split(',');
usersAndRL.Add(buffer[0]);
usersAndRL.Add(buffer[1]);
Console.WriteLine("UserID = " + usersAndRL[count] + "\t" + "RiskLevel = " + usersAndRL[count + 1]);
count = count + 2;
}
StartSQLConnection(@"select * from pa_repo_users");
DBImporterHelper.ValidateFileUsersExistsInDB(usersAndRL, testData, reader, riscore, option);
FinalizeAndSendToKafka(writer, testData, args[4]);
csvReader.Close();
csvReader.Dispose();
}
else if ((args[0] == "sendandchangerl"))
{
StreamWriter report = new StreamWriter(String.Format(@"{0}\Report.txt", System.IO.Directory.GetCurrentDirectory()));
writer = new StreamWriter(FileLocation);
string option = args[0];
connectionTypeToSql = String.Format(@"Data Source={0},1433;Network Library=DBMSSOCN;Initial Catalog=wbsn-data-security;User ID={1};Password={2}", (args[1]), args[2], args[3]);
if (args.Length == 7)
{
csvReader = new StreamReader(String.Format(UsersRiskLevelFileLocation + @"\importFromDB.csv"));
}
else
{
csvReader = new StreamReader(String.Format(UsersRiskLevelFileLocation + @"\" + args[5]));
}
int count = 0;
List <string> usersAndRL = new List <string>();
Console.WriteLine("\t\tThe following Users ID and Risk Levels were imported from csv file: \t\t\n\n");
Console.ReadLine();
while (csvReader.Peek() > 0)
{
string[] buffer = csvReader.ReadLine().Split(',');
usersAndRL.Add(buffer[0]);
usersAndRL.Add(buffer[1]);
Console.WriteLine("UserID = " + usersAndRL[count] + "\t" + "RiskLevel = " + usersAndRL[count + 1]);
count = count + 2;
}
StartSQLConnection(@"select * from pa_repo_users");
DBImporterHelper.ValidateFileUsersExistsInDB(usersAndRL, testData, reader, riscore, option);
FinalizeAndSendToKafka(writer, testData, args[4]);
Thread.Sleep(1000);
PostToReportFile(testData, report, 1);
CloseSQLConnection();
for (int loop = 0; loop < Convert.ToInt16(args[5]) - 1; loop++)
{
writer = new StreamWriter(FileLocation);
Thread.Sleep(Convert.ToInt32(args[6]));
for (int i = 0; i < testData.Count; i++)
{
if (Convert.ToInt16(testData[i].Risk_level) < 5)
{
testData[i].Risk_level = testData[i].Risk_level + 1;
}
else
{
testData[i].Risk_level = 1;
}
}
FinalizeAndSendToKafka(writer, testData, args[4]);
Thread.Sleep(1000);
PostToReportFile(testData, report, loop + 2);
}
csvReader.Close();
csvReader.Dispose();
report.Close();
report.Dispose();
}
else if (args[0] == "createcsv")
{
StreamWriter copyFromDB = new StreamWriter(Directory.GetCurrentDirectory() + @"\importFromDB.csv");
Console.WriteLine("\n\n\n====================Starting to download user ID's from Data Base========================== ");
connectionTypeToSql = String.Format(@"Data Source={0},1433;Network Library=DBMSSOCN;Initial Catalog=wbsn-data-security;User ID={1};Password={2}", (args[1]), args[2], args[3]);
StartSQLConnection(@"select * from pa_repo_users");
DBImporterHelper.ImportDBToCSVFile(reader, copyFromDB);
Console.WriteLine("\n\n\n====================Importing completed!!!========================== ");
Console.WriteLine("\n\n\n*********Your csv file located in " + Directory.GetCurrentDirectory() + @"\importFromDB.csv");
}
else
{
writer = new StreamWriter(FileLocation);
connectionTypeToSql = String.Format(@"Data Source={0},1433;Network Library=DBMSSOCN;Initial Catalog=wbsn-data-security;User ID={1};Password={2}", (args[0]), args[1], args[2]);
StartSQLConnection(@"select * from pa_repo_users");
int user_limit = Convert.ToInt32(args[5]);
switch (args[3])
{
case "EXTERNAL":
DBImporterHelper.ScanOnlyExternalUsers(testData, reader, user_limit, riscore);
break;
case "INTERNAL":
DBImporterHelper.ScanOnlyInternalUsers(testData, reader, user_limit, riscore);
break;
default:
DBImporterHelper.ScanAllUsers(testData, reader, user_limit, riscore);
break;
}
FinalizeAndSendToKafka(writer, testData, args[4]);
}
}