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.28 by root, Sat May 24 22:27:11 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
173	our (%RH, %WH);
174
175	sub register_read_type($$) {
176	$RH{$_[0]} = $_[1];
177	}
178
179	sub register_write_type($$) {
180	$WH{$_[0]} = $_[1];
181	}
182		289
183	sub new {	290	sub new {
184	my $class = shift;	291	my $class = shift;
185		292
186	my $self = bless { @_ }, $class;	293	my $self = bless { @_ }, $class;
187		294
188	$self->{fh} or Carp::croak "mandatory argument fh is missing";	295	$self->{fh} or Carp::croak "mandatory argument fh is missing";
189		296
190	AnyEvent::Util::fh_nonblocking $self->{fh}, 1;	297	AnyEvent::Util::fh_nonblocking $self->{fh}, 1;
191		298
192	if ($self->{tls}) {
193	require Net::SSLeay;
194	$self->starttls (delete $self->{tls}, delete $self->{tls_ctx});	299	$self->starttls (delete $self->{tls}, delete $self->{tls_ctx})
195	}	300	if $self->{tls};
196		301
197	$self->on_eof (delete $self->{on_eof} ) if $self->{on_eof};	302	$self->{_activity} = AnyEvent->now;
198	$self->on_error (delete $self->{on_error}) if $self->{on_error};	303	$self->_timeout;
		304
199	$self->on_drain (delete $self->{on_drain}) if $self->{on_drain};	305	$self->on_drain (delete $self->{on_drain}) if exists $self->{on_drain};
200	$self->on_read (delete $self->{on_read} ) if $self->{on_read};	306	$self->no_delay (delete $self->{no_delay}) if exists $self->{no_delay};
201		307
202	$self->start_read;	308	$self->start_read
		309	if $self->{on_read};
203		310
204	$self	311	$self
205	}	312	}
206		313
207	sub _shutdown {	314	sub _shutdown {
208	my ($self) = @_;	315	my ($self) = @_;
209		316
210	delete $self->{rw};	317	delete @$self{qw(_tw _rw _ww fh rbuf wbuf on_read _queue)};
211	delete $self->{ww};
212	delete $self->{fh};
213	}
214		318
		319	&_freetls;
		320	}
		321
