crdcache.cpp
上传用户:zhongxx05
上传日期:2007-06-06
资源大小:33641k
文件大小:6k
- /* ***** 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 "hxtypes.h"
- #include "hxstring.h"
- #include "hxmap.h"
- #include "hxbuffer.h"
- #include "chxpckts.h"
- #include "crdcache.h"
- #define DEFAULTCREDENTIALENTRY "defaultCredentialEntry"
- CHXCredentialsCache::CHXCredentialsCache()
- {
- m_lRefCount = 0;
- }
- CHXCredentialsCache::~CHXCredentialsCache()
- {
- Close();
- }
- STDMETHODIMP
- CHXCredentialsCache::QueryInterface(REFIID riid, void**ppvObj)
- {
- QInterfaceList qiList[] =
- {
- { GET_IIDHANDLE(IID_IUnknown), this },
- { GET_IIDHANDLE(IID_IHXCredentialsCache), (IHXCredentialsCache*) this },
- };
- return QIFind(qiList, QILISTSIZE(qiList), riid, ppvObj);
- }
- /////////////////////////////////////////////////////////////////////////
- // Method:
- // IUnknown::AddRef
- // Purpose:
- // Everyone usually implements this the same... feel free to use
- // this implementation.
- //
- STDMETHODIMP_(ULONG32) CHXCredentialsCache::AddRef()
- {
- return InterlockedIncrement(&m_lRefCount);
- }
- /////////////////////////////////////////////////////////////////////////
- // Method:
- // IUnknown::Release
- // Purpose:
- // Everyone usually implements this the same... feel free to use
- // this implementation.
- //
- STDMETHODIMP_(ULONG32) CHXCredentialsCache::Release()
- {
- if (InterlockedDecrement(&m_lRefCount) > 0)
- {
- return m_lRefCount;
- }
- delete this;
- return 0;
- }
- STDMETHODIMP_(BOOL)
- CHXCredentialsCache::IsEmpty(IHXBuffer* pBuffer)
- {
- const char* pKey = NULL;
- CredentialEntry* pCredentialEntry = NULL;
- if (!pBuffer)
- {
- pKey = DEFAULTCREDENTIALENTRY;
- }
- else
- {
- pKey = (const char*)pBuffer->GetBuffer();
- }
- if (m_credentialMap.Lookup(pKey, (void*&)pCredentialEntry) &&
- pCredentialEntry)
- {
- return FALSE;
- }
- else
- {
- return TRUE;
- }
- }
- STDMETHODIMP
- CHXCredentialsCache::Empty(IHXBuffer* pBuffer)
- {
- HX_RESULT hr = HXR_OK;
- const char* pKey = NULL;
- CredentialEntry* pCredentialEntry = NULL;
- if (!pBuffer)
- {
- pKey = DEFAULTCREDENTIALENTRY;
- }
- else
- {
- pKey = (const char*)pBuffer->GetBuffer();
- }
- if (m_credentialMap.Lookup(pKey, (void*&)pCredentialEntry) &&
- pCredentialEntry)
- {
- HX_DELETE(pCredentialEntry);
-
- m_credentialMap.RemoveKey(pKey);
- }
- return hr;
- }
- STDMETHODIMP
- CHXCredentialsCache::FillCredentials(REF(IHXValues*) pValues)
- {
- HX_RESULT hr = HXR_FAILED;
- IHXBuffer* pRealm = NULL;
- CredentialEntry* pCredentialEntry = NULL;
- if (!pValues)
- {
- goto cleanup;
- }
-
- if (HXR_OK != pValues->GetPropertyCString("Realm", pRealm))
- {
- pRealm = new CHXBuffer();
- pRealm->AddRef();
- pRealm->Set((const unsigned char*)DEFAULTCREDENTIALENTRY, strlen(DEFAULTCREDENTIALENTRY)+1);
- }
-
- if (m_credentialMap.Lookup((const char*)pRealm->GetBuffer(), (void*&)pCredentialEntry) &&
- pCredentialEntry)
- {
- pValues->SetPropertyCString("Username", pCredentialEntry->m_pUserName);
- pValues->SetPropertyCString("Password", pCredentialEntry->m_pPassword);
- hr = HXR_OK;
- }
- cleanup:
- HX_RELEASE(pRealm);
- return hr;
- }
- STDMETHODIMP
- CHXCredentialsCache::SetCredentials(IHXValues* pValues)
- {
- HX_RESULT hr = HXR_OK;
- IHXBuffer* pRealm = NULL;
- IHXBuffer* pUserName = NULL;
- IHXBuffer* pPassword = NULL;
- CredentialEntry* pCredentialEntry = NULL;
- if (!pValues)
- {
- goto cleanup;
- }
-
- if (HXR_OK != pValues->GetPropertyCString("Realm", pRealm))
- {
- pRealm = new CHXBuffer();
- pRealm->AddRef();
- pRealm->Set((const unsigned char*)DEFAULTCREDENTIALENTRY, strlen(DEFAULTCREDENTIALENTRY)+1);
- }
-
- Empty(pRealm);
- if (HXR_OK == pValues->GetPropertyCString("Username", pUserName) && pUserName &&
- HXR_OK == pValues->GetPropertyCString("Password", pPassword) && pPassword)
- {
- pCredentialEntry = new CredentialEntry((char*)pUserName->GetBuffer(),
- (char*)pPassword->GetBuffer());
- m_credentialMap.SetAt((const char*)pRealm->GetBuffer(), (void*)pCredentialEntry);
- }
- cleanup:
- HX_RELEASE(pRealm);
- HX_RELEASE(pUserName);
- HX_RELEASE(pPassword);
- return hr;
- }
- void
- CHXCredentialsCache::Close()
- {
- CHXString key;
- POSITION p;
- CredentialEntry* pCredentialEntry = NULL;
-
- p = m_credentialMap.GetStartPosition();
- while (p)
- {
- m_credentialMap.GetNextAssoc(p, key, (void*&)pCredentialEntry);
- HX_DELETE(pCredentialEntry);
- }
- m_credentialMap.RemoveAll();
- }