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.152 by root, Tue Jul 14 00:32:27 2009 UTC

1	#include "xthread.h"	1	#include "libeio/xthread.h"
2		2
3	#include <errno.h>	3	#include <errno.h>
4		4
5	#include "EXTERN.h"	5	#include "EXTERN.h"
6	#include "perl.h"	6	#include "perl.h"
7	#include "XSUB.h"	7	#include "XSUB.h"
		8
		9	#include "schmorp.h"
8		10
9	#include <stddef.h>	11	#include <stddef.h>
10	#include <stdlib.h>	12	#include <stdlib.h>
11	#include <errno.h>	13	#include <errno.h>
12	#include <sys/types.h>	14	#include <sys/types.h>
13	#include <sys/stat.h>	15	#include <sys/stat.h>
14	#include <limits.h>	16	#include <limits.h>
15	#include <fcntl.h>	17	#include <fcntl.h>
16	#include <sched.h>	18	#include <sched.h>
17		19
		20	/* perl namespace pollution */
		21	#undef VERSION
		22
18	#ifdef _WIN32	23	#ifdef _WIN32
19		24
20	# define SIGIO 0	25	# define EIO_STRUCT_DIRENT Direntry_t
21	typedef Direntry_t X_DIRENT;
22	#undef malloc	26	# undef malloc
23	#undef free	27	# undef free
24		28
25	// perl overrides all those nice win32 functions	29	// perl overrides all those nice win32 functions
26	# undef open	30	# undef open
27	# undef read	31	# undef read
28	# undef write	32	# undef write
…		…
54	# define closedir(fd) (errno = ENOSYS, -1)	58	# define closedir(fd) (errno = ENOSYS, -1)
55	# define mkdir(a,b) mkdir (a)	59	# define mkdir(a,b) mkdir (a)
56		60
57	#else	61	#else
58		62
59	# include "autoconf/config.h"
60	# include <sys/time.h>	63	# include <sys/time.h>
61	# include <sys/select.h>	64	# include <sys/select.h>
62	# include <unistd.h>	65	# include <unistd.h>
63	# include <utime.h>	66	# include <utime.h>
64	# include <signal.h>	67	# include <signal.h>
65	typedef struct dirent X_DIRENT;	68	# define EIO_STRUCT_DIRENT struct dirent
66		69
67	#endif
68
69	#if HAVE_SENDFILE
70	# if __linux
71	# include <sys/sendfile.h>
72	# elif __freebsd
73	# include <sys/socket.h>
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	70	#endif
82	#endif
83		71
84	/* number of seconds after which idle threads exit */	72	/* perl stupidly overrides readdir and maybe others */
85	#define IDLE_TIMEOUT 10	73	/* with thread-unsafe versions, imagine that :( */
		74	#undef readdir
		75	#undef opendir
		76	#undef closedir
86		77
87	/* used for struct dirent, AIX doesn't provide it */	78	#define EIO_STRUCT_STAT Stat_t
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	#define dBUF \	91	/*****************************************************************************/
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		92
		93	#if __GNUC__ >= 3
		94	# define expect(expr,value) __builtin_expect ((expr),(value))
		95	#else
		96	# define expect(expr,value) (expr)
		97	#endif
		98
		99	#define expect_false(expr) expect ((expr) != 0, 0)
		100	#define expect_true(expr) expect ((expr) != 0, 1)
		101
		102	/*****************************************************************************/
		103
		104	static HV *stash;
110	typedef SV SV8; /* byte-sv, used for argument-checking */	105	typedef SV SV8; /* byte-sv, used for argument-checking */
111		106	typedef int aio_rfd; /* read file desriptor */
112	enum {	107	typedef int aio_wfd; /* write file descriptor */
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		108
130	#define AIO_REQ_KLASS "IO::AIO::REQ"	109	#define AIO_REQ_KLASS "IO::AIO::REQ"
131	#define AIO_GRP_KLASS "IO::AIO::GRP"	110	#define AIO_GRP_KLASS "IO::AIO::GRP"
132		111
133	typedef struct aio_cb	112	#define EIO_REQ_MEMBERS \
134	{
135	struct aio_cb *volatile next;
136
137	SV *callback;	113	SV *callback; \
138	SV *sv1, *sv2;	114	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;	115	STRLEN stroffset; \
146	int type;	116	SV *self;
147	int int1, int2, int3;
148	int errorno;
149	mode_t mode; /* open */
150		117
151	unsigned char flags;	118	#define EIO_NO_WRAPPERS 1
152	unsigned char pri;
153		119
154	SV *self; /* the perl counterpart of this request, if any */	120	#include "libeio/eio.h"
155	struct aio_cb *grp, *grp_prev, *grp_next, *grp_first;	121
156	} aio_cb;	122	#ifndef POSIX_FADV_NORMAL
		123	# define POSIX_FADV_NORMAL 0
		124	# define NO_FADVISE 1
		125	#endif
		126
		127	#ifndef POSIX_FADV_SEQUENTIAL
		128	# define POSIX_FADV_SEQUENTIAL 0
		129	#endif
		130
		131	#ifndef POSIX_FADV_RANDOM
		132	# define POSIX_FADV_RANDOM 0
		133	#endif
		134
		135	#ifndef POSIX_FADV_NOREUSE
		136	# define POSIX_FADV_NOREUSE 0
		137	#endif
		138
		139	#ifndef POSIX_FADV_WILLNEED
		140	# define POSIX_FADV_WILLNEED 0
		141	#endif
		142
		143	#ifndef POSIX_FADV_DONTNEED
		144	# define POSIX_FADV_DONTNEED 0
		145	#endif
		146
		147	static int req_invoke (eio_req *req);
		148	#define EIO_FINISH(req) req_invoke (req)
		149	static void req_destroy (eio_req *grp);
		150	#define EIO_DESTROY(req) req_destroy (req)
157		151
158	enum {	152	enum {
159	FLAG_CANCELLED = 0x01, /* request was cancelled */
160	FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */	153	FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
161	FLAG_PTR2_FREE = 0x80, /* need to free(ptr2) */
162	};	154	};
163		155
		156	#include "libeio/eio.c"
		157
164	typedef aio_cb *aio_req;	158	typedef eio_req *aio_req;
165	typedef aio_cb *aio_req_ornot;	159	typedef eio_req *aio_req_ornot;
166		160
167	enum {	161	static SV *on_next_submit;
168	PRI_MIN = -4,	162	static int next_pri = EIO_PRI_DEFAULT;
169	PRI_MAX = 4,	163	static int max_outstanding;
170		164
171	DEFAULT_PRI = 0,	165	static int respipe_osf [2], respipe [2] = { -1, -1 };
172	PRI_BIAS = -PRI_MIN,
173	NUM_PRI = PRI_MAX + PRI_BIAS + 1,
174	};
175		166
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);	167	static void req_destroy (aio_req req);
375	static void req_cancel (aio_req req);	168	static void req_cancel (aio_req req);
		169
		170	static void want_poll (void)
		171	{
		172	/* write a dummy byte to the pipe so fh becomes ready */
		173	respipe_write (respipe_osf [1], (const void *)&respipe_osf, 1);
		174	}
		175
		176	static void done_poll (void)
		177	{
		178	/* read any signals sent by the worker threads */
		179	char buf [4];
		180	while (respipe_read (respipe [0], buf, 4) == 4)
		181	;
		182	}
