public static ISofOperation PrepareRebootResults()
{
ISofOperation operation = new SofOperation();
var location = Path.Combine(Settings.AgentDirectory, "rebootoperation.data");
if (!File.Exists(location))
{
return(null);
}
var operationId = File.ReadAllText(location);
Logger.Log("Found reboot operation, preparing to send back results for operation id {0}", LogLevel.Info, operationId);
var rebooted = (IsBootUp().ToLower() == "yes") ? true.ToString().ToLower() : false.ToString().ToLower();
var json = new JObject();
json["operation"] = "reboot";
json["operation_id"] = operationId;
json["success"] = (String.IsNullOrEmpty(rebooted)) ? "no" : rebooted;
operation.Id = operationId;
operation.Api = ApiCalls.CoreRebootResults();
operation.Type = "reboot";
operation.RawResult = json.ToString();
File.Delete(location);
Logger.Log("Deleted reboot operation file, sending back results.");
return(operation);
}
private void SendNewUpdatesHandler(object sender, ElapsedEventArgs e)
{
var operation = new SofOperation
{
Plugin = "rv",
Type = RvOperationValue.UpdatesAndApplications,
Api = ApiCalls.RvUpdatesApplications()
};
RegisterOperation(operation);
RunOperation(operation);
}
private string RestoreInfoResult(SofOperation operation)
{
JObject json = new JObject();
JArray jsonArray = new JArray();
json.Add(OperationKey.Operation, OperationValue.WindowsRestoreInfo);
json.Add(OperationKey.OperationId, operation.OperationId);
json.Add(OperationKey.AgentId, AgentSettings.AgentId);
// If there are no system restores, then return empty list.
if (operation.RestoreList == null)
{
JObject jObject = JObject.Parse(String.Empty);
jsonArray.Add(jObject);
}
else
{
foreach (WindowsRestoreData data in operation.RestoreList)
{
StringBuilder builder = new StringBuilder("{");
builder.AppendFormat(@" ""description"" : ""{0}"", ", data.Description);
builder.AppendFormat(@" ""creation_time"" : ""{0}"", ", data.CreationTime);
builder.AppendFormat(@" ""sequence_number"" : ""{0}"", ", data.SequenceNumber);
builder.Append("}");
JObject jObject = JObject.Parse(builder.ToString());
jsonArray.Add(jObject);
}
}
json.Add(OperationKey.Data, jsonArray);
return json.ToString();
}
/// <summary>
/// Find the Update that corresponds to the vendor ID by matching TopPatchIDs.
/// </summary>
/// <param name="operation">The operation used to iterate over the UpdateList.</param>
/// <param name="vendorId">String used for the VendorId.</param>
/// <returns></returns>
private Update FindUpdateByVendorId(SofOperation operation, string vendorId)
{
// Find the Update that corresponds to the vendor ID by matching UpdateIds.
return operation.UpdateList.Find(
delegate(Update up)
{
return up.VendorId == vendorId;
});
}
/// <summary>
/// Method called in the background to actually install the downloaded updates.
/// </summary>
/// <param name="operation"></param>
/// <param name="updatesToInstall"></param>
/// <returns></returns>
private SofOperation Install(SofOperation operation, UpdateCollection updatesToInstall)
{
AgentSettings.Log("Installing...");
foreach (IUpdate update in updatesToInstall)
{
// TODO Need a way to send back to server for approval of EULA...
if (update.IsDownloaded)
{
if (update.EulaAccepted == false)
{
update.AcceptEula();
}
}
}
IUpdateInstaller2 installer = (IUpdateInstaller2)session.CreateUpdateInstaller();
installer.ForceQuiet = true;
installer.Updates = updatesToInstall;
IInstallationResult installationRes = installer.Install();
for (int i = 0; i < updatesToInstall.Count; i++)
{
// Find the Update that corresponds to updatesToInstall[i] by matching UpdateIds.
Update update = FindUpdateByVendorId(operation, updatesToInstall[i].Identity.UpdateID);
SofResult results;
results.TopPatchId = update.TopPatchId;
if (installationRes.GetUpdateResult(i).HResult == 0)
{ // Success!
results.Successful = true;
results.Restart = installationRes.GetUpdateResult(i).RebootRequired;
results.SpecificMessage = null;
// Update SQLiteDB
SQLiteHelper.UpdateToggle(SQLiteHelper.WindowsUpdateColumn.Installed, true, results.TopPatchId);
SQLiteHelper.UpdateDate(SQLiteHelper.WindowsUpdateColumn.DateInstalled, DateTime.Now.ToShortDateString(), results.TopPatchId);
}
else
{
// Failed...
results.Successful = false;
results.Restart = false;
results.SpecificMessage = installationRes.GetUpdateResult(i).HResult.ToString();
// Update SQLiteDB
SQLiteHelper.UpdateToggle(SQLiteHelper.WindowsUpdateColumn.Installed, false, results.TopPatchId);
}
operation.SetResult(results);
}
return operation;
}
/// <summary>
/// Determines whether to perform a system restore or send data back to the server. It use WMI.
/// </summary>
/// <param name="operation"></param>
/// <returns></returns>
public SofOperation WindowsRestore(SofOperation operation)
{
SofResult results;
ManagementClass restoreClass = new ManagementClass("\\\\.\\root\\default", "systemrestore", new System.Management.ObjectGetOptions());
ManagementObjectCollection restoreCollection = restoreClass.GetInstances();
foreach (ManagementObject restoreItem in restoreCollection)
{
// Possible properties. See: http://msdn.microsoft.com/en-us/library/windows/desktop/aa378925(v=vs.85).aspx
//(string)restoreItem["Description"]
//(uint)restoreItem["RestorePointType"]).ToString()
//(uint)restoreItem["EventType"]).ToString()
//(uint)restoreItem["SequenceNumber"]).ToString()
//(string)restoreItem["CreationTime"]
// Crazy way to call a method for a WMI class through System.Management.
// See: http://msdn.microsoft.com/en-us/library/ms257364(v=vs.80).aspx
if (((uint)restoreItem["SequenceNumber"]) == operation.WindowsRestoreSequence)
{
ManagementBaseObject inputParameters = restoreClass.GetMethodParameters("Restore");
inputParameters["SequenceNumber"] = operation.WindowsRestoreSequence;
try
{
ManagementBaseObject outputParameters = restoreClass.InvokeMethod("Restore", inputParameters, null);
if (Convert.ToInt32(outputParameters["returnValue"]) == 0)
{
// Success! Restart system for restore point can take affect.
RestartSystem();
return null;
}
else
{
// Failed...
results.TopPatchId = null;
results.Successful = false;
results.Restart = false;
// Ummmm from the docs: "If the method succeeds, the return value is S_OK (0).
// Otherwise, the method returns one of the COM error codes defined in WinError.h."
// Yayyyyy.... (/s + April face)
int exitCode = Convert.ToInt32(outputParameters["returnValue"]);
results.SpecificMessage = "Win32 Error: " + new Win32Exception(exitCode).Message;
operation.SetResult(results);
}
}
catch (ManagementException e)
{
AgentSettings.Log("Exception: {0}", AgentLogLevel.Error, e.Message);
if (e.InnerException != null)
{
AgentSettings.Log("Inner exception: {0}", AgentLogLevel.Error, e.InnerException.Message);
}
AgentSettings.Log("Failed to perform a system restore.", AgentLogLevel.Error);
results.TopPatchId = null;
results.Successful = false;
results.Restart = false;
results.SpecificMessage = String.Format("ManagementException Error: ", e);
}
operation.SetResult(results);
return operation;
}
}
results.TopPatchId = null;
results.Successful = false;
results.Restart = false;
results.SpecificMessage = String.Format("No restore point with sequence number {0} was found.", operation.WindowsRestoreSequence);
operation.SetResult(results);
return operation;
}
public void UninstallUpdates(SofOperation operation)
{
// Can't use IUpdateInstaller.Uninstall() because it will fail if it does not go through WSUS.
// Work around by using hacks provided in the nested Uninstaller class
AgentSettings.Log("Uninstalling...");
UninstallerResults uninstallerResults;
SofResult results;
foreach (Update update in operation.UpdateList)
{
// Make sure to set all results properties or the previous value will roll over
// to the next update's results!!
results.TopPatchId = update.TopPatchId;
uninstallerResults = windowsUninstaller.Uninstall(update);
if (uninstallerResults.Sucess)
{ // Success!
results.Successful = true;
results.Restart = uninstallerResults.Restart;
results.SpecificMessage = uninstallerResults.Message;
// Update SQLiteDB
SQLiteHelper.UpdateToggle(SQLiteHelper.WindowsUpdateColumn.Installed, false, results.TopPatchId);
}
else
{ // Fail...
results.Successful = false;
results.Restart = uninstallerResults.Restart;
results.SpecificMessage = uninstallerResults.Message;
}
operation.SetResult(results);
}
SQLiteHelper.RecreateUpdatesTable();
CheckForInstalledUpdates();
GetPendingUpdates();
if (UninstallOperationCompleted != null)
UninstallOperationCompleted(operation);
}
/// <summary>
/// "Un-hides" updates that are hidden and/or ignored.
/// </summary>
/// <param name="operation"></param>
public void ShowUpdates(SofOperation operation)
{
UpdateCollection updateCollection = RetrieveUpdates(operation.UpdateList, OperationValue.Show);
Update tpUpdate;
foreach (IUpdate iUpdate in updateCollection)
{
// This is all it takes...
iUpdate.IsHidden = false;
tpUpdate = FindUpdateByVendorId(operation, iUpdate.Identity.UpdateID);
// Seems like a lot for a simple operation.
SofResult results;
results.TopPatchId = tpUpdate.TopPatchId;
results.Successful = true;
results.Restart = false;
results.SpecificMessage = null;
operation.SetResult(results);
// Update SQLiteDB
SQLiteHelper.UpdateToggle(SQLiteHelper.WindowsUpdateColumn.Hidden, false, results.TopPatchId);
}
if (ShowOperationCompleted != null)
ShowOperationCompleted(operation);
}
public void InstallUpdates(SofOperation operation)
{
if (updateWorker.IsBusy != true)
{
updateWorker.RunWorkerAsync(operation);
}
AgentSettings.Log("UpdateWorker is busy.");
}
public string CustomDataResults(SofOperation operation)
{
string jsonString = String.Empty;
switch (operation.Operation)
{
case OperationValue.WindowsRestoreInfo:
jsonString = RestoreInfoResult(operation);
break;
}
return jsonString;
}
