指定 BITS 傳輸作業的伺服器驗證認證

發行項
02/14/2024

您可以為背景智慧型傳送服務（BITS）傳輸作業指定不同的驗證配置。

下列程式會取得驗證配置，並建立驗證身分識別結構。然後應用程式會建立 BITS 下載作業，並設定作業的認證。建立作業之後，應用程式會取得回呼介面的指標，並輪詢 BITS 傳輸作業的狀態。作業完成之後，它會從佇列中移除。

此範例使用範例：一般類別中定義的標頭和實作。

指定 BITS 傳輸作業的伺服器驗證認證

建立容器結構，將BG_AUTH_SCHEME值對應至配置名稱。

struct
{
    LPCWSTR        Name;
    BG_AUTH_SCHEME Scheme;
}
SchemeNames[] =
{
    { L"basic",      BG_AUTH_SCHEME_BASIC },
    { L"digest",     BG_AUTH_SCHEME_DIGEST },
    { L"ntlm",       BG_AUTH_SCHEME_NTLM },
    { L"negotiate",  BG_AUTH_SCHEME_NEGOTIATE },
    { L"passport",   BG_AUTH_SCHEME_PASSPORT },

    { NULL,         BG_AUTH_SCHEME_BASIC }
};

呼叫 CCoInitializer 函式來初始化 COM 參數。如需 CCoInitializer 函式的詳細資訊，請參閱範例：一般類別。
準備BG_AUTH_CREDENTIALS結構。此範例會使用 SecureZeroMemory 函式來清除與敏感性資訊相關聯的記憶體位置。 SecureZeroMemory 函式定義於 WinBase.h 中。

使用 GetScheme 函式取得用來連線到伺服器的驗證配置。

bool GetScheme( LPCWSTR s, BG_AUTH_SCHEME *scheme )
{
    int i;

    i = 0;
    while (SchemeNames[i].Name != NULL)
    {
        if (0 == _wcsicmp( s, SchemeNames[i].Name ))
        {

            *scheme = SchemeNames[i].Scheme;
            return true;
        }

        ++i;
    }

    return false;
}

使用 GetScheme 函式所傳回的驗證配置，以及傳入 ServerAuthentication 函式的使用者名稱和密碼，填入BG_AUTH_CREDENTIALS結構。
取得IBackgroundCopyJob 介面（pJob）的指標。
呼叫 CoInitializeSecurity 來初始化 COM 程序安全性。 BITS 至少需要仿真的 IMPERSONATE 層級。如果未設定正確的模擬層級，BITS 會失敗並E_ACCESSDENIED。
取得IBackgroundCopyManager 介面的指標，並藉由呼叫CoCreateInstance函式來取得BITS的初始定位器。
呼叫IBackgroundCopyManager：：CreateJob方法，以建立BITS傳輸作業。
取得 CNotifyInterface 回呼介面的指標，並呼叫 IBackgroundCopyJob：：SetNotifyInterface 方法來接收作業相關事件的通知。如需 CNotifyInterface 的詳細資訊，請參閱範例：一般類別。
呼叫IBackgroundCopyJob：：SetNotifyFlags 方法來設定要接收的通知類型。在此範例中，會 設定BG_NOTIFY_JOB_TRANSFERRED 和 BG_NOTIFY_JOB_ERROR 旗標。
取得IBackgroundCopyJob2 介面的指標。 使用IBackgroundCopyJob2指標來設定作業的其他屬性。此程式會使用 IBackgroundCopyJob2：：SetCredentials 方法來設定 BITS 傳輸作業的認證。
呼叫 IBackgroundCopyJob：：AddFile，將檔案新增至 BITS 傳輸作業。在此範例中 ，IBackgroundCopyJob：：AddFile 方法會使用傳遞至 ServerAuthentication 函式的 remoteFile 和 localFile 變數。
新增檔案之後，請呼叫 IBackgroundCopyJob：：Resume 以繼續作業。

設定 while 迴圈，以在作業傳輸時從回呼介面等候結束訊息。

注意

只有在 COM Apartment 是單個線程 Apartment 時，才需要此步驟。如需詳細資訊，請參閱單個線程 Apartment。

// Wait for QuitMessage from CallBack
    DWORD dwLimit = GetTickCount() + (15 * 60 * 1000);  // set 15 minute limit
    while (dwLimit > GetTickCount())
    {
        MSG msg;

        while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) 
        { 
            // If it is a quit message, exit.
            if (msg.message == WM_QUIT) 
            {
                return;
            }

            // Otherwise, dispatch the message.
            DispatchMessage(&msg); 
        } // End of PeekMessage while loop
    }

上述迴圈會使用 GetTickCount 函式來擷取自作業開始傳輸之後經過的毫秒數。

BITS 傳輸作業完成之後，請呼叫 IBackgroundCopyJob：：Complete，從佇列中移除作業。

下列程式代碼範例會指定 BITS 傳輸作業的伺服器驗證認證。

#include <bits.h>
#include <bits4_0.h>
#include <stdio.h>
#include <tchar.h>
#include <lm.h>
#include <iostream>
#include <exception>
#include <string>
#include <atlbase.h>
#include <memory>
#include <new>
#include "CommonCode.h"

