ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/AnyEvent/lib/AnyEvent/Handle.pm

Comparing AnyEvent/lib/AnyEvent/Handle.pm (file contents):
Revision 1.27 by root, Sat May 24 15:26:04 2008 UTC vs.
Revision 1.129 by root, Mon Jun 29 11:04:09 2009 UTC

1	package AnyEvent::Handle;	1	package AnyEvent::Handle;
2		2
3	no warnings;	3	no warnings;
4	use strict;	4	use strict qw(subs vars);
5		5
6	use AnyEvent ();	6	use AnyEvent ();
7	use AnyEvent::Util ();	7	use AnyEvent::Util qw(WSAEWOULDBLOCK);
8	use Scalar::Util ();	8	use Scalar::Util ();
9	use Carp ();	9	use Carp ();
10	use Fcntl ();	10	use Fcntl ();
11	use Errno qw/EAGAIN EINTR/;	11	use Errno qw(EAGAIN EINTR);
12		12
13	=head1 NAME	13	=head1 NAME
14		14
15	AnyEvent::Handle - non-blocking I/O on file handles via AnyEvent	15	AnyEvent::Handle - non-blocking I/O on file handles via AnyEvent
16		16
17	This module is experimental.
18
19	=cut	17	=cut
20		18
21	our $VERSION = '0.04';	19	our $VERSION = 4.42;
22		20
23	=head1 SYNOPSIS	21	=head1 SYNOPSIS
24		22
25	use AnyEvent;	23	use AnyEvent;
26	use AnyEvent::Handle;	24	use AnyEvent::Handle;
27		25
28	my $cv = AnyEvent->condvar;	26	my $cv = AnyEvent->condvar;
29		27
30	my $ae_fh = AnyEvent::Handle->new (fh => \*STDIN);	28	my $handle =
31
32	#TODO
33
34	# or use the constructor to pass the callback:
35
36	my $ae_fh2 =
37	AnyEvent::Handle->new (	29	AnyEvent::Handle->new (
38	fh => \*STDIN,	30	fh => \*STDIN,
39	on_eof => sub {	31	on_eof => sub {
40	$cv->broadcast;	32	$cv->send;
41	},	33	},
42	#TODO
43	);	34	);
44		35
45	$cv->wait;	36	# send some request line
		37	$handle->push_write ("getinfo\015\012");
		38
		39	# read the response line
		40	$handle->push_read (line => sub {
		41	my ($handle, $line) = @_;
		42	warn "read line <$line>\n";
		43	$cv->send;
		44	});
		45
		46	$cv->recv;
46		47
47	=head1 DESCRIPTION	48	=head1 DESCRIPTION
48		49
49	This module is a helper module to make it easier to do event-based I/O on	50	This module is a helper module to make it easier to do event-based I/O on
50	filehandles. For utility functions for doing non-blocking connects and accepts	51	filehandles. For utility functions for doing non-blocking connects and accepts
51	on sockets see L<AnyEvent::Util>.	52	on sockets see L<AnyEvent::Util>.
52		53
		54	The L<AnyEvent::Intro> tutorial contains some well-documented
		55	AnyEvent::Handle examples.
		56
53	In the following, when the documentation refers to of "bytes" then this	57	In the following, when the documentation refers to of "bytes" then this
54	means characters. As sysread and syswrite are used for all I/O, their	58	means characters. As sysread and syswrite are used for all I/O, their
55	treatment of characters applies to this module as well.	59	treatment of characters applies to this module as well.
56		60
57	All callbacks will be invoked with the handle object as their first	61	All callbacks will be invoked with the handle object as their first
…		…
69		73
70	=item fh => $filehandle [MANDATORY]	74	=item fh => $filehandle [MANDATORY]
71		75
72	The filehandle this L<AnyEvent::Handle> object will operate on.	76	The filehandle this L<AnyEvent::Handle> object will operate on.
73		77
74	NOTE: The filehandle will be set to non-blocking (using	78	NOTE: The filehandle will be set to non-blocking mode (using
75	AnyEvent::Util::fh_nonblocking).	79	C<AnyEvent::Util::fh_nonblocking>) by the constructor and needs to stay in
		80	that mode.
76		81
77	=item on_eof => $cb->($self)	82	=item on_eof => $cb->($handle)
78		83
79	Set the callback to be called on EOF.	84	Set the callback to be called when an end-of-file condition is detected,
		85	i.e. in the case of a socket, when the other side has closed the
		86	connection cleanly.
80		87
		88	For sockets, this just means that the other side has stopped sending data,
		89	you can still try to write data, and, in fact, one can return from the EOF
		90	callback and continue writing data, as only the read part has been shut
		91	down.
		92
81	While not mandatory, it is highly recommended to set an eof callback,	93	While not mandatory, it is I<highly> recommended to set an EOF callback,
82	otherwise you might end up with a closed socket while you are still	94	otherwise you might end up with a closed socket while you are still
83	waiting for data.	95	waiting for data.
84		96
		97	If an EOF condition has been detected but no C<on_eof> callback has been
		98	set, then a fatal error will be raised with C<$!> set to <0>.
		99
85	=item on_error => $cb->($self)	100	=item on_error => $cb->($handle, $fatal)
86		101
87	This is the fatal error callback, that is called when, well, a fatal error	102	This is the error callback, which is called when, well, some error
88	occurs, such as not being able to resolve the hostname, failure to connect	103	occured, such as not being able to resolve the hostname, failure to
89	or a read error.	104	connect or a read error.
90		105
91	The object will not be in a usable state when this callback has been	106	Some errors are fatal (which is indicated by C<$fatal> being true). On
92	called.	107	fatal errors the handle object will be shut down and will not be usable
		108	(but you are free to look at the current C<< ->rbuf >>). Examples of fatal
		109	errors are an EOF condition with active (but unsatisifable) read watchers
		110	(C<EPIPE>) or I/O errors.
		111
		112	Non-fatal errors can be retried by simply returning, but it is recommended
		113	to simply ignore this parameter and instead abondon the handle object
		114	when this callback is invoked. Examples of non-fatal errors are timeouts
		115	C<ETIMEDOUT>) or badly-formatted data (C<EBADMSG>).
93		116
94	On callback entrance, the value of C<$!> contains the operating system	117	On callback entrance, the value of C<$!> contains the operating system
95	error (or C<ENOSPC> or C<EPIPE>).	118	error (or C<ENOSPC>, C<EPIPE>, C<ETIMEDOUT> or C<EBADMSG>).
96		119
97	While not mandatory, it is I<highly> recommended to set this callback, as	120	While not mandatory, it is I<highly> recommended to set this callback, as
98	you will not be notified of errors otherwise. The default simply calls	121	you will not be notified of errors otherwise. The default simply calls
99	die.	122	C<croak>.
100		123
101	=item on_read => $cb->($self)	124	=item on_read => $cb->($handle)
102		125
103	This sets the default read callback, which is called when data arrives	126	This sets the default read callback, which is called when data arrives
104	and no read request is in the queue.	127	and no read request is in the queue (unlike read queue callbacks, this
		128	callback will only be called when at least one octet of data is in the
		129	read buffer).
105		130
106	To access (and remove data from) the read buffer, use the C<< ->rbuf >>	131	To access (and remove data from) the read buffer, use the C<< ->rbuf >>
107	method or access the C<$self->{rbuf}> member directly.	132	method or access the C<$handle->{rbuf}> member directly. Note that you
		133	must not enlarge or modify the read buffer, you can only remove data at
		134	the beginning from it.
108		135
109	When an EOF condition is detected then AnyEvent::Handle will first try to	136	When an EOF condition is detected then AnyEvent::Handle will first try to
110	feed all the remaining data to the queued callbacks and C<on_read> before	137	feed all the remaining data to the queued callbacks and C<on_read> before
111	calling the C<on_eof> callback. If no progress can be made, then a fatal	138	calling the C<on_eof> callback. If no progress can be made, then a fatal
112	error will be raised (with C<$!> set to C<EPIPE>).	139	error will be raised (with C<$!> set to C<EPIPE>).
113		140
114	=item on_drain => $cb->()	141	=item on_drain => $cb->($handle)
115		142
116	This sets the callback that is called when the write buffer becomes empty	143	This sets the callback that is called when the write buffer becomes empty
117	(or when the callback is set and the buffer is empty already).	144	(or when the callback is set and the buffer is empty already).
118		145
119	To append to the write buffer, use the C<< ->push_write >> method.	146	To append to the write buffer, use the C<< ->push_write >> method.
120		147
		148	This callback is useful when you don't want to put all of your write data
		149	into the queue at once, for example, when you want to write the contents
		150	of some file to the socket you might not want to read the whole file into
		151	memory and push it into the queue, but instead only read more data from
		152	the file when the write queue becomes empty.
		153
		154	=item timeout => $fractional_seconds
		155
		156	If non-zero, then this enables an "inactivity" timeout: whenever this many
		157	seconds pass without a successful read or write on the underlying file
		158	handle, the C<on_timeout> callback will be invoked (and if that one is
		159	missing, a non-fatal C<ETIMEDOUT> error will be raised).
		160
		161	Note that timeout processing is also active when you currently do not have
		162	any outstanding read or write requests: If you plan to keep the connection
		163	idle then you should disable the timout temporarily or ignore the timeout
		164	in the C<on_timeout> callback, in which case AnyEvent::Handle will simply
		165	restart the timeout.
		166
		167	Zero (the default) disables this timeout.
		168
		169	=item on_timeout => $cb->($handle)
		170
		171	Called whenever the inactivity timeout passes. If you return from this
		172	callback, then the timeout will be reset as if some activity had happened,
		173	so this condition is not fatal in any way.
		174
