dll/directx/msdvbnp/ethernetfilter.cpp

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS BDA Proxy
   4  * FILE:            dll/directx/msdvbnp/ethernetfilter.cpp
   5  * PURPOSE:         IBDA_EthernetFilter interface
   6  *
   7  * PROGRAMMERS:     Johannes Anderwald (janderwald@reactos.org)
   8  */
   9 #include "precomp.h"
  10
  11 class CEthernetFilter : public IBDA_EthernetFilter
  12 {
  13 public:
  14     STDMETHODIMP QueryInterface( REFIID InterfaceId, PVOID* Interface);
  15
  16     STDMETHODIMP_(ULONG) AddRef();
  17     STDMETHODIMP_(ULONG) Release();
  18
  19     //IBDA_EthernetFilter
  20     HRESULT STDMETHODCALLTYPE GetMulticastListSize(ULONG *pulcbAddresses);
  21     HRESULT STDMETHODCALLTYPE PutMulticastList(ULONG ulcbAddresses, BYTE * pAddressList);
  22     HRESULT STDMETHODCALLTYPE GetMulticastList(ULONG *pulcbAddresses, BYTE *pAddressList);
  23     HRESULT STDMETHODCALLTYPE PutMulticastMode(ULONG ulModeMask);
  24     HRESULT STDMETHODCALLTYPE GetMulticastMode(ULONG *pulModeMask);
  25
  26
  27     CEthernetFilter(IBDA_NetworkProvider * pNetworkProvider);
  28     virtual ~CEthernetFilter();
  29
  30 protected:
  31     IBDA_NetworkProvider * m_pNetworkProvider;
  32     ULONG m_ulcbAddresses;
  33     BYTE * m_pAddressList;
  34     ULONG m_ulModeMask;
  35 };
  36
  37 CEthernetFilter::CEthernetFilter(
  38     IBDA_NetworkProvider * pNetworkProvider) : m_pNetworkProvider(pNetworkProvider),
  39                                                m_ulcbAddresses(0),
  40                                                m_pAddressList(0),
  41                                                m_ulModeMask(0)
  42 {
  43 }
  44
  45 CEthernetFilter::~CEthernetFilter()
  46 {
  47     if (m_pAddressList)
  48         CoTaskMemFree(m_pAddressList);
  49 }
  50
  51 HRESULT
  52 STDMETHODCALLTYPE
  53 CEthernetFilter::QueryInterface(
  54     REFIID InterfaceId,
  55     PVOID* Interface)
  56 {
  57     assert(m_pNetworkProvider);
  58     return m_pNetworkProvider->QueryInterface(InterfaceId, Interface);
  59 }
  60
  61 ULONG
  62 STDMETHODCALLTYPE
  63 CEthernetFilter::AddRef()
  64 {
  65     assert(m_pNetworkProvider);
  66     return m_pNetworkProvider->AddRef();
  67 }
  68
  69 ULONG
  70 STDMETHODCALLTYPE
  71 CEthernetFilter::Release()
  72 {
  73     assert(m_pNetworkProvider);
  74     return m_pNetworkProvider->Release();
  75 }
  76
  77 //-------------------------------------------------------------------
  78 //IBDA_EthernetFilter
  79 //
  80
  81 HRESULT
  82 STDMETHODCALLTYPE
  83 CEthernetFilter::PutMulticastList(
  84     ULONG ulcbAddresses,
  85     BYTE * pAddressList)
  86 {
  87     if (!ulcbAddresses || !pAddressList)
  88         return E_POINTER;
  89
  90     if (m_pAddressList)
  91         CoTaskMemFree(m_pAddressList);
  92
  93     m_pAddressList = (BYTE*)CoTaskMemAlloc(ulcbAddresses);
  94     if (!m_pAddressList)
  95         return E_OUTOFMEMORY;
  96
  97     CopyMemory(m_pAddressList, pAddressList, ulcbAddresses);
  98     m_ulcbAddresses = ulcbAddresses;
  99     return NOERROR;
 100 }
 101
 102 HRESULT
 103 STDMETHODCALLTYPE
 104 CEthernetFilter::GetMulticastList(
 105     ULONG *pulcbAddresses,
 106     BYTE *pAddressList)
 107 {
 108     if (!pulcbAddresses || !pAddressList)
 109         return E_POINTER;
 110
 111     if (*pulcbAddresses < m_ulcbAddresses)
 112         return E_OUTOFMEMORY;
 113
 114     CopyMemory(pAddressList, m_pAddressList, m_ulcbAddresses);
 115     *pulcbAddresses = m_ulcbAddresses;
 116     return NOERROR;
 117 }
 118
 119 HRESULT
 120 STDMETHODCALLTYPE
 121 CEthernetFilter::GetMulticastListSize(
 122     ULONG *pulcbAddresses)
 123 {
 124     if (!pulcbAddresses)
 125         return E_POINTER;
 126
 127     *pulcbAddresses = m_ulcbAddresses;
 128     return NOERROR;
 129 }
 130
 131 HRESULT
 132 STDMETHODCALLTYPE
 133 CEthernetFilter::PutMulticastMode(
 134     ULONG ulModeMask)
 135 {
 136     m_ulModeMask = ulModeMask;
 137     return NOERROR;
 138 }
 139
 140 HRESULT
 141 STDMETHODCALLTYPE
 142 CEthernetFilter::GetMulticastMode(
 143     ULONG *pulModeMask)
 144 {
 145     *pulModeMask = m_ulModeMask;
 146     return NOERROR;
 147 }
 148
 149 HRESULT
 150 WINAPI
 151 CEthernetFilter_fnConstructor(
 152     IBDA_NetworkProvider * pNetworkProvider,
 153     REFIID riid,
 154     LPVOID * ppv)
 155 {
 156     CEthernetFilter * filter = new CEthernetFilter(pNetworkProvider);
 157
 158     if (!filter)
 159         return E_OUTOFMEMORY;
 160
 161     if (FAILED(filter->QueryInterface(riid, ppv)))
 162     {
 163         /* not supported */
 164         delete filter;
 165         return E_NOINTERFACE;
 166     }
 167
 168     return NOERROR;
 169 }