215	sub error {	322	sub _error {
216	my ($self) = @_;	323	my ($self, $errno, $fatal) = @_;
217		324
218	{
219	local $!;
220	$self->_shutdown;	325	$self->_shutdown
221	}	326	if $fatal;
		327
		328	$! = $errno;
222		329
223	if ($self->{on_error}) {	330	if ($self->{on_error}) {
224	$self->{on_error}($self);	331	$self->{on_error}($self, $fatal);
225	} else {	332	} elsif ($self->{fh}) {
226	die "AnyEvent::Handle uncaught fatal error: $!";	333	Carp::croak "AnyEvent::Handle uncaught error: $!";
227	}	334	}
228	}	335	}
229		336
230	=item $fh = $handle->fh	337	=item $fh = $handle->fh
231		338
232	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.
233		340
234	=cut	341	=cut
235		342
236	sub fh { $_[0]->{fh} }	343	sub fh { $_[0]{fh} }
237		344
238	=item $handle->on_error ($cb)	345	=item $handle->on_error ($cb)
239		346
240	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).
241		348
…		…
253		360
254	sub on_eof {	361	sub on_eof {
255	$_[0]{on_eof} = $_[1];	362	$_[0]{on_eof} = $_[1];
256	}	363	}
257		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
258	#############################################################################	459	#############################################################################
259		460
260	=back	461	=back
261		462
262	=head2 WRITE QUEUE	463	=head2 WRITE QUEUE
…		…
283	my ($self, $cb) = @_;	484	my ($self, $cb) = @_;
284		485
285	$self->{on_drain} = $cb;	486	$self->{on_drain} = $cb;
286		487
287	$cb->($self)	488	$cb->($self)
288	if $cb && $self->{low_water_mark} >= length $self->{wbuf};	489	if $cb && $self->{low_water_mark} >= (length $self->{wbuf}) + (length $self->{_tls_wbuf});
289	}	490	}
290		491
291	=item $handle->push_write ($data)	492	=item $handle->push_write ($data)
292		493
293	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
…		…
297	=cut	498	=cut
298		499
299	sub _drain_wbuf {	500	sub _drain_wbuf {
300	my ($self) = @_;	501	my ($self) = @_;
301		502
302	unless ($self->{ww}) {	503	if (!$self->{_ww} && length $self->{wbuf}) {
		504
303	Scalar::Util::weaken $self;	505	Scalar::Util::weaken $self;
		506
304	my $cb = sub {	507	my $cb = sub {
305	my $len = syswrite $self->{fh}, $self->{wbuf};	508	my $len = syswrite $self->{fh}, $self->{wbuf};
306		509
307	if ($len > 0) {	510	if ($len >= 0) {
308	substr $self->{wbuf}, 0, $len, "";	511	substr $self->{wbuf}, 0, $len, "";
309		512
		513	$self->{_activity} = AnyEvent->now;
		514
310	$self->{on_drain}($self)	515	$self->{on_drain}($self)
311	if $self->{low_water_mark} >= length $self->{wbuf}	516	if $self->{low_water_mark} >= (length $self->{wbuf}) + (length $self->{_tls_wbuf})
312	&& $self->{on_drain};	517	&& $self->{on_drain};
313		518
314	delete $self->{ww} unless length $self->{wbuf};	519	delete $self->{_ww} unless length $self->{wbuf};
315	} elsif ($! != EAGAIN && $! != EINTR) {	520	} elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
316	$self->error;	521	$self->_error ($!, 1);
317	}	522	}
318	};	523	};
319		524
		525	# try to write data immediately
		526	$cb->() unless $self->{autocork};
		527
		528	# if still data left in wbuf, we need to poll
320	$self->{ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb);	529	$self->{_ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb)
321		530	if length $self->{wbuf};
322	$cb->($self);
323	};	531	};
		532	}
		533
		534	our %WH;
		535
		536	sub register_write_type($$) {
		537	$WH{$_[0]} = $_[1];
324	}	538	}
325		539
326	sub push_write {	540	sub push_write {
327	my $self = shift;	541	my $self = shift;
328		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
329	if ($self->{filter_w}) {	550	if ($self->{tls}) {
330	$self->{filter_w}->($self, \$_[0]);	551	$self->{_tls_wbuf} .= $_[0];
		552
		553	&_dotls ($self);
331	} else {	554	} else {
332	$self->{wbuf} .= $_[0];	555	$self->{wbuf} .= $_[0];
333	$self->_drain_wbuf;	556	$self->_drain_wbuf;
334	}	557	}
335	}	558	}
336		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
337	#############################################################################	670	#############################################################################
338		671
339	=back	672	=back
340		673
341	=head2 READ QUEUE	674	=head2 READ QUEUE
…		…
347	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
348	a queue.	681	a queue.
349		682
350	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
351	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
352	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
353	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).
354		688
355	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
356	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
357	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
358	below).	692	done its job (see C<push_read>, below).
359		693
360	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
361	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.
362		696
363	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
364	the specified number of bytes which give an XML datagram.	698	the specified number of bytes which give an XML datagram.
365		699
366	# in the default state, expect some header bytes	700	# in the default state, expect some header bytes
367	$handle->on_read (sub {	701	$handle->on_read (sub {
368	# 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)
369	shift->unshift_read_chunk (4, sub {	703	shift->unshift_read (chunk => 4, sub {
370	# header arrived, decode	704	# header arrived, decode
371	my $len = unpack "N", $_[1];	705	my $len = unpack "N", $_[1];
372		706
373	# now read the payload	707	# now read the payload
374	shift->unshift_read_chunk ($len, sub {	708	shift->unshift_read (chunk => $len, sub {
375	my $xml = $_[1];	709	my $xml = $_[1];
376	# handle xml	710	# handle xml
377	});	711	});
378	});	712	});
379	});	713	});
380		714
381	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"
382	"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
383	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
384	pipeline sending both requests and manipulate the queue as necessary in	718	just pipeline sending both requests and manipulate the queue as necessary
385	the callbacks:	719	in the callbacks.
386		720
387	# 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"
388	$handle->push_write ("request 1\015\012");	726	$handle->push_write ("request 1\015\012");
389		727
390	# 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
391	$handle->push_read_line (sub {	729	$handle->push_read (line => sub {
392	# if we got an "OK", we have to _prepend_ another line,	730	# if we got an "OK", we have to _prepend_ another line,
393	# 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
394	# which are already in the queue when this callback is called	732	# which are already in the queue when this callback is called
395	# we don't do this in case we got an error	733	# we don't do this in case we got an error
396	if ($_[1] eq "OK") {	734	if ($_[1] eq "OK") {
397	$_[0]->unshift_read_line (sub {	735	$_[0]->unshift_read (line => sub {
398	my $response = $_[1];	736	my $response = $_[1];
399	...	737	...
400	});	738	});
401	}	739	}
402	});	740	});
403		741
404	# request two	742	# request two, simply returns 64 octets
405	$handle->push_write ("request 2\015\012");	743	$handle->push_write ("request 2\015\012");
406		744
407	# simply read 64 bytes, always	745	# simply read 64 bytes, always
408	$handle->push_read_chunk (64, sub {	746	$handle->push_read (chunk => 64, sub {
409	my $response = $_[1];	747	my $response = $_[1];
410	...	748	...
411	});	749	});
412		750
413	=over 4	751	=over 4
414		752
415	=cut	753	=cut
416		754
417	sub _drain_rbuf {	755	sub _drain_rbuf {
418	my ($self) = @_;	756	my ($self) = @_;
		757
		758	local $self->{_in_drain} = 1;
419		759
420	if (	760	if (
421	defined $self->{rbuf_max}	761	defined $self->{rbuf_max}
422	&& $self->{rbuf_max} < length $self->{rbuf}	762	&& $self->{rbuf_max} < length $self->{rbuf}
423	) {	763	) {
424	$! = &Errno::ENOSPC; return $self->error;	764	$self->_error (&Errno::ENOSPC, 1), return;
425	}	765	}
426		766
427	return if $self->{in_drain};	767	while () {
428	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};
429		771
430	while (my $len = length $self->{rbuf}) {	772	my $len = length $self->{rbuf};
431	no strict 'refs';	773
432	if (my $cb = shift @{ $self->{queue} }) {	774	if (my $cb = shift @{ $self->{_queue} }) {
433	if (!$cb->($self)) {	775	unless ($cb->($self)) {
434	if ($self->{eof}) {	776	if ($self->{_eof}) {
435	# 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)
436	$! = &Errno::EPIPE; return $self->error;	778	$self->_error (&Errno::EPIPE, 1), return;
437	}	779	}
438		780
439	unshift @{ $self->{queue} }, $cb;	781	unshift @{ $self->{_queue} }, $cb;
440	return;	782	last;
441	}	783	}
442	} elsif ($self->{on_read}) {	784	} elsif ($self->{on_read}) {
		785	last unless $len;
		786
443	$self->{on_read}($self);	787	$self->{on_read}($self);
444		788
445	if (	789	if (
446	$self->{eof} # if no further data will arrive
447	&& $len == length $self->{rbuf} # and no data has been consumed	790	$len == length $self->{rbuf} # if no data has been consumed
448	&& !@{ $self->{queue} } # and the queue is still empty	791	&& !@{ $self->{_queue} } # and the queue is still empty
449	&& $self->{on_read} # and we still want to read data	792	&& $self->{on_read} # but we still have on_read
450	) {	793	) {
		794	# no further data will arrive
451	# then no progress can be made	795	# so no progress can be made
452	$! = &Errno::EPIPE; return $self->error;	796	$self->_error (&Errno::EPIPE, 1), return
		797	if $self->{_eof};
		798
		799	last; # more data might arrive
453	}	800	}
454	} else {	801	} else {
455	# read side becomes idle	802	# read side becomes idle
456	delete $self->{rw};	803	delete $self->{_rw} unless $self->{tls};
457	return;	804	last;
458	}	805	}
459	}	806	}
460		807
461	if ($self->{eof}) {	808	if ($self->{_eof}) {
462	$self->_shutdown;	809	if ($self->{on_eof}) {
463	$self->{on_eof}($self)	810	$self->{on_eof}($self)
464	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} };
465	}	820	}
466	}	821	}
467		822
468	=item $handle->on_read ($cb)	823	=item $handle->on_read ($cb)
469		824
…		…
475		830
476	sub on_read {	831	sub on_read {
477	my ($self, $cb) = @_;	832	my ($self, $cb) = @_;
478		833
479	$self->{on_read} = $cb;	834	$self->{on_read} = $cb;
		835	$self->_drain_rbuf if $cb && !$self->{_in_drain};
480	}	836	}
481		837
482	=item $handle->rbuf	838	=item $handle->rbuf
483		839
484	Returns the read buffer (as a modifiable lvalue).	840	Returns the read buffer (as a modifiable lvalue).
485		841
486	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} >>
487	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.
488		847
489	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>,
490	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
491	automatically manage the read buffer.	850	automatically manage the read buffer.
492		851
…		…
515	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
516	true, it will be removed from the queue.	875	true, it will be removed from the queue.
517		876
518	=cut	877	=cut
519		878
		879	our %RH;
		880
		881	sub register_read_type($$) {
		882	$RH{$_[0]} = $_[1];
		883	}
		884
