ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/IO-AIO/AIO.xs

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.114 by root, Sat Apr 26 12:00:23 2008 UTC vs.
Revision 1.155 by root, Fri Jul 17 00:56:54 2009 UTC

1	#include "xthread.h"	1	#include "libeio/xthread.h"
2		2
3	#include <errno.h>	3	#include <errno.h>
4		4
5	#include "EXTERN.h"	5	#include "EXTERN.h"
6	#include "perl.h"	6	#include "perl.h"
7	#include "XSUB.h"	7	#include "XSUB.h"
		8
		9	#include "schmorp.h"
8		10
9	#include <stddef.h>	11	#include <stddef.h>
10	#include <stdlib.h>	12	#include <stdlib.h>
11	#include <errno.h>	13	#include <errno.h>
12	#include <sys/types.h>	14	#include <sys/types.h>
13	#include <sys/stat.h>	15	#include <sys/stat.h>
14	#include <limits.h>	16	#include <limits.h>
15	#include <fcntl.h>	17	#include <fcntl.h>
16	#include <sched.h>	18	#include <sched.h>
17		19
		20	/* perl namespace pollution */
		21	#undef VERSION
		22
18	#ifdef _WIN32	23	#ifdef _WIN32
19		24
20	# define SIGIO 0	25	# define EIO_STRUCT_DIRENT Direntry_t
21	typedef Direntry_t X_DIRENT;
22	#undef malloc	26	# undef malloc
23	#undef free	27	# undef free
24		28
25	// perl overrides all those nice win32 functions	29	// perl overrides all those nice win32 functions
26	# undef open	30	# undef open
27	# undef read	31	# undef read
28	# undef write	32	# undef write
…		…
54	# define closedir(fd) (errno = ENOSYS, -1)	58	# define closedir(fd) (errno = ENOSYS, -1)
55	# define mkdir(a,b) mkdir (a)	59	# define mkdir(a,b) mkdir (a)
56		60
57	#else	61	#else
58		62
59	# include "autoconf/config.h"
60	# include <sys/time.h>	63	# include <sys/time.h>
61	# include <sys/select.h>	64	# include <sys/select.h>
62	# include <unistd.h>	65	# include <unistd.h>
63	# include <utime.h>	66	# include <utime.h>
64	# include <signal.h>	67	# include <signal.h>
65	typedef struct dirent X_DIRENT;	68	# define EIO_STRUCT_DIRENT struct dirent
66		69
67	#endif	70	#endif
68		71
69	#if HAVE_SENDFILE	72	/* perl stupidly overrides readdir and maybe others */
70	# if __linux	73	/* with thread-unsafe versions, imagine that :( */
71	# include <sys/sendfile.h>	74	#undef readdir
72	# elif __freebsd	75	#undef opendir
73	# include <sys/socket.h>	76	#undef closedir
74	# include <sys/uio.h>
75	# elif __hpux
76	# include <sys/socket.h>
77	# elif __solaris /* not yet */
78	# include <sys/sendfile.h>
79	# else
80	# error sendfile support requested but not available
81	# endif
82	#endif
83		77
84	/* number of seconds after which idle threads exit */	78	#define EIO_STRUCT_STAT Stat_t
85	#define IDLE_TIMEOUT 10
86
87	/* used for struct dirent, AIX doesn't provide it */
88	#ifndef NAME_MAX
89	# define NAME_MAX 4096
90	#endif
91
92	/* buffer size for various temporary buffers */
93	#define AIO_BUFSIZE 65536
94		79
95	/* use NV for 32 bit perls as it allows larger offsets */	80	/* use NV for 32 bit perls as it allows larger offsets */
96	#if IVSIZE >= 8	81	#if IVSIZE >= 8
		82	# define VAL64 IV
97	# define SvVAL64 SvIV	83	# define SvVAL64 SvIV
		84	# define newSVval64 newSViv
98	#else	85	#else
		86	# define VAL64 NV
99	# define SvVAL64 SvNV	87	# define SvVAL64 SvNV
		88	# define newSVval64 newSVnv
100	#endif	89	#endif
101		90
		91	/*****************************************************************************/
		92
		93	#if __GNUC__ >= 3
		94	# define expect(expr,value) __builtin_expect ((expr),(value))
		95	#else
		96	# define expect(expr,value) (expr)
		97	#endif
		98
		99	#define expect_false(expr) expect ((expr) != 0, 0)
		100	#define expect_true(expr) expect ((expr) != 0, 1)
		101
		102	/*****************************************************************************/
		103
102	static HV *stash;	104	static HV *stash;
103
104	#define dBUF \
105	char *aio_buf; \
106	X_LOCK (wrklock); \
107	self->dbuf = aio_buf = malloc (AIO_BUFSIZE); \
108	X_UNLOCK (wrklock); \
109	if (!aio_buf) \
110	return -1;
111
112	typedef SV SV8; /* byte-sv, used for argument-checking */	105	typedef SV SV8; /* byte-sv, used for argument-checking */
113		106	typedef int aio_rfd; /* read file desriptor */
114	enum {	107	typedef int aio_wfd; /* write file descriptor */
115	REQ_QUIT,
116	REQ_OPEN, REQ_CLOSE,
117	REQ_READ, REQ_WRITE,
118	REQ_READAHEAD, REQ_SENDFILE,
119	REQ_STAT, REQ_LSTAT, REQ_FSTAT,
120	REQ_TRUNCATE, REQ_FTRUNCATE,
121	REQ_UTIME, REQ_FUTIME,
122	REQ_CHMOD, REQ_FCHMOD,
123	REQ_CHOWN, REQ_FCHOWN,
124	REQ_SYNC, REQ_FSYNC, REQ_FDATASYNC,
125	REQ_UNLINK, REQ_RMDIR, REQ_MKDIR, REQ_RENAME,
126	REQ_MKNOD, REQ_READDIR,
127	REQ_LINK, REQ_SYMLINK, REQ_READLINK,
128	REQ_GROUP, REQ_NOP,
129	REQ_BUSY,
130	};
131		108
132	#define AIO_REQ_KLASS "IO::AIO::REQ"	109	#define AIO_REQ_KLASS "IO::AIO::REQ"
133	#define AIO_GRP_KLASS "IO::AIO::GRP"	110	#define AIO_GRP_KLASS "IO::AIO::GRP"
134		111
135	typedef struct aio_cb	112	#define EIO_REQ_MEMBERS \
136	{
137	struct aio_cb *volatile next;
138
139	SV *callback;	113	SV *callback; \
140	SV *sv1, *sv2;	114	SV *sv1, *sv2; \
141	void *ptr1, *ptr2;
142	off_t offs;
143	size_t size;
144	ssize_t result;
145	double nv1, nv2;
146
147	STRLEN stroffset;	115	STRLEN stroffset; \
148	int type;	116	SV *self;
149	int int1, int2, int3;
150	int errorno;
151	mode_t mode; /* open */
152		117
153	unsigned char flags;	118	#define EIO_NO_WRAPPERS 1
154	unsigned char pri;
155		119
156	SV *self; /* the perl counterpart of this request, if any */	120	#include "libeio/eio.h"
157	struct aio_cb *grp, *grp_prev, *grp_next, *grp_first;	121
158	} aio_cb;	122	#ifndef POSIX_FADV_NORMAL
		123	# define POSIX_FADV_NORMAL 0
		124	# define NO_FADVISE 1
		125	#endif
		126
		127	#ifndef POSIX_FADV_SEQUENTIAL
		128	# define POSIX_FADV_SEQUENTIAL 0
		129	#endif
		130
		131	#ifndef POSIX_FADV_RANDOM
		132	# define POSIX_FADV_RANDOM 0
		133	#endif
		134
		135	#ifndef POSIX_FADV_NOREUSE
		136	# define POSIX_FADV_NOREUSE 0
		137	#endif
		138
		139	#ifndef POSIX_FADV_WILLNEED
		140	# define POSIX_FADV_WILLNEED 0
		141	#endif
		142
		143	#ifndef POSIX_FADV_DONTNEED
		144	# define POSIX_FADV_DONTNEED 0
		145	#endif
		146
		147	static int req_invoke (eio_req *req);
		148	#define EIO_FINISH(req) req_invoke (req)
		149	static void req_destroy (eio_req *grp);
		150	#define EIO_DESTROY(req) req_destroy (req)
159		151
160	enum {	152	enum {
161	FLAG_CANCELLED = 0x01, /* request was cancelled */
162	FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */	153	FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
163	FLAG_PTR2_FREE = 0x80, /* need to free(ptr2) */
164	};	154	};
165		155
		156	#include "libeio/eio.c"
		157
