Enumerating Logon Sessions

#ifndef __LOGONSESSIONDATA_H__
#define __LOGONSESSIONDATA_H__

/*
Copyright 2004 by Stefan Kuhr
*/


#include <ntsecapi.h>
#include <wtsapi32.h>

// The following constant may be defined by including NtStatus.h.

#ifndef STATUS_SUCCESS
#define STATUS_SUCCESS ((NTSTATUS)0x00000000L)
#endif

#define HAVE_RECENT_PLATFORMSDK 1

#if !HAVE_RECENT_PLATFORMSDK
typedef struct _SECURITY_LOGON_SESSION_DATA {
    ULONG               Size ;
    LUID                LogonId ;
    LSA_UNICODE_STRING  UserName ;
    LSA_UNICODE_STRING  LogonDomain ;
    LSA_UNICODE_STRING  AuthenticationPackage ;
    ULONG               LogonType ;
    ULONG               Session ;
    PSID                Sid ;
    LARGE_INTEGER       LogonTime ;

    //
    // new for whistler:
    //

    LSA_UNICODE_STRING  LogonServer ;
    LSA_UNICODE_STRING  DnsDomainName ;
    LSA_UNICODE_STRING  Upn ;
} SECURITY_LOGON_SESSION_DATA, * PSECURITY_LOGON_SESSION_DATA ;
#endif

class CLogonSessionData
{
    /// class member variables:
    LPWSTR m_szUserName;
    LPWSTR m_szLogonDomain;
    LPWSTR m_szAuthenticationPackage;;
    ULONG m_ulLogonType;
    ULONG m_ulSession;
    LPWSTR m_szUserSID;
    LARGE_INTEGER m_liLogonTime;
    DWORD m_dwFlags;
    LUID m_Luid;

    /// operators and member functions
    void SetFlags(DWORD dw){m_dwFlags|=dw;}
    BOOL Initialize(const PSECURITY_LOGON_SESSION_DATA);
    BOOL Initialize(HANDLE hToken, PLUID pLogonId, PWTS_PROCESS_INFO pWtsInfo);

protected:
    /// 
    /// copy ctor and assignement operator made protected: If you really
    /// insist on tampering with CLogonSessionData objects, then derive
    /// your own class from it and do whatever you want!
    /// 

    CLogonSessionData(const CLogonSessionData &);
    CLogonSessionData & operator = (const CLogonSessionData &me);

public:
    CLogonSessionData(void);    
    ~CLogonSessionData(void);    
    DWORD GetFlags(void)const{return m_dwFlags;}

    LPCWSTR GetUserName(void)const {return m_szUserName;}
    LPCWSTR GetLogonDomain(void)const {return m_szLogonDomain;}
    LPCWSTR GetAuthenticationPackage(void)const {return m_szAuthenticationPackage;}
    LPCWSTR GetUserSID(void)const {return m_szUserSID;}
    ULONG GetLogonType(void)const {return m_ulLogonType;}
    ULONG GetTSSession(void) const {return m_ulSession;}
    LARGE_INTEGER GetLogonTime(void)const{return m_liLogonTime;}
    LUID GetLuid(void)const{return m_Luid;}
    enum Flags
    {
        VOID_STATE  = 0,          
        INITIALIZED = 1,
        STALE       = 2
    };

    /// 
    /// We make EnumLogonSessions/EnumNT4StyleInteractiveSessions a "friend"  
    /// function in order to be able to make SetFlags and Initialize private. 
    /// This way, ctors and dtor are the only  non-const public member 
    /// functions which is a very desirable effect.
    /// 

    friend BOOL EnumLogonSessions(CLogonSessionData *pLogonSessionData, PULONG lpulSessions);
    friend BOOL EnumNT4StyleInteractiveSessions(CLogonSessionData *pLogonSessionData, PULONG lpulSessions);
};



#endif /// __LOGONSESSIONDATA_H__