kc3-lang/libressl/tests/compat/pipe2.c

• Show log

  Commit

• Hash : dbd5fb28
  Author :
  Date : 2017-01-15T10:17:13
  

  add windows versions of socketpair / pipe
  

Download

• tests/compat/pipe2.c

• /*
   * Public domain
   *
   * pipe2/pipe/socketpair emulation
   * Brent Cook <bcook@openbsd.org>
   */
  
  #include <errno.h>
  #include <fcntl.h>
  #include <unistd.h>
  #include <sys/socket.h>
  
  #undef socketpair
  
  #ifdef _WIN32
  
  static int setfd(int fd, int flag)
  {
  	int rc = -1;
  	if (flag & FD_CLOEXEC) {
  		HANDLE h = (HANDLE)_get_osfhandle(fd);
  		if (h != NULL)
  			rc = SetHandleInformation(h, HANDLE_FLAG_INHERIT, 0) == 0 ? -1 : 0;
  	}
  	return rc;
  }
  
  static int setfl(int fd, int flag)
  {
  	int rc = -1;
  	if (flag & O_NONBLOCK) {
  		long mode = 1;
  		rc = ioctlsocket(fd, FIONBIO, &mode);
  	}
  	return rc;
  }
  
  int socketpair(int domain, int type, int protocol, int socket_vector[2])
  {
  	if (domain != AF_UNIX || !(type & SOCK_STREAM) || protocol != PF_UNSPEC)
  		return -1;
  
  	socket_vector[0] = -1;
  	socket_vector[1] = -1;
  
  	int listener = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
  	if (listener == -1) {
  		return -1;
  	}
  
  	struct sockaddr_in addr = {
  		.sin_family = AF_INET,
  		.sin_addr.s_addr = htonl(INADDR_LOOPBACK),
  		.sin_port = 0,
  	};
  
  	int yes = 1, e;
  	if (setsockopt(listener, SOL_SOCKET, SO_REUSEADDR,
  			(void *)&yes, sizeof yes) == -1)
  		goto err;
  
  	if (bind(listener, (struct sockaddr *)&addr, sizeof addr) != 0)
  		goto err;
  
  	memset(&addr, 0, sizeof addr);
  	socklen_t addrlen = sizeof addr;
  	if (getsockname(listener, (struct sockaddr *)&addr, &addrlen) != 0)
  		goto err;
  
  	addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
  	addr.sin_family = AF_INET;
  
  	if (listen(listener, 1) != 0)
  		goto err;
  
  	socket_vector[0] = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, 0);
  	if (socket_vector[0] == -1)
  		goto err;
  
  	if (connect(socket_vector[0], (struct sockaddr *)&addr, sizeof addr) != 0)
  		goto err;
  
  	socket_vector[1] = accept(listener, NULL, NULL);
  	if (socket_vector[1] == -1)
  		goto err;
  
  	closesocket(listener);
  	return 0;
  
  err:
  	e = WSAGetLastError();
  	closesocket(listener);
  	closesocket(socket_vector[0]);
  	closesocket(socket_vector[1]);
  	WSASetLastError(e);
  	socket_vector[0] = -1;
  	socket_vector[1] = -1;
  	return -1;
  }
  
  int pipe(int fildes[2])
  {
  	return socketpair(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK, PF_UNSPEC, fildes);
  }
  
  #else
  
  static int setfd(int fd, int flag)
  {
  	int flags = fcntl(fd, F_GETFD);
  	flags |= flag;
  	return fcntl(fd, F_SETFD, flags);
  }
  
  static int setfl(int fd, int flag)
  {
  	int flags = fcntl(fd, F_GETFL);
  	flags |= flag;
  	return fcntl(fd, F_SETFL, flags);
  }
  #endif
  
  int pipe2(int fildes[2], int flags)
  {
  	int rc = pipe(fildes);
  	if (rc == 0) {
  		if (flags & O_NONBLOCK) {
  			rc |= setfl(fildes[0], O_NONBLOCK);
  			rc |= setfl(fildes[1], O_NONBLOCK);
  		}
  		if (flags & O_CLOEXEC) {
  			rc |= setfd(fildes[0], FD_CLOEXEC);
  			rc |= setfd(fildes[1], FD_CLOEXEC);
  		}
  		if (rc != 0) {
  			int e = errno;
  			close(fildes[0]);
  			close(fildes[1]);
  			errno = e;
  		}
  	}
  	return rc;
  }
  
  int bsd_socketpair(int domain, int type, int protocol, int socket_vector[2])
  {
  	int flags = type & ~0xf;
  	type &= 0xf;
  	int rc = socketpair(domain, type, protocol, socket_vector);
  	if (rc == 0) {
  		if (flags & SOCK_NONBLOCK) {
  			rc |= setfl(socket_vector[0], O_NONBLOCK);
  			rc |= setfl(socket_vector[1], O_NONBLOCK);
  		}
  		if (flags & SOCK_CLOEXEC) {
  			rc |= setfd(socket_vector[0], FD_CLOEXEC);
  			rc |= setfd(socket_vector[1], FD_CLOEXEC);
  		}
  		if (rc != 0) {
  			int e = errno;
  			close(socket_vector[0]);
  			close(socket_vector[1]);
  			errno = e;
  		}
  	}
  	return rc;
  }
  

Download