376		183
377	/* must be called at most once */	184	/* must be called at most once */
378	static SV *req_sv (aio_req req, const char *klass)	185	static SV *req_sv (aio_req req, const char *klass)
379	{	186	{
380	if (!req->self)	187	if (!req->self)
…		…
398	return mg ? (aio_req)mg->mg_ptr : 0;	205	return mg ? (aio_req)mg->mg_ptr : 0;
399	}	206	}
400		207
401	static void aio_grp_feed (aio_req grp)	208	static void aio_grp_feed (aio_req grp)
402	{	209	{
403	block_sig ();	210	if (grp->sv2 && SvOK (grp->sv2))
404
405	while (grp->size < grp->int2 && !(grp->flags & FLAG_CANCELLED))
406	{	211	{
407	int old_len = grp->size;
408
409	if (grp->sv2 && SvOK (grp->sv2))
410	{
411	dSP;	212	dSP;
412		213
413	ENTER;	214	ENTER;
414	SAVETMPS;	215	SAVETMPS;
415	PUSHMARK (SP);	216	PUSHMARK (SP);
416	XPUSHs (req_sv (grp, AIO_GRP_KLASS));	217	XPUSHs (req_sv (grp, AIO_GRP_KLASS));
417	PUTBACK;	218	PUTBACK;
418	call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);	219	call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
419	SPAGAIN;	220	SPAGAIN;
420	FREETMPS;	221	FREETMPS;
421	LEAVE;	222	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	}	223	}
432
433	unblock_sig ();
434	}	224	}
435		225
436	static void aio_grp_dec (aio_req grp)	226	static void req_submit (eio_req *req)
437	{	227	{
438	--grp->size;	228	eio_submit (req);
439		229
440	/* call feeder, if applicable */	230	if (expect_false (on_next_submit))
441	aio_grp_feed (grp);
442
443	/* finish, if done */
444	if (!grp->size && grp->int1)
445	{	231	{
446	block_sig ();	232	dSP;
		233	SV *cb = sv_2mortal (on_next_submit);
447		234
448	if (!req_invoke (grp))	235	on_next_submit = 0;
449	{
450	req_destroy (grp);
451	unblock_sig ();
452	croak (0);
453	}
454		236
455	req_destroy (grp);	237	PUSHMARK (SP);
456	unblock_sig ();	238	PUTBACK;
		239	call_sv (cb, G_DISCARD | G_EVAL);
457	}	240	}
458	}	241	}
459		242
460	static int req_invoke (aio_req req)	243	static int req_invoke (eio_req *req)
461	{	244	{
462	dSP;	245	dSP;
463		246
464	if (req->flags & FLAG_SV2_RO_OFF)	247	if (req->flags & FLAG_SV2_RO_OFF)
465	SvREADONLY_off (req->sv2);	248	SvREADONLY_off (req->sv2);
466		249
467	if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback))	250	if (!EIO_CANCELLED (req) && req->callback)
468	{	251	{
469	ENTER;	252	ENTER;
470	SAVETMPS;	253	SAVETMPS;
471	PUSHMARK (SP);	254	PUSHMARK (SP);
472	EXTEND (SP, 1);	255	EXTEND (SP, 1);
473		256
474	switch (req->type)	257	switch (req->type)
475	{	258	{
476	case REQ_READDIR:	259	case EIO_READDIR:
477	{	260	{
478	SV *rv = &PL_sv_undef;	261	SV *rv = &PL_sv_undef;
479		262
480	if (req->result >= 0)	263	if (req->result >= 0)
481	{	264	{
482	int i;	265	int i;
483	char *buf = req->ptr2;	266	char *names = (char *)req->ptr2;
		267	eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
484	AV *av = newAV ();	268	AV *av = newAV ();
485		269
486	av_extend (av, req->result - 1);	270	av_extend (av, req->result - 1);
487		271
488	for (i = 0; i < req->result; ++i)	272	for (i = 0; i < req->result; ++i)
489	{	273	{
490	SV *sv = newSVpv (buf, 0);	274	if (req->int1 & EIO_READDIR_DENTS)
		275	{
		276	SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
491		277
		278	if (req->int1 & EIO_READDIR_CUSTOM2)
		279	{
		280	static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */
		281	AV *avent = newAV ();
		282
		283	av_extend (avent, 2);
		284
		285	if (!sv_type [ent->type])
		286	{
		287	sv_type [ent->type] = newSViv (ent->type);
		288	SvREADONLY_on (sv_type [ent->type]);
		289	}
		290
		291	av_store (avent, 0, namesv);
		292	av_store (avent, 1, SvREFCNT_inc (sv_type [ent->type]));
		293	av_store (avent, 2, IVSIZE >= 8 ? newSVuv (ent->inode) : newSVnv (ent->inode));
		294
		295	av_store (av, i, newRV_noinc ((SV *)avent));
		296	}
		297	else
492	av_store (av, i, sv);	298	av_store (av, i, namesv);
		299
		300	++ent;
		301	}
		302	else
		303	{
		304	SV *name = newSVpv (names, 0);
		305	av_store (av, i, name);
493	buf += SvCUR (sv) + 1;	306	names += SvCUR (name) + 1;
		307	}
494	}	308	}
495		309
496	rv = sv_2mortal (newRV_noinc ((SV *)av));	310	rv = sv_2mortal (newRV_noinc ((SV *)av));
497	}	311	}
498		312
499	PUSHs (rv);	313	PUSHs (rv);
		314
		315	if (req->int1 & EIO_READDIR_CUSTOM1)
		316	XPUSHs (sv_2mortal (newSViv (req->int1 & ~(EIO_READDIR_CUSTOM1 | EIO_READDIR_CUSTOM2))));
500	}	317	}
501	break;	318	break;
502		319
503	case REQ_OPEN:	320	case EIO_OPEN:
504	{	321	{
505	/* convert fd to fh */	322	/* convert fd to fh */
506	SV *fh;	323	SV *fh = &PL_sv_undef;
507		324
508	PUSHs (sv_2mortal (newSViv (req->result)));	325	if (req->result >= 0)
509	PUTBACK;	326	{
510	call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);	327	GV *gv = (GV *)sv_newmortal ();
511	SPAGAIN;	328	int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
		329	char sym [64];
		330	int symlen;
		331
		332	symlen = snprintf (sym, sizeof (sym), "fd#%d", (int)req->result);
		333	gv_init (gv, stash, sym, symlen, 0);
512		334
513	fh = POPs;	335	symlen = snprintf (
514	PUSHMARK (SP);	336	sym,
		337	sizeof (sym),
		338	"%s&=%d",
		339	flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
		340	(int)req->result
		341	);
		342
		343	if (do_open (gv, sym, symlen, 0, 0, 0, 0))
		344	fh = (SV *)gv;
		345	}
		346
515	XPUSHs (fh);	347	PUSHs (fh);
516	}	348	}
517	break;	349	break;
518		350
519	case REQ_GROUP:	351	case EIO_GROUP:
520	req->int1 = 2; /* mark group as finished */	352	req->int1 = 2; /* mark group as finished */
521		353
522	if (req->sv1)	354	if (req->sv1)
523	{	355	{
524	int i;	356	int i;
…		…
528	for (i = 0; i <= AvFILL (av); ++i)	360	for (i = 0; i <= AvFILL (av); ++i)
529	PUSHs (*av_fetch (av, i, 0));	361	PUSHs (*av_fetch (av, i, 0));
530	}	362	}
531	break;	363	break;
532		364
533	case REQ_NOP:	365	case EIO_NOP:
534	case REQ_BUSY:	366	case EIO_BUSY:
535	break;	367	break;
536		368
537	case REQ_READLINK:	369	case EIO_READLINK:
538	if (req->result > 0)	370	if (req->result > 0)
539	{	371	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;	372	break;
545		373
546	case REQ_STAT:	374	case EIO_STAT:
547	case REQ_LSTAT:	375	case EIO_LSTAT:
548	case REQ_FSTAT:	376	case EIO_FSTAT:
549	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;	377	PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT;
550	PL_laststatval = req->result;	378	PL_laststatval = req->result;
551	PL_statcache = *(Stat_t *)(req->ptr2);	379	PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
552	PUSHs (sv_2mortal (newSViv (req->result)));	380	PUSHs (sv_2mortal (newSViv (req->result)));
553	break;	381	break;
554		382
555	case REQ_READ:	383	case EIO_READ:
		384	{
556	SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));	385	SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
557	*SvEND (req->sv2) = 0;	386	*SvEND (req->sv2) = 0;
		387	SvPOK_only (req->sv2);
		388	SvSETMAGIC (req->sv2);
558	PUSHs (sv_2mortal (newSViv (req->result)));	389	PUSHs (sv_2mortal (newSViv (req->result)));
		390	}
559	break;	391	break;
		392
		393	case EIO_DUP2:
		394	if (req->result > 0)
		395	req->result = 0;
		396	/* FALLTHROUGH */
560		397
561	default:	398	default:
562	PUSHs (sv_2mortal (newSViv (req->result)));	399	PUSHs (sv_2mortal (newSViv (req->result)));
563	break;	400	break;
564	}	401	}
565		402
566	errno = req->errorno;	403	errno = req->errorno;
567		404
568	PUTBACK;	405	PUTBACK;
569	call_sv (req->callback, G_VOID | G_EVAL);	406	call_sv (req->callback, G_VOID | G_EVAL | G_DISCARD);
570	SPAGAIN;	407	SPAGAIN;
571		408
572	FREETMPS;	409	FREETMPS;
573	LEAVE;	410	LEAVE;
		411
		412	PUTBACK;
