Program Listing for File module_deleter.h

↰ Return to documentation for file (/home/docs/checkouts/readthedocs.org/user_builds/canonical-mir-documentation/checkouts/2.26/include/common/mir/module_deleter.h)

/*
 * Copyright © Canonical Ltd.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version 2 or 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MIR_MODULE_DELETER_H_
#define MIR_MODULE_DELETER_H_

#include <memory>

namespace mir
{
class SharedLibrary;

namespace detail
{
class RefCountedLibrary
{
public:
    RefCountedLibrary(void* address);
    RefCountedLibrary(RefCountedLibrary const&);
    ~RefCountedLibrary();
    RefCountedLibrary& operator=(RefCountedLibrary const&);
private:
    std::shared_ptr<mir::SharedLibrary> internal_state;
};
}

template<typename T>
struct ModuleDeleter : std::default_delete<T>
{
    ModuleDeleter() : library(nullptr) {}
    template<typename U>
    ModuleDeleter(ModuleDeleter<U> const& other)
        : std::default_delete<T>{other},
        library{other.get_library()}
    {
    }

    detail::RefCountedLibrary get_library() const
    {
        return library;
    }

protected:
    ModuleDeleter(void *address_in_module)
        : library{address_in_module}
    {
    }
private:
    detail::RefCountedLibrary library;
};

template<typename T>
using UniqueModulePtr = std::unique_ptr<T,ModuleDeleter<T>>;

namespace
{
template<typename Type, typename... Args>
inline auto make_module_ptr(Args&&... args)
-> UniqueModulePtr<Type>
{
    struct Deleter : ModuleDeleter<Type>
    {
        Deleter(void* address)
            : ModuleDeleter<Type>(address) {}
    } deleter(reinterpret_cast<void*>(&make_module_ptr<Type, Args...>));

    return UniqueModulePtr<Type>(new Type(std::forward<Args>(args)...), std::move(deleter));
}
}

}
#endif