keys, dns: Use kmalloc_flex to improve dns_resolver_preparse

Use kmalloc_flex() when allocating a new 'struct user_key_payload' in
dns_resolver_preparse() to replace the open-coded size arithmetic.

Signed-off-by: Thorsten Blum <thorsten.blum@linux.dev>
Link: https://patch.msgid.link/20260226214930.785423-3-thorsten.blum@linux.dev
Signed-off-by: Jakub Kicinski <kuba@kernel.org>

diff --git a/net/dns_resolver/dns_key.c b/net/dns_resolver/dns_key.c
index c42ddd8..891287a 100644 (file)
--- a/net/dns_resolver/dns_key.c
+++ b/net/dns_resolver/dns_key.c
@@ -215,7 +215,7 @@ store_result:
        kdebug("store result");
        prep->quotalen = result_len;
 
-       upayload = kmalloc(sizeof(*upayload) + result_len + 1, GFP_KERNEL);
+       upayload = kmalloc_flex(*upayload, data, result_len + 1);
        if (!upayload) {
                kleave(" = -ENOMEM");
                return -ENOMEM;