[ViewVC] Diff of: cvs/IO-AIO/AIO.xs

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.38 by root, Sun Aug 28 10:51:33 2005 UTC vs.
Revision 1.64 by root, Mon Oct 23 23:54:41 2006 UTC

…		…
		1	#if __linux
		2	# define _GNU_SOURCE
		3	#endif
		4
1	#define _REENTRANT 1	5	#define _REENTRANT 1
		6
2	#include <errno.h>	7	#include <errno.h>
3		8
4	#include "EXTERN.h"	9	#include "EXTERN.h"
5	#include "perl.h"	10	#include "perl.h"
6	#include "XSUB.h"	11	#include "XSUB.h"
8	#include "autoconf/config.h"	13	#include "autoconf/config.h"
9		14
10	#include <pthread.h>	15	#include <pthread.h>
11		16
12	#include <stddef.h>	17	#include <stddef.h>
		18	#include <errno.h>
		19	#include <sys/time.h>
		20	#include <sys/select.h>
13	#include <sys/types.h>	21	#include <sys/types.h>
14	#include <sys/stat.h>	22	#include <sys/stat.h>
15	#include <limits.h>	23	#include <limits.h>
16	#include <unistd.h>	24	#include <unistd.h>
17	#include <fcntl.h>	25	#include <fcntl.h>
…		…
31	# else	39	# else
32	# error sendfile support requested but not available	40	# error sendfile support requested but not available
33	# endif	41	# endif
34	#endif	42	#endif
35		43
		44	/* used for struct dirent, AIX doesn't provide it */
		45	#ifndef NAME_MAX
		46	# define NAME_MAX 4096
		47	#endif
		48
36	#if __ia64	49	#if __ia64
37	# define STACKSIZE 65536	50	# define STACKSIZE 65536
38	#else	51	#else
39	# define STACKSIZE 8192	52	# define STACKSIZE 8192
40	#endif	53	#endif
…		…
44	REQ_OPEN, REQ_CLOSE,	57	REQ_OPEN, REQ_CLOSE,
45	REQ_READ, REQ_WRITE, REQ_READAHEAD,	58	REQ_READ, REQ_WRITE, REQ_READAHEAD,
46	REQ_SENDFILE,	59	REQ_SENDFILE,
47	REQ_STAT, REQ_LSTAT, REQ_FSTAT,	60	REQ_STAT, REQ_LSTAT, REQ_FSTAT,
48	REQ_FSYNC, REQ_FDATASYNC,	61	REQ_FSYNC, REQ_FDATASYNC,
49	REQ_UNLINK, REQ_RMDIR,	62	REQ_UNLINK, REQ_RMDIR, REQ_RENAME,
50	REQ_READDIR,	63	REQ_READDIR,
51	REQ_SYMLINK,	64	REQ_LINK, REQ_SYMLINK,
		65	REQ_GROUP, REQ_NOP,
		66	REQ_SLEEP,
52	};	67	};
53		68
		69	#define AIO_REQ_KLASS "IO::AIO::REQ"
		70	#define AIO_GRP_KLASS "IO::AIO::GRP"
		71
54	typedef struct aio_cb {	72	typedef struct aio_cb
		73	{
55	struct aio_cb *volatile next;	74	struct aio_cb *volatile next;
56		75
57	int type;	76	SV data, callback;
58		77	SV fh, fh2;
59	/* should receive a cleanup, with unions */	78	void dataptr, data2ptr;
60	int fd, fd2;	79	Stat_t *statdata;
61	off_t offset;	80	off_t offset;
62	size_t length;	81	size_t length;
63	ssize_t result;	82	ssize_t result;
		83
		84	STRLEN dataoffset;
		85	int type;
		86	int fd, fd2;
		87	int errorno;
64	mode_t mode; /* open */	88	mode_t mode; /* open */
65	int errorno;
66	SV data, callback;
67	SV fh, fh2;
68	void dataptr, data2ptr;
69	STRLEN dataoffset;
70		89
71	Stat_t *statdata;	90	unsigned char flags;
		91	unsigned char pri;
		92
		93	SV self; / the perl counterpart of this request, if any */
		94	struct aio_cb grp, grp_prev, grp_next, grp_first;
72	} aio_cb;	95	} aio_cb;
73		96
		97	enum {
		98	FLAG_CANCELLED = 0x01,
		99	};
		100
74	typedef aio_cb *aio_req;	101	typedef aio_cb *aio_req;
		102	typedef aio_cb *aio_req_ornot;
		103
		104	enum {
		105	PRI_MIN = -4,
		106	PRI_MAX = 4,
		107
		108	DEFAULT_PRI = 0,
		109	PRI_BIAS = -PRI_MIN,
		110	};
		111
		112	static int next_pri = DEFAULT_PRI + PRI_BIAS;
75		113
76	static int started, wanted;	114	static int started, wanted;
77	static volatile int nreqs;	115	static volatile int nreqs;
78	static int max_outstanding = 1<<30;	116	static int max_outstanding = 1<<30;
79	static int respipe [2];	117	static int respipe [2];
80		118
		119	#if __linux && defined (PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP)
		120	# define AIO_MUTEX_INIT PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP
		121	#else
		122	# define AIO_MUTEX_INIT PTHREAD_MUTEX_INITIALIZER
		123	#endif
		124
