public override Structs.TaskQueue GetMessages(Apollo.Agent agent)
{
Structs.TaskQueue result;
List <Task> finalTaskList = new List <Task>();
List <Structs.DelegateMessage> finalDelegateList = new List <Structs.DelegateMessage>();
Structs.CheckTaskingResponse resp = JsonConvert.DeserializeObject <Structs.CheckTaskingResponse>(GetTaskingMessage());
foreach (Task task in resp.tasks)
{
Debug.WriteLine("[-] CheckTasking - NEW TASK with ID: " + task.id);
finalTaskList.Add(task);
}
if (resp.delegates != null)
{
foreach (Dictionary <string, string> delegateMessage in resp.delegates)
{
foreach (KeyValuePair <string, string> item in delegateMessage)
{
finalDelegateList.Add(new Structs.DelegateMessage()
{
UUID = item.Key,
Message = item.Value
});
}
}
}
result = new Structs.TaskQueue()
{
Tasks = finalTaskList.ToArray(),
Delegates = finalDelegateList.ToArray()
};
//result.Add("tasks", finalTaskList.ToArray());
//result.Add("delegates", finalDelegateList.ToArray());
//SCTask task = JsonConvert.DeserializeObject<SCTask>(Post(json));
return(result);
}
public override Structs.TaskQueue GetMessages(Apollo.Agent agent)
{
Structs.TaskQueue result;
List <Task> finalTaskList = new List <Task>();
List <Structs.DelegateMessage> finalDelegateList = new List <Structs.DelegateMessage>();
Structs.CheckTaskingRequest req = new Structs.CheckTaskingRequest()
{
action = "get_tasking",
tasking_size = 1
};
if (DelegateMessageRequestQueue.Count > 0)
{
DelegateMessageRequestMutex.WaitOne();
req.delegates = DelegateMessageRequestQueue.ToArray();
DelegateMessageRequestQueue.Clear();
//DelegateMessageQueue = new List<Dictionary<string, string>>();
DelegateMessageRequestMutex.ReleaseMutex();
}
// Could add delegate post messages
string json = JsonConvert.SerializeObject(req);
string taskingId = Guid.NewGuid().ToString();
if (Send(taskingId, json))
{
string response = (string)Inbox.GetMessage(taskingId);
Mythic.Structs.CheckTaskingResponse resp = JsonConvert.DeserializeObject <Mythic.Structs.CheckTaskingResponse>(response);
foreach (Task task in resp.tasks)
{
Debug.WriteLine("[-] CheckTasking - NEW TASK with ID: " + task.id);
finalTaskList.Add(task);
}
if (resp.delegates != null)
{
foreach (Dictionary <string, string> delegateMessage in resp.delegates)
{
foreach (KeyValuePair <string, string> item in delegateMessage)
{
finalDelegateList.Add(new Structs.DelegateMessage()
{
UUID = item.Key,
Message = item.Value
});
}
}
}
}
result = new Structs.TaskQueue()
{
Tasks = finalTaskList.ToArray(),
Delegates = finalDelegateList.ToArray()
};
//result.Add("tasks", finalTaskList.ToArray());
//result.Add("delegates", finalDelegateList.ToArray());
//SCTask task = JsonConvert.DeserializeObject<SCTask>(Post(json));
return(result);
}
