public CollectExecution CreateCollect(IDocumentSession session)
{
CollectExecution collect = new CollectExecution();
collect.SetDateStartCollect();
collect.SetDateEndCollect();
session.Store(collect);
return collect;
}
public CollectExecution CreateACollectExecutionWithError(IDocumentSession session)
{
CollectExecution collectExecution = new CollectExecution();
collectExecution.SetDateStartCollect();
ProbeExecution probeExecution = this.GetProbeExecutionWithError(session);
collectExecution.ProbeExecutions.Add(probeExecution);
collectExecution.SetDateEndCollect();
return collectExecution;
}
public CollectExecution CreateCollectExecution(IDocumentSession session, CollectRequest collectRequest)
{
CollectExecution collect = new CollectExecution();
collect.RequestId = collectRequest.Oid;
session.Store(collect);
session.SaveChanges();
return collect;
}
public CollectExecution CreateCollectExecution(IDocumentSession session, CollectRequest collectRequest)
{
CollectExecution collect = new CollectExecution();
collect.RequestId = collectRequest.Oid;
session.Store(collect);
session.SaveChanges();
return(collect);
}
private void StartCollect(IDocumentSession session, CollectRequest collectRequest, CollectExecution collectExecution, FamilyEnumeration plataform, ExecutionLogBuilder executionLog)
{
int attemptsOfEvaluateVariables = 1;
bool anErrorOccured = false;
bool allObjectWasCollect = false;
IEnumerable<ObjectType> objectTypes = collectRequest.GetObjectTypesWasNotCollected(session);
session.SaveChanges();
while (!collectTimeOut.IsExceededTheMaxAttemptsOfEvaluateVariables(attemptsOfEvaluateVariables) &&
!allObjectWasCollect &&
!anErrorOccured &&
!Interrupted)
{
IEnumerable<StateType> states = collectRequest.GetStates(session);
IEnumerable<SelectedProbe> selectedProbes = ProbeManager.GetProbesFor(objectTypes, plataform);
VariablesEvaluated variables = variableEvaluatorService.Evaluate(collectRequest, session);
session.SaveChanges();
try
{
this.UpdateSystemInformationOfTarget(session, collectRequest, plataform, executionLog);
this.ExecuteCollectWithProbes(session, selectedProbes, collectRequest, variables, states, collectExecution, executionLog);
session.SaveChanges();
}
catch (SystemInformationException ex)
{
string logMessage;
if (ex.InnerException == null)
{
logMessage = String.Format(EXECUTION_ERROR_MESSAGE, ex.Message, ex.StackTrace);
}
else
{
logMessage = String.Format(EXECUTION_ERROR_MESSAGE_WITH_INNER_EXCEPTION, ex.Message, ex.InnerException.StackTrace, ex.StackTrace);
}
Logger.Error(logMessage);
collectRequest.SetResultError();
collectRequest.Close();
anErrorOccured = true;
session.SaveChanges();
break;
}
catch
{
// only ends the loop. The error already was treated
anErrorOccured = true;
}
objectTypes = collectRequest.GetObjectTypesWasNotCollected(session);
allObjectWasCollect = objectTypes.Count() == 0;
attemptsOfEvaluateVariables++;
}
if (Interrupted)
{
collectRequest.Status = CollectRequestStatus.Canceled;
}
else
{
if ((!allObjectWasCollect) && (!anErrorOccured))
{
CloseExecutionIfIncomplete(session, collectRequest, collectExecution, plataform, objectTypes);
}
}
session.SaveChanges();
}
/// <summary>
/// Execute the collect for the probes selected
/// </summary>
/// <param name="selectedProbes">The selected probes.</param>
/// <param name="collectRequest">The request collect.</param>
private void ExecuteCollectWithProbes(IDocumentSession session, IEnumerable<SelectedProbe> selectedProbes, CollectRequest collectRequest,
VariablesEvaluated variables, IEnumerable<StateType> states, CollectExecution collectExecution, ExecutionLogBuilder executionLog)
{
foreach (SelectedProbe probe in selectedProbes)
{
if (Interrupted)
{
break;
}
try
{
ProbeExecution probeExecution = this.ExecuteCollect(session, probe, collectRequest, variables, states, executionLog);
collectExecution.ProbeExecutions.Add(probeExecution);
}
catch (RecoverableProbeException ex)
{
this.ConfigureTheCollectRequestWithAnErrorProbeExecute(collectRequest, probe.Capability.OvalObject, collectExecution, ex, executionLog);
session.SaveChanges();
throw ex;
}
catch (Exception ex)
{
EndsACollectRequestBecauseThisErrorIsUnrecoverable(collectRequest, probe.Capability.OvalObject, collectExecution, ex, executionLog);
session.SaveChanges();
throw ex;
}
}
session.SaveChanges();
}
/// <summary>
/// Ends the request collect because this error is unrecoverable.
/// This request was not execute because exists some error that is not possible perform the collect.
/// Ex.: Invalid credentials, bad format of oval xmls, etc.
/// </summary>
/// <param name="collectRequest">The request collect.</param>
/// <param name="probe">The probe.</param>
/// <param name="collectExecution">the Collect execution object.</param>
/// <param name="error">the exception that represents an error.</param>
private void EndsACollectRequestBecauseThisErrorIsUnrecoverable(CollectRequest collectRequest, string probeCapability, CollectExecution collectExecution, Exception error, ExecutionLogBuilder executionLog)
{
ConfigureTheCollectRequestWithAnErrorProbeExecute(collectRequest, probeCapability, collectExecution, error, executionLog);
collectRequest.Close();
}
private void EndCollect(IDocumentSession session, CollectRequest collectRequest, CollectExecution collectExecution)
{
if (collectTimeOut.IsExceededTheMaxOfExecutionsDefined(collectRequest.GetNumberOfExecutions(session)))
{
collectRequest.Close();
}
else
{
if (!collectRequest.isClosed())
{
if (!collectExecution.ExistsExecutionsWithError())
{
collectRequest.TryClose(session);
}
else
{
collectRequest.Reopen();
}
}
}
collectExecution.SetDateEndCollect();
session.SaveChanges();
}
private void CreateCollectedObjectsForNotSupportedObjects(IDocumentSession session, IEnumerable<SelectedProbe> objectsNotSupported, CollectRequest collectRequest, CollectExecution collectExecution)
{
foreach (var probe in objectsNotSupported)
{
ProbeResult probeResult = probe.CreateCollectedObjectForNotSupportedObjects(probe.ObjectTypes);
ProbeExecution probeExecution = this.CreateTheProbeExecution(probeResult, probe);
collectRequest.UpdateSystemCharacteristics(session);
collectExecution.ProbeExecutions.Add(probeExecution);
session.SaveChanges();
}
}
/// <summary>
/// Configures the request collect with one probeExecute with the error status.
/// </summary>
/// <param name="collectRequest">The request collect.</param>
/// <param name="probe">The probe.</param>
/// <param name="collectExecution">The collect execution.</param>
/// <param name="error">The error.</param>
private void ConfigureTheCollectRequestWithAnErrorProbeExecute(CollectRequest collectRequest, string probeCapability, CollectExecution collectExecution, Exception error, ExecutionLogBuilder executionLog)
{
executionLog.AnErrorOccurred(error.Message);
executionLog.EndCollect();
ProbeExecution executionWithError = collectFactory.CreateAProbeExecutionWithError(probeCapability, executionLog.BuildExecutionLogs());
collectExecution.ProbeExecutions.Add(executionWithError);
}
private void CloseExecutionIfIncomplete(IDocumentSession session, CollectRequest collectRequest, CollectExecution collectExecution, FamilyEnumeration plataform, IEnumerable<ObjectType> objectTypes)
{
IEnumerable<SelectedProbe> objectsNotSupported = ProbeManager.GetNotSupportedObjects(objectTypes, plataform);
var objectNotSupported = this.GetObjectTypesFromSelectedProbes(objectsNotSupported);
var objectSupportedNotCollect = objectTypes.Except(objectNotSupported);
this.CreateCollectedObjectsForNotSupportedObjects(session, objectsNotSupported, collectRequest, collectExecution);
if (objectSupportedNotCollect.Count() > 0)
{
IEnumerable<SelectedProbe> objectsSupportedNotCollected = ProbeManager.GetProbesFor(objectSupportedNotCollect, plataform);
this.CreateCollectedObjectsForNotSupportedObjects(session, objectsSupportedNotCollected, collectRequest, collectExecution);
}
collectRequest.SetResultComplete(session);
collectRequest.Close();
session.SaveChanges();
}
public void Should_be_possible_finalize_the_collectRequest_if_the_number_of_execution_exceeds_of_limits()
{
IDocumentSession fakeSession = provider.GetSession();
CollectRequest collectRequest = this.GetCollectRequest();
oval_definitions definitions = new OvalDocumentLoader().GetFakeOvalDefinitions("definitionsWithLocalVariable.xml");
var objList = new List<ObjectType>();
objList.Add(new family_object() { id = "oval:org.mitre.oval:obj:1000" });
objList.Add(new file_object() { id = "oval:org.mitre.oval:obj:5000" });
definitions.objects = objList.ToArray();
var newDefinitiondoc = new DefinitionDocument() { OriginalId = "01", Text = definitions.GetDefinitionsXml() };
session.Store(newDefinitiondoc);
collectRequest.OvalDefinitionsId = newDefinitiondoc.Oid;
CollectExecutionManagerFactory factory = new CollectExecutionManagerFactory(collectRequest, session);
List<string> registryObjectsThatNotProcessInFirstTime = new List<string>();
List<string> familyObjectsThatNotProcessInFirstTime = new List<string>();
CollectExecutionManager executionManager = factory.CreateExecutionManagerForTheSuccessScenario(this.GetResultForRegistry(collectRequest, registryObjectsThatNotProcessInFirstTime),
this.GetResultForFamily(collectRequest, familyObjectsThatNotProcessInFirstTime));
session.Store(collectRequest);
var newExecution1 = new CollectExecution() { RequestId = collectRequest.Oid };
session.Store(newExecution1);
var newExecution2 = new CollectExecution() { RequestId = collectRequest.Oid };
session.Store(newExecution2);
var newExecution3 = new CollectExecution() { RequestId = collectRequest.Oid };
session.Store(newExecution3);
executionManager.ExecuteCollect(fakeSession, collectRequest, FamilyEnumeration.windows);
session.SaveChanges();
CollectRequest collectRequestAfterExecute = session.Load<CollectRequest>(collectRequest.Oid.ToString());
Assert.IsNotNull(collectRequest);
Assert.AreEqual(4, collectRequestAfterExecute.GetNumberOfExecutions(session));
Assert.IsTrue(collectRequestAfterExecute.isClosed());
}
public CollectRequest CreateCollectRequestIncompleteAfterCollect(IDocumentSession session)
{
CollectRequest collectRequest = new CollectRequest();
string definitionsInXml = GetOvalObjectInXML(".definitions.fdcc_xpfirewall_oval.xml");
StringBuilder builder = new StringBuilder(definitionsInXml);
var newDefinitiondoc = new DefinitionDocument()
{
OriginalId = "01",
Text = builder.ToString()
};
session.Store(newDefinitiondoc);
collectRequest.OvalDefinitionsId = newDefinitiondoc.Oid ;
session.Store(collectRequest);
CollectExecution collectExecution = new CollectExecution();
collectExecution.RequestId = collectRequest.Oid;
session.Store(collectExecution);
collectExecution.SetDateStartCollect();
collectExecution.SetDateEndCollect();
//collectRequest.Collects.Add(collectExecution);
ProbeResult probeResult = new ProbeResultFactory().CreateProbeResultForRegistryCollect();
CollectFactory collectFactory = new CollectFactory(session);
ProbeExecution probeExecution = collectFactory.CreateAProbeExecution(probeResult, "registry");
CollectResult collectResult = new CollectResultFactory().CreateCollectResultForTheProbeExecution(probeResult);
string systemCharacteristicsXml = this.GetSystemCharacteristicsInXML(".system_characteristics.fdcc_xpfirewall_oval.sc_incomplete.xml");
collectResult.SystemCharacteristics = systemCharacteristicsXml;
probeExecution.SystemCharacteristics = collectResult.SystemCharacteristics;
collectExecution.ProbeExecutions.Add(probeExecution);
// collectRequest.Collects.Add(collectExecution);
this.CreateTarget(session, collectRequest);
session.SaveChanges();
return collectRequest;
}
private static CollectExecution CreateCollectExecution(IDocumentSession session, CollectRequest collectRequest)
{
var collectExecution =
new CollectExecution() { RequestId = collectRequest.Oid };
collectExecution.SetDateStartCollect();
collectExecution.SetDateEndCollect();
return collectExecution;
}
public CollectRequest CreateCollectRequestCompleteAfterCollect(IDocumentSession session)
{
CollectRequest collectRequest = new CollectRequest();
session.Store(collectRequest);
CollectExecution collectExecution = new CollectExecution();
session.Store(collectExecution);
collectExecution.RequestId = collectRequest.Oid;
collectExecution.SetDateStartCollect();
collectExecution.SetDateEndCollect();
ProbeResult probeResult = new ProbeResultFactory().CreateProbeResultForRegistryCollect();
CollectFactory collectFactory = new CollectFactory(session);
ProbeExecution probeExecution = collectFactory.CreateAProbeExecution(probeResult, "registry");
CollectResult collectResult = new CollectResultFactory().CreateCollectResultForTheProbeExecution(probeResult);
collectRequest.Result = collectResult;
string systemCharacteristicsXml = this.GetSystemCharacteristicsInXML(".system_characteristics.oval.org.mitre.oval.sc.5368.xml");
collectResult.SystemCharacteristics = systemCharacteristicsXml;
probeExecution.SystemCharacteristics = collectResult.SystemCharacteristics;
collectExecution.ProbeExecutions.Add(probeExecution);
this.CreateTarget(session, collectRequest);
//session.Store(collectRequest);
collectExecution.RequestId = collectRequest.Oid;
return collectRequest;
}
