WbemServices.hpp

Go to the documentation of this file.

/* 
 * Confdump-Agent - Dump static and runtime system configuration
 * Copyright (C) 2009-2012  Straton IT, SAS
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 as
 * published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef INCLUDED_CONFDUMP_SYSTEM_WMI_WBEMSERVICES_HPP
#define INCLUDED_CONFDUMP_SYSTEM_WMI_WBEMSERVICES_HPP

#include "boost/range.hpp"
#include "boost/iterator/iterator_facade.hpp"

#include "QString"

#include "Confdump/System/Wmi/ComContext.hpp"
#include "Confdump/System/Wmi/WmiLists.hpp"
#include "Confdump/System/Wmi/Query.hpp"

struct IWbemServices;
struct IEnumWbemClassObject;
struct IWbemClassObject;

namespace Confdump
{
namespace System
{
namespace Wmi
{

class WbemQueryIterator
 : public boost::iterator_facade<
    WbemQueryIterator,
    boost::shared_ptr<IWbemClassObject>,
    boost::forward_traversal_tag,
    boost::shared_ptr<IWbemClassObject> >
{
public:
    WbemQueryIterator();
    explicit WbemQueryIterator( boost::shared_ptr<IEnumWbemClassObject> enumerator );
    ~WbemQueryIterator();

private:
    friend class boost::iterator_core_access;
    template <class> friend class node_iter;

    bool equal( const WbemQueryIterator &other ) const;
    void increment();
    boost::shared_ptr<IWbemClassObject> dereference() const { return element_; }

    boost::shared_ptr<IEnumWbemClassObject> enumerator_;
    boost::shared_ptr<IWbemClassObject> element_;
};

class WbemServices
{
public:
    WbemServices( ComContext::Guard comGuard );
    ~WbemServices();

    void connect();

    void setWmiNamespace( QString ns );

    boost::iterator_range<WbemQueryIterator> execQuery( const Query & );

    boost::iterator_range<WbemQueryIterator> listAllClasses();

private:
    WbemServices();

    ComContext::Guard comGuard_;
    boost::shared_ptr<IWbemServices> services_;
    QString wmiNamespace_;
    WmiLists wmiLists_;
};

}
}
}

#endif /* INCLUDED_CONFDUMP_SYSTEM_WMI_WBEMSERVICES_HPP */