121	=item rbuf_max => <bytes>	175	=item rbuf_max => <bytes>
122		176
123	If defined, then a fatal error will be raised (with C<$!> set to C<ENOSPC>)	177	If defined, then a fatal error will be raised (with C<$!> set to C<ENOSPC>)
124	when the read buffer ever (strictly) exceeds this size. This is useful to	178	when the read buffer ever (strictly) exceeds this size. This is useful to
125	avoid denial-of-service attacks.	179	avoid some forms of denial-of-service attacks.
126		180
127	For example, a server accepting connections from untrusted sources should	181	For example, a server accepting connections from untrusted sources should
128	be configured to accept only so-and-so much data that it cannot act on	182	be configured to accept only so-and-so much data that it cannot act on
129	(for example, when expecting a line, an attacker could send an unlimited	183	(for example, when expecting a line, an attacker could send an unlimited
130	amount of data without a callback ever being called as long as the line	184	amount of data without a callback ever being called as long as the line
131	isn't finished).	185	isn't finished).
132		186
		187	=item autocork => <boolean>
		188
		189	When disabled (the default), then C<push_write> will try to immediately
		190	write the data to the handle, if possible. This avoids having to register
		191	a write watcher and wait for the next event loop iteration, but can
		192	be inefficient if you write multiple small chunks (on the wire, this
		193	disadvantage is usually avoided by your kernel's nagle algorithm, see
		194	C<no_delay>, but this option can save costly syscalls).
		195
		196	When enabled, then writes will always be queued till the next event loop
		197	iteration. This is efficient when you do many small writes per iteration,
		198	but less efficient when you do a single write only per iteration (or when
		199	the write buffer often is full). It also increases write latency.
		200
		201	=item no_delay => <boolean>
		202
		203	When doing small writes on sockets, your operating system kernel might
		204	wait a bit for more data before actually sending it out. This is called
		205	the Nagle algorithm, and usually it is beneficial.
		206
		207	In some situations you want as low a delay as possible, which can be
		208	accomplishd by setting this option to a true value.
		209
		210	The default is your opertaing system's default behaviour (most likely
		211	enabled), this option explicitly enables or disables it, if possible.
		212
