ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/IO-AIO/AIO.xs

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.2 by root, Sun Jul 10 18:16:49 2005 UTC vs.
Revision 1.127 by root, Tue Sep 30 13:17:32 2008 UTC

1	#define PERL_NO_GET_CONTEXT	1	#include "libeio/xthread.h"
		2
		3	#include <errno.h>
2		4
3	#include "EXTERN.h"	5	#include "EXTERN.h"
4	#include "perl.h"	6	#include "perl.h"
5	#include "XSUB.h"	7	#include "XSUB.h"
6		8
		9	#include <stddef.h>
		10	#include <stdlib.h>
		11	#include <errno.h>
7	#include <sys/types.h>	12	#include <sys/types.h>
8	#include <sys/stat.h>	13	#include <sys/stat.h>
9	#include <unistd.h>	14	#include <limits.h>
10	#include <fcntl.h>	15	#include <fcntl.h>
11	#include <signal.h>
12	#include <sched.h>	16	#include <sched.h>
13	#include <endian.h>
14		17
15	#include <pthread.h>	18	/* perl namespace pollution */
16	#include <sys/syscall.h>	19	#undef VERSION
17		20
18	typedef void *InputStream; /* hack, but 5.6.1 is simply toooo old ;) */	21	#ifdef _WIN32
19	typedef void *OutputStream; /* hack, but 5.6.1 is simply toooo old ;) */
20	typedef void *InOutStream; /* hack, but 5.6.1 is simply toooo old ;) */
21		22
22	#if __i386 || __amd64	23	# define EIO_STRUCT_DIRENT Direntry_t
23	# define STACKSIZE ( 256 * sizeof (long))	24	# undef malloc
24	#elif __ia64	25	# undef free
25	# define STACKSIZE (8192 * sizeof (long))	26
		27	// perl overrides all those nice win32 functions
		28	# undef open
		29	# undef read
		30	# undef write
		31	# undef send
		32	# undef recv
		33	# undef stat
		34	# undef fstat
		35	# define lstat stat
		36	# undef truncate
		37	# undef ftruncate
		38	# undef open
		39	# undef close
		40	# undef unlink
		41	# undef rmdir
		42	# undef rename
		43	# undef lseek
		44
		45	# define chown(a,b,c) (errno = ENOSYS, -1)
		46	# define fchown(a,b,c) (errno = ENOSYS, -1)
		47	# define fchmod(a,b) (errno = ENOSYS, -1)
		48	# define symlink(a,b) (errno = ENOSYS, -1)
		49	# define readlink(a,b,c) (errno = ENOSYS, -1)
		50	# define mknod(a,b,c) (errno = ENOSYS, -1)
		51	# define truncate(a,b) (errno = ENOSYS, -1)
		52	# define ftruncate(fd,o) chsize ((fd), (o))
		53	# define fsync(fd) _commit (fd)
		54	# define opendir(fd) (errno = ENOSYS, 0)
		55	# define readdir(fd) (errno = ENOSYS, -1)
		56	# define closedir(fd) (errno = ENOSYS, -1)
		57	# define mkdir(a,b) mkdir (a)
		58
26	#else	59	#else
27	# define STACKSIZE ( 512 * sizeof (long))	60
		61	# include <sys/time.h>
		62	# include <sys/select.h>
		63	# include <unistd.h>
		64	# include <utime.h>
		65	# include <signal.h>
		66	# define EIO_STRUCT_DIRENT struct dirent
		67
28	#endif	68	#endif
29		69
		70	/* perl stupidly overrides readdir and maybe others */
		71	/* with thread-unsafe versions, imagine that :( */
		72	#undef readdir
		73	#undef opendir
		74	#undef closedir
		75
		76	#define EIO_STRUCT_STAT Stat_t
		77
		78	/* use NV for 32 bit perls as it allows larger offsets */
		79	#if IVSIZE >= 8
		80	# define SvVAL64 SvIV
		81	#else
		82	# define SvVAL64 SvNV
		83	#endif
		84
		85	/*****************************************************************************/
		86
		87	#if __GNUC__ >= 3
		88	# define expect(expr,value) __builtin_expect ((expr),(value))
		89	#else
		90	# define expect(expr,value) (expr)
		91	#endif
		92
		93	#define expect_false(expr) expect ((expr) != 0, 0)
		94	#define expect_true(expr) expect ((expr) != 0, 1)
		95
		96	/*****************************************************************************/
		97
		98	static HV *stash;
		99	typedef SV SV8; /* byte-sv, used for argument-checking */
		100
		101	#define AIO_REQ_KLASS "IO::AIO::REQ"
		102	#define AIO_GRP_KLASS "IO::AIO::GRP"
		103
		104	#define EIO_REQ_MEMBERS \
		105	SV *callback; \
		106	SV *sv1, *sv2; \
		107	STRLEN stroffset; \
		108	SV *self;
		109
		110	#define EIO_NO_WRAPPERS 1
		111
		112	#include "libeio/eio.h"
		113
		114	static int req_invoke (eio_req *req);
		115	#define EIO_FINISH(req) req_invoke (req)
		116	static void aio_grp_feed (eio_req *grp);
		117	#define EIO_FEED(req) aio_grp_feed (req)
		118	static void req_destroy (eio_req *grp);
		119	#define EIO_DESTROY(req) req_destroy (req)
		120
