Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

111 lines
2.4 KiB

/*++
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);
}
};