windows-nt/Source/XPSP1/NT/termsrv/admtools/tscc/snapin/dataobj.cpp

//Copyright (c) 1998 - 1999 Microsoft Corporation
#include "stdafx.h"
#include "tscc.h"
#include "dataobj.h"
#include "resource.h"
#include "rnodes.h"

extern const GUID GUID_ResultNode;

static UINT s_cfInternal;// = RegisterClipboardFormat( TEXT( "TSCC" ) );   

static UINT s_cfDisplayName;// = RegisterClipboardFormat( CCF_DISPLAY_NAME );

static UINT s_cfNodeType;// = RegisterClipboardFormat( CCF_NODETYPE );

static UINT s_cfSnapinClsid;// = RegisterClipboardFormat( CCF_SNAPIN_CLASSID );

static UINT s_cfSZNodeType;// = RegisterClipboardFormat( CCF_SZNODETYPE );

static UINT s_cfSZWinstaName;


//--------------------------------------------------------------------------
// ctor
//--------------------------------------------------------------------------
CBaseNode::CBaseNode( )
{
    // The ndmgr gets the dataobj via IComponent and then calls release
    // so dataobj should have an implicit addref

    m_cref = 1;

    if( s_cfInternal == 0 )
    {
        s_cfInternal = RegisterClipboardFormat( TEXT( "TSCC" ) );   
    }

    if( s_cfSZWinstaName == 0 )
    {
        s_cfSZWinstaName = RegisterClipboardFormat( TEXT( "TSCC_WINSTANAME" ) );
    }

    if( s_cfDisplayName == 0 )
    {
        s_cfDisplayName = RegisterClipboardFormat( CCF_DISPLAY_NAME );
    }

    if( s_cfNodeType == 0 )
    {
        s_cfNodeType = RegisterClipboardFormat( CCF_NODETYPE );
    }

    if( s_cfSnapinClsid == 0 )
    {
        s_cfSnapinClsid = RegisterClipboardFormat( CCF_SNAPIN_CLASSID );
    }

    if( s_cfSZNodeType == 0 )
    {
        s_cfSZNodeType = RegisterClipboardFormat( CCF_SZNODETYPE );
    }

	m_nNodeType = 0;
	
}

//--------------------------------------------------------------------------
// Standard QI behavior
//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::QueryInterface( REFIID riid , PVOID *ppv )
{
    if( riid == IID_IUnknown )
    {
        *ppv = ( LPUNKNOWN )this;
    }
    else if( riid == IID_IDataObject )
    {
        *ppv = ( LPDATAOBJECT )this;
    }
    else
    {
        *ppv = NULL;

        return E_NOINTERFACE;
    }

    AddRef( );

    return S_OK;
}

//--------------------------------------------------------------------------
// Standard addref
//--------------------------------------------------------------------------
STDMETHODIMP_( ULONG )CBaseNode::AddRef(  )
{
    return InterlockedIncrement( ( LPLONG )&m_cref );
}

//--------------------------------------------------------------------------
// Same as addref no need for cs
//--------------------------------------------------------------------------
STDMETHODIMP_( ULONG )CBaseNode::Release( )
{
    if( InterlockedDecrement( ( LPLONG )&m_cref ) == 0 )
    {
         ODS( L"CBaseNode -- Releasing Dataobj\n" );

        delete this;

        return 0;
    }

    return m_cref;
}

