ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/IO-AIO/AIO.xs

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

Diff Legend

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