小小地实现一个 whoami

Windows 里有一个程序叫做 whoami.exe，它用来显示调用者的用户身份。whoami.exe 的功能很多，在这里我只实现它最基本的功能——显示当前登录的用户（NTLM 格式）。

思路：首先知道 LookupAccountSid 函数可以获取我们所需的用户名和登陆域->LookupAccountSid 通过 SID 来查询信息->可以通过 GetTokenInformation 查询进程令牌里的 SID->用 OpenProcessToken 可以获取进程的令牌。

代码(在 Delphi XE2 上编译通过)：

function GetProcessIdentity(): String;
var
  hToken: THandle;
  UserName, UserDomain: String;
  cbName, cbDomainName: ULONG;
  ReturnLength: DWORD;
  Buff: array of Byte;
  tu: PTokenUser;
  peUse: SID_NAME_USE;
label
  Cleanup;
begin
  Result := '';
  // 打开进程令牌
  if not OpenProcessToken(GetCurrentProcess, MAXIMUM_ALLOWED, hToken) then Exit;

  // 查询用户账户令牌

  // 首先获取需要的缓冲区大小
  if not GetTokenInformation(hToken, TokenUser, nil, 0, ReturnLength) then
    if GetLastError = ERROR_INSUFFICIENT_BUFFER then
    begin
      // 设置缓冲区大小
      SetLength(Buff, ReturnLength);
      // 获取数据
      GetTokenInformation(hToken, TokenUser, @Buff[0], ReturnLength, ReturnLength);
      tu := PTokenUser(@Buff[0]);
    end
    else
      goto Cleanup
  else
    goto Cleanup;

  // 通过 SID 查询用户名及登陆域
  cbName := 0;
  cbDomainName := 0;
  // 获取需要的缓冲区大小
  if not LookupAccountSid(nil, tu.User.Sid, nil, cbName, nil, cbDomainName, peUse) then
    if GetLastError = ERROR_INSUFFICIENT_BUFFER then
    begin
      // 设置字符串长度(包含 NULL 字符)
      SetLength(UserName, cbName);
      SetLength(UserDomain, cbDomainName);
      // 获取数据
      if LookupAccountSid(nil, tu.User.Sid, @UserName[1], cbName, @UserDomain[1], cbDomainName, peUse) then
      begin
        // 截去最后的 NULL 字符
        SetLength(UserName, cbName);
        SetLength(UserDomain, cbDomainName);
      end
      else
        goto Cleanup;
    end
    else
      goto Cleanup
  else
    goto Cleanup;

  // 组合信息
  Result := UserDomain + '\' + UserName;

Cleanup:
  // 关闭令牌
  CloseHandle(hToken);
end;

运行结果： 

whoami.exe 的其他功能都是通过 GetTokenInformation 和 LookupAccountSid 这两个函数完成的，至于其他功能的实现，我将在以后更新。