Building Rich COMponents with Attributed ATL

// smServicesEnum.cpp : Implementation of CsmServicesEnum

#include "stdafx.h"
#include "smServicesEnum.h"

#include "smService.h"

// CsmServicesEnum

STDMETHODIMP CsmServicesEnum::Next(unsigned long celt, VARIANT *rgvar, unsigned long *pceltFetched)
{
    if (pceltFetched)
        *pceltFetched = 0;

    if (!rgvar)
        return E_INVALIDARG;

    for (int i = 0; i < celt; i++)
        VariantInit(&rgvar[i]);

    unsigned long fetched = 0;
    while (m_Idx < m_Services.size() && fetched < celt)
    {
        rgvar[fetched].vt = VT_DISPATCH;

        //  Create service
        CService *pService;
        CService::CreateInstance(&pService);

        pService->AddRef();

        pService->Init(m_Services[m_Idx]);
        HRESULT hr = pService->QueryInterface(&rgvar[fetched].pdispVal);
        
        pService->Release();

        if (FAILED(hr))
            break;

        m_Idx++;
        fetched++;
    }

    if (pceltFetched)
        *pceltFetched = fetched;

    return (celt == fetched) ? S_OK : S_FALSE;
}

STDMETHODIMP CsmServicesEnum::Skip(unsigned long celt)
{
    unsigned long i = 0;
    while (m_Idx < m_Services.size() && i < celt)
    {
        m_Idx++;
        i++;
    }

    return (celt == i) ? S_OK : S_FALSE;
}

STDMETHODIMP CsmServicesEnum::Reset()
{
    m_Idx = 0;

    return S_OK;
}

STDMETHODIMP CsmServicesEnum::Clone(IEnumVARIANT **ppenum)
{
    CServicesEnum *pEnum;
    CServicesEnum::CreateInstance(&pEnum);

    pEnum->AddRef();

    pEnum->CloneServices(&m_Services);
    HRESULT hr = pEnum->QueryInterface(ppenum);

    pEnum->Release();

    return hr;
}