windows specific session, fix conn count (#7133)

Signed-off-by: 21pages <pages21@163.com>
This commit is contained in:
21pages
2024-02-15 10:44:32 +08:00
committed by GitHub
parent 4bf3764b5d
commit 73d429d064
2 changed files with 7 additions and 3 deletions

View File

@@ -464,7 +464,7 @@ extern "C"
void get_available_session_ids(PWSTR buf, uint32_t bufSize, BOOL include_rdp) {
std::vector<std::wstring> sessionIds;
PWTS_SESSION_INFOA pInfos;
PWTS_SESSION_INFOA pInfos = NULL;
DWORD count;
if (WTSEnumerateSessionsA(WTS_CURRENT_SERVER_HANDLE, 0, 1, &pInfos, &count)) {
@@ -498,7 +498,7 @@ extern "C"
}
if (buf && !tmpStr.empty() && tmpStr.size() < bufSize) {
memcpy(buf, tmpStr.c_str(), (tmpStr.size() + 1) * sizeof(wchar_t));
wcsncpy_s(buf, bufSize, tmpStr.c_str(), tmpStr.size());
}
}