81	static pthread_mutex_t reslock = PTHREAD_MUTEX_INITIALIZER;	125	static pthread_mutex_t reslock = AIO_MUTEX_INIT;
82	static pthread_mutex_t reqlock = PTHREAD_MUTEX_INITIALIZER;	126	static pthread_mutex_t reqlock = AIO_MUTEX_INIT;
83	static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;	127	static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
84		128
85	static volatile aio_req reqs, reqe; /* queue start, queue end */	129	static volatile aio_req reqs, reqe; /* queue start, queue end */
86	static volatile aio_req ress, rese; /* queue start, queue end */	130	static volatile aio_req ress, rese; /* queue start, queue end */
87		131
		132	static void req_invoke (aio_req req);
		133	static void req_free (aio_req req);
		134
		135	/* must be called at most once */
		136	static SV req_sv (aio_req req, const char klass)
		137	{
		138	if (!req->self)
		139	{
		140	req->self = (SV *)newHV ();
		141	sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
		142	}
		143
		144	return sv_2mortal (sv_bless (newRV_inc (req->self), gv_stashpv (klass, 1)));
		145	}
		146
		147	static aio_req SvAIO_REQ (SV *sv)
		148	{
		149	MAGIC *mg;
		150
		151	if (!sv_derived_from (sv, AIO_REQ_KLASS) \|\| !SvROK (sv))
		152	croak ("object of class " AIO_REQ_KLASS " expected");
		153
		154	mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
		155
		156	return mg ? (aio_req)mg->mg_ptr : 0;
		157	}
		158
		159	static void aio_grp_feed (aio_req grp)
		160	{
		161	while (grp->length < grp->fd2 && !(grp->flags & FLAG_CANCELLED))
		162	{
		163	int old_len = grp->length;
		164
		165	if (grp->fh2 && SvOK (grp->fh2))
		166	{
		167	dSP;
		168
		169	ENTER;
		170	SAVETMPS;
		171	PUSHMARK (SP);
		172	XPUSHs (req_sv (grp, AIO_GRP_KLASS));
		173	PUTBACK;
		174	call_sv (grp->fh2, G_VOID \| G_EVAL);
		175	SPAGAIN;
		176	FREETMPS;
		177	LEAVE;
		178	}
		179
		180	/* stop if no progress has been made */
		181	if (old_len == grp->length)
		182	{
		183	SvREFCNT_dec (grp->fh2);
		184	grp->fh2 = 0;
		185	break;
		186	}
		187	}
		188	}
		189
		190	static void aio_grp_dec (aio_req grp)
		191	{
		192	--grp->length;
		193
		194	/* call feeder, if applicable */
		195	aio_grp_feed (grp);
		196
		197	/* finish, if done */
		198	if (!grp->length && grp->fd)
		199	{
		200	req_invoke (grp);
		201	req_free (grp);
		202	}
		203	}
		204
		205	static void poll_wait ()
		206	{
		207	fd_set rfd;
		208
		209	while (nreqs)
		210	{
		211	aio_req req;
		212	#if !(__x86 \|\| __x86_64) /* safe without sempahore on this archs */
		213	pthread_mutex_lock (&reslock);
		214	#endif
		215	req = ress;
		216	#if !(__x86 \|\| __x86_64) /* safe without sempahore on this archs */
		217	pthread_mutex_unlock (&reslock);
		218	#endif
		219
		220	if (req)
		221	return;
		222
		223	FD_ZERO(&rfd);
		224	FD_SET(respipe [0], &rfd);
		225
		226	select (respipe [0] + 1, &rfd, 0, 0, 0);
		227	}
		228	}
		229
		230	static void req_invoke (aio_req req)
		231	{
		232	dSP;
		233	int errorno = errno;
		234
		235	if (req->flags & FLAG_CANCELLED \|\| !SvOK (req->callback))
		236	return;
		237
		238	errno = req->errorno;
		239
		240	ENTER;
		241	SAVETMPS;
		242	PUSHMARK (SP);
		243	EXTEND (SP, 1);
		244
		245	switch (req->type)
		246	{
		247	case REQ_READDIR:
		248	{
		249	SV *rv = &PL_sv_undef;
		250
		251	if (req->result >= 0)
		252	{
		253	char *buf = req->data2ptr;
		254	AV *av = newAV ();
		255
		256	while (req->result)
		257	{
		258	SV *sv = newSVpv (buf, 0);
		259
		260	av_push (av, sv);
		261	buf += SvCUR (sv) + 1;
		262	req->result--;
		263	}
		264
		265	rv = sv_2mortal (newRV_noinc ((SV *)av));
		266	}
		267
		268	PUSHs (rv);
		269	}
		270	break;
		271
		272	case REQ_OPEN:
		273	{
		274	/* convert fd to fh */
		275	SV *fh;
		276
		277	PUSHs (sv_2mortal (newSViv (req->result)));
		278	PUTBACK;
		279	call_pv ("IO::AIO::_fd2fh", G_SCALAR \| G_EVAL);
		280	SPAGAIN;
		281
		282	fh = SvREFCNT_inc (POPs);
		283
		284	PUSHMARK (SP);
		285	XPUSHs (sv_2mortal (fh));
		286	}
		287	break;
		288
		289	case REQ_GROUP:
		290	req->fd = 2; /* mark group as finished */
		291
		292	if (req->data)
		293	{
		294	int i;
		295	AV av = (AV )req->data;
		296
		297	EXTEND (SP, AvFILL (av) + 1);
		298	for (i = 0; i <= AvFILL (av); ++i)
		299	PUSHs (*av_fetch (av, i, 0));
		300	}
		301	break;
		302
		303	case REQ_NOP:
		304	case REQ_SLEEP:
		305	break;
		306
		307	default:
		308	PUSHs (sv_2mortal (newSViv (req->result)));
		309	break;
		310	}
		311
		312
		313	PUTBACK;
		314	call_sv (req->callback, G_VOID \| G_EVAL);
		315	SPAGAIN;
		316
		317	FREETMPS;
		318	LEAVE;
		319
		320	errno = errorno;
		321
		322	if (SvTRUE (ERRSV))
		323	{
		324	req_free (req);
		325	croak (0);
		326	}
		327	}
		328