520	sub push_read {	885	sub push_read {
521	my $self = shift;	886	my $self = shift;
522	my $cb = pop;	887	my $cb = pop;
523		888
524	if (@_) {	889	if (@_) {
…		…
526		891
527	$cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_read")	892	$cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_read")
528	->($self, $cb, @_);	893	->($self, $cb, @_);
529	}	894	}
530		895
531	push @{ $self->{queue} }, $cb;	896	push @{ $self->{_queue} }, $cb;
532	$self->_drain_rbuf;	897	$self->_drain_rbuf unless $self->{_in_drain};
533	}	898	}
534		899
535	sub unshift_read {	900	sub unshift_read {
536	my $self = shift;	901	my $self = shift;
537	my $cb = pop;	902	my $cb = pop;
…		…
542	$cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::unshift_read")	907	$cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::unshift_read")
543	->($self, $cb, @_);	908	->($self, $cb, @_);
544	}	909	}
545		910
546		911
547	unshift @{ $self->{queue} }, $cb;	912	unshift @{ $self->{_queue} }, $cb;
548	$self->_drain_rbuf;	913	$self->_drain_rbuf unless $self->{_in_drain};
549	}	914	}
550		915
551	=item $handle->push_read (type => @args, $cb)	916	=item $handle->push_read (type => @args, $cb)
552		917
553	=item $handle->unshift_read (type => @args, $cb)	918	=item $handle->unshift_read (type => @args, $cb)
554		919
555	Instead of providing a callback that parses the data itself you can chose	920	Instead of providing a callback that parses the data itself you can chose
556	between a number of predefined parsing formats, for chunks of data, lines	921	between a number of predefined parsing formats, for chunks of data, lines
557	etc.	922	etc.
558		923
559	The types currently supported are:	924	Predefined types are (if you have ideas for additional types, feel free to
		925	drop by and tell us):
