src/resolve/resolved-dns-query.h

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 #pragma once
   4
   5 /***
   6   This file is part of systemd.
   7
   8   Copyright 2014 Lennart Poettering
   9
  10   systemd is free software; you can redistribute it and/or modify it
  11   under the terms of the GNU Lesser General Public License as published by
  12   the Free Software Foundation; either version 2.1 of the License, or
  13   (at your option) any later version.
  14
  15   systemd is distributed in the hope that it will be useful, but
  16   WITHOUT ANY WARRANTY; without even the implied warranty of
  17   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18   Lesser General Public License for more details.
  19
  20   You should have received a copy of the GNU Lesser General Public License
  21   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  22 ***/
  23
  24 #include <inttypes.h>
  25
  26 #include "sd-bus.h"
  27 #include "util.h"
  28
  29 typedef struct DnsQuery DnsQuery;
  30 typedef struct DnsQueryTransaction DnsQueryTransaction;
  31
  32 #include "resolved.h"
  33 #include "resolved-dns-scope.h"
  34 #include "resolved-dns-rr.h"
  35 #include "resolved-dns-packet.h"
  36
  37 typedef enum DnsQueryState {
  38         DNS_QUERY_NULL,
  39         DNS_QUERY_PENDING,
  40         DNS_QUERY_FAILURE,
  41         DNS_QUERY_SUCCESS,
  42         DNS_QUERY_NO_SERVERS,
  43         DNS_QUERY_TIMEOUT,
  44         DNS_QUERY_ATTEMPTS_MAX,
  45         DNS_QUERY_INVALID_REPLY,
  46         DNS_QUERY_RESOURCES
  47 } DnsQueryState;
  48
  49 struct DnsQueryTransaction {
  50         DnsQuery *query;
  51         DnsScope *scope;
  52
  53         DnsQueryState state;
  54         uint16_t id;
  55
  56         sd_event_source *timeout_event_source;
  57         unsigned n_attempts;
  58
  59         DnsPacket *sent, *received;
  60
  61         /* TCP connection logic */
  62         int tcp_fd;
  63         sd_event_source *tcp_event_source;
  64         size_t tcp_written, tcp_read;
  65         be16_t tcp_read_size;
  66
  67         /* Data from cache */
  68         DnsResourceRecord **cached_rrs;
  69         unsigned n_cached_rrs;
  70
  71         LIST_FIELDS(DnsQueryTransaction, transactions_by_query);
  72         LIST_FIELDS(DnsQueryTransaction, transactions_by_scope);
  73 };
  74
  75 struct DnsQuery {
  76         Manager *manager;
  77
  78         DnsResourceKey *keys;
  79         unsigned n_keys;
  80
  81         DnsQueryState state;
  82         unsigned n_cname;
  83
  84         sd_event_source *timeout_event_source;
  85
  86         /* Discovered data */
  87         DnsPacket *received;
  88         DnsResourceRecord **cached_rrs;
  89         unsigned n_cached_rrs;
  90
  91         /* Bus client information */
  92         sd_bus_message *request;
  93         int request_family;
  94         const char *request_hostname;
  95         union in_addr_union request_address;
  96
  97         /* Completion callback */
  98         void (*complete)(DnsQuery* q);
  99         unsigned block_finish;
 100
 101         LIST_HEAD(DnsQueryTransaction, transactions);
 102         LIST_FIELDS(DnsQuery, queries);
 103 };
 104
 105 DnsQueryTransaction* dns_query_transaction_free(DnsQueryTransaction *t);
 106 void dns_query_transaction_reply(DnsQueryTransaction *t, DnsPacket *p);
 107
 108 int dns_query_new(Manager *m, DnsQuery **q, DnsResourceKey *keys, unsigned n_keys);
 109 DnsQuery *dns_query_free(DnsQuery *q);
 110
 111 int dns_query_go(DnsQuery *q);
 112 int dns_query_cname_redirect(DnsQuery *q, const char *name);
 113 void dns_query_finish(DnsQuery *q);
 114
 115 int dns_query_matches_rr(DnsQuery *q, DnsResourceRecord *rr);
 116 int dns_query_matches_cname(DnsQuery *q, DnsResourceRecord *rr);
 117
 118 /* What we found */
 119 int dns_query_get_rrs(DnsQuery *q, DnsResourceRecord *** rrs);
 120 int dns_query_get_rcode(DnsQuery *q);
 121 int dns_query_get_ifindex(DnsQuery *q);
 122
 123 DEFINE_TRIVIAL_CLEANUP_FUNC(DnsQuery*, dns_query_free);