ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/AnyEvent/lib/AnyEvent/Handle.pm
(Generate patch)

Comparing AnyEvent/lib/AnyEvent/Handle.pm (file contents):
Revision 1.12 by elmex, Thu May 15 09:03:43 2008 UTC vs.
Revision 1.69 by root, Sun Jun 15 21:44:56 2008 UTC

2 2
3no warnings; 3no warnings;
4use strict; 4use strict;
5 5
6use AnyEvent (); 6use AnyEvent ();
7use AnyEvent::Util (); 7use AnyEvent::Util qw(WSAEWOULDBLOCK);
8use Scalar::Util (); 8use Scalar::Util ();
9use Carp (); 9use Carp ();
10use Fcntl (); 10use Fcntl ();
11use Errno qw/EAGAIN EINTR/; 11use Errno qw(EAGAIN EINTR);
12 12
13=head1 NAME 13=head1 NAME
14 14
15AnyEvent::Handle - non-blocking I/O on filehandles via AnyEvent 15AnyEvent::Handle - non-blocking I/O on file handles via AnyEvent
16 16
17=cut 17=cut
18 18
19our $VERSION = '0.02'; 19our $VERSION = 4.151;
20 20
21=head1 SYNOPSIS 21=head1 SYNOPSIS
22 22
23 use AnyEvent; 23 use AnyEvent;
24 use AnyEvent::Handle; 24 use AnyEvent::Handle;
25 25
26 my $cv = AnyEvent->condvar; 26 my $cv = AnyEvent->condvar;
27 27
28 my $ae_fh = AnyEvent::Handle->new (fh => \*STDIN); 28 my $handle =
29
30 #TODO
31
32 # or use the constructor to pass the callback:
33
34 my $ae_fh2 =
35 AnyEvent::Handle->new ( 29 AnyEvent::Handle->new (
36 fh => \*STDIN, 30 fh => \*STDIN,
37 on_eof => sub { 31 on_eof => sub {
38 $cv->broadcast; 32 $cv->broadcast;
39 }, 33 },
40 #TODO
41 ); 34 );
42 35
43 $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;
44 47
45=head1 DESCRIPTION 48=head1 DESCRIPTION
46 49
47This module is a helper module to make it easier to do event-based I/O on 50This module is a helper module to make it easier to do event-based I/O on
48filehandles (and sockets, see L<AnyEvent::Socket> for an easy way to make 51filehandles. For utility functions for doing non-blocking connects and accepts
49non-blocking resolves and connects). 52on sockets see L<AnyEvent::Util>.
50 53
51In the following, when the documentation refers to of "bytes" then this 54In the following, when the documentation refers to of "bytes" then this
52means characters. As sysread and syswrite are used for all I/O, their 55means characters. As sysread and syswrite are used for all I/O, their
53treatment of characters applies to this module as well. 56treatment of characters applies to this module as well.
54 57
70The filehandle this L<AnyEvent::Handle> object will operate on. 73The filehandle this L<AnyEvent::Handle> object will operate on.
71 74
72NOTE: The filehandle will be set to non-blocking (using 75NOTE: The filehandle will be set to non-blocking (using
73AnyEvent::Util::fh_nonblocking). 76AnyEvent::Util::fh_nonblocking).
74 77
75=item on_eof => $cb->($self) [MANDATORY]
76
77Set the callback to be called on EOF.
78
79=item on_error => $cb->($self) 78=item on_eof => $cb->($handle)
80 79
80Set the callback to be called when an end-of-file condition is detcted,
81i.e. in the case of a socket, when the other side has closed the
82connection cleanly.
83
84While not mandatory, it is highly recommended to set an eof callback,
85otherwise you might end up with a closed socket while you are still
86waiting for data.
87
88=item on_error => $cb->($handle, $fatal)
89
81This is the fatal error callback, that is called when, well, a fatal error 90This is the error callback, which is called when, well, some error
82ocurs, such as not being able to resolve the hostname, failure to connect 91occured, such as not being able to resolve the hostname, failure to
83or a read error. 92connect or a read error.
84 93
85The object will not be in a usable state when this callback has been 94Some errors are fatal (which is indicated by C<$fatal> being true). On
86called. 95fatal errors the handle object will be shut down and will not be
96usable. Non-fatal errors can be retried by simply returning, but it is
97recommended to simply ignore this parameter and instead abondon the handle
98object when this callback is invoked.
87 99
88On callback entrance, the value of C<$!> contains the operating system 100On callback entrance, the value of C<$!> contains the operating system
89error (or C<ENOSPC> or C<EPIPE>). 101error (or C<ENOSPC>, C<EPIPE>, C<ETIMEDOUT> or C<EBADMSG>).
90 102
91While not mandatory, it is I<highly> recommended to set this callback, as 103While not mandatory, it is I<highly> recommended to set this callback, as
92you will not be notified of errors otherwise. The default simply calls 104you will not be notified of errors otherwise. The default simply calls
93die. 105C<croak>.
94 106
95=item on_read => $cb->($self) 107=item on_read => $cb->($handle)
96 108
97This sets the default read callback, which is called when data arrives 109This sets the default read callback, which is called when data arrives
98and no read request is in the queue. 110and no read request is in the queue (unlike read queue callbacks, this
111callback will only be called when at least one octet of data is in the
112read buffer).
99 113
100To access (and remove data from) the read buffer, use the C<< ->rbuf >> 114To access (and remove data from) the read buffer, use the C<< ->rbuf >>
101method or acces sthe C<$self->{rbuf}> member directly. 115method or access the C<$handle->{rbuf}> member directly.
102 116
103When an EOF condition is detected then AnyEvent::Handle will first try to 117When an EOF condition is detected then AnyEvent::Handle will first try to
104feed all the remaining data to the queued callbacks and C<on_read> before 118feed all the remaining data to the queued callbacks and C<on_read> before
105calling the C<on_eof> callback. If no progress can be made, then a fatal 119calling the C<on_eof> callback. If no progress can be made, then a fatal
106error will be raised (with C<$!> set to C<EPIPE>). 120error will be raised (with C<$!> set to C<EPIPE>).
107 121
108=item on_drain => $cb->() 122=item on_drain => $cb->($handle)
109 123
110This sets the callback that is called when the write buffer becomes empty 124This sets the callback that is called when the write buffer becomes empty
111(or when the callback is set and the buffer is empty already). 125(or when the callback is set and the buffer is empty already).
112 126
113To append to the write buffer, use the C<< ->push_write >> method. 127To append to the write buffer, use the C<< ->push_write >> method.
128
129This callback is useful when you don't want to put all of your write data
130into the queue at once, for example, when you want to write the contents
131of some file to the socket you might not want to read the whole file into
132memory and push it into the queue, but instead only read more data from
133the file when the write queue becomes empty.
134
135=item timeout => $fractional_seconds
136
137If non-zero, then this enables an "inactivity" timeout: whenever this many
138seconds pass without a successful read or write on the underlying file
139handle, the C<on_timeout> callback will be invoked (and if that one is
140missing, an C<ETIMEDOUT> error will be raised).
141
142Note that timeout processing is also active when you currently do not have
143any outstanding read or write requests: If you plan to keep the connection
144idle then you should disable the timout temporarily or ignore the timeout
145in the C<on_timeout> callback.
146
147Zero (the default) disables this timeout.
148
149=item on_timeout => $cb->($handle)
150
151Called whenever the inactivity timeout passes. If you return from this
152callback, then the timeout will be reset as if some activity had happened,
153so this condition is not fatal in any way.
114 154
115=item rbuf_max => <bytes> 155=item rbuf_max => <bytes>
116 156
117If defined, then a fatal error will be raised (with C<$!> set to C<ENOSPC>) 157If defined, then a fatal error will be raised (with C<$!> set to C<ENOSPC>)
118when the read buffer ever (strictly) exceeds this size. This is useful to 158when the read buffer ever (strictly) exceeds this size. This is useful to
125isn't finished). 165isn't finished).
126 166
127=item read_size => <bytes> 167=item read_size => <bytes>
128 168
129The default read block size (the amount of bytes this module will try to read 169The default read block size (the amount of bytes this module will try to read
130on each [loop iteration). Default: C<4096>. 170during each (loop iteration). Default: C<8192>.
131 171
132=item low_water_mark => <bytes> 172=item low_water_mark => <bytes>
133 173
134Sets the amount of bytes (default: C<0>) that make up an "empty" write 174Sets the amount of bytes (default: C<0>) that make up an "empty" write
135buffer: If the write reaches this size or gets even samller it is 175buffer: If the write reaches this size or gets even samller it is
136considered empty. 176considered empty.
137 177
178=item linger => <seconds>
179
180If non-zero (default: C<3600>), then the destructor of the
181AnyEvent::Handle object will check wether there is still outstanding write
182data and will install a watcher that will write out this data. No errors
183will be reported (this mostly matches how the operating system treats
184outstanding data at socket close time).
185
186This will not work for partial TLS data that could not yet been
187encoded. This data will be lost.
188
189=item tls => "accept" | "connect" | Net::SSLeay::SSL object
190
191When this parameter is given, it enables TLS (SSL) mode, that means it
192will start making tls handshake and will transparently encrypt/decrypt
193data.
194
195TLS mode requires Net::SSLeay to be installed (it will be loaded
196automatically when you try to create a TLS handle).
197
198For the TLS server side, use C<accept>, and for the TLS client side of a
199connection, use C<connect> mode.
200
201You can also provide your own TLS connection object, but you have
202to make sure that you call either C<Net::SSLeay::set_connect_state>
203or C<Net::SSLeay::set_accept_state> on it before you pass it to
204AnyEvent::Handle.
205
206See the C<starttls> method if you need to start TLs negotiation later.
207
208=item tls_ctx => $ssl_ctx
209
210Use the given Net::SSLeay::CTX object to create the new TLS connection
211(unless a connection object was specified directly). If this parameter is
212missing, then AnyEvent::Handle will use C<AnyEvent::Handle::TLS_CTX>.
213
214=item json => JSON or JSON::XS object
215
216This is the json coder object used by the C<json> read and write types.
217
218If you don't supply it, then AnyEvent::Handle will create and use a
219suitable one, which will write and expect UTF-8 encoded JSON texts.
220
221Note that you are responsible to depend on the JSON module if you want to
222use this functionality, as AnyEvent does not have a dependency itself.
223
224=item filter_r => $cb
225
226=item filter_w => $cb
227
228These exist, but are undocumented at this time.
229
138=back 230=back
139 231
140=cut 232=cut
141 233
142sub new { 234sub new {
146 238
147 $self->{fh} or Carp::croak "mandatory argument fh is missing"; 239 $self->{fh} or Carp::croak "mandatory argument fh is missing";
148 240
149 AnyEvent::Util::fh_nonblocking $self->{fh}, 1; 241 AnyEvent::Util::fh_nonblocking $self->{fh}, 1;
150 242
151 $self->on_eof ((delete $self->{on_eof} ) or Carp::croak "mandatory argument on_eof is missing"); 243 if ($self->{tls}) {
244 require Net::SSLeay;
245 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx});
246 }
152 247
153 $self->on_error (delete $self->{on_error}) if $self->{on_error}; 248 $self->{_activity} = AnyEvent->now;
249 $self->_timeout;
250
154 $self->on_drain (delete $self->{on_drain}) if $self->{on_drain}; 251 $self->on_drain (delete $self->{on_drain}) if $self->{on_drain};
155 $self->on_read (delete $self->{on_read} ) if $self->{on_read};
156 252
157 $self->start_read; 253 $self->start_read
254 if $self->{on_read};
158 255
159 $self 256 $self
160} 257}
161 258
162sub _shutdown { 259sub _shutdown {
163 my ($self) = @_; 260 my ($self) = @_;
164 261
262 delete $self->{_tw};
165 delete $self->{rw}; 263 delete $self->{_rw};
166 delete $self->{ww}; 264 delete $self->{_ww};
167 delete $self->{fh}; 265 delete $self->{fh};
168}
169 266
267 $self->stoptls;
268}
269
170sub error { 270sub _error {
171 my ($self) = @_; 271 my ($self, $errno, $fatal) = @_;
172 272
173 {
174 local $!;
175 $self->_shutdown; 273 $self->_shutdown
176 } 274 if $fatal;
275
276 $! = $errno;
177 277
178 if ($self->{on_error}) { 278 if ($self->{on_error}) {
179 $self->{on_error}($self); 279 $self->{on_error}($self, $fatal);
180 } else { 280 } else {
181 die "AnyEvent::Handle uncaught fatal error: $!"; 281 Carp::croak "AnyEvent::Handle uncaught error: $!";
182 } 282 }
183} 283}
184 284
185=item $fh = $handle->fh 285=item $fh = $handle->fh
186 286
187This method returns the filehandle of the L<AnyEvent::Handle> object. 287This method returns the file handle of the L<AnyEvent::Handle> object.
188 288
189=cut 289=cut
190 290
191sub fh { $_[0]->{fh} } 291sub fh { $_[0]{fh} }
192 292
193=item $handle->on_error ($cb) 293=item $handle->on_error ($cb)
194 294
195Replace the current C<on_error> callback (see the C<on_error> constructor argument). 295Replace the current C<on_error> callback (see the C<on_error> constructor argument).
196 296
208 308
209sub on_eof { 309sub on_eof {
210 $_[0]{on_eof} = $_[1]; 310 $_[0]{on_eof} = $_[1];
211} 311}
212 312
313=item $handle->on_timeout ($cb)
314
315Replace the current C<on_timeout> callback, or disables the callback
316(but not the timeout) if C<$cb> = C<undef>. See C<timeout> constructor
317argument.
318
319=cut
320
321sub on_timeout {
322 $_[0]{on_timeout} = $_[1];
323}
324
325#############################################################################
326
327=item $handle->timeout ($seconds)
328
329Configures (or disables) the inactivity timeout.
330
331=cut
332
333sub timeout {
334 my ($self, $timeout) = @_;
335
336 $self->{timeout} = $timeout;
337 $self->_timeout;
338}
339
340# reset the timeout watcher, as neccessary
341# also check for time-outs
342sub _timeout {
343 my ($self) = @_;
344
345 if ($self->{timeout}) {
346 my $NOW = AnyEvent->now;
347
348 # when would the timeout trigger?
349 my $after = $self->{_activity} + $self->{timeout} - $NOW;
350
351 # now or in the past already?
352 if ($after <= 0) {
353 $self->{_activity} = $NOW;
354
355 if ($self->{on_timeout}) {
356 $self->{on_timeout}($self);
357 } else {
358 $self->_error (&Errno::ETIMEDOUT);
359 }
360
361 # callback could have changed timeout value, optimise
362 return unless $self->{timeout};
363
364 # calculate new after
365 $after = $self->{timeout};
366 }
367
368 Scalar::Util::weaken $self;
369 return unless $self; # ->error could have destroyed $self
370
371 $self->{_tw} ||= AnyEvent->timer (after => $after, cb => sub {
372 delete $self->{_tw};
373 $self->_timeout;
374 });
375 } else {
376 delete $self->{_tw};
377 }
378}
379
213############################################################################# 380#############################################################################
214 381
215=back 382=back
216 383
217=head2 WRITE QUEUE 384=head2 WRITE QUEUE
220for reading. 387for reading.
221 388
222The write queue is very simple: you can add data to its end, and 389The write queue is very simple: you can add data to its end, and
223AnyEvent::Handle will automatically try to get rid of it for you. 390AnyEvent::Handle will automatically try to get rid of it for you.
224 391
225When data could be writtena nd the write buffer is shorter then the low 392When data could be written and the write buffer is shorter then the low
226water mark, the C<on_drain> callback will be invoked. 393water mark, the C<on_drain> callback will be invoked.
227 394
228=over 4 395=over 4
229 396
230=item $handle->on_drain ($cb) 397=item $handle->on_drain ($cb)
249want (only limited by the available memory), as C<AnyEvent::Handle> 416want (only limited by the available memory), as C<AnyEvent::Handle>
250buffers it independently of the kernel. 417buffers it independently of the kernel.
251 418
252=cut 419=cut
253 420
254sub push_write { 421sub _drain_wbuf {
255 my ($self, $data) = @_; 422 my ($self) = @_;
256 423
257 $self->{wbuf} .= $data; 424 if (!$self->{_ww} && length $self->{wbuf}) {
258 425
259 unless ($self->{ww}) {
260 Scalar::Util::weaken $self; 426 Scalar::Util::weaken $self;
427
261 my $cb = sub { 428 my $cb = sub {
262 my $len = syswrite $self->{fh}, $self->{wbuf}; 429 my $len = syswrite $self->{fh}, $self->{wbuf};
263 430
264 if ($len > 0) { 431 if ($len >= 0) {
265 substr $self->{wbuf}, 0, $len, ""; 432 substr $self->{wbuf}, 0, $len, "";
266 433
434 $self->{_activity} = AnyEvent->now;
267 435
268 $self->{on_drain}($self) 436 $self->{on_drain}($self)
269 if $self->{low_water_mark} >= length $self->{wbuf} 437 if $self->{low_water_mark} >= length $self->{wbuf}
270 && $self->{on_drain}; 438 && $self->{on_drain};
271 439
272 delete $self->{ww} unless length $self->{wbuf}; 440 delete $self->{_ww} unless length $self->{wbuf};
273 } elsif ($! != EAGAIN && $! != EINTR) { 441 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
274 $self->error; 442 $self->_error ($!, 1);
275 } 443 }
276 }; 444 };
277 445
446 # try to write data immediately
447 $cb->();
448
449 # if still data left in wbuf, we need to poll
278 $self->{ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb); 450 $self->{_ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb)
279 451 if length $self->{wbuf};
280 $cb->($self);
281 }; 452 };
282} 453}
454
455our %WH;
456
457sub register_write_type($$) {
458 $WH{$_[0]} = $_[1];
459}
460
461sub push_write {
462 my $self = shift;
463
464 if (@_ > 1) {
465 my $type = shift;
466
467 @_ = ($WH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_write")
468 ->($self, @_);
469 }
470
471 if ($self->{filter_w}) {
472 $self->{filter_w}($self, \$_[0]);
473 } else {
474 $self->{wbuf} .= $_[0];
475 $self->_drain_wbuf;
476 }
477}
478
479=item $handle->push_write (type => @args)
480
481Instead of formatting your data yourself, you can also let this module do
482the job by specifying a type and type-specific arguments.
483
484Predefined types are (if you have ideas for additional types, feel free to
485drop by and tell us):
486
487=over 4
488
489=item netstring => $string
490
491Formats the given value as netstring
492(http://cr.yp.to/proto/netstrings.txt, this is not a recommendation to use them).
493
494=cut
495
496register_write_type netstring => sub {
497 my ($self, $string) = @_;
498
499 sprintf "%d:%s,", (length $string), $string
500};
501
502=item packstring => $format, $data
503
504An octet string prefixed with an encoded length. The encoding C<$format>
505uses the same format as a Perl C<pack> format, but must specify a single
506integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
507optional C<!>, C<< < >> or C<< > >> modifier).
508
509=cut
510
511register_write_type packstring => sub {
512 my ($self, $format, $string) = @_;
513
514 pack "$format/a*", $string
515};
516
517=item json => $array_or_hashref
518
519Encodes the given hash or array reference into a JSON object. Unless you
520provide your own JSON object, this means it will be encoded to JSON text
521in UTF-8.
522
523JSON objects (and arrays) are self-delimiting, so you can write JSON at
524one end of a handle and read them at the other end without using any
525additional framing.
526
527The generated JSON text is guaranteed not to contain any newlines: While
528this module doesn't need delimiters after or between JSON texts to be
529able to read them, many other languages depend on that.
530
531A simple RPC protocol that interoperates easily with others is to send
532JSON arrays (or objects, although arrays are usually the better choice as
533they mimic how function argument passing works) and a newline after each
534JSON text:
535
536 $handle->push_write (json => ["method", "arg1", "arg2"]); # whatever
537 $handle->push_write ("\012");
538
539An AnyEvent::Handle receiver would simply use the C<json> read type and
540rely on the fact that the newline will be skipped as leading whitespace:
541
542 $handle->push_read (json => sub { my $array = $_[1]; ... });
543
544Other languages could read single lines terminated by a newline and pass
545this line into their JSON decoder of choice.
546
547=cut
548
549register_write_type json => sub {
550 my ($self, $ref) = @_;
551
552 require JSON;
553
554 $self->{json} ? $self->{json}->encode ($ref)
555 : JSON::encode_json ($ref)
556};
557
558=item storable => $reference
559
560Freezes the given reference using L<Storable> and writes it to the
561handle. Uses the C<nfreeze> format.
562
563=cut
564
565register_write_type storable => sub {
566 my ($self, $ref) = @_;
567
568 require Storable;
569
570 pack "w/a*", Storable::nfreeze ($ref)
571};
572
573=back
574
575=item AnyEvent::Handle::register_write_type type => $coderef->($handle, @args)
576
577This function (not method) lets you add your own types to C<push_write>.
578Whenever the given C<type> is used, C<push_write> will invoke the code
579reference with the handle object and the remaining arguments.
580
581The code reference is supposed to return a single octet string that will
582be appended to the write buffer.
583
584Note that this is a function, and all types registered this way will be
585global, so try to use unique names.
586
587=cut
283 588
284############################################################################# 589#############################################################################
285 590
286=back 591=back
287 592
294ways, the "simple" way, using only C<on_read> and the "complex" way, using 599ways, the "simple" way, using only C<on_read> and the "complex" way, using
295a queue. 600a queue.
296 601
297In the simple case, you just install an C<on_read> callback and whenever 602In the simple case, you just install an C<on_read> callback and whenever
298new data arrives, it will be called. You can then remove some data (if 603new data arrives, it will be called. You can then remove some data (if
299enough is there) from the read buffer (C<< $handle->rbuf >>) if you want 604enough is there) from the read buffer (C<< $handle->rbuf >>). Or you cna
300or not. 605leave the data there if you want to accumulate more (e.g. when only a
606partial message has been received so far).
301 607
302In the more complex case, you want to queue multiple callbacks. In this 608In the more complex case, you want to queue multiple callbacks. In this
303case, AnyEvent::Handle will call the first queued callback each time new 609case, AnyEvent::Handle will call the first queued callback each time new
304data arrives and removes it when it has done its job (see C<push_read>, 610data arrives (also the first time it is queued) and removes it when it has
305below). 611done its job (see C<push_read>, below).
306 612
307This way you can, for example, push three line-reads, followed by reading 613This way you can, for example, push three line-reads, followed by reading
308a chunk of data, and AnyEvent::Handle will execute them in order. 614a chunk of data, and AnyEvent::Handle will execute them in order.
309 615
310Example 1: EPP protocol parser. EPP sends 4 byte length info, followed by 616Example 1: EPP protocol parser. EPP sends 4 byte length info, followed by
311the specified number of bytes which give an XML datagram. 617the specified number of bytes which give an XML datagram.
312 618
313 # in the default state, expect some header bytes 619 # in the default state, expect some header bytes
314 $handle->on_read (sub { 620 $handle->on_read (sub {
315 # some data is here, now queue the length-header-read (4 octets) 621 # some data is here, now queue the length-header-read (4 octets)
316 shift->unshift_read_chunk (4, sub { 622 shift->unshift_read (chunk => 4, sub {
317 # header arrived, decode 623 # header arrived, decode
318 my $len = unpack "N", $_[1]; 624 my $len = unpack "N", $_[1];
319 625
320 # now read the payload 626 # now read the payload
321 shift->unshift_read_chunk ($len, sub { 627 shift->unshift_read (chunk => $len, sub {
322 my $xml = $_[1]; 628 my $xml = $_[1];
323 # handle xml 629 # handle xml
324 }); 630 });
325 }); 631 });
326 }); 632 });
327 633
328Example 2: Implement a client for a protocol that replies either with 634Example 2: Implement a client for a protocol that replies either with "OK"
329"OK" and another line or "ERROR" for one request, and 64 bytes for the 635and another line or "ERROR" for the first request that is sent, and 64
330second request. Due tot he availability of a full queue, we can just 636bytes for the second request. Due to the availability of a queue, we can
331pipeline sending both requests and manipulate the queue as necessary in 637just pipeline sending both requests and manipulate the queue as necessary
332the callbacks: 638in the callbacks.
333 639
334 # request one 640When the first callback is called and sees an "OK" response, it will
641C<unshift> another line-read. This line-read will be queued I<before> the
64264-byte chunk callback.
643
644 # request one, returns either "OK + extra line" or "ERROR"
335 $handle->push_write ("request 1\015\012"); 645 $handle->push_write ("request 1\015\012");
336 646
337 # we expect "ERROR" or "OK" as response, so push a line read 647 # we expect "ERROR" or "OK" as response, so push a line read
338 $handle->push_read_line (sub { 648 $handle->push_read (line => sub {
339 # if we got an "OK", we have to _prepend_ another line, 649 # if we got an "OK", we have to _prepend_ another line,
340 # so it will be read before the second request reads its 64 bytes 650 # so it will be read before the second request reads its 64 bytes
341 # which are already in the queue when this callback is called 651 # which are already in the queue when this callback is called
342 # we don't do this in case we got an error 652 # we don't do this in case we got an error
343 if ($_[1] eq "OK") { 653 if ($_[1] eq "OK") {
344 $_[0]->unshift_read_line (sub { 654 $_[0]->unshift_read (line => sub {
345 my $response = $_[1]; 655 my $response = $_[1];
346 ... 656 ...
347 }); 657 });
348 } 658 }
349 }); 659 });
350 660
351 # request two 661 # request two, simply returns 64 octets
352 $handle->push_write ("request 2\015\012"); 662 $handle->push_write ("request 2\015\012");
353 663
354 # simply read 64 bytes, always 664 # simply read 64 bytes, always
355 $handle->push_read_chunk (64, sub { 665 $handle->push_read (chunk => 64, sub {
356 my $response = $_[1]; 666 my $response = $_[1];
357 ... 667 ...
358 }); 668 });
359 669
360=over 4 670=over 4
362=cut 672=cut
363 673
364sub _drain_rbuf { 674sub _drain_rbuf {
365 my ($self) = @_; 675 my ($self) = @_;
366 676
367 return if $self->{in_drain};
368 local $self->{in_drain} = 1; 677 local $self->{_in_drain} = 1;
369 678
370 while (my $len = length $self->{rbuf}) { 679 if (
680 defined $self->{rbuf_max}
681 && $self->{rbuf_max} < length $self->{rbuf}
682 ) {
683 return $self->_error (&Errno::ENOSPC, 1);
684 }
685
686 while () {
371 no strict 'refs'; 687 no strict 'refs';
688
689 my $len = length $self->{rbuf};
690
372 if (my $cb = shift @{ $self->{queue} }) { 691 if (my $cb = shift @{ $self->{_queue} }) {
373 if (!$cb->($self)) { 692 unless ($cb->($self)) {
374 if ($self->{eof}) { 693 if ($self->{_eof}) {
375 # no progress can be made (not enough data and no data forthcoming) 694 # no progress can be made (not enough data and no data forthcoming)
376 $! = &Errno::EPIPE; return $self->error; 695 $self->_error (&Errno::EPIPE, 1), last;
377 } 696 }
378 697
379 unshift @{ $self->{queue} }, $cb; 698 unshift @{ $self->{_queue} }, $cb;
380 return; 699 last;
381 } 700 }
382 } elsif ($self->{on_read}) { 701 } elsif ($self->{on_read}) {
702 last unless $len;
703
383 $self->{on_read}($self); 704 $self->{on_read}($self);
384 705
385 if ( 706 if (
386 $self->{eof} # if no further data will arrive
387 && $len == length $self->{rbuf} # and no data has been consumed 707 $len == length $self->{rbuf} # if no data has been consumed
388 && !@{ $self->{queue} } # and the queue is still empty 708 && !@{ $self->{_queue} } # and the queue is still empty
389 && $self->{on_read} # and we still want to read data 709 && $self->{on_read} # but we still have on_read
390 ) { 710 ) {
711 # no further data will arrive
391 # then no progress can be made 712 # so no progress can be made
392 $! = &Errno::EPIPE; return $self->error; 713 $self->_error (&Errno::EPIPE, 1), last
714 if $self->{_eof};
715
716 last; # more data might arrive
393 } 717 }
394 } else { 718 } else {
395 # read side becomes idle 719 # read side becomes idle
396 delete $self->{rw}; 720 delete $self->{_rw};
397 return; 721 last;
398 } 722 }
399 } 723 }
400 724
401 if ($self->{eof}) {
402 $self->_shutdown;
403 $self->{on_eof}($self); 725 $self->{on_eof}($self)
726 if $self->{_eof} && $self->{on_eof};
727
728 # may need to restart read watcher
729 unless ($self->{_rw}) {
730 $self->start_read
731 if $self->{on_read} || @{ $self->{_queue} };
404 } 732 }
405} 733}
406 734
407=item $handle->on_read ($cb) 735=item $handle->on_read ($cb)
408 736
414 742
415sub on_read { 743sub on_read {
416 my ($self, $cb) = @_; 744 my ($self, $cb) = @_;
417 745
418 $self->{on_read} = $cb; 746 $self->{on_read} = $cb;
747 $self->_drain_rbuf if $cb && !$self->{_in_drain};
419} 748}
420 749
421=item $handle->rbuf 750=item $handle->rbuf
422 751
423Returns the read buffer (as a modifiable lvalue). 752Returns the read buffer (as a modifiable lvalue).
442Append the given callback to the end of the queue (C<push_read>) or 771Append the given callback to the end of the queue (C<push_read>) or
443prepend it (C<unshift_read>). 772prepend it (C<unshift_read>).
444 773
445The callback is called each time some additional read data arrives. 774The callback is called each time some additional read data arrives.
446 775
447It must check wether enough data is in the read buffer already. 776It must check whether enough data is in the read buffer already.
448 777
449If not enough data is available, it must return the empty list or a false 778If not enough data is available, it must return the empty list or a false
450value, in which case it will be called repeatedly until enough data is 779value, in which case it will be called repeatedly until enough data is
451available (or an error condition is detected). 780available (or an error condition is detected).
452 781
454interested in (which can be none at all) and return a true value. After returning 783interested in (which can be none at all) and return a true value. After returning
455true, it will be removed from the queue. 784true, it will be removed from the queue.
456 785
457=cut 786=cut
458 787
788our %RH;
789
790sub register_read_type($$) {
791 $RH{$_[0]} = $_[1];
792}
793
459sub push_read { 794sub push_read {
460 my ($self, $cb) = @_; 795 my $self = shift;
796 my $cb = pop;
461 797
798 if (@_) {
799 my $type = shift;
800
801 $cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_read")
802 ->($self, $cb, @_);
803 }
804
462 push @{ $self->{queue} }, $cb; 805 push @{ $self->{_queue} }, $cb;
463 $self->_drain_rbuf; 806 $self->_drain_rbuf unless $self->{_in_drain};
464} 807}
465 808
466sub unshift_read { 809sub unshift_read {
467 my ($self, $cb) = @_; 810 my $self = shift;
811 my $cb = pop;
468 812
813 if (@_) {
814 my $type = shift;
815
816 $cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::unshift_read")
817 ->($self, $cb, @_);
818 }
819
820
469 push @{ $self->{queue} }, $cb; 821 unshift @{ $self->{_queue} }, $cb;
470 $self->_drain_rbuf; 822 $self->_drain_rbuf unless $self->{_in_drain};
471} 823}
472 824
473=item $handle->push_read_chunk ($len, $cb->($self, $data)) 825=item $handle->push_read (type => @args, $cb)
474 826
475=item $handle->unshift_read_chunk ($len, $cb->($self, $data)) 827=item $handle->unshift_read (type => @args, $cb)
476 828
477Append the given callback to the end of the queue (C<push_read_chunk>) or 829Instead of providing a callback that parses the data itself you can chose
478prepend it (C<unshift_read_chunk>). 830between a number of predefined parsing formats, for chunks of data, lines
831etc.
479 832
480The callback will be called only once C<$len> bytes have been read, and 833Predefined types are (if you have ideas for additional types, feel free to
481these C<$len> bytes will be passed to the callback. 834drop by and tell us):
482 835
483=cut 836=over 4
484 837
485sub _read_chunk($$) { 838=item chunk => $octets, $cb->($handle, $data)
839
840Invoke the callback only once C<$octets> bytes have been read. Pass the
841data read to the callback. The callback will never be called with less
842data.
843
844Example: read 2 bytes.
845
846 $handle->push_read (chunk => 2, sub {
847 warn "yay ", unpack "H*", $_[1];
848 });
849
850=cut
851
852register_read_type chunk => sub {
486 my ($self, $len, $cb) = @_; 853 my ($self, $cb, $len) = @_;
487 854
488 sub { 855 sub {
489 $len <= length $_[0]{rbuf} or return; 856 $len <= length $_[0]{rbuf} or return;
490 $cb->($_[0], substr $_[0]{rbuf}, 0, $len, ""); 857 $cb->($_[0], substr $_[0]{rbuf}, 0, $len, "");
491 1 858 1
492 } 859 }
493} 860};
494 861
862# compatibility with older API
495sub push_read_chunk { 863sub push_read_chunk {
496 $_[0]->push_read (&_read_chunk); 864 $_[0]->push_read (chunk => $_[1], $_[2]);
497} 865}
498
499 866
500sub unshift_read_chunk { 867sub unshift_read_chunk {
501 $_[0]->unshift_read (&_read_chunk); 868 $_[0]->unshift_read (chunk => $_[1], $_[2]);
502} 869}
503 870
504=item $handle->push_read_line ([$eol, ]$cb->($self, $line, $eol)) 871=item line => [$eol, ]$cb->($handle, $line, $eol)
505
506=item $handle->unshift_read_line ([$eol, ]$cb->($self, $line, $eol))
507
508Append the given callback to the end of the queue (C<push_read_line>) or
509prepend it (C<unshift_read_line>).
510 872
511The callback will be called only once a full line (including the end of 873The callback will be called only once a full line (including the end of
512line marker, C<$eol>) has been read. This line (excluding the end of line 874line marker, C<$eol>) has been read. This line (excluding the end of line
513marker) will be passed to the callback as second argument (C<$line>), and 875marker) will be passed to the callback as second argument (C<$line>), and
514the end of line marker as the third argument (C<$eol>). 876the end of line marker as the third argument (C<$eol>).
525Partial lines at the end of the stream will never be returned, as they are 887Partial lines at the end of the stream will never be returned, as they are
526not marked by the end of line marker. 888not marked by the end of line marker.
527 889
528=cut 890=cut
529 891
530sub _read_line($$) { 892register_read_type line => sub {
531 my $self = shift; 893 my ($self, $cb, $eol) = @_;
532 my $cb = pop;
533 my $eol = @_ ? shift : qr|(\015?\012)|;
534 my $pos;
535 894
895 $eol = qr|(\015?\012)| if @_ < 3;
536 $eol = qr|(\Q$eol\E)| unless ref $eol; 896 $eol = quotemeta $eol unless ref $eol;
537 $eol = qr|^(.*?)($eol)|; 897 $eol = qr|^(.*?)($eol)|s;
538 898
539 sub { 899 sub {
540 $_[0]{rbuf} =~ s/$eol// or return; 900 $_[0]{rbuf} =~ s/$eol// or return;
541 901
542 $cb->($_[0], $1, $2); 902 $cb->($_[0], $1, $2);
543 1 903 1
544 } 904 }
545} 905};
546 906
907# compatibility with older API
547sub push_read_line { 908sub push_read_line {
548 $_[0]->push_read (&_read_line); 909 my $self = shift;
910 $self->push_read (line => @_);
549} 911}
550 912
551sub unshift_read_line { 913sub unshift_read_line {
552 $_[0]->unshift_read (&_read_line); 914 my $self = shift;
915 $self->unshift_read (line => @_);
553} 916}
917
918=item regex => $accept[, $reject[, $skip], $cb->($handle, $data)
919
920Makes a regex match against the regex object C<$accept> and returns
921everything up to and including the match.
922
923Example: read a single line terminated by '\n'.
924
925 $handle->push_read (regex => qr<\n>, sub { ... });
926
927If C<$reject> is given and not undef, then it determines when the data is
928to be rejected: it is matched against the data when the C<$accept> regex
929does not match and generates an C<EBADMSG> error when it matches. This is
930useful to quickly reject wrong data (to avoid waiting for a timeout or a
931receive buffer overflow).
932
933Example: expect a single decimal number followed by whitespace, reject
934anything else (not the use of an anchor).
935
936 $handle->push_read (regex => qr<^[0-9]+\s>, qr<[^0-9]>, sub { ... });
937
938If C<$skip> is given and not C<undef>, then it will be matched against
939the receive buffer when neither C<$accept> nor C<$reject> match,
940and everything preceding and including the match will be accepted
941unconditionally. This is useful to skip large amounts of data that you
942know cannot be matched, so that the C<$accept> or C<$reject> regex do not
943have to start matching from the beginning. This is purely an optimisation
944and is usually worth only when you expect more than a few kilobytes.
945
946Example: expect a http header, which ends at C<\015\012\015\012>. Since we
947expect the header to be very large (it isn't in practise, but...), we use
948a skip regex to skip initial portions. The skip regex is tricky in that
949it only accepts something not ending in either \015 or \012, as these are
950required for the accept regex.
951
952 $handle->push_read (regex =>
953 qr<\015\012\015\012>,
954 undef, # no reject
955 qr<^.*[^\015\012]>,
956 sub { ... });
957
958=cut
959
960register_read_type regex => sub {
961 my ($self, $cb, $accept, $reject, $skip) = @_;
962
963 my $data;
964 my $rbuf = \$self->{rbuf};
965
966 sub {
967 # accept
968 if ($$rbuf =~ $accept) {
969 $data .= substr $$rbuf, 0, $+[0], "";
970 $cb->($self, $data);
971 return 1;
972 }
973
974 # reject
975 if ($reject && $$rbuf =~ $reject) {
976 $self->_error (&Errno::EBADMSG);
977 }
978
979 # skip
980 if ($skip && $$rbuf =~ $skip) {
981 $data .= substr $$rbuf, 0, $+[0], "";
982 }
983
984 ()
985 }
986};
987
988=item netstring => $cb->($handle, $string)
989
990A netstring (http://cr.yp.to/proto/netstrings.txt, this is not an endorsement).
991
992Throws an error with C<$!> set to EBADMSG on format violations.
993
994=cut
995
996register_read_type netstring => sub {
997 my ($self, $cb) = @_;
998
999 sub {
1000 unless ($_[0]{rbuf} =~ s/^(0|[1-9][0-9]*)://) {
1001 if ($_[0]{rbuf} =~ /[^0-9]/) {
1002 $self->_error (&Errno::EBADMSG);
1003 }
1004 return;
1005 }
1006
1007 my $len = $1;
1008
1009 $self->unshift_read (chunk => $len, sub {
1010 my $string = $_[1];
1011 $_[0]->unshift_read (chunk => 1, sub {
1012 if ($_[1] eq ",") {
1013 $cb->($_[0], $string);
1014 } else {
1015 $self->_error (&Errno::EBADMSG);
1016 }
1017 });
1018 });
1019
1020 1
1021 }
1022};
1023
1024=item packstring => $format, $cb->($handle, $string)
1025
1026An octet string prefixed with an encoded length. The encoding C<$format>
1027uses the same format as a Perl C<pack> format, but must specify a single
1028integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
1029optional C<!>, C<< < >> or C<< > >> modifier).
1030
1031DNS over TCP uses a prefix of C<n>, EPP uses a prefix of C<N>.
1032
1033Example: read a block of data prefixed by its length in BER-encoded
1034format (very efficient).
1035
1036 $handle->push_read (packstring => "w", sub {
1037 my ($handle, $data) = @_;
1038 });
1039
1040=cut
1041
1042register_read_type packstring => sub {
1043 my ($self, $cb, $format) = @_;
1044
1045 sub {
1046 # when we can use 5.10 we can use ".", but for 5.8 we use the re-pack method
1047 defined (my $len = eval { unpack $format, $_[0]->{rbuf} })
1048 or return;
1049
1050 # remove prefix
1051 substr $_[0]->{rbuf}, 0, (length pack $format, $len), "";
1052
1053 # read rest
1054 $_[0]->unshift_read (chunk => $len, $cb);
1055
1056 1
1057 }
1058};
1059
1060=item json => $cb->($handle, $hash_or_arrayref)
1061
1062Reads a JSON object or array, decodes it and passes it to the callback.
1063
1064If a C<json> object was passed to the constructor, then that will be used
1065for the final decode, otherwise it will create a JSON coder expecting UTF-8.
1066
1067This read type uses the incremental parser available with JSON version
10682.09 (and JSON::XS version 2.2) and above. You have to provide a
1069dependency on your own: this module will load the JSON module, but
1070AnyEvent does not depend on it itself.
1071
1072Since JSON texts are fully self-delimiting, the C<json> read and write
1073types are an ideal simple RPC protocol: just exchange JSON datagrams. See
1074the C<json> write type description, above, for an actual example.
1075
1076=cut
1077
1078register_read_type json => sub {
1079 my ($self, $cb) = @_;
1080
1081 require JSON;
1082
1083 my $data;
1084 my $rbuf = \$self->{rbuf};
1085
1086 my $json = $self->{json} ||= JSON->new->utf8;
1087
1088 sub {
1089 my $ref = $json->incr_parse ($self->{rbuf});
1090
1091 if ($ref) {
1092 $self->{rbuf} = $json->incr_text;
1093 $json->incr_text = "";
1094 $cb->($self, $ref);
1095
1096 1
1097 } else {
1098 $self->{rbuf} = "";
1099 ()
1100 }
1101 }
1102};
1103
1104=item storable => $cb->($handle, $ref)
1105
1106Deserialises a L<Storable> frozen representation as written by the
1107C<storable> write type (BER-encoded length prefix followed by nfreeze'd
1108data).
1109
1110Raises C<EBADMSG> error if the data could not be decoded.
1111
1112=cut
1113
1114register_read_type storable => sub {
1115 my ($self, $cb) = @_;
1116
1117 require Storable;
1118
1119 sub {
1120 # when we can use 5.10 we can use ".", but for 5.8 we use the re-pack method
1121 defined (my $len = eval { unpack "w", $_[0]->{rbuf} })
1122 or return;
1123
1124 # remove prefix
1125 substr $_[0]->{rbuf}, 0, (length pack "w", $len), "";
1126
1127 # read rest
1128 $_[0]->unshift_read (chunk => $len, sub {
1129 if (my $ref = eval { Storable::thaw ($_[1]) }) {
1130 $cb->($_[0], $ref);
1131 } else {
1132 $self->_error (&Errno::EBADMSG);
1133 }
1134 });
1135 }
1136};
1137
1138=back
1139
1140=item AnyEvent::Handle::register_read_type type => $coderef->($handle, $cb, @args)
1141
1142This function (not method) lets you add your own types to C<push_read>.
1143
1144Whenever the given C<type> is used, C<push_read> will invoke the code
1145reference with the handle object, the callback and the remaining
1146arguments.
1147
1148The code reference is supposed to return a callback (usually a closure)
1149that works as a plain read callback (see C<< ->push_read ($cb) >>).
1150
1151It should invoke the passed callback when it is done reading (remember to
1152pass C<$handle> as first argument as all other callbacks do that).
1153
1154Note that this is a function, and all types registered this way will be
1155global, so try to use unique names.
1156
1157For examples, see the source of this module (F<perldoc -m AnyEvent::Handle>,
1158search for C<register_read_type>)).
554 1159
555=item $handle->stop_read 1160=item $handle->stop_read
556 1161
557=item $handle->start_read 1162=item $handle->start_read
558 1163
559In rare cases you actually do not want to read anything form the 1164In rare cases you actually do not want to read anything from the
560socket. In this case you can call C<stop_read>. Neither C<on_read> no 1165socket. In this case you can call C<stop_read>. Neither C<on_read> nor
561any queued callbacks will be executed then. To start readign again, call 1166any queued callbacks will be executed then. To start reading again, call
562C<start_read>. 1167C<start_read>.
1168
1169Note that AnyEvent::Handle will automatically C<start_read> for you when
1170you change the C<on_read> callback or push/unshift a read callback, and it
1171will automatically C<stop_read> for you when neither C<on_read> is set nor
1172there are any read requests in the queue.
563 1173
564=cut 1174=cut
565 1175
566sub stop_read { 1176sub stop_read {
567 my ($self) = @_; 1177 my ($self) = @_;
568 1178
569 delete $self->{rw}; 1179 delete $self->{_rw};
570} 1180}
571 1181
572sub start_read { 1182sub start_read {
573 my ($self) = @_; 1183 my ($self) = @_;
574 1184
575 unless ($self->{rw} || $self->{eof}) { 1185 unless ($self->{_rw} || $self->{_eof}) {
576 Scalar::Util::weaken $self; 1186 Scalar::Util::weaken $self;
577 1187
578 $self->{rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub { 1188 $self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {
1189 my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf};
579 my $len = sysread $self->{fh}, $self->{rbuf}, $self->{read_size} || 8192, length $self->{rbuf}; 1190 my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf;
580 1191
581 if ($len > 0) { 1192 if ($len > 0) {
582 if (defined $self->{rbuf_max}) { 1193 $self->{_activity} = AnyEvent->now;
583 if ($self->{rbuf_max} < length $self->{rbuf}) { 1194
584 $! = &Errno::ENOSPC; return $self->error; 1195 $self->{filter_r}
585 } 1196 ? $self->{filter_r}($self, $rbuf)
586 } 1197 : $self->{_in_drain} || $self->_drain_rbuf;
587 1198
588 } elsif (defined $len) { 1199 } elsif (defined $len) {
589 $self->{eof} = 1;
590 delete $self->{rw}; 1200 delete $self->{_rw};
1201 $self->{_eof} = 1;
1202 $self->_drain_rbuf unless $self->{_in_drain};
591 1203
592 } elsif ($! != EAGAIN && $! != EINTR) { 1204 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
593 return $self->error; 1205 return $self->_error ($!, 1);
594 } 1206 }
595
596 $self->_drain_rbuf;
597 }); 1207 });
598 } 1208 }
599} 1209}
600 1210
1211sub _dotls {
1212 my ($self) = @_;
1213
1214 my $buf;
1215
1216 if (length $self->{_tls_wbuf}) {
1217 while ((my $len = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) {
1218 substr $self->{_tls_wbuf}, 0, $len, "";
1219 }
1220 }
1221
1222 if (length ($buf = Net::SSLeay::BIO_read ($self->{_wbio}))) {
1223 $self->{wbuf} .= $buf;
1224 $self->_drain_wbuf;
1225 }
1226
1227 while (defined ($buf = Net::SSLeay::read ($self->{tls}))) {
1228 if (length $buf) {
1229 $self->{rbuf} .= $buf;
1230 $self->_drain_rbuf unless $self->{_in_drain};
1231 } else {
1232 # let's treat SSL-eof as we treat normal EOF
1233 $self->{_eof} = 1;
1234 $self->_shutdown;
1235 return;
1236 }
1237 }
1238
1239 my $err = Net::SSLeay::get_error ($self->{tls}, -1);
1240
1241 if ($err!= Net::SSLeay::ERROR_WANT_READ ()) {
1242 if ($err == Net::SSLeay::ERROR_SYSCALL ()) {
1243 return $self->_error ($!, 1);
1244 } elsif ($err == Net::SSLeay::ERROR_SSL ()) {
1245 return $self->_error (&Errno::EIO, 1);
1246 }
1247
1248 # all others are fine for our purposes
1249 }
1250}
1251
1252=item $handle->starttls ($tls[, $tls_ctx])
1253
1254Instead of starting TLS negotiation immediately when the AnyEvent::Handle
1255object is created, you can also do that at a later time by calling
1256C<starttls>.
1257
1258The first argument is the same as the C<tls> constructor argument (either
1259C<"connect">, C<"accept"> or an existing Net::SSLeay object).
1260
1261The second argument is the optional C<Net::SSLeay::CTX> object that is
1262used when AnyEvent::Handle has to create its own TLS connection object.
1263
1264The TLS connection object will end up in C<< $handle->{tls} >> after this
1265call and can be used or changed to your liking. Note that the handshake
1266might have already started when this function returns.
1267
1268=cut
1269
1270sub starttls {
1271 my ($self, $ssl, $ctx) = @_;
1272
1273 $self->stoptls;
1274
1275 if ($ssl eq "accept") {
1276 $ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
1277 Net::SSLeay::set_accept_state ($ssl);
1278 } elsif ($ssl eq "connect") {
1279 $ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
1280 Net::SSLeay::set_connect_state ($ssl);
1281 }
1282
1283 $self->{tls} = $ssl;
1284
1285 # basically, this is deep magic (because SSL_read should have the same issues)
1286 # but the openssl maintainers basically said: "trust us, it just works".
1287 # (unfortunately, we have to hardcode constants because the abysmally misdesigned
1288 # and mismaintained ssleay-module doesn't even offer them).
1289 # http://www.mail-archive.com/openssl-dev@openssl.org/msg22420.html
1290 Net::SSLeay::CTX_set_mode ($self->{tls},
1291 (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)
1292 | (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));
1293
1294 $self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
1295 $self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
1296
1297 Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio});
1298
1299 $self->{filter_w} = sub {
1300 $_[0]{_tls_wbuf} .= ${$_[1]};
1301 &_dotls;
1302 };
1303 $self->{filter_r} = sub {
1304 Net::SSLeay::BIO_write ($_[0]{_rbio}, ${$_[1]});
1305 &_dotls;
1306 };
1307}
1308
1309=item $handle->stoptls
1310
1311Destroys the SSL connection, if any. Partial read or write data will be
1312lost.
1313
1314=cut
1315
1316sub stoptls {
1317 my ($self) = @_;
1318
1319 Net::SSLeay::free (delete $self->{tls}) if $self->{tls};
1320
1321 delete $self->{_rbio};
1322 delete $self->{_wbio};
1323 delete $self->{_tls_wbuf};
1324 delete $self->{filter_r};
1325 delete $self->{filter_w};
1326}
1327
1328sub DESTROY {
1329 my $self = shift;
1330
1331 $self->stoptls;
1332
1333 my $linger = exists $self->{linger} ? $self->{linger} : 3600;
1334
1335 if ($linger && length $self->{wbuf}) {
1336 my $fh = delete $self->{fh};
1337 my $wbuf = delete $self->{wbuf};
1338
1339 my @linger;
1340
1341 push @linger, AnyEvent->io (fh => $fh, poll => "w", cb => sub {
1342 my $len = syswrite $fh, $wbuf, length $wbuf;
1343
1344 if ($len > 0) {
1345 substr $wbuf, 0, $len, "";
1346 } else {
1347 @linger = (); # end
1348 }
1349 });
1350 push @linger, AnyEvent->timer (after => $linger, cb => sub {
1351 @linger = ();
1352 });
1353 }
1354}
1355
1356=item AnyEvent::Handle::TLS_CTX
1357
1358This function creates and returns the Net::SSLeay::CTX object used by
1359default for TLS mode.
1360
1361The context is created like this:
1362
1363 Net::SSLeay::load_error_strings;
1364 Net::SSLeay::SSLeay_add_ssl_algorithms;
1365 Net::SSLeay::randomize;
1366
1367 my $CTX = Net::SSLeay::CTX_new;
1368
1369 Net::SSLeay::CTX_set_options $CTX, Net::SSLeay::OP_ALL
1370
1371=cut
1372
1373our $TLS_CTX;
1374
1375sub TLS_CTX() {
1376 $TLS_CTX || do {
1377 require Net::SSLeay;
1378
1379 Net::SSLeay::load_error_strings ();
1380 Net::SSLeay::SSLeay_add_ssl_algorithms ();
1381 Net::SSLeay::randomize ();
1382
1383 $TLS_CTX = Net::SSLeay::CTX_new ();
1384
1385 Net::SSLeay::CTX_set_options ($TLS_CTX, Net::SSLeay::OP_ALL ());
1386
1387 $TLS_CTX
1388 }
1389}
1390
601=back 1391=back
602 1392
1393=head1 SUBCLASSING AnyEvent::Handle
1394
1395In many cases, you might want to subclass AnyEvent::Handle.
1396
1397To make this easier, a given version of AnyEvent::Handle uses these
1398conventions:
1399
1400=over 4
1401
1402=item * all constructor arguments become object members.
1403
1404At least initially, when you pass a C<tls>-argument to the constructor it
1405will end up in C<< $handle->{tls} >>. Those members might be changes or
1406mutated later on (for example C<tls> will hold the TLS connection object).
1407
1408=item * other object member names are prefixed with an C<_>.
1409
1410All object members not explicitly documented (internal use) are prefixed
1411with an underscore character, so the remaining non-C<_>-namespace is free
1412for use for subclasses.
1413
1414=item * all members not documented here and not prefixed with an underscore
1415are free to use in subclasses.
1416
1417Of course, new versions of AnyEvent::Handle may introduce more "public"
1418member variables, but thats just life, at least it is documented.
1419
1420=back
1421
603=head1 AUTHOR 1422=head1 AUTHOR
604 1423
605Robin Redeker C<< <elmex at ta-sa.org> >>, Marc Lehmann <schmorp@schmorp.de>. 1424Robin Redeker C<< <elmex at ta-sa.org> >>, Marc Lehmann <schmorp@schmorp.de>.
606 1425
607=cut 1426=cut

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines