[ViewVC] Diff of: 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.35 by root, Tue Aug 23 01:16:50 2005 UTC

…		…
1	#define PERL_NO_GET_CONTEXT	1	#define _REENTRANT 1
		2	#include <errno.h>
2		3
3	#include "EXTERN.h"	4	#include "EXTERN.h"
4	#include "perl.h"	5	#include "perl.h"
5	#include "XSUB.h"	6	#include "XSUB.h"
6		7
		8	#include "autoconf/config.h"
		9
		10	#include <pthread.h>
		11
7	#include <sys/types.h>	12	#include <sys/types.h>
8	#include <sys/stat.h>	13	#include <sys/stat.h>
		14
9	#include <unistd.h>	15	#include <unistd.h>
10	#include <fcntl.h>	16	#include <fcntl.h>
11	#include <signal.h>	17	#include <signal.h>
12	#include <sched.h>	18	#include <sched.h>
13	#include <endian.h>
14		19
15	#include <pthread.h>	20	#if HAVE_SENDFILE
		21	# if __linux
		22	# include <sys/sendfile.h>
		23	# elif __freebsd
16	#include <sys/syscall.h>	24	# include <sys/socket.h>
		25	# include <sys/uio.h>
		26	# elif __hpux
		27	# include <sys/socket.h>
		28	# elif __solaris /* not yet */
		29	# include <sys/sendfile.h>
		30	# else
		31	# error sendfile support requested but not available
		32	# endif
		33	#endif
17		34
18	typedef void InputStream; / hack, but 5.6.1 is simply toooo old ;) */
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	#if __i386 \|\| __amd64
23	# define STACKSIZE ( 256 * sizeof (long))
24	#elif __ia64	35	#if __ia64
25	# define STACKSIZE (8192 * sizeof (long))	36	# define STACKSIZE 65536
26	#else	37	#else
27	# define STACKSIZE ( 512 * sizeof (long))	38	# define STACKSIZE 4096
28	#endif	39	#endif
29		40
30	enum {	41	enum {
31	REQ_QUIT,	42	REQ_QUIT,
32	REQ_OPEN, REQ_CLOSE,	43	REQ_OPEN, REQ_CLOSE,
33	REQ_READ, REQ_WRITE, REQ_READAHEAD,	44	REQ_READ, REQ_WRITE, REQ_READAHEAD,
		45	REQ_SENDFILE,
34	REQ_STAT, REQ_LSTAT, REQ_FSTAT, REQ_UNLINK,	46	REQ_STAT, REQ_LSTAT, REQ_FSTAT,
35	REQ_FSYNC, REQ_FDATASYNC,	47	REQ_FSYNC, REQ_FDATASYNC,
		48	REQ_UNLINK, REQ_RMDIR,
		49	REQ_SYMLINK,
36	};	50	};
37		51
38	typedef struct aio_cb {	52	typedef struct aio_cb {
39	struct aio_cb *next;	53	struct aio_cb *volatile next;
40		54
41	int type;	55	int type;
42		56
43	int fd;	57	int fd, fd2;
44	off_t offset;	58	off_t offset;
45	size_t length;	59	size_t length;
46	ssize_t result;	60	ssize_t result;
47	mode_t mode; /* open */	61	mode_t mode; /* open */
48	int errorno;	62	int errorno;
49	SV data, callback;	63	SV data, callback;
50	void *dataptr;	64	SV fh, fh2;
		65	void dataptr, data2ptr;
51	STRLEN dataoffset;	66	STRLEN dataoffset;
52		67
53	Stat_t *statdata;	68	Stat_t *statdata;
54	} aio_cb;	69	} aio_cb;
55		70
56	typedef aio_cb *aio_req;	71	typedef aio_cb *aio_req;
57		72
58	static int started;	73	static int started, wanted;
59	static int nreqs;	74	static volatile int nreqs;
		75	static int max_outstanding = 1<<30;
60	static int reqpipe[2], respipe[2];	76	static int respipe [2];
61		77
		78	static pthread_mutex_t reslock = PTHREAD_MUTEX_INITIALIZER;
		79	static pthread_mutex_t reqlock = PTHREAD_MUTEX_INITIALIZER;
		80	static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
		81
62	static aio_req qs, qe; /* queue start, queue end */	82	static volatile aio_req reqs, reqe; /* queue start, queue end */
		83	static volatile aio_req ress, rese; /* queue start, queue end */
63		84
64	static void aio_proc(void arg);	85	static void free_req (aio_req req)
65
66	static void
67	start_thread (void)
68	{	86	{
69	sigset_t fullsigset, oldsigset;	87	if (req->data)
70	pthread_t tid;	88	SvREFCNT_dec (req->data);
71	pthread_attr_t attr;
72		89
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	}
85
86	static void
87	send_reqs (void)
88	{
89	/* this write is atomic */
90	while (qs && write (reqpipe[1], &qs, sizeof qs) == sizeof qs)
91	{
92	qs = qs->next;
93	if (!qs) qe = 0;
94	}
95	}
96
97	static void
98	send_req (aio_req req)
99	{
100	nreqs++;
101	req->next = 0;
102
103	if (qe)
104	{
105	qe->next = req;
106	qe = req;
107	}
108	else
109	qe = qs = req;
110
111	send_reqs ();
112	}
113
114	static void
115	end_thread (void)
116	{
117	aio_req req;
118	New (0, req, 1, aio_cb);
119	req->type = REQ_QUIT;
120
121	send_req (req);
122	}
123
124	static void
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
133	SvUPGRADE (data, SVt_PV);
134	SvPOK_on (data);
135
136	if (dataoffset < 0)
137	dataoffset += svlen;
138
139	if (dataoffset < 0 \|\| dataoffset > svlen)
140	croak ("data offset outside of string");
141
142	if (dowrite)
143	{
144	/* write: check length and adjust. */
145	if (length < 0 \|\| length + dataoffset > svlen)
146	length = svlen - dataoffset;
147	}
148	else
149	{
150	/* read: grow scalar as necessary */
151	svptr = SvGROW (data, length + dataoffset);
152	}
153
154	if (length < 0)
155	croak ("length must not be negative");
156
157	Newz (0, req, 1, aio_cb);
158
159	if (!req)	90	if (req->fh)
160	croak ("out of memory during aio_req allocation");	91	SvREFCNT_dec (req->fh);
161		92
162	req->type = dowrite ? REQ_WRITE : REQ_READ;	93	if (req->fh2)
163	req->fd = fd;	94	SvREFCNT_dec (req->fh2);
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		95
170	send_req (req);	96	if (req->statdata)
		97	Safefree (req->statdata);
		98
		99	if (req->callback)
		100	SvREFCNT_dec (req->callback);
		101
		102	Safefree (req);
171	}	103	}
172		104
173	static void	105	static void
174	poll_wait ()	106	poll_wait ()
175	{	107	{
		108	if (nreqs && !ress)
		109	{
176	fd_set rfd;	110	fd_set rfd;
177	FD_ZERO(&rfd);	111	FD_ZERO(&rfd);
178	FD_SET(respipe[0], &rfd);	112	FD_SET(respipe [0], &rfd);
179		113
180	select (respipe[0] + 1, &rfd, 0, 0, 0);	114	select (respipe [0] + 1, &rfd, 0, 0, 0);
		115	}
181	}	116	}
182		117
183	static int	118	static int
184	poll_cb (pTHX)	119	poll_cb ()
185	{	120	{
186	dSP;	121	dSP;
187	int count = 0;	122	int count = 0;
		123	int do_croak = 0;
188	aio_req req;	124	aio_req req;
189		125
190	while (read (respipe[0], (void *)&req, sizeof (req)) == sizeof (req))	126	for (;;)
191	{	127	{
		128	pthread_mutex_lock (&reslock);
		129	req = ress;
		130
		131	if (req)
		132	{
		133	ress = req->next;
		134
		135	if (!ress)
		136	{
		137	/* read any signals sent by the worker threads */
		138	char buf [32];
		139	while (read (respipe [0], buf, 32) == 32)
		140	;
		141
		142	rese = 0;
		143	}
		144	}
		145
		146	pthread_mutex_unlock (&reslock);
		147
		148	if (!req)
		149	break;
		150
192	nreqs--;	151	nreqs--;
193		152
194	if (req->type == REQ_QUIT)	153	if (req->type == REQ_QUIT)
195	started--;	154	started--;
196	else	155	else
197	{	156	{
198	int errorno = errno;	157	int errorno = errno;
199	errno = req->errorno;	158	errno = req->errorno;
200		159
201	if (req->type == REQ_READ)	160	if (req->type == REQ_READ)
202	SvCUR_set (req->data, req->dataoffset	161	SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
203	+ req->result > 0 ? req->result : 0);
204		162
		163	if (req->data2ptr && (req->type == REQ_READ \|\| req->type == REQ_WRITE))
		164	SvREADONLY_off (req->data);
		165
205	if (req->data)	166	if (req->statdata)
206	SvREFCNT_dec (req->data);
207
208	if (req->type == REQ_STAT \|\| req->type == REQ_LSTAT \|\| req->type == REQ_FSTAT)
209	{	167	{
210	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;	168	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
211	PL_laststatval = req->result;	169	PL_laststatval = req->result;
212	PL_statcache = *(req->statdata);	170	PL_statcache = *(req->statdata);
213
214	Safefree (req->statdata);
215	}	171	}
216		172
		173	ENTER;
217	PUSHMARK (SP);	174	PUSHMARK (SP);
218	XPUSHs (sv_2mortal (newSViv (req->result)));	175	XPUSHs (sv_2mortal (newSViv (req->result)));
219		176
220	if (req->type == REQ_OPEN)	177	if (req->type == REQ_OPEN)
221	{	178	{
224		181
225	PUTBACK;	182	PUTBACK;
226	call_pv ("IO::AIO::_fd2fh", G_SCALAR \| G_EVAL);	183	call_pv ("IO::AIO::_fd2fh", G_SCALAR \| G_EVAL);
227	SPAGAIN;	184	SPAGAIN;
228		185
229	fh = POPs;	186	fh = SvREFCNT_inc (POPs);
230		187
231	PUSHMARK (SP);	188	PUSHMARK (SP);
232	XPUSHs (fh);	189	XPUSHs (sv_2mortal (fh));
233	}	190	}
234		191
		192	if (SvOK (req->callback))
		193	{
235	PUTBACK;	194	PUTBACK;
236	call_sv (req->callback, G_VOID \| G_EVAL);	195	call_sv (req->callback, G_VOID \| G_EVAL);
237	SPAGAIN;	196	SPAGAIN;
		197
		198	if (SvTRUE (ERRSV))
		199	{
		200	free_req (req);
		201	croak (0);
		202	}
238		203	}
239	if (req->callback)	204
240	SvREFCNT_dec (req->callback);	205	LEAVE;
241		206
242	errno = errorno;	207	errno = errorno;
243	count++;	208	count++;
244	}	209	}
245		210
246	Safefree (req);	211	free_req (req);
247	}	212	}
248		213
249	if (qs)
250	send_reqs ();
251
252	return count;	214	return count;
253	}	215	}
		216
		217	static void aio_proc(void arg);
		218
		219	static void
		220	start_thread (void)
		221	{
		222	sigset_t fullsigset, oldsigset;
		223	pthread_t tid;
		224	pthread_attr_t attr;
		225
		226	pthread_attr_init (&attr);
		227	pthread_attr_setstacksize (&attr, STACKSIZE);
		228	pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
		229
		230	sigfillset (&fullsigset);
		231	sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
		232
		233	if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
		234	started++;
		235
		236	sigprocmask (SIG_SETMASK, &oldsigset, 0);
		237	}
		238
		239	static void
		240	send_req (aio_req req)
		241	{
		242	while (started < wanted && nreqs >= started)
		243	start_thread ();
		244
		245	nreqs++;
		246
		247	pthread_mutex_lock (&reqlock);
		248
		249	req->next = 0;
		250
		251	if (reqe)
		252	{
		253	reqe->next = req;
		254	reqe = req;
		255	}
		256	else
		257	reqe = reqs = req;
		258
		259	pthread_cond_signal (&reqwait);
		260	pthread_mutex_unlock (&reqlock);
		261
		262	if (nreqs > max_outstanding)
		263	for (;;)
		264	{
		265	poll_cb ();
		266
		267	if (nreqs <= max_outstanding)
		268	break;
		269
		270	poll_wait ();
		271	}
		272	}
		273
		274	static void
		275	end_thread (void)
		276	{
		277	aio_req req;
		278	Newz (0, req, 1, aio_cb);
		279	req->type = REQ_QUIT;
		280
		281	send_req (req);
		282	}
		283
		284	static void min_parallel (int nthreads)
		285	{
		286	if (wanted < nthreads)
		287	wanted = nthreads;
		288	}
		289
		290	static void max_parallel (int nthreads)
		291	{
		292	int cur = started;
		293
		294	if (wanted > nthreads)
		295	wanted = nthreads;
		296
		297	while (cur > wanted)
		298	{
		299	end_thread ();
		300	cur--;
		301	}
		302
		303	while (started > wanted)
		304	{
		305	poll_wait ();
		306	poll_cb ();
		307	}
		308	}
		309
		310	static void create_pipe ()
		311	{
		312	if (pipe (respipe))
		313	croak ("unable to initialize result pipe");
		314
		315	if (fcntl (respipe [0], F_SETFL, O_NONBLOCK))
		316	croak ("cannot set result pipe to nonblocking mode");
		317
		318	if (fcntl (respipe [1], F_SETFL, O_NONBLOCK))
		319	croak ("cannot set result pipe to nonblocking mode");
		320	}
		321
		322	static void atfork_prepare (void)
		323	{
		324	pthread_mutex_lock (&reqlock);
		325	pthread_mutex_lock (&reslock);
		326	}
		327
		328	static void atfork_parent (void)
		329	{
		330	pthread_mutex_unlock (&reslock);
		331	pthread_mutex_unlock (&reqlock);
		332	}
		333
		334	static void atfork_child (void)
		335	{
		336	aio_req prv;
		337
		338	started = 0;
		339
		340	while (reqs)
		341	{
		342	prv = reqs;
		343	reqs = prv->next;
		344	free_req (prv);
		345	}
		346
		347	reqs = reqe = 0;
		348
		349	while (ress)
		350	{
		351	prv = ress;
		352	ress = prv->next;
		353	free_req (prv);
		354	}
		355
		356	ress = rese = 0;
		357
		358	close (respipe [0]);
		359	close (respipe [1]);
		360	create_pipe ();
		361
		362	atfork_parent ();
		363	}
		364
		365	/* currently noops */
		366	#define LOCK_FD(fd) do { } while (0)
		367	#define UNLOCK_FD(fd) do { } while (0)
		368
		369	/*****************************************************************************/
		370	/* work around various missing functions */
		371
		372	#if !HAVE_PREADWRITE
		373	# define pread aio_pread
		374	# define pwrite aio_pwrite
		375
		376	/*
		377	* make our pread/pwrite safe against themselves, but not against
		378	* normal read/write by using a mutex. slows down execution a lot,
		379	* but that's your problem, not mine.
		380	*/
		381	static pthread_mutex_t iolock = PTHREAD_MUTEX_INITIALIZER;
		382
		383	static ssize_t
		384	pread (int fd, void *buf, size_t count, off_t offset)
		385	{
		386	ssize_t res;
		387	off_t ooffset;
		388
		389	LOCK_FD (fd);
		390	pthread_mutex_lock (&iolock); /* replace by LOCK_FD and private buffer */
		391	ooffset = lseek (fd, 0, SEEK_CUR);
		392	lseek (fd, offset, SEEK_SET);
		393	res = read (fd, buf, count);
		394	lseek (fd, ooffset, SEEK_SET);
		395	pthread_mutex_unlock (&iolock);
		396	UNLOCK_FD (d);
		397
		398	return res;
		399	}
		400
		401	static ssize_t
		402	pwrite (int fd, void *buf, size_t count, off_t offset)
		403	{
		404	ssize_t res;
		405	off_t ooffset;
		406
		407	LOCK_FD (fd);
		408	pthread_mutex_lock (&iolock); /* replace by LOCK_FD and private buffer */
		409	ooffset = lseek (fd, 0, SEEK_CUR);
		410	lseek (fd, offset, SEEK_SET);
		411	res = write (fd, buf, count);
		412	lseek (fd, offset, SEEK_SET);
		413	pthread_mutex_unlock (&iolock);
		414	UNLOCK_FD (d);
		415
		416	return res;
		417	}
		418	#endif
		419
		420	#if !HAVE_FDATASYNC
		421	# define fdatasync fsync
		422	#endif
		423
		424	#if !HAVE_READAHEAD
		425	# define readahead aio_readahead
		426
		427	static char readahead_buf[4096];
		428
		429	static ssize_t
		430	readahead (int fd, off_t offset, size_t count)
		431	{
		432	while (count > 0)
		433	{
		434	size_t len = count < sizeof (readahead_buf) ? count : sizeof (readahead_buf);
		435
		436	pread (fd, readahead_buf, len, offset);
		437	offset += len;
		438	count -= len;
		439	}
		440
		441	errno = 0;
		442	}
		443	#endif
		444
		445	/* sendfile always needs emulation */
		446	static ssize_t
		447	sendfile_ (int ofd, int ifd, off_t offset, size_t count)
		448	{
		449	ssize_t res;
		450
		451	if (!count)
		452	return 0;
		453
		454	LOCK_FD (ofd);
		455
		456	#if HAVE_SENDFILE
		457	# if __linux
		458	res = sendfile (ofd, ifd, &offset, count);
		459
		460	# elif __freebsd
		461	/*
		462	* Of course, the freebsd sendfile is a dire hack with no thoughts
		463	* wasted on making it similar to other I/O functions.
		464	*/
		465	{
		466	off_t sbytes;
		467	res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
		468
		469	if (res < 0 && sbytes)
		470	/* maybe only on EAGAIN only: as usual, the manpage leaves you guessing */
		471	res = sbytes;
		472	}
		473
		474	# elif __hpux
		475	res = sendfile (ofd, ifd, offset, count, 0, 0);
		476
		477	# elif __solaris
		478	{
		479	struct sendfilevec vec;
		480	size_t sbytes;
		481
		482	vec.sfv_fd = ifd;
		483	vec.sfv_flag = 0;
		484	vec.sfv_off = offset;
		485	vec.sfv_len = count;
		486
		487	res = sendfilev (ofd, &vec, 1, &sbytes);
		488
		489	if (res < 0 && sbytes)
		490	res = sbytes;
		491	}
		492
		493	# else
		494	res = -1;
		495	errno = ENOSYS;
		496	# endif
		497	#endif
		498
		499	if (res < 0
		500	&& (errno == ENOSYS \|\| errno == EINVAL \|\| errno == ENOTSOCK
		501	#if __solaris
		502	\|\| errno == EAFNOSUPPORT \|\| errno == EPROTOTYPE
		503	#endif
		504	)
		505	)
		506	{
		507	/* emulate sendfile. this is a major pain in the ass */
		508	char *buf = malloc (4096);
		509	res = 0;
		510
		511	for (;;)
		512	{
		513	ssize_t cnt;
		514
		515	cnt = pread (ifd, buf, 4096, offset);
		516
		517	if (cnt <= 0)
		518	{
		519	if (cnt && !res) res = -1;
		520	break;
		521	}
		522
		523	cnt = write (ofd, buf, cnt);
		524
		525	if (cnt <= 0)
		526	{
		527	if (cnt && !res) res = -1;
		528	break;
		529	}
		530
		531	offset += cnt;
		532	res += cnt;
		533	}
		534
		535	{
		536	int errorno = errno;
		537	free (buf);
		538	errno = errorno;
		539	}
		540	}
		541
		542	UNLOCK_FD (ofd);
		543
		544	return res;
		545	}
		546
		547	/*****************************************************************************/
