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.10 by root, Thu Jun 5 13:06:43 2008 UTC vs.
Revision 1.53 by root, Sat Dec 5 15:37:07 2009 UTC

3AnyEvent::HTTP - simple but non-blocking HTTP/HTTPS client 3AnyEvent::HTTP - simple but non-blocking HTTP/HTTPS client
4 4
5=head1 SYNOPSIS 5=head1 SYNOPSIS
6 6
7 use AnyEvent::HTTP; 7 use AnyEvent::HTTP;
8
9 http_get "http://www.nethype.de/", sub { print $_[1] };
10
11 # ... do something else here
8 12
9=head1 DESCRIPTION 13=head1 DESCRIPTION
10 14
11This module is an L<AnyEvent> user, you need to make sure that you use and 15This module is an L<AnyEvent> user, you need to make sure that you use and
12run a supported event loop. 16run a supported event loop.
13 17
18This module implements a simple, stateless and non-blocking HTTP
19client. It supports GET, POST and other request methods, cookies and more,
20all on a very low level. It can follow redirects supports proxies and
21automatically limits the number of connections to the values specified in
22the RFC.
23
24It should generally be a "good client" that is enough for most HTTP
25tasks. Simple tasks should be simple, but complex tasks should still be
26possible as the user retains control over request and response headers.
27
28The caller is responsible for authentication management, cookies (if
29the simplistic implementation in this module doesn't suffice), referer
30and other high-level protocol details for which this module offers only
31limited support.
32
14=head2 METHODS 33=head2 METHODS
15 34
16=over 4 35=over 4
17 36
18=cut 37=cut
20package AnyEvent::HTTP; 39package AnyEvent::HTTP;
21 40
22use strict; 41use strict;
23no warnings; 42no warnings;
24 43
25use Carp; 44use Errno ();
26 45
27use AnyEvent (); 46use AnyEvent 5.0 ();
28use AnyEvent::Util (); 47use AnyEvent::Util ();
29use AnyEvent::Socket (); 48use AnyEvent::Socket ();
30use AnyEvent::Handle (); 49use AnyEvent::Handle ();
31 50
32use base Exporter::; 51use base Exporter::;
33 52
34our $VERSION = '1.0'; 53our $VERSION = '1.44';
35 54
36our @EXPORT = qw(http_get http_request); 55our @EXPORT = qw(http_get http_post http_head http_request);
37 56
38our $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)";
39our $MAX_RECURSE = 10; 58our $MAX_RECURSE = 10;
40our $MAX_PERSISTENT = 8; 59our $MAX_PERSISTENT = 8;
41our $PERSISTENT_TIMEOUT = 2; 60our $PERSISTENT_TIMEOUT = 2;
42our $TIMEOUT = 300; 61our $TIMEOUT = 300;
43 62
44# changing these is evil 63# changing these is evil
45our $MAX_PERSISTENT_PER_HOST = 2; 64our $MAX_PERSISTENT_PER_HOST = 0;
46our $MAX_PER_HOST = 4; # not respected yet :( 65our $MAX_PER_HOST = 4;
47 66
48our $PROXY; 67our $PROXY;
68our $ACTIVE = 0;
49 69
50my %KA_COUNT; # number of open keep-alive connections per host 70my %KA_COUNT; # number of open keep-alive connections per host
71my %CO_SLOT; # number of open connections, and wait queue, per host
51 72
52=item http_get $url, key => value..., $cb->($data, $headers) 73=item http_get $url, key => value..., $cb->($data, $headers)
53 74
54Executes an HTTP-GET request. See the http_request function for details on 75Executes an HTTP-GET request. See the http_request function for details on
55additional parameters. 76additional parameters and the return value.
56 77
57=item http_head $url, key => value..., $cb->($data, $headers) 78=item http_head $url, key => value..., $cb->($data, $headers)
58 79
59Executes an HTTP-HEAD request. See the http_request function for details on 80Executes an HTTP-HEAD request. See the http_request function for details
60additional parameters. 81on additional parameters and the return value.
61 82
62=item http_post $url, $body, key => value..., $cb->($data, $headers) 83=item http_post $url, $body, key => value..., $cb->($data, $headers)
63 84
64Executes an HTTP-POST request with a request body of C<$bod>. See the 85Executes an HTTP-POST request with a request body of C<$body>. See the
65http_request function for details on additional parameters. 86http_request function for details on additional parameters and the return
87value.
66 88
67=item http_request $method => $url, key => value..., $cb->($data, $headers) 89=item http_request $method => $url, key => value..., $cb->($data, $headers)
68 90
69Executes a HTTP request of type C<$method> (e.g. C<GET>, C<POST>). The URL 91Executes a HTTP request of type C<$method> (e.g. C<GET>, C<POST>). The URL
70must be an absolute http or https URL. 92must be an absolute http or https URL.
71 93
94When called in void context, nothing is returned. In other contexts,
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
97destroyed before the callbakc is called, the request will be cancelled.
98
72The callback will be called with the response data as first argument 99The callback will be called with the response body data as first argument
73(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
74response headers as second argument. 101second argument.
75 102
76All 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
77headers, the three "pseudo-headers" C<HTTPVersion>, C<Status> and 104headers, the "pseudo-headers" C<HTTPVersion>, C<Status> and C<Reason>
78C<Reason> contain the three parts of the HTTP Status-Line of the same 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
107requested URL when following redirects).
108
79name. If the server sends a header multiple lines, then their contents 109If the server sends a header multiple times, then their contents will be
80will be joined together with C<\x00>. 110joined together with a comma (C<,>), as per the HTTP spec.
81 111
82If 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,
83then 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>
84and 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.
85 116
86A typical callback might look like this: 117A typical callback might look like this:
87 118
88 sub { 119 sub {
89 my ($body, $hdr) = @_; 120 my ($body, $hdr) = @_;
105Whether to recurse requests or not, e.g. on redirects, authentication 136Whether to recurse requests or not, e.g. on redirects, authentication
106retries and so on, and how often to do so. 137retries and so on, and how often to do so.
107 138
108=item headers => hashref 139=item headers => hashref
109 140
110The request headers to use. 141The request headers to use. Currently, C<http_request> may provide its
142own C<Host:>, C<Content-Length:>, C<Connection:> and C<Cookie:> headers
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).
111 146
112=item timeout => $seconds 147=item timeout => $seconds
113 148
114The 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
115the 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.
116 154
117=item proxy => [$host, $port[, $scheme]] or undef 155=item proxy => [$host, $port[, $scheme]] or undef
118 156
119Use 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
120default proxy (as specified by C<$ENV{http_proxy}>) is used. 158default proxy (as specified by C<$ENV{http_proxy}>) is used.
121 159
122C<$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
123HTTPS. 161HTTPS.
124 162
125=item body => $string 163=item body => $string
126 164
127The 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
133based on the original netscape specification. 171based on the original netscape specification.
134 172
135The 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
136get updated automatically. It is possible to save the cookie_jar to 174get updated automatically. It is possible to save the cookie_jar to
137persistent storage with something like JSON or Storable, but this is not 175persistent storage with something like JSON or Storable, but this is not
138recommended, as expire times are currently being ignored. 176recommended, as expiry times are currently being ignored.
139 177
140Note that this cookie implementation is not of very high quality, nor 178Note that this cookie implementation is not of very high quality, nor
141meant 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
142do 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
143cookie-using sites working. Cookies are a privacy disaster, do not use 181cookie-using sites working. Cookies are a privacy disaster, do not use
144them 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.
145 266
146=back 267=back
147 268
148Example: make a simple HTTP GET request for http://www.nethype.de/ 269Example: make a simple HTTP GET request for http://www.nethype.de/
149 270
163 use Data::Dumper; 284 use Data::Dumper;
164 print Dumper $hdr; 285 print Dumper $hdr;
165 } 286 }
166 ; 287 ;
167 288
289Example: make another simple HTTP GET request, but immediately try to
290cancel it.
291
292 my $request = http_request GET => "http://www.nethype.de/", sub {
293 my ($body, $hdr) = @_;
294 print "$body\n";
295 };
296
297 undef $request;
298
168=cut 299=cut
169 300
301sub _slot_schedule;
302sub _slot_schedule($) {
303 my $host = shift;
304
305 while ($CO_SLOT{$host}[0] < $MAX_PER_HOST) {
306 if (my $cb = shift @{ $CO_SLOT{$host}[1] }) {
307 # somebody wants that slot
308 ++$CO_SLOT{$host}[0];
309 ++$ACTIVE;
310
311 $cb->(AnyEvent::Util::guard {
312 --$ACTIVE;
313 --$CO_SLOT{$host}[0];
314 _slot_schedule $host;
315 });
316 } else {
317 # nobody wants the slot, maybe we can forget about it
318 delete $CO_SLOT{$host} unless $CO_SLOT{$host}[0];
319 last;
320 }
321 }
322}
323
324# wait for a free slot on host, call callback
325sub _get_slot($$) {
326 push @{ $CO_SLOT{$_[0]}[1] }, $_[1];
327
328 _slot_schedule $_[0];
329}
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
170sub http_request($$$;@) { 337sub http_request($$@) {
171 my $cb = pop; 338 my $cb = pop;
172 my ($method, $url, %arg) = @_; 339 my ($method, $url, %arg) = @_;
173 340
174 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";
175 345
176 $method = uc $method; 346 $method = uc $method;
177 347
178 if (my $hdr = $arg{headers}) { 348 if (my $hdr = $arg{headers}) {
179 while (my ($k, $v) = each %$hdr) { 349 while (my ($k, $v) = each %$hdr) {
180 $hdr{lc $k} = $v; 350 $hdr{lc $k} = $v;
181 } 351 }
182 } 352 }
183 353
184 my $recurse = exists $arg{recurse} ? $arg{recurse} : $MAX_RECURSE; 354 my $recurse = exists $arg{recurse} ? delete $arg{recurse} : $MAX_RECURSE;
185 355
186 return $cb->(undef, { Status => 599, Reason => "recursion limit reached" }) 356 return $cb->(undef, { Status => 599, Reason => "Too many redirections", URL => $url })
187 if $recurse < 0; 357 if $recurse < 0;
188 358
189 my $proxy = $arg{proxy} || $PROXY; 359 my $proxy = $arg{proxy} || $PROXY;
190 my $timeout = $arg{timeout} || $TIMEOUT; 360 my $timeout = $arg{timeout} || $TIMEOUT;
191 361
192 $hdr{"user-agent"} ||= $USERAGENT;
193
194 my ($scheme, $authority, $upath, $query, $fragment) = 362 my ($uscheme, $uauthority, $upath, $query, $fragment) =
195 $url =~ m|(?:([^:/?#]+):)?(?://([^/?#]*))?([^?#]*)(?:\?([^#]*))?(?:#(.*))?|; 363 $url =~ m|(?:([^:/?#]+):)?(?://([^/?#]*))?([^?#]*)(?:\?([^#]*))?(?:#(.*))?|;
196 364
197 $scheme = lc $scheme; 365 $uscheme = lc $uscheme;
198 366
199 my $uport = $scheme eq "http" ? 80 367 my $uport = $uscheme eq "http" ? 80
200 : $scheme eq "https" ? 443 368 : $uscheme eq "https" ? 443
201 : return $cb->(undef, { Status => 599, Reason => "only http and https URL schemes supported" }); 369 : return $cb->(undef, { Status => 599, Reason => "Only http and https URL schemes supported", URL => $url });
202 370
203 $authority =~ /^(?: .*\@ )? ([^\@:]+) (?: : (\d+) )?$/x 371 $uauthority =~ /^(?: .*\@ )? ([^\@:]+) (?: : (\d+) )?$/x
204 or return $cb->(undef, { Status => 599, Reason => "unparsable URL" }); 372 or return $cb->(undef, { Status => 599, Reason => "Unparsable URL", URL => $url });
205 373
206 my $uhost = $1; 374 my $uhost = $1;
207 $uport = $2 if defined $2; 375 $uport = $2 if defined $2;
208 376
377 $hdr{host} = defined $2 ? "$uhost:$2" : "$uhost"
378 unless exists $hdr{host};
379
209 $uhost =~ s/^\[(.*)\]$/$1/; 380 $uhost =~ s/^\[(.*)\]$/$1/;
210 $upath .= "?$query" if length $query; 381 $upath .= "?$query" if length $query;
211 382
212 $upath =~ s%^/?%/%; 383 $upath =~ s%^/?%/%;
213 384
214 # cookie processing 385 # cookie processing
215 if (my $jar = $arg{cookie_jar}) { 386 if (my $jar = $arg{cookie_jar}) {
216 %$jar = () if $jar->{version} < 1; 387 %$jar = () if $jar->{version} != 1;
217 388
218 my @cookie; 389 my @cookie;
219 390
220 while (my ($chost, $v) = each %$jar) { 391 while (my ($chost, $v) = each %$jar) {
392 if ($chost =~ /^\./) {
221 next unless $chost eq substr $uhost, -length $chost; 393 next unless $chost eq substr $uhost, -length $chost;
222 next unless $chost =~ /^\./; 394 } elsif ($chost =~ /\./) {
395 next unless $chost eq $uhost;
396 } else {
397 next;
398 }
223 399
224 while (my ($cpath, $v) = each %$v) { 400 while (my ($cpath, $v) = each %$v) {
225 next unless $cpath eq substr $upath, 0, length $cpath; 401 next unless $cpath eq substr $upath, 0, length $cpath;
226 402
227 while (my ($k, $v) = each %$v) { 403 while (my ($k, $v) = each %$v) {
228 next if $scheme ne "https" && exists $v->{secure}; 404 next if $uscheme ne "https" && exists $v->{secure};
405 my $value = $v->{value};
406 $value =~ s/([\\"])/\\$1/g;
229 push @cookie, "$k=$v->{value}"; 407 push @cookie, "$k=\"$value\"";
230 } 408 }
231 } 409 }
232 } 410 }
233 411
234 $hdr{cookie} = join "; ", @cookie 412 $hdr{cookie} = join "; ", @cookie
235 if @cookie; 413 if @cookie;
236 } 414 }
237 415
238 my ($rhost, $rport, $rpath); # request host, port, path 416 my ($rhost, $rport, $rscheme, $rpath); # request host, port, path
239 417
240 if ($proxy) { 418 if ($proxy) {
241 ($rhost, $rport, $scheme) = @$proxy; 419 ($rpath, $rhost, $rport, $rscheme) = ($url, @$proxy);
242 $rpath = $url; 420
421 $rscheme = "http" unless defined $rscheme;
422
423 # don't support https requests over https-proxy transport,
424 # can't be done with tls as spec'ed, unless you double-encrypt.
425 $rscheme = "http" if $uscheme eq "https" && $rscheme eq "https";
243 } else { 426 } else {
244 ($rhost, $rport, $rpath) = ($uhost, $uport, $upath); 427 ($rhost, $rport, $rscheme, $rpath) = ($uhost, $uport, $uscheme, $upath);
245 $hdr{host} = $uhost;
246 } 428 }
247 429
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
248 $hdr{"content-length"} = length $arg{body}; 434 $hdr{"content-length"} = length $arg{body}
435 if length $arg{body} || $method ne "GET";
249 436
250 my %state; 437 my %state = (connect_guard => 1);
251 438
439 _get_slot $uhost, sub {
440 $state{slot_guard} = shift;
441
442 return unless $state{connect_guard};
443
252 $state{connect_guard} = AnyEvent::Socket::tcp_connect $rhost, $rport, sub { 444 $state{connect_guard} = AnyEvent::Socket::tcp_connect $rhost, $rport, sub {
253 $state{fh} = shift 445 $state{fh} = shift
254 or return $cb->(undef, { Status => 599, Reason => "$!" });
255
256 delete $state{connect_guard}; # reduce memory usage, save a tree
257
258 # get handle
259 $state{handle} = new AnyEvent::Handle
260 fh => $state{fh},
261 ($scheme eq "https" ? (tls => "connect") : ());
262
263 # limit the number of persistent connections
264 if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) {
265 ++$KA_COUNT{$_[1]};
266 $state{handle}{ka_count_guard} = AnyEvent::Util::guard { --$KA_COUNT{$_[1]} };
267 $hdr{connection} = "keep-alive";
268 delete $hdr{connection}; # keep-alive not yet supported
269 } else {
270 delete $hdr{connection};
271 }
272
273 # (re-)configure handle
274 $state{handle}->timeout ($timeout);
275 $state{handle}->on_error (sub {
276 %state = ();
277 $cb->(undef, { Status => 599, Reason => "$!" });
278 });
279 $state{handle}->on_eof (sub {
280 %state = ();
281 $cb->(undef, { Status => 599, Reason => "unexpected end-of-file" });
282 });
283
284 # send request
285 $state{handle}->push_write (
286 "$method $rpath HTTP/1.0\015\012"
287 . (join "", map "$_: $hdr{$_}\015\012", keys %hdr)
288 . "\015\012"
289 . (delete $arg{body})
290 );
291
292 %hdr = (); # reduce memory usage, save a kitten
293
294 # status line
295 $state{handle}->push_read (line => qr/\015?\012/, sub {
296 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) \s+ ([^\015\012]+)/ix
297 or return (%state = (), $cb->(undef, { Status => 599, Reason => "invalid server response ($_[1])" }));
298
299 my %hdr = ( # response headers
300 HTTPVersion => "\x00$1",
301 Status => "\x00$2",
302 Reason => "\x00$3",
303 );
304
305 # headers, could be optimized a bit
306 $state{handle}->unshift_read (line => qr/\015?\012\015?\012/, sub {
307 for ("$_[1]\012") {
308 # we support spaces in field names, as lotus domino
309 # creates them.
310 $hdr{lc $1} .= "\x00$2"
311 while /\G
312 ([^:\000-\037]+):
313 [\011\040]*
314 ((?: [^\015\012]+ | \015?\012[\011\040] )*)
315 \015?\012
316 /gxc;
317
318 /\G$/
319 or return (%state = (), $cb->(undef, { Status => 599, Reason => "garbled response headers" }));
320 } 446 or do {
321 447 my $err = "$!";
322 substr $_, 0, 1, ""
323 for values %hdr;
324
325 my $finish = sub {
326 %state = (); 448 %state = ();
449 return $cb->(undef, { Status => 599, Reason => $err, URL => $url });
450 };
327 451
328 # set-cookie processing 452 pop; # free memory, save a tree
329 if ($arg{cookie_jar} && exists $hdr{"set-cookie"}) { 453
330 for (split /\x00/, $hdr{"set-cookie"}) { 454 return unless delete $state{connect_guard};
331 my ($cookie, @arg) = split /;\s*/; 455
332 my ($name, $value) = split /=/, $cookie, 2; 456 # get handle
333 my %kv = (value => $value, map { split /=/, $_, 2 } @arg); 457 $state{handle} = new AnyEvent::Handle
334 458 fh => $state{fh},
335 my $cdom = (delete $kv{domain}) || $uhost; 459 timeout => $timeout,
336 my $cpath = (delete $kv{path}) || "/"; 460 peername => $rhost,
337 461 tls_ctx => $arg{tls_ctx};
338 $cdom =~ s/^.?/./; # make sure it starts with a "." 462
339 463 # limit the number of persistent connections
340 my $ndots = $cdom =~ y/.//; 464 # keepalive not yet supported
341 next if $ndots < ($cdom =~ /[^.]{3}$/ ? 2 : 3); 465 if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) {
342 466 ++$KA_COUNT{$_[1]};
343 # store it 467 $state{handle}{ka_count_guard} = AnyEvent::Util::guard {
344 $arg{cookie_jar}{version} = 1; 468 --$KA_COUNT{$_[1]}
345 $arg{cookie_jar}{$cdom}{$cpath}{$name} = \%kv; 469 };
470 $hdr{connection} = "keep-alive";
471 } else {
472 delete $hdr{connection};
473 }
474
475 # (re-)configure handle
476 $state{handle}->on_error (sub {
477 %state = ();
478 $cb->(undef, { Status => 599, Reason => $_[2], URL => $url });
479 });
480 $state{handle}->on_eof (sub {
481 %state = ();
482 $cb->(undef, { Status => 599, Reason => "Unexpected end-of-file", URL => $url });
483 });
484
485 $state{handle}->starttls ("connect") if $rscheme eq "https";
486
487 # handle actual, non-tunneled, request
488 my $handle_actual_request = sub {
489 $state{handle}->starttls ("connect") if $uscheme eq "https" && !exists $state{handle}{tls};
490
491 # send request
492 $state{handle}->push_write (
493 "$method $rpath HTTP/1.0\015\012"
494 . (join "", map "\u$_: $hdr{$_}\015\012", grep defined $hdr{$_}, keys %hdr)
495 . "\015\012"
496 . (delete $arg{body})
497 );
498
499 %hdr = (); # reduce memory usage, save a kitten
500
501 # status line
502 $state{handle}->push_read (line => $qr_nl, sub {
503 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix
504 or return (%state = (), $cb->(undef, { Status => 599, Reason => "Invalid server response ($_[1])", URL => $url }));
505
506 my %hdr = ( # response headers
507 HTTPVersion => ",$1",
508 Status => ",$2",
509 Reason => ",$3",
510 URL => ",$url"
511 );
512
513 # headers, could be optimized a bit
514 $state{handle}->unshift_read (line => $qr_nlnl, sub {
515 for ("$_[1]") {
516 y/\015//d; # weed out any \015, as they show up in the weirdest of places.
517
518 # things seen, not parsed:
519 # p3pP="NON CUR OTPi OUR NOR UNI"
520
521 $hdr{lc $1} .= ",$2"
522 while /\G
523 ([^:\000-\037]*):
524 [\011\040]*
525 ((?: [^\012]+ | \012[\011\040] )*)
526 \012
527 /gxc;
528
529 /\G$/
530 or return (%state = (), $cb->(undef, { Status => 599, Reason => "Garbled response headers", URL => $url }));
346 } 531 }
532
533 substr $_, 0, 1, ""
534 for values %hdr;
535
536 # redirect handling
537 # microsoft and other shitheads don't give a shit for following standards,
538 # try to support some common forms of broken Location headers.
539 if ($hdr{location} !~ /^(?: $ | [^:\/?\#]+ : )/x) {
540 $hdr{location} =~ s/^\.\/+//;
541
542 my $url = "$rscheme://$uhost:$uport";
543
544 unless ($hdr{location} =~ s/^\///) {
545 $url .= $upath;
546 $url =~ s/\/[^\/]*$//;
547 }
548
549 $hdr{location} = "$url/$hdr{location}";
550 }
551
552 my $redirect;
553
554 if ($recurse) {
555 if ($hdr{Status} =~ /^30[12]$/ && $method ne "POST") {
556 # apparently, mozilla et al. just change POST to GET here
557 # more research is needed before we do the same
558 $redirect = 1;
559 } elsif ($hdr{Status} == 303) {
560 # even http/1.1 is unclear on how to mutate the method
561 $method = "GET" unless $method eq "HEAD";
562 $redirect = 1;
563 } elsif ($hdr{Status} == 307 && $method =~ /^(?:GET|HEAD)$/) {
564 $redirect = 1;
565 }
566 }
567
568 my $finish = sub {
569 $state{handle}->destroy if $state{handle};
570 %state = ();
571
572 # set-cookie processing
573 if ($arg{cookie_jar}) {
574 for ($_[1]{"set-cookie"}) {
575 # parse NAME=VALUE
576 my @kv;
577
578 while (/\G\s* ([^=;,[:space:]]+) \s*=\s* (?: "((?:[^\\"]+|\\.)*)" | ([^=;,[:space:]]*) )/gcxs) {
579 my $name = $1;
580 my $value = $3;
581
582 unless ($value) {
583 $value = $2;
584 $value =~ s/\\(.)/$1/gs;
585 }
586
587 push @kv, $name => $value;
588
589 last unless /\G\s*;/gc;
590 }
591
592 last unless @kv;
593
594 my $name = shift @kv;
595 my %kv = (value => shift @kv, @kv);
596
597 my $cdom;
598 my $cpath = (delete $kv{path}) || "/";
599
600 if (exists $kv{domain}) {
601 $cdom = delete $kv{domain};
602
603 $cdom =~ s/^\.?/./; # make sure it starts with a "."
604
605 next if $cdom =~ /\.$/;
606
607 # this is not rfc-like and not netscape-like. go figure.
608 my $ndots = $cdom =~ y/.//;
609 next if $ndots < ($cdom =~ /\.[^.][^.]\.[^.][^.]$/ ? 3 : 2);
610 } else {
611 $cdom = $uhost;
612 }
613
614 # store it
615 $arg{cookie_jar}{version} = 1;
616 $arg{cookie_jar}{$cdom}{$cpath}{$name} = \%kv;
617
618 redo if /\G\s*,/gc;
619 }
620 }
621
622 if ($redirect && exists $hdr{location}) {
623 # we ignore any errors, as it is very common to receive
624 # Content-Length != 0 but no actual body
625 # we also access %hdr, as $_[1] might be an erro
626 http_request ($method => $hdr{location}, %arg, recurse => $recurse - 1, $cb);
627 } else {
628 $cb->($_[0], $_[1]);
629 }
630 };
631
632 my $len = $hdr{"content-length"};
633
634 if (!$redirect && $arg{on_header} && !$arg{on_header}(\%hdr)) {
635 $finish->(undef, { Status => 598, Reason => "Request cancelled by on_header", URL => $url });
636 } elsif (
637 $hdr{Status} =~ /^(?:1..|[23]04)$/
638 or $method eq "HEAD"
639 or (defined $len && !$len)
640 ) {
641 # no body
642 $finish->("", \%hdr);
643 } else {
644 # body handling, four different code paths
645 # for want_body_handle, on_body (2x), normal (2x)
646 # we might read too much here, but it does not matter yet (no pers. connections)
647 if (!$redirect && $arg{want_body_handle}) {
648 $_[0]->on_eof (undef);
649 $_[0]->on_error (undef);
650 $_[0]->on_read (undef);
651
652 $finish->(delete $state{handle}, \%hdr);
653
654 } elsif ($arg{on_body}) {
655 $_[0]->on_error (sub { $finish->(undef, { Status => 599, Reason => $_[2], URL => $url }) });
656 if ($len) {
657 $_[0]->on_eof (undef);
658 $_[0]->on_read (sub {
659 $len -= length $_[0]{rbuf};
660
661 $arg{on_body}(delete $_[0]{rbuf}, \%hdr)
662 or $finish->(undef, { Status => 598, Reason => "Request cancelled by on_body", URL => $url });
663
664 $len > 0
665 or $finish->("", \%hdr);
666 });
667 } else {
668 $_[0]->on_eof (sub {
669 $finish->("", \%hdr);
670 });
671 $_[0]->on_read (sub {
672 $arg{on_body}(delete $_[0]{rbuf}, \%hdr)
673 or $finish->(undef, { Status => 598, Reason => "Request cancelled by on_body", URL => $url });
674 });
675 }
676 } else {
677 $_[0]->on_eof (undef);
678
679 if ($len) {
680 $_[0]->on_error (sub { $finish->(undef, { Status => 599, Reason => $_[2], URL => $url }) });
681 $_[0]->on_read (sub {
682 $finish->((substr delete $_[0]{rbuf}, 0, $len, ""), \%hdr)
683 if $len <= length $_[0]{rbuf};
684 });
685 } else {
686 $_[0]->on_error (sub {
687 $! == Errno::EPIPE || !$!
688 ? $finish->(delete $_[0]{rbuf}, \%hdr)
689 : $finish->(undef, { Status => 599, Reason => $_[2], URL => $url });
690 });
691 $_[0]->on_read (sub { });
692 }
693 }
694 }
695 });
696 });
697 };
698
699 # now handle proxy-CONNECT method
700 if ($proxy && $uscheme eq "https") {
701 # oh dear, we have to wrap it into a connect request
702
703 # maybe re-use $uauthority with patched port?
704 $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012Host: $uhost\015\012\015\012");
705 $state{handle}->push_read (line => $qr_nlnl, sub {
706 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix
707 or return (%state = (), $cb->(undef, { Status => 599, Reason => "Invalid proxy connect response ($_[1])", URL => $url }));
708
709 if ($2 == 200) {
710 $rpath = $upath;
711 &$handle_actual_request;
712 } else {
713 %state = ();
714 $cb->(undef, { Status => $2, Reason => $3, URL => $url });
347 } 715 }
348
349 if ($_[1]{Status} =~ /^x30[12]$/ && $recurse) {
350 # microsoft and other assholes don't give a shit for following standards,
351 # try to support a common form of broken Location header.
352 $_[1]{location} =~ s%^/%$scheme://$uhost:$uport/%;
353
354 http_request ($method, $_[1]{location}, %arg, recurse => $recurse - 1, $cb);
355 } else {
356 $cb->($_[0], $_[1]);
357 }
358 }; 716 });
359
360 if ($hdr{Status} =~ /^(?:1..|204|304)$/ or $method eq "HEAD") {
361 $finish->(undef, \%hdr);
362 } else { 717 } else {
363 if (exists $hdr{"content-length"}) { 718 &$handle_actual_request;
364 $_[0]->unshift_read (chunk => $hdr{"content-length"}, sub {
365 # could cache persistent connection now
366 if ($hdr{connection} =~ /\bkeep-alive\b/i) {
367 # but we don't, due to misdesigns, this is annoyingly complex
368 };
369
370 $finish->($_[1], \%hdr);
371 });
372 } else {
373 # too bad, need to read until we get an error or EOF,
374 # no way to detect winged data.
375 $_[0]->on_error (sub {
376 $finish->($_[0]{rbuf}, \%hdr);
377 });
378 $_[0]->on_eof (undef);
379 $_[0]->on_read (sub { });
380 }
381 }
382 }); 719 }
383 }); 720
384 }, sub { 721 }, $arg{on_prepare} || sub { $timeout };
385 $timeout
386 }; 722 };
387 723
388 defined wantarray && AnyEvent::Util::guard { %state = () } 724 defined wantarray && AnyEvent::Util::guard { %state = () }
389} 725}
390 726
391sub http_get($$;@) { 727sub http_get($@) {
392 unshift @_, "GET"; 728 unshift @_, "GET";
393 &http_request 729 &http_request
394} 730}
395 731
396sub http_head($$;@) { 732sub http_head($@) {
397 unshift @_, "HEAD"; 733 unshift @_, "HEAD";
398 &http_request 734 &http_request
399} 735}
400 736
401sub http_post($$$;@) { 737sub http_post($$@) {
738 my $url = shift;
402 unshift @_, "POST", "body"; 739 unshift @_, "POST", $url, "body";
403 &http_request 740 &http_request
404} 741}
405 742
406=back 743=back
407 744
410=over 4 747=over 4
411 748
412=item AnyEvent::HTTP::set_proxy "proxy-url" 749=item AnyEvent::HTTP::set_proxy "proxy-url"
413 750
414Sets the default proxy server to use. The proxy-url must begin with a 751Sets the default proxy server to use. The proxy-url must begin with a
415string of the form C<http://host:port> (optionally C<https:...>). 752string of the form C<http://host:port> (optionally C<https:...>), croaks
753otherwise.
754
755To clear an already-set proxy, use C<undef>.
416 756
417=item $AnyEvent::HTTP::MAX_RECURSE 757=item $AnyEvent::HTTP::MAX_RECURSE
418 758
419The default value for the C<recurse> request parameter (default: C<10>). 759The default value for the C<recurse> request parameter (default: C<10>).
420 760
421=item $AnyEvent::HTTP::USERAGENT 761=item $AnyEvent::HTTP::USERAGENT
422 762
423The default value for the C<User-Agent> header (the default is 763The default value for the C<User-Agent> header (the default is
424C<Mozilla/5.0 (compatible; AnyEvent::HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)>). 764C<Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)>).
425 765
426=item $AnyEvent::HTTP::MAX_PERSISTENT 766=item $AnyEvent::HTTP::MAX_PER_HOST
427 767
428The maximum number of persistent connections to keep open (default: 8). 768The maximum number of concurrent connections to the same host (identified
769by the hostname). If the limit is exceeded, then the additional requests
770are queued until previous connections are closed.
429 771
430Not implemented currently. 772The default value for this is C<4>, and it is highly advisable to not
773increase it.
431 774
432=item $AnyEvent::HTTP::PERSISTENT_TIMEOUT 775=item $AnyEvent::HTTP::ACTIVE
433 776
434The maximum time to cache a persistent connection, in seconds (default: 2). 777The number of active connections. This is not the number of currently
435 778running requests, but the number of currently open and non-idle TCP
436Not implemented currently. 779connections. This number of can be useful for load-leveling.
437 780
438=back 781=back
439 782
440=cut 783=cut
441 784
442sub set_proxy($) { 785sub set_proxy($) {
786 if (length $_[0]) {
443 $PROXY = [$2, $3 || 3128, $1] if $_[0] =~ m%^(https?):// ([^:/]+) (?: : (\d*) )?%ix; 787 $_[0] =~ m%^(https?):// ([^:/]+) (?: : (\d*) )?%ix
788 or Carp::croak "$_[0]: invalid proxy URL";
789 $PROXY = [$2, $3 || 3128, $1]
790 } else {
791 undef $PROXY;
792 }
444} 793}
445 794
446# initialise proxy from environment 795# initialise proxy from environment
796eval {
447set_proxy $ENV{http_proxy}; 797 set_proxy $ENV{http_proxy};
798};
448 799
449=head1 SEE ALSO 800=head1 SEE ALSO
450 801
451L<AnyEvent>. 802L<AnyEvent>.
452 803
453=head1 AUTHOR 804=head1 AUTHOR
454 805
455 Marc Lehmann <schmorp@schmorp.de> 806 Marc Lehmann <schmorp@schmorp.de>
456 http://home.schmorp.de/ 807 http://home.schmorp.de/
808
809With many thanks to Дмитрий Шалашов, who provided countless
810testcases and bugreports.
457 811
458=cut 812=cut
459 813
4601 8141
461 815

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines