windows-server-2003/termsrv/sessdir/tssdwmi/sdwmi.cpp

/******************************************************************
   Copyright (C) 2001 Microsoft Corp.

   sdwmi.cpp -- WMI provider class implementation

   Generated by Microsoft WMI Code Generation Engine
  
   TO DO: - See individual function headers
          - When linking, make sure you link to framedyd.lib & 
            msvcrtd.lib (debug) or framedyn.lib & msvcrt.lib (retail).            

   Description: 
   
  
  
******************************************************************/
#include "stdafx.h"
#include <fwcommon.h>
#include "sdwmi.h"
#include "smartptr.h"
#include <windows.h>

LPWSTR pszUuid = L"6af13c8b-0844-4c83-9064-1892ba825527"; // From JETRPC.IDL

#define SD_QUERY_ENDPOINT_NAME L"TSSessionDirectoryQueryApi"

/****************************************************************************/
// MIDL_user_allocate
// MIDL_user_free
//
// RPC-required allocation functions.
/****************************************************************************/
void __RPC_FAR * __RPC_USER MIDL_user_allocate(size_t Size)
{
    return LocalAlloc(LMEM_FIXED, Size);
}

void __RPC_USER MIDL_user_free(void __RPC_FAR *p)
{
    LocalFree(p);
}

// Get the handle of the SD RPC server
RPC_BINDING_HANDLE OpenSDWMIServer()
{

    HRESULT hr;
    WCHAR *pBindingString = NULL;
    RPC_BINDING_HANDLE hRPCBinding = NULL;

    hr = RpcStringBindingCompose(/*(WCHAR *)g_RPCUUID,*/
                0,
                L"ncalrpc", NULL,
                SD_QUERY_ENDPOINT_NAME,    // Endpoint
                NULL, &pBindingString);

    if (hr == RPC_S_OK) {
        // Generate the RPC binding from the canonical RPC binding string.
        hr = RpcBindingFromStringBinding(pBindingString, &hRPCBinding);
        if (hr != RPC_S_OK) {
            ERR((TB,"SDWMI OpenSDWMIServer: Error %d in RpcBindingFromStringBinding\n", hr));
            goto HandleError;
        } 
    }
    else {
        ERR((TB,"SDWMI OpenSDWMIServer: Error %d in RpcStringBindingCompose\n", hr));
        goto HandleError;
    }

    hr = RpcEpResolveBinding(hRPCBinding, TSSDQUERYRPC_ClientIfHandle);
    if (hr != RPC_S_OK) {
        ERR((TB, "SDWMI OpenSDWMIServer: Error %d in RpcEpResolveBinding", hr));
        goto HandleError;
    }
HandleError:
    if (pBindingString != NULL) {
        RpcStringFree(&pBindingString);
    }
    return hRPCBinding;
}


// wrapper to make RPC call: TSSDRpcQueryAllClusterInfo
BOOL SDWMIQueryAllClusterInfo(DWORD *pNumberOfClusters,
                              TSSD_ClusterInfo **ppClusterInfo)
{
    BOOL rc = FALSE;
    RPC_BINDING_HANDLE rpcHandle;
    DWORD result;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQueryAllClusterInfo(rpcHandle, pNumberOfClusters, ppClusterInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQueryAllClusterInfo: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE; 
    }
    else {
        ERR((TB,"TSSDRpcQueryAllClusterInfo: RPC call failed, result=0x%X", result));
    }

HandleError:
    return rc;
}


// wrapper to make RPC call: TSSDRpcQueryClusterInfo
BOOL SDWMIQueryClusterInfo(WCHAR *ClusterName, 
                           DWORD *pNumberOfClusters,
                           TSSD_ClusterInfo **ppClusterInfo)
{
    BOOL rc = FALSE;
    DWORD result;
    RPC_BINDING_HANDLE rpcHandle;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQueryClusterInfo(rpcHandle, ClusterName,
                                         pNumberOfClusters, ppClusterInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQueryClusterInfo: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE;
    }
    else {
        ERR((TB,"TSSDRpcQueryClusterInfo: RPC call failed, result=0x%X", result));
    }
HandleError:
    return rc;
}


// wrapper to make RPC call: TSSDRpcQueryServerByName
BOOL SDWMIQueryServerByName(WCHAR *ServerName,
                            DWORD *pNumberOfServers,
                            TSSD_ServerInfo **ppServerInfo)
{
    BOOL rc = FALSE;
    RPC_BINDING_HANDLE rpcHandle;
    DWORD result;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQueryServerByName(rpcHandle, ServerName, pNumberOfServers, ppServerInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQueryServerByName: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE;    
    }
    else {
        ERR((TB,"TSSDRpcQueryServerByName: RPC call failed, result=0x%X", result));
    }

HandleError:
    return rc;
}

// wrapper to make RPC call: TSSDRpcQueryServersInCluster
BOOL SDWMIQueryServersInCluster(WCHAR *ClusterName,
                               DWORD *pNumberOfServers,
                               TSSD_ServerInfo **ppServerInfo)
{
    BOOL rc = FALSE;
    RPC_BINDING_HANDLE rpcHandle;
    DWORD result;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQueryServersInCluster(rpcHandle, ClusterName, pNumberOfServers, ppServerInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQueryServersInCluster: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE;    
    }
    else {
        ERR((TB,"TSSDRpcQueryServersInCluster: RPC call failed, result=0x%X", result));
    }

HandleError:
    return rc;
}


// wrapper to make RPC call: TSSDRpcQueryAllServers
BOOL SDWMIQueryAllServers(DWORD *pNumberOfServers,
                          TSSD_ServerInfo **ppServerInfo)
{
    BOOL rc = FALSE;
    RPC_BINDING_HANDLE rpcHandle;
    DWORD result;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQueryAllServers(rpcHandle, pNumberOfServers, ppServerInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQueryAllServers: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE;    
    }
    else {
        ERR((TB,"TSSDRpcQueryAllServers: RPC call failed, result=0x%X", result));
    }

HandleError:
    return rc;
}



// wrapper to make RPC call: TSSDRpcQuerySessionInfoByUserName
BOOL SDWMIQuerySessionInfoByUserName(WCHAR *UserName,
                                     WCHAR *DomainName,
                                     DWORD *pNumberOfSessions,
                                     TSSD_SessionInfo **ppSessionInfo)
{
    BOOL rc = FALSE;
    RPC_BINDING_HANDLE rpcHandle;
    DWORD result;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQuerySessionInfoByUserName(rpcHandle, UserName, DomainName, 
                                                         pNumberOfSessions, ppSessionInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQuerySessionInfoByUserName: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE;    
    }
    else {
        ERR((TB,"TSSDRpcQuerySessionInfoByUserName: RPC call failed, result=0x%X", result));
    }

HandleError:
    return rc;
}


// wrapper to make RPC call: TSSDRpcQuerySessionInfoByServer
BOOL SDWMIQuerySessionInfoByServer(WCHAR *ServerName,
                                   DWORD *pNumberOfSessions,
                                   TSSD_SessionInfo **ppSessionInfo)
{
    BOOL rc = FALSE;
    RPC_BINDING_HANDLE rpcHandle;
    DWORD result;
    unsigned long RpcException;

    rpcHandle = OpenSDWMIServer();
    if (rpcHandle == NULL) {
        goto HandleError;
    }

    RpcTryExcept {
        // Make the call.
        result = TSSDRpcQuerySessionInfoByServer(rpcHandle, ServerName, 
                                                       pNumberOfSessions, ppSessionInfo);
    }
    RpcExcept(TRUE) {
        RpcException = RpcExceptionCode();
        ERR((TB,"TSSDRpcQuerySessionInfoByServer: RPC Exception %d\n", RpcException));
        result = E_FAIL;
    } 
    RpcEndExcept

    if (SUCCEEDED(result)) {
        rc = TRUE;    
    }
    else {
        ERR((TB,"TSSDRpcQuerySessionInfoByServer: RPC call failed, result=0x%X", result));
    }

HandleError:
    return rc;
}




// TO DO: Replace "NameSpace" with the appropriate namespace for your
//        provider instance.  For instance:  "root\\default or "root\\cimv2".
//===================================================================


/***********************************************************************
    CWin32_SessionDirectoryCluster
************************************************************************/


/*****************************************************************************
 *
 *  FUNCTION    :   CWin32_SessionDirectoryCluster::CWin32_SessionDirectoryCluster
 *                  This class reads properties such as NumberOfServers, etc.
 *
 *****************************************************************************/
CWin32_SessionDirectoryCluster::CWin32_SessionDirectoryCluster (LPCWSTR lpwszName, LPCWSTR lpwszNameSpace ) : Provider( lpwszName, lpwszNameSpace )
{
    TRC2((TB, "CWin32_SessionDirectoryCluster_ctor"));       

    _tcscpy(m_szClusterName, _T("ClusterName"));

    _tcscpy(m_szNumberOfServers, _T("NumberOfServers"));

    _tcscpy(m_szSingleSessionMode, _T("SingleSessionMode"));

    m_pClusterInfo = NULL;
}

//=-------------
/*****************************************************************************
 *
 *  FUNCTION    :   CWin32_SessionDirectoryCluster::~CWin32_SessionDirectoryCluster
 *****************************************************************************/
CWin32_SessionDirectoryCluster::~CWin32_SessionDirectoryCluster ()
{
}
//=-------------
/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryCluster::EnumerateInstances
*
*  DESCRIPTION :    Returns all the instances of this class.
*
*  INPUTS      :    A pointer to the MethodContext for communication with WinMgmt.
*                   A long that contains the flags described in 
*                   IWbemServices::CreateInstanceEnumAsync.  Note that the following
*                   flags are handled by (and filtered out by) WinMgmt:
*                   WBEM_FLAG_DEEP, WBEM_FLAG_SHALLOW, WBEM_FLAG_RETURN_IMMEDIATELY, 
*                   WBEM_FLAG_FORWARD_ONLY, WBEM_FLAG_BIDIRECTIONAL
*
*  RETURNS     :    WBEM_S_NO_ERROR if successful
*
*  COMMENTS    :    All instances on the machine are returned here and
*                   all properties that this class knows how to populate must 
*                   be filled in.  If there are no instances, return 
*                   WBEM_S_NO_ERROR.  
*****************************************************************************/

HRESULT CWin32_SessionDirectoryCluster::EnumerateInstances (MethodContext* pMethodContext, long lFlags )
{
    HRESULT hr = WBEM_E_NOT_FOUND; 
        
    DWORD dwRequiredProperties = 0;
    //CHStringArray asNames;
    CHStringArray aszNames;
    DWORD i, NumberOfClusters = 0;
    WCHAR ClusterName[TSSD_NameLength];
    BOOL rc;

    ERR((TB, "EnumerateInstances in SDCluster"));

    rc = SDWMIQueryAllClusterInfo(&NumberOfClusters, &m_pClusterInfo);
    if (!rc || (NumberOfClusters == 0)) {
        goto HandleError;
    }

    // 
    for(i=0; i<NumberOfClusters; i++)
    {
        CInstance* pInstance = CreateNewInstance(pMethodContext);

        if( pInstance == NULL)
        {
           ERR((TB, "CWin32_SessionDirectoryCluster@ExecQuery: CreateNewInstance failed"));

           hr = WBEM_E_OUT_OF_MEMORY;
           goto HandleError;
        }            
  
        hr = LoadPropertyValues(pInstance, i, BIT_ALL_PROPERTIES);

        if( SUCCEEDED( hr ) )
        {
           hr = pInstance->Commit();
        }

        pInstance->Release();  
    }

    if (m_pClusterInfo != NULL) {
        MIDL_user_free(m_pClusterInfo);
        m_pClusterInfo = NULL;
    }
HandleError:
    return hr ;
}

//=---------
/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryCluster::GetObject
*
*  DESCRIPTION :    Find a single instance based on the key property, the TerminalName. 
*
*  INPUTS      :    A pointer to a CInstance object containing the key properties. 
*                   A long that contains the flags described in 
*                   IWbemServices::GetObjectAsync.  
*
*  RETURNS     :    WBEM_S_NO_ERROR if the instance can be found
*                   WBEM_E_NOT_FOUND if the instance described by the key properties 
*                   could not be found
*                   WBEM_E_FAILED if the instance could be found but another error 
*                   occurred. 
*
*****************************************************************************/

HRESULT CWin32_SessionDirectoryCluster::GetObject ( CInstance* pInstance, long lFlags, CFrameworkQuery &Query )
{
    HRESULT hr = WBEM_E_NOT_FOUND;
    CHString StrClusterName;
    DWORD dwRequiredProperties = 0;
    DWORD NumberOfClusters = 0;
    WCHAR ClusterName[TSSD_NameLength];
    BOOL rc;

    ERR((TB, "GetObject in SDCluster"));
    if( Query.IsPropertyRequired(m_szClusterName))
       dwRequiredProperties |= BIT_CLUSTERNAME;
 
    if (Query.IsPropertyRequired(m_szNumberOfServers))
       dwRequiredProperties |= BIT_NUMBEROFSERVERS;

    if (Query.IsPropertyRequired(m_szSingleSessionMode))
       dwRequiredProperties |= BIT_SINGLESESSIONMODE;

    // Get the key property
    pInstance->GetCHString(m_szClusterName, StrClusterName);

    wcsncpy(ClusterName, StrClusterName, TSSD_NameLength);
    ClusterName[TSSD_NameLength - 1] = L'\0';    
    ERR((TB,"TSSDRpcQueryClusterInfo: Query cluster name: %S", ClusterName));
    rc = SDWMIQueryClusterInfo(ClusterName, &NumberOfClusters, &m_pClusterInfo);
    ERR((TB,"ExecQuery: Get numCluster is %d", NumberOfClusters));
    if (!rc || (NumberOfClusters == 0)) {
        goto HandleError;
    }

  
    hr = LoadPropertyValues(pInstance, 0, dwRequiredProperties);

    if (m_pClusterInfo != NULL) {
        MIDL_user_free(m_pClusterInfo);
        m_pClusterInfo = NULL;
    }

HandleError:
    return hr ;
}
//=---------

