ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/IO-AIO/AIO.xs

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.41 by root, Sat Jun 24 19:14:04 2006 UTC vs.
Revision 1.235 by root, Thu Jun 25 15:47:01 2015 UTC

1	#define _REENTRANT 1	1	#include "libeio/xthread.h"
		2
2	#include <errno.h>	3	#include <errno.h>
3		4
4	#include "EXTERN.h"	5	#include "EXTERN.h"
5	#include "perl.h"	6	#include "perl.h"
6	#include "XSUB.h"	7	#include "XSUB.h"
7		8
8	#include "autoconf/config.h"	9	#include "schmorp.h"
9
10	#include <pthread.h>
11		10
12	#include <stddef.h>	11	#include <stddef.h>
		12	#include <stdlib.h>
13	#include <errno.h>	13	#include <errno.h>
14	#include <sys/types.h>	14	#include <sys/types.h>
15	#include <sys/stat.h>	15	#include <sys/stat.h>
16	#include <limits.h>	16	#include <limits.h>
17	#include <unistd.h>
18	#include <fcntl.h>	17	#include <fcntl.h>
19	#include <signal.h>
20	#include <sched.h>	18	#include <sched.h>
21		19
22	#if HAVE_SENDFILE	20	#if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
23	# if __linux
24	# include <sys/sendfile.h>
25	# elif __freebsd
26	# include <sys/socket.h>
27	# include <sys/uio.h>	21	# include <sys/mman.h>
28	# elif __hpux	22	#endif
		23
		24	/* the incompetent fool that created musl keeps __linux__, refuses
		25	* to implement any linux standard apis, and also has no way to test
		26	* for his broken iplementation. on't complain if this fails for you.
		27	*/
		28	#if __linux__ && (defined __GLIBC__ || defined __UCLIBC__)
		29	# include <linux/fs.h>
		30	# ifdef FS_IOC_FIEMAP
		31	# include <linux/types.h>
		32	# include <linux/fiemap.h>
		33	# define HAVE_FIEMAP 1
		34	# endif
		35	#endif
		36
		37	/* perl namespace pollution */
		38	#undef VERSION
		39
		40	/* perl stupidly overrides readdir and maybe others */
		41	/* with thread-unsafe versions, imagine that :( */
		42	#undef readdir
		43	#undef opendir
		44	#undef closedir
		45
		46	#ifdef _WIN32
		47
		48	// perl overrides all those nice libc functions
		49
		50	#undef malloc
		51	#undef free
		52	#undef open
		53	#undef read
		54	#undef write
		55	#undef send
		56	#undef recv
		57	#undef stat
		58	#undef lstat
		59	#undef fstat
		60	#undef truncate
		61	#undef ftruncate
		62	#undef open
		63	#undef link
		64	#undef close
		65	#undef unlink
		66	#undef mkdir
		67	#undef rmdir
		68	#undef rename
		69	#undef lseek
		70	#undef opendir
		71	#undef readdir
		72	#undef closedir
		73	#undef chmod
		74	#undef fchmod
		75	#undef dup
		76	#undef dup2
		77	#undef abort
		78	#undef pipe
		79
		80	#define EIO_STRUCT_STAT struct _stati64
		81	#define EIO_STRUCT_STATI64
		82
		83	#else
		84
29	# include <sys/socket.h>	85	#include <sys/time.h>
30	# elif __solaris /* not yet */
31	# include <sys/sendfile.h>	86	#include <sys/select.h>
		87	#include <unistd.h>
		88	#include <utime.h>
		89	#include <signal.h>
		90
		91	#define EIO_STRUCT_STAT Stat_t
		92
		93	#endif
		94
		95	/*****************************************************************************/
		96
		97	#if __GNUC__ >= 3
		98	# define expect(expr,value) __builtin_expect ((expr),(value))
		99	#else
		100	# define expect(expr,value) (expr)
		101	#endif
		102
		103	#define expect_false(expr) expect ((expr) != 0, 0)
		104	#define expect_true(expr) expect ((expr) != 0, 1)
		105
		106	/*****************************************************************************/
		107
		108	typedef SV SV8; /* byte-sv, used for argument-checking */
		109	typedef int aio_rfd; /* read file desriptor */
		110	typedef int aio_wfd; /* write file descriptor */
		111
		112	static HV *aio_stash, *aio_req_stash, *aio_grp_stash, *aio_wd_stash;
		113
		114	#define EIO_REQ_MEMBERS \
		115	SV *callback; \
		116	SV *sv1, *sv2; \
		117	SV *sv3, *sv4; \
		118	STRLEN stroffset; \
		119	SV *self;
		120
		121	#define EIO_NO_WRAPPERS 1
		122
		123	#include "libeio/config.h"
		124	#include "libeio/eio.h"
		125
		126	static int req_invoke (eio_req *req);
		127	#define EIO_FINISH(req) req_invoke (req)
		128	static void req_destroy (eio_req *grp);
		129	#define EIO_DESTROY(req) req_destroy (req)
		130
		131	#include "libeio/eio.c"
		132
		133	#if !HAVE_POSIX_FADVISE
		134	# define posix_fadvise(a,b,c,d) errno = ENOSYS /* also return ENOSYS */
		135	#endif
		136
		137	#if !HAVE_POSIX_MADVISE
		138	# define posix_madvise(a,b,c) errno = ENOSYS /* also return ENOSYS */
		139	#endif
		140
		141	#ifndef MAP_ANONYMOUS
		142	# ifdef MAP_ANON
		143	# define MAP_ANONYMOUS MAP_ANON
32	# else	144	# else
33	# error sendfile support requested but not available	145	# define MAP_ANONYMOUS MAP_FIXED /* and hope this fails */
34	# endif	146	# endif
35	#endif	147	#endif
36		148
37	/* used for struct dirent, AIX doesn't provide it */	149	/* defines all sorts of constants to 0 unless they are already defined */
38	#ifndef NAME_MAX	150	/* also provides const_iv_ and const_niv_ macros for them */
39	# define NAME_MAX 4096	151	#include "def0.h"
40	#endif
41		152
42	#if __ia64	153	#ifndef makedev
43	# define STACKSIZE 65536	154	# define makedev(maj,min) (((maj) << 8) | (min))
44	#else
45	# define STACKSIZE 8192
46	#endif	155	#endif
		156	#ifndef major
		157	# define major(dev) ((dev) >> 8)
		158	#endif
		159	#ifndef minor
		160	# define minor(dev) ((dev) & 0xff)
		161	#endif
