112 lines
2.4 KiB
C++
112 lines
2.4 KiB
C++
/*++
|
|
|
|
Copyright (c) 1994 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
username.hxx
|
|
|
|
Abstract:
|
|
|
|
Class for managing user name
|
|
|
|
Author:
|
|
|
|
Richard L Firth (rfirth) 30-Aug-1997
|
|
|
|
|
|
Revision History:
|
|
|
|
30-Aug-1997 rfirth
|
|
Created
|
|
|
|
--*/
|
|
|
|
class CUserName {
|
|
|
|
private:
|
|
|
|
CRITICAL_SECTION m_CritSec;
|
|
ICSTRING m_UserName;
|
|
BOOL m_fInitialized;
|
|
|
|
public:
|
|
|
|
CUserName() {
|
|
InitializeCriticalSection(&m_CritSec);
|
|
m_UserName = NULL;
|
|
m_fInitialized = TRUE;
|
|
}
|
|
|
|
~CUserName() {
|
|
if (m_fInitialized) {
|
|
// CRT may result in the destructor being called twice.
|
|
// Prevent the delete critsec from happening a second time.
|
|
DeleteCriticalSection(&m_CritSec);
|
|
m_fInitialized = FALSE;
|
|
}
|
|
}
|
|
|
|
VOID Set(VOID) {
|
|
|
|
EnterCriticalSection(&m_CritSec);
|
|
if (!m_UserName.HaveString()) {
|
|
|
|
char buf[256];
|
|
DWORD length = sizeof(buf);
|
|
|
|
if (GetUserName(buf, &length)) {
|
|
m_UserName = buf;
|
|
}
|
|
}
|
|
LeaveCriticalSection(&m_CritSec);
|
|
}
|
|
|
|
LPSTR Get(LPSTR lpszBuffer, DWORD dwLength) {
|
|
|
|
EnterCriticalSection(&m_CritSec);
|
|
if (!m_UserName.HaveString()) {
|
|
Set();
|
|
}
|
|
if ((m_UserName.StringLength() < (int)dwLength) && (dwLength > 0)) {
|
|
m_UserName.CopyTo(lpszBuffer, dwLength);
|
|
} else {
|
|
if (dwLength > 0) {
|
|
*lpszBuffer = '\0';
|
|
}
|
|
lpszBuffer = "";
|
|
}
|
|
LeaveCriticalSection(&m_CritSec);
|
|
return lpszBuffer;
|
|
}
|
|
|
|
LPSTR Get(LPSTR lpszBuffer, LPDWORD lpdwLength) {
|
|
|
|
EnterCriticalSection(&m_CritSec);
|
|
if (!m_UserName.HaveString()) {
|
|
Set();
|
|
}
|
|
|
|
DWORD dwLength = *lpdwLength;
|
|
|
|
if ((m_UserName.StringLength() < (int)dwLength) && (dwLength != 0)) {
|
|
m_UserName.CopyTo(lpszBuffer, lpdwLength);
|
|
} else {
|
|
if (dwLength > 0) {
|
|
*lpszBuffer = '\0';
|
|
}
|
|
lpszBuffer = "";
|
|
*lpdwLength = 0;
|
|
}
|
|
LeaveCriticalSection(&m_CritSec);
|
|
return lpszBuffer;
|
|
}
|
|
|
|
VOID Clear(VOID) {
|
|
|
|
EnterCriticalSection(&m_CritSec);
|
|
m_UserName = NULL;
|
|
LeaveCriticalSection(&m_CritSec);
|
|
}
|
|
};
|