使用CreateProcessAsUser和CreateEnvironmentBlock创建进程后未设置客户端名称

Question

我已经编写了在本地系统帐户下运行的C＃服务。我用它在用户登录到终端服务器时产生一个进程。该服务执行OnSessionChange方法并接收具有相应SessionID的SessionChangeDescription消息。

我使用此SessionID从WTSQueryUserToken获得用户的访问令牌。我将此令牌转换为主令牌并将其传递给CreateEnvironmentBlock以检索指向用户环境变量的指针。经过一些进一步的准备工作后，我打电话给CreateProcessAsUser函数，最终在他的winsta0\default桌面上生成我最近登录的用户的进程。

当我使用ProcessExplorer调查流程时，我发现流程上下文中没有CLIENTNAME环境变量。然而，应用程序需要这个变量。

我想知道我做错了什么。或者也许我错过了一些东西。应该加载用户配置文件，因为我在用户登录时会作出反应。

是否有可能存在计时问题？还是CLIENTNAME var以任何其他方式应用于进程？

这里是我如何调用该函数CreateEnvironmentBlock：

private static IntPtr GetEnvironmentFromToken(IntPtr token) 
    { 
     // Get a pointer to the environment variables from the specified user access token 
     IntPtr newEnvironment = IntPtr.Zero; 
     if (!WinApi.CreateEnvironmentBlock(ref newEnvironment, token, false)) 
     { 
      newEnvironment = IntPtr.Zero; 
     } 

     return newEnvironment; 
    } 

如果您需要了解更多信息或代码样本，随意问。

Answer 1

的环境变量不仅取决于从用户SID但的SessionID太多，因为有些变量是每个会话。

我们可以在注册表中查看HKEY_USERS\<SID>\Volatile Environment用户环境变量。和SessionId这里存在子密钥。下的子键 - 每个会话变量

所以CreateEnvironmentBlock必须做下一个 - 获取用户SID从令牌，开放HKEY_USERS\<SID>\Volatile Environment键，查询它的值。

则必须查询通过GetTokenInformation(hToken, TokenSessionId,)和查询Volatile Environment\SessionId子键SessionId从令牌。

但由于错误系统使用SessionId从当前进程PEB而不是从它获取它的标记。下一个代码是在系统DLL：

WCHAR buf[MAX_PATH]; 
StringCchPrintfW(buf, RTL_NUMBER_OF(buf), 
    L"%s\\%d", L"Volatile Environment", RtlGetCurrentPeb()->SessionId); 

MOV RAX，GS：[60H] // RAX - > PEB
2c0h这在PEB偏移的SessionID的

当您从服务中执行应用程序 - PEB中的SessionId为0时，结果CLIENTNAME和SESSIONNAME未添加到环境块中。

这是常见的系统错误。为测试你可以运行两个cmd.exe - 一个没有提升（执行从explorer.exe）和一个作为管理员（执行从svchost.exe -k netsvcs），然后运行在两个set命令 - 它显示环境字符串。你可以注意到在没有提升cmd.exe存在字符串SESSIONNAME=Console（或SESSIONNAME=RDP-Tcp#N如果你从rdp运行它）和，如果你在rdp，CLIENTNAME=DESKTOP-xxx。但在提升（作为管理员运行）cmd.exe - 没有这个字符串。这是因为从的svchost.exe -k netsvcs具有SessionId == 0

用于修复被称为CreateEnvironmentBlock这可以是2路：

容易，但不正确：

 _PEB* peb = RtlGetCurrentPeb(); 
     DWORD _SessionId = peb->SessionId, SessionId, rcb; 

     if (GetTokenInformation(hToken, TokenSessionId, &SessionId, sizeof(SessionId), &rcb)) 
     { 
      peb->SessionId = SessionId; 
     } 

     PVOID Environment; 
     BOOL fOk = CreateEnvironmentBlock(&Environment, hToken, FALSE); 

     peb->SessionId = _SessionId; 

想法这里 - 临时替代SessionIdPEB到SessionId从令牌。这将是工作。 bu在这里坏 - 如果另一个并行线程将使用SessionId在PEB？

另一种方式，相对较大的代码，但正确 - 你自己通过SessionId子键和扩展环境块。

void AddSessionEnv(HANDLE hToken, PVOID Environment, PVOID* pNewEnvironment) 
{ 
    SIZE_T cb = 1, len; 
    PWSTR sz = (PWSTR)Environment; 
    while (*sz) 
    { 
     len = wcslen(sz) + 1; 
     sz += len; 
     cb += len; 
    } 

    DWORD SessionId, rcb; 
    if (GetTokenInformation(hToken, TokenSessionId, &SessionId, sizeof(SessionId), &rcb)) 
    { 
     PROFILEINFO pi = { sizeof(pi), PI_NOUI, L"*" }; 
     if (LoadUserProfileW(hToken, &pi)) 
     { 
      WCHAR SubKey[48]; 
      swprintf(SubKey, L"Volatile Environment\\%d", SessionId); 
      HKEY hKey; 

      if (ERROR_SUCCESS == RegOpenKeyExW((HKEY)pi.hProfile, SubKey, 0, KEY_READ, &hKey)) 
      { 
       cb *= sizeof(WCHAR); 

       ULONG cbNeed = 0x200, cbAllocated; 
       PVOID NewEnvironment; 
       do 
       { 
        if (NewEnvironment = LocalAlloc(0, cb + (cbAllocated = cbNeed))) 
        { 
         cbNeed = AddSessionEnv(hKey, (PWSTR)NewEnvironment, cbAllocated); 

         if (cbNeed && cbAllocated >= cbNeed) 
         { 
          memcpy((PBYTE)NewEnvironment + cbNeed, Environment, cb); 
          *pNewEnvironment = NewEnvironment; 
          break; 
         } 

         LocalFree(NewEnvironment); 
        } 

       } while (cbNeed); 

       RegCloseKey(hKey); 
      } 
      UnloadUserProfile(hToken, pi.hProfile); 
     } 
    } 
} 

