Return to WMICollector.h CVS log

Up to [Pegasus] / pegasus / src / WMIMapper / WMIProvider

BUG#: 6697
TITLE: WMI Mapper code needs chksrc adjusments.

DESCRIPTION: Reformatted the files according chksrc requirements.

//%2006////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2000, 2001, 2002 BMC Software; Hewlett-Packard Development
// Company, L.P.; IBM Corp.; The Open Group; Tivoli Systems.
// Copyright (c) 2003 BMC Software; Hewlett-Packard Development Company, L.P.;
// IBM Corp.; EMC Corporation, The Open Group.
// Copyright (c) 2004 BMC Software; Hewlett-Packard Development Company, L.P.;
// IBM Corp.; EMC Corporation; VERITAS Software Corporation; The Open Group.
// Copyright (c) 2005 Hewlett-Packard Development Company, L.P.; IBM Corp.;
// EMC Corporation; VERITAS Software Corporation; The Open Group.
// Copyright (c) 2006 Hewlett-Packard Development Company, L.P.; IBM Corp.;
// EMC Corporation; Symantec Corporation; The Open Group.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// 
// THE ABOVE COPYRIGHT NOTICE AND THIS PERMISSION NOTICE SHALL BE INCLUDED IN
// ALL COPIES OR SUBSTANTIAL PORTIONS OF THE SOFTWARE. THE SOFTWARE IS PROVIDED
// "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
// LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
// PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
// ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
//=============================================================================
//
// Author: Barbara Packard (barbara_packard@hp.com)
//
// Modified By:    Adriano Zanuz (adriano.zanuz@hp.com)
//              Terry Martin, Hewlett-Packard Company (terry.martin@hp.com)
//
//%////////////////////////////////////////////////////////////////////////////


#ifndef Pegasus_WMICollector_h
#define Pegasus_WMICollector_h

#include <Pegasus/Common/Array.h>
#include <Pegasus/Common/String.h>
#include <Pegasus/Common/CIMPropertyList.h>
#include <Pegasus/Common/CIMObjectPath.h>
#include <Pegasus/Common/CIMInstance.h>
#include <Pegasus/Common/CIMClass.h>

#include <Pegasus/Common/CIMValue.h>
#include <Pegasus/Common/CIMDateTime.h>
#include <Pegasus/Common/CIMProperty.h>
#include <Pegasus/Common/CIMPropertyList.h>
#include <Pegasus/Common/CIMPropertyList.h>
#include <Pegasus/Common/CIMParamValue.h>

//#include <Ntsecapi.h>

PEGASUS_NAMESPACE_BEGIN

class WMICollector 
{
public:
    WMICollector(bool bLocal = FALSE);
    virtual ~WMICollector(void);

    virtual void terminate(void);

    bool getInstanceEnum(
        IEnumWbemClassObject **ppInstances,
        const String & sClassName,
        Boolean deepInheritance);

    bool getClassEnum(
        IEnumWbemClassObject **ppClasses,
        const String & sSuperClass,
        Boolean deepInheritance);

    bool getQueryResult(
        IEnumWbemClassObject **ppInstances, 
        const String & query, 
        const String & queryLanguage);

    void setNamespace(const char * sNamespace)
    {
        m_bsNamespace = (LPCTSTR)sNamespace;
    }

    void setNamespace(const String & sNamespace);
    void setUserName(const String & sUserName);
    void setPassword(const String & sPassword);
    bool Connect(IWbemServices **ppServices);
    bool setup();

    bool getObject(IWbemClassObject **ppObject, const String & sObjectName);

    bool getCIMInstance(IWbemClassObject *pObject, 
        CIMInstance & cimInst,
        Boolean localOnly,
        Boolean includeQualifiers,
        Boolean includeClassOrigin,
        const CIMPropertyList& propertyList = CIMPropertyList(),
        Boolean getKeyProperties = FALSE);    

    bool getCIMObject(IWbemClassObject *pObject, 
        CIMObject & cimObj,
        Boolean localOnly,
        Boolean includeQualifiers,
        Boolean includeClassOrigin,
        const CIMPropertyList& propertyList = CIMPropertyList(),
        Boolean getKeyProperties = FALSE);    


    bool getCIMClass(IWbemClassObject *pObject, 
        CIMClass & cimClass,
        Boolean localOnly,
        Boolean includeQualifiers,
        Boolean includeClassOrigin,
        const CIMPropertyList& propertyList = CIMPropertyList());    

    static String getClassName(IWbemClassObject *pObject);
    String getSuperClass(IWbemClassObject *pClass);
    String getRelativePath(IWbemClassObject *pObject);

    bool isInstance(IWbemClassObject *pObject);

    bool getObjectProperties(IWbemClassObject * pObject, 
        CIMObject & cimObj,
        Boolean localOnly,
        Boolean includeQualifiers,
        Boolean includeClassOrigin,
        const CIMPropertyList& propertyList,
        Boolean bGetKeyProperties = FALSE);
    
    static CIMProperty getProperty(IWbemClassObject *pClass, 
        const CComBSTR & bsName, 
        const CComVariant & vValue, // this will be NULL for class objects
        CIMTYPE type, 
        Boolean includeClassOrigin, 
        Boolean includeQualifiers,
        Boolean bPropagated);

    bool getClassMethods(IWbemClassObject *pObject, 
        CIMClass & cimClass,
        Boolean localOnly,
        Boolean includeQualifiers,
        Boolean includeClassOrigin);

    static CIMMethod getMethod(IWbemClassObject *pClass, 
        const CComBSTR &bsName, 
        const CComPtr<IWbemClassObject> &inParameters,
        const CComPtr<IWbemClassObject> &outParameters,
        Boolean includeClassOrigin, 
        Boolean includeQualifiers,
        Boolean bPropagated);

    bool setProxySecurity(IUnknown * pProxy);

    bool isLocalNamespace()
        {return m_bIsLocalNamespace;}

    bool isLocalConnection() 
        { return m_bLocalConnection; }

private:
    static String getStringProperty(
        IWbemClassObject *pObject, 
        const CComBSTR & bsPropertyName);

    bool isArrayType(VARTYPE vt)
        {return (vt & VT_ARRAY) ? true : false;}

    bool isReferenceType(VARTYPE vt)
        {return (vt & VT_BYREF) ? true : false;}

    void logonUser();

    void revertToSelf();

    /*
    LSA_HANDLE GetPolicyHandle();
    bool GetAccountSid(LPTSTR, LPTSTR, PSID*);
    void AddPrivileges(PSID, LSA_HANDLE);
    void InitLsaString(PLSA_UNICODE_STRING, LPWSTR);
    */

    bool    m_bInitialized;
    bool    m_bIsLocalNamespace;
    bool    m_bImpersonate;
    bool    m_bLocalConnection;

    // WMI interfaces
    CComBSTR                m_bsNamespace;
    CComBSTR                m_bsUserName;
    CComBSTR                m_bsPassword;
    CComBSTR                m_bsDomain;
};

PEGASUS_NAMESPACE_END

#endif