47		162
48	enum {	163	#if PAGESIZE <= 0
49	REQ_QUIT,	164	# define PAGESIZE sysconf (_SC_PAGESIZE)
50	REQ_OPEN, REQ_CLOSE,	165	#endif
51	REQ_READ, REQ_WRITE, REQ_READAHEAD,
52	REQ_SENDFILE,
53	REQ_STAT, REQ_LSTAT, REQ_FSTAT,
54	REQ_FSYNC, REQ_FDATASYNC,
55	REQ_UNLINK, REQ_RMDIR, REQ_RENAME,
56	REQ_READDIR,
57	REQ_LINK, REQ_SYMLINK,
58	};
59		166
60	typedef struct aio_cb {	167	/*****************************************************************************/
61	struct aio_cb *volatile next;
62
63	int type;
64
65	/* should receive a cleanup, with unions */
66	int fd, fd2;
67	off_t offset;
68	size_t length;
69	ssize_t result;
70	mode_t mode; /* open */
71	int errorno;
72	SV *data, *callback;
73	SV *fh, *fh2;
74	void *dataptr, *data2ptr;
75	STRLEN dataoffset;
76
77	Stat_t *statdata;
78	} aio_cb;
79
80	typedef aio_cb *aio_req;
81
82	static int started, wanted;
83	static volatile int nreqs;
84	static int max_outstanding = 1<<30;
85	static int respipe [2];
86
87	static pthread_mutex_t reslock = PTHREAD_MUTEX_INITIALIZER;
88	static pthread_mutex_t reqlock = PTHREAD_MUTEX_INITIALIZER;
89	static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
90
91	static volatile aio_req reqs, reqe; /* queue start, queue end */
92	static volatile aio_req ress, rese; /* queue start, queue end */
93
94	static void free_req (aio_req req)
95	{
96	if (req->data)
97	SvREFCNT_dec (req->data);
98
99	if (req->fh)
100	SvREFCNT_dec (req->fh);
101
102	if (req->fh2)
103	SvREFCNT_dec (req->fh2);
104
105	if (req->statdata)
106	Safefree (req->statdata);
107
108	if (req->callback)
109	SvREFCNT_dec (req->callback);
110
111	if (req->type == REQ_READDIR && req->result >= 0)
112	free (req->data2ptr);
113
114	Safefree (req);
115	}
116		168
117	static void	169	static void
118	poll_wait ()	170	fiemap (eio_req *req)
119	{	171	{
120	if (nreqs && !ress)	172	req->result = -1;
121	{
122	fd_set rfd;
123	FD_ZERO(&rfd);
124	FD_SET(respipe [0], &rfd);
125		173
126	select (respipe [0] + 1, &rfd, 0, 0, 0);	174	#if HAVE_FIEMAP
127	}	175	/* assume some c99 */
128	}	176	struct fiemap *fiemap = 0;
		177	size_t end_offset;
		178	int count = req->int3;
129		179
130	static int	180	req->flags |= EIO_FLAG_PTR1_FREE;
131	poll_cb ()	181
132	{	182	/* heuristic: start with 512 bytes (8 extents), and if that isn't enough, */
133	dSP;	183	/* increase in 3.5kb steps */
		184	if (count < 0)
134	int count = 0;	185	count = 8;
135	int do_croak = 0;	186
136	aio_req req;	187	fiemap = malloc (sizeof (*fiemap) + sizeof (struct fiemap_extent) * count);
		188	errno = ENOMEM;
		189	if (!fiemap)
		190	return;
		191
		192	req->ptr1 = fiemap;
		193
		194	fiemap->fm_start = req->offs;
		195	fiemap->fm_length = req->size;
		196	fiemap->fm_flags = req->int2;
		197	fiemap->fm_extent_count = count;
		198
		199	if (ioctl (req->int1, FS_IOC_FIEMAP, fiemap) < 0)
		200	return;
		201
		202	if (req->int3 >= 0 /* not autosizing */
		203	|| !fiemap->fm_mapped_extents /* no more extents */
		204	|| fiemap->fm_extents [fiemap->fm_mapped_extents - 1].fe_flags & FIEMAP_EXTENT_LAST /* hit eof */)
		205	goto done;
		206
		207	/* else we have to loop -
		208	* it would be tempting (actually I tried that first) to just query the
		209	* number of extents needed, but linux often feels like not returning all
		210	* extents, without telling us it left any out. this complicates
		211	* this quite a bit.
		212	*/
		213
		214	end_offset = fiemap->fm_length + (fiemap->fm_length == FIEMAP_MAX_OFFSET ? 0 : fiemap->fm_start);
137		215
138	for (;;)	216	for (;;)
139	{	217	{
140	pthread_mutex_lock (&reslock);	218	/* we go in 54 extent steps - 3kb, in the hope that this fits nicely on the eio stack (normally 16+ kb) */
141	req = ress;	219	char scratch[3072];
		220	struct fiemap *incmap = (struct fiemap *)scratch;
142		221
		222	incmap->fm_start = fiemap->fm_extents [fiemap->fm_mapped_extents - 1].fe_logical
		223	+ fiemap->fm_extents [fiemap->fm_mapped_extents - 1].fe_length;
		224	incmap->fm_length = fiemap->fm_length - (incmap->fm_start - fiemap->fm_start);
		225	incmap->fm_flags = fiemap->fm_flags;
		226	incmap->fm_extent_count = (sizeof (scratch) - sizeof (struct fiemap)) / sizeof (struct fiemap_extent);
		227
		228	if (ioctl (req->int1, FS_IOC_FIEMAP, incmap) < 0)
		229	return;
		230
		231	if (!incmap->fm_mapped_extents)
		232	goto done;
		233
		234	count = fiemap->fm_mapped_extents + incmap->fm_mapped_extents;
		235	fiemap = realloc (fiemap, sizeof (*fiemap) + sizeof (struct fiemap_extent) * count);
		236	errno = ENOMEM;
143	if (req)	237	if (!fiemap)
		238	return;
		239
		240	req->ptr1 = fiemap;
		241
		242	for (count = 0; count < incmap->fm_mapped_extents; ++count)
144	{	243	{
145	ress = req->next;	244	struct fiemap_extent *e = incmap->fm_extents + count;
146		245
147	if (!ress)	246	if (e->fe_logical + e->fe_length >= end_offset)
148	{	247	goto done;
149	/* read any signals sent by the worker threads */
150	char buf [32];
151	while (read (respipe [0], buf, 32) == 32)
152	;
153		248
154	rese = 0;	249	fiemap->fm_extents [fiemap->fm_mapped_extents++] = *e;
155	}
156	}
157		250
158	pthread_mutex_unlock (&reslock);	251	if (e->fe_flags & FIEMAP_EXTENT_LAST)
		252	goto done;
159		253
160	if (!req)
161	break;
162
163	nreqs--;
164
165	if (req->type == REQ_QUIT)
166	started--;
167	else
168	{
169	int errorno = errno;
170	errno = req->errorno;
171
172	if (req->type == REQ_READ)
173	SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
174
175	if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
176	SvREADONLY_off (req->data);
177
178	if (req->statdata)
179	{
180	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
181	PL_laststatval = req->result;
182	PL_statcache = *(req->statdata);
183	}
184
185	ENTER;
186	PUSHMARK (SP);
187
188	if (req->type == REQ_READDIR)
189	{
190	SV *rv = &PL_sv_undef;
191
192	if (req->result >= 0)
193	{
194	char *buf = req->data2ptr;
195	AV *av = newAV ();
196
197	while (req->result)
198	{
199	SV *sv = newSVpv (buf, 0);
200
201	av_push (av, sv);
202	buf += SvCUR (sv) + 1;
203	req->result--;
204	}
205
206	rv = sv_2mortal (newRV_noinc ((SV *)av));
207	}
208
209	XPUSHs (rv);
210	}
211	else
212	{
213	XPUSHs (sv_2mortal (newSViv (req->result)));
214
215	if (req->type == REQ_OPEN)
216	{
217	/* convert fd to fh */
218	SV *fh;
219
220	PUTBACK;
221	call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);
222	SPAGAIN;
223
224	fh = SvREFCNT_inc (POPs);
225
226	PUSHMARK (SP);
227	XPUSHs (sv_2mortal (fh));
228	}
229	}
230
231	if (SvOK (req->callback))
232	{
233	PUTBACK;
234	call_sv (req->callback, G_VOID | G_EVAL);
235	SPAGAIN;
236
237	if (SvTRUE (ERRSV))
238	{
239	free_req (req);
240	croak (0);
241	}
242	}
243
244	LEAVE;
245
246	errno = errorno;
247	count++;
248	}
249
250	free_req (req);
251	}
252
253	return count;
254	}
255
256	static void *aio_proc(void *arg);
257
258	static void
259	start_thread (void)
260	{
261	sigset_t fullsigset, oldsigset;
262	pthread_t tid;
263	pthread_attr_t attr;
264
265	pthread_attr_init (&attr);
266	pthread_attr_setstacksize (&attr, STACKSIZE);
267	pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
268
269	sigfillset (&fullsigset);
270	sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
271
272	if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
273	started++;
274
275	sigprocmask (SIG_SETMASK, &oldsigset, 0);
276	}
277
278	static void
279	send_req (aio_req req)
280	{
281	while (started < wanted && nreqs >= started)
282	start_thread ();
283
284	nreqs++;
285
286	pthread_mutex_lock (&reqlock);
287
288	req->next = 0;
289
290	if (reqe)
291	{
292	reqe->next = req;
293	reqe = req;
294	}
295	else
296	reqe = reqs = req;
297
298	pthread_cond_signal (&reqwait);
299	pthread_mutex_unlock (&reqlock);
300
301	if (nreqs > max_outstanding)
302	for (;;)
303	{
304	poll_cb ();
305
306	if (nreqs <= max_outstanding)
307	break;
308
309	poll_wait ();
310	}
311	}
312
313	static void
314	end_thread (void)
315	{
316	aio_req req;
317	Newz (0, req, 1, aio_cb);
318	req->type = REQ_QUIT;
319
320	send_req (req);
321	}
322
323	static void min_parallel (int nthreads)
324	{
325	if (wanted < nthreads)
326	wanted = nthreads;
327	}
328
329	static void max_parallel (int nthreads)
330	{
331	int cur = started;
332
333	if (wanted > nthreads)
334	wanted = nthreads;
335
336	while (cur > wanted)
337	{
338	end_thread ();
339	cur--;
340	}
341
342	while (started > wanted)
343	{
344	poll_wait ();
345	poll_cb ();
346	}
347	}
348
349	static void create_pipe ()
350	{
351	if (pipe (respipe))
352	croak ("unable to initialize result pipe");
353
354	if (fcntl (respipe [0], F_SETFL, O_NONBLOCK))
355	croak ("cannot set result pipe to nonblocking mode");
356
357	if (fcntl (respipe [1], F_SETFL, O_NONBLOCK))
358	croak ("cannot set result pipe to nonblocking mode");
359	}
360
361	/*****************************************************************************/
362	/* work around various missing functions */
363
364	#if !HAVE_PREADWRITE
365	# define pread aio_pread
366	# define pwrite aio_pwrite
367
368	/*
369	* make our pread/pwrite safe against themselves, but not against
370	* normal read/write by using a mutex. slows down execution a lot,
371	* but that's your problem, not mine.
372	*/
373	static pthread_mutex_t preadwritelock = PTHREAD_MUTEX_INITIALIZER;
374
375	static ssize_t
376	pread (int fd, void *buf, size_t count, off_t offset)
377	{
378	ssize_t res;
379	off_t ooffset;
380
381	pthread_mutex_lock (&preadwritelock);
382	ooffset = lseek (fd, 0, SEEK_CUR);
383	lseek (fd, offset, SEEK_SET);
384	res = read (fd, buf, count);
385	lseek (fd, ooffset, SEEK_SET);
386	pthread_mutex_unlock (&preadwritelock);
387
388	return res;
389	}
390
391	static ssize_t
392	pwrite (int fd, void *buf, size_t count, off_t offset)
393	{
394	ssize_t res;
395	off_t ooffset;
396
397	pthread_mutex_lock (&preadwritelock);
398	ooffset = lseek (fd, 0, SEEK_CUR);
399	lseek (fd, offset, SEEK_SET);
400	res = write (fd, buf, count);
401	lseek (fd, offset, SEEK_SET);
402	pthread_mutex_unlock (&preadwritelock);
403
404	return res;
405	}
406	#endif
407
408	#if !HAVE_FDATASYNC
409	# define fdatasync fsync
410	#endif
411
412	#if !HAVE_READAHEAD
413	# define readahead aio_readahead
414
415	static ssize_t
416	readahead (int fd, off_t offset, size_t count)
417	{
418	char readahead_buf[4096];
419
420	while (count > 0)
421	{
422	size_t len = count < sizeof (readahead_buf) ? count : sizeof (readahead_buf);
423
424	pread (fd, readahead_buf, len, offset);
425	offset += len;
426	count -= len;
427	}
428
429	errno = 0;
430	}
431	#endif
432
433	#if !HAVE_READDIR_R
434	# define readdir_r aio_readdir_r
435
436	static pthread_mutex_t readdirlock = PTHREAD_MUTEX_INITIALIZER;
437
438	static int
439	readdir_r (DIR *dirp, struct dirent *ent, struct dirent **res)
440	{
441	struct dirent *e;
442	int errorno;
443
444	pthread_mutex_lock (&readdirlock);
445
446	e = readdir (dirp);
447	errorno = errno;
448
449	if (e)
450	{
451	*res = ent;
452	strcpy (ent->d_name, e->d_name);
453	}
454	else
455	*res = 0;
456
457	pthread_mutex_unlock (&readdirlock);
458
459	errno = errorno;
460	return e ? 0 : -1;
461	}
462	#endif
463
464	/* sendfile always needs emulation */
465	static ssize_t
466	sendfile_ (int ofd, int ifd, off_t offset, size_t count)
467	{
468	ssize_t res;
469
470	if (!count)
471	return 0;
472
473	#if HAVE_SENDFILE
474	# if __linux
475	res = sendfile (ofd, ifd, &offset, count);
476
477	# elif __freebsd
478	/*
479	* Of course, the freebsd sendfile is a dire hack with no thoughts
480	* wasted on making it similar to other I/O functions.
481	*/
482	{
483	off_t sbytes;
484	res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
485
486	if (res < 0 && sbytes)
487	/* maybe only on EAGAIN only: as usual, the manpage leaves you guessing */
488	res = sbytes;
489	}
490
491	# elif __hpux
492	res = sendfile (ofd, ifd, offset, count, 0, 0);
493
494	# elif __solaris
495	{
496	struct sendfilevec vec;
497	size_t sbytes;
498
499	vec.sfv_fd = ifd;
500	vec.sfv_flag = 0;
501	vec.sfv_off = offset;
502	vec.sfv_len = count;
503
504	res = sendfilev (ofd, &vec, 1, &sbytes);
505
506	if (res < 0 && sbytes)
507	res = sbytes;
508	}
509
510	# endif
511	#else
512	res = -1;
513	errno = ENOSYS;
514	#endif
515
516	if (res < 0
517	&& (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
518	#if __solaris
519	|| errno == EAFNOSUPPORT || errno == EPROTOTYPE
520	#endif
521	)
522	)
523	{
524	/* emulate sendfile. this is a major pain in the ass */
525	char buf[4096];
526	res = 0;
527
528	while (count)
529	{
530	ssize_t cnt;
531
532	cnt = pread (ifd, buf, count > 4096 ? 4096 : count, offset);
533
534	if (cnt <= 0)
535	{
536	if (cnt && !res) res = -1;
537	break;
538	}
539
540	cnt = write (ofd, buf, cnt);
541
542	if (cnt <= 0)
543	{
544	if (cnt && !res) res = -1;
545	break;
546	}
547
548	offset += cnt;
549	res += cnt;
550	count -= cnt;
551	}	254	}
552	}	255	}
553		256
554	return res;	257	done:
555	}	258	req->result = 0;
556		259
557	/* read a full directory */	260	#else
		261	errno = ENOSYS;
		262	#endif
		263	}
		264
		265	/*****************************************************************************/
		266
		267	enum {
		268	FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
		269	};
		270
		271	typedef eio_req *aio_req;
		272	typedef eio_req *aio_req_ornot;
		273	typedef eio_wd aio_wd;
		274
		275	static SV *on_next_submit;
		276	static int next_pri = EIO_PRI_DEFAULT;
		277	static int max_outstanding;
		278
		279	static s_epipe respipe;
		280
		281	static void req_destroy (aio_req req);
		282	static void req_cancel (aio_req req);
		283
		284	static void
		285	want_poll (void)
		286	{
		287	/* write a dummy byte to the pipe so fh becomes ready */
		288	s_epipe_signal (&respipe);
		289	}
		290
		291	static void
		292	done_poll (void)
		293	{
		294	/* read any signals sent by the worker threads */
		295	s_epipe_drain (&respipe);
		296	}
		297
		298	/* must be called at most once */
		299	static SV *
		300	req_sv (aio_req req, HV *stash)
		301	{
		302	if (!req->self)
		303	{
		304	req->self = (SV *)newHV ();
		305	sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
		306	}
		307
		308	return sv_2mortal (sv_bless (newRV_inc (req->self), stash));
		309	}
		310
		311	static SV *
		312	newSVaio_wd (aio_wd wd)
		313	{
		314	return sv_bless (newRV_noinc (newSViv ((intptr_t)wd)), aio_wd_stash);
		315	}
		316
		317	static aio_req
		318	SvAIO_REQ (SV *sv)
		319	{
		320	MAGIC *mg;
		321
		322	if (!SvROK (sv)
		323	/* for speed reasons, we do not verify that SvROK actually has a stash ptr */
		324	|| (SvSTASH (SvRV (sv)) != aio_grp_stash
		325	&& SvSTASH (SvRV (sv)) != aio_req_stash
		326	&& !sv_derived_from (sv, "IO::AIO::REQ")))
		327	croak ("object of class IO::AIO::REQ expected");
		328
		329	mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
		330
		331	return mg ? (aio_req)mg->mg_ptr : 0;
		332	}
		333
		334	static aio_wd
		335	SvAIO_WD (SV *sv)
		336	{
		337	if (!SvROK (sv)
		338	|| SvTYPE (SvRV (sv)) != SVt_PVMG
		339	|| SvSTASH (SvRV (sv)) != aio_wd_stash)
		340	croak ("IO::AIO: expected a working directory object as returned by aio_wd");
		341
		342	return (aio_wd)(long)SvIVX (SvRV (sv));
		343	}
		344
		345	static SV *
		346	newmortalFH (int fd, int flags)
		347	{
		348	if (fd < 0)
		349	return &PL_sv_undef;
		350
		351	GV *gv = (GV *)sv_newmortal ();
		352	char sym[64];
		353	int symlen;
		354
		355	symlen = snprintf (sym, sizeof (sym), "fd#%d", fd);
		356	gv_init (gv, aio_stash, sym, symlen, 0);
		357
		358	symlen = snprintf (
		359	sym,
		360	sizeof (sym),
		361	"%s&=%d",
		362	flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
		363	fd
		364	);
		365
		366	return do_open (gv, sym, symlen, 0, 0, 0, 0)
		367	? (SV *)gv : &PL_sv_undef;
		368	}
		369
		370	static void
		371	aio_grp_feed (aio_req grp)
		372	{
		373	if (grp->sv2 && SvOK (grp->sv2))
		374	{
		375	dSP;
		376
		377	ENTER;
		378	SAVETMPS;
		379	PUSHMARK (SP);
		380	XPUSHs (req_sv (grp, aio_grp_stash));
		381	PUTBACK;
		382	call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
		383	SPAGAIN;
		384	FREETMPS;
		385	LEAVE;
		386	}
		387	}
		388
		389	static void
		390	req_submit (eio_req *req)
		391	{
		392	eio_submit (req);
		393
		394	if (expect_false (on_next_submit))
		395	{
		396	dSP;
		397	SV *cb = sv_2mortal (on_next_submit);
		398
		399	on_next_submit = 0;
		400
		401	PUSHMARK (SP);
		402	PUTBACK;
		403	call_sv (cb, G_DISCARD | G_EVAL);
		404	}
		405	}
		406
