ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/IO-AIO/AIO.xs

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

Diff Legend

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