133	=item read_size => <bytes>	213	=item read_size => <bytes>
134		214
135	The default read block size (the amount of bytes this module will try to read	215	The default read block size (the amount of bytes this module will
136	on each [loop iteration). Default: C<4096>.	216	try to read during each loop iteration, which affects memory
		217	requirements). Default: C<8192>.
137		218
138	=item low_water_mark => <bytes>	219	=item low_water_mark => <bytes>
139		220
140	Sets the amount of bytes (default: C<0>) that make up an "empty" write	221	Sets the amount of bytes (default: C<0>) that make up an "empty" write
141	buffer: If the write reaches this size or gets even samller it is	222	buffer: If the write reaches this size or gets even samller it is
142	considered empty.	223	considered empty.
143		224
		225	Sometimes it can be beneficial (for performance reasons) to add data to
		226	the write buffer before it is fully drained, but this is a rare case, as
		227	the operating system kernel usually buffers data as well, so the default
		228	is good in almost all cases.
		229
		230	=item linger => <seconds>
		231
		232	If non-zero (default: C<3600>), then the destructor of the
		233	AnyEvent::Handle object will check whether there is still outstanding
		234	write data and will install a watcher that will write this data to the
		235	socket. No errors will be reported (this mostly matches how the operating
		236	system treats outstanding data at socket close time).
		237
		238	This will not work for partial TLS data that could not be encoded
		239	yet. This data will be lost. Calling the C<stoptls> method in time might
		240	help.
		241
144	=item tls => "accept" | "connect" | Net::SSLeay::SSL object	242	=item tls => "accept" | "connect" | Net::SSLeay::SSL object
145		243
146	When this parameter is given, it enables TLS (SSL) mode, that means it	244	When this parameter is given, it enables TLS (SSL) mode, that means
147	will start making tls handshake and will transparently encrypt/decrypt	245	AnyEvent will start a TLS handshake as soon as the conenction has been
148	data.	246	established and will transparently encrypt/decrypt data afterwards.
149		247
150	TLS mode requires Net::SSLeay to be installed (it will be loaded	248	TLS mode requires Net::SSLeay to be installed (it will be loaded
151	automatically when you try to create a TLS handle).	249	automatically when you try to create a TLS handle): this module doesn't
		250	have a dependency on that module, so if your module requires it, you have
		251	to add the dependency yourself.
152		252
153	For the TLS server side, use C<accept>, and for the TLS client side of a	253	Unlike TCP, TLS has a server and client side: for the TLS server side, use
154	connection, use C<connect> mode.	254	C<accept>, and for the TLS client side of a connection, use C<connect>
		255	mode.
155		256
156	You can also provide your own TLS connection object, but you have	257	You can also provide your own TLS connection object, but you have
157	to make sure that you call either C<Net::SSLeay::set_connect_state>	258	to make sure that you call either C<Net::SSLeay::set_connect_state>
158	or C<Net::SSLeay::set_accept_state> on it before you pass it to	259	or C<Net::SSLeay::set_accept_state> on it before you pass it to
159	AnyEvent::Handle.	260	AnyEvent::Handle.
160		261
		262	B<IMPORTANT:> since Net::SSLeay "objects" are really only integers,
		263	passing in the wrong integer will lead to certain crash. This most often
		264	happens when one uses a stylish C<< tls => 1 >> and is surprised about the
		265	segmentation fault.
		266
161	See the C<starttls> method if you need to start TLs negotiation later.	267	See the C<< ->starttls >> method for when need to start TLS negotiation later.
162		268
163	=item tls_ctx => $ssl_ctx	269	=item tls_ctx => $ssl_ctx
164		270
165	Use the given Net::SSLeay::CTX object to create the new TLS connection	271	Use the given C<Net::SSLeay::CTX> object to create the new TLS connection
166	(unless a connection object was specified directly). If this parameter is	272	(unless a connection object was specified directly). If this parameter is
167	missing, then AnyEvent::Handle will use C<AnyEvent::Handle::TLS_CTX>.	273	missing, then AnyEvent::Handle will use C<AnyEvent::Handle::TLS_CTX>.
168		274
		275	=item json => JSON or JSON::XS object
		276
		277	This is the json coder object used by the C<json> read and write types.
		278
		279	If you don't supply it, then AnyEvent::Handle will create and use a
		280	suitable one (on demand), which will write and expect UTF-8 encoded JSON
		281	texts.
		282
		283	Note that you are responsible to depend on the JSON module if you want to
		284	use this functionality, as AnyEvent does not have a dependency itself.
		285
169	=back	286	=back
170		287
171	=cut	288	=cut
172		289
173	sub new {	290	sub new {
…		…
177		294
178	$self->{fh} or Carp::croak "mandatory argument fh is missing";	295	$self->{fh} or Carp::croak "mandatory argument fh is missing";
179		296
180	AnyEvent::Util::fh_nonblocking $self->{fh}, 1;	297	AnyEvent::Util::fh_nonblocking $self->{fh}, 1;
181		298
182	if ($self->{tls}) {
183	require Net::SSLeay;
184	$self->starttls (delete $self->{tls}, delete $self->{tls_ctx});	299	$self->starttls (delete $self->{tls}, delete $self->{tls_ctx})
185	}	300	if $self->{tls};
186		301
187	$self->on_eof (delete $self->{on_eof} ) if $self->{on_eof};	302	$self->{_activity} = AnyEvent->now;
188	$self->on_error (delete $self->{on_error}) if $self->{on_error};	303	$self->_timeout;
		304
189	$self->on_drain (delete $self->{on_drain}) if $self->{on_drain};	305	$self->on_drain (delete $self->{on_drain}) if exists $self->{on_drain};
190	$self->on_read (delete $self->{on_read} ) if $self->{on_read};	306	$self->no_delay (delete $self->{no_delay}) if exists $self->{no_delay};
191		307
192	$self->start_read;	308	$self->start_read
		309	if $self->{on_read};
193		310
194	$self	311	$self
195	}	312	}
196		313
197	sub _shutdown {	314	sub _shutdown {
198	my ($self) = @_;	315	my ($self) = @_;
199		316
200	delete $self->{rw};	317	delete @$self{qw(_tw _rw _ww fh rbuf wbuf on_read _queue)};
201	delete $self->{ww};
202	delete $self->{fh};
203	}
204		318
		319	&_freetls;
		320	}
		321
205	sub error {	322	sub _error {
206	my ($self) = @_;	323	my ($self, $errno, $fatal) = @_;
207		324
208	{
209	local $!;
210	$self->_shutdown;	325	$self->_shutdown
211	}	326	if $fatal;
		327
		328	$! = $errno;
212		329
213	if ($self->{on_error}) {	330	if ($self->{on_error}) {
214	$self->{on_error}($self);	331	$self->{on_error}($self, $fatal);
215	} else {	332	} elsif ($self->{fh}) {
216	die "AnyEvent::Handle uncaught fatal error: $!";	333	Carp::croak "AnyEvent::Handle uncaught error: $!";
217	}	334	}
218	}	335	}
219		336
220	=item $fh = $handle->fh	337	=item $fh = $handle->fh
221		338
222	This method returns the file handle of the L<AnyEvent::Handle> object.	339	This method returns the file handle used to create the L<AnyEvent::Handle> object.
223		340
224	=cut	341	=cut
225		342
226	sub fh { $_[0]->{fh} }	343	sub fh { $_[0]{fh} }
227		344
228	=item $handle->on_error ($cb)	345	=item $handle->on_error ($cb)
229		346
230	Replace the current C<on_error> callback (see the C<on_error> constructor argument).	347	Replace the current C<on_error> callback (see the C<on_error> constructor argument).
231		348
…		…
243		360
244	sub on_eof {	361	sub on_eof {
245	$_[0]{on_eof} = $_[1];	362	$_[0]{on_eof} = $_[1];
246	}	363	}
247		364
		365	=item $handle->on_timeout ($cb)
		366
		367	Replace the current C<on_timeout> callback, or disables the callback (but
		368	not the timeout) if C<$cb> = C<undef>. See the C<timeout> constructor
		369	argument and method.
		370
		371	=cut
		372
		373	sub on_timeout {
		374	$_[0]{on_timeout} = $_[1];
		375	}
		376
		377	=item $handle->autocork ($boolean)
		378
		379	Enables or disables the current autocork behaviour (see C<autocork>
		380	constructor argument). Changes will only take effect on the next write.
		381
		382	=cut
		383
		384	sub autocork {
		385	$_[0]{autocork} = $_[1];
		386	}
		387
		388	=item $handle->no_delay ($boolean)
		389
		390	Enables or disables the C<no_delay> setting (see constructor argument of
		391	the same name for details).
		392
		393	=cut
		394
		395	sub no_delay {
		396	$_[0]{no_delay} = $_[1];
		397
		398	eval {
		399	local $SIG{__DIE__};
		400	setsockopt $_[0]{fh}, &Socket::IPPROTO_TCP, &Socket::TCP_NODELAY, int $_[1];
		401	};
		402	}
		403
		404	#############################################################################
		405
		406	=item $handle->timeout ($seconds)
		407
		408	Configures (or disables) the inactivity timeout.
		409
		410	=cut
		411
		412	sub timeout {
		413	my ($self, $timeout) = @_;
		414
		415	$self->{timeout} = $timeout;
		416	$self->_timeout;
		417	}
		418
		419	# reset the timeout watcher, as neccessary
		420	# also check for time-outs
		421	sub _timeout {
		422	my ($self) = @_;
		423
		424	if ($self->{timeout}) {
		425	my $NOW = AnyEvent->now;
		426
		427	# when would the timeout trigger?
		428	my $after = $self->{_activity} + $self->{timeout} - $NOW;
		429
		430	# now or in the past already?
		431	if ($after <= 0) {
		432	$self->{_activity} = $NOW;
		433
		434	if ($self->{on_timeout}) {
		435	$self->{on_timeout}($self);
		436	} else {
		437	$self->_error (&Errno::ETIMEDOUT);
		438	}
		439
		440	# callback could have changed timeout value, optimise
		441	return unless $self->{timeout};
		442
		443	# calculate new after
		444	$after = $self->{timeout};
		445	}
		446
		447	Scalar::Util::weaken $self;
		448	return unless $self; # ->error could have destroyed $self
		449
		450	$self->{_tw} ||= AnyEvent->timer (after => $after, cb => sub {
		451	delete $self->{_tw};
		452	$self->_timeout;
		453	});
		454	} else {
		455	delete $self->{_tw};
		456	}
		457	}
		458
248	#############################################################################	459	#############################################################################
249		460
250	=back	461	=back
251		462
252	=head2 WRITE QUEUE	463	=head2 WRITE QUEUE
…		…
273	my ($self, $cb) = @_;	484	my ($self, $cb) = @_;
274		485
275	$self->{on_drain} = $cb;	486	$self->{on_drain} = $cb;
276		487
277	$cb->($self)	488	$cb->($self)
278	if $cb && $self->{low_water_mark} >= length $self->{wbuf};	489	if $cb && $self->{low_water_mark} >= (length $self->{wbuf}) + (length $self->{_tls_wbuf});
279	}	490	}
280		491
281	=item $handle->push_write ($data)	492	=item $handle->push_write ($data)
282		493
283	Queues the given scalar to be written. You can push as much data as you	494	Queues the given scalar to be written. You can push as much data as you
…		…
287	=cut	498	=cut
288		499
289	sub _drain_wbuf {	500	sub _drain_wbuf {
290	my ($self) = @_;	501	my ($self) = @_;
291		502
292	unless ($self->{ww}) {	503	if (!$self->{_ww} && length $self->{wbuf}) {
		504
293	Scalar::Util::weaken $self;	505	Scalar::Util::weaken $self;
		506
294	my $cb = sub {	507	my $cb = sub {
295	my $len = syswrite $self->{fh}, $self->{wbuf};	508	my $len = syswrite $self->{fh}, $self->{wbuf};
296		509
297	if ($len > 0) {	510	if ($len >= 0) {
298	substr $self->{wbuf}, 0, $len, "";	511	substr $self->{wbuf}, 0, $len, "";
299		512
		513	$self->{_activity} = AnyEvent->now;
		514
300	$self->{on_drain}($self)	515	$self->{on_drain}($self)
301	if $self->{low_water_mark} >= length $self->{wbuf}	516	if $self->{low_water_mark} >= (length $self->{wbuf}) + (length $self->{_tls_wbuf})
302	&& $self->{on_drain};	517	&& $self->{on_drain};
303		518
304	delete $self->{ww} unless length $self->{wbuf};	519	delete $self->{_ww} unless length $self->{wbuf};
305	} elsif ($! != EAGAIN && $! != EINTR) {	520	} elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
306	$self->error;	521	$self->_error ($!, 1);
307	}	522	}
308	};	523	};
309		524
		525	# try to write data immediately
		526	$cb->() unless $self->{autocork};
		527
		528	# if still data left in wbuf, we need to poll
310	$self->{ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb);	529	$self->{_ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb)
311		530	if length $self->{wbuf};
312	$cb->($self);
313	};	531	};
		532	}
		533
		534	our %WH;
		535
		536	sub register_write_type($$) {
		537	$WH{$_[0]} = $_[1];
314	}	538	}
315		539
316	sub push_write {	540	sub push_write {
317	my $self = shift;	541	my $self = shift;
318		542
		543	if (@_ > 1) {
		544	my $type = shift;
		545
		546	@_ = ($WH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_write")
		547	->($self, @_);
		548	}
		549
319	if ($self->{filter_w}) {	550	if ($self->{tls}) {
320	$self->{filter_w}->($self, \$_[0]);	551	$self->{_tls_wbuf} .= $_[0];
		552
		553	&_dotls ($self);
321	} else {	554	} else {
322	$self->{wbuf} .= $_[0];	555	$self->{wbuf} .= $_[0];
323	$self->_drain_wbuf;	556	$self->_drain_wbuf;
324	}	557	}
325	}	558	}
326		559
		560	=item $handle->push_write (type => @args)
		561
		562	Instead of formatting your data yourself, you can also let this module do
		563	the job by specifying a type and type-specific arguments.
		564
		565	Predefined types are (if you have ideas for additional types, feel free to
		566	drop by and tell us):
		567
		568	=over 4
		569
		570	=item netstring => $string
		571
		572	Formats the given value as netstring
		573	(http://cr.yp.to/proto/netstrings.txt, this is not a recommendation to use them).
		574
		575	=cut
		576
		577	register_write_type netstring => sub {
		578	my ($self, $string) = @_;
		579
		580	(length $string) . ":$string,"
		581	};
		582
		583	=item packstring => $format, $data
		584
		585	An octet string prefixed with an encoded length. The encoding C<$format>
		586	uses the same format as a Perl C<pack> format, but must specify a single
		587	integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
		588	optional C<!>, C<< < >> or C<< > >> modifier).
		589
		590	=cut
		591
		592	register_write_type packstring => sub {
		593	my ($self, $format, $string) = @_;
		594
		595	pack "$format/a*", $string
		596	};
		597
		598	=item json => $array_or_hashref
		599
		600	Encodes the given hash or array reference into a JSON object. Unless you
		601	provide your own JSON object, this means it will be encoded to JSON text
		602	in UTF-8.
		603
		604	JSON objects (and arrays) are self-delimiting, so you can write JSON at
		605	one end of a handle and read them at the other end without using any
		606	additional framing.
		607
		608	The generated JSON text is guaranteed not to contain any newlines: While
		609	this module doesn't need delimiters after or between JSON texts to be
		610	able to read them, many other languages depend on that.
		611
		612	A simple RPC protocol that interoperates easily with others is to send
		613	JSON arrays (or objects, although arrays are usually the better choice as
		614	they mimic how function argument passing works) and a newline after each
		615	JSON text:
		616
		617	$handle->push_write (json => ["method", "arg1", "arg2"]); # whatever
		618	$handle->push_write ("\012");
		619
		620	An AnyEvent::Handle receiver would simply use the C<json> read type and
		621	rely on the fact that the newline will be skipped as leading whitespace:
		622
		623	$handle->push_read (json => sub { my $array = $_[1]; ... });
		624
		625	Other languages could read single lines terminated by a newline and pass
		626	this line into their JSON decoder of choice.
		627
		628	=cut
		629
		630	register_write_type json => sub {
		631	my ($self, $ref) = @_;
		632
		633	require JSON;
		634
		635	$self->{json} ? $self->{json}->encode ($ref)
		636	: JSON::encode_json ($ref)
		637	};
		638
		639	=item storable => $reference
		640
		641	Freezes the given reference using L<Storable> and writes it to the
		642	handle. Uses the C<nfreeze> format.
		643
		644	=cut
		645
		646	register_write_type storable => sub {
		647	my ($self, $ref) = @_;
		648
		649	require Storable;
		650
		651	pack "w/a*", Storable::nfreeze ($ref)
		652	};
		653
		654	=back
		655
		656	=item AnyEvent::Handle::register_write_type type => $coderef->($handle, @args)
		657
		658	This function (not method) lets you add your own types to C<push_write>.
		659	Whenever the given C<type> is used, C<push_write> will invoke the code
		660	reference with the handle object and the remaining arguments.
		661
		662	The code reference is supposed to return a single octet string that will
		663	be appended to the write buffer.
		664
		665	Note that this is a function, and all types registered this way will be
		666	global, so try to use unique names.
		667
		668	=cut
		669
327	#############################################################################	670	#############################################################################
328		671
329	=back	672	=back
330		673
331	=head2 READ QUEUE	674	=head2 READ QUEUE
…		…
337	ways, the "simple" way, using only C<on_read> and the "complex" way, using	680	ways, the "simple" way, using only C<on_read> and the "complex" way, using
338	a queue.	681	a queue.
339		682
340	In the simple case, you just install an C<on_read> callback and whenever	683	In the simple case, you just install an C<on_read> callback and whenever
341	new data arrives, it will be called. You can then remove some data (if	684	new data arrives, it will be called. You can then remove some data (if
342	enough is there) from the read buffer (C<< $handle->rbuf >>) if you want	685	enough is there) from the read buffer (C<< $handle->rbuf >>). Or you cna
343	or not.	686	leave the data there if you want to accumulate more (e.g. when only a
		687	partial message has been received so far).
344		688
345	In the more complex case, you want to queue multiple callbacks. In this	689	In the more complex case, you want to queue multiple callbacks. In this
346	case, AnyEvent::Handle will call the first queued callback each time new	690	case, AnyEvent::Handle will call the first queued callback each time new
347	data arrives and removes it when it has done its job (see C<push_read>,	691	data arrives (also the first time it is queued) and removes it when it has
348	below).	692	done its job (see C<push_read>, below).
349		693
350	This way you can, for example, push three line-reads, followed by reading	694	This way you can, for example, push three line-reads, followed by reading
351	a chunk of data, and AnyEvent::Handle will execute them in order.	695	a chunk of data, and AnyEvent::Handle will execute them in order.
352		696
353	Example 1: EPP protocol parser. EPP sends 4 byte length info, followed by	697	Example 1: EPP protocol parser. EPP sends 4 byte length info, followed by
354	the specified number of bytes which give an XML datagram.	698	the specified number of bytes which give an XML datagram.
355		699
356	# in the default state, expect some header bytes	700	# in the default state, expect some header bytes
357	$handle->on_read (sub {	701	$handle->on_read (sub {
358	# some data is here, now queue the length-header-read (4 octets)	702	# some data is here, now queue the length-header-read (4 octets)
359	shift->unshift_read_chunk (4, sub {	703	shift->unshift_read (chunk => 4, sub {
360	# header arrived, decode	704	# header arrived, decode
361	my $len = unpack "N", $_[1];	705	my $len = unpack "N", $_[1];
362		706
363	# now read the payload	707	# now read the payload
364	shift->unshift_read_chunk ($len, sub {	708	shift->unshift_read (chunk => $len, sub {
365	my $xml = $_[1];	709	my $xml = $_[1];
366	# handle xml	710	# handle xml
367	});	711	});
368	});	712	});
369	});	713	});
370		714
371	Example 2: Implement a client for a protocol that replies either with	715	Example 2: Implement a client for a protocol that replies either with "OK"
372	"OK" and another line or "ERROR" for one request, and 64 bytes for the	716	and another line or "ERROR" for the first request that is sent, and 64
373	second request. Due tot he availability of a full queue, we can just	717	bytes for the second request. Due to the availability of a queue, we can
374	pipeline sending both requests and manipulate the queue as necessary in	718	just pipeline sending both requests and manipulate the queue as necessary
375	the callbacks:	719	in the callbacks.
376		720
377	# request one	721	When the first callback is called and sees an "OK" response, it will
		722	C<unshift> another line-read. This line-read will be queued I<before> the
		723	64-byte chunk callback.
		724
		725	# request one, returns either "OK + extra line" or "ERROR"
378	$handle->push_write ("request 1\015\012");	726	$handle->push_write ("request 1\015\012");
379		727
380	# we expect "ERROR" or "OK" as response, so push a line read	728	# we expect "ERROR" or "OK" as response, so push a line read
381	$handle->push_read_line (sub {	729	$handle->push_read (line => sub {
382	# if we got an "OK", we have to _prepend_ another line,	730	# if we got an "OK", we have to _prepend_ another line,
383	# so it will be read before the second request reads its 64 bytes	731	# so it will be read before the second request reads its 64 bytes
384	# which are already in the queue when this callback is called	732	# which are already in the queue when this callback is called
385	# we don't do this in case we got an error	733	# we don't do this in case we got an error
386	if ($_[1] eq "OK") {	734	if ($_[1] eq "OK") {
387	$_[0]->unshift_read_line (sub {	735	$_[0]->unshift_read (line => sub {
388	my $response = $_[1];	736	my $response = $_[1];
389	...	737	...
390	});	738	});
391	}	739	}
392	});	740	});
393		741
394	# request two	742	# request two, simply returns 64 octets
395	$handle->push_write ("request 2\015\012");	743	$handle->push_write ("request 2\015\012");
396		744
397	# simply read 64 bytes, always	745	# simply read 64 bytes, always
398	$handle->push_read_chunk (64, sub {	746	$handle->push_read (chunk => 64, sub {
399	my $response = $_[1];	747	my $response = $_[1];
400	...	748	...
401	});	749	});
402		750
403	=over 4	751	=over 4
404		752
405	=cut	753	=cut
406		754
407	sub _drain_rbuf {	755	sub _drain_rbuf {
408	my ($self) = @_;	756	my ($self) = @_;
		757
		758	local $self->{_in_drain} = 1;
409		759
410	if (	760	if (
411	defined $self->{rbuf_max}	761	defined $self->{rbuf_max}
412	&& $self->{rbuf_max} < length $self->{rbuf}	762	&& $self->{rbuf_max} < length $self->{rbuf}
413	) {	763	) {
414	$! = &Errno::ENOSPC; return $self->error;	764	$self->_error (&Errno::ENOSPC, 1), return;
415	}	765	}
416		766
417	return if $self->{in_drain};	767	while () {
418	local $self->{in_drain} = 1;	768	# we need to use a separate tls read buffer, as we must not receive data while
		769	# we are draining the buffer, and this can only happen with TLS.
		770	$self->{rbuf} .= delete $self->{_tls_rbuf} if exists $self->{_tls_rbuf};
419		771
420	while (my $len = length $self->{rbuf}) {	772	my $len = length $self->{rbuf};
421	no strict 'refs';	773
422	if (my $cb = shift @{ $self->{queue} }) {	774	if (my $cb = shift @{ $self->{_queue} }) {
423	if (!$cb->($self)) {	775	unless ($cb->($self)) {
424	if ($self->{eof}) {	776	if ($self->{_eof}) {
425	# no progress can be made (not enough data and no data forthcoming)	777	# no progress can be made (not enough data and no data forthcoming)
426	$! = &Errno::EPIPE; return $self->error;	778	$self->_error (&Errno::EPIPE, 1), return;
427	}	779	}
428		780
429	unshift @{ $self->{queue} }, $cb;	781	unshift @{ $self->{_queue} }, $cb;
430	return;	782	last;
431	}	783	}
432	} elsif ($self->{on_read}) {	784	} elsif ($self->{on_read}) {
		785	last unless $len;
		786
433	$self->{on_read}($self);	787	$self->{on_read}($self);
434		788
435	if (	789	if (
436	$self->{eof} # if no further data will arrive
437	&& $len == length $self->{rbuf} # and no data has been consumed	790	$len == length $self->{rbuf} # if no data has been consumed
438	&& !@{ $self->{queue} } # and the queue is still empty	791	&& !@{ $self->{_queue} } # and the queue is still empty
439	&& $self->{on_read} # and we still want to read data	792	&& $self->{on_read} # but we still have on_read
440	) {	793	) {
		794	# no further data will arrive
441	# then no progress can be made	795	# so no progress can be made
442	$! = &Errno::EPIPE; return $self->error;	796	$self->_error (&Errno::EPIPE, 1), return
		797	if $self->{_eof};
		798
		799	last; # more data might arrive
443	}	800	}
444	} else {	801	} else {
445	# read side becomes idle	802	# read side becomes idle
446	delete $self->{rw};	803	delete $self->{_rw} unless $self->{tls};
447	return;	804	last;
448	}	805	}
449	}	806	}
450		807
451	if ($self->{eof}) {	808	if ($self->{_eof}) {
452	$self->_shutdown;	809	if ($self->{on_eof}) {
453	$self->{on_eof}($self)	810	$self->{on_eof}($self)
454	if $self->{on_eof};	811	} else {
		812	$self->_error (0, 1);
		813	}
		814	}
		815
		816	# may need to restart read watcher
		817	unless ($self->{_rw}) {
		818	$self->start_read
		819	if $self->{on_read} || @{ $self->{_queue} };
455	}	820	}
456	}	821	}
457		822
458	=item $handle->on_read ($cb)	823	=item $handle->on_read ($cb)
459		824
…		…
465		830
466	sub on_read {	831	sub on_read {
467	my ($self, $cb) = @_;	832	my ($self, $cb) = @_;
468		833
469	$self->{on_read} = $cb;	834	$self->{on_read} = $cb;
		835	$self->_drain_rbuf if $cb && !$self->{_in_drain};
470	}	836	}
471		837
472	=item $handle->rbuf	838	=item $handle->rbuf
473		839
474	Returns the read buffer (as a modifiable lvalue).	840	Returns the read buffer (as a modifiable lvalue).
475		841
476	You can access the read buffer directly as the C<< ->{rbuf} >> member, if	842	You can access the read buffer directly as the C<< ->{rbuf} >>
477	you want.	843	member, if you want. However, the only operation allowed on the
		844	read buffer (apart from looking at it) is removing data from its
		845	beginning. Otherwise modifying or appending to it is not allowed and will
		846	lead to hard-to-track-down bugs.
478		847
479	NOTE: The read buffer should only be used or modified if the C<on_read>,	848	NOTE: The read buffer should only be used or modified if the C<on_read>,
480	C<push_read> or C<unshift_read> methods are used. The other read methods	849	C<push_read> or C<unshift_read> methods are used. The other read methods
481	automatically manage the read buffer.	850	automatically manage the read buffer.
482		851
…		…
505	interested in (which can be none at all) and return a true value. After returning	874	interested in (which can be none at all) and return a true value. After returning
506	true, it will be removed from the queue.	875	true, it will be removed from the queue.
507		876
508	=cut	877	=cut
509		878
		879	our %RH;
		880
		881	sub register_read_type($$) {
		882	$RH{$_[0]} = $_[1];
		883	}
		884
510	sub push_read {	885	sub push_read {
511	my ($self, $cb) = @_;	886	my $self = shift;
		887	my $cb = pop;
512		888
		889	if (@_) {
		890	my $type = shift;
		891
		892	$cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_read")
		893	->($self, $cb, @_);
		894	}
		895
513	push @{ $self->{queue} }, $cb;	896	push @{ $self->{_queue} }, $cb;
514	$self->_drain_rbuf;	897	$self->_drain_rbuf unless $self->{_in_drain};
515	}	898	}
516		899
517	sub unshift_read {	900	sub unshift_read {
518	my ($self, $cb) = @_;	901	my $self = shift;
		902	my $cb = pop;
519		903
		904	if (@_) {
		905	my $type = shift;
		906
		907	$cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::unshift_read")
		908	->($self, $cb, @_);
		909	}
		910
		911
520	push @{ $self->{queue} }, $cb;	912	unshift @{ $self->{_queue} }, $cb;
521	$self->_drain_rbuf;	913	$self->_drain_rbuf unless $self->{_in_drain};
522	}	914	}
523		915
524	=item $handle->push_read_chunk ($len, $cb->($self, $data))	916	=item $handle->push_read (type => @args, $cb)
525		917
526	=item $handle->unshift_read_chunk ($len, $cb->($self, $data))	918	=item $handle->unshift_read (type => @args, $cb)
527		919
528	Append the given callback to the end of the queue (C<push_read_chunk>) or	920	Instead of providing a callback that parses the data itself you can chose
529	prepend it (C<unshift_read_chunk>).	921	between a number of predefined parsing formats, for chunks of data, lines
		922	etc.
530		923
531	The callback will be called only once C<$len> bytes have been read, and	924	Predefined types are (if you have ideas for additional types, feel free to
532	these C<$len> bytes will be passed to the callback.	925	drop by and tell us):
533		926
534	=cut	927	=over 4
535		928
536	sub _read_chunk($$) {	929	=item chunk => $octets, $cb->($handle, $data)
		930
		931	Invoke the callback only once C<$octets> bytes have been read. Pass the
		932	data read to the callback. The callback will never be called with less
		933	data.
		934
		935	Example: read 2 bytes.
		936
		937	$handle->push_read (chunk => 2, sub {
		938	warn "yay ", unpack "H*", $_[1];
		939	});
		940
		941	=cut
		942
		943	register_read_type chunk => sub {
537	my ($self, $len, $cb) = @_;	944	my ($self, $cb, $len) = @_;
538		945
539	sub {	946	sub {
540	$len <= length $_[0]{rbuf} or return;	947	$len <= length $_[0]{rbuf} or return;
541	$cb->($_[0], substr $_[0]{rbuf}, 0, $len, "");	948	$cb->($_[0], substr $_[0]{rbuf}, 0, $len, "");
542	1	949	1
543	}	950	}
544	}	951	};
545		952
546	sub push_read_chunk {	953	=item line => [$eol, ]$cb->($handle, $line, $eol)
547	$_[0]->push_read (&_read_chunk);
548	}
549
550
551	sub unshift_read_chunk {
552	$_[0]->unshift_read (&_read_chunk);
553	}
554
555	=item $handle->push_read_line ([$eol, ]$cb->($self, $line, $eol))
556
557	=item $handle->unshift_read_line ([$eol, ]$cb->($self, $line, $eol))
558
559	Append the given callback to the end of the queue (C<push_read_line>) or
560	prepend it (C<unshift_read_line>).
561		954
562	The callback will be called only once a full line (including the end of	955	The callback will be called only once a full line (including the end of
563	line marker, C<$eol>) has been read. This line (excluding the end of line	956	line marker, C<$eol>) has been read. This line (excluding the end of line
564	marker) will be passed to the callback as second argument (C<$line>), and	957	marker) will be passed to the callback as second argument (C<$line>), and
565	the end of line marker as the third argument (C<$eol>).	958	the end of line marker as the third argument (C<$eol>).
…		…
576	Partial lines at the end of the stream will never be returned, as they are	969	Partial lines at the end of the stream will never be returned, as they are
577	not marked by the end of line marker.	970	not marked by the end of line marker.
578		971
579	=cut	972	=cut
580		973
581	sub _read_line($$) {	974	register_read_type line => sub {
582	my $self = shift;	975	my ($self, $cb, $eol) = @_;
583	my $cb = pop;
584	my $eol = @_ ? shift : qr|(\015?\012)|;
585	my $pos;
586		976
		977	if (@_ < 3) {
		978	# this is more than twice as fast as the generic code below
		979	sub {
		980	$_[0]{rbuf} =~ s/^([^\015\012]*)(\015?\012)// or return;
		981
		982	$cb->($_[0], $1, $2);
		983	1
		984	}
		985	} else {
587	$eol = quotemeta $eol unless ref $eol;	986	$eol = quotemeta $eol unless ref $eol;
588	$eol = qr|^(.*?)($eol)|s;	987	$eol = qr|^(.*?)($eol)|s;
		988
		989	sub {
		990	$_[0]{rbuf} =~ s/$eol// or return;
		991
		992	$cb->($_[0], $1, $2);
		993	1
		994	}
		995	}
		996	};
		997
		998	=item regex => $accept[, $reject[, $skip], $cb->($handle, $data)
		999
		1000	Makes a regex match against the regex object C<$accept> and returns
		1001	everything up to and including the match.
		1002
		1003	Example: read a single line terminated by '\n'.
		1004
		1005	$handle->push_read (regex => qr<\n>, sub { ... });
		1006
		1007	If C<$reject> is given and not undef, then it determines when the data is
		1008	to be rejected: it is matched against the data when the C<$accept> regex
		1009	does not match and generates an C<EBADMSG> error when it matches. This is
		1010	useful to quickly reject wrong data (to avoid waiting for a timeout or a
		1011	receive buffer overflow).
		1012
		1013	Example: expect a single decimal number followed by whitespace, reject
		1014	anything else (not the use of an anchor).
		1015
		1016	$handle->push_read (regex => qr<^[0-9]+\s>, qr<[^0-9]>, sub { ... });
		1017
		1018	If C<$skip> is given and not C<undef>, then it will be matched against
		1019	the receive buffer when neither C<$accept> nor C<$reject> match,
		1020	and everything preceding and including the match will be accepted
		1021	unconditionally. This is useful to skip large amounts of data that you
		1022	know cannot be matched, so that the C<$accept> or C<$reject> regex do not
		1023	have to start matching from the beginning. This is purely an optimisation
		1024	and is usually worth only when you expect more than a few kilobytes.
		1025
		1026	Example: expect a http header, which ends at C<\015\012\015\012>. Since we
		1027	expect the header to be very large (it isn't in practise, but...), we use
		1028	a skip regex to skip initial portions. The skip regex is tricky in that
		1029	it only accepts something not ending in either \015 or \012, as these are
		1030	required for the accept regex.
		1031
		1032	$handle->push_read (regex =>
		1033	qr<\015\012\015\012>,
		1034	undef, # no reject
		1035	qr<^.*[^\015\012]>,
		1036	sub { ... });
		1037
		1038	=cut
		1039
		1040	register_read_type regex => sub {
		1041	my ($self, $cb, $accept, $reject, $skip) = @_;
		1042
		1043	my $data;
		1044	my $rbuf = \$self->{rbuf};
589		1045
590	sub {	1046	sub {
591	$_[0]{rbuf} =~ s/$eol// or return;	1047	# accept
		1048	if ($$rbuf =~ $accept) {
		1049	$data .= substr $$rbuf, 0, $+[0], "";
		1050	$cb->($self, $data);
		1051	return 1;
		1052	}
		1053
		1054	# reject
		1055	if ($reject && $$rbuf =~ $reject) {
		1056	$self->_error (&Errno::EBADMSG);
		1057	}
592		1058
593	$cb->($_[0], $1, $2);	1059	# skip
		1060	if ($skip && $$rbuf =~ $skip) {
		1061	$data .= substr $$rbuf, 0, $+[0], "";
		1062	}
		1063
		1064	()
		1065	}
		1066	};
		1067
		1068	=item netstring => $cb->($handle, $string)
		1069
		1070	A netstring (http://cr.yp.to/proto/netstrings.txt, this is not an endorsement).
		1071
		1072	Throws an error with C<$!> set to EBADMSG on format violations.
		1073
		1074	=cut
		1075
		1076	register_read_type netstring => sub {
		1077	my ($self, $cb) = @_;
		1078
		1079	sub {
		1080	unless ($_[0]{rbuf} =~ s/^(0|[1-9][0-9]*)://) {
		1081	if ($_[0]{rbuf} =~ /[^0-9]/) {
		1082	$self->_error (&Errno::EBADMSG);
		1083	}
		1084	return;
		1085	}
		1086
		1087	my $len = $1;
		1088
		1089	$self->unshift_read (chunk => $len, sub {
		1090	my $string = $_[1];
		1091	$_[0]->unshift_read (chunk => 1, sub {
		1092	if ($_[1] eq ",") {
		1093	$cb->($_[0], $string);
		1094	} else {
		1095	$self->_error (&Errno::EBADMSG);
		1096	}
		1097	});
		1098	});
		1099
594	1	1100	1
595	}	1101	}
596	}	1102	};
597		1103
598	sub push_read_line {	1104	=item packstring => $format, $cb->($handle, $string)
599	$_[0]->push_read (&_read_line);
600	}
601		1105
602	sub unshift_read_line {	1106	An octet string prefixed with an encoded length. The encoding C<$format>
603	$_[0]->unshift_read (&_read_line);	1107	uses the same format as a Perl C<pack> format, but must specify a single
604	}	1108	integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
		1109	optional C<!>, C<< < >> or C<< > >> modifier).
		1110
		1111	For example, DNS over TCP uses a prefix of C<n> (2 octet network order),
		1112	EPP uses a prefix of C<N> (4 octtes).
		1113
		1114	Example: read a block of data prefixed by its length in BER-encoded
		1115	format (very efficient).
		1116
		1117	$handle->push_read (packstring => "w", sub {
		1118	my ($handle, $data) = @_;
		1119	});
		1120
		1121	=cut
		1122
		1123	register_read_type packstring => sub {
		1124	my ($self, $cb, $format) = @_;
		1125
		1126	sub {
		1127	# when we can use 5.10 we can use ".", but for 5.8 we use the re-pack method
		1128	defined (my $len = eval { unpack $format, $_[0]{rbuf} })
		1129	or return;
		1130
		1131	$format = length pack $format, $len;
		1132
		1133	# bypass unshift if we already have the remaining chunk
		1134	if ($format + $len <= length $_[0]{rbuf}) {
		1135	my $data = substr $_[0]{rbuf}, $format, $len;
		1136	substr $_[0]{rbuf}, 0, $format + $len, "";
		1137	$cb->($_[0], $data);
		1138	} else {
		1139	# remove prefix
		1140	substr $_[0]{rbuf}, 0, $format, "";
		1141
		1142	# read remaining chunk
		1143	$_[0]->unshift_read (chunk => $len, $cb);
		1144	}
		1145
		1146	1
		1147	}
		1148	};
		1149
		1150	=item json => $cb->($handle, $hash_or_arrayref)
		1151
		1152	Reads a JSON object or array, decodes it and passes it to the
		1153	callback. When a parse error occurs, an C<EBADMSG> error will be raised.
		1154
		1155	If a C<json> object was passed to the constructor, then that will be used
		1156	for the final decode, otherwise it will create a JSON coder expecting UTF-8.
		1157
		1158	This read type uses the incremental parser available with JSON version
		1159	2.09 (and JSON::XS version 2.2) and above. You have to provide a
		1160	dependency on your own: this module will load the JSON module, but
		1161	AnyEvent does not depend on it itself.
		1162
		1163	Since JSON texts are fully self-delimiting, the C<json> read and write
		1164	types are an ideal simple RPC protocol: just exchange JSON datagrams. See
		1165	the C<json> write type description, above, for an actual example.
		1166
		1167	=cut
		1168
		1169	register_read_type json => sub {
		1170	my ($self, $cb) = @_;
		1171
		1172	require JSON;
		1173
		1174	my $data;
		1175	my $rbuf = \$self->{rbuf};
		1176
		1177	my $json = $self->{json} ||= JSON->new->utf8;
		1178
		1179	sub {
		1180	my $ref = eval { $json->incr_parse ($self->{rbuf}) };
		1181
		1182	if ($ref) {
		1183	$self->{rbuf} = $json->incr_text;
		1184	$json->incr_text = "";
		1185	$cb->($self, $ref);
		1186
		1187	1
		1188	} elsif ($@) {
		1189	# error case
		1190	$json->incr_skip;
		1191
		1192	$self->{rbuf} = $json->incr_text;
		1193	$json->incr_text = "";
		1194
		1195	$self->_error (&Errno::EBADMSG);
		1196
		1197	()
		1198	} else {
		1199	$self->{rbuf} = "";
		1200
		1201	()
		1202	}
		1203	}
		1204	};
		1205
		1206	=item storable => $cb->($handle, $ref)
		1207
		1208	Deserialises a L<Storable> frozen representation as written by the
		1209	C<storable> write type (BER-encoded length prefix followed by nfreeze'd
		1210	data).
		1211
		1212	Raises C<EBADMSG> error if the data could not be decoded.
		1213
		1214	=cut
		1215
		1216	register_read_type storable => sub {
		1217	my ($self, $cb) = @_;
		1218
		1219	require Storable;
		1220
		1221	sub {
		1222	# when we can use 5.10 we can use ".", but for 5.8 we use the re-pack method
		1223	defined (my $len = eval { unpack "w", $_[0]{rbuf} })
		1224	or return;
		1225
		1226	my $format = length pack "w", $len;
		1227
		1228	# bypass unshift if we already have the remaining chunk
		1229	if ($format + $len <= length $_[0]{rbuf}) {
		1230	my $data = substr $_[0]{rbuf}, $format, $len;
		1231	substr $_[0]{rbuf}, 0, $format + $len, "";
		1232	$cb->($_[0], Storable::thaw ($data));
		1233	} else {
		1234	# remove prefix
		1235	substr $_[0]{rbuf}, 0, $format, "";
		1236
		1237	# read remaining chunk
		1238	$_[0]->unshift_read (chunk => $len, sub {
		1239	if (my $ref = eval { Storable::thaw ($_[1]) }) {
		1240	$cb->($_[0], $ref);
		1241	} else {
		1242	$self->_error (&Errno::EBADMSG);
		1243	}
		1244	});
		1245	}
		1246
		1247	1
		1248	}
		1249	};
		1250
		1251	=back
		1252
		1253	=item AnyEvent::Handle::register_read_type type => $coderef->($handle, $cb, @args)
		1254
		1255	This function (not method) lets you add your own types to C<push_read>.
		1256
		1257	Whenever the given C<type> is used, C<push_read> will invoke the code
		1258	reference with the handle object, the callback and the remaining
		1259	arguments.
		1260
		1261	The code reference is supposed to return a callback (usually a closure)
		1262	that works as a plain read callback (see C<< ->push_read ($cb) >>).
		1263
		1264	It should invoke the passed callback when it is done reading (remember to
		1265	pass C<$handle> as first argument as all other callbacks do that).
		1266
		1267	Note that this is a function, and all types registered this way will be
		1268	global, so try to use unique names.
		1269
		1270	For examples, see the source of this module (F<perldoc -m AnyEvent::Handle>,
		1271	search for C<register_read_type>)).
605		1272
606	=item $handle->stop_read	1273	=item $handle->stop_read
607		1274
608	=item $handle->start_read	1275	=item $handle->start_read
609		1276
610	In rare cases you actually do not want to read anything from the	1277	In rare cases you actually do not want to read anything from the
611	socket. In this case you can call C<stop_read>. Neither C<on_read> no	1278	socket. In this case you can call C<stop_read>. Neither C<on_read> nor
612	any queued callbacks will be executed then. To start reading again, call	1279	any queued callbacks will be executed then. To start reading again, call
613	C<start_read>.	1280	C<start_read>.
614		1281
		1282	Note that AnyEvent::Handle will automatically C<start_read> for you when
		1283	you change the C<on_read> callback or push/unshift a read callback, and it
		1284	will automatically C<stop_read> for you when neither C<on_read> is set nor
		1285	there are any read requests in the queue.
		1286
		1287	These methods will have no effect when in TLS mode (as TLS doesn't support
		1288	half-duplex connections).
		1289
615	=cut	1290	=cut
616		1291
617	sub stop_read {	1292	sub stop_read {
618	my ($self) = @_;	1293	my ($self) = @_;
619		1294
620	delete $self->{rw};	1295	delete $self->{_rw} unless $self->{tls};
621	}	1296	}
622		1297
623	sub start_read {	1298	sub start_read {
624	my ($self) = @_;	1299	my ($self) = @_;
625		1300
626	unless ($self->{rw} || $self->{eof}) {	1301	unless ($self->{_rw} || $self->{_eof}) {
627	Scalar::Util::weaken $self;	1302	Scalar::Util::weaken $self;
628		1303
629	$self->{rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {	1304	$self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {
630	my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf};	1305	my $rbuf = \($self->{tls} ? my $buf : $self->{rbuf});
631	my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf;	1306	my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf;
632		1307
633	if ($len > 0) {	1308	if ($len > 0) {
634	$self->{filter_r}	1309	$self->{_activity} = AnyEvent->now;
635	? $self->{filter_r}->($self, $rbuf)	1310
636	: $self->_drain_rbuf;	1311	if ($self->{tls}) {
		1312	Net::SSLeay::BIO_write ($self->{_rbio}, $$rbuf);
		1313
		1314	&_dotls ($self);
		1315	} else {
		1316	$self->_drain_rbuf unless $self->{_in_drain};
		1317	}
637		1318
638	} elsif (defined $len) {	1319	} elsif (defined $len) {
639	delete $self->{rw};	1320	delete $self->{_rw};
640	$self->{eof} = 1;	1321	$self->{_eof} = 1;
641	$self->_drain_rbuf;	1322	$self->_drain_rbuf unless $self->{_in_drain};
642		1323
643	} elsif ($! != EAGAIN && $! != EINTR) {	1324	} elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
644	return $self->error;	1325	return $self->_error ($!, 1);
645	}	1326	}
646	});	1327	});
647	}	1328	}
648	}	1329	}
649		1330
		1331	# poll the write BIO and send the data if applicable
