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

Comparing AnyEvent-HTTP/HTTP.pm (file contents):
Revision 1.32 by root, Fri Oct 24 01:27:29 2008 UTC vs.
Revision 1.54 by root, Wed Jun 16 18:09:52 2010 UTC

39package AnyEvent::HTTP; 39package AnyEvent::HTTP;
40 40
41use strict; 41use strict;
42no warnings; 42no warnings;
43 43
44use Carp; 44use Errno ();
45 45
46use AnyEvent (); 46use AnyEvent 5.0 ();
47use AnyEvent::Util (); 47use AnyEvent::Util ();
48use AnyEvent::Socket (); 48use AnyEvent::Socket ();
49use AnyEvent::Handle (); 49use AnyEvent::Handle ();
50 50
51use base Exporter::; 51use base Exporter::;
52 52
53our $VERSION = '1.05'; 53our $VERSION = '1.44';
54 54
55our @EXPORT = qw(http_get http_post http_head http_request); 55our @EXPORT = qw(http_get http_post http_head http_request);
56 56
57our $USERAGENT = "Mozilla/5.0 (compatible; AnyEvent::HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)"; 57our $USERAGENT = "Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)";
58our $MAX_RECURSE = 10; 58our $MAX_RECURSE = 10;
59our $MAX_PERSISTENT = 8; 59our $MAX_PERSISTENT = 8;
60our $PERSISTENT_TIMEOUT = 2; 60our $PERSISTENT_TIMEOUT = 2;
61our $TIMEOUT = 300; 61our $TIMEOUT = 300;
62 62
63# changing these is evil 63# changing these is evil
64our $MAX_PERSISTENT_PER_HOST = 2; 64our $MAX_PERSISTENT_PER_HOST = 0;
65our $MAX_PER_HOST = 4; 65our $MAX_PER_HOST = 4;
66 66
67our $PROXY; 67our $PROXY;
68our $ACTIVE = 0; 68our $ACTIVE = 0;
69 69
94When called in void context, nothing is returned. In other contexts, 94When called in void context, nothing is returned. In other contexts,
95C<http_request> returns a "cancellation guard" - you have to keep the 95C<http_request> returns a "cancellation guard" - you have to keep the
96object at least alive until the callback get called. If the object gets 96object at least alive until the callback get called. If the object gets
97destroyed before the callbakc is called, the request will be cancelled. 97destroyed before the callbakc is called, the request will be cancelled.
98 98
99The callback will be called with the response data as first argument 99The callback will be called with the response body data as first argument
100(or C<undef> if it wasn't available due to errors), and a hash-ref with 100(or C<undef> if an error occured), and a hash-ref with response headers as
101response headers as second argument. 101second argument.
102 102
103All the headers in that hash are lowercased. In addition to the response 103All the headers in that hash are lowercased. In addition to the response
104headers, the "pseudo-headers" C<HTTPVersion>, C<Status> and C<Reason> 104headers, the "pseudo-headers" C<HTTPVersion>, C<Status> and C<Reason>
105contain the three parts of the HTTP Status-Line of the same name. The 105contain the three parts of the HTTP Status-Line of the same name. The
106pseudo-header C<URL> contains the original URL (which can differ from the 106pseudo-header C<URL> contains the original URL (which can differ from the
108 108
109If the server sends a header multiple times, then their contents will be 109If the server sends a header multiple times, then their contents will be
110joined together with a comma (C<,>), as per the HTTP spec. 110joined together with a comma (C<,>), as per the HTTP spec.
111 111
112If an internal error occurs, such as not being able to resolve a hostname, 112If an internal error occurs, such as not being able to resolve a hostname,
113then C<$data> will be C<undef>, C<< $headers->{Status} >> will be C<599> 113then C<$data> will be C<undef>, C<< $headers->{Status} >> will be C<59x>
114and the C<Reason> pseudo-header will contain an error message. 114(usually C<599>) and the C<Reason> pseudo-header will contain an error
115message.
115 116
116A typical callback might look like this: 117A typical callback might look like this:
117 118
118 sub { 119 sub {
119 my ($body, $hdr) = @_; 120 my ($body, $hdr) = @_;
137 138
138=item headers => hashref 139=item headers => hashref
139 140
140The request headers to use. Currently, C<http_request> may provide its 141The request headers to use. Currently, C<http_request> may provide its
141own C<Host:>, C<Content-Length:>, C<Connection:> and C<Cookie:> headers 142own C<Host:>, C<Content-Length:>, C<Connection:> and C<Cookie:> headers
142and will provide defaults for C<User-Agent:> and C<Referer:>. 143and will provide defaults for C<User-Agent:> and C<Referer:> (this can be
144suppressed by using C<undef> for these headers in which case they won't be
145sent at all).
143 146
144=item timeout => $seconds 147=item timeout => $seconds
145 148
146The time-out to use for various stages - each connect attempt will reset 149The time-out to use for various stages - each connect attempt will reset
147the timeout, as will read or write activity. Default timeout is 5 minutes. 150the timeout, as will read or write activity, i.e. this is not an overall
151timeout.
152
153Default timeout is 5 minutes.
148 154
149=item proxy => [$host, $port[, $scheme]] or undef 155=item proxy => [$host, $port[, $scheme]] or undef
150 156
151Use the given http proxy for all requests. If not specified, then the 157Use the given http proxy for all requests. If not specified, then the
152default proxy (as specified by C<$ENV{http_proxy}>) is used. 158default proxy (as specified by C<$ENV{http_proxy}>) is used.
153 159
154C<$scheme> must be either missing or C<http> for HTTP, or C<https> for 160C<$scheme> must be either missing, C<http> for HTTP or C<https> for
155HTTPS. 161HTTPS.
156 162
157=item body => $string 163=item body => $string
158 164
159The request body, usually empty. Will be-sent as-is (future versions of 165The request body, usually empty. Will be-sent as-is (future versions of
165based on the original netscape specification. 171based on the original netscape specification.
166 172
167The C<$hash_ref> must be an (initially empty) hash reference which will 173The C<$hash_ref> must be an (initially empty) hash reference which will
168get updated automatically. It is possible to save the cookie_jar to 174get updated automatically. It is possible to save the cookie_jar to
169persistent storage with something like JSON or Storable, but this is not 175persistent storage with something like JSON or Storable, but this is not
170recommended, as expire times are currently being ignored. 176recommended, as expiry times are currently being ignored.
171 177
172Note that this cookie implementation is not of very high quality, nor 178Note that this cookie implementation is not of very high quality, nor
173meant to be complete. If you want complete cookie management you have to 179meant to be complete. If you want complete cookie management you have to
174do that on your own. C<cookie_jar> is meant as a quick fix to get some 180do that on your own. C<cookie_jar> is meant as a quick fix to get some
175cookie-using sites working. Cookies are a privacy disaster, do not use 181cookie-using sites working. Cookies are a privacy disaster, do not use
176them unless required to. 182them unless required to.
183
184=item tls_ctx => $scheme | $tls_ctx
185
186Specifies the AnyEvent::TLS context to be used for https connections. This
187parameter follows the same rules as the C<tls_ctx> parameter to
188L<AnyEvent::Handle>, but additionally, the two strings C<low> or
189C<high> can be specified, which give you a predefined low-security (no
190verification, highest compatibility) and high-security (CA and common-name
191verification) TLS context.
192
193The default for this option is C<low>, which could be interpreted as "give
194me the page, no matter what".
195
196=item on_prepare => $callback->($fh)
197
198In rare cases you need to "tune" the socket before it is used to
199connect (for exmaple, to bind it on a given IP address). This parameter
200overrides the prepare callback passed to C<AnyEvent::Socket::tcp_connect>
201and behaves exactly the same way (e.g. it has to provide a
202timeout). See the description for the C<$prepare_cb> argument of
203C<AnyEvent::Socket::tcp_connect> for details.
204
205=item on_header => $callback->($headers)
206
207When specified, this callback will be called with the header hash as soon
208as headers have been successfully received from the remote server (not on
209locally-generated errors).
210
211It has to return either true (in which case AnyEvent::HTTP will continue),
212or false, in which case AnyEvent::HTTP will cancel the download (and call
213the finish callback with an error code of C<598>).
214
215This callback is useful, among other things, to quickly reject unwanted
216content, which, if it is supposed to be rare, can be faster than first
217doing a C<HEAD> request.
218
219Example: cancel the request unless the content-type is "text/html".
220
221 on_header => sub {
222 $_[0]{"content-type"} =~ /^text\/html\s*(?:;|$)/
223 },
224
225=item on_body => $callback->($partial_body, $headers)
226
227When specified, all body data will be passed to this callback instead of
228to the completion callback. The completion callback will get the empty
229string instead of the body data.
230
231It has to return either true (in which case AnyEvent::HTTP will continue),
232or false, in which case AnyEvent::HTTP will cancel the download (and call
233the completion callback with an error code of C<598>).
234
235This callback is useful when the data is too large to be held in memory
236(so the callback writes it to a file) or when only some information should
237be extracted, or when the body should be processed incrementally.
238
239It is usually preferred over doing your own body handling via
240C<want_body_handle>, but in case of streaming APIs, where HTTP is
241only used to create a connection, C<want_body_handle> is the better
242alternative, as it allows you to install your own event handler, reducing
243resource usage.
244
245=item want_body_handle => $enable
246
247When enabled (default is disabled), the behaviour of AnyEvent::HTTP
248changes considerably: after parsing the headers, and instead of
249downloading the body (if any), the completion callback will be
250called. Instead of the C<$body> argument containing the body data, the
251callback will receive the L<AnyEvent::Handle> object associated with the
252connection. In error cases, C<undef> will be passed. When there is no body
253(e.g. status C<304>), the empty string will be passed.
254
255The handle object might or might not be in TLS mode, might be connected to
256a proxy, be a persistent connection etc., and configured in unspecified
257ways. The user is responsible for this handle (it will not be used by this
258module anymore).
259
260This is useful with some push-type services, where, after the initial
261headers, an interactive protocol is used (typical example would be the
262push-style twitter API which starts a JSON/XML stream).
263
264If you think you need this, first have a look at C<on_body>, to see if
265that doesn't solve your problem in a better way.
177 266
178=back 267=back
179 268
180Example: make a simple HTTP GET request for http://www.nethype.de/ 269Example: make a simple HTTP GET request for http://www.nethype.de/
181 270
237 push @{ $CO_SLOT{$_[0]}[1] }, $_[1]; 326 push @{ $CO_SLOT{$_[0]}[1] }, $_[1];
238 327
239 _slot_schedule $_[0]; 328 _slot_schedule $_[0];
240} 329}
241 330
331our $qr_nl = qr{\015?\012};
332our $qr_nlnl = qr{(?<![^\012])\015?\012};
333
334our $TLS_CTX_LOW = { cache => 1, sslv2 => 1 };
335our $TLS_CTX_HIGH = { cache => 1, verify => 1, verify_peername => "https" };
336
242sub http_request($$@) { 337sub http_request($$@) {
243 my $cb = pop; 338 my $cb = pop;
244 my ($method, $url, %arg) = @_; 339 my ($method, $url, %arg) = @_;
245 340
246 my %hdr; 341 my %hdr;
342
343 $arg{tls_ctx} = $TLS_CTX_LOW if $arg{tls_ctx} eq "low" || !exists $arg{tls_ctx};
344 $arg{tls_ctx} = $TLS_CTX_HIGH if $arg{tls_ctx} eq "high";
247 345
248 $method = uc $method; 346 $method = uc $method;
249 347
250 if (my $hdr = $arg{headers}) { 348 if (my $hdr = $arg{headers}) {
251 while (my ($k, $v) = each %$hdr) { 349 while (my ($k, $v) = each %$hdr) {
253 } 351 }
254 } 352 }
255 353
256 my $recurse = exists $arg{recurse} ? delete $arg{recurse} : $MAX_RECURSE; 354 my $recurse = exists $arg{recurse} ? delete $arg{recurse} : $MAX_RECURSE;
257 355
258 return $cb->(undef, { Status => 599, Reason => "recursion limit reached", URL => $url }) 356 return $cb->(undef, { Status => 599, Reason => "Too many redirections", URL => $url })
259 if $recurse < 0; 357 if $recurse < 0;
260 358
261 my $proxy = $arg{proxy} || $PROXY; 359 my $proxy = $arg{proxy} || $PROXY;
262 my $timeout = $arg{timeout} || $TIMEOUT; 360 my $timeout = $arg{timeout} || $TIMEOUT;
263 361
264 $hdr{"user-agent"} ||= $USERAGENT;
265
266 my ($uscheme, $uauthority, $upath, $query, $fragment) = 362 my ($uscheme, $uauthority, $upath, $query, $fragment) =
267 $url =~ m|(?:([^:/?#]+):)?(?://([^/?#]*))?([^?#]*)(?:\?([^#]*))?(?:#(.*))?|; 363 $url =~ m|(?:([^:/?#]+):)?(?://([^/?#]*))?([^?#]*)(?:\?([^#]*))?(?:#(.*))?|;
268 364
269 $uscheme = lc $uscheme; 365 $uscheme = lc $uscheme;
270 366
271 my $uport = $uscheme eq "http" ? 80 367 my $uport = $uscheme eq "http" ? 80
272 : $uscheme eq "https" ? 443 368 : $uscheme eq "https" ? 443
273 : return $cb->(undef, { Status => 599, Reason => "only http and https URL schemes supported", URL => $url }); 369 : return $cb->(undef, { Status => 599, Reason => "Only http and https URL schemes supported", URL => $url });
274
275 $hdr{referer} ||= "$uscheme://$uauthority$upath"; # leave out fragment and query string, just a heuristic
276 370
277 $uauthority =~ /^(?: .*\@ )? ([^\@:]+) (?: : (\d+) )?$/x 371 $uauthority =~ /^(?: .*\@ )? ([^\@:]+) (?: : (\d+) )?$/x
278 or return $cb->(undef, { Status => 599, Reason => "unparsable URL", URL => $url }); 372 or return $cb->(undef, { Status => 599, Reason => "Unparsable URL", URL => $url });
279 373
280 my $uhost = $1; 374 my $uhost = $1;
281 $uport = $2 if defined $2; 375 $uport = $2 if defined $2;
376
377 $hdr{host} = defined $2 ? "$uhost:$2" : "$uhost"
378 unless exists $hdr{host};
282 379
283 $uhost =~ s/^\[(.*)\]$/$1/; 380 $uhost =~ s/^\[(.*)\]$/$1/;
284 $upath .= "?$query" if length $query; 381 $upath .= "?$query" if length $query;
285 382
286 $upath =~ s%^/?%/%; 383 $upath =~ s%^/?%/%;
317 } 414 }
318 415
319 my ($rhost, $rport, $rscheme, $rpath); # request host, port, path 416 my ($rhost, $rport, $rscheme, $rpath); # request host, port, path
320 417
321 if ($proxy) { 418 if ($proxy) {
322 ($rhost, $rport, $rscheme, $rpath) = (@$proxy, $url); 419 ($rpath, $rhost, $rport, $rscheme) = ($url, @$proxy);
420
421 $rscheme = "http" unless defined $rscheme;
323 422
324 # don't support https requests over https-proxy transport, 423 # don't support https requests over https-proxy transport,
325 # can't be done with tls as spec'ed. 424 # can't be done with tls as spec'ed, unless you double-encrypt.
326 $rscheme = "http" if $uscheme eq "https" && $rscheme eq "https"; 425 $rscheme = "http" if $uscheme eq "https" && $rscheme eq "https";
327 } else { 426 } else {
328 ($rhost, $rport, $rscheme, $rpath) = ($uhost, $uport, $uscheme, $upath); 427 ($rhost, $rport, $rscheme, $rpath) = ($uhost, $uport, $uscheme, $upath);
329 } 428 }
330 429
331 $hdr{host} = $uhost; 430 # leave out fragment and query string, just a heuristic
431 $hdr{referer} ||= "$uscheme://$uauthority$upath" unless exists $hdr{referer};
432 $hdr{"user-agent"} ||= $USERAGENT unless exists $hdr{"user-agent"};
433
332 $hdr{"content-length"} = length $arg{body}; 434 $hdr{"content-length"} = length $arg{body}
435 if length $arg{body} || $method ne "GET";
333 436
334 my %state = (connect_guard => 1); 437 my %state = (connect_guard => 1);
335 438
336 _get_slot $uhost, sub { 439 _get_slot $uhost, sub {
337 $state{slot_guard} = shift; 440 $state{slot_guard} = shift;
338 441
339 return unless $state{connect_guard}; 442 return unless $state{connect_guard};
340 443
341 $state{connect_guard} = AnyEvent::Socket::tcp_connect $rhost, $rport, sub { 444 $state{connect_guard} = AnyEvent::Socket::tcp_connect $rhost, $rport, sub {
342 $state{fh} = shift 445 $state{fh} = shift
446 or do {
447 my $err = "$!";
448 %state = ();
343 or return $cb->(undef, { Status => 599, Reason => "$!", URL => $url }); 449 return $cb->(undef, { Status => 599, Reason => $err, URL => $url });
450 };
344 451
345 delete $state{connect_guard}; # reduce memory usage, save a tree 452 pop; # free memory, save a tree
453
454 return unless delete $state{connect_guard};
346 455
347 # get handle 456 # get handle
348 $state{handle} = new AnyEvent::Handle 457 $state{handle} = new AnyEvent::Handle
349 fh => $state{fh}; 458 fh => $state{fh},
459 timeout => $timeout,
460 peername => $rhost,
461 tls_ctx => $arg{tls_ctx};
350 462
351 # limit the number of persistent connections 463 # limit the number of persistent connections
464 # keepalive not yet supported
352 if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) { 465 if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) {
353 ++$KA_COUNT{$_[1]}; 466 ++$KA_COUNT{$_[1]};
354 $state{handle}{ka_count_guard} = AnyEvent::Util::guard { --$KA_COUNT{$_[1]} }; 467 $state{handle}{ka_count_guard} = AnyEvent::Util::guard {
468 --$KA_COUNT{$_[1]}
469 };
355 $hdr{connection} = "keep-alive"; 470 $hdr{connection} = "keep-alive";
356 delete $hdr{connection}; # keep-alive not yet supported
357 } else { 471 } else {
358 delete $hdr{connection}; 472 delete $hdr{connection};
359 } 473 }
360 474
361 # (re-)configure handle 475 # (re-)configure handle
362 $state{handle}->timeout ($timeout);
363 $state{handle}->on_error (sub { 476 $state{handle}->on_error (sub {
364 my $errno = "$!";
365 %state = (); 477 %state = ();
366 $cb->(undef, { Status => 599, Reason => $errno, URL => $url }); 478 $cb->(undef, { Status => 599, Reason => $_[2], URL => $url });
367 }); 479 });
368 $state{handle}->on_eof (sub { 480 $state{handle}->on_eof (sub {
369 %state = (); 481 %state = ();
370 $cb->(undef, { Status => 599, Reason => "unexpected end-of-file", URL => $url }); 482 $cb->(undef, { Status => 599, Reason => "Unexpected end-of-file", URL => $url });
371 }); 483 });
372 484
373 $state{handle}->starttls ("connect") if $rscheme eq "https"; 485 $state{handle}->starttls ("connect") if $rscheme eq "https";
374 486
375 # handle actual, non-tunneled, request 487 # handle actual, non-tunneled, request
376 my $handle_actual_request = sub { 488 my $handle_actual_request = sub {
377# $state{handle}->starttls ("connect") if $uscheme eq "https"; 489 $state{handle}->starttls ("connect") if $uscheme eq "https" && !exists $state{handle}{tls};
378 490
379 # send request 491 # send request
380 $state{handle}->push_write ( 492 $state{handle}->push_write (
381 "$method $rpath HTTP/1.0\015\012" 493 "$method $rpath HTTP/1.0\015\012"
382 . (join "", map "\u$_: $hdr{$_}\015\012", keys %hdr) 494 . (join "", map "\u$_: $hdr{$_}\015\012", grep defined $hdr{$_}, keys %hdr)
383 . "\015\012" 495 . "\015\012"
384 . (delete $arg{body}) 496 . (delete $arg{body})
385 ); 497 );
386 498
499 # return if error occured during push_write()
500 return unless %state;
501
387 %hdr = (); # reduce memory usage, save a kitten 502 %hdr = (); # reduce memory usage, save a kitten
388 503
389 # status line 504 # status line
390 $state{handle}->push_read (line => qr/\015?\012/, sub { 505 $state{handle}->push_read (line => $qr_nl, sub {
391 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix 506 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix
392 or return (%state = (), $cb->(undef, { Status => 599, Reason => "invalid server response ($_[1])", URL => $url })); 507 or return (%state = (), $cb->(undef, { Status => 599, Reason => "Invalid server response ($_[1])", URL => $url }));
393 508
394 my %hdr = ( # response headers 509 my %hdr = ( # response headers
395 HTTPVersion => ",$1", 510 HTTPVersion => ",$1",
396 Status => ",$2", 511 Status => ",$2",
397 Reason => ",$3", 512 Reason => ",$3",
398 URL => ",$url" 513 URL => ",$url"
399 ); 514 );
400 515
401 # headers, could be optimized a bit 516 # headers, could be optimized a bit
402 $state{handle}->unshift_read (line => qr/\015?\012\015?\012/, sub { 517 $state{handle}->unshift_read (line => $qr_nlnl, sub {
403 for ("$_[1]\012") { 518 for ("$_[1]") {
404 y/\015//d; # weed out any \015, as they show up in the weirdest of places. 519 y/\015//d; # weed out any \015, as they show up in the weirdest of places.
405 520
406 # we support spaces in field names, as lotus domino 521 # things seen, not parsed:
407 # creates them (actually spaces around seperators 522 # p3pP="NON CUR OTPi OUR NOR UNI"
408 # are strictly allowed in http, they are a security issue). 523
409 $hdr{lc $1} .= ",$2" 524 $hdr{lc $1} .= ",$2"
410 while /\G 525 while /\G
411 ([^:\000-\037]+): 526 ([^:\000-\037]*):
412 [\011\040]* 527 [\011\040]*
413 ((?: [^\012]+ | \012[\011\040] )*) 528 ((?: [^\012]+ | \012[\011\040] )*)
414 \012 529 \012
415 /gxc; 530 /gxc;
416 531
417 /\G$/ 532 /\G$/
418 or return (%state = (), $cb->(undef, { Status => 599, Reason => "garbled response headers", URL => $url })); 533 or return (%state = (), $cb->(undef, { Status => 599, Reason => "Garbled response headers", URL => $url }));
419 } 534 }
420 535
421 substr $_, 0, 1, "" 536 substr $_, 0, 1, ""
422 for values %hdr; 537 for values %hdr;
423 538
539 # redirect handling
540 # microsoft and other shitheads don't give a shit for following standards,
541 # try to support some common forms of broken Location headers.
542 if ($hdr{location} !~ /^(?: $ | [^:\/?\#]+ : )/x) {
543 $hdr{location} =~ s/^\.\/+//;
544
545 my $url = "$rscheme://$uhost:$uport";
546
547 unless ($hdr{location} =~ s/^\///) {
548 $url .= $upath;
549 $url =~ s/\/[^\/]*$//;
550 }
551
552 $hdr{location} = "$url/$hdr{location}";
553 }
554
555 my $redirect;
556
557 if ($recurse) {
558 if ($hdr{Status} =~ /^30[12]$/ && $method ne "POST") {
559 # apparently, mozilla et al. just change POST to GET here
560 # more research is needed before we do the same
561 $redirect = 1;
562 } elsif ($hdr{Status} == 303) {
563 # even http/1.1 is unclear on how to mutate the method
564 $method = "GET" unless $method eq "HEAD";
565 $redirect = 1;
566 } elsif ($hdr{Status} == 307 && $method =~ /^(?:GET|HEAD)$/) {
567 $redirect = 1;
568 }
569 }
570
424 my $finish = sub { 571 my $finish = sub {
425 # TODO: use destroy method, when/if available
426 #$state{handle}->destroy; 572 $state{handle}->destroy if $state{handle};
427 $state{handle}->on_eof (undef);
428 $state{handle}->on_error (undef);
429 %state = (); 573 %state = ();
430 574
431 # set-cookie processing 575 # set-cookie processing
432 if ($arg{cookie_jar}) { 576 if ($arg{cookie_jar}) {
433 for ($hdr{"set-cookie"}) { 577 for ($_[1]{"set-cookie"}) {
434 # parse NAME=VALUE 578 # parse NAME=VALUE
435 my @kv; 579 my @kv;
436 580
437 while (/\G\s* ([^=;,[:space:]]+) \s*=\s* (?: "((?:[^\\"]+|\\.)*)" | ([^=;,[:space:]]*) )/gcxs) { 581 while (/\G\s* ([^=;,[:space:]]+) \s*=\s* (?: "((?:[^\\"]+|\\.)*)" | ([^=;,[:space:]]*) )/gcxs) {
438 my $name = $1; 582 my $name = $1;
476 620
477 redo if /\G\s*,/gc; 621 redo if /\G\s*,/gc;
478 } 622 }
479 } 623 }
480 624
481 # microsoft and other shitheads don't give a shit for following standards, 625 if ($redirect && exists $hdr{location}) {
482 # try to support some common forms of broken Location headers. 626 # we ignore any errors, as it is very common to receive
483 if ($_[1]{location} !~ /^(?: $ | [^:\/?\#]+ : )/x) { 627 # Content-Length != 0 but no actual body
484 $_[1]{location} =~ s/^\.\/+//; 628 # we also access %hdr, as $_[1] might be an erro
485
486 my $url = "$rscheme://$uhost:$uport";
487
488 unless ($_[1]{location} =~ s/^\///) {
489 $url .= $upath;
490 $url =~ s/\/[^\/]*$//;
491 }
492
493 $_[1]{location} = "$url/$_[1]{location}";
494 }
495
496 if ($_[1]{Status} =~ /^30[12]$/ && $recurse && $method ne "POST") {
497 # apparently, mozilla et al. just change POST to GET here
498 # more research is needed before we do the same
499 http_request ($method, $_[1]{location}, %arg, recurse => $recurse - 1, $cb);
500 } elsif ($_[1]{Status} == 303 && $recurse) {
501 # even http/1.1 is unclear on how to mutate the method
502 $method = "GET" unless $method eq "HEAD";
503 http_request ($method => $_[1]{location}, %arg, recurse => $recurse - 1, $cb); 629 http_request ($method => $hdr{location}, %arg, recurse => $recurse - 1, $cb);
504 } elsif ($_[1]{Status} == 307 && $recurse && $method =~ /^(?:GET|HEAD)$/) {
505 http_request ($method => $_[1]{location}, %arg, recurse => $recurse - 1, $cb);
506 } else { 630 } else {
507 $cb->($_[0], $_[1]); 631 $cb->($_[0], $_[1]);
508 } 632 }
509 }; 633 };
510 634
511 if ($hdr{Status} =~ /^(?:1..|204|304)$/ or $method eq "HEAD") { 635 my $len = $hdr{"content-length"};
636
637 if (!$redirect && $arg{on_header} && !$arg{on_header}(\%hdr)) {
638 $finish->(undef, { Status => 598, Reason => "Request cancelled by on_header", URL => $url });
639 } elsif (
640 $hdr{Status} =~ /^(?:1..|[23]04)$/
641 or $method eq "HEAD"
642 or (defined $len && !$len)
643 ) {
644 # no body
512 $finish->(undef, \%hdr); 645 $finish->("", \%hdr);
513 } else { 646 } else {
514 if (exists $hdr{"content-length"}) { 647 # body handling, four different code paths
515 $_[0]->unshift_read (chunk => $hdr{"content-length"}, sub { 648 # for want_body_handle, on_body (2x), normal (2x)
516 # could cache persistent connection now 649 # we might read too much here, but it does not matter yet (no pers. connections)
517 if ($hdr{connection} =~ /\bkeep-alive\b/i) { 650 if (!$redirect && $arg{want_body_handle}) {
518 # but we don't, due to misdesigns, this is annoyingly complex 651 $_[0]->on_eof (undef);
652 $_[0]->on_error (undef);
653 $_[0]->on_read (undef);
654
655 $finish->(delete $state{handle}, \%hdr);
656
657 } elsif ($arg{on_body}) {
658 $_[0]->on_error (sub { $finish->(undef, { Status => 599, Reason => $_[2], URL => $url }) });
659 if ($len) {
660 $_[0]->on_eof (undef);
661 $_[0]->on_read (sub {
662 $len -= length $_[0]{rbuf};
663
664 $arg{on_body}(delete $_[0]{rbuf}, \%hdr)
665 or $finish->(undef, { Status => 598, Reason => "Request cancelled by on_body", URL => $url });
666
667 $len > 0
668 or $finish->("", \%hdr);
519 }; 669 });
520 670 } else {
671 $_[0]->on_eof (sub {
521 $finish->($_[1], \%hdr); 672 $finish->("", \%hdr);
673 });
674 $_[0]->on_read (sub {
675 $arg{on_body}(delete $_[0]{rbuf}, \%hdr)
676 or $finish->(undef, { Status => 598, Reason => "Request cancelled by on_body", URL => $url });
677 });
522 }); 678 }
523 } else { 679 } else {
524 # too bad, need to read until we get an error or EOF,
525 # no way to detect winged data.
526 $_[0]->on_error (sub {
527 $finish->($_[0]{rbuf}, \%hdr);
528 });
529 $_[0]->on_eof (undef); 680 $_[0]->on_eof (undef);
681
682 if ($len) {
683 $_[0]->on_error (sub { $finish->(undef, { Status => 599, Reason => $_[2], URL => $url }) });
684 $_[0]->on_read (sub {
685 $finish->((substr delete $_[0]{rbuf}, 0, $len, ""), \%hdr)
686 if $len <= length $_[0]{rbuf};
687 });
688 } else {
689 $_[0]->on_error (sub {
690 $! == Errno::EPIPE || !$!
691 ? $finish->(delete $_[0]{rbuf}, \%hdr)
692 : $finish->(undef, { Status => 599, Reason => $_[2], URL => $url });
693 });
530 $_[0]->on_read (sub { }); 694 $_[0]->on_read (sub { });
695 }
531 } 696 }
532 } 697 }
533 }); 698 });
534 }); 699 });
535 }; 700 };
538 if ($proxy && $uscheme eq "https") { 703 if ($proxy && $uscheme eq "https") {
539 # oh dear, we have to wrap it into a connect request 704 # oh dear, we have to wrap it into a connect request
540 705
541 # maybe re-use $uauthority with patched port? 706 # maybe re-use $uauthority with patched port?
542 $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012Host: $uhost\015\012\015\012"); 707 $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012Host: $uhost\015\012\015\012");
543 $state{handle}->push_read (line => qr/\015?\012\015?\012/, sub { 708 $state{handle}->push_read (line => $qr_nlnl, sub {
544 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix 709 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix
545 or return (%state = (), $cb->(undef, { Status => 599, Reason => "invalid proxy connect response ($_[1])", URL => $url })); 710 or return (%state = (), $cb->(undef, { Status => 599, Reason => "Invalid proxy connect response ($_[1])", URL => $url }));
546 711
547 if ($2 == 200) { 712 if ($2 == 200) {
548 $rpath = $upath; 713 $rpath = $upath;
549 &$handle_actual_request; 714 &$handle_actual_request;
550 } else { 715 } else {
554 }); 719 });
555 } else { 720 } else {
556 &$handle_actual_request; 721 &$handle_actual_request;
557 } 722 }
558 723
559 }, sub { 724 }, $arg{on_prepare} || sub { $timeout };
560 $timeout
561 };
562 }; 725 };
563 726
564 defined wantarray && AnyEvent::Util::guard { %state = () } 727 defined wantarray && AnyEvent::Util::guard { %state = () }
565} 728}
566 729
587=over 4 750=over 4
588 751
589=item AnyEvent::HTTP::set_proxy "proxy-url" 752=item AnyEvent::HTTP::set_proxy "proxy-url"
590 753
591Sets the default proxy server to use. The proxy-url must begin with a 754Sets the default proxy server to use. The proxy-url must begin with a
592string of the form C<http://host:port> (optionally C<https:...>). 755string of the form C<http://host:port> (optionally C<https:...>), croaks
756otherwise.
757
758To clear an already-set proxy, use C<undef>.
593 759
594=item $AnyEvent::HTTP::MAX_RECURSE 760=item $AnyEvent::HTTP::MAX_RECURSE
595 761
596The default value for the C<recurse> request parameter (default: C<10>). 762The default value for the C<recurse> request parameter (default: C<10>).
597 763
598=item $AnyEvent::HTTP::USERAGENT 764=item $AnyEvent::HTTP::USERAGENT
599 765
600The default value for the C<User-Agent> header (the default is 766The default value for the C<User-Agent> header (the default is
601C<Mozilla/5.0 (compatible; AnyEvent::HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)>). 767C<Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)>).
602 768
603=item $AnyEvent::HTTP::MAX_PERSISTENT 769=item $AnyEvent::HTTP::MAX_PER_HOST
604 770
605The maximum number of persistent connections to keep open (default: 8). 771The maximum number of concurrent connections to the same host (identified
772by the hostname). If the limit is exceeded, then the additional requests
773are queued until previous connections are closed.
606 774
607Not implemented currently. 775The default value for this is C<4>, and it is highly advisable to not
608 776increase it.
609=item $AnyEvent::HTTP::PERSISTENT_TIMEOUT
610
611The maximum time to cache a persistent connection, in seconds (default: 2).
612
613Not implemented currently.
614 777
615=item $AnyEvent::HTTP::ACTIVE 778=item $AnyEvent::HTTP::ACTIVE
616 779
617The number of active connections. This is not the number of currently 780The number of active connections. This is not the number of currently
618running requests, but the number of currently open and non-idle TCP 781running requests, but the number of currently open and non-idle TCP
621=back 784=back
622 785
623=cut 786=cut
624 787
625sub set_proxy($) { 788sub set_proxy($) {
789 if (length $_[0]) {
626 $PROXY = [$2, $3 || 3128, $1] if $_[0] =~ m%^(https?):// ([^:/]+) (?: : (\d*) )?%ix; 790 $_[0] =~ m%^(https?):// ([^:/]+) (?: : (\d*) )?%ix
791 or Carp::croak "$_[0]: invalid proxy URL";
792 $PROXY = [$2, $3 || 3128, $1]
793 } else {
794 undef $PROXY;
795 }
627} 796}
628 797
629# initialise proxy from environment 798# initialise proxy from environment
799eval {
630set_proxy $ENV{http_proxy}; 800 set_proxy $ENV{http_proxy};
801};
631 802
632=head1 SEE ALSO 803=head1 SEE ALSO
633 804
634L<AnyEvent>. 805L<AnyEvent>.
635 806
636=head1 AUTHOR 807=head1 AUTHOR
637 808
638 Marc Lehmann <schmorp@schmorp.de> 809 Marc Lehmann <schmorp@schmorp.de>
639 http://home.schmorp.de/ 810 http://home.schmorp.de/
640 811
812With many thanks to Дмитрий Шалашов, who provided countless
813testcases and bugreports.
814
641=cut 815=cut
642 816
6431 8171
644 818

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines