示例:将 SSPI 身份验证编码与 BITS 配合使用
可以使用安全支持提供程序接口 (SSPI) 身份验证和后台智能传输服务 (BITS) 方法从用户获取凭据、对凭据进行编码,并在 BITS 传输作业上设置编码凭据。 编码是将凭据结构转换为可以传递给 BITS 传输作业的字符串所必需的。
有关 SSPI 身份验证和方法的详细信息,请参阅 SSPI。
以下过程使用 Negotiate 安全包提示用户提供凭据。 程序创建身份验证标识结构,并使用表示用户用户名、域和密码的编码字符串填充结构。 然后,程序创建 BITS 下载作业,并将编码的用户名和密码设置为作业的凭据。 程序在不再需要身份验证标识结构后释放它。
此示例使用示例:公共类中定义的标头和实现。
将 SSPI 身份验证编码与 BITS 传输作业配合使用
- 通过调用 CCoInitializer 函数初始化 COM 参数。 有关 CCoInitializer 函数的详细信息,请参阅示例:公共类。
- 获取 IBackgroundCopyManager、IBackgroundCopyJob、IBackgroundCopyJob2 接口的指针。 此示例使用 CComPtr 类管理 COM 接口指针。
- 创建一个 CREDUI_INFO 结构,其中包含用于自定义 SspiPromptForCredentials Function 函数对话框外观的信息。 然后提示用户输入凭据。 有关详细信息,请参阅 SspiPromptForCredentials 函数。
- 使用 SspiEncodeAuthIdentityAsStrings 函数将凭据结构编码为可传递给 BITS 传输作业的字符串。
- 准备 BG_AUTH_CREDENTIALS 结构。
- 通过调用 CoInitializeSecurity 初始化 COM 进程安全性。 BITS 至少需要模拟级别的模拟。 如果未设置正确的模拟级别,BITS 将以 E_ACCESSDENIED 失败。
- 通过调用 CoCreateInstance 函数获取 BITS 的初始定位符。
- 通过调用 IBackgroundCopyManager::CreateJob 方法创建 BITS 传输作业。
- 获取 IBackgroundCopyJob2 接口的标识符,并调用 IBackgroundCopyJob::QueryInterface 方法。
- 使用编码的用户名和密码字符串填充 BG_AUTH_CREDENTIALS 结构,并将身份验证方案设置为“协商”(BG_AUTH_SCHEME_NEGOTIATE)。
- 使用 IBackgroundCopyJob2 指针向 BITS 发出请求。 此程序使用 IBackgroundCopyJob2::SetCredentials 方法设置 BITS 传输作业的凭据。
- 添加文件、修改属性或恢复 BITS 传输作业。
- BITS 传输作业完成后,通过调用 IBackgroundCopyJob::Complete 从队列中删除该作业。
- 最后,通过调用 SspiFreeAuthIdentity 函数释放身份验证标识结构。
下面的代码示例演示如何将 SSPI 身份验证编码与 BITS 传输作业配合使用。
#define SECURITY_WIN32
#define _SEC_WINNT_AUTH_TYPES
#include <windows.h>
#include <ntsecapi.h>
#include <bits.h>
#include <sspi.h>
#include <wincred.h>
#include <iostream>
#include <atlbase.h>
#include "CommonCode.h"
void PromptForCredentials(PWSTR pwTargetName)
{
HRESULT hr;
// If CoInitializeEx fails, the exception is unhandled and the program terminates
CCoInitializer coInitializer(COINIT_APARTMENTTHREADED);
CComPtr<IBackgroundCopyManager> pQueueMgr;
CComPtr<IBackgroundCopyJob> pJob;
CComPtr<IBackgroundCopyJob2> pJob2;
PSEC_WINNT_AUTH_IDENTITY_OPAQUE pAuthIdentityEx2 = NULL;
DWORD dwFlags = 0;
BOOL fSave = FALSE;
BOOL bReturn = TRUE;
CREDUI_INFO creduiInfo = { 0 };
creduiInfo.cbSize = sizeof(creduiInfo);
// Change the message text and caption to the actual text for your dialog.
creduiInfo.pszMessageText = pwTargetName;
creduiInfo.pszCaptionText = L"SSPIPFC title for the dialog box";
try {
// Prompt for credentials from user using Negotiate security package.
DWORD dwRet = SspiPromptForCredentials(
pwTargetName,
&creduiInfo,
0,
L"Negotiate",
NULL,
&pAuthIdentityEx2,
&fSave,
dwFlags
);
if (SEC_E_OK != dwRet)
{
// Prompt for credentials failed.
throw MyException(dwRet, L"SspiPromptForCredentials");
}
if (NULL != pAuthIdentityEx2)
{
GUID guidJob;
BG_AUTH_CREDENTIALS authCreds;
PCWSTR pwUserName = NULL;
PCWSTR pwDomainName = NULL;
PCWSTR pwPassword = NULL;
// Encode credential structure as strings that can
// be passed to a BITS job.
SECURITY_STATUS secStatus = SspiEncodeAuthIdentityAsStrings(
pAuthIdentityEx2,
&pwUserName,
&pwDomainName,
&pwPassword
);
if(SEC_E_OK != secStatus)
{
// Encode authentication identity as strings.
throw MyException(secStatus, L"SspiEncodeAuthIdentityAsStrings");
}
// Show the encoded user name and domain name.
wprintf(
L"User Name: %s\nDomain Name: %s",
pwUserName,
pwDomainName
);
//The impersonation level must be at least RPC_C_IMP_LEVEL_IMPERSONATE.
HRESULT hr = CoInitializeSecurity(
NULL,
-1,
NULL,
NULL,
RPC_C_AUTHN_LEVEL_CONNECT,
RPC_C_IMP_LEVEL_IMPERSONATE,
NULL,
EOAC_DYNAMIC_CLOAKING,
0
);
if (FAILED(hr))
{
throw MyException(hr, L"CoInitializeSecurity");
}
// Connect to BITS.
hr = CoCreateInstance(__uuidof(BackgroundCopyManager), NULL,
CLSCTX_LOCAL_SERVER,
__uuidof(IBackgroundCopyManager),
(void**) &pQueueMgr);
if (FAILED(hr))
{
// Failed to connect.
throw MyException(hr, L"CoCreateInstance");
}
// Create a job.
hr = pQueueMgr->CreateJob(
L"EncodeSample",
BG_JOB_TYPE_DOWNLOAD,
&guidJob,
&pJob
);
if(FAILED(hr))
{
// Failed to create a BITS job.
throw MyException(hr, L"CreateJob");
}
// Get IBackgroundCopyJob2 interface.
hr = pJob->QueryInterface(__uuidof(IBackgroundCopyJob2), (void**)&pJob2);
if (FAILED(hr))
{
// Failed to get a reference to the IBackgroundCopyJob2 interface.
throw MyException(hr, L"QueryInterface(IBackgroundCopyJob2)");
}
// Create a BITS authentication structure from the encoded strings.
authCreds.Target = BG_AUTH_TARGET_SERVER;
authCreds.Scheme = BG_AUTH_SCHEME_NEGOTIATE;
authCreds.Credentials.Basic.UserName = (LPWSTR)pwUserName;
authCreds.Credentials.Basic.Password = (LPWSTR)pwPassword;
// Set the credentials for the job.
hr = pJob2->SetCredentials(&authCreds);
if (FAILED(hr))
{
// Failed to set credentials.
throw MyException(hr, L"SetCredentials");
}
// Modify the job's property values.
// Add files to the job.
// Activate (resume) the job in the transfer queue.
// Remove the job from the transfer queue.
hr = pJob->Complete();
if (FAILED(hr))
{
// Failed to complete the job.
throw MyException(hr, L"Complete");
}
}
}
catch(std::bad_alloc &)
{
wprintf(L"Memory allocation failed");
if (pJob != NULL)
{
pJob->Cancel();
}
}
catch(MyException &ex)
{
wprintf(L"Error %x occurred during operation", ex.Error);
if (pJob != NULL)
{
pJob->Cancel();
}
}
// Free the auth identity structure.
if (NULL != pAuthIdentityEx2)
{
SspiFreeAuthIdentity(pAuthIdentityEx2);
pAuthIdentityEx2 = NULL;
}
return;
}
void _cdecl _tmain(int argc, LPWSTR* argv)
{
PromptForCredentials(L"Target");
}
相关主题