166	typedef aio_cb *aio_req;	158	typedef eio_req *aio_req;
167	typedef aio_cb *aio_req_ornot;	159	typedef eio_req *aio_req_ornot;
168		160
169	enum {	161	static SV *on_next_submit;
170	PRI_MIN = -4,	162	static int next_pri = EIO_PRI_DEFAULT;
171	PRI_MAX = 4,	163	static int max_outstanding;
172		164
173	DEFAULT_PRI = 0,	165	static s_epipe respipe;
174	PRI_BIAS = -PRI_MIN,
175	NUM_PRI = PRI_MAX + PRI_BIAS + 1,
176	};
177		166
178	#define AIO_TICKS ((1000000 + 1023) >> 10)
179
180	static unsigned int max_poll_time = 0;
181	static unsigned int max_poll_reqs = 0;
182
183	/* calculcate time difference in ~1/AIO_TICKS of a second */
184	static int tvdiff (struct timeval *tv1, struct timeval *tv2)
185	{
186	return (tv2->tv_sec - tv1->tv_sec ) * AIO_TICKS
187	+ ((tv2->tv_usec - tv1->tv_usec) >> 10);
188	}
189
190	static thread_t main_tid;
191	static int main_sig;
192	static int block_sig_level;
193
194	void block_sig (void)
195	{
196	sigset_t ss;
197
198	if (block_sig_level++)
199	return;
200
201	if (!main_sig)
202	return;
203
204	sigemptyset (&ss);
205	sigaddset (&ss, main_sig);
206	pthread_sigmask (SIG_BLOCK, &ss, 0);
207	}
208
209	void unblock_sig (void)
210	{
211	sigset_t ss;
212
213	if (--block_sig_level)
214	return;
215
216	if (!main_sig)
217	return;
218
219	sigemptyset (&ss);
220	sigaddset (&ss, main_sig);
221	pthread_sigmask (SIG_UNBLOCK, &ss, 0);
222	}
223
224	static int next_pri = DEFAULT_PRI + PRI_BIAS;
225
226	static unsigned int started, idle, wanted;
227
228	/* worker threads management */
229	static mutex_t wrklock = X_MUTEX_INIT;
230
231	typedef struct worker {
232	/* locked by wrklock */
233	struct worker *prev, *next;
234
235	thread_t tid;
236
237	/* locked by reslock, reqlock or wrklock */
238	aio_req req; /* currently processed request */
239	void *dbuf;
240	DIR *dirp;
241	} worker;
242
243	static worker wrk_first = { &wrk_first, &wrk_first, 0 };
244
245	static void worker_clear (worker *wrk)
246	{
247	if (wrk->dirp)
248	{
249	closedir (wrk->dirp);
250	wrk->dirp = 0;
251	}
252
253	if (wrk->dbuf)
254	{
255	free (wrk->dbuf);
256	wrk->dbuf = 0;
257	}
258	}
259
260	static void worker_free (worker *wrk)
261	{
262	wrk->next->prev = wrk->prev;
263	wrk->prev->next = wrk->next;
264
265	free (wrk);
266	}
267
268	static volatile unsigned int nreqs, nready, npending;
269	static volatile unsigned int max_idle = 4;
270	static volatile unsigned int max_outstanding = 0xffffffff;
271	static int respipe_osf [2], respipe [2] = { -1, -1 };
272
273	static mutex_t reslock = X_MUTEX_INIT;
274	static mutex_t reqlock = X_MUTEX_INIT;
275	static cond_t reqwait = X_COND_INIT;
276
277	#if WORDACCESS_UNSAFE
278
279	static unsigned int get_nready (void)
280	{
281	unsigned int retval;
282
283	X_LOCK (reqlock);
284	retval = nready;
285	X_UNLOCK (reqlock);
286
287	return retval;
288	}
289
290	static unsigned int get_npending (void)
291	{
292	unsigned int retval;
293
294	X_LOCK (reslock);
295	retval = npending;
296	X_UNLOCK (reslock);
297
298	return retval;
299	}
300
301	static unsigned int get_nthreads (void)
302	{
303	unsigned int retval;
304
305	X_LOCK (wrklock);
306	retval = started;
307	X_UNLOCK (wrklock);
308
309	return retval;
310	}
311
312	#else
313
314	# define get_nready() nready
315	# define get_npending() npending
316	# define get_nthreads() started
317
318	#endif
319
320	/*
321	* a somewhat faster data structure might be nice, but
322	* with 8 priorities this actually needs <20 insns
323	* per shift, the most expensive operation.
324	*/
325	typedef struct {
326	aio_req qs[NUM_PRI], qe[NUM_PRI]; /* qstart, qend */
327	int size;
328	} reqq;
329
330	static reqq req_queue;
331	static reqq res_queue;
332
333	int reqq_push (reqq *q, aio_req req)
334	{
335	int pri = req->pri;
336	req->next = 0;
337
338	if (q->qe[pri])
339	{
340	q->qe[pri]->next = req;
341	q->qe[pri] = req;
342	}
343	else
344	q->qe[pri] = q->qs[pri] = req;
345
346	return q->size++;
347	}
348
349	aio_req reqq_shift (reqq *q)
350	{
351	int pri;
352
353	if (!q->size)
354	return 0;
355
356	--q->size;
357
358	for (pri = NUM_PRI; pri--; )
359	{
360	aio_req req = q->qs[pri];
361
362	if (req)
363	{
364	if (!(q->qs[pri] = req->next))
365	q->qe[pri] = 0;
366
367	return req;
368	}
369	}
370
371	abort ();
372	}
373
374	static int poll_cb (void);
375	static int req_invoke (aio_req req);
376	static void req_destroy (aio_req req);	167	static void req_destroy (aio_req req);
377	static void req_cancel (aio_req req);	168	static void req_cancel (aio_req req);
		169
		170	static void want_poll (void)
		171	{
		172	/* write a dummy byte to the pipe so fh becomes ready */
		173	s_epipe_signal (&respipe);
		174	}
		175
		176	static void done_poll (void)
		177	{
		178	/* read any signals sent by the worker threads */
		179	s_epipe_drain (&respipe);
		180	}
378		181
379	/* must be called at most once */	182	/* must be called at most once */
380	static SV *req_sv (aio_req req, const char *klass)	183	static SV *req_sv (aio_req req, const char *klass)
381	{	184	{
382	if (!req->self)	185	if (!req->self)
…		…
400	return mg ? (aio_req)mg->mg_ptr : 0;	203	return mg ? (aio_req)mg->mg_ptr : 0;
401	}	204	}
402		205
403	static void aio_grp_feed (aio_req grp)	206	static void aio_grp_feed (aio_req grp)
404	{	207	{
405	block_sig ();	208	if (grp->sv2 && SvOK (grp->sv2))
406
407	while (grp->size < grp->int2 && !(grp->flags & FLAG_CANCELLED))
408	{	209	{
409	int old_len = grp->size;
410
411	if (grp->sv2 && SvOK (grp->sv2))
412	{
413	dSP;	210	dSP;
414		211
415	ENTER;	212	ENTER;
416	SAVETMPS;	213	SAVETMPS;
417	PUSHMARK (SP);	214	PUSHMARK (SP);
418	XPUSHs (req_sv (grp, AIO_GRP_KLASS));	215	XPUSHs (req_sv (grp, AIO_GRP_KLASS));
419	PUTBACK;	216	PUTBACK;
420	call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);	217	call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
421	SPAGAIN;	218	SPAGAIN;
422	FREETMPS;	219	FREETMPS;
423	LEAVE;	220	LEAVE;
424	}
425
426	/* stop if no progress has been made */
427	if (old_len == grp->size)
428	{
429	SvREFCNT_dec (grp->sv2);
430	grp->sv2 = 0;
431	break;
432	}
433	}	221	}
434
435	unblock_sig ();
436	}	222	}
437		223
438	static void aio_grp_dec (aio_req grp)	224	static void req_submit (eio_req *req)
439	{	225	{
440	--grp->size;	226	eio_submit (req);
441		227
442	/* call feeder, if applicable */	228	if (expect_false (on_next_submit))
443	aio_grp_feed (grp);
444
445	/* finish, if done */
446	if (!grp->size && grp->int1)
447	{	229	{
448	block_sig ();	230	dSP;
		231	SV *cb = sv_2mortal (on_next_submit);
449		232
450	if (!req_invoke (grp))	233	on_next_submit = 0;
451	{
452	req_destroy (grp);
453	unblock_sig ();
454	croak (0);
455	}
456		234
457	req_destroy (grp);	235	PUSHMARK (SP);
458	unblock_sig ();	236	PUTBACK;
		237	call_sv (cb, G_DISCARD | G_EVAL);
459	}	238	}
460	}	239	}
461		240
462	static int req_invoke (aio_req req)	241	static int req_invoke (eio_req *req)
463	{	242	{
464	dSP;	243	dSP;
465		244
466	if (req->flags & FLAG_SV2_RO_OFF)	245	if (req->flags & FLAG_SV2_RO_OFF)
467	SvREADONLY_off (req->sv2);	246	SvREADONLY_off (req->sv2);
468		247
469	if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback))	248	if (!EIO_CANCELLED (req) && req->callback)
470	{	249	{
471	ENTER;	250	ENTER;
472	SAVETMPS;	251	SAVETMPS;
473	PUSHMARK (SP);	252	PUSHMARK (SP);
474	EXTEND (SP, 1);	253	EXTEND (SP, 1);
475		254
476	switch (req->type)	255	switch (req->type)
477	{	256	{
478	case REQ_READDIR:	257	case EIO_READDIR:
479	{	258	{
480	SV *rv = &PL_sv_undef;	259	SV *rv = &PL_sv_undef;
481		260
482	if (req->result >= 0)	261	if (req->result >= 0)
483	{	262	{
484	int i;	263	int i;
485	char *buf = req->ptr2;	264	char *names = (char *)req->ptr2;
		265	eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
486	AV *av = newAV ();	266	AV *av = newAV ();
487		267
488	av_extend (av, req->result - 1);	268	av_extend (av, req->result - 1);
489		269
490	for (i = 0; i < req->result; ++i)	270	for (i = 0; i < req->result; ++i)
491	{	271	{
492	SV *sv = newSVpv (buf, 0);	272	if (req->int1 & EIO_READDIR_DENTS)
		273	{
		274	SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
493		275
		276	if (req->int1 & EIO_READDIR_CUSTOM2)
		277	{
		278	static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */
		279	AV *avent = newAV ();
		280
		281	av_extend (avent, 2);
		282
		283	if (!sv_type [ent->type])
		284	{
		285	sv_type [ent->type] = newSViv (ent->type);
		286	SvREADONLY_on (sv_type [ent->type]);
		287	}
		288
		289	av_store (avent, 0, namesv);
		290	av_store (avent, 1, SvREFCNT_inc (sv_type [ent->type]));
		291	av_store (avent, 2, IVSIZE >= 8 ? newSVuv (ent->inode) : newSVnv (ent->inode));
		292
		293	av_store (av, i, newRV_noinc ((SV *)avent));
		294	}
		295	else
494	av_store (av, i, sv);	296	av_store (av, i, namesv);
		297
		298	++ent;
		299	}
		300	else
		301	{
		302	SV *name = newSVpv (names, 0);
		303	av_store (av, i, name);
495	buf += SvCUR (sv) + 1;	304	names += SvCUR (name) + 1;
		305	}
496	}	306	}
497		307
498	rv = sv_2mortal (newRV_noinc ((SV *)av));	308	rv = sv_2mortal (newRV_noinc ((SV *)av));
499	}	309	}
500		310
501	PUSHs (rv);	311	PUSHs (rv);
		312
		313	if (req->int1 & EIO_READDIR_CUSTOM1)
		314	XPUSHs (sv_2mortal (newSViv (req->int1 & ~(EIO_READDIR_CUSTOM1 | EIO_READDIR_CUSTOM2))));
502	}	315	}
503	break;	316	break;
504		317
505	case REQ_OPEN:	318	case EIO_OPEN:
506	{	319	{
507	/* convert fd to fh */	320	/* convert fd to fh */
508	SV *fh = &PL_sv_undef;	321	SV *fh = &PL_sv_undef;
509		322
510	if (req->result >= 0)	323	if (req->result >= 0)
…		…
512	GV *gv = (GV *)sv_newmortal ();	325	GV *gv = (GV *)sv_newmortal ();
513	int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);	326	int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
514	char sym [64];	327	char sym [64];
515	int symlen;	328	int symlen;
516		329
517	symlen = snprintf (sym, sizeof (sym), "fd#%d", req->result);	330	symlen = snprintf (sym, sizeof (sym), "fd#%d", (int)req->result);
518	gv_init (gv, stash, sym, symlen, 0);	331	gv_init (gv, stash, sym, symlen, 0);
519		332
520	symlen = snprintf (	333	symlen = snprintf (
521	sym,	334	sym,
522	sizeof (sym),	335	sizeof (sym),
523	"%s&=%d",	336	"%s&=%d",
524	flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",	337	flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
525	req->result	338	(int)req->result
526	);	339	);
527		340
528	if (do_open (gv, sym, symlen, 0, 0, 0, 0))	341	if (do_open (gv, sym, symlen, 0, 0, 0, 0))
529	fh = (SV *)gv;	342	fh = (SV *)gv;
530	}	343	}
531		344
532	PUSHs (fh);	345	PUSHs (fh);
533	}	346	}
534	break;	347	break;
535		348
536	case REQ_GROUP:	349	case EIO_GROUP:
537	req->int1 = 2; /* mark group as finished */	350	req->int1 = 2; /* mark group as finished */
538		351
539	if (req->sv1)	352	if (req->sv1)
540	{	353	{
541	int i;	354	int i;
…		…
545	for (i = 0; i <= AvFILL (av); ++i)	358	for (i = 0; i <= AvFILL (av); ++i)
546	PUSHs (*av_fetch (av, i, 0));	359	PUSHs (*av_fetch (av, i, 0));
547	}	360	}
548	break;	361	break;
549		362
550	case REQ_NOP:	363	case EIO_NOP:
551	case REQ_BUSY:	364	case EIO_BUSY:
552	break;	365	break;
553		366
554	case REQ_READLINK:	367	case EIO_READLINK:
555	if (req->result > 0)	368	if (req->result > 0)
556	{	369	PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
557	SvCUR_set (req->sv2, req->result);
558	*SvEND (req->sv2) = 0;
559	PUSHs (req->sv2);
560	}
561	break;	370	break;
562		371
563	case REQ_STAT:	372	case EIO_STAT:
564	case REQ_LSTAT:	373	case EIO_LSTAT:
565	case REQ_FSTAT:	374	case EIO_FSTAT:
566	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;	375	PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT;
567	PL_laststatval = req->result;	376	PL_laststatval = req->result;
568	PL_statcache = *(Stat_t *)(req->ptr2);	377	PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
569	PUSHs (sv_2mortal (newSViv (req->result)));	378	PUSHs (sv_2mortal (newSViv (req->result)));
570	break;	379	break;
571		380
572	case REQ_READ:	381	case EIO_READ:
		382	{
573	SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));	383	SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
574	*SvEND (req->sv2) = 0;	384	*SvEND (req->sv2) = 0;
		385	SvPOK_only (req->sv2);
		386	SvSETMAGIC (req->sv2);