254		548
255	static void *	549	static void *
256	aio_proc (void *thr_arg)	550	aio_proc (void *thr_arg)
257	{	551	{
258	aio_req req;	552	aio_req req;
		553	int type;
259		554
260	/* then loop */	555	do
261	while (read (reqpipe[0], (void *)&req, sizeof (req)) == sizeof (req))
262	{	556	{
		557	pthread_mutex_lock (&reqlock);
		558
		559	for (;;)
		560	{
		561	req = reqs;
		562
		563	if (reqs)
		564	{
		565	reqs = reqs->next;
		566	if (!reqs) reqe = 0;
		567	}
		568
		569	if (req)
		570	break;
		571
		572	pthread_cond_wait (&reqwait, &reqlock);
		573	}
		574
		575	pthread_mutex_unlock (&reqlock);
		576
263	errno = 0; /* strictly unnecessary */	577	errno = 0; /* strictly unnecessary */
264		578
		579	type = req->type;
		580
265	switch (req->type)	581	switch (type)
266	{	582	{
267	case REQ_READ: req->result = pread64 (req->fd, req->dataptr, req->length, req->offset); break;	583	case REQ_READ: req->result = pread (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;	584	case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
269	#if SYS_readahead	585
270	case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;	586	case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
271	#else	587	case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length); break;
272	case REQ_READAHEAD: req->result = -1; errno = ENOSYS; break;
273	#endif
274		588
275	case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;	589	case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
276	case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;	590	case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
277	case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;	591	case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
278		592
279	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;	593	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
280	case REQ_CLOSE: req->result = close (req->fd); break;	594	case REQ_CLOSE: req->result = close (req->fd); break;
281	case REQ_UNLINK: req->result = unlink (req->dataptr); break;	595	case REQ_UNLINK: req->result = unlink (req->dataptr); break;
		596	case REQ_RMDIR: req->result = rmdir (req->dataptr); break;
		597	case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;
282		598
		599	case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
283	case REQ_FSYNC: req->result = fsync (req->fd); break;	600	case REQ_FSYNC: req->result = fsync (req->fd); break;
284	case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
285		601
286	case REQ_QUIT:	602	case REQ_QUIT:
287	write (respipe[1], (void *)&req, sizeof (req));	603	break;
288	return 0;
289		604
290	default:	605	default:
291	req->result = ENOSYS;	606	req->result = ENOSYS;
292	break;	607	break;
293	}	608	}
294		609
295	req->errorno = errno;	610	req->errorno = errno;
296	write (respipe[1], (void *)&req, sizeof (req));	611
		612	pthread_mutex_lock (&reslock);
		613
		614	req->next = 0;
		615
		616	if (rese)
		617	{
		618	rese->next = req;
		619	rese = req;
		620	}
		621	else
		622	{
		623	rese = ress = req;
		624
		625	/* write a dummy byte to the pipe so fh becomes ready */
		626	write (respipe [1], &respipe, 1);
		627	}
		628
		629	pthread_mutex_unlock (&reslock);
297	}	630	}
		631	while (type != REQ_QUIT);
298		632
299	return 0;	633	return 0;
300	}	634	}
301		635
		636	#define dREQ \
		637	aio_req req; \
		638	\
		639	if (SvOK (callback) && !SvROK (callback)) \
		640	croak ("clalback must be undef or of reference type"); \
		641	\
		642	Newz (0, req, 1, aio_cb); \
		643	if (!req) \
		644	croak ("out of memory during aio_req allocation"); \
		645	\
		646	req->callback = newSVsv (callback);
		647
302	MODULE = IO::AIO PACKAGE = IO::AIO	648	MODULE = IO::AIO PACKAGE = IO::AIO
303		649
		650	PROTOTYPES: ENABLE
		651
304	BOOT:	652	BOOT:
305	{	653	{
306	if (pipe (reqpipe) \|\| pipe (respipe))	654	create_pipe ();
307	croak ("unable to initialize request or result pipe");	655	pthread_atfork (atfork_prepare, atfork_parent, atfork_child);
308
309	if (fcntl (reqpipe[1], F_SETFL, O_NONBLOCK))
310	croak ("cannot set result pipe to nonblocking mode");
311
312	if (fcntl (respipe[0], F_SETFL, O_NONBLOCK))
313	croak ("cannot set result pipe to nonblocking mode");
314	}	656	}
315		657
316	void	658	void
317	min_parallel(nthreads)	659	min_parallel(nthreads)
318	int nthreads	660	int nthreads
319	PROTOTYPE: $	661	PROTOTYPE: $
320	CODE:
321	while (nthreads > started)
322	start_thread ();
323		662
324	void	663	void
325	max_parallel(nthreads)	664	max_parallel(nthreads)
326	int nthreads	665	int nthreads
327	PROTOTYPE: $	666	PROTOTYPE: $
		667
		668	int
		669	max_outstanding(nreqs)
		670	int nreqs
		671	PROTOTYPE: $
328	CODE:	672	CODE:
329	{	673	RETVAL = max_outstanding;
330	int cur = started;	674	max_outstanding = nreqs;
331	while (cur > nthreads)
332	{
333	end_thread ();
334	cur--;
335	}
336		675
337	while (started > nthreads)
338	{
339	poll_wait ();
340	poll_cb (aTHX);
341	}
342	}
343
344	void	676	void
345	aio_open(pathname,flags,mode,callback)	677	aio_open(pathname,flags,mode,callback=&PL_sv_undef)
346	SV * pathname	678	SV * pathname
347	int flags	679	int flags
348	int mode	680	int mode
349	SV * callback	681	SV * callback
350	PROTOTYPE: $$$$	682	PROTOTYPE: $$$;$
351	CODE:	683	CODE:
352	{	684	{
353	aio_req req;	685	dREQ;
354
355	Newz (0, req, 1, aio_cb);
356
357	if (!req)
358	croak ("out of memory during aio_req allocation");
359		686
360	req->type = REQ_OPEN;	687	req->type = REQ_OPEN;
361	req->data = newSVsv (pathname);	688	req->data = newSVsv (pathname);
362	req->dataptr = SvPV_nolen (req->data);	689	req->dataptr = SvPVbyte_nolen (req->data);
363	req->fd = flags;	690	req->fd = flags;
364	req->mode = mode;	691	req->mode = mode;
365	req->callback = SvREFCNT_inc (callback);
366		692
367	send_req (req);	693	send_req (req);
368	}	694	}
369		695
370	void	696	void
371	aio_close(fh,callback)	697	aio_close(fh,callback=&PL_sv_undef)
372	InputStream fh	698	SV * fh
373	SV * callback	699	SV * callback
374	PROTOTYPE: $$	700	PROTOTYPE: $;$
375	ALIAS:	701	ALIAS:
376	aio_close = REQ_CLOSE	702	aio_close = REQ_CLOSE
377	aio_fsync = REQ_FSYNC	703	aio_fsync = REQ_FSYNC
378	aio_fdatasync = REQ_FDATASYNC	704	aio_fdatasync = REQ_FDATASYNC
379	CODE:	705	CODE:
380	{	706	{
		707	dREQ;
		708
		709	req->type = ix;
		710	req->fh = newSVsv (fh);
		711	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
		712
		713	send_req (req);
		714	}
		715
		716	void
		717	aio_read(fh,offset,length,data,dataoffset,callback=&PL_sv_undef)
		718	SV * fh
		719	UV offset
		720	UV length
		721	SV * data
		722	UV dataoffset
		723	SV * callback
		724	ALIAS:
		725	aio_read = REQ_READ
		726	aio_write = REQ_WRITE
		727	PROTOTYPE: $$$$$;$
		728	CODE:
		729	{
381	aio_req req;	730	aio_req req;
		731	STRLEN svlen;
		732	char *svptr = SvPVbyte (data, svlen);
382		733
383	Newz (0, req, 1, aio_cb);	734	SvUPGRADE (data, SVt_PV);
		735	SvPOK_on (data);
384		736
385	if (!req)	737	if (dataoffset < 0)
386	croak ("out of memory during aio_req allocation");	738	dataoffset += svlen;
387		739
388	req->type = ix;	740	if (dataoffset < 0 \|\| dataoffset > svlen)
389	req->fd = PerlIO_fileno (fh);	741	croak ("data offset outside of string");
390	req->callback = SvREFCNT_inc (callback);
391		742
392	send_req (req);	743	if (ix == REQ_WRITE)
393	}	744	{
394		745	/* write: check length and adjust. */
395	void	746	if (length < 0 \|\| length + dataoffset > svlen)
396	aio_read(fh,offset,length,data,dataoffset,callback)	747	length = svlen - dataoffset;
397	InputStream fh	748	}
398	UV offset	749	else
399	IV length	750	{
400	SV * data	751	/* read: grow scalar as necessary */
401	IV dataoffset	752	svptr = SvGROW (data, length + dataoffset);
402	SV * callback	753	}
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		754
430	if (length < 0)	755	if (length < 0)
431	croak ("length must not be negative");	756	croak ("length must not be negative");
432		757
433	Newz (0, req, 1, aio_cb);	758	{
		759	dREQ;
434		760
435	if (!req)	761	req->type = ix;
436	croak ("out of memory during aio_req allocation");	762	req->fh = newSVsv (fh);
		763	req->fd = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))
		764	: IoOFP (sv_2io (fh)));
		765	req->offset = offset;
		766	req->length = length;
		767	req->data = SvREFCNT_inc (data);
		768	req->dataptr = (char *)svptr + dataoffset;
		769
		770	if (!SvREADONLY (data))
		771	{
		772	SvREADONLY_on (data);
		773	req->data2ptr = (void *)data;
		774	}
		775
		776	send_req (req);
		777	}
		778	}
		779
		780	void
		781	aio_sendfile(out_fh,in_fh,in_offset,length,callback=&PL_sv_undef)
		782	SV * out_fh
		783	SV * in_fh
		784	UV in_offset
		785	UV length
		786	SV * callback
		787	PROTOTYPE: $$$$;$
		788	CODE:
		789	{
		790	dREQ;
		791
		792	req->type = REQ_SENDFILE;
		793	req->fh = newSVsv (out_fh);
		794	req->fd = PerlIO_fileno (IoIFP (sv_2io (out_fh)));
		795	req->fh2 = newSVsv (in_fh);
		796	req->fd2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));
		797	req->offset = in_offset;
		798	req->length = length;
		799
		800	send_req (req);
		801	}
		802
		803	void
		804	aio_readahead(fh,offset,length,callback=&PL_sv_undef)
		805	SV * fh
		806	UV offset
		807	IV length
		808	SV * callback
		809	PROTOTYPE: $$$;$
		810	CODE:
		811	{
		812	dREQ;
437		813
438	req->type = REQ_READAHEAD;	814	req->type = REQ_READAHEAD;
		815	req->fh = newSVsv (fh);
439	req->fd = PerlIO_fileno (fh);	816	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
440	req->offset = offset;	817	req->offset = offset;
441	req->length = length;	818	req->length = length;
442	req->callback = SvREFCNT_inc (callback);
443		819
444	send_req (req);	820	send_req (req);
445	}	821	}
446		822
447	void	823	void
448	aio_stat(fh_or_path,callback)	824	aio_stat(fh_or_path,callback=&PL_sv_undef)
449	SV * fh_or_path	825	SV * fh_or_path
450	SV * callback	826	SV * callback
451	PROTOTYPE: $$
452	ALIAS:	827	ALIAS:
		828	aio_stat = REQ_STAT
