… | |
… | |
61 | |
61 | |
62 | use common::sense; |
62 | use common::sense; |
63 | |
63 | |
64 | use Carp; |
64 | use Carp; |
65 | |
65 | |
66 | our $VERSION = '0.3'; |
66 | our $VERSION = 0.4; |
67 | |
67 | |
68 | use Scalar::Util (); |
68 | use Scalar::Util (); |
69 | |
69 | |
70 | use AnyEvent; |
70 | use AnyEvent; |
71 | use AnyEvent::Handle; |
71 | use AnyEvent::Handle; |
… | |
… | |
86 | 1 while s/([^_])(SVK|CHK|URI|FCP|DS|MIME|DDA)/$1\_$2/; |
86 | 1 while s/([^_])(SVK|CHK|URI|FCP|DS|MIME|DDA)/$1\_$2/; |
87 | s/(?<=[a-z])(?=[A-Z])/_/g; |
87 | s/(?<=[a-z])(?=[A-Z])/_/g; |
88 | lc |
88 | lc |
89 | } |
89 | } |
90 | |
90 | |
91 | =item $fcp = new AnyEvent::FCP [host => $host][, port => $port][, name => $name] |
91 | =item $fcp = new AnyEvent::FCP key => value...; |
92 | |
92 | |
93 | Create a new FCP connection to the given host and port (default |
93 | Create a new FCP connection to the given host and port (default |
94 | 127.0.0.1:9481, or the environment variables C<FREDHOST> and C<FREDPORT>). |
94 | 127.0.0.1:9481, or the environment variables C<FREDHOST> and C<FREDPORT>). |
95 | |
95 | |
96 | If no C<name> was specified, then AnyEvent::FCP will generate a |
96 | If no C<name> was specified, then AnyEvent::FCP will generate a |
97 | (hopefully) unique client name for you. |
97 | (hopefully) unique client name for you. |
|
|
98 | |
|
|
99 | The following keys can be specified (they are all optional): |
|
|
100 | |
|
|
101 | =over 4 |
|
|
102 | |
|
|
103 | =item name => $string |
|
|
104 | |
|
|
105 | A unique name to identify this client. If none is specified, a randomly |
|
|
106 | generated name will be used. |
|
|
107 | |
|
|
108 | =item host => $hostname |
|
|
109 | |
|
|
110 | The hostname or IP address of the freenet node. Default is C<$ENV{FREDHOST}> |
|
|
111 | or C<127.0.0.1>. |
|
|
112 | |
|
|
113 | =item port => $portnumber |
|
|
114 | |
|
|
115 | The port number of the FCP port. Default is C<$ENV{FREDPORT}> or C<9481>. |
|
|
116 | |
|
|
117 | =item timeout => $seconds |
|
|
118 | |
|
|
119 | The timeout, in seconds, after which a connection error is assumed when |
|
|
120 | there is no activity. Default is C<7200>, i.e. two hours. |
|
|
121 | |
|
|
122 | =item keepalive => $seconds |
|
|
123 | |
|
|
124 | The interval, in seconds, at which keepalive messages will be |
|
|
125 | sent. Default is C<540>, i.e. nine minutes. |
|
|
126 | |
|
|
127 | These keepalive messages are useful both to detect that a connection is |
|
|
128 | no longer working and to keep any (home) routers from expiring their |
|
|
129 | masquerading entry. |
|
|
130 | |
|
|
131 | =item on_error => $callback->($fcp, $message) |
|
|
132 | |
|
|
133 | Invoked on any (fatal) errors, such as unexpected connection close. The |
|
|
134 | callback receives the FCP object and a textual error message. |
|
|
135 | |
|
|
136 | =item on_failure => $callback->($fcp, $backtrace, $args, $error) |
|
|
137 | |
|
|
138 | Invoked when an FCP request fails that didn't have a failure callback. See |
|
|
139 | L<FCP REQUESTS> for details. |
|
|
140 | |
|
|
141 | =back |
98 | |
142 | |
99 | =cut |
143 | =cut |
100 | |
144 | |
101 | sub new { |
145 | sub new { |
102 | my $class = shift; |
146 | my $class = shift; |
… | |
… | |
105 | |
149 | |
106 | my $self = bless { |
150 | my $self = bless { |
107 | host => $ENV{FREDHOST} || "127.0.0.1", |
151 | host => $ENV{FREDHOST} || "127.0.0.1", |
108 | port => $ENV{FREDPORT} || 9481, |
152 | port => $ENV{FREDPORT} || 9481, |
109 | timeout => 3600 * 2, |
153 | timeout => 3600 * 2, |
|
|
154 | keepalive => 9 * 60, |
110 | name => time.rand.rand.rand, # lame |
155 | name => time.rand.rand.rand, # lame |
111 | @_, |
156 | @_, |
112 | queue => [], |
157 | queue => [], |
113 | req => {}, |
158 | req => {}, |
114 | prefix => "..:aefcpid:$rand:", |
159 | prefix => "..:aefcpid:$rand:", |
… | |
… | |
116 | }, $class; |
161 | }, $class; |
117 | |
162 | |
118 | { |
163 | { |
119 | Scalar::Util::weaken (my $self = $self); |
164 | Scalar::Util::weaken (my $self = $self); |
120 | |
165 | |
|
|
166 | $self->{kw} = AE::timer $self->{keepalive}, $self->{keepalive}, sub { |
|
|
167 | $self->{hdl}->push_write ("\n"); |
|
|
168 | }; |
|
|
169 | |
|
|
170 | our $ENDMESSAGE = qr<\012(EndMessage|Data)\012>; |
|
|
171 | |
|
|
172 | # these are declared here for performance reasons |
|
|
173 | my ($k, $v, $type); |
|
|
174 | my $rdata; |
|
|
175 | |
|
|
176 | my $on_read = sub { |
|
|
177 | my ($hdl) = @_; |
|
|
178 | |
|
|
179 | # we only carve out whole messages here |
|
|
180 | while ($hdl->{rbuf} =~ /\012(EndMessage|Data)\012/) { |
|
|
181 | # remember end marker |
|
|
182 | $rdata = $1 eq "Data" |
|
|
183 | or $1 eq "EndMessage" |
|
|
184 | or return $self->fatal ("protocol error, expected message end, got $1\n"); |
|
|
185 | |
|
|
186 | my @lines = split /\012/, substr $hdl->{rbuf}, 0, $-[0]; |
|
|
187 | |
|
|
188 | substr $hdl->{rbuf}, 0, $+[0], ""; # remove pkg |
|
|
189 | |
|
|
190 | $type = shift @lines; |
|
|
191 | $type = ($TOLC{$type} ||= tolc $type); |
|
|
192 | |
|
|
193 | my %kv; |
|
|
194 | |
|
|
195 | for (@lines) { |
|
|
196 | ($k, $v) = split /=/, $_, 2; |
|
|
197 | $k = ($TOLC{$k} ||= tolc $k); |
|
|
198 | |
|
|
199 | if ($k =~ /\./) { |
|
|
200 | # generic, slow case |
|
|
201 | my @k = split /\./, $k; |
|
|
202 | my $ro = \\%kv; |
|
|
203 | |
|
|
204 | while (@k) { |
|
|
205 | $k = shift @k; |
|
|
206 | if ($k =~ /^\d+$/) { |
|
|
207 | $ro = \$$ro->[$k]; |
|
|
208 | } else { |
|
|
209 | $ro = \$$ro->{$k}; |
|
|
210 | } |
|
|
211 | } |
|
|
212 | |
|
|
213 | $$ro = $v; |
|
|
214 | |
|
|
215 | next; |
|
|
216 | } |
|
|
217 | |
|
|
218 | # special comon case, for performance only |
|
|
219 | $kv{$k} = $v; |
|
|
220 | } |
|
|
221 | |
|
|
222 | if ($rdata) { |
|
|
223 | $_[0]->push_read (chunk => delete $kv{data_length}, sub { |
|
|
224 | $rdata = \$_[1]; |
|
|
225 | $self->recv ($type, \%kv, $rdata); |
|
|
226 | }); |
|
|
227 | |
|
|
228 | last; # do not tgry to parse more messages |
|
|
229 | } else { |
|
|
230 | $self->recv ($type, \%kv); |
|
|
231 | } |
|
|
232 | } |
|
|
233 | }; |
|
|
234 | |
121 | $self->{hdl} = new AnyEvent::Handle |
235 | $self->{hdl} = new AnyEvent::Handle |
122 | connect => [$self->{host} => $self->{port}], |
236 | connect => [$self->{host} => $self->{port}], |
123 | timeout => $self->{timeout}, |
237 | timeout => $self->{timeout}, |
|
|
238 | on_read => $on_read, |
|
|
239 | on_eof => $self->{on_eof}, |
124 | on_error => sub { |
240 | on_error => sub { |
125 | warn "@_\n";#d# |
241 | $self->fatal ($_[2]); |
126 | exit 1; |
|
|
127 | }, |
242 | }, |
128 | on_read => sub { $self->on_read (@_) }, |
243 | ; |
129 | on_eof => $self->{on_eof} || sub { }; |
|
|
130 | |
244 | |
131 | Scalar::Util::weaken ($self->{hdl}{fcp} = $self); |
245 | Scalar::Util::weaken ($self->{hdl}{fcp} = $self); |
132 | } |
246 | } |
133 | |
247 | |
134 | $self->send_msg (client_hello => |
248 | $self->send_msg (client_hello => |
135 | name => $self->{name}, |
249 | name => $self->{name}, |
136 | expected_version => "2.0", |
250 | expected_version => "2.0", |
137 | ); |
251 | ); |
138 | |
252 | |
139 | $self |
253 | $self |
|
|
254 | } |
|
|
255 | |
|
|
256 | sub fatal { |
|
|
257 | my ($self, $msg) = @_; |
|
|
258 | |
|
|
259 | $self->{hdl}->shutdown; |
|
|
260 | delete $self->{kw}; |
|
|
261 | |
|
|
262 | if ($self->{on_error}) { |
|
|
263 | $self->{on_error}->($self, $msg); |
|
|
264 | } else { |
|
|
265 | die $msg; |
|
|
266 | } |
140 | } |
267 | } |
141 | |
268 | |
142 | sub identifier { |
269 | sub identifier { |
143 | $_[0]{prefix} . ++$_[0]{idseq} |
270 | $_[0]{prefix} . ++$_[0]{idseq} |
144 | } |
271 | } |
… | |
… | |
253 | } else { |
380 | } else { |
254 | $self->default_recv ($type, $kv, @extra); |
381 | $self->default_recv ($type, $kv, @extra); |
255 | } |
382 | } |
256 | } |
383 | } |
257 | |
384 | |
258 | sub on_read { |
|
|
259 | my ($self) = @_; |
|
|
260 | |
|
|
261 | my ($k, $v, $type); |
|
|
262 | my %kv; |
|
|
263 | my $rdata; |
|
|
264 | |
|
|
265 | my $hdr_cb; $hdr_cb = sub { |
|
|
266 | if (($v = index $_[1], "=") >= 0) { |
|
|
267 | $k = substr $_[1], 0, $v; |
|
|
268 | $v = substr $_[1], $v + 1; |
|
|
269 | $k = ($TOLC{$k} ||= tolc $k); |
|
|
270 | |
|
|
271 | if ($k !~ /\./) { |
|
|
272 | # special case common case, for performance only |
|
|
273 | $kv{$k} = $v; |
|
|
274 | } else { |
|
|
275 | my @k = split /\./, $k; |
|
|
276 | my $ro = \\%kv; |
|
|
277 | |
|
|
278 | while (@k) { |
|
|
279 | $k = shift @k; |
|
|
280 | if ($k =~ /^\d+$/) { |
|
|
281 | $ro = \$$ro->[$k]; |
|
|
282 | } else { |
|
|
283 | $ro = \$$ro->{$k}; |
|
|
284 | } |
|
|
285 | } |
|
|
286 | |
|
|
287 | $$ro = $v; |
|
|
288 | } |
|
|
289 | |
|
|
290 | $_[0]->push_read (line => $hdr_cb); |
|
|
291 | } elsif ($_[1] eq "Data") { |
|
|
292 | $_[0]->push_read (chunk => delete $kv{data_length}, sub { |
|
|
293 | $rdata = \$_[1]; |
|
|
294 | $self->recv ($type, \%kv, $rdata); |
|
|
295 | }); |
|
|
296 | } elsif ($_[1] eq "EndMessage") { |
|
|
297 | $self->recv ($type, \%kv); |
|
|
298 | } else { |
|
|
299 | die "protocol error, expected message end, got $_[1]\n";#d# |
|
|
300 | } |
|
|
301 | }; |
|
|
302 | |
|
|
303 | $self->{hdl}->push_read (line => sub { |
|
|
304 | $type = ($TOLC{$_[1]} ||= tolc $_[1]); |
|
|
305 | $_[0]->push_read (line => $hdr_cb); |
|
|
306 | }); |
|
|
307 | } |
|
|
308 | |
|
|
309 | sub default_recv { |
385 | sub default_recv { |
310 | my ($self, $type, $kv, $rdata) = @_; |
386 | my ($self, $type, $kv, $rdata) = @_; |
311 | |
387 | |
312 | if ($type eq "node_hello") { |
388 | if ($type eq "node_hello") { |
313 | $self->{node_hello} = $kv; |
389 | $self->{node_hello} = $kv; |
… | |
… | |
350 | =over 4 |
426 | =over 4 |
351 | |
427 | |
352 | =item A code reference (or rather anything not matching some other alternative) |
428 | =item A code reference (or rather anything not matching some other alternative) |
353 | |
429 | |
354 | This code reference will be invoked with the result on success. On an |
430 | This code reference will be invoked with the result on success. On an |
|
|
431 | error, it will invoke the C<on_failure> callback of the FCP object, or, |
355 | error, it will die (in the event loop) with a backtrace of the call site. |
432 | if none was defined, will die (in the event loop) with a backtrace of the |
|
|
433 | call site. |
356 | |
434 | |
357 | This is a popular choice, but it makes handling errors hard - make sure |
435 | This is a popular choice, but it makes handling errors hard - make sure |
358 | you never generate protocol errors! |
436 | you never generate protocol errors! |
|
|
437 | |
|
|
438 | If an C<on_failure> hook exists, it will be invoked with the FCP object, |
|
|
439 | a (textual) backtrace as generated by C<Carp::longmess>, and arrayref |
|
|
440 | containing the arguments from the original request invocation and the |
|
|
441 | error object from the server, in this order, e.g.: |
|
|
442 | |
|
|
443 | on_failure => sub { |
|
|
444 | my ($fcp, $backtrace, $orig_args, $error_object) = @_; |
|
|
445 | ... |
|
|
446 | }, |
359 | |
447 | |
360 | =item A condvar (as returned by e.g. C<< AnyEvent->condvar >>) |
448 | =item A condvar (as returned by e.g. C<< AnyEvent->condvar >>) |
361 | |
449 | |
362 | When a condvar is passed, it is sent (C<< $cv->send ($results) >>) the |
450 | When a condvar is passed, it is sent (C<< $cv->send ($results) >>) the |
363 | results when the request has finished. Should an error occur, the error |
451 | results when the request has finished. Should an error occur, the error |
… | |
… | |
368 | =item An array with two callbacks C<[$success, $failure]> |
456 | =item An array with two callbacks C<[$success, $failure]> |
369 | |
457 | |
370 | The C<$success> callback will be invoked with the results, while the |
458 | The C<$success> callback will be invoked with the results, while the |
371 | C<$failure> callback will be invoked on any errors. |
459 | C<$failure> callback will be invoked on any errors. |
372 | |
460 | |
|
|
461 | The C<$failure> callback will be invoked with the error object from the |
|
|
462 | server. |
|
|
463 | |
373 | =item C<undef> |
464 | =item C<undef> |
374 | |
465 | |
375 | This is the same thing as specifying C<sub { }> as callback, i.e. on |
466 | This is the same thing as specifying C<sub { }> as callback, i.e. on |
376 | success, the results are ignored, while on failure, you the module dies |
467 | success, the results are ignored, while on failure, the C<on_failure> hook |
377 | with a backtrace. |
468 | is invoked or the module dies with a backtrace. |
378 | |
469 | |
379 | This is good for quick scripts, or when you really aren't interested in |
470 | This is good for quick scripts, or when you really aren't interested in |
380 | the results. |
471 | the results. |
381 | |
472 | |
382 | =back |
473 | =back |
… | |
… | |
402 | if (ARRAY:: eq ref $ok) { |
493 | if (ARRAY:: eq ref $ok) { |
403 | ($ok, $err) = @$ok; |
494 | ($ok, $err) = @$ok; |
404 | } elsif (UNIVERSAL::isa $ok, AnyEvent::CondVar::) { |
495 | } elsif (UNIVERSAL::isa $ok, AnyEvent::CondVar::) { |
405 | $err = sub { $ok->croak ($_[0]{extra_description}) }; |
496 | $err = sub { $ok->croak ($_[0]{extra_description}) }; |
406 | } else { |
497 | } else { |
407 | my $bt = Carp::longmess ""; |
498 | my $bt = Carp::longmess "AnyEvent::FCP request $name"; |
|
|
499 | Scalar::Util::weaken (my $self = $_[0]); |
|
|
500 | my $args = [@_]; shift @$args; |
408 | $err = sub { |
501 | $err = sub { |
|
|
502 | if ($self->{on_failure}) { |
|
|
503 | $self->{on_failure}($self, $args, $bt, $_[0]); |
|
|
504 | } else { |
409 | die "$_[0]{code_description} ($_[0]{extra_description})$bt"; |
505 | die "$_[0]{code_description} ($_[0]{extra_description})$bt"; |
|
|
506 | } |
410 | }; |
507 | }; |
411 | } |
508 | } |
412 | |
509 | |
413 | $ok ||= $NOP_CB; |
510 | $ok ||= $NOP_CB; |
414 | |
511 | |