前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >WMI技术介绍和应用——执行方法

WMI技术介绍和应用——执行方法

作者头像
方亮
发布2019-01-16 15:01:20
6740
发布2019-01-16 15:01:20
举报
文章被收录于专栏:方亮

        在之前的博文中,我们主要介绍了如何使用WMI查询信息和接收事件。本文将介绍WMI的另一种用法——执行方法。(转载请指明出于breaksoftware的csdn博客)

        这块的内容在msdn中有详细的介绍,如果想看原版的可以参阅《Example: Calling a Provider Method》

        本文将基于《WMI技术介绍和应用——VC开发WMI应用的基本步骤》中介绍的基类CWMI,在继承类中重写Excute函数,实现执行方法的功能。

        之所以继承于CWMI,是为了将WMI逻辑中相同部分提炼出来。不同的使用方式只要完成其核心功能即可,回顾下CWMI类的执行主体

代码语言:javascript
复制
HRESULT CWMI::ExcuteFun()
{
    HRESULT hr = E_FAIL;
    CComPtr<IWbemLocator> pLoc = NULL;
    CComPtr<IWbemServices> pSvc = NULL;

    do {
        hr = InitialCom();
        CHECKHR(hr);

        hr = SetComSecLevels();
        CHECKHR(hr);

        hr = ObtainLocator2WMI(pLoc);
        CHECKHR(hr);

        hr = Connect2WMI(pLoc, pSvc);
        CHECKHR(hr);

        hr = SetProxySecLevels(pSvc);
        CHECKHR(hr);

        hr = Excute(pSvc);
        CHECKHR(hr);

    } while (0);
    return hr;
}

        除Excute方法之外的其他方法,我们都是公用的。那我们看下执行方法的类是如何实现的。

        首先我们定义一个map,用于保存执行函数的参数。因为不同函数的参数名不同,类型不同,所以我们要做如下定义

代码语言:javascript
复制
typedef std::map<std::wstring, CComVariant> ParamsMap;

class CExcuteMethod : public CWMI
{
public:
    CExcuteMethod(const wstring& wszNamespace, const std::wstring& wstrClass,
        const std::wstring& wstrInstanceName, const std::wstring& wstrMethod, const std::wstring& wstrRet, const ParamsMap& params);
    ~CExcuteMethod(void);
private:
    HRESULT Excute(CComPtr<IWbemServices> pSvc);
private:
    std::wstring m_wstrInstanceName;
    std::wstring m_wstrClassName;
    std::wstring m_wstrMethod;
    std::wstring m_wstrRet;
    ParamsMap m_params;
};

        CComVariant类型保证我们每个参数可以由使用者自己定义。

        在构造函数中,我们需要传入WMI类名(非C++类名),调用方法名,返回值名,参数map。

        在执行的主体函数Excute中,我们首先使用WMI类名获取类

代码语言:javascript
复制
HRESULT CExcuteMethod::Excute( CComPtr<IWbemServices> pSvc )
{
    HRESULT hr = WBEM_S_FALSE;

    do {
        CComBSTR bstrClassName = m_wstrClassName.c_str();
        CComPtr<IWbemClassObject> spClassObject = NULL;
        hr = pSvc->GetObject(bstrClassName, 0, NULL, &spClassObject, NULL);
        if (FAILED(hr) || !spClassObject) {
            break;
        }

        然后通过方法名,获取这个类中函数的入参定义

代码语言:javascript
复制
        CComBSTR bstrMethodName = m_wstrMethod.c_str();
        CComPtr<IWbemClassObject> spInParamsDefinition = NULL;
        hr = spClassObject->GetMethod(bstrMethodName, 0, &spInParamsDefinition, NULL);
        if (FAILED(hr) || !spInParamsDefinition) {
            break;
        }

        实例化入参定义的一个实例,然后对这个入参实例进行参数赋值。注意一下,这个地方的赋值,非常符合脚本语言的问题,即通过名称进行赋值,而不是按照C语言风格的入参顺序进行赋值

代码语言:javascript
复制
        CComPtr<IWbemClassObject> spParamsInstance = NULL;
        hr = spInParamsDefinition->SpawnInstance(0, &spParamsInstance);
        if (FAILED(hr) || !spParamsInstance) {
            break;
        }

        for (ParamsMap::iterator it = m_params.begin(); it != m_params.end(); it++) {
            if (!it->first.empty()) {
                CComVariant value = it->second;
                hr = spParamsInstance->Put(it->first.c_str(), 0, &value, 0);
            }
        }

        最后,我们通过类名、方法名、参数实例去执行方法名获取返回的数据实例

代码语言:javascript
复制
        CComPtr<IWbemClassObject> spOutParams = NULL;
        if (!m_wstrInstanceName.empty()) {
            //CComBSTR(L"ClassInstance.name='InstanceName'")
            CComBSTR bstrInstanceName = m_wstrInstanceName.c_str();
            hr = pSvc->ExecMethod(bstrInstanceName, bstrMethodName, 0, NULL, spParamsInstance, &spOutParams, NULL);
        }
        else if (!m_wstrClassName.empty()) {
            hr = pSvc->ExecMethod(bstrClassName, bstrMethodName, 0, NULL, spParamsInstance, &spOutParams, NULL);
        }

        if (SUCCEEDED(hr) && !m_wstrRet.empty()) {
            CComVariant varRet;
            hr = spOutParams->Get(CComBSTR(m_wstrRet.c_str()), 0, &varRet, NULL, 0);
        }

        如此便可以执行一个方法调用。这儿有个地方需要注意下,就是调用ExecMethod方式存在两种方式:

  1. 类的静态方法直接使用类名调用
  2. 类的非静态方法使用对象名调用,这种调用我们将在之后的讲解WMI Provider时介绍。

        我们在main函数中进行如下测试

代码语言:javascript
复制
    CExcuteMethod* excute_method = NULL;
    {
        ParamsMap params;
        {
            CComVariant vt;
            vt.vt = VT_BSTR;
            vt.bstrVal = CComBSTR("notepad.exe");
            params[L"CommandLine"] = vt;
        }

        excute_method = new CExcuteMethod(L"root\\CIMV2", L"", L"Win32_Process", L"Create", L"ReturnValue", params);
        excute_method->ExcuteFun();
    }
    delete excute_method;

        这段代码将启动一个记事本程序,我们没有指定对象名,说明Ceate方法是Win32_Process的静态方法。这段代码需要注意下,就是我为什么要使用动态创建指针的方式,而不是直接定义一个对象?还有就是为什么要种{}控制除指针申明和销毁之外的逻辑?这其中主要的原因是我们CWMI类中控制了COM组件的初始化和卸载操作。如果直接使用对象,则对象的消亡和Main函数中使用的CComVariant类型数据的消亡顺序将不可控制,会导致崩溃(实际的确是CComVariant后释放从而出现异常)。而我们使用动态创建对象和使用{}控制CComVariant的生命周期的方式,实现了对象消亡顺序的可控。当然这不是一种好的设计,于是我们似乎应该将WMI的初始化和卸载交由调用者控制。

        工程源码见《WMI技术介绍和应用——WMI概述》结尾。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2016年01月21日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档