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

Comparing Linux-AIO/AIO.xs (file contents):
Revision 1.8 by root, Thu Aug 16 02:43:46 2001 UTC vs.
Revision 1.23 by root, Wed Jun 29 15:28:14 2005 UTC

…		…
		1	#define PERL_NO_GET_CONTEXT
		2
1	#include "EXTERN.h"	3	#include "EXTERN.h"
2	#include "perl.h"	4	#include "perl.h"
3	#include "XSUB.h"	5	#include "XSUB.h"
4		6
5	#include <sys/types.h>	7	#include <sys/types.h>
		8	#include <sys/stat.h>
6	#include <unistd.h>	9	#include <unistd.h>
7	#include <fcntl.h>	10	#include <fcntl.h>
8	#include <signal.h>	11	#include <signal.h>
9	#include <sched.h>	12	#include <sched.h>
10		13
11	#define STACKSIZE 1024 /* yeah */	14	typedef void InputStream; / hack, but 5.6.1 is simply toooo old ;) */
		15	typedef void OutputStream; / hack, but 5.6.1 is simply toooo old ;) */
		16	typedef void InOutStream; / hack, but 5.6.1 is simply toooo old ;) */
12		17
13	enum { REQ_QUIT, REQ_READ, REQ_WRITE, REQ_OPEN };	18	#define STACKSIZE (128 * sizeof (long)) /* yeah */
		19
		20	enum {
		21	REQ_QUIT,
		22	REQ_OPEN, REQ_CLOSE, REQ_READ, REQ_WRITE,
		23	REQ_STAT, REQ_LSTAT, REQ_FSTAT, REQ_UNLINK
		24	};
14		25
15	typedef struct {	26	typedef struct {
16	char stack[STACKSIZE];	27	char stack[STACKSIZE];
17	} aio_thread;	28	} aio_thread;
18		29
19	typedef struct {	30	typedef struct aio_cb {
		31	struct aio_cb *next;
		32
20	int type;	33	int type;
21	aio_thread *thread;	34	aio_thread *thread;
22		35
23	/* read/write */
24	int fd;	36	int fd;
25	off_t offset;	37	off_t offset;
26	size_t length;	38	size_t length;
27	ssize_t result;	39	ssize_t result;
28	mode_t mode; /* open */	40	mode_t mode; /* open */
29	int errorno;	41	int errorno;
30	SV data, callback;	42	SV data, callback;
31	void *dataptr;	43	void *dataptr;
32	STRLEN dataoffset;	44	STRLEN dataoffset;
		45
		46	Stat_t *statdata;
33	} aio_cb;	47	} aio_cb;
34		48
35	typedef aio_cb *aio_req;	49	typedef aio_cb *aio_req;
36		50
37	static int started;	51	static int started;
38	static int nreqs;	52	static int nreqs;
39	static int reqpipe[2], respipe[2];	53	static int reqpipe[2], respipe[2];
40		54
		55	static aio_req qs, qe; /* queue start, queue end */
		56
41	static int aio_proc(void *arg);	57	static int aio_proc(void *arg);
42		58
43	static void	59	static void
44	start_thread(void)	60	start_thread (void)
45	{	61	{
46	aio_thread *thr;	62	aio_thread *thr;
47		63
48	New (0, thr, 1, aio_thread);	64	New (0, thr, 1, aio_thread);
49		65
50	if (clone (aio_proc,	66	if (clone (aio_proc,
51	&(thr->stack[STACKSIZE]),	67	&(thr->stack[STACKSIZE - sizeof (long)]),
52	CLONE_VM\|CLONE_FS\|CLONE_FILES\|CLONE_SIGHAND,	68	CLONE_VM\|CLONE_FS\|CLONE_FILES,
53	thr) >= 0)	69	thr) >= 0)
54	started++;	70	started++;
55	else	71	else
56	Safefree (thr);	72	Safefree (thr);
57	}	73	}
58		74
59	static void	75	static void
		76	send_reqs (void)
		77	{
		78	/* this write is atomic */
		79	while (qs && write (reqpipe[1], &qs, sizeof qs) == sizeof qs)
		80	{
		81	qs = qs->next;
		82	if (!qs) qe = 0;
		83	}
		84	}
		85
		86	static void
		87	send_req (aio_req req)
		88	{
		89	nreqs++;
		90	req->next = 0;
		91
		92	if (qe)
		93	{
		94	qe->next = req;
		95	qe = req;
		96	}
		97	else
		98	qe = qs = req;
		99
		100	send_reqs ();
		101	}
		102
		103	static void