558	static int	407	static int
559	scandir_ (const char *path, void **namesp)	408	req_invoke (eio_req *req)
560	{	409	{
561	DIR *dirp = opendir (path);	410	if (req->flags & FLAG_SV2_RO_OFF)
562	union	411	SvREADONLY_off (req->sv2);
563	{
564	struct dirent d;
565	char b [offsetof (struct dirent, d_name) + NAME_MAX + 1];
566	} u;
567	struct dirent *entp;
568	char *name, *names;
569	int memlen = 4096;
570	int memofs = 0;
571	int res = 0;
572	int errorno;
573		412
574	if (!dirp)	413	if (!EIO_CANCELLED (req) && req->callback)
575	return -1;
576
577	names = malloc (memlen);
578
579	for (;;)
580	{	414	{
581	errno = 0, readdir_r (dirp, &u.d, &entp);	415	dSP;
		416	static SV *sv_result_cache; /* caches the result integer SV */
		417	SV *sv_result;
582		418
583	if (!entp)	419	ENTER;
584	break;	420	SAVETMPS;
		421	PUSHMARK (SP);
		422	EXTEND (SP, 1);
585		423
586	name = entp->d_name;	424	/* do not recreate the result IV from scratch each time */
587		425	if (expect_true (sv_result_cache))
588	if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
589	{	426	{
590	int len = strlen (name) + 1;	427	sv_result = sv_result_cache; sv_result_cache = 0;
591		428	SvIV_set (sv_result, req->result);
592	res++;	429	SvIOK_only (sv_result);
593
594	while (memofs + len > memlen)
595	{
596	memlen *= 2;
597	names = realloc (names, memlen);
598	if (!names)
599	break;
600	}
601
602	memcpy (names + memofs, name, len);
603	memofs += len;
604	}
605	}
606
607	errorno = errno;
608	closedir (dirp);
609
610	if (errorno)
611	{
612	free (names);
613	errno = errorno;
614	res = -1;
615	}
616
617	*namesp = (void *)names;
618	return res;
619	}
620
621	/*****************************************************************************/
622
623	static void *
624	aio_proc (void *thr_arg)
625	{
626	aio_req req;
627	int type;
628
629	do
630	{
631	pthread_mutex_lock (&reqlock);
632
633	for (;;)
634	{
635	req = reqs;
636
637	if (reqs)
638	{
639	reqs = reqs->next;
640	if (!reqs) reqe = 0;
641	}
642
643	if (req)
644	break;
645
646	pthread_cond_wait (&reqwait, &reqlock);
647	}
648
649	pthread_mutex_unlock (&reqlock);
650
651	errno = 0; /* strictly unnecessary */
652
653	type = req->type;
654
655	switch (type)
656	{
657	case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
658	case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
659
660	case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
661	case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length); break;
662
663	case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
664	case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
665	case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
666
667	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
668	case REQ_CLOSE: req->result = close (req->fd); break;
669	case REQ_UNLINK: req->result = unlink (req->dataptr); break;
670	case REQ_RMDIR: req->result = rmdir (req->dataptr); break;
671	case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break;
672	case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break;
673	case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;
674
675	case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
676	case REQ_FSYNC: req->result = fsync (req->fd); break;
677	case REQ_READDIR: req->result = scandir_ (req->dataptr, &req->data2ptr); break;
678
679	case REQ_QUIT:
680	break;
681
682	default:
683	req->result = ENOSYS;
684	break;
685	}
686
687	req->errorno = errno;
688
689	pthread_mutex_lock (&reslock);
690
691	req->next = 0;
692
693	if (rese)
694	{
695	rese->next = req;
696	rese = req;
697	}	430	}
698	else	431	else
699	{	432	{
700	rese = ress = req;	433	sv_result = newSViv (req->result);
701		434	SvREADONLY_on (sv_result);
702	/* write a dummy byte to the pipe so fh becomes ready */
703	write (respipe [1], &respipe, 1);
704	}	435	}
705		436
706	pthread_mutex_unlock (&reslock);	437	switch (req->type)
		438	{
		439	case EIO_WD_OPEN:
		440	PUSHs (req->result ? &PL_sv_undef : sv_2mortal (newSVaio_wd (req->wd)));
		441	break;
		442
		443	case EIO_READDIR:
		444	{
		445	SV *rv = &PL_sv_undef;
		446
		447	if (req->result >= 0)
		448	{
		449	int i;
		450	char *names = (char *)req->ptr2;
		451	eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
		452	AV *av = newAV ();
		453
		454	av_extend (av, req->result - 1);
		455
		456	for (i = 0; i < req->result; ++i)
		457	{
		458	if (req->int1 & EIO_READDIR_DENTS)
		459	{
		460	SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
		461
		462	if (req->int1 & EIO_READDIR_CUSTOM2)
		463	{
		464	static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */
		465	AV *avent = newAV ();
		466
		467	av_extend (avent, 2);
		468
		469	if (!sv_type [ent->type])
		470	{
		471	sv_type [ent->type] = newSViv (ent->type);
		472	SvREADONLY_on (sv_type [ent->type]);
		473	}
		474
		475	av_store (avent, 0, namesv);
		476	av_store (avent, 1, SvREFCNT_inc (sv_type [ent->type]));
		477	av_store (avent, 2, IVSIZE >= 8 ? newSVuv (ent->inode) : newSVnv (ent->inode));
		478
		479	av_store (av, i, newRV_noinc ((SV *)avent));
		480	}
		481	else
		482	av_store (av, i, namesv);
		483
		484	++ent;
		485	}
		486	else
		487	{
		488	SV *name = newSVpv (names, 0);
		489	av_store (av, i, name);
		490	names += SvCUR (name) + 1;
		491	}
		492	}
		493
		494	rv = sv_2mortal (newRV_noinc ((SV *)av));
		495	}
		496
		497	PUSHs (rv);
		498
		499	if (req->int1 & EIO_READDIR_CUSTOM1)
		500	XPUSHs (sv_2mortal (newSViv (req->int1 & ~(EIO_READDIR_CUSTOM1 | EIO_READDIR_CUSTOM2))));
		501	}
		502	break;
		503
		504	case EIO_OPEN:
		505	PUSHs (newmortalFH (req->result, req->int1 & (O_RDONLY | O_WRONLY | O_RDWR)));
		506	break;
		507
		508	case EIO_STATVFS:
		509	case EIO_FSTATVFS:
		510	{
		511	SV *rv = &PL_sv_undef;
		512
		513	#ifndef _WIN32
		514	if (req->result >= 0)
		515	{
		516	EIO_STRUCT_STATVFS *f = EIO_STATVFS_BUF (req);
		517	HV *hv = newHV ();
		518	/* POSIX requires fsid to be unsigned long, but AIX in its infinite wisdom
		519	* chooses to make it a struct.
		520	*/
		521	unsigned long fsid = 0;
		522	memcpy (&fsid, &f->f_fsid, sizeof (unsigned long) < sizeof (f->f_fsid) ? sizeof (unsigned long) : sizeof (f->f_fsid));
		523
		524	rv = sv_2mortal (newRV_noinc ((SV *)hv));
		525
		526	hv_store (hv, "bsize" , sizeof ("bsize" ) - 1, newSVval64 (f->f_bsize ), 0);
		527	hv_store (hv, "frsize" , sizeof ("frsize" ) - 1, newSVval64 (f->f_frsize ), 0);
		528	hv_store (hv, "blocks" , sizeof ("blocks" ) - 1, newSVval64 (f->f_blocks ), 0);
		529	hv_store (hv, "bfree" , sizeof ("bfree" ) - 1, newSVval64 (f->f_bfree ), 0);
		530	hv_store (hv, "bavail" , sizeof ("bavail" ) - 1, newSVval64 (f->f_bavail ), 0);
		531	hv_store (hv, "files" , sizeof ("files" ) - 1, newSVval64 (f->f_files ), 0);
		532	hv_store (hv, "ffree" , sizeof ("ffree" ) - 1, newSVval64 (f->f_ffree ), 0);
		533	hv_store (hv, "favail" , sizeof ("favail" ) - 1, newSVval64 (f->f_favail ), 0);
		534	hv_store (hv, "fsid" , sizeof ("fsid" ) - 1, newSVval64 (fsid ), 0);
		535	hv_store (hv, "flag" , sizeof ("flag" ) - 1, newSVval64 (f->f_flag ), 0);
		536	hv_store (hv, "namemax", sizeof ("namemax") - 1, newSVval64 (f->f_namemax), 0);
		537	}
		538	#endif
		539
		540	PUSHs (rv);
		541	}
		542
		543	break;
		544
		545	case EIO_GROUP:
		546	req->int1 = 2; /* mark group as finished */
		547
		548	if (req->sv1)
		549	{
		550	int i;
		551	AV *av = (AV *)req->sv1;
		552
		553	EXTEND (SP, AvFILL (av) + 1);
		554	for (i = 0; i <= AvFILL (av); ++i)
		555	PUSHs (*av_fetch (av, i, 0));
		556	}
		557	break;
		558
		559	case EIO_NOP:
		560	case EIO_BUSY:
		561	break;
		562
		563	case EIO_READLINK:
		564	case EIO_REALPATH:
		565	if (req->result > 0)
		566	PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
		567	break;
		568
		569	case EIO_STAT:
		570	case EIO_LSTAT:
		571	case EIO_FSTAT:
		572	PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT;
		573
		574	if (!(PL_laststatval = req->result))
		575	/* if compilation fails here then perl's Stat_t is not struct _stati64 */
		576	PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
		577
		578	PUSHs (sv_result);
		579	break;
		580
		581	case EIO_SEEK:
		582	PUSHs (req->result ? sv_result : sv_2mortal (newSVval64 (req->offs)));
		583	break;
		584
		585	case EIO_READ:
		586	{
		587	SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
		588	*SvEND (req->sv2) = 0;
		589	SvPOK_only (req->sv2);
		590	SvSETMAGIC (req->sv2);
		591	PUSHs (sv_result);
		592	}
		593	break;
		594
		595	case EIO_CUSTOM:
		596	if (req->feed == fiemap)
		597	{
		598	#if HAVE_FIEMAP
		599	if (!req->result)
		600	{
		601	struct fiemap *fiemap = (struct fiemap *)req->ptr1;
		602
		603	if (fiemap->fm_extent_count)
		604	{
		605	AV *av = newAV ();
		606	int i;
		607
		608	while (fiemap->fm_mapped_extents)
		609	{
		610	struct fiemap_extent *extent = &fiemap->fm_extents [--fiemap->fm_mapped_extents];
		611	AV *ext_av = newAV ();
		612
		613	av_store (ext_av, 3, newSVuv (extent->fe_flags));
		614	av_store (ext_av, 2, newSVval64 (extent->fe_length));
		615	av_store (ext_av, 1, newSVval64 (extent->fe_physical));
		616	av_store (ext_av, 0, newSVval64 (extent->fe_logical));
		617
		618	av_store (av, fiemap->fm_mapped_extents, newRV_noinc ((SV *)ext_av));
		619	}
		620
		621	PUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
		622	}
		623	else
		624	{
		625	SvIV_set (sv_result, fiemap->fm_mapped_extents);
		626	PUSHs (sv_result);
		627	}
		628	}
		629	#endif
		630	}
		631	else
		632	PUSHs (sv_result);
		633	break;
		634
		635	case EIO_DUP2: /* EIO_DUP2 actually means aio_close(), so fudge result value */
		636	if (req->result > 0)
		637	SvIV_set (sv_result, 0);
		638	/* FALLTHROUGH */
		639
		640	default:
		641	PUSHs (sv_result);
		642	break;
		643	}
		644
		645	errno = req->errorno;
		646
		647	PUTBACK;
		648	call_sv (req->callback, G_VOID | G_EVAL | G_DISCARD);
		649	SPAGAIN;
		650
		651	if (expect_false (SvREFCNT (sv_result) != 1 || sv_result_cache))
		652	SvREFCNT_dec (sv_result);
		653	else
		654	sv_result_cache = sv_result;
		655
		656	FREETMPS;
		657	LEAVE;
		658
		659	PUTBACK;
707	}	660	}
708	while (type != REQ_QUIT);	661
		662	return !!SvTRUE (ERRSV);
		663	}
		664
		665	static void
		666	req_destroy (aio_req req)
		667	{
		668	if (req->self)
		669	{
		670	sv_unmagic (req->self, PERL_MAGIC_ext);
		671	SvREFCNT_dec (req->self);
		672	}
		673
		674	SvREFCNT_dec (req->sv1);
		675	SvREFCNT_dec (req->sv2);
		676	SvREFCNT_dec (req->sv3);
		677	SvREFCNT_dec (req->sv4);
		678	SvREFCNT_dec (req->callback);
		679
		680	free (req);
		681	}
		682
		683	static void
		684	req_cancel_subs (aio_req grp)
		685	{
		686	if (grp->type != EIO_GROUP)
		687	return;
		688
		689	SvREFCNT_dec (grp->sv2);
		690	grp->sv2 = 0;
		691
		692	eio_grp_cancel (grp);
		693	}
		694
		695	static void ecb_cold
		696	create_respipe (void)
		697	{
		698	if (s_epipe_renew (&respipe))
		699	croak ("IO::AIO: unable to initialize result pipe");
		700	}
		701
		702	static void
		703	poll_wait (void)
		704	{
		705	while (eio_nreqs ())
		706	{
		707	int size;
		708
		709	X_LOCK (reslock);
		710	size = res_queue.size;
		711	X_UNLOCK (reslock);
		712
		713	if (size)
		714	return;
		715
		716	etp_maybe_start_thread ();
		717
		718	s_epipe_wait (&respipe);
		719	}
		720	}
		721
		722	static int
		723	poll_cb (void)
		724	{
		725	for (;;)
		726	{
		727	int res = eio_poll ();
		728
		729	if (res > 0)
		730	croak (0);
		731
		732	if (!max_outstanding || max_outstanding > eio_nreqs ())
		733	return res;
		734
		735	poll_wait ();
		736	}
		737	}
		738
		739	static void ecb_cold
		740	reinit (void)
		741	{
		742	create_respipe ();
		743
		744	if (eio_init (want_poll, done_poll) < 0)
		745	croak ("IO::AIO: unable to initialise eio library");
		746	}
		747
		748	/*****************************************************************************/
		749
		750	#if !_POSIX_MAPPED_FILES
		751	# define mmap(addr,length,prot,flags,fd,offs) EIO_ENOSYS ()
		752	# define munmap(addr,length) EIO_ENOSYS ()
		753	#endif
		754
		755	#if !_POSIX_MEMORY_PROTECTION
		756	# define mprotect(addr,len,prot) EIO_ENOSYS ()
		757	#endif
		758
		759	#define MMAP_MAGIC PERL_MAGIC_ext
		760
		761	static int ecb_cold
		762	mmap_free (pTHX_ SV *sv, MAGIC *mg)
		763	{
		764	int old_errno = errno;
		765	munmap (mg->mg_ptr, (size_t)mg->mg_obj);
		766	errno = old_errno;
		767
		768	mg->mg_obj = 0; /* just in case */
		769
		770	SvREADONLY_off (sv);
		771
		772	if (SvPVX (sv) != mg->mg_ptr)
		773	croak ("ERROR: IO::AIO::mmap-mapped scalar changed location, detected");
		774
		775	SvCUR_set (sv, 0);
		776	SvPVX (sv) = 0;
		777	SvOK_off (sv);
709		778
710	return 0;	779	return 0;
711	}	780	}
712		781
		782	static MGVTBL mmap_vtbl = {
		783	0, 0, 0, 0, mmap_free
		784	};
		785
