IO-FDPass/FDPass.xs

#ifdef __sun
  #define _XOPEN_SOURCE 1
  #define _XOPEN_SOURCE_EXTENDED 1
  #define __EXTENSIONS__ 1
#endif

#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"

#if WIN32

  /* perl probably did this already */
  #include <windows.h>

#elif __CYGWIN__

  #include <windows.h>
  #include <io.h>
  #include <sys/cygwin.h>

  #define _open_osfhandle(h,m) cygwin_attach_handle_to_fd ("/dev/tcp", -1, (HANDLE)h, 1, GENERIC_READ | GENERIC_WRITE)
  typedef int SOCKET;

#else

  #include <stddef.h> // needed by broken bsds for NULL used in sys/uio.h
  #include <stdlib.h>
  #include <errno.h>

  /* send_fd/recv_fd taken from libptytty */
  #include <sys/types.h>
  #include <sys/uio.h>
  #include <sys/socket.h>

  #ifndef CMSG_SPACE
  # define CMSG_SPACE(len) (sizeof (cmsghdr) + len)
  #endif

  #ifndef CMSG_LEN
  # define CMSG_LEN(len) (sizeof (cmsghdr) + len)
  #endif

#endif

#if defined(WIN32)
/* the rub is this: win32 doesn't seem to have a way to query whether a socket */
/* is non-blocking or not. so we assume it is blocking, make it so if it isn't */
/* and reset it afterwards */
static int
rw (int wr, int fd, char *buf, int len)
{
  u_long nbio = 0;
  int got = 0;

  while (got != len)
    {
      int sze = wr
              ? send ((SOCKET)fd, buf, len - got, 0)  /* we assume send and recv are macros with arguments */
              : recv ((SOCKET)fd, buf, len - got, 0); /* to be on the safe side */

      if (sze < 0)
        {
          if (errno == EAGAIN || errno == WSAEWOULDBLOCK)
            {
              ioctl (fd, FIONBIO, (void *)&nbio);
              nbio = 1;
            }
          else
            break;
        }
      else if (sze == 0)
        break;
      else
        got += sze;
    }

  if (nbio)
    ioctl (fd, FIONBIO, (void *)&nbio);

  return got == len;
}
#endif

static int
fd_send (int socket, int fd)
{
#if defined(WIN32)
  DWORD pid;
  HANDLE hdl;
  
  pid = GetCurrentProcessId ();

  if (!rw (1, socket, (char *)&pid, sizeof (pid)))
    return 0;

  errno = EBADF;
  if (!DuplicateHandle ((HANDLE)-1, (HANDLE)_get_osfhandle (fd), (HANDLE)-1, &hdl, 0, FALSE, DUPLICATE_SAME_ACCESS))
    return 0;

  if (!rw (1, socket, (char *)&hdl, sizeof (hdl)))
    {
      CloseHandle (hdl);
      return 0;
    }

  return 1;

#else
  void *buf = malloc (CMSG_SPACE (sizeof (int)));

  if (!buf)
    return 0;

  struct msghdr msg;
  struct iovec iov;
  struct cmsghdr *cmsg;
  char data = 0;

  iov.iov_base = &data;
  iov.iov_len  = 1;

  msg.msg_name       = 0;
  msg.msg_namelen    = 0;
  msg.msg_iov        = &iov;
  msg.msg_iovlen     = 1;
  msg.msg_control    = buf;
  msg.msg_controllen = CMSG_SPACE (sizeof (int));

  cmsg = CMSG_FIRSTHDR (&msg);
  cmsg->cmsg_level = SOL_SOCKET;
  cmsg->cmsg_type  = SCM_RIGHTS;
  cmsg->cmsg_len   = CMSG_LEN (sizeof (int));

  *(int *)CMSG_DATA (cmsg) = fd;

  ssize_t result = sendmsg (socket, &msg, 0);

  free (buf);

  return result >= 0;
#endif
}

