Return to wmiquery.h CVS log

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

smartmontools

    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,v 1.1.1.1 2012/02/21 16:32:16 misho Exp $"
   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