实现 IMFNetCredentialManager
在 IMFNetCredentialManager::BeginGetCredentials 方法中,执行以下操作。
- 如果还没有 IMFNetCredentialCache 指针,请调用 MFCreateCredentialCache 来创建凭据缓存对象。 存储此指针。
- 调用 IMFNetCredentialCache::GetCredential。 按如下所示在 dwAuthenticationFlags 参数中设置标志:
- 如果 MFNetCredentialManagerGetParam 结构的 hrOp 成员等于 NS_E_PROXY_ACCESSDENIED,请设置 MFNET_AUTHENTICATION_PROXY 标志。
- 如果 fClearTextPackage 为 TRUE,请设置 MFNET_AUTHENTICATION_CLEAR_TEXT 标志。
- 如果 fAllowLoggedOnUser 为 TRUE,请设置 MFNET_AUTHENTICATION_LOGGED_ON_USER 标志。
- GetCredential 方法返回 IMFNetCredential 指针,可能返回REQUIRE_PROMPT标志。 存储 IMFNetCredential 指针。
- 如果 GetCredential 未返回 REQUIRE_PROMPT 标志,则表示已完成。 跳到步骤 9。
- 否则,如果 GetCredential 返回 REQUIRE_PROMPT 标志,则必须提示用户输入其用户名和密码。
- 如果 fClearTextPackage 为 FALSE,请加密凭据。
- 调用 IMFNetCredential::SetUser 和 IMFNetCredential::SetPassword ,以在凭据对象上设置用户名和密码。
- (可选)调用 IMFNetCredentialCache::SetUserOptions 以使用用户存储和发送凭据的首选项更新凭据缓存对象。
- 通过调用 MFInvokeCallback 调用 IMFAsyncCallback 回调。
在 IMFNetCredentialManager::EndGetCredentials 方法中,返回在 BeginGetCredentials 方法中获取的 IMFNetCredential 指针。
在 IMFNetCredentialManager::SetGood 方法中,将输入参数直接传递到 IMFNetCredentialCache::SetGood 方法。 这会通知凭据缓存服务器是否接受凭据。
如果需要提示用户 (步骤 5) 或加密 (步骤 6) 的凭据,则应在工作队列线程上执行此操作,以免阻止 Microsoft Media Foundation 管道。 调用 MFPutWorkItem ,然后在工作队列回调中执行剩余步骤。
注意
请注意,在创建网络源时,可能会调用 BeginGetCredentials 。 因此,如果通过调用同步 的 IMFSourceResolver::CreateObjectFromURL 方法创建网络源,则调用线程可能会在获取凭据时阻塞。 因此,建议改用异步 IMFSourceResolver::BeginCreateObjectFromURL 方法。
示例
此示例演示凭据管理器可以提供的一种行为类型。
下面是实现 IMFNetCredentialManager 的类的声明。
class CCredentialManager : public IMFNetCredentialManager, IMFAsyncCallback
{
long m_cRef;
IMFNetCredentialCache *m_pCredentialCache;
public:
CCredentialManager () : m_cRef(1), m_pCredentialCache(NULL)
{
}
~CCredentialManager()
{
SafeRelease(&m_pCredentialCache);
}
STDMETHODIMP QueryInterface(REFIID riid, void** ppv)
{
static const QITAB qit[] =
{
QITABENT(CCredentialManager, IMFNetCredentialManager),
QITABENT(CCredentialManager, IMFAsyncCallback),
{ 0 }
};
return QISearch(this, qit, riid, ppv);
}
STDMETHODIMP_(ULONG) AddRef()
{
return InterlockedIncrement(&m_cRef);
}
STDMETHODIMP_(ULONG) Release()
{
LONG cRef = InterlockedDecrement(&m_cRef);
if (cRef == 0)
{
delete this;
}
return cRef;
}
STDMETHODIMP BeginGetCredentials(
MFNetCredentialManagerGetParam* pParam,
IMFAsyncCallback* pCallback,
IUnknown* pState
);
STDMETHODIMP EndGetCredentials(
IMFAsyncResult* pResult,
IMFNetCredential** ppCred);
STDMETHODIMP SetGood(IMFNetCredential* pCred, BOOL fGood)
{
if (!pCred)
{
return E_POINTER;
}
return m_pCredentialCache->SetGood(pCred, fGood);
}
STDMETHODIMP GetParameters(DWORD* pdwFlags, DWORD* pdwQueue)
{
return E_NOTIMPL;
}
STDMETHODIMP Invoke(IMFAsyncResult* pResult);
};
为了跟踪 BeginGetCredentials 操作的状态,类使用以下帮助程序对象:
// Holds state information for the GetCredentials operation, so that work can
// be moved to a work-queue thread.
struct CredentialOp : public IUnknown
{
long m_cRef;
IMFNetCredential *m_pCredential;
DWORD m_dwFlags;
CredentialOp(IMFNetCredential *pCredential)
: m_cRef(1), m_dwFlags(0), m_pCredential(pCredential)
{
m_pCredential->AddRef();
}
~CredentialOp()
{
SafeRelease(&m_pCredential);
}
STDMETHODIMP QueryInterface(REFIID riid, void** ppv)
{
static const QITAB qit[] =
{
QITABENT(CredentialOp, IUnknown),
{ 0 }
};
return QISearch(this, qit, riid, ppv);
}
STDMETHODIMP_(ULONG) AddRef()
{
return InterlockedIncrement(&m_cRef);
}
STDMETHODIMP_(ULONG) Release()
{
LONG cRef = InterlockedDecrement(&m_cRef);
if (cRef == 0)
{
delete this;
}
return cRef;
}
};
BeginGetCredentials 方法创建凭据缓存并获取 IMFNetCredential 指针。 如果必须提示用户 (由 REQUIRE_PROMPT 标志) 指示,该方法将调用 MFPutWorkItem 以将新工作项排队:
STDMETHODIMP CCredentialManager::BeginGetCredentials(
MFNetCredentialManagerGetParam* pParam,
IMFAsyncCallback* pCallback,
IUnknown* pState
)
{
if (!pParam || !pCallback)
{
return E_POINTER;
}
DWORD dwAuthenticationFlags = 0;
DWORD dwRequirementFlags = 0;
if (pParam->hrOp == NS_E_PROXY_ACCESSDENIED)
{
dwAuthenticationFlags |= MFNET_AUTHENTICATION_PROXY;
}
if (pParam->fAllowLoggedOnUser)
{
dwAuthenticationFlags |= MFNET_AUTHENTICATION_LOGGED_ON_USER;
}
if (pParam->fClearTextPackage)
{
dwAuthenticationFlags |= MFNET_AUTHENTICATION_CLEAR_TEXT;
}
IMFNetCredential *pCredential = NULL;
IMFAsyncResult* pResult = NULL;
HRESULT hr = S_OK;
if (m_pCredentialCache == NULL)
{
hr = MFCreateCredentialCache(&m_pCredentialCache);
if (FAILED(hr))
{
goto done;
}
}
hr = m_pCredentialCache->GetCredential(
pParam->pszUrl,
pParam->pszRealm,
dwAuthenticationFlags,
&pCredential,
&dwRequirementFlags
);
if (FAILED(hr))
{
goto done;
}
if( ( dwRequirementFlags & REQUIRE_PROMPT ) == 0 )
{
// The credential is good to use. Prompting the user is not required.
hr = S_OK;
goto done;
}
// The credential requires prompting the user.
CredentialOp *pOp = new (std::nothrow) CredentialOp(pCredential);
if (pOp == NULL)
{
hr = E_OUTOFMEMORY;
goto done;
}
// Set flags. Use these to inform the user if the credentials will
// be sent in plaintext or saved in the credential cache.
if (pParam->fClearTextPackage)
{
// Notify the user that credentials will be sent in plaintext.
pOp->m_dwFlags |= MFNET_CREDENTIAL_ALLOW_CLEAR_TEXT;
}
if(dwRequirementFlags & REQUIRE_SAVE_SELECTED )
{
// Credentials will be saved in the cache by default.
pOp->m_dwFlags |= MFNET_CREDENTIAL_SAVE;
}
// NOTE: The application should enable to user to deselect these two flags;
// for example, through check boxes in the prompt dialog.
// Now queue the work item.
hr = MFCreateAsyncResult(pOp, pCallback, pState, &pResult);
if (FAILED(hr))
{
goto done;
}
hr = MFPutWorkItem(MFASYNC_CALLBACK_QUEUE_LONG_FUNCTION, this, pResult);
done:
SafeRelease(&pResult);
SafeRelease(&pCredential);
SafeRelease(&pOp);
return hr;
}
工作队列线程调用 Invoke,这会提示用户,然后调用 MFInvokeCallback 来调用 BeginGetCredentials 中提供的回调指针。
STDMETHODIMP CCredentialManager::Invoke(IMFAsyncResult* pResult)
{
IUnknown *pState = NULL;
IMFAsyncResult *pGetCredentialsResult = NULL;
IUnknown *pOpState = NULL;
CredentialOp *pOp = NULL; // not AddRef'd
HRESULT hr = pResult->GetState(&pState);
if (SUCCEEDED(hr))
{
hr = pState->QueryInterface(IID_PPV_ARGS(&pGetCredentialsResult));
}
if (SUCCEEDED(hr))
{
hr = pGetCredentialsResult->GetObject(&pOpState);
}
if (SUCCEEDED(hr))
{
pOp = static_cast<CredentialOp*>(pOpState);
// Display a dialog for the user to enter user name and password.
hr = PromptUserCredentials(pOp);
}
if (SUCCEEDED(hr) && m_pCredentialCache)
{
// Update with options set by the user.
hr = m_pCredentialCache->SetUserOptions(
pOp->m_pCredential,
pOp->m_dwFlags
);
}
if (pGetCredentialsResult)
{
pGetCredentialsResult->SetStatus(hr);
MFInvokeCallback(pGetCredentialsResult);
}
SafeRelease(&pState);
SafeRelease(&pGetCredentialsResult);
SafeRelease(&pOpState);
return S_OK;
}
EndGetCredentials 方法通过返回指向调用方的 IMFNetCredential 指针来完成操作。
STDMETHODIMP CCredentialManager::EndGetCredentials(
IMFAsyncResult* pResult,
IMFNetCredential** ppCred
)
{
if (!pResult || !ppCred)
{
return E_POINTER;
}
*ppCred = NULL;
IUnknown *pUnk = NULL;
// Check the result of the asynchronous operation.
HRESULT hr = pResult->GetStatus();
if (FAILED(hr))
{
// The operation failed.
goto done;
}
hr = pResult->GetObject(&pUnk);
if (FAILED(hr))
{
goto done;
}
CredentialOp *pOp = static_cast<CredentialOp*>(pUnk);
*ppCred = pOp->m_pCredential;
pOp->m_pCredential = NULL;
done:
SafeRelease(&pUnk);
return hr;
}
相关主题