574	}	413	}
575		414
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);	415	return !!SvTRUE (ERRSV);
591	}	416	}
592		417
593	static void req_destroy (aio_req req)	418	static void req_destroy (aio_req req)
594	{	419	{
595	if (req->self)	420	if (req->self)
…		…
600		425
601	SvREFCNT_dec (req->sv1);	426	SvREFCNT_dec (req->sv1);
602	SvREFCNT_dec (req->sv2);	427	SvREFCNT_dec (req->sv2);
603	SvREFCNT_dec (req->callback);	428	SvREFCNT_dec (req->callback);
604		429
605	if (req->flags & FLAG_PTR2_FREE)
606	free (req->ptr2);
607
608	Safefree (req);	430	Safefree (req);
609	}	431	}
610		432
611	static void req_cancel_subs (aio_req grp)	433	static void req_cancel_subs (aio_req grp)
612	{	434	{
613	aio_req sub;	435	aio_req sub;
614		436
615	if (grp->type != REQ_GROUP)	437	if (grp->type != EIO_GROUP)
616	return;	438	return;
617		439
618	SvREFCNT_dec (grp->sv2);	440	SvREFCNT_dec (grp->sv2);
619	grp->sv2 = 0;	441	grp->sv2 = 0;
620		442
621	for (sub = grp->grp_first; sub; sub = sub->grp_next)	443	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	}	444	}
631		445
632	#ifdef USE_SOCKETS_AS_HANDLES	446	#ifdef USE_SOCKETS_AS_HANDLES
633	# define TO_SOCKET(x) (win32_get_osfhandle (x))	447	# define TO_SOCKET(x) (win32_get_osfhandle (x))
634	#else	448	#else
635	# define TO_SOCKET(x) (x)	449	# define TO_SOCKET(x) (x)
636	#endif	450	#endif
637		451
638	static void	452	static void
639	create_pipe (int fd[2])	453	create_respipe (void)
640	{	454	{
		455	int old_readfd = respipe [0];
		456
		457	if (respipe [1] >= 0)
		458	respipe_close (TO_SOCKET (respipe [1]));
		459
		460	#ifdef _WIN32
		461	if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, respipe))
		462	#else
		463	if (pipe (respipe))
		464	#endif
		465	croak ("unable to initialize result pipe");
		466
		467	if (old_readfd >= 0)
		468	{
		469	if (dup2 (TO_SOCKET (respipe [0]), TO_SOCKET (old_readfd)) < 0)
		470	croak ("unable to initialize result pipe(2)");
		471
		472	respipe_close (respipe [0]);
		473	respipe [0] = old_readfd;
		474	}
		475