453	aio_lstat = 1	829	aio_lstat = REQ_LSTAT
454	CODE:	830	CODE:
455	{	831	{
456	aio_req req;	832	dREQ;
457
458	Newz (0, req, 1, aio_cb);
459
460	if (!req)
461	croak ("out of memory during aio_req allocation");
462		833
463	New (0, req->statdata, 1, Stat_t);	834	New (0, req->statdata, 1, Stat_t);
464
465	if (!req->statdata)	835	if (!req->statdata)
		836	{
		837	free_req (req);
466	croak ("out of memory during aio_req->statdata allocation");	838	croak ("out of memory during aio_req->statdata allocation");
		839	}
467		840
468	if (SvPOK (fh_or_path))	841	if (SvPOK (fh_or_path))
469	{	842	{
470	req->type = ix ? REQ_LSTAT : REQ_STAT;	843	req->type = ix;
471	req->data = newSVsv (fh_or_path);	844	req->data = newSVsv (fh_or_path);
472	req->dataptr = SvPV_nolen (req->data);	845	req->dataptr = SvPVbyte_nolen (req->data);
473	}	846	}
474	else	847	else
475	{	848	{
476	req->type = REQ_FSTAT;	849	req->type = REQ_FSTAT;
		850	req->fh = newSVsv (fh_or_path);
477	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));	851	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
478	}	852	}
479		853
480	req->callback = SvREFCNT_inc (callback);
481
482	send_req (req);	854	send_req (req);
483	}	855	}
484		856
485	void	857	void
486	aio_unlink(pathname,callback)	858	aio_unlink(pathname,callback=&PL_sv_undef)
487	SV * pathname	859	SV * pathname
488	SV * callback	860	SV * callback
489	PROTOTYPE: $$	861	ALIAS:
		862	aio_unlink = REQ_UNLINK
		863	aio_rmdir = REQ_RMDIR
