protected override void Seed(Context.DataContext context)
{
// This method will be called after migrating to the latest version.
// You can use the DbSet<T>.AddOrUpdate() helper extension method
// to avoid creating duplicate seed data.
}
/// <summary>
/// wipe out all data from database
/// </summary>
public static void ClearData()
{
Context.DataContext dbContext = new Context.DataContext(EstablishValidConnection());
//delete data and reset ids
dbContext.ExecuteCommand("TRUNCATE TABLE Relateds");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[Relateds]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM BaselineSecurityMappings");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[BaselineSecurityMappings]', RESEED, 0)");
dbContext.ExecuteCommand("TRUNCATE TABLE MapTypesCapabilitiesControls");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[MapTypesCapabilitiesControls]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM TICMappings");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[TICMappings]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM Specs");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[Specs]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM Controls");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[Controls]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM Capabilities");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[Capabilities]', RESEED, 0)");
}
public static void wipeBaselines()
{
Context.DataContext dbContext = new Context.DataContext(EstablishValidConnection());
dbContext.ExecuteCommand("DELETE FROM BaselineSecurityMappings");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[BaselineSecurityMappings]', RESEED, 0)");
}
//set up lists
public controlSpecChooser()
{
InitializeComponent();
dbcontext = new Context.DataContext(DataConnecter.EstablishValidConnection());
selected = new List <string>();
populateList(this.fullList);
}
/// <summary>
/// sets up connection
/// </summary>
protected void InitDataModel()
{
theCapability = new Context.Capabilities();
if (dbContext == null)
{
dbContext = new Context.DataContext(DataConnecter.EstablishValidConnection());
}
}
public VisualizationTool()
{
InitializeComponent();
ColorMaps.colorAll();
dbcontext = new Context.DataContext(Models.DataConnecter.EstablishValidConnection());
this.WindowState = System.Windows.WindowState.Maximized;
}
/// <summary>
/// sets controls on canvas
/// </summary>
public CapabilityChooser()
{
//maximized
InitializeComponent();
ResizeMode = System.Windows.ResizeMode.NoResize;
capsSelected = new List <Context.Capabilities>();
if (dbContext == null)
{
dbContext = new Context.DataContext(DataConnecter.EstablishValidConnection());
}
this.WindowState = System.Windows.WindowState.Maximized;
//add buttons
Button can = new Button();
can.Content = "Cancel";
can.IsCancel = true;
can.Margin = new Thickness(85, 15, 0, 0);
root.Children.Add(can);
Button ok = new Button();
ok.Content = "Done";
ok.Name = "Done";
ok.IsDefault = true;
ok.Margin = new Thickness(15, 15, 0, 0);
ok.Click += Done;
root.Children.Add(ok);
root.RegisterName("Done", ok);
Button all = new Button();
all.Content = "Selet all";
all.Name = "all";
all.Margin = new Thickness(168, 15, 0, 0);
all.Click += new RoutedEventHandler(SelectAll);
root.Children.Add(all);
root.RegisterName("all", all);
Label directions = new Label();
directions.Content = "Click once to add to report, again to remove. Read description by hovering on capability." +
"Click done to create report.";
directions.Margin = new Thickness(220, 15, 0, 0);
root.Children.Add(directions);
SetUpPallate();
try
{
//place capabilities
Addcaps();
}
catch (Exception e) {
Console.WriteLine(e);
Close();
}
}
/// <summary>
/// 新增实体
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="log"></param>
/// <returns></returns>
public async static Task <int> Insert <T>(T log)
{
using var context = new Context.DataContext();
context.Entry(log).State = Microsoft.EntityFrameworkCore.EntityState.Added;
await context.SaveChangesAsync();
return(-1);
}
/// <summary>
/// 更新单个推送实体
/// </summary>
/// <param name="obj">更新对象</param>
/// <param name="updateColumn">更新字段</param>
/// <returns></returns>
internal async static Task <int> Update(PushMessageDTO obj, string updateColumn)
{
using var context = new Context.DataContext();
//TODO Update
await context.SaveChangesAsync();
return(-1);
}
public ControlChooser()
{
InitializeComponent();
if (dbContext == null)
{
dbContext = new Context.DataContext(DataConnecter.EstablishValidConnection());
}
populateList();
}
/// <summary>
/// 更新多条推送消息
/// </summary>
/// <param name="exp"></param>
/// <param name="msgFunc"></param>
/// <returns></returns>
public async static Task <int> UpdateMany(
Expression <Func <PushMessageDTO, bool> > exp,
Expression <Func <PushMessageDTO, Message> > msgFunc)
{
using var context = new Context.DataContext();
//TODO Update many
await context.SaveChangesAsync();
return(-1);
}
public async static Task <int> Update <T>(T updateObj, string column) where T : class, new()
{
using var context = new Context.DataContext();
var entry = context.Entry(updateObj);
entry.Property(column).IsModified = true;
await context.SaveChangesAsync();
return(-1);
}
/// <summary>
/// clear capablities data
/// </summary>
public static void wipeCape()
{
Context.DataContext dbContext = new Context.DataContext(EstablishValidConnection());
dbContext.ExecuteCommand("DELETE FROM MapTypesCapabilitiesControls");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[MapTypesCapabilitiesControls]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM TICMappings");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[TICMappings]', RESEED, 0)");
dbContext.ExecuteCommand("DELETE FROM Capabilities");
dbContext.ExecuteCommand("DBCC CHECKIDENT ('[ModelDb].[dbo].[Capabilities]', RESEED, 0)");
}
/// <summary>
/// setup a new database with all tables
/// </summary>
public static void FirstUse()
{
Context.DataContext dbContext = new Context.DataContext(EstablishValidConnection());
if (!(dbContext.DatabaseExists()))
{
dbContext.CreateDatabase();
}
else
{
//ClearData();
}
//DropInitDB();
}
private static void DropInitDB()
{
//RunScript("schemaInit.sql");
Context.DataContext dbContext = new Context.DataContext(EstablishValidConnection());
if (!(dbContext.DatabaseExists()))
{
dbContext.CreateDatabase();
}
else
{
ClearData();
}
RunScript("dataInit.sql");
}
/// <summary>
/// esenbles a valid conection string to be passed to context.dataContext
/// </summary>
/// <returns>The connection string</returns>
public static string EstablishValidConnection()
{
if (Properties.Settings.Default.Connection != "")
{
return(Properties.Settings.Default.Connection);
}
//trying computer name
string CompName = System.Environment.MachineName;
string conectstr = @"Data Source=" + CompName + @"\SQLEXPRESS;Initial Catalog=ModelDB;Integrated Security=True;Persist Security Info=True";
string serversection = @"Data Source=" + CompName + @"\SQLEXPRESS;Integrated Security=True;Persist Security Info=True";
Context.DataContext connection = new Context.DataContext(serversection);
try
{
connection.Connection.Open();
}
catch (Exception e)
{
//not normal name
bool accept = false;
string prompt = "Enter SQL server Name: ";
//error not default. need to repeativle prompt
while (!accept)
{
string server = Interaction.InputBox(prompt);
conectstr = @"Data Source=" + CompName + "\\" + server + @";Initial Catalog=ModelDB;Integrated Security=True;Persist Security Info=True";
connection = new Context.DataContext(conectstr);
try
{
connection.Connection.Open();
accept = true;
}
catch (Exception ex)
{
prompt = "Connection failed. Enter SQL server Name: ";
}
}
}
//save database info
Properties.Settings.Default.Connection = conectstr;
Properties.Settings.Default.Save();
return(conectstr);
}
private static void RunScript(string fileName)
{
Context.DataContext dbContext = new Context.DataContext(EstablishValidConnection());
string initScript = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, fileName);
string[] script = File.ReadAllLines(initScript);
foreach (string command in script)
{
try
{
dbContext.ExecuteCommand(command);
}
catch (Exception ex)
{
string s = ex.Message;
}
}
}
public bool LoginUser(string name, string pass)
{
Context.User result;
using (var context = new Context.DataContext())
{
result = context.Users.Where(x => x.Name == name && x.Pass == _encrypter.EncripterPass(pass) && x.Active == true).FirstOrDefault();
}
if (result == null)
{
return(false);
}
else
{
return(true);
}
}
/// <summary>
/// change database location
/// </summary>
/// <param name="sender"></param>
/// <param name="e"></param>
private void EditDB(object sender, RoutedEventArgs e)
{
bool accept = false;
string prompt = "Enter SQL server Name: ", conectstr = "";
string CompName = System.Environment.MachineName;
while (!accept)
{
//continue till connection is made
string server = Interaction.InputBox(prompt);
if (server == string.Empty)
{
return;
}
conectstr = @"Data Source=" + CompName + "\\" + server + @";Initial Catalog=ModelDB;Integrated Security=True;Persist Security Info=True";
string serversection = @"Data Source=" + CompName + @"\" + server + ";Integrated Security=True;Persist Security Info=True";
Context.DataContext connection = new Context.DataContext(serversection);
try
{
connection.Connection.Open();
accept = true;
}
catch (Exception ex)
{
prompt = "Connection failed. Enter SQL server name: ";
}
}
Properties.Settings.Default.Connection = conectstr;
Properties.Settings.Default.Save();
//set up tables and values
Waiting load = new Waiting();
load.Topmost = true;
load.Show();
load.Topmost = false;
DataConnecter.FirstUse();
load.Close();
MessageBox.Show("Connection updated.");
ChangeReportStatus(true);
}
//set button and run first time set up
public MainWindow()
{
InitializeComponent();
this.ResizeMode = System.Windows.ResizeMode.CanMinimize;
TextBlock cap = new TextBlock()
{
Text = "This report allows you to choose what capabilities to create a shortened report of these capabilities.", Width = 200, TextWrapping = TextWrapping.Wrap
};
this.capReport.ToolTip = new ToolTip()
{
Content = cap
};
TextBlock cont = new TextBlock()
{
Text = "This report allows you to choose what control to list in the report. For each, there will be a list of all capabilities it is selected for."
, Width = 200, TextWrapping = TextWrapping.Wrap
};
this.conReport.ToolTip = new ToolTip()
{
Content = cont
};
TextBlock tic = new TextBlock()
{
Text = "This report show every TIC with a list of all capabilities with that marking.", Width = 200, TextWrapping = TextWrapping.Wrap
};
this.TICReport.ToolTip = new ToolTip()
{
Content = tic
};
//setup
if (Properties.Settings.Default.FirstRun)
{
MessageBox.Show(
"Welcome to the Cloud Security Manager. \n\tBefore you can use the program, there is some simple set up to do. \n\n\t First, you must choose where " +
"to put the folders that will hold input and output files. The program will create a containing folder, so just pick a easily accessible place.\n\n "
+ "Then, you may need to set up database connection.");
Waiting load = new Waiting();
SetPath();
load.Topmost = true;
load.Show();
load.Topmost = false;
try
{
Properties.Settings.Default.FirstRun = false;
Properties.Settings.Default.Save();
DataConnecter.FirstUse();
load.Close();
}
catch (Exception e)
{
load.error.Content = e.Message;
Console.WriteLine(e.Message);
}
}
else
{
apppath = Properties.Settings.Default.appFolders;
}
dbContent = new Context.DataContext(DataConnecter.EstablishValidConnection());
CSRC.Models.Constants.ReadValues();
var con = from p in dbContent.Controls
select p;
if (!con.Any())
{
this.updateCaps.IsEnabled = false;
this.updateBaselines.IsEnabled = false;
this.menuUpdateCapabilities.IsEnabled = false;
this.menuUpdateBaseines.IsEnabled = false;
}
var ret = from p in dbContent.Capabilities
select new { p.Id };
if (!(ret.Any()))
{
ChangeReportStatus(false);
}
var retdata = from p in dbContent.BaselineSecurityMappings
select new { p.Id };
if (!retdata.Any())
{
ChangeReportStatus(false);
}
CSRC.Models.Constants.ReadValues();
if (CSRC.Models.Constants.capFile3Cols)
{
this.cap3col.IsChecked = true;
this.cap9col.IsChecked = false;
}
}
public AuthorRepository(Context.DataContext context)
{
_context = context;
}
public BooksRepository(Context.DataContext context)
{
_context = context;
}
