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.
 
 
 
 
 
 

163 lines
3.6 KiB

//
// enumsrvmru.cpp: Implements IEnumStr for the server MRU list
// used by autocomplete code
//
// Copyright Microsoft Corporation 2000
#include "stdafx.h"
#ifndef OS_WINCE
#define TRC_GROUP TRC_GROUP_UI
#define TRC_FILE "enumsrvmru"
#include <atrcapi.h>
#include "enumsrvmru.h"
STDMETHODIMP CEnumSrvMru::QueryInterface(
REFIID riid,
void **ppvObject)
{
DC_BEGIN_FN("QueryInterface");
TRC_ASSERT(ppvObject, (TB,_T("ppvObject is NULL\n")));
if(!ppvObject)
{
return E_INVALIDARG;
}
if ( IID_IEnumString == riid )
*ppvObject = (void *)((IEnumString*)this);
else if ( IID_IUnknown == riid )
*ppvObject = (void *)((IUnknown *)this);
else
{
*ppvObject = 0;
return E_NOINTERFACE;
}
AddRef();
DC_END_FN();
return S_OK;
} //QueryInterface
STDMETHODIMP_(ULONG) CEnumSrvMru::AddRef()
{
return InterlockedIncrement(&_refCount);
} //AddRef
STDMETHODIMP_(ULONG) CEnumSrvMru::Release()
{
DC_BEGIN_FN("Release");
TRC_ASSERT(_refCount > 0, (TB,_T("_refCount invalid %d"), _refCount));
LONG refCount = InterlockedDecrement(&_refCount);
if ( refCount <= 0 )
delete this;
DC_END_FN();
return (ULONG) refCount;
} //Release
//Clone a copy of this object
STDMETHODIMP CEnumSrvMru::Clone(IEnumString ** ppEnumStr)
{
return E_NOTIMPL;
}
//
// Next enum
// celt - number of elements requested
// rgelt - array of elements to return
// pceltFetched - pointer to number of elements actually supplied
//
STDMETHODIMP CEnumSrvMru::Next( ULONG celt,
LPOLESTR * rgelt,
ULONG * pceltFetched )
{
DC_BEGIN_FN("Next");
*pceltFetched = 0;
while( _iCurrEnum < SH_NUM_SERVER_MRU &&
*pceltFetched < celt)
{
//Need to allocate powerful COM memory
//caller frees
LPOLESTR pwzMRU= (LPOLESTR)CoTaskMemAlloc(SH_MAX_ADDRESS_LENGTH*sizeof(OLECHAR));
if(!pwzMRU)
{
return E_OUTOFMEMORY;
}
DC_WSTRNCPY(pwzMRU, _szMRU[_iCurrEnum++], SH_MAX_ADDRESS_LENGTH);
rgelt[(*pceltFetched)++] = pwzMRU;
}
//
// Fill in remaining request items with NULLS
//
ULONG cAdded = *pceltFetched;
while (cAdded < celt)
{
rgelt[cAdded++] = NULL;
}
DC_END_FN();
return *pceltFetched == celt ? S_OK : S_FALSE;
}
//
// Skips celt elements
// if cannot skip as many as requested don't skip any
//
STDMETHODIMP CEnumSrvMru::Skip( ULONG celt )
{
DC_BEGIN_FN("Next");
TRC_ASSERT(_iCurrEnum < SH_NUM_SERVER_MRU, (TB,_T("_iCurEnum out of range: %d"),
_iCurrEnum));
if(_iCurrEnum + celt < SH_NUM_SERVER_MRU)
{
_iCurrEnum += celt;
return S_OK;
}
DC_END_FN();
return S_FALSE;
}
//
// Initialize the string collection with strings
// from the TscSettings's server MRU list
//
BOOL CEnumSrvMru::InitializeFromTscSetMru( CTscSettings* pTscSet)
{
DC_BEGIN_FN("InitializeFromSHMru");
USES_CONVERSION;
TRC_ASSERT(pTscSet, (TB,_T("pTscSet NULL")));
if(!pTscSet)
{
return FALSE;
}
for(int i=0; i<TSC_NUM_SERVER_MRU; i++)
{
PWCHAR wszServer = T2W( (LPTSTR)pTscSet->GetMRUServer(i));
if(!wszServer)
{
return FALSE;
}
DC_WSTRNCPY(_szMRU[i], wszServer, SH_MAX_ADDRESS_LENGTH);
}
DC_END_FN();
return TRUE;
}
#endif //OS_WINCE