windows-nt/Source/XPSP1/NT/com/svcdlls/trksvcs/trkwks/stubs.cxx

540 lines
15 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
// Copyright (c) 1996-1999 Microsoft Corporation
//+-------------------------------------------------------------------------
//
// Microsoft Windows
//
// File: stubs.cxx
//
// Contents: RPC stub routines that call CTrkWksSvc
//
// Classes:
//
// Functions:
//
//
//
// History: 18-Nov-96 BillMo Created.
//
// Notes:
//
//--------------------------------------------------------------------------
#include "pch.cxx"
#pragma hdrstop
#include "trkwks.hxx"
#define THIS_FILE_NUMBER STUBS_CXX_FILE_NO
//+----------------------------------------------------------------------------
//
// StubLnkCallSvrMessage
//
// Calls CTrkWksSvc::CallSvrMessage.
//
//+----------------------------------------------------------------------------
HRESULT StubLnkCallSvrMessage(
/* [in] */ handle_t IDL_handle,
/* [switch_is][out][in] */ TRKSVR_MESSAGE_UNION __RPC_FAR *pMsg)
{
HRESULT hr;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
state = InitializeThreadFromPool();
hr = g_ptrkwks->CallSvrMessage( IDL_handle, pMsg );
}
__except (BreakOnDebuggableException())
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return(hr);
}
//+----------------------------------------------------------------------------
//
// Stubold_LnkCallSvrMessage
//
// Backward compatibility, calls StubLnkCallSvrMessage with new msg
// structure.
//
//+----------------------------------------------------------------------------
HRESULT Stubold_LnkCallSvrMessage(
/* [in] */ handle_t IDL_handle,
/* [out][in] */ TRKSVR_MESSAGE_UNION_OLD __RPC_FAR *pMsg)
{
TRKSVR_MESSAGE_UNION Msg2;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
Msg2.MessageType = pMsg->MessageType;
Msg2.Priority = PRI_5;
switch (Msg2.MessageType)
{
case (SEARCH):
Msg2.Search = pMsg->Search;
break;
case (MOVE_NOTIFICATION):
Msg2.MoveNotification = pMsg->MoveNotification;
break;
case (REFRESH):
Msg2.Refresh = pMsg->Refresh;
break;
case (SYNC_VOLUMES):
Msg2.SyncVolumes = pMsg->SyncVolumes;
break;
case (DELETE_NOTIFY):
Msg2.Delete = pMsg->Delete;
break;
}
Msg2.ptszMachineID = pMsg->ptszMachineID;
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return StubLnkCallSvrMessage( IDL_handle, &Msg2 );
}
//+----------------------------------------------------------------------------
//
// StubLnkMendLink
//
// Calls CTrkWksSvc::MendLink. This stub is caled from within the local machine.
//
//+----------------------------------------------------------------------------
/*
// Version 1.2 (added pdroidBirthCurrent)
HRESULT
StubLnkMendLink(RPC_BINDING_HANDLE IDL_handle,
FILETIME ftLimit,
DWORD RestrictionsIn,
const CDomainRelativeObjId *pdroidBirthLast,
const CDomainRelativeObjId *pdroidLast,
const CMachineId *pmcidLast,
CDomainRelativeObjId *pdroidBirthCurrent,
CDomainRelativeObjId *pdroidCurrent,
CMachineId *pmcidCurrent,
ULONG *pcbPath,
WCHAR *pwsz )
{
HRESULT hr = g_ptrkwks->MendLink( IDL_handle, static_cast<CFILETIME>(ftLimit), RestrictionsIn,
*pdroidBirthLast, *pdroidLast, *pmcidLast,
pdroidBirthCurrent, pdroidCurrent, pmcidCurrent,
pcbPath, pwsz );
TrkAssert( TRK_E_POTENTIAL_FILE_FOUND != hr
||
*pdroidBirthLast != *pdroidBirthCurrent );
TrkAssert( FAILED(hr) || *pdroidBirthLast == *pdroidBirthCurrent
|| *pdroidBirthLast == CDomainRelativeObjId() );
return( MapTR2HR(hr) );
}
*/
// Version 1.1 (added pmcidLast and pmcidCurrent)
void
StubLnkMendLink(PRPC_ASYNC_STATE pAsync_handle,
RPC_BINDING_HANDLE IDL_handle,
FILETIME ftLimit,
DWORD RestrictionsIn,
const CDomainRelativeObjId *pdroidBirth,
const CDomainRelativeObjId *pdroidLast,
const CMachineId * pmcidLast,
CDomainRelativeObjId * pdroidCurrent,
CMachineId * pmcidCurrent,
ULONG * pcbPath,
WCHAR * wsz)
{
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
CDomainRelativeObjId droidBirthCurrent;
HRESULT hr = S_OK;
SThreadFromPoolState state;
__try
{
state = InitializeThreadFromPool();
// Convert the time limit into a tick-count limit, so that we're reslient
// to clock updates. Perf: Since this is always a intra-machine call,
// the interface really ought to be changed so that it just passes
// in a tick count, but it's not worth changing the interface just
// for that.
CFILETIME cftNow, cftLimit(ftLimit);
DWORD dwTickCountDeadline = GetTickCount();
if( cftLimit > cftNow )
dwTickCountDeadline += (DWORD) ( (cftLimit - cftNow)/10000 );
hr = g_ptrkwks->MendLink( IDL_handle, dwTickCountDeadline, RestrictionsIn,
*pdroidBirth, *pdroidLast, *pmcidLast,
&droidBirthCurrent, pdroidCurrent, pmcidCurrent,
pcbPath, wsz );
TrkAssert( FAILED(hr)
||
*pdroidBirth == droidBirthCurrent );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
}
__except( EXCEPTION_EXECUTE_HANDLER )
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
hr = MapTR2HR(hr);
// If this request came in on Async RPC, complete the call and
// pass back the return code.
if( NULL != pAsync_handle )
{
HRESULT hrT = RpcAsyncCompleteCall( pAsync_handle, &hr );
#if DBG
if( ERROR_SUCCESS != hrT )
TrkLog(( TRKDBG_ERROR, TEXT("Failed RpcAsyncCompleteCall (%lu)"), hrT ));
#endif
}
}
// Version 1.0
HRESULT Stubold_LnkMendLink(
/* [in] */ handle_t IDL_handle,
/* [in] */ FILETIME ftLimit,
/* [in] */ ULONG Restrictions,
/* [in] */ const CDomainRelativeObjId __RPC_FAR *pdroidBirth,
/* [in] */ const CDomainRelativeObjId __RPC_FAR *pdroidLast,
/* [out] */ CDomainRelativeObjId __RPC_FAR *pdroidCurrent,
/* [string][out] */ WCHAR __RPC_FAR wsz[ MAX_PATH + 1 ] )
{
TrkLog(( TRKDBG_ERROR, TEXT("Stubold_LnkMendLink was called") ));
return( E_FAIL );
}
//+----------------------------------------------------------------------------
//
// StubLnkSearchMachine
//
// Calls CTrkWksSvc::SearchMachine. This is called from the trkwks service
// on another machine, or directly (i.e. not by RPC) from within this
// service.
//
//+----------------------------------------------------------------------------
// Version 1.2 (added pdroidBirthLast, pdroidBirthNext)
// S_OK || TRK_E_REFERRAL || TRK_E_NOT_FOUND || TRK_E_POTENTIAL_FILE_FOUND
HRESULT StubLnkSearchMachine(RPC_BINDING_HANDLE IDL_handle,
ULONG RestrictionsIn,
const CDomainRelativeObjId *pdroidBirthLast,
const CDomainRelativeObjId *pdroidLast,
CDomainRelativeObjId *pdroidBirthNext,
CDomainRelativeObjId *pdroidNext,
CMachineId *pmcidNext,
TCHAR *ptsz )
{
HRESULT hr;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
state = InitializeThreadFromPool();
hr = g_ptrkwks->SearchMachine(
IDL_handle,
RestrictionsIn,
*pdroidBirthLast, *pdroidLast,
pdroidBirthNext, pdroidNext, pmcidNext, ptsz
);
}
__except( BreakOnDebuggableException() )
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return( hr );
}
// Version 1.1 (added pmcidNext)
HRESULT Stubold2_LnkSearchMachine( RPC_BINDING_HANDLE IDL_handle,
ULONG RestrictionsIn,
const CDomainRelativeObjId *pdroidLast,
CDomainRelativeObjId *pdroidNext,
CMachineId *pmcidNext,
TCHAR *tsz )
{
CDomainRelativeObjId droidBirthLast, droidBirthNext;
return( StubLnkSearchMachine( IDL_handle, RestrictionsIn,
&droidBirthLast, pdroidLast,
&droidBirthNext, pdroidNext, pmcidNext,
tsz ));
}
// Version 1.0
HRESULT Stubold_LnkSearchMachine(
/* [in] */ handle_t IDL_handle,
/* [in] */ ULONG Restrictions,
/* [in] */ const CDomainRelativeObjId __RPC_FAR *pdroidLast,
/* [out] */ CDomainRelativeObjId __RPC_FAR *pdroidReferral,
/* [string][out] */ TCHAR __RPC_FAR tsz[ MAX_PATH + 1 ])
{
CMachineId mcidNext;
return Stubold2_LnkSearchMachine( IDL_handle, Restrictions, pdroidLast, pdroidReferral, &mcidNext, tsz );
}
HRESULT
StubLnkGetBackup(
/* [in] */ handle_t IDL_handle,
/* [out][in] */ DWORD __RPC_FAR *pcVolumes,
/* [size_is][size_is][out] */ VolumeMapEntry __RPC_FAR *__RPC_FAR *ppVolumeChanges,
/* [out] */ FILETIME __RPC_FAR *pft)
{
return(E_NOTIMPL);
}
HRESULT
StubGetFileTrackingInformation( RPC_BINDING_HANDLE IDL_handle,
/*[in]*/ CDomainRelativeObjId droidCurrent,
/*[in]*/ TrkInfoScope scope,
/*[out]*/ TRK_FILE_TRACKING_INFORMATION_PIPE pipeFileInfo )
{
HRESULT hr = E_FAIL;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
TCHAR tszUncPath[ MAX_PATH + 1 ];
ULONG cbPath = sizeof(tszUncPath);
state = InitializeThreadFromPool();
hr = g_ptrkwks->GetFileTrackingInformation( droidCurrent, scope, pipeFileInfo );
}
__except( BreakOnDebuggableException() )
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return( hr );
} // StubGetFileTrackingInformation()
HRESULT
StubGetVolumeTrackingInformation( RPC_BINDING_HANDLE IDL_handle,
/*[in]*/ CVolumeId volid,
/*[in]*/ TrkInfoScope scope,
/*[out]*/ TRK_VOLUME_TRACKING_INFORMATION_PIPE pipeVolInfo )
{
HRESULT hr = E_FAIL;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
state = InitializeThreadFromPool();
hr = g_ptrkwks->GetVolumeTrackingInformation( volid, scope, pipeVolInfo );
}
__except( BreakOnDebuggableException() )
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return( hr );
} // StubGetVolumes()
HRESULT StubLnkOnRestore(/*[in]*/ RPC_BINDING_HANDLE IDL_handle)
{
HRESULT hr;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
state = InitializeThreadFromPool();
hr = g_ptrkwks->OnRestore();
}
__except( BreakOnDebuggableException() )
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return hr;
}
HRESULT StubLnkRestartDcSynchronization(
RPC_BINDING_HANDLE IDL_handle
)
{
return(E_NOTIMPL);
}
HRESULT StubLnkSetVolumeId(
handle_t IDL_handle,
ULONG iVolume,
const CVolumeId VolId)
{
HRESULT hr;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
state = InitializeThreadFromPool();
hr = g_ptrkwks->SetVolumeId( iVolume, VolId );
}
__except (BreakOnDebuggableException())
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return(hr);
}
HRESULT
StubTriggerVolumeClaims( RPC_BINDING_HANDLE IDL_handle,
/*[in]*/ ULONG cVolumes,
/*[in]*/ const CVolumeId *rgvolid )
{
HRESULT hr = E_FAIL;
SThreadFromPoolState state;
#if DBG
InterlockedIncrement( &g_cTrkWksRpcThreads );
TrkAssert( NULL != g_ptrkwks && CTRKWKSSVC_SIG == g_ptrkwks->GetSignature() );
#endif
__try
{
state = InitializeThreadFromPool();
hr = g_ptrkwks->TriggerVolumeClaims( cVolumes, rgvolid );
}
__except( BreakOnDebuggableException() )
{
hr = GetExceptionCode();
}
UnInitializeThreadFromPool( state );
#if DBG
InterlockedDecrement( &g_cTrkWksRpcThreads );
TrkAssert( 0 <= g_cTrkWksRpcThreads );
#endif
return( hr );
} // StubTriggerVolumeClaims