/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryCluster::ExecQuery
*
*  DESCRIPTION :    The method context is passed for use in the creation of 
*                   instances that satisfy the query, and a CFrameworkQuery 
*                   which describes the query.  Create and populate all 
*                   instances which satisfy the query.  You may return more 
*                   instances or more properties than are requested and WinMgmt 
*                   will post filter out any that do not apply.
*
*  INPUTS      :    A pointer to the MethodContext for communication with WinMgmt.
*                   A query object describing the query to satisfy.
*                   A long that contains the flags described in 
*                   IWbemServices::CreateInstanceEnumAsync.  Note that the following
*                   flags are handled by (and filtered out by) WinMgmt:
*                       WBEM_FLAG_FORWARD_ONLY
*                       WBEM_FLAG_BIDIRECTIONAL
*                       WBEM_FLAG_ENSURE_LOCATABLE
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if queries not supported for 
*                       this class or if the query is too complex for this class
*                       to interpret.  The framework will call the EnumerateInstances
*                       function instead and let Winmgmt post filter.
*                   WBEM_E_FAILED if the query failed
*                   WBEM_S_NO_ERROR if query was successful 
*
*  COMMENTS    : TO DO: Most providers will not need to implement this method.  If you don't, WinMgmt 
*                       will call your enumerate function to get all the instances and perform the 
*                       filtering for you.  Unless you expect SIGNIFICANT savings from implementing 
*                       queries, you should remove this method.  You should also remove this method
*                       if you are implementing a 'method only' provider.
*
*****************************************************************************/
HRESULT CWin32_SessionDirectoryCluster::ExecQuery (MethodContext *pMethodContext, CFrameworkQuery& Query, long lFlags)
{

    HRESULT hr = WBEM_E_NOT_FOUND, result;
    DWORD dwRequiredProperties = 0;
    //CHStringArray asNames;
    CHStringArray aszNames;
    DWORD i, NumberOfClusters = 0;
    WCHAR ClusterName[TSSD_NameLength];
    BOOL rc;

    //Query.GetValuesForProp(m_szClusterName, asNames);

    //BOOL bGetAllInstances = asNames.GetSize() == 0;

    // Method 1
    if (Query.IsPropertyRequired(m_szClusterName))
        dwRequiredProperties |= BIT_CLUSTERNAME;

    if (Query.IsPropertyRequired(m_szNumberOfServers))
       dwRequiredProperties |= BIT_NUMBEROFSERVERS;

    if (Query.IsPropertyRequired(m_szSingleSessionMode))
       dwRequiredProperties |= BIT_SINGLESESSIONMODE;
    
    result = Query.GetValuesForProp(m_szClusterName, aszNames);
    if ((result != WBEM_S_NO_ERROR) ||
        (aszNames.GetSize() == 0)) {
        // No query found
        goto HandleError;
    }
        
    wcsncpy(ClusterName, aszNames.GetAt(0), TSSD_NameLength);
    ClusterName[TSSD_NameLength - 1] = L'\0';
    ERR((TB,"TSSDRpcQueryClusterInfo: Query cluster name: %S", ClusterName));

    rc = SDWMIQueryClusterInfo(ClusterName, &NumberOfClusters, &m_pClusterInfo);
    ERR((TB,"ExecQuery: Get numCluster is %d", NumberOfClusters));
    if (!rc || (NumberOfClusters == 0)) {
        goto HandleError;
    }

    // 
    for(i=0; i<NumberOfClusters; i++)
    {
        CInstance* pInstance = CreateNewInstance(pMethodContext);

        if( pInstance == NULL)
        {
           ERR((TB, "CWin32_SessionDirectoryCluster@ExecQuery: CreateNewInstance failed"));

           hr = WBEM_E_OUT_OF_MEMORY;
           goto HandleError;
        }            
  
        hr = LoadPropertyValues(pInstance, i, dwRequiredProperties);

        if( SUCCEEDED( hr ) )
        {
           hr = pInstance->Commit();
        }

        pInstance->Release();  
    }

    if (m_pClusterInfo != NULL) {
        MIDL_user_free(m_pClusterInfo);
        m_pClusterInfo = NULL;
    }

HandleError:
    return hr;
}



BOOL CWin32_SessionDirectoryCluster::IsInList(const CHStringArray &asArray, LPCWSTR pszString)
{
    DWORD dwSize = asArray.GetSize();

    for (DWORD x=0; x < dwSize; x++)
    {
        if( asArray[x].CompareNoCase(pszString) == 0 )
        {
            return TRUE;
        }
    }

    return FALSE;
}


//=---------
/*************************************************************************************
*
*  FUNCTION    : CWin32_SessionDirectoryCluster::PutInstance
*
*  DESCRIPTION :    PutInstance is in provider classes that can 
*                   write instance information back to the registry.
*
*  INPUTS      :    A pointer to a CInstance object containing the key 
*                   property - ClusterName
* 
*                   A long that contains the flags described in 
*                   IWbemServices::PutInstanceAsync.  
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if PutInstance is not available
*                   WBEM_E_FAILED if there is an error delivering the instance
*                   WBEM_E_INVALID_PARAMETER if any of the instance properties 
*                   are incorrect.
*                   WBEM_S_NO_ERROR if instance is properly delivered
*
*  COMMENTS    :    
*
***************************************************************************************/

HRESULT CWin32_SessionDirectoryCluster::PutInstance ( const CInstance &Instance, long lFlags)
{   
    return (WBEM_E_PROVIDER_NOT_CAPABLE);    
}


//=---------
/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryCluster::DeleteInstance
*
*  DESCRIPTION :    DeleteInstance, like PutInstance, actually writes information
*                   to the software or hardware.  For most hardware devices, 
*                   DeleteInstance should not be implemented, but for software
*                   configuration, DeleteInstance implementation is plausible.
*
*  INPUTS      :    A pointer to a CInstance object containing the key properties. 
*                   A long that contains the flags described in 
*                   IWbemServices::DeleteInstanceAsync.  
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if DeleteInstance is not available.
*                   WBEM_E_FAILED if there is an error deleting the instance.
*                   WBEM_E_INVALID_PARAMETER if any of the instance properties 
*                   are incorrect.
*                   WBEM_S_NO_ERROR if instance is properly deleted.
*
*  COMMENTS    : TO DO: If you don't intend to support deleting instances or are
*                       creating a 'method only' provider, remove this method.
*
*****************************************************************************/
HRESULT CWin32_SessionDirectoryCluster::DeleteInstance ( const CInstance &Instance,  long lFlags )
{
    

    return (WBEM_E_PROVIDER_NOT_CAPABLE);

}