60	end_thread(void)	104	end_thread (void)
61	{	105	{
62	aio_req req;	106	aio_req req;
63	New (0, req, 1, aio_cb);	107	New (0, req, 1, aio_cb);
64	req->type = REQ_QUIT;	108	req->type = REQ_QUIT;
65	write (reqpipe[1], &req, sizeof (aio_req));	109
		110	send_req (req);
66	}	111	}
67		112
68	static void	113	static void
69	send_req (aio_req req)	114	read_write (pTHX_
70	{
71	nreqs++;
72	write (reqpipe[1], &req, sizeof (aio_req));
73	}
74
75	static void
76	read_write (pTHX_ int dowrite, int fd, off_t offset, size_t length,	115	int dowrite, int fd, off_t offset, size_t length,
77	SV data, STRLEN dataoffset, SVcallback)	116	SV data, STRLEN dataoffset, SV callback)
78	{	117	{
79	aio_req req;	118	aio_req req;
80	STRLEN svlen;	119	STRLEN svlen;
81	char *svptr = SvPV (data, svlen);	120	char *svptr = SvPV (data, svlen);
		121
		122	SvUPGRADE (data, SVt_PV);
		123	SvPOK_on (data);
82		124
83	if (dataoffset < 0)	125	if (dataoffset < 0)
84	dataoffset += svlen;	126	dataoffset += svlen;
85		127
86	if (dataoffset < 0 \|\| dataoffset > svlen)	128	if (dataoffset < 0 \|\| dataoffset > svlen)
99	}	141	}
100		142
101	if (length < 0)	143	if (length < 0)
102	croak ("length must not be negative");	144	croak ("length must not be negative");
103		145
104	New (0, req, 1, aio_cb);	146	Newz (0, req, 1, aio_cb);
105		147
106	if (!req)	148	if (!req)
107	croak ("out of memory during aio_req allocation");	149	croak ("out of memory during aio_req allocation");
108		150
109	req->type = dowrite ? REQ_WRITE : REQ_READ;	151	req->type = dowrite ? REQ_WRITE : REQ_READ;
…		…
115	req->callback = SvREFCNT_inc (callback);	157	req->callback = SvREFCNT_inc (callback);
116		158
117	send_req (req);	159	send_req (req);
118	}	160	}
119		161
		162	static void
		163	poll_wait ()
		164	{
		165	fd_set rfd;
		166	FD_ZERO(&rfd);
		167	FD_SET(respipe[0], &rfd);
		168
		169	select (respipe[0] + 1, &rfd, 0, 0, 0);
		170	}
		171