641	#ifdef _WIN32	476	#ifdef _WIN32
642	int arg = 1;	477	int arg = 1;
643	if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, fd)
644	|| ioctlsocket (TO_SOCKET (fd [0]), FIONBIO, &arg)	478	if (ioctlsocket (TO_SOCKET (respipe [0]), FIONBIO, &arg)
645	|| ioctlsocket (TO_SOCKET (fd [1]), FIONBIO, &arg))	479	|| ioctlsocket (TO_SOCKET (respipe [1]), FIONBIO, &arg))
646	#else	480	#else
647	if (pipe (fd)
648	|| fcntl (fd [0], F_SETFL, O_NONBLOCK)	481	if (fcntl (respipe [0], F_SETFL, O_NONBLOCK)
649	|| fcntl (fd [1], F_SETFL, O_NONBLOCK))	482	|| fcntl (respipe [1], F_SETFL, O_NONBLOCK))
650	#endif	483	#endif
651	croak ("unable to initialize result pipe");	484	croak ("unable to initialize result pipe(3)");
652		485
653	respipe_osf [0] = TO_SOCKET (respipe [0]);	486	respipe_osf [0] = TO_SOCKET (respipe [0]);
654	respipe_osf [1] = TO_SOCKET (respipe [1]);	487	respipe_osf [1] = TO_SOCKET (respipe [1]);
655	}	488	}
656		489
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 ()	490	static void poll_wait (void)
753	{	491	{
754	fd_set rfd;	492	fd_set rfd;
755		493
756	while (nreqs)	494	while (eio_nreqs ())
757	{	495	{
758	int size;	496	int size;
759	if (WORDACCESS_UNSAFE) X_LOCK (reslock);	497
		498	X_LOCK (reslock);
760	size = res_queue.size;	499	size = res_queue.size;
761	if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);	500	X_UNLOCK (reslock);
762		501
763	if (size)	502	if (size)
764	return;	503	return;
765		504
766	maybe_start_thread ();	505	etp_maybe_start_thread ();
767		506
768	FD_ZERO (&rfd);	507	FD_ZERO (&rfd);
769	FD_SET (respipe [0], &rfd);	508	FD_SET (respipe [0], &rfd);
770		509
771	PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);	510	PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);
772	}	511	}
773	}	512	}
774		513
775	static int poll_cb ()	514	static int poll_cb (void)
776	{	515	{
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 (;;)	516	for (;;)
790	{	517	{
791	for (;;)	518	int res = eio_poll ();
792	{
793	maybe_start_thread ();
794		519
795	X_LOCK (reslock);	520	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);	521	croak (0);
830	}
831		522
832	count++;	523	if (!max_outstanding || max_outstanding > eio_nreqs ())
833	}	524	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		525
852	poll_wait ();	526	poll_wait ();
853
854	++maxreqs;
855	}	527	}
856
857	unblock_sig ();
858	return count;
859	}
860
861	/*****************************************************************************/
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	{
877	ssize_t res;
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	}
889
890	static ssize_t pwrite (int fd, void *buf, size_t count, off_t offset)
891	{
892	ssize_t res;
893	off_t ooffset;
894
895	X_LOCK (preadwritelock);
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;
998
999	#if HAVE_SENDFILE
1000	# if __linux
1001	res = sendfile (ofd, ifd, &offset, count);
1002
1003	# elif __freebsd
1004	/*
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
1012	if (res < 0 && sbytes)
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	}	528	}
1347		529
1348	static void atfork_child (void)	530	static void atfork_child (void)
1349	{	531	{
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);	532	create_respipe ();
		533	}
1379		534
1380	atfork_parent ();	535	static SV *
		536	get_cb (SV *cb_sv)
		537	{
		538	SvGETMAGIC (cb_sv);
		539	return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0;
1381	}	540	}
1382		541
1383	#define dREQ \	542	#define dREQ \
		543	SV *cb_cv; \
1384	aio_req req; \	544	aio_req req; \
1385	int req_pri = next_pri; \	545	int req_pri = next_pri; \
1386	next_pri = DEFAULT_PRI + PRI_BIAS; \	546	next_pri = EIO_PRI_DEFAULT; \
1387	\	547	\
1388	if (SvOK (callback) && !SvROK (callback)) \	548	cb_cv = get_cb (callback); \
1389	croak ("callback must be undef or of reference type"); \
1390	\	549	\
1391	Newz (0, req, 1, aio_cb); \	550	Newz (0, req, 1, eio_req); \
1392	if (!req) \	551	if (!req) \
1393	croak ("out of memory during aio_req allocation"); \	552	croak ("out of memory during eio_req allocation"); \
1394	\	553	\
1395	req->callback = newSVsv (callback); \	554	req->callback = SvREFCNT_inc (cb_cv); \
1396	req->pri = req_pri	555	req->pri = req_pri
1397		556
1398	#define REQ_SEND \	557	#define REQ_SEND \
		558	PUTBACK; \
1399	req_send (req); \	559	req_submit (req); \
		560	SPAGAIN; \
1400	\	561	\
1401	if (GIMME_V != G_VOID) \	562	if (GIMME_V != G_VOID) \
1402	XPUSHs (req_sv (req, AIO_REQ_KLASS));	563	XPUSHs (req_sv (req, AIO_REQ_KLASS));
1403		564
1404	MODULE = IO::AIO PACKAGE = IO::AIO	565	MODULE = IO::AIO PACKAGE = IO::AIO
1405		566
1406	PROTOTYPES: ENABLE	567	PROTOTYPES: ENABLE
1407		568
1408	BOOT:	569	BOOT:
1409	{	570	{
1410	HV *stash = gv_stashpv ("IO::AIO", 1);	571	static const struct {
1411		572	const char *name;
1412	newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));	573	IV iv;
1413	newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));	574	} *civ, const_iv[] = {
1414	newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));	575	# define const_iv(name, value) { # name, (IV) value },
1415	newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));	576	# define const_eio(name) { # name, (IV) EIO_ ## name },
1416	newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));	577	const_iv (EXDEV , EXDEV)
		578	const_iv (ENOSYS , ENOSYS)
		579	const_iv (O_RDONLY, O_RDONLY)
		580	const_iv (O_WRONLY, O_WRONLY)
		581	const_iv (O_CREAT , O_CREAT)
		582	const_iv (O_TRUNC , O_TRUNC)
1417	#ifdef _WIN32	583	#ifndef _WIN32
1418	X_MUTEX_CHECK (wrklock);	584	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	585	#endif
		586	const_iv (FADV_NORMAL , POSIX_FADV_NORMAL)
		587	const_iv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
		588	const_iv (FADV_RANDOM , POSIX_FADV_RANDOM)
		589	const_iv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
		590	const_iv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
		591	const_iv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
1430		592
1431	create_pipe (respipe);	593	const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
		594	const_eio (SYNC_FILE_RANGE_WRITE)
		595	const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
1432		596
1433	X_THREAD_ATFORK (atfork_prepare, atfork_parent, atfork_child);	597	const_eio (READDIR_DENTS)
		598	const_eio (READDIR_DIRS_FIRST)
		599	const_eio (READDIR_STAT_ORDER)
		600	const_eio (READDIR_FOUND_UNKNOWN)
		601
		602	const_eio (DT_UNKNOWN)
		603	const_eio (DT_FIFO)
		604	const_eio (DT_CHR)
		605	const_eio (DT_DIR)
		606	const_eio (DT_BLK)
		607	const_eio (DT_REG)
		608	const_eio (DT_LNK)
		609	const_eio (DT_SOCK)
		610	const_eio (DT_WHT)
		611	};
		612
		613	stash = gv_stashpv ("IO::AIO", 1);
		614
		615	for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
		616	newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
		617
		618	create_respipe ();
		619
		620	if (eio_init (want_poll, done_poll) < 0)
		621	croak ("IO::AIO: unable to initialise eio library");
		622
		623	/* atfork child called in fifo order, so before eio's handler */
		624	X_THREAD_ATFORK (0, 0, atfork_child);
1434	}	625	}
1435		626
1436	void	627	void
1437	max_poll_reqs (int nreqs)	628	max_poll_reqs (int nreqs)
1438	PROTOTYPE: $	629	PROTOTYPE: $
1439	CODE:	630	CODE:
1440	max_poll_reqs = nreqs;	631	eio_set_max_poll_reqs (nreqs);
1441		632
1442	void	633	void
1443	max_poll_time (double nseconds)	634	max_poll_time (double nseconds)
1444	PROTOTYPE: $	635	PROTOTYPE: $
1445	CODE:	636	CODE:
1446	max_poll_time = nseconds * AIO_TICKS;	637	eio_set_max_poll_time (nseconds);
1447		638
1448	void	639	void
1449	min_parallel (int nthreads)	640	min_parallel (int nthreads)
1450	PROTOTYPE: $	641	PROTOTYPE: $
		642	CODE:
		643	eio_set_min_parallel (nthreads);
1451		644
1452	void	645	void
1453	max_parallel (int nthreads)	646	max_parallel (int nthreads)
1454	PROTOTYPE: $	647	PROTOTYPE: $
		648	CODE:
		649	eio_set_max_parallel (nthreads);
