Return to wmiquery.h CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / smartmontools / os_win32

Annotation of embedaddon/smartmontools/os_win32/wmiquery.h, revision 1.1

1.1     ! misho       1: /*
        !             2:  * os_win32/wmiquery.h
        !             3:  *
        !             4:  * Home page of code is: http://smartmontools.sourceforge.net
        !             5:  *
        !             6:  * Copyright (C) 2011 Christian Franke <smartmontools-support@lists.sourceforge.net>
        !             7:  *
        !             8:  * This program is free software; you can redistribute it and/or modify
        !             9:  * it under the terms of the GNU General Public License as published by
        !            10:  * the Free Software Foundation; either version 2, or (at your option)
        !            11:  * any later version.
        !            12:  *
        !            13:  * You should have received a copy of the GNU General Public License
        !            14:  * (for example COPYING); If not, see <http://www.gnu.org/licenses/>.
        !            15:  *
        !            16:  */
        !            17: 
        !            18: #ifndef WMIQUERY_H
        !            19: #define WMIQUERY_H
        !            20: 
        !            21: #define WMIQUERY_H_CVSID "$Id: wmiquery.h 3243 2011-01-19 20:03:47Z chrfranke $"
        !            22: 
        !            23: #ifdef HAVE_WBEMCLI_H
        !            24: #include <wbemcli.h>
        !            25: #else
        !            26: #include "wbemcli_small.h"
        !            27: #endif
        !            28: 
        !            29: #include <string>
        !            30: 
        !            31: #if !defined(__GNUC__) && !defined(__attribute__)
        !            32: #define __attribute__(x)  /**/
        !            33: #endif
        !            34: 
        !            35: /////////////////////////////////////////////////////////////////////////////
        !            36: // com_bstr
        !            37: 
        !            38: /// Wrapper class for COM BSTR
        !            39: class com_bstr
        !            40: {
        !            41: public:
        !            42:   /// Construct from string.
        !            43:   com_bstr(const char * str);
        !            44: 
        !            45:   /// Destructor frees BSTR.
        !            46:   ~com_bstr()
        !            47:     { SysFreeString(m_bstr); }
        !            48: 
        !            49:   /// Implicit conversion to BSTR.
        !            50:   operator BSTR()
        !            51:     { return m_bstr; }
        !            52: 
        !            53:   /// Convert BSTR back to std::string.
        !            54:   static bool to_str(const BSTR & bstr, std::string & str);
        !            55: 
        !            56: private:
        !            57:   BSTR m_bstr;
        !            58: 
        !            59:   com_bstr(const com_bstr &);
        !            60:   void operator=(const com_bstr &);
        !            61: };
        !            62: 
        !            63: 
        !            64: /////////////////////////////////////////////////////////////////////////////
        !            65: // com_intf_ptr
        !            66: 
        !            67: /// Wrapper class for COM Interface pointer
        !            68: template <class T>
        !            69: class com_intf_ptr
        !            70: {
        !            71: public:
        !            72:   /// Construct empty object
        !            73:   com_intf_ptr()
        !            74:     : m_ptr(0) { }
        !            75: 
        !            76:   /// Destructor releases the interface.
        !            77:   ~com_intf_ptr()
        !            78:     { reset(); }
        !            79: 
        !            80:   /// Release interface and clear the pointer.
        !            81:   void reset()
        !            82:     {
        !            83:       if (m_ptr) {
        !            84:         m_ptr->Release(); m_ptr = 0;
        !            85:       }
        !            86:     }
        !            87: 
        !            88:   /// Return the pointer.
        !            89:   T * get()
        !            90:     { return m_ptr; }
        !            91: 
        !            92:   /// Pointer dereferencing.
        !            93:   T * operator->()
        !            94:     { return m_ptr; }
        !            95: 
        !            96:   /// Return address of pointer for replacement.
        !            97:   T * * replace()
        !            98:     { reset(); return &m_ptr; }
        !            99: 
        !           100:   /// For (ptr != 0) check.
        !           101:   operator bool() const
        !           102:     { return !!m_ptr; }
        !           103: 
        !           104:   /// For (ptr == 0) check.
        !           105:   bool operator!() const
        !           106:     { return !m_ptr; }
        !           107: 
        !           108: private:
        !           109:   T * m_ptr;
        !           110: 
        !           111:   com_intf_ptr(const com_intf_ptr &);
        !           112:   void operator=(const com_intf_ptr &);
        !           113: };
        !           114: 
        !           115: 
        !           116: /////////////////////////////////////////////////////////////////////////////
        !           117: // wbem_object
        !           118: 
        !           119: class wbem_enumerator;
        !           120: 
        !           121: /// Wrapper class for IWbemClassObject
        !           122: class wbem_object
        !           123: {
        !           124: public:
        !           125:   /// Get string representation.
        !           126:   std::string get_str(const char * name) /*const*/;
        !           127: 
        !           128: private:
        !           129:   /// Contents is set by wbem_enumerator.
        !           130:   friend class wbem_enumerator;
        !           131:   com_intf_ptr<IWbemClassObject> m_intf;
        !           132: };
        !           133: 
        !           134: 
        !           135: /////////////////////////////////////////////////////////////////////////////
        !           136: // wbem_enumerator
        !           137: 
        !           138: class wbem_services;
        !           139: 
        !           140: /// Wrapper class for IEnumWbemClassObject
        !           141: class wbem_enumerator
        !           142: {
        !           143: public:
        !           144:   /// Get next object, return false if none or error.
        !           145:   bool next(wbem_object & obj);
        !           146: 
        !           147: private:
        !           148:   /// Contents is set by wbem_services.
        !           149:   friend class wbem_services;
        !           150:   com_intf_ptr<IEnumWbemClassObject> m_intf;
        !           151: };
        !           152: 
        !           153: 
        !           154: /////////////////////////////////////////////////////////////////////////////
        !           155: // wbem_services
        !           156: 
        !           157: /// Wrapper class for IWbemServices
        !           158: class wbem_services
        !           159: {
        !           160: public:
        !           161:   /// Connect to service, return false on error.
        !           162:   bool connect();
        !           163: 
        !           164:   /// Execute query, get result list.
        !           165:   /// Return false on error.
        !           166:   bool vquery(wbem_enumerator & result, const char * qstr, va_list args) /*const*/;
        !           167: 
        !           168:   /// Execute query, get single result object.
        !           169:   /// Return false on error or result size != 1.
        !           170:   bool vquery1(wbem_object & obj, const char * qstr, va_list args) /*const*/;
        !           171: 
        !           172:   /// Version of vquery() with printf() formatting.
        !           173:   bool query(wbem_enumerator & result, const char * qstr, ...) /*const*/
        !           174:        __attribute__ ((format (printf, 3, 4)));
        !           175: 
        !           176:   /// Version of vquery1() with printf() formatting.
        !           177:   bool query1(wbem_object & obj, const char * qstr, ...) /*const*/
        !           178:        __attribute__ ((format (printf, 3, 4)));
        !           179: 
        !           180: private:
        !           181:   com_intf_ptr<IWbemServices> m_intf;
        !           182: };
        !           183: 
        !           184: #endif // WMIQUERY_H