From 87f33d6a03d8c25393ad6472fcad5b69bb60aa6b Mon Sep 17 00:00:00 2001
From: Yu Watanabe <watanabe.yu+github@gmail.com>
Date: Sat, 5 Feb 2022 22:04:42 +0900
Subject: [PATCH] resolve: use _cleanup_ attribute for freeing DnsQuery

(cherry picked from commit c704288c473fa08820566fdb16c38726d24db026)
(cherry picked from commit 0533d1aab61b6a797d07c4c861acf5e87f8191e8)

Conflict:NA
Reference:https://github.com/systemd/systemd/commit/87f33d6a03d8c25393ad6472fcad5b69bb60aa6b
---
 src/resolve/resolved-bus.c      | 112 +++++++++++++++-----------------
 src/resolve/resolved-dns-stub.c |  27 +++-----
 src/resolve/resolved-varlink.c  |  38 +++++------
 3 files changed, 79 insertions(+), 98 deletions(-)

diff --git a/src/resolve/resolved-bus.c b/src/resolve/resolved-bus.c
index 78d1acf5e0..ed06895ed9 100644
--- a/src/resolve/resolved-bus.c
+++ b/src/resolve/resolved-bus.c
@@ -179,9 +179,10 @@ static int append_address(sd_bus_message *reply, DnsResourceRecord *rr, int ifin
         return 0;
 }
 