88	static void free_req (aio_req req)	329	static void req_free (aio_req req)
89	{	330	{
90	if (req->data)	331	if (req->grp)
		332	{
		333	aio_req grp = req->grp;
		334
		335	/* unlink request */
		336	if (req->grp_next) req->grp_next->grp_prev = req->grp_prev;
		337	if (req->grp_prev) req->grp_prev->grp_next = req->grp_next;
		338
		339	if (grp->grp_first == req)
		340	grp->grp_first = req->grp_next;
		341
		342	aio_grp_dec (grp);
		343	}
		344
		345	if (req->self)
		346	{
		347	sv_unmagic (req->self, PERL_MAGIC_ext);
		348	SvREFCNT_dec (req->self);
		349	}
		350
91	SvREFCNT_dec (req->data);	351	SvREFCNT_dec (req->data);
92
93	if (req->fh)
94	SvREFCNT_dec (req->fh);	352	SvREFCNT_dec (req->fh);
95
96	if (req->fh2)
97	SvREFCNT_dec (req->fh2);	353	SvREFCNT_dec (req->fh2);
98
99	if (req->statdata)
100	Safefree (req->statdata);
101
102	if (req->callback)
103	SvREFCNT_dec (req->callback);	354	SvREFCNT_dec (req->callback);
		355	Safefree (req->statdata);
104		356
105	if (req->type == REQ_READDIR && req->result >= 0)	357	if (req->type == REQ_READDIR && req->result >= 0)
106	free (req->data2ptr);	358	free (req->data2ptr);
107		359
108	Safefree (req);	360	Safefree (req);
109	}	361	}
110		362
111	static void	363	static void req_cancel (aio_req req)
112	poll_wait ()
113	{	364	{
114	if (nreqs && !ress)	365	req->flags \|= FLAG_CANCELLED;
115	{
116	fd_set rfd;
117	FD_ZERO(&rfd);
118	FD_SET(respipe [0], &rfd);
119		366
120	select (respipe [0] + 1, &rfd, 0, 0, 0);	367	if (req->type == REQ_GROUP)
121	}	368	{
122	}	369	aio_req sub;
123		370
124	static int	371	for (sub = req->grp_first; sub; sub = sub->grp_next)
125	poll_cb ()	372	req_cancel (sub);
		373	}
		374	}
		375
		376	static int poll_cb ()