30	enum {	121	enum {
31	REQ_QUIT,	122	FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
32	REQ_OPEN, REQ_CLOSE,
33	REQ_READ, REQ_WRITE, REQ_READAHEAD,
34	REQ_STAT, REQ_LSTAT, REQ_FSTAT, REQ_UNLINK,
35	REQ_FSYNC, REQ_FDATASYNC,
36	};	123	};
37		124
38	typedef struct aio_cb {	125	#include "libeio/eio.c"
39	struct aio_cb *next;
40		126
41	int type;
42
43	int fd;
44	off_t offset;
45	size_t length;
46	ssize_t result;
47	mode_t mode; /* open */
48	int errorno;
49	SV *data, *callback;
50	void *dataptr;
51	STRLEN dataoffset;
52
53	Stat_t *statdata;
54	} aio_cb;
55
56	typedef aio_cb *aio_req;	127	typedef eio_req *aio_req;
		128	typedef eio_req *aio_req_ornot;
57		129
58	static int started;	130	static SV *on_next_submit;
59	static int nreqs;	131	static int next_pri = EIO_PRI_DEFAULT;
60	static int reqpipe[2], respipe[2];	132	static int max_outstanding;
61		133
62	static aio_req qs, qe; /* queue start, queue end */	134	static int respipe_osf [2], respipe [2] = { -1, -1 };
63		135
64	static void *aio_proc(void *arg);	136	static void req_destroy (aio_req req);
		137	static void req_cancel (aio_req req);