//--------------------------------------------------------------------------
// Trust me ndmgr will call this with a fury
//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::GetDataHere( LPFORMATETC pF , LPSTGMEDIUM pMedium)
{
    HRESULT hr = DV_E_FORMATETC;
        
    const CLIPFORMAT cf = pF->cfFormat;
  
    IStream *pStream = NULL;
  
    pMedium->pUnkForRelease = NULL;
    

    hr = CreateStreamOnHGlobal( pMedium->hGlobal, FALSE, &pStream );

    if( SUCCEEDED( hr ) ) 
    {
        if( cf == s_cfDisplayName )
        {
            TCHAR szDispname[ 128 ]; // = TEXT("Terminal Server Connection Configuration" );

            VERIFY_E( 0 , LoadString( _Module.GetResourceInstance( ) , IDS_NAMESTRING , szDispname , SIZE_OF_BUFFER( szDispname ) ) );

            // Include null terminator 

            hr = pStream->Write( szDispname , SIZE_OF_BUFFER( szDispname )/* + sizeof( TCHAR )*/ , NULL );
        }
        else if( cf == s_cfInternal )
        {
            // The nodemgr will use this copy and pass it back to us in
            // functions such as ::Notify

            ODS( L"GetDataHere -- s_cfInternal used\n" );

            hr = pStream->Write( this , sizeof( CBaseNode ) , NULL );
        }
        else if( cf == s_cfSZWinstaName )
        {
            CResultNode *pNode = dynamic_cast< CResultNode *>( this );

            hr = E_FAIL; // generic failure

            // if we're talking about a connection base node get the winstation name

            ODS( L"GetDataHere -- current winstaname\n" );

            if( pNode != NULL )
            {
                LPTSTR szConName = pNode->GetConName( );

                hr = pStream->Write( szConName , lstrlen( szConName ) * sizeof( TCHAR ) + sizeof( TCHAR ) , NULL  );
            }
        }            
        else if( cf == s_cfNodeType )
        {
            const GUID *pGuid = NULL;

			if( GetNodeType( ) == MAIN_NODE )
			{
			    ODS( L"GetDataHere -- NodeType is MAIN_NODE\n" );

				pGuid = &GUID_MainNode;
			}
            else if( GetNodeType( ) == SETTINGS_NODE )
            {
                ODS( L"GetDataHere -- NodeType is SETTINGS_NODE\n" );

                pGuid = &GUID_SettingsNode;
            }
			else if( GetNodeType( ) == RESULT_NODE )
			{
				ODS( L"GetDataHere -- NodeType is RESULT_NODE\n" );

				pGuid = &GUID_ResultNode;
			}
			else
			{
				ODS( L"GetDataHere -- NodeType is userdefined\n ");

				pGuid = &GUID_ResultNode;
			}

            hr = pStream->Write( ( PVOID )pGuid , sizeof( GUID ) , NULL );
        }
		else if( cf == s_cfSZNodeType )
        {
            TCHAR szGUID[ 40 ];

			if( GetNodeType( ) == MAIN_NODE )
			{
				StringFromGUID2( GUID_MainNode , szGUID , SIZE_OF_BUFFER( szGUID ) );
			}
            else if( GetNodeType( ) == SETTINGS_NODE )
            {
                StringFromGUID2( GUID_SettingsNode , szGUID , SIZE_OF_BUFFER( szGUID ) );
            }                
			else if( GetNodeType( ) == RESULT_NODE )
			{
				StringFromGUID2( GUID_ResultNode , szGUID , SIZE_OF_BUFFER( szGUID ) );
			}
			else
			{
				StringFromGUID2( GUID_ResultNode , szGUID , SIZE_OF_BUFFER( szGUID ) );
			}

            // write nodetype in String format -- ok

            hr = pStream->Write( szGUID , sizeof( szGUID ) , NULL );
        }
        else if( cf == s_cfSnapinClsid )
        {
            // write out snapin's clsid

            hr = pStream->Write( &CLSID_Compdata , sizeof( CLSID ) , NULL );
        }

        pStream->Release( );

    } // CreateStreamOnHGlobal
     
    return hr;
   
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::GetData( LPFORMATETC , LPSTGMEDIUM )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::QueryGetData( LPFORMATETC )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::GetCanonicalFormatEtc( LPFORMATETC , LPFORMATETC )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::SetData( LPFORMATETC , LPSTGMEDIUM , BOOL )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::EnumFormatEtc( DWORD , LPENUMFORMATETC * )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::DAdvise( LPFORMATETC , ULONG , LPADVISESINK , PULONG )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::DUnadvise( DWORD )
{
    return E_NOTIMPL;
}

//--------------------------------------------------------------------------
STDMETHODIMP CBaseNode::EnumDAdvise( LPENUMSTATDATA * )
{
    return E_NOTIMPL;
}