575	PUSHs (sv_2mortal (newSViv (req->result)));	387	PUSHs (sv_2mortal (newSViv (req->result)));
		388	}
576	break;	389	break;
		390
		391	case EIO_DUP2:
		392	if (req->result > 0)
		393	req->result = 0;
		394	/* FALLTHROUGH */
577		395
578	default:	396	default:
579	PUSHs (sv_2mortal (newSViv (req->result)));	397	PUSHs (sv_2mortal (newSViv (req->result)));
580	break;	398	break;
581	}	399	}
…		…
590	LEAVE;	408	LEAVE;
591		409
592	PUTBACK;	410	PUTBACK;
593	}	411	}
594		412
595	if (req->grp)
596	{
597	aio_req grp = req->grp;
598
599	/* unlink request */
600	if (req->grp_next) req->grp_next->grp_prev = req->grp_prev;
601	if (req->grp_prev) req->grp_prev->grp_next = req->grp_next;
602
603	if (grp->grp_first == req)
604	grp->grp_first = req->grp_next;
605
606	aio_grp_dec (grp);
607	}
608
609	return !SvTRUE (ERRSV);	413	return !!SvTRUE (ERRSV);
610	}	414	}
611		415
612	static void req_destroy (aio_req req)	416	static void req_destroy (aio_req req)
613	{	417	{
614	if (req->self)	418	if (req->self)
…		…
619		423
620	SvREFCNT_dec (req->sv1);	424	SvREFCNT_dec (req->sv1);
621	SvREFCNT_dec (req->sv2);	425	SvREFCNT_dec (req->sv2);
622	SvREFCNT_dec (req->callback);	426	SvREFCNT_dec (req->callback);
623		427
624	if (req->flags & FLAG_PTR2_FREE)
625	free (req->ptr2);
626
627	Safefree (req);	428	Safefree (req);
628	}	429	}
629		430
630	static void req_cancel_subs (aio_req grp)	431	static void req_cancel_subs (aio_req grp)
631	{	432	{
632	aio_req sub;	433	aio_req sub;
633		434
634	if (grp->type != REQ_GROUP)	435	if (grp->type != EIO_GROUP)
635	return;	436	return;
636		437
637	SvREFCNT_dec (grp->sv2);	438	SvREFCNT_dec (grp->sv2);
638	grp->sv2 = 0;	439	grp->sv2 = 0;
639		440
640	for (sub = grp->grp_first; sub; sub = sub->grp_next)	441	eio_grp_cancel (grp);
641	req_cancel (sub);
642	}	442	}
643
644	static void req_cancel (aio_req req)
645	{
646	req->flags |= FLAG_CANCELLED;
647
648	req_cancel_subs (req);
649	}
650
651	#ifdef USE_SOCKETS_AS_HANDLES
652	# define TO_SOCKET(x) (win32_get_osfhandle (x))
653	#else
654	# define TO_SOCKET(x) (x)
655	#endif
656		443
657	static void	444	static void
658	create_respipe (void)	445	create_respipe (void)
659	{	446	{
660	int old_readfd = respipe [0];
661
662	if (respipe [1] >= 0)
663	respipe_close (TO_SOCKET (respipe [1]));
664
665	#ifdef _WIN32
666	if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, respipe))
667	#else
668	if (pipe (respipe))	447	if (s_epipe_renew (&respipe))
669	#endif
670	croak ("unable to initialize result pipe");	448	croak ("IO::AIO: unable to initialize result pipe");
671
672	if (old_readfd >= 0)
673	{
674	if (dup2 (TO_SOCKET (respipe [0]), TO_SOCKET (old_readfd)) < 0)
675	croak ("unable to initialize result pipe(2)");
676
677	respipe_close (respipe [0]);
678	respipe [0] = old_readfd;
679	}
680
681	#ifdef _WIN32
682	int arg = 1;
683	if (ioctlsocket (TO_SOCKET (respipe [0]), FIONBIO, &arg)
684	|| ioctlsocket (TO_SOCKET (respipe [1]), FIONBIO, &arg))
685	#else
686	if (fcntl (respipe [0], F_SETFL, O_NONBLOCK)
687	|| fcntl (respipe [1], F_SETFL, O_NONBLOCK))
688	#endif
689	croak ("unable to initialize result pipe(3)");
690
691	respipe_osf [0] = TO_SOCKET (respipe [0]);
692	respipe_osf [1] = TO_SOCKET (respipe [1]);
693	}
694
695	X_THREAD_PROC (aio_proc);
696
697	static void start_thread (void)
698	{
699	worker *wrk = calloc (1, sizeof (worker));
700
701	if (!wrk)
702	croak ("unable to allocate worker thread data");
703
704	X_LOCK (wrklock);
705
706	if (thread_create (&wrk->tid, aio_proc, (void *)wrk))
707	{
708	wrk->prev = &wrk_first;
709	wrk->next = wrk_first.next;
710	wrk_first.next->prev = wrk;
711	wrk_first.next = wrk;
712	++started;
713	}
714	else
715	free (wrk);
716
717	X_UNLOCK (wrklock);
718	}
719
720	static void maybe_start_thread (void)
721	{
722	if (get_nthreads () >= wanted)
723	return;
724
725	/* todo: maybe use idle here, but might be less exact */
726	if (0 <= (int)get_nthreads () + (int)get_npending () - (int)nreqs)
727	return;
728
729	start_thread ();
730	}
731
732	static void req_send (aio_req req)
733	{
734	block_sig ();
735
736	++nreqs;
737
738	X_LOCK (reqlock);
739	++nready;
740	reqq_push (&req_queue, req);
741	X_COND_SIGNAL (reqwait);
742	X_UNLOCK (reqlock);
743
744	unblock_sig ();
745
746	maybe_start_thread ();
747	}
748
749	static void end_thread (void)
750	{
751	aio_req req;
752
753	Newz (0, req, 1, aio_cb);
754
755	req->type = REQ_QUIT;
756	req->pri = PRI_MAX + PRI_BIAS;
757
758	X_LOCK (reqlock);
759	reqq_push (&req_queue, req);
760	X_COND_SIGNAL (reqwait);
761	X_UNLOCK (reqlock);
762
763	X_LOCK (wrklock);
764	--started;
765	X_UNLOCK (wrklock);
766	}
767
768	static void set_max_idle (int nthreads)
769	{
770	if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
771	max_idle = nthreads <= 0 ? 1 : nthreads;
772	if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
773	}
774
775	static void min_parallel (int nthreads)
776	{
777	if (wanted < nthreads)
778	wanted = nthreads;
779	}
780
781	static void max_parallel (int nthreads)
782	{
783	if (wanted > nthreads)
784	wanted = nthreads;
785
786	while (started > wanted)
787	end_thread ();
788	}	449	}
789		450
790	static void poll_wait (void)	451	static void poll_wait (void)
791	{	452	{
792	fd_set rfd;
793
794	while (nreqs)	453	while (eio_nreqs ())
795	{	454	{
796	int size;	455	int size;
797	if (WORDACCESS_UNSAFE) X_LOCK (reslock);	456
		457	X_LOCK (reslock);
798	size = res_queue.size;	458	size = res_queue.size;
799	if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);	459	X_UNLOCK (reslock);
800		460
801	if (size)	461	if (size)
802	return;	462	return;
803		463
804	maybe_start_thread ();	464	etp_maybe_start_thread ();
805		465
806	FD_ZERO (&rfd);	466	s_epipe_wait (&respipe);
807	FD_SET (respipe [0], &rfd);
808
809	PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);
810	}	467	}
811	}	468	}
812		469
813	static int poll_cb (void)	470	static int poll_cb (void)
814	{	471	{
815	dSP;
816	int count = 0;
817	int maxreqs = max_poll_reqs;
818	int do_croak = 0;
819	struct timeval tv_start, tv_now;
820	aio_req req;
821
822	if (max_poll_time)
823	gettimeofday (&tv_start, 0);
824
825	block_sig ();
826
827	for (;;)	472	for (;;)
828	{	473	{
829	for (;;)	474	int res = eio_poll ();
830	{
831	maybe_start_thread ();
832		475
833	X_LOCK (reslock);	476	if (res > 0)
834	req = reqq_shift (&res_queue);
835
836	if (req)
837	{
838	--npending;
839
840	if (!res_queue.size)
841	{
842	/* read any signals sent by the worker threads */
843	char buf [4];
844	while (respipe_read (respipe [0], buf, 4) == 4)
845	;
846	}
847	}
848
849	X_UNLOCK (reslock);
850
851	if (!req)
852	break;
853
854	--nreqs;
855
856	if (req->type == REQ_GROUP && req->size)
857	{
858	req->int1 = 1; /* mark request as delayed */
859	continue;
860	}
861	else
862	{
863	if (!req_invoke (req))
864	{
865	req_destroy (req);
866	unblock_sig ();
867	croak (0);	477	croak (0);
868	}
869		478
870	count++;	479	if (!max_outstanding || max_outstanding > eio_nreqs ())
871	}	480	return res;
872
873	req_destroy (req);
874
875	if (maxreqs && !--maxreqs)
876	break;
877
878	if (max_poll_time)
879	{
880	gettimeofday (&tv_now, 0);
881
882	if (tvdiff (&tv_start, &tv_now) >= max_poll_time)
883	break;
884	}
885	}
886
887	if (nreqs <= max_outstanding)
888	break;
889		481
890	poll_wait ();	482	poll_wait ();
891
892	++maxreqs;
893	}	483	}
894
895	unblock_sig ();
896	return count;
897	}
898
899	/*****************************************************************************/
900	/* work around various missing functions */
901
902	#if !HAVE_PREADWRITE
903	# define pread aio_pread
904	# define pwrite aio_pwrite
905
906	/*
907	* make our pread/pwrite safe against themselves, but not against
908	* normal read/write by using a mutex. slows down execution a lot,
909	* but that's your problem, not mine.
910	*/
911	static mutex_t preadwritelock = X_MUTEX_INIT;
912
913	static ssize_t pread (int fd, void *buf, size_t count, off_t offset)
914	{
915	ssize_t res;
916	off_t ooffset;
917
918	X_LOCK (preadwritelock);
919	ooffset = lseek (fd, 0, SEEK_CUR);
920	lseek (fd, offset, SEEK_SET);
921	res = read (fd, buf, count);
922	lseek (fd, ooffset, SEEK_SET);
923	X_UNLOCK (preadwritelock);
924
925	return res;
926	}
927
928	static ssize_t pwrite (int fd, void *buf, size_t count, off_t offset)
929	{
930	ssize_t res;
931	off_t ooffset;
932
933	X_LOCK (preadwritelock);
934	ooffset = lseek (fd, 0, SEEK_CUR);
935	lseek (fd, offset, SEEK_SET);
936	res = write (fd, buf, count);
937	lseek (fd, offset, SEEK_SET);
938	X_UNLOCK (preadwritelock);
939
940	return res;
941	}
942	#endif
943
944	#ifndef HAVE_FUTIMES
945
946	# define utimes(path,times) aio_utimes (path, times)
947	# define futimes(fd,times) aio_futimes (fd, times)
948
949	int aio_utimes (const char *filename, const struct timeval times[2])
950	{
951	if (times)
952	{
953	struct utimbuf buf;
954
955	buf.actime = times[0].tv_sec;
956	buf.modtime = times[1].tv_sec;
957
958	return utime (filename, &buf);
959	}
960	else
961	return utime (filename, 0);
962	}
963
964	int aio_futimes (int fd, const struct timeval tv[2])
965	{
966	errno = ENOSYS;
967	return -1;
968	}
969
970	#endif
971
972	#if !HAVE_FDATASYNC
973	# define fdatasync fsync
974	#endif
975
976	#if !HAVE_READAHEAD
977	# define readahead(fd,offset,count) aio_readahead (fd, offset, count, self)
978
979	static ssize_t aio_readahead (int fd, off_t offset, size_t count, worker *self)
980	{
981	size_t todo = count;
982	dBUF;
983
984	while (todo > 0)
985	{
986	size_t len = todo < AIO_BUFSIZE ? todo : AIO_BUFSIZE;
987
988	pread (fd, aio_buf, len, offset);
989	offset += len;
990	todo -= len;
991	}
992
993	errno = 0;
994	return count;
995	}
996
997	#endif
998
999	#if !HAVE_READDIR_R
1000	# define readdir_r aio_readdir_r
1001
1002	static mutex_t readdirlock = X_MUTEX_INIT;
1003
1004	static int readdir_r (DIR *dirp, X_DIRENT *ent, X_DIRENT **res)
1005	{
1006	X_DIRENT *e;
1007	int errorno;
1008
1009	X_LOCK (readdirlock);
1010
1011	e = readdir (dirp);
1012	errorno = errno;
1013
1014	if (e)
1015	{
1016	*res = ent;
1017	strcpy (ent->d_name, e->d_name);
1018	}
1019	else
1020	*res = 0;
1021
1022	X_UNLOCK (readdirlock);
1023
1024	errno = errorno;
1025	return e ? 0 : -1;
1026	}
1027	#endif
1028
1029	/* sendfile always needs emulation */
1030	static ssize_t sendfile_ (int ofd, int ifd, off_t offset, size_t count, worker *self)
1031	{
1032	ssize_t res;
1033
1034	if (!count)
1035	return 0;
1036
1037	#if HAVE_SENDFILE
1038	# if __linux
1039	res = sendfile (ofd, ifd, &offset, count);
1040
1041	# elif __freebsd
1042	/*
1043	* Of course, the freebsd sendfile is a dire hack with no thoughts
1044	* wasted on making it similar to other I/O functions.
1045	*/
1046	{
1047	off_t sbytes;
1048	res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
1049
1050	if (res < 0 && sbytes)
1051	/* maybe only on EAGAIN: as usual, the manpage leaves you guessing */
1052	res = sbytes;
1053	}
1054
1055	# elif __hpux
1056	res = sendfile (ofd, ifd, offset, count, 0, 0);
1057
1058	# elif __solaris
1059	{
1060	struct sendfilevec vec;
1061	size_t sbytes;
1062
1063	vec.sfv_fd = ifd;
1064	vec.sfv_flag = 0;
1065	vec.sfv_off = offset;
1066	vec.sfv_len = count;
1067
1068	res = sendfilev (ofd, &vec, 1, &sbytes);
1069
1070	if (res < 0 && sbytes)
1071	res = sbytes;
1072	}
1073
1074	# endif
1075	#else
1076	res = -1;
1077	errno = ENOSYS;
1078	#endif
1079
1080	if (res < 0
1081	&& (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
1082	#if __solaris
1083	|| errno == EAFNOSUPPORT || errno == EPROTOTYPE
1084	#endif
1085	)
1086	)
1087	{
1088	/* emulate sendfile. this is a major pain in the ass */
1089	dBUF;
1090
1091	res = 0;
1092
1093	while (count)
1094	{
1095	ssize_t cnt;
1096
1097	cnt = pread (ifd, aio_buf, count > AIO_BUFSIZE ? AIO_BUFSIZE : count, offset);
1098
1099	if (cnt <= 0)
1100	{
1101	if (cnt && !res) res = -1;
1102	break;
1103	}
1104
1105	cnt = write (ofd, aio_buf, cnt);
1106
1107	if (cnt <= 0)
1108	{
1109	if (cnt && !res) res = -1;
1110	break;
1111	}
1112
1113	offset += cnt;
1114	res += cnt;
1115	count -= cnt;
1116	}
1117	}
1118
1119	return res;
1120	}
1121
1122	/* read a full directory */
1123	static void scandir_ (aio_req req, worker *self)
1124	{
1125	DIR *dirp;
1126	union
1127	{
1128	X_DIRENT d;
1129	char b [offsetof (X_DIRENT, d_name) + NAME_MAX + 1];
1130	} *u;
1131	X_DIRENT *entp;
1132	char *name, *names;
1133	int memlen = 4096;
1134	int memofs = 0;
1135	int res = 0;
1136
1137	X_LOCK (wrklock);
1138	self->dirp = dirp = opendir (req->ptr1);
1139	self->dbuf = u = malloc (sizeof (*u));
1140	req->flags |= FLAG_PTR2_FREE;
1141	req->ptr2 = names = malloc (memlen);
1142	X_UNLOCK (wrklock);
1143
1144	if (dirp && u && names)
1145	for (;;)
1146	{
1147	errno = 0;
1148	readdir_r (dirp, &u->d, &entp);
1149
1150	if (!entp)
1151	break;
1152
1153	name = entp->d_name;
1154
1155	if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
1156	{
1157	int len = strlen (name) + 1;
1158
1159	res++;
1160
1161	while (memofs + len > memlen)
1162	{
1163	memlen *= 2;
1164	X_LOCK (wrklock);
1165	req->ptr2 = names = realloc (names, memlen);
1166	X_UNLOCK (wrklock);
1167
1168	if (!names)
1169	break;
1170	}
1171
1172	memcpy (names + memofs, name, len);
1173	memofs += len;
1174	}
1175	}
1176
1177	if (errno)
1178	res = -1;
1179
1180	req->result = res;
1181	}
1182
1183	static int
1184	aio_close (int fd)
1185	{
1186	static int close_pipe = -1; /* dummy fd to close fds via dup2 */
1187
1188	X_LOCK (wrklock);
1189
1190	if (close_pipe < 0)
1191	{
1192	int pipefd [2];
1193
1194	if (pipe (pipefd) < 0
1195	|| close (pipefd [1]) < 0
1196	|| fcntl (pipefd [0], F_SETFD, FD_CLOEXEC) < 0)
1197	{
1198	X_UNLOCK (wrklock);
1199	return -1;
1200	}
1201
1202	close_pipe = pipefd [0];
1203	}
1204
1205	X_UNLOCK (wrklock);
1206
1207	return dup2 (close_pipe, fd) < 0 ? -1 : 0;
1208	}
1209
1210	/*****************************************************************************/
1211
1212	X_THREAD_PROC (aio_proc)
1213	{
1214	aio_req req;
1215	struct timespec ts;
1216	worker *self = (worker *)thr_arg;
1217
1218	/* try to distribute timeouts somewhat randomly */
1219	ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL);
1220
1221	for (;;)
1222	{
1223	ts.tv_sec = time (0) + IDLE_TIMEOUT;
1224
1225	X_LOCK (reqlock);
1226
1227	for (;;)
1228	{
1229	self->req = req = reqq_shift (&req_queue);
1230
1231	if (req)
1232	break;
1233
1234	++idle;
1235
1236	if (X_COND_TIMEDWAIT (reqwait, reqlock, ts)
1237	== ETIMEDOUT)
1238	{
1239	if (idle > max_idle)
1240	{
1241	--idle;
1242	X_UNLOCK (reqlock);
1243	X_LOCK (wrklock);
1244	--started;
1245	X_UNLOCK (wrklock);
1246	goto quit;
1247	}
1248
1249	/* we are allowed to idle, so do so without any timeout */
1250	X_COND_WAIT (reqwait, reqlock);
1251	ts.tv_sec = time (0) + IDLE_TIMEOUT;
1252	}
1253
1254	--idle;
1255	}
1256
1257	--nready;
1258
1259	X_UNLOCK (reqlock);
1260
1261	errno = 0; /* strictly unnecessary */
1262
1263	if (!(req->flags & FLAG_CANCELLED))
1264	switch (req->type)
1265	{
1266	case REQ_READ: req->result = req->offs >= 0
1267	? pread (req->int1, req->ptr1, req->size, req->offs)
1268	: read (req->int1, req->ptr1, req->size); break;
1269	case REQ_WRITE: req->result = req->offs >= 0
1270	? pwrite (req->int1, req->ptr1, req->size, req->offs)
1271	: write (req->int1, req->ptr1, req->size); break;
1272
1273	case REQ_READAHEAD: req->result = readahead (req->int1, req->offs, req->size); break;
1274	case REQ_SENDFILE: req->result = sendfile_ (req->int1, req->int2, req->offs, req->size, self); break;
1275
1276	case REQ_STAT: req->result = stat (req->ptr1, (Stat_t *)req->ptr2); break;
1277	case REQ_LSTAT: req->result = lstat (req->ptr1, (Stat_t *)req->ptr2); break;
1278	case REQ_FSTAT: req->result = fstat (req->int1, (Stat_t *)req->ptr2); break;
1279
1280	case REQ_CHOWN: req->result = chown (req->ptr1, req->int2, req->int3); break;
1281	case REQ_FCHOWN: req->result = fchown (req->int1, req->int2, req->int3); break;
1282	case REQ_CHMOD: req->result = chmod (req->ptr1, req->mode); break;
1283	case REQ_FCHMOD: req->result = fchmod (req->int1, req->mode); break;
1284	case REQ_TRUNCATE: req->result = truncate (req->ptr1, req->offs); break;
1285	case REQ_FTRUNCATE: req->result = ftruncate (req->int1, req->offs); break;
1286
1287	case REQ_OPEN: req->result = open (req->ptr1, req->int1, req->mode); break;
1288	case REQ_CLOSE: req->result = aio_close (req->int1); break;
1289	case REQ_UNLINK: req->result = unlink (req->ptr1); break;
1290	case REQ_RMDIR: req->result = rmdir (req->ptr1); break;
1291	case REQ_MKDIR: req->result = mkdir (req->ptr1, req->mode); break;
1292	case REQ_RENAME: req->result = rename (req->ptr2, req->ptr1); break;
1293	case REQ_LINK: req->result = link (req->ptr2, req->ptr1); break;
1294	case REQ_SYMLINK: req->result = symlink (req->ptr2, req->ptr1); break;
1295	case REQ_MKNOD: req->result = mknod (req->ptr2, req->mode, (dev_t)req->offs); break;
1296	case REQ_READLINK: req->result = readlink (req->ptr2, req->ptr1, NAME_MAX); break;
1297
1298	case REQ_SYNC: req->result = 0; sync (); break;
1299	case REQ_FSYNC: req->result = fsync (req->int1); break;
1300	case REQ_FDATASYNC: req->result = fdatasync (req->int1); break;
1301
1302	case REQ_READDIR: scandir_ (req, self); break;
1303
1304	case REQ_BUSY:
1305	#ifdef _WIN32
1306	Sleep (req->nv1 * 1000.);
1307	#else
1308	{
1309	struct timeval tv;
1310
1311	tv.tv_sec = req->nv1;
1312	tv.tv_usec = (req->nv1 - tv.tv_sec) * 1000000.;
1313
1314	req->result = select (0, 0, 0, 0, &tv);
1315	}
1316	#endif
1317	break;
1318
1319	case REQ_UTIME:
1320	case REQ_FUTIME:
1321	{
1322	struct timeval tv[2];
1323	struct timeval *times;
1324
1325	if (req->nv1 != -1. || req->nv2 != -1.)
1326	{
1327	tv[0].tv_sec = req->nv1;
1328	tv[0].tv_usec = (req->nv1 - tv[0].tv_sec) * 1000000.;
1329	tv[1].tv_sec = req->nv2;
1330	tv[1].tv_usec = (req->nv2 - tv[1].tv_sec) * 1000000.;
1331
1332	times = tv;
1333	}
1334	else
1335	times = 0;
1336
1337
1338	req->result = req->type == REQ_FUTIME
1339	? futimes (req->int1, times)
1340	: utimes (req->ptr1, times);
1341	}
1342
1343	case REQ_GROUP:
1344	case REQ_NOP:
1345	break;
1346
1347	case REQ_QUIT:
1348	goto quit;
1349
1350	default:
1351	req->result = -1;
1352	break;
1353	}
1354
1355	req->errorno = errno;
1356
1357	X_LOCK (reslock);
1358
1359	++npending;
1360
1361	if (!reqq_push (&res_queue, req))
1362	{
1363	/* write a dummy byte to the pipe so fh becomes ready */
1364	respipe_write (respipe_osf [1], (const void *)&respipe_osf, 1);
1365
1366	/* optionally signal the main thread asynchronously */
1367	if (main_sig)
1368	pthread_kill (main_tid, main_sig);
1369	}
1370
1371	self->req = 0;
1372	worker_clear (self);
1373
1374	X_UNLOCK (reslock);
1375	}
1376
1377	quit:
1378	X_LOCK (wrklock);
1379	worker_free (self);
1380	X_UNLOCK (wrklock);
1381
1382	return 0;
1383	}
1384
1385	/*****************************************************************************/
1386
1387	static void atfork_prepare (void)
1388	{
1389	X_LOCK (wrklock);
1390	X_LOCK (reqlock);
1391	X_LOCK (reslock);
1392	#if !HAVE_PREADWRITE
1393	X_LOCK (preadwritelock);
1394	#endif
1395	#if !HAVE_READDIR_R
1396	X_LOCK (readdirlock);
1397	#endif
1398	}
1399
1400	static void atfork_parent (void)
1401	{
1402	#if !HAVE_READDIR_R
1403	X_UNLOCK (readdirlock);
1404	#endif
1405	#if !HAVE_PREADWRITE
1406	X_UNLOCK (preadwritelock);
1407	#endif
1408	X_UNLOCK (reslock);
1409	X_UNLOCK (reqlock);
1410	X_UNLOCK (wrklock);
1411	}	484	}
1412		485
1413	static void atfork_child (void)	486	static void atfork_child (void)
1414	{	487	{
1415	aio_req prv;
1416
1417	while (prv = reqq_shift (&req_queue))
1418	req_destroy (prv);
1419
1420	while (prv = reqq_shift (&res_queue))
1421	req_destroy (prv);
1422
1423	while (wrk_first.next != &wrk_first)
1424	{
1425	worker *wrk = wrk_first.next;
1426
1427	if (wrk->req)
1428	req_destroy (wrk->req);
1429
1430	worker_clear (wrk);
1431	worker_free (wrk);
1432	}
1433
1434	started = 0;
1435	idle = 0;
1436	nreqs = 0;
1437	nready = 0;
1438	npending = 0;
1439
1440	create_respipe ();	488	create_respipe ();
		489	}
1441		490
1442	atfork_parent ();	491	static SV *
		492	get_cb (SV *cb_sv)
		493	{
		494	SvGETMAGIC (cb_sv);
		495	return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0;
1443	}	496	}
1444		497
1445	#define dREQ \	498	#define dREQ \
		499	SV *cb_cv; \
