1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
// Copyright 2014 Renato Tegon Forti, Antony Polukhin.
// Copyright Antony Polukhin, 2015-2025.
//
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt
// or copy at http://www.boost.org/LICENSE_1_0.txt)
// MinGW related workaround
#define BOOST_DLL_FORCE_ALIAS_INSTANTIATION
#include "../b2_workarounds.hpp"
#include "../tutorial4/static_plugin.hpp"
#include <boost/dll/runtime_symbol_info.hpp> // for program_location()
#include <boost/dll/shared_library.hpp>
#include <boost/filesystem.hpp>
#include <map>
#include <iostream>
//[plugcpp_plugins_collector_def
namespace dll = boost::dll;
class plugins_collector {
// Name => plugin
using plugins_t = std::map<std::string, dll::shared_library>;
boost::dll::fs::path plugins_directory_;
plugins_t plugins_;
// loads all plugins in plugins_directory_
void load_all();
// Gets `my_plugin_api` instance using "create_plugin" or "plugin" imports,
// stores plugin with its name in the `plugins_` map.
void insert_plugin(dll::shared_library&& lib);
public:
plugins_collector(const boost::dll::fs::path& plugins_directory)
: plugins_directory_(plugins_directory)
{
load_all();
}
void print_plugins() const;
std::size_t count() const;
// ...
};
//]
//[plugcpp_plugins_collector_load_all
void plugins_collector::load_all() {
namespace fs = ::boost::dll::fs;
const auto extension = dll::shared_library::suffix().native();
// Searching a folder for files with '.so' or '.dll' extension
fs::recursive_directory_iterator endit;
for (fs::recursive_directory_iterator it(plugins_directory_); it != endit; ++it) {
if (!fs::is_regular_file(*it)) {
continue;
}
/*<-*/
if ( !b2_workarounds::is_shared_library((*it).path()) ) {
continue;
}
/*->*/
// We found a file. Trying to load it
std::error_code error;
dll::shared_library plugin(it->path(), error);
if (error) {
continue;
}
std::cout << "Loaded (" << plugin.native() << "):" << it->path() << '\n';
// Gets plugin using "create_plugin" or "plugin" function
insert_plugin(std::move(plugin));
}
dll::shared_library plugin(dll::program_location());
std::cout << "Loaded self\n";
insert_plugin(std::move(plugin));
}
//]
//[plugcpp_plugins_collector_insert_plugin
void plugins_collector::insert_plugin(dll::shared_library&& lib) {
std::string plugin_name;
if (lib.has("create_plugin")) {
plugin_name = lib.get_alias<std::shared_ptr<my_plugin_api>()>("create_plugin")()->name();
} else if (lib.has("plugin")) {
plugin_name = lib.get<my_plugin_api>("plugin").name();
} else {
return;
}
if (plugins_.find(plugin_name) == plugins_.cend()) {
plugins_[plugin_name] = std::move(lib);
}
}
//]
void plugins_collector::print_plugins() const {
plugins_t::const_iterator const end = plugins_.cend();
for (plugins_t::const_iterator it = plugins_.cbegin(); it != end; ++it) {
std::cout << '(' << it->second.native() << "): " << it->first << '\n';
}
}
std::size_t plugins_collector::count() const {
return plugins_.size();
}
//[plugcpp_load_all
int main(int argc, char* argv[]) {
/*<-*/
BOOST_ASSERT(argc >= 3);
boost::dll::fs::path path1(argv[1]);
for (int i = 2; i < argc; ++i) {
boost::dll::fs::path path2(argv[i]);
boost::dll::fs::path res;
for (boost::dll::fs::path::iterator it1 = path1.begin(), it2 = path2.begin();
it1 != path1.end() && it2 != path2.end() && *it1 == *it2;
++it1, ++it2)
{
res /= *it1;
}
path1 = res;
}
std::string new_argv = path1.string();
std::cout << "\nPlugins path: " << new_argv << ":\n";
argv[1] = &new_argv[0];
/*->*/
plugins_collector plugins(argv[1]);
std::cout << "\n\nUnique plugins " << plugins.count() << ":\n";
plugins.print_plugins();
// ...
//]
BOOST_ASSERT(plugins.count() >= 3);
}
|