//=---------
/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryCluster::ExecMethod
*
*  DESCRIPTION :    Override this function to provide support for methods.  
*                   A method is an entry point for the user of your provider 
*                   to request your class perform some function above and 
*                   beyond a change of state.  (A change of state should be 
*                   handled by PutInstance() )
*
*  INPUTS      :    A pointer to a CInstance containing the instance the method was executed against.
*                   A string containing the method name
*                   A pointer to the CInstance which contains the IN parameters.
*                   A pointer to the CInstance to contain the OUT parameters.
*                   A set of specialized method flags
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if not implemented for this class
*                   WBEM_S_NO_ERROR if method executes successfully
*                   WBEM_E_FAILED if error occurs executing method 
*
*  COMMENTS    :    Provides method to configure the License type base on
*                   the Terminal server mode, UseTempFolders, DeleteTempFolders
*                   and Help that are group policy based.
*
*****************************************************************************/
/*HRESULT CWin32_SessionDirectoryCluster::ExecMethod ( const CInstance& Inst,
                        const BSTR bstrMethodName,
                        CInstance *pInParams,
                        CInstance *pOutParams,
                        long lFlags)

{
    
    
    DWORD dwData = 0;
    BOOL fRet = FALSE;
    bool bRet;	
    HRESULT hr = WBEM_S_NO_ERROR;
    

	//If the method takes one or more parameters, require to check for validity of pInParams

    if(pInParams == NULL)
    {
        return WBEM_E_INVALID_METHOD_PARAMETERS;
    }
    

    if( _wcsicmp(bstrMethodName, m_szSetNumberOfSessions) == 0 )
    {   
        bRet = pInParams->GetDWORD(m_szNumberOfSessions, dwData);
        
        if ( !bRet )
        {
            return WBEM_E_INVALID_PARAMETER;
        }                    
                        
        
        // RPC function that sets the Number of Sessions                             
    
        TRC2( (TB, "CWin32_SessionDirectoryClusterSetting@ExecMethod returned 0x%x \n" , hr) );  
    
        if( ERROR_SUCCESS == hr && pOutParams != NULL )
        {
            pOutParams->SetDWORD(L"ReturnValue", WBEM_S_NO_ERROR);
        }
        else
        {                        
            hr = WBEM_E_INVALID_OPERATION;                        
        }                            
    }

    return hr;
}
        */
//=---------

HRESULT CWin32_SessionDirectoryCluster::LoadPropertyValues( CInstance *pInstance, DWORD dwIndex, DWORD dwRequiredProperties)
{      
    dwIndex;
    DWORD dwData = 0;
    DWORD dwSize = 0;
    BOOL bData = 0;
    BOOL bActivate = 0;
    DWORD dwStatus = 0;
    
    HRESULT hr = WBEM_E_INVALID_PARAMETER;
    
    // Your RPC call that gets all the data for the instance of interest. You could override the LoadPropertyValues method
    // to pass the data structure you got in GetObject, EnumerateInstances or ExecQuery

    if( dwRequiredProperties & BIT_CLUSTERNAME )
    {
        pInstance->SetCHString(m_szClusterName, m_pClusterInfo[dwIndex].ClusterName);  
        ERR((TB,"UserName is %S", m_pClusterInfo[dwIndex].ClusterName));
    }

    if( dwRequiredProperties & BIT_NUMBEROFSERVERS )
    {        
        pInstance->SetDWORD(m_szNumberOfServers, m_pClusterInfo[dwIndex].NumberOfServers);
        ERR((TB,"NumberofServers is %d", m_pClusterInfo[dwIndex].NumberOfServers));
    }

    if( dwRequiredProperties & BIT_SINGLESESSIONMODE )
    {        
        pInstance->SetDWORD(m_szSingleSessionMode, m_pClusterInfo[dwIndex].SingleSessionMode ? 1 : 0);
    }
    
    return S_OK;
}





/***********************************************************************
    CWin32_SessionDirectoryServer
************************************************************************/
    
    
    
    
/*****************************************************************************
 *
 *  FUNCTION    :   CWin32_SessionDirectoryServer::CWin32_SessionDirectoryServer
 *                  
 *
 *****************************************************************************/
CWin32_SessionDirectoryServer::CWin32_SessionDirectoryServer (LPCWSTR lpwszName, LPCWSTR lpwszNameSpace ) : Provider( lpwszName, lpwszNameSpace )
{
        TRC2((TB, "CWin32_SessionDirectoryServer_ctor"));       

        _tcscpy(m_szServerName, _T("ServerName"));

        _tcscpy(m_szServerIPAddress, _T("ServerIPAddress"));

        _tcscpy(m_szClusterName, _T("ClusterName"));

        _tcscpy(m_szNumberOfSessions, _T("NumberOfSessions"));

        _tcscpy(m_szSingleSessionMode, _T("SingleSessionMode"));

        m_pServerInfo = NULL;
}


/*****************************************************************************
 *
 *  FUNCTION    :   CWin32_SessionDirectoryServer::~CWin32_SessionDirectoryServer
 *****************************************************************************/
CWin32_SessionDirectoryServer::~CWin32_SessionDirectoryServer ()
{

}

/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryServer::EnumerateInstances
*
*  DESCRIPTION :    Returns all the instances of this class.
*
*  INPUTS      :    A pointer to the MethodContext for communication with WinMgmt.
*                   A long that contains the flags described in 
*                   IWbemServices::CreateInstanceEnumAsync.  Note that the following
*                   flags are handled by (and filtered out by) WinMgmt:
*                   WBEM_FLAG_DEEP, WBEM_FLAG_SHALLOW, WBEM_FLAG_RETURN_IMMEDIATELY, 
*                   WBEM_FLAG_FORWARD_ONLY, WBEM_FLAG_BIDIRECTIONAL
*
*  RETURNS     :    WBEM_S_NO_ERROR if successful
*
*  COMMENTS    :    All instances on the machine are returned here and
*                   all properties that this class knows how to populate must 
*                   be filled in.  If there are no instances, return 
*                   WBEM_S_NO_ERROR.  
*****************************************************************************/

HRESULT CWin32_SessionDirectoryServer::EnumerateInstances (MethodContext* pMethodContext, long lFlags )
{
    HRESULT hr = WBEM_E_NOT_FOUND; 
        
    RPC_BINDING_HANDLE rpcHandle = NULL;
    DWORD i, NumberOfServers = 0;
    BOOL rc;

    ERR((TB, "EnumerateInstances in SDServer"));

    rc = SDWMIQueryAllServers(&NumberOfServers, &m_pServerInfo);
    if (!rc || (NumberOfServers == 0)) {
        goto HandleError;
    }

    for(i=0; i<NumberOfServers; i++)
    {
        CInstance* pInstance = CreateNewInstance(pMethodContext);

        if( pInstance == NULL)
        {
           ERR((TB, "CWin32_SessionDirectoryServer@ExecQuery: CreateNewInstance failed"));

           hr = WBEM_E_OUT_OF_MEMORY;
           goto HandleError;
        }            
  
        hr = LoadPropertyValues(pInstance, i, BIT_ALL_PROPERTIES);

        if( SUCCEEDED( hr ) )
        {
           hr = pInstance->Commit();
        }

        pInstance->Release();  
    }

    if (m_pServerInfo != NULL) {
        MIDL_user_free(m_pServerInfo);
        m_pServerInfo = NULL;
    }
HandleError: 
    return hr ;
}


/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryServer::GetObject
*
*  DESCRIPTION :    Find a single instance based on the key property, the TerminalName. 
*
*  INPUTS      :    A pointer to a CInstance object containing the key properties. 
*                   A long that contains the flags described in 
*                   IWbemServices::GetObjectAsync.  
*
*  RETURNS     :    WBEM_S_NO_ERROR if the instance can be found
*                   WBEM_E_NOT_FOUND if the instance described by the key properties 
*                   could not be found
*                   WBEM_E_FAILED if the instance could be found but another error 
*                   occurred. 
*
*****************************************************************************/

