plugins/dnsproxy.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 <errno.h>
  27 #include <unistd.h>
  28 #include <string.h>
  29 #include <stdint.h>
  30 #include <arpa/inet.h>
  31 #include <netinet/in.h>
  32
  33 #define CONNMAN_API_SUBJECT_TO_CHANGE
  34 #include <connman/plugin.h>
  35 #include <connman/resolver.h>
  36 #include <connman/notifier.h>
  37 #include <connman/log.h>
  38
  39 #include <glib.h>
  40
  41 #if __BYTE_ORDER == __LITTLE_ENDIAN
  42 struct domain_hdr {
  43         uint16_t id;
  44         uint8_t rd:1;
  45         uint8_t tc:1;
  46         uint8_t aa:1;
  47         uint8_t opcode:4;
  48         uint8_t qr:1;
  49         uint8_t rcode:4;
  50         uint8_t z:3;
  51         uint8_t ra:1;
  52         uint16_t qdcount;
  53         uint16_t ancount;
  54         uint16_t nscount;
  55         uint16_t arcount;
  56 } __attribute__ ((packed));
  57 #elif __BYTE_ORDER == __BIG_ENDIAN
  58 struct domain_hdr {
  59         uint16_t id;
  60         uint8_t qr:1;
  61         uint8_t opcode:4;
  62         uint8_t aa:1;
  63         uint8_t tc:1;
  64         uint8_t rd:1;
  65         uint8_t ra:1;
  66         uint8_t z:3;
  67         uint8_t rcode:4;
  68         uint16_t qdcount;
  69         uint16_t ancount;
  70         uint16_t nscount;
  71         uint16_t arcount;
  72 } __attribute__ ((packed));
  73 #else
  74 #error "Unknown byte order"
  75 #endif
  76
  77 struct server_data {
  78         char *interface;
  79         char *domain;
  80         char *server;
  81         GIOChannel *channel;
  82         guint watch;
  83         gboolean enabled;
  84 };
  85
  86 struct request_data {
  87         struct sockaddr_in sin;
  88         socklen_t len;
  89         guint16 srcid;
  90         guint16 dstid;
  91         guint16 altid;
  92         guint timeout;
  93         guint numserv;
  94         guint numresp;
  95         gpointer resp;
  96         gsize resplen;
  97 };
  98
  99 static GSList *server_list = NULL;
 100 static GSList *request_list = NULL;
 101 static guint16 request_id = 0x0000;
 102
 103 static GIOChannel *listener_channel = NULL;
 104 static guint listener_watch = 0;
 105
 106 static struct request_data *find_request(guint16 id)
 107 {
 108         GSList *list;
 109
 110         for (list = request_list; list; list = list->next) {
 111                 struct request_data *req = list->data;
 112
 113                 if (req->dstid == id || req->altid == id)
 114                         return req;
 115         }
 116
 117         return NULL;
 118 }
 119
 120 static struct server_data *find_server(const char *interface,
 121                                         const char *domain, const char *server)
 122 {
 123         GSList *list;
 124
 125         DBG("interface %s server %s", interface, server);
 126
 127         for (list = server_list; list; list = list->next) {
 128                 struct server_data *data = list->data;
 129
 130                 if (data->interface == NULL || data->server == NULL)
 131                         continue;
 132
 133                 if (g_str_equal(data->interface, interface) == TRUE &&
 134                                 g_str_equal(data->server, server) == TRUE) {
 135                         if (domain == NULL) {
 136                                 if (data->domain == NULL)
 137                                         return data;
 138                                 continue;
 139                         }
 140
 141                         if (g_str_equal(data->domain, domain) == TRUE)
 142                                 return data;
 143                 }
 144         }
 145
 146         return NULL;
 147 }
 148
 149 static gboolean server_event(GIOChannel *channel, GIOCondition condition,
 150                                                         gpointer user_data)
 151 {
 152         struct server_data *data = user_data;
 153         struct request_data *req;
 154         unsigned char buf[768];
 155         struct domain_hdr *hdr = (void *) &buf;
 156         int sk, err, len;
 157
 158         if (condition & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
 159                 connman_error("Error with server channel");
 160                 data->watch = 0;
 161                 return FALSE;
 162         }
 163
 164         sk = g_io_channel_unix_get_fd(channel);
 165
 166         len = recv(sk, buf, sizeof(buf), 0);
 167         if (len < 12)
 168                 return TRUE;
 169
 170         DBG("Received %d bytes (id 0x%04x)", len, buf[0] | buf[1] << 8);
 171
 172         req = find_request(buf[0] | buf[1] << 8);
 173         if (req == NULL)
 174                 return TRUE;
 175
 176         DBG("id 0x%04x rcode %d", hdr->id, hdr->rcode);
 177
 178         buf[0] = req->srcid & 0xff;
 179         buf[1] = req->srcid >> 8;
 180
 181         req->numresp++;
 182
 183         if (hdr->rcode == 0 || req->resp == NULL) {
 184                 g_free(req->resp);
 185                 req->resplen = 0;
 186
 187                 req->resp = g_try_malloc(len);
 188                 if (req->resp == NULL)
 189                         return TRUE;
 190
 191                 memcpy(req->resp, buf, len);
 192                 req->resplen = len;
 193         }
 194
 195         if (hdr->rcode > 0 && req->numresp < req->numserv)
 196                 return TRUE;
 197
 198         if (req->timeout > 0)
 199                 g_source_remove(req->timeout);
 200
 201         request_list = g_slist_remove(request_list, req);
 202
 203         sk = g_io_channel_unix_get_fd(listener_channel);
 204
 205         err = sendto(sk, req->resp, req->resplen, 0,
 206                                 (struct sockaddr *) &req->sin, req->len);
 207
 208         g_free(req->resp);
 209         g_free(req);
 210
 211         return TRUE;
 212 }
 213
 214 static struct server_data *create_server(const char *interface,
 215                                         const char *domain, const char *server)
 216 {
 217         struct server_data *data;
 218         struct sockaddr_in sin;
 219         int sk;
 220
 221         DBG("interface %s server %s", interface, server);
 222
 223         sk = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
 224         if (sk < 0) {
 225                 connman_error("Failed to create server %s socket", server);
 226                 return NULL;
 227         }
 228
 229         if (setsockopt(sk, SOL_SOCKET, SO_BINDTODEVICE,
 230                                 interface, strlen(interface) + 1) < 0) {
 231                 connman_error("Failed to bind server %s to interface %s",
 232                                                         server, interface);
 233                 close(sk);
 234                 return NULL;
 235         }
 236
 237         memset(&sin, 0, sizeof(sin));
 238         sin.sin_family = AF_INET;
 239         sin.sin_port = htons(53);
 240         sin.sin_addr.s_addr = inet_addr(server);
 241
 242         if (connect(sk, (struct sockaddr *) &sin, sizeof(sin)) < 0) {
 243                 connman_error("Failed to connect server %s", server);
 244                 close(sk);
 245                 return NULL;
 246         }
 247
 248         data = g_try_new0(struct server_data, 1);
 249         if (data == NULL) {
 250                 connman_error("Failed to allocate server %s data", server);
 251                 close(sk);
 252                 return NULL;
 253         }
 254
 255         data->channel = g_io_channel_unix_new(sk);
 256         if (data->channel == NULL) {
 257                 connman_error("Failed to create server %s channel", server);
 258                 close(sk);
 259                 g_free(data);
 260                 return NULL;
 261         }
 262
 263         g_io_channel_set_close_on_unref(data->channel, TRUE);
 264
 265         data->watch = g_io_add_watch(data->channel, G_IO_IN,
 266                                                         server_event, data);
 267
 268         data->interface = g_strdup(interface);
 269         data->domain = g_strdup(domain);
 270         data->server = g_strdup(server);
 271
 272         /* Enable new servers by default */
 273         data->enabled = TRUE;
 274
 275         return data;
 276 }
 277
 278 static void destroy_server(struct server_data *data)
 279 {
 280         DBG("interface %s server %s", data->interface, data->server);
 281
 282         if (data->watch > 0)
 283                 g_source_remove(data->watch);
 284
 285         g_io_channel_unref(data->channel);
 286
 287         g_free(data->server);
 288         g_free(data->domain);
 289         g_free(data->interface);
 290         g_free(data);
 291 }
 292
 293 static int dnsproxy_append(const char *interface, const char *domain,
 294                                                         const char *server)
 295 {
 296         struct server_data *data;
 297
 298         DBG("interface %s server %s", interface, server);
 299
 300         if (g_str_equal(server, "127.0.0.1") == TRUE)
 301                 return -ENODEV;
 302
 303         data = create_server(interface, domain, server);
 304         if (data == NULL)
 305                 return -EIO;
 306
 307         server_list = g_slist_append(server_list, data);
 308
 309         return 0;
 310 }
 311
 312 static int dnsproxy_remove(const char *interface, const char *domain,
 313                                                         const char *server)
 314 {
 315         struct server_data *data;
 316
 317         DBG("interface %s server %s", interface, server);
 318
 319         if (g_str_equal(server, "127.0.0.1") == TRUE)
 320                 return -ENODEV;
 321
 322         data = find_server(interface, domain, server);
 323         if (data == NULL)
 324                 return 0;
 325
 326         server_list = g_slist_remove(server_list, data);
 327
 328         destroy_server(data);
 329
 330         return 0;
 331 }
 332
 333 static struct connman_resolver dnsproxy_resolver = {
 334         .name           = "dnsproxy",
 335         .priority       = CONNMAN_RESOLVER_PRIORITY_HIGH,
 336         .append         = dnsproxy_append,
 337         .remove         = dnsproxy_remove,
 338 };
 339
 340 static void dnsproxy_offline_mode(connman_bool_t enabled)
 341 {
 342         GSList *list;
 343
 344         DBG("enabled %d", enabled);
 345
 346         for (list = server_list; list; list = list->next) {
 347                 struct server_data *data = list->data;
 348
 349                 data->enabled = enabled ? FALSE : TRUE;
 350         }
 351 }
 352
 353 static void dnsproxy_default_changed(struct connman_service *service)
 354 {
 355         GSList *list;
 356         char *interface;
 357
 358         DBG("service %p", service);
 359
 360         if (service == NULL) {
 361                 /* When no services are active, then disable DNS proxying */
 362                 dnsproxy_offline_mode(TRUE);
 363                 return;
 364         }
 365
 366         interface = connman_service_get_interface(service);
 367         if (interface == NULL)
 368                 return;
 369
 370         for (list = server_list; list; list = list->next) {
 371                 struct server_data *data = list->data;
 372
 373                 if (g_strcmp0(data->interface, interface) == 0)
 374                         data->enabled = TRUE;
 375                 else
 376                         data->enabled = FALSE;
 377         }
 378
 379         g_free(interface);
 380 }
 381
 382 static struct connman_notifier dnsproxy_notifier = {
 383         .name                   = "dnsproxy",
 384         .default_changed        = dnsproxy_default_changed,
 385         .offline_mode           = dnsproxy_offline_mode,
 386 };
 387
 388 static int parse_request(unsigned char *buf, int len,
 389                                         char *name, unsigned int size)
 390 {
 391         struct domain_hdr *hdr = (void *) buf;
 392         uint16_t qdcount = ntohs(hdr->qdcount);
 393         unsigned char *ptr;
 394         char *last_label = NULL;
 395         int label_count = 0;
 396         unsigned int remain, used = 0;
 397
 398         if (len < 12)
 399                 return -EINVAL;
 400
 401         DBG("id 0x%04x qr %d opcode %d qdcount %d",
 402                                 hdr->id, hdr->qr, hdr->opcode, qdcount);
 403
 404         if (hdr->qr != 0 || qdcount != 1)
 405                 return -EINVAL;
 406
 407         memset(name, 0, size);
 408
 409         ptr = buf + 12;
 410         remain = len - 12;
 411
 412         while (remain > 0) {
 413                 uint8_t len = *ptr;
 414
 415                 if (len == 0x00) {
 416                         if (label_count > 0)
 417                                 last_label = (char *) (ptr + 1);
 418                         break;
 419                 }
 420
 421                 label_count++;
 422
 423                 if (used + len + 1 > size)
 424                         return -ENOBUFS;
 425
 426                 strncat(name, (char *) (ptr + 1), len);
 427                 strcat(name, ".");
 428
 429                 used += len + 1;
 430
 431                 ptr += len + 1;
 432                 remain -= len + 1;
 433         }
 434
 435         DBG("query %s (%d labels)", name, label_count);
 436
 437         return 0;
 438 }
 439
 440 static void send_response(int sk, unsigned char *buf, int len,
 441                                 const struct sockaddr *to, socklen_t tolen)
 442 {
 443         struct domain_hdr *hdr = (void *) buf;
 444         int err;
 445
 446         if (len < 12)
 447                 return;
 448
 449         DBG("id 0x%04x qr %d opcode %d", hdr->id, hdr->qr, hdr->opcode);
 450
 451         hdr->qr = 1;
 452         hdr->rcode = 2;
 453
 454         hdr->ancount = 0;
 455         hdr->nscount = 0;
 456         hdr->arcount = 0;
 457
 458         err = sendto(sk, buf, len, 0, to, tolen);
 459 }
 460
 461 static int append_query(unsigned char *buf, unsigned int size,
 462                                 const char *query, const char *domain)
 463 {
 464         unsigned char *ptr = buf;
 465         char *offset;
 466
 467         DBG("query %s domain %s", query, domain);
 468
 469         offset = (char *) query;
 470         while (offset != NULL) {
 471                 char *tmp;
 472
 473                 tmp = strchr(offset, '.');
 474                 if (tmp == NULL) {
 475                         if (strlen(offset) == 0)
 476                                 break;
 477                         *ptr = strlen(offset);
 478                         memcpy(ptr + 1, offset, strlen(offset));
 479                         ptr += strlen(offset) + 1;
 480                         break;
 481                 }
 482
 483                 *ptr = tmp - offset;
 484                 memcpy(ptr + 1, offset, tmp - offset);
 485                 ptr += tmp - offset + 1;
 486
 487                 offset = tmp + 1;
 488         }
 489
 490         offset = (char *) domain;
 491         while (offset != NULL) {
 492                 char *tmp;
 493
 494                 tmp = strchr(offset, '.');
 495                 if (tmp == NULL) {
 496                         if (strlen(offset) == 0)
 497                                 break;
 498                         *ptr = strlen(offset);
 499                         memcpy(ptr + 1, offset, strlen(offset));
 500                         ptr += strlen(offset) + 1;
 501                         break;
 502                 }
 503
 504                 *ptr = tmp - offset;
 505                 memcpy(ptr + 1, offset, tmp - offset);
 506                 ptr += tmp - offset + 1;
 507
 508                 offset = tmp + 1;
 509         }
 510
 511         *ptr++ = 0x00;
 512
 513         return ptr - buf;
 514 }
 515
 516 static gboolean request_timeout(gpointer user_data)
 517 {
 518         struct request_data *req = user_data;
 519
 520         DBG("id 0x%04x", req->srcid);
 521
 522         request_list = g_slist_remove(request_list, req);
 523
 524         if (req->resplen > 0 && req->resp != NULL) {
 525                 int sk, err;
 526
 527                 sk = g_io_channel_unix_get_fd(listener_channel);
 528
 529                 err = sendto(sk, req->resp, req->resplen, 0,
 530                                 (struct sockaddr *) &req->sin, req->len);
 531         }
 532
 533         g_free(req->resp);
 534         g_free(req);
 535
 536         return FALSE;
 537 }
 538
 539 static gboolean listener_event(GIOChannel *channel, GIOCondition condition,
 540                                                         gpointer user_data)
 541 {
 542         GSList *list;
 543         unsigned char buf[768];
 544         char query[512];
 545         struct request_data *req;
 546         struct sockaddr_in sin;
 547         socklen_t size = sizeof(sin);
 548         int sk, err, len;
 549
 550         if (condition & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
 551                 connman_error("Error with listener channel");
 552                 listener_watch = 0;
 553                 return FALSE;
 554         }
 555
 556         sk = g_io_channel_unix_get_fd(channel);
 557
 558         memset(&sin, 0, sizeof(sin));
 559         len = recvfrom(sk, buf, sizeof(buf), 0,
 560                                         (struct sockaddr *) &sin, &size);
 561         if (len < 2)
 562                 return TRUE;
 563
 564         DBG("Received %d bytes (id 0x%04x)", len, buf[0] | buf[1] << 8);
 565
 566         err = parse_request(buf, len, query, sizeof(query));
 567         if (err < 0 || g_slist_length(server_list) == 0) {
 568                 send_response(sk, buf, len, (struct sockaddr *) &sin, size);
 569                 return TRUE;
 570         }
 571
 572         req = g_try_new0(struct request_data, 1);
 573         if (req == NULL)
 574                 return TRUE;
 575
 576         memcpy(&req->sin, &sin, sizeof(sin));
 577         req->len = size;
 578
 579         request_id += 2;
 580         if (request_id == 0x0000 || request_id == 0xffff)
 581                 request_id += 2;
 582
 583         req->srcid = buf[0] | (buf[1] << 8);
 584         req->dstid = request_id;
 585         req->altid = request_id + 1;
 586
 587         buf[0] = req->dstid & 0xff;
 588         buf[1] = req->dstid >> 8;
 589
 590         request_list = g_slist_append(request_list, req);
 591
 592         req->numserv = 0;
 593         req->timeout = g_timeout_add_seconds(5, request_timeout, req);
 594
 595         for (list = server_list; list; list = list->next) {
 596                 struct server_data *data = list->data;
 597
 598                 DBG("server %s domain %s enabled %d",
 599                                 data->server, data->domain, data->enabled);
 600
 601                 if (data->enabled == FALSE)
 602                         continue;
 603
 604                 sk = g_io_channel_unix_get_fd(data->channel);
 605
 606                 err = send(sk, buf, len, 0);
 607
 608                 req->numserv++;
 609
 610                 if (data->domain != NULL) {
 611                         unsigned char alt[1024];
 612                         struct domain_hdr *hdr = (void *) &alt;
 613                         int altlen, domlen;
 614
 615                         domlen = strlen(data->domain) + 1;
 616                         if (domlen < 5)
 617                                 continue;
 618
 619                         alt[0] = req->altid & 0xff;
 620                         alt[1] = req->altid >> 8;
 621
 622                         memcpy(alt + 2, buf + 2, 10);
 623                         hdr->qdcount = htons(1);
 624
 625                         altlen = append_query(alt + 12, sizeof(alt) - 12,
 626                                                         query, data->domain);
 627                         if (altlen < 0)
 628                                 continue;
 629
 630                         altlen += 12;
 631
 632                         memcpy(alt + altlen, buf + altlen - domlen,
 633                                                         len - altlen + domlen);
 634
 635                         err = send(sk, alt, altlen + 4, 0);
 636
 637                         req->numserv++;
 638                 }
 639         }
 640
 641         return TRUE;
 642 }
 643
 644 static int create_listener(void)
 645 {
 646         const char *ifname = "lo";
 647         struct sockaddr_in sin;
 648         int sk;
 649
 650         DBG("");
 651
 652         sk = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
 653         if (sk < 0) {
 654                 connman_error("Failed to create listener socket");
 655                 return -EIO;
 656         }
 657
 658         //setsockopt(sk, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
 659         //setsockopt(sk, SOL_IP, IP_PKTINFO, &opt, sizeof(opt));
 660
 661         if (setsockopt(sk, SOL_SOCKET, SO_BINDTODEVICE,
 662                                         ifname, strlen(ifname) + 1) < 0) {
 663                 connman_error("Failed to bind listener interface");
 664                 close(sk);
 665                 return -EIO;
 666         }
 667
 668         memset(&sin, 0, sizeof(sin));
 669         sin.sin_family = AF_INET;
 670         sin.sin_port = htons(53);
 671         sin.sin_addr.s_addr = inet_addr("127.0.0.1");
 672         //sin.sin_addr.s_addr = INADDR_ANY;
 673
 674         if (bind(sk, (struct sockaddr *) &sin, sizeof(sin)) < 0) {
 675                 connman_error("Failed to bind listener socket");
 676                 close(sk);
 677                 return -EIO;
 678         }
 679
 680         listener_channel = g_io_channel_unix_new(sk);
 681         if (listener_channel == NULL) {
 682                 connman_error("Failed to create listener channel");
 683                 close(sk);
 684                 return -EIO;
 685         }
 686
 687         g_io_channel_set_close_on_unref(listener_channel, TRUE);
 688
 689         listener_watch = g_io_add_watch(listener_channel, G_IO_IN,
 690                                                         listener_event, NULL);
 691
 692         connman_resolver_append("lo", NULL, "127.0.0.1");
 693
 694         return 0;
 695 }
 696
 697 static void destroy_listener(void)
 698 {
 699         GSList *list;
 700
 701         DBG("");
 702
 703         connman_resolver_remove_all("lo");
 704
 705         if (listener_watch > 0)
 706                 g_source_remove(listener_watch);
 707
 708         for (list = request_list; list; list = list->next) {
 709                 struct request_data *req = list->data;
 710
 711                 DBG("Dropping request (id 0x%04x -> 0x%04x)",
 712                                                 req->srcid, req->dstid);
 713
 714                 g_free(req->resp);
 715                 g_free(req);
 716                 list->data = NULL;
 717         }
 718
 719         g_slist_free(request_list);
 720         request_list = NULL;
 721
 722         g_io_channel_unref(listener_channel);
 723 }
 724
 725 static int dnsproxy_init(void)
 726 {
 727         int err;
 728
 729         err = create_listener();
 730         if (err < 0)
 731                 return err;
 732
 733         err = connman_resolver_register(&dnsproxy_resolver);
 734         if (err < 0)
 735                 goto destroy;
 736
 737         err = connman_notifier_register(&dnsproxy_notifier);
 738         if (err < 0)
 739                 goto unregister;
 740
 741         return 0;
 742
 743 unregister:
 744         connman_resolver_unregister(&dnsproxy_resolver);
 745
 746 destroy:
 747         destroy_listener();
 748
 749         return err;
 750 }
 751
 752 static void dnsproxy_exit(void)
 753 {
 754         connman_notifier_unregister(&dnsproxy_notifier);
 755
 756         connman_resolver_unregister(&dnsproxy_resolver);
 757
 758         destroy_listener();
 759 }
 760
 761 CONNMAN_PLUGIN_DEFINE(dnsproxy, "DNS proxy resolver plugin", VERSION,
 762                  CONNMAN_PLUGIN_PRIORITY_DEFAULT, dnsproxy_init, dnsproxy_exit)