131 lines
4.2 KiB
Plaintext
131 lines
4.2 KiB
Plaintext
#if 0
|
|
[Code]
|
|
#endif
|
|
|
|
function Quote(const S: String): String;
|
|
begin
|
|
Result := '"' + S + '"';
|
|
end;
|
|
|
|
|
|
function AddParam(const S, P, V: String): String;
|
|
begin
|
|
if V <> '""' then
|
|
Result := S + ' /' + P + '=' + V;
|
|
end;
|
|
|
|
|
|
function AddSimpleParam(const S, P: String): String;
|
|
begin
|
|
Result := S + ' /' + P;
|
|
end;
|
|
|
|
|
|
procedure CreateRunOnceEntry;
|
|
var RunOnceData, SetupRestartData: String;
|
|
i: Integer;
|
|
begin
|
|
DebugMsg('CreateRunOnceEntry','Preparing for restart');
|
|
|
|
//RunOnce command-line is limited to 256 characters, so keep it to the bare minimum required to start setup
|
|
RunOnceData := Quote(ExpandConstant('{srcexe}')) + ' /resumeinstall=1';
|
|
RunOnceData := AddParam(RunOnceData, 'LANG', ExpandConstant('{language}'));
|
|
|
|
SetupRestartData := Quote(ExpandConstant('{srcexe}')) + ' /resumeinstall=2';
|
|
SetupRestartData := AddParam(SetupRestartData, 'LANG', ExpandConstant('{language}'));
|
|
SetupRestartData := AddParam(SetupRestartData, 'DIR', Quote(WizardDirValue));
|
|
//SetupRestartData := AddParam(SetupRestartData, 'GROUP', Quote(WizardGroupValue));
|
|
//if WizardNoIcons then
|
|
// SetupRestartData := AddSimpleParam(SetupRestartData, 'NOICONS');
|
|
SetupRestartData := AddParam(SetupRestartData, 'TYPE', Quote(WizardSetupType(False)));
|
|
SetupRestartData := AddParam(SetupRestartData, 'COMPONENTS', Quote(WizardSelectedComponents(False)));
|
|
SetupRestartData := AddParam(SetupRestartData, 'TASKS', Quote(WizardSelectedTasks(False)));
|
|
|
|
if Force32bitInstall then
|
|
SetupRestartData := AddSimpleParam(SetupRestartData, '32');
|
|
|
|
if ExpandConstant('{param:log|*}') <> '*' then
|
|
begin
|
|
SetupRestartData := AddParam(SetupRestartData, 'LOG', Quote(ExpandConstant('{param:log|*}')));
|
|
end else
|
|
begin
|
|
for i := 0 to ParamCount() do
|
|
if LowerCase(ParamStr(i)) = '/log' then
|
|
begin
|
|
RunOnceData := AddSimpleParam(RunOnceData,'LOG'); //multiple logs are created in %TEMP% when no filename is given
|
|
SetupRestartData := AddSimpleParam(SetupRestartData,'LOG');
|
|
break;
|
|
end;
|
|
end;
|
|
|
|
DebugMsg('CreateRunOnceEntry','RunOnce: ' + RunOnceData);
|
|
RegWriteStringValue(HKLM, 'Software\Microsoft\Windows\CurrentVersion\RunOnce', RunOnceName, RunOnceData);
|
|
|
|
DebugMsg('CreateRunOnceEntry','RunOnce: ' + SetupRestartData);
|
|
RegWriteStringValue(HKLM, 'Software\' + RunOnceName, '', SetupRestartData);
|
|
end;
|
|
|
|
|
|
procedure RestartMyself();
|
|
var CmdLine: String;
|
|
ResultCode: Integer;
|
|
begin
|
|
DebugMsg('RestartMyself','Continuing install after reboot (reexecuting setup)');
|
|
|
|
if RegValueExists(HKLM, 'Software\' + RunOnceName, '') then
|
|
begin
|
|
if RegQueryStringValue(HKLM, 'Software\' + RunOnceName, '', CmdLine) then
|
|
begin
|
|
RegDeleteKeyIncludingSubkeys(HKLM, 'Software\' + RunOnceName); //clean up
|
|
if not Exec('>',CmdLine,'',SW_SHOW,ewNoWait,ResultCode) then //bonus: don't block shell from loading, since RunOnce installer exits immediately
|
|
MsgBox(FmtMessage(CustomMessage('ErrorRestartingSetup'),[IntToStr(ResultCode)]), mbError, mb_Ok);
|
|
|
|
DebugMsg('RestartMyself','Result of running ' + CmdLine + ': ' + IntToStr(ResultCode));
|
|
|
|
end else
|
|
begin
|
|
MsgBox(FmtMessage(CustomMessage('ErrorRestartingSetup'),['-2']), mbError, mb_Ok);
|
|
DebugMsg('RestartMyself','Error reading HKLM\'+RunOnceName);
|
|
end;
|
|
end else
|
|
begin
|
|
MsgBox(FmtMessage(CustomMessage('ErrorRestartingSetup'),['-1']), mbError, mb_Ok);
|
|
DebugMsg('RestartMyself','HKLM\'+RunOnceName + ' not found in Registy');
|
|
end;
|
|
end;
|
|
|
|
|
|
function RestartSetupAfterReboot(): Boolean;
|
|
begin
|
|
|
|
if ExpandConstant('{param:resumeinstall|0}') = '1' then //called from RunOnce
|
|
begin
|
|
Result := False; //setup will just re-execute itself in this run
|
|
|
|
RestartMyself();
|
|
|
|
DebugMsg('RestartSetupAfterReboot','Phase 1 complete, exiting');
|
|
exit;
|
|
|
|
end else
|
|
if ExpandConstant('{param:resumeinstall|0}') = '2' then //setup re-executed itself
|
|
begin
|
|
|
|
Result := True;
|
|
InstallMode := imRebootContinue;
|
|
DebugMsg('RestartSetupAfterReboot','Continuing install after reboot');
|
|
|
|
end else
|
|
begin
|
|
Result := True; //normal install
|
|
end;
|
|
|
|
if InstallMode <> imRebootContinue then
|
|
if RegValueExists(HKLM, 'Software\Microsoft\Windows\CurrentVersion\RunOnce', RunOnceName) then
|
|
begin
|
|
DebugMsg('RestartSetupAfterReboot','System must be restarted first');
|
|
MsgBox(CustomMessage('RebootRequiredFirst'), mbError, mb_Ok);
|
|
Result := False;
|
|
end;
|
|
end;
|
