gl/mesa/src/gallium/auxiliary/util/u_network.c

   1
   2 #include "pipe/p_compiler.h"
   3 #include "util/u_network.h"
   4 #include "util/u_debug.h"
   5
   6 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER)
   7 #  include <winsock2.h>
   8 #  include <windows.h>
   9 #elif defined(PIPE_OS_LINUX) || defined(PIPE_OS_HAIKU) || \
  10    defined(PIPE_OS_APPLE) || defined(PIPE_OS_CYGWIN) || defined(PIPE_OS_SOLARIS)
  11 #  include <sys/socket.h>
  12 #  include <netinet/in.h>
  13 #  include <unistd.h>
  14 #  include <fcntl.h>
  15 #  include <netdb.h>
  16 #else
  17 #  warning "No socket implementation"
  18 #endif
  19
  20 boolean
  21 u_socket_init()
  22 {
  23 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER)
  24    WORD wVersionRequested;
  25    WSADATA wsaData;
  26    int err;
  27
  28    /* Use the MAKEWORD(lowbyte, highbyte) macro declared in Windef.h */
  29    wVersionRequested = MAKEWORD(1, 1);
  30
  31    err = WSAStartup(wVersionRequested, &wsaData);
  32    if (err != 0) {
  33       debug_printf("WSAStartup failed with error: %d\n", err);
  34       return FALSE;
  35    }
  36    return TRUE;
  37 #elif defined(PIPE_HAVE_SOCKETS)
  38    return TRUE;
  39 #else
  40    return FALSE;
  41 #endif
  42 }
  43
  44 void
  45 u_socket_stop()
  46 {
  47 #if defined(PIPE_SUBSYSTEM_WINDOWS_USER)
  48    WSACleanup();
  49 #endif
  50 }
  51
  52 void
  53 u_socket_close(int s)
  54 {
  55    if (s < 0)
  56       return;
  57
  58 #if defined(PIPE_OS_LINUX) || defined(PIPE_OS_HAIKU) \
  59     || defined(PIPE_OS_APPLE) || defined(PIPE_OS_SOLARIS)
  60    shutdown(s, SHUT_RDWR);
  61    close(s);
  62 #elif defined(PIPE_SUBSYSTEM_WINDOWS_USER)
  63    shutdown(s, SD_BOTH);
  64    closesocket(s);
  65 #else
  66    assert(0);
  67 #endif
  68 }
  69
  70 int u_socket_accept(int s)
  71 {
  72 #if defined(PIPE_HAVE_SOCKETS)
  73    return accept(s, NULL, NULL);
  74 #else
  75    return -1;
  76 #endif
  77 }
  78
  79 int
  80 u_socket_send(int s, void *data, size_t size)
  81 {
  82 #if defined(PIPE_HAVE_SOCKETS)
  83    return send(s, data, size, 0);
  84 #else
  85    return -1;
  86 #endif
  87 }
  88
  89 int
  90 u_socket_peek(int s, void *data, size_t size)
  91 {
  92 #if defined(PIPE_HAVE_SOCKETS)
  93    return recv(s, data, size, MSG_PEEK);
  94 #else
  95    return -1;
  96 #endif
  97 }
  98
  99 int
 100 u_socket_recv(int s, void *data, size_t size)
 101 {
 102 #if defined(PIPE_HAVE_SOCKETS)
 103    return recv(s, data, size, 0);
 104 #else
 105    return -1;
 106 #endif
 107 }
 108
 109 int
 110 u_socket_connect(const char *hostname, uint16_t port)
 111 {
 112 #if defined(PIPE_HAVE_SOCKETS)
 113    int s;
 114    struct sockaddr_in sa;
 115    struct hostent *host = NULL;
 116
 117    memset(&sa, 0, sizeof(struct sockaddr_in));
 118    host = gethostbyname(hostname);
 119    if (!host)
 120       return -1;
 121
 122    memcpy((char *)&sa.sin_addr,host->h_addr_list[0],host->h_length);
 123    sa.sin_family= host->h_addrtype;
 124    sa.sin_port = htons(port);
 125
 126    s = socket(host->h_addrtype, SOCK_STREAM, IPPROTO_TCP);
 127    if (s < 0)
 128       return -1;
 129
 130    if (connect(s, (struct sockaddr *)&sa, sizeof(sa))) {
 131       u_socket_close(s);
 132       return -1;
 133    }
 134
 135    return s;
 136 #else
 137    assert(0);
 138    return -1;
 139 #endif
 140 }
 141
 142 int
 143 u_socket_listen_on_port(uint16_t portnum)
 144 {
 145 #if defined(PIPE_HAVE_SOCKETS)
 146    int s;
 147    struct sockaddr_in sa;
 148    memset(&sa, 0, sizeof(struct sockaddr_in));
 149
 150    sa.sin_family = AF_INET;
 151    sa.sin_port = htons(portnum);
 152
 153    s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
 154    if (s < 0)
 155       return -1;
 156
 157    if (bind(s, (struct sockaddr *)&sa, sizeof(struct sockaddr_in)) == -1) {
 158       u_socket_close(s);
 159       return -1;
 160    }
 161
 162    listen(s, 0);
 163
 164    return s;
 165 #else
 166    assert(0);
 167    return -1;
 168 #endif
 169 }
 170
 171 void
 172 u_socket_block(int s, boolean block)
 173 {
 174 #if defined(PIPE_OS_LINUX) || defined(PIPE_OS_HAIKU) \
 175     || defined(PIPE_OS_APPLE) || defined(PIPE_OS_SOLARIS)
 176    int old = fcntl(s, F_GETFL, 0);
 177    if (old == -1)
 178       return;
 179
 180    /* TODO obey block */
 181    if (block)
 182       fcntl(s, F_SETFL, old & ~O_NONBLOCK);
 183    else
 184       fcntl(s, F_SETFL, old | O_NONBLOCK);
 185 #elif defined(PIPE_SUBSYSTEM_WINDOWS_USER)
 186    u_long iMode = block ? 0 : 1;
 187    ioctlsocket(s, FIONBIO, &iMode);
 188 #else
 189    assert(0);
 190 #endif
 191 }