560		926
561	=over 4	927	=over 4
562		928
563	=item chunk => $octets, $cb->($self, $data)	929	=item chunk => $octets, $cb->($handle, $data)
564		930
565	Invoke the callback only once C<$octets> bytes have been read. Pass the	931	Invoke the callback only once C<$octets> bytes have been read. Pass the
566	data read to the callback. The callback will never be called with less	932	data read to the callback. The callback will never be called with less
567	data.	933	data.
568		934
…		…
582	$cb->($_[0], substr $_[0]{rbuf}, 0, $len, "");	948	$cb->($_[0], substr $_[0]{rbuf}, 0, $len, "");
583	1	949	1
584	}	950	}
585	};	951	};
586		952
587	# compatibility with older API
588	sub push_read_chunk {
589	$_[0]->push_read (chunk => $_[1], $_[2]);
590	}
591
592	sub unshift_read_chunk {
593	$_[0]->unshift_read (chunk => $_[1], $_[2]);
594	}
595
596	=item line => [$eol, ]$cb->($self, $line, $eol)	953	=item line => [$eol, ]$cb->($handle, $line, $eol)
597		954
598	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
599	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
600	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
601	the end of line marker as the third argument (C<$eol>).	958	the end of line marker as the third argument (C<$eol>).
…		…
615	=cut	972	=cut
616		973
617	register_read_type line => sub {	974	register_read_type line => sub {
618	my ($self, $cb, $eol) = @_;	975	my ($self, $cb, $eol) = @_;
619		976
620	$eol = qr|(\015?\012)| if @_ < 3;	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 {
621	$eol = quotemeta $eol unless ref $eol;	986	$eol = quotemeta $eol unless ref $eol;
622	$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};
623		1045
624	sub {	1046	sub {
625	$_[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	}
626		1058
627	$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
628	1	1100	1
629	}	1101	}
630	};	1102	};
631		1103
632	# compatibility with older API	1104	=item packstring => $format, $cb->($handle, $string)
633	sub push_read_line {
634	my $self = shift;
635	$self->push_read (line => @_);
636	}
637		1105
638	sub unshift_read_line {	1106	An octet string prefixed with an encoded length. The encoding C<$format>
639	my $self = shift;	1107	uses the same format as a Perl C<pack> format, but must specify a single
640	$self->unshift_read (line => @_);	1108	integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
641	}	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	};
642		1250
643	=back	1251	=back
644		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>)).
		1272
