src/net/socket.c

   1 /* ncmpc (Ncurses MPD Client)
   2    (c) 2004-2017 The Music Player Daemon Project
   3    Project homepage: http://musicpd.org
   4
   5    Redistribution and use in source and binary forms, with or without
   6    modification, are permitted provided that the following conditions
   7    are met:
   8
   9    - Redistributions of source code must retain the above copyright
  10    notice, this list of conditions and the following disclaimer.
  11
  12    - Redistributions in binary form must reproduce the above copyright
  13    notice, this list of conditions and the following disclaimer in the
  14    documentation and/or other materials provided with the distribution.
  15
  16    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  19    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
  20    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  21    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  22    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  23    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  24    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  25    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  26    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27 */
  28
  29 #include "socket.h"
  30
  31 #ifndef WIN32
  32 #include <fcntl.h>
  33 #endif
  34
  35 static void
  36 socket_set_cloexec(socket_t fd)
  37 {
  38 #ifndef WIN32
  39         fcntl(fd, F_SETFD, FD_CLOEXEC);
  40 #else
  41         (void)fd;
  42 #endif
  43 }
  44
  45 static void
  46 socket_set_nonblock(socket_t fd)
  47 {
  48 #ifdef WIN32
  49         u_long val = 1;
  50         ioctlsocket(fd, FIONBIO, &val);
  51 #else
  52         int flags = fcntl(fd, F_GETFL);
  53         if (flags >= 0)
  54                 fcntl(fd, F_SETFL, flags | O_NONBLOCK);
  55 #endif
  56 }
  57
  58 socket_t
  59 create_socket(int domain, int type, int protocol)
  60 {
  61         socket_t fd;
  62
  63 #if defined(SOCK_CLOEXEC) && defined(SOCK_NONBLOCK)
  64         fd = socket(domain, type | SOCK_CLOEXEC | SOCK_NONBLOCK, protocol);
  65         if (fd != INVALID_SOCKET || errno != EINVAL)
  66                 return fd;
  67 #endif
  68
  69         fd = socket(domain, type, protocol);
  70         if (fd != INVALID_SOCKET) {
  71                 socket_set_cloexec(fd);
  72                 socket_set_nonblock(fd);
  73         }
  74
  75         return fd;
  76 }