source: nscp/include/com_helpers.hpp @ d05c3f0

#pragma once

#define _WIN32_DCOM
#define _WIN32_WINNT 0x0400

#include <error.hpp>
#include <windows.h>
#include <objbase.h>
#include <atlbase.h>
#include <atlsafe.h>

namespace com_helper {

        class com_exception {
                std::wstring error_;
                HRESULT result_;
        public:
                com_exception(std::wstring error) : error_(error) {}
                com_exception(std::wstring error, HRESULT result) : error_(error), result_(result) {
                        error_ += error::format::from_system(result);
                }
                std::wstring getMessage() {
                        return error_;
                }

        };

        class initialize_com {
                bool isInitialized_;
        public:
                initialize_com() : isInitialized_(false) {}
                ~initialize_com() {
                        unInitialize();
                }

                void initialize() {
                        HRESULT hRes = CoInitializeEx(NULL, COINIT_MULTITHREADED);
                        if (FAILED(hRes))
                                throw com_exception(_T("CoInitialize failed: "), hRes);
                        isInitialized_ = true;
                        hRes = CoInitializeSecurity(NULL,-1,NULL,NULL,RPC_C_AUTHN_LEVEL_PKT,RPC_C_IMP_LEVEL_IMPERSONATE,NULL,EOAC_NONE,NULL);
                        if (FAILED(hRes)) 
                                throw com_exception(_T("CoInitializeSecurity failed: "), hRes);
                }
                void unInitialize() {
                        if (!isInitialized_)
                                return;
                        CoUninitialize();
                        isInitialized_ = false;
                }
                bool isInitialized() const {
                        return isInitialized_;
                }

        };

};