hxnetif.cpp
上传用户:zhongxx05
上传日期:2007-06-06
资源大小:33641k
文件大小:10k
- /* ***** BEGIN LICENSE BLOCK *****
- * Version: RCSL 1.0/RPSL 1.0
- *
- * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved.
- *
- * The contents of this file, and the files included with this file, are
- * subject to the current version of the RealNetworks Public Source License
- * Version 1.0 (the "RPSL") available at
- * http://www.helixcommunity.org/content/rpsl unless you have licensed
- * the file under the RealNetworks Community Source License Version 1.0
- * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl,
- * in which case the RCSL will apply. You may also obtain the license terms
- * directly from RealNetworks. You may not use this file except in
- * compliance with the RPSL or, if you have a valid RCSL with RealNetworks
- * applicable to this file, the RCSL. Please see the applicable RPSL or
- * RCSL for the rights, obligations and limitations governing use of the
- * contents of the file.
- *
- * This file is part of the Helix DNA Technology. RealNetworks is the
- * developer of the Original Code and owns the copyrights in the portions
- * it created.
- *
- * This file, and the files included with this file, is distributed and made
- * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
- * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES,
- * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS
- * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
- *
- * Technology Compatibility Kit Test Suite(s) Location:
- * http://www.helixcommunity.org/content/tck
- *
- * Contributor(s):
- *
- * ***** END LICENSE BLOCK ***** */
- #include "hxcom.h"
- #include <string.h>
- #include <stdio.h>
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <time.h>
- #include "hxresult.h"
- #include "hxslist.h"
- #include "netbyte.h"
- #include "hxengin.h"
- #include "hxnetif.h"
- #ifdef _UNIX
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <netinet/in.h>
- #include <sys/socket.h>
- #if defined _SOLARIS || defined _FREEBSD || defined _OPENBSD || defined _NETBSD
- #include <sys/sockio.h>
- #endif
- #include <net/if.h>
- #include <sys/ioctl.h>
- #endif
- #if defined (_UNIX) && !defined(_SUN) && !defined(_SCO_UW) && !defined(_HPUX) && !defined(_IRIX) && !defined(_OSF1)
- #include <sys/file.h>
- #endif /* UNIX */
- HXNetInterface::HXNetInterface(IUnknown* pContext)
- : m_lRefCount(0)
- , m_bInitialized(FALSE)
- , m_pNetInterfaceList(NULL)
- , m_pSinkList(NULL)
- {
- }
- HXNetInterface::~HXNetInterface()
- {
- Close();
- }
- STDMETHODIMP
- HXNetInterface::QueryInterface(REFIID riid, void**ppvObj)
- {
- if (IsEqualIID(riid, IID_IUnknown))
- {
- AddRef();
- *ppvObj = this;
- return HXR_OK;
- }
- else if (IsEqualIID(riid, IID_IHXNetInterfaces))
- {
- AddRef();
- *ppvObj = (IHXNetInterfaces*)this;
- return HXR_OK;
- }
- *ppvObj = NULL;
- return HXR_NOINTERFACE;
- }
- /////////////////////////////////////////////////////////////////////////
- // Method:
- // IUnknown::AddRef
- // Purpose:
- // Everyone usually implements this the same... feel free to use
- // this implementation.
- //
- STDMETHODIMP_(ULONG32)
- HXNetInterface::AddRef()
- {
- return InterlockedIncrement(&m_lRefCount);
- }
- /////////////////////////////////////////////////////////////////////////
- // Method:
- // IUnknown::Release
- // Purpose:
- // Everyone usually implements this the same... feel free to use
- // this implementation.
- //
- STDMETHODIMP_(ULONG32)
- HXNetInterface::Release()
- {
- if (InterlockedDecrement(&m_lRefCount) > 0)
- {
- return m_lRefCount;
- }
- delete this;
- return 0;
- }
- STDMETHODIMP
- HXNetInterface::UpdateNetInterfaces(void)
- {
- HX_RESULT rc = HXR_OK;
- BOOL bChanged = FALSE;
- bChanged = IsNetInterfaceChanged();
- if (!m_bInitialized)
- {
- m_bInitialized = TRUE;
- }
- else if (bChanged && m_pSinkList)
- {
- CHXSimpleList::Iterator ndx = m_pSinkList->Begin();
- for (; ndx != m_pSinkList->End(); ++ndx)
- {
- IHXNetInterfacesAdviseSink* pSink = (IHXNetInterfacesAdviseSink*) (*ndx);
- pSink->NetInterfacesUpdated();
- }
- }
-
- return rc;
- }
- HX_RESULT
- HXNetInterface::RetrieveNetInterface(CHXSimpleList*& pNetInterfaceList)
- {
- HX_RESULT rc = HXR_OK;
- int i, fd, num;
- char buf[BUFSIZ]; /* Flawfinder: ignore */
- long lFlags = 0;
- NIInfo* pNIInfo = NULL;
- struct ifconf ifc;
- struct ifreq* ifr;
- fd = socket(AF_INET, SOCK_DGRAM, 0);
- if (-1 == fd)
- {
- rc = HXR_FAILED;
- goto cleanup;
- }
- ifc.ifc_len = BUFSIZ;
- ifc.ifc_buf = buf;
- // get network interface configuration
- if (-1 == ioctl(fd, SIOCGIFCONF, &ifc))
- {
- rc = HXR_FAILED;
- goto cleanup;
- }
- num = ifc.ifc_len / sizeof(struct ifreq);
- // iterate through the network interface
- for (i = 0, ifr = ifc.ifc_req; i < num; i++, ifr += 1)
- {
- if (-1 != ioctl(fd, SIOCGIFFLAGS, ifr))
- {
- pNIInfo = new NIInfo;
-
- lFlags = ifr->ifr_flags;
- if (lFlags & IFF_LOOPBACK)
- {
- pNIInfo->type = LOOPBACK;
- }
- else if (lFlags & IFF_POINTOPOINT)
- {
- pNIInfo->type = POINTTOPOINT;
- }
- else if (lFlags & IFF_BROADCAST)
- {
- pNIInfo->type = BROADCAST;
- }
- if (lFlags & IFF_UP)
- {
- pNIInfo->bActive = TRUE;
- }
- if (-1 != ioctl(fd, SIOCGIFADDR, ifr))
- {
- pNIInfo->ulNetAddress =
- ((struct sockaddr_in *)&ifr->ifr_broadaddr)->sin_addr.s_addr;
- }
-
- if (-1 != ioctl(fd, SIOCGIFNETMASK, ifr))
- {
- pNIInfo->ulNetMask =
- ((struct sockaddr_in *)&ifr->ifr_broadaddr)->sin_addr.s_addr;
- }
- if (!pNetInterfaceList)
- {
- pNetInterfaceList = new CHXSimpleList;
- }
- pNetInterfaceList->AddTail(pNIInfo);
- }
- }
- cleanup:
- close(fd);
-
- return rc;
- }
- STDMETHODIMP_(UINT32)
- HXNetInterface::GetNumOfNetInterfaces()
- {
- if (!m_bInitialized)
- {
- UpdateNetInterfaces();
- }
- return m_pNetInterfaceList ? m_pNetInterfaceList->GetCount() : 0;
- }
- STDMETHODIMP HXNetInterface::GetNetInterfaces(UINT16 lIndex,
- REF(NIInfo*) pNIInfo)
- {
- HX_RESULT rc = HXR_OK;
- int i = 0;
- CHXSimpleList::Iterator iter;
-
- pNIInfo = NULL;
-
- if (!m_bInitialized)
- {
- UpdateNetInterfaces();
- }
-
- if (m_pNetInterfaceList)
- {
- iter = m_pNetInterfaceList->Begin();
- for (; iter != m_pNetInterfaceList->End(); ++iter, ++i)
- {
- NIInfo* pInfo = (NIInfo*)(*iter);
- if (i == lIndex)
- {
- pNIInfo = pInfo;
- break;
- }
- }
- }
-
- if (!pNIInfo)
- {
- rc = HXR_FAILED;
- }
-
- cleanup:
-
- return rc;
- }
- STDMETHODIMP
- HXNetInterface::AddAdviseSink(IHXNetInterfacesAdviseSink* pSink)
- {
- HX_RESULT rc = HXR_OK;
- if (!m_pSinkList)
- {
- m_pSinkList = new CHXSimpleList();
- }
- pSink->AddRef();
- m_pSinkList->AddTail(pSink);
- return rc;
- }
- STDMETHODIMP
- HXNetInterface::RemoveAdviseSink(IHXNetInterfacesAdviseSink* pSink)
- {
- HX_RESULT rc = HXR_OK;
- LISTPOSITION lPosition = m_pSinkList->Find(pSink);
- if (!lPosition)
- {
- rc = HXR_UNEXPECTED;
- goto cleanup;
- }
- m_pSinkList->RemoveAt(lPosition);
- pSink->Release();
- cleanup:
- return rc;
- }
- BOOL HXNetInterface::IsNetInterfaceChanged(void)
- {
- BOOL bResult = FALSE;
- CHXSimpleList* pTempNetInterfaceList = new CHXSimpleList();
- RetrieveNetInterface(pTempNetInterfaceList);
-
- if (pTempNetInterfaceList && m_pNetInterfaceList)
- {
- if (pTempNetInterfaceList->GetCount() != m_pNetInterfaceList->GetCount())
- {
- bResult = TRUE;
- }
- else
- {
- CHXSimpleList::Iterator ndx0 = pTempNetInterfaceList->Begin();
- CHXSimpleList::Iterator ndx1 = m_pNetInterfaceList->Begin();
- for (; ndx0 != pTempNetInterfaceList->End() && ndx1 != m_pNetInterfaceList->End(); ++ndx0, ++ndx1)
- {
- NIInfo* pInfo0 = (NIInfo*)(*ndx0);
- NIInfo* pInfo1 = (NIInfo*)(*ndx1);
-
- if (pInfo0->ulNetAddress != pInfo1->ulNetAddress ||
- pInfo0->ulNetMask != pInfo1->ulNetMask)
- {
- bResult = TRUE;
- }
- }
- }
- }
- else if (pTempNetInterfaceList != m_pNetInterfaceList)
- {
- bResult = TRUE;
- }
-
- if (bResult)
- {
- Reset(m_pNetInterfaceList);
- HX_DELETE(m_pNetInterfaceList);
-
- m_pNetInterfaceList = pTempNetInterfaceList;
- }
- else
- {
- Reset(pTempNetInterfaceList);
- HX_DELETE(pTempNetInterfaceList);
- }
-
- return bResult;
- }
- void
- HXNetInterface::Reset(CHXSimpleList* pNetInterfaceList)
- {
- if (pNetInterfaceList)
- {
- while (pNetInterfaceList->GetCount())
- {
- NIInfo* pNIInfo = (NIInfo*)pNetInterfaceList->RemoveHead();
- HX_DELETE(pNIInfo);
- }
- }
- }
- void
- HXNetInterface::Close(void)
- {
- Reset(m_pNetInterfaceList);
- HX_DELETE(m_pNetInterfaceList);
- if (m_pSinkList)
- {
- HX_ASSERT(m_pSinkList->GetCount() == 0);
- CHXSimpleList::Iterator ndx = m_pSinkList->Begin();
- for (; ndx != m_pSinkList->End(); ++ndx)
- {
- IHXNetInterfacesAdviseSink* pSink = (IHXNetInterfacesAdviseSink*) (*ndx);
- HX_RELEASE(pSink);
- }
- HX_DELETE(m_pSinkList);
- }
- }