8

我有 2 个应用程序,program.exe 和 updater.exe,它们都是用 Delphi5 编写的。程序在没有管理员权限(也没有清单)的情况下运行,更新程序有一个带有“requireAdministrator”的清单,因为他必须能够在 Program-Folder 写入来更新 program.exe。

问题是启动更新程序并让他等到程序关闭。我在网上找到了不同的方法,但没有一种方法有效(在大多数情况下,第一个应用程序启动第二个应用程序并等待第二个应用程序的结束,在我的情况下,第二个应用程序应该等待第一个应用程序的结束)。

更新程序应该等待,这很容易
updater.exe

{$R manifest.res}
label.caption:='Wait for program.exe closing';
repeat
    sleep(1000);
until File is not open
ProgramHandle := Read Handle from File
WaitForSingleObject(ProgramHandle,INFINITE);
label.caption:='program.exe CLOSED';
Do updates

方式 1
使用 CreateProcess 启动更新程序:
program.exe

FillChar(siInfo, SizeOf(siInfo), 0);
siInfo.cb := SizeOf(siInfo);

saProcessAttributes.nLength := SizeOf(saProcessAttributes);
saProcessAttributes.lpSecurityDescriptor := nil;
saProcessAttributes.bInheritHandle := TRUE;

saThreadAttributes.nLength := SizeOf(saThreadAttributes);
saThreadAttributes.lpSecurityDescriptor := nil;
saThreadAttributes.bInheritHandle := True;

if CreateProcess(nil,
           PChar('updater.exe'),
           @saProcessAttributes,
           @saThreadAttributes,
           TRUE, NORMAL_PRIORITY_CLASS, nil,
           PChar(ExtractFilePath(Application.ExeName)),
           siInfo, piInfo) then
begin
DuplicateHandle(GetCurrentProcess, GetCurrentProcess,
                piInfo.hProcess, @MyHandle,
                PROCESS_QUERY_INFORMATION, TRUE,
                DUPLICATE_SAME_ACCESS) then
Write MyHandle in a File
end;
Close program

不做任何事情,仅当更新程序没有带有 requireAdministrator 的清单时才有效。如果我使用 explizit 管理员权限运行程序,它也可以工作。

方式 2 使用 ShellExecuteEx 启动更新程序:
program.exe

  FillChar(Info, SizeOf(Info), Chr(0));
  Info.cbSize := SizeOf(Info);
  Info.fMask := SEE_MASK_NOCLOSEPROCESS;
  Info.lpVerb := PChar('runas');
  Info.lpFile := PChar('update.exe');
  Info.lpDirectory := nil;
  Info.nShow := SW_RESTORE;
  ShellExecuteEx(@Info);

  MyHandle:=OpenProcess(PROCESS_ALL_ACCESS, False, GetCurrentProcessId())));
  Write MyHandle in a File
  Close program

不起作用,MyHandle 每次运行此过程时都有不同的值(无需重新启动程序),因此更新程序无法使用它。

所以我不知道如何启动 updater.exe 并将 program.exe 的句柄写入文件中。

我对编程的这些部分不太熟悉......有人对我的问题有想法吗?

4

3 回答 3

7

您的代码不起作用,因为句柄表是每个进程的,这意味着第二个进程可能具有指向另一个内核对象的相同句柄。下面,有许多可能的解决方案之一:

创建进程 2 时,将进程 1 的 PID 作为参数传递:

procedure CreateUpdater;
var
  Info: TShellExecuteInfo;
begin  
  FillChar(Info, SizeOf(TShellExecuteInfo), 0);
  Info.cbSize := SizeOf(TShellExecuteInfo);
  Info.fMask  := SEE_MASK_NOCLOSEPROCESS;
  Info.lpVerb := PChar('runas');
  Info.lpFile := PChar('Update.exe');
  Info.lpParameters := PChar(IntToStr(GetCurrentProcessId));
  Info.lpDirectory := nil;
  Info.nShow := SW_RESTORE;
  ShellExecuteEx(@Info);
  //NOTE: MISSING ERROR CHECKING!
end;

在 Updater 中,等待 process1 终止:

procedure WaitForAndClose;
var
  PID: String;
  AHandle: Cardinal;
  Ret: longbool;
  ExitNumber: DWORD;