-static void bus_method_resolve_hostname_complete(DnsQuery *q) {
+static void bus_method_resolve_hostname_complete(DnsQuery *query) {
         _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *canonical = NULL;
         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         _cleanup_free_ char *normalized = NULL;
         DnsQuestion *question;
         DnsResourceRecord *rr;
@@ -202,8 +203,11 @@ static void bus_method_resolve_hostname_complete(DnsQuery *q) {
         }
         if (r < 0)
                 goto finish;
-        if (r == DNS_QUERY_CNAME) /* This was a cname, and the query was restarted. */
+        if (r == DNS_QUERY_CNAME) {
+                /* This was a cname, and the query was restarted. */
+                TAKE_PTR(q);
                 return;
+        }
 
         r = sd_bus_message_new_method_return(q->bus_request, &reply);
         if (r < 0)
@@ -264,8 +268,6 @@ finish:
                 log_error_errno(r, "Failed to send hostname reply: %m");
                 sd_bus_reply_method_errno(q->bus_request, r, NULL);
         }
-
-        dns_query_free(q);
 }
 
 static int validate_and_mangle_flags(
@@ -403,11 +405,11 @@ void bus_client_log(sd_bus_message *m, const char *what) {
 
 static int bus_method_resolve_hostname(sd_bus_message *message, void *userdata, sd_bus_error *error) {
         _cleanup_(dns_question_unrefp) DnsQuestion *question_idna = NULL, *question_utf8 = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = NULL;
         Manager *m = userdata;
         const char *hostname;
         int family, ifindex;
         uint64_t flags;
-        DnsQuery *q;
         int r;
 
         assert(message);
@@ -459,21 +461,19 @@ static int bus_method_resolve_hostname(sd_bus_message *message, void *userdata,
 
         r = dns_query_bus_track(q, message);
         if (r < 0)
-                goto fail;
+                return r;
 
         r = dns_query_go(q);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(q);
         return 1;
-
-fail:
-        dns_query_free(q);
-        return r;
 }
 
-static void bus_method_resolve_address_complete(DnsQuery *q) {
+static void bus_method_resolve_address_complete(DnsQuery *query) {
         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         DnsQuestion *question;
         DnsResourceRecord *rr;
         unsigned added = 0;
@@ -493,8 +493,11 @@ static void bus_method_resolve_address_complete(DnsQuery *q) {
         }
         if (r < 0)
                 goto finish;
-        if (r == DNS_QUERY_CNAME) /* This was a cname, and the query was restarted. */
+        if (r == DNS_QUERY_CNAME) {
+                /* This was a cname, and the query was restarted. */
+                TAKE_PTR(q);
                 return;
+        }
 
         r = sd_bus_message_new_method_return(q->bus_request, &reply);
         if (r < 0)
@@ -550,17 +553,15 @@ finish:
                 log_error_errno(r, "Failed to send address reply: %m");
                 sd_bus_reply_method_errno(q->bus_request, r, NULL);
         }
-
-        dns_query_free(q);
 }
 
 static int bus_method_resolve_address(sd_bus_message *message, void *userdata, sd_bus_error *error) {
         _cleanup_(dns_question_unrefp) DnsQuestion *question = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = NULL;
         Manager *m = userdata;
         union in_addr_union a;
         int family, ifindex;
         uint64_t flags;
-        DnsQuery *q;
         int r;
 
         assert(message);
@@ -604,17 +605,14 @@ static int bus_method_resolve_address(sd_bus_message *message, void *userdata, s
 
         r = dns_query_bus_track(q, message);
         if (r < 0)
-                goto fail;
+                return r;
 
         r = dns_query_go(q);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(q);
         return 1;
-
-fail:
-        dns_query_free(q);
-        return r;
 }
 
 static int bus_message_append_rr(sd_bus_message *m, DnsResourceRecord *rr, int ifindex) {
@@ -645,8 +643,9 @@ static int bus_message_append_rr(sd_bus_message *m, DnsResourceRecord *rr, int i
         return sd_bus_message_close_container(m);
 }
 
-static void bus_method_resolve_record_complete(DnsQuery *q) {
+static void bus_method_resolve_record_complete(DnsQuery *query) {
         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         DnsResourceRecord *rr;
         DnsQuestion *question;
         unsigned added = 0;
@@ -667,8 +666,11 @@ static void bus_method_resolve_record_complete(DnsQuery *q) {
         }
         if (r < 0)
                 goto finish;
-        if (r == DNS_QUERY_CNAME) /* This was a cname, and the query was restarted. */
+        if (r == DNS_QUERY_CNAME) {
+                /* This was a cname, and the query was restarted. */
+                TAKE_PTR(q);
                 return;
+        }
 
         r = sd_bus_message_new_method_return(q->bus_request, &reply);
         if (r < 0)
@@ -714,19 +716,17 @@ finish:
                 log_error_errno(r, "Failed to send record reply: %m");
                 sd_bus_reply_method_errno(q->bus_request, r, NULL);
         }
-
-        dns_query_free(q);
 }
 
 static int bus_method_resolve_record(sd_bus_message *message, void *userdata, sd_bus_error *error) {
         _cleanup_(dns_resource_key_unrefp) DnsResourceKey *key = NULL;
         _cleanup_(dns_question_unrefp) DnsQuestion *question = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = NULL;
         Manager *m = userdata;
         uint16_t class, type;
         const char *name;
         int r, ifindex;
         uint64_t flags;
-        DnsQuery *q;
 
         assert(message);
         assert(m);
@@ -782,17 +782,14 @@ static int bus_method_resolve_record(sd_bus_message *message, void *userdata, sd
 
         r = dns_query_bus_track(q, message);
         if (r < 0)
-                goto fail;
+                return r;
 
         r = dns_query_go(q);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(q);
         return 1;
-
-fail:
-        dns_query_free(q);
-        return r;
 }
 
 static int append_srv(DnsQuery *q, sd_bus_message *reply, DnsResourceRecord *rr) {
@@ -952,10 +949,11 @@ static int append_txt(sd_bus_message *reply, DnsResourceRecord *rr) {
         return 1;
 }
 
-static void resolve_service_all_complete(DnsQuery *q) {
+static void resolve_service_all_complete(DnsQuery *query) {
         _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *canonical = NULL;
         _cleanup_(sd_bus_message_unrefp) sd_bus_message *reply = NULL;
         _cleanup_free_ char *name = NULL, *type = NULL, *domain = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         DnsQuestion *question;
         DnsResourceRecord *rr;
         unsigned added = 0;
@@ -964,8 +962,10 @@ static void resolve_service_all_complete(DnsQuery *q) {
 
         assert(q);
 
-        if (q->block_all_complete > 0)
+        if (q->block_all_complete > 0) {
+                TAKE_PTR(q);
                 return;
+        }
 
         if ((q->flags & SD_RESOLVED_NO_ADDRESS) == 0) {
                 DnsQuery *bad = NULL;
@@ -977,6 +977,7 @@ static void resolve_service_all_complete(DnsQuery *q) {
 
                         case DNS_TRANSACTION_PENDING:
                                 /* If an auxiliary query is still pending, let's wait */
+                                TAKE_PTR(q);
                                 return;
 
                         case DNS_TRANSACTION_SUCCESS:
@@ -1093,8 +1094,6 @@ finish:
                 log_error_errno(r, "Failed to send service reply: %m");
                 sd_bus_reply_method_errno(q->bus_request, r, NULL);
         }
-
-        dns_query_free(q);
 }
 
 static void resolve_service_hostname_complete(DnsQuery *q) {
@@ -1119,7 +1118,7 @@ static void resolve_service_hostname_complete(DnsQuery *q) {
 
 static int resolve_service_hostname(DnsQuery *q, DnsResourceRecord *rr, int ifindex) {
         _cleanup_(dns_question_unrefp) DnsQuestion *question = NULL;
-        DnsQuery *aux;
+        _cleanup_(dns_query_freep) DnsQuery *aux = NULL;
         int r;
 
         assert(q);
@@ -1142,32 +1141,27 @@ static int resolve_service_hostname(DnsQuery *q, DnsResourceRecord *rr, int ifin
         aux->complete = resolve_service_hostname_complete;
 
         r = dns_query_make_auxiliary(aux, q);
-        if (r == -EAGAIN) {
+        if (r == -EAGAIN)
                 /* Too many auxiliary lookups? If so, don't complain,
                  * let's just not add this one, we already have more
                  * than enough */
-
-                dns_query_free(aux);
                 return 0;
-        }
         if (r < 0)
-                goto fail;
+                return r;
 
         /* Note that auxiliary queries do not track the original bus
          * client, only the primary request does that. */
 
         r = dns_query_go(aux);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(aux);
         return 1;
-
-fail:
-        dns_query_free(aux);
-        return r;
 }
 
-static void bus_method_resolve_service_complete(DnsQuery *q) {
+static void bus_method_resolve_service_complete(DnsQuery *query) {
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         bool has_root_domain = false;
         DnsResourceRecord *rr;
         DnsQuestion *question;
@@ -1188,8 +1182,11 @@ static void bus_method_resolve_service_complete(DnsQuery *q) {
         }
         if (r < 0)
                 goto finish;
-        if (r == DNS_QUERY_CNAME) /* This was a cname, and the query was restarted. */
+        if (r == DNS_QUERY_CNAME) {
+                /* This was a cname, and the query was restarted. */
+                TAKE_PTR(q);
                 return;
+        }
 
         question = dns_query_question_for_protocol(q, q->answer_protocol);
 
@@ -1237,7 +1234,7 @@ static void bus_method_resolve_service_complete(DnsQuery *q) {
         }
 
         /* Maybe we are already finished? check now... */
-        resolve_service_all_complete(q);
+        resolve_service_all_complete(TAKE_PTR(q));
         return;
 
 finish:
@@ -1245,17 +1242,15 @@ finish:
                 log_error_errno(r, "Failed to send service reply: %m");
                 sd_bus_reply_method_errno(q->bus_request, r, NULL);
         }
-
-        dns_query_free(q);
 }
 
 static int bus_method_resolve_service(sd_bus_message *message, void *userdata, sd_bus_error *error) {
         _cleanup_(dns_question_unrefp) DnsQuestion *question_idna = NULL, *question_utf8 = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = NULL;
         const char *name, *type, *domain;
         Manager *m = userdata;
         int family, ifindex;
         uint64_t flags;
-        DnsQuery *q;
         int r;
 
         assert(message);
@@ -1316,17 +1311,14 @@ static int bus_method_resolve_service(sd_bus_message *message, void *userdata, s
 
         r = dns_query_bus_track(q, message);
         if (r < 0)
-                goto fail;
+                return r;
 
         r = dns_query_go(q);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(q);
         return 1;
-
-fail:
-        dns_query_free(q);
-        return r;
 }
 
 int bus_dns_server_append(sd_bus_message *reply, DnsServer *s, bool with_ifindex, bool extended) {
diff --git a/src/resolve/resolved-dns-stub.c b/src/resolve/resolved-dns-stub.c
index 8040db70f9..400e741031 100644
--- a/src/resolve/resolved-dns-stub.c
+++ b/src/resolve/resolved-dns-stub.c
@@ -684,7 +684,8 @@ static int dns_stub_patch_bypass_reply_packet(
         return 0;
 }
 
-static void dns_stub_query_complete(DnsQuery *q) {
+static void dns_stub_query_complete(DnsQuery *query) {
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         int r;
 
         assert(q);
@@ -705,7 +706,6 @@ static void dns_stub_query_complete(DnsQuery *q) {
                         else
                                 (void) dns_stub_send(q->manager, q->stub_listener_extra, q->request_stream, q->request_packet, reply);
 
-                        dns_query_free(q);
                         return;
                 }
         }
@@ -717,11 +717,8 @@ static void dns_stub_query_complete(DnsQuery *q) {
                         q,
                         dns_query_question_for_protocol(q, DNS_PROTOCOL_DNS),
                         dns_stub_reply_with_edns0_do(q));
-        if (r < 0) {
-                log_debug_errno(r, "Failed to assign sections: %m");
-                dns_query_free(q);
-                return;
-        }
+        if (r < 0)
+                return (void) log_debug_errno(r, "Failed to assign sections: %m");
 
         switch (q->state) {
 
@@ -755,11 +752,10 @@ static void dns_stub_query_complete(DnsQuery *q) {
                                  * packet doesn't answer our question. In that case let's restart the query,
                                  * now with the redirected question. We'll */
                                 r = dns_query_go(q);
-                                if (r < 0) {
-                                        log_debug_errno(r, "Failed to restart query: %m");
-                                        dns_query_free(q);
-                                }
+                                if (r < 0)
+                                        return (void) log_debug_errno(r, "Failed to restart query: %m");
 
+                                TAKE_PTR(q);
                                 return;
                         }
 
@@ -767,11 +763,8 @@ static void dns_stub_query_complete(DnsQuery *q) {
                                         q,
                                         dns_query_question_for_protocol(q, DNS_PROTOCOL_DNS),
                                         dns_stub_reply_with_edns0_do(q));
-                        if (r < 0) {
-                                log_debug_errno(r, "Failed to assign sections: %m");
-                                dns_query_free(q);
-                                return;
-                        }
+                        if (r < 0)
+                                return (void) log_debug_errno(r, "Failed to assign sections: %m");
 
                         if (cname_result == DNS_QUERY_MATCH) /* A match? Then we are done, let's return what we got */
                                 break;
@@ -817,8 +810,6 @@ static void dns_stub_query_complete(DnsQuery *q) {
         default:
                 assert_not_reached("Impossible state");
         }
-
-        dns_query_free(q);
 }
 
 static int dns_stub_stream_complete(DnsStream *s, int error) {
diff --git a/src/resolve/resolved-varlink.c b/src/resolve/resolved-varlink.c
index 27d8c8967e..6f4ab425ef 100644
--- a/src/resolve/resolved-varlink.c
+++ b/src/resolve/resolved-varlink.c
@@ -143,9 +143,10 @@ static bool validate_and_mangle_flags(
         return true;
 }
 
-static void vl_method_resolve_hostname_complete(DnsQuery *q) {
+static void vl_method_resolve_hostname_complete(DnsQuery *query) {
         _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *canonical = NULL;
         _cleanup_(json_variant_unrefp) JsonVariant *array = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         _cleanup_free_ char *normalized = NULL;
         DnsResourceRecord *rr;
         DnsQuestion *question;
@@ -165,8 +166,11 @@ static void vl_method_resolve_hostname_complete(DnsQuery *q) {
         }
         if (r < 0)
                 goto finish;
-        if (r == DNS_QUERY_CNAME) /* This was a cname, and the query was restarted. */
+        if (r == DNS_QUERY_CNAME) {
+                /* This was a cname, and the query was restarted. */
+                TAKE_PTR(q);
                 return;
+        }
 
         question = dns_query_question_for_protocol(q, q->answer_protocol);
 
@@ -228,8 +232,6 @@ finish:
                 log_error_errno(r, "Failed to send hostname reply: %m");
                 r = varlink_error_errno(q->varlink_request, r);
         }
-
-        dns_query_free(q);
 }
 
 static int parse_as_address(Varlink *link, LookupParameters *p) {
@@ -284,7 +286,7 @@ static int vl_method_resolve_hostname(Varlink *link, JsonVariant *parameters, Va
         _cleanup_(lookup_parameters_destroy) LookupParameters p = {
                 .family = AF_UNSPEC,
         };
-        DnsQuery *q;
+        _cleanup_(dns_query_freep) DnsQuery *q = NULL;
         Manager *m;
         int r;
 
@@ -338,13 +340,10 @@ static int vl_method_resolve_hostname(Varlink *link, JsonVariant *parameters, Va
 
         r = dns_query_go(q);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(q);
         return 1;
-
-fail:
-        dns_query_free(q);
-        return r;
 }
 
 static int json_dispatch_address(const char *name, JsonVariant *variant, JsonDispatchFlags flags, void *userdata) {
@@ -382,8 +381,9 @@ static int json_dispatch_address(const char *name, JsonVariant *variant, JsonDis
         return 0;
 }
 
-static void vl_method_resolve_address_complete(DnsQuery *q) {
+static void vl_method_resolve_address_complete(DnsQuery *query) {
         _cleanup_(json_variant_unrefp) JsonVariant *array = NULL;
+        _cleanup_(dns_query_freep) DnsQuery *q = query;
         DnsQuestion *question;
         DnsResourceRecord *rr;
         int ifindex, r;
@@ -402,8 +402,11 @@ static void vl_method_resolve_address_complete(DnsQuery *q) {
         }
         if (r < 0)
                 goto finish;
-        if (r == DNS_QUERY_CNAME) /* This was a cname, and the query was restarted. */
+        if (r == DNS_QUERY_CNAME) {
+                /* This was a cname, and the query was restarted. */
+                TAKE_PTR(q);
                 return;
+        }
 
         question = dns_query_question_for_protocol(q, q->answer_protocol);
 
@@ -447,8 +450,6 @@ finish:
                 log_error_errno(r, "Failed to send address reply: %m");
                 r = varlink_error_errno(q->varlink_request, r);
         }
-
-        dns_query_free(q);
 }
 
 static int vl_method_resolve_address(Varlink *link, JsonVariant *parameters, VarlinkMethodFlags flags, void *userdata) {
@@ -464,7 +465,7 @@ static int vl_method_resolve_address(Varlink *link, JsonVariant *parameters, Var
         _cleanup_(lookup_parameters_destroy) LookupParameters p = {
                 .family = AF_UNSPEC,
         };
-        DnsQuery *q;
+        _cleanup_(dns_query_freep) DnsQuery *q = NULL;
         Manager *m;
         int r;
 
@@ -509,13 +510,10 @@ static int vl_method_resolve_address(Varlink *link, JsonVariant *parameters, Var
 
         r = dns_query_go(q);
         if (r < 0)
-                goto fail;
+                return r;
 
+        TAKE_PTR(q);
         return 1;
-
-fail:
-        dns_query_free(q);
-        return r;
 }
 
 int manager_varlink_init(Manager *m) {
-- 
2.33.0