Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

175 lines
3.8 KiB

//+------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1993.
//
// File: bm_regis.cxx
//
// Contents: Ole object registration test
//
// Classes: COleRegistrationTest
//
// History: 12-July-93 t-martig Created
//
//--------------------------------------------------------------------------
#include <headers.cxx>
#pragma hdrstop
#include <bm_regis.hxx>
#include <oletest.hxx>
extern COleTestClassFactory theFactory;
const CLSID *apClsid[] = {&CLSID_COleTestClass1,
&CLSID_COleTestClass2,
&CLSID_COleTestClass3,
&CLSID_COleTestClass4,
&CLSID_COleTestClass5,
&CLSID_COleTestClass6,
&CLSID_COleTestClass7,
&CLSID_COleTestClass8};
TCHAR *COleRegistrationTest::Name ()
{
return TEXT("ObjRegistration");
}
SCODE COleRegistrationTest::Setup (CTestInput *pInput)
{
HRESULT sc;
CTestBase::Setup(pInput);
// get the iteration count
m_ulIterations = pInput->GetIterations(Name());
// for the inproc case. load a class factory object to register.
// this is temporary to fix a problem (ambiguity) in inproc
// registration.
sc = pInput->GetGUID(&m_Clsid[0], Name(), apszClsIDName[0]);
if (FAILED(sc))
{
Log (TEXT("Setup - GetClassID failed."), sc);
return sc;
}
sc = pInput->GetGUID(&m_Clsid[1], Name(), apszClsIDName[1]);
if (FAILED(sc))
{
Log (TEXT("Setup - GetClassID failed."), sc);
return sc;
}
sc = InitCOM();
if (FAILED(sc))
{
Log (TEXT("Setup - CoInitialize failed."), sc);
return sc;
}
// get the class factory
sc = CoGetClassObject(m_Clsid[0], dwaClsCtx[0], NULL,
IID_IClassFactory, (void **)&m_apICF[0]);
if (FAILED(sc))
{
Log (TEXT("Setup - CoGetClassObject failed."), sc);
return sc;
}
// for local server, register an internal class.
m_apICF[1] = (IClassFactory *)&theFactory;
// init internal state
for (ULONG iCtx=0; iCtx<CNT_CLSCTX; iCtx++)
{
INIT_RESULTS(m_ulRegisterCtx[iCtx]);
INIT_RESULTS(m_ulRevokeCtx[iCtx]);
}
// _pInput = pInput;
return S_OK;
}
SCODE COleRegistrationTest::Cleanup ()
{
m_apICF[0]->Release();
UninitCOM();
return S_OK;
}
SCODE COleRegistrationTest::Run ()
{
CStopWatch sw;
DWORD dwReg;
for (ULONG iCtx=0; iCtx<CNT_CLSCTX; iCtx++)
{
for (ULONG iIter=0; iIter<m_ulIterations; iIter++)
{
sw.Reset ();
SCODE sc = CoRegisterClassObject(m_Clsid[iCtx],
m_apICF[iCtx],
dwaClsCtx[iCtx],
REGCLS_MULTIPLEUSE, &dwReg);
m_ulRegisterCtx[iCtx][iIter] = sw.Read ();
Log (TEXT("CoRegisterClassObject"),sc);
// NOTE: Revoke has different behaviour depending on
// whether single use or multiuse is specified in the
// registration.
if (SUCCEEDED(sc))
{
// Revoke
sw.Reset ();
sc = CoRevokeClassObject(dwReg);
m_ulRevokeCtx[iCtx][iIter] = sw.Read();
if (Log (TEXT("CoRevokeClassObject"), sc))
m_ulRevokeCtx[iCtx][iIter] = NOTAVAIL;
}
else
{
m_ulRegisterCtx[iCtx][iIter] = NOTAVAIL;
}
// _pInput->Pause(IP_ITERPAUSE);
}
}
return S_OK;
}
SCODE COleRegistrationTest::Report (CTestOutput &output)
{
output.WriteSectionHeader (Name(),
TEXT("CoRegisterClassObject / CoRevokeClassObject"), *m_pInput);
for (ULONG iCtx=0; iCtx<CNT_CLSCTX; iCtx++)
{
output.WriteString (TEXT("\n"));
output.WriteClassID(&m_Clsid[iCtx]);
output.WriteString (apszClsCtx[iCtx]);
output.WriteString (TEXT("\n"));
output.WriteResults (TEXT("Register"), m_ulIterations,
m_ulRegisterCtx[iCtx]);
output.WriteResults (TEXT("Revoke "), m_ulIterations,
m_ulRevokeCtx[iCtx]);
}
return S_OK;
}