1 Star 0 Fork 91

xuxinyu / systemd

forked from src-openEuler / systemd 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
backport-resolve-use-_cleanup_-attribute-for-freeing-DnsQuery.patch 19.68 KB
一键复制 编辑 原始数据 按行查看 历史
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563
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
1
https://gitee.com/xuiny/systemd.git
git@gitee.com:xuiny/systemd.git
xuiny
systemd
systemd
master

搜索帮助