713	/*****************************************************************************/	786	/*****************************************************************************/
714		787
715	static void atfork_prepare (void)	788	static SV *
		789	get_cb (SV *cb_sv)
716	{	790	{
717	pthread_mutex_lock (&reqlock);	791	SvGETMAGIC (cb_sv);
718	pthread_mutex_lock (&reslock);	792	return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0;
719	#if !HAVE_PREADWRITE
720	pthread_mutex_lock (&preadwritelock);
721	#endif
722	#if !HAVE_READDIR_R
723	pthread_mutex_lock (&readdirlock);
724	#endif
725	}	793	}
726		794
727	static void atfork_parent (void)	795	static aio_req ecb_noinline
		796	dreq (SV *callback)
728	{	797	{
729	#if !HAVE_READDIR_R	798	SV *cb_cv;
730	pthread_mutex_unlock (&readdirlock);
731	#endif
732	#if !HAVE_PREADWRITE
733	pthread_mutex_unlock (&preadwritelock);
734	#endif
735	pthread_mutex_unlock (&reslock);
736	pthread_mutex_unlock (&reqlock);
737	}
738
739	static void atfork_child (void)
740	{
741	aio_req prv;	799	aio_req req;
		800	int req_pri = next_pri;
		801	next_pri = EIO_PRI_DEFAULT;
742		802
743	started = 0;	803	cb_cv = get_cb (callback);
744		804
745	while (reqs)	805	req = calloc (sizeof (*req), 1);
		806	if (!req)
		807	croak ("out of memory during eio_req allocation");
		808
		809	req->callback = SvREFCNT_inc (cb_cv);
		810	req->pri = req_pri;
		811
		812	return req;
		813	}
		814
		815	#define dREQ \
		816	aio_req req = dreq (callback); \
		817
		818	#define REQ_SEND \
		819	PUTBACK; \
		820	req_submit (req); \
		821	SPAGAIN; \
		822	\
		823	if (GIMME_V != G_VOID) \
		824	XPUSHs (req_sv (req, aio_req_stash));
		825
		826	ecb_inline void
		827	req_set_path (aio_req req, SV *path, SV **wdsv, SV **pathsv, eio_wd *wd, void **ptr)
		828	{
		829	if (expect_false (SvROK (path)))
746	{	830	{
747	prv = reqs;	831	SV *rv = SvRV (path);
748	reqs = prv->next;	832	SV *wdob;
749	free_req (prv);	833
		834	if (SvTYPE (rv) == SVt_PVAV && AvFILLp (rv) == 1)
		835	{
		836	path = AvARRAY (rv)[1];
		837	wdob = AvARRAY (rv)[0];
		838
		839	if (SvOK (wdob))
		840	{
		841	*wd = SvAIO_WD (wdob);
		842	*wdsv = SvREFCNT_inc_NN (SvRV (wdob));
		843	}
		844	else
		845	*wd = EIO_INVALID_WD;
		846	}
		847	else if (SvTYPE (rv) == SVt_PVMG && SvSTASH (rv) == aio_wd_stash)
		848	{
		849	*wd = (aio_wd)(long)SvIVX (rv);
		850	*wdsv = SvREFCNT_inc_NN (rv);
		851	*ptr = ".";
		852	return; /* path set to "." */
		853	}
		854	else
		855	croak ("IO::AIO: pathname arguments must be specified as a string, an IO::AIO::WD object or a [IO::AIO::WD, path] pair");
750	}	856	}
751		857
752	reqs = reqe = 0;	858	*pathsv = newSVsv (path);
753		859	*ptr = SvPVbyte_nolen (*pathsv);
754	while (ress)	860	}
		861
		862	static void ecb_noinline
		863	req_set_path1 (aio_req req, SV *path)
		864	{
		865	req_set_path (req, path, &req->sv1, &req->sv3, &req->wd, &req->ptr1);
		866	}
		867
		868	static void ecb_noinline
		869	req_set_fh_or_path (aio_req req, int type_path, int type_fh, SV *fh_or_path)
		870	{
		871	SV *rv = SvROK (fh_or_path) ? SvRV (fh_or_path) : fh_or_path;
		872
		873	switch (SvTYPE (rv))
755	{	874	{
756	prv = ress;	875	case SVt_PVIO:
757	ress = prv->next;	876	case SVt_PVLV:
758	free_req (prv);	877	case SVt_PVGV:
		878	req->type = type_fh;
		879	req->sv1 = newSVsv (fh_or_path);
		880	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		881	break;
		882
		883	default:
		884	req->type = type_path;
		885	req_set_path1 (req, fh_or_path);
		886	break;
759	}	887	}
760
761	ress = rese = 0;
762
763	close (respipe [0]);
764	close (respipe [1]);
765	create_pipe ();
766
767	atfork_parent ();
768	}	888	}
769		889
770	#define dREQ \	890	XS(boot_IO__AIO) ecb_cold;
771	aio_req req; \	891
772	\
773	if (SvOK (callback) && !SvROK (callback)) \
774	croak ("clalback must be undef or of reference type"); \
775	\
776	Newz (0, req, 1, aio_cb); \
777	if (!req) \
778	croak ("out of memory during aio_req allocation"); \
779	\
780	req->callback = newSVsv (callback);
781
782	MODULE = IO::AIO PACKAGE = IO::AIO	892	MODULE = IO::AIO PACKAGE = IO::AIO
783		893
784	PROTOTYPES: ENABLE	894	PROTOTYPES: ENABLE
785		895
786	BOOT:	896	BOOT:
787	{	897	{
		898	static const struct {
		899	const char *name;
		900	IV iv;
		901	} *civ, const_iv[] = {
		902	# define const_niv(name, value) { # name, (IV) value },
		903	# define const_iv(name) { # name, (IV) name },
		904	# define const_eio(name) { # name, (IV) EIO_ ## name },
		905
		906	/* you have to re-run ./gendef0 after adding/removing any constants here */
		907	/* the first block can be undef if missing */
		908	const_iv (ENOSYS)
		909	const_iv (EXDEV)
		910	const_iv (EBADR)
		911
		912	/* for lseek */
		913	const_iv (SEEK_DATA)
		914	const_iv (SEEK_HOLE)
		915
		916	const_niv (FADV_NORMAL , POSIX_FADV_NORMAL)
		917	const_niv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
		918	const_niv (FADV_RANDOM , POSIX_FADV_RANDOM)
		919	const_niv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
		920	const_niv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
		921	const_niv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
		922
		923	const_niv (MADV_NORMAL , POSIX_MADV_NORMAL)
		924	const_niv (MADV_SEQUENTIAL, POSIX_MADV_SEQUENTIAL)
		925	const_niv (MADV_RANDOM , POSIX_MADV_RANDOM)
		926	const_niv (MADV_WILLNEED , POSIX_MADV_WILLNEED)
		927	const_niv (MADV_DONTNEED , POSIX_MADV_DONTNEED)
		928
		929	/* the second block will be 0 when missing */
		930	const_iv (O_RDONLY)
		931	const_iv (O_WRONLY)
		932	const_iv (O_RDWR)
		933	const_iv (O_CREAT)
		934	const_iv (O_TRUNC)
		935	const_iv (O_EXCL)
		936	const_iv (O_APPEND)
		937
		938	const_iv (O_ASYNC)
		939	const_iv (O_DIRECT)
		940	const_iv (O_NOATIME)
		941
		942	const_iv (O_CLOEXEC)
		943	const_iv (O_NOCTTY)
		944	const_iv (O_NOFOLLOW)
		945	const_iv (O_NONBLOCK)
		946	const_iv (O_EXEC)
		947	const_iv (O_SEARCH)
		948	const_iv (O_DIRECTORY)
		949	const_iv (O_DSYNC)
		950	const_iv (O_RSYNC)
		951	const_iv (O_SYNC)
		952	const_iv (O_PATH)
		953	const_iv (O_TMPFILE)
		954	const_iv (O_TTY_INIT)
		955
		956	const_iv (S_IFIFO)
		957	const_iv (S_IFCHR)
		958	const_iv (S_IFBLK)
		959	const_iv (S_IFLNK)
		960	const_iv (S_IFREG)
		961	const_iv (S_IFDIR)
		962	const_iv (S_IFWHT)
		963	const_iv (S_IFSOCK)
		964	const_iv (S_IFMT)
		965
		966	const_iv (ST_RDONLY)
		967	const_iv (ST_NOSUID)
		968	const_iv (ST_NODEV)
		969	const_iv (ST_NOEXEC)
		970	const_iv (ST_SYNCHRONOUS)
		971	const_iv (ST_MANDLOCK)
		972	const_iv (ST_WRITE)
		973	const_iv (ST_APPEND)
		974	const_iv (ST_IMMUTABLE)
		975	const_iv (ST_NOATIME)
		976	const_iv (ST_NODIRATIME)
		977	const_iv (ST_RELATIME)
		978
		979	const_iv (PROT_NONE)
		980	const_iv (PROT_EXEC)
		981	const_iv (PROT_READ)
		982	const_iv (PROT_WRITE)
		983
		984	const_iv (MAP_PRIVATE)
		985	const_iv (MAP_SHARED)
		986	const_iv (MAP_FIXED)
		987	const_iv (MAP_ANONYMOUS)
		988
		989	/* linuxish */
		990	const_iv (MAP_HUGETLB)
		991	const_iv (MAP_LOCKED)
		992	const_iv (MAP_NORESERVE)
		993	const_iv (MAP_POPULATE)
		994	const_iv (MAP_NONBLOCK)
		995	const_iv (MAP_GROWSDOWN)
		996	const_iv (MAP_32BIT)
		997	const_iv (MAP_HUGETLB)
		998	const_iv (MAP_STACK)
		999
		1000	const_iv (FIEMAP_FLAG_SYNC)
		1001	const_iv (FIEMAP_FLAG_XATTR)
		1002	const_iv (FIEMAP_FLAGS_COMPAT)
		1003	const_iv (FIEMAP_EXTENT_LAST)
		1004	const_iv (FIEMAP_EXTENT_UNKNOWN)
		1005	const_iv (FIEMAP_EXTENT_DELALLOC)
		1006	const_iv (FIEMAP_EXTENT_ENCODED)
		1007	const_iv (FIEMAP_EXTENT_DATA_ENCRYPTED)
		1008	const_iv (FIEMAP_EXTENT_NOT_ALIGNED)
		1009	const_iv (FIEMAP_EXTENT_DATA_INLINE)
		1010	const_iv (FIEMAP_EXTENT_DATA_TAIL)
		1011	const_iv (FIEMAP_EXTENT_UNWRITTEN)
		1012	const_iv (FIEMAP_EXTENT_MERGED)
		1013	const_iv (FIEMAP_EXTENT_SHARED)
		1014
		1015	const_iv (SPLICE_F_MOVE)
		1016	const_iv (SPLICE_F_NONBLOCK)
		1017	const_iv (SPLICE_F_MORE)
		1018	const_iv (SPLICE_F_GIFT)
		1019
		1020	/* these are libeio constants, and are independent of gendef0 */
		1021	const_eio (SEEK_SET)
		1022	const_eio (SEEK_CUR)
		1023	const_eio (SEEK_END)
		1024
		1025	const_eio (MCL_FUTURE)
		1026	const_eio (MCL_CURRENT)
		1027
		1028	const_eio (MS_ASYNC)
		1029	const_eio (MS_INVALIDATE)
		1030	const_eio (MS_SYNC)
		1031
		1032	const_eio (MT_MODIFY)
		1033
		1034	const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
		1035	const_eio (SYNC_FILE_RANGE_WRITE)
		1036	const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
		1037
		1038	const_eio (FALLOC_FL_KEEP_SIZE)
		1039	const_eio (FALLOC_FL_PUNCH_HOLE)
		1040	const_eio (FALLOC_FL_COLLAPSE_RANGE)
		1041	const_eio (FALLOC_FL_ZERO_RANGE)
		1042
		1043	const_eio (READDIR_DENTS)
		1044	const_eio (READDIR_DIRS_FIRST)
		1045	const_eio (READDIR_STAT_ORDER)
		1046	const_eio (READDIR_FOUND_UNKNOWN)
		1047
		1048	const_eio (DT_UNKNOWN)
		1049	const_eio (DT_FIFO)
		1050	const_eio (DT_CHR)
		1051	const_eio (DT_DIR)
		1052	const_eio (DT_BLK)
		1053	const_eio (DT_REG)
		1054	const_eio (DT_LNK)
		1055	const_eio (DT_SOCK)
		1056	const_eio (DT_WHT)
		1057	};
		1058
788	HV *stash = gv_stashpv ("IO::AIO", 1);	1059	aio_stash = gv_stashpv ("IO::AIO" , 1);
789	newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));	1060	aio_req_stash = gv_stashpv ("IO::AIO::REQ", 1);
790	newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));	1061	aio_grp_stash = gv_stashpv ("IO::AIO::GRP", 1);
791	newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));	1062	aio_wd_stash = gv_stashpv ("IO::AIO::WD" , 1);
792		1063
793	create_pipe ();	1064	for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ > const_iv; civ--)
794	pthread_atfork (atfork_prepare, atfork_parent, atfork_child);	1065	newCONSTSUB (aio_stash, (char *)civ[-1].name, newSViv (civ[-1].iv));
795	}
796		1066
		1067	newCONSTSUB (aio_stash, "PAGESIZE", newSViv (PAGESIZE));
		1068
		1069	reinit ();
		1070	}
		1071
797	void	1072	void
798	min_parallel (nthreads)	1073	reinit ()
799	int nthreads	1074	PROTOTYPE:
		1075
		1076	void
		1077	max_poll_reqs (unsigned int nreqs)
800	PROTOTYPE: $	1078	PROTOTYPE: $
		1079	CODE:
		1080	eio_set_max_poll_reqs (nreqs);
801		1081
802	void	1082	void
803	max_parallel (nthreads)	1083	max_poll_time (double nseconds)
804	int nthreads
805	PROTOTYPE: $	1084	PROTOTYPE: $
806
807	int
808	max_outstanding (nreqs)
809	int nreqs
810	PROTOTYPE: $
811	CODE:	1085	CODE:
812	RETVAL = max_outstanding;	1086	eio_set_max_poll_time (nseconds);
		1087
		1088	void
		1089	min_parallel (unsigned int nthreads)
		1090	PROTOTYPE: $
		1091	CODE:
		1092	eio_set_min_parallel (nthreads);
		1093
		1094	void
		1095	max_parallel (unsigned int nthreads)
		1096	PROTOTYPE: $
		1097	CODE:
		1098	eio_set_max_parallel (nthreads);
		1099
		1100	void
		1101	max_idle (unsigned int nthreads)
		1102	PROTOTYPE: $
		1103	CODE:
		1104	eio_set_max_idle (nthreads);
		1105
		1106	void
		1107	idle_timeout (unsigned int seconds)
		1108	PROTOTYPE: $
		1109	CODE:
		1110	eio_set_idle_timeout (seconds);
		1111
		1112	void
		1113	max_outstanding (unsigned int maxreqs)
		1114	PROTOTYPE: $
		1115	CODE:
813	max_outstanding = nreqs;	1116	max_outstanding = maxreqs;
814		1117
815	void	1118	void
816	aio_open (pathname,flags,mode,callback=&PL_sv_undef)	1119	aio_wd (SV8 *pathname, SV *callback = &PL_sv_undef)
817	SV * pathname
818	int flags
819	int mode
820	SV * callback
821	PROTOTYPE: $$$;$
822	CODE:	1120	PPCODE:
823	{	1121	{
824	dREQ;	1122	dREQ;
825		1123
		1124	req->type = EIO_WD_OPEN;
		1125	req_set_path1 (req, pathname);
		1126
		1127	REQ_SEND;
		1128	}
		1129
		1130	void
		1131	aio_open (SV8 *pathname, int flags, int mode, SV *callback = &PL_sv_undef)
		1132	PPCODE:
		1133	{
		1134	dREQ;
		1135
826	req->type = REQ_OPEN;	1136	req->type = EIO_OPEN;
827	req->data = newSVsv (pathname);	1137	req_set_path1 (req, pathname);
828	req->dataptr = SvPVbyte_nolen (req->data);
829	req->fd = flags;	1138	req->int1 = flags;
830	req->mode = mode;	1139	req->int2 = mode;
831		1140
832	send_req (req);	1141	REQ_SEND;
833	}	1142	}
834		1143
835	void	1144	void
836	aio_close (fh,callback=&PL_sv_undef)	1145	aio_fsync (SV *fh, SV *callback = &PL_sv_undef)
837	SV * fh
838	SV * callback
839	PROTOTYPE: $;$
840	ALIAS:	1146	ALIAS:
841	aio_close = REQ_CLOSE
842	aio_fsync = REQ_FSYNC	1147	aio_fsync = EIO_FSYNC
843	aio_fdatasync = REQ_FDATASYNC	1148	aio_fdatasync = EIO_FDATASYNC
		1149	aio_syncfs = EIO_SYNCFS