126	{	377	{
127	dSP;	378	dSP;
128	int count = 0;	379	int count = 0;
129	int do_croak = 0;	380	int do_croak = 0;
130	aio_req req;	381	aio_req req;
…		…
152	pthread_mutex_unlock (&reslock);	403	pthread_mutex_unlock (&reslock);
153		404
154	if (!req)	405	if (!req)
155	break;	406	break;
156		407
157	nreqs--;	408	--nreqs;
158		409
159	if (req->type == REQ_QUIT)	410	if (req->type == REQ_QUIT)
160	started--;	411	started--;
		412	else if (req->type == REQ_GROUP && req->length)
		413	{
		414	req->fd = 1; /* mark request as delayed */
		415	continue;
		416	}
161	else	417	else
162	{	418	{
163	int errorno = errno;
164	errno = req->errorno;
165
166	if (req->type == REQ_READ)	419	if (req->type == REQ_READ)
167	SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));	420	SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
168		421
169	if (req->data2ptr && (req->type == REQ_READ \|\| req->type == REQ_WRITE))	422	if (req->data2ptr && (req->type == REQ_READ \|\| req->type == REQ_WRITE))
170	SvREADONLY_off (req->data);	423	SvREADONLY_off (req->data);
…		…
174	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;	427	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
175	PL_laststatval = req->result;	428	PL_laststatval = req->result;
176	PL_statcache = *(req->statdata);	429	PL_statcache = *(req->statdata);
177	}	430	}
178		431
179	ENTER;	432	req_invoke (req);
180	PUSHMARK (SP);
181		433
182	if (req->type == REQ_READDIR)
183	{
184	SV *rv = &PL_sv_undef;
185
186	if (req->result >= 0)
187	{
188	char *buf = req->data2ptr;
189	AV *av = newAV ();
190
191	while (req->result)
192	{
193	SV *sv = newSVpv (buf, 0);
194
195	av_push (av, sv);
196	buf += SvCUR (sv) + 1;
197	req->result--;
198	}
199
200	rv = sv_2mortal (newRV_noinc ((SV *)av));
201	}
202
203	XPUSHs (rv);
204	}
205	else
206	{
207	XPUSHs (sv_2mortal (newSViv (req->result)));
208
209	if (req->type == REQ_OPEN)
210	{
211	/* convert fd to fh */
212	SV *fh;
213
214	PUTBACK;
215	call_pv ("IO::AIO::_fd2fh", G_SCALAR \| G_EVAL);
216	SPAGAIN;
217
218	fh = SvREFCNT_inc (POPs);
219
220	PUSHMARK (SP);
221	XPUSHs (sv_2mortal (fh));
222	}
223	}
224
225	if (SvOK (req->callback))
226	{
227	PUTBACK;
228	call_sv (req->callback, G_VOID \| G_EVAL);
229	SPAGAIN;
230
231	if (SvTRUE (ERRSV))
232	{
233	free_req (req);
234	croak (0);
235	}
236	}
237
238	LEAVE;
239
240	errno = errorno;
241	count++;	434	count++;
242	}	435	}
243		436
244	free_req (req);	437	req_free (req);
245	}	438	}
246		439
247	return count;	440	return count;
248	}	441	}
249		442
250	static void aio_proc(void arg);	443	static void aio_proc(void arg);
251		444
252	static void
253	start_thread (void)	445	static void start_thread (void)
254	{	446	{
255	sigset_t fullsigset, oldsigset;	447	sigset_t fullsigset, oldsigset;
256	pthread_t tid;	448	pthread_t tid;
257	pthread_attr_t attr;	449	pthread_attr_t attr;
258		450
…		…
267	started++;	459	started++;
268		460
269	sigprocmask (SIG_SETMASK, &oldsigset, 0);	461	sigprocmask (SIG_SETMASK, &oldsigset, 0);
270	}	462	}
271		463
272	static void	464	static void req_send (aio_req req)
273	send_req (aio_req req)
274	{	465	{
275	while (started < wanted && nreqs >= started)	466	while (started < wanted && nreqs >= started)
276	start_thread ();	467	start_thread ();
277		468
278	nreqs++;	469	++nreqs;
279		470
280	pthread_mutex_lock (&reqlock);	471	pthread_mutex_lock (&reqlock);
281		472
282	req->next = 0;	473	req->next = 0;
283		474
…		…
302		493
303	poll_wait ();	494	poll_wait ();
304	}	495	}
305	}	496	}
306		497
307	static void	498	static void end_thread (void)
308	end_thread (void)
309	{	499	{
310	aio_req req;	500	aio_req req;
311	Newz (0, req, 1, aio_cb);	501	Newz (0, req, 1, aio_cb);
312	req->type = REQ_QUIT;	502	req->type = REQ_QUIT;
313		503
314	send_req (req);	504	req_send (req);
315	}	505	}
316		506
317	static void min_parallel (int nthreads)	507	static void min_parallel (int nthreads)
318	{	508	{
319	if (wanted < nthreads)	509	if (wanted < nthreads)
…		…
364	* normal read/write by using a mutex. slows down execution a lot,	554	* normal read/write by using a mutex. slows down execution a lot,
365	* but that's your problem, not mine.	555	* but that's your problem, not mine.
366	*/	556	*/
367	static pthread_mutex_t preadwritelock = PTHREAD_MUTEX_INITIALIZER;	557	static pthread_mutex_t preadwritelock = PTHREAD_MUTEX_INITIALIZER;
368		558
369	static ssize_t	559	static ssize_t pread (int fd, void *buf, size_t count, off_t offset)
370	pread (int fd, void *buf, size_t count, off_t offset)
371	{	560	{
372	ssize_t res;	561	ssize_t res;
373	off_t ooffset;	562	off_t ooffset;
374		563
375	pthread_mutex_lock (&preadwritelock);	564	pthread_mutex_lock (&preadwritelock);
…		…
380	pthread_mutex_unlock (&preadwritelock);	569	pthread_mutex_unlock (&preadwritelock);
381		570
382	return res;	571	return res;
383	}	572	}
384		573
385	static ssize_t
386	pwrite (int fd, void *buf, size_t count, off_t offset)	574	static ssize_t pwrite (int fd, void *buf, size_t count, off_t offset)
387	{	575	{
388	ssize_t res;	576	ssize_t res;
389	off_t ooffset;	577	off_t ooffset;
390		578
391	pthread_mutex_lock (&preadwritelock);	579	pthread_mutex_lock (&preadwritelock);
…		…
404	#endif	592	#endif
405		593
406	#if !HAVE_READAHEAD	594	#if !HAVE_READAHEAD
407	# define readahead aio_readahead	595	# define readahead aio_readahead
408		596
409	static ssize_t
410	readahead (int fd, off_t offset, size_t count)	597	static ssize_t readahead (int fd, off_t offset, size_t count)
411	{	598	{
412	char readahead_buf[4096];	599	char readahead_buf[4096];
413		600
414	while (count > 0)	601	while (count > 0)
415	{	602	{
…		…
427	#if !HAVE_READDIR_R	614	#if !HAVE_READDIR_R
428	# define readdir_r aio_readdir_r	615	# define readdir_r aio_readdir_r
429		616
430	static pthread_mutex_t readdirlock = PTHREAD_MUTEX_INITIALIZER;	617	static pthread_mutex_t readdirlock = PTHREAD_MUTEX_INITIALIZER;
431		618
432	static int
433	readdir_r (DIR dirp, struct dirent ent, struct dirent **res)	619	static int readdir_r (DIR dirp, struct dirent ent, struct dirent **res)
434	{	620	{
435	struct dirent *e;	621	struct dirent *e;
436	int errorno;	622	int errorno;
437		623
438	pthread_mutex_lock (&readdirlock);	624	pthread_mutex_lock (&readdirlock);
…		…
454	return e ? 0 : -1;	640	return e ? 0 : -1;
455	}	641	}
456	#endif	642	#endif
457		643
458	/* sendfile always needs emulation */	644	/* sendfile always needs emulation */
459	static ssize_t
460	sendfile_ (int ofd, int ifd, off_t offset, size_t count)	645	static ssize_t sendfile_ (int ofd, int ifd, off_t offset, size_t count)
461	{	646	{
462	ssize_t res;	647	ssize_t res;
463		648
464	if (!count)	649	if (!count)
465	return 0;	650	return 0;
…		…
547		732
548	return res;	733	return res;
549	}	734	}
550		735
551	/* read a full directory */	736	/* read a full directory */
552	static int
553	scandir_ (const char path, void *namesp)	737	static int scandir_ (const char path, void *namesp)
554	{	738	{
555	DIR *dirp = opendir (path);	739	DIR *dirp = opendir (path);
556	union	740	union
557	{	741	{
558	struct dirent d;	742	struct dirent d;
…		…
612	return res;	796	return res;
613	}	797	}
614		798
615	/*****************************************************************************/	799	/*****************************************************************************/
616		800
617	static void *
618	aio_proc (void *thr_arg)	801	static void aio_proc (void thr_arg)
619	{	802	{
620	aio_req req;	803	aio_req req;
621	int type;	804	int type;
622		805
623	do	806	do
…		…
641	}	824	}
642		825
643	pthread_mutex_unlock (&reqlock);	826	pthread_mutex_unlock (&reqlock);
644		827
645	errno = 0; /* strictly unnecessary */	828	errno = 0; /* strictly unnecessary */
		829	type = req->type; /* remember type for QUIT check */
646		830
647	type = req->type;	831	if (!(req->flags & FLAG_CANCELLED))
648
649	switch (type)	832	switch (type)
650	{	833	{
651	case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;	834	case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
652	case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;	835	case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
653		836
654	case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;	837	case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
655	case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length); break;	838	case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length); break;
656		839
657	case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;	840	case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
658	case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;	841	case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
659	case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;	842	case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
660		843
661	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;	844	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
662	case REQ_CLOSE: req->result = close (req->fd); break;	845	case REQ_CLOSE: req->result = close (req->fd); break;
663	case REQ_UNLINK: req->result = unlink (req->dataptr); break;	846	case REQ_UNLINK: req->result = unlink (req->dataptr); break;
664	case REQ_RMDIR: req->result = rmdir (req->dataptr); break;	847	case REQ_RMDIR: req->result = rmdir (req->dataptr); break;
		848	case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break;
		849	case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break;
665	case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;	850	case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;
666		851
667	case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;	852	case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
668	case REQ_FSYNC: req->result = fsync (req->fd); break;	853	case REQ_FSYNC: req->result = fsync (req->fd); break;
669	case REQ_READDIR: req->result = scandir_ (req->dataptr, &req->data2ptr); break;	854	case REQ_READDIR: req->result = scandir_ (req->dataptr, &req->data2ptr); break;
670		855
		856	case REQ_SLEEP:
		857	{
		858	struct timeval tv;
		859
		860	tv.tv_sec = req->fd;
		861	tv.tv_usec = req->fd2;
		862
		863	req->result = select (0, 0, 0, 0, &tv);
		864	}
		865
		866	case REQ_GROUP:
		867	case REQ_NOP:
671	case REQ_QUIT:	868	case REQ_QUIT:
672	break;	869	break;
673		870
674	default:	871	default:
675	req->result = ENOSYS;	872	req->result = ENOSYS;
676	break;	873	break;
677	}	874	}
678		875
679	req->errorno = errno;	876	req->errorno = errno;
680		877
681	pthread_mutex_lock (&reslock);	878	pthread_mutex_lock (&reslock);
682		879
…		…
736		933
737	while (reqs)	934	while (reqs)
738	{	935	{
739	prv = reqs;	936	prv = reqs;
740	reqs = prv->next;	937	reqs = prv->next;
741	free_req (prv);	938	req_free (prv);
742	}	939	}
743		940
744	reqs = reqe = 0;	941	reqs = reqe = 0;
745		942
746	while (ress)	943	while (ress)
747	{	944	{
748	prv = ress;	945	prv = ress;
749	ress = prv->next;	946	ress = prv->next;
750	free_req (prv);	947	req_free (prv);
751	}	948	}
752		949
753	ress = rese = 0;	950	ress = rese = 0;
754		951
755	close (respipe [0]);	952	close (respipe [0]);
…		…
759	atfork_parent ();	956	atfork_parent ();
760	}	957	}
761		958
762	#define dREQ \	959	#define dREQ \
763	aio_req req; \	960	aio_req req; \
		961	int req_pri = next_pri; \
		962	next_pri = DEFAULT_PRI + PRI_BIAS; \
764	\	963	\
765	if (SvOK (callback) && !SvROK (callback)) \	964	if (SvOK (callback) && !SvROK (callback)) \
766	croak ("clalback must be undef or of reference type"); \	965	croak ("callback must be undef or of reference type"); \
767	\	966	\
768	Newz (0, req, 1, aio_cb); \	967	Newz (0, req, 1, aio_cb); \
769	if (!req) \	968	if (!req) \
770	croak ("out of memory during aio_req allocation"); \	969	croak ("out of memory during aio_req allocation"); \
771	\	970	\
772	req->callback = newSVsv (callback);	971	req->callback = newSVsv (callback); \
		972	req->pri = req_pri
		973
		974	#define REQ_SEND \
		975	req_send (req); \
		976	\
		977	if (GIMME_V != G_VOID) \
		978	XPUSHs (req_sv (req, AIO_REQ_KLASS));
773		979
774	MODULE = IO::AIO PACKAGE = IO::AIO	980	MODULE = IO::AIO PACKAGE = IO::AIO
775		981
776	PROTOTYPES: ENABLE	982	PROTOTYPES: ENABLE
777		983
778	BOOT:	984	BOOT:
779	{	985	{
		986	HV *stash = gv_stashpv ("IO::AIO", 1);
		987	newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));
		988	newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));
		989	newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));
		990
780	create_pipe ();	991	create_pipe ();
781	pthread_atfork (atfork_prepare, atfork_parent, atfork_child);	992	pthread_atfork (atfork_prepare, atfork_parent, atfork_child);
782	}	993	}
783		994
784	void	995	void
785	min_parallel(nthreads)	996	min_parallel (nthreads)
786	int nthreads	997	int nthreads
787	PROTOTYPE: $	998	PROTOTYPE: $
788		999
789	void	1000	void
790	max_parallel(nthreads)	1001	max_parallel (nthreads)
791	int nthreads	1002	int nthreads
792	PROTOTYPE: $	1003	PROTOTYPE: $
793		1004
794	int	1005	int
795	max_outstanding(nreqs)	1006	max_outstanding (nreqs)
796	int nreqs	1007	int nreqs
797	PROTOTYPE: $	1008	PROTOTYPE: $
798	CODE:	1009	CODE:
799	RETVAL = max_outstanding;	1010	RETVAL = max_outstanding;
800	max_outstanding = nreqs;	1011	max_outstanding = nreqs;
801		1012
802	void	1013	void
803	aio_open(pathname,flags,mode,callback=&PL_sv_undef)	1014	aio_open (pathname,flags,mode,callback=&PL_sv_undef)
804	SV * pathname	1015	SV * pathname
805	int flags	1016	int flags
806	int mode	1017	int mode
807	SV * callback	1018	SV * callback
808	PROTOTYPE: $$$;$	1019	PROTOTYPE: $$$;$
809	CODE:	1020	PPCODE:
810	{	1021	{
811	dREQ;	1022	dREQ;
812		1023
813	req->type = REQ_OPEN;	1024	req->type = REQ_OPEN;
814	req->data = newSVsv (pathname);	1025	req->data = newSVsv (pathname);
815	req->dataptr = SvPVbyte_nolen (req->data);	1026	req->dataptr = SvPVbyte_nolen (req->data);
816	req->fd = flags;	1027	req->fd = flags;
817	req->mode = mode;	1028	req->mode = mode;
818		1029
819	send_req (req);	1030	REQ_SEND;
820	}	1031	}
821		1032
822	void	1033	void
823	aio_close(fh,callback=&PL_sv_undef)	1034	aio_close (fh,callback=&PL_sv_undef)
824	SV * fh	1035	SV * fh
825	SV * callback	1036	SV * callback
826	PROTOTYPE: $;$	1037	PROTOTYPE: $;$
827	ALIAS:	1038	ALIAS:
828	aio_close = REQ_CLOSE	1039	aio_close = REQ_CLOSE
829	aio_fsync = REQ_FSYNC	1040	aio_fsync = REQ_FSYNC
830	aio_fdatasync = REQ_FDATASYNC	1041	aio_fdatasync = REQ_FDATASYNC
831	CODE:	1042	PPCODE:
832	{	1043	{
833	dREQ;	1044	dREQ;
834		1045
835	req->type = ix;	1046	req->type = ix;
836	req->fh = newSVsv (fh);	1047	req->fh = newSVsv (fh);
837	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));	1048	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
838		1049
839	send_req (req);	1050	REQ_SEND (req);
840	}	1051	}
841		1052
842	void	1053	void
843	aio_read(fh,offset,length,data,dataoffset,callback=&PL_sv_undef)	1054	aio_read (fh,offset,length,data,dataoffset,callback=&PL_sv_undef)
844	SV * fh	1055	SV * fh
845	UV offset	1056	UV offset
846	UV length	1057	UV length
847	SV * data	1058	SV * data
848	UV dataoffset	1059	UV dataoffset
849	SV * callback	1060	SV * callback
850	ALIAS:	1061	ALIAS:
851	aio_read = REQ_READ	1062	aio_read = REQ_READ
852	aio_write = REQ_WRITE	1063	aio_write = REQ_WRITE
853	PROTOTYPE: $$$$$;$	1064	PROTOTYPE: $$$$$;$
854	CODE:	1065	PPCODE:
855	{	1066	{
856	aio_req req;	1067	aio_req req;
857	STRLEN svlen;	1068	STRLEN svlen;
858	char *svptr = SvPVbyte (data, svlen);	1069	char *svptr = SvPVbyte (data, svlen);
859		1070
…		…
897	{	1108	{
898	SvREADONLY_on (data);	1109	SvREADONLY_on (data);
899	req->data2ptr = (void *)data;	1110	req->data2ptr = (void *)data;
900	}	1111	}
901		1112
902	send_req (req);	1113	REQ_SEND;
903	}	1114	}
904	}	1115	}
905		1116
906	void	1117	void
907	aio_sendfile(out_fh,in_fh,in_offset,length,callback=&PL_sv_undef)	1118	aio_sendfile (out_fh,in_fh,in_offset,length,callback=&PL_sv_undef)
908	SV * out_fh	1119	SV * out_fh
909	SV * in_fh	1120	SV * in_fh
910	UV in_offset	1121	UV in_offset
911	UV length	1122	UV length
912	SV * callback	1123	SV * callback
913	PROTOTYPE: $$$$;$	1124	PROTOTYPE: $$$$;$
914	CODE:	1125	PPCODE:
915	{	1126	{
916	dREQ;	1127	dREQ;
917		1128
918	req->type = REQ_SENDFILE;	1129	req->type = REQ_SENDFILE;
919	req->fh = newSVsv (out_fh);	1130	req->fh = newSVsv (out_fh);
…		…
921	req->fh2 = newSVsv (in_fh);	1132	req->fh2 = newSVsv (in_fh);
922	req->fd2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));	1133	req->fd2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));
923	req->offset = in_offset;	1134	req->offset = in_offset;
924	req->length = length;	1135	req->length = length;
925		1136
926	send_req (req);	1137	REQ_SEND;
927	}	1138	}
928		1139
929	void	1140	void
930	aio_readahead(fh,offset,length,callback=&PL_sv_undef)	1141	aio_readahead (fh,offset,length,callback=&PL_sv_undef)
931	SV * fh	1142	SV * fh
932	UV offset	1143	UV offset
933	IV length	1144	IV length
934	SV * callback	1145	SV * callback
935	PROTOTYPE: $$$;$	1146	PROTOTYPE: $$$;$
936	CODE:	1147	PPCODE:
937	{	1148	{
938	dREQ;	1149	dREQ;
939		1150
940	req->type = REQ_READAHEAD;	1151	req->type = REQ_READAHEAD;
941	req->fh = newSVsv (fh);	1152	req->fh = newSVsv (fh);
942	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));	1153	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
943	req->offset = offset;	1154	req->offset = offset;
944	req->length = length;	1155	req->length = length;
945		1156
946	send_req (req);	1157	REQ_SEND;
947	}	1158	}
948		1159
949	void	1160	void
950	aio_stat(fh_or_path,callback=&PL_sv_undef)	1161	aio_stat (fh_or_path,callback=&PL_sv_undef)
951	SV * fh_or_path	1162	SV * fh_or_path
952	SV * callback	1163	SV * callback
953	ALIAS:	1164	ALIAS:
954	aio_stat = REQ_STAT	1165	aio_stat = REQ_STAT
955	aio_lstat = REQ_LSTAT	1166	aio_lstat = REQ_LSTAT
956	CODE:	1167	PPCODE:
957	{	1168	{
958	dREQ;	1169	dREQ;
959		1170
960	New (0, req->statdata, 1, Stat_t);	1171	New (0, req->statdata, 1, Stat_t);
961	if (!req->statdata)	1172	if (!req->statdata)
962	{	1173	{
963	free_req (req);	1174	req_free (req);
964	croak ("out of memory during aio_req->statdata allocation");	1175	croak ("out of memory during aio_req->statdata allocation");
965	}	1176	}
966		1177
967	if (SvPOK (fh_or_path))	1178	if (SvPOK (fh_or_path))
968	{	1179	{
…		…
975	req->type = REQ_FSTAT;	1186	req->type = REQ_FSTAT;
976	req->fh = newSVsv (fh_or_path);	1187	req->fh = newSVsv (fh_or_path);
977	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	1188	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
978	}	1189	}
979		1190
980	send_req (req);	1191	REQ_SEND;
981	}	1192	}
982		1193
983	void	1194	void
984	aio_unlink(pathname,callback=&PL_sv_undef)	1195	aio_unlink (pathname,callback=&PL_sv_undef)
985	SV * pathname	1196	SV * pathname
986	SV * callback	1197	SV * callback
987	ALIAS:	1198	ALIAS:
988	aio_unlink = REQ_UNLINK	1199	aio_unlink = REQ_UNLINK
989	aio_rmdir = REQ_RMDIR	1200	aio_rmdir = REQ_RMDIR
		1201	aio_readdir = REQ_READDIR
