Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 1 | #pragma once |
Ed Tanous | 3ccb3ad | 2023-01-13 17:40:03 -0800 | [diff] [blame] | 2 | #include "dbus_singleton.hpp" |
| 3 | #include "logging.hpp" |
| 4 | |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 5 | #include <boost/asio/ip/address.hpp> |
| 6 | #include <boost/asio/ip/basic_endpoint.hpp> |
Carson Labrado | 7e8890c | 2022-11-23 23:58:01 +0000 | [diff] [blame] | 7 | #include <boost/asio/ip/tcp.hpp> |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 8 | #include <sdbusplus/message.hpp> |
| 9 | |
| 10 | #include <charconv> |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 11 | #include <memory> |
| 12 | |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 13 | namespace async_resolve |
| 14 | { |
| 15 | |
Ed Tanous | e1452be | 2021-10-04 17:03:52 -0700 | [diff] [blame] | 16 | inline bool endpointFromResolveTuple(const std::vector<uint8_t>& ipAddress, |
| 17 | boost::asio::ip::tcp::endpoint& endpoint) |
| 18 | { |
| 19 | if (ipAddress.size() == 4) // ipv4 address |
| 20 | { |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 21 | BMCWEB_LOG_DEBUG("ipv4 address"); |
Ed Tanous | e1452be | 2021-10-04 17:03:52 -0700 | [diff] [blame] | 22 | boost::asio::ip::address_v4 ipv4Addr( |
| 23 | {ipAddress[0], ipAddress[1], ipAddress[2], ipAddress[3]}); |
| 24 | endpoint.address(ipv4Addr); |
| 25 | } |
| 26 | else if (ipAddress.size() == 16) // ipv6 address |
| 27 | { |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 28 | BMCWEB_LOG_DEBUG("ipv6 address"); |
Ed Tanous | e1452be | 2021-10-04 17:03:52 -0700 | [diff] [blame] | 29 | boost::asio::ip::address_v6 ipv6Addr( |
| 30 | {ipAddress[0], ipAddress[1], ipAddress[2], ipAddress[3], |
| 31 | ipAddress[4], ipAddress[5], ipAddress[6], ipAddress[7], |
| 32 | ipAddress[8], ipAddress[9], ipAddress[10], ipAddress[11], |
| 33 | ipAddress[12], ipAddress[13], ipAddress[14], ipAddress[15]}); |
| 34 | endpoint.address(ipv6Addr); |
| 35 | } |
| 36 | else |
| 37 | { |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 38 | BMCWEB_LOG_ERROR("Resolve failed to fetch the IP address"); |
Ed Tanous | e1452be | 2021-10-04 17:03:52 -0700 | [diff] [blame] | 39 | return false; |
| 40 | } |
| 41 | return true; |
| 42 | } |
| 43 | |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 44 | class Resolver |
| 45 | { |
| 46 | public: |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 47 | // unused io param used to keep interface identical to |
| 48 | // boost::asio::tcp:::resolver |
| 49 | explicit Resolver(boost::asio::io_context& /*io*/) {} |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 50 | |
| 51 | ~Resolver() = default; |
| 52 | |
Ed Tanous | ecd6a3a | 2022-01-07 09:18:40 -0800 | [diff] [blame] | 53 | Resolver(const Resolver&) = delete; |
| 54 | Resolver(Resolver&&) = delete; |
| 55 | Resolver& operator=(const Resolver&) = delete; |
| 56 | Resolver& operator=(Resolver&&) = delete; |
| 57 | |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 58 | using results_type = std::vector<boost::asio::ip::tcp::endpoint>; |
| 59 | |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 60 | template <typename ResolveHandler> |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 61 | // This function is kept using snake case so that it is interoperable with |
| 62 | // boost::asio::ip::tcp::resolver |
| 63 | // NOLINTNEXTLINE(readability-identifier-naming) |
Ed Tanous | a716aa7 | 2023-08-01 11:35:53 -0700 | [diff] [blame] | 64 | void async_resolve(std::string_view host, std::string_view port, |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 65 | ResolveHandler&& handler) |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 66 | { |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 67 | BMCWEB_LOG_DEBUG("Trying to resolve: {}:{}", host, port); |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 68 | |
| 69 | uint16_t portNum = 0; |
| 70 | |
| 71 | auto it = std::from_chars(&*port.begin(), &*port.end(), portNum); |
| 72 | if (it.ec != std::errc()) |
| 73 | { |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 74 | BMCWEB_LOG_ERROR("Failed to get the Port"); |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 75 | handler(std::make_error_code(std::errc::invalid_argument), |
| 76 | results_type{}); |
| 77 | |
| 78 | return; |
| 79 | } |
| 80 | |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 81 | uint64_t flag = 0; |
| 82 | crow::connections::systemBus->async_method_call( |
Ed Tanous | a716aa7 | 2023-08-01 11:35:53 -0700 | [diff] [blame] | 83 | [host{std::string(host)}, portNum, |
Ed Tanous | 8cb2c02 | 2024-03-27 16:31:46 -0700 | [diff] [blame] | 84 | handler = std::forward<ResolveHandler>(handler)]( |
Ed Tanous | 5e7e2dc | 2023-02-16 10:37:01 -0800 | [diff] [blame] | 85 | const boost::system::error_code& ec, |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 86 | const std::vector< |
| 87 | std::tuple<int32_t, int32_t, std::vector<uint8_t>>>& resp, |
| 88 | const std::string& hostName, const uint64_t flagNum) { |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 89 | results_type endpointList; |
Ed Tanous | 002d39b | 2022-05-31 08:59:27 -0700 | [diff] [blame] | 90 | if (ec) |
| 91 | { |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 92 | BMCWEB_LOG_ERROR("Resolve failed: {}", ec.message()); |
Ed Tanous | 002d39b | 2022-05-31 08:59:27 -0700 | [diff] [blame] | 93 | handler(ec, endpointList); |
| 94 | return; |
| 95 | } |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 96 | BMCWEB_LOG_DEBUG("ResolveHostname returned: {}:{}", hostName, |
| 97 | flagNum); |
Ed Tanous | 002d39b | 2022-05-31 08:59:27 -0700 | [diff] [blame] | 98 | // Extract the IP address from the response |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 99 | for (const std::tuple<int32_t, int32_t, std::vector<uint8_t>>& |
| 100 | resolveList : resp) |
Ed Tanous | 002d39b | 2022-05-31 08:59:27 -0700 | [diff] [blame] | 101 | { |
Ed Tanous | 002d39b | 2022-05-31 08:59:27 -0700 | [diff] [blame] | 102 | boost::asio::ip::tcp::endpoint endpoint; |
Ed Tanous | f8ca6d7 | 2022-06-28 12:12:03 -0700 | [diff] [blame] | 103 | endpoint.port(portNum); |
Ed Tanous | e1452be | 2021-10-04 17:03:52 -0700 | [diff] [blame] | 104 | if (!endpointFromResolveTuple(std::get<2>(resolveList), |
| 105 | endpoint)) |
| 106 | { |
| 107 | boost::system::error_code ecErr = make_error_code( |
| 108 | boost::system::errc::address_not_available); |
| 109 | handler(ecErr, endpointList); |
| 110 | } |
Ed Tanous | 62598e3 | 2023-07-17 17:06:25 -0700 | [diff] [blame] | 111 | BMCWEB_LOG_DEBUG("resolved endpoint is : {}", |
| 112 | endpoint.address().to_string()); |
Ed Tanous | 002d39b | 2022-05-31 08:59:27 -0700 | [diff] [blame] | 113 | endpointList.push_back(endpoint); |
| 114 | } |
| 115 | // All the resolved data is filled in the endpointList |
| 116 | handler(ec, endpointList); |
Patrick Williams | 5a39f77 | 2023-10-20 11:20:21 -0500 | [diff] [blame] | 117 | }, |
Sunitha Harish | 29a82b0 | 2021-02-18 15:54:16 +0530 | [diff] [blame] | 118 | "org.freedesktop.resolve1", "/org/freedesktop/resolve1", |
| 119 | "org.freedesktop.resolve1.Manager", "ResolveHostname", 0, host, |
| 120 | AF_UNSPEC, flag); |
| 121 | } |
| 122 | }; |
| 123 | |
| 124 | } // namespace async_resolve |