844	CODE:	1150	PPCODE:
845	{	1151	{
		1152	int fd = s_fileno_croak (fh, 0);
846	dREQ;	1153	dREQ;
847		1154
848	req->type = ix;	1155	req->type = ix;
849	req->fh = newSVsv (fh);	1156	req->sv1 = newSVsv (fh);
850	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));	1157	req->int1 = fd;
851		1158
852	send_req (req);	1159	REQ_SEND;
853	}	1160	}
854		1161
855	void	1162	void
		1163	aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback = &PL_sv_undef)
		1164	PPCODE:
		1165	{
		1166	int fd = s_fileno_croak (fh, 0);
		1167	dREQ;
		1168
		1169	req->type = EIO_SYNC_FILE_RANGE;
		1170	req->sv1 = newSVsv (fh);
		1171	req->int1 = fd;
		1172	req->offs = offset;
		1173	req->size = nbytes;
		1174	req->int2 = flags;
		1175
		1176	REQ_SEND;
		1177	}
		1178
		1179	void
		1180	aio_allocate (SV *fh, int mode, off_t offset, size_t len, SV *callback = &PL_sv_undef)
		1181	PPCODE:
		1182	{
		1183	int fd = s_fileno_croak (fh, 0);
		1184	dREQ;
		1185
		1186	req->type = EIO_FALLOCATE;
		1187	req->sv1 = newSVsv (fh);
		1188	req->int1 = fd;
		1189	req->int2 = mode;
		1190	req->offs = offset;
		1191	req->size = len;
		1192
		1193	REQ_SEND;
		1194	}
		1195
		1196	void
		1197	aio_close (SV *fh, SV *callback = &PL_sv_undef)
		1198	PPCODE:
		1199	{
		1200	static int close_fd = -1; /* dummy fd to close fds via dup2 */
		1201	int fd = s_fileno_croak (fh, 0);
		1202	dREQ;
		1203
		1204	if (expect_false (close_fd < 0))
		1205	{
		1206	int pipefd [2];
		1207
		1208	if (
		1209	#ifdef _WIN32
		1210	_pipe (pipefd, 1, _O_BINARY) < 0
		1211	#else
		1212	pipe (pipefd) < 0
		1213	|| fcntl (pipefd [0], F_SETFD, FD_CLOEXEC) < 0
		1214	#endif
		1215	|| close (pipefd [1]) < 0
		1216	)
		1217	abort (); /*D*/
		1218
		1219	close_fd = pipefd [0];
		1220	}
		1221
		1222	req->type = EIO_DUP2;
		1223	req->int1 = close_fd;
		1224	req->sv2 = newSVsv (fh);
		1225	req->int2 = fd;
		1226
		1227	REQ_SEND;
		1228	}
		1229
		1230	void
		1231	aio_seek (SV *fh, SV *offset, int whence, SV *callback = &PL_sv_undef)
		1232	PPCODE:
		1233	{
		1234	int fd = s_fileno_croak (fh, 0);
		1235	dREQ;
		1236
		1237	req->type = EIO_SEEK;
		1238	req->sv1 = newSVsv (fh);
		1239	req->int1 = fd;
		1240	req->offs = SvVAL64 (offset);
		1241	req->int2 = whence;
		1242
		1243	REQ_SEND;
		1244	}
		1245
		1246	void
856	aio_read (fh,offset,length,data,dataoffset,callback=&PL_sv_undef)	1247	aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback = &PL_sv_undef)
857	SV * fh
858	UV offset
859	UV length
860	SV * data
861	UV dataoffset
862	SV * callback
863	ALIAS:	1248	ALIAS:
864	aio_read = REQ_READ	1249	aio_read = EIO_READ
865	aio_write = REQ_WRITE	1250	aio_write = EIO_WRITE
866	PROTOTYPE: $$$$$;$
867	CODE:	1251	PPCODE:
868	{	1252	{
869	aio_req req;
870	STRLEN svlen;	1253	STRLEN svlen;
		1254	int fd = s_fileno_croak (fh, ix == EIO_WRITE);
871	char *svptr = SvPVbyte (data, svlen);	1255	char *svptr = SvPVbyte (data, svlen);
872		1256	UV len = SvUV (length);
873	SvUPGRADE (data, SVt_PV);
874	SvPOK_on (data);
875		1257
876	if (dataoffset < 0)	1258	if (dataoffset < 0)
877	dataoffset += svlen;	1259	dataoffset += svlen;
878		1260
879	if (dataoffset < 0 || dataoffset > svlen)	1261	if (dataoffset < 0 || dataoffset > svlen)
880	croak ("data offset outside of string");	1262	croak ("dataoffset outside of data scalar");
881		1263
882	if (ix == REQ_WRITE)	1264	if (ix == EIO_WRITE)
883	{	1265	{
884	/* write: check length and adjust. */	1266	/* write: check length and adjust. */
885	if (length < 0 || length + dataoffset > svlen)	1267	if (!SvOK (length) || len + dataoffset > svlen)
886	length = svlen - dataoffset;	1268	len = svlen - dataoffset;
887	}	1269	}
888	else	1270	else
889	{	1271	{
890	/* read: grow scalar as necessary */	1272	/* read: check type and grow scalar as necessary */
		1273	if (!SvPOK (data) || SvLEN (data) >= SvCUR (data))
891	svptr = SvGROW (data, length + dataoffset);	1274	svptr = sv_grow (data, len + dataoffset + 1);
		1275	else if (SvCUR (data) < len + dataoffset)
		1276	croak ("length + dataoffset outside of scalar, and cannot grow");
892	}	1277	}
893
894	if (length < 0)
895	croak ("length must not be negative");
896		1278
897	{	1279	{
898	dREQ;	1280	dREQ;
899		1281
900	req->type = ix;	1282	req->type = ix;
901	req->fh = newSVsv (fh);	1283	req->sv1 = newSVsv (fh);
902	req->fd = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))	1284	req->int1 = fd;
903	: IoOFP (sv_2io (fh)));	1285	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
904	req->offset = offset;	1286	req->size = len;
905	req->length = length;
906	req->data = SvREFCNT_inc (data);	1287	req->sv2 = SvREFCNT_inc (data);
907	req->dataptr = (char *)svptr + dataoffset;	1288	req->ptr2 = (char *)svptr + dataoffset;
		1289	req->stroffset = dataoffset;
908		1290
909	if (!SvREADONLY (data))	1291	if (!SvREADONLY (data))
910	{	1292	{
911	SvREADONLY_on (data);	1293	SvREADONLY_on (data);
912	req->data2ptr = (void *)data;	1294	req->flags |= FLAG_SV2_RO_OFF;
913	}	1295	}
914		1296
915	send_req (req);	1297	REQ_SEND;
916	}	1298	}
917	}	1299	}
918		1300
919	void	1301	void
920	aio_sendfile (out_fh,in_fh,in_offset,length,callback=&PL_sv_undef)	1302	aio_readlink (SV8 *pathname, SV *callback = &PL_sv_undef)
921	SV * out_fh	1303	ALIAS:
922	SV * in_fh	1304	aio_readlink = EIO_READLINK
923	UV in_offset	1305	aio_realpath = EIO_REALPATH
924	UV length
925	SV * callback
926	PROTOTYPE: $$$$;$
927	CODE:	1306	PPCODE:
928	{	1307	{
		1308	dREQ;
		1309
		1310	req->type = ix;
		1311	req_set_path1 (req, pathname);
		1312
		1313	REQ_SEND;
		1314	}
		1315
		1316	void
		1317	aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback = &PL_sv_undef)
		1318	PPCODE:
		1319	{
		1320	int ifd = s_fileno_croak (in_fh , 0);
		1321	int ofd = s_fileno_croak (out_fh, 1);
929	dREQ;	1322	dREQ;
930		1323
931	req->type = REQ_SENDFILE;	1324	req->type = EIO_SENDFILE;
932	req->fh = newSVsv (out_fh);	1325	req->sv1 = newSVsv (out_fh);
933	req->fd = PerlIO_fileno (IoIFP (sv_2io (out_fh)));	1326	req->int1 = ofd;
934	req->fh2 = newSVsv (in_fh);	1327	req->sv2 = newSVsv (in_fh);
935	req->fd2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));	1328	req->int2 = ifd;
936	req->offset = in_offset;	1329	req->offs = in_offset;
937	req->length = length;	1330	req->size = length;
938		1331
939	send_req (req);	1332	REQ_SEND;
940	}	1333	}
941		1334
942	void	1335	void
943	aio_readahead (fh,offset,length,callback=&PL_sv_undef)	1336	aio_readahead (SV *fh, off_t offset, size_t length, SV *callback = &PL_sv_undef)
944	SV * fh
945	UV offset
946	IV length
947	SV * callback
948	PROTOTYPE: $$$;$
949	CODE:	1337	PPCODE:
950	{	1338	{
		1339	int fd = s_fileno_croak (fh, 0);
951	dREQ;	1340	dREQ;
952		1341
953	req->type = REQ_READAHEAD;	1342	req->type = EIO_READAHEAD;
954	req->fh = newSVsv (fh);	1343	req->sv1 = newSVsv (fh);
955	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));	1344	req->int1 = fd;
956	req->offset = offset;	1345	req->offs = offset;
957	req->length = length;	1346	req->size = length;
958		1347
959	send_req (req);	1348	REQ_SEND;
960	}	1349	}
961		1350
962	void	1351	void
963	aio_stat (fh_or_path,callback=&PL_sv_undef)	1352	aio_stat (SV8 *fh_or_path, SV *callback = &PL_sv_undef)
964	SV * fh_or_path
965	SV * callback
966	ALIAS:	1353	ALIAS:
967	aio_stat = REQ_STAT	1354	aio_stat = EIO_STAT
968	aio_lstat = REQ_LSTAT	1355	aio_lstat = EIO_LSTAT
		1356	aio_statvfs = EIO_STATVFS
		1357	PPCODE:
		1358	{
		1359	dREQ;
		1360
		1361	req_set_fh_or_path (req, ix, ix == EIO_STATVFS ? EIO_FSTATVFS : EIO_FSTAT, fh_or_path);
		1362
		1363	REQ_SEND;
		1364	}
		1365
		1366	UV
		1367	major (UV dev)
		1368	ALIAS:
		1369	minor = 1
969	CODE:	1370	CODE:
		1371	RETVAL = ix ? major (dev) : minor (dev);
		1372	OUTPUT:
		1373	RETVAL
		1374
		1375	UV
		1376	makedev (UV maj, UV min)
		1377	CODE:
		1378	RETVAL = makedev (maj, min);
		1379	OUTPUT:
		1380	RETVAL
		1381
		1382	void
		1383	aio_utime (SV8 *fh_or_path, SV *atime, SV *mtime, SV *callback = &PL_sv_undef)
		1384	PPCODE:
