public static void Main(string[] args)
{
IPersistenceConfigurer config = PostgreSQLConfiguration.PostgreSQL82.ConnectionString ("Server=" + ConfigurationManager.AppSettings ["PostgreSQL"] + ";Port=5432;Database=autoassess;User Id=postgres;Password=password;SSL=true;");
ISessionFactory factory = Fluently.Configure ()
.Database (config)
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentCVE> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentNessusScan> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentNexposeScan> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentOpenVASTask> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentOneSixtyOneResults> ())
.BuildSessionFactory ();
List<PersistentCVE> cves = new List<PersistentCVE> ();
using (ISession session = factory.OpenSession()) {
string xml = System.IO.File.ReadAllText ("/home/bperry/tmp/cve/allitems.xml");
XmlDocument doc = new XmlDocument ();
doc.LoadXml (xml);
using (ITransaction trans = session.BeginTransaction()) {
foreach (XmlNode child in doc.LastChild.ChildNodes) {
PersistentCVE cve = new PersistentCVE (new CVE (child));
cve.SetCreationInfo (Guid.Empty);
foreach (PersistentCVEReference reference in cve.PersistentReferences) {
reference.CVE = cve;
reference.SetCreationInfo (Guid.Empty);
}
foreach (PersistentCVEComment comment in cve.PersistentComments) {
comment.CVE = cve;
comment.SetCreationInfo (Guid.Empty);
}
Console.WriteLine ("Saving " + cve.Name);
session.Save (cve);
cves.Add(cve);
}
List<string> nvdExports = new List<string>();
foreach (FileInfo file in new System.IO.DirectoryInfo(ConfigurationManager.AppSettings["nvdExportDir"]).EnumerateFiles())
nvdExports.Add (file.FullName);
foreach (string export in nvdExports) {
xml = System.IO.File.ReadAllText (export);
doc = new XmlDocument ();
doc.LoadXml (xml);
foreach (XmlNode child in doc.LastChild.ChildNodes) {
PersistentNVD nvd = new PersistentNVD (new NVD (child));
if (cves.Where (c => c.Name == nvd.CVEID).Count() != 1)
continue;
if (!string.IsNullOrEmpty (nvd.CVEID)) {
PersistentCVE cve = cves.Where (c => c.Name == nvd.CVEID).Single ();
if (cve == null)
throw new Exception ("CVE " + nvd.CVEID + " doesn't exist.");
nvd.CVE = cve;
}
Console.WriteLine (nvd.NVDID);
nvd.SetCreationInfo (Guid.Empty);
if (nvd.CVSS != null)
nvd.CVSS.SetCreationInfo (Guid.Empty);
if (nvd.References != null)
foreach (PersistentNVDReference reference in nvd.References)
reference.SetCreationInfo (Guid.Empty);
if (nvd.VulnerableSoftware != null)
foreach (PersistentVulnerableSoftware vs in nvd.VulnerableSoftware)
vs.SetCreationInfo (Guid.Empty);
session.SaveOrUpdate (nvd);
}
}
try {
Console.WriteLine ("Committing...");
trans.Commit ();
} catch (Exception ex) {
trans.Rollback ();
throw ex;
}
}
}
}
public static void Main(string[] args)
{
IPersistenceConfigurer config = PostgreSQLConfiguration.PostgreSQL82.ConnectionString ("Server=" + ConfigurationManager.AppSettings ["PostgreSQL"] + ";Port=5432;Database=rising_sun;User Id=" +
ConfigurationManager.AppSettings ["PostgreSQLUser"] + ";Password="******"PostgreSQLPassword"] + ";SSL=true;");
ISessionFactory factory = Fluently.Configure ()
.Database (config)
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentNVD> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentNessusScan> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentNexposeScan> ())
.Mappings (m =>
m.FluentMappings.AddFromAssemblyOf<PersistentOpenVASTask> ())
.BuildSessionFactory ();
using (ISession session = factory.OpenSession()) {
List<string> nvdExports = new List<string>();
foreach (FileInfo file in new System.IO.DirectoryInfo(ConfigurationManager.AppSettings["nvdExportDir"]).EnumerateFiles())
nvdExports.Add(file.FullName);
foreach (string export in nvdExports) {
using (ITransaction trans = session.BeginTransaction()) {
string xml = System.IO.File.ReadAllText (export);
XmlDocument doc = new XmlDocument ();
doc.LoadXml (xml);
foreach (XmlNode child in doc.LastChild.ChildNodes) {
PersistentNVD nvd = new PersistentNVD (new NVD (child));
// bool exists = session.CreateCriteria<PersistentNVD> ()
// .Add (NHibernate.Criterion.Restrictions.Eq ("CVEID", nvd.CVEID))
// .List<PersistentNVD> ()
// .Any ();
//
// if (exists) {
// Console.WriteLine ("Skipping NVD: " + nvd.CVEID);
// continue;
// }
if (!string.IsNullOrEmpty (nvd.CVEID)) {
PersistentCVE cve = session.CreateCriteria<PersistentCVE> ()
.Add (NHibernate.Criterion.Restrictions.Eq ("Name", nvd.CVEID))
.List<PersistentCVE> ()
.SingleOrDefault ();
if (cve == null)
throw new Exception ("CVE " + nvd.CVEID + " doesn't exist.");
nvd.CVE = cve;
}
Console.WriteLine (nvd.NVDID);
nvd.SetCreationInfo (Guid.Empty);
if (nvd.CVSS != null)
nvd.CVSS.SetCreationInfo (Guid.Empty);
if (nvd.References != null)
foreach (PersistentNVDReference reference in nvd.References)
reference.SetCreationInfo (Guid.Empty);
if (nvd.VulnerableSoftware != null)
foreach (PersistentVulnerableSoftware vs in nvd.VulnerableSoftware)
vs.SetCreationInfo (Guid.Empty);
session.SaveOrUpdate (nvd);
}
try {
Console.WriteLine ("Committing...");
trans.Commit ();
} catch (Exception ex) {
trans.Rollback ();
throw ex;
}
}
}
}
}