1446	aio_req req; \	500	aio_req req; \
1447	int req_pri = next_pri; \	501	int req_pri = next_pri; \
1448	next_pri = DEFAULT_PRI + PRI_BIAS; \	502	next_pri = EIO_PRI_DEFAULT; \
1449	\	503	\
1450	if (SvOK (callback) && !SvROK (callback)) \	504	cb_cv = get_cb (callback); \
1451	croak ("callback must be undef or of reference type"); \
1452	\	505	\
1453	Newz (0, req, 1, aio_cb); \	506	Newz (0, req, 1, eio_req); \
1454	if (!req) \	507	if (!req) \
1455	croak ("out of memory during aio_req allocation"); \	508	croak ("out of memory during eio_req allocation"); \
1456	\	509	\
1457	req->callback = newSVsv (callback); \	510	req->callback = SvREFCNT_inc (cb_cv); \
1458	req->pri = req_pri	511	req->pri = req_pri
1459		512
1460	#define REQ_SEND \	513	#define REQ_SEND \
		514	PUTBACK; \
1461	req_send (req); \	515	req_submit (req); \
		516	SPAGAIN; \
1462	\	517	\
1463	if (GIMME_V != G_VOID) \	518	if (GIMME_V != G_VOID) \
1464	XPUSHs (req_sv (req, AIO_REQ_KLASS));	519	XPUSHs (req_sv (req, AIO_REQ_KLASS));
1465		520
1466	MODULE = IO::AIO PACKAGE = IO::AIO	521	MODULE = IO::AIO PACKAGE = IO::AIO
1467		522
1468	PROTOTYPES: ENABLE	523	PROTOTYPES: ENABLE
1469		524
1470	BOOT:	525	BOOT:
1471	{	526	{
1472	stash = gv_stashpv ("IO::AIO", 1);	527	static const struct {
1473		528	const char *name;
1474	newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));	529	IV iv;
1475	newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));	530	} *civ, const_iv[] = {
1476	newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));	531	# define const_iv(name, value) { # name, (IV) value },
1477	newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));	532	# define const_eio(name) { # name, (IV) EIO_ ## name },
1478	newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));	533	const_iv (EXDEV , EXDEV)
		534	const_iv (ENOSYS , ENOSYS)
		535	const_iv (O_RDONLY, O_RDONLY)
		536	const_iv (O_WRONLY, O_WRONLY)
		537	const_iv (O_CREAT , O_CREAT)
		538	const_iv (O_TRUNC , O_TRUNC)
