lib/socket-util.c

   1 /* Copyright (c) 2008 The Board of Trustees of The Leland Stanford
   2  * Junior University
   3  *
   4  * We are making the OpenFlow specification and associated documentation
   5  * (Software) available for public use and benefit with the expectation
   6  * that others will use, modify and enhance the Software and contribute
   7  * those enhancements back to the community. However, since we would
   8  * like to make the Software available for broadest use, with as few
   9  * restrictions as possible permission is hereby granted, free of
  10  * charge, to any person obtaining a copy of this Software to deal in
  11  * the Software under the copyrights without restriction, including
  12  * without limitation the rights to use, copy, modify, merge, publish,
  13  * distribute, sublicense, and/or sell copies of the Software, and to
  14  * permit persons to whom the Software is furnished to do so, subject to
  15  * the following conditions:
  16  *
  17  * The above copyright notice and this permission notice shall be
  18  * included in all copies or substantial portions of the Software.
  19  *
  20  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23  * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  24  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  25  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  26  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  27  * SOFTWARE.
  28  *
  29  * The name and trademarks of copyright holder(s) may NOT be used in
  30  * advertising or publicity pertaining to the Software or any
  31  * derivatives without specific, written prior permission.
  32  */
  33
  34 #include "socket-util.h"
  35 #include <arpa/inet.h>
  36 #include <errno.h>
  37 #include <fcntl.h>
  38 #include <netdb.h>
  39 #include <poll.h>
  40 #include <stdio.h>
  41 #include <string.h>
  42
  43 #include "vlog.h"
  44 #define THIS_MODULE VLM_socket_util
  45
  46 /* Sets 'fd' to non-blocking mode.  Returns 0 if successful, otherwise a
  47  * positive errno value. */
  48 int
  49 set_nonblocking(int fd)
  50 {
  51     int flags = fcntl(fd, F_GETFL, 0);
  52     if (flags != -1) {
  53         if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) != -1) {
  54             return 0;
  55         } else {
  56             VLOG_ERR("fcntl(F_SETFL) failed: %s", strerror(errno));
  57             return errno;
  58         }
  59     } else {
  60         VLOG_ERR("fcntl(F_GETFL) failed: %s", strerror(errno));
  61         return errno;
  62     }
  63 }
  64
  65 /* Translates 'host_name', which may be a DNS name or an IP address, into a
  66  * numeric IP address in '*addr'.  Returns 0 if successful, otherwise a
  67  * positive errno value. */
  68 int
  69 lookup_ip(const char *host_name, struct in_addr *addr)
  70 {
  71     if (!inet_aton(host_name, addr)) {
  72         struct hostent *he = gethostbyname(host_name);
  73         if (he == NULL) {
  74             VLOG_ERR("gethostbyname(%s): %s", host_name,
  75                      (h_errno == HOST_NOT_FOUND ? "host not found"
  76                       : h_errno == TRY_AGAIN ? "try again"
  77                       : h_errno == NO_RECOVERY ? "non-recoverable error"
  78                       : h_errno == NO_ADDRESS ? "no address"
  79                       : "unknown error"));
  80             return ENOENT;
  81         }
  82         addr->s_addr = *(uint32_t *) he->h_addr;
  83     }
  84     return 0;
  85 }
  86
  87 /* Returns the error condition associated with socket 'fd' and resets the
  88  * socket's error status. */
  89 int
  90 get_socket_error(int fd)
  91 {
  92     int error;
  93     socklen_t len = sizeof(error);
  94     if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &error, &len) < 0) {
  95         error = errno;
  96         VLOG_ERR("getsockopt(SO_ERROR): %s", strerror(error));
  97     }
  98     return error;
  99 }
 100
 101 int
 102 check_connection_completion(int fd)
 103 {
 104     struct pollfd pfd;
 105     int retval;
 106
 107     pfd.fd = fd;
 108     pfd.events = POLLOUT;
 109     do {
 110         retval = poll(&pfd, 1, 0);
 111     } while (retval < 0 && errno == EINTR);
 112     if (retval == 1) {
 113         return get_socket_error(fd);
 114     } else if (retval < 0) {
 115         VLOG_ERR("poll: %s", strerror(errno));
 116         return errno;
 117     } else {
 118         return EAGAIN;
 119     }
 120 }