645	=item $handle->stop_read	1273	=item $handle->stop_read
646		1274
647	=item $handle->start_read	1275	=item $handle->start_read
648		1276
649	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
650	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
651	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
652	C<start_read>.	1280	C<start_read>.
653		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
654	=cut	1290	=cut
655		1291
656	sub stop_read {	1292	sub stop_read {
657	my ($self) = @_;	1293	my ($self) = @_;
658		1294
659	delete $self->{rw};	1295	delete $self->{_rw} unless $self->{tls};
660	}	1296	}
661		1297
662	sub start_read {	1298	sub start_read {
663	my ($self) = @_;	1299	my ($self) = @_;
664		1300
665	unless ($self->{rw} || $self->{eof}) {	1301	unless ($self->{_rw} || $self->{_eof}) {
666	Scalar::Util::weaken $self;	1302	Scalar::Util::weaken $self;
667		1303
668	$self->{rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {	1304	$self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {
669	my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf};	1305	my $rbuf = \($self->{tls} ? my $buf : $self->{rbuf});
670	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;
671		1307
672	if ($len > 0) {	1308	if ($len > 0) {
673	$self->{filter_r}	1309	$self->{_activity} = AnyEvent->now;
674	? $self->{filter_r}->($self, $rbuf)	1310
675	: $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	}
676		1318
677	} elsif (defined $len) {	1319	} elsif (defined $len) {
678	delete $self->{rw};	1320	delete $self->{_rw};
679	$self->{eof} = 1;	1321	$self->{_eof} = 1;
680	$self->_drain_rbuf;	1322	$self->_drain_rbuf unless $self->{_in_drain};
681		1323
682	} elsif ($! != EAGAIN && $! != EINTR) {	1324	} elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
683	return $self->error;	1325	return $self->_error ($!, 1);
684	}	1326	}
685	});	1327	});
686	}	1328	}
687	}	1329	}
688		1330
		1331	# poll the write BIO and send the data if applicable