HRESULT CWin32_SessionDirectoryServer::GetObject ( CInstance* pInstance, long lFlags, CFrameworkQuery &Query )
{
    HRESULT hr = WBEM_E_NOT_FOUND, result;
    DWORD dwRequiredProperties = 0;
    CHString StrServerName;
    DWORD i, NumberOfServers = 0;
    WCHAR ServerName[TSSD_NameLength];
    BOOL rc;

    if (Query.IsPropertyRequired(m_szServerName))
        dwRequiredProperties |= BIT_SERVERNAME;

    if (Query.IsPropertyRequired(m_szServerIPAddress))
        dwRequiredProperties |= BIT_SERVERIPADDRESS;

    if (Query.IsPropertyRequired(m_szClusterName))
        dwRequiredProperties |= BIT_CLUSTERNAME;

    if (Query.IsPropertyRequired(m_szNumberOfSessions))
       dwRequiredProperties |= BIT_NUMBEROFSESSIONS;

    if (Query.IsPropertyRequired(m_szSingleSessionMode))
       dwRequiredProperties |= BIT_SINGLESESSIONMODE;
    
    // Get the key property
    pInstance->GetCHString(m_szServerName, StrServerName);

    wcsncpy(ServerName, StrServerName, TSSD_NameLength);
    ServerName[TSSD_NameLength - 1] = L'\0';

    // Query by Server Name
    ERR((TB,"SDQueryServerInfo: Query server name: %S", ServerName));
                 
    rc = SDWMIQueryServerByName(ServerName, &NumberOfServers, &m_pServerInfo);
    ERR((TB,"ExecQuery: Get numServer is %d", NumberOfServers));
    if (!rc || (NumberOfServers == 0)) {
        // Not found
            goto HandleError;
    }

    hr = LoadPropertyValues(pInstance, 0, dwRequiredProperties);

    if (m_pServerInfo != NULL) {
        MIDL_user_free(m_pServerInfo);
        m_pServerInfo = NULL;
    }

HandleError:
    return hr;
}


/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryServer::ExecQuery
*
*  DESCRIPTION :    The method context is passed for use in the creation of 
*                   instances that satisfy the query, and a CFrameworkQuery 
*                   which describes the query.  Create and populate all 
*                   instances which satisfy the query.  You may return more 
*                   instances or more properties than are requested and WinMgmt 
*                   will post filter out any that do not apply.
*
*  INPUTS      :    A pointer to the MethodContext for communication with WinMgmt.
*                   A query object describing the query to satisfy.
*                   A long that contains the flags described in 
*                   IWbemServices::CreateInstanceEnumAsync.  Note that the following
*                   flags are handled by (and filtered out by) WinMgmt:
*                       WBEM_FLAG_FORWARD_ONLY
*                       WBEM_FLAG_BIDIRECTIONAL
*                       WBEM_FLAG_ENSURE_LOCATABLE
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if queries not supported for 
*                       this class or if the query is too complex for this class
*                       to interpret.  The framework will call the EnumerateInstances
*                       function instead and let Winmgmt post filter.
*                   WBEM_E_FAILED if the query failed
*                   WBEM_S_NO_ERROR if query was successful 
*
*  COMMENTS    : TO DO: Most providers will not need to implement this method.  If you don't, WinMgmt 
*                       will call your enumerate function to get all the instances and perform the 
*                       filtering for you.  Unless you expect SIGNIFICANT savings from implementing 
*                       queries, you should remove this method.  You should also remove this method
*                       if you are implementing a 'method only' provider.
*
*****************************************************************************/
HRESULT CWin32_SessionDirectoryServer::ExecQuery (MethodContext *pMethodContext, CFrameworkQuery& Query, long lFlags)
{

    HRESULT hr = WBEM_E_NOT_FOUND, result;
    DWORD dwRequiredProperties = 0;
    //CHStringArray asNames;
    CHStringArray aszNames;
    DWORD i, NumberOfServers = 0;
    WCHAR ServerName[TSSD_NameLength], ClusterName[TSSD_NameLength];
    BOOL rc;

    if (Query.IsPropertyRequired(m_szServerName))
        dwRequiredProperties |= BIT_SERVERNAME;

    if (Query.IsPropertyRequired(m_szServerIPAddress))
        dwRequiredProperties |= BIT_SERVERIPADDRESS;

    if (Query.IsPropertyRequired(m_szClusterName))
        dwRequiredProperties |= BIT_CLUSTERNAME;

    if (Query.IsPropertyRequired(m_szNumberOfSessions))
       dwRequiredProperties |= BIT_NUMBEROFSESSIONS;

    if (Query.IsPropertyRequired(m_szSingleSessionMode))
       dwRequiredProperties |= BIT_SINGLESESSIONMODE;
    
    // Query by Server Name
    result = Query.GetValuesForProp(m_szServerName, aszNames);
    if ((result == WBEM_S_NO_ERROR) &&
        (aszNames.GetSize() != 0)) {
        wcsncpy(ServerName, aszNames.GetAt(0), TSSD_NameLength);
        ServerName[TSSD_NameLength - 1] = L'\0';
        ERR((TB,"SDQueryServerInfo: Query server name: %S", ServerName));
                 
        rc = SDWMIQueryServerByName(ServerName, &NumberOfServers, &m_pServerInfo);
        ERR((TB,"ExecQuery: Get numServer is %d", NumberOfServers));
        if (!rc || (NumberOfServers == 0)) {
            // Not found
            goto HandleError;
        }
    }
    else {
        result = Query.GetValuesForProp(m_szClusterName, aszNames);
        if ((result == WBEM_S_NO_ERROR) &&
            (aszNames.GetSize() != 0)) {
            wcsncpy(ClusterName, aszNames.GetAt(0), TSSD_NameLength);
            ClusterName[TSSD_NameLength - 1] = L'\0';
            ERR((TB,"SDQueryServerInfo: Query Cluster name: %S", ClusterName));
                 
            rc = SDWMIQueryServersInCluster(ClusterName, &NumberOfServers, &m_pServerInfo);
            ERR((TB,"ExecQuery: Get numServer is %d", NumberOfServers));
            if (!rc || (NumberOfServers == 0)) {
                // Not found
                goto HandleError;
            }
        }
    }
    
    // 
    for(i=0; i<NumberOfServers; i++)
    {
        CInstance* pInstance = CreateNewInstance(pMethodContext);

        if( pInstance == NULL)
        {
           ERR((TB, "CWin32_SessionDirectoryServer@ExecQuery: CreateNewInstance failed"));

           hr = WBEM_E_OUT_OF_MEMORY;
           goto HandleError;
        }            
  
        hr = LoadPropertyValues(pInstance, i, dwRequiredProperties);

        if( SUCCEEDED( hr ) )
        {
           hr = pInstance->Commit();
        }

        pInstance->Release();  
    }

    if (m_pServerInfo != NULL) {
        MIDL_user_free(m_pServerInfo);
        m_pServerInfo = NULL;
    }

HandleError:
    return hr;
}