650	sub _dotls {	1332	sub _dotls {
651	my ($self) = @_;	1333	my ($self) = @_;
652		1334
		1335	my $tmp;
		1336
653	if (length $self->{tls_wbuf}) {	1337	if (length $self->{_tls_wbuf}) {
654	while ((my $len = Net::SSLeay::write ($self->{tls}, $self->{tls_wbuf})) > 0) {	1338	while (($tmp = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) {
655	substr $self->{tls_wbuf}, 0, $len, "";	1339	substr $self->{_tls_wbuf}, 0, $tmp, "";
656	}	1340	}
657	}	1341	}
658		1342
659	if (defined (my $buf = Net::SSLeay::BIO_read ($self->{tls_wbio}))) {	1343	while (defined ($tmp = Net::SSLeay::read ($self->{tls}))) {
		1344	unless (length $tmp) {
		1345	# let's treat SSL-eof as we treat normal EOF
		1346	delete $self->{_rw};
		1347	$self->{_eof} = 1;
		1348	&_freetls;
		1349	}
		1350
		1351	$self->{_tls_rbuf} .= $tmp;
		1352	$self->_drain_rbuf unless $self->{_in_drain};
		1353	$self->{tls} or return; # tls session might have gone away in callback
		1354	}
		1355
		1356	$tmp = Net::SSLeay::get_error ($self->{tls}, -1);
		1357
		1358	if ($tmp != Net::SSLeay::ERROR_WANT_READ ()) {
		1359	if ($tmp == Net::SSLeay::ERROR_SYSCALL ()) {
		1360	return $self->_error ($!, 1);
		1361	} elsif ($tmp == Net::SSLeay::ERROR_SSL ()) {
		1362	return $self->_error (&Errno::EIO, 1);
		1363	}
		1364
		1365	# all other errors are fine for our purposes
		1366	}
		1367
		1368	while (length ($tmp = Net::SSLeay::BIO_read ($self->{_wbio}))) {
660	$self->{wbuf} .= $buf;	1369	$self->{wbuf} .= $tmp;
661	$self->_drain_wbuf;	1370	$self->_drain_wbuf;
662	}
663
664	while (defined (my $buf = Net::SSLeay::read ($self->{tls}))) {
665	$self->{rbuf} .= $buf;
666	$self->_drain_rbuf;
667	}
668
669	my $err = Net::SSLeay::get_error ($self->{tls}, -1);
670
671	if ($err!= Net::SSLeay::ERROR_WANT_READ ()) {
672	if ($err == Net::SSLeay::ERROR_SYSCALL ()) {
673	$self->error;
674	} elsif ($err == Net::SSLeay::ERROR_SSL ()) {
675	$! = &Errno::EIO;
676	$self->error;
677	}
678
679	# all others are fine for our purposes
680	}	1371	}
681	}	1372	}
682		1373
683	=item $handle->starttls ($tls[, $tls_ctx])	1374	=item $handle->starttls ($tls[, $tls_ctx])
684		1375
…		…
690	C<"connect">, C<"accept"> or an existing Net::SSLeay object).	1381	C<"connect">, C<"accept"> or an existing Net::SSLeay object).
691		1382
692	The second argument is the optional C<Net::SSLeay::CTX> object that is	1383	The second argument is the optional C<Net::SSLeay::CTX> object that is
693	used when AnyEvent::Handle has to create its own TLS connection object.	1384	used when AnyEvent::Handle has to create its own TLS connection object.
694		1385
695	=cut	1386	The TLS connection object will end up in C<< $handle->{tls} >> after this
		1387	call and can be used or changed to your liking. Note that the handshake
		1388	might have already started when this function returns.