begin
  PID:= ParamStr(1);
  if PID <> '' then
  begin
    AHandle:= OpenProcess(PROCESS_QUERY_INFORMATION, False, StrToInt(PID));  
    //NOTE: MISSING ERROR CHECKING!
    try
      repeat
        Ret:= GetExitCodeProcess(AHandle, ExitNumber);
        //NOTE: MISSING ERROR CHECKING!
        Sleep(1000); //define a time to poolling
      until (ExitNumber <> STILL_ACTIVE);
    finally
      CloseHandle(AHandle);
    end;
    //Terminate the process;
    Application.Terminate;        
  end;
end;

您还可以使用WaitForSingleObject来避免轮询:

WaitForSingleObject(AHandle, INFINITE);
//NOTE: MISSING ERROR CHECKING!

但是您需要 SYNCHRONIZE 访问权限才能打开该进程:

AHandle:= OpenProcess(SYNCHRONIZE, False, StrToInt(PID));
//NOTE: MISSING ERROR CHECKING!

注意:这里没有错误检查。您应该阅读文档并正确检查错误。

注意 2:我想让您注意您正在泄漏手柄的事实。当你使用SEE_MASK_NOCLOSEPROCESS调用者负责关闭 calee 的句柄。在你的情况下,我认为你根本不需要那个面具。我会删除它。

于 2015-06-10T15:51:18.330 回答
1

以下是如何使用事件实现此目的的基本示例:

program.exe

  // manual-reset event, non-signaled 
  Event := CreateEvent(nil, True, False, 'MyUniqueName');
  ExecuteUpdater; // via ShellExecuteEx with runas
  // synchronize - wait for the event to be signaled
  WaitForSingleObject(Event, INFINITE);
  // WAIT_OBJECT_0 = The state of the specified object is signaled.  
  CloseHandle(Event);

updater.exe

  Event := CreateEvent(nil, True, False, 'MyUniqueName');
  if Event = 0 then RaiseLastWin32Error;
  SetEvent(Event); // sets the event object to the signaled state
  CloseHandle(Event);

您还应该向program.exe( requestedExecutionLevelshould be level="asInvoker") 添加清单以避免虚拟化。

于 2015-06-11T08:50:18.767 回答
0

我在两个事件的不确定顺序中看到了主要问题:关闭程序和启动更新程序主代码。

解决它的一种可能方法是使用事件 - https://msdn.microsoft.com/ru-ru/library/windows/desktop/ms686670(v=vs.85).aspx

程序创建事件(带有一个让孩子继承它的选项),然后启动更新程序(通过命令行将事件和程序进程的句柄作为整数传递给它!),然后在事件的 WaitForSingleObject 中冻结.

这确保程序在更新程序准备好监视它之前不会退出,因此 PID 不会变得无效。

然后更新程序在从命令行获得的程序 PID 上调用 OpenProcess,然后调用 SignalAndWait 敲击事件(从命令行获得)和冻结句柄(从 OpenProcess 获得) - https://msdn.microsoft.com/ ru-ru/库/windows/desktop/ms686293(v=vs.85).aspx

现在从等待事件中释放的程序终止。进程的终止发出信号,所以现在更新程序被依次释放并可以开始做主要工作。


C++建议的另一种方法,如何确定 Windows 进程是否正在运行?正在查询程序ProcessID的退出代码- 据说当程序仍在运行时,会有一个特定的错误代码,您可以 Sleep(100) 然后再试一次。任何其他结果意味着程序已经完成。

程序在启动更新程序后立即退出,无需等待它开始监控。

这似乎是不错的方法,只是我现在不保证不会重复使用 PID 值。机会微乎其微,但仍不为零。


就个人而言,我可能会使用标志文件。CreateFile API 有一个非常有趣的标志——临时文件模式。这意味着,Windows 将在进程结束后自动删除该文件。那么那么

  1. 该程序使用 Windows API(或使用 Crypto API 的新随机值)创建一个新的 GUID。
  2. 程序在临时文件夹中创建临时模式文件,其名称基于 GUID 或随机值。如果运气好的话,这样的文件已经存在 - 您只需获得一个新的 GUID 或随机值。
  3. 程序启动更新程序并通过命令行将文件名传递给它
  4. 程序在启动更新程序后立即退出,无需等待它开始监控。
  5. 更新程序不断检查文件是否存在(在尝试之间暂停)。当文件不再存在时 - 这意味着程序已经完成并且 Windows 自动删除了它。

同样,在程序终止和更新程序再次检查之间,其他进程创建具有完全相同名称的标志文件的可能性很小,但这在实践中几乎是不可能的

于 2015-06-10T15:50:44.770 回答