BOOL CWin32_SessionDirectoryServer::IsInList(const CHStringArray &asArray, LPCWSTR pszString)
{
    DWORD dwSize = asArray.GetSize();

    for (DWORD x=0; x < dwSize; x++)
    {
        if( asArray[x].CompareNoCase(pszString) == 0 )
        {
            return TRUE;
        }
    }

    return FALSE;
}


/*************************************************************************************
*
*  FUNCTION    : CWin32_SessionDirectoryServer::PutInstance
*
*  DESCRIPTION :    PutInstance is in provider classes that can 
*                   write instance information back to the registry.
*
*  INPUTS      :    A pointer to a CInstance object containing the key 
*                   property - ClusterName
* 
*                   A long that contains the flags described in 
*                   IWbemServices::PutInstanceAsync.  
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if PutInstance is not available
*                   WBEM_E_FAILED if there is an error delivering the instance
*                   WBEM_E_INVALID_PARAMETER if any of the instance properties 
*                   are incorrect.
*                   WBEM_S_NO_ERROR if instance is properly delivered
*
*  COMMENTS    :    
*
***************************************************************************************/

HRESULT CWin32_SessionDirectoryServer::PutInstance ( const CInstance &Instance, long lFlags)
{   
    return (WBEM_E_PROVIDER_NOT_CAPABLE);    
}


/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectoryServer::DeleteInstance
*
*  DESCRIPTION :    DeleteInstance, like PutInstance, actually writes information
*                   to the software or hardware.  For most hardware devices, 
*                   DeleteInstance should not be implemented, but for software
*                   configuration, DeleteInstance implementation is plausible.
*
*  INPUTS      :    A pointer to a CInstance object containing the key properties. 
*                   A long that contains the flags described in 
*                   IWbemServices::DeleteInstanceAsync.  
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if DeleteInstance is not available.
*                   WBEM_E_FAILED if there is an error deleting the instance.
*                   WBEM_E_INVALID_PARAMETER if any of the instance properties 
*                   are incorrect.
*                   WBEM_S_NO_ERROR if instance is properly deleted.
*
*  COMMENTS    : TO DO: If you don't intend to support deleting instances or are
*                       creating a 'method only' provider, remove this method.
*
*****************************************************************************/
HRESULT CWin32_SessionDirectoryServer::DeleteInstance ( const CInstance &Instance,  long lFlags )
{
    

    return (WBEM_E_PROVIDER_NOT_CAPABLE);

}



HRESULT CWin32_SessionDirectoryServer::LoadPropertyValues( CInstance *pInstance, DWORD dwIndex, DWORD dwRequiredProperties)
{      
    DWORD dwData = 0;
    DWORD dwSize = 0;
    BOOL bData = 0;
    BOOL bActivate = 0;
    DWORD dwStatus = 0;
    
    HRESULT hr = WBEM_E_INVALID_PARAMETER;
    
    if( dwRequiredProperties & BIT_SERVERNAME )
    {
        pInstance->SetCHString(m_szServerName, m_pServerInfo[dwIndex].ServerName);  
        ERR((TB,"Server is %S", m_pServerInfo[dwIndex].ServerName));
    }

    if( dwRequiredProperties & BIT_SERVERIPADDRESS )
    {
        pInstance->SetCHString(m_szServerIPAddress, m_pServerInfo[dwIndex].ServerIPAddress);  
        ERR((TB,"ServerIP is %S", m_pServerInfo[dwIndex].ServerIPAddress));
    }

    if( dwRequiredProperties & BIT_CLUSTERNAME )
    {
        pInstance->SetCHString(m_szClusterName, m_pServerInfo[dwIndex].ClusterName);  
        ERR((TB,"ClusterName is %S", m_pServerInfo[dwIndex].ClusterName));
    }

    if( dwRequiredProperties & BIT_NUMBEROFSESSIONS )
    {        
        pInstance->SetDWORD(m_szNumberOfSessions, m_pServerInfo[dwIndex].NumberOfSessions);
        ERR((TB,"NumberofSessions is %d", m_pServerInfo[dwIndex].NumberOfSessions));
    }

    if( dwRequiredProperties & BIT_SINGLESESSIONMODE )
    {        
        pInstance->SetDWORD(m_szSingleSessionMode, m_pServerInfo[dwIndex].SingleSessionMode ? 1 : 0);
    }
    
    return S_OK;
}





/***********************************************************************
    CWin32_SessionDirectorySession
************************************************************************/
    
    
    
    
/*****************************************************************************
 *
 *  FUNCTION    :   CWin32_SessionDirectorySession::CWin32_SessionDirectorySession
 *                  
 *
 *****************************************************************************/
CWin32_SessionDirectorySession::CWin32_SessionDirectorySession (LPCWSTR lpwszName, LPCWSTR lpwszNameSpace ) : Provider( lpwszName, lpwszNameSpace )
{
        TRC2((TB, "CWin32_SessionDirectorySession_ctor"));       

        _tcscpy(m_szServerName, _T("ServerName"));

        _tcscpy(m_szSessionID, _T("SessionID"));

        _tcscpy(m_szUserName, _T("UserName"));

        _tcscpy(m_szDomainName, _T("DomainName"));

        _tcscpy(m_szServerIPAddress, _T("ServerIPAddress"));

        _tcscpy(m_szTSProtocol, _T("TSProtocol"));

        _tcscpy(m_szApplicationType, _T("ApplicationType"));

        _tcscpy(m_szResolutionWidth, _T("ResolutionWidth"));

        _tcscpy(m_szResolutionHeight, _T("ResolutionHeight"));

        _tcscpy(m_szColorDepth, _T("ColorDepth"));

        _tcscpy(m_szCreateTime, _T("CreateTime"));

        _tcscpy(m_szDisconnectTime, _T("DisconnectTime"));

        _tcscpy(m_szSessionState, _T("SessionState"));

        m_pSessionInfo = NULL;
}


/*****************************************************************************
 *
 *  FUNCTION    :   CWin32_SessionDirectorySession::~CWin32_SessionDirectorySession
 *****************************************************************************/
CWin32_SessionDirectorySession::~CWin32_SessionDirectorySession ()
{

}

/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectorySession::EnumerateInstances
*
*  DESCRIPTION :    Returns all the instances of this class.
*
*  INPUTS      :    A pointer to the MethodContext for communication with WinMgmt.
*                   A long that contains the flags described in 
*                   IWbemServices::CreateInstanceEnumAsync.  Note that the following
*                   flags are handled by (and filtered out by) WinMgmt:
*                   WBEM_FLAG_DEEP, WBEM_FLAG_SHALLOW, WBEM_FLAG_RETURN_IMMEDIATELY, 
*                   WBEM_FLAG_FORWARD_ONLY, WBEM_FLAG_BIDIRECTIONAL
*
*  RETURNS     :    WBEM_S_NO_ERROR if successful
*
*  COMMENTS    :    All instances on the machine are returned here and
*                   all properties that this class knows how to populate must 
*                   be filled in.  If there are no instances, return 
*                   WBEM_S_NO_ERROR.  
*****************************************************************************/