1479	#ifdef _WIN32	539	#ifndef _WIN32
1480	X_MUTEX_CHECK (wrklock);	540	const_iv (S_IFIFO , S_IFIFO)
1481	X_MUTEX_CHECK (reslock);
1482	X_MUTEX_CHECK (reqlock);
1483	X_MUTEX_CHECK (reqwait);
1484	X_MUTEX_CHECK (preadwritelock);
1485	X_MUTEX_CHECK (readdirlock);
1486
1487	X_COND_CHECK (reqwait);
1488	#else
1489	newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
1490	newCONSTSUB (stash, "SIGIO", newSViv (SIGIO));
1491	#endif	541	#endif
		542	const_iv (FADV_NORMAL , POSIX_FADV_NORMAL)
		543	const_iv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
		544	const_iv (FADV_RANDOM , POSIX_FADV_RANDOM)
		545	const_iv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
		546	const_iv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
		547	const_iv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
1492		548
		549	const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
		550	const_eio (SYNC_FILE_RANGE_WRITE)
		551	const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
		552
		553	const_eio (READDIR_DENTS)
		554	const_eio (READDIR_DIRS_FIRST)
		555	const_eio (READDIR_STAT_ORDER)
		556	const_eio (READDIR_FOUND_UNKNOWN)
		557
		558	const_eio (DT_UNKNOWN)
		559	const_eio (DT_FIFO)
		560	const_eio (DT_CHR)
		561	const_eio (DT_DIR)
		562	const_eio (DT_BLK)
		563	const_eio (DT_REG)
		564	const_eio (DT_LNK)
		565	const_eio (DT_SOCK)
		566	const_eio (DT_WHT)
		567	};
		568
		569	stash = gv_stashpv ("IO::AIO", 1);
		570
		571	for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
		572	newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
		573
1493	create_respipe ();	574	create_respipe ();
1494		575
1495	X_THREAD_ATFORK (atfork_prepare, atfork_parent, atfork_child);	576	if (eio_init (want_poll, done_poll) < 0)
		577	croak ("IO::AIO: unable to initialise eio library");
		578
		579	/* atfork child called in fifo order, so before eio's handler */
		580	X_THREAD_ATFORK (0, 0, atfork_child);