1455		650
1456	void	651	void
1457	max_idle (int nthreads)	652	max_idle (int nthreads)
1458	PROTOTYPE: $	653	PROTOTYPE: $
1459	CODE:	654	CODE:
1460	set_max_idle (nthreads);	655	eio_set_max_idle (nthreads);
1461		656
1462	int	657	void
1463	max_outstanding (int maxreqs)	658	max_outstanding (int maxreqs)
1464	PROTOTYPE: $	659	PROTOTYPE: $
1465	CODE:	660	CODE:
1466	RETVAL = max_outstanding;
1467	max_outstanding = maxreqs;	661	max_outstanding = maxreqs;
1468	OUTPUT:
1469	RETVAL
1470		662
1471	void	663	void
1472	aio_open (SV8 *pathname, int flags, int mode, SV *callback=&PL_sv_undef)	664	aio_open (SV8 *pathname, int flags, int mode, SV *callback=&PL_sv_undef)
1473	PROTOTYPE: $$$;$	665	PROTOTYPE: $$$;$
1474	PPCODE:	666	PPCODE:
1475	{	667	{
1476	dREQ;	668	dREQ;
1477		669
1478	req->type = REQ_OPEN;	670	req->type = EIO_OPEN;
1479	req->sv1 = newSVsv (pathname);	671	req->sv1 = newSVsv (pathname);
1480	req->ptr1 = SvPVbyte_nolen (req->sv1);	672	req->ptr1 = SvPVbyte_nolen (req->sv1);
1481	req->int1 = flags;	673	req->int1 = flags;
1482	req->mode = mode;	674	req->int2 = mode;
1483		675
1484	REQ_SEND;	676	REQ_SEND;
		677	}
		678
		679	void
		680	aio_fsync (SV *fh, SV *callback=&PL_sv_undef)
		681	PROTOTYPE: $;$
		682	ALIAS:
		683	aio_fsync = EIO_FSYNC
		684	aio_fdatasync = EIO_FDATASYNC
		685	PPCODE:
		686	{
		687	int fd = s_fileno_croak (fh, 0);
		688	dREQ;
		689
		690	req->type = ix;
		691	req->sv1 = newSVsv (fh);
		692	req->int1 = fd;
		693
		694	REQ_SEND (req);
		695	}
		696
		697	void
		698	aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef)
		699	PROTOTYPE: $$$$;$
		700	PPCODE:
		701	{
		702	int fd = s_fileno_croak (fh, 0);
		703	dREQ;
		704
		705	req->type = EIO_SYNC_FILE_RANGE;
		706	req->sv1 = newSVsv (fh);
		707	req->int1 = fd;
		708	req->offs = offset;
		709	req->size = nbytes;
		710	req->int2 = flags;
		711
		712	REQ_SEND (req);
1485	}	713	}
1486		714
1487	void	715	void
1488	aio_close (SV *fh, SV *callback=&PL_sv_undef)	716	aio_close (SV *fh, SV *callback=&PL_sv_undef)
1489	PROTOTYPE: $;$	717	PROTOTYPE: $;$
1490	ALIAS:
1491	aio_close = REQ_CLOSE
1492	aio_fsync = REQ_FSYNC
1493	aio_fdatasync = REQ_FDATASYNC
1494	PPCODE:	718	PPCODE:
1495	{	719	{
		720	static int close_pipe = -1; /* dummy fd to close fds via dup2 */
		721	int fd = s_fileno_croak (fh, 0);
1496	dREQ;	722	dREQ;
1497		723
		724	if (close_pipe < 0)
		725	{
		726	int pipefd [2];
		727
		728	if (pipe (pipefd) < 0
		729	|| close (pipefd [1]) < 0
		730	|| fcntl (pipefd [0], F_SETFD, FD_CLOEXEC) < 0)
		731	abort (); /*D*/
		732
		733	close_pipe = pipefd [0];
		734	}
		735
1498	req->type = ix;	736	req->type = EIO_DUP2;
		737	req->int1 = close_pipe;
1499	req->sv1 = newSVsv (fh);	738	req->sv2 = newSVsv (fh);
1500	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));	739	req->int2 = fd;
1501		740
1502	REQ_SEND (req);	741	REQ_SEND (req);
1503	}	742	}
1504		743
1505	void	744	void
1506	aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback=&PL_sv_undef)	745	aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback=&PL_sv_undef)
1507	ALIAS:	746	ALIAS:
1508	aio_read = REQ_READ	747	aio_read = EIO_READ
1509	aio_write = REQ_WRITE	748	aio_write = EIO_WRITE
1510	PROTOTYPE: $$$$$;$	749	PROTOTYPE: $$$$$;$
1511	PPCODE:	750	PPCODE:
1512	{	751	{
1513	STRLEN svlen;	752	STRLEN svlen;
		753	int fd = s_fileno_croak (fh, ix == EIO_WRITE);
1514	char *svptr = SvPVbyte (data, svlen);	754	char *svptr = SvPVbyte (data, svlen);
1515	UV len = SvUV (length);	755	UV len = SvUV (length);
1516		756
1517	SvUPGRADE (data, SVt_PV);
1518	SvPOK_on (data);
1519
1520	if (dataoffset < 0)	757	if (dataoffset < 0)
1521	dataoffset += svlen;	758	dataoffset += svlen;
1522		759
1523	if (dataoffset < 0 || dataoffset > svlen)	760	if (dataoffset < 0 || dataoffset > svlen)
1524	croak ("dataoffset outside of data scalar");	761	croak ("dataoffset outside of data scalar");
1525		762
1526	if (ix == REQ_WRITE)	763	if (ix == EIO_WRITE)
1527	{	764	{
1528	/* write: check length and adjust. */	765	/* write: check length and adjust. */
1529	if (!SvOK (length) || len + dataoffset > svlen)	766	if (!SvOK (length) || len + dataoffset > svlen)
1530	len = svlen - dataoffset;	767	len = svlen - dataoffset;
1531	}	768	}
1532	else	769	else
1533	{	770	{
1534	/* read: grow scalar as necessary */	771	/* read: check type and grow scalar as necessary */
		772	SvUPGRADE (data, SVt_PV);
1535	svptr = SvGROW (data, len + dataoffset + 1);	773	svptr = SvGROW (data, len + dataoffset + 1);
1536	}	774	}
1537
1538	if (len < 0)
1539	croak ("length must not be negative");
1540		775
1541	{	776	{
1542	dREQ;	777	dREQ;
1543		778
1544	req->type = ix;	779	req->type = ix;
1545	req->sv1 = newSVsv (fh);	780	req->sv1 = newSVsv (fh);
1546	req->int1 = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))	781	req->int1 = fd;
1547	: IoOFP (sv_2io (fh)));
1548	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;	782	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
1549	req->size = len;	783	req->size = len;
1550	req->sv2 = SvREFCNT_inc (data);	784	req->sv2 = SvREFCNT_inc (data);
1551	req->ptr1 = (char *)svptr + dataoffset;	785	req->ptr2 = (char *)svptr + dataoffset;
1552	req->stroffset = dataoffset;	786	req->stroffset = dataoffset;
1553		787
1554	if (!SvREADONLY (data))	788	if (!SvREADONLY (data))
1555	{	789	{
1556	SvREADONLY_on (data);	790	SvREADONLY_on (data);
…		…
1567	PPCODE:	801	PPCODE:
1568	{	802	{
1569	SV *data;	803	SV *data;
1570	dREQ;	804	dREQ;
1571		805
1572	data = newSV (NAME_MAX);
1573	SvPOK_on (data);
1574
1575	req->type = REQ_READLINK;	806	req->type = EIO_READLINK;
1576	req->sv1 = newSVsv (path);	807	req->sv1 = newSVsv (path);
1577	req->ptr2 = SvPVbyte_nolen (req->sv1);	808	req->ptr1 = SvPVbyte_nolen (req->sv1);
1578	req->sv2 = data;
1579	req->ptr1 = SvPVbyte_nolen (data);
1580		809
1581	REQ_SEND;	810	REQ_SEND;
1582	}	811	}
1583		812
1584	void	813	void
1585	aio_sendfile (SV *out_fh, SV *in_fh, SV *in_offset, UV length, SV *callback=&PL_sv_undef)	814	aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef)
1586	PROTOTYPE: $$$$;$	815	PROTOTYPE: $$$$;$
1587	PPCODE:	816	PPCODE:
1588	{	817	{
		818	int ifd = s_fileno_croak (in_fh , 0);
		819	int ofd = s_fileno_croak (out_fh, 1);
1589	dREQ;	820	dREQ;
1590		821
1591	req->type = REQ_SENDFILE;	822	req->type = EIO_SENDFILE;
1592	req->sv1 = newSVsv (out_fh);	823	req->sv1 = newSVsv (out_fh);
1593	req->int1 = PerlIO_fileno (IoIFP (sv_2io (out_fh)));	824	req->int1 = ofd;
1594	req->sv2 = newSVsv (in_fh);	825	req->sv2 = newSVsv (in_fh);
1595	req->int2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));	826	req->int2 = ifd;
1596	req->offs = SvVAL64 (in_offset);	827	req->offs = in_offset;
1597	req->size = length;	828	req->size = length;
1598		829
1599	REQ_SEND;	830	REQ_SEND;
1600	}	831	}
1601		832
1602	void	833	void
1603	aio_readahead (SV *fh, SV *offset, IV length, SV *callback=&PL_sv_undef)	834	aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef)
1604	PROTOTYPE: $$$;$	835	PROTOTYPE: $$$;$
1605	PPCODE:	836	PPCODE:
1606	{	837	{
		838	int fd = s_fileno_croak (fh, 0);
1607	dREQ;	839	dREQ;
1608		840
1609	req->type = REQ_READAHEAD;	841	req->type = EIO_READAHEAD;
1610	req->sv1 = newSVsv (fh);	842	req->sv1 = newSVsv (fh);
1611	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));	843	req->int1 = fd;
1612	req->offs = SvVAL64 (offset);	844	req->offs = offset;
1613	req->size = length;	845	req->size = length;
1614		846
1615	REQ_SEND;	847	REQ_SEND;
1616	}	848	}
1617		849
1618	void	850	void
1619	aio_stat (SV8 *fh_or_path, SV *callback=&PL_sv_undef)	851	aio_stat (SV8 *fh_or_path, SV *callback=&PL_sv_undef)
1620	ALIAS:	852	ALIAS:
1621	aio_stat = REQ_STAT	853	aio_stat = EIO_STAT
1622	aio_lstat = REQ_LSTAT	854	aio_lstat = EIO_LSTAT
1623	PPCODE:	855	PPCODE:
1624	{	856	{
1625	dREQ;	857	dREQ;
1626		858
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);	859	req->sv1 = newSVsv (fh_or_path);
1636		860
1637	if (SvPOK (fh_or_path))	861	if (SvPOK (req->sv1))
1638	{	862	{
1639	req->type = ix;	863	req->type = ix;
1640	req->ptr1 = SvPVbyte_nolen (req->sv1);	864	req->ptr1 = SvPVbyte_nolen (req->sv1);
1641	}	865	}
1642	else	866	else
1643	{	867	{
1644	req->type = REQ_FSTAT;	868	req->type = EIO_FSTAT;
1645	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	869	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1646	}	870	}
1647		871
1648	REQ_SEND;	872	REQ_SEND;
1649	}	873	}
…		…
1656		880
1657	req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;	881	req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
1658	req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;	882	req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
1659	req->sv1 = newSVsv (fh_or_path);	883	req->sv1 = newSVsv (fh_or_path);
1660		884
1661	if (SvPOK (fh_or_path))	885	if (SvPOK (req->sv1))
1662	{	886	{
1663	req->type = REQ_UTIME;	887	req->type = EIO_UTIME;
1664	req->ptr1 = SvPVbyte_nolen (req->sv1);	888	req->ptr1 = SvPVbyte_nolen (req->sv1);
1665	}	889	}
1666	else	890	else
1667	{	891	{
1668	req->type = REQ_FUTIME;	892	req->type = EIO_FUTIME;
1669	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	893	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1670	}	894	}
1671		895
1672	REQ_SEND;	896	REQ_SEND;
1673	}	897	}
…		…
1679	dREQ;	903	dREQ;
1680		904
1681	req->sv1 = newSVsv (fh_or_path);	905	req->sv1 = newSVsv (fh_or_path);
1682	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;	906	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
1683		907
1684	if (SvPOK (fh_or_path))	908	if (SvPOK (req->sv1))
1685	{	909	{
1686	req->type = REQ_TRUNCATE;	910	req->type = EIO_TRUNCATE;
1687	req->ptr1 = SvPVbyte_nolen (req->sv1);	911	req->ptr1 = SvPVbyte_nolen (req->sv1);
1688	}	912	}
1689	else	913	else
1690	{	914	{
1691	req->type = REQ_FTRUNCATE;	915	req->type = EIO_FTRUNCATE;
1692	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	916	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1693	}	917	}
1694		918
1695	REQ_SEND;	919	REQ_SEND;
1696	}	920	}
1697		921
1698	void	922	void
1699	aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)	923	aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
		924	ALIAS:
		925	aio_chmod = EIO_CHMOD
		926	aio_mkdir = EIO_MKDIR
1700	PPCODE:	927	PPCODE:
1701	{	928	{
1702	dREQ;	929	dREQ;
1703		930
1704	req->mode = mode;	931	req->int2 = mode;
1705	req->sv1 = newSVsv (fh_or_path);	932	req->sv1 = newSVsv (fh_or_path);
1706		933
1707	if (SvPOK (fh_or_path))	934	if (SvPOK (req->sv1))
1708	{	935	{
1709	req->type = REQ_CHMOD;	936	req->type = ix;
1710	req->ptr1 = SvPVbyte_nolen (req->sv1);	937	req->ptr1 = SvPVbyte_nolen (req->sv1);
1711	}	938	}
1712	else	939	else
1713	{	940	{
1714	req->type = REQ_FCHMOD;	941	req->type = EIO_FCHMOD;
1715	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	942	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1716	}	943	}
1717		944
1718	REQ_SEND;	945	REQ_SEND;
1719	}	946	}
…		…
1726		953
1727	req->int2 = SvOK (uid) ? SvIV (uid) : -1;	954	req->int2 = SvOK (uid) ? SvIV (uid) : -1;
1728	req->int3 = SvOK (gid) ? SvIV (gid) : -1;	955	req->int3 = SvOK (gid) ? SvIV (gid) : -1;
1729	req->sv1 = newSVsv (fh_or_path);	956	req->sv1 = newSVsv (fh_or_path);
1730		957
1731	if (SvPOK (fh_or_path))	958	if (SvPOK (req->sv1))
1732	{	959	{
1733	req->type = REQ_CHOWN;	960	req->type = EIO_CHOWN;
1734	req->ptr1 = SvPVbyte_nolen (req->sv1);	961	req->ptr1 = SvPVbyte_nolen (req->sv1);
1735	}	962	}
1736	else	963	else
1737	{	964	{
1738	req->type = REQ_FCHOWN;	965	req->type = EIO_FCHOWN;
1739	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	966	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1740	}	967	}
1741		968
1742	REQ_SEND;	969	REQ_SEND;
		970	}
		971
		972	void
		973	aio_readdirx (SV8 *pathname, IV flags, SV *callback=&PL_sv_undef)
		974	PPCODE:
		975	{
		976	dREQ;
		977
		978	req->type = EIO_READDIR;
		979	req->sv1 = newSVsv (pathname);
		980	req->ptr1 = SvPVbyte_nolen (req->sv1);
		981	req->int1 = flags | EIO_READDIR_DENTS | EIO_READDIR_CUSTOM1;
		982
		983	if (flags & EIO_READDIR_DENTS)
		984	req->int1 |= EIO_READDIR_CUSTOM2;
		985
		986	REQ_SEND;
1743	}	987	}
1744		988
1745	void	989	void
1746	aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)	990	aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)
1747	ALIAS:	991	ALIAS:
1748	aio_unlink = REQ_UNLINK	992	aio_unlink = EIO_UNLINK
1749	aio_rmdir = REQ_RMDIR	993	aio_rmdir = EIO_RMDIR
1750	aio_readdir = REQ_READDIR	994	aio_readdir = EIO_READDIR
1751	PPCODE:	995	PPCODE:
1752	{	996	{
1753	dREQ;	997	dREQ;
1754		998
1755	req->type = ix;	999	req->type = ix;
…		…
1758		1002
1759	REQ_SEND;	1003	REQ_SEND;
1760	}	1004	}
1761		1005
1762	void	1006	void
1763	aio_mkdir (SV8 *pathname, int mode, SV *callback=&PL_sv_undef)	1007	aio_link (SV8 *oldpath, SV8 *newpath, SV *callback=&PL_sv_undef)
		1008	ALIAS:
		1009	aio_link = EIO_LINK
		1010	aio_symlink = EIO_SYMLINK
		1011	aio_rename = EIO_RENAME
