70 lines
1.8 KiB
C++
70 lines
1.8 KiB
C++
|
|
||
|
|
|
||
|
|
#include "pch.cxx"
|
||
|
|
#pragma hdrstop
|
||
|
|
#include "teststub.cxx"
|
||
|
|
|
||
|
|
#define TRKDATA_ALLOCATE
|
||
|
|
#include "trksvr.hxx"
|
||
|
|
|
||
|
|
#include "itrkadmn.h"
|
||
|
|
|
||
|
|
DWORD g_Debug = 0xffffffff & ~TRKDBG_WORKMAN;
|
||
|
|
|
||
|
|
const TCHAR tszKeyNameLinkTrack[] = TEXT("System\\CurrentControlSet\\Services\\TrkWks\\Parameters");
|
||
|
|
|
||
|
|
EXTERN_C void __cdecl _tmain( ULONG cArgs, TCHAR *rgtszArgs[] )
|
||
|
|
{
|
||
|
|
TCHAR tsz[ MAX_PATH + 1 ];
|
||
|
|
|
||
|
|
IBindCtx *pBC = NULL;
|
||
|
|
IMoniker* pMnk = NULL;
|
||
|
|
ITrkRestoreNotify *pTrkRestore = NULL;
|
||
|
|
ULONG cchEaten = 0;
|
||
|
|
HRESULT hr = E_FAIL;
|
||
|
|
|
||
|
|
|
||
|
|
TrkDebugCreate( TRK_DBG_FLAGS_WRITE_TO_DBG | TRK_DBG_FLAGS_WRITE_TO_STDOUT, "TRestore" );
|
||
|
|
CoInitialize(NULL);
|
||
|
|
|
||
|
|
if( 1 >= cArgs )
|
||
|
|
{
|
||
|
|
printf( "\n"
|
||
|
|
"This test takes a path and calls ITrkRestoreNotify on that machine\n" );
|
||
|
|
printf( "Usage: TRestore <path>\n" );
|
||
|
|
printf( "E.g.: TRestore \\\\machine\\share\\dir\\path.ext\n" );
|
||
|
|
goto Exit;
|
||
|
|
}
|
||
|
|
|
||
|
|
__try
|
||
|
|
{
|
||
|
|
_tcscpy( tsz, TEXT("@LinkTrack@") );
|
||
|
|
_tcscat( tsz, rgtszArgs[1] );
|
||
|
|
|
||
|
|
// Get the moniker
|
||
|
|
CreateBindCtx( 0, &pBC );
|
||
|
|
hr = MkParseDisplayName( pBC, tsz, &cchEaten, &pMnk );
|
||
|
|
if( FAILED(hr) ) TrkRaiseException( hr );
|
||
|
|
|
||
|
|
// Get the ITrkRestoreNotify object
|
||
|
|
hr = pMnk->BindToObject( pBC, NULL, IID_ITrkRestoreNotify, reinterpret_cast<void**>(&pTrkRestore) );
|
||
|
|
if( FAILED(hr) ) TrkRaiseException( hr );
|
||
|
|
|
||
|
|
// Do the notification
|
||
|
|
hr = pTrkRestore->OnRestore();
|
||
|
|
if( FAILED(hr) ) TrkRaiseException( hr );
|
||
|
|
}
|
||
|
|
__except( BreakOnDebuggableException() )
|
||
|
|
{
|
||
|
|
TrkAssert( STATUS_ACCESS_VIOLATION != GetExceptionCode() );
|
||
|
|
TrkLog(( TRKDBG_ERROR, TEXT("Exception %d caught in TRestore\n"), GetExceptionCode() ));
|
||
|
|
}
|
||
|
|
|
||
|
|
Exit:
|
||
|
|
|
||
|
|
if( SUCCEEDED(hr) ) printf("Passed\n");
|
||
|
|
|
||
|
|
return;
|
||
|
|
|
||
|
|
}
|