static int
fd_recv (int socket)
{
#if defined(WIN32)
  DWORD pid;
  HANDLE source, rhd, lhd;

  if (!rw (0, socket, (char *)&pid, sizeof (pid)))
    return -1;

  if (!rw (0, socket, (char *)&rhd, sizeof (rhd)))
    return -1;

  source = OpenProcess (PROCESS_DUP_HANDLE, FALSE, pid);
  errno = EACCES;
  if (!source)
    return -1;

  pid = DuplicateHandle (source, rhd, (HANDLE)-1, &lhd,
                         0, FALSE, DUPLICATE_SAME_ACCESS | DUPLICATE_CLOSE_SOURCE);

  CloseHandle (source);

  errno = EBADF;
  if (!pid)
    return -1;

  return _open_osfhandle ((intptr_t)lhd, 0);
#else
  void *buf = malloc (CMSG_SPACE (sizeof (int)));

  if (!buf)
    return -1;

  struct msghdr msg;
  struct iovec iov;
  char data = 1;

  iov.iov_base = &data;
  iov.iov_len  = 1;

  msg.msg_name       = 0;
  msg.msg_namelen    = 0;
  msg.msg_iov        = &iov;
  msg.msg_iovlen     = 1;
  msg.msg_control    = buf;
  msg.msg_controllen = CMSG_SPACE (sizeof (int));

  if (recvmsg (socket, &msg, 0) <= 0)
    return -1;

  int fd = -1;

  errno = EDOM;

  if (
     data == 0
#if __OpenBSD__
     && msg.msg_controllen >= CMSG_LEN   (sizeof (int)) /* work around a bug in at least openbsd 4.5 and 4.8 */
#else
     && msg.msg_controllen >= CMSG_SPACE (sizeof (int))
#endif
  ) {
      struct cmsghdr *cmsg = CMSG_FIRSTHDR (&msg);

      if (cmsg->cmsg_level   == SOL_SOCKET
          && cmsg->cmsg_type == SCM_RIGHTS
          && cmsg->cmsg_len  >= CMSG_LEN (sizeof (int)))
        fd = *(int *)CMSG_DATA (cmsg);
    }

  free (buf);

  return fd;
#endif
}

MODULE = IO::FDPass             PACKAGE = IO::FDPass            PREFIX = fd_

PROTOTYPES: DISABLE

int
fd_send (int socket, int fd)

int
fd_recv (int socket)