490	CODE:	864	CODE:
491	{	865	{
492	aio_req req;	866	dREQ;
493		867
494	Newz (0, req, 1, aio_cb);	868	req->type = ix;
495
496	if (!req)
497	croak ("out of memory during aio_req allocation");
498
499	req->type = REQ_UNLINK;
500	req->data = newSVsv (pathname);	869	req->data = newSVsv (pathname);
501	req->dataptr = SvPV_nolen (req->data);	870	req->dataptr = SvPVbyte_nolen (req->data);
502	req->callback = SvREFCNT_inc (callback);
503		871
504	send_req (req);	872	send_req (req);
505	}	873	}
		874
		875	void
		876	aio_symlink(oldpath,newpath,callback=&PL_sv_undef)
		877	SV * oldpath
		878	SV * newpath
		879	SV * callback
		880	CODE:
		881	{
		882	dREQ;
		883
		884	req->type = REQ_SYMLINK;
		885	req->fh = newSVsv (oldpath);
		886	req->data2ptr = SvPVbyte_nolen (req->fh);
		887	req->data = newSVsv (newpath);
		888	req->dataptr = SvPVbyte_nolen (req->data);
		889
		890	send_req (req);
		891	}
		892
		893	void
		894	flush()
		895	PROTOTYPE:
		896	CODE:
		897	while (nreqs)
		898	{
		899	poll_wait ();
		900	poll_cb ();
		901	}
		902
		903	void
		904	poll()
		905	PROTOTYPE:
		906	CODE:
		907	if (nreqs)
		908	{
		909	poll_wait ();
		910	poll_cb ();
		911	}