696		1389
697	# TODO: maybe document...	1390	If it an error to start a TLS handshake more than once per
		1391	AnyEvent::Handle object (this is due to bugs in OpenSSL).
		1392
		1393	=cut
		1394
698	sub starttls {	1395	sub starttls {
699	my ($self, $ssl, $ctx) = @_;	1396	my ($self, $ssl, $ctx) = @_;
700		1397
701	$self->stoptls;	1398	require Net::SSLeay;
702		1399
		1400	Carp::croak "it is an error to call starttls more than once on an AnyEvent::Handle object"
		1401	if $self->{tls};
		1402
703	if ($ssl eq "accept") {	1403	if ($ssl eq "accept") {
704	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());	1404	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
705	Net::SSLeay::set_accept_state ($ssl);	1405	Net::SSLeay::set_accept_state ($ssl);
706	} elsif ($ssl eq "connect") {	1406	} elsif ($ssl eq "connect") {
707	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());	1407	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
…		…
713	# basically, this is deep magic (because SSL_read should have the same issues)	1413	# basically, this is deep magic (because SSL_read should have the same issues)
714	# but the openssl maintainers basically said: "trust us, it just works".	1414	# but the openssl maintainers basically said: "trust us, it just works".
715	# (unfortunately, we have to hardcode constants because the abysmally misdesigned	1415	# (unfortunately, we have to hardcode constants because the abysmally misdesigned
716	# and mismaintained ssleay-module doesn't even offer them).	1416	# and mismaintained ssleay-module doesn't even offer them).
717	# http://www.mail-archive.com/openssl-dev@openssl.org/msg22420.html	1417	# http://www.mail-archive.com/openssl-dev@openssl.org/msg22420.html
		1418	#
		1419	# in short: this is a mess.
		1420	#
		1421	# note that we do not try to keep the length constant between writes as we are required to do.
		1422	# we assume that most (but not all) of this insanity only applies to non-blocking cases,
		1423	# and we drive openssl fully in blocking mode here. Or maybe we don't - openssl seems to
		1424	# have identity issues in that area.
718	Net::SSLeay::CTX_set_mode ($self->{tls},	1425	Net::SSLeay::CTX_set_mode ($self->{tls},
719	(eval { Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)	1426	(eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)
720	| (eval { Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));	1427	| (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));
721		1428
722	$self->{tls_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());	1429	$self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
723	$self->{tls_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());	1430	$self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
724		1431
725	Net::SSLeay::set_bio ($ssl, $self->{tls_rbio}, $self->{tls_wbio});	1432	Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio});
726		1433
727	$self->{filter_w} = sub {	1434	&_dotls; # need to trigger the initial handshake
728	$_[0]{tls_wbuf} .= ${$_[1]};	1435	$self->start_read; # make sure we actually do read
729	&_dotls;
730	};
731	$self->{filter_r} = sub {
732	Net::SSLeay::BIO_write ($_[0]{tls_rbio}, ${$_[1]});
733	&_dotls;
734	};
735	}	1436	}
736		1437
737	=item $handle->stoptls	1438	=item $handle->stoptls
738		1439
739	Destroys the SSL connection, if any. Partial read or write data will be	1440	Shuts down the SSL connection - this makes a proper EOF handshake by
740	lost.	1441	sending a close notify to the other side, but since OpenSSL doesn't
		1442	support non-blocking shut downs, it is not possible to re-use the stream
		1443	afterwards.
