ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/EV/schmorp.h
(Generate patch)

Comparing EV/schmorp.h (file contents):
Revision 1.1 by root, Tue Jul 14 00:09:59 2009 UTC vs.
Revision 1.5 by root, Fri Jul 17 14:49:33 2009 UTC

1#ifndef SCHMORP_PERL_H_ 1#ifndef SCHMORP_PERL_H_
2#define SCHMORP_PERL_H_ 2#define SCHMORP_PERL_H_
3 3
4/* WARNING
5 * This header file is a shared resource between many modules.
6 */
7
8#include <signal.h>
9#include <errno.h>
10
11#ifndef _WIN32
12# include <poll.h>
13#endif
14
4/* useful stuff, used by schmorp mostly */ 15/* useful stuff, used by schmorp mostly */
16
17#include "patchlevel.h"
5 18
6#define PERL_VERSION_ATLEAST(a,b,c) \ 19#define PERL_VERSION_ATLEAST(a,b,c) \
7 (PERL_REVISION > (a) \ 20 (PERL_REVISION > (a) \
8 || (PERL_REVISION == (a) \ 21 || (PERL_REVISION == (a) \
9 && (PERL_VERSION > (b) \ 22 && (PERL_VERSION > (b) \
68#ifndef SIG_SIZE 81#ifndef SIG_SIZE
69 /* kudos to Slaven Rezic for the idea */ 82 /* kudos to Slaven Rezic for the idea */
70 static char sig_size [] = { SIG_NUM }; 83 static char sig_size [] = { SIG_NUM };
71# define SIG_SIZE (sizeof (sig_size) + 1) 84# define SIG_SIZE (sizeof (sig_size) + 1)
72#endif 85#endif
86 dTHX;
73 int signum; 87 int signum;
74 88
75 SvGETMAGIC (sig); 89 SvGETMAGIC (sig);
76 90
77 for (signum = 1; signum < SIG_SIZE; ++signum) 91 for (signum = 1; signum < SIG_SIZE; ++signum)
90s_signum_croak (SV *sig) 104s_signum_croak (SV *sig)
91{ 105{
92 int signum = s_signum (sig); 106 int signum = s_signum (sig);
93 107
94 if (signum < 0) 108 if (signum < 0)
109 {
110 dTHX;
95 croak ("%s: invalid signal name or number", SvPV_nolen (sig)); 111 croak ("%s: invalid signal name or number", SvPV_nolen (sig));
112 }
96 113
97 return signum; 114 return signum;
98} 115}
99 116
100static int 117static int
101s_fileno (SV *fh, int wr) 118s_fileno (SV *fh, int wr)
102{ 119{
120 dTHX;
103 SvGETMAGIC (fh); 121 SvGETMAGIC (fh);
104 122
105 if (SvROK (fh)) 123 if (SvROK (fh))
106 { 124 {
107 fh = SvRV (fh); 125 fh = SvRV (fh);
121s_fileno_croak (SV *fh, int wr) 139s_fileno_croak (SV *fh, int wr)
122{ 140{
123 int fd = s_fileno (fh, wr); 141 int fd = s_fileno (fh, wr);
124 142
125 if (fd < 0) 143 if (fd < 0)
144 {
145 dTHX;
126 croak ("%s: illegal fh argument, either not an OS file or read/write mode mismatch", SvPV_nolen (fh)); 146 croak ("%s: illegal fh argument, either not an OS file or read/write mode mismatch", SvPV_nolen (fh));
147 }
127 148
128 return fd; 149 return fd;
129} 150}
130 151
131static SV * 152static SV *
132s_get_cv (SV *cb_sv) 153s_get_cv (SV *cb_sv)
133{ 154{
155 dTHX;
134 HV *st; 156 HV *st;
135 GV *gvp; 157 GV *gvp;
158
136 CV *cv = sv_2cv (cb_sv, &st, &gvp, 0); 159 return (SV *)sv_2cv (cb_sv, &st, &gvp, 0);
137
138 return (SV *)cv;
139} 160}
140 161
141static SV * 162static SV *
142s_get_cv_croak (SV *cb_sv) 163s_get_cv_croak (SV *cb_sv)
143{ 164{
144 cb_sv = s_get_cv (cb_sv); 165 SV *cv = s_get_cv (cb_sv);
145 166
146 if (!cb_sv) 167 if (!cv)
168 {
169 dTHX;
147 croak ("%s: callback must be a CODE reference or another callable object", SvPV_nolen (cb_sv)); 170 croak ("%s: callback must be a CODE reference or another callable object", SvPV_nolen (cb_sv));
171 }
148 172
149 return cb_sv; 173 return cv;
150} 174}
151 175
152/*****************************************************************************/ 176/*****************************************************************************/
153/* gensub: simple closure generation utility */ 177/* gensub: simple closure generation utility */
154 178
170 S_GENSUB_ARG = arg; 194 S_GENSUB_ARG = arg;
171 195
172 return newRV_noinc ((SV *)cv); 196 return newRV_noinc ((SV *)cv);
173} 197}
174 198
199/*****************************************************************************/
200/* portable pipe/socketpair */
201
202#ifdef USE_SOCKETS_AS_HANDLES
203# define S_TO_HANDLE(x) ((HANDLE)win32_get_osfhandle (x))
204#else
205# define S_TO_HANDLE(x) ((HANDLE)x)
206#endif
207
208#ifdef _WIN32
209/* taken almost verbatim from libev's ev_win32.c */
210/* oh, the humanity! */
211static int
212s_pipe (int filedes [2])
213{
214 struct sockaddr_in addr = { 0 };
215 int addr_size = sizeof (addr);
216 struct sockaddr_in adr2;
217 int adr2_size = sizeof (adr2);
218 SOCKET listener;
219 SOCKET sock [2] = { -1, -1 };
220
221 if ((listener = socket (AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET)
222 return -1;
223
224 addr.sin_family = AF_INET;
225 addr.sin_addr.s_addr = htonl (INADDR_LOOPBACK);
226 addr.sin_port = 0;
227
228 if (bind (listener, (struct sockaddr *)&addr, addr_size))
229 goto fail;
230
231 if (getsockname (listener, (struct sockaddr *)&addr, &addr_size))
232 goto fail;
233
234 if (listen (listener, 1))
235 goto fail;
236
237 if ((sock [0] = socket (AF_INET, SOCK_STREAM, 0)) == INVALID_SOCKET)
238 goto fail;
239
240 if (connect (sock [0], (struct sockaddr *)&addr, addr_size))
241 goto fail;
242
243 if ((sock [1] = accept (listener, 0, 0)) < 0)
244 goto fail;
245
246 /* windows vista returns fantasy port numbers for getpeername.
247 * example for two interconnected tcp sockets:
248 *
249 * (Socket::unpack_sockaddr_in getsockname $sock0)[0] == 53364
250 * (Socket::unpack_sockaddr_in getpeername $sock0)[0] == 53363
251 * (Socket::unpack_sockaddr_in getsockname $sock1)[0] == 53363
252 * (Socket::unpack_sockaddr_in getpeername $sock1)[0] == 53365
253 *
254 * wow! tridirectional sockets!
255 *
256 * this way of checking ports seems to work:
257 */
258 if (getpeername (sock [0], (struct sockaddr *)&addr, &addr_size))
259 goto fail;
260
261 if (getsockname (sock [1], (struct sockaddr *)&adr2, &adr2_size))
262 goto fail;
263
264 errno = WSAEINVAL;
265 if (addr_size != adr2_size
266 || addr.sin_addr.s_addr != adr2.sin_addr.s_addr /* just to be sure, I mean, it's windows */
267 || addr.sin_port != adr2.sin_port)
268 goto fail;
269
270 closesocket (listener);
271
272#ifdef USE_SOCKETS_AS_HANDLES
273 /* when select isn't winsocket, we also expect socket, connect, accept etc.
274 * to work on fds */
275 filedes [0] = sock [0];
276 filedes [1] = sock [1];
277#else
278 filedes [0] = _open_osfhandle (sock [0], 0);
279 filedes [1] = _open_osfhandle (sock [1], 0);
280#endif
281
282 return 0;
283
284fail:
285 closesocket (listener);
286
287 if (sock [0] != INVALID_SOCKET) closesocket (sock [0]);
288 if (sock [1] != INVALID_SOCKET) closesocket (sock [1]);
289
290 return -1;
291}
292
293#define s_socketpair(domain,type,protocol,filedes) s_pipe (filedes)
294
295static int
296s_fd_blocking (int fd, int blocking)
297{
298 u_long nonblocking = !blocking;
299
300 return ioctlsocket ((SOCKET)S_TO_HANDLE (fd), FIONBIO, &nonblocking);
301}
302
303#define s_fd_prepare(fd) s_fd_blocking (fd, 0)
304
305#else
306
307#define s_socketpair(domain,type,protocol,filedes) socketpair (domain, type, protocol, filedes)
308#define s_pipe(filedes) pipe (filedes)
309
310static int
311s_fd_blocking (int fd, int blocking)
312{
313 return fcntl (fd, F_SETFL, blocking ? 0 : O_NONBLOCK);
314}
315
316static int
317s_fd_prepare (int fd)
318{
319 return s_fd_blocking (fd, 0)
320 || fcntl (fd, F_SETFD, FD_CLOEXEC);
321}
322
323#endif
324
325#if __linux && (__GLIBC__ > 2 || (__GLIBC__ == 2 && __GLIBC_MINOR__ >= 7))
326/* our minimum requirement is glibc 2.7 which has the stub, but not the header */
327# include <stdint.h>
328# ifdef __cplusplus
329extern "C" {
175#endif 330# endif
331 int eventfd (unsigned int initval, int flags);
332# ifdef __cplusplus
333}
334# endif
335#else
336# define eventfd(initval,flags) -1
337#endif
176 338
339typedef struct {
340 int fd[2]; /* read, write fd, might be equal */
341 int len; /* write length (1 pipe/socket, 8 eventfd) */
342 volatile sig_atomic_t sent;
343} s_epipe;
344
345static int
346s_epipe_new (s_epipe *epp)
347{
348 s_epipe ep;
349
350 ep.fd [0] = ep.fd [1] = eventfd (0, 0);
351
352 if (ep.fd [0] >= 0)
353 {
354 s_fd_prepare (ep.fd [0]);
355 ep.len = 8;
356 }
357 else
358 {
359 if (s_pipe (ep.fd))
360 return -1;
361
362 if (s_fd_prepare (ep.fd [0])
363 || s_fd_prepare (ep.fd [1]))
364 {
365 close (ep.fd [0]);
366 close (ep.fd [1]);
367 return -1;
368 }
369
370 ep.len = 1;
371 }
372
373 ep.sent = 0;
374 *epp = ep;
375 return 0;
376}
377
378static void
379s_epipe_destroy (s_epipe *epp)
380{
381 close (epp->fd [0]);
382
383 if (epp->fd [1] != epp->fd [0])
384 close (epp->fd [1]);
385
386 epp->len = 0;
387}
388
389static void
390s_epipe_signal (s_epipe *epp)
391{
392 if (epp->sent)
393 return;
394
395 epp->sent = 1;
396 {
397#ifdef _WIN32
398 /* perl overrides send with a function that crashes in other threads.
399 * unfortunately, it overrides it with an argument-less macro, so
400 * there is no way to force usage of the real send function.
401 * incompetent windows programmers - is this redundant?
402 */
403 DWORD dummy;
404 WriteFile (S_TO_HANDLE (epp->fd [1]), (LPCVOID)&dummy, 1, &dummy, 0);
405#else
406 static uint64_t counter = 1;
407 /* some modules accept fd's from outside, support eventfd here */
408 if (write (epp->fd [1], &counter, epp->len) < 0
409 && errno == EINVAL
410 && epp->len != 8)
411 write (epp->fd [1], &counter, (epp->len = 8));
412#endif
413 }
414}
415
416static void
417s_epipe_drain (s_epipe *epp)
418{
419 char buf [9];
420
421#ifdef _WIN32
422 recv (epp->fd [0], buf, sizeof (buf), 0);
423#else
424 read (epp->fd [0], buf, sizeof (buf));
425#endif
426
427 epp->sent = 0;
428}
429
430/* like new, but dups over old */
431static int
432s_epipe_renew (s_epipe *epp)
433{
434 s_epipe epn;
435
436 if (epp->fd [1] != epp->fd [0])
437 close (epp->fd [1]);
438
439 if (s_epipe_new (&epn))
440 return -1;
441
442 if (epp->len)
443 {
444 if (dup2 (epn.fd [0], epp->fd [0]) < 0)
445 croak ("unable to dup over old event pipe"); /* should not croak */
446
447 if (epp->fd [1] != epp->fd [0])
448 close (epn.fd [0]);
449
450 epn.fd [0] = epp->fd [0];
451 }
452
453 *epp = epn;
454
455 return 0;
456}
457
458#define s_epipe_fd(epp) ((epp)->fd [0])
459
460static int
461s_epipe_wait (s_epipe *epp)
462{
463#ifdef _WIN32
464 fd_set rfd;
465 int fd = s_epipe_fd (epp);
466
467 FD_ZERO (&rfd);
468 FD_SET (fd, &rfd);
469
470 return PerlSock_select (fd + 1, &rfd, 0, 0, 0);
471#else
472 /* poll is preferable on posix systems */
473 struct pollfd pfd;
474
475 pfd.fd = s_epipe_fd (epp);
476 pfd.events = POLLIN;
477
478 return poll (&pfd, 1, 0);
479#endif
480}
481
482#endif
483

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines