public void ReceivedResult(DynamicCommandEvent dynamicCommandEvent, Client client)
{
_cacheManager.AddCommandEvent(dynamicCommandEvent);
if (dynamicCommandEvent.Status == ActivityType.Active || dynamicCommandEvent.Status == ActivityType.Stopped)
{
lock (_activeCommandsLock)
{
var activeCommand =
ActiveCommands.FirstOrDefault(x => x.DynamicCommand.Id == dynamicCommandEvent.DynamicCommand);
if (activeCommand == null)
{
var dynamicCommand =
_databaseManager.GetDynamicCommandById(dynamicCommandEvent.DynamicCommand);
if (dynamicCommand == null)
{
//when there is no command on this server with the id, we stop it because it may be removed by an administrator
client.StopActiveCommand(dynamicCommandEvent.DynamicCommand);
return;
}
ActiveCommands.Add(activeCommand = new ActiveCommandInfo(dynamicCommand));
}
switch (dynamicCommandEvent.Status)
{
case ActivityType.Active:
lock (activeCommand.ClientsLock)
activeCommand.Clients.Add(client);
ActiveCommandEventManager.AddClient(activeCommand, client);
CheckClientExecuteActiveCommand(activeCommand, client);
break;
case ActivityType.Stopped:
lock (activeCommand.ClientsLock)
activeCommand.Clients.Remove(client);
ActiveCommandEventManager.RemoveClient(activeCommand, client);
if (activeCommand.Clients.Count == 0)
{
ActiveCommands.Remove(activeCommand);
if (activeCommand.DynamicCommand.Status == DynamicCommandStatus.Active)
{
//don't change the status when stopped
activeCommand.DynamicCommand.Status = DynamicCommandStatus.Done;
}
ActiveCommandEventManager.RemoveActiveCommand(activeCommand);
}
break;
}
}
}
}
public void RemoveDynamicCommand(RegisteredDynamicCommand dynamicCommand)
{
Logger.Debug("Remove dynamic command with id {0}", dynamicCommand.Id);
if (dynamicCommand.CommandType == CommandType.Active)
{
lock (_activeCommandsLock)
{
var activeCommand = ActiveCommands.FirstOrDefault(x => x.DynamicCommand.Id == dynamicCommand.Id);
if (activeCommand != null)
{
StopActiveCommand(activeCommand);
}
}
}
_dynamicCommandScheduler.RemoveDynamicCommand(dynamicCommand);
lock (_dynamicCommandsLock)
DynamicCommands.Remove(dynamicCommand);
_databaseManager.RemoveDynamicCommand(dynamicCommand.Id);
}
public List <RegisteredDynamicCommand> GetDynamicCommands()
{
Logger.Debug("Get all dynamic commands");
var dynamicCommands = _databaseManager.GetDynamicCommands();
lock (_activeCommandsLock)
{
foreach (var registeredDynamicCommand in dynamicCommands)
{
var activeCommandInfo =
ActiveCommands.FirstOrDefault(x => x.DynamicCommand.Id == registeredDynamicCommand.Id);
if (activeCommandInfo != null)
{
registeredDynamicCommand.Status = DynamicCommandStatus.Active;
lock (activeCommandInfo.ClientsLock)
registeredDynamicCommand.ExecutingClientIds =
activeCommandInfo.Clients.Select(x => x.Id).ToArray();
}
}
}
return(dynamicCommands);
}
public ActiveCommandInfo GetActiveCommandById(int id)
{
lock (_activeCommandsLock)
return(ActiveCommands.FirstOrDefault(x => x.DynamicCommand.Id == id));
}
public void OnClientJoin(Client client)
{
Logger.Debug("Client CI-{0} joined, check if there are commands to execute", client.Id);
lock (_dynamicCommandsLock)
{
foreach (
var command in
DynamicCommands.Where(
x =>
(x.TransmissionEvent.GetType() == typeof(OnJoinTransmissionEvent)) &&
IsClientInTargets(client, x.Target) &&
CheckConditions(client.GetOnlineClientInformation(), client.ComputerInformation.ClientConfig,
x.Conditions)))
{
if (ExecuteStaticCommand(new[] { client },
DynamicCommandToPotentialCommand(command,
_databaseManager.GetDynamicCommandParameter(command.Id)))
.Count == 1)
{
_databaseManager.AddDynamicCommandEvent(command.Id, client.Id, ActivityType.Sent, null);
}
}
foreach (
var command in
DynamicCommands.Where(
x =>
x.TransmissionEvent.GetType() == typeof(EveryClientOnceTransmissionEvent) &&
IsClientInTargets(client, x.Target) &&
CheckConditions(client.GetOnlineClientInformation(), client.ComputerInformation.ClientConfig,
x.Conditions)))
{
if (!_databaseManager.ClientCommandExecuted(client.Id, command.Id))
{
if (ExecuteStaticCommand(new[] { client },
DynamicCommandToPotentialCommand(command,
_databaseManager.GetDynamicCommandParameter(command.Id))).Count == 1)
{
_databaseManager.AddDynamicCommandEvent(command.Id, client.Id, ActivityType.Sent, null);
}
}
}
}
if (client.ComputerInformation.ActiveCommands?.Count > 0)
{
lock (_activeCommandsLock)
foreach (var activeCommandId in client.ComputerInformation.ActiveCommands)
{
var activeCommand = ActiveCommands.FirstOrDefault(x => x.DynamicCommand.Id == activeCommandId);
if (activeCommand == null)
{
var dynamicCommand =
_databaseManager.GetDynamicCommandById(activeCommandId);
if (dynamicCommand == null)
{
//when there is no command on this server with the id, we stop it because it may be removed by an administrator
client.StopActiveCommand(activeCommandId);
return;
}
ActiveCommands.Add(activeCommand = new ActiveCommandInfo(dynamicCommand));
ActiveCommandEventManager.AddActiveCommand(activeCommand);
}
lock (activeCommand.ClientsLock)
activeCommand.Clients.Add(client);
CheckClientExecuteActiveCommand(activeCommand, client);
ActiveCommandEventManager.AddClient(activeCommand, client);
}
}
}