static volatile UCHAR guz; 

ULONG AddSessionEnv(HANDLE hKey, PWSTR sz, ULONG Length) 
{ 
    LONG status; 

    PVOID stack = alloca(guz); 

    ULONG TotalLength = 0, DataLength, Index = 0, cb = 0, rcb = sizeof(KEY_VALUE_FULL_INFORMATION) + 256; 

    union { 
     PVOID buf; 
     PKEY_VALUE_FULL_INFORMATION pkvfi; 
    }; 

    do 
    { 
     do 
     { 
      if (cb < rcb) cb = RtlPointerToOffset(buf = alloca(rcb - cb), stack); 

      if (0 <= (status = ZwEnumerateValueKey(hKey, Index, KeyValueFullInformation, buf, cb, &rcb)) && 
       pkvfi->Type == REG_SZ && 
       (DataLength = pkvfi->DataLength) && 
       !(DataLength & (sizeof(WCHAR) - 1))) 
      { 
       static const UNICODE_STRING CharSet = { 2 * sizeof(WCHAR), 2 * sizeof(WCHAR), L"="}; 

       USHORT NonInclusivePrefixLength; 
       UNICODE_STRING Name = { (USHORT)pkvfi->NameLength, Name.Length, pkvfi->Name }; 

       // not add strings which containing 0 or `=` symbol or emply 
       if (Name.Length && RtlFindCharInUnicodeString(0, &Name, &CharSet, &NonInclusivePrefixLength) == STATUS_NOT_FOUND) 
       { 
        UNICODE_STRING Value = { 
         (USHORT)DataLength, 
         Value.Length, 
         (PWSTR)RtlOffsetToPointer(pkvfi, pkvfi->DataOffset) 
        }; 

        PWSTR szEnd = (PWSTR)RtlOffsetToPointer(Value.Buffer, DataLength - sizeof(WCHAR)); 

        if (!*szEnd) Value.Length -= sizeof(WCHAR); 

        // not add empty strings or containing 0 or `=` symbol 
        if (Value.Length && RtlFindCharInUnicodeString(0, &Value, &CharSet, &NonInclusivePrefixLength) == STATUS_NOT_FOUND) 
        { 
         ULONG cbNeed = Name.Length + 2 * sizeof(WCHAR) + Value.Length; 

         if (Length >= cbNeed) 
         { 
          sz += 1 + swprintf(sz, L"%wZ=%wZ", &Name, &Value), Length -= cbNeed; 
         } 
         else 
         { 
          Length = 0; 
         } 

         TotalLength += cbNeed; 
        } 
       } 
      } 

     } while (status == STATUS_BUFFER_OVERFLOW || status == STATUS_BUFFER_TOO_SMALL); 

     Index++; 

    } while (status != STATUS_NO_MORE_ENTRIES); 

    return TotalLength; 
} 

，并使用此代码：

 PVOID Environment, NewEnvironment = 0; 

     if (CreateEnvironmentBlock(&Environment, hToken, FALSE)) 
     { 
      AddSessionEnv(hToken, Environment, &NewEnvironment); 

      CreateProcessAsUserW(hToken, *, CREATE_UNICODE_ENVIRONMENT, 
       NewEnvironment ? NewEnvironment : Environment, *); 

      if (NewEnvironment) 
      { 
       LocalFree(NewEnvironment); 
      } 
      DestroyEnvironmentBlock(Environment); 
     } 

的RtlFindCharInUnicodeString的定义，我用，对于舒适

enum { 
    RTL_FIND_CHAR_IN_UNICODE_STRING_START_AT_END = 1, 
    RTL_FIND_CHAR_IN_UNICODE_STRING_COMPLEMENT_CHAR_SET = 2, 
    RTL_FIND_CHAR_IN_UNICODE_STRING_CASE_INSENSITIVE = 4 
}; 

NTSYSAPI 
NTSTATUS 
NTAPI 
RtlFindCharInUnicodeString(
           ULONG Flags, 
           PCUNICODE_STRING StringToSearch, 
           PCUNICODE_STRING CharSet, 
           USHORT *NonInclusivePrefixLength 
           ); 

Answer 2

经过一些实验，看起来CreateEnvironmentBlock只设置CLIENTNAME如果进程在与令牌关联的同一个远程桌面会话中运行。假冒没有任何区别。这可以说是Windows中的一个错误。

要解决此问题，您可以自己将CLIENTNAME添加到环境块中，也可以在用户会话中启动进程以代表您调用CreateEnvironmentBlock函数。