65		138
66	static void	139	static void want_poll (void)
67	start_thread (void)
68	{	140	{
69	sigset_t fullsigset, oldsigset;	141	/* write a dummy byte to the pipe so fh becomes ready */
70	pthread_t tid;	142	respipe_write (respipe_osf [1], (const void *)&respipe_osf, 1);
71	pthread_attr_t attr;
72
73	pthread_attr_init (&attr);
74	pthread_attr_setstacksize (&attr, STACKSIZE);
75	pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
76
77	sigfillset (&fullsigset);
78	sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
79
80	if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
81	started++;
82
83	sigprocmask (SIG_SETMASK, &oldsigset, 0);
84	}	143	}
85		144
86	static void	145	static void done_poll (void)
87	send_reqs (void)
88	{	146	{
89	/* this write is atomic */	147	/* read any signals sent by the worker threads */
90	while (qs && write (reqpipe[1], &qs, sizeof qs) == sizeof qs)	148	char buf [4];
91	{	149	while (respipe_read (respipe [0], buf, 4) == 4)
92	qs = qs->next;	150	;
93	if (!qs) qe = 0;
94	}
95	}	151	}
96		152
97	static void	153	/* must be called at most once */
98	send_req (aio_req req)	154	static SV *req_sv (aio_req req, const char *klass)
99	{	155	{
100	nreqs++;	156	if (!req->self)
101	req->next = 0;
102
103	if (qe)
104	{	157	{
105	qe->next = req;	158	req->self = (SV *)newHV ();
106	qe = req;	159	sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
107	}	160	}
108	else
109	qe = qs = req;
110		161
111	send_reqs ();	162	return sv_2mortal (sv_bless (newRV_inc (req->self), gv_stashpv (klass, 1)));
112	}	163	}
113		164
114	static void	165	static aio_req SvAIO_REQ (SV *sv)
115	end_thread (void)
116	{	166	{
117	aio_req req;	167	MAGIC *mg;
118	New (0, req, 1, aio_cb);
119	req->type = REQ_QUIT;
120		168
121	send_req (req);	169	if (!sv_derived_from (sv, AIO_REQ_KLASS) || !SvROK (sv))
122	}	170	croak ("object of class " AIO_REQ_KLASS " expected");
123		171
124	static void	172	mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
125	read_write (pTHX_
126	int dowrite, int fd, off_t offset, size_t length,
127	SV *data, STRLEN dataoffset, SV *callback)
128	{
129	aio_req req;
130	STRLEN svlen;
131	char *svptr = SvPV (data, svlen);
132		173
133	SvUPGRADE (data, SVt_PV);	174	return mg ? (aio_req)mg->mg_ptr : 0;
134	SvPOK_on (data);	175	}
135		176
136	if (dataoffset < 0)	177	static void aio_grp_feed (aio_req grp)
137	dataoffset += svlen;	178	{
138		179	if (grp->sv2 && SvOK (grp->sv2))
139	if (dataoffset < 0 || dataoffset > svlen)
140	croak ("data offset outside of string");
141
142	if (dowrite)
143	{	180	{
144	/* write: check length and adjust. */	181	dSP;
145	if (length < 0 || length + dataoffset > svlen)	182
146	length = svlen - dataoffset;	183	ENTER;
		184	SAVETMPS;
		185	PUSHMARK (SP);
		186	XPUSHs (req_sv (grp, AIO_GRP_KLASS));
		187	PUTBACK;
		188	call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
		189	SPAGAIN;
		190	FREETMPS;
		191	LEAVE;
147	}	192	}
148	else	193	}
		194
		195	static void req_submit (eio_req *req)
		196	{
		197	eio_submit (req);
		198
		199	if (expect_false (on_next_submit))
149	{	200	{
150	/* read: grow scalar as necessary */	201	dSP;
151	svptr = SvGROW (data, length + dataoffset);	202	SV *cb = sv_2mortal (on_next_submit);
		203
		204	on_next_submit = 0;
		205
		206	PUSHMARK (SP);
		207	PUTBACK;
		208	call_sv (cb, G_DISCARD | G_EVAL);
152	}	209	}
153
154	if (length < 0)
155	croak ("length must not be negative");
156
157	Newz (0, req, 1, aio_cb);
158
159	if (!req)
160	croak ("out of memory during aio_req allocation");
161
162	req->type = dowrite ? REQ_WRITE : REQ_READ;
163	req->fd = fd;
164	req->offset = offset;
165	req->length = length;
166	req->data = SvREFCNT_inc (data);
167	req->dataptr = (char *)svptr + dataoffset;
168	req->callback = SvREFCNT_inc (callback);
169
170	send_req (req);
171	}	210	}
172		211
173	static void	212	static int req_invoke (eio_req *req)
174	poll_wait ()
175	{
176	fd_set rfd;
177	FD_ZERO(&rfd);
178	FD_SET(respipe[0], &rfd);
179
180	select (respipe[0] + 1, &rfd, 0, 0, 0);
181	}
182
183	static int
184	poll_cb (pTHX)
185	{	213	{
186	dSP;	214	dSP;
187	int count = 0;
188	aio_req req;
189		215
190	while (read (respipe[0], (void *)&req, sizeof (req)) == sizeof (req))	216	if (req->flags & FLAG_SV2_RO_OFF)
		217	SvREADONLY_off (req->sv2);
		218
		219	if (!EIO_CANCELLED (req) && SvOK (req->callback))
191	{	220	{
192	nreqs--;	221	ENTER;
		222	SAVETMPS;
		223	PUSHMARK (SP);
		224	EXTEND (SP, 1);
193		225
194	if (req->type == REQ_QUIT)	226	switch (req->type)
195	started--;
196	else
197	{	227	{
198	int errorno = errno;	228	case EIO_READDIR:
199	errno = req->errorno;
200
201	if (req->type == REQ_READ)
202	SvCUR_set (req->data, req->dataoffset
203	+ req->result > 0 ? req->result : 0);
204
205	if (req->data)
206	SvREFCNT_dec (req->data);
207
208	if (req->type == REQ_STAT || req->type == REQ_LSTAT || req->type == REQ_FSTAT)
209	{	229	{
210	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;	230	SV *rv = &PL_sv_undef;
211	PL_laststatval = req->result;
212	PL_statcache = *(req->statdata);
213		231
214	Safefree (req->statdata);	232	if (req->result >= 0)
		233	{
		234	int i;
		235	char *buf = req->ptr2;
		236	AV *av = newAV ();
		237
		238	av_extend (av, req->result - 1);
		239
		240	for (i = 0; i < req->result; ++i)
		241	{
		242	SV *sv = newSVpv (buf, 0);
		243
		244	av_store (av, i, sv);
		245	buf += SvCUR (sv) + 1;
		246	}
		247
		248	rv = sv_2mortal (newRV_noinc ((SV *)av));
		249	}
		250
		251	PUSHs (rv);
215	}	252	}
		253	break;
216		254
217	PUSHMARK (SP);	255	case EIO_OPEN:
218	XPUSHs (sv_2mortal (newSViv (req->result)));
219
220	if (req->type == REQ_OPEN)
221	{	256	{
222	/* convert fd to fh */	257	/* convert fd to fh */
223	SV *fh;	258	SV *fh = &PL_sv_undef;
224		259
225	PUTBACK;	260	if (req->result >= 0)
226	call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);	261	{
227	SPAGAIN;	262	GV *gv = (GV *)sv_newmortal ();
		263	int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
		264	char sym [64];
		265	int symlen;
		266
		267	symlen = snprintf (sym, sizeof (sym), "fd#%d", req->result);
		268	gv_init (gv, stash, sym, symlen, 0);
228		269
229	fh = POPs;	270	symlen = snprintf (
		271	sym,
		272	sizeof (sym),
		273	"%s&=%d",
		274	flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
		275	req->result
		276	);
230		277
231	PUSHMARK (SP);	278	if (do_open (gv, sym, symlen, 0, 0, 0, 0))
		279	fh = (SV *)gv;
		280	}
		281
232	XPUSHs (fh);	282	PUSHs (fh);
233	}	283	}
		284	break;
234		285
235	PUTBACK;	286	case EIO_GROUP:
236	call_sv (req->callback, G_VOID | G_EVAL);	287	req->int1 = 2; /* mark group as finished */
237	SPAGAIN;	288
		289	if (req->sv1)
238		290	{
239	if (req->callback)	291	int i;
240	SvREFCNT_dec (req->callback);	292	AV *av = (AV *)req->sv1;
241		293
242	errno = errorno;	294	EXTEND (SP, AvFILL (av) + 1);
243	count++;	295	for (i = 0; i <= AvFILL (av); ++i)
		296	PUSHs (*av_fetch (av, i, 0));
244	}	297	}
		298	break;
245		299
246	Safefree (req);	300	case EIO_NOP:
247	}
248
249	if (qs)
250	send_reqs ();
251
252	return count;
253	}
254
255	static void *
256	aio_proc (void *thr_arg)
257	{
258	aio_req req;
259
260	/* then loop */
261	while (read (reqpipe[0], (void *)&req, sizeof (req)) == sizeof (req))
262	{
263	errno = 0; /* strictly unnecessary */
264
265	switch (req->type)
266	{
267	case REQ_READ: req->result = pread64 (req->fd, req->dataptr, req->length, req->offset); break;
268	case REQ_WRITE: req->result = pwrite64 (req->fd, req->dataptr, req->length, req->offset); break;
269	#if SYS_readahead
270	case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
271	#else
272	case REQ_READAHEAD: req->result = -1; errno = ENOSYS; break;
273	#endif
274
275	case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
276	case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
277	case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
278
279	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
280	case REQ_CLOSE: req->result = close (req->fd); break;
281	case REQ_UNLINK: req->result = unlink (req->dataptr); break;
282
283	case REQ_FSYNC: req->result = fsync (req->fd); break;
284	case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
285
286	case REQ_QUIT:	301	case EIO_BUSY:
287	write (respipe[1], (void *)&req, sizeof (req));	302	break;
288	return 0;	303
		304	case EIO_READLINK:
		305	if (req->result > 0)
		306	PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
		307	break;
		308
		309	case EIO_STAT:
		310	case EIO_LSTAT:
		311	case EIO_FSTAT:
		312	PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT;
		313	PL_laststatval = req->result;
		314	PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
		315	PUSHs (sv_2mortal (newSViv (req->result)));
		316	break;
		317
		318	case EIO_READ:
		319	SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
		320	*SvEND (req->sv2) = 0;
		321	PUSHs (sv_2mortal (newSViv (req->result)));
		322	break;
		323
		324	case EIO_DUP2:
		325	if (req->result > 0)
		326	req->result = 0;
		327	/* FALLTHROUGH */
289		328
290	default:	329	default:
291	req->result = ENOSYS;	330	PUSHs (sv_2mortal (newSViv (req->result)));
292	break;	331	break;
293	}	332	}
294		333
295	req->errorno = errno;	334	errno = req->errorno;
296	write (respipe[1], (void *)&req, sizeof (req));	335
		336	PUTBACK;
		337	call_sv (req->callback, G_VOID | G_EVAL | G_DISCARD);
		338	SPAGAIN;
		339
		340	FREETMPS;
		341	LEAVE;
		342
		343	PUTBACK;
297	}	344	}
298		345
		346	return !!SvTRUE (ERRSV);
		347	}
		348
		349	static void req_destroy (aio_req req)
		350	{
		351	if (req->self)
		352	{
		353	sv_unmagic (req->self, PERL_MAGIC_ext);
		354	SvREFCNT_dec (req->self);
		355	}
		356
		357	SvREFCNT_dec (req->sv1);
		358	SvREFCNT_dec (req->sv2);
		359	SvREFCNT_dec (req->callback);
		360
		361	Safefree (req);
		362	}
		363
		364	static void req_cancel_subs (aio_req grp)
		365	{
		366	aio_req sub;
		367
		368	if (grp->type != EIO_GROUP)
299	return 0;	369	return;
300	}
301		370
		371	SvREFCNT_dec (grp->sv2);
		372	grp->sv2 = 0;
		373
		374	eio_grp_cancel (grp);
		375	}
		376
		377	#ifdef USE_SOCKETS_AS_HANDLES
		378	# define TO_SOCKET(x) (win32_get_osfhandle (x))
		379	#else
		380	# define TO_SOCKET(x) (x)
		381	#endif
		382
		383	static void
		384	create_respipe (void)
		385	{
		386	int old_readfd = respipe [0];
		387
		388	if (respipe [1] >= 0)
		389	respipe_close (TO_SOCKET (respipe [1]));
		390
		391	#ifdef _WIN32
		392	if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, respipe))
		393	#else
		394	if (pipe (respipe))
		395	#endif
		396	croak ("unable to initialize result pipe");
		397
		398	if (old_readfd >= 0)
		399	{
		400	if (dup2 (TO_SOCKET (respipe [0]), TO_SOCKET (old_readfd)) < 0)
		401	croak ("unable to initialize result pipe(2)");
		402
		403	respipe_close (respipe [0]);
		404	respipe [0] = old_readfd;
		405	}
		406
		407	#ifdef _WIN32
		408	int arg = 1;
		409	if (ioctlsocket (TO_SOCKET (respipe [0]), FIONBIO, &arg)
		410	|| ioctlsocket (TO_SOCKET (respipe [1]), FIONBIO, &arg))
		411	#else
		412	if (fcntl (respipe [0], F_SETFL, O_NONBLOCK)
		413	|| fcntl (respipe [1], F_SETFL, O_NONBLOCK))
		414	#endif
		415	croak ("unable to initialize result pipe(3)");
		416
		417	respipe_osf [0] = TO_SOCKET (respipe [0]);
		418	respipe_osf [1] = TO_SOCKET (respipe [1]);
		419	}
		420
		421	static void poll_wait (void)
		422	{
		423	fd_set rfd;
		424
		425	while (eio_nreqs ())
		426	{
		427	int size;
		428
		429	X_LOCK (reslock);
		430	size = res_queue.size;
		431	X_UNLOCK (reslock);
		432
		433	if (size)
		434	return;
		435
		436	etp_maybe_start_thread ();
		437
		438	FD_ZERO (&rfd);
		439	FD_SET (respipe [0], &rfd);
		440
		441	PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);
		442	}
		443	}
		444
		445	static int poll_cb (void)
		446	{
		447	for (;;)
		448	{
		449	int res = eio_poll ();
		450
		451	if (res > 0)
		452	croak (0);
		453
		454	if (!max_outstanding || max_outstanding > eio_nreqs ())
		455	return res;
		456
		457	poll_wait ();
		458	}
		459	}
		460
		461	static void atfork_child (void)
		462	{
		463	create_respipe ();
		464	}
		465
		466	#define dREQ \
		467	aio_req req; \
		468	int req_pri = next_pri; \
		469	next_pri = EIO_PRI_DEFAULT; \
		470	\
		471	if (SvOK (callback) && !SvROK (callback)) \
		472	croak ("callback must be undef or of reference type"); \
		473	\
		474	Newz (0, req, 1, eio_req); \
		475	if (!req) \
		476	croak ("out of memory during eio_req allocation"); \
		477	\
		478	req->callback = newSVsv (callback); \
		479	req->pri = req_pri
		480
		481	#define REQ_SEND \
		482	PUTBACK; \
		483	req_submit (req); \
		484	SPAGAIN; \
		485	\
		486	if (GIMME_V != G_VOID) \
		487	XPUSHs (req_sv (req, AIO_REQ_KLASS));
		488