1764	PPCODE:	1012	PPCODE:
1765	{	1013	{
1766	dREQ;	1014	dREQ;
1767		1015
		1016	req->type = ix;
		1017	req->sv1 = newSVsv (oldpath);
		1018	req->ptr1 = SvPVbyte_nolen (req->sv1);
		1019	req->sv2 = newSVsv (newpath);
		1020	req->ptr2 = SvPVbyte_nolen (req->sv2);
		1021
		1022	REQ_SEND;
		1023	}
		1024
		1025	void
		1026	aio_mknod (SV8 *pathname, int mode, UV dev, SV *callback=&PL_sv_undef)
		1027	PPCODE:
		1028	{
		1029	dREQ;
		1030
1768	req->type = REQ_MKDIR;	1031	req->type = EIO_MKNOD;
1769	req->sv1 = newSVsv (pathname);	1032	req->sv1 = newSVsv (pathname);
1770	req->ptr1 = SvPVbyte_nolen (req->sv1);	1033	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;	1034	req->int2 = (mode_t)mode;
1805	req->offs = dev;	1035	req->offs = dev;
1806		1036
1807	REQ_SEND;	1037	REQ_SEND;
1808	}	1038	}
1809		1039
…		…
1811	aio_busy (double delay, SV *callback=&PL_sv_undef)	1041	aio_busy (double delay, SV *callback=&PL_sv_undef)
1812	PPCODE:	1042	PPCODE:
1813	{	1043	{
1814	dREQ;	1044	dREQ;
1815		1045
1816	req->type = REQ_BUSY;	1046	req->type = EIO_BUSY;
1817	req->nv1 = delay < 0. ? 0. : delay;	1047	req->nv1 = delay < 0. ? 0. : delay;
1818		1048
1819	REQ_SEND;	1049	REQ_SEND;
1820	}	1050	}
1821		1051
…		…
1824	PROTOTYPE: ;$	1054	PROTOTYPE: ;$
1825	PPCODE:	1055	PPCODE:
1826	{	1056	{
1827	dREQ;	1057	dREQ;
1828		1058
1829	req->type = REQ_GROUP;	1059	req->type = EIO_GROUP;
1830		1060
1831	req_send (req);	1061	req_submit (req);
1832	XPUSHs (req_sv (req, AIO_GRP_KLASS));	1062	XPUSHs (req_sv (req, AIO_GRP_KLASS));
1833	}	1063	}
1834		1064
1835	void	1065	void
1836	aio_nop (SV *callback=&PL_sv_undef)	1066	aio_nop (SV *callback=&PL_sv_undef)
		1067	ALIAS:
		1068	aio_nop = EIO_NOP
		1069	aio_sync = EIO_SYNC