HRESULT CWin32_SessionDirectorySession::EnumerateInstances (MethodContext* pMethodContext, long lFlags )
{
    return (WBEM_E_PROVIDER_NOT_CAPABLE);
}


/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectorySession::GetObject
*
*  DESCRIPTION :    Find a single instance based on the key property, the TerminalName. 
*
*  INPUTS      :    A pointer to a CInstance object containing the key properties. 
*                   A long that contains the flags described in 
*                   IWbemServices::GetObjectAsync.  
*
*  RETURNS     :    WBEM_S_NO_ERROR if the instance can be found
*                   WBEM_E_NOT_FOUND if the instance described by the key properties 
*                   could not be found
*                   WBEM_E_FAILED if the instance could be found but another error 
*                   occurred. 
*
*****************************************************************************/

HRESULT CWin32_SessionDirectorySession::GetObject ( CInstance* pInstance, long lFlags, CFrameworkQuery &Query )
{
    return (WBEM_E_PROVIDER_NOT_CAPABLE);
}


/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectorySession::ExecQuery
*
*  DESCRIPTION :    The method context is passed for use in the creation of 
*                   instances that satisfy the query, and a CFrameworkQuery 
*                   which describes the query.  Create and populate all 
*                   instances which satisfy the query.  You may return more 
*                   instances or more properties than are requested and WinMgmt 
*                   will post filter out any that do not apply.
*
*  INPUTS      :    A pointer to the MethodContext for communication with WinMgmt.
*                   A query object describing the query to satisfy.
*                   A long that contains the flags described in 
*                   IWbemServices::CreateInstanceEnumAsync.  Note that the following
*                   flags are handled by (and filtered out by) WinMgmt:
*                       WBEM_FLAG_FORWARD_ONLY
*                       WBEM_FLAG_BIDIRECTIONAL
*                       WBEM_FLAG_ENSURE_LOCATABLE
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if queries not supported for 
*                       this class or if the query is too complex for this class
*                       to interpret.  The framework will call the EnumerateInstances
*                       function instead and let Winmgmt post filter.
*                   WBEM_E_FAILED if the query failed
*                   WBEM_S_NO_ERROR if query was successful 
*
*  COMMENTS    : TO DO: Most providers will not need to implement this method.  If you don't, WinMgmt 
*                       will call your enumerate function to get all the instances and perform the 
*                       filtering for you.  Unless you expect SIGNIFICANT savings from implementing 
*                       queries, you should remove this method.  You should also remove this method
*                       if you are implementing a 'method only' provider.
*
*****************************************************************************/
HRESULT CWin32_SessionDirectorySession::ExecQuery (MethodContext *pMethodContext, CFrameworkQuery& Query, long lFlags)
{

    HRESULT hr = WBEM_E_NOT_FOUND, result;
    DWORD dwRequiredProperties = 0;
    //CHStringArray asNames;
    CHStringArray aszNames;
    DWORD i, NumberOfSessions = 0;
    WCHAR UserName[TSSD_NameLength], DomainName[TSSD_NameLength], ServerName[TSSD_NameLength];
    BOOL rc;
    BOOL bQueryByServerName = FALSE;


    if (Query.IsPropertyRequired(m_szServerName))
        dwRequiredProperties |= BIT_SERVERNAME;

    if (Query.IsPropertyRequired(m_szSessionID))
        dwRequiredProperties |= BIT_SESSIONID;

    if (Query.IsPropertyRequired(m_szUserName))
        dwRequiredProperties |= BIT_USERNAME;

    if (Query.IsPropertyRequired(m_szDomainName))
        dwRequiredProperties |= BIT_DOMAINNAME;

    if (Query.IsPropertyRequired(m_szServerIPAddress))
       dwRequiredProperties |= BIT_SERVERIPADDRESS;

    if (Query.IsPropertyRequired(m_szTSProtocol))
       dwRequiredProperties |= BIT_TSPROTOCOL;

    if (Query.IsPropertyRequired(m_szApplicationType))
        dwRequiredProperties |= BIT_APPLICATIONTYPE;

    if (Query.IsPropertyRequired(m_szResolutionWidth))
        dwRequiredProperties |= BIT_RESOLUTIONWIDTH;

    if (Query.IsPropertyRequired(m_szResolutionHeight))
        dwRequiredProperties |= BIT_RESOLUTIONHEIGHT;

    if (Query.IsPropertyRequired(m_szColorDepth))
        dwRequiredProperties |= BIT_COLORDEPTH;

    if (Query.IsPropertyRequired(m_szCreateTime))
        dwRequiredProperties |= BIT_CREATETIME;

    if (Query.IsPropertyRequired(m_szDisconnectTime))
        dwRequiredProperties |= BIT_DISCONNECTTIME;

    if (Query.IsPropertyRequired(m_szSessionState))
        dwRequiredProperties |= BIT_SESSIONSTATE;

    // Get the server name
    result = Query.GetValuesForProp(m_szServerName, aszNames);
    if ((result == WBEM_S_NO_ERROR) &&
        (aszNames.GetSize() != 0)) {
        bQueryByServerName = TRUE;
        wcsncpy(ServerName, aszNames.GetAt(0), TSSD_NameLength);
        ServerName[TSSD_NameLength - 1] = L'\0';
        ERR((TB, "Query sessions for servername %S", ServerName));

        rc = SDWMIQuerySessionInfoByServer(ServerName, &NumberOfSessions, &m_pSessionInfo);
        ERR((TB,"ExecQuery: Get numSession is %d", NumberOfSessions));
        if (!rc || (NumberOfSessions == 0)) {
            goto HandleError;
        }
    }
    else {
        // Get the user name
        result = Query.GetValuesForProp(m_szUserName, aszNames);
        if ((result != WBEM_S_NO_ERROR) ||
            (aszNames.GetSize() == 0)) {
            // No query found
            goto HandleError;
        }
        wcsncpy(UserName, aszNames.GetAt(0), TSSD_NameLength);
        UserName[TSSD_NameLength - 1] = L'\0';
        ERR((TB, "Query UserName is %S", UserName));

        // Get the domain name
        result = Query.GetValuesForProp(m_szDomainName, aszNames);
        if ((result != WBEM_S_NO_ERROR) ||
            (aszNames.GetSize() == 0)) {
            // No query found
            goto HandleError;
        }
        wcsncpy(DomainName, aszNames.GetAt(0), TSSD_NameLength);
        DomainName[TSSD_NameLength - 1] = L'\0';
        ERR((TB, "Query DomainName is %S", DomainName));
    
        rc = SDWMIQuerySessionInfoByUserName(UserName, DomainName, &NumberOfSessions, &m_pSessionInfo);
        ERR((TB,"ExecQuery: Get numSession is %d", NumberOfSessions));
        if (!rc || (NumberOfSessions == 0)) {
            goto HandleError;
        }
    }

    // 
    for(i=0; i<NumberOfSessions; i++)
    {
        CInstance* pInstance = CreateNewInstance(pMethodContext);

        if( pInstance == NULL)
        {
           ERR((TB, "CWin32_SessionDirectorySession@ExecQuery: CreateNewInstance failed"));

           hr = WBEM_E_OUT_OF_MEMORY;
           goto HandleError;
        }            
  
        hr = LoadPropertyValues(pInstance, i, dwRequiredProperties);

        if( SUCCEEDED( hr ) )
        {
           hr = pInstance->Commit();
        }

        pInstance->Release();  
    }

    if (m_pSessionInfo != NULL) {
        MIDL_user_free(m_pSessionInfo);
        m_pSessionInfo = NULL;
    }

HandleError: 
    return hr;
}



