cosmopolitan/libc/sock/socket-nt.c

/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :vi│
╞══════════════════════════════════════════════════════════════════════════════╡
│ Copyright 2020 Justine Alexandra Roberts Tunney                              │
│                                                                              │
│ Permission to use, copy, modify, and/or distribute this software for         │
│ any purpose with or without fee is hereby granted, provided that the         │
│ above copyright notice and this permission notice appear in all copies.      │
│                                                                              │
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL                │
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED                │
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE             │
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL         │
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR        │
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER               │
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR             │
│ PERFORMANCE OF THIS SOFTWARE.                                                │
╚─────────────────────────────────────────────────────────────────────────────*/
#include "libc/assert.h"
#include "libc/calls/internal.h"
#include "libc/calls/state.internal.h"
#include "libc/errno.h"
#include "libc/mem/mem.h"
#include "libc/nt/enum/fileflagandattributes.h"
#include "libc/nt/iphlpapi.h"
#include "libc/nt/thunk/msabi.h"
#include "libc/nt/winsock.h"
#include "libc/sock/internal.h"
#include "libc/sock/yoink.inc"
#include "libc/str/str.h"
#include "libc/sysv/consts/af.h"
#include "libc/sysv/consts/ipproto.h"
#include "libc/sysv/consts/o.h"
#include "libc/sysv/consts/so.h"
#include "libc/sysv/consts/sock.h"
#include "libc/sysv/consts/sol.h"

__msabi extern typeof(__sys_setsockopt_nt) *const __imp_setsockopt;

/*
 * ioctl(SIOCGIFCONFIG) for Windows need to access the following
 * functions through weak reference. This ensure those symbols are not
 * stripped during final link.
 */
__static_yoink("GetAdaptersAddresses");
__static_yoink("tprecode16to8");
__static_yoink("_dupsockfd");

textwindows int sys_socket_nt(int family, int type, int protocol) {
  int64_t h;
  struct SockFd *sockfd;
  int fd, oflags, truetype, yes = 1;
  fd = __reservefd(-1);
  if (fd == -1) return -1;
  truetype = type & ~(SOCK_CLOEXEC | SOCK_NONBLOCK);
  if ((h = WSASocket(family, truetype, protocol, NULL, 0,
                     kNtWsaFlagOverlapped)) != -1) {

    // sets SO_EXCLUSIVEADDRUSE on all sockets so they won't get pilfered
    // you can read a blog post on this subject in the find_unused_port()
    // pydoc of this file third_party/python/Lib/test/support/__init__.py
    // this needs to happen right after socket is called or it won't work
    if (family == AF_INET || family == AF_INET6) {
      unassert(__imp_setsockopt(h, SOL_SOCKET, -5, &yes, 4) != -1);
    }

    oflags = O_RDWR;
    if (type & SOCK_CLOEXEC) oflags |= O_CLOEXEC;
    if (type & SOCK_NONBLOCK) oflags |= O_NONBLOCK;
    sockfd = calloc(1, sizeof(struct SockFd));
    sockfd->family = family;
    sockfd->type = truetype;
    sockfd->protocol = protocol;
    __fds_lock();
    g_fds.p[fd].kind = kFdSocket;
    g_fds.p[fd].flags = oflags;
    g_fds.p[fd].mode = 0140666;
    g_fds.p[fd].handle = h;
    g_fds.p[fd].extra = (uintptr_t)sockfd;
    __fds_unlock();

    return fd;
  } else {
    __releasefd(fd);
    return __winsockerr();
  }
}