185 lines
3.0 KiB
C++
185 lines
3.0 KiB
C++
/*
|
|
* Session.h
|
|
*
|
|
* Author: BreenH
|
|
*
|
|
* The Session class provides a level of separation between the winstation
|
|
* structure and the policy modules. It is just a wrapper class; it does not
|
|
* create or destroy winstation stuctures.
|
|
*/
|
|
|
|
#ifndef __LC_SESSION_H__
|
|
#define __LC_SESSION_H__
|
|
|
|
|
|
/*
|
|
* Typedefs
|
|
*/
|
|
|
|
typedef struct {
|
|
CRITICAL_SECTION CritSec;
|
|
class CPolicy *pPolicy;
|
|
ULONG ulClientProtocolVersion;
|
|
HANDLE hProtocolLibContext;
|
|
BOOL fTsLicense;
|
|
BOOL fLlsLicense;
|
|
LS_HANDLE hLlsLicense;
|
|
LPARAM lPrivate;
|
|
} LCCONTEXT, *LPLCCONTEXT;
|
|
|
|
/*
|
|
* Class Definition
|
|
*/
|
|
|
|
class CSession
|
|
{
|
|
public:
|
|
|
|
/*
|
|
* Creation Functions
|
|
*/
|
|
|
|
CSession(
|
|
PWINSTATION pWinStation
|
|
)
|
|
{
|
|
m_pWinStation = pWinStation;
|
|
}
|
|
|
|
~CSession(
|
|
)
|
|
{
|
|
m_pWinStation = NULL;
|
|
}
|
|
|
|
/*
|
|
* Get Functions
|
|
*/
|
|
|
|
inline HANDLE
|
|
GetIcaStack(
|
|
) const
|
|
{
|
|
return(m_pWinStation->hStack);
|
|
}
|
|
|
|
inline LPLCCONTEXT
|
|
GetLicenseContext(
|
|
) const
|
|
{
|
|
return((LPLCCONTEXT)(m_pWinStation->lpLicenseContext));
|
|
}
|
|
|
|
inline ULONG
|
|
GetLogonId(
|
|
) const
|
|
{
|
|
return(m_pWinStation->LogonId);
|
|
}
|
|
|
|
inline LPCWSTR
|
|
GetUserDomain(
|
|
) const
|
|
{
|
|
return((LPCWSTR)(m_pWinStation->Domain));
|
|
}
|
|
|
|
inline LPCWSTR
|
|
GetUserName(
|
|
) const
|
|
{
|
|
return((LPCWSTR)(m_pWinStation->UserName));
|
|
}
|
|
|
|
/*
|
|
* Is Functions
|
|
*/
|
|
|
|
inline BOOLEAN
|
|
IsConsoleSession(
|
|
) const
|
|
{
|
|
return((BOOLEAN)(GetCurrentConsoleId() == m_pWinStation->LogonId));
|
|
}
|
|
|
|
inline BOOLEAN
|
|
IsSessionZero(
|
|
) const
|
|
{
|
|
return((BOOLEAN)((0 == m_pWinStation->LogonId)
|
|
|| (m_pWinStation->bClientSupportsRedirection
|
|
&& m_pWinStation->bRequestedSessionIDFieldValid
|
|
&& (0 == m_pWinStation->RequestedSessionID))));
|
|
}
|
|
|
|
inline BOOLEAN
|
|
IsUserAdmin(
|
|
) const
|
|
{
|
|
return(m_pWinStation->fUserIsAdmin);
|
|
}
|
|
|
|
inline BOOL
|
|
IsUserHelpAssistant(
|
|
) const
|
|
{
|
|
return TSIsSessionHelpSession( m_pWinStation, NULL );
|
|
}
|
|
|
|
|
|
/*
|
|
* Do Functions
|
|
*/
|
|
|
|
inline NTSTATUS
|
|
SendWinStationCommand(
|
|
PWINSTATION_APIMSG pMsg
|
|
)
|
|
{
|
|
//
|
|
// Wait time must be zero, or termsrv will release the winstation,
|
|
// causing who knows what to happen to our state.
|
|
//
|
|
|
|
return(::SendWinStationCommand(m_pWinStation, pMsg, 0));
|
|
}
|
|
|
|
//
|
|
// ASSUMPTION: This function will be
|
|
// called with the stack lock already held
|
|
//
|
|
inline NTSTATUS
|
|
SetErrorInfo(
|
|
UINT32 dwErr
|
|
)
|
|
{
|
|
if(m_pWinStation->pWsx &&
|
|
m_pWinStation->pWsx->pWsxSetErrorInfo &&
|
|
m_pWinStation->pWsxContext)
|
|
{
|
|
return m_pWinStation->pWsx->pWsxSetErrorInfo(
|
|
m_pWinStation->pWsxContext,
|
|
dwErr,
|
|
TRUE); //lock already held
|
|
}
|
|
else
|
|
{
|
|
return STATUS_INVALID_PARAMETER;
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
* Set Functions
|
|
*/
|
|
|
|
|
|
private:
|
|
|
|
PWINSTATION m_pWinStation;
|
|
|
|
};
|
|
|
|
#endif
|
|
|