ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/AnyEvent-MP/bin/aemp
Revision: 1.63
Committed: Tue Mar 27 00:41:36 2012 UTC (12 years, 2 months ago) by root
Branch: MAIN
Changes since 1.62: +1 -1 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 root 1.38 #!/opt/bin/perl
2 root 1.1
3     =head1 NAME
4    
5     aemp - AnyEvent:MP utility
6    
7     =head1 SYNOPSIS
8    
9     aemp command args...
10    
11     # protocol commands
12 root 1.19 aemp snd <port> <arg...> # send a message
13     aemp mon <port> # wait till port is killed
14 root 1.41 aemp cal <port> <arg...> # send message, append reply
15 root 1.19 aemp eval <node> <expr...> # evaluate expression
16 root 1.42 aemp shell [<node>] # run an interactive shell
17 root 1.37 aemp trace <node> # trace the network topology
18 root 1.8
19     # run a node
20 root 1.26 aemp run configure_args... # run a node
21 root 1.37 aemp restart <node> # restart a node if running under watchdog
22 root 1.8
23 root 1.23 # node configuration: node ID
24 root 1.19 aemp setnodeid <nodeid> # configure the real node id
25     aemp delnodeid # reset node id to default (= inherit)
26 root 1.1
27 root 1.47 # node configuration: authentication
28 root 1.19 aemp gensecret # generate a random shared secret
29     aemp setsecret <secret> # set the shared secret
30     aemp delsecret # remove the secret (= inherit)
31     aemp gencert # generate a random certificate
32     aemp setcert <file> # set a certificate (key.pem + certificate.pem)
33     aemp delcert # remove certificate (= inherit)
34 root 1.8
35 root 1.17 # node configuration: seed addresses for bootstrapping
36 root 1.19 aemp setseeds <host:port>,... # set seeds
37     aemp delseeds # clear all seeds (= inherit)
38     aemp addseed <host:port> # add a seed
39     aemp delseed <host:port> # remove seed
40 root 1.17
41     # node configuration: bind addresses
42 root 1.19 aemp setbinds <host:port>,... # set binds
43     aemp delbinds # clear all binds (= inherit)
44     aemp addbind <host:port> # add a bind address
45     aemp delbind <host:port> # remove a bind address
46 root 1.8
47     # node configuration: services
48 root 1.19 aemp setservices initfunc,... # set service functions
49     aemp delservices # clear all services (= inherit)
50     aemp addservice <initfunc> # add an instance of a service
51     aemp delservice <initfunc> # delete one instance of a service
52 root 1.6
53 root 1.47 # profile management
54 root 1.10 aemp profile <name> <command>... # apply command to profile only
55 root 1.26 aemp setparent <name> # specify a parent profile
56 root 1.25 aemp delparent # clear parent again
57 root 1.19 aemp delprofile <name> # eradicate the named profile
58     aemp showprofile <name> # display given profile
59 root 1.25 aemp showconfig <name> ... # display effective config
60 root 1.10
61 root 1.47 # node configuration: low-level protocol
62 root 1.54 aemp [set|del]secure <boolean>
63 root 1.47 aemp [set|del]monitor_timeout <seconds>
64     aemp [set|del]connect_interval <seconds>
65     aemp [set|del]framing_format [array]
66     aemp [set|del]auth_offer [array]
67     aemp [set|del]auth_accept [array]
68     aemp [set|del]autocork <boolean>
69     aemp [set|del]nodelay <boolean>
70    
71 root 1.1 =head1 DESCRIPTION
72    
73 root 1.8 With aemp you can configure various aspects of AnyEvent::MP and its
74 root 1.22 protocol, send various messages and even run a node.
75 root 1.1
76 root 1.23 The F<aemp> utility works like F<cvs>, F<svn> or other commands: the first
77     argument defines which operation (subcommand) is requested, after which
78     arguments for this operation are expected. When a subcommand does not eat
79     all remaining arguments, the remaining arguments will again be interpreted
80     as subcommand and so on.
81    
82     This means you can chain multiple commands, which is handy for profile
83     configuration, e.g.:
84    
85 root 1.27 aemp gensecret profile xyzzy binds 4040,4041 nodeid anon/
86    
87     Please note that all C<setxxx> subcommands have an alias without the
88     C<set> prefix.
89 root 1.23
90 root 1.34 All configuration data is stored in a human-readable (JSON) config file
91 root 1.46 stored in F<~/.perl-anyevent-mp> (or F<%appdata%/perl-anyevent-mp> on
92 root 1.49 loser systems, or wherever C<$ENV{PERL_ANYEVENT_MP_RC}> points to). Feel
93 root 1.46 free to look at it or edit it, the format is relatively simple.
94 root 1.34
95 root 1.35 =head2 SPECIFYING ARGUMENTS
96    
97     Arguments can be specified just as with any other shell command, with a
98     few special cases:
99    
100     If the I<first> argument starts with a literal C<[>-character, then it is
101     interpreted as a UTF-8 encoded JSON text. The resulting array replaces all
102     arguments.
103    
104     Otherwise, if I<any> argument starts with one of C<[>, C<{> or C<">, then
105     it is interpreted as UTF-8 encoded JSON text (or a single value in case of
106     C<">), and the resulting reference or scalar replaces the argument.
107    
108     This allows you, for example, to specify binds in F<aemp run> (using POSIX
109     shell syntax):
110    
111     aemp run binds '["*:4040"]'
112    
113 root 1.23 =head2 RUNNING A NODE
114    
115     This can be used to run a node - together with some services, this makes
116 elmex 1.32 it unnecessary to write any wrapper programs.
117 root 1.23
118     =over 4
119    
120 root 1.30 =item run <configure_args>...
121 root 1.23
122 root 1.26 Runs a node by calling C<AnyEvent::MP::Kernel::configure> with the given
123 root 1.23 arguments. The node runs under L<AnyEvent::Watchdog>, can be restarted
124 root 1.30 (and autorestarted, see the L<AnyEvent::Watchdog> manual). A very common
125 root 1.33 invocation is to just specify a profile using the profile name
126 root 1.30
127 root 1.33 aemp run database-backend
128 root 1.30
129     ... but you can use most arguments that C<configure> understands:
130    
131     aemp run nodeid mynode2 profile someprofile
132 root 1.23
133     Care has been taken to load (almost) no modules other than
134     L<AnyEvent::Watchdog> and the modules it loads, so everything (including
135     the L<AnyEvent::MP> modules themselves) will be freshly loaded on restart,
136     which makes upgrading everything except the perl binary easy.
137    
138 root 1.37 =item restart <node>
139    
140     Restarts the node using C<AnyEvent::Watchdog::Util::restart>. This works
141     for nodes started by C<aemp run>, but also for any other node that uses
142     L<AnyEvent::Watchdog>.
143    
144 root 1.23 =back
145    
146     =head2 PROTOCOL COMMANDS
147    
148     These commands actually communicate with other nodes. They all use a node
149 elmex 1.32 profile name of C<aemp> (specifying a default node ID of C<anon/> and a
150 root 1.25 binds list containing C<*:*> only).
151 root 1.23
152     They all use a timeout of five seconds, after which they give up.
153    
154     =over 4
155    
156     =item snd <port> <arguments...>
157    
158     Simply send a message to the given port - where you get the port ID from
159     is your problem.
160    
161     Exits after ensuring that the message has been delivered to its node.
162    
163 elmex 1.32 Most useful to take advantage of some undocumented functionality inside
164 root 1.23 nodes, such as node ports being able to call any method:
165    
166     aemp snd doomed AnyEvent::Watchdog::restart 1
167    
168 root 1.41 =item cal <port> <arg...>
169 root 1.23
170 root 1.41 Like F<aemp cal>: appends a local reply port to the message and waits for
171     a message to it.
172 root 1.23
173     Any return values will be JSON-encoded and printed separated by commas
174     (kind of like a JSON array without []-brackets).
175    
176 root 1.41 Example: ask the (undocumented) time service of a node for its current
177 root 1.23 time.
178    
179 root 1.41 aemp cal mynode time
180 root 1.23
181     =item mon <port>
182    
183     Monitors the port and exits when it's monitorign callback is called. Most
184     useful to monitor node ports.
185    
186     Example: monitor some node.
187    
188     aemp mon doomed
189    
190     =item eval <node> <expr...>
191    
192     Joins all remaining arguments into a string and evaluates it on the given
193 root 1.41 node. Return values are handled as with F<aemp cal>.
194 root 1.23
195     Example: find the unix process ID of the node called posicks.
196    
197     aemp eval posicks '$$'
198    
199     =item trace <node>
200    
201     Asks the given node for all currently connected nodes, then asks those
202     nodes for the same, thus tracing all node connections.
203    
204 root 1.31 =back
205 root 1.23
206     =head2 CONFIGURATION/NODE ID/SECRET/CERTIFICATE
207    
208     These commands deal with rather basic settings, the node ID, the shared
209     secret and the TLS certificate.
210    
211     =over 4
212    
213     =item setnodeid <nodeid>
214    
215 root 1.50 Set the node ID to the given string. If it ends with a slash (C</>), then
216     a random string is appended to make it unique.
217    
218     If no nodeid is specified in any profile, then the profile name, plus
219     appended slash, is used.
220 root 1.23
221     =item delnodeid
222    
223     Removes the node ID again, which means it is inherited again from it's
224     parent profile, or stays unset.
225    
226     =item gensecret
227    
228 root 1.51 Generates a random shared secret (currently 1071 bits) and sets it. The
229     shared secret is used to authenticate nodes to each other when TLS is not
230     required.
231 root 1.23
232     =item setsecret <secret>
233    
234 root 1.51 Sets the shared secret to the given string, which can be anything.
235 root 1.23
236     =item delsecret
237    
238     Removes the shared secret again, which means it is inherited again from
239     it's parent profile, or stays unset.
240    
241     =item gencert
242    
243 elmex 1.32 Generates a self-signed certificate and key, and sets it. This works
244 root 1.23 similarly to a shared secret: when all nodes have it, TLS will be used to
245     authenticate and encrypt all traffic.
246    
247     =item setcert <file>
248    
249     Set a node certificate (and optionally any CA certificates) from the given
250     file. The file must contain the key, followed by the certificate, followed
251     by any CA certificates you want to trust, all in PEM format.
252    
253     See L<AnyEvent::TLS> for some more details - this sets the C<cert> and
254     C<ca_cert> options.
255    
256     =item delcert
257    
258     Removes the certificate(s) again, which means it is inherited again from
259     it's parent profile, or stays unset.
260    
261     =back
262    
263     =head2 CONFIGURATION/SEEDS
264    
265     To discover the network you have to specify some seed addresses, which are
266     basically C<host:port> pairs where you expect some long-running nodes. It
267     does no harm to have a node as its own seed (they will eventually be
268     ignored).
269    
270 root 1.31 =over 4
271    
272 root 1.23 =item setseeds <host:port>,...
273    
274     Sets or replaces the list of seeds, which must be specified as a
275     comma-separated list of C<host:port> pairs. The C<host> can be a hostname,
276 root 1.36 an IP address, or C<*> to signify all local host addresses (which makes
277     little sense for seeds, outside some examples, but a lot of sense for
278     binds).
279 root 1.23
280     An empty list is allowed.
281    
282     Example: use C<doomed> with default port as only seednode.
283    
284     aemp setseeds doomed
285    
286     =item delseeds
287    
288     Removes the seed list again, which means it is inherited again from it's
289     parent profile, or stays unset.
290    
291     =item addseed <host:port>
292    
293     Adds a single seed address.
294    
295     =item delseed <host:port>
296    
297     Deletes the given seed address, if it exists.
298    
299     =back
300    
301     =head2 CONFIGURATION/BINDS
302    
303     To be able to be reached from other nodes, a node must I<bind> itself
304     to some listening socket(s). The list of these can either bs specified
305     manually, or AnyEvent::MP can guess them. Nodes without any binds are
306     possible to some extent.
307    
308     =over 4
309    
310     =item setbinds <host:port>,...
311    
312     Sets the list of bind addresses explicitly - see the F<aemp setseeds>
313 root 1.36 command for the exact syntax. In addition, a value of C<*> for the port,
314     or not specifying a port, means to use a dynamically-assigned port.
315 root 1.23
316 root 1.36 Note that the C<*>, C<*:*> or C<*:port> patterns are very useful here.
317 root 1.23
318 root 1.36 Example: bind on a ephemeral port on all local interfaces.
319 root 1.23
320     aemp setbinds "*"
321    
322     Example: bind on a random port on all local interfaces.
323    
324 root 1.25 aemp setbinds "*:*"
325 root 1.23
326     Example: resolve "doomed.mydomain" and try to bind on port C<4040> of all
327     IP addressess returned.
328    
329 root 1.36 aep setbinds doomed.mydomain:4040
330 root 1.23
331     =item delbinds
332    
333     Removes the bind list again, which means it is inherited again from it's
334     parent profile, or stays unset.
335    
336     =item addbind <host:port>
337    
338     Adds a single bind address.
339    
340     =item delbind <host:port>
341    
342     Deletes the given bind address, if it exists.
343    
344     =back
345    
346     =head2 CONFIGURATION/SERVICES
347    
348     Services are modules (or functions) that are automatically loaded (or
349     executed) when a node starts. They are especially useful when used in
350     conjunction with F<aemp run>, to configure which services a node should
351     run.
352    
353 root 1.47 Despite the daunting name, services really I<are> nothing more than a
354     module name or a function name with arguments. The "service" aspect comes
355     only from the behaviour of the module or function, which is supposed to
356     implement, well, some kind of service for the node, network etc.
357    
358     Instead of writing a standalone program for each different node type in
359     your network, you can simply put your code into a module, and then let the
360     configuration decide which node runs which "services". This also makes it
361     easy to combine multiple different services within the same node.
362    
363 root 1.23 =over 4
364    
365     =item setservices <initfunc>...
366    
367     Sets or replaces the list of services, which must be specified as a
368 root 1.37 comma-separated list or a JSON array.
369    
370     Each string entry in the list is interpreted as either a module name to
371     load (when it ends with C<::>) or a function to call (all other cases).
372    
373     Each entry which is an array itself (you need to use JSON format to
374     specify those) is interpreted as a function name and the arguments to
375     pass.
376 root 1.23
377 root 1.37 The algorithm to find the function is the same as used for C<<
378 root 1.23 L<AnyEvent::MP>::spawn >>.
379    
380 root 1.58 Example: run the global service.
381 root 1.37
382     aemp setservices AnyEvent::MP::Global::
383    
384     Example: call the mymod::myfun function with arguments 1, 2 and 3.
385    
386     aemp setservices '[["mymod::myfun", 1,2,3]]'
387    
388 root 1.23 =item delservices
389    
390     Removes the service list again, which means it is inherited again from
391     it's parent profile, or stays unset.
392    
393     =item addservice <initfunc>
394    
395     Adds a single service.
396    
397     =item delservice <initfunc>
398    
399     Deletes the given service, if it exists.
400    
401 root 1.58 =item seteval <string>
402    
403     Sometimes, all you need is to evaluate a small perl snippet to bring a
404     node up. This sets a perl string that is eval'ed after the node has been
405     configured.
406    
407     =item deleval
408    
409     Delete any eval string set with seteval.
410    
411 root 1.23 =back
412    
413     =head2 CONFIGURATION/PROFILE MANAGEMENT
414    
415     All the above configuration functions by default affect the I<global
416     default configuration>, which is basically used to augment every profile
417     and node configuration.
418    
419     =over 4
420    
421     =item profile <name> ...
422    
423     This subcommand makes the following subcommands act only on a specific
424     named profile, instead of on the global default. The profile is created if
425     necessary.
426    
427     Example: create a C<server> profile, give it a random node name, some seed
428     nodes and bind it on an unspecified port on all local interfaces. You
429     should add some services then and run the node...
430    
431 root 1.30 aemp profile server nodeid anon/ seeds doomed,10.0.0.2:5000 binds "*:*"
432 root 1.23
433     =item delprofile <name>
434    
435     Deletes the profile of the given name.
436    
437 root 1.26 =item setparent <name>
438 root 1.24
439     Sets the parent profile to use - values not specified in a profile will be
440     taken from the parent profile (even recursively, with the global default
441     config being the default parent). This is useful to configure profile
442     I<classes> and then to inherit from them for individual nodes.
443    
444     Note that you can specify circular parent chains and even a parent for the
445     global configuration. Neither will do you any good, however.
446    
447     Example: inherit all values not specified in the C<doomed> profile from
448     the C<server> profile.
449    
450 root 1.26 aemp profile doomed setparent server
451 root 1.24
452     =item delparent
453    
454     Removes the parent again from the profile, if any was set, so the profile
455     inherits directly from the global default config again.
456    
457 root 1.23 =item showprofile <name>
458    
459     Shows the values of the given profile, and only those, no inherited
460     values.
461    
462 root 1.25 =item showconfig <name> <key value...>
463 root 1.23
464 root 1.24 Shows the I<effective> config, i.e. the values as used by a node started
465 root 1.25 with the given profile name. Any additional key-value pairs specified
466 root 1.26 augment the configuration, just as with C<configure>.
467 root 1.24
468 root 1.25 If all arguments are omitted, show the global default config.
469 root 1.23
470     =back
471    
472 root 1.47 =head2 LOW-LEVEL TRANSPORT PROTOCOL
473    
474     The low-level transport protocol betwene two nodes also has a number of
475     configurable options, most of which should not be touched unless you know
476     what you are doing.
477    
478     =over 4
479    
480 root 1.54 =item [set|del]secure <boolean>
481    
482     Normally, nodes allow anything to be done to them by remote nodes,
483     including remotely-triggered execution of code.
484    
485     Sometimes a more secure mode is desired - this can be achieved by setting
486     the secure option to a true value.
487    
488 root 1.56 When secure mode is enabled, then the node will not execute code locally,
489     at least not via the normal node protocol. All other messages are still
490 root 1.62 allowed. This means remote nodes can monitor, kill or snd to local ports
491     (port names can also be easily guessed).
492 root 1.54
493 root 1.62 At the moment, this setting ignores C<snd_on>, C<eval_on> and C<spawn>
494     requests.
495 root 1.54
496 root 1.47 =item [set|del]monitor_timeout <seconds>
497    
498     Sets the default monitor timeout, that is, when a connection to a node
499     cannot be established within this many seconds, the node is declared
500     unreachable and all monitors will fire.
501    
502     C<30> seconds are usually a good time span for this.
503    
504     =item [set|del]connect_interval <seconds>
505    
506     When a connection cannot be established successfully within this many
507     seconds, try the next transport address (e.g. the next IP address). If
508     your nodes have a lot of transports, you might have to set this to a low
509     value so that they will actually all be tried within the monitor timeout
510     interval.
511    
512     C<2> is usually a good value, unless you live in new zealand.
513    
514     =item [set|del]framing_format [array]
515    
516     Configures the list of framing formats offered to the other side. This is
517     simply a list of formatted read/write types used with L<AnyEvent::Handle>,
518     in order of decreasing preference.
519    
520     Nodes support both C<json> and C<storable> framing formats for data
521     packets out of the box, and usually choose C<json> because it is first in
522     the list.
523    
524     Example: prefer the C<My::Personal::Format> framing format over JSON over
525     Storable.
526    
527     aemp setframing_format '["My::Personal::Format", "json", "storable"]'
528    
529     =item [set|del]auth_offer [array]
530    
531     Configures the list of authentication types that the node offers to the
532     other side as acceptable, in order of decreasing preference. Only auth
533     methods that the node can actually support will be offered.
534    
535     The default is '["tls_md6_64_256", "hmac_md6_64_256"]' and is usually good
536     enough.
537    
538     =item [set|del]auth_accept [array]
539    
540     Configures the list of authentication types that remote nodes can use to
541     authenticate, in order of decreasing preference.
542    
543     The default is '["tls_md6_64_256", "hmac_md6_64_256", "tls_anon",
544     "cleartext"]' and is usually good enough.
545    
546     =item [set|del]autocork <boolean>
547    
548     Sets the default C<autocork> option value for the L<AnyEvent::Handle>
549     object used by transports. By default, autocorking is off.
550    
551     =item [set|del]nodelay <boolean>
552    
553     Sets the default C<nodelay> option value for the L<AnyEvent::Handle>
554     object used by transports. By default, nodelay is on.
555    
556     =back
557    
558 root 1.1 =cut
559    
560     use common::sense;
561    
562 root 1.19 # should come before anything else, so all modules
563     # will be loaded on each restart
564 root 1.22 BEGIN {
565 root 1.35 if (@ARGV == 1 && $ARGV[0] =~ /^\[/) {
566     require JSON::XS;
567     @ARGV = @{ JSON::XS->new->utf8->decode (shift) };
568     } else {
569     for (@ARGV) {
570     if (/^[\[\{\"]/) {
571     require JSON::XS;
572     $_ = JSON::XS->new->utf8->allow_nonref->decode ($_);
573     }
574     }
575     }
576    
577 root 1.8 if ($ARGV[0] eq "run") {
578     shift;
579    
580 root 1.13 # d'oh
581 root 1.22 require AnyEvent::Watchdog;
582 root 1.48 # only now can we load additional modules
583    
584     require AnyEvent;
585 root 1.19
586 root 1.29 require AnyEvent::Watchdog::Util;
587     AnyEvent::Watchdog::Util::autorestart (1);
588     AnyEvent::Watchdog::Util::heartbeat (300);
589 root 1.13
590 root 1.25 require AnyEvent::MP::Kernel;
591 root 1.26 AnyEvent::MP::Kernel::configure (@ARGV);
592 root 1.8
593 root 1.16 AnyEvent::detect () eq "AnyEvent::Impl::EV"
594 root 1.8 ? EV::loop ()
595     : AE::cv ()->recv;
596     }
597 root 1.22 }
598 root 1.8
599 root 1.1 use Carp ();
600    
601 root 1.17 use JSON::XS;
602    
603 root 1.1 use AnyEvent;
604 root 1.4 use AnyEvent::Util;
605    
606 root 1.8 use AnyEvent::MP;
607 root 1.1 use AnyEvent::MP::Config;
608    
609     sub my_run_cmd {
610     my ($cmd) = @_;
611    
612     my $cv = &run_cmd;
613     my $status = $cv->recv;
614    
615     $status
616     and die "@$cmd: command failed with exit status $status.";
617     }
618    
619     sub gen_cert {
620 root 1.2 my_run_cmd [qw(openssl req
621     -new -nodes -x509 -days 3650
622     -newkey rsa:2048 -keyout /dev/fd/3
623     -batch -subj /CN=AnyEvent::MP
624     )],
625 root 1.5 "<", "/dev/null",
626 root 1.1 ">" , \my $cert,
627     "3>", \my $key,
628 root 1.4 "2>", "/dev/null";
629 root 1.1
630     "$cert$key"
631     }
632    
633 root 1.25 sub init {
634 root 1.53 configure profile => "aemp", nodeid => "aemp/%n/%u";
635 root 1.25 }
636    
637 root 1.10 our $cfg = AnyEvent::MP::Config::config;
638 root 1.8 our $profile = $cfg;
639 root 1.1
640 root 1.11 sub trace {
641 root 1.56 my ($seed) = @_;
642 root 1.11 my $cv = AE::cv;
643     my %seen;
644 root 1.57 my $exit;
645 root 1.11
646 root 1.57 my %to;
647 root 1.11
648 root 1.25 init;
649 root 1.11
650     my $reply = port {
651 root 1.55 my ($node, undef, @neigh) = @_;
652 root 1.11
653 root 1.57 delete $to{$node};
654    
655 root 1.12 @neigh = grep $_ ne $NODE, @neigh;
656    
657 root 1.55 print $node, " -> ", (join " ", @neigh), "\n";
658 root 1.12
659 root 1.57 for my $neigh (@neigh) {
660     unless ($seen{$neigh}++) {
661 root 1.11 $cv->begin;
662 root 1.57 $to{$neigh} = AE::timer 15, 0, sub {
663     print "$neigh (timeout)\n";
664     $exit = 1;
665     $cv->end;
666     };
667     AnyEvent::MP::Kernel::eval_on $neigh, "AnyEvent::MP::Kernel::up_nodes" => $SELF => $neigh;
668 root 1.11 }
669     }
670    
671     $cv->end;
672     };
673    
674     $cv->begin;
675 root 1.56 snd $reply, seed => undef, $seed;
676 root 1.11
677     $cv->recv;
678 root 1.57
679     exit $exit;
680 root 1.11 }
681    
682 root 1.42 sub shell {
683     init;
684    
685     my $node = shift @ARGV || $NODE;
686     $| = 1;
687    
688     print <<EOF;
689     Entering interactive shell - no commandline editing of course (use rlfe etc.).
690    
691 root 1.43 \= display a list of nodes
692     \=name switch to another node
693 root 1.51 package P switch to package P when evaluating
694     \$ECHO contains the name of a port that echos everything sent to it
695 root 1.42
696     EOF
697 root 1.59 my $json = JSON::XS->new->pretty->ascii;
698 root 1.52 my $pkg = "AnyEvent::MP::Kernel";
699 root 1.42 my $cv = AE::cv;
700 root 1.51 my $echo = port {
701 root 1.61 print "\nECHO<$AnyEvent::MP::Kernel::SRCNODE> ", $json->encode (\@_), "\n$node $pkg> ";
702 root 1.51 };
703 root 1.52 print "$node $pkg> ";
704 root 1.42 my $t = AE::io *STDIN, 0, sub {
705     chomp (my $line = <STDIN>);
706    
707     if ($line =~ s/^=//) {
708     if (length $line) {
709     $node = $line;
710     } else {
711 root 1.61 db_keys "'l" => sub {
712     print "\nnodes: ", (join " ", sort @{ $_[0] }), "\n$node $pkg> ";
713     };
714 root 1.42 }
715 root 1.51 } elsif ($line =~ /^\s*package\s+(\S+)\s*;?\s*$/) {
716 root 1.52 $pkg = $1;
717 root 1.51 } elsif ($line =~ /\S/) {
718 root 1.42 my $time = AE::time;
719 root 1.52 AnyEvent::MP::Kernel::eval_on
720     $node,
721     "package $pkg; my \$ECHO = '$echo'; $line",
722     port {
723     kil $SELF;
724    
725     my ($err, @res) = @_;
726    
727     $time = AE::time - $time;
728    
729 root 1.60 print "\n$node: $line\n";
730     printf "%0.3fs\n", $time;
731 root 1.52 if (length $err) {
732 root 1.60 print "$err @res";
733 root 1.52 } else {
734 root 1.60 print $json->encode(\@res);
735 root 1.52 }
736 root 1.60 print "\n$node $pkg> ";
737 root 1.42 }
738 root 1.52 ;
739 root 1.42 }
740    
741 root 1.52 print "$node $pkg> ";
742 root 1.42 };
743     $cv->recv;
744     }
745    
746 root 1.37 sub node_eval {
747     my ($node, $expr) = @_;
748    
749     init;
750    
751     my $cv = AE::cv;
752     my $to = AE::timer 5, 0, sub { exit 1 };
753     AnyEvent::MP::Kernel::eval_on $node, $expr, port { &$cv };
754     mon $node, $cv;
755    
756     my ($err, @res) = $cv->recv;
757    
758 root 1.39 die "$err @res" if length $err;
759 root 1.37
760     print +(substr JSON::XS->new->encode (\@res), 1, -1), "\n";
761     }
762    
763 root 1.10 sub docmd;
764    
765 root 1.1 our %CMD = (
766     snd => sub {
767 root 1.17 my $port = shift @ARGV;
768 root 1.25 init;
769 root 1.1
770 root 1.7 snd $port, @ARGV; @ARGV = ();
771 root 1.1
772     my $cv = AE::cv;
773 root 1.28 my $to = AE::timer 5, 0, sub { exit 1 };
774 root 1.1 mon $port, $cv;
775 root 1.19 my $reply = port sub { &$cv };
776     snd node_of $port, snd => $reply, "message sent successfully";
777 root 1.1
778     print join " ", $cv->recv, "\n";
779     },
780    
781 root 1.41 cal => sub {
782 root 1.17 my $port = shift @ARGV;
783 root 1.25 init;
784 root 1.7
785     my $cv = AE::cv;
786 root 1.41 cal $port, @ARGV, sub { &$cv }; @ARGV = ();
787 root 1.7
788 root 1.23 print +(substr JSON::XS->new->encode ([$cv->recv]), 1, -1), "\n";
789 root 1.7 },
790    
791 root 1.1 mon => sub {
792 root 1.17 my $port = shift @ARGV;
793 root 1.25 init;
794 root 1.1
795     mon $port, my $cv = AE::cv;
796     print join " ", $cv->recv, "\n";
797     },
798    
799 root 1.19 eval => sub {
800     my $node = node_of shift @ARGV;
801     my $expr = join " ", @ARGV; @ARGV = ();
802 root 1.37 node_eval $node, $expr;
803 root 1.19 },
804    
805 root 1.42 shell => \&shell,
806    
807 root 1.11 trace => sub {
808     @ARGV >= 1
809 root 1.17 or die "node id missing\n";
810 root 1.11
811 root 1.17 trace shift @ARGV;
812 root 1.11 },
813 root 1.37 restart => sub {
814     my $node = node_of shift @ARGV;
815 root 1.44 node_eval $node, 'my $w; $w = AE::idle sub { '
816     . 'undef $w; '
817     . 'use AnyEvent::Watchdog::Util ();'
818     . 'AnyEvent::Watchdog::Util::restart'
819     . '}; ()';
820 root 1.37 },
821 root 1.11
822 root 1.17 setnodeid => sub {
823 root 1.8 @ARGV >= 1
824     or die "shared secret missing\n";
825    
826 root 1.17 $profile->{nodeid} = shift @ARGV;
827 root 1.8 ++$cfg->{dirty};
828     },
829 root 1.17 delnodeid => sub {
830     delete $profile->{nodeid};
831 root 1.8 ++$cfg->{dirty};
832     },
833    
834 root 1.1 setsecret => sub {
835 root 1.8 @ARGV >= 1
836 root 1.1 or die "shared secret missing\n";
837    
838 root 1.8 $profile->{secret} = shift @ARGV;
839 root 1.1 ++$cfg->{dirty};
840     },
841     gensecret => sub {
842 root 1.51 $profile->{secret} = AnyEvent::MP::Kernel::nonce62 180; # ~1071 bits
843 root 1.1 ++$cfg->{dirty};
844     },
845 root 1.17 delsecret => sub {
846 root 1.8 delete $profile->{secret};
847 root 1.1 ++$cfg->{dirty};
848     },
849    
850     setcert => sub {
851 root 1.8 @ARGV >= 1
852 root 1.1 or die "key+certificate pem filename missing\n";
853    
854 elmex 1.45 my $certfile = shift @ARGV;
855    
856     open my $fh, "<", $certfile
857     or die "$certfile: $!";
858 root 1.1
859     local $/;
860 root 1.8 $profile->{cert} = <$fh>;
861 root 1.1 ++$cfg->{dirty};
862     },
863     gencert => sub {
864 root 1.8 $profile->{cert} = gen_cert;
865 root 1.1 ++$cfg->{dirty};
866     },
867 root 1.17 delcert => sub {
868 root 1.8 delete $profile->{cert};
869 root 1.1 ++$cfg->{dirty};
870     },
871 root 1.6
872 root 1.17 setbinds => sub {
873 root 1.19 @ARGV >= 1
874     or die "bind addresses missing\n";
875 root 1.38 my $list = shift @ARGV;
876     $profile->{binds} = ref $list ? $list : [split /,/, $list];
877 root 1.17 ++$cfg->{dirty};
878     },
879     delbinds => sub {
880     delete $profile->{binds};
881     ++$cfg->{dirty};
882     },
883     addbind => sub {
884     @ARGV >= 1
885     or die "bind address missing\n";
886     my $bind = shift @ARGV;
887    
888     @{ $profile->{binds} } = grep $_ ne $bind, @{ $profile->{binds} };
889     push @{ $profile->{binds} }, $bind;
890     ++$cfg->{dirty};
891     },
892     delbind => sub {
893     @ARGV >= 1
894     or die "bind address missing\n";
895     my $bind = shift @ARGV;
896    
897     @{ $profile->{binds} } = grep $_ ne $bind, @{ $profile->{binds} };
898     ++$cfg->{dirty};
899     },
900    
901 root 1.6 setseeds => sub {
902 root 1.19 @ARGV >= 1
903     or die "seed addresses missing\n";
904 root 1.38 my $list = shift @ARGV;
905     $profile->{seeds} = ref $list ? $list : [split /,/, $list];
906 root 1.6 ++$cfg->{dirty};
907     },
908 root 1.17 delseeds => sub {
909     delete $profile->{seeds};
910     ++$cfg->{dirty};
911     },
912 root 1.6 addseed => sub {
913 root 1.10 @ARGV >= 1
914 root 1.17 or die "seed address missing\n";
915 root 1.6 my $seed = shift @ARGV;
916 root 1.10
917 root 1.8 @{ $profile->{seeds} } = grep $_ ne $seed, @{ $profile->{seeds} };
918     push @{ $profile->{seeds} }, $seed;
919     ++$cfg->{dirty};
920     },
921     delseed => sub {
922 root 1.10 @ARGV >= 1
923 root 1.17 or die "seed address missing\n";
924 root 1.8 my $seed = shift @ARGV;
925 root 1.10
926 root 1.8 @{ $profile->{seeds} } = grep $_ ne $seed, @{ $profile->{seeds} };
927     ++$cfg->{dirty};
928     },
929    
930     setservices => sub {
931 root 1.19 @ARGV >= 1
932     or die "service specifications missing\n";
933 root 1.38 my $list = shift @ARGV;
934     $profile->{services} = ref $list ? $list : [split /,/, $list];
935 root 1.6 ++$cfg->{dirty};
936     },
937 root 1.17 delservices => sub {
938     delete $profile->{services};
939     ++$cfg->{dirty};
940     },
941 root 1.9 addservice => sub {
942 root 1.10 @ARGV >= 1
943     or die "service specification missing\n";
944 root 1.8 my $service = shift @ARGV;
945     push @{ $profile->{services} }, $service;
946     ++$cfg->{dirty};
947     },
948 root 1.9 delservice => sub {
949 root 1.10 @ARGV >= 1
950     or die "service specification missing\n";
951 root 1.8 my $service = shift @ARGV;
952     for (0 .. $#{ $profile->{services} }) {
953     next unless $profile->{services}[$_] eq $service;
954     splice @{ $profile->{services} }, $_, 1;
955     last;
956     }
957 root 1.6 ++$cfg->{dirty};
958     },
959 root 1.58 seteval => sub {
960     @ARGV >= 1
961     or die "eval string missing\n";
962    
963     $profile->{eval} = shift @ARGV;
964     ++$cfg->{dirty};
965     },
966     deleval => sub {
967     delete $profile->{eval};
968     ++$cfg->{dirty};
969     },
970 root 1.10
971     profile => sub {
972 root 1.19 @ARGV >= 1
973     or die "profile name is missing\n";
974 root 1.10 my $name = shift @ARGV;
975    
976     $profile = $cfg->{profile}{$name} ||= {};
977 root 1.14 ++$cfg->{dirty};
978 root 1.10 },
979     delprofile => sub {
980     @ARGV >= 1
981     or die "profile name is missing\n";
982     my $name = shift @ARGV;
983    
984     delete $cfg->{profile}{$name};
985 root 1.14 ++$cfg->{dirty};
986 root 1.10 },
987 root 1.26 setparent => sub {
988 root 1.24 @ARGV >= 1
989     or die "profile name is missing\n";
990    
991     $profile->{parent} = shift @ARGV;
992     ++$cfg->{dirty};
993     },
994     delparent => sub {
995     delete $profile->{parent};
996     ++$cfg->{dirty};
997     },
998 root 1.17 showprofile => sub {
999     @ARGV >= 1
1000     or die "profile name is missing\n";
1001     my $name = shift @ARGV;
1002    
1003     print JSON::XS->new->pretty->encode ($cfg->{profile}{$name} || {});
1004     },
1005     showconfig => sub {
1006 root 1.53 my $name = @ARGV ? shift @ARGV : AnyEvent::MP::Kernel::nodename;
1007 root 1.17
1008 root 1.25 my $profile = AnyEvent::MP::Config::find_profile $name, @ARGV;
1009     @ARGV = ();
1010 root 1.24
1011     # make it look nicer:
1012     delete $profile->{profile};
1013     delete $profile->{parent};
1014    
1015     print JSON::XS->new->pretty->encode ($profile);
1016 root 1.17 },
1017 root 1.1 );
1018    
1019 root 1.47 for my $attr (qw(
1020     monitor_timeout connect_interval framing_format auth_offer
1021 root 1.54 auth_accept autocork nodelay secure
1022 root 1.47 )) {
1023     $CMD{"set$attr"} = sub {
1024     @ARGV >= 1
1025     or die "$attr value is missing\n";
1026    
1027     $profile->{$attr} = shift @ARGV;
1028     ++$cfg->{dirty};
1029     };
1030     $CMD{"del$attr"} = sub {
1031     delete $profile->{$attr};
1032     ++$cfg->{dirty};
1033     };
1034     }
1035    
1036 root 1.27 for (keys %CMD) {
1037 root 1.63 $CMD{$1} ||= $CMD{$_} if /^set(.*)$/;
1038 root 1.27 }
1039    
1040 root 1.1 sub docmd {
1041     my $cmd = shift @ARGV;
1042    
1043     $CMD{$cmd}
1044 root 1.23 or die "$cmd: no such aemp command (try perldoc aemp, or man aemp)";
1045 root 1.1
1046     $CMD{$cmd}();
1047     }
1048    
1049     @ARGV
1050 root 1.23 or die "Usage: aemp subcommand ... (try perldoc aemp, or man aemp)\n";
1051 root 1.1
1052 root 1.19 docmd while @ARGV;
1053 root 1.1
1054