blob: a1e6be857c865854b54700b40888247febeb118d [file] [log] [blame]
Matt Spinlerdd945862018-09-07 12:41:05 -05001#include "src/argument.hpp"
2
Ed Tanous60520632018-06-11 17:46:52 -07003#include <tinyxml2.h>
4
5#include <atomic>
6#include <boost/algorithm/string/predicate.hpp>
7#include <boost/container/flat_map.hpp>
8#include <boost/container/flat_set.hpp>
9#include <chrono>
10#include <iomanip>
11#include <iostream>
12#include <sdbusplus/asio/connection.hpp>
13#include <sdbusplus/asio/object_server.hpp>
14
15constexpr const char* OBJECT_MAPPER_DBUS_NAME =
16 "xyz.openbmc_project.ObjectMapper";
17constexpr const char* ASSOCIATIONS_INTERFACE = "org.openbmc.Associations";
18
19// interface_map_type is the underlying datastructure the mapper uses.
20// The 3 levels of map are
21// object paths
22// connection names
23// interface names
24using interface_map_type = boost::container::flat_map<
25 std::string, boost::container::flat_map<
26 std::string, boost::container::flat_set<std::string>>>;
27
28using Association = std::tuple<std::string, std::string, std::string>;
29
30boost::container::flat_map<std::string,
31 std::shared_ptr<sdbusplus::asio::dbus_interface>>
32 associationInterfaces;
33
Matt Spinlerdd945862018-09-07 12:41:05 -050034static boost::container::flat_set<std::string> service_whitelist;
35static boost::container::flat_set<std::string> service_blacklist;
36static boost::container::flat_set<std::string> iface_whitelist;
37
Ed Tanous60520632018-06-11 17:46:52 -070038/** Exception thrown when a path is not found in the object list. */
39struct NotFoundException final : public sdbusplus::exception_t
40{
41 const char* name() const noexcept override
42 {
43 return "org.freedesktop.DBus.Error.FileNotFound";
44 };
45 const char* description() const noexcept override
46 {
47 return "path or object not found";
48 };
49 const char* what() const noexcept override
50 {
51 return "org.freedesktop.DBus.Error.FileNotFound: "
52 "The requested object was not found";
53 };
54};
55
56bool get_well_known(
57 boost::container::flat_map<std::string, std::string>& owners,
58 const std::string& request, std::string& well_known)
59{
60 // If it's already a well known name, just return
61 if (!boost::starts_with(request, ":"))
62 {
63 well_known = request;
64 return true;
65 }
66
67 auto it = owners.find(request);
68 if (it == owners.end())
69 {
70 return false;
71 }
72 well_known = it->second;
73 return true;
74}
75
76void update_owners(sdbusplus::asio::connection* conn,
77 boost::container::flat_map<std::string, std::string>& owners,
78 const std::string& new_object)
79{
80 if (boost::starts_with(new_object, ":"))
81 {
82 return;
83 }
84 conn->async_method_call(
85 [&, new_object](const boost::system::error_code ec,
86 const std::string& nameOwner) {
87 if (ec)
88 {
89 std::cerr << "Error getting owner of " << new_object << " : "
90 << ec << "\n";
91 return;
92 }
93 owners[nameOwner] = new_object;
94 },
95 "org.freedesktop.DBus", "/", "org.freedesktop.DBus", "GetNameOwner",
96 new_object);
97}
98
99void send_introspection_complete_signal(sdbusplus::asio::connection* system_bus,
100 const std::string& process_name)
101{
102 // TODO(ed) This signal doesn't get exposed properly in the
103 // introspect right now. Find out how to register signals in
104 // sdbusplus
105 sdbusplus::message::message m = system_bus->new_signal(
106 "/xyz/openbmc_project/object_mapper",
107 "xyz.openbmc_project.ObjectMapper.Private", "IntrospectionComplete");
108 m.append(process_name);
109 m.signal_send();
110}
111
112struct InProgressIntrospect
113{
114 InProgressIntrospect(
115 sdbusplus::asio::connection* system_bus, boost::asio::io_service& io,
116 const std::string& process_name,
117 std::shared_ptr<std::chrono::time_point<std::chrono::steady_clock>>
118 global_start_time) :
119 system_bus(system_bus),
120 io(io), process_name(process_name),
121 global_start_time(global_start_time),
122 process_start_time(std::chrono::steady_clock::now())
123 {
124 }
125 ~InProgressIntrospect()
126 {
127 send_introspection_complete_signal(system_bus, process_name);
128 std::chrono::duration<float> diff =
129 std::chrono::steady_clock::now() - process_start_time;
130 std::cout << std::setw(50) << process_name << " scan took "
131 << diff.count() << " seconds\n";
132
133 // If we're the last outstanding caller globally, calculate the
134 // time it took
135 if (global_start_time != nullptr && global_start_time.use_count() == 1)
136 {
137 diff = std::chrono::steady_clock::now() - *global_start_time;
138 std::cout << "Total scan took " << diff.count()
139 << " seconds to complete\n";
140 }
141 }
142 sdbusplus::asio::connection* system_bus;
143 boost::asio::io_service& io;
144 std::string process_name;
145
146 std::shared_ptr<std::chrono::time_point<std::chrono::steady_clock>>
147 global_start_time;
148 std::chrono::time_point<std::chrono::steady_clock> process_start_time;
149};
150
151static const boost::container::flat_set<std::string> ignored_interfaces{
152 "org.freedesktop.DBus.Introspectable", "org.freedesktop.DBus.Peer",
153 "org.freedesktop.DBus.Properties"};
154
155void do_getmanagedobjects(sdbusplus::asio::connection* system_bus,
156 std::shared_ptr<InProgressIntrospect> transaction,
157 interface_map_type& interface_map, std::string path)
158{
159 // note, the variant type doesn't matter, as we don't actually track
160 // property names as of yet. variant<bool> seemed like the most simple.
161 using ManagedObjectType = std::vector<std::pair<
162 sdbusplus::message::object_path,
163 boost::container::flat_map<
164 std::string, boost::container::flat_map<
165 std::string, sdbusplus::message::variant<bool>>>>>;
166
167 system_bus->async_method_call(
168 [&interface_map, system_bus, transaction,
169 path](const boost::system::error_code ec,
170 const ManagedObjectType& objects) {
171 if (ec)
172 {
173 std::cerr << "GetMangedObjects call failed" << ec << "\n";
174 return;
175 }
176
177 interface_map.reserve(interface_map.size() + objects.size());
178 for (const std::pair<
179 sdbusplus::message::object_path,
180 boost::container::flat_map<
181 std::string,
182 boost::container::flat_map<
183 std::string, sdbusplus::message::variant<bool>>>>&
184 object : objects)
185 {
186 const std::string& path_name = object.first.str;
187 auto& this_path_map =
188 interface_map[path_name][transaction->process_name];
189 for (auto& interface_it : object.second)
190 {
191 this_path_map.insert(interface_it.first);
192 }
193 }
194 },
195 transaction->process_name, path, "org.freedesktop.DBus.ObjectManager",
196 "GetManagedObjects");
197}
198
199void addAssociation(sdbusplus::asio::object_server& objectServer,
200 const std::vector<Association>& associations,
201 const std::string& path)
202{
203 boost::container::flat_map<std::string,
204 boost::container::flat_set<std::string>>
205 objects;
206 for (const Association& association : associations)
207 {
208 std::string forward;
209 std::string reverse;
210 std::string endpoint;
211 std::tie(forward, reverse, endpoint) = association;
212
213 if (forward.size())
214 {
215 objects[path + "/" + forward].emplace(endpoint);
216 }
217 if (reverse.size())
218 {
219 if (endpoint.empty())
220 {
221 std::cerr << "Found invalid association on path " << path
222 << "\n";
223 continue;
224 }
225 objects[endpoint + "/" + reverse].emplace(path);
226 }
227 }
228 for (const auto& object : objects)
229 {
230 // the mapper exposes the new association interface but intakes
231 // the old
232
233 auto& iface = associationInterfaces[object.first];
234 iface = objectServer.add_interface(object.first,
235 "xyz.openbmc_project.Association");
236 iface->register_property("endpoints",
237 std::vector<std::string>(object.second.begin(),
238 object.second.end()));
239 iface->initialize();
240 }
241}
242
243void do_associations(sdbusplus::asio::connection* system_bus,
244 sdbusplus::asio::object_server& objectServer,
245 const std::string& processName, const std::string& path)
246{
247 system_bus->async_method_call(
248 [&objectServer,
249 path](const boost::system::error_code ec,
250 const sdbusplus::message::variant<std::vector<Association>>&
251 variantAssociations) {
252 if (ec)
253 {
254 std::cerr << "Error getting associations from " << path << "\n";
255 }
256 std::vector<Association> associations =
257 sdbusplus::message::variant_ns::get<std::vector<Association>>(
258 variantAssociations);
259 addAssociation(objectServer, associations, path);
260 },
261 processName, path, "org.freedesktop.DBus.Properties", "Get",
262 ASSOCIATIONS_INTERFACE, "associations");
263}
264
265void do_introspect(sdbusplus::asio::connection* system_bus,
266 std::shared_ptr<InProgressIntrospect> transaction,
267 interface_map_type& interface_map,
268 sdbusplus::asio::object_server& objectServer,
269 std::string path)
270{
271 system_bus->async_method_call(
272 [&interface_map, &objectServer, transaction, path,
273 system_bus](const boost::system::error_code ec,
274 const std::string& introspect_xml) {
275 if (ec)
276 {
277 std::cerr << "Introspect call failed with error: " << ec << ", "
278 << ec.message()
279 << " on process: " << transaction->process_name
280 << " path: " << path << "\n";
281 return;
282 }
283
284 tinyxml2::XMLDocument doc;
285
286 tinyxml2::XMLError e = doc.Parse(introspect_xml.c_str());
287 if (e != tinyxml2::XMLError::XML_SUCCESS)
288 {
289 std::cerr << "XML parsing failed\n";
290 return;
291 }
292
293 tinyxml2::XMLNode* pRoot = doc.FirstChildElement("node");
294 if (pRoot == nullptr)
295 {
296 std::cerr << "XML document did not contain any data\n";
297 return;
298 }
299 auto& thisPathMap = interface_map[path];
300 bool handling_via_objectmanager = false;
301 tinyxml2::XMLElement* pElement =
302 pRoot->FirstChildElement("interface");
303 while (pElement != nullptr)
304 {
305 const char* iface_name = pElement->Attribute("name");
306 if (iface_name == nullptr)
307 {
308 continue;
309 }
310
Matt Spinlerdd945862018-09-07 12:41:05 -0500311 std::string iface{iface_name};
312
313 if (((ignored_interfaces.find(iface) ==
314 ignored_interfaces.end()) &&
315 (std::find_if(iface_whitelist.begin(),
316 iface_whitelist.end(),
317 [iface](const auto& prefix) {
318 return boost::starts_with(iface, prefix);
319 }) != iface_whitelist.end())) ||
320 (iface == "org.freedesktop.DBus.ObjectManager"))
Ed Tanous60520632018-06-11 17:46:52 -0700321 {
322 thisPathMap[transaction->process_name].emplace(iface_name);
323 }
324 if (std::strcmp(iface_name, ASSOCIATIONS_INTERFACE) == 0)
325 {
326 do_associations(system_bus, objectServer,
327 transaction->process_name, path);
328 }
329 else if (std::strcmp(iface_name,
330 "org.freedesktop.DBus.ObjectManager") == 0)
331 {
332 // TODO(ed) in the current implementation,
333 // introspect is actually faster than
334 // getmanagedObjects, but I suspect it will be
335 // faster when needing to deal with
336 // associations, so leave the code here for now
337
338 // handling_via_objectmanager = true;
339 // do_getmanagedobjects(system_bus, transaction,
340 // interface_map, path);
341 }
342
343 pElement = pElement->NextSiblingElement("interface");
344 }
345
346 if (!handling_via_objectmanager)
347 {
348 pElement = pRoot->FirstChildElement("node");
349 while (pElement != nullptr)
350 {
351 const char* child_path = pElement->Attribute("name");
352 if (child_path != nullptr)
353 {
354 std::string parent_path(path);
355 if (parent_path == "/")
356 {
357 parent_path.clear();
358 }
359
360 do_introspect(system_bus, transaction, interface_map,
361 objectServer,
362 parent_path + "/" + child_path);
363 }
364 pElement = pElement->NextSiblingElement("node");
365 }
366 }
367 },
368 transaction->process_name, path, "org.freedesktop.DBus.Introspectable",
369 "Introspect");
370}
371
372bool need_to_introspect(const std::string& process_name)
373{
Matt Spinlerdd945862018-09-07 12:41:05 -0500374 auto inWhitelist =
375 std::find_if(service_whitelist.begin(), service_whitelist.end(),
376 [&process_name](const auto& prefix) {
377 return boost::starts_with(process_name, prefix);
378 }) != service_whitelist.end();
379
380 // This holds full service names, not prefixes
381 auto inBlacklist =
382 service_blacklist.find(process_name) != service_blacklist.end();
383
384 return inWhitelist && !inBlacklist;
Ed Tanous60520632018-06-11 17:46:52 -0700385}
386
387void start_new_introspect(
388 sdbusplus::asio::connection* system_bus, boost::asio::io_service& io,
389 interface_map_type& interface_map, const std::string& process_name,
390 std::shared_ptr<std::chrono::time_point<std::chrono::steady_clock>>
391 global_start_time,
392 sdbusplus::asio::object_server& objectServer)
393{
394 if (need_to_introspect(process_name))
395 {
396
397 std::cerr << "starting introspect on " << process_name << "\n";
398 std::shared_ptr<InProgressIntrospect> transaction =
399 std::make_shared<InProgressIntrospect>(system_bus, io, process_name,
400 global_start_time);
401
402 do_introspect(system_bus, transaction, interface_map, objectServer,
403 "/");
404 }
405}
406
407// TODO(ed) replace with std::set_intersection once c++17 is available
408template <class InputIt1, class InputIt2>
409bool intersect(InputIt1 first1, InputIt1 last1, InputIt2 first2, InputIt2 last2)
410{
411 while (first1 != last1 && first2 != last2)
412 {
413 if (*first1 < *first2)
414 {
415 ++first1;
416 continue;
417 }
418 if (*first2 < *first1)
419 {
420 ++first2;
421 continue;
422 }
423 return true;
424 }
425 return false;
426}
427
428void doListNames(
429 boost::asio::io_service& io, interface_map_type& interface_map,
430 sdbusplus::asio::connection* system_bus,
431 boost::container::flat_map<std::string, std::string>& name_owners,
432 sdbusplus::asio::object_server& objectServer)
433{
434 system_bus->async_method_call(
435 [&io, &interface_map, &name_owners, &objectServer,
436 system_bus](const boost::system::error_code ec,
437 std::vector<std::string> process_names) {
438 if (ec)
439 {
440 std::cerr << "Error getting names: " << ec << "\n";
441 std::exit(EXIT_FAILURE);
442 return;
443 }
444 std::cerr << "ListNames returned " << process_names.size()
445 << " entries\n";
446 // Try to make startup consistent
447 std::sort(process_names.begin(), process_names.end());
448 std::shared_ptr<std::chrono::time_point<std::chrono::steady_clock>>
449 global_start_time = std::make_shared<
450 std::chrono::time_point<std::chrono::steady_clock>>(
451 std::chrono::steady_clock::now());
452 for (const std::string& process_name : process_names)
453 {
454 if (need_to_introspect(process_name))
455 {
456 start_new_introspect(system_bus, io, interface_map,
457 process_name, global_start_time,
458 objectServer);
459 update_owners(system_bus, name_owners, process_name);
460 }
461 }
462 },
463 "org.freedesktop.DBus", "/org/freedesktop/DBus", "org.freedesktop.DBus",
464 "ListNames");
465}
466
Matt Spinlerdd945862018-09-07 12:41:05 -0500467void splitArgs(const std::string& stringArgs,
468 boost::container::flat_set<std::string>& listArgs)
469{
470 std::istringstream args;
471 std::string arg;
472
473 args.str(stringArgs);
474
475 while (!args.eof())
476 {
477 args >> arg;
478 if (!arg.empty())
479 {
480 listArgs.insert(arg);
481 }
482 }
483}
484
Matt Spinler9f0958e2018-09-11 08:26:10 -0500485void addSubtreeResult(
486 std::vector<interface_map_type::value_type>& subtree,
487 const std::string& objectPath,
488 const std::pair<std::string, boost::container::flat_set<std::string>>&
489 interfaceMap)
490{
491 // Adds an object path/service name/interface list entry to
492 // the results of GetSubTree.
493 // If an entry for the object path already exists, just add the
494 // service name and interfaces to that entry, otherwise create
495 // a new entry.
496 auto entry = std::find_if(
497 subtree.begin(), subtree.end(),
498 [&objectPath](const auto& i) { return objectPath == i.first; });
499
500 if (entry != subtree.end())
501 {
502 entry->second.emplace(interfaceMap);
503 }
504 else
505 {
506 interface_map_type::value_type object;
507 object.first = objectPath;
508 object.second.emplace(interfaceMap);
509 subtree.push_back(object);
510 }
511}
512
Ed Tanous60520632018-06-11 17:46:52 -0700513int main(int argc, char** argv)
514{
515 std::cerr << "started\n";
Matt Spinlerdd945862018-09-07 12:41:05 -0500516 auto options = ArgumentParser(argc, argv);
Ed Tanous60520632018-06-11 17:46:52 -0700517 boost::asio::io_service io;
518 std::shared_ptr<sdbusplus::asio::connection> system_bus =
519 std::make_shared<sdbusplus::asio::connection>(io);
520
Matt Spinlerdd945862018-09-07 12:41:05 -0500521 splitArgs(options["service-namespaces"], service_whitelist);
522 splitArgs(options["interface-namespaces"], iface_whitelist);
523 splitArgs(options["service-blacklists"], service_blacklist);
524
Ed Tanous60520632018-06-11 17:46:52 -0700525 system_bus->request_name(OBJECT_MAPPER_DBUS_NAME);
526 sdbusplus::asio::object_server server(system_bus);
527
528 // Construct a signal set registered for process termination.
529 boost::asio::signal_set signals(io, SIGINT, SIGTERM);
530 signals.async_wait([&io](const boost::system::error_code& error,
531 int signal_number) { io.stop(); });
532
533 interface_map_type interface_map;
534 boost::container::flat_map<std::string, std::string> name_owners;
535
536 std::function<void(sdbusplus::message::message & message)>
537 nameChangeHandler = [&interface_map, &io, &name_owners, &server,
538 system_bus](sdbusplus::message::message& message) {
539 std::string name;
540 std::string old_owner;
541 std::string new_owner;
542
543 message.read(name, old_owner, new_owner);
544
545 if (!old_owner.empty())
546 {
547 if (boost::starts_with(old_owner, ":"))
548 {
549 auto it = name_owners.find(old_owner);
550 if (it != name_owners.end())
551 {
552 name_owners.erase(it);
553 }
554 }
555 // Connection removed
556 interface_map_type::iterator path_it = interface_map.begin();
557 while (path_it != interface_map.end())
558 {
559 path_it->second.erase(name);
560 if (path_it->second.empty())
561 {
562 // If the last connection to the object is gone,
563 // delete the top level object
564 path_it = interface_map.erase(path_it);
565 continue;
566 }
567 path_it++;
568 }
569 }
570
571 if (!new_owner.empty())
572 {
573 auto transaction = std::make_shared<
574 std::chrono::time_point<std::chrono::steady_clock>>(
575 std::chrono::steady_clock::now());
576 // New daemon added
577 if (need_to_introspect(name))
578 {
579 name_owners[new_owner] = name;
580 start_new_introspect(system_bus.get(), io, interface_map,
581 name, transaction, server);
582 }
583 }
584 };
585
586 sdbusplus::bus::match::match nameOwnerChanged(
587 static_cast<sdbusplus::bus::bus&>(*system_bus),
588 sdbusplus::bus::match::rules::nameOwnerChanged(), nameChangeHandler);
589
590 std::function<void(sdbusplus::message::message & message)>
591 interfacesAddedHandler = [&interface_map, &name_owners, &server](
592 sdbusplus::message::message& message) {
593 sdbusplus::message::object_path obj_path;
594 std::vector<std::pair<
595 std::string, std::vector<std::pair<
596 std::string, sdbusplus::message::variant<
597 std::vector<Association>>>>>>
598 interfaces_added;
599 message.read(obj_path, interfaces_added);
600 std::string well_known;
601 if (!get_well_known(name_owners, message.get_sender(), well_known))
602 {
603 return; // only introspect well-known
604 }
605 if (need_to_introspect(well_known))
606 {
607 auto& iface_list = interface_map[obj_path.str];
608
609 for (const std::pair<
610 std::string,
611 std::vector<std::pair<std::string,
612 sdbusplus::message::variant<
613 std::vector<Association>>>>>&
614 interface_pair : interfaces_added)
615 {
616 iface_list[well_known].emplace(interface_pair.first);
617
618 if (interface_pair.first == ASSOCIATIONS_INTERFACE)
619 {
620 const sdbusplus::message::variant<
621 std::vector<Association>>* variantAssociations =
622 nullptr;
623 for (const auto& interface : interface_pair.second)
624 {
625 if (interface.first == "associations")
626 {
627 variantAssociations = &(interface.second);
628 }
629 }
630 if (variantAssociations == nullptr)
631 {
632 std::cerr << "Illegal association found on "
633 << well_known << "\n";
634 continue;
635 }
636 std::vector<Association> associations =
637 sdbusplus::message::variant_ns::get<
638 std::vector<Association>>(*variantAssociations);
639 addAssociation(server, associations, obj_path.str);
640 }
641 }
642 }
643 };
644
645 sdbusplus::bus::match::match interfacesAdded(
646 static_cast<sdbusplus::bus::bus&>(*system_bus),
647 sdbusplus::bus::match::rules::interfacesAdded(),
648 interfacesAddedHandler);
649
650 std::function<void(sdbusplus::message::message & message)>
651 interfacesRemovedHandler = [&interface_map, &name_owners, &server](
652 sdbusplus::message::message& message) {
653 sdbusplus::message::object_path obj_path;
654 std::vector<std::string> interfaces_removed;
655 message.read(obj_path, interfaces_removed);
656 auto connection_map = interface_map.find(obj_path.str);
657 if (connection_map == interface_map.end())
658 {
659 return;
660 }
661
662 std::string sender;
663 if (!get_well_known(name_owners, message.get_sender(), sender))
664 {
665 return;
666 }
667 for (const std::string& interface : interfaces_removed)
668 {
669 auto interface_set = connection_map->second.find(sender);
670 if (interface_set == connection_map->second.end())
671 {
672 continue;
673 }
674
675 if (interface == ASSOCIATIONS_INTERFACE)
676 {
677 auto findAssociation =
678 associationInterfaces.find(interface);
679 if (findAssociation != associationInterfaces.end())
680 {
681 server.remove_interface(findAssociation->second);
682 findAssociation->second = nullptr;
683 }
684 }
685
686 interface_set->second.erase(interface);
687 // If this was the last interface on this connection,
688 // erase the connection
689 if (interface_set->second.empty())
690 {
691 connection_map->second.erase(interface_set);
692 }
693 }
694 // If this was the last connection on this object path,
695 // erase the object path
696 if (connection_map->second.empty())
697 {
698 interface_map.erase(connection_map);
699 }
700 };
701
702 sdbusplus::bus::match::match interfacesRemoved(
703 static_cast<sdbusplus::bus::bus&>(*system_bus),
704 sdbusplus::bus::match::rules::interfacesRemoved(),
705 interfacesRemovedHandler);
706
707 std::function<void(sdbusplus::message::message & message)>
708 associationChangedHandler =
709 [&server](sdbusplus::message::message& message) {
710 std::string objectName;
711 boost::container::flat_map<
712 std::string,
713 sdbusplus::message::variant<std::vector<Association>>>
714 values;
715 message.read(objectName, values);
716 auto findAssociations = values.find("associations");
717 if (findAssociations != values.end())
718 {
719 std::vector<Association> associations =
720 sdbusplus::message::variant_ns::get<
721 std::vector<Association>>(findAssociations->second);
722 addAssociation(server, associations, message.get_path());
723 }
724 };
725 sdbusplus::bus::match::match associationChanged(
726 static_cast<sdbusplus::bus::bus&>(*system_bus),
727 sdbusplus::bus::match::rules::interface(
728 "org.freedesktop.DBus.Properties") +
729 sdbusplus::bus::match::rules::member("PropertiesChanged") +
730 sdbusplus::bus::match::rules::argN(0, ASSOCIATIONS_INTERFACE),
731 associationChangedHandler);
732
733 std::shared_ptr<sdbusplus::asio::dbus_interface> iface =
734 server.add_interface("/xyz/openbmc_project/object_mapper",
735 "xyz.openbmc_project.ObjectMapper");
736
737 iface->register_method(
738 "GetAncestors", [&interface_map](const std::string& req_path,
739 std::vector<std::string>& interfaces) {
740 // Interfaces need to be sorted for intersect to function
741 std::sort(interfaces.begin(), interfaces.end());
742
743 std::vector<interface_map_type::value_type> ret;
744 for (auto& object_path : interface_map)
745 {
746 auto& this_path = object_path.first;
747 if (boost::starts_with(req_path, this_path))
748 {
749 if (interfaces.empty())
750 {
751 ret.emplace_back(object_path);
752 }
753 else
754 {
755 for (auto& interface_map : object_path.second)
756 {
757
758 if (intersect(interfaces.begin(), interfaces.end(),
759 interface_map.second.begin(),
760 interface_map.second.end()))
761 {
762 ret.emplace_back(object_path);
763 break;
764 }
765 }
766 }
767 }
768 }
769 if (ret.empty())
770 {
771 throw NotFoundException();
772 }
773
774 return ret;
775 });
776
777 iface->register_method(
778 "GetObject", [&interface_map](const std::string& path,
779 std::vector<std::string>& interfaces) {
780 // Interfaces need to be sorted for intersect to function
781 std::sort(interfaces.begin(), interfaces.end());
782 auto path_ref = interface_map.find(path);
783 if (path_ref == interface_map.end())
784 {
785 throw NotFoundException();
786 }
787 if (interfaces.empty())
788 {
789 return path_ref->second;
790 }
791 for (auto& interface_map : path_ref->second)
792 {
793 if (intersect(interfaces.begin(), interfaces.end(),
794 interface_map.second.begin(),
795 interface_map.second.end()))
796 {
797 return path_ref->second;
798 }
799 }
800 // Unable to find intersection, return default constructed
801 // object
802 throw NotFoundException();
803 });
804
805 iface->register_method(
806 "GetSubTree",
807 [&interface_map](const std::string& req_path, int32_t depth,
808 std::vector<std::string>& interfaces) {
809 if (depth <= 0)
810 {
811 depth = std::numeric_limits<int32_t>::max();
812 }
813 // Interfaces need to be sorted for intersect to function
814 std::sort(interfaces.begin(), interfaces.end());
815 std::vector<interface_map_type::value_type> ret;
816
817 for (auto& object_path : interface_map)
818 {
819 auto& this_path = object_path.first;
820 if (boost::starts_with(this_path, req_path))
821 {
822 // count the number of slashes past the search term
823 int32_t this_depth =
824 std::count(this_path.begin() + req_path.size(),
825 this_path.end(), '/');
826 if (this_depth <= depth)
827 {
Ed Tanous60520632018-06-11 17:46:52 -0700828 for (auto& interface_map : object_path.second)
829 {
830 if (intersect(interfaces.begin(), interfaces.end(),
831 interface_map.second.begin(),
Matt Spinler9f0958e2018-09-11 08:26:10 -0500832 interface_map.second.end()) ||
833 interfaces.empty())
Ed Tanous60520632018-06-11 17:46:52 -0700834 {
Matt Spinler9f0958e2018-09-11 08:26:10 -0500835 addSubtreeResult(ret, this_path, interface_map);
836
837 // if not just adding every interface, then done
838 if (!interfaces.empty())
839 {
840 break;
841 }
Ed Tanous60520632018-06-11 17:46:52 -0700842 }
843 }
Ed Tanous60520632018-06-11 17:46:52 -0700844 }
845 }
846 }
847 if (ret.empty())
848 {
849 throw NotFoundException();
850 }
851 return ret;
852 });
853
854 iface->register_method(
855 "GetSubTreePaths",
856 [&interface_map](const std::string& req_path, int32_t depth,
857 std::vector<std::string>& interfaces) {
858 if (depth <= 0)
859 {
860 depth = std::numeric_limits<int32_t>::max();
861 }
862 // Interfaces need to be sorted for intersect to function
863 std::sort(interfaces.begin(), interfaces.end());
864 std::vector<std::string> ret;
865 for (auto& object_path : interface_map)
866 {
867 auto& this_path = object_path.first;
868 if (boost::starts_with(this_path, req_path))
869 {
870 // count the number of slashes past the search term
871 int this_depth =
872 std::count(this_path.begin() + req_path.size(),
873 this_path.end(), '/');
874 if (this_depth <= depth)
875 {
876 bool add = interfaces.empty();
877 for (auto& interface_map : object_path.second)
878 {
879 if (intersect(interfaces.begin(), interfaces.end(),
880 interface_map.second.begin(),
881 interface_map.second.end()))
882 {
883 add = true;
884 break;
885 }
886 }
887 if (add)
888 {
889 // TODO(ed) this is a copy
890 ret.emplace_back(this_path);
891 }
892 }
893 }
894 }
895 if (ret.empty())
896 {
897 throw NotFoundException();
898 }
899 return ret;
900 });
901
902 iface->initialize();
903
904 io.post([&]() {
905 doListNames(io, interface_map, system_bus.get(), name_owners, server);
906 });
907
908 std::cerr << "starting event loop\n";
909 io.run();
910}