//
// Retrieve BG_AUTH_SCHEME from scheme name.
//
//
struct
{
    LPCWSTR        Name;
    BG_AUTH_SCHEME Scheme;
}
SchemeNames[] =
{
    { L"basic",      BG_AUTH_SCHEME_BASIC },
    { L"digest",     BG_AUTH_SCHEME_DIGEST },
    { L"ntlm",       BG_AUTH_SCHEME_NTLM },
    { L"negotiate",  BG_AUTH_SCHEME_NEGOTIATE },
    { L"passport",   BG_AUTH_SCHEME_PASSPORT },

    { NULL,         BG_AUTH_SCHEME_BASIC }
};

bool GetScheme( LPCWSTR s, BG_AUTH_SCHEME *scheme )
{
    int i;

    i = 0;
    while (SchemeNames[i].Name != NULL)
    {
        if (0 == _wcsicmp( s, SchemeNames[i].Name ))
        {

            *scheme = SchemeNames[i].Scheme;
            return true;
        }

        ++i;
    }

    return false;
}

void ServerAuthentication(const LPWSTR &remoteFile, const LPWSTR &localFile, const LPWSTR &scheme, const LPWSTR &username, const LPWSTR &password)
{

 // If CoInitializeEx fails, the exception is unhandled and the program terminates  
 CCoInitializer coInitializer(COINIT_APARTMENTTHREADED);
    
    // Prepare the credentials structure.
    BG_AUTH_CREDENTIALS cred;
    ZeroMemory(&cred, sizeof(cred));
    if (!GetScheme(scheme,&cred.Scheme))
    {
        wprintf(L"Invalid authentication scheme specified\n");
        return;
    }

    cred.Target = BG_AUTH_TARGET_SERVER;
    cred.Credentials.Basic.UserName = username;
    if (0 == _wcsicmp(cred.Credentials.Basic.UserName, L"NULL"))
    {
        cred.Credentials.Basic.UserName = NULL;
    }

    cred.Credentials.Basic.Password = password;
    if (0 == _wcsicmp(cred.Credentials.Basic.Password, L"NULL"))
    {
        cred.Credentials.Basic.Password = NULL;
    }

    CComPtr<IBackgroundCopyJob> pJob;
    try
    {
        //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.
        CComPtr<IBackgroundCopyManager> pQueueMgr;
        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.
        wprintf(L"Creating Job...\n");

        GUID guidJob;
        hr = pQueueMgr->CreateJob(L"BitsAuthSample",
            BG_JOB_TYPE_DOWNLOAD,
            &guidJob,
            &pJob);

        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"CreateJob");
        }

        // Set the File Completed call.
        CComPtr<CNotifyInterface> pNotify;
        pNotify = new CNotifyInterface();
        hr = pJob->SetNotifyInterface(pNotify);
        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"SetNotifyInterface");
        }
        hr = pJob->SetNotifyFlags(BG_NOTIFY_JOB_TRANSFERRED | 
            BG_NOTIFY_JOB_ERROR);

        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"SetNotifyFlags");
        }

        // Set credentials.
        CComPtr<IBackgroundCopyJob2> job2;
        hr = pJob.QueryInterface(&job2);
        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"QueryInterface");
        }

        hr = job2->SetCredentials(&cred);
        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"SetCredentials");
        }

        // Add a file.
        wprintf(L"Adding File to Job\n");
        hr = pJob->AddFile(remoteFile, localFile);
        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"AddFile");
        }

        //Resume the job.
        wprintf(L"Resuming Job...\n");
        hr = pJob->Resume();
        if (FAILED(hr))
        {
            // Failed to connect.
            throw MyException(hr, L"Resume");
        }
    }
    catch(const std::bad_alloc &)
    {
        wprintf(L"Memory allocation failed");
        if (pJob)
        {
            pJob->Cancel();
        }

        return;
    }
    catch(const MyException &ex)
    {
        wprintf(L"Error %x occurred during operation", ex.Error);
        if (pJob)
        {
            pJob->Cancel();
        }

        return;
    }

    wprintf(L"Transferring file and waiting for callback.\n");

    // Wait for QuitMessage from CallBack.
    DWORD dwLimit = GetTickCount() + (15 * 60 * 1000);  // set 15 minute limit
    while (dwLimit > GetTickCount())
    {
        MSG msg;

        while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) 
        { 
            // If it is a quit message, exit.
            if (msg.message == WM_QUIT) 
            {
                return;
            }

            // Otherwise, dispatch the message.
            DispatchMessage(&msg); 
        } // End of PeekMessage while loop
    }

    pJob->Cancel();
    return;
}

void _cdecl _tmain(int argc, LPWSTR* argv)
{   
    if (argc != 6)
    {
        wprintf(L"Usage:");
        wprintf(L"%s", argv[0]);
        wprintf(L" [remote name] [local name] [server authentication scheme =\"NTLM\"|\"NEGOTIATE\"|\"BASIC\"|\"DIGEST\"] [username] [password]\n");
        return;
    }

    ServerAuthentication(argv[1], argv[2], argv[3], argv[4], argv[5]); 

}

IBackgroundCopyManager
IBackgroundCopyJob
IBackgroundCopyJob2
範例：一般類別

共用方式為

指定 BITS 傳輸作業的伺服器驗證認證

意見反應

其他資源

共用方式為

指定 BITS 傳輸作業的伺服器驗證認證

相關主題

意見反應

其他資源