void bw_DoWorkPreExecute(object sender, DoWorkEventArgs e)
{
// simply update the progress bar to show the 3rd step is entirely complete
bw.ReportProgress(0, new object[] { GetRelativeProgess(3, 0), 0, string.Empty, ProgressStatus.None, null });
List <UninstallFileInfo> rollbackCOM = new List <UninstallFileInfo>();
Exception except = null;
for (int i = 0; i < UpdtDetails.UpdateFiles.Count; i++)
{
bool unregister = (UpdtDetails.UpdateFiles[i].RegisterCOMDll &
(COMRegistration.UnRegister | COMRegistration.PreviouslyRegistered)) != 0;
// skip non-executing files, skip execute "after" updates
if ((!UpdtDetails.UpdateFiles[i].Execute || !UpdtDetails.UpdateFiles[i].ExBeforeUpdate) && !unregister)
{
continue;
}
// form the absolute path of the file to execute or unregister
string fullFile = FixUpdateDetailsPaths(UpdtDetails.UpdateFiles[i].RelativePath);
if (string.IsNullOrEmpty(fullFile))
{
continue;
}
if (!unregister)
{
try
{
// we only support starting non-elevated on Vista+
// And the user must be an admin (otherwise starting as the same elevation as wyUpdate is ample)
if (UpdtDetails.UpdateFiles[i].ElevationType == ElevationType.NotElevated &&
IsAdmin &&
VistaTools.AtLeastVista())
{
int exitCode = (int)LimitedProcess.Start(fullFile,
string.IsNullOrEmpty(UpdtDetails.UpdateFiles[i].CommandLineArgs)
? null
: ParseText(UpdtDetails.UpdateFiles[i].CommandLineArgs), false,
UpdtDetails.UpdateFiles[i].WaitForExecution,
UpdtDetails.UpdateFiles[i].ProcessWindowStyle);
// if we're rolling back on non-zero return codes, the return code is non-zero, and it's not in the exception list
if (UpdtDetails.UpdateFiles[i].RollbackOnNonZeroRet && exitCode != 0 && (UpdtDetails.UpdateFiles[i].RetExceptions == null ||
!UpdtDetails.UpdateFiles[i].RetExceptions.Contains(exitCode)))
{
except = new Exception("\"" + fullFile + "\" returned " + exitCode + ".");
break;
}
}
else // Same as wyUpdate or elevated
{
ProcessStartInfo psi = new ProcessStartInfo
{
FileName = fullFile,
WindowStyle = UpdtDetails.UpdateFiles[i].ProcessWindowStyle
};
// command line arguments
if (!string.IsNullOrEmpty(UpdtDetails.UpdateFiles[i].CommandLineArgs))
{
psi.Arguments = ParseText(UpdtDetails.UpdateFiles[i].CommandLineArgs);
}
// only elevate if the current process isn't already elevated
if (!IsAdmin && UpdtDetails.UpdateFiles[i].ElevationType == ElevationType.Elevated)
{
psi.Verb = "runas";
psi.ErrorDialog = true;
psi.ErrorDialogParentHandle = MainWindowHandle;
}
//start the process
Process p = Process.Start(psi);
if (UpdtDetails.UpdateFiles[i].WaitForExecution && p != null)
{
p.WaitForExit();
// if we're rolling back on non-zero return codes, the return code is non-zero, and it's not in the exception list
if (UpdtDetails.UpdateFiles[i].RollbackOnNonZeroRet && p.ExitCode != 0 && (UpdtDetails.UpdateFiles[i].RetExceptions == null ||
!UpdtDetails.UpdateFiles[i].RetExceptions.Contains(p.ExitCode)))
{
except = new Exception("\"" + psi.FileName + "\" returned " + p.ExitCode + ".");
break;
}
}
}
}
catch (Exception ex)
{
// failure when executing the file
except = new Exception("Failed to execute the file \"" + fullFile + "\": " + ex.Message, ex);
break;
}
}
else // unregistering DLL
{
try
{
RegisterDllServer(fullFile, true);
// add to the rollback list
rollbackCOM.Add(new UninstallFileInfo {
Path = fullFile, RegisterCOMDll = COMRegistration.Register
});
}
catch (Exception ex)
{
except = ex;
break;
}
}
}
// save rollback info
RollbackUpdate.WriteRollbackCOM(Path.Combine(TempDirectory, "backup\\unreggedComList.bak"), rollbackCOM);
if (IsCancelled() || except != null)
{
// rollback unregged COM
bw.ReportProgress(1, false);
RollbackUpdate.RollbackUnregedCOM(TempDirectory);
// rollback stopped services
RollbackUpdate.RollbackStoppedServices(TempDirectory);
bw.ReportProgress(0, new object[] { -1, -1, string.Empty, ProgressStatus.Failure, except });
}
else
{
// registry modification completed sucessfully
bw.ReportProgress(0, new object[] { -1, -1, string.Empty, ProgressStatus.Success, null });
}
}
void bw_DoWorkOptimizeExecute(object sender, DoWorkEventArgs e)
{
// simply update the progress bar to show the 6th step is entirely complete
bw.ReportProgress(0, new object[] { GetRelativeProgess(6, 0), 0, string.Empty, ProgressStatus.None, null });
List <UninstallFileInfo> rollbackCOM = new List <UninstallFileInfo>();
List <string> startedServices = new List <string>();
Exception except = null;
//optimize everything but "temp" files
for (int i = 0; i < UpdtDetails.UpdateFiles.Count; i++)
{
if (UpdtDetails.UpdateFiles[i].IsNETAssembly || (UpdtDetails.UpdateFiles[i].RegisterCOMDll & COMRegistration.Register) == COMRegistration.Register)
{
if (IsCancelled())
{
break;
}
//if not a temp file
if (UpdtDetails.UpdateFiles[i].RelativePath.Length >= 4 &&
UpdtDetails.UpdateFiles[i].RelativePath.Substring(0, 4) != "temp")
{
//optimize (ngen) the file
string filename = FixUpdateDetailsPaths(UpdtDetails.UpdateFiles[i].RelativePath);
if (UpdtDetails.UpdateFiles[i].IsNETAssembly)
{
//TODO: add proper rolling back of newly NGENed files
if (!string.IsNullOrEmpty(filename))
{
NGenInstall(filename, UpdtDetails.UpdateFiles[i]); //optimize the file
}
}
else
{
try
{
RegisterDllServer(filename, false);
// add to the rollback list
rollbackCOM.Add(new UninstallFileInfo {
Path = filename, RegisterCOMDll = COMRegistration.UnRegister
});
}
catch (Exception ex)
{
except = ex;
break;
}
}
}
}
}
RollbackUpdate.WriteRollbackCOM(Path.Combine(TempDirectory, "backup\\reggedComList.bak"), rollbackCOM);
bw.ReportProgress(0, new object[] { GetRelativeProgess(6, 50), 50, string.Empty, ProgressStatus.None, null });
if (!IsCancelled() && except == null)
{
// execute files
for (int i = 0; i < UpdtDetails.UpdateFiles.Count; i++)
{
// skip non-executing files, skip execute "before" updates
if (!UpdtDetails.UpdateFiles[i].Execute || UpdtDetails.UpdateFiles[i].ExBeforeUpdate)
{
continue;
}
// form the absolute path of the file to execute
string fileToExec = FixUpdateDetailsPaths(UpdtDetails.UpdateFiles[i].RelativePath);
if (string.IsNullOrEmpty(fileToExec))
{
continue;
}
try
{
// we only support starting non-elevated on Vista+
// And the user must be an admin (otherwise starting as the same elevation as wyUpdate is ample)
if (UpdtDetails.UpdateFiles[i].ElevationType == ElevationType.NotElevated &&
IsAdmin &&
VistaTools.AtLeastVista())
{
int exitCode = (int)LimitedProcess.Start(fileToExec,
string.IsNullOrEmpty(UpdtDetails.UpdateFiles[i].CommandLineArgs)
? null
: ParseText(UpdtDetails.UpdateFiles[i].CommandLineArgs), false,
UpdtDetails.UpdateFiles[i].WaitForExecution,
UpdtDetails.UpdateFiles[i].ProcessWindowStyle);
// if we're rolling back on non-zero return codes, the return code is non-zero, and it's not in the exception list
if (UpdtDetails.UpdateFiles[i].RollbackOnNonZeroRet && exitCode != 0 && (UpdtDetails.UpdateFiles[i].RetExceptions == null ||
!UpdtDetails.UpdateFiles[i].RetExceptions.Contains(exitCode)))
{
except = new Exception("\"" + fileToExec + "\" returned " + exitCode + ".");
break;
}
}
else
{
ProcessStartInfo psi = new ProcessStartInfo
{
FileName = fileToExec,
WindowStyle = UpdtDetails.UpdateFiles[i].ProcessWindowStyle
};
// command line arguments
if (!string.IsNullOrEmpty(UpdtDetails.UpdateFiles[i].CommandLineArgs))
{
psi.Arguments = ParseText(UpdtDetails.UpdateFiles[i].CommandLineArgs);
}
// only elevate if the current process isn't already elevated
if (!IsAdmin && UpdtDetails.UpdateFiles[i].ElevationType == ElevationType.Elevated)
{
psi.Verb = "runas";
psi.ErrorDialog = true;
psi.ErrorDialogParentHandle = MainWindowHandle;
}
// start the process
Process p = Process.Start(psi);
if (UpdtDetails.UpdateFiles[i].WaitForExecution && p != null)
{
p.WaitForExit();
// if we're rolling back on non-zero return codes, the return code is non-zero, and it's not in the exception list
if (UpdtDetails.UpdateFiles[i].RollbackOnNonZeroRet && p.ExitCode != 0 && (UpdtDetails.UpdateFiles[i].RetExceptions == null ||
!UpdtDetails.UpdateFiles[i].RetExceptions.Contains(p.ExitCode)))
{
except = new Exception("\"" + psi.FileName + "\" returned " + p.ExitCode + ".");
break;
}
}
}
}
catch (Exception ex)
{
// failure when executing the file
except = new Exception("Failed to execute the file \"" + fileToExec + "\": " + ex.Message, ex);
break;
}
}
}
if (!IsCancelled() && except == null)
{
try
{
// try to start services
foreach (StartService service in UpdtDetails.ServicesToStart)
{
// skip the start service if it will be started as part of the auto-update process
if (SkipStartService != null && string.Compare(SkipStartService, service.Name, StringComparison.OrdinalIgnoreCase) == 0)
{
continue;
}
using (ServiceController srvc = new ServiceController(service.Name))
{
ServiceControllerStatus status = srvc.Status;
if (status != ServiceControllerStatus.Running)
{
if (service.Arguments != null)
{
// parse the arguments for variables
for (int i = 0; i < service.Arguments.Length; i++)
{
service.Arguments[i] = ParseText(service.Arguments[i]);
}
// start the service with the arguments
srvc.Start(service.Arguments);
}
else // no arguments
{
srvc.Start();
}
// report that we're waiting for the service to start so the user knows what's going on
bw.ReportProgress(0, new object[] { GetRelativeProgess(6, 50), 50, "Waiting for service to start: " + srvc.DisplayName, ProgressStatus.None, null });
//TODO: periodically check if the user has cancelled -- for services that fail to start
//srvc.WaitForStatus(ServiceControllerStatus.Running, TimeSpan.FromSeconds(5));
srvc.WaitForStatus(ServiceControllerStatus.Running);
startedServices.Add(service.Name);
}
}
}
}
catch (Exception ex)
{
except = ex;
}
// save rollback info
RollbackUpdate.WriteRollbackServices(Path.Combine(TempDirectory, "backup\\startedServices.bak"), startedServices);
}
if (IsCancelled() || except != null)
{
// tell the main window we're rolling back registry
bw.ReportProgress(1, true);
// rollback started services
RollbackUpdate.RollbackStartedServices(TempDirectory);
// rollback newly regged COM dlls
RollbackUpdate.RollbackRegedCOM(TempDirectory);
// rollback the registry
RollbackUpdate.RollbackRegistry(TempDirectory);
//rollback files
bw.ReportProgress(1, false);
RollbackUpdate.RollbackFiles(TempDirectory, ProgramDirectory);
// rollback unregged COM
RollbackUpdate.RollbackUnregedCOM(TempDirectory);
// rollback stopped services
RollbackUpdate.RollbackStoppedServices(TempDirectory);
bw.ReportProgress(0, new object[] { -1, -1, string.Empty, ProgressStatus.Failure, except });
}
else
{
bw.ReportProgress(0, new object[] { -1, -1, string.Empty, ProgressStatus.Success, null });
}
}