120	static int	172	static int
121	poll_cb (pTHX)	173	poll_cb (pTHX)
122	{	174	{
123	dSP;	175	dSP;
124	int count = 0;	176	int count = 0;
125	aio_req req;	177	aio_req req;
126		178
127	while (read (respipe[0], (void *)&req, sizeof (req)) == sizeof (req))	179	while (read (respipe[0], (void *)&req, sizeof (req)) == sizeof (req))
128	{	180	{
		181	nreqs--;
		182
129	if (req->type == REQ_QUIT)	183	if (req->type == REQ_QUIT)
130	{	184	{
131	Safefree (req->thread);	185	Safefree (req->thread);
132	started--;	186	started--;
133	}	187	}
…		…
138		192
139	if (req->type == REQ_READ)	193	if (req->type == REQ_READ)
140	SvCUR_set (req->data, req->dataoffset	194	SvCUR_set (req->data, req->dataoffset
141	+ req->result > 0 ? req->result : 0);	195	+ req->result > 0 ? req->result : 0);
142		196
		197	if (req->data)
		198	SvREFCNT_dec (req->data);
		199
		200	if (req->type == REQ_STAT \|\| req->type == REQ_LSTAT \|\| req->type == REQ_FSTAT)
		201	{
		202	PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
		203	PL_laststatval = req->result;
		204	PL_statcache = *(req->statdata);
		205
		206	Safefree (req->statdata);
		207	}
		208
143	PUSHMARK (SP);	209	PUSHMARK (SP);
144	XPUSHs (sv_2mortal (newSViv (req->result)));	210	XPUSHs (sv_2mortal (newSViv (req->result)));
145	PUTBACK;	211	PUTBACK;
146	call_sv (req->callback, G_VOID);	212	call_sv (req->callback, G_VOID);
147	SPAGAIN;	213	SPAGAIN;
148		214
149	SvREFCNT_dec (req->data);	215	if (req->callback)
150	SvREFCNT_dec (req->callback);	216	SvREFCNT_dec (req->callback);
151		217
152	errno = errorno;	218	errno = errorno;
153	nreqs--;
154	count++;	219	count++;
155	}	220	}
156		221
157	Safefree (req);	222	Safefree (req);
158	}	223	}
159		224
		225	if (qs)
		226	send_reqs ();
		227
160	return count;	228	return count;
161	}	229	}
162		230
163	static sigset_t fullsigset;	231	static sigset_t fullsigset;
164		232
165	#undef errno	233	#undef errno
166	#include <asm/unistd.h>	234	#include <asm/unistd.h>
		235	#include <sys/prctl.h>
		236
		237	#define COPY_STATDATA \
		238	req->statdata->st_dev = statdata.st_dev; \
		239	req->statdata->st_ino = statdata.st_ino; \
		240	req->statdata->st_mode = statdata.st_mode; \
		241	req->statdata->st_nlink = statdata.st_nlink; \
		242	req->statdata->st_uid = statdata.st_uid; \
		243	req->statdata->st_gid = statdata.st_gid; \
		244	req->statdata->st_rdev = statdata.st_rdev; \
		245	req->statdata->st_size = statdata.st_size; \
		246	req->statdata->st_atime = statdata.st_atime; \
		247	req->statdata->st_mtime = statdata.st_mtime; \
		248	req->statdata->st_ctime = statdata.st_ctime; \
		249	req->statdata->st_blksize = statdata.st_blksize; \
		250	req->statdata->st_blocks = statdata.st_blocks; \