970	{	1385	{
971	dREQ;	1386	dREQ;
972		1387
973	New (0, req->statdata, 1, Stat_t);	1388	req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
974	if (!req->statdata)	1389	req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
975	{	1390	req_set_fh_or_path (req, EIO_UTIME, EIO_FUTIME, fh_or_path);
976	free_req (req);
977	croak ("out of memory during aio_req->statdata allocation");
978	}
979		1391
980	if (SvPOK (fh_or_path))	1392	REQ_SEND;
981	{
982	req->type = ix;
983	req->data = newSVsv (fh_or_path);
984	req->dataptr = SvPVbyte_nolen (req->data);
985	}
986	else
987	{
988	req->type = REQ_FSTAT;
989	req->fh = newSVsv (fh_or_path);
990	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
991	}
992
993	send_req (req);
994	}	1393	}
995		1394
996	void	1395	void
		1396	aio_truncate (SV8 *fh_or_path, SV *offset, SV *callback = &PL_sv_undef)
		1397	PPCODE:
		1398	{
		1399	dREQ;
		1400
		1401	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
		1402	req_set_fh_or_path (req, EIO_TRUNCATE, EIO_FTRUNCATE, fh_or_path);
		1403
		1404	REQ_SEND;
		1405	}
		1406
		1407	void
		1408	aio_chmod (SV8 *fh_or_path, int mode, SV *callback = &PL_sv_undef)
		1409	PPCODE:
		1410	{
		1411	dREQ;
		1412
		1413	req->int2 = mode;
		1414	req_set_fh_or_path (req, EIO_CHMOD, EIO_FCHMOD, fh_or_path);
		1415
		1416	REQ_SEND;
		1417	}
		1418
		1419	void
		1420	aio_chown (SV8 *fh_or_path, SV *uid, SV *gid, SV *callback = &PL_sv_undef)
		1421	PPCODE:
		1422	{
		1423	dREQ;
		1424
		1425	req->int2 = SvOK (uid) ? SvIV (uid) : -1;
		1426	req->int3 = SvOK (gid) ? SvIV (gid) : -1;
		1427	req_set_fh_or_path (req, EIO_CHOWN, EIO_FCHOWN, fh_or_path);
		1428
		1429	REQ_SEND;
		1430	}
		1431
		1432	void
		1433	aio_readdirx (SV8 *pathname, IV flags, SV *callback = &PL_sv_undef)
		1434	PPCODE:
		1435	{
		1436	dREQ;
		1437
		1438	req->type = EIO_READDIR;
		1439	req->int1 = flags | EIO_READDIR_DENTS | EIO_READDIR_CUSTOM1;
		1440
		1441	if (flags & EIO_READDIR_DENTS)
		1442	req->int1 |= EIO_READDIR_CUSTOM2;
		1443
		1444	req_set_path1 (req, pathname);
		1445
		1446	REQ_SEND;
		1447	}
		1448
		1449	void
		1450	aio_mkdir (SV8 *pathname, int mode, SV *callback = &PL_sv_undef)
		1451	PPCODE:
		1452	{
		1453	dREQ;
		1454
		1455	req->type = EIO_MKDIR;
		1456	req->int2 = mode;
		1457	req_set_path1 (req, pathname);
		1458
		1459	REQ_SEND;
		1460	}
		1461
		1462	void
997	aio_unlink (pathname,callback=&PL_sv_undef)	1463	aio_unlink (SV8 *pathname, SV *callback = &PL_sv_undef)
998	SV * pathname
999	SV * callback
1000	ALIAS:	1464	ALIAS:
1001	aio_unlink = REQ_UNLINK	1465	aio_unlink = EIO_UNLINK
1002	aio_rmdir = REQ_RMDIR	1466	aio_rmdir = EIO_RMDIR
1003	aio_readdir = REQ_READDIR	1467	aio_readdir = EIO_READDIR
1004	CODE:	1468	PPCODE:
1005	{	1469	{
1006	dREQ;	1470	dREQ;
1007		1471
1008	req->type = ix;	1472	req->type = ix;
1009	req->data = newSVsv (pathname);	1473	req_set_path1 (req, pathname);
1010	req->dataptr = SvPVbyte_nolen (req->data);
1011
1012	send_req (req);
1013	}
1014		1474
		1475	REQ_SEND;
		1476	}
		1477
1015	void	1478	void
1016	aio_link (oldpath,newpath,callback=&PL_sv_undef)	1479	aio_link (SV8 *oldpath, SV8 *newpath, SV *callback = &PL_sv_undef)
1017	SV * oldpath
1018	SV * newpath
1019	SV * callback
1020	ALIAS:	1480	ALIAS:
1021	aio_link = REQ_LINK	1481	aio_link = EIO_LINK
1022	aio_symlink = REQ_SYMLINK	1482	aio_symlink = EIO_SYMLINK
1023	aio_rename = REQ_RENAME	1483	aio_rename = EIO_RENAME
1024	CODE:	1484	PPCODE:
1025	{	1485	{
		1486	eio_wd wd2 = 0;
1026	dREQ;	1487	dREQ;
1027		1488
1028	req->type = ix;	1489	req->type = ix;
1029	req->fh = newSVsv (oldpath);	1490	req_set_path1 (req, oldpath);
1030	req->data2ptr = SvPVbyte_nolen (req->fh);	1491	req_set_path (req, newpath, &req->sv2, &req->sv4, &wd2, &req->ptr2);
1031	req->data = newSVsv (newpath);	1492	req->int3 = (long)wd2;
1032	req->dataptr = SvPVbyte_nolen (req->data);
1033		1493
1034	send_req (req);	1494	REQ_SEND;
1035	}	1495	}
		1496
		1497	void
		1498	aio_mknod (SV8 *pathname, int mode, UV dev, SV *callback = &PL_sv_undef)
		1499	PPCODE:
		1500	{
		1501	dREQ;
		1502
		1503	req->type = EIO_MKNOD;
		1504	req->int2 = (mode_t)mode;
		1505	req->offs = dev;
		1506	req_set_path1 (req, pathname);
		1507
		1508	REQ_SEND;
		1509	}
		1510
		1511	void
		1512	aio_mtouch (SV8 *data, IV offset = 0, SV *length = &PL_sv_undef, int flags = 0, SV *callback = &PL_sv_undef)
		1513	ALIAS:
		1514	aio_mtouch = EIO_MTOUCH
		1515	aio_msync = EIO_MSYNC
		1516	PPCODE:
		1517	{
		1518	STRLEN svlen;
		1519	char *svptr = SvPVbyte (data, svlen);
		1520	UV len = SvUV (length);
		1521
		1522	if (offset < 0)
		1523	offset += svlen;
		1524
		1525	if (offset < 0 || offset > svlen)
		1526	croak ("offset outside of scalar");
		1527
		1528	if (!SvOK (length) || len + offset > svlen)
		1529	len = svlen - offset;
		1530
		1531	{
		1532	dREQ;
		1533
		1534	req->type = ix;
		1535	req->sv2 = SvREFCNT_inc (data);
		1536	req->ptr2 = (char *)svptr + offset;
		1537	req->size = len;
		1538	req->int1 = flags;
		1539
		1540	REQ_SEND;
		1541	}
		1542	}
		1543
		1544	void
		1545	aio_mlock (SV8 *data, IV offset = 0, SV *length = &PL_sv_undef, SV *callback = &PL_sv_undef)
		1546	PPCODE:
		1547	{
		1548	STRLEN svlen;
		1549	char *svptr = SvPVbyte (data, svlen);
		1550	UV len = SvUV (length);
		1551
		1552	if (offset < 0)
		1553	offset += svlen;
		1554
		1555	if (offset < 0 || offset > svlen)
		1556	croak ("offset outside of scalar");
		1557
		1558	if (!SvOK (length) || len + offset > svlen)
		1559	len = svlen - offset;
		1560
		1561	{
		1562	dREQ;
		1563
		1564	req->type = EIO_MLOCK;
		1565	req->sv2 = SvREFCNT_inc (data);
		1566	req->ptr2 = (char *)svptr + offset;
		1567	req->size = len;
		1568
		1569	REQ_SEND;
		1570	}
		1571	}
		1572
		1573	void
		1574	aio_mlockall (IV flags, SV *callback = &PL_sv_undef)
		1575	PPCODE:
		1576	{
		1577	dREQ;
		1578
		1579	req->type = EIO_MLOCKALL;
		1580	req->int1 = flags;
		1581
		1582	REQ_SEND;
		1583	}
		1584
		1585	void
		1586	aio_fiemap (SV *fh, off_t start, SV *length, U32 flags, SV *count, SV *callback = &PL_sv_undef)
		1587	PPCODE:
		1588	{
		1589	int fd = s_fileno_croak (fh, 0);
		1590	dREQ;
		1591
		1592	req->type = EIO_CUSTOM;
		1593	req->sv1 = newSVsv (fh);
		1594	req->int1 = fd;
		1595
		1596	req->feed = fiemap;
		1597	#if HAVE_FIEMAP
		1598	/* keep our fingers crossed that the next two types are 64 bit */
		1599	req->offs = start;
		1600	req->size = SvOK (length) ? SvVAL64 (length) : ~0ULL;
		1601	req->int2 = flags;
		1602	req->int3 = SvOK (count) ? SvIV (count) : -1;
		1603	#endif
		1604
		1605	REQ_SEND;
		1606	}
		1607
		1608	void
		1609	aio_busy (double delay, SV *callback = &PL_sv_undef)
		1610	PPCODE:
		1611	{
		1612	dREQ;
		1613
		1614	req->type = EIO_BUSY;
		1615	req->nv1 = delay < 0. ? 0. : delay;
		1616
		1617	REQ_SEND;
		1618	}
		1619
		1620	void
		1621	aio_group (SV *callback = &PL_sv_undef)
		1622	PPCODE:
		1623	{
		1624	dREQ;
		1625
		1626	req->type = EIO_GROUP;
		1627
		1628	PUTBACK;
		1629	req_submit (req);
		1630	SPAGAIN;
		1631
		1632	XPUSHs (req_sv (req, aio_grp_stash));
		1633	}
		1634
		1635	void
		1636	aio_nop (SV *callback = &PL_sv_undef)
		1637	ALIAS:
		1638	aio_nop = EIO_NOP
		1639	aio_sync = EIO_SYNC
		1640	PPCODE:
		1641	{
		1642	dREQ;
		1643
		1644	req->type = ix;
		1645
		1646	REQ_SEND;
		1647	}
		1648
		1649	int
		1650	aioreq_pri (int pri = NO_INIT)
		1651	CODE:
		1652	RETVAL = next_pri;
		1653	if (items > 0)
		1654	{
		1655	if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
		1656	if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
		1657	next_pri = pri;
		1658	}
		1659	OUTPUT:
		1660	RETVAL
		1661
		1662	void
		1663	aioreq_nice (int nice = 0)
		1664	CODE:
		1665	nice = next_pri - nice;
		1666	if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
		1667	if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
		1668	next_pri = nice;
1036		1669
1037	void	1670	void
1038	flush ()	1671	flush ()
1039	PROTOTYPE:
1040	CODE:	1672	CODE:
1041	while (nreqs)	1673	while (eio_nreqs ())
1042	{	1674	{
1043	poll_wait ();	1675	poll_wait ();
1044	poll_cb ();	1676	poll_cb ();
1045	}	1677	}
1046		1678
1047	void	1679	int
1048	poll()	1680	poll ()
1049	PROTOTYPE:
1050	CODE:	1681	CODE:
1051	if (nreqs)
1052	{
1053	poll_wait ();	1682	poll_wait ();
1054	poll_cb ();	1683	RETVAL = poll_cb ();
1055	}
1056
1057	int
1058	poll_fileno()
1059	PROTOTYPE:
1060	CODE:
1061	RETVAL = respipe [0];
1062	OUTPUT:	1684	OUTPUT:
1063	RETVAL	1685	RETVAL
1064		1686
1065	int	1687	int
		1688	poll_fileno ()
		1689	CODE:
		1690	RETVAL = s_epipe_fd (&respipe);
		1691	OUTPUT:
		1692	RETVAL
		1693
		1694	int
