public bool CheckWhitelistDupe(Whitelists wl)
{
if (!OpenCon())
{
return(true);
}
String q1 = "SELECT * FROM dbo.Whitelist Where Serial = @cereal";
SqlParameter serialParam = new SqlParameter("cereal", SqlDbType.VarChar);
serialParam.Value = wl.SerialNumber1;
SqlCommand comm = new SqlCommand(q1, cnn);
comm.Parameters.Add(serialParam);
using (var results = comm.ExecuteReader())
{
while (results.Read())
{
if (results["serial"].ToString().Equals(wl.SerialNumber1) && results["pid"].ToString().Equals(wl.Pid1) && results["vid"].ToString().Equals(wl.Vid1))
{
cnn.Close();
return(true);
}
}
cnn.Close();
return(false);
}
}
//Whitelist Methods
public List <Whitelists> GetWhitelists()
{
if (!OpenCon())
{
return(new List <Whitelists>());
}
List <Whitelists> wls = new List <Whitelists>();
SqlCommand cmd = new SqlCommand();
SqlDataReader reader;
cmd.CommandText = "SELECT * FROM dbo.Whitelist ORDER BY WhitelistID desc";
cmd.CommandType = CommandType.Text;
cmd.Connection = cnn;
using (reader = cmd.ExecuteReader())
{
while (reader.Read())
{
Whitelists wl = new Whitelists(
(DateTime)reader["Datetime"],
reader["Vid"].ToString(),
reader["Pid"].ToString(),
reader["Serial"].ToString(),
(int)reader["WhitelistID"]);
wls.Add(wl);
}
return(wls);
}
}
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
public void Configure(IApplicationBuilder app, IHostingEnvironment env, Whitelists whitelists, IOptions <S3Options> options)
{
Console.WriteLine(options.Value.Region);
whitelists.SetWhiteList(
Environment.GetEnvironmentVariable("PDF_WHITELIST"),
WhiteListType.PDF
);
whitelists.SetWhiteList(
Environment.GetEnvironmentVariable("IMAGE_WHITELIST"),
WhiteListType.Image
);
whitelists.SetWhiteList(
Environment.GetEnvironmentVariable("KEY_WHITELIST"),
WhiteListType.Key
);
if (env.IsDevelopment())
{
app.UseDeveloperExceptionPage();
}
app.UseMvc();
}
public void RemoveWhitelist(Whitelists s)
{
if (OpenCon())
{
using (SqlCommand sql = new SqlCommand("DELETE FROM dbo.Whitelist WHERE WhitelistID=@wid", cnn))
{
sql.Parameters.AddWithValue("wid", s.WhitelistID1);
sql.ExecuteNonQuery();
}
}
}
private void BtnRemoveSelected_Click(object sender, RoutedEventArgs e)
{
WLTableRow row = (WLTableRow)WhitelistTable.Items.GetItemAt(WhitelistTable.SelectedIndex);
MSSQL ms = new MSSQL();
Whitelists wl = new Whitelists();
wl.WhitelistID1 = row.WhitelistID;
ms.RemoveWhitelist(wl);
ms.SetLastUpdated(DateTime.Now, MSSQL.GET_WHITELIST_LIST);
WhitelistTable.Items.Clear();
LoadTable();
}
public void AddWhitelist(Whitelists wl)
{
using (SQLiteCommand cmd = new SQLiteCommand("INSERT INTO Whitelists (PID, VID, Serial, WhitelistID, DateTime) values ($PID, $VID, $Serial, $WhitelistID, $DateTime) ", m_dbConnection))
{
cmd.Parameters.AddWithValue("$PID", wl.Pid1);
cmd.Parameters.AddWithValue("$VID", wl.Vid1);
cmd.Parameters.AddWithValue("$Serial", wl.SerialNumber1);
cmd.Parameters.AddWithValue("$WhitelistID", wl.WhitelistID1);
cmd.Parameters.AddWithValue("$DateTime", wl.Datetime1);
cmd.ExecuteNonQuery();
}
}
private void BtnAddOne_Click(object sender, RoutedEventArgs e)
{
Whitelists wl;
if ((VidAddOne.Text.Trim().Length == 0 || VidAddOne.Text.Trim().Length > 4) && (SerialAddOne.Text.Trim().Length > 255 || SerialAddOne.Text.Trim().Length == 0) && (PidAddOne.Text.Trim().Length == 0 || PidAddOne.Text.Trim().Length > 4))
{
VidAddOne.BorderBrush = new SolidColorBrush(Colors.Red);
PidAddOne.BorderBrush = new SolidColorBrush(Colors.Red);
SerialAddOne.BorderBrush = new SolidColorBrush(Colors.Red);
}
else
{
wl = new Whitelists(DateTime.Now,
VidAddOne.Text.Trim().ToUpper(),
PidAddOne.Text.Trim().ToUpper(),
SerialAddOne.Text.Trim());
MSSQL ms = new MSSQL();
if (ms.AddWhitelist(wl) == 0)
{
//Set Success
ms.SetLastUpdated(DateTime.Now, MSSQL.GET_WHITELIST_LIST);
VidAddOne.ClearValue(Border.BorderBrushProperty);
PidAddOne.ClearValue(Border.BorderBrushProperty);
SerialAddOne.ClearValue(Border.BorderBrushProperty);
VidAddOne.Text = "";
PidAddOne.Text = "";
SerialAddOne.Text = "";
MsgAddOne.Content = "Success!";
WhitelistTable.Items.Clear();
LoadTable();
}
else if (ms.AddWhitelist(wl) == 1)
{
MsgAddOne.Content = "Database Error";
}
else
{
MsgAddOne.Content = "Unknown Error";
}
ms.CloseCon();
}
}
public List <Whitelists> GetWhitelists()
{
List <Whitelists> wls = new List <Whitelists>();
using (SQLiteCommand cmd = new SQLiteCommand("SELECT * From Whitelists", m_dbConnection))
{
using (var reader = cmd.ExecuteReader())
{
while (reader.Read())
{
Whitelists wl = new Whitelists(DateTime.Parse(reader["DateTime"].ToString(), System.Globalization.CultureInfo.InvariantCulture), reader["Vid"].ToString(), reader["Pid"].ToString(), reader["Serial"].ToString(), (int)reader["WhitelistID"]);
wls.Add(wl);
}
return(wls);
}
}
}
public byte AddWhitelist(Whitelists wl)
{
if (!CheckWhitelistDupe(wl))
{
if (!OpenCon())
{
return(2);
}
String q2 = "INSERT INTO dbo.Whitelist (DateTime,vid,pid,serial) Values (@dt, @v, @p, @cereal2)";
SqlCommand comm2 = new SqlCommand(q2, cnn);
SqlParameter datetimeParam = new SqlParameter("dt", SqlDbType.DateTime);
datetimeParam.Value = wl.Datetime1;
SqlParameter vParam = new SqlParameter("v", SqlDbType.VarChar);
vParam.Value = wl.Vid1;
SqlParameter pParam = new SqlParameter("p", SqlDbType.VarChar);
pParam.Value = wl.Pid1;
SqlParameter cereal2Param = new SqlParameter("cereal2", SqlDbType.VarChar);
cereal2Param.Value = wl.SerialNumber1;
comm2.Parameters.Add(datetimeParam);
comm2.Parameters.Add(vParam);
comm2.Parameters.Add(pParam);
comm2.Parameters.Add(cereal2Param);
comm2.ExecuteNonQuery();
cnn.Close();
return(0);
}
else
{
cnn.Close();
return(1);
}
}
public List <Whitelists> FilteredWhitelists(String pid, String vid, String srl, DateTime dt1, DateTime dt2)
{
if (!OpenCon())
{
return(new List <Whitelists>());
}
List <Whitelists> wls = new List <Whitelists>();
SqlCommand cmd = new SqlCommand();
SqlDataReader reader;
String sqltxt = "SELECT * FROM dbo.Whitelist WHERE "
+ "DateTime BETWEEN @dt1 AND @dt2 ";
cmd.CommandType = CommandType.Text;
cmd.Connection = cnn;
if (!pid.Equals(""))
{
sqltxt += "AND Pid LIKE @pid ";
cmd.CommandText = sqltxt;
cmd.Parameters.AddWithValue("@pid", "%" + pid + "%");
}
if (!vid.Equals(""))
{
sqltxt += "AND Vid LIKE @vid ";
cmd.CommandText = sqltxt;
cmd.Parameters.AddWithValue("@vid", "%" + vid + "%");
}
if (!srl.Equals(""))
{
sqltxt += "AND Serial LIKE @srl ";
cmd.CommandText = sqltxt;
cmd.Parameters.AddWithValue("@srl", "%" + srl + "%");
}
sqltxt += "ORDER BY WhitelistID desc";
cmd.CommandText = sqltxt;
SqlParameter dtt1 = new SqlParameter("dt1", SqlDbType.DateTime);
dtt1.Value = dt1;
cmd.Parameters.Add(dtt1);
SqlParameter dtt2 = new SqlParameter("dt2", SqlDbType.DateTime);
dtt2.Value = dt2;
cmd.Parameters.Add(dtt2);
using (reader = cmd.ExecuteReader())
{
while (reader.Read())
{
Whitelists wl = new Whitelists(
(DateTime)reader["datetime"],
reader["Vid"].ToString(),
reader["Pid"].ToString(),
reader["Serial"].ToString(),
(int)reader["WhitelistID"]);
wls.Add(wl);
}
cnn.Close();
return(wls);
}
}
public DocumentController(IDocumentRepository documentRepository, IS3Repository s3Repository, Whitelists whitelists)
{
this.documentRepository = documentRepository;
this.s3Repository = s3Repository;
this.whitelists = whitelists;
}
private IEnumerable <IWhitelist <TEntity> > GetWhitelists <TEntity>()
{
return(Whitelists.Where(d => d is IWhitelist <TEntity>).Cast <IWhitelist <TEntity> >());
}
public HealthCheckController(IConfiguration configuration, Whitelists whitelists)
{
this.configuration = configuration;
this.whitelists = whitelists;
}
