/* * client.cxx */ #include "client.hxx" #define MIN_TEST_NUMBER 1 #define MAX_TEST_NUMBER 27 BOOL UnimplementedTest(); BOOL UnsupportedTest(); BOOL GenericCITest(REFCLSID clsid, REFIID iid, WCHAR * szServerName, int n, DWORD ctxt); BOOL CGCOLocalEXE(); BOOL CGCOExplicitActivator(); BOOL CILocalDLL(); BOOL CILocalEXE(); BOOL CIExplicitActivator(); BOOL CI3LocalEXE(); BOOL CI3ExplicitActivator(); BOOL CGIFFLocalEXE(); BOOL CGIFFExplicitActivator(); BOOL CGIFFRegistryActivator(); BOOL CGIFFAtStorageActivator(); BOOL IMBLocalEXE(); BOOL IMBLocalService(); BOOL IMBAtStorageActivator(); BOOL IMBAtStorageService(); BOOL IMBAtStoragePreCon(); BOOL IMBAtStorageUser(); BOOL CIFromStgLocalEXE(); BOOL CIFromStgActivator(); BOOL CICustomLocalDLL(); BOOL CICustomLocalEXE(); BOOL CILocalPreCon(); BOOL CIExplicitPreCon(); BOOL CILocalUser(); BOOL CIExplicitUser(); BOOL CILocalPreConACL(); BOOL CIExplicitPreConACL(); BOOL CILocalService(); BOOL CIExplicitService(); BOOL MyStartService(WCHAR * wszServiceName, WCHAR *pwszRegServiceArgs); BOOL MyStopService(WCHAR * wszServiceName); #ifdef NO_DCOM LPTESTFUNC rgTest[] = { NULL, CGCOLocalEXE, UnsupportedTest, CILocalDLL, CILocalEXE, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest /* New tests UnsupportedTest, UnsupportedTest, end new tests */ CI3LocalEXE, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, IMBLocalEXE, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, UnsupportedTest, NULL }; #else LPTESTFUNC rgTest[] = { NULL, CGCOLocalEXE, CGCOExplicitActivator, CILocalDLL, CILocalEXE, CIExplicitActivator, CILocalPreCon, CIExplicitPreCon, CILocalUser, CIExplicitUser, CILocalService, CIExplicitService, /* new tests CILocalPreConACL, CIExplicitPreConACL, end new tests */ CI3LocalEXE, CI3ExplicitActivator, CICustomLocalDLL, CICustomLocalEXE, CGIFFLocalEXE, CGIFFExplicitActivator, CGIFFRegistryActivator, CGIFFAtStorageActivator, IMBLocalEXE, IMBLocalService, IMBAtStorageActivator, IMBAtStorageService, IMBAtStoragePreCon, IMBAtStorageUser, CIFromStgLocalEXE, CIFromStgActivator, NULL }; #endif // NO_DCOM char * TestName[] = { NULL, "CoGetClassObject local EXE", "CoGetClassObject explicit activator", "CoCreateInstance local DLL", "CoCreateInstance local EXE", "CoCreateInstance explicit activator", "CoCreateInstance local pre-configured", "CoCreateInstance explicit pre-configured", "CoCreateInstance local user", "CoCreateInstance explicit user", "CoCreateInstance local service", "CoCreateInstance explicit service", /* new tests "CoCreateInstance local pre-configured ACL", "CoCreateInstance explicit pre-configured ACL", end new tests */ "CoCreateInstance (3 IIDs) local EXE", "CoCreateInstance (3 IIDs) explicit activator", "CoCreateInstance from custom itf. local DLL", "CoCreateInstance from custom itf. local EXE", "CoGetInstanceFromFile local EXE", "CoGetInstanceFromFile explicit activator", "CoGetInstanceFromFile registry activator", "CoGetInstanceFromFile AtStorage activator", "IMoniker::BindToObject local EXE", "IMoniker::BindToObject local service", "IMoniker::BindToObject AtStorage activator", "IMoniker::BindToObject AtStorage service", "IMoniker::BindToObject AtStorage pre-configured", "IMoniker::BindToObject AtStorage user", "CoGetInstanceFromIStorage local EXE", "CoGetInstanceFromIStorage explicit activator", NULL }; char RunTest[] = { -1, // CoGetClassObject YES, YES, // CoCreateInstance YES, YES, YES, YES, YES, YES, YES, YES, YES, /* new tests YES, YES, end new tests */ // CoCreateInstance (3 IIDs) YES, YES, // CoGetInstanceFromFile YES, YES, YES, YES, // IMoniker:Bind YES, YES, YES, YES, YES, YES, YES, YES, YES, YES, -1 }; char RunLocalTest[] = { -1, // CoGetClassObject YES, NO, // CreateInstance YES, YES, NO, YES, NO, YES, NO, YES, NO, /* new tests YES, NO, end new tests */ // CreateInstance (3 IIDs) YES, NO, // CreateInstance from custom Itf. YES, YES, // CoGetInstanceFromFile YES, NO, NO, NO, // IMoniker:Bind YES, YES, NO, NO, NO, NO, // CoGetInstanceFromIStorage YES, NO, -1 }; WCHAR ServerName[32]; WCHAR RemoteFileName[256]; WCHAR * LocalFileName = L"c:\\acttest.dat"; WCHAR * StorageName = L"c:\\acttest.stg"; LARGE_INTEGER liPerfFreq; LARGE_INTEGER liStart; LARGE_INTEGER liStop; LARGE_INTEGER liElapsedTime; #define RESET_CLOCK liElapsedTime.LowPart = liElapsedTime.HighPart = 0 #define START_CLOCK QueryPerformanceCounter(&liStart) #define STOP_CLOCK QueryPerformanceCounter(&liStop); \ liElapsedTime.QuadPart += liStop.QuadPart - liStart.QuadPart #define DUMP_CLOCK DisplayElapsedTime() #define START_LOOP for (unsigned sl_n = uIterations+1; sl_n--;){ #define STOP_LOOP if (uIterations == sl_n ) RESET_CLOCK;} #define SLEEP_IF_LOOPING if (sl_n && !gfHoldServer) Sleep(1000) unsigned uIterations = 0; BOOL gfRegister = TRUE; BOOL gfHoldServer = FALSE; BOOL gfLocal = FALSE; BOOL gfNolocal = FALSE; BOOL gfSpecificTest = FALSE; DWORD dwWaitHint = 0; void DisplayElapsedTime(void) { LONGLONG MicroPerIter; liElapsedTime.QuadPart /= uIterations; MicroPerIter = liElapsedTime.QuadPart * 1000000; MicroPerIter /= liPerfFreq.QuadPart; printf( "Time: %d microseconds per iteration", (DWORD) MicroPerIter ); } BOOL AllLocal() { for (int x = MIN_TEST_NUMBER; x<= MAX_TEST_NUMBER; x++) { if (RunTest[x] && !RunLocalTest[x]) return(FALSE); } return(TRUE); } extern WCHAR ServiceName[]; void _cdecl main( int argc, char ** argv ) { HRESULT HResult; HANDLE hFile; int n; BOOL f; if (argc > 1) { if (argc == 3 && (!strcmp(argv[1], "-t"))) { dwWaitHint = atoi(argv[2]); } else { printf("usage: %s [-t #]\n # number of milliseconds between QueryServiceStatus calls\n", argv[0]); return; } } if ((!QueryPerformanceFrequency(&liPerfFreq)) && uIterations > 0) { printf("No high performance counter.\nTests cannot be timed.\nAborting.\n"); return; } uIterations = 1; printf("Starting service...\n"); RESET_CLOCK; START_CLOCK; f = MyStartService(L"ActTestService", L""); STOP_CLOCK; if (f) printf("succeeded - "); else printf("failed - "); DUMP_CLOCK; printf("\n"); printf("Stopping service...\n"); RESET_CLOCK; START_CLOCK; f = MyStopService(L"ActTestService"); STOP_CLOCK; if (f) printf("succeeded - "); else printf("failed - "); DUMP_CLOCK; printf("\n"); return; #ifndef NO_DCOM if ( argc == 1 ) PrintUsageAndExit( FALSE ); #endif // NO_DCOM if ( argc > 1 && strcmp(argv[1],"-?") == 0 ) PrintUsageAndExit( TRUE ); n = 1; while ( (n < argc) && (*argv[n] == '-') ) { if ( (n < argc) && strcmp(argv[n],"-local") == 0 ) { if (gfLocal | gfNolocal) { PrintUsageAndExit( FALSE ); } gfLocal = TRUE; memcpy(RunTest, RunLocalTest, MAX_TEST_NUMBER + MIN_TEST_NUMBER); n++; } if ( (n < argc) && strcmp(argv[n],"-nolocal") == 0 ) { if (gfLocal | gfNolocal) { PrintUsageAndExit( FALSE ); } gfNolocal = TRUE; for (int x = MIN_TEST_NUMBER; x<= MAX_TEST_NUMBER; x++) { RunTest[x] = !RunLocalTest[x]; } n++; } if ( (n < argc) && strcmp(argv[n],"-noreg") == 0 ) { gfRegister = FALSE; n++; continue; } if ( (n < argc) && strcmp(argv[n],"-hold") == 0 ) { gfHoldServer = TRUE; n++; continue; } if ( (n < argc) && strcmp(argv[n],"-n") == 0 ) { if ( ++n >= argc ) PrintUsageAndExit(FALSE); uIterations = atoi(argv[n++]); } if ( (n < argc) && strcmp(argv[n],"-t") == 0 ) { long TestNum1, TestNum2; if ( ++n >= argc ) PrintUsageAndExit(FALSE); TestNum1 = atoi(argv[n++]); if ( (n < argc) && ((TestNum2 = atoi(argv[n])) != 0) ) n++; else TestNum2 = TestNum1; if ( (TestNum1 < MIN_TEST_NUMBER) || (TestNum2 > MAX_TEST_NUMBER) ) { printf( "Test number(s) must be between %d and %d.\n", MIN_TEST_NUMBER, MAX_TEST_NUMBER ); return; } if ( TestNum1 > TestNum2 ) { printf( "Second test number must be greater than the first.\n" ); return; } if (!gfSpecificTest) { gfSpecificTest = TRUE; // only do this the first time -t is found on the command line memset(RunTest,NO,sizeof(RunTest)); } memset(&RunTest[TestNum1],YES,sizeof(char)*(TestNum2-TestNum1+1)); } } #ifndef NO_DCOM if ( n != argc - 1 && !AllLocal()) { printf("ERROR - Selected tests require a server name.\n"); ExitThread(0); } if ( n < argc ) { MultiByteToWideChar( CP_ACP, 0, argv[n], -1, ServerName, sizeof(ServerName) / sizeof(WCHAR) ); } else ServerName[0] = 0; #endif // NO_DCOM hFile = CreateFile( LocalFileName, GENERIC_WRITE, FILE_SHARE_READ, 0, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0 ); if ( hFile == INVALID_HANDLE_VALUE ) { printf("CreateFile failed while creating local file: %d\n", GetLastError()); return; } CloseHandle( hFile ); #ifndef NO_DCOM if ( ServerName[0] != 0 ) { RemoteFileName[0] = 0; if ( ServerName[0] != L'\\' ) wcscat( RemoteFileName, L"\\\\" ); wcscat( RemoteFileName, ServerName ); wcscat( RemoteFileName, L"\\c$\\acttest.dat" ); hFile = CreateFile( RemoteFileName, GENERIC_WRITE, FILE_SHARE_READ, 0, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0 ); if ( hFile == INVALID_HANDLE_VALUE ) { printf("CreateFile failed while creating remote file: %d\n", GetLastError()); return; } CloseHandle( hFile ); } #endif // NO_DCOM HResult = CoInitialize(NULL); if( FAILED(HResult) ) { printf( "Client CoInitialize failed Ox%x!\n", HResult ); return; } if ((!QueryPerformanceFrequency(&liPerfFreq)) && uIterations > 0) { printf("No high performance counter.\nTests cannot be timed.\nAborting.\n"); } else { if ( ! Tests() ) printf("\nTests FAILED\n"); else printf("\nTests SUCCEEDED\n"); } CoUninitialize(); DeleteFile( LocalFileName ); #ifndef NO_DCOM DeleteFile( RemoteFileName ); #endif // NO_DCOM } BOOL Tests() { HRESULT HResult; long RegStatus; if (gfRegister) { DeleteClsidKey( ClsidGoober32String ); DeleteClsidKey( ClsidActLocalString ); DeleteClsidKey( ClsidActRemoteString ); DeleteClsidKey( ClsidActAtStorageString ); DeleteClsidKey( ClsidActInprocString ); DeleteClsidKey( ClsidActPreConfigString ); DeleteClsidKey( ClsidActRunAsLoggedOnString ); DeleteClsidKey( ClsidActServiceString ); DeleteClsidKey( ClsidActServerOnlyString ); if ( (RegStatus = InitializeRegistryForInproc()) != ERROR_SUCCESS ) { printf("InitializeRegistryForInproc failed %d.\n", RegStatus); return FALSE; } if ( (RegStatus = InitializeRegistryForLocal()) != ERROR_SUCCESS ) { printf("InitializeRegistryForLocal failed %d.\n", RegStatus); return FALSE; } if ( (RegStatus = InitializeRegistryForCustom()) != ERROR_SUCCESS ) { printf("InitializeRegistryForCustom failed %d.\n", RegStatus); return FALSE; } #ifndef NO_DCOM if ( (RegStatus = InitializeRegistryForRemote()) != ERROR_SUCCESS ) { printf("InitializeRegistryForRemote failed %d.\n", RegStatus); return FALSE; } if ( (RegStatus = InitializeRegistryForService()) != ERROR_SUCCESS ) { printf("InitializeRegistryForService failed %d.\n", RegStatus); return FALSE; } #endif // NO_DCOM } BOOL fAllTests = TRUE; for (int x = MIN_TEST_NUMBER; x <= MAX_TEST_NUMBER; x++) { if (RunTest[x]) { printf("\nTest %2d: %s\n",x, TestName[x]); // prime the system once to ensure the test is fully cached if (rgTest[x]()) { printf("PASSED"); if (uIterations) { printf(" - "); DUMP_CLOCK; } printf("\n"); } else { printf("FAILED\n"); fAllTests = FALSE; } } } return(fAllTests); } void PrintUsageAndExit( BOOL bListTests ) { #ifdef NO_DCOM printf("Usage : %s [-hold] [-noreg] [-n #] [-t # [#]]\n", "actclt"); printf("\t-n # : Run specific number of timed iterations (default is 0).\n"); printf("\t-noreg : Don't update registry\n"); printf("\t-t # : Run specific test number or a range of tests.\n"); printf("\t-? : Print usage plus test descriptions.\n"); #else printf("Usage : %s [-hold] [-noreg] [[-local] | [-nolocal]] [-n #] [-t # [#]] [server_name]\n", "actclt"); printf("\t-hold : Hold the server up during all iterations.\n"); printf("\t-local : Run only local activation tests.\n"); printf("\t-n # : Run specific number of timed iterations (default is 0).\n"); printf("\t-nolocal : Run only remote activation tests.\n"); printf("\t-noreg : Don't update registry\n"); printf("\t-t # : Run specific test number or a range of tests.\n"); printf("\t-? : Print usage plus test descriptions.\n"); #endif // NO_DCOM if ( bListTests ) { long n; printf("\nTests :\n"); printf("\t # Method Location Security\n"); printf("\t-- ----------------------------------- ----------- --------------\n"); for ( n = MIN_TEST_NUMBER; n <= MAX_TEST_NUMBER; n++ ) { printf("\t%2d %s\n", n, TestName[n]); } } ExitThread(0); } void * __RPC_API MIDL_user_allocate(size_t len) { return malloc(len); } void __RPC_API MIDL_user_free(void * vp) { free(vp); } BOOL StampFileWithCLSID(WCHAR * szFilename, CLSID & clsid) { HRESULT hr; IStorage * pStg; hr = StgCreateDocfile( szFilename, STGM_READWRITE | STGM_DIRECT | STGM_SHARE_EXCLUSIVE | STGM_CREATE, 0, &pStg); if (FAILED (hr)) { printf("StgCreateDocfile returned 0x%x\n", hr); return(FALSE); } hr = WriteClassStg(pStg, clsid); pStg->Release(); if (FAILED (hr)) { printf("WriteClassStg returned 0x%x\n", hr); return(FALSE); } return(TRUE); } // // Test Procedures: // BOOL UnimplementedTest() { printf("Not implemented at this time.\n"); return(TRUE); } BOOL UnsupportedTest() { printf("Not supported by this version.\n"); return(TRUE); } BOOL GenericCITest(REFCLSID clsid, REFIID iid, WCHAR * szServerName, int n, DWORD ctxt) { COSERVERINFO ServerInfo; COSERVERINFO *pServerInfo; IUnknown * punkHeld = NULL; if (szServerName) { memset( &ServerInfo, 0, sizeof(COSERVERINFO) ); ServerInfo.pwszName = szServerName; pServerInfo = &ServerInfo; } else { pServerInfo = NULL; } MULTI_QI QIStruct[10]; int x; HRESULT hr; BOOL fReturn = TRUE; START_LOOP; for (x = n; x--;) { QIStruct[x].pItf = NULL; QIStruct[x].pIID = (IID *) &iid; } START_CLOCK; hr = CoCreateInstanceEx( clsid, NULL, ctxt, pServerInfo, n, QIStruct); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoCreateInstanceEx returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld && SUCCEEDED(QIStruct[0].hr)) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } for (x = 0; x < n; x++) { if (FAILED(QIStruct[x].hr)) { printf("CoCreateInstanceEx returned 0x%x for interface %d\n", QIStruct[x].hr, x); fReturn = FALSE; } else QIStruct[x].pItf->Release(); } if (!fReturn) { if (punkHeld) punkHeld->Release(); return(fReturn); } STOP_LOOP; if (punkHeld) punkHeld->Release(); return fReturn; } BOOL CGCOLocalEXE() { IClassFactory * pClassFactory; IUnknown * pUnknown; IUnknown * punkHeld = NULL; HRESULT hr; START_LOOP; START_CLOCK; hr = CoGetClassObject( CLSID_ActLocal, CLSCTX_LOCAL_SERVER, NULL, IID_IClassFactory, (void **) &pClassFactory ); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoGetClassObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } hr = pClassFactory->CreateInstance( NULL, IID_IUnknown, (void **) &pUnknown ); pClassFactory->Release(); if ( FAILED(hr) ) { printf("CreateInstance returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnknown; punkHeld->AddRef(); } pUnknown->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return TRUE; } #ifndef NO_DCOM BOOL CGCOExplicitActivator() { COSERVERINFO ServerInfo; memset( &ServerInfo, 0, sizeof(COSERVERINFO) ); ServerInfo.pwszName = ServerName; IClassFactory * pClassFactory; IUnknown * pUnknown; IUnknown * punkHeld = NULL; HRESULT hr; START_LOOP; START_CLOCK; hr = CoGetClassObject( CLSID_ActLocal, CLSCTX_REMOTE_SERVER, &ServerInfo, IID_IClassFactory, (void **) &pClassFactory ); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoGetClassObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } hr = pClassFactory->CreateInstance( NULL, IID_IUnknown, (void **) &pUnknown ); pClassFactory->Release(); if ( FAILED(hr) ) { printf("CreateInstance returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnknown; punkHeld->AddRef(); } pUnknown->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return TRUE; } #endif // NO_DCOM #ifdef NO_DCOM BOOL CILocalDLL() { IUnknown * pUnknown; IUnknown * punkHeld = NULL; HRESULT hr; START_LOOP; START_CLOCK; hr = CoCreateInstance( CLSID_ActInproc, NULL, CLSCTX_INPROC_SERVER, IID_IUnknown, (void **) &pUnknown ); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoCreateInstance returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnknown; punkHeld->AddRef(); } pUnknown->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return TRUE; } BOOL CILocalEXE() { IUnknown * pUnknown; IUnknown * punkHeld = NULL; HRESULT hr; START_LOOP; START_CLOCK; hr = CoCreateInstance( CLSID_ActLocal, NULL, CLSCTX_LOCAL_SERVER, IID_IUnknown, (void **) &pUnknown ); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoCreateInstance returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnknown; punkHeld->AddRef(); } pUnknown->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return TRUE; } #else BOOL CILocalDLL() { return GenericCITest(CLSID_ActInproc, IID_IUnknown, NULL, 1, CLSCTX_INPROC_SERVER); } BOOL CILocalEXE() { return GenericCITest(CLSID_ActLocal, IID_IUnknown, NULL, 1, CLSCTX_LOCAL_SERVER); } BOOL CIExplicitActivator() { return GenericCITest(CLSID_ActLocal, IID_IUnknown, ServerName, 1, CLSCTX_REMOTE_SERVER); } #endif // NO_DCOM #ifdef NO_DCOM // Pre-DCOM version of CI3LocalEXE which calls CoCreateInstance // once followed by 2 calls to QueryInterface // instead of calling CoCreateInstanceEx. BOOL CI3LocalEXE() { IUnknown * pUnknown; IUnknwon * punkHeld = NULL; IPersist * pPersist; IPersistFile * pPersistFile; HRESULT hr; START_LOOP; START_CLOCK; hr = CoCreateInstance( CLSID_ActLocal, NULL, CLSCTX_LOCAL_SERVER, IID_IUnknown, (void **) &pUnknown ); if ( FAILED(hr) ) { printf("CoCreateInstance returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnknown; punkHeld->AddRef(); } pUnknown->QueryInterface( IID_IPersist, (void **)&pPersist ); pUnknown->QueryInterface( IID_IPersist, (void **)&pPersistFile ); STOP_CLOCK; pUnknown->Release(); pPersist->Release(); pPersistFile->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return TRUE; } #else // NO_DCOM BOOL CI3LocalEXE() { return GenericCITest(CLSID_ActLocal, IID_IPersist, NULL, 3, CLSCTX_LOCAL_SERVER); /* MULTI_QI QIStruct[3]; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[1].pIID = (IID *)&IID_IPersist; QIStruct[2].pIID = (IID *)&IID_IPersistFile; QIStruct[0].pItf = NULL; QIStruct[1].pItf = NULL; QIStruct[2].pItf = NULL; */ } BOOL CI3ExplicitActivator() { return GenericCITest(CLSID_ActLocal, IID_IPersist, ServerName, 3, CLSCTX_REMOTE_SERVER); /* MULTI_QI QIStruct[3]; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[1].pIID = (IID *)&IID_IPersist; QIStruct[2].pIID = (IID *)&IID_IPersistFile; QIStruct[0].pItf = NULL; QIStruct[1].pItf = NULL; QIStruct[2].pItf = NULL; */ } BOOL CGIFFLocalEXE() { MULTI_QI QIStruct[1]; IUnknown * punkHeld = NULL; START_LOOP; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[0].pItf = NULL; START_CLOCK; HRESULT HResult = CoGetInstanceFromFile( NULL, &CLSID_ActLocal, NULL, CLSCTX_LOCAL_SERVER, STGM_READWRITE, LocalFileName, 1, QIStruct ); STOP_CLOCK; if ( FAILED(HResult) ) { printf("CoGetInstanceFromFile returned 0x%x\n", HResult); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } QIStruct[0].pItf->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CGIFFExplicitActivator() { COSERVERINFO ServerInfo; IUnknown * punkHeld = NULL; memset( &ServerInfo, 0, sizeof(COSERVERINFO) ); ServerInfo.pwszName = ServerName; MULTI_QI QIStruct[1]; START_LOOP; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[0].pItf = NULL; START_CLOCK; HRESULT HResult = CoGetInstanceFromFile( &ServerInfo, &CLSID_ActLocal, NULL, CLSCTX_REMOTE_SERVER, STGM_READWRITE, RemoteFileName, 1, QIStruct ); STOP_CLOCK; if ( FAILED(HResult) ) { printf("CoGetInstanceFromFile returned 0x%x\n", HResult); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } QIStruct[0].pItf->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CGIFFRegistryActivator() { MULTI_QI QIStruct[1]; IUnknown * punkHeld = NULL; START_LOOP; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[0].pItf = NULL; START_CLOCK; HRESULT HResult = CoGetInstanceFromFile( NULL, &CLSID_ActRemote, NULL, CLSCTX_REMOTE_SERVER, STGM_READWRITE, RemoteFileName, 1, QIStruct ); STOP_CLOCK; if ( FAILED(HResult) ) { printf("CoGetInstanceFromFile returned 0x%x\n", HResult); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } QIStruct[0].pItf->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CGIFFAtStorageActivator() { MULTI_QI QIStruct[1]; IUnknown * punkHeld = NULL; START_LOOP; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[0].pItf = NULL; START_CLOCK; HRESULT HResult = CoGetInstanceFromFile( NULL, &CLSID_ActAtStorage, NULL, CLSCTX_REMOTE_SERVER, STGM_READWRITE, RemoteFileName, 1, QIStruct ); STOP_CLOCK; if ( FAILED(HResult) ) { printf("CoGetInstanceFromFile returned 0x%x\n", HResult); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } QIStruct[0].pItf->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return(TRUE); } #endif // NO_DCOM BOOL IMBLocalEXE() { IBindCtx * pBc; IUnknown * pUnk; IUnknown * punkHeld = NULL; HRESULT hr; IMoniker *pMon; if (!StampFileWithCLSID(LocalFileName, CLSID_ActLocal)) { return(FALSE); } hr = CreateFileMoniker(LocalFileName, &pMon); if (FAILED(hr)) { printf("CreateFileMoniker returned 0x%x\n", hr); } hr = CreateBindCtx(0, &pBc); if (FAILED(hr)) { printf("CreateBindCtx returned 0x%x\n", hr); pMon->Release(); return(FALSE); } START_LOOP; START_CLOCK; hr = pMon->BindToObject( pBc, NULL, IID_IUnknown, (void **) &pUnk); STOP_CLOCK; if (FAILED(hr)) { printf("IMoniker::BindToObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return(FALSE); } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnk; punkHeld->AddRef(); } pUnk->Release(); STOP_LOOP pMon->Release(); pBc->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } #ifndef NO_DCOM BOOL IMBLocalService() { IBindCtx * pBc; IUnknown * pUnk; IUnknown * punkHeld = NULL; HRESULT hr; IMoniker *pMon; if (!StampFileWithCLSID(LocalFileName, CLSID_ActService)) { return(FALSE); } hr = CreateFileMoniker(LocalFileName, &pMon); if (FAILED(hr)) { printf("CreateFileMoniker returned 0x%x\n", hr); } hr = CreateBindCtx(0, &pBc); if (FAILED(hr)) { printf("CreateBindCtx returned 0x%x\n", hr); pMon->Release(); return(FALSE); } START_LOOP; START_CLOCK; hr = pMon->BindToObject( pBc, NULL, IID_IUnknown, (void **) &pUnk); STOP_CLOCK; if (FAILED(hr)) { printf("IMoniker::BindToObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return(FALSE); } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnk; punkHeld->AddRef(); } pUnk->Release(); SLEEP_IF_LOOPING; STOP_LOOP; pMon->Release(); pBc->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL IMBAtStorageActivator() { IBindCtx * pBc; IUnknown * pUnk; IUnknown * punkHeld = NULL; HRESULT hr; IMoniker *pMon; if (!StampFileWithCLSID(RemoteFileName, CLSID_ActAtStorage)) { return(FALSE); } hr = CreateFileMoniker(RemoteFileName, &pMon); if (FAILED(hr)) { printf("CreateFileMoniker returned 0x%x\n", hr); } hr = CreateBindCtx(0, &pBc); if (FAILED(hr)) { printf("CreateBindCtx returned 0x%x\n", hr); pMon->Release(); return(FALSE); } START_LOOP; START_CLOCK; hr = pMon->BindToObject( pBc, NULL, IID_IUnknown, (void **) &pUnk); STOP_CLOCK; if (FAILED(hr)) { printf("IMoniker::BindToObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return(FALSE); } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnk; punkHeld->AddRef(); } pUnk->Release(); STOP_LOOP; pMon->Release(); pBc->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL IMBAtStorageService() { IBindCtx * pBc; IUnknown * pUnk; IUnknown * punkHeld = NULL; HRESULT hr; IMoniker *pMon; if (!StampFileWithCLSID(RemoteFileName, CLSID_ActService)) { return(FALSE); } hr = CreateFileMoniker(RemoteFileName, &pMon); if (FAILED(hr)) { printf("CreateFileMoniker returned 0x%x\n", hr); } hr = CreateBindCtx(0, &pBc); if (FAILED(hr)) { printf("CreateBindCtx returned 0x%x\n", hr); pMon->Release(); return(FALSE); } START_LOOP; START_CLOCK; hr = pMon->BindToObject( pBc, NULL, IID_IUnknown, (void **) &pUnk); STOP_CLOCK; if (FAILED(hr)) { printf("IMoniker::BindToObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return(FALSE); } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnk; punkHeld->AddRef(); } pUnk->Release(); SLEEP_IF_LOOPING; STOP_LOOP; pMon->Release(); pBc->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL IMBAtStoragePreCon() { IBindCtx * pBc; IUnknown * pUnk; IUnknown * punkHeld = NULL; HRESULT hr; IMoniker *pMon; if (!StampFileWithCLSID(RemoteFileName, CLSID_ActPreConfig)) { return(FALSE); } hr = CreateFileMoniker(RemoteFileName, &pMon); if (FAILED(hr)) { printf("CreateFileMoniker returned 0x%x\n", hr); } hr = CreateBindCtx(0, &pBc); if (FAILED(hr)) { printf("CreateBindCtx returned 0x%x\n", hr); pMon->Release(); return(FALSE); } START_LOOP; START_CLOCK; hr = pMon->BindToObject( pBc, NULL, IID_IUnknown, (void **) &pUnk); STOP_CLOCK; if (FAILED(hr)) { printf("IMoniker::BindToObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return(FALSE); } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnk; punkHeld->AddRef(); } pUnk->Release(); STOP_LOOP; pMon->Release(); pBc->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL IMBAtStorageUser() { IBindCtx * pBc; IUnknown * pUnk; IUnknown * punkHeld = NULL; HRESULT hr; IMoniker *pMon; if (!StampFileWithCLSID(RemoteFileName, CLSID_ActRunAsLoggedOn)) { return(FALSE); } hr = CreateFileMoniker(RemoteFileName, &pMon); if (FAILED(hr)) { printf("CreateFileMoniker returned 0x%x\n", hr); } hr = CreateBindCtx(0, &pBc); if (FAILED(hr)) { printf("CreateBindCtx returned 0x%x\n", hr); pMon->Release(); return(FALSE); } START_LOOP; START_CLOCK; hr = pMon->BindToObject( pBc, NULL, IID_IUnknown, (void **) &pUnk); STOP_CLOCK; if (FAILED(hr)) { printf("IMoniker::BindToObject returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return(FALSE); } if (gfHoldServer && NULL == punkHeld) { punkHeld = pUnk; punkHeld->AddRef(); } pUnk->Release(); STOP_LOOP; pMon->Release(); pBc->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CIFromStgLocalEXE() { IStorage * pStorage; MULTI_QI QIStruct[10]; IUnknown * punkHeld = NULL; pStorage = 0; DeleteFile( StorageName ); HRESULT HResult = StgCreateDocfile( StorageName, STGM_READWRITE | STGM_DIRECT | STGM_SHARE_EXCLUSIVE | STGM_CREATE, 0, &pStorage ); if ( FAILED(HResult) ) { printf("StgCreateDocfile returned 0x%x\n", HResult); return FALSE; } START_LOOP; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[0].pItf = NULL; START_CLOCK; HResult = CoGetInstanceFromIStorage( NULL, &CLSID_ActLocal, NULL, CLSCTX_SERVER, pStorage, 1, QIStruct ); STOP_CLOCK; if ( FAILED(HResult) ) { printf("CoGetInstanceFromIStorage returned 0x%x\n", HResult); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } QIStruct[0].pItf->Release(); QIStruct[0].pItf = 0; STOP_LOOP; pStorage->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CIFromStgActivator() { IStorage * pStorage; MULTI_QI QIStruct[10]; COSERVERINFO ServerInfo; IUnknown * punkHeld = NULL; memset( &ServerInfo, 0, sizeof(COSERVERINFO) ); ServerInfo.pwszName = ServerName; pStorage = 0; DeleteFile( StorageName ); HRESULT HResult = StgCreateDocfile( StorageName, STGM_READWRITE | STGM_DIRECT | STGM_SHARE_EXCLUSIVE | STGM_CREATE, 0, &pStorage ); if ( FAILED(HResult) ) { printf("StgCreateDocfile returned 0x%x\n", HResult); return FALSE; } START_LOOP; QIStruct[0].pIID = (IID *)&IID_IUnknown; QIStruct[0].pItf = NULL; START_CLOCK; HResult = CoGetInstanceFromIStorage( &ServerInfo, &CLSID_ActRemote, NULL, CLSCTX_REMOTE_SERVER, pStorage, 1, QIStruct ); STOP_CLOCK; if ( FAILED(HResult) ) { printf("CoGetInstanceFromIStorage returned 0x%x\n", HResult); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } QIStruct[0].pItf->Release(); QIStruct[0].pItf = 0; STOP_LOOP; pStorage->Release(); if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CICustomLocalDLL() { IGoober * pGoober; MULTI_QI QIStruct[10]; HRESULT hr; IUnknown * punkHeld = NULL; START_LOOP; QIStruct[0].pItf = NULL; QIStruct[0].pIID = (IID *) &IID_IGoober; START_CLOCK; hr = CoCreateInstanceEx( CLSID_ActInproc, NULL, CLSCTX_INPROC_SERVER, NULL, 1, QIStruct ); STOP_CLOCK; pGoober = (IGoober *)QIStruct[0].pItf; if ( FAILED(hr) ) { printf("CoCreateInstanceEx returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } START_CLOCK; hr = pGoober->Ping(); STOP_CLOCK; if ( hr != S_OK ) { printf("IGoober::Ping returned %d\n", hr); pGoober->Release(); if (punkHeld) punkHeld->Release(); return FALSE; } pGoober->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CICustomLocalEXE() { IGoober * pGoober; MULTI_QI QIStruct[10]; HRESULT hr; IUnknown * punkHeld = NULL; START_LOOP; QIStruct[0].pItf = NULL; QIStruct[0].pIID = (IID *) &IID_IGoober; START_CLOCK; hr = CoCreateInstanceEx( CLSID_ActLocal, NULL, CLSCTX_LOCAL_SERVER, NULL, 1, QIStruct ); STOP_CLOCK; pGoober = (IGoober *)QIStruct[0].pItf; if ( FAILED(hr) ) { printf("CoCreateInstanceEx returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } START_CLOCK; hr = pGoober->Ping(); STOP_CLOCK; if ( hr != S_OK ) { printf("IGoober::Ping returned %d\n", hr); pGoober->Release(); if (punkHeld) punkHeld->Release(); return FALSE; } pGoober->Release(); STOP_LOOP; if (punkHeld) punkHeld->Release(); return(TRUE); } BOOL CILocalPreCon() { return GenericCITest(CLSID_ActPreConfig, IID_IUnknown, NULL, 1, CLSCTX_LOCAL_SERVER); } BOOL CIExplicitPreCon() { return GenericCITest(CLSID_ActPreConfig, IID_IUnknown, ServerName, 1, CLSCTX_REMOTE_SERVER); } BOOL CILocalUser() { return GenericCITest(CLSID_ActRunAsLoggedOn, IID_IUnknown, NULL, 1, CLSCTX_LOCAL_SERVER); } BOOL CIExplicitUser() { return GenericCITest(CLSID_ActRunAsLoggedOn, IID_IUnknown, ServerName, 1, CLSCTX_REMOTE_SERVER); } BOOL CILocalService() { COSERVERINFO ServerInfo; COSERVERINFO *pServerInfo; IUnknown * punkHeld = NULL; MULTI_QI QIStruct[1]; HRESULT hr; BOOL fReturn = TRUE; START_LOOP; QIStruct[0].pItf = NULL; QIStruct[0].pIID = (IID *) &IID_IUnknown; START_CLOCK; hr = CoCreateInstanceEx( CLSID_ActService, NULL, CLSCTX_LOCAL_SERVER, NULL, 1, QIStruct); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoCreateInstanceEx returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld && SUCCEEDED(QIStruct[0].hr)) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } if (FAILED(QIStruct[0].hr)) { printf("CoCreateInstanceEx returned 0x%x\n", QIStruct[0].hr); fReturn = FALSE; } else QIStruct[0].pItf->Release(); if (!fReturn) { if (punkHeld) punkHeld->Release(); return(fReturn); } SLEEP_IF_LOOPING; STOP_LOOP; if (punkHeld) punkHeld->Release(); return fReturn; } BOOL CIExplicitService() { COSERVERINFO ServerInfo; COSERVERINFO *pServerInfo; IUnknown * punkHeld = NULL; memset( &ServerInfo, 0, sizeof(COSERVERINFO) ); ServerInfo.pwszName = ServerName; pServerInfo = &ServerInfo; MULTI_QI QIStruct[1]; HRESULT hr; BOOL fReturn = TRUE; START_LOOP; QIStruct[0].pItf = NULL; QIStruct[0].pIID = (IID *) &IID_IUnknown; START_CLOCK; hr = CoCreateInstanceEx( CLSID_ActService, NULL, CLSCTX_REMOTE_SERVER, pServerInfo, 1, QIStruct); STOP_CLOCK; if ( FAILED(hr) ) { printf("CoCreateInstanceEx returned 0x%x\n", hr); if (punkHeld) punkHeld->Release(); return FALSE; } if (gfHoldServer && NULL == punkHeld && SUCCEEDED(QIStruct[0].hr)) { punkHeld = QIStruct[0].pItf; punkHeld->AddRef(); } if (FAILED(QIStruct[0].hr)) { printf("CoCreateInstanceEx returned 0x%x\n", QIStruct[0].hr); fReturn = FALSE; } else QIStruct[0].pItf->Release(); if (!fReturn) { if (punkHeld) punkHeld->Release(); return(fReturn); } SLEEP_IF_LOOPING; STOP_LOOP; if (punkHeld) punkHeld->Release(); return fReturn; } BOOL CILocalPreConACL(); BOOL CIExplicitPreConACL(); #define MAX_SERVICE_ARGS 10 BOOL MyStartService(WCHAR * wszServiceName, WCHAR *pwszRegServiceArgs) { SC_HANDLE hSCManager; SC_HANDLE hService; WCHAR *pwszServiceArgs = NULL; ULONG cArgs = 0; WCHAR *apwszArgs[MAX_SERVICE_ARGS]; // Get a handle to the Service Control Manager if (hSCManager = OpenSCManager(NULL, NULL, GENERIC_EXECUTE | SERVICE_QUERY_STATUS )) { // Open a handle to the requested service if (hService = OpenService(hSCManager, wszServiceName, GENERIC_EXECUTE | SERVICE_QUERY_STATUS )) { // Close the service manager's database CloseServiceHandle(hSCManager); // Formulate the arguments (if any) if (pwszRegServiceArgs) { UINT k = 0; // Make a copy of the service arguments pwszServiceArgs = new WCHAR[(lstrlenW(pwszRegServiceArgs) + 1)]; if (pwszServiceArgs == NULL) { CloseServiceHandle(hService); return FALSE; } lstrcpyW(pwszServiceArgs, pwszRegServiceArgs); // Scan the arguments do { // Scan to the next non-whitespace character while(pwszServiceArgs[k] && (pwszServiceArgs[k] == L' ' || pwszServiceArgs[k] == L'\t')) { k++; } // Store the next argument if (pwszServiceArgs[k]) { apwszArgs[cArgs++] = &pwszServiceArgs[k]; } // Scan to the next whitespace char while(pwszServiceArgs[k] && pwszServiceArgs[k] != L' ' && pwszServiceArgs[k] != L'\t') { k++; } // Null terminate the previous argument if (pwszServiceArgs[k]) { pwszServiceArgs[k++] = L'\0'; } } while(pwszServiceArgs[k]); } // Start the service if (StartService(hService, cArgs, cArgs > 0 ? (LPCTSTR *) apwszArgs : NULL)) { SERVICE_STATUS ss; do { QueryServiceStatus(hService, &ss); if (dwWaitHint) { Sleep(dwWaitHint); } else Sleep(ss.dwWaitHint); } while (ss.dwCurrentState == SERVICE_START_PENDING); CloseServiceHandle(hService); delete [] pwszServiceArgs; return TRUE; } else { CloseServiceHandle(hService); delete [] pwszServiceArgs; } } else { CloseServiceHandle(hSCManager); } } DWORD err = GetLastError(); return FALSE; } BOOL MyStopService(WCHAR * wszServiceName) { SC_HANDLE hSCManager; SC_HANDLE hService; SERVICE_STATUS ss; // Get a handle to the Service Control Manager if (hSCManager = OpenSCManager(NULL, NULL, GENERIC_EXECUTE | SERVICE_QUERY_STATUS )) { // Open a handle to the requested service if (hService = OpenService(hSCManager, wszServiceName, GENERIC_EXECUTE | SERVICE_QUERY_STATUS )) { // Close the service manager's database CloseServiceHandle(hSCManager); // Stop the service if (ControlService(hService, SERVICE_CONTROL_STOP, &ss)) { while (ss.dwCurrentState == SERVICE_STOP_PENDING) { if (dwWaitHint) { Sleep(dwWaitHint); } else Sleep(ss.dwWaitHint); QueryServiceStatus(hService, &ss); }; CloseServiceHandle(hService); return TRUE; } else { CloseServiceHandle(hService); } } else { CloseServiceHandle(hSCManager); } } DWORD err = GetLastError(); return FALSE; } #endif // NO_DCOM