Revision:	1.5
Committed:	Fri Apr 5 08:20:36 2013 UTC (11 years, 1 month ago) by root
Branch:	MAIN
CVS Tags:	rel-1_0, rel-0_2
Changes since 1.4:	+38 -24 lines
Log Message:	0.2
#	User	Rev	Content
1	root	1.1	#ifdef __sun
2			#define _XOPEN_SOURCE 1
3			#define _XOPEN_SOURCE_EXTENDED 1
4			#define __EXTENSIONS__ 1
5			#endif
6
7			#include "EXTERN.h"
8			#include "perl.h"
9			#include "XSUB.h"
10
11			#if WIN32
12
13			/* perl probably did this already */
14			#include <windows.h>
15
16			#elif __CYGWIN__
17
18			#include <windows.h>
19			#include <io.h>
20			#include <sys/cygwin.h>
21
22	root	1.4	#define _open_osfhandle(h,m) cygwin_attach_handle_to_fd ("/dev/tcp", -1, (HANDLE)h, 1, GENERIC_READ \| GENERIC_WRITE)
23	root	1.1	typedef int SOCKET;
24
25			#else
26
27			#include <stddef.h> // needed by broken bsds for NULL used in sys/uio.h
28			#include <stdlib.h>
29			#include <errno.h>
30
31			/* send_fd/recv_fd taken from libptytty */
32			#include <sys/types.h>
33			#include <sys/uio.h>
34			#include <sys/socket.h>
35
36			#ifndef CMSG_SPACE
37			# define CMSG_SPACE(len) (sizeof (cmsghdr) + len)
38			#endif
39
40			#ifndef CMSG_LEN
41			# define CMSG_LEN(len) (sizeof (cmsghdr) + len)
42			#endif
43
44			#endif
45
46			#if defined(WIN32)
47			/* the rub is this: win32 doesn't seem to have a way to query whether a socket */
48			/* is non-blocking or not. so we assume it is blocking, make it so if it isn't */
49			/* and reset it afterwards */
50			static int
51			rw (int wr, int fd, char *buf, int len)
52			{
53			u_long nbio = 0;
54			int got = 0;
55
56			while (got != len)
57			{
58			int sze = wr
59	root	1.5	? send ((SOCKET)fd, buf, len - got, 0) /* we assume send and recv are macros with arguments */
60			: recv ((SOCKET)fd, buf, len - got, 0); /* to be on the safe side */
61	root	1.1
62	root	1.5	if (sze < 0)
63	root	1.1	{
64	root	1.2	if (errno == EAGAIN \|\| errno == WSAEWOULDBLOCK)
65	root	1.1	{
66	root	1.5	ioctl (fd, FIONBIO, (void *)&nbio);
67	root	1.1	nbio = 1;
68			}
69			else
70			break;
71			}
72	root	1.5	else if (sze == 0)
73			break;
74	root	1.1	else
75			got += sze;
76			}
77
78			if (nbio)
79	root	1.5	ioctl (fd, FIONBIO, (void *)&nbio);
80	root	1.1
81			return got == len;
82			}
83			#endif
84
85			static int
86			fd_send (int socket, int fd)
87			{
88			#if defined(WIN32)
89			DWORD pid;
90	root	1.5	HANDLE hdl;
91
92			pid = GetCurrentProcessId ();
93	root	1.1
94	root	1.5	if (!rw (1, socket, (char *)&pid, sizeof (pid)))
95	root	1.1	return 0;
96
97	root	1.5	errno = EBADF;
98			if (!DuplicateHandle ((HANDLE)-1, (HANDLE)_get_osfhandle (fd), (HANDLE)-1, &hdl, 0, FALSE, DUPLICATE_SAME_ACCESS))
99			return 0;
100	root	1.1
101	root	1.5	if (!rw (1, socket, (char *)&hdl, sizeof (hdl)))
102			{
103			CloseHandle (hdl);
104			return 0;
105			}
106	root	1.1
107			return 1;
108
109			#else
110			void *buf = malloc (CMSG_SPACE (sizeof (int)));
111
112			if (!buf)
113			return 0;
114
115			struct msghdr msg;
116			struct iovec iov;
117			struct cmsghdr *cmsg;
118			char data = 0;
119
120			iov.iov_base = &data;
121			iov.iov_len = 1;
122
123			msg.msg_name = 0;
124			msg.msg_namelen = 0;
125			msg.msg_iov = &iov;
126			msg.msg_iovlen = 1;
127			msg.msg_control = buf;
128			msg.msg_controllen = CMSG_SPACE (sizeof (int));
129
130			cmsg = CMSG_FIRSTHDR (&msg);
131			cmsg->cmsg_level = SOL_SOCKET;
132			cmsg->cmsg_type = SCM_RIGHTS;
133			cmsg->cmsg_len = CMSG_LEN (sizeof (int));
134
135			(int )CMSG_DATA (cmsg) = fd;
136
137			ssize_t result = sendmsg (socket, &msg, 0);
138
139			free (buf);
140
141			return result >= 0;
142			#endif
143			}
144
145			static int
146			fd_recv (int socket)
147			{
148			#if defined(WIN32)
149	root	1.5	DWORD pid;
150			HANDLE source, rhd, lhd;
151	root	1.1
152	root	1.5	if (!rw (0, socket, (char *)&pid, sizeof (pid)))
153			return -1;
154
155			if (!rw (0, socket, (char *)&rhd, sizeof (rhd)))
156			return -1;
157
158			source = OpenProcess (PROCESS_DUP_HANDLE, FALSE, pid);
159			errno = EACCES;
160			if (!source)
161	root	1.1	return -1;
162
163	root	1.5	pid = DuplicateHandle (source, rhd, (HANDLE)-1, &lhd,
164			0, FALSE, DUPLICATE_SAME_ACCESS \| DUPLICATE_CLOSE_SOURCE);
165
166			CloseHandle (source);
167
168			errno = EBADF;
169			if (!pid)
170	root	1.1	return -1;
171
172	root	1.5	return _open_osfhandle ((intptr_t)lhd, 0);
173	root	1.1	#else
174			void *buf = malloc (CMSG_SPACE (sizeof (int)));
175
176			if (!buf)
177			return -1;
178
179			struct msghdr msg;
180			struct iovec iov;
181			char data = 1;
182
183			iov.iov_base = &data;
184			iov.iov_len = 1;
185
186			msg.msg_name = 0;
187			msg.msg_namelen = 0;
188			msg.msg_iov = &iov;
189			msg.msg_iovlen = 1;
190			msg.msg_control = buf;
191			msg.msg_controllen = CMSG_SPACE (sizeof (int));
192
193			if (recvmsg (socket, &msg, 0) <= 0)
194			return -1;
195
196			int fd = -1;
197
198			errno = EDOM;
199
200			if (
201			data == 0
202			#if __OpenBSD__
203			&& msg.msg_controllen >= CMSG_LEN (sizeof (int)) /* work around a bug in at least openbsd 4.5 and 4.8 */
204			#else
205			&& msg.msg_controllen >= CMSG_SPACE (sizeof (int))
206			#endif
207			) {
208			struct cmsghdr *cmsg = CMSG_FIRSTHDR (&msg);
209
210			if (cmsg->cmsg_level == SOL_SOCKET
211			&& cmsg->cmsg_type == SCM_RIGHTS
212			&& cmsg->cmsg_len >= CMSG_LEN (sizeof (int)))
213			fd = (int )CMSG_DATA (cmsg);
214			}
215
216			free (buf);
217
218			return fd;
219			#endif
220			}
221
222			MODULE = IO::FDPass PACKAGE = IO::FDPass PREFIX = fd_
223
224			PROTOTYPES: DISABLE
225
226			int
227			fd_send (int socket, int fd)
228
229			int
230			fd_recv (int socket)
231