1 #include <sys/socket.h>
9 static gint timeval_cmp(const GTimeVal *a, const GTimeVal *b) {
13 if (a->tv_sec < b->tv_sec)
16 if (a->tv_sec > b->tv_sec)
19 if (a->tv_usec < b->tv_usec)
22 if (a->tv_usec > b->tv_usec)
28 static gint query_job_instance_compare(gpointer a, gpointer b) {
29 flxQueryJobInstance *j = a, *k = b;
33 return timeval_cmp(&j->job->time, &k->job->time);
36 static gint response_job_instance_compare(gpointer a, gpointer b) {
37 flxResponseJobInstance *j = a, *k = b;
41 return timeval_cmp(&j->job->time, &k->job->time);
44 flxServer *flx_server_new(GMainContext *c) {
45 flxServer *s = g_new(flxServer, 1);
48 g_main_context_ref(c);
51 s->context = g_main_context_default();
54 s->rrset_by_id = g_hash_table_new(g_int_hash, g_int_equal);
55 s->rrset_by_name = g_hash_table_new(g_str_hash, g_str_equal);
58 s->query_job_queue = flx_prio_queue_new(query_job_instance_compare);
59 s->response_job_queue = flx_prio_queue_new(response_job_instance_compare);
61 s->monitor = flx_interface_monitor_new(s->context);
66 void flx_server_free(flxServer* s) {
69 while (s->query_job_queue->last)
70 flx_server_remove_query_job_instance(s, s->query_job_queue->last->data);
72 flx_prio_queue_free(s->query_job_queue);
73 flx_prio_queue_free(s->response_job_queue);
75 flx_interface_monitor_free(s->monitor);
77 flx_server_remove(s, 0);
79 g_hash_table_destroy(s->rrset_by_id);
80 g_hash_table_destroy(s->rrset_by_name);
81 g_main_context_unref(s->context);
85 gint flx_server_get_next_id(flxServer *s) {
88 return s->current_id++;
91 void flx_server_add_rr(flxServer *s, gint id, gint interface, guchar protocol, const flxRecord *rr) {
99 e = g_new(flxEntry, 1);
100 flx_record_copy_normalize(&e->rr, rr);
102 e->interface = interface;
103 e->protocol = protocol;
105 /* Insert into linked list */
107 if ((e->next = s->entries))
111 /* Insert into hash table indexed by id */
112 e->prev_by_id = NULL;
113 if ((e->next_by_id = g_hash_table_lookup(s->rrset_by_id, &id)))
114 e->next_by_id->prev = e;
115 g_hash_table_replace(s->rrset_by_id, &e->id, e);
117 /* Insert into hash table indexed by name */
118 e->prev_by_name = NULL;
119 if ((e->next_by_name = g_hash_table_lookup(s->rrset_by_name, e->rr.name)))
120 e->next_by_name->prev = e;
121 g_hash_table_replace(s->rrset_by_name, e->rr.name, e);
124 void flx_server_add(flxServer *s, gint id, gint interface, guchar protocol, const gchar *name, guint16 type, gconstpointer data, guint size) {
131 rr.name = (gchar*) name;
133 rr.class = FLX_DNS_CLASS_IN;
134 rr.data = (gpointer) data;
136 rr.ttl = FLX_DEFAULT_TTL;
137 flx_server_add_rr(s, id, interface, protocol, &rr);
140 const flxRecord *flx_server_iterate(flxServer *s, gint id, void **state) {
141 flxEntry **e = (flxEntry**) state;
146 *e = id > 0 ? (*e)->next_by_id : (*e)->next;
148 *e = id > 0 ? g_hash_table_lookup(s->rrset_by_id, &id) : s->entries;
156 static void free_entry(flxServer*s, flxEntry *e) {
159 /* Remove from linked list */
161 e->prev->next = e->next;
163 s->entries = e->next;
166 e->next->prev = e->prev;
168 /* Remove from hash table indexed by id */
170 e->prev_by_id = e->next_by_id;
173 g_hash_table_replace(s->rrset_by_id, &e->next_by_id->id, e->next_by_id);
175 g_hash_table_remove(s->rrset_by_id, &e->id);
179 e->next_by_id->prev_by_id = e->prev_by_id;
181 /* Remove from hash table indexed by name */
183 e->prev_by_name = e->next_by_name;
186 g_hash_table_replace(s->rrset_by_name, &e->next_by_name->rr.name, e->next_by_name);
188 g_hash_table_remove(s->rrset_by_name, &e->rr.name);
192 e->next_by_name->prev_by_name = e->prev_by_name;
195 void flx_server_remove(flxServer *s, gint id) {
200 free_entry(s, s->entries);
204 while ((e = g_hash_table_lookup(s->rrset_by_id, &id)))
209 flxRecord *flx_record_copy_normalize(flxRecord *ret_dest, const flxRecord*src) {
214 ret_dest->name = flx_normalize_name(src->name);
215 ret_dest->data = g_memdup(src->data, src->size);
220 static const gchar *dns_class_to_string(guint16 class) {
221 if (class == FLX_DNS_CLASS_IN)
227 static const gchar *dns_type_to_string(guint16 type) {
231 case FLX_DNS_TYPE_AAAA:
233 case FLX_DNS_TYPE_PTR:
235 case FLX_DNS_TYPE_HINFO:
237 case FLX_DNS_TYPE_TXT:
244 void flx_server_dump(flxServer *s, FILE *f) {
249 for (e = s->entries; e; e = e->next) {
251 fprintf(f, "%i.%u: %-40s %-8s %-8s ", e->interface, e->protocol, e->rr.name, dns_class_to_string(e->rr.class), dns_type_to_string(e->rr.type));
255 if (e->rr.class == FLX_DNS_CLASS_IN) {
256 if (e->rr.type == FLX_DNS_TYPE_A)
257 inet_ntop(AF_INET, e->rr.data, t, sizeof(t));
258 else if (e->rr.type == FLX_DNS_TYPE_AAAA)
259 inet_ntop(AF_INET6, e->rr.data, t, sizeof(t));
260 else if (e->rr.type == FLX_DNS_TYPE_PTR)
261 g_strlcpy(t, e->rr.data, sizeof(t));
262 else if (e->rr.type == FLX_DNS_TYPE_HINFO) {
265 if ((s2 = memchr(e->rr.data, 0, e->rr.size))) {
267 if (memchr(s2, 0, e->rr.size - ((char*) s2 - (char*) e->rr.data)))
268 snprintf(t, sizeof(t), "'%s' '%s'", (char*) e->rr.data, s2);
274 fprintf(f, "%s\n", t);
278 void flx_server_add_address(flxServer *s, gint id, gint interface, guchar protocol, const gchar *name, flxAddress *a) {
284 n = flx_normalize_name(name);
286 if (a->family == AF_INET) {
289 flx_server_add(s, id, interface, protocol, n, FLX_DNS_TYPE_A, &a->ipv4, sizeof(a->ipv4));
291 r = flx_reverse_lookup_name_ipv4(&a->ipv4);
293 flx_server_add(s, id, interface, protocol, r, FLX_DNS_TYPE_PTR, n, strlen(n)+1);
299 flx_server_add(s, id, interface, protocol, n, FLX_DNS_TYPE_AAAA, &a->ipv6, sizeof(a->ipv6));
301 r = flx_reverse_lookup_name_ipv6_arpa(&a->ipv6);
303 flx_server_add(s, id, interface, protocol, r, FLX_DNS_TYPE_PTR, n, strlen(n)+1);
306 r = flx_reverse_lookup_name_ipv6_int(&a->ipv6);
308 flx_server_add(s, id, interface, protocol, r, FLX_DNS_TYPE_PTR, n, strlen(n)+1);
315 flxQueryJob* flx_query_job_new(void) {
316 flxQueryJob *job = g_new(flxQueryJob, 1);
317 job->query.name = NULL;
318 job->query.class = 0;
321 job->time.tv_sec = 0;
322 job->time.tv_usec = 0;
326 flxQueryJob* flx_query_job_ref(flxQueryJob *job) {
328 g_assert(job->ref >= 1);
333 void flx_query_job_unref(flxQueryJob *job) {
335 g_assert(job->ref >= 1);
340 static gboolean query_job_exists(flxServer *s, gint interface, guchar protocol, flxQuery *q) {
345 for (n = s->query_job_queue->root; n; n = n->next)
346 if (flx_query_equal(&((flxQueryJobInstance*) n->data)->job->query, q))
352 static void post_query_job(flxServer *s, gint interface, guchar protocol, flxQueryJob *job) {
356 if (interface <= 0) {
357 const flxInterface *i;
359 for (i = flx_interface_monitor_get_first(s->monitor); i; i = i->next)
360 post_query_job(s, i->index, protocol, job);
361 } else if (protocol == AF_UNSPEC) {
362 post_query_job(s, interface, AF_INET, job);
363 post_query_job(s, interface, AF_INET6, job);
365 flxQueryJobInstance *i;
367 if (query_job_exists(s, interface, protocol, &job->query))
370 i = g_new(flxQueryJobInstance, 1);
371 i->job = flx_query_job_ref(job);
372 i->interface = interface;
373 i->protocol = protocol;
374 i->node = flx_prio_queue_put(s->query_job_queue, i);
378 void flx_server_post_query_job(flxServer *s, gint interface, guchar protocol, const GTimeVal *tv, const flxQuery *q) {
383 job = flx_query_job_new();
384 job->query.name = g_strdup(q->name);
385 job->query.class = q->class;
386 job->query.type = q->type;
389 post_query_job(s, interface, protocol, job);
392 void flx_server_drop_query_job(flxServer *s, gint interface, guchar protocol, const flxQuery *q) {
393 flxPrioQueueNode *n, *next;
395 g_assert(interface > 0);
396 g_assert(protocol != AF_UNSPEC);
399 for (n = s->query_job_queue->root; n; n = next) {
402 if (flx_query_equal(&((flxQueryJobInstance*) n->data)->job->query, q))
403 flx_server_remove_query_job_instance(s, n->data);
407 void flx_server_remove_query_job_instance(flxServer *s, flxQueryJobInstance *i) {
412 flx_prio_queue_remove(s->query_job_queue, i->node);
413 flx_query_job_unref(i->job);
417 gboolean flx_query_equal(const flxQuery *a, const flxQuery *b) {
418 return strcmp(a->name, b->name) == 0 && a->type == b->type && a->class == b->class;