discover.h

Go to the documentation of this file.

/*
        discover.h

        Copyright 2012 - 2017 Brenton Ross

This file is part of VICI.

VICI is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

VICI 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 VICI. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef CFI_DISCOVER_H
#define CFI_DISCOVER_H

#include "stringy.h"
#include <map>
#include <vector>
#include <memory>

namespace VICI
{

namespace cfi
{


struct DiscoverPointer
{
        void * addr;    // reference

        DiscoverPointer() : addr(nullptr) {}


        DiscoverPointer(void *a) : addr(a) {}
};

typedef std::shared_ptr< DiscoverPointer > DiscoverSharedPointer;

typedef std::weak_ptr< DiscoverPointer > DiscoverWeakPointer;


class Discoverable
{
protected:
        DiscoverSharedPointer discover;
};


class DiscoveryMgr
{
private:
        std::map<std::string, std::vector<DiscoverWeakPointer> > objects;
        DiscoveryMgr() {}
public:
        static DiscoveryMgr & instance();


        void save(const char *prettyName, DiscoverSharedPointer p);


        void fetch(csr name, std::vector<void *> & results);
};

#define DISCOVERABLE \
        VICI::cfi::DiscoveryMgr::instance().save( __PRETTY_FUNCTION__, \
                        (discover.reset( new VICI::cfi::DiscoverPointer( this )), discover) );

} // namespace cfi
} // namespace VICI

#endif /* CFI_DISCOVER_H_ */