111 lines
2.7 KiB
C++
111 lines
2.7 KiB
C++
|
|
||
|
|
#include "MiniUsoClient.h"
|
||
|
|
|
||
|
|
#pragma comment(lib, "rpcrt4.lib")
|
||
|
|
|
||
|
|
MiniUsoClient::MiniUsoClient()
|
||
|
|
{
|
||
|
|
HRESULT hResult;
|
||
|
|
|
||
|
|
hResult = CoInitializeEx(0, COINIT_MULTITHREADED);
|
||
|
|
if (FAILED(hResult))
|
||
|
|
{
|
||
|
|
//wprintf_s(L" |__ CoInitializeEx() failed. Error code = 0x%08X\n", hResult);
|
||
|
|
_ready = false;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
_ready = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
MiniUsoClient::~MiniUsoClient()
|
||
|
|
{
|
||
|
|
CoUninitialize();
|
||
|
|
}
|
||
|
|
|
||
|
|
void MiniUsoClient::ThrowOnError(HRESULT hResult)
|
||
|
|
{
|
||
|
|
if (hResult != 0)
|
||
|
|
{
|
||
|
|
throw _com_error(hResult);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
bool MiniUsoClient::Run(UsoAction action)
|
||
|
|
{
|
||
|
|
HRESULT hResult;
|
||
|
|
|
||
|
|
if (this->_ready)
|
||
|
|
{
|
||
|
|
//wprintf_s(L" |__ Creating instance of 'UpdateSessionOrchestrator'... ");
|
||
|
|
|
||
|
|
GUID CLSID_UpdateSessionOrchestrator = { 0xb91d5831, 0xb1bd, 0x4608, { 0x81, 0x98, 0xd7, 0x2e, 0x15, 0x50, 0x20, 0xf7 } };
|
||
|
|
IUpdateSessionOrchestratorPtr updateSessionOrchestrator;
|
||
|
|
hResult = CoCreateInstance(CLSID_UpdateSessionOrchestrator, nullptr, CLSCTX_LOCAL_SERVER, IID_PPV_ARGS(&updateSessionOrchestrator));
|
||
|
|
if (FAILED(hResult))
|
||
|
|
{
|
||
|
|
//wprintf_s(L"\n |__ CoCreateInstance() failed. Error code = 0x%08X\n", hResult);
|
||
|
|
CoUninitialize();
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
//wprintf_s(L"Done.\n");
|
||
|
|
|
||
|
|
/*
|
||
|
|
try
|
||
|
|
{
|
||
|
|
ThrowOnError(updateSessionOrchestrator->LogTaskRunning(L"StartScan"));
|
||
|
|
}
|
||
|
|
catch (const _com_error& error)
|
||
|
|
{
|
||
|
|
//wprintf(L" |__ LogTaskRunning() - Return code: 0x%08X (\"%s\")\n", error.Error(), error.ErrorMessage());
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
IUsoSessionCommonPtr usoSessionCommon;
|
||
|
|
GUID IID_IUsoSessionCommon = { 0xfccc288d, 0xb47e, 0x41fa, { 0x97, 0x0c, 0x93, 0x5e, 0xc9, 0x52, 0xf4, 0xa4 } };
|
||
|
|
try
|
||
|
|
{
|
||
|
|
//wprintf_s(L" |__ Creating a new Update Session... ");
|
||
|
|
updateSessionOrchestrator->CreateUpdateSession(1, &IID_IUsoSessionCommon, &usoSessionCommon);
|
||
|
|
//wprintf_s(L"Done.\n");
|
||
|
|
|
||
|
|
//wprintf_s(L" |__ Calling 'CoSetProxyBlanket()'... ");
|
||
|
|
ThrowOnError(CoSetProxyBlanket(usoSessionCommon, RPC_C_AUTHN_DEFAULT, RPC_C_AUTHZ_DEFAULT, COLE_DEFAULT_PRINCIPAL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, nullptr, NULL));
|
||
|
|
//wprintf_s(L"Done.\n");
|
||
|
|
|
||
|
|
switch (action)
|
||
|
|
{
|
||
|
|
case USO_STARTSCAN:
|
||
|
|
//wprintf(L" |__ Calling 'StartScan'... ");
|
||
|
|
ThrowOnError(usoSessionCommon->Proc21(0, 0, L"ScanTriggerUsoClient"));
|
||
|
|
//wprintf(L"Done.\n");
|
||
|
|
break;
|
||
|
|
case USO_STARTDOWNLOAD:
|
||
|
|
//wprintf(L" |__ Calling 'StartDownload'... ");
|
||
|
|
ThrowOnError(usoSessionCommon->Proc22(0));
|
||
|
|
//wprintf(L"Done.\n");
|
||
|
|
break;
|
||
|
|
case USO_STARTINTERACTIVESCAN:
|
||
|
|
//wprintf(L" |__ Calling 'StartInteractiveScan'... ");
|
||
|
|
ThrowOnError(usoSessionCommon->Proc21(-1, 0, L"ScanTriggerUsoClientInteractive"));
|
||
|
|
//wprintf(L"Done.\n");
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
catch (const _com_error&)
|
||
|
|
{
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|