506		912
507	int	913	int
508	poll_fileno()	914	poll_fileno()
509	PROTOTYPE:	915	PROTOTYPE:
510	CODE:	916	CODE:
511	RETVAL = respipe[0];	917	RETVAL = respipe [0];
512	OUTPUT:	918	OUTPUT:
513	RETVAL	919	RETVAL
514		920
515	int	921	int
516	poll_cb(...)	922	poll_cb(...)
517	PROTOTYPE:	923	PROTOTYPE:
518	CODE:	924	CODE:
519	RETVAL = poll_cb (aTHX);	925	RETVAL = poll_cb ();
520	OUTPUT:	926	OUTPUT:
521	RETVAL	927	RETVAL
522		928
523	void	929	void
524	poll_wait()	930	poll_wait()
525	PROTOTYPE:	931	PROTOTYPE:
526	CODE:	932	CODE:
		933	if (nreqs)
527	poll_wait ();	934	poll_wait ();
528		935
529	int	936	int
530	nreqs()	937	nreqs()
531	PROTOTYPE:	938	PROTOTYPE:
532	CODE:	939	CODE:

Diff Legend

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

Comparing IO-AIO/AIO.xs (file contents): Revision 1.2 by root, Sun Jul 10 18:16:49 2005 UTC vs. Revision 1.35 by root, Tue Aug 23 01:16:50 2005 UTC

Diff Legend

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.2 by root, Sun Jul 10 18:16:49 2005 UTC vs.
Revision 1.35 by root, Tue Aug 23 01:16:50 2005 UTC