git.meshlink.io Git - catta/blob - avahi-core/socket.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of avahi.
   5
   6   avahi is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as
   8   published by the Free Software Foundation; either version 2.1 of the
   9   License, or (at your option) any later version.
  10
  11   avahi is distributed in the hope that it will be useful, but WITHOUT
  12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  14   Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with avahi; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <inttypes.h>
  27 #include <errno.h>
  28 #include <string.h>
  29 #include <stdio.h>
  30 #include <sys/socket.h>
  31 #include <netinet/in.h>
  32 #include <arpa/inet.h>
  33 #include <unistd.h>
  34 #include <fcntl.h>
  35 #include <sys/time.h>
  36 #include <net/if.h>
  37 #include <sys/ioctl.h>
  38 #include <assert.h>
  39
  40 #include "dns.h"
  41 #include "fdutil.h"
  42 #include "socket.h"
  43 #include "log.h"
  44
  45 static void mdns_mcast_group_ipv4(struct sockaddr_in *ret_sa) {
  46     assert(ret_sa);
  47
  48     memset(ret_sa, 0, sizeof(struct sockaddr_in));
  49     ret_sa->sin_family = AF_INET;
  50     ret_sa->sin_port = htons(AVAHI_MDNS_PORT);
  51     inet_pton(AF_INET, AVAHI_IPV4_MCAST_GROUP, &ret_sa->sin_addr);
  52 }
  53
  54 static void mdns_mcast_group_ipv6(struct sockaddr_in6 *ret_sa) {
  55     assert(ret_sa);
  56
  57     memset(ret_sa, 0, sizeof(struct sockaddr_in6));
  58     ret_sa->sin6_family = AF_INET6;
  59     ret_sa->sin6_port = htons(AVAHI_MDNS_PORT);
  60     inet_pton(AF_INET6, AVAHI_IPV6_MCAST_GROUP, &ret_sa->sin6_addr);
  61 }
  62
  63 static void ipv4_address_to_sockaddr(struct sockaddr_in *ret_sa, const AvahiIPv4Address *a, uint16_t port) {
  64     assert(ret_sa);
  65     assert(a);
  66     assert(port > 0);
  67
  68     memset(ret_sa, 0, sizeof(struct sockaddr_in));
  69     ret_sa->sin_family = AF_INET;
  70     ret_sa->sin_port = htons(port);
  71     memcpy(&ret_sa->sin_addr, a, sizeof(AvahiIPv4Address));
  72 }
  73
  74 static void ipv6_address_to_sockaddr(struct sockaddr_in6 *ret_sa, const AvahiIPv6Address *a, uint16_t port) {
  75     assert(ret_sa);
  76     assert(a);
  77     assert(port > 0);
  78
  79     memset(ret_sa, 0, sizeof(struct sockaddr_in6));
  80     ret_sa->sin6_family = AF_INET6;
  81     ret_sa->sin6_port = htons(port);
  82     memcpy(&ret_sa->sin6_addr, a, sizeof(AvahiIPv6Address));
  83 }
  84
  85 int avahi_mdns_mcast_join_ipv4(int fd, int idx) {
  86     struct ip_mreqn mreq;
  87     struct sockaddr_in sa;
  88
  89     mdns_mcast_group_ipv4 (&sa);
  90
  91     memset(&mreq, 0, sizeof(mreq));
  92     mreq.imr_multiaddr = sa.sin_addr;
  93     mreq.imr_ifindex = idx;
  94
  95     if (setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) < 0) {
  96         avahi_log_warn("IP_ADD_MEMBERSHIP failed: %s", strerror(errno));
  97         return -1;
  98     }
  99
 100     return 0;
 101 }
 102
 103 int avahi_mdns_mcast_join_ipv6(int fd, int idx) {
 104     struct ipv6_mreq mreq6;
 105     struct sockaddr_in6 sa6;
 106
 107     mdns_mcast_group_ipv6 (&sa6);
 108
 109     memset(&mreq6, 0, sizeof(mreq6));
 110     mreq6.ipv6mr_multiaddr = sa6.sin6_addr;
 111     mreq6.ipv6mr_interface = idx;
 112
 113     if (setsockopt(fd, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, &mreq6, sizeof(mreq6)) < 0) {
 114         avahi_log_warn("IPV6_ADD_MEMBERSHIP failed: %s", strerror(errno));
 115         return -1;
 116     }
 117
 118     return 0;
 119 }
 120
 121 int avahi_mdns_mcast_leave_ipv4(int fd, int idx) {
 122     struct ip_mreqn mreq;
 123     struct sockaddr_in sa;
 124
 125     mdns_mcast_group_ipv4 (&sa);
 126
 127     memset(&mreq, 0, sizeof(mreq));
 128     mreq.imr_multiaddr = sa.sin_addr;
 129     mreq.imr_ifindex = idx;
 130
 131     if (setsockopt(fd, IPPROTO_IP, IP_DROP_MEMBERSHIP, &mreq, sizeof(mreq)) < 0) {
 132         avahi_log_warn("IP_DROP_MEMBERSHIP failed: %s", strerror(errno));
 133         return -1;
 134     }
 135
 136     return 0;
 137 }
 138
 139 int avahi_mdns_mcast_leave_ipv6(int fd, int idx) {
 140     struct ipv6_mreq mreq6;
 141     struct sockaddr_in6 sa6;
 142
 143     mdns_mcast_group_ipv6 (&sa6);
 144
 145     memset(&mreq6, 0, sizeof(mreq6));
 146     mreq6.ipv6mr_multiaddr = sa6.sin6_addr;
 147     mreq6.ipv6mr_interface = idx;
 148
 149     if (setsockopt(fd, IPPROTO_IPV6, IPV6_DROP_MEMBERSHIP, &mreq6, sizeof(mreq6)) < 0) {
 150         avahi_log_warn("IPV6_DROP_MEMBERSHIP failed: %s", strerror(errno));
 151         return -1;
 152     }
 153
 154     return 0;
 155 }
 156
 157 static int bind_with_warn(int fd, const struct sockaddr *sa, socklen_t l) {
 158     int yes;
 159
 160     assert(fd >= 0);
 161     assert(sa);
 162     assert(l > 0);
 163
 164     if (bind(fd, sa, l) < 0) {
 165
 166         if (errno != EADDRINUSE) {
 167             avahi_log_warn("bind() failed: %s", strerror(errno));
 168             return -1;
 169         }
 170
 171         avahi_log_warn("*** WARNING: Detected another %s mDNS stack running on this host. This makes mDNS unreliable and is thus not recommended. ***",
 172                        sa->sa_family == AF_INET ? "IPv4" : "IPv6");
 173
 174         /* Try again, this time with SO_REUSEADDR set */
 175         yes = 1;
 176         if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0) {
 177             avahi_log_warn("SO_REUSEADDR failed: %s", strerror(errno));
 178             return -1;
 179         }
 180
 181         if (bind(fd, sa, l) < 0) {
 182             avahi_log_warn("bind() failed: %s", strerror(errno));
 183             return -1;
 184         }
 185     } else {
 186
 187         /* We enable SO_REUSEADDR afterwards, to make sure that the
 188          * user may run other mDNS implementations if he really
 189          * wants. */
 190
 191         yes = 1;
 192         if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0) {
 193             avahi_log_warn("SO_REUSEADDR failed: %s", strerror(errno));
 194             return -1;
 195         }
 196     }
 197
 198     return 0;
 199 }
 200
 201 int avahi_open_socket_ipv4(int no_reuse) {
 202     struct sockaddr_in local;
 203     int fd = -1, ttl, yes, r;
 204
 205     if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
 206         avahi_log_warn("socket() failed: %s", strerror(errno));
 207         goto fail;
 208     }
 209
 210     ttl = 255;
 211     if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl)) < 0) {
 212         avahi_log_warn("IP_MULTICAST_TTL failed: %s", strerror(errno));
 213         goto fail;
 214     }
 215
 216     ttl = 255;
 217     if (setsockopt(fd, IPPROTO_IP, IP_TTL, &ttl, sizeof(ttl)) < 0) {
 218         avahi_log_warn("IP_TTL failed: %s", strerror(errno));
 219         goto fail;
 220     }
 221
 222     yes = 1;
 223     if (setsockopt(fd, IPPROTO_IP, IP_MULTICAST_LOOP, &yes, sizeof(yes)) < 0) {
 224         avahi_log_warn("IP_MULTICAST_LOOP failed: %s", strerror(errno));
 225         goto fail;
 226     }
 227
 228     memset(&local, 0, sizeof(local));
 229     local.sin_family = AF_INET;
 230     local.sin_port = htons(AVAHI_MDNS_PORT);
 231
 232     if (no_reuse)
 233         r = bind(fd, (struct sockaddr*) &local, sizeof(local));
 234     else
 235         r = bind_with_warn(fd, (struct sockaddr*) &local, sizeof(local));
 236
 237     if (r < 0)
 238         goto fail;
 239
 240     yes = 1;
 241     if (setsockopt(fd, IPPROTO_IP, IP_RECVTTL, &yes, sizeof(yes)) < 0) {
 242         avahi_log_warn("IP_RECVTTL failed: %s", strerror(errno));
 243         goto fail;
 244     }
 245
 246     yes = 1;
 247     if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &yes, sizeof(yes)) < 0) {
 248         avahi_log_warn("IP_PKTINFO failed: %s", strerror(errno));
 249         goto fail;
 250     }
 251
 252     if (avahi_set_cloexec(fd) < 0) {
 253         avahi_log_warn("FD_CLOEXEC failed: %s", strerror(errno));
 254         goto fail;
 255     }
 256
 257     if (avahi_set_nonblock(fd) < 0) {
 258         avahi_log_warn("O_NONBLOCK failed: %s", strerror(errno));
 259         goto fail;
 260     }
 261
 262     return fd;
 263
 264 fail:
 265     if (fd >= 0)
 266         close(fd);
 267
 268     return -1;
 269 }
 270
 271 int avahi_open_socket_ipv6(int no_reuse) {
 272     struct sockaddr_in6 sa, local;
 273     int fd = -1, ttl, yes, r;
 274
 275     mdns_mcast_group_ipv6(&sa);
 276
 277     if ((fd = socket(AF_INET6, SOCK_DGRAM, 0)) < 0) {
 278         avahi_log_warn("socket() failed: %s", strerror(errno));
 279         goto fail;
 280     }
 281
 282     ttl = 255;
 283     if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &ttl, sizeof(ttl)) < 0) {
 284         avahi_log_warn("IPV6_MULTICAST_HOPS failed: %s", strerror(errno));
 285         goto fail;
 286     }
 287
 288     ttl = 255;
 289     if (setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &ttl, sizeof(ttl)) < 0) {
 290         avahi_log_warn("IPV6_UNICAST_HOPS failed: %s", strerror(errno));
 291         goto fail;
 292     }
 293
 294     yes = 1;
 295     if (setsockopt(fd, IPPROTO_IPV6, IPV6_V6ONLY, &yes, sizeof(yes)) < 0) {
 296         avahi_log_warn("IPV6_V6ONLY failed: %s", strerror(errno));
 297         goto fail;
 298     }
 299
 300     yes = 1;
 301     if (setsockopt(fd, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, &yes, sizeof(yes)) < 0) {
 302         avahi_log_warn("IPV6_MULTICAST_LOOP failed: %s", strerror(errno));
 303         goto fail;
 304     }
 305
 306     memset(&local, 0, sizeof(local));
 307     local.sin6_family = AF_INET6;
 308     local.sin6_port = htons(AVAHI_MDNS_PORT);
 309
 310     if (no_reuse)
 311         r = bind(fd, (struct sockaddr*) &local, sizeof(local));
 312     else
 313         r = bind_with_warn(fd, (struct sockaddr*) &local, sizeof(local));
 314
 315     if (r < 0)
 316         goto fail;
 317
 318     yes = 1;
 319     if (setsockopt(fd, IPPROTO_IPV6, IPV6_HOPLIMIT, &yes, sizeof(yes)) < 0) {
 320         avahi_log_warn("IPV6_HOPLIMIT failed: %s", strerror(errno));
 321         goto fail;
 322     }
 323
 324     yes = 1;
 325     if (setsockopt(fd, IPPROTO_IPV6, IPV6_PKTINFO, &yes, sizeof(yes)) < 0) {
 326         avahi_log_warn("IPV6_PKTINFO failed: %s", strerror(errno));
 327         goto fail;
 328     }
 329
 330     if (avahi_set_cloexec(fd) < 0) {
 331         avahi_log_warn("FD_CLOEXEC failed: %s", strerror(errno));
 332         goto fail;
 333     }
 334
 335     if (avahi_set_nonblock(fd) < 0) {
 336         avahi_log_warn("O_NONBLOCK failed: %s", strerror(errno));
 337         goto fail;
 338     }
 339
 340     return fd;
 341
 342 fail:
 343     if (fd >= 0)
 344         close(fd);
 345
 346     return -1;
 347 }
 348
 349 static int sendmsg_loop(int fd, struct msghdr *msg, int flags) {
 350     assert(fd >= 0);
 351     assert(msg);
 352
 353     for (;;) {
 354
 355         if (sendmsg(fd, msg, flags) >= 0)
 356             break;
 357
 358         if (errno != EAGAIN) {
 359             avahi_log_debug("sendmsg() failed: %s", strerror(errno));
 360             return -1;
 361         }
 362
 363         if (avahi_wait_for_write(fd) < 0)
 364             return -1;
 365     }
 366
 367     return 0;
 368 }
 369
 370 int avahi_send_dns_packet_ipv4(int fd, int interface, AvahiDnsPacket *p, const AvahiIPv4Address *a, uint16_t port) {
 371     struct sockaddr_in sa;
 372     struct msghdr msg;
 373     struct iovec io;
 374     struct cmsghdr *cmsg;
 375     uint8_t cmsg_data[sizeof(struct cmsghdr) + sizeof(struct in_pktinfo)];
 376
 377     assert(fd >= 0);
 378     assert(p);
 379     assert(avahi_dns_packet_check_valid(p) >= 0);
 380     assert(!a || port > 0);
 381
 382     if (!a)
 383         mdns_mcast_group_ipv4(&sa);
 384     else
 385         ipv4_address_to_sockaddr(&sa, a, port);
 386
 387     memset(&io, 0, sizeof(io));
 388     io.iov_base = AVAHI_DNS_PACKET_DATA(p);
 389     io.iov_len = p->size;
 390
 391     memset(&msg, 0, sizeof(msg));
 392     msg.msg_name = &sa;
 393     msg.msg_namelen = sizeof(sa);
 394     msg.msg_iov = &io;
 395     msg.msg_iovlen = 1;
 396     msg.msg_flags = 0;
 397
 398     if (interface >= 0) {
 399         struct in_pktinfo *pkti;
 400
 401         memset(cmsg_data, 0, sizeof(cmsg_data));
 402         cmsg = (struct cmsghdr*) cmsg_data;
 403         cmsg->cmsg_len = sizeof(cmsg_data);
 404         cmsg->cmsg_level = IPPROTO_IP;
 405         cmsg->cmsg_type = IP_PKTINFO;
 406
 407         pkti = (struct in_pktinfo*) (cmsg_data + sizeof(struct cmsghdr));
 408         pkti->ipi_ifindex = interface;
 409
 410         msg.msg_control = cmsg_data;
 411         msg.msg_controllen = sizeof(cmsg_data);
 412     } else {
 413         msg.msg_control = NULL;
 414         msg.msg_controllen = 0;
 415     }
 416
 417     return sendmsg_loop(fd, &msg, 0);
 418 }
 419
 420 int avahi_send_dns_packet_ipv6(int fd, int interface, AvahiDnsPacket *p, const AvahiIPv6Address *a, uint16_t port) {
 421     struct sockaddr_in6 sa;
 422     struct msghdr msg;
 423     struct iovec io;
 424     struct cmsghdr *cmsg;
 425     uint8_t cmsg_data[sizeof(struct cmsghdr) + sizeof(struct in6_pktinfo)];
 426
 427     assert(fd >= 0);
 428     assert(p);
 429     assert(avahi_dns_packet_check_valid(p) >= 0);
 430
 431     if (!a)
 432         mdns_mcast_group_ipv6(&sa);
 433     else
 434         ipv6_address_to_sockaddr(&sa, a, port);
 435
 436     memset(&io, 0, sizeof(io));
 437     io.iov_base = AVAHI_DNS_PACKET_DATA(p);
 438     io.iov_len = p->size;
 439
 440
 441     memset(&msg, 0, sizeof(msg));
 442     msg.msg_name = &sa;
 443     msg.msg_namelen = sizeof(sa);
 444     msg.msg_iov = &io;
 445     msg.msg_iovlen = 1;
 446     msg.msg_flags = 0;
 447
 448     if (interface >= 0) {
 449         struct in6_pktinfo *pkti;
 450
 451         memset(cmsg_data, 0, sizeof(cmsg_data));
 452         cmsg = (struct cmsghdr*) cmsg_data;
 453         cmsg->cmsg_len = sizeof(cmsg_data);
 454         cmsg->cmsg_level = IPPROTO_IPV6;
 455         cmsg->cmsg_type = IPV6_PKTINFO;
 456
 457         pkti = (struct in6_pktinfo*) (cmsg_data + sizeof(struct cmsghdr));
 458         pkti->ipi6_ifindex = interface;
 459
 460         msg.msg_control = cmsg_data;
 461         msg.msg_controllen = sizeof(cmsg_data);
 462     } else {
 463         msg.msg_control = NULL;
 464         msg.msg_controllen = 0;
 465     }
 466
 467     return sendmsg_loop(fd, &msg, 0);
 468 }
 469
 470 AvahiDnsPacket* avahi_recv_dns_packet_ipv4(int fd, struct sockaddr_in *ret_sa, AvahiIPv4Address *ret_dest_address, int *ret_iface, uint8_t* ret_ttl) {
 471     AvahiDnsPacket *p= NULL;
 472     struct msghdr msg;
 473     struct iovec io;
 474     uint8_t aux[1024];
 475     ssize_t l;
 476     struct cmsghdr *cmsg;
 477     int found_ttl = 0, found_iface = 0;
 478     int ms;
 479
 480     assert(fd >= 0);
 481
 482     if (ioctl(fd, FIONREAD, &ms) < 0) {
 483         avahi_log_warn("ioctl(): %s", strerror(errno));
 484         goto fail;
 485     }
 486
 487     p = avahi_dns_packet_new(ms + AVAHI_DNS_PACKET_EXTRA_SIZE);
 488
 489     io.iov_base = AVAHI_DNS_PACKET_DATA(p);
 490     io.iov_len = p->max_size;
 491
 492     memset(&msg, 0, sizeof(msg));
 493     if (ret_sa) {
 494         msg.msg_name = ret_sa;
 495         msg.msg_namelen = sizeof(struct sockaddr_in);
 496     } else {
 497         msg.msg_name = NULL;
 498         msg.msg_namelen = 0;
 499     }
 500     msg.msg_iov = &io;
 501     msg.msg_iovlen = 1;
 502     msg.msg_control = aux;
 503     msg.msg_controllen = sizeof(aux);
 504     msg.msg_flags = 0;
 505
 506     if ((l = recvmsg(fd, &msg, 0)) < 0) {
 507         avahi_log_warn("recvmsg(): %s", strerror(errno));
 508         goto fail;
 509     }
 510
 511     if (ret_sa && ret_sa->sin_addr.s_addr == INADDR_ANY) {
 512         /* Linux 2.4 behaves very strangely sometimes! */
 513
 514         /*avahi_hexdump(AVAHI_DNS_PACKET_DATA(p), l); */
 515         goto fail;
 516     }
 517
 518     assert(!(msg.msg_flags & MSG_CTRUNC));
 519     assert(!(msg.msg_flags & MSG_TRUNC));
 520     p->size = (size_t) l;
 521
 522     if (ret_ttl)
 523         *ret_ttl = 0;
 524
 525     for (cmsg = CMSG_FIRSTHDR(&msg); cmsg != NULL; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
 526
 527         if (cmsg->cmsg_level == IPPROTO_IP) {
 528
 529             if (cmsg->cmsg_type == IP_TTL) {
 530                 if (ret_ttl)
 531                     *ret_ttl = (uint8_t) (*(int *) CMSG_DATA(cmsg));
 532                 found_ttl = 1;
 533             } else if (cmsg->cmsg_type == IP_PKTINFO) {
 534                 struct in_pktinfo *i = (struct in_pktinfo*) CMSG_DATA(cmsg);
 535
 536                 if (ret_iface)
 537                     *ret_iface = (int) i->ipi_ifindex;
 538
 539                 if (ret_dest_address)
 540                     ret_dest_address->address = i->ipi_addr.s_addr;
 541                 found_iface = 1;
 542             }
 543         }
 544     }
 545
 546     assert(found_iface);
 547     assert(found_ttl);
 548
 549     return p;
 550
 551 fail:
 552     if (p)
 553         avahi_dns_packet_free(p);
 554
 555     return NULL;
 556 }
 557
 558 AvahiDnsPacket* avahi_recv_dns_packet_ipv6(int fd, struct sockaddr_in6 *ret_sa, AvahiIPv6Address *ret_dest_address, int *ret_iface, uint8_t* ret_ttl) {
 559     AvahiDnsPacket *p = NULL;
 560     struct msghdr msg;
 561     struct iovec io;
 562     uint8_t aux[64];
 563     ssize_t l;
 564     int ms;
 565
 566     struct cmsghdr *cmsg;
 567     int found_ttl = 0, found_iface = 0;
 568
 569     assert(fd >= 0);
 570
 571     if (ioctl(fd, FIONREAD, &ms) < 0) {
 572         avahi_log_warn("ioctl(): %s", strerror(errno));
 573         goto fail;
 574     }
 575
 576     p = avahi_dns_packet_new(ms + AVAHI_DNS_PACKET_EXTRA_SIZE);
 577
 578     io.iov_base = AVAHI_DNS_PACKET_DATA(p);
 579     io.iov_len = p->max_size;
 580
 581     memset(&msg, 0, sizeof(msg));
 582     if (ret_sa) {
 583         msg.msg_name = ret_sa;
 584         msg.msg_namelen = sizeof(struct sockaddr_in6);
 585     } else {
 586         msg.msg_name = NULL;
 587         msg.msg_namelen = 0;
 588     }
 589
 590     msg.msg_iov = &io;
 591     msg.msg_iovlen = 1;
 592     msg.msg_control = aux;
 593     msg.msg_controllen = sizeof(aux);
 594     msg.msg_flags = 0;
 595
 596     if ((l = recvmsg(fd, &msg, 0)) < 0) {
 597         avahi_log_warn("recvmsg(): %s", strerror(errno));
 598         goto fail;
 599     }
 600
 601     p->size = (size_t) l;
 602
 603     if (ret_ttl)
 604         *ret_ttl = 0;
 605
 606     for (cmsg = CMSG_FIRSTHDR(&msg); cmsg != NULL; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
 607         if (cmsg->cmsg_level == IPPROTO_IPV6 && cmsg->cmsg_type == IPV6_HOPLIMIT) {
 608
 609             if (ret_ttl)
 610                 *ret_ttl = (uint8_t) (*(int *) CMSG_DATA(cmsg));
 611
 612             found_ttl = 1;
 613         }
 614
 615         if (cmsg->cmsg_level == IPPROTO_IPV6 && cmsg->cmsg_type == IPV6_PKTINFO) {
 616             struct in6_pktinfo *i = (struct in6_pktinfo*) CMSG_DATA(cmsg);
 617
 618             if (ret_iface)
 619                 *ret_iface = i->ipi6_ifindex;
 620
 621             if (ret_dest_address)
 622                 memcpy(ret_dest_address->address, i->ipi6_addr.s6_addr, 16);
 623             found_iface = 1;
 624         }
 625     }
 626
 627     assert(found_iface);
 628     assert(found_ttl);
 629
 630     return p;
 631
 632 fail:
 633     if (p)
 634         avahi_dns_packet_free(p);
 635
 636     return NULL;
 637 }
 638
 639 int avahi_open_unicast_socket_ipv4(void) {
 640     struct sockaddr_in local;
 641     int fd = -1, yes;
 642
 643     if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
 644         avahi_log_warn("socket() failed: %s", strerror(errno));
 645         goto fail;
 646     }
 647
 648     memset(&local, 0, sizeof(local));
 649     local.sin_family = AF_INET;
 650
 651     if (bind(fd, (struct sockaddr*) &local, sizeof(local)) < 0) {
 652         avahi_log_warn("bind() failed: %s", strerror(errno));
 653         goto fail;
 654     }
 655
 656     yes = 1;
 657     if (setsockopt(fd, IPPROTO_IP, IP_RECVTTL, &yes, sizeof(yes)) < 0) {
 658         avahi_log_warn("IP_RECVTTL failed: %s", strerror(errno));
 659         goto fail;
 660     }
 661
 662     yes = 1;
 663     if (setsockopt(fd, IPPROTO_IP, IP_PKTINFO, &yes, sizeof(yes)) < 0) {
 664         avahi_log_warn("IP_PKTINFO failed: %s", strerror(errno));
 665         goto fail;
 666     }
 667
 668     if (avahi_set_cloexec(fd) < 0) {
 669         avahi_log_warn("FD_CLOEXEC failed: %s", strerror(errno));
 670         goto fail;
 671     }
 672
 673     if (avahi_set_nonblock(fd) < 0) {
 674         avahi_log_warn("O_NONBLOCK failed: %s", strerror(errno));
 675         goto fail;
 676     }
 677
 678     return fd;
 679
 680 fail:
 681     if (fd >= 0)
 682         close(fd);
 683
 684     return -1;
 685 }
 686
 687 int avahi_open_unicast_socket_ipv6(void) {
 688     struct sockaddr_in6 local;
 689     int fd = -1, yes;
 690
 691     if ((fd = socket(AF_INET6, SOCK_DGRAM, 0)) < 0) {
 692         avahi_log_warn("socket() failed: %s", strerror(errno));
 693         goto fail;
 694     }
 695
 696     memset(&local, 0, sizeof(local));
 697     local.sin6_family = AF_INET6;
 698
 699     if (bind(fd, (struct sockaddr*) &local, sizeof(local)) < 0) {
 700         avahi_log_warn("bind() failed: %s", strerror(errno));
 701         goto fail;
 702     }
 703
 704     yes = 1;
 705     if (setsockopt(fd, IPPROTO_IPV6, IPV6_HOPLIMIT, &yes, sizeof(yes)) < 0) {
 706         avahi_log_warn("IPV6_HOPLIMIT failed: %s", strerror(errno));
 707         goto fail;
 708     }
 709
 710     yes = 1;
 711     if (setsockopt(fd, IPPROTO_IPV6, IPV6_PKTINFO, &yes, sizeof(yes)) < 0) {
 712         avahi_log_warn("IPV6_PKTINFO failed: %s", strerror(errno));
 713         goto fail;
 714     }
 715
 716     if (avahi_set_cloexec(fd) < 0) {
 717         avahi_log_warn("FD_CLOEXEC failed: %s", strerror(errno));
 718         goto fail;
 719     }
 720
 721     if (avahi_set_nonblock(fd) < 0) {
 722         avahi_log_warn("O_NONBLOCK failed: %s", strerror(errno));
 723         goto fail;
 724     }
 725
 726     return fd;
 727
 728 fail:
 729     if (fd >= 0)
 730         close(fd);
 731
 732     return -1;
 733 }