1066	poll_cb(...)	1695	poll_cb (...)
1067	PROTOTYPE:	1696	PROTOTYPE:
1068	CODE:	1697	CODE:
1069	RETVAL = poll_cb ();	1698	RETVAL = poll_cb ();
1070	OUTPUT:	1699	OUTPUT:
1071	RETVAL	1700	RETVAL
1072		1701
1073	void	1702	void
1074	poll_wait()	1703	poll_wait ()
1075	PROTOTYPE:
1076	CODE:	1704	CODE:
1077	if (nreqs)
1078	poll_wait ();	1705	poll_wait ();
1079		1706
1080	int	1707	int
1081	nreqs()	1708	nreqs ()
1082	PROTOTYPE:
1083	CODE:	1709	CODE:
1084	RETVAL = nreqs;	1710	RETVAL = eio_nreqs ();
1085	OUTPUT:	1711	OUTPUT:
1086	RETVAL	1712	RETVAL
1087		1713
		1714	int
		1715	nready ()
		1716	CODE:
		1717	RETVAL = eio_nready ();
		1718	OUTPUT:
		1719	RETVAL
		1720
		1721	int
		1722	npending ()
		1723	CODE:
		1724	RETVAL = eio_npending ();
		1725	OUTPUT:
		1726	RETVAL
		1727
		1728	int
		1729	nthreads ()
		1730	CODE:
		1731	RETVAL = eio_nthreads ();
		1732	OUTPUT:
		1733	RETVAL
		1734
		1735	int
		1736	fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
		1737	CODE:
		1738	RETVAL = posix_fadvise (fh, offset, length, advice);
		1739	OUTPUT:
		1740	RETVAL
		1741
		1742	IV
		1743	sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
		1744	CODE:
		1745	RETVAL = eio_sendfile_sync (ofh, ifh, offset, count);
		1746	OUTPUT:
		1747	RETVAL
		1748
		1749	void
		1750	mmap (SV *scalar, STRLEN length, int prot, int flags, SV *fh = &PL_sv_undef, off_t offset = 0)
		1751	PPCODE:
		1752	sv_unmagic (scalar, MMAP_MAGIC);
		1753	{
		1754	int fd = SvOK (fh) ? s_fileno_croak (fh, flags & PROT_WRITE) : -1;
		1755	void *addr = (void *)mmap (0, length, prot, flags, fd, offset);
		1756	if (addr == (void *)-1)
		1757	XSRETURN_NO;
		1758
		1759	sv_force_normal (scalar);
		1760
		1761	/* we store the length in mg_obj, as namlen is I32 :/ */
		1762	sv_magicext (scalar, 0, MMAP_MAGIC, &mmap_vtbl, (char *)addr, 0)
		1763	->mg_obj = (SV *)length;
		1764
		1765	SvUPGRADE (scalar, SVt_PV); /* nop... */
		1766
		1767	if (!(prot & PROT_WRITE))
		1768	SvREADONLY_on (scalar);
		1769
		1770	if (SvLEN (scalar))
		1771	Safefree (SvPVX (scalar));
		1772
		1773	SvPVX (scalar) = (char *)addr;
		1774	SvCUR_set (scalar, length);
		1775	SvLEN_set (scalar, 0);
		1776	SvPOK_only (scalar);
		1777
		1778	XSRETURN_YES;
		1779	}
		1780
		1781	void
		1782	munmap (SV *scalar)
		1783	CODE:
		1784	sv_unmagic (scalar, MMAP_MAGIC);
		1785
		1786	int
		1787	madvise (SV *scalar, STRLEN offset = 0, SV *length = &PL_sv_undef, IV advice_or_prot)
		1788	ALIAS:
		1789	mprotect = 1
		1790	CODE:
		1791	{
		1792	STRLEN svlen;
		1793	void *addr = SvPVbyte (scalar, svlen);
		1794	STRLEN len = SvUV (length);
		1795
		1796	if (offset < 0)
		1797	offset += svlen;
		1798
		1799	if (offset < 0 || offset > svlen)
		1800	croak ("offset outside of scalar");
		1801
		1802	if (!SvOK (length) || len + offset > svlen)
		1803	len = svlen - offset;
		1804
		1805	addr = (void *)(((intptr_t)addr) + offset);
		1806	eio_page_align (&addr, &len);
		1807
		1808	switch (ix)
		1809	{
		1810	case 0: RETVAL = posix_madvise (addr, len, advice_or_prot); break;
		1811	case 1: RETVAL = mprotect (addr, len, advice_or_prot); break;
		1812	}
		1813	}
		1814	OUTPUT:
		1815	RETVAL
		1816
		1817	int
		1818	munlock (SV *scalar, STRLEN offset = 0, SV *length = &PL_sv_undef)
		1819	CODE:
		1820	{
		1821	STRLEN svlen;
		1822	void *addr = SvPVbyte (scalar, svlen);
		1823	size_t len = SvUV (length);
		1824
		1825	if (offset < 0)
		1826	offset += svlen;
		1827
		1828	if (offset < 0 || offset > svlen)
		1829	croak ("offset outside of scalar");
		1830
		1831	if (!SvOK (length) || len + offset > svlen)
		1832	len = svlen - offset;
		1833
		1834	addr = (void *)(((intptr_t)addr) + offset);
		1835	eio_page_align (&addr, &len);
		1836	#if _POSIX_MEMLOCK_RANGE
		1837	RETVAL = munlock (addr, len);
		1838	#else
		1839	RETVAL = EIO_ENOSYS ();
		1840	#endif
		1841	}
		1842	OUTPUT:
		1843	RETVAL
		1844
		1845	int
		1846	munlockall ()
		1847	CODE:
		1848	#if _POSIX_MEMLOCK
		1849	munlockall ();
		1850	#else
		1851	RETVAL = EIO_ENOSYS ();
		1852	#endif
		1853	OUTPUT:
		1854	RETVAL
		1855
		1856	int
		1857	splice (aio_rfd rfh, SV *off_in, aio_wfd wfh, SV *off_out, size_t length, unsigned int flags)
		1858	CODE:
		1859	{
		1860	#if HAVE_LINUX_SPLICE
		1861	loff_t off_in_, off_out_;
		1862	RETVAL = splice (
		1863	rfh, SvOK (off_in ) ? (off_in_ = SvVAL64 (off_in )), &off_in_ : 0,
		1864	wfh, SvOK (off_out) ? (off_out_ = SvVAL64 (off_out)), &off_out_ : 0,
		1865	length, flags
		1866	);
		1867	#else
		1868	RETVAL = EIO_ENOSYS ();
		1869	#endif
		1870	}
		1871	OUTPUT:
		1872	RETVAL
		1873
		1874	int
		1875	tee (aio_rfd rfh, aio_wfd wfh, size_t length, unsigned int flags)
		1876	CODE:
		1877	#if HAVE_LINUX_SPLICE
		1878	RETVAL = tee (rfh, wfh, length, flags);
		1879	#else
		1880	RETVAL = EIO_ENOSYS ();
		1881	#endif
		1882	OUTPUT:
		1883	RETVAL
		1884
		1885	int
		1886	pipesize (aio_rfd rfh, int new_size = -1)
		1887	PROTOTYPE: $;$
		1888	CODE:
		1889	#if defined(F_SETPIPE_SZ) && defined(F_GETPIPE_SZ)
		1890	if (new_size >= 0)
		1891	RETVAL = fcntl (rfh, F_SETPIPE_SZ, new_size);
		1892	else
		1893	RETVAL = fcntl (rfh, F_GETPIPE_SZ);
		1894	#else
		1895	errno = ENOSYS;
		1896	RETVAL = -1;
		1897	#endif
		1898	OUTPUT:
		1899	RETVAL
		1900
		1901	void
		1902	pipe2 (int flags = 0)
		1903	PROTOTYPE: ;$
		1904	PPCODE:
		1905	{
		1906	int fd[2];
		1907	int res;
		1908
		1909	if (flags)
		1910	#if HAVE_PIPE2
		1911	res = pipe2 (fd, flags);
		1912	#else
		1913	res = (errno = ENOSYS, -1);
		1914	#endif
		1915	else
		1916	res = pipe (fd);
		1917
		1918	if (!res)
		1919	{
		1920	EXTEND (SP, 2);
		1921	PUSHs (newmortalFH (fd[0], O_RDONLY));
		1922	PUSHs (newmortalFH (fd[1], O_WRONLY));
		1923	}
		1924	}
		1925
		1926	void _on_next_submit (SV *cb)
		1927	CODE:
		1928	SvREFCNT_dec (on_next_submit);
		1929	on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
		1930
		1931	PROTOTYPES: DISABLE
		1932
		1933	MODULE = IO::AIO PACKAGE = IO::AIO::WD
		1934
		1935	BOOT:
		1936	{
		1937	newCONSTSUB (aio_stash, "CWD" , newSVaio_wd (EIO_CWD ));
		1938	newCONSTSUB (aio_stash, "INVALID_WD", newSVaio_wd (EIO_INVALID_WD));
		1939	}
		1940
		1941	void
		1942	DESTROY (SV *self)
		1943	CODE:
		1944	{
		1945	aio_wd wd = SvAIO_WD (self);
		1946	#if HAVE_AT
		1947	{
		1948	SV *callback = &PL_sv_undef;
		1949	dREQ; /* clobbers next_pri :/ */
		1950	next_pri = req->pri; /* restore next_pri */
		1951	req->pri = EIO_PRI_MAX; /* better use max. priority to conserve fds */
		1952	req->type = EIO_WD_CLOSE;
		1953	req->wd = wd;
		1954	REQ_SEND;
		1955	}
		1956	#else
		1957	eio_wd_close_sync (wd);
		1958	#endif
		1959	}
		1960
		1961	MODULE = IO::AIO PACKAGE = IO::AIO::REQ
		1962
		1963	void
		1964	cancel (aio_req_ornot req)
		1965	CODE:
		1966	eio_cancel (req);
		1967
		1968	void
		1969	cb (aio_req_ornot req, SV *callback = NO_INIT)
		1970	PPCODE:
		1971	{
		1972	if (GIMME_V != G_VOID)
		1973	XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
		1974
		1975	if (items > 1)
		1976	{
		1977	SV *cb_cv = get_cb (callback);
		1978
		1979	SvREFCNT_dec (req->callback);
		1980	req->callback = SvREFCNT_inc (cb_cv);
		1981	}
		1982	}
		1983
		1984	MODULE = IO::AIO PACKAGE = IO::AIO::GRP
		1985
		1986	void
		1987	add (aio_req grp, ...)
		1988	PPCODE:
		1989	{
		1990	int i;
		1991
		1992	if (grp->int1 == 2)
		1993	croak ("cannot add requests to IO::AIO::GRP after the group finished");
		1994
		1995	for (i = 1; i < items; ++i )
		1996	{
		1997	aio_req req;
		1998
		1999	if (GIMME_V != G_VOID)
		2000	XPUSHs (sv_2mortal (newSVsv (ST (i))));
		2001
		2002	req = SvAIO_REQ (ST (i));
		2003
		2004	if (req)
		2005	eio_grp_add (grp, req);
		2006	}
		2007	}
		2008
		2009	void
		2010	cancel_subs (aio_req_ornot req)
		2011	CODE:
		2012	req_cancel_subs (req);
		2013
		2014	void
		2015	result (aio_req grp, ...)
		2016	CODE:
		2017	{
		2018	int i;
		2019	AV *av;
		2020
		2021	grp->errorno = errno;
		2022
		2023	av = newAV ();
		2024	av_extend (av, items - 1);
		2025
		2026	for (i = 1; i < items; ++i )
		2027	av_push (av, newSVsv (ST (i)));
		2028
		2029	SvREFCNT_dec (grp->sv1);
		2030	grp->sv1 = (SV *)av;
		2031	}
		2032
		2033	void
		2034	errno (aio_req grp, int errorno = errno)
		2035	CODE:
		2036	grp->errorno = errorno;
		2037
		2038	void
		2039	limit (aio_req grp, int limit)
		2040	CODE:
		2041	eio_grp_limit (grp, limit);
		2042
		2043	void
		2044	feed (aio_req grp, SV *callback = &PL_sv_undef)
		2045	CODE:
		2046	{
		2047	SvREFCNT_dec (grp->sv2);
		2048	grp->sv2 = newSVsv (callback);
		2049	grp->feed = aio_grp_feed;
		2050
		2051	if (grp->int2 <= 0)
		2052	grp->int2 = 2;
		2053
		2054	eio_grp_limit (grp, grp->int2);
		2055	}
		2056

Diff Legend

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