1496	}	581	}
1497		582
1498	void	583	void
1499	max_poll_reqs (int nreqs)	584	max_poll_reqs (int nreqs)
1500	PROTOTYPE: $	585	PROTOTYPE: $
1501	CODE:	586	CODE:
1502	max_poll_reqs = nreqs;	587	eio_set_max_poll_reqs (nreqs);
1503		588
1504	void	589	void
1505	max_poll_time (double nseconds)	590	max_poll_time (double nseconds)
1506	PROTOTYPE: $	591	PROTOTYPE: $
1507	CODE:	592	CODE:
1508	max_poll_time = nseconds * AIO_TICKS;	593	eio_set_max_poll_time (nseconds);
1509		594
1510	void	595	void
1511	min_parallel (int nthreads)	596	min_parallel (int nthreads)
1512	PROTOTYPE: $	597	PROTOTYPE: $
		598	CODE:
		599	eio_set_min_parallel (nthreads);
1513		600
1514	void	601	void
1515	max_parallel (int nthreads)	602	max_parallel (int nthreads)
1516	PROTOTYPE: $	603	PROTOTYPE: $
		604	CODE:
		605	eio_set_max_parallel (nthreads);
1517		606
1518	void	607	void
1519	max_idle (int nthreads)	608	max_idle (int nthreads)
1520	PROTOTYPE: $	609	PROTOTYPE: $
1521	CODE:	610	CODE:
1522	set_max_idle (nthreads);	611	eio_set_max_idle (nthreads);
1523		612
1524	int	613	void
1525	max_outstanding (int maxreqs)	614	max_outstanding (int maxreqs)
1526	PROTOTYPE: $	615	PROTOTYPE: $
1527	CODE:	616	CODE:
1528	RETVAL = max_outstanding;
1529	max_outstanding = maxreqs;	617	max_outstanding = maxreqs;
1530	OUTPUT:
1531	RETVAL
1532		618
1533	void	619	void
1534	aio_open (SV8 *pathname, int flags, int mode, SV *callback=&PL_sv_undef)	620	aio_open (SV8 *pathname, int flags, int mode, SV *callback=&PL_sv_undef)
1535	PROTOTYPE: $$$;$	621	PROTOTYPE: $$$;$
1536	PPCODE:	622	PPCODE:
1537	{	623	{
1538	dREQ;	624	dREQ;
1539		625
1540	req->type = REQ_OPEN;	626	req->type = EIO_OPEN;
1541	req->sv1 = newSVsv (pathname);	627	req->sv1 = newSVsv (pathname);
1542	req->ptr1 = SvPVbyte_nolen (req->sv1);	628	req->ptr1 = SvPVbyte_nolen (req->sv1);
1543	req->int1 = flags;	629	req->int1 = flags;
1544	req->mode = mode;	630	req->int2 = mode;
1545		631
1546	REQ_SEND;	632	REQ_SEND;
1547	}	633	}
1548		634
1549	void	635	void
1550	aio_fsync (SV *fh, SV *callback=&PL_sv_undef)	636	aio_fsync (SV *fh, SV *callback=&PL_sv_undef)
1551	PROTOTYPE: $;$	637	PROTOTYPE: $;$
1552	ALIAS:	638	ALIAS:
1553	aio_fsync = REQ_FSYNC	639	aio_fsync = EIO_FSYNC
1554	aio_fdatasync = REQ_FDATASYNC	640	aio_fdatasync = EIO_FDATASYNC
1555	PPCODE:	641	PPCODE:
1556	{	642	{
		643	int fd = s_fileno_croak (fh, 0);
1557	dREQ;	644	dREQ;
1558		645
1559	req->type = ix;	646	req->type = ix;
1560	req->sv1 = newSVsv (fh);	647	req->sv1 = newSVsv (fh);
1561	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));	648	req->int1 = fd;
		649
		650	REQ_SEND (req);
		651	}
		652
		653	void
		654	aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef)
		655	PROTOTYPE: $$$$;$
		656	PPCODE:
		657	{
		658	int fd = s_fileno_croak (fh, 0);
		659	dREQ;
		660
		661	req->type = EIO_SYNC_FILE_RANGE;
		662	req->sv1 = newSVsv (fh);
		663	req->int1 = fd;
		664	req->offs = offset;
		665	req->size = nbytes;
		666	req->int2 = flags;
1562		667
1563	REQ_SEND (req);	668	REQ_SEND (req);
1564	}	669	}
1565		670
1566	void	671	void
1567	aio_close (SV *fh, SV *callback=&PL_sv_undef)	672	aio_close (SV *fh, SV *callback=&PL_sv_undef)
1568	PROTOTYPE: $;$	673	PROTOTYPE: $;$
1569	PPCODE:	674	PPCODE:
1570	{	675	{
		676	static int close_pipe = -1; /* dummy fd to close fds via dup2 */
		677	int fd = s_fileno_croak (fh, 0);
1571	dREQ;	678	dREQ;
1572		679
		680	if (close_pipe < 0)
		681	{
		682	int pipefd [2];
		683
		684	if (pipe (pipefd) < 0
		685	|| close (pipefd [1]) < 0
		686	|| fcntl (pipefd [0], F_SETFD, FD_CLOEXEC) < 0)
		687	abort (); /*D*/
		688
		689	close_pipe = pipefd [0];
		690	}
		691
1573	req->type = REQ_CLOSE;	692	req->type = EIO_DUP2;
		693	req->int1 = close_pipe;
1574	req->sv1 = newSVsv (fh);	694	req->sv2 = newSVsv (fh);
1575	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));	695	req->int2 = fd;
1576		696
1577	REQ_SEND (req);	697	REQ_SEND (req);
1578	}	698	}
1579		699
1580	void	700	void
1581	aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback=&PL_sv_undef)	701	aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback=&PL_sv_undef)
1582	ALIAS:	702	ALIAS:
1583	aio_read = REQ_READ	703	aio_read = EIO_READ
1584	aio_write = REQ_WRITE	704	aio_write = EIO_WRITE
1585	PROTOTYPE: $$$$$;$	705	PROTOTYPE: $$$$$;$
1586	PPCODE:	706	PPCODE:
1587	{	707	{
1588	STRLEN svlen;	708	STRLEN svlen;
		709	int fd = s_fileno_croak (fh, ix == EIO_WRITE);
1589	char *svptr = SvPVbyte (data, svlen);	710	char *svptr = SvPVbyte (data, svlen);
1590	UV len = SvUV (length);	711	UV len = SvUV (length);
1591		712
1592	SvUPGRADE (data, SVt_PV);
1593	SvPOK_on (data);
1594
1595	if (dataoffset < 0)	713	if (dataoffset < 0)
1596	dataoffset += svlen;	714	dataoffset += svlen;
1597		715
1598	if (dataoffset < 0 || dataoffset > svlen)	716	if (dataoffset < 0 || dataoffset > svlen)
1599	croak ("dataoffset outside of data scalar");	717	croak ("dataoffset outside of data scalar");
1600		718
1601	if (ix == REQ_WRITE)	719	if (ix == EIO_WRITE)
1602	{	720	{
1603	/* write: check length and adjust. */	721	/* write: check length and adjust. */
1604	if (!SvOK (length) || len + dataoffset > svlen)	722	if (!SvOK (length) || len + dataoffset > svlen)
1605	len = svlen - dataoffset;	723	len = svlen - dataoffset;
1606	}	724	}
1607	else	725	else
1608	{	726	{
1609	/* read: grow scalar as necessary */	727	/* read: check type and grow scalar as necessary */
		728	SvUPGRADE (data, SVt_PV);
1610	svptr = SvGROW (data, len + dataoffset + 1);	729	svptr = SvGROW (data, len + dataoffset + 1);
1611	}	730	}
1612
1613	if (len < 0)
1614	croak ("length must not be negative");
1615		731
1616	{	732	{
1617	dREQ;	733	dREQ;
1618		734
1619	req->type = ix;	735	req->type = ix;
1620	req->sv1 = newSVsv (fh);	736	req->sv1 = newSVsv (fh);
1621	req->int1 = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))	737	req->int1 = fd;
1622	: IoOFP (sv_2io (fh)));
1623	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;	738	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
1624	req->size = len;	739	req->size = len;
1625	req->sv2 = SvREFCNT_inc (data);	740	req->sv2 = SvREFCNT_inc (data);
1626	req->ptr1 = (char *)svptr + dataoffset;	741	req->ptr2 = (char *)svptr + dataoffset;
1627	req->stroffset = dataoffset;	742	req->stroffset = dataoffset;
1628		743
1629	if (!SvREADONLY (data))	744	if (!SvREADONLY (data))
1630	{	745	{
1631	SvREADONLY_on (data);	746	SvREADONLY_on (data);
…		…
1642	PPCODE:	757	PPCODE:
1643	{	758	{
1644	SV *data;	759	SV *data;
1645	dREQ;	760	dREQ;
1646		761
1647	data = newSV (NAME_MAX);
1648	SvPOK_on (data);
1649
1650	req->type = REQ_READLINK;	762	req->type = EIO_READLINK;
1651	req->sv1 = newSVsv (path);	763	req->sv1 = newSVsv (path);
1652	req->ptr2 = SvPVbyte_nolen (req->sv1);	764	req->ptr1 = SvPVbyte_nolen (req->sv1);
1653	req->sv2 = data;
1654	req->ptr1 = SvPVbyte_nolen (data);
1655		765
1656	REQ_SEND;	766	REQ_SEND;
1657	}	767	}
1658		768
1659	void	769	void
1660	aio_sendfile (SV *out_fh, SV *in_fh, SV *in_offset, UV length, SV *callback=&PL_sv_undef)	770	aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef)
1661	PROTOTYPE: $$$$;$	771	PROTOTYPE: $$$$;$
1662	PPCODE:	772	PPCODE:
1663	{	773	{
		774	int ifd = s_fileno_croak (in_fh , 0);
		775	int ofd = s_fileno_croak (out_fh, 1);
1664	dREQ;	776	dREQ;
1665		777
1666	req->type = REQ_SENDFILE;	778	req->type = EIO_SENDFILE;
1667	req->sv1 = newSVsv (out_fh);	779	req->sv1 = newSVsv (out_fh);
1668	req->int1 = PerlIO_fileno (IoIFP (sv_2io (out_fh)));	780	req->int1 = ofd;
1669	req->sv2 = newSVsv (in_fh);	781	req->sv2 = newSVsv (in_fh);
1670	req->int2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));	782	req->int2 = ifd;
1671	req->offs = SvVAL64 (in_offset);	783	req->offs = in_offset;
1672	req->size = length;	784	req->size = length;
1673		785
1674	REQ_SEND;	786	REQ_SEND;
1675	}	787	}
1676		788
1677	void	789	void
1678	aio_readahead (SV *fh, SV *offset, IV length, SV *callback=&PL_sv_undef)	790	aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef)
1679	PROTOTYPE: $$$;$	791	PROTOTYPE: $$$;$
1680	PPCODE:	792	PPCODE:
1681	{	793	{
		794	int fd = s_fileno_croak (fh, 0);
1682	dREQ;	795	dREQ;
1683		796
1684	req->type = REQ_READAHEAD;	797	req->type = EIO_READAHEAD;
1685	req->sv1 = newSVsv (fh);	798	req->sv1 = newSVsv (fh);
1686	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));	799	req->int1 = fd;
1687	req->offs = SvVAL64 (offset);	800	req->offs = offset;
1688	req->size = length;	801	req->size = length;
1689		802
1690	REQ_SEND;	803	REQ_SEND;
1691	}	804	}
1692		805
1693	void	806	void
1694	aio_stat (SV8 *fh_or_path, SV *callback=&PL_sv_undef)	807	aio_stat (SV8 *fh_or_path, SV *callback=&PL_sv_undef)
1695	ALIAS:	808	ALIAS:
1696	aio_stat = REQ_STAT	809	aio_stat = EIO_STAT
1697	aio_lstat = REQ_LSTAT	810	aio_lstat = EIO_LSTAT
1698	PPCODE:	811	PPCODE:
1699	{	812	{
1700	dREQ;	813	dREQ;
1701		814
1702	req->ptr2 = malloc (sizeof (Stat_t));
1703	if (!req->ptr2)
1704	{
1705	req_destroy (req);
1706	croak ("out of memory during aio_stat statdata allocation");
1707	}
1708
1709	req->flags |= FLAG_PTR2_FREE;
1710	req->sv1 = newSVsv (fh_or_path);	815	req->sv1 = newSVsv (fh_or_path);
1711		816
1712	if (SvPOK (fh_or_path))	817	if (SvPOK (req->sv1))
1713	{	818	{
1714	req->type = ix;	819	req->type = ix;
1715	req->ptr1 = SvPVbyte_nolen (req->sv1);	820	req->ptr1 = SvPVbyte_nolen (req->sv1);
1716	}	821	}
1717	else	822	else
1718	{	823	{
1719	req->type = REQ_FSTAT;	824	req->type = EIO_FSTAT;
1720	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	825	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1721	}	826	}
1722		827
1723	REQ_SEND;	828	REQ_SEND;
1724	}	829	}
…		…
1731		836
1732	req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;	837	req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
1733	req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;	838	req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
1734	req->sv1 = newSVsv (fh_or_path);	839	req->sv1 = newSVsv (fh_or_path);
1735		840
1736	if (SvPOK (fh_or_path))	841	if (SvPOK (req->sv1))
1737	{	842	{
1738	req->type = REQ_UTIME;	843	req->type = EIO_UTIME;
1739	req->ptr1 = SvPVbyte_nolen (req->sv1);	844	req->ptr1 = SvPVbyte_nolen (req->sv1);
1740	}	845	}
1741	else	846	else
1742	{	847	{
1743	req->type = REQ_FUTIME;	848	req->type = EIO_FUTIME;
1744	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	849	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1745	}	850	}
1746		851
1747	REQ_SEND;	852	REQ_SEND;
1748	}	853	}
…		…
1754	dREQ;	859	dREQ;
1755		860
1756	req->sv1 = newSVsv (fh_or_path);	861	req->sv1 = newSVsv (fh_or_path);
1757	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;	862	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
1758		863
1759	if (SvPOK (fh_or_path))	864	if (SvPOK (req->sv1))
1760	{	865	{
1761	req->type = REQ_TRUNCATE;	866	req->type = EIO_TRUNCATE;
1762	req->ptr1 = SvPVbyte_nolen (req->sv1);	867	req->ptr1 = SvPVbyte_nolen (req->sv1);
1763	}	868	}
1764	else	869	else
1765	{	870	{
1766	req->type = REQ_FTRUNCATE;	871	req->type = EIO_FTRUNCATE;
1767	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	872	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1768	}	873	}
1769		874
1770	REQ_SEND;	875	REQ_SEND;
1771	}	876	}
1772		877
1773	void	878	void
1774	aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)	879	aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
		880	ALIAS:
		881	aio_chmod = EIO_CHMOD
		882	aio_mkdir = EIO_MKDIR
