usr.sbin/dntpd/socket.c

   1 /*
   2  * Copyright (c) 2005 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Matthew Dillon <dillon@backplane.com>
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the
  16  *    distribution.
  17  * 3. Neither the name of The DragonFly Project nor the names of its
  18  *    contributors may be used to endorse or promote products derived
  19  *    from this software without specific, prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $DragonFly: src/usr.sbin/dntpd/socket.c,v 1.4 2007/06/25 21:33:36 dillon Exp $
  35  */
  36
  37 #include "defs.h"
  38
  39 int
  40 udp_socket(const char *target, int port, struct sockaddr *sam,
  41            dns_error_policy_t dns_error_policy)
  42 {
  43     struct addrinfo hints, *res, *res0;
  44     char servname[128];
  45     const char *cause = NULL;
  46     int error;
  47     int fd;
  48     int tos;
  49
  50     memset(&hints, 0, sizeof(hints));
  51     hints.ai_family = family;
  52     hints.ai_socktype = SOCK_DGRAM;
  53     snprintf(servname, sizeof(servname), "%d", port);
  54     error = getaddrinfo(target, servname, &hints, &res0);
  55     if (error) {
  56         if (dns_error_policy == LOG_DNS_ERROR)
  57             logerr("getaddrinfo (%s) init error", target, gai_strerror(error));
  58         return(-1);
  59     }
  60
  61     fd = -1;
  62     for (res = res0; res; res = res->ai_next) {
  63         fd = socket(res->ai_family, res->ai_socktype,
  64         res->ai_protocol);
  65         if (fd < 0) {
  66            cause = "socket";
  67            continue;
  68         }
  69
  70         if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0) {
  71             logerr("%s: unable to set non-blocking mode", target);
  72             close(fd);
  73             fd = -1;
  74             continue;
  75         }
  76
  77         if (connect(fd, res->ai_addr, res->ai_addrlen) < 0) {
  78            cause = "connect";
  79            close(fd);
  80            fd = -1;
  81            continue;
  82         }
  83
  84         break;  /* okay we got one */
  85     }
  86
  87     if (fd < 0) {
  88         logerr("Unable to establish a connection with %s: %s", target, cause);
  89         return(-1);
  90     }
  91     memcpy(sam, res->ai_addr, res->ai_addr->sa_len);
  92     freeaddrinfo(res0);
  93
  94 #ifdef IPTOS_LOWDELAY
  95     tos = IPTOS_LOWDELAY;
  96     setsockopt(fd, IPPROTO_IP, IP_TOS, &tos, sizeof(tos));
  97 #endif
  98 #if 0
  99 #ifdef IP_PORTRANGE
 100     tos = IP_PORTRANGE_HIGH;
 101     setsockopt(fd, IPPROTO_IP, IP_PORTRANGE, &tos, sizeof(tos));
 102 #endif
 103 #endif
 104     return(fd);
 105 }