]> git.meshlink.io Git - meshlink/blob - src/adns.c
Send RST packets when receiving data after we closed a UDP channel.
[meshlink] / src / adns.c
1 /*
2     dns.c -- hostname resolving functions
3     Copyright (C) 2019 Guus Sliepen <guus@meshlink.io>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "system.h"
21
22 #include <pthread.h>
23
24 #include "adns.h"
25 #include "devtools.h"
26 #include "logger.h"
27 #include "xalloc.h"
28
29 typedef struct adns_item {
30         adns_cb_t cb;
31         void *data;
32         time_t deadline;
33         struct addrinfo *ai;
34         int err;
35         char *host;
36         char *serv;
37 } adns_item_t;
38
39 static void *adns_loop(void *data) {
40         meshlink_handle_t *mesh = data;
41
42         while(true) {
43                 adns_item_t *item = meshlink_queue_pop_cond(&mesh->adns_queue, &mesh->adns_cond);
44
45                 if(!item) {
46                         break;
47                 }
48
49                 if(time(NULL) < item->deadline) {
50                         logger(mesh, MESHLINK_DEBUG, "Resolving %s port %s", item->host, item->serv);
51                         devtool_adns_resolve_probe();
52                         int result = getaddrinfo(item->host, item->serv, NULL, &item->ai);
53
54                         if(result) {
55                                 item->ai = NULL;
56                                 item->err = errno;
57                         }
58                 } else {
59                         logger(mesh, MESHLINK_WARNING, "Deadline passed for DNS request %s port %s", item->host, item->serv);
60                         item->ai = NULL;
61                         item->err = ETIMEDOUT;
62                 }
63
64                 if(meshlink_queue_push(&mesh->adns_done_queue, item)) {
65                         signal_trigger(&mesh->loop, &mesh->adns_signal);
66                 } else {
67                         free(item->host);
68                         free(item->serv);
69                         free(item);
70                 }
71         }
72
73         return NULL;
74 }
75
76 static void adns_cb_handler(event_loop_t *loop, void *data) {
77         (void)loop;
78         meshlink_handle_t *mesh = data;
79
80         for(adns_item_t *item; (item = meshlink_queue_pop(&mesh->adns_done_queue));) {
81                 item->cb(mesh, item->host, item->serv, item->data, item->ai, item->err);
82                 free(item);
83         }
84 }
85
86 void init_adns(meshlink_handle_t *mesh) {
87         meshlink_queue_init(&mesh->adns_queue);
88         meshlink_queue_init(&mesh->adns_done_queue);
89         signal_add(&mesh->loop, &mesh->adns_signal, adns_cb_handler, mesh, 1);
90         pthread_create(&mesh->adns_thread, NULL, adns_loop, mesh);
91 }
92
93 void exit_adns(meshlink_handle_t *mesh) {
94         if(!mesh->adns_signal.cb) {
95                 return;
96         }
97
98         /* Drain the queue of any pending ADNS requests */
99         for(adns_item_t *item; (item = meshlink_queue_pop(&mesh->adns_queue));) {
100                 free(item->host);
101                 free(item->serv);
102                 free(item);
103         }
104
105         /* Signal the ADNS thread to stop */
106         if(!meshlink_queue_push(&mesh->adns_queue, NULL)) {
107                 abort();
108         }
109
110         pthread_cond_signal(&mesh->adns_cond);
111
112         pthread_join(mesh->adns_thread, NULL);
113         meshlink_queue_exit(&mesh->adns_queue);
114         signal_del(&mesh->loop, &mesh->adns_signal);
115 }
116
117 void adns_queue(meshlink_handle_t *mesh, char *host, char *serv, adns_cb_t cb, void *data, int timeout) {
118         adns_item_t *item = xmalloc(sizeof(*item));
119         item->cb = cb;
120         item->data = data;
121         item->deadline = time(NULL) + timeout;
122         item->host = host;
123         item->serv = serv;
124
125         logger(mesh, MESHLINK_DEBUG, "Enqueueing DNS request for %s port %s", item->host, item->serv);
126
127         if(!meshlink_queue_push(&mesh->adns_queue, item)) {
128                 abort();
129         }
130
131         pthread_cond_signal(&mesh->adns_cond);
132 }
133
134 struct adns_blocking_info {
135         meshlink_handle_t *mesh;
136         pthread_mutex_t mutex;
137         pthread_cond_t cond;
138         char *host;
139         char *serv;
140         struct addrinfo *ai;
141         int socktype;
142         bool done;
143 };
144
145 static void *adns_blocking_handler(void *data) {
146         struct adns_blocking_info *info = data;
147
148         logger(info->mesh, MESHLINK_DEBUG, "Resolving %s port %s", info->host, info->serv);
149         devtool_adns_resolve_probe();
150
151         struct addrinfo hint = {
152                 .ai_family = AF_UNSPEC,
153                 .ai_socktype = info->socktype,
154         };
155
156         if(getaddrinfo(info->host, info->serv, &hint, &info->ai)) {
157                 info->ai = NULL;
158         }
159
160         if(pthread_mutex_lock(&info->mutex) != 0) {
161                 abort();
162         }
163
164         bool cleanup = info->done;
165
166         if(!info->done) {
167                 info->done = true;
168                 pthread_cond_signal(&info->cond);
169         }
170
171         pthread_mutex_unlock(&info->mutex);
172
173         if(cleanup) {
174                 free(info->host);
175                 free(info->serv);
176                 free(info);
177         }
178
179         return NULL;
180 }
181
182 struct addrinfo *adns_blocking_request(meshlink_handle_t *mesh, char *host, char *serv, int socktype, int timeout) {
183         struct adns_blocking_info *info = xzalloc(sizeof(*info));
184
185         info->mesh = mesh;
186         info->host = host;
187         info->serv = serv;
188         info->socktype = socktype;
189         pthread_mutex_init(&info->mutex, NULL);
190         pthread_cond_init(&info->cond, NULL);
191
192         struct timespec deadline;
193         clock_gettime(CLOCK_REALTIME, &deadline);
194         deadline.tv_sec += timeout;
195
196         pthread_t thread;
197
198         if(pthread_create(&thread, NULL, adns_blocking_handler, info)) {
199                 free(info->host);
200                 free(info->serv);
201                 free(info);
202                 return NULL;
203         } else {
204                 pthread_detach(thread);
205         }
206
207         if(pthread_mutex_lock(&info->mutex) != 0) {
208                 abort();
209         }
210
211         pthread_cond_timedwait(&info->cond, &info->mutex, &deadline);
212
213         struct addrinfo *result = NULL;
214         bool cleanup = info->done;
215
216         if(info->done) {
217                 result = info->ai;
218         } else {
219                 logger(mesh, MESHLINK_WARNING, "Deadline passed for DNS request %s port %s", host, serv);
220                 info->done = true;
221         }
222
223         pthread_mutex_unlock(&info->mutex);
224
225         if(cleanup) {
226                 free(info->host);
227                 free(info->serv);
228                 free(info);
229         }
230
231         return result;
232 }