1775	PPCODE:	883	PPCODE:
1776	{	884	{
1777	dREQ;	885	dREQ;
1778		886
1779	req->mode = mode;	887	req->int2 = mode;
1780	req->sv1 = newSVsv (fh_or_path);	888	req->sv1 = newSVsv (fh_or_path);
1781		889
1782	if (SvPOK (fh_or_path))	890	if (SvPOK (req->sv1))
1783	{	891	{
1784	req->type = REQ_CHMOD;	892	req->type = ix;
1785	req->ptr1 = SvPVbyte_nolen (req->sv1);	893	req->ptr1 = SvPVbyte_nolen (req->sv1);
1786	}	894	}
1787	else	895	else
1788	{	896	{
1789	req->type = REQ_FCHMOD;	897	req->type = EIO_FCHMOD;
1790	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	898	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1791	}	899	}
1792		900
1793	REQ_SEND;	901	REQ_SEND;
1794	}	902	}
…		…
1801		909
1802	req->int2 = SvOK (uid) ? SvIV (uid) : -1;	910	req->int2 = SvOK (uid) ? SvIV (uid) : -1;
1803	req->int3 = SvOK (gid) ? SvIV (gid) : -1;	911	req->int3 = SvOK (gid) ? SvIV (gid) : -1;
1804	req->sv1 = newSVsv (fh_or_path);	912	req->sv1 = newSVsv (fh_or_path);
1805		913
1806	if (SvPOK (fh_or_path))	914	if (SvPOK (req->sv1))
1807	{	915	{
1808	req->type = REQ_CHOWN;	916	req->type = EIO_CHOWN;
1809	req->ptr1 = SvPVbyte_nolen (req->sv1);	917	req->ptr1 = SvPVbyte_nolen (req->sv1);
1810	}	918	}
1811	else	919	else
1812	{	920	{
1813	req->type = REQ_FCHOWN;	921	req->type = EIO_FCHOWN;
1814	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	922	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1815	}	923	}
1816		924
1817	REQ_SEND;	925	REQ_SEND;
		926	}
		927
		928	void
		929	aio_readdirx (SV8 *pathname, IV flags, SV *callback=&PL_sv_undef)
		930	PPCODE:
		931	{
		932	dREQ;
		933
		934	req->type = EIO_READDIR;
		935	req->sv1 = newSVsv (pathname);
		936	req->ptr1 = SvPVbyte_nolen (req->sv1);
		937	req->int1 = flags | EIO_READDIR_DENTS | EIO_READDIR_CUSTOM1;
		938
		939	if (flags & EIO_READDIR_DENTS)
		940	req->int1 |= EIO_READDIR_CUSTOM2;
		941
		942	REQ_SEND;
1818	}	943	}
1819		944
1820	void	945	void
1821	aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)	946	aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)
1822	ALIAS:	947	ALIAS:
1823	aio_unlink = REQ_UNLINK	948	aio_unlink = EIO_UNLINK
1824	aio_rmdir = REQ_RMDIR	949	aio_rmdir = EIO_RMDIR
1825	aio_readdir = REQ_READDIR	950	aio_readdir = EIO_READDIR
1826	PPCODE:	951	PPCODE:
1827	{	952	{
1828	dREQ;	953	dREQ;
1829		954
1830	req->type = ix;	955	req->type = ix;
…		…
1833		958
1834	REQ_SEND;	959	REQ_SEND;
1835	}	960	}
1836		961
1837	void	962	void
1838	aio_mkdir (SV8 *pathname, int mode, SV *callback=&PL_sv_undef)	963	aio_link (SV8 *oldpath, SV8 *newpath, SV *callback=&PL_sv_undef)
		964	ALIAS:
		965	aio_link = EIO_LINK
		966	aio_symlink = EIO_SYMLINK
		967	aio_rename = EIO_RENAME
