]>
Commit | Line | Data |
---|---|---|
52ad194e | 1 | /* SPDX-License-Identifier: LGPL-2.1+ */ |
5eef597e MP |
2 | #pragma once |
3 | ||
5eef597e MP |
4 | #include "socket-util.h" |
5 | ||
6 | typedef struct DnsStream DnsStream; | |
7 | ||
8 | #include "resolved-dns-packet.h" | |
9 | #include "resolved-dns-transaction.h" | |
8a584da2 | 10 | #include "resolved-manager.h" |
b012e921 | 11 | #if ENABLE_DNS_OVER_TLS |
6e866b33 | 12 | #include "resolved-dnstls.h" |
b012e921 MB |
13 | #endif |
14 | ||
6e866b33 MB |
15 | #define DNS_STREAM_WRITE_TLS_DATA 1 |
16 | ||
5a920b42 MP |
17 | /* Streams are used by three subsystems: |
18 | * | |
19 | * 1. The normal transaction logic when doing a DNS or LLMNR lookup via TCP | |
20 | * 2. The LLMNR logic when accepting a TCP-based lookup | |
21 | * 3. The DNS stub logic when accepting a TCP-based lookup | |
22 | */ | |
23 | ||
5eef597e MP |
24 | struct DnsStream { |
25 | Manager *manager; | |
6e866b33 | 26 | unsigned n_ref; |
5eef597e MP |
27 | |
28 | DnsProtocol protocol; | |
29 | ||
30 | int fd; | |
31 | union sockaddr_union peer; | |
32 | socklen_t peer_salen; | |
33 | union sockaddr_union local; | |
34 | socklen_t local_salen; | |
35 | int ifindex; | |
36 | uint32_t ttl; | |
37 | bool identified; | |
38 | ||
b012e921 MB |
39 | /* only when using TCP fast open */ |
40 | union sockaddr_union tfo_address; | |
41 | socklen_t tfo_salen; | |
42 | ||
43 | #if ENABLE_DNS_OVER_TLS | |
6e866b33 MB |
44 | DnsTlsStreamData dnstls_data; |
45 | int dnstls_events; | |
b012e921 MB |
46 | #endif |
47 | ||
5eef597e MP |
48 | sd_event_source *io_event_source; |
49 | sd_event_source *timeout_event_source; | |
50 | ||
51 | be16_t write_size, read_size; | |
52 | DnsPacket *write_packet, *read_packet; | |
53 | size_t n_written, n_read; | |
b012e921 | 54 | OrderedSet *write_queue; |
5eef597e MP |
55 | |
56 | int (*on_packet)(DnsStream *s); | |
57 | int (*complete)(DnsStream *s, int error); | |
58 | ||
b012e921 MB |
59 | LIST_HEAD(DnsTransaction, transactions); /* when used by the transaction logic */ |
60 | DnsServer *server; /* when used by the transaction logic */ | |
6e866b33 | 61 | DnsQuery *query; /* when used by the DNS stub logic */ |
5eef597e | 62 | |
b012e921 MB |
63 | /* used when DNS-over-TLS is enabled */ |
64 | bool encrypted:1; | |
65 | ||
5eef597e MP |
66 | LIST_FIELDS(DnsStream, streams); |
67 | }; | |
68 | ||
b012e921 MB |
69 | int dns_stream_new(Manager *m, DnsStream **s, DnsProtocol protocol, int fd, const union sockaddr_union *tfo_address); |
70 | #if ENABLE_DNS_OVER_TLS | |
6e866b33 | 71 | int dns_stream_connect_tls(DnsStream *s, void *tls_session); |
b012e921 | 72 | #endif |
5a920b42 MP |
73 | DnsStream *dns_stream_unref(DnsStream *s); |
74 | DnsStream *dns_stream_ref(DnsStream *s); | |
5eef597e | 75 | |
b012e921 MB |
76 | DEFINE_TRIVIAL_CLEANUP_FUNC(DnsStream*, dns_stream_unref); |
77 | ||
5eef597e | 78 | int dns_stream_write_packet(DnsStream *s, DnsPacket *p); |
6e866b33 | 79 | ssize_t dns_stream_writev(DnsStream *s, const struct iovec *iov, size_t iovcnt, int flags); |
5a920b42 MP |
80 | |
81 | static inline bool DNS_STREAM_QUEUED(DnsStream *s) { | |
82 | assert(s); | |
83 | ||
84 | if (s->fd < 0) /* already stopped? */ | |
85 | return false; | |
86 | ||
87 | return !!s->write_packet; | |
88 | } | |
6e866b33 MB |
89 | |
90 | DnsPacket *dns_stream_take_read_packet(DnsStream *s); | |
91 | ||
92 | void dns_stream_detach(DnsStream *s); |