使用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函數。