481369e203c85da3790d0a7b54c956d95d005b29
[connman] / src / resolver.c
1 /*
2  *
3  *  Connection Manager
4  *
5  *  Copyright (C) 2007-2009  Intel Corporation. All rights reserved.
6  *
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License version 2 as
9  *  published by the Free Software Foundation.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program; if not, write to the Free Software
18  *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19  *
20  */
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <stdio.h>
27 #include <errno.h>
28 #include <fcntl.h>
29 #include <unistd.h>
30 #include <string.h>
31 #include <sys/stat.h>
32
33 #include "connman.h"
34
35 struct entry_data {
36         struct connman_resolver *resolver;
37         char *interface;
38         char *domain;
39         char *server;
40 };
41
42 static GSList *entry_list = NULL;
43 static GSList *resolver_list = NULL;
44
45 static void remove_entries(GSList *entries)
46 {
47         GSList *list;
48
49         for (list = entries; list; list = list->next) {
50                 struct entry_data *entry = list->data;
51                 struct connman_resolver *resolver = entry->resolver;
52
53                 entry_list = g_slist_remove(entry_list, entry);
54
55                 if (resolver->remove)
56                         resolver->remove(entry->interface, entry->domain,
57                                                                 entry->server);
58
59                 g_free(entry->server);
60                 g_free(entry->domain);
61                 g_free(entry->interface);
62                 g_free(entry);
63         }
64
65         g_slist_free(entries);
66 }
67
68 static gint compare_priority(gconstpointer a, gconstpointer b)
69 {
70         const struct connman_resolver *resolver1 = a;
71         const struct connman_resolver *resolver2 = b;
72
73         return resolver2->priority - resolver1->priority;
74 }
75
76 /**
77  * connman_resolver_register:
78  * @resolver: resolver module
79  *
80  * Register a new resolver module
81  *
82  * Returns: %0 on success
83  */
84 int connman_resolver_register(struct connman_resolver *resolver)
85 {
86         GSList *list;
87
88         DBG("resolver %p name %s", resolver, resolver->name);
89
90         resolver_list = g_slist_insert_sorted(resolver_list, resolver,
91                                                         compare_priority);
92
93         if (resolver->append == NULL)
94                 return 0;
95
96         for (list = entry_list; list; list = list->next) {
97                 struct entry_data *entry = list->data;
98
99                 if (entry->resolver)
100                         continue;
101
102                 if (resolver->append(entry->interface, entry->domain,
103                                                         entry->server) == 0)
104                         entry->resolver = resolver;
105         }
106
107         return 0;
108 }
109
110 /**
111  * connman_resolver_unregister:
112  * @resolver: resolver module
113  *
114  * Remove a previously registered resolver module
115  */
116 void connman_resolver_unregister(struct connman_resolver *resolver)
117 {
118         GSList *list, *matches = NULL;
119
120         DBG("resolver %p name %s", resolver, resolver->name);
121
122         resolver_list = g_slist_remove(resolver_list, resolver);
123
124         for (list = entry_list; list; list = list->next) {
125                 struct entry_data *entry = list->data;
126
127                 if (entry->resolver != resolver)
128                         continue;
129
130                 matches = g_slist_append(matches, entry);
131         }
132
133         remove_entries(matches);
134 }
135
136 /**
137  * connman_resolver_append:
138  * @interface: network interface
139  * @domain: domain limitation
140  * @server: server address
141  *
142  * Append resolver server address to current list
143  */
144 int connman_resolver_append(const char *interface, const char *domain,
145                                                         const char *server)
146 {
147         struct entry_data *entry;
148         GSList *list;
149
150         DBG("interface %s domain %s server %s", interface, domain, server);
151
152         entry = g_try_new0(struct entry_data, 1);
153         if (entry == NULL)
154                 return -ENOMEM;
155
156         entry->interface = g_strdup(interface);
157         entry->domain = g_strdup(domain);
158         entry->server = g_strdup(server);
159
160         entry_list = g_slist_append(entry_list, entry);
161
162         for (list = resolver_list; list; list = list->next) {
163                 struct connman_resolver *resolver = list->data;
164
165                 if (resolver->append == NULL)
166                         continue;
167
168                 if (resolver->append(interface, domain, server) == 0) {
169                         entry->resolver = resolver;
170                         break;
171                 }
172         }
173
174         return 0;
175 }
176
177 /**
178  * connman_resolver_remove_all:
179  * @interface: network interface
180  *
181  * Remove all resolver server address for the specified interface
182  */
183 int connman_resolver_remove_all(const char *interface)
184 {
185         GSList *list, *matches = NULL;
186
187         DBG("interface %s", interface);
188
189         for (list = entry_list; list; list = list->next) {
190                 struct entry_data *entry = list->data;
191
192                 if (g_str_equal(entry->interface, interface) == FALSE)
193                         continue;
194
195                 matches = g_slist_append(matches, entry);
196         }
197
198         remove_entries(matches);
199
200         return 0;
201 }
202
203 static int selftest_append(const char *interface, const char *domain,
204                                                         const char *server)
205 {
206         DBG("server %s", server);
207
208         return 0;
209 }
210
211 static int selftest_remove(const char *interface, const char *domain,
212                                                         const char *server)
213 {
214         DBG("server %s", server);
215
216         return 0;
217 }
218
219 static struct connman_resolver selftest_resolver = {
220         .name     = "selftest",
221         .priority = CONNMAN_RESOLVER_PRIORITY_HIGH + 42,
222         .append   = selftest_append,
223         .remove   = selftest_remove,
224 };
225
226 int __connman_resolver_selftest(void)
227 {
228         connman_resolver_append("wlan0", "lwn.net", "192.168.0.1");
229
230         connman_resolver_register(&selftest_resolver);
231
232         connman_resolver_append("eth0", "moblin.org", "192.168.42.1");
233         connman_resolver_append("wlan0", "lwn.net", "192.168.0.2");
234
235         connman_resolver_remove_all("wlan0");
236
237         connman_resolver_unregister(&selftest_resolver);
238
239         return 0;
240 }
241
242 static int resolvfile_append(const char *interface, const char *domain,
243                                                         const char *server)
244 {
245         char *cmd;
246         int fd, len, err;
247
248         DBG("interface %s server %s", interface, server);
249
250         fd = open("/etc/resolv.conf", O_RDWR | O_CREAT,
251                                         S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
252         if (fd < 0)
253                 return -errno;
254
255         err = ftruncate(fd, 0);
256
257         cmd = g_strdup_printf("# Generated by Connection Manager\n"
258                                                 "domain localdomain\n"
259                                                 "search localdomain\n"
260                                                 "nameserver %s\n", server);
261
262         len = write(fd, cmd, strlen(cmd));
263
264         g_free(cmd);
265
266         close(fd);
267
268         return 0;
269 }
270
271 static int resolvfile_remove(const char *interface, const char *domain,
272                                                         const char *server)
273 {
274         DBG("interface %s server %s", interface, server);
275
276         return 0;
277 }
278
279 static struct connman_resolver resolvfile_resolver = {
280         .name           = "resolvfile",
281         .priority       = CONNMAN_RESOLVER_PRIORITY_LOW,
282         .append         = resolvfile_append,
283         .remove         = resolvfile_remove,
284 };
285
286 int __connman_resolver_init(void)
287 {
288         DBG("");
289
290         return connman_resolver_register(&resolvfile_resolver);
291 }
292
293 void __connman_resolver_cleanup(void)
294 {
295         DBG("");
296
297         connman_resolver_unregister(&resolvfile_resolver);
298 }