1 /* 2 * Copyright (C) 2020 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include "all.h" 18 19 #include "generic/Generic.h" 20 #include "route/Route.h" 21 22 #include <map> 23 24 namespace android::nl::protocols { 25 26 // This should be a map of unique_ptr, but it's not trivial to uniformly initialize such a map toMap(std::initializer_list<std::shared_ptr<NetlinkProtocol>> l)27static std::map<int, std::shared_ptr<NetlinkProtocol>> toMap( 28 std::initializer_list<std::shared_ptr<NetlinkProtocol>> l) { 29 std::map<int, std::shared_ptr<NetlinkProtocol>> map; 30 for (auto p : l) { 31 map[p->getProtocol()] = p; 32 } 33 return map; 34 } 35 get(int protocol)36std::optional<std::reference_wrapper<NetlinkProtocol>> get(int protocol) { 37 static auto all = toMap({ 38 std::make_unique<generic::Generic>(), 39 std::make_unique<route::Route>(), 40 }); 41 42 if (all.count(protocol) == 0) return std::nullopt; 43 return *all.find(protocol)->second.get(); 44 } 45 46 } // namespace android::nl::protocols 47