689	sub _dotls {	1332	sub _dotls {
690	my ($self) = @_;	1333	my ($self) = @_;
691		1334
		1335	my $tmp;
		1336
692	if (length $self->{tls_wbuf}) {	1337	if (length $self->{_tls_wbuf}) {
693	while ((my $len = Net::SSLeay::write ($self->{tls}, $self->{tls_wbuf})) > 0) {	1338	while (($tmp = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) {
694	substr $self->{tls_wbuf}, 0, $len, "";	1339	substr $self->{_tls_wbuf}, 0, $tmp, "";
695	}	1340	}
696	}	1341	}
697		1342
698	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}))) {
699	$self->{wbuf} .= $buf;	1369	$self->{wbuf} .= $tmp;
700	$self->_drain_wbuf;	1370	$self->_drain_wbuf;
701	}
702
703	while (defined (my $buf = Net::SSLeay::read ($self->{tls}))) {
704	$self->{rbuf} .= $buf;
705	$self->_drain_rbuf;
706	}
707
708	my $err = Net::SSLeay::get_error ($self->{tls}, -1);
709
710	if ($err!= Net::SSLeay::ERROR_WANT_READ ()) {
711	if ($err == Net::SSLeay::ERROR_SYSCALL ()) {
712	$self->error;
713	} elsif ($err == Net::SSLeay::ERROR_SSL ()) {
714	$! = &Errno::EIO;
715	$self->error;
716	}
717
718	# all others are fine for our purposes
719	}	1371	}
720	}	1372	}
721		1373
722	=item $handle->starttls ($tls[, $tls_ctx])	1374	=item $handle->starttls ($tls[, $tls_ctx])
723		1375
…		…
729	C<"connect">, C<"accept"> or an existing Net::SSLeay object).	1381	C<"connect">, C<"accept"> or an existing Net::SSLeay object).
730		1382
731	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
732	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.
733		1385
734	=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.
735		1389
736	# 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
737	sub starttls {	1395	sub starttls {
738	my ($self, $ssl, $ctx) = @_;	1396	my ($self, $ssl, $ctx) = @_;
739		1397
740	$self->stoptls;	1398	require Net::SSLeay;
741		1399
		1400	Carp::croak "it is an error to call starttls more than once on an AnyEvent::Handle object"
		1401	if $self->{tls};
		1402
742	if ($ssl eq "accept") {	1403	if ($ssl eq "accept") {
743	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());	1404	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
744	Net::SSLeay::set_accept_state ($ssl);	1405	Net::SSLeay::set_accept_state ($ssl);
745	} elsif ($ssl eq "connect") {	1406	} elsif ($ssl eq "connect") {
746	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());	1407	$ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
…		…
752	# 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)
753	# but the openssl maintainers basically said: "trust us, it just works".	1414	# but the openssl maintainers basically said: "trust us, it just works".
754	# (unfortunately, we have to hardcode constants because the abysmally misdesigned	1415	# (unfortunately, we have to hardcode constants because the abysmally misdesigned
755	# and mismaintained ssleay-module doesn't even offer them).	1416	# and mismaintained ssleay-module doesn't even offer them).
756	# 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.
757	Net::SSLeay::CTX_set_mode ($self->{tls},	1425	Net::SSLeay::CTX_set_mode ($self->{tls},
758	(eval { Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)	1426	(eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)
759	| (eval { Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));	1427	| (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));
760		1428
761	$self->{tls_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());	1429	$self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
762	$self->{tls_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());	1430	$self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
763		1431
764	Net::SSLeay::set_bio ($ssl, $self->{tls_rbio}, $self->{tls_wbio});	1432	Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio});
765		1433
766	$self->{filter_w} = sub {	1434	&_dotls; # need to trigger the initial handshake
767	$_[0]{tls_wbuf} .= ${$_[1]};	1435	$self->start_read; # make sure we actually do read
768	&_dotls;
769	};
770	$self->{filter_r} = sub {
771	Net::SSLeay::BIO_write ($_[0]{tls_rbio}, ${$_[1]});
772	&_dotls;
773	};
774	}	1436	}
775		1437
776	=item $handle->stoptls	1438	=item $handle->stoptls
777		1439
778	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
779	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.
780		1444
781	=cut	1445	=cut
782		1446
783	sub stoptls {	1447	sub stoptls {
784	my ($self) = @_;	1448	my ($self) = @_;
785		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
786	Net::SSLeay::free (delete $self->{tls}) if $self->{tls};	1466	Net::SSLeay::free (delete $self->{tls});
787	delete $self->{tls_rbio};	1467
788	delete $self->{tls_wbio};	1468	delete @$self{qw(_rbio _wbio _tls_wbuf)};
789	delete $self->{tls_wbuf};
790	delete $self->{filter_r};
791	delete $self->{filter_w};
792	}	1469	}
793		1470
794	sub DESTROY {	1471	sub DESTROY {
795	my $self = shift;	1472	my ($self) = @_;
796		1473
797	$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 = ();
798	}	1522	}
799		1523
800	=item AnyEvent::Handle::TLS_CTX	1524	=item AnyEvent::Handle::TLS_CTX
801		1525
802	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
…		…
832	}	1556	}
833	}	1557	}
834		1558
835	=back	1559	=back
836		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
837	=head1 AUTHOR	1662	=head1 AUTHOR
838		1663
839	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>.
840		1665
841	=cut	1666	=cut

Diff Legend

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