BOOL CWin32_SessionDirectorySession::IsInList(const CHStringArray &asArray, LPCWSTR pszString)
{
    DWORD dwSize = asArray.GetSize();

    for (DWORD x=0; x < dwSize; x++)
    {
        if( asArray[x].CompareNoCase(pszString) == 0 )
        {
            return TRUE;
        }
    }

    return FALSE;
}


/*************************************************************************************
*
*  FUNCTION    : CWin32_SessionDirectorySession::PutInstance
*
*  DESCRIPTION :    PutInstance is in provider classes that can 
*                   write instance information back to the registry.
*
*  INPUTS      :    A pointer to a CInstance object containing the key 
*                   property - ClusterName
* 
*                   A long that contains the flags described in 
*                   IWbemServices::PutInstanceAsync.  
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if PutInstance is not available
*                   WBEM_E_FAILED if there is an error delivering the instance
*                   WBEM_E_INVALID_PARAMETER if any of the instance properties 
*                   are incorrect.
*                   WBEM_S_NO_ERROR if instance is properly delivered
*
*  COMMENTS    :    
*
***************************************************************************************/

HRESULT CWin32_SessionDirectorySession::PutInstance ( const CInstance &Instance, long lFlags)
{   
    return (WBEM_E_PROVIDER_NOT_CAPABLE);    
}


/*****************************************************************************
*
*  FUNCTION    :    CWin32_SessionDirectorySession::DeleteInstance
*
*  DESCRIPTION :    DeleteInstance, like PutInstance, actually writes information
*                   to the software or hardware.  For most hardware devices, 
*                   DeleteInstance should not be implemented, but for software
*                   configuration, DeleteInstance implementation is plausible.
*
*  INPUTS      :    A pointer to a CInstance object containing the key properties. 
*                   A long that contains the flags described in 
*                   IWbemServices::DeleteInstanceAsync.  
*
*  RETURNS     :    WBEM_E_PROVIDER_NOT_CAPABLE if DeleteInstance is not available.
*                   WBEM_E_FAILED if there is an error deleting the instance.
*                   WBEM_E_INVALID_PARAMETER if any of the instance properties 
*                   are incorrect.
*                   WBEM_S_NO_ERROR if instance is properly deleted.
*
*  COMMENTS    : TO DO: If you don't intend to support deleting instances or are
*                       creating a 'method only' provider, remove this method.
*
*****************************************************************************/
HRESULT CWin32_SessionDirectorySession::DeleteInstance ( const CInstance &Instance,  long lFlags )
{ 
    return (WBEM_E_PROVIDER_NOT_CAPABLE);
}



HRESULT CWin32_SessionDirectorySession::LoadPropertyValues( CInstance *pInstance, DWORD dwIndex, DWORD dwRequiredProperties)
{      
    BOOL rc;
    
    HRESULT hr = WBEM_E_INVALID_PARAMETER;
    
    if( dwRequiredProperties & BIT_SERVERNAME )
    {
        pInstance->SetCHString(m_szServerName, m_pSessionInfo[dwIndex].ServerName);  
        ERR((TB,"Server is %S", m_pSessionInfo[dwIndex].ServerName));
    }

    if( dwRequiredProperties & BIT_SESSIONID )
    {
        pInstance->SetDWORD(m_szSessionID, m_pSessionInfo[dwIndex].SessionID);  
        ERR((TB,"Session ID is %d", m_pSessionInfo[dwIndex].SessionID));
    }

    if( dwRequiredProperties & BIT_USERNAME )
    {
        pInstance->SetCHString(m_szUserName, m_pSessionInfo[dwIndex].UserName);  
        ERR((TB,"UserName is %S", m_pSessionInfo[dwIndex].UserName));
    }

    if( dwRequiredProperties & BIT_DOMAINNAME )
    {
        pInstance->SetCHString(m_szDomainName, m_pSessionInfo[dwIndex].DomainName);  
        ERR((TB,"DomainName is %S", m_pSessionInfo[dwIndex].DomainName));
    }

    if( dwRequiredProperties & BIT_SERVERIPADDRESS )
    {
        pInstance->SetCHString(m_szServerIPAddress, m_pSessionInfo[dwIndex].ServerIPAddress);  
        ERR((TB,"ServerIP is %S", m_pSessionInfo[dwIndex].ServerIPAddress));
    }

    if( dwRequiredProperties & BIT_TSPROTOCOL )
    {
        pInstance->SetDWORD(m_szTSProtocol, m_pSessionInfo[dwIndex].TSProtocol);  
        ERR((TB,"TSProtocol is %d", m_pSessionInfo[dwIndex].TSProtocol));
    }

    if( dwRequiredProperties & BIT_APPLICATIONTYPE )
    {
        pInstance->SetCHString(m_szApplicationType, m_pSessionInfo[dwIndex].ApplicationType);  
        ERR((TB,"AppType is %S", m_pSessionInfo[dwIndex].ApplicationType));
    }

    if( dwRequiredProperties & BIT_RESOLUTIONWIDTH )
    {
        pInstance->SetDWORD(m_szResolutionWidth, m_pSessionInfo[dwIndex].ResolutionWidth);  
        ERR((TB,"ResolutionWidth is %d", m_pSessionInfo[dwIndex].ResolutionWidth));
    }

    if( dwRequiredProperties & BIT_RESOLUTIONHEIGHT )
    {
        pInstance->SetDWORD(m_szResolutionHeight, m_pSessionInfo[dwIndex].ResolutionHeight);  
        ERR((TB,"ResolutionHeight is %d", m_pSessionInfo[dwIndex].ResolutionHeight));
    }

    if( dwRequiredProperties & BIT_COLORDEPTH )
    {
        pInstance->SetDWORD(m_szColorDepth, m_pSessionInfo[dwIndex].ColorDepth);  
        ERR((TB,"ColorDepth is %d", m_pSessionInfo[dwIndex].ColorDepth));
    }

    if( dwRequiredProperties & BIT_CREATETIME )
    {
        rc = pInstance->SetDateTime(m_szCreateTime, WBEMTime(m_pSessionInfo[dwIndex].CreateTime));  
    }

    if( dwRequiredProperties & BIT_DISCONNECTTIME )
    {
        // Only set Disconnect time when the session is disconnected
        if (m_pSessionInfo[dwIndex].SessionState != 0) {
            pInstance->SetDateTime(m_szDisconnectTime, WBEMTime(m_pSessionInfo[dwIndex].DisconnectTime));  
        }
    }

    if( dwRequiredProperties & BIT_SESSIONSTATE )
    {
        pInstance->SetDWORD(m_szSessionState, m_pSessionInfo[dwIndex].SessionState ? 1 : 0);  
        ERR((TB,"SessionState is %d", m_pSessionInfo[dwIndex].SessionState));
    }   
    return S_OK;
}