990	CODE:	1202	PPCODE:
991	{	1203	{
992	dREQ;	1204	dREQ;
993		1205
994	req->type = ix;	1206	req->type = ix;
995	req->data = newSVsv (pathname);	1207	req->data = newSVsv (pathname);
996	req->dataptr = SvPVbyte_nolen (req->data);	1208	req->dataptr = SvPVbyte_nolen (req->data);
997		1209
998	send_req (req);	1210	REQ_SEND;
999	}	1211	}
1000		1212
1001	void	1213	void
1002	aio_symlink(oldpath,newpath,callback=&PL_sv_undef)	1214	aio_link (oldpath,newpath,callback=&PL_sv_undef)
1003	SV * oldpath	1215	SV * oldpath
1004	SV * newpath	1216	SV * newpath
1005	SV * callback	1217	SV * callback
		1218	ALIAS:
		1219	aio_link = REQ_LINK
		1220	aio_symlink = REQ_SYMLINK
		1221	aio_rename = REQ_RENAME
1006	CODE:	1222	PPCODE:
1007	{	1223	{
1008	dREQ;	1224	dREQ;
1009		1225
1010	req->type = REQ_SYMLINK;	1226	req->type = ix;
1011	req->fh = newSVsv (oldpath);	1227	req->fh = newSVsv (oldpath);
1012	req->data2ptr = SvPVbyte_nolen (req->fh);	1228	req->data2ptr = SvPVbyte_nolen (req->fh);
1013	req->data = newSVsv (newpath);	1229	req->data = newSVsv (newpath);
1014	req->dataptr = SvPVbyte_nolen (req->data);	1230	req->dataptr = SvPVbyte_nolen (req->data);
1015		1231
1016	send_req (req);	1232	REQ_SEND;
1017	}	1233	}
1018		1234
1019	void	1235	void
1020	aio_readdir(pathname,callback=&PL_sv_undef)	1236	aio_sleep (delay,callback=&PL_sv_undef)
1021	SV * pathname	1237	double delay
1022	SV * callback	1238	SV * callback
		1239	PPCODE:
		1240	{
		1241	dREQ;
		1242
		1243	req->type = REQ_SLEEP;
		1244	req->fd = delay < 0. ? 0 : delay;
		1245	req->fd2 = delay < 0. ? 0 : 1000. * (delay - req->fd);
		1246
		1247	REQ_SEND;
		1248	}
		1249
		1250	void
		1251	aio_group (callback=&PL_sv_undef)
		1252	SV * callback
		1253	PROTOTYPE: ;$
		1254	PPCODE:
		1255	{
		1256	dREQ;
		1257
		1258	req->type = REQ_GROUP;
		1259	req_send (req);
		1260
		1261	XPUSHs (req_sv (req, AIO_GRP_KLASS));
		1262	}
		1263
		1264	void
		1265	aio_nop (callback=&PL_sv_undef)
		1266	SV * callback
		1267	PPCODE:
		1268	{
		1269	dREQ;
		1270
		1271	req->type = REQ_NOP;
		1272
		1273	REQ_SEND;
		1274	}
		1275
		1276	#if 0
		1277
		1278	void
		1279	aio_pri (int pri = DEFAULT_PRI)