167		251
168	static int	252	static int
169	aio_proc(void *thr_arg)	253	aio_proc (void *thr_arg)
170	{	254	{
171	aio_thread *thr = thr_arg;	255	aio_thread *thr = thr_arg;
172	int sig;	256	aio_req req;
173	int errno;	257	int errno;
174	aio_req req;	258
		259	/* this is very much kernel-specific :(:(:( */
		260	/* we rely on gcc's ability to create closures. */
		261	_syscall3(int,read,int,fd,char *,buf,size_t,count)
		262	_syscall3(int,write,int,fd,char *,buf,size_t,count)
		263
		264	_syscall3(int,open,char *,pathname,int,flags,mode_t,mode)
		265	_syscall1(int,close,int,fd)
		266
		267	#ifdef __NR_pread64
		268	_syscall5(int,pread64,int,fd,char *,buf,size_t,count,unsigned int,offset_lo,unsigned int,offset_hi)
		269	_syscall5(int,pwrite64,int,fd,char *,buf,size_t,count,unsigned int,offset_lo,unsigned int,offset_hi)
		270	#elif __NR_pread
		271	_syscall4(int,pread,int,fd,char *,buf,size_t,count,offset_t,offset)
		272	_syscall4(int,pwrite,int,fd,char *,buf,size_t,count,offset_t,offset)
		273	#else
		274	# error "neither pread nor pread64 defined"
		275	#endif
		276
		277
		278	#ifdef __NR_stat64
		279	_syscall2(int,stat64, const char , filename, struct stat64 , buf)
		280	_syscall2(int,lstat64, const char , filename, struct stat64 , buf)
		281	_syscall2(int,fstat64, int, fd, struct stat64 *, buf)
		282	#elif __NR_stat
		283	_syscall2(int,stat, const char , filename, struct stat , buf)
		284	_syscall2(int,lstat, const char , filename, struct stat , buf)
		285	_syscall2(int,fstat, int, fd, struct stat *, buf)
		286	#else
		287	# error "neither stat64 nor stat defined"
		288	#endif
		289
		290	_syscall1(int,unlink, char *, filename);
175		291
176	sigprocmask (SIG_SETMASK, &fullsigset, 0);	292	sigprocmask (SIG_SETMASK, &fullsigset, 0);
177		293	prctl (PR_SET_PDEATHSIG, SIGKILL);
178	/* we rely on gcc's ability to create closures. */
179	_syscall3(int,lseek,int,fd,off_t,offset,int,whence);
180	_syscall3(int,read,int,fd,char *,buf,off_t,count);
181	_syscall3(int,write,int,fd,char *,buf,off_t,count);
182	_syscall3(int,open,char *,pathname,int,flags,mode_t,mode);
183		294
184	/* then loop */	295	/* then loop */
185	while (read (reqpipe[0], (void *)&req, sizeof (req)) == sizeof (req))	296	while (read (reqpipe[0], (void *)&req, sizeof (req)) == sizeof (req))
186	{	297	{
187	req->thread = thr;	298	req->thread = thr;
188	errno = 0;	299	errno = 0; /* strictly unnecessary */
189		300
190	if (req->type == REQ_READ \|\| req->type == REQ_WRITE)	301	switch (req->type)
191	{	302	{
192	if (lseek (req->fd, req->offset, SEEK_SET) == req->offset)	303	#ifdef __NR_pread64
193	{	304	case REQ_READ: req->result = pread64 (req->fd, req->dataptr, req->length, req->offset & 0xffffffff, req->offset >> 32); break;
194	if (req->type == REQ_READ)	305	case REQ_WRITE: req->result = pwrite64(req->fd, req->dataptr, req->length, req->offset & 0xffffffff, req->offset >> 32); break;
		306	#else
195	req->result = read (req->fd, req->dataptr, req->length);	307	case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
196	else
197	req->result = write(req->fd, req->dataptr, req->length);	308	case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
198	}	309	#endif
199	}	310	#ifdef __NR_stat64
200	else if (req->type == REQ_OPEN)	311	struct stat64 statdata;
201	{	312	case REQ_STAT: req->result = stat64 (req->dataptr, &statdata); COPY_STATDATA; break;
		313	case REQ_LSTAT: req->result = lstat64 (req->dataptr, &statdata); COPY_STATDATA; break;
		314	case REQ_FSTAT: req->result = fstat64 (req->fd, &statdata); COPY_STATDATA; break;
		315	#else
		316	struct stat statdata;
		317	case REQ_STAT: req->result = stat (req->dataptr, &statdata); COPY_STATDATA; break;
		318	case REQ_LSTAT: req->result = lstat (req->dataptr, &statdata); COPY_STATDATA; break;
		319	case REQ_FSTAT: req->result = fstat (req->fd, &statdata); COPY_STATDATA; break;
		320	#endif
202	req->result = open (req->dataptr, req->fd, req->mode);	321	case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
203	}	322	case REQ_CLOSE: req->result = close (req->fd); break;
204	else	323	case REQ_UNLINK: req->result = unlink (req->dataptr); break;
205	{	324
		325	case REQ_QUIT:
		326	default:
206	write (respipe[1], (void *)&req, sizeof (req));	327	write (respipe[1], (void *)&req, sizeof (req));
207	break;	328	return 0;
208	}	329	}
209		330
210	req->errorno = errno;	331	req->errorno = errno;
211	write (respipe[1], (void *)&req, sizeof (req));	332	write (respipe[1], (void *)&req, sizeof (req));
212	}	333	}
…		…
217	MODULE = Linux::AIO PACKAGE = Linux::AIO	338	MODULE = Linux::AIO PACKAGE = Linux::AIO
218		339
219	BOOT:	340	BOOT:
220	{	341	{
221	sigfillset (&fullsigset);	342	sigfillset (&fullsigset);
		343	sigdelset (&fullsigset, SIGTERM);
		344	sigdelset (&fullsigset, SIGQUIT);
		345	sigdelset (&fullsigset, SIGABRT);
		346	sigdelset (&fullsigset, SIGINT);
222		347
223	if (pipe (reqpipe) \|\| pipe (respipe))	348	if (pipe (reqpipe) \|\| pipe (respipe))
224	croak ("unable to initialize request or result pipe");	349	croak ("unable to initialize request or result pipe");
		350
		351	if (fcntl (reqpipe[1], F_SETFL, O_NONBLOCK))
		352	croak ("cannot set result pipe to nonblocking mode");
225		353
226	if (fcntl (respipe[0], F_SETFL, O_NONBLOCK))	354	if (fcntl (respipe[0], F_SETFL, O_NONBLOCK))
227	croak ("cannot set result pipe to nonblocking mode");	355	croak ("cannot set result pipe to nonblocking mode");
228	}	356	}
229		357
…		…
245	{	373	{
246	end_thread ();	374	end_thread ();
247	cur--;	375	cur--;
248	}	376	}
249		377
250	poll_cb ();
251	while (started > nthreads)	378	while (started > nthreads)
252	{	379	{
253	sched_yield ();	380	poll_wait ();
254	poll_cb ();	381	poll_cb (aTHX);
255	}	382	}
256		383
257	void	384	void
258	aio_read(fh,offset,length,data,dataoffset,callback)
259	PerlIO * fh
260	UV offset
261	STRLEN length
262	SV * data
263	STRLEN dataoffset
264	SV * callback
265	PROTOTYPE: $$$$$$
266	ALIAS:
267	aio_write = 1
268	CODE:
269	SvUPGRADE (data, SVt_PV);
270	SvPOK_on (data);
271	read_write (aTHX_ ix, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
272
273	void
274	aio_open(pathname,flags,mode,callback)	385	aio_open(pathname,flags,mode,callback)
275	char * pathname	386	SV * pathname
276	int flags	387	int flags
277	int mode	388	int mode
278	SV * callback	389	SV * callback
279	PROTOTYPE: $$$$	390	PROTOTYPE: $$$$
280	CODE:	391	CODE:
281	aio_req req;	392	aio_req req;
282		393
283	New (0, req, 1, aio_cb);	394	Newz (0, req, 1, aio_cb);
284		395
285	if (!req)	396	if (!req)
286	croak ("out of memory during aio_req allocation");	397	croak ("out of memory during aio_req allocation");
287		398
288	req->type = REQ_OPEN;	399	req->type = REQ_OPEN;
289	req->dataptr = pathname;	400	req->data = newSVsv (pathname);
		401	req->dataptr = SvPV_nolen (req->data);
290	req->fd = flags;	402	req->fd = flags;
291	req->mode = mode;	403	req->mode = mode;
292	req->callback = SvREFCNT_inc (callback);	404	req->callback = SvREFCNT_inc (callback);
293		405
294	send_req (req);	406	send_req (req);
295		407
		408	void
		409	aio_close(fh,callback)
		410	InputStream fh
		411	SV * callback
		412	PROTOTYPE: $$
		413	CODE:
		414	aio_req req;
		415
		416	Newz (0, req, 1, aio_cb);
		417
		418	if (!req)
		419	croak ("out of memory during aio_req allocation");
		420
		421	req->type = REQ_CLOSE;
		422	req->fd = PerlIO_fileno (fh);
		423	req->callback = SvREFCNT_inc (callback);
		424
		425	send_req (req);
		426
		427	void
		428	aio_read(fh,offset,length,data,dataoffset,callback)
		429	InputStream fh
		430	UV offset
		431	IV length
		432	SV * data
		433	IV dataoffset
		434	SV * callback
		435	PROTOTYPE: $$$$$$
		436	CODE:
		437	read_write (aTHX_ 0, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
		438
		439	void
		440	aio_write(fh,offset,length,data,dataoffset,callback)
		441	OutputStream fh
		442	UV offset
		443	IV length
		444	SV * data
		445	IV dataoffset
		446	SV * callback
		447	PROTOTYPE: $$$$$$
		448	CODE:
		449	read_write (aTHX_ 1, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
		450
		451	void
		452	aio_stat(fh_or_path,callback)
		453	SV * fh_or_path
		454	SV * callback
		455	PROTOTYPE: $$
		456	ALIAS:
		457	aio_lstat = 1
		458	CODE:
		459	aio_req req;
		460
		461	Newz (0, req, 1, aio_cb);
		462
		463	if (!req)
		464	croak ("out of memory during aio_req allocation");
		465
		466	New (0, req->statdata, 1, Stat_t);
		467
		468	if (!req->statdata)
		469	croak ("out of memory during aio_req->statdata allocation");
		470
		471	if (SvPOK (fh_or_path))
		472	{
		473	req->type = ix ? REQ_LSTAT : REQ_STAT;
		474	req->data = newSVsv (fh_or_path);
		475	req->dataptr = SvPV_nolen (req->data);
		476	}
		477	else
		478	{
		479	req->type = REQ_FSTAT;
		480	req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
		481	}
		482
		483	req->callback = SvREFCNT_inc (callback);
		484
		485	send_req (req);
		486
		487	void
		488	aio_unlink(pathname,callback)
		489	SV * pathname
		490	SV * callback
		491	PROTOTYPE: $$
		492	CODE:
		493	aio_req req;
		494
		495	Newz (0, req, 1, aio_cb);
		496
		497	if (!req)
		498	croak ("out of memory during aio_req allocation");
		499
		500	req->type = REQ_UNLINK;
		501	req->data = newSVsv (pathname);
		502	req->dataptr = SvPV_nolen (req->data);
		503	req->callback = SvREFCNT_inc (callback);
		504
		505	send_req (req);
		506
296	int	507	int
297	poll_fileno()	508	poll_fileno()
298	PROTOTYPE:	509	PROTOTYPE:
299	CODE:	510	CODE:
300	RETVAL = respipe[0];	511	RETVAL = respipe[0];
…		…
307	CODE:	518	CODE:
308	RETVAL = poll_cb (aTHX);	519	RETVAL = poll_cb (aTHX);
309	OUTPUT:	520	OUTPUT:
310	RETVAL	521	RETVAL
311		522
		523	void
		524	poll_wait()
		525	PROTOTYPE:
		526	CODE:
		527	poll_wait ();
		528
312	int	529	int
313	nreqs()	530	nreqs()
314	PROTOTYPE:	531	PROTOTYPE:
315	CODE:	532	CODE:
316	RETVAL = nreqs;	533	RETVAL = nreqs;

Diff Legend

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

Comparing Linux-AIO/AIO.xs (file contents): Revision 1.8 by root, Thu Aug 16 02:43:46 2001 UTC vs. Revision 1.23 by root, Wed Jun 29 15:28:14 2005 UTC

Diff Legend

Comparing Linux-AIO/AIO.xs (file contents):
Revision 1.8 by root, Thu Aug 16 02:43:46 2001 UTC vs.
Revision 1.23 by root, Wed Jun 29 15:28:14 2005 UTC