741		1444
742	=cut	1445	=cut
743		1446
744	sub stoptls {	1447	sub stoptls {
745	my ($self) = @_;	1448	my ($self) = @_;
746		1449
		1450	if ($self->{tls}) {
		1451	Net::SSLeay::shutdown ($self->{tls});
		1452
		1453	&_dotls;
		1454
		1455	# we don't give a shit. no, we do, but we can't. no...
		1456	# we, we... have to use openssl :/
		1457	&_freetls;
		1458	}
		1459	}
		1460
		1461	sub _freetls {
		1462	my ($self) = @_;
		1463
		1464	return unless $self->{tls};
		1465
747	Net::SSLeay::free (delete $self->{tls}) if $self->{tls};	1466	Net::SSLeay::free (delete $self->{tls});
748	delete $self->{tls_rbio};	1467
749	delete $self->{tls_wbio};	1468	delete @$self{qw(_rbio _wbio _tls_wbuf)};
750	delete $self->{tls_wbuf};
751	delete $self->{filter_r};
752	delete $self->{filter_w};
753	}	1469	}
754		1470
755	sub DESTROY {	1471	sub DESTROY {
756	my $self = shift;	1472	my ($self) = @_;
757		1473
758	$self->stoptls;	1474	&_freetls;
		1475
		1476	my $linger = exists $self->{linger} ? $self->{linger} : 3600;
		1477
		1478	if ($linger && length $self->{wbuf}) {
		1479	my $fh = delete $self->{fh};
		1480	my $wbuf = delete $self->{wbuf};
		1481
		1482	my @linger;
		1483
		1484	push @linger, AnyEvent->io (fh => $fh, poll => "w", cb => sub {
		1485	my $len = syswrite $fh, $wbuf, length $wbuf;
		1486
		1487	if ($len > 0) {
		1488	substr $wbuf, 0, $len, "";
		1489	} else {
		1490	@linger = (); # end
		1491	}
		1492	});
		1493	push @linger, AnyEvent->timer (after => $linger, cb => sub {
		1494	@linger = ();
		1495	});
		1496	}
		1497	}
		1498
		1499	=item $handle->destroy
		1500
		1501	Shuts down the handle object as much as possible - this call ensures that
		1502	no further callbacks will be invoked and resources will be freed as much
		1503	as possible. You must not call any methods on the object afterwards.
		1504
		1505	Normally, you can just "forget" any references to an AnyEvent::Handle
		1506	object and it will simply shut down. This works in fatal error and EOF
		1507	callbacks, as well as code outside. It does I<NOT> work in a read or write
		1508	callback, so when you want to destroy the AnyEvent::Handle object from
		1509	within such an callback. You I<MUST> call C<< ->destroy >> explicitly in
		1510	that case.
		1511
		1512	The handle might still linger in the background and write out remaining
		1513	data, as specified by the C<linger> option, however.
		1514
		1515	=cut
		1516
		1517	sub destroy {
		1518	my ($self) = @_;
		1519
		1520	$self->DESTROY;
		1521	%$self = ();
759	}	1522	}
760		1523
761	=item AnyEvent::Handle::TLS_CTX	1524	=item AnyEvent::Handle::TLS_CTX
762		1525
763	This function creates and returns the Net::SSLeay::CTX object used by	1526	This function creates and returns the Net::SSLeay::CTX object used by
…		…
793	}	1556	}
794	}	1557	}
795		1558
796	=back	1559	=back
797		1560
		1561
		1562	=head1 NONFREQUENTLY ASKED QUESTIONS
		1563
		1564	=over 4
		1565
		1566	=item I C<undef> the AnyEvent::Handle reference inside my callback and
		1567	still get further invocations!
		1568
		1569	That's because AnyEvent::Handle keeps a reference to itself when handling
		1570	read or write callbacks.
		1571
		1572	It is only safe to "forget" the reference inside EOF or error callbacks,
		1573	from within all other callbacks, you need to explicitly call the C<<
		1574	->destroy >> method.
		1575
		1576	=item I get different callback invocations in TLS mode/Why can't I pause
		1577	reading?
		1578
		1579	Unlike, say, TCP, TLS connections do not consist of two independent
		1580	communication channels, one for each direction. Or put differently. The
		1581	read and write directions are not independent of each other: you cannot
		1582	write data unless you are also prepared to read, and vice versa.
		1583
		1584	This can mean than, in TLS mode, you might get C<on_error> or C<on_eof>
		1585	callback invocations when you are not expecting any read data - the reason
		1586	is that AnyEvent::Handle always reads in TLS mode.
		1587
		1588	During the connection, you have to make sure that you always have a
		1589	non-empty read-queue, or an C<on_read> watcher. At the end of the
		1590	connection (or when you no longer want to use it) you can call the
		1591	C<destroy> method.
		1592
		1593	=item How do I read data until the other side closes the connection?
		1594
		1595	If you just want to read your data into a perl scalar, the easiest way
		1596	to achieve this is by setting an C<on_read> callback that does nothing,
		1597	clearing the C<on_eof> callback and in the C<on_error> callback, the data
		1598	will be in C<$_[0]{rbuf}>:
		1599
		1600	$handle->on_read (sub { });
		1601	$handle->on_eof (undef);
		1602	$handle->on_error (sub {
		1603	my $data = delete $_[0]{rbuf};
		1604	undef $handle;
		1605	});
		1606
		1607	The reason to use C<on_error> is that TCP connections, due to latencies
		1608	and packets loss, might get closed quite violently with an error, when in
		1609	fact, all data has been received.
		1610
		1611	It is usually better to use acknowledgements when transferring data,
		1612	to make sure the other side hasn't just died and you got the data
		1613	intact. This is also one reason why so many internet protocols have an
		1614	explicit QUIT command.
		1615
		1616	=item I don't want to destroy the handle too early - how do I wait until
		1617	all data has been written?
		1618
		1619	After writing your last bits of data, set the C<on_drain> callback
		1620	and destroy the handle in there - with the default setting of
		1621	C<low_water_mark> this will be called precisely when all data has been
		1622	written to the socket:
		1623
		1624	$handle->push_write (...);
		1625	$handle->on_drain (sub {
		1626	warn "all data submitted to the kernel\n";
		1627	undef $handle;
		1628	});
		1629
		1630	=back
		1631
		1632
		1633	=head1 SUBCLASSING AnyEvent::Handle
		1634
		1635	In many cases, you might want to subclass AnyEvent::Handle.
		1636
		1637	To make this easier, a given version of AnyEvent::Handle uses these
		1638	conventions:
		1639
		1640	=over 4
		1641
		1642	=item * all constructor arguments become object members.
		1643
		1644	At least initially, when you pass a C<tls>-argument to the constructor it
		1645	will end up in C<< $handle->{tls} >>. Those members might be changed or
		1646	mutated later on (for example C<tls> will hold the TLS connection object).
		1647
		1648	=item * other object member names are prefixed with an C<_>.
		1649
		1650	All object members not explicitly documented (internal use) are prefixed
		1651	with an underscore character, so the remaining non-C<_>-namespace is free
		1652	for use for subclasses.
		1653
		1654	=item * all members not documented here and not prefixed with an underscore
		1655	are free to use in subclasses.
		1656
		1657	Of course, new versions of AnyEvent::Handle may introduce more "public"
		1658	member variables, but thats just life, at least it is documented.
		1659
		1660	=back
		1661
798	=head1 AUTHOR	1662	=head1 AUTHOR
799		1663
800	Robin Redeker C<< <elmex at ta-sa.org> >>, Marc Lehmann <schmorp@schmorp.de>.	1664	Robin Redeker C<< <elmex at ta-sa.org> >>, Marc Lehmann <schmorp@schmorp.de>.
801		1665
802	=cut	1666	=cut

Diff Legend

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