1837	PPCODE:	1070	PPCODE:
1838	{	1071	{
1839	dREQ;	1072	dREQ;
1840		1073
1841	req->type = REQ_NOP;	1074	req->type = ix;
1842		1075
1843	REQ_SEND;	1076	REQ_SEND;
1844	}	1077	}
1845		1078
1846	int	1079	int
1847	aioreq_pri (int pri = 0)	1080	aioreq_pri (int pri = 0)
1848	PROTOTYPE: ;$	1081	PROTOTYPE: ;$
1849	CODE:	1082	CODE:
1850	RETVAL = next_pri - PRI_BIAS;	1083	RETVAL = next_pri;
1851	if (items > 0)	1084	if (items > 0)
1852	{	1085	{
1853	if (pri < PRI_MIN) pri = PRI_MIN;	1086	if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
1854	if (pri > PRI_MAX) pri = PRI_MAX;	1087	if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
1855	next_pri = pri + PRI_BIAS;	1088	next_pri = pri;
1856	}	1089	}
1857	OUTPUT:	1090	OUTPUT:
1858	RETVAL	1091	RETVAL
1859		1092
1860	void	1093	void
1861	aioreq_nice (int nice = 0)	1094	aioreq_nice (int nice = 0)
1862	CODE:	1095	CODE:
1863	nice = next_pri - nice;	1096	nice = next_pri - nice;
1864	if (nice < PRI_MIN) nice = PRI_MIN;	1097	if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
1865	if (nice > PRI_MAX) nice = PRI_MAX;	1098	if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
1866	next_pri = nice + PRI_BIAS;	1099	next_pri = nice;
1867		1100
1868	void	1101	void
1869	flush ()	1102	flush ()
1870	PROTOTYPE:	1103	PROTOTYPE:
1871	CODE:	1104	CODE:
1872	while (nreqs)	1105	while (eio_nreqs ())
1873	{	1106	{
1874	poll_wait ();	1107	poll_wait ();
1875	poll_cb ();	1108	poll_cb ();
1876	}	1109	}
1877		1110
…		…
1904	poll_wait()	1137	poll_wait()
1905	PROTOTYPE:	1138	PROTOTYPE:
1906	CODE:	1139	CODE:
1907	poll_wait ();	1140	poll_wait ();
1908		1141
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	1142	int
1947	nreqs()	1143	nreqs()
1948	PROTOTYPE:	1144	PROTOTYPE:
1949	CODE:	1145	CODE:
1950	RETVAL = nreqs;	1146	RETVAL = eio_nreqs ();
1951	OUTPUT:	1147	OUTPUT:
1952	RETVAL	1148	RETVAL
1953		1149
1954	int	1150	int
1955	nready()	1151	nready()
1956	PROTOTYPE:	1152	PROTOTYPE:
1957	CODE:	1153	CODE:
1958	RETVAL = get_nready ();	1154	RETVAL = eio_nready ();
1959	OUTPUT:	1155	OUTPUT:
1960	RETVAL	1156	RETVAL
1961		1157
1962	int	1158	int
1963	npending()	1159	npending()
1964	PROTOTYPE:	1160	PROTOTYPE:
1965	CODE:	1161	CODE:
1966	RETVAL = get_npending ();	1162	RETVAL = eio_npending ();
1967	OUTPUT:	1163	OUTPUT:
1968	RETVAL	1164	RETVAL
1969		1165
1970	int	1166	int
1971	nthreads()	1167	nthreads()
1972	PROTOTYPE:	1168	PROTOTYPE:
1973	CODE:	1169	CODE:
1974	if (WORDACCESS_UNSAFE) X_LOCK (wrklock);	1170	RETVAL = eio_nthreads ();
1975	RETVAL = started;
1976	if (WORDACCESS_UNSAFE) X_UNLOCK (wrklock);
1977	OUTPUT:	1171	OUTPUT:
1978	RETVAL	1172	RETVAL
1979		1173
		1174	int
		1175	fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
		1176	PROTOTYPE: $$$$
		1177	CODE:
		1178	#if _XOPEN_SOURCE >= 600 && !NO_FADVISE
		1179	RETVAL = posix_fadvise (fh, offset, length, advice);
		1180	#else
		1181	RETVAL = errno = ENOSYS;
		1182	#endif
		1183	OUTPUT:
		1184	RETVAL
		1185
		1186	ssize_t
		1187	sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
		1188	PROTOTYPE: $$$$
		1189	CODE:
		1190	eio_sendfile_sync (ofh, ifh, offset, count);
		1191
		1192	void _on_next_submit (SV *cb)
		1193	CODE:
		1194	SvREFCNT_dec (on_next_submit);
		1195	on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
		1196
1980	PROTOTYPES: DISABLE	1197	PROTOTYPES: DISABLE
1981		1198
1982	MODULE = IO::AIO PACKAGE = IO::AIO::REQ	1199	MODULE = IO::AIO PACKAGE = IO::AIO::REQ
1983		1200
1984	void	1201	void
1985	cancel (aio_req_ornot req)	1202	cancel (aio_req_ornot req)
1986	CODE:	1203	CODE:
1987	req_cancel (req);	1204	eio_cancel (req);
1988		1205
1989	void	1206	void
1990	cb (aio_req_ornot req, SV *callback=&PL_sv_undef)	1207	cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
1991	CODE:	1208	PPCODE:
		1209	{
		1210	if (GIMME_V != G_VOID)
		1211	XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
		1212
		1213	if (items > 1)
		1214	{
		1215	SV *cb_cv =get_cb (callback);
		1216
1992	SvREFCNT_dec (req->callback);	1217	SvREFCNT_dec (req->callback);
1993	req->callback = newSVsv (callback);	1218	req->callback = SvREFCNT_inc (cb_cv);
		1219	}
		1220	}
1994		1221
1995	MODULE = IO::AIO PACKAGE = IO::AIO::GRP	1222	MODULE = IO::AIO PACKAGE = IO::AIO::GRP
1996		1223
1997	void	1224	void
1998	add (aio_req grp, ...)	1225	add (aio_req grp, ...)
1999	PPCODE:	1226	PPCODE:
2000	{	1227	{
2001	int i;	1228	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		1229
2007	if (grp->int1 == 2)	1230	if (grp->int1 == 2)
2008	croak ("cannot add requests to IO::AIO::GRP after the group finished");	1231	croak ("cannot add requests to IO::AIO::GRP after the group finished");
2009		1232
2010	for (i = 1; i < items; ++i )	1233	for (i = 1; i < items; ++i )
2011	{	1234	{
		1235	aio_req req;
		1236
2012	if (GIMME_V != G_VOID)	1237	if (GIMME_V != G_VOID)
2013	XPUSHs (sv_2mortal (newSVsv (ST (i))));	1238	XPUSHs (sv_2mortal (newSVsv (ST (i))));
2014		1239
2015	req = SvAIO_REQ (ST (i));	1240	req = SvAIO_REQ (ST (i));
2016		1241
2017	if (req)	1242	if (req)
2018	{	1243	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	}	1244	}
2031	}	1245	}
2032		1246
2033	void	1247	void
2034	cancel_subs (aio_req_ornot req)	1248	cancel_subs (aio_req_ornot req)
…		…
2043	AV *av;	1257	AV *av;
2044		1258
2045	grp->errorno = errno;	1259	grp->errorno = errno;
2046		1260
2047	av = newAV ();	1261	av = newAV ();
		1262	av_extend (av, items - 1);
2048		1263
2049	for (i = 1; i < items; ++i )	1264	for (i = 1; i < items; ++i )
2050	av_push (av, newSVsv (ST (i)));	1265	av_push (av, newSVsv (ST (i)));
2051		1266
2052	SvREFCNT_dec (grp->sv1);	1267	SvREFCNT_dec (grp->sv1);
…		…
2059	grp->errorno = errorno;	1274	grp->errorno = errorno;
2060		1275
2061	void	1276	void
2062	limit (aio_req grp, int limit)	1277	limit (aio_req grp, int limit)
2063	CODE:	1278	CODE:
2064	grp->int2 = limit;	1279	eio_grp_limit (grp, limit);
2065	aio_grp_feed (grp);
2066		1280
2067	void	1281	void
2068	feed (aio_req grp, SV *callback=&PL_sv_undef)	1282	feed (aio_req grp, SV *callback=&PL_sv_undef)
2069	CODE:	1283	CODE:
2070	{	1284	{
2071	SvREFCNT_dec (grp->sv2);	1285	SvREFCNT_dec (grp->sv2);
2072	grp->sv2 = newSVsv (callback);	1286	grp->sv2 = newSVsv (callback);
		1287	grp->feed = aio_grp_feed;
2073		1288
2074	if (grp->int2 <= 0)	1289	if (grp->int2 <= 0)
2075	grp->int2 = 2;	1290	grp->int2 = 2;
2076		1291
2077	aio_grp_feed (grp);	1292	eio_grp_limit (grp, grp->int2);
2078	}	1293	}
2079		1294

Diff Legend

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