实现 IMFNetCredentialManager

项目
06/13/2023

在 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;
}

网络源身份验证

Share via

实现 IMFNetCredentialManager

示例

反馈

反馈

其他资源

Share via

实现 IMFNetCredentialManager

示例

相关主题

反馈

反馈

其他资源