302	MODULE = IO::AIO PACKAGE = IO::AIO	489	MODULE = IO::AIO PACKAGE = IO::AIO
303		490
		491	PROTOTYPES: ENABLE
		492
304	BOOT:	493	BOOT:
305	{	494	{
306	if (pipe (reqpipe) || pipe (respipe))	495	stash = gv_stashpv ("IO::AIO", 1);
307	croak ("unable to initialize request or result pipe");
308		496
309	if (fcntl (reqpipe[1], F_SETFL, O_NONBLOCK))	497	newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));
310	croak ("cannot set result pipe to nonblocking mode");	498	newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));
		499	newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));
		500	newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));
		501	newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));
		502	#ifndef _WIN32
		503	newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
		504	#endif
311		505
312	if (fcntl (respipe[0], F_SETFL, O_NONBLOCK))	506	create_respipe ();
313	croak ("cannot set result pipe to nonblocking mode");
314	}
315		507
		508	if (eio_init (want_poll, done_poll) < 0)
		509	croak ("IO::AIO: unable to initialise eio library");
		510
		511	/* atfork child called in fifo order, so before eio's handler */
		512	X_THREAD_ATFORK (0, 0, atfork_child);
		513	}
		514
316	void	515	void
317	min_parallel(nthreads)	516	max_poll_reqs (int nreqs)
318	int nthreads
319	PROTOTYPE: $	517	PROTOTYPE: $
320	CODE:	518	CODE:
321	while (nthreads > started)	519	eio_set_max_poll_reqs (nreqs);
322	start_thread ();
323		520
324	void	521	void
325	max_parallel(nthreads)	522	max_poll_time (double nseconds)
326	int nthreads
327	PROTOTYPE: $	523	PROTOTYPE: $
328	CODE:	524	CODE:
		525	eio_set_max_poll_time (nseconds);
		526
		527	void
		528	min_parallel (int nthreads)
		529	PROTOTYPE: $
		530	CODE:
		531	eio_set_min_parallel (nthreads);
		532
		533	void
		534	max_parallel (int nthreads)
		535	PROTOTYPE: $
		536	CODE:
		537	eio_set_max_parallel (nthreads);
		538
		539	void
		540	max_idle (int nthreads)
		541	PROTOTYPE: $
		542	CODE:
		543	eio_set_max_idle (nthreads);
		544
		545	void
		546	max_outstanding (int maxreqs)
		547	PROTOTYPE: $
		548	CODE:
		549	max_outstanding = maxreqs;
		550
		551	void
		552	aio_open (SV8 *pathname, int flags, int mode, SV *callback=&PL_sv_undef)
		553	PROTOTYPE: $$$;$
		554	PPCODE:
329	{	555	{
330	int cur = started;	556	dREQ;
331	while (cur > nthreads)	557
332	{	558	req->type = EIO_OPEN;
333	end_thread ();	559	req->sv1 = newSVsv (pathname);
		560	req->ptr1 = SvPVbyte_nolen (req->sv1);
		561	req->int1 = flags;
		562	req->int2 = mode;
		563
		564	REQ_SEND;
		565	}
		566
		567	void
		568	aio_fsync (SV *fh, SV *callback=&PL_sv_undef)
		569	PROTOTYPE: $;$
		570	ALIAS:
		571	aio_fsync = EIO_FSYNC
		572	aio_fdatasync = EIO_FDATASYNC
		573	PPCODE:
		574	{
		575	dREQ;
		576
		577	req->type = ix;
		578	req->sv1 = newSVsv (fh);
		579	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));
		580
		581	REQ_SEND (req);
		582	}
		583
		584	void
		585	aio_close (SV *fh, SV *callback=&PL_sv_undef)
		586	PROTOTYPE: $;$
		587	PPCODE:
		588	{
		589	static int close_pipe = -1; /* dummy fd to close fds via dup2 */
		590	dREQ;
		591
		592	if (close_pipe < 0)
		593	{
		594	int pipefd [2];
		595
		596	if (pipe (pipefd) < 0
		597	|| close (pipefd [1]) < 0
		598	|| fcntl (pipefd [0], F_SETFD, FD_CLOEXEC) < 0)
		599	abort (); /*D*/
		600
		601	close_pipe = pipefd [0];
		602	}
		603
		604	req->type = EIO_DUP2;
		605	req->int1 = close_pipe;
		606	req->sv2 = newSVsv (fh);
		607	req->int2 = PerlIO_fileno (IoIFP (sv_2io (fh)));
		608
		609	REQ_SEND (req);
		610	}
		611
		612	void
		613	aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback=&PL_sv_undef)
		614	ALIAS:
		615	aio_read = EIO_READ
		616	aio_write = EIO_WRITE
		617	PROTOTYPE: $$$$$;$
		618	PPCODE:
		619	{
		620	STRLEN svlen;
		621	char *svptr = SvPVbyte (data, svlen);
		622	UV len = SvUV (length);
		623
		624	SvUPGRADE (data, SVt_PV);
		625	SvPOK_on (data);
		626
		627	if (dataoffset < 0)
		628	dataoffset += svlen;
		629
		630	if (dataoffset < 0 || dataoffset > svlen)
		631	croak ("dataoffset outside of data scalar");
		632
		633	if (ix == EIO_WRITE)
		634	{
		635	/* write: check length and adjust. */
		636	if (!SvOK (length) || len + dataoffset > svlen)
		637	len = svlen - dataoffset;
		638	}
		639	else
		640	{
		641	/* read: grow scalar as necessary */
		642	svptr = SvGROW (data, len + dataoffset + 1);
		643	}
		644
		645	if (len < 0)
		646	croak ("length must not be negative");
		647
		648	{
		649	dREQ;
		650
		651	req->type = ix;
		652	req->sv1 = newSVsv (fh);
		653	req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh))
		654	: IoOFP (sv_2io (fh)));
		655	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
		656	req->size = len;
		657	req->sv2 = SvREFCNT_inc (data);
		658	req->ptr2 = (char *)svptr + dataoffset;
		659	req->stroffset = dataoffset;
		660
		661	if (!SvREADONLY (data))
334	cur--;	662	{
		663	SvREADONLY_on (data);
		664	req->flags |= FLAG_SV2_RO_OFF;
335	}	665	}
336		666
337	while (started > nthreads)	667	REQ_SEND;
		668	}
		669	}
		670
		671	void
		672	aio_readlink (SV8 *path, SV *callback=&PL_sv_undef)
		673	PROTOTYPE: $$;$
		674	PPCODE:
		675	{
		676	SV *data;
		677	dREQ;
		678
		679	req->type = EIO_READLINK;
		680	req->sv1 = newSVsv (path);
		681	req->ptr1 = SvPVbyte_nolen (req->sv1);
		682
		683	REQ_SEND;
		684	}
		685
		686	void
		687	aio_sendfile (SV *out_fh, SV *in_fh, SV *in_offset, UV length, SV *callback=&PL_sv_undef)
		688	PROTOTYPE: $$$$;$
		689	PPCODE:
		690	{
		691	dREQ;
		692
		693	req->type = EIO_SENDFILE;
		694	req->sv1 = newSVsv (out_fh);
		695	req->int1 = PerlIO_fileno (IoIFP (sv_2io (out_fh)));
		696	req->sv2 = newSVsv (in_fh);
		697	req->int2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));
		698	req->offs = SvVAL64 (in_offset);
		699	req->size = length;
		700
		701	REQ_SEND;
		702	}
		703
		704	void
		705	aio_readahead (SV *fh, SV *offset, IV length, SV *callback=&PL_sv_undef)
		706	PROTOTYPE: $$$;$
		707	PPCODE:
		708	{
		709	dREQ;
		710
		711	req->type = EIO_READAHEAD;
		712	req->sv1 = newSVsv (fh);
		713	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));
		714	req->offs = SvVAL64 (offset);
		715	req->size = length;
		716
		717	REQ_SEND;
		718	}
		719
		720	void
		721	aio_stat (SV8 *fh_or_path, SV *callback=&PL_sv_undef)
		722	ALIAS:
		723	aio_stat = EIO_STAT
		724	aio_lstat = EIO_LSTAT
		725	PPCODE:
		726	{
		727	dREQ;
		728
		729	req->sv1 = newSVsv (fh_or_path);
		730
		731	if (SvPOK (req->sv1))
		732	{
		733	req->type = ix;
		734	req->ptr1 = SvPVbyte_nolen (req->sv1);
		735	}
		736	else
		737	{
		738	req->type = EIO_FSTAT;
		739	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		740	}
		741
		742	REQ_SEND;
		743	}
		744
		745	void
		746	aio_utime (SV8 *fh_or_path, SV *atime, SV *mtime, SV *callback=&PL_sv_undef)
		747	PPCODE:
		748	{
		749	dREQ;
		750
		751	req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
		752	req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
		753	req->sv1 = newSVsv (fh_or_path);
		754
		755	if (SvPOK (req->sv1))
		756	{
		757	req->type = EIO_UTIME;
		758	req->ptr1 = SvPVbyte_nolen (req->sv1);
		759	}
		760	else
		761	{
		762	req->type = EIO_FUTIME;
		763	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		764	}
		765
		766	REQ_SEND;
		767	}
		768
		769	void
		770	aio_truncate (SV8 *fh_or_path, SV *offset, SV *callback=&PL_sv_undef)
		771	PPCODE:
		772	{
		773	dREQ;
		774
		775	req->sv1 = newSVsv (fh_or_path);
		776	req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
		777
		778	if (SvPOK (req->sv1))
		779	{
		780	req->type = EIO_TRUNCATE;
		781	req->ptr1 = SvPVbyte_nolen (req->sv1);
		782	}
		783	else
		784	{
		785	req->type = EIO_FTRUNCATE;
		786	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		787	}
		788
		789	REQ_SEND;
		790	}
		791
		792	void
		793	aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
		794	ALIAS:
		795	aio_chmod = EIO_CHMOD
		796	aio_mkdir = EIO_MKDIR
		797	PPCODE:
		798	{
		799	dREQ;
		800
		801	req->int2 = mode;
		802	req->sv1 = newSVsv (fh_or_path);
		803
		804	if (SvPOK (req->sv1))
		805	{
		806	req->type = ix;
		807	req->ptr1 = SvPVbyte_nolen (req->sv1);
		808	}
		809	else
		810	{
		811	req->type = EIO_FCHMOD;
		812	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		813	}
		814
		815	REQ_SEND;
		816	}
		817
		818	void
		819	aio_chown (SV8 *fh_or_path, SV *uid, SV *gid, SV *callback=&PL_sv_undef)
		820	PPCODE:
		821	{
		822	dREQ;
		823
		824	req->int2 = SvOK (uid) ? SvIV (uid) : -1;
		825	req->int3 = SvOK (gid) ? SvIV (gid) : -1;
		826	req->sv1 = newSVsv (fh_or_path);
		827
		828	if (SvPOK (req->sv1))
		829	{
		830	req->type = EIO_CHOWN;
		831	req->ptr1 = SvPVbyte_nolen (req->sv1);
		832	}
		833	else
		834	{
		835	req->type = EIO_FCHOWN;
		836	req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		837	}
		838
		839	REQ_SEND;
		840	}
		841
		842	void
		843	aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)
		844	ALIAS:
		845	aio_unlink = EIO_UNLINK
		846	aio_rmdir = EIO_RMDIR
		847	aio_readdir = EIO_READDIR
		848	PPCODE:
		849	{
		850	dREQ;
		851
		852	req->type = ix;
		853	req->sv1 = newSVsv (pathname);
		854	req->ptr1 = SvPVbyte_nolen (req->sv1);
		855
		856	REQ_SEND;
		857	}
		858
		859	void
		860	aio_link (SV8 *oldpath, SV8 *newpath, SV *callback=&PL_sv_undef)
		861	ALIAS:
		862	aio_link = EIO_LINK
		863	aio_symlink = EIO_SYMLINK
		864	aio_rename = EIO_RENAME
		865	PPCODE:
		866	{
		867	dREQ;
		868
		869	req->type = ix;
		870	req->sv1 = newSVsv (oldpath);
		871	req->ptr1 = SvPVbyte_nolen (req->sv1);
		872	req->sv2 = newSVsv (newpath);
		873	req->ptr2 = SvPVbyte_nolen (req->sv2);
		874
		875	REQ_SEND;
		876	}
		877
		878	void
		879	aio_mknod (SV8 *pathname, int mode, UV dev, SV *callback=&PL_sv_undef)
		880	PPCODE:
		881	{
		882	dREQ;
		883
		884	req->type = EIO_MKNOD;
		885	req->sv1 = newSVsv (pathname);
		886	req->ptr1 = SvPVbyte_nolen (req->sv1);
		887	req->int2 = (mode_t)mode;
		888	req->offs = dev;
		889
		890	REQ_SEND;
		891	}
		892
		893	void
		894	aio_busy (double delay, SV *callback=&PL_sv_undef)
		895	PPCODE:
		896	{
		897	dREQ;
		898
		899	req->type = EIO_BUSY;
		900	req->nv1 = delay < 0. ? 0. : delay;
		901
		902	REQ_SEND;
		903	}
		904
		905	void
		906	aio_group (SV *callback=&PL_sv_undef)
		907	PROTOTYPE: ;$
		908	PPCODE:
		909	{
		910	dREQ;
		911
		912	req->type = EIO_GROUP;
		913
		914	req_submit (req);
		915	XPUSHs (req_sv (req, AIO_GRP_KLASS));
		916	}
		917
		918	void
		919	aio_nop (SV *callback=&PL_sv_undef)
		920	ALIAS:
		921	aio_nop = EIO_NOP
		922	aio_sync = EIO_SYNC
		923	PPCODE:
		924	{
		925	dREQ;
		926
		927	req->type = ix;
		928
		929	REQ_SEND;
		930	}
		931
		932	int
		933	aioreq_pri (int pri = 0)
		934	PROTOTYPE: ;$
		935	CODE:
		936	RETVAL = next_pri;
		937	if (items > 0)
		938	{
		939	if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
		940	if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
		941	next_pri = pri;
		942	}
		943	OUTPUT:
		944	RETVAL
		945
		946	void
		947	aioreq_nice (int nice = 0)
		948	CODE:
		949	nice = next_pri - nice;
		950	if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
		951	if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
		952	next_pri = nice;
		953
		954	void
		955	flush ()
		956	PROTOTYPE:
		957	CODE:
		958	while (eio_nreqs ())
338	{	959	{
339	poll_wait ();	960	poll_wait ();
340	poll_cb (aTHX);	961	poll_cb ();
341	}
342	}
343
344	void
345	aio_open(pathname,flags,mode,callback)
346	SV * pathname
347	int flags
348	int mode
349	SV * callback
350	PROTOTYPE: $$$$
351	CODE:
352	{
353	aio_req req;
354
355	Newz (0, req, 1, aio_cb);
356
357	if (!req)
358	croak ("out of memory during aio_req allocation");
359
360	req->type = REQ_OPEN;
361	req->data = newSVsv (pathname);
362	req->dataptr = SvPV_nolen (req->data);
363	req->fd = flags;
364	req->mode = mode;
365	req->callback = SvREFCNT_inc (callback);
366
367	send_req (req);
368	}
369
370	void
371	aio_close(fh,callback)
372	InputStream fh
373	SV * callback
374	PROTOTYPE: $$
375	ALIAS:
376	aio_close = REQ_CLOSE
377	aio_fsync = REQ_FSYNC
378	aio_fdatasync = REQ_FDATASYNC
379	CODE:
380	{
381	aio_req req;
382
383	Newz (0, req, 1, aio_cb);
384
385	if (!req)
386	croak ("out of memory during aio_req allocation");
387
388	req->type = ix;
389	req->fd = PerlIO_fileno (fh);
390	req->callback = SvREFCNT_inc (callback);
391
392	send_req (req);
393	}
394
395	void
396	aio_read(fh,offset,length,data,dataoffset,callback)
397	InputStream fh
398	UV offset
399	IV length
400	SV * data
401	IV dataoffset
402	SV * callback
403	PROTOTYPE: $$$$$$
404	CODE:
405	read_write (aTHX_ 0, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
406
407	void
408	aio_write(fh,offset,length,data,dataoffset,callback)
409	OutputStream fh
410	UV offset
411	IV length
412	SV * data
413	IV dataoffset
414	SV * callback
415	PROTOTYPE: $$$$$$
416	CODE:
417	read_write (aTHX_ 1, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
418
419	void
420	aio_readahead(fh,offset,length,callback)
421	InputStream fh
422	UV offset
423	IV length
424	SV * callback
425	PROTOTYPE: $$$$
426	CODE:
427	{
428	aio_req req;
429
430	if (length < 0)
431	croak ("length must not be negative");
432
433	Newz (0, req, 1, aio_cb);
434
435	if (!req)
436	croak ("out of memory during aio_req allocation");
437
438	req->type = REQ_READAHEAD;
439	req->fd = PerlIO_fileno (fh);
440	req->offset = offset;
441	req->length = length;
442	req->callback = SvREFCNT_inc (callback);
443
444	send_req (req);
445	}
446
447	void
448	aio_stat(fh_or_path,callback)
449	SV * fh_or_path
450	SV * callback
451	PROTOTYPE: $$
452	ALIAS:
453	aio_lstat = 1
454	CODE:
455	{
456	aio_req req;
457
458	Newz (0, req, 1, aio_cb);
459
460	if (!req)
461	croak ("out of memory during aio_req allocation");
462
463	New (0, req->statdata, 1, Stat_t);
464
465	if (!req->statdata)
466	croak ("out of memory during aio_req->statdata allocation");
467
468	if (SvPOK (fh_or_path))
469	{	962	}
470	req->type = ix ? REQ_LSTAT : REQ_STAT;
471	req->data = newSVsv (fh_or_path);
472	req->dataptr = SvPV_nolen (req->data);
473	}
474	else
475	{
476	req->type = REQ_FSTAT;
477	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
478	}
479		963
480	req->callback = SvREFCNT_inc (callback);	964	int
481		965	poll()
482	send_req (req);
483	}
484
485	void
486	aio_unlink(pathname,callback)
487	SV * pathname
488	SV * callback
489	PROTOTYPE: $$	966	PROTOTYPE:
490	CODE:	967	CODE:
491	{	968	poll_wait ();
492	aio_req req;	969	RETVAL = poll_cb ();
493		970	OUTPUT:
494	Newz (0, req, 1, aio_cb);	971	RETVAL
495
496	if (!req)
497	croak ("out of memory during aio_req allocation");
498
499	req->type = REQ_UNLINK;
500	req->data = newSVsv (pathname);
501	req->dataptr = SvPV_nolen (req->data);
502	req->callback = SvREFCNT_inc (callback);
503
504	send_req (req);
505	}
506		972
507	int	973	int
508	poll_fileno()	974	poll_fileno()
509	PROTOTYPE:	975	PROTOTYPE:
510	CODE:	976	CODE:
511	RETVAL = respipe[0];	977	RETVAL = respipe [0];
512	OUTPUT:	978	OUTPUT:
513	RETVAL	979	RETVAL
514		980
515	int	981	int
516	poll_cb(...)	982	poll_cb(...)
517	PROTOTYPE:	983	PROTOTYPE:
518	CODE:	984	CODE:
519	RETVAL = poll_cb (aTHX);	985	RETVAL = poll_cb ();
520	OUTPUT:	986	OUTPUT:
521	RETVAL	987	RETVAL
522		988
523	void	989	void
524	poll_wait()	990	poll_wait()
…		…
528		994
529	int	995	int
530	nreqs()	996	nreqs()
531	PROTOTYPE:	997	PROTOTYPE:
532	CODE:	998	CODE:
533	RETVAL = nreqs;	999	RETVAL = eio_nreqs ();
534	OUTPUT:	1000	OUTPUT:
535	RETVAL	1001	RETVAL
536		1002
		1003	int
		1004	nready()
		1005	PROTOTYPE:
		1006	CODE:
		1007	RETVAL = eio_nready ();
		1008	OUTPUT:
		1009	RETVAL
		1010
		1011	int
		1012	npending()
		1013	PROTOTYPE:
		1014	CODE:
		1015	RETVAL = eio_npending ();
		1016	OUTPUT:
		1017	RETVAL
		1018
		1019	int
		1020	nthreads()
		1021	PROTOTYPE:
		1022	CODE:
		1023	RETVAL = eio_nthreads ();
		1024	OUTPUT:
		1025	RETVAL
		1026
		1027	void _on_next_submit (SV *cb)
		1028	CODE:
		1029	SvREFCNT_dec (on_next_submit);
		1030	on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
		1031
		1032	PROTOTYPES: DISABLE
		1033
		1034	MODULE = IO::AIO PACKAGE = IO::AIO::REQ
		1035
		1036	void
		1037	cancel (aio_req_ornot req)
		1038	CODE:
		1039	eio_cancel (req);
		1040
		1041	void
		1042	cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
		1043	CODE:
		1044	SvREFCNT_dec (req->callback);
		1045	req->callback = newSVsv (callback);
		1046
		1047	MODULE = IO::AIO PACKAGE = IO::AIO::GRP
		1048
		1049	void
		1050	add (aio_req grp, ...)
		1051	PPCODE:
		1052	{
		1053	int i;
		1054
		1055	if (grp->int1 == 2)
		1056	croak ("cannot add requests to IO::AIO::GRP after the group finished");
		1057
		1058	for (i = 1; i < items; ++i )
		1059	{
		1060	aio_req req;
		1061
		1062	if (GIMME_V != G_VOID)
		1063	XPUSHs (sv_2mortal (newSVsv (ST (i))));
		1064
		1065	req = SvAIO_REQ (ST (i));
		1066
		1067	if (req)
		1068	eio_grp_add (grp, req);
		1069	}
		1070	}
		1071
		1072	void
		1073	cancel_subs (aio_req_ornot req)
		1074	CODE:
		1075	req_cancel_subs (req);
		1076
		1077	void
		1078	result (aio_req grp, ...)
		1079	CODE:
		1080	{
		1081	int i;
		1082	AV *av;
		1083
		1084	grp->errorno = errno;
		1085
		1086	av = newAV ();
		1087
		1088	for (i = 1; i < items; ++i )
		1089	av_push (av, newSVsv (ST (i)));
		1090
		1091	SvREFCNT_dec (grp->sv1);
		1092	grp->sv1 = (SV *)av;
		1093	}
		1094
		1095	void
		1096	errno (aio_req grp, int errorno = errno)
		1097	CODE:
		1098	grp->errorno = errorno;
		1099
		1100	void
		1101	limit (aio_req grp, int limit)
		1102	CODE:
		1103	eio_grp_limit (grp, limit);
		1104
		1105	void
		1106	feed (aio_req grp, SV *callback=&PL_sv_undef)
		1107	CODE:
		1108	{
		1109	SvREFCNT_dec (grp->sv2);
		1110	grp->sv2 = newSVsv (callback);
		1111
		1112	if (grp->int2 <= 0)
		1113	grp->int2 = 2;
		1114
		1115	eio_grp_limit (grp, grp->int2);
		1116	}
		1117

Diff Legend

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