Fix build on latest c-ares

an-tao · Apr 3, 2024 · 560a7e4 · 560a7e4
1 parent c922801
commit 560a7e4
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 18 deletions.
diff --git a/trantor/net/inner/AresResolver.cc b/trantor/net/inner/AresResolver.cc
@@ -52,10 +52,17 @@ bool Resolver::isCAresUsed()
 AresResolver::LibraryInitializer::LibraryInitializer()
 {
     ares_library_init(ARES_LIB_INIT_ALL);
+
+    hints_ = new ares_addrinfo_hints;
+    hints_->ai_flags = 0;
+    hints_->ai_family = AF_INET;
+    hints_->ai_socktype = 0;
+    hints_->ai_protocol = 0;
 }
 AresResolver::LibraryInitializer::~LibraryInitializer()
 {
     ares_library_cleanup();
+    delete hints_;
 }
 
 AresResolver::LibraryInitializer AresResolver::libraryInitializer_;
@@ -124,11 +131,12 @@ void AresResolver::resolveInLoop(const std::string& hostname,
 #endif
     init();
     QueryData* queryData = new QueryData(this, cb, hostname);
-    ares_gethostbyname(ctx_,
-                       hostname.c_str(),
-                       AF_INET,
-                       &AresResolver::ares_hostcallback_,
-                       queryData);
+    ares_getaddrinfo(ctx_,
+                     hostname.c_str(),
+                     NULL,
+                     libraryInitializer_.hints_,
+                     &AresResolver::ares_hostcallback_,
+                     queryData);
     struct timeval tv;
     struct timeval* tvp = ares_timeout(ctx_, NULL, &tv);
     double timeout = getSeconds(tvp);
@@ -166,23 +174,33 @@ void AresResolver::onTimer()
 }
 
 void AresResolver::onQueryResult(int status,
-                                 struct hostent* result,
+                                 struct ares_addrinfo* result,
                                  const std::string& hostname,
                                  const ResolverResultsCallback& callback)
 {
     LOG_TRACE << "onQueryResult " << status;
     auto inets_ptr = std::make_shared<std::vector<trantor::InetAddress>>();
     if (result)
     {
-        auto pptr = (struct in_addr**)result->h_addr_list;
-        for (; *pptr != nullptr; pptr++)
+        auto pptr = (struct ares_addrinfo_node*)result->nodes;
+        for (; pptr != NULL; pptr = pptr->ai_next)
         {
-            struct sockaddr_in addr;
-            memset(&addr, 0, sizeof addr);
-            addr.sin_family = AF_INET;
-            addr.sin_port = 0;
-            addr.sin_addr = *reinterpret_cast<in_addr*>(*pptr);
-            inets_ptr->emplace_back(trantor::InetAddress{addr});
+            trantor::InetAddress inet;
+            if (pptr->ai_family == AF_INET)
+            {
+                struct sockaddr_in* addr4 = (struct sockaddr_in*)pptr->ai_addr;
+                inets_ptr->emplace_back(trantor::InetAddress{*addr4});
+            }
+            else if (pptr->ai_family == AF_INET6)
+            {
+                struct sockaddr_in6* addr6 =
+                    (struct sockaddr_in6*)pptr->ai_addr;
+                inets_ptr->emplace_back(trantor::InetAddress{*addr6});
+            }
+            else
+            {
+                // TODO: Handle unknown family?
+            }
         }
     }
     if (inets_ptr->empty())
@@ -237,7 +255,7 @@ void AresResolver::onSockStateChange(int sockfd, bool read, bool write)
 void AresResolver::ares_hostcallback_(void* data,
                                       int status,
                                       int timeouts,
-                                      struct hostent* hostent)
+                                      struct ares_addrinfo* hostent)
 {
     (void)timeouts;
     QueryData* query = static_cast<QueryData*>(data);

diff --git a/trantor/net/inner/AresResolver.h b/trantor/net/inner/AresResolver.h
@@ -15,8 +15,9 @@
 
 extern "C"
 {
-    struct hostent;
+    struct ares_addrinfo;
     struct ares_channeldata;
+    struct ares_addrinfo_hints;
     using ares_channel = struct ares_channeldata*;
 }
 namespace trantor
@@ -157,7 +158,7 @@ class AresResolver : public Resolver,
     void onRead(int sockfd);
     void onTimer();
     void onQueryResult(int status,
-                       struct hostent* result,
+                       struct ares_addrinfo* result,
                        const std::string& hostname,
                        const ResolverResultsCallback& callback);
     void onSockCreate(int sockfd, int type);
@@ -166,7 +167,7 @@ class AresResolver : public Resolver,
     static void ares_hostcallback_(void* data,
                                    int status,
                                    int timeouts,
-                                   struct hostent* hostent);
+                                   struct ares_addrinfo* hostent);
 #ifdef _WIN32
     static int ares_sock_createcallback_(SOCKET sockfd, int type, void* data);
 #else
@@ -184,6 +185,7 @@ class AresResolver : public Resolver,
     {
         LibraryInitializer();
         ~LibraryInitializer();
+        ares_addrinfo_hints* hints_;
     };
     static LibraryInitializer libraryInitializer_;
 };