1023	CODE:	1280	CODE:
1024	{	1281	if (pri < PRI_MIN) pri = PRI_MIN;
1025	dREQ;	1282	if (pri > PRI_MAX) pri = PRI_MAX;
1026		1283	next_pri = pri + PRI_BIAS;
1027	req->type = REQ_READDIR;
1028	req->data = newSVsv (pathname);
1029	req->dataptr = SvPVbyte_nolen (req->data);
1030
1031	send_req (req);
1032	}
1033		1284
		1285	#endif
		1286
1034	void	1287	void
1035	flush()	1288	flush ()
1036	PROTOTYPE:	1289	PROTOTYPE:
1037	CODE:	1290	CODE:
1038	while (nreqs)	1291	while (nreqs)
1039	{	1292	{
1040	poll_wait ();	1293	poll_wait ();
…		…
1080	CODE:	1333	CODE:
1081	RETVAL = nreqs;	1334	RETVAL = nreqs;
1082	OUTPUT:	1335	OUTPUT:
1083	RETVAL	1336	RETVAL
1084		1337
		1338	PROTOTYPES: DISABLE
		1339
		1340	MODULE = IO::AIO PACKAGE = IO::AIO::REQ
		1341
		1342	void
		1343	cancel (aio_req_ornot req)
		1344	PROTOTYPE:
		1345	CODE:
		1346	req_cancel (req);
		1347
		1348	void
		1349	cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
		1350	CODE:
		1351	SvREFCNT_dec (req->callback);
		1352	req->callback = newSVsv (callback);
		1353
		1354	MODULE = IO::AIO PACKAGE = IO::AIO::GRP
		1355
		1356	void
		1357	add (aio_req grp, ...)
		1358	PPCODE:
		1359	{
		1360	int i;
		1361	aio_req req;
		1362
		1363	if (grp->fd == 2)
		1364	croak ("cannot add requests to IO::AIO::GRP after the group finished");
		1365
		1366	for (i = 1; i < items; ++i )
		1367	{
		1368	if (GIMME_V != G_VOID)
		1369	XPUSHs (sv_2mortal (newSVsv (ST (i))));
		1370
		1371	req = SvAIO_REQ (ST (i));
		1372
		1373	if (req)
		1374	{
		1375	++grp->length;
		1376	req->grp = grp;
		1377
		1378	req->grp_prev = 0;
		1379	req->grp_next = grp->grp_first;
		1380
		1381	if (grp->grp_first)
		1382	grp->grp_first->grp_prev = req;
		1383
		1384	grp->grp_first = req;
		1385	}
		1386	}
		1387	}
		1388
		1389	void
		1390	result (aio_req grp, ...)
		1391	CODE:
		1392	{
		1393	int i;
		1394	AV *av = newAV ();
		1395
		1396	for (i = 1; i < items; ++i )
		1397	av_push (av, newSVsv (ST (i)));
		1398
		1399	SvREFCNT_dec (grp->data);
		1400	grp->data = (SV *)av;
		1401	}
		1402
		1403	void
		1404	feed_limit (aio_req grp, int limit)
		1405	CODE:
		1406	grp->fd2 = limit;
		1407	aio_grp_feed (grp);
		1408
		1409	void
		1410	feed (aio_req grp, SV *callback=&PL_sv_undef)
		1411	CODE:
		1412	{
		1413	SvREFCNT_dec (grp->fh2);
		1414	grp->fh2 = newSVsv (callback);
		1415
		1416	if (grp->fd2 <= 0)
		1417	grp->fd2 = 2;
		1418
		1419	aio_grp_feed (grp);
		1420	}
		1421

Diff Legend

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

Comparing IO-AIO/AIO.xs (file contents): Revision 1.38 by root, Sun Aug 28 10:51:33 2005 UTC vs. Revision 1.64 by root, Mon Oct 23 23:54:41 2006 UTC

Diff Legend

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.38 by root, Sun Aug 28 10:51:33 2005 UTC vs.
Revision 1.64 by root, Mon Oct 23 23:54:41 2006 UTC