1839	PPCODE:	968	PPCODE:
1840	{	969	{
1841	dREQ;	970	dREQ;
1842		971
		972	req->type = ix;
		973	req->sv1 = newSVsv (oldpath);
		974	req->ptr1 = SvPVbyte_nolen (req->sv1);
		975	req->sv2 = newSVsv (newpath);
		976	req->ptr2 = SvPVbyte_nolen (req->sv2);
		977
		978	REQ_SEND;
		979	}
		980
		981	void
		982	aio_mknod (SV8 *pathname, int mode, UV dev, SV *callback=&PL_sv_undef)
		983	PPCODE:
		984	{
		985	dREQ;
		986
1843	req->type = REQ_MKDIR;	987	req->type = EIO_MKNOD;
1844	req->sv1 = newSVsv (pathname);	988	req->sv1 = newSVsv (pathname);
1845	req->ptr1 = SvPVbyte_nolen (req->sv1);	989	req->ptr1 = SvPVbyte_nolen (req->sv1);
1846	req->mode = mode;
1847
1848	REQ_SEND;
1849	}
1850
1851	void
1852	aio_link (SV8 *oldpath, SV8 *newpath, SV *callback=&PL_sv_undef)
1853	ALIAS:
1854	aio_link = REQ_LINK
1855	aio_symlink = REQ_SYMLINK
1856	aio_rename = REQ_RENAME
1857	PPCODE:
1858	{
1859	dREQ;
1860
1861	req->type = ix;
1862	req->sv2 = newSVsv (oldpath);
1863	req->ptr2 = SvPVbyte_nolen (req->sv2);
1864	req->sv1 = newSVsv (newpath);
1865	req->ptr1 = SvPVbyte_nolen (req->sv1);
1866
1867	REQ_SEND;
1868	}
1869
1870	void
1871	aio_mknod (SV8 *pathname, int mode, UV dev, SV *callback=&PL_sv_undef)
1872	PPCODE:
1873	{
1874	dREQ;
1875
1876	req->type = REQ_MKNOD;
1877	req->sv1 = newSVsv (pathname);
1878	req->ptr1 = SvPVbyte_nolen (req->sv1);
1879	req->mode = (mode_t)mode;	990	req->int2 = (mode_t)mode;
1880	req->offs = dev;	991	req->offs = dev;
1881		992
1882	REQ_SEND;	993	REQ_SEND;
1883	}	994	}
1884		995
…		…
1886	aio_busy (double delay, SV *callback=&PL_sv_undef)	997	aio_busy (double delay, SV *callback=&PL_sv_undef)
1887	PPCODE:	998	PPCODE:
1888	{	999	{
1889	dREQ;	1000	dREQ;
1890		1001
1891	req->type = REQ_BUSY;	1002	req->type = EIO_BUSY;
1892	req->nv1 = delay < 0. ? 0. : delay;	1003	req->nv1 = delay < 0. ? 0. : delay;
1893		1004
1894	REQ_SEND;	1005	REQ_SEND;
1895	}	1006	}
1896		1007
…		…
1899	PROTOTYPE: ;$	1010	PROTOTYPE: ;$
1900	PPCODE:	1011	PPCODE:
1901	{	1012	{
1902	dREQ;	1013	dREQ;
1903		1014
1904	req->type = REQ_GROUP;	1015	req->type = EIO_GROUP;
1905		1016
1906	req_send (req);	1017	req_submit (req);
1907	XPUSHs (req_sv (req, AIO_GRP_KLASS));	1018	XPUSHs (req_sv (req, AIO_GRP_KLASS));
1908	}	1019	}
1909		1020
1910	void	1021	void
1911	aio_nop (SV *callback=&PL_sv_undef)	1022	aio_nop (SV *callback=&PL_sv_undef)
1912	ALIAS:	1023	ALIAS:
1913	aio_nop = REQ_NOP	1024	aio_nop = EIO_NOP
1914	aio_sync = REQ_SYNC	1025	aio_sync = EIO_SYNC
1915	PPCODE:	1026	PPCODE:
1916	{	1027	{
1917	dREQ;	1028	dREQ;
1918		1029
1919	req->type = ix;	1030	req->type = ix;
…		…
1923		1034
1924	int	1035	int
1925	aioreq_pri (int pri = 0)	1036	aioreq_pri (int pri = 0)
1926	PROTOTYPE: ;$	1037	PROTOTYPE: ;$
1927	CODE:	1038	CODE:
1928	RETVAL = next_pri - PRI_BIAS;	1039	RETVAL = next_pri;
1929	if (items > 0)	1040	if (items > 0)
1930	{	1041	{
1931	if (pri < PRI_MIN) pri = PRI_MIN;	1042	if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
1932	if (pri > PRI_MAX) pri = PRI_MAX;	1043	if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
1933	next_pri = pri + PRI_BIAS;	1044	next_pri = pri;
1934	}	1045	}
1935	OUTPUT:	1046	OUTPUT:
1936	RETVAL	1047	RETVAL
1937		1048
1938	void	1049	void
1939	aioreq_nice (int nice = 0)	1050	aioreq_nice (int nice = 0)
1940	CODE:	1051	CODE:
1941	nice = next_pri - nice;	1052	nice = next_pri - nice;
1942	if (nice < PRI_MIN) nice = PRI_MIN;	1053	if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
1943	if (nice > PRI_MAX) nice = PRI_MAX;	1054	if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
1944	next_pri = nice + PRI_BIAS;	1055	next_pri = nice;
1945		1056
1946	void	1057	void
1947	flush ()	1058	flush ()
1948	PROTOTYPE:	1059	PROTOTYPE:
1949	CODE:	1060	CODE:
1950	while (nreqs)	1061	while (eio_nreqs ())
1951	{	1062	{
1952	poll_wait ();	1063	poll_wait ();
1953	poll_cb ();	1064	poll_cb ();
1954	}	1065	}
1955		1066
…		…
1964		1075
1965	int	1076	int
1966	poll_fileno()	1077	poll_fileno()
1967	PROTOTYPE:	1078	PROTOTYPE:
1968	CODE:	1079	CODE:
1969	RETVAL = respipe [0];	1080	RETVAL = s_epipe_fd (&respipe);
1970	OUTPUT:	1081	OUTPUT:
1971	RETVAL	1082	RETVAL
1972		1083
1973	int	1084	int
1974	poll_cb(...)	1085	poll_cb(...)
…		…
1982	poll_wait()	1093	poll_wait()
1983	PROTOTYPE:	1094	PROTOTYPE:
1984	CODE:	1095	CODE:
1985	poll_wait ();	1096	poll_wait ();
1986		1097
1987	void
1988	setsig (int signum = SIGIO)
1989	PROTOTYPE: ;$
1990	CODE:
1991	{
1992	if (block_sig_level)
1993	croak ("cannot call IO::AIO::setsig from within aio_block/callback");
1994
1995	X_LOCK (reslock);
1996	main_tid = pthread_self ();
1997	main_sig = signum;
1998	X_UNLOCK (reslock);
1999
2000	if (main_sig && npending)
2001	pthread_kill (main_tid, main_sig);
2002	}
2003
2004	void
2005	aio_block (SV *cb)
2006	PROTOTYPE: &
2007	PPCODE:
2008	{
2009	int count;
2010
2011	block_sig ();
2012	PUSHMARK (SP);
2013	PUTBACK;
2014	count = call_sv (cb, GIMME_V | G_NOARGS | G_EVAL);
2015	unblock_sig ();
2016
2017	if (SvTRUE (ERRSV))
2018	croak (0);
2019
2020	XSRETURN (count);
2021	}
2022
2023	int	1098	int
2024	nreqs()	1099	nreqs()
2025	PROTOTYPE:	1100	PROTOTYPE:
2026	CODE:	1101	CODE:
2027	RETVAL = nreqs;	1102	RETVAL = eio_nreqs ();
2028	OUTPUT:	1103	OUTPUT:
2029	RETVAL	1104	RETVAL
2030		1105
2031	int	1106	int
2032	nready()	1107	nready()
2033	PROTOTYPE:	1108	PROTOTYPE:
2034	CODE:	1109	CODE:
2035	RETVAL = get_nready ();	1110	RETVAL = eio_nready ();
2036	OUTPUT:	1111	OUTPUT:
2037	RETVAL	1112	RETVAL
2038		1113
2039	int	1114	int
2040	npending()	1115	npending()
2041	PROTOTYPE:	1116	PROTOTYPE:
2042	CODE:	1117	CODE:
2043	RETVAL = get_npending ();	1118	RETVAL = eio_npending ();
2044	OUTPUT:	1119	OUTPUT:
2045	RETVAL	1120	RETVAL
2046		1121
2047	int	1122	int
2048	nthreads()	1123	nthreads()
2049	PROTOTYPE:	1124	PROTOTYPE:
2050	CODE:	1125	CODE:
2051	if (WORDACCESS_UNSAFE) X_LOCK (wrklock);	1126	RETVAL = eio_nthreads ();
2052	RETVAL = started;
2053	if (WORDACCESS_UNSAFE) X_UNLOCK (wrklock);
2054	OUTPUT:	1127	OUTPUT:
2055	RETVAL	1128	RETVAL
2056		1129
		1130	int
		1131	fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
		1132	PROTOTYPE: $$$$
		1133	CODE:
		1134	#if _XOPEN_SOURCE >= 600 && !NO_FADVISE
		1135	RETVAL = posix_fadvise (fh, offset, length, advice);
		1136	#else
		1137	RETVAL = errno = ENOSYS;
		1138	#endif
		1139	OUTPUT:
		1140	RETVAL
		1141
		1142	ssize_t
		1143	sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
		1144	PROTOTYPE: $$$$
		1145	CODE:
		1146	eio_sendfile_sync (ofh, ifh, offset, count);
		1147
		1148	void _on_next_submit (SV *cb)
		1149	CODE:
		1150	SvREFCNT_dec (on_next_submit);
		1151	on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
		1152
2057	PROTOTYPES: DISABLE	1153	PROTOTYPES: DISABLE
2058		1154
2059	MODULE = IO::AIO PACKAGE = IO::AIO::REQ	1155	MODULE = IO::AIO PACKAGE = IO::AIO::REQ
2060		1156
2061	void	1157	void
2062	cancel (aio_req_ornot req)	1158	cancel (aio_req_ornot req)
2063	CODE:	1159	CODE:
2064	req_cancel (req);	1160	eio_cancel (req);
2065		1161
2066	void	1162	void
2067	cb (aio_req_ornot req, SV *callback=&PL_sv_undef)	1163	cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
2068	CODE:	1164	PPCODE:
		1165	{
		1166	if (GIMME_V != G_VOID)
		1167	XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
		1168
		1169	if (items > 1)
		1170	{
		1171	SV *cb_cv =get_cb (callback);
		1172
2069	SvREFCNT_dec (req->callback);	1173	SvREFCNT_dec (req->callback);
2070	req->callback = newSVsv (callback);	1174	req->callback = SvREFCNT_inc (cb_cv);
		1175	}
		1176	}
2071		1177
2072	MODULE = IO::AIO PACKAGE = IO::AIO::GRP	1178	MODULE = IO::AIO PACKAGE = IO::AIO::GRP
2073		1179
2074	void	1180	void
2075	add (aio_req grp, ...)	1181	add (aio_req grp, ...)
2076	PPCODE:	1182	PPCODE:
2077	{	1183	{
2078	int i;	1184	int i;
2079	aio_req req;
2080
2081	if (main_sig && !block_sig_level)
2082	croak ("aio_group->add called outside aio_block/callback context while IO::AIO::setsig is in use");
2083		1185
2084	if (grp->int1 == 2)	1186	if (grp->int1 == 2)
2085	croak ("cannot add requests to IO::AIO::GRP after the group finished");	1187	croak ("cannot add requests to IO::AIO::GRP after the group finished");
2086		1188
2087	for (i = 1; i < items; ++i )	1189	for (i = 1; i < items; ++i )
2088	{	1190	{
		1191	aio_req req;
		1192
2089	if (GIMME_V != G_VOID)	1193	if (GIMME_V != G_VOID)
2090	XPUSHs (sv_2mortal (newSVsv (ST (i))));	1194	XPUSHs (sv_2mortal (newSVsv (ST (i))));
2091		1195
2092	req = SvAIO_REQ (ST (i));	1196	req = SvAIO_REQ (ST (i));
2093		1197
2094	if (req)	1198	if (req)
2095	{	1199	eio_grp_add (grp, req);
2096	++grp->size;
2097	req->grp = grp;
2098
2099	req->grp_prev = 0;
2100	req->grp_next = grp->grp_first;
2101
2102	if (grp->grp_first)
2103	grp->grp_first->grp_prev = req;
2104
2105	grp->grp_first = req;
2106	}
2107	}	1200	}
2108	}	1201	}
2109		1202
2110	void	1203	void
2111	cancel_subs (aio_req_ornot req)	1204	cancel_subs (aio_req_ornot req)
…		…
2120	AV *av;	1213	AV *av;
2121		1214
2122	grp->errorno = errno;	1215	grp->errorno = errno;
2123		1216
2124	av = newAV ();	1217	av = newAV ();
		1218	av_extend (av, items - 1);
2125		1219
2126	for (i = 1; i < items; ++i )	1220	for (i = 1; i < items; ++i )
2127	av_push (av, newSVsv (ST (i)));	1221	av_push (av, newSVsv (ST (i)));
2128		1222
2129	SvREFCNT_dec (grp->sv1);	1223	SvREFCNT_dec (grp->sv1);
…		…
2136	grp->errorno = errorno;	1230	grp->errorno = errorno;
2137		1231
2138	void	1232	void
2139	limit (aio_req grp, int limit)	1233	limit (aio_req grp, int limit)
2140	CODE:	1234	CODE:
2141	grp->int2 = limit;	1235	eio_grp_limit (grp, limit);
2142	aio_grp_feed (grp);
2143		1236
2144	void	1237	void
2145	feed (aio_req grp, SV *callback=&PL_sv_undef)	1238	feed (aio_req grp, SV *callback=&PL_sv_undef)
2146	CODE:	1239	CODE:
2147	{	1240	{
2148	SvREFCNT_dec (grp->sv2);	1241	SvREFCNT_dec (grp->sv2);
2149	grp->sv2 = newSVsv (callback);	1242	grp->sv2 = newSVsv (callback);
		1243	grp->feed = aio_grp_feed;
2150		1244
2151	if (grp->int2 <= 0)	1245	if (grp->int2 <= 0)
2152	grp->int2 = 2;	1246	grp->int2 = 2;
2153		1247
2154	aio_grp_feed (grp);	1248	eio_grp_limit (grp, grp->int2);
2155	}	1249	}
2156		1250

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines