Change the 'This Node' column of the machinereadable output for 'ctdb status -Y'...
[obnox/ctdb.git] / tools / ctdb.c
1 /* 
2    ctdb control tool
3
4    Copyright (C) Andrew Tridgell  2007
5    Copyright (C) Ronnie Sahlberg  2007
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 3 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, see <http://www.gnu.org/licenses/>.
19 */
20
21 #include "includes.h"
22 #include "lib/tevent/tevent.h"
23 #include "system/time.h"
24 #include "system/filesys.h"
25 #include "system/network.h"
26 #include "system/locale.h"
27 #include "popt.h"
28 #include "cmdline.h"
29 #include "../include/ctdb.h"
30 #include "../include/ctdb_client.h"
31 #include "../include/ctdb_private.h"
32 #include "../common/rb_tree.h"
33 #include "db_wrap.h"
34
35 #define ERR_TIMEOUT     20      /* timed out trying to reach node */
36 #define ERR_NONODE      21      /* node does not exist */
37 #define ERR_DISNODE     22      /* node is disconnected */
38
39 struct ctdb_connection *ctdb_connection;
40
41 static void usage(void);
42
43 static struct {
44         int timelimit;
45         uint32_t pnn;
46         int machinereadable;
47         int verbose;
48         int maxruntime;
49 } options;
50
51 #define TIMELIMIT() timeval_current_ofs(options.timelimit, 0)
52 #define LONGTIMELIMIT() timeval_current_ofs(options.timelimit*10, 0)
53
54 #ifdef CTDB_VERS
55 static int control_version(struct ctdb_context *ctdb, int argc, const char **argv)
56 {
57 #define STR(x) #x
58 #define XSTR(x) STR(x)
59         printf("CTDB version: %s\n", XSTR(CTDB_VERS));
60         return 0;
61 }
62 #endif
63
64
65 /*
66   verify that a node exists and is reachable
67  */
68 static void verify_node(struct ctdb_context *ctdb)
69 {
70         int ret;
71         struct ctdb_node_map *nodemap=NULL;
72
73         if (options.pnn == CTDB_CURRENT_NODE) {
74                 return;
75         }
76         if (options.pnn == CTDB_BROADCAST_ALL) {
77                 return;
78         }
79
80         /* verify the node exists */
81         if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
82                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
83                 exit(10);
84         }
85         if (options.pnn >= nodemap->num) {
86                 DEBUG(DEBUG_ERR, ("Node %u does not exist\n", options.pnn));
87                 exit(ERR_NONODE);
88         }
89         if (nodemap->nodes[options.pnn].flags & NODE_FLAGS_DELETED) {
90                 DEBUG(DEBUG_ERR, ("Node %u is DELETED\n", options.pnn));
91                 exit(ERR_DISNODE);
92         }
93         if (nodemap->nodes[options.pnn].flags & NODE_FLAGS_DISCONNECTED) {
94                 DEBUG(DEBUG_ERR, ("Node %u is DISCONNECTED\n", options.pnn));
95                 exit(ERR_DISNODE);
96         }
97
98         /* verify we can access the node */
99         ret = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), options.pnn);
100         if (ret == -1) {
101                 DEBUG(DEBUG_ERR,("Can not access node. Node is not operational.\n"));
102                 exit(10);
103         }
104 }
105
106 /*
107  check if a database exists
108 */
109 static int db_exists(struct ctdb_context *ctdb, const char *db_name, bool *persistent)
110 {
111         int i, ret;
112         struct ctdb_dbid_map *dbmap=NULL;
113
114         ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, ctdb, &dbmap);
115         if (ret != 0) {
116                 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n", options.pnn));
117                 return -1;
118         }
119
120         for(i=0;i<dbmap->num;i++){
121                 const char *name;
122
123                 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &name);
124                 if (!strcmp(name, db_name)) {
125                         if (persistent) {
126                                 *persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
127                         }
128                         return 0;
129                 }
130         }
131
132         return -1;
133 }
134
135 /*
136   see if a process exists
137  */
138 static int control_process_exists(struct ctdb_context *ctdb, int argc, const char **argv)
139 {
140         uint32_t pnn, pid;
141         int ret;
142         if (argc < 1) {
143                 usage();
144         }
145
146         if (sscanf(argv[0], "%u:%u", &pnn, &pid) != 2) {
147                 DEBUG(DEBUG_ERR, ("Badly formed pnn:pid\n"));
148                 return -1;
149         }
150
151         ret = ctdb_ctrl_process_exists(ctdb, pnn, pid);
152         if (ret == 0) {
153                 printf("%u:%u exists\n", pnn, pid);
154         } else {
155                 printf("%u:%u does not exist\n", pnn, pid);
156         }
157         return ret;
158 }
159
160 /*
161   display statistics structure
162  */
163 static void show_statistics(struct ctdb_statistics *s, int show_header)
164 {
165         TALLOC_CTX *tmp_ctx = talloc_new(NULL);
166         int i;
167         const char *prefix=NULL;
168         int preflen=0;
169         int tmp, days, hours, minutes, seconds;
170         const struct {
171                 const char *name;
172                 uint32_t offset;
173         } fields[] = {
174 #define STATISTICS_FIELD(n) { #n, offsetof(struct ctdb_statistics, n) }
175                 STATISTICS_FIELD(num_clients),
176                 STATISTICS_FIELD(frozen),
177                 STATISTICS_FIELD(recovering),
178                 STATISTICS_FIELD(num_recoveries),
179                 STATISTICS_FIELD(client_packets_sent),
180                 STATISTICS_FIELD(client_packets_recv),
181                 STATISTICS_FIELD(node_packets_sent),
182                 STATISTICS_FIELD(node_packets_recv),
183                 STATISTICS_FIELD(keepalive_packets_sent),
184                 STATISTICS_FIELD(keepalive_packets_recv),
185                 STATISTICS_FIELD(node.req_call),
186                 STATISTICS_FIELD(node.reply_call),
187                 STATISTICS_FIELD(node.req_dmaster),
188                 STATISTICS_FIELD(node.reply_dmaster),
189                 STATISTICS_FIELD(node.reply_error),
190                 STATISTICS_FIELD(node.req_message),
191                 STATISTICS_FIELD(node.req_control),
192                 STATISTICS_FIELD(node.reply_control),
193                 STATISTICS_FIELD(client.req_call),
194                 STATISTICS_FIELD(client.req_message),
195                 STATISTICS_FIELD(client.req_control),
196                 STATISTICS_FIELD(timeouts.call),
197                 STATISTICS_FIELD(timeouts.control),
198                 STATISTICS_FIELD(timeouts.traverse),
199                 STATISTICS_FIELD(total_calls),
200                 STATISTICS_FIELD(pending_calls),
201                 STATISTICS_FIELD(lockwait_calls),
202                 STATISTICS_FIELD(pending_lockwait_calls),
203                 STATISTICS_FIELD(childwrite_calls),
204                 STATISTICS_FIELD(pending_childwrite_calls),
205                 STATISTICS_FIELD(memory_used),
206                 STATISTICS_FIELD(max_hop_count),
207         };
208         tmp = s->statistics_current_time.tv_sec - s->statistics_start_time.tv_sec;
209         seconds = tmp%60;
210         tmp    /= 60;
211         minutes = tmp%60;
212         tmp    /= 60;
213         hours   = tmp%24;
214         tmp    /= 24;
215         days    = tmp;
216
217         if (options.machinereadable){
218                 if (show_header) {
219                         printf("CTDB version:");
220                         printf("Current time of statistics:");
221                         printf("Statistics collected since:");
222                         for (i=0;i<ARRAY_SIZE(fields);i++) {
223                                 printf("%s:", fields[i].name);
224                         }
225                         printf("num_reclock_ctdbd_latency:");
226                         printf("min_reclock_ctdbd_latency:");
227                         printf("avg_reclock_ctdbd_latency:");
228                         printf("max_reclock_ctdbd_latency:");
229
230                         printf("num_reclock_recd_latency:");
231                         printf("min_reclock_recd_latency:");
232                         printf("avg_reclock_recd_latency:");
233                         printf("max_reclock_recd_latency:");
234
235                         printf("num_call_latency:");
236                         printf("min_call_latency:");
237                         printf("avg_call_latency:");
238                         printf("max_call_latency:");
239
240                         printf("num_lockwait_latency:");
241                         printf("min_lockwait_latency:");
242                         printf("avg_lockwait_latency:");
243                         printf("max_lockwait_latency:");
244
245                         printf("num_childwrite_latency:");
246                         printf("min_childwrite_latency:");
247                         printf("avg_childwrite_latency:");
248                         printf("max_childwrite_latency:");
249                         printf("\n");
250                 }
251                 printf("%d:", CTDB_VERSION);
252                 printf("%d:", (int)s->statistics_current_time.tv_sec);
253                 printf("%d:", (int)s->statistics_start_time.tv_sec);
254                 for (i=0;i<ARRAY_SIZE(fields);i++) {
255                         printf("%d:", *(uint32_t *)(fields[i].offset+(uint8_t *)s));
256                 }
257                 printf("%d:", s->reclock.ctdbd.num);
258                 printf("%.6f:", s->reclock.ctdbd.min);
259                 printf("%.6f:", s->reclock.ctdbd.num?s->reclock.ctdbd.total/s->reclock.ctdbd.num:0.0);
260                 printf("%.6f:", s->reclock.ctdbd.max);
261
262                 printf("%d:", s->reclock.recd.num);
263                 printf("%.6f:", s->reclock.recd.min);
264                 printf("%.6f:", s->reclock.recd.num?s->reclock.recd.total/s->reclock.recd.num:0.0);
265                 printf("%.6f:", s->reclock.recd.max);
266
267                 printf("%d:", s->call_latency.num);
268                 printf("%.6f:", s->call_latency.min);
269                 printf("%.6f:", s->call_latency.num?s->call_latency.total/s->call_latency.num:0.0);
270                 printf("%.6f:", s->call_latency.max);
271
272                 printf("%d:", s->lockwait_latency.num);
273                 printf("%.6f:", s->lockwait_latency.min);
274                 printf("%.6f:", s->lockwait_latency.num?s->lockwait_latency.total/s->lockwait_latency.num:0.0);
275                 printf("%.6f:", s->lockwait_latency.max);
276
277                 printf("%d:", s->childwrite_latency.num);
278                 printf("%.6f:", s->childwrite_latency.min);
279                 printf("%.6f:", s->childwrite_latency.num?s->childwrite_latency.total/s->childwrite_latency.num:0.0);
280                 printf("%.6f:", s->childwrite_latency.max);
281                 printf("\n");
282         } else {
283                 printf("CTDB version %u\n", CTDB_VERSION);
284                 printf("Current time of statistics  :                %s", ctime(&s->statistics_current_time.tv_sec));
285                 printf("Statistics collected since  : (%03d %02d:%02d:%02d) %s", days, hours, minutes, seconds, ctime(&s->statistics_start_time.tv_sec));
286
287                 for (i=0;i<ARRAY_SIZE(fields);i++) {
288                         if (strchr(fields[i].name, '.')) {
289                                 preflen = strcspn(fields[i].name, ".")+1;
290                                 if (!prefix || strncmp(prefix, fields[i].name, preflen) != 0) {
291                                         prefix = fields[i].name;
292                                         printf(" %*.*s\n", preflen-1, preflen-1, fields[i].name);
293                                 }
294                         } else {
295                                 preflen = 0;
296                         }
297                         printf(" %*s%-22s%*s%10u\n", 
298                                preflen?4:0, "",
299                                fields[i].name+preflen, 
300                                preflen?0:4, "",
301                                *(uint32_t *)(fields[i].offset+(uint8_t *)s));
302                 }
303                 printf(" %-30s     %.6f/%.6f/%.6f sec out of %d\n", "reclock_ctdbd       MIN/AVG/MAX", s->reclock.ctdbd.min, s->reclock.ctdbd.num?s->reclock.ctdbd.total/s->reclock.ctdbd.num:0.0, s->reclock.ctdbd.max, s->reclock.ctdbd.num);
304
305                 printf(" %-30s     %.6f/%.6f/%.6f sec out of %d\n", "reclock_recd       MIN/AVG/MAX", s->reclock.recd.min, s->reclock.recd.num?s->reclock.recd.total/s->reclock.recd.num:0.0, s->reclock.recd.max, s->reclock.recd.num);
306
307                 printf(" %-30s     %.6f/%.6f/%.6f sec out of %d\n", "call_latency       MIN/AVG/MAX", s->call_latency.min, s->call_latency.num?s->call_latency.total/s->call_latency.num:0.0, s->call_latency.max, s->call_latency.num);
308                 printf(" %-30s     %.6f/%.6f/%.6f sec out of %d\n", "lockwait_latency   MIN/AVG/MAX", s->lockwait_latency.min, s->lockwait_latency.num?s->lockwait_latency.total/s->lockwait_latency.num:0.0, s->lockwait_latency.max, s->lockwait_latency.num);
309                 printf(" %-30s     %.6f/%.6f/%.6f sec out of %d\n", "childwrite_latency MIN/AVG/MAX", s->childwrite_latency.min, s->childwrite_latency.num?s->childwrite_latency.total/s->childwrite_latency.num:0.0, s->childwrite_latency.max, s->childwrite_latency.num);
310         }
311
312         talloc_free(tmp_ctx);
313 }
314
315 /*
316   display remote ctdb statistics combined from all nodes
317  */
318 static int control_statistics_all(struct ctdb_context *ctdb)
319 {
320         int ret, i;
321         struct ctdb_statistics statistics;
322         uint32_t *nodes;
323         uint32_t num_nodes;
324
325         nodes = ctdb_get_connected_nodes(ctdb, TIMELIMIT(), ctdb, &num_nodes);
326         CTDB_NO_MEMORY(ctdb, nodes);
327         
328         ZERO_STRUCT(statistics);
329
330         for (i=0;i<num_nodes;i++) {
331                 struct ctdb_statistics s1;
332                 int j;
333                 uint32_t *v1 = (uint32_t *)&s1;
334                 uint32_t *v2 = (uint32_t *)&statistics;
335                 uint32_t num_ints = 
336                         offsetof(struct ctdb_statistics, __last_counter) / sizeof(uint32_t);
337                 ret = ctdb_ctrl_statistics(ctdb, nodes[i], &s1);
338                 if (ret != 0) {
339                         DEBUG(DEBUG_ERR, ("Unable to get statistics from node %u\n", nodes[i]));
340                         return ret;
341                 }
342                 for (j=0;j<num_ints;j++) {
343                         v2[j] += v1[j];
344                 }
345                 statistics.max_hop_count = 
346                         MAX(statistics.max_hop_count, s1.max_hop_count);
347                 statistics.call_latency.max = 
348                         MAX(statistics.call_latency.max, s1.call_latency.max);
349                 statistics.lockwait_latency.max = 
350                         MAX(statistics.lockwait_latency.max, s1.lockwait_latency.max);
351         }
352         talloc_free(nodes);
353         printf("Gathered statistics for %u nodes\n", num_nodes);
354         show_statistics(&statistics, 1);
355         return 0;
356 }
357
358 /*
359   display remote ctdb statistics
360  */
361 static int control_statistics(struct ctdb_context *ctdb, int argc, const char **argv)
362 {
363         int ret;
364         struct ctdb_statistics statistics;
365
366         if (options.pnn == CTDB_BROADCAST_ALL) {
367                 return control_statistics_all(ctdb);
368         }
369
370         ret = ctdb_ctrl_statistics(ctdb, options.pnn, &statistics);
371         if (ret != 0) {
372                 DEBUG(DEBUG_ERR, ("Unable to get statistics from node %u\n", options.pnn));
373                 return ret;
374         }
375         show_statistics(&statistics, 1);
376         return 0;
377 }
378
379
380 /*
381   reset remote ctdb statistics
382  */
383 static int control_statistics_reset(struct ctdb_context *ctdb, int argc, const char **argv)
384 {
385         int ret;
386
387         ret = ctdb_statistics_reset(ctdb, options.pnn);
388         if (ret != 0) {
389                 DEBUG(DEBUG_ERR, ("Unable to reset statistics on node %u\n", options.pnn));
390                 return ret;
391         }
392         return 0;
393 }
394
395
396 /*
397   display remote ctdb rolling statistics
398  */
399 static int control_stats(struct ctdb_context *ctdb, int argc, const char **argv)
400 {
401         int ret;
402         struct ctdb_statistics_wire *stats;
403         int i, num_records = -1;
404
405         if (argc ==1) {
406                 num_records = atoi(argv[0]) - 1;
407         }
408
409         ret = ctdb_ctrl_getstathistory(ctdb, TIMELIMIT(), options.pnn, ctdb, &stats);
410         if (ret != 0) {
411                 DEBUG(DEBUG_ERR, ("Unable to get rolling statistics from node %u\n", options.pnn));
412                 return ret;
413         }
414         for (i=0;i<stats->num;i++) {
415                 if (stats->stats[i].statistics_start_time.tv_sec == 0) {
416                         continue;
417                 }
418                 show_statistics(&stats->stats[i], i==0);
419                 if (i == num_records) {
420                         break;
421                 }
422         }
423         return 0;
424 }
425
426
427 /*
428   display uptime of remote node
429  */
430 static int control_uptime(struct ctdb_context *ctdb, int argc, const char **argv)
431 {
432         int ret;
433         struct ctdb_uptime *uptime = NULL;
434         int tmp, days, hours, minutes, seconds;
435
436         ret = ctdb_ctrl_uptime(ctdb, ctdb, TIMELIMIT(), options.pnn, &uptime);
437         if (ret != 0) {
438                 DEBUG(DEBUG_ERR, ("Unable to get uptime from node %u\n", options.pnn));
439                 return ret;
440         }
441
442         if (options.machinereadable){
443                 printf(":Current Node Time:Ctdb Start Time:Last Recovery/Failover Time:Last Recovery/IPFailover Duration:\n");
444                 printf(":%u:%u:%u:%lf\n",
445                         (unsigned int)uptime->current_time.tv_sec,
446                         (unsigned int)uptime->ctdbd_start_time.tv_sec,
447                         (unsigned int)uptime->last_recovery_finished.tv_sec,
448                         timeval_delta(&uptime->last_recovery_finished,
449                                       &uptime->last_recovery_started)
450                 );
451                 return 0;
452         }
453
454         printf("Current time of node          :                %s", ctime(&uptime->current_time.tv_sec));
455
456         tmp = uptime->current_time.tv_sec - uptime->ctdbd_start_time.tv_sec;
457         seconds = tmp%60;
458         tmp    /= 60;
459         minutes = tmp%60;
460         tmp    /= 60;
461         hours   = tmp%24;
462         tmp    /= 24;
463         days    = tmp;
464         printf("Ctdbd start time              : (%03d %02d:%02d:%02d) %s", days, hours, minutes, seconds, ctime(&uptime->ctdbd_start_time.tv_sec));
465
466         tmp = uptime->current_time.tv_sec - uptime->last_recovery_finished.tv_sec;
467         seconds = tmp%60;
468         tmp    /= 60;
469         minutes = tmp%60;
470         tmp    /= 60;
471         hours   = tmp%24;
472         tmp    /= 24;
473         days    = tmp;
474         printf("Time of last recovery/failover: (%03d %02d:%02d:%02d) %s", days, hours, minutes, seconds, ctime(&uptime->last_recovery_finished.tv_sec));
475         
476         printf("Duration of last recovery/failover: %lf seconds\n",
477                 timeval_delta(&uptime->last_recovery_finished,
478                               &uptime->last_recovery_started));
479
480         return 0;
481 }
482
483 /*
484   show the PNN of the current node
485  */
486 static int control_pnn(struct ctdb_context *ctdb, int argc, const char **argv)
487 {
488         uint32_t mypnn;
489         bool ret;
490
491         ret = ctdb_getpnn(ctdb_connection, options.pnn, &mypnn);
492         if (!ret) {
493                 DEBUG(DEBUG_ERR, ("Unable to get pnn from node."));
494                 return -1;
495         }
496
497         printf("PNN:%d\n", mypnn);
498         return 0;
499 }
500
501
502 struct pnn_node {
503         struct pnn_node *next;
504         const char *addr;
505         int pnn;
506 };
507
508 static struct pnn_node *read_nodes_file(TALLOC_CTX *mem_ctx)
509 {
510         const char *nodes_list;
511         int nlines;
512         char **lines;
513         int i, pnn;
514         struct pnn_node *pnn_nodes = NULL;
515         struct pnn_node *pnn_node;
516         struct pnn_node *tmp_node;
517
518         /* read the nodes file */
519         nodes_list = getenv("CTDB_NODES");
520         if (nodes_list == NULL) {
521                 nodes_list = "/etc/ctdb/nodes";
522         }
523         lines = file_lines_load(nodes_list, &nlines, mem_ctx);
524         if (lines == NULL) {
525                 return NULL;
526         }
527         while (nlines > 0 && strcmp(lines[nlines-1], "") == 0) {
528                 nlines--;
529         }
530         for (i=0, pnn=0; i<nlines; i++) {
531                 char *node;
532
533                 node = lines[i];
534                 /* strip leading spaces */
535                 while((*node == ' ') || (*node == '\t')) {
536                         node++;
537                 }
538                 if (*node == '#') {
539                         pnn++;
540                         continue;
541                 }
542                 if (strcmp(node, "") == 0) {
543                         continue;
544                 }
545                 pnn_node = talloc(mem_ctx, struct pnn_node);
546                 pnn_node->pnn = pnn++;
547                 pnn_node->addr = talloc_strdup(pnn_node, node);
548                 pnn_node->next = pnn_nodes;
549                 pnn_nodes = pnn_node;
550         }
551
552         /* swap them around so we return them in incrementing order */
553         pnn_node = pnn_nodes;
554         pnn_nodes = NULL;
555         while (pnn_node) {
556                 tmp_node = pnn_node;
557                 pnn_node = pnn_node->next;
558
559                 tmp_node->next = pnn_nodes;
560                 pnn_nodes = tmp_node;
561         }
562
563         return pnn_nodes;
564 }
565
566 /*
567   show the PNN of the current node
568   discover the pnn by loading the nodes file and try to bind to all
569   addresses one at a time until the ip address is found.
570  */
571 static int control_xpnn(struct ctdb_context *ctdb, int argc, const char **argv)
572 {
573         TALLOC_CTX *mem_ctx = talloc_new(NULL);
574         struct pnn_node *pnn_nodes;
575         struct pnn_node *pnn_node;
576
577         pnn_nodes = read_nodes_file(mem_ctx);
578         if (pnn_nodes == NULL) {
579                 DEBUG(DEBUG_ERR,("Failed to read nodes file\n"));
580                 talloc_free(mem_ctx);
581                 return -1;
582         }
583
584         for(pnn_node=pnn_nodes;pnn_node;pnn_node=pnn_node->next) {
585                 ctdb_sock_addr addr;
586
587                 if (parse_ip(pnn_node->addr, NULL, 63999, &addr) == 0) {
588                         DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s' in nodes file\n", pnn_node->addr));
589                         talloc_free(mem_ctx);
590                         return -1;
591                 }
592
593                 if (ctdb_sys_have_ip(&addr)) {
594                         printf("PNN:%d\n", pnn_node->pnn);
595                         talloc_free(mem_ctx);
596                         return 0;
597                 }
598         }
599
600         printf("Failed to detect which PNN this node is\n");
601         talloc_free(mem_ctx);
602         return -1;
603 }
604
605 /*
606   display remote ctdb status
607  */
608 static int control_status(struct ctdb_context *ctdb, int argc, const char **argv)
609 {
610         int i, ret;
611         struct ctdb_vnn_map *vnnmap=NULL;
612         struct ctdb_node_map *nodemap=NULL;
613         uint32_t recmode, recmaster;
614         int mypnn;
615
616         mypnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), options.pnn);
617         if (mypnn == -1) {
618                 return -1;
619         }
620
621         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb, &nodemap);
622         if (ret != 0) {
623                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
624                 return ret;
625         }
626
627         if (options.machinereadable) {
628                 printf(":Node:IP:Disconnected:Banned:Disabled:Unhealthy:Stopped"
629                        ":Inactive:PartiallyOnline:ThisNode:\n");
630                 for (i=0;i<nodemap->num;i++) {
631                         int partially_online = 0;
632                         int j;
633
634                         if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
635                                 continue;
636                         }
637                         if (nodemap->nodes[i].flags == 0) {
638                                 struct ctdb_control_get_ifaces *ifaces;
639
640                                 ret = ctdb_ctrl_get_ifaces(ctdb, TIMELIMIT(),
641                                                            nodemap->nodes[i].pnn,
642                                                            ctdb, &ifaces);
643                                 if (ret == 0) {
644                                         for (j=0; j < ifaces->num; j++) {
645                                                 if (ifaces->ifaces[j].link_state != 0) {
646                                                         continue;
647                                                 }
648                                                 partially_online = 1;
649                                                 break;
650                                         }
651                                         talloc_free(ifaces);
652                                 }
653                         }
654                         printf(":%d:%s:%d:%d:%d:%d:%d:%d:%d:%c:\n", nodemap->nodes[i].pnn,
655                                 ctdb_addr_to_str(&nodemap->nodes[i].addr),
656                                !!(nodemap->nodes[i].flags&NODE_FLAGS_DISCONNECTED),
657                                !!(nodemap->nodes[i].flags&NODE_FLAGS_BANNED),
658                                !!(nodemap->nodes[i].flags&NODE_FLAGS_PERMANENTLY_DISABLED),
659                                !!(nodemap->nodes[i].flags&NODE_FLAGS_UNHEALTHY),
660                                !!(nodemap->nodes[i].flags&NODE_FLAGS_STOPPED),
661                                !!(nodemap->nodes[i].flags&NODE_FLAGS_INACTIVE),
662                                partially_online,
663                                (nodemap->nodes[i].pnn == mypnn)?'Y':'N');
664                 }
665                 return 0;
666         }
667
668         printf("Number of nodes:%d\n", nodemap->num);
669         for(i=0;i<nodemap->num;i++){
670                 static const struct {
671                         uint32_t flag;
672                         const char *name;
673                 } flag_names[] = {
674                         { NODE_FLAGS_DISCONNECTED,          "DISCONNECTED" },
675                         { NODE_FLAGS_PERMANENTLY_DISABLED,  "DISABLED" },
676                         { NODE_FLAGS_BANNED,                "BANNED" },
677                         { NODE_FLAGS_UNHEALTHY,             "UNHEALTHY" },
678                         { NODE_FLAGS_DELETED,               "DELETED" },
679                         { NODE_FLAGS_STOPPED,               "STOPPED" },
680                         { NODE_FLAGS_INACTIVE,              "INACTIVE" },
681                 };
682                 char *flags_str = NULL;
683                 int j;
684
685                 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
686                         continue;
687                 }
688                 if (nodemap->nodes[i].flags == 0) {
689                         struct ctdb_control_get_ifaces *ifaces;
690
691                         ret = ctdb_ctrl_get_ifaces(ctdb, TIMELIMIT(),
692                                                    nodemap->nodes[i].pnn,
693                                                    ctdb, &ifaces);
694                         if (ret == 0) {
695                                 for (j=0; j < ifaces->num; j++) {
696                                         if (ifaces->ifaces[j].link_state != 0) {
697                                                 continue;
698                                         }
699                                         flags_str = talloc_strdup(ctdb, "PARTIALLYONLINE");
700                                         break;
701                                 }
702                                 talloc_free(ifaces);
703                         }
704                 }
705                 for (j=0;j<ARRAY_SIZE(flag_names);j++) {
706                         if (nodemap->nodes[i].flags & flag_names[j].flag) {
707                                 if (flags_str == NULL) {
708                                         flags_str = talloc_strdup(ctdb, flag_names[j].name);
709                                 } else {
710                                         flags_str = talloc_asprintf_append(flags_str, "|%s",
711                                                                            flag_names[j].name);
712                                 }
713                                 CTDB_NO_MEMORY_FATAL(ctdb, flags_str);
714                         }
715                 }
716                 if (flags_str == NULL) {
717                         flags_str = talloc_strdup(ctdb, "OK");
718                         CTDB_NO_MEMORY_FATAL(ctdb, flags_str);
719                 }
720                 printf("pnn:%d %-16s %s%s\n", nodemap->nodes[i].pnn,
721                        ctdb_addr_to_str(&nodemap->nodes[i].addr),
722                        flags_str,
723                        nodemap->nodes[i].pnn == mypnn?" (THIS NODE)":"");
724                 talloc_free(flags_str);
725         }
726
727         ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn, ctdb, &vnnmap);
728         if (ret != 0) {
729                 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n", options.pnn));
730                 return ret;
731         }
732         if (vnnmap->generation == INVALID_GENERATION) {
733                 printf("Generation:INVALID\n");
734         } else {
735                 printf("Generation:%d\n",vnnmap->generation);
736         }
737         printf("Size:%d\n",vnnmap->size);
738         for(i=0;i<vnnmap->size;i++){
739                 printf("hash:%d lmaster:%d\n", i, vnnmap->map[i]);
740         }
741
742         if (!ctdb_getrecmode(ctdb_connection, options.pnn, &recmode)) {
743                 DEBUG(DEBUG_ERR, ("Unable to get recmode from node %u\n", options.pnn));
744                 return -1;
745         }
746         printf("Recovery mode:%s (%d)\n",recmode==CTDB_RECOVERY_NORMAL?"NORMAL":"RECOVERY",recmode);
747
748         if (!ctdb_getrecmaster(ctdb_connection, options.pnn, &recmaster)) {
749                 DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
750                 return -1;
751         }
752         printf("Recovery master:%d\n",recmaster);
753
754         return 0;
755 }
756
757
758 struct natgw_node {
759         struct natgw_node *next;
760         const char *addr;
761 };
762
763 /*
764   display the list of nodes belonging to this natgw configuration
765  */
766 static int control_natgwlist(struct ctdb_context *ctdb, int argc, const char **argv)
767 {
768         int i, ret;
769         uint32_t capabilities;
770         const char *natgw_list;
771         int nlines;
772         char **lines;
773         struct natgw_node *natgw_nodes = NULL;
774         struct natgw_node *natgw_node;
775         struct ctdb_node_map *nodemap=NULL;
776
777
778         /* read the natgw nodes file into a linked list */
779         natgw_list = getenv("NATGW_NODES");
780         if (natgw_list == NULL) {
781                 natgw_list = "/etc/ctdb/natgw_nodes";
782         }
783         lines = file_lines_load(natgw_list, &nlines, ctdb);
784         if (lines == NULL) {
785                 ctdb_set_error(ctdb, "Failed to load natgw node list '%s'\n", natgw_list);
786                 return -1;
787         }
788         while (nlines > 0 && strcmp(lines[nlines-1], "") == 0) {
789                 nlines--;
790         }
791         for (i=0;i<nlines;i++) {
792                 char *node;
793
794                 node = lines[i];
795                 /* strip leading spaces */
796                 while((*node == ' ') || (*node == '\t')) {
797                         node++;
798                 }
799                 if (*node == '#') {
800                         continue;
801                 }
802                 if (strcmp(node, "") == 0) {
803                         continue;
804                 }
805                 natgw_node = talloc(ctdb, struct natgw_node);
806                 natgw_node->addr = talloc_strdup(natgw_node, node);
807                 CTDB_NO_MEMORY(ctdb, natgw_node->addr);
808                 natgw_node->next = natgw_nodes;
809                 natgw_nodes = natgw_node;
810         }
811
812         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
813         if (ret != 0) {
814                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node.\n"));
815                 return ret;
816         }
817
818         i=0;
819         while(i<nodemap->num) {
820                 for(natgw_node=natgw_nodes;natgw_node;natgw_node=natgw_node->next) {
821                         if (!strcmp(natgw_node->addr, ctdb_addr_to_str(&nodemap->nodes[i].addr))) {
822                                 break;
823                         }
824                 }
825
826                 /* this node was not in the natgw so we just remove it from
827                  * the list
828                  */
829                 if ((natgw_node == NULL) 
830                 ||  (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) ) {
831                         int j;
832
833                         for (j=i+1; j<nodemap->num; j++) {
834                                 nodemap->nodes[j-1] = nodemap->nodes[j];
835                         }
836                         nodemap->num--;
837                         continue;
838                 }
839
840                 i++;
841         }               
842
843         /* pick a node to be natgwmaster
844          * we dont allow STOPPED, DELETED, BANNED or UNHEALTHY nodes to become the natgwmaster
845          */
846         for(i=0;i<nodemap->num;i++){
847                 if (!(nodemap->nodes[i].flags & (NODE_FLAGS_DISCONNECTED|NODE_FLAGS_STOPPED|NODE_FLAGS_DELETED|NODE_FLAGS_BANNED|NODE_FLAGS_UNHEALTHY))) {
848                         ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, &capabilities);
849                         if (ret != 0) {
850                                 DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", nodemap->nodes[i].pnn));
851                                 return ret;
852                         }
853                         if (!(capabilities&CTDB_CAP_NATGW)) {
854                                 continue;
855                         }
856                         printf("%d %s\n", nodemap->nodes[i].pnn,ctdb_addr_to_str(&nodemap->nodes[i].addr));
857                         break;
858                 }
859         }
860         /* we couldnt find any healthy node, try unhealthy ones */
861         if (i == nodemap->num) {
862                 for(i=0;i<nodemap->num;i++){
863                         if (!(nodemap->nodes[i].flags & (NODE_FLAGS_DISCONNECTED|NODE_FLAGS_STOPPED|NODE_FLAGS_DELETED))) {
864                                 ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, &capabilities);
865                                 if (ret != 0) {
866                                         DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", nodemap->nodes[i].pnn));
867                                         return ret;
868                                 }
869                                 if (!(capabilities&CTDB_CAP_NATGW)) {
870                                         continue;
871                                 }
872                                 printf("%d %s\n", nodemap->nodes[i].pnn,ctdb_addr_to_str(&nodemap->nodes[i].addr));
873                                 break;
874                         }
875                 }
876         }
877         /* unless all nodes are STOPPED, when we pick one anyway */
878         if (i == nodemap->num) {
879                 for(i=0;i<nodemap->num;i++){
880                         if (!(nodemap->nodes[i].flags & (NODE_FLAGS_DISCONNECTED|NODE_FLAGS_DELETED))) {
881                                 ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, &capabilities);
882                                 if (ret != 0) {
883                                         DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", nodemap->nodes[i].pnn));
884                                         return ret;
885                                 }
886                                 if (!(capabilities&CTDB_CAP_NATGW)) {
887                                         continue;
888                                 }
889                                 printf("%d %s\n", nodemap->nodes[i].pnn, ctdb_addr_to_str(&nodemap->nodes[i].addr));
890                                 break;
891                         }
892                 }
893                 /* or if we still can not find any */
894                 if (i == nodemap->num) {
895                         printf("-1 0.0.0.0\n");
896                         ret = 2; /* matches ENOENT */
897                 }
898         }
899
900         /* print the pruned list of nodes belonging to this natgw list */
901         for(i=0;i<nodemap->num;i++){
902                 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
903                         continue;
904                 }
905                 printf(":%d:%s:%d:%d:%d:%d:%d\n", nodemap->nodes[i].pnn,
906                         ctdb_addr_to_str(&nodemap->nodes[i].addr),
907                        !!(nodemap->nodes[i].flags&NODE_FLAGS_DISCONNECTED),
908                        !!(nodemap->nodes[i].flags&NODE_FLAGS_BANNED),
909                        !!(nodemap->nodes[i].flags&NODE_FLAGS_PERMANENTLY_DISABLED),
910                        !!(nodemap->nodes[i].flags&NODE_FLAGS_UNHEALTHY),
911                        !!(nodemap->nodes[i].flags&NODE_FLAGS_STOPPED));
912         }
913
914         return ret;
915 }
916
917 /*
918   display the status of the scripts for monitoring (or other events)
919  */
920 static int control_one_scriptstatus(struct ctdb_context *ctdb,
921                                     enum ctdb_eventscript_call type)
922 {
923         struct ctdb_scripts_wire *script_status;
924         int ret, i;
925
926         ret = ctdb_ctrl_getscriptstatus(ctdb, TIMELIMIT(), options.pnn, ctdb, type, &script_status);
927         if (ret != 0) {
928                 DEBUG(DEBUG_ERR, ("Unable to get script status from node %u\n", options.pnn));
929                 return ret;
930         }
931
932         if (script_status == NULL) {
933                 if (!options.machinereadable) {
934                         printf("%s cycle never run\n",
935                                ctdb_eventscript_call_names[type]);
936                 }
937                 return 0;
938         }
939
940         if (!options.machinereadable) {
941                 printf("%d scripts were executed last %s cycle\n",
942                        script_status->num_scripts,
943                        ctdb_eventscript_call_names[type]);
944         }
945         for (i=0; i<script_status->num_scripts; i++) {
946                 const char *status = NULL;
947
948                 switch (script_status->scripts[i].status) {
949                 case -ETIME:
950                         status = "TIMEDOUT";
951                         break;
952                 case -ENOEXEC:
953                         status = "DISABLED";
954                         break;
955                 case 0:
956                         status = "OK";
957                         break;
958                 default:
959                         if (script_status->scripts[i].status > 0)
960                                 status = "ERROR";
961                         break;
962                 }
963                 if (options.machinereadable) {
964                         printf(":%s:%s:%i:%s:%lu.%06lu:%lu.%06lu:%s:\n",
965                                ctdb_eventscript_call_names[type],
966                                script_status->scripts[i].name,
967                                script_status->scripts[i].status,
968                                status,
969                                (long)script_status->scripts[i].start.tv_sec,
970                                (long)script_status->scripts[i].start.tv_usec,
971                                (long)script_status->scripts[i].finished.tv_sec,
972                                (long)script_status->scripts[i].finished.tv_usec,
973                                script_status->scripts[i].output);
974                         continue;
975                 }
976                 if (status)
977                         printf("%-20s Status:%s    ",
978                                script_status->scripts[i].name, status);
979                 else
980                         /* Some other error, eg from stat. */
981                         printf("%-20s Status:CANNOT RUN (%s)",
982                                script_status->scripts[i].name,
983                                strerror(-script_status->scripts[i].status));
984
985                 if (script_status->scripts[i].status >= 0) {
986                         printf("Duration:%.3lf ",
987                         timeval_delta(&script_status->scripts[i].finished,
988                               &script_status->scripts[i].start));
989                 }
990                 if (script_status->scripts[i].status != -ENOEXEC) {
991                         printf("%s",
992                                ctime(&script_status->scripts[i].start.tv_sec));
993                         if (script_status->scripts[i].status != 0) {
994                                 printf("   OUTPUT:%s\n",
995                                        script_status->scripts[i].output);
996                         }
997                 } else {
998                         printf("\n");
999                 }
1000         }
1001         return 0;
1002 }
1003
1004
1005 static int control_scriptstatus(struct ctdb_context *ctdb,
1006                                 int argc, const char **argv)
1007 {
1008         int ret;
1009         enum ctdb_eventscript_call type, min, max;
1010         const char *arg;
1011
1012         if (argc > 1) {
1013                 DEBUG(DEBUG_ERR, ("Unknown arguments to scriptstatus\n"));
1014                 return -1;
1015         }
1016
1017         if (argc == 0)
1018                 arg = ctdb_eventscript_call_names[CTDB_EVENT_MONITOR];
1019         else
1020                 arg = argv[0];
1021
1022         for (type = 0; type < CTDB_EVENT_MAX; type++) {
1023                 if (strcmp(arg, ctdb_eventscript_call_names[type]) == 0) {
1024                         min = type;
1025                         max = type+1;
1026                         break;
1027                 }
1028         }
1029         if (type == CTDB_EVENT_MAX) {
1030                 if (strcmp(arg, "all") == 0) {
1031                         min = 0;
1032                         max = CTDB_EVENT_MAX;
1033                 } else {
1034                         DEBUG(DEBUG_ERR, ("Unknown event type %s\n", argv[0]));
1035                         return -1;
1036                 }
1037         }
1038
1039         if (options.machinereadable) {
1040                 printf(":Type:Name:Code:Status:Start:End:Error Output...:\n");
1041         }
1042
1043         for (type = min; type < max; type++) {
1044                 ret = control_one_scriptstatus(ctdb, type);
1045                 if (ret != 0) {
1046                         return ret;
1047                 }
1048         }
1049
1050         return 0;
1051 }
1052
1053 /*
1054   enable an eventscript
1055  */
1056 static int control_enablescript(struct ctdb_context *ctdb, int argc, const char **argv)
1057 {
1058         int ret;
1059
1060         if (argc < 1) {
1061                 usage();
1062         }
1063
1064         ret = ctdb_ctrl_enablescript(ctdb, TIMELIMIT(), options.pnn, argv[0]);
1065         if (ret != 0) {
1066           DEBUG(DEBUG_ERR, ("Unable to enable script %s on node %u\n", argv[0], options.pnn));
1067                 return ret;
1068         }
1069
1070         return 0;
1071 }
1072
1073 /*
1074   disable an eventscript
1075  */
1076 static int control_disablescript(struct ctdb_context *ctdb, int argc, const char **argv)
1077 {
1078         int ret;
1079
1080         if (argc < 1) {
1081                 usage();
1082         }
1083
1084         ret = ctdb_ctrl_disablescript(ctdb, TIMELIMIT(), options.pnn, argv[0]);
1085         if (ret != 0) {
1086           DEBUG(DEBUG_ERR, ("Unable to disable script %s on node %u\n", argv[0], options.pnn));
1087                 return ret;
1088         }
1089
1090         return 0;
1091 }
1092
1093 /*
1094   display the pnn of the recovery master
1095  */
1096 static int control_recmaster(struct ctdb_context *ctdb, int argc, const char **argv)
1097 {
1098         uint32_t recmaster;
1099
1100         if (!ctdb_getrecmaster(ctdb_connection, options.pnn, &recmaster)) {
1101                 DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
1102                 return -1;
1103         }
1104         printf("%d\n",recmaster);
1105
1106         return 0;
1107 }
1108
1109 /*
1110   add a tickle to a public address
1111  */
1112 static int control_add_tickle(struct ctdb_context *ctdb, int argc, const char **argv)
1113 {
1114         struct ctdb_tcp_connection t;
1115         TDB_DATA data;
1116         int ret;
1117
1118         if (argc < 2) {
1119                 usage();
1120         }
1121
1122         if (parse_ip_port(argv[0], &t.src_addr) == 0) {
1123                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1124                 return -1;
1125         }
1126         if (parse_ip_port(argv[1], &t.dst_addr) == 0) {
1127                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[1]));
1128                 return -1;
1129         }
1130
1131         data.dptr = (uint8_t *)&t;
1132         data.dsize = sizeof(t);
1133
1134         /* tell all nodes about this tcp connection */
1135         ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_TCP_ADD_DELAYED_UPDATE,
1136                            0, data, ctdb, NULL, NULL, NULL, NULL);
1137         if (ret != 0) {
1138                 DEBUG(DEBUG_ERR,("Failed to add tickle\n"));
1139                 return -1;
1140         }
1141         
1142         return 0;
1143 }
1144
1145
1146 /*
1147   delete a tickle from a node
1148  */
1149 static int control_del_tickle(struct ctdb_context *ctdb, int argc, const char **argv)
1150 {
1151         struct ctdb_tcp_connection t;
1152         TDB_DATA data;
1153         int ret;
1154
1155         if (argc < 2) {
1156                 usage();
1157         }
1158
1159         if (parse_ip_port(argv[0], &t.src_addr) == 0) {
1160                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1161                 return -1;
1162         }
1163         if (parse_ip_port(argv[1], &t.dst_addr) == 0) {
1164                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[1]));
1165                 return -1;
1166         }
1167
1168         data.dptr = (uint8_t *)&t;
1169         data.dsize = sizeof(t);
1170
1171         /* tell all nodes about this tcp connection */
1172         ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_TCP_REMOVE,
1173                            0, data, ctdb, NULL, NULL, NULL, NULL);
1174         if (ret != 0) {
1175                 DEBUG(DEBUG_ERR,("Failed to remove tickle\n"));
1176                 return -1;
1177         }
1178         
1179         return 0;
1180 }
1181
1182
1183 /*
1184   get a list of all tickles for this pnn
1185  */
1186 static int control_get_tickles(struct ctdb_context *ctdb, int argc, const char **argv)
1187 {
1188         struct ctdb_control_tcp_tickle_list *list;
1189         ctdb_sock_addr addr;
1190         int i, ret;
1191         unsigned port = 0;
1192
1193         if (argc < 1) {
1194                 usage();
1195         }
1196
1197         if (argc == 2) {
1198                 port = atoi(argv[1]);
1199         }
1200
1201         if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
1202                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1203                 return -1;
1204         }
1205
1206         ret = ctdb_ctrl_get_tcp_tickles(ctdb, TIMELIMIT(), options.pnn, ctdb, &addr, &list);
1207         if (ret == -1) {
1208                 DEBUG(DEBUG_ERR, ("Unable to list tickles\n"));
1209                 return -1;
1210         }
1211
1212         if (options.machinereadable){
1213                 printf(":source ip:port:destination ip:port:\n");
1214                 for (i=0;i<list->tickles.num;i++) {
1215                         if (port && port != ntohs(list->tickles.connections[i].dst_addr.ip.sin_port)) {
1216                                 continue;
1217                         }
1218                         printf(":%s:%u", ctdb_addr_to_str(&list->tickles.connections[i].src_addr), ntohs(list->tickles.connections[i].src_addr.ip.sin_port));
1219                         printf(":%s:%u:\n", ctdb_addr_to_str(&list->tickles.connections[i].dst_addr), ntohs(list->tickles.connections[i].dst_addr.ip.sin_port));
1220                 }
1221         } else {
1222                 printf("Tickles for ip:%s\n", ctdb_addr_to_str(&list->addr));
1223                 printf("Num tickles:%u\n", list->tickles.num);
1224                 for (i=0;i<list->tickles.num;i++) {
1225                         if (port && port != ntohs(list->tickles.connections[i].dst_addr.ip.sin_port)) {
1226                                 continue;
1227                         }
1228                         printf("SRC: %s:%u   ", ctdb_addr_to_str(&list->tickles.connections[i].src_addr), ntohs(list->tickles.connections[i].src_addr.ip.sin_port));
1229                         printf("DST: %s:%u\n", ctdb_addr_to_str(&list->tickles.connections[i].dst_addr), ntohs(list->tickles.connections[i].dst_addr.ip.sin_port));
1230                 }
1231         }
1232
1233         talloc_free(list);
1234         
1235         return 0;
1236 }
1237
1238
1239 static int move_ip(struct ctdb_context *ctdb, ctdb_sock_addr *addr, uint32_t pnn)
1240 {
1241         struct ctdb_all_public_ips *ips;
1242         struct ctdb_public_ip ip;
1243         int i, ret;
1244         uint32_t *nodes;
1245         uint32_t disable_time;
1246         TDB_DATA data;
1247         struct ctdb_node_map *nodemap=NULL;
1248         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1249
1250         disable_time = 30;
1251         data.dptr  = (uint8_t*)&disable_time;
1252         data.dsize = sizeof(disable_time);
1253         ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, CTDB_SRVID_DISABLE_IP_CHECK, data);
1254         if (ret != 0) {
1255                 DEBUG(DEBUG_ERR,("Failed to send message to disable ipcheck\n"));
1256                 return -1;
1257         }
1258
1259
1260
1261         /* read the public ip list from the node */
1262         ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), pnn, ctdb, &ips);
1263         if (ret != 0) {
1264                 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", pnn));
1265                 talloc_free(tmp_ctx);
1266                 return -1;
1267         }
1268
1269         for (i=0;i<ips->num;i++) {
1270                 if (ctdb_same_ip(addr, &ips->ips[i].addr)) {
1271                         break;
1272                 }
1273         }
1274         if (i==ips->num) {
1275                 DEBUG(DEBUG_ERR, ("Node %u can not host ip address '%s'\n",
1276                         pnn, ctdb_addr_to_str(addr)));
1277                 talloc_free(tmp_ctx);
1278                 return -1;
1279         }
1280
1281         ip.pnn  = pnn;
1282         ip.addr = *addr;
1283
1284         data.dptr  = (uint8_t *)&ip;
1285         data.dsize = sizeof(ip);
1286
1287         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &nodemap);
1288         if (ret != 0) {
1289                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1290                 talloc_free(tmp_ctx);
1291                 return ret;
1292         }
1293
1294         nodes = list_of_active_nodes_except_pnn(ctdb, nodemap, tmp_ctx, pnn);
1295         ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_RELEASE_IP,
1296                                         nodes, 0,
1297                                         LONGTIMELIMIT(),
1298                                         false, data,
1299                                         NULL, NULL,
1300                                         NULL);
1301         if (ret != 0) {
1302                 DEBUG(DEBUG_ERR,("Failed to release IP on nodes\n"));
1303                 talloc_free(tmp_ctx);
1304                 return -1;
1305         }
1306
1307         ret = ctdb_ctrl_takeover_ip(ctdb, LONGTIMELIMIT(), pnn, &ip);
1308         if (ret != 0) {
1309                 DEBUG(DEBUG_ERR,("Failed to take over IP on node %d\n", pnn));
1310                 talloc_free(tmp_ctx);
1311                 return -1;
1312         }
1313
1314         /* update the recovery daemon so it now knows to expect the new
1315            node assignment for this ip.
1316         */
1317         ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, CTDB_SRVID_RECD_UPDATE_IP, data);
1318         if (ret != 0) {
1319                 DEBUG(DEBUG_ERR,("Failed to send message to update the ip on the recovery master.\n"));
1320                 return -1;
1321         }
1322
1323         talloc_free(tmp_ctx);
1324         return 0;
1325 }
1326
1327 /*
1328   move/failover an ip address to a specific node
1329  */
1330 static int control_moveip(struct ctdb_context *ctdb, int argc, const char **argv)
1331 {
1332         uint32_t pnn;
1333         int ret, retries = 0;
1334         ctdb_sock_addr addr;
1335
1336         if (argc < 2) {
1337                 usage();
1338                 return -1;
1339         }
1340
1341         if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
1342                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1343                 return -1;
1344         }
1345
1346
1347         if (sscanf(argv[1], "%u", &pnn) != 1) {
1348                 DEBUG(DEBUG_ERR, ("Badly formed pnn\n"));
1349                 return -1;
1350         }
1351
1352         do {
1353                 ret = move_ip(ctdb, &addr, pnn);
1354                 if (ret != 0) {
1355                         DEBUG(DEBUG_ERR,("Failed to move ip to node %d. Wait 3 second and try again.\n", pnn));
1356                         sleep(3);
1357                         retries++;
1358                 }
1359         } while (retries < 5 && ret != 0);
1360         if (ret != 0) {
1361                 DEBUG(DEBUG_ERR,("Failed to move ip to node %d. Giving up.\n", pnn));
1362                 return -1;
1363         }
1364
1365         return 0;
1366 }
1367
1368 void getips_store_callback(void *param, void *data)
1369 {
1370         struct ctdb_public_ip *node_ip = (struct ctdb_public_ip *)data;
1371         struct ctdb_all_public_ips *ips = param;
1372         int i;
1373
1374         i = ips->num++;
1375         ips->ips[i].pnn  = node_ip->pnn;
1376         ips->ips[i].addr = node_ip->addr;
1377 }
1378
1379 void getips_count_callback(void *param, void *data)
1380 {
1381         uint32_t *count = param;
1382
1383         (*count)++;
1384 }
1385
1386 #define IP_KEYLEN       4
1387 static uint32_t *ip_key(ctdb_sock_addr *ip)
1388 {
1389         static uint32_t key[IP_KEYLEN];
1390
1391         bzero(key, sizeof(key));
1392
1393         switch (ip->sa.sa_family) {
1394         case AF_INET:
1395                 key[0]  = ip->ip.sin_addr.s_addr;
1396                 break;
1397         case AF_INET6:
1398                 key[0]  = ip->ip6.sin6_addr.s6_addr32[3];
1399                 key[1]  = ip->ip6.sin6_addr.s6_addr32[2];
1400                 key[2]  = ip->ip6.sin6_addr.s6_addr32[1];
1401                 key[3]  = ip->ip6.sin6_addr.s6_addr32[0];
1402                 break;
1403         default:
1404                 DEBUG(DEBUG_ERR, (__location__ " ERROR, unknown family passed :%u\n", ip->sa.sa_family));
1405                 return key;
1406         }
1407
1408         return key;
1409 }
1410
1411 static void *add_ip_callback(void *parm, void *data)
1412 {
1413         return parm;
1414 }
1415
1416 static int
1417 control_get_all_public_ips(struct ctdb_context *ctdb, TALLOC_CTX *tmp_ctx, struct ctdb_all_public_ips **ips)
1418 {
1419         struct ctdb_all_public_ips *tmp_ips;
1420         struct ctdb_node_map *nodemap=NULL;
1421         trbt_tree_t *ip_tree;
1422         int i, j, len, ret;
1423         uint32_t count;
1424
1425         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
1426         if (ret != 0) {
1427                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1428                 return ret;
1429         }
1430
1431         ip_tree = trbt_create(tmp_ctx, 0);
1432
1433         for(i=0;i<nodemap->num;i++){
1434                 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
1435                         continue;
1436                 }
1437                 if (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) {
1438                         continue;
1439                 }
1440
1441                 /* read the public ip list from this node */
1442                 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &tmp_ips);
1443                 if (ret != 0) {
1444                         DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", nodemap->nodes[i].pnn));
1445                         return -1;
1446                 }
1447         
1448                 for (j=0; j<tmp_ips->num;j++) {
1449                         struct ctdb_public_ip *node_ip;
1450
1451                         node_ip = talloc(tmp_ctx, struct ctdb_public_ip);
1452                         node_ip->pnn  = tmp_ips->ips[j].pnn;
1453                         node_ip->addr = tmp_ips->ips[j].addr;
1454
1455                         trbt_insertarray32_callback(ip_tree,
1456                                 IP_KEYLEN, ip_key(&tmp_ips->ips[j].addr),
1457                                 add_ip_callback,
1458                                 node_ip);
1459                 }
1460                 talloc_free(tmp_ips);
1461         }
1462
1463         /* traverse */
1464         count = 0;
1465         trbt_traversearray32(ip_tree, IP_KEYLEN, getips_count_callback, &count);
1466
1467         len = offsetof(struct ctdb_all_public_ips, ips) + 
1468                 count*sizeof(struct ctdb_public_ip);
1469         tmp_ips = talloc_zero_size(tmp_ctx, len);
1470         trbt_traversearray32(ip_tree, IP_KEYLEN, getips_store_callback, tmp_ips);
1471
1472         *ips = tmp_ips;
1473
1474         return 0;
1475 }
1476
1477
1478 /* 
1479  * scans all other nodes and returns a pnn for another node that can host this 
1480  * ip address or -1
1481  */
1482 static int
1483 find_other_host_for_public_ip(struct ctdb_context *ctdb, ctdb_sock_addr *addr)
1484 {
1485         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1486         struct ctdb_all_public_ips *ips;
1487         struct ctdb_node_map *nodemap=NULL;
1488         int i, j, ret;
1489
1490         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
1491         if (ret != 0) {
1492                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1493                 talloc_free(tmp_ctx);
1494                 return ret;
1495         }
1496
1497         for(i=0;i<nodemap->num;i++){
1498                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
1499                         continue;
1500                 }
1501                 if (nodemap->nodes[i].pnn == options.pnn) {
1502                         continue;
1503                 }
1504
1505                 /* read the public ip list from this node */
1506                 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &ips);
1507                 if (ret != 0) {
1508                         DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", nodemap->nodes[i].pnn));
1509                         return -1;
1510                 }
1511
1512                 for (j=0;j<ips->num;j++) {
1513                         if (ctdb_same_ip(addr, &ips->ips[j].addr)) {
1514                                 talloc_free(tmp_ctx);
1515                                 return nodemap->nodes[i].pnn;
1516                         }
1517                 }
1518                 talloc_free(ips);
1519         }
1520
1521         talloc_free(tmp_ctx);
1522         return -1;
1523 }
1524
1525 static uint32_t ipreallocate_finished;
1526
1527 /*
1528   handler for receiving the response to ipreallocate
1529 */
1530 static void ip_reallocate_handler(struct ctdb_context *ctdb, uint64_t srvid, 
1531                              TDB_DATA data, void *private_data)
1532 {
1533         ipreallocate_finished = 1;
1534 }
1535
1536 static void ctdb_every_second(struct event_context *ev, struct timed_event *te, struct timeval t, void *p)
1537 {
1538         struct ctdb_context *ctdb = talloc_get_type(p, struct ctdb_context);
1539
1540         event_add_timed(ctdb->ev, ctdb, 
1541                                 timeval_current_ofs(1, 0),
1542                                 ctdb_every_second, ctdb);
1543 }
1544
1545 /*
1546   ask the recovery daemon on the recovery master to perform a ip reallocation
1547  */
1548 static int control_ipreallocate(struct ctdb_context *ctdb, int argc, const char **argv)
1549 {
1550         int i, ret;
1551         TDB_DATA data;
1552         struct takeover_run_reply rd;
1553         uint32_t recmaster;
1554         struct ctdb_node_map *nodemap=NULL;
1555         int retries=0;
1556         struct timeval tv = timeval_current();
1557
1558         /* we need some events to trigger so we can timeout and restart
1559            the loop
1560         */
1561         event_add_timed(ctdb->ev, ctdb, 
1562                                 timeval_current_ofs(1, 0),
1563                                 ctdb_every_second, ctdb);
1564
1565         rd.pnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE);
1566         if (rd.pnn == -1) {
1567                 DEBUG(DEBUG_ERR, ("Failed to get pnn of local node\n"));
1568                 return -1;
1569         }
1570         rd.srvid = getpid();
1571
1572         /* register a message port for receiveing the reply so that we
1573            can receive the reply
1574         */
1575         ctdb_client_set_message_handler(ctdb, rd.srvid, ip_reallocate_handler, NULL);
1576
1577         data.dptr = (uint8_t *)&rd;
1578         data.dsize = sizeof(rd);
1579
1580 again:
1581         /* check that there are valid nodes available */
1582         if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb, &nodemap) != 0) {
1583                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
1584                 return -1;
1585         }
1586         for (i=0; i<nodemap->num;i++) {
1587                 if ((nodemap->nodes[i].flags & (NODE_FLAGS_DELETED|NODE_FLAGS_BANNED|NODE_FLAGS_STOPPED)) == 0) {
1588                         break;
1589                 }
1590         }
1591         if (i==nodemap->num) {
1592                 DEBUG(DEBUG_ERR,("No recmaster available, no need to wait for cluster convergence\n"));
1593                 return 0;
1594         }
1595
1596
1597         if (!ctdb_getrecmaster(ctdb_connection, options.pnn, &recmaster)) {
1598                 DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
1599                 return -1;
1600         }
1601
1602         /* verify the node exists */
1603         if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), recmaster, ctdb, &nodemap) != 0) {
1604                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
1605                 return -1;
1606         }
1607
1608
1609         /* check tha there are nodes available that can act as a recmaster */
1610         for (i=0; i<nodemap->num; i++) {
1611                 if (nodemap->nodes[i].flags & (NODE_FLAGS_DELETED|NODE_FLAGS_BANNED|NODE_FLAGS_STOPPED)) {
1612                         continue;
1613                 }
1614                 break;
1615         }
1616         if (i == nodemap->num) {
1617                 DEBUG(DEBUG_ERR,("No possible nodes to host addresses.\n"));
1618                 return 0;
1619         }
1620
1621         /* verify the recovery master is not STOPPED, nor BANNED */
1622         if (nodemap->nodes[recmaster].flags & (NODE_FLAGS_DELETED|NODE_FLAGS_BANNED|NODE_FLAGS_STOPPED)) {
1623                 DEBUG(DEBUG_ERR,("No suitable recmaster found. Try again\n"));
1624                 retries++;
1625                 sleep(1);
1626                 goto again;
1627         } 
1628         
1629         /* verify the recovery master is not STOPPED, nor BANNED */
1630         if (nodemap->nodes[recmaster].flags & (NODE_FLAGS_DELETED|NODE_FLAGS_BANNED|NODE_FLAGS_STOPPED)) {
1631                 DEBUG(DEBUG_ERR,("No suitable recmaster found. Try again\n"));
1632                 retries++;
1633                 sleep(1);
1634                 goto again;
1635         } 
1636
1637         ipreallocate_finished = 0;
1638         ret = ctdb_client_send_message(ctdb, recmaster, CTDB_SRVID_TAKEOVER_RUN, data);
1639         if (ret != 0) {
1640                 DEBUG(DEBUG_ERR,("Failed to send ip takeover run request message to %u\n", options.pnn));
1641                 return -1;
1642         }
1643
1644         tv = timeval_current();
1645         /* this loop will terminate when we have received the reply */
1646         while (timeval_elapsed(&tv) < 5.0 && ipreallocate_finished == 0) {
1647                 event_loop_once(ctdb->ev);
1648         }
1649         if (ipreallocate_finished == 1) {
1650                 return 0;
1651         }
1652
1653         retries++;
1654         sleep(1);
1655         goto again;
1656
1657         return 0;
1658 }
1659
1660
1661 /*
1662   add a public ip address to a node
1663  */
1664 static int control_addip(struct ctdb_context *ctdb, int argc, const char **argv)
1665 {
1666         int i, ret;
1667         int len, retries = 0;
1668         unsigned mask;
1669         ctdb_sock_addr addr;
1670         struct ctdb_control_ip_iface *pub;
1671         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1672         struct ctdb_all_public_ips *ips;
1673
1674
1675         if (argc != 2) {
1676                 talloc_free(tmp_ctx);
1677                 usage();
1678         }
1679
1680         if (!parse_ip_mask(argv[0], argv[1], &addr, &mask)) {
1681                 DEBUG(DEBUG_ERR, ("Badly formed ip/mask : %s\n", argv[0]));
1682                 talloc_free(tmp_ctx);
1683                 return -1;
1684         }
1685
1686         /* read the public ip list from the node */
1687         ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ips);
1688         if (ret != 0) {
1689                 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", options.pnn));
1690                 talloc_free(tmp_ctx);
1691                 return -1;
1692         }
1693         for (i=0;i<ips->num;i++) {
1694                 if (ctdb_same_ip(&addr, &ips->ips[i].addr)) {
1695                         DEBUG(DEBUG_ERR,("Can not add ip to node. Node already hosts this ip\n"));
1696                         return 0;
1697                 }
1698         }
1699
1700
1701
1702         /* Dont timeout. This command waits for an ip reallocation
1703            which sometimes can take wuite a while if there has
1704            been a recent recovery
1705         */
1706         alarm(0);
1707
1708         len = offsetof(struct ctdb_control_ip_iface, iface) + strlen(argv[1]) + 1;
1709         pub = talloc_size(tmp_ctx, len); 
1710         CTDB_NO_MEMORY(ctdb, pub);
1711
1712         pub->addr  = addr;
1713         pub->mask  = mask;
1714         pub->len   = strlen(argv[1])+1;
1715         memcpy(&pub->iface[0], argv[1], strlen(argv[1])+1);
1716
1717         do {
1718                 ret = ctdb_ctrl_add_public_ip(ctdb, TIMELIMIT(), options.pnn, pub);
1719                 if (ret != 0) {
1720                         DEBUG(DEBUG_ERR, ("Unable to add public ip to node %u. Wait 3 seconds and try again.\n", options.pnn));
1721                         sleep(3);
1722                         retries++;
1723                 }
1724         } while (retries < 5 && ret != 0);
1725         if (ret != 0) {
1726                 DEBUG(DEBUG_ERR, ("Unable to add public ip to node %u. Giving up.\n", options.pnn));
1727                 talloc_free(tmp_ctx);
1728                 return ret;
1729         }
1730
1731         do {
1732                 ret = control_ipreallocate(ctdb, argc, argv);
1733                 if (ret != 0) {
1734                         DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u. Wait 3 seconds and try again.\n", options.pnn));
1735                         sleep(3);
1736                         retries++;
1737                 }
1738         } while (retries < 5 && ret != 0);
1739         if (ret != 0) {
1740                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u. Giving up.\n", options.pnn));
1741                 talloc_free(tmp_ctx);
1742                 return ret;
1743         }
1744
1745         talloc_free(tmp_ctx);
1746         return 0;
1747 }
1748
1749 static int control_delip(struct ctdb_context *ctdb, int argc, const char **argv);
1750
1751 static int control_delip_all(struct ctdb_context *ctdb, int argc, const char **argv, ctdb_sock_addr *addr)
1752 {
1753         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1754         struct ctdb_node_map *nodemap=NULL;
1755         struct ctdb_all_public_ips *ips;
1756         int ret, i, j;
1757
1758         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
1759         if (ret != 0) {
1760                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from current node\n"));
1761                 return ret;
1762         }
1763
1764         /* remove it from the nodes that are not hosting the ip currently */
1765         for(i=0;i<nodemap->num;i++){
1766                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
1767                         continue;
1768                 }
1769                 if (ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &ips) != 0) {
1770                         DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %d\n", nodemap->nodes[i].pnn));
1771                         continue;
1772                 }
1773
1774                 for (j=0;j<ips->num;j++) {
1775                         if (ctdb_same_ip(addr, &ips->ips[j].addr)) {
1776                                 break;
1777                         }
1778                 }
1779                 if (j==ips->num) {
1780                         continue;
1781                 }
1782
1783                 if (ips->ips[j].pnn == nodemap->nodes[i].pnn) {
1784                         continue;
1785                 }
1786
1787                 options.pnn = nodemap->nodes[i].pnn;
1788                 control_delip(ctdb, argc, argv);
1789         }
1790
1791
1792         /* remove it from every node (also the one hosting it) */
1793         for(i=0;i<nodemap->num;i++){
1794                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
1795                         continue;
1796                 }
1797                 if (ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &ips) != 0) {
1798                         DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %d\n", nodemap->nodes[i].pnn));
1799                         continue;
1800                 }
1801
1802                 for (j=0;j<ips->num;j++) {
1803                         if (ctdb_same_ip(addr, &ips->ips[j].addr)) {
1804                                 break;
1805                         }
1806                 }
1807                 if (j==ips->num) {
1808                         continue;
1809                 }
1810
1811                 options.pnn = nodemap->nodes[i].pnn;
1812                 control_delip(ctdb, argc, argv);
1813         }
1814
1815         talloc_free(tmp_ctx);
1816         return 0;
1817 }
1818         
1819 /*
1820   delete a public ip address from a node
1821  */
1822 static int control_delip(struct ctdb_context *ctdb, int argc, const char **argv)
1823 {
1824         int i, ret;
1825         int retries = 0;
1826         ctdb_sock_addr addr;
1827         struct ctdb_control_ip_iface pub;
1828         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1829         struct ctdb_all_public_ips *ips;
1830
1831         if (argc != 1) {
1832                 talloc_free(tmp_ctx);
1833                 usage();
1834         }
1835
1836         if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
1837                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1838                 return -1;
1839         }
1840
1841         if (options.pnn == CTDB_BROADCAST_ALL) {
1842                 return control_delip_all(ctdb, argc, argv, &addr);
1843         }
1844
1845         pub.addr  = addr;
1846         pub.mask  = 0;
1847         pub.len   = 0;
1848
1849         ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ips);
1850         if (ret != 0) {
1851                 DEBUG(DEBUG_ERR, ("Unable to get public ip list from cluster\n"));
1852                 talloc_free(tmp_ctx);
1853                 return ret;
1854         }
1855         
1856         for (i=0;i<ips->num;i++) {
1857                 if (ctdb_same_ip(&addr, &ips->ips[i].addr)) {
1858                         break;
1859                 }
1860         }
1861
1862         if (i==ips->num) {
1863                 DEBUG(DEBUG_ERR, ("This node does not support this public address '%s'\n",
1864                         ctdb_addr_to_str(&addr)));
1865                 talloc_free(tmp_ctx);
1866                 return -1;
1867         }
1868
1869         if (ips->ips[i].pnn == options.pnn) {
1870                 ret = find_other_host_for_public_ip(ctdb, &addr);
1871                 if (ret != -1) {
1872                         do {
1873                                 ret = move_ip(ctdb, &addr, ret);
1874                                 if (ret != 0) {
1875                                         DEBUG(DEBUG_ERR,("Failed to move ip to node %d. Wait 3 seconds and try again.\n", options.pnn));
1876                                         sleep(3);
1877                                         retries++;
1878                                 }
1879                         } while (retries < 5 && ret != 0);
1880                         if (ret != 0) {
1881                                 DEBUG(DEBUG_ERR,("Failed to move ip to node %d. Giving up.\n", options.pnn));
1882                                 return -1;
1883                         }
1884                 }
1885         }
1886
1887         ret = ctdb_ctrl_del_public_ip(ctdb, TIMELIMIT(), options.pnn, &pub);
1888         if (ret != 0) {
1889                 DEBUG(DEBUG_ERR, ("Unable to del public ip from node %u\n", options.pnn));
1890                 talloc_free(tmp_ctx);
1891                 return ret;
1892         }
1893
1894         talloc_free(tmp_ctx);
1895         return 0;
1896 }
1897
1898 /*
1899   kill a tcp connection
1900  */
1901 static int kill_tcp(struct ctdb_context *ctdb, int argc, const char **argv)
1902 {
1903         int ret;
1904         struct ctdb_control_killtcp killtcp;
1905
1906         if (argc < 2) {
1907                 usage();
1908         }
1909
1910         if (!parse_ip_port(argv[0], &killtcp.src_addr)) {
1911                 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[0]));
1912                 return -1;
1913         }
1914
1915         if (!parse_ip_port(argv[1], &killtcp.dst_addr)) {
1916                 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[1]));
1917                 return -1;
1918         }
1919
1920         ret = ctdb_ctrl_killtcp(ctdb, TIMELIMIT(), options.pnn, &killtcp);
1921         if (ret != 0) {
1922                 DEBUG(DEBUG_ERR, ("Unable to killtcp from node %u\n", options.pnn));
1923                 return ret;
1924         }
1925
1926         return 0;
1927 }
1928
1929
1930 /*
1931   send a gratious arp
1932  */
1933 static int control_gratious_arp(struct ctdb_context *ctdb, int argc, const char **argv)
1934 {
1935         int ret;
1936         ctdb_sock_addr addr;
1937
1938         if (argc < 2) {
1939                 usage();
1940         }
1941
1942         if (!parse_ip(argv[0], NULL, 0, &addr)) {
1943                 DEBUG(DEBUG_ERR, ("Bad IP '%s'\n", argv[0]));
1944                 return -1;
1945         }
1946
1947         ret = ctdb_ctrl_gratious_arp(ctdb, TIMELIMIT(), options.pnn, &addr, argv[1]);
1948         if (ret != 0) {
1949                 DEBUG(DEBUG_ERR, ("Unable to send gratious_arp from node %u\n", options.pnn));
1950                 return ret;
1951         }
1952
1953         return 0;
1954 }
1955
1956 /*
1957   register a server id
1958  */
1959 static int regsrvid(struct ctdb_context *ctdb, int argc, const char **argv)
1960 {
1961         int ret;
1962         struct ctdb_server_id server_id;
1963
1964         if (argc < 3) {
1965                 usage();
1966         }
1967
1968         server_id.pnn       = strtoul(argv[0], NULL, 0);
1969         server_id.type      = strtoul(argv[1], NULL, 0);
1970         server_id.server_id = strtoul(argv[2], NULL, 0);
1971
1972         ret = ctdb_ctrl_register_server_id(ctdb, TIMELIMIT(), &server_id);
1973         if (ret != 0) {
1974                 DEBUG(DEBUG_ERR, ("Unable to register server_id from node %u\n", options.pnn));
1975                 return ret;
1976         }
1977         DEBUG(DEBUG_ERR,("Srvid registered. Sleeping for 999 seconds\n"));
1978         sleep(999);
1979         return -1;
1980 }
1981
1982 /*
1983   unregister a server id
1984  */
1985 static int unregsrvid(struct ctdb_context *ctdb, int argc, const char **argv)
1986 {
1987         int ret;
1988         struct ctdb_server_id server_id;
1989
1990         if (argc < 3) {
1991                 usage();
1992         }
1993
1994         server_id.pnn       = strtoul(argv[0], NULL, 0);
1995         server_id.type      = strtoul(argv[1], NULL, 0);
1996         server_id.server_id = strtoul(argv[2], NULL, 0);
1997
1998         ret = ctdb_ctrl_unregister_server_id(ctdb, TIMELIMIT(), &server_id);
1999         if (ret != 0) {
2000                 DEBUG(DEBUG_ERR, ("Unable to unregister server_id from node %u\n", options.pnn));
2001                 return ret;
2002         }
2003         return -1;
2004 }
2005
2006 /*
2007   check if a server id exists
2008  */
2009 static int chksrvid(struct ctdb_context *ctdb, int argc, const char **argv)
2010 {
2011         uint32_t status;
2012         int ret;
2013         struct ctdb_server_id server_id;
2014
2015         if (argc < 3) {
2016                 usage();
2017         }
2018
2019         server_id.pnn       = strtoul(argv[0], NULL, 0);
2020         server_id.type      = strtoul(argv[1], NULL, 0);
2021         server_id.server_id = strtoul(argv[2], NULL, 0);
2022
2023         ret = ctdb_ctrl_check_server_id(ctdb, TIMELIMIT(), options.pnn, &server_id, &status);
2024         if (ret != 0) {
2025                 DEBUG(DEBUG_ERR, ("Unable to check server_id from node %u\n", options.pnn));
2026                 return ret;
2027         }
2028
2029         if (status) {
2030                 printf("Server id %d:%d:%d EXISTS\n", server_id.pnn, server_id.type, server_id.server_id);
2031         } else {
2032                 printf("Server id %d:%d:%d does NOT exist\n", server_id.pnn, server_id.type, server_id.server_id);
2033         }
2034         return 0;
2035 }
2036
2037 /*
2038   get a list of all server ids that are registered on a node
2039  */
2040 static int getsrvids(struct ctdb_context *ctdb, int argc, const char **argv)
2041 {
2042         int i, ret;
2043         struct ctdb_server_id_list *server_ids;
2044
2045         ret = ctdb_ctrl_get_server_id_list(ctdb, ctdb, TIMELIMIT(), options.pnn, &server_ids);
2046         if (ret != 0) {
2047                 DEBUG(DEBUG_ERR, ("Unable to get server_id list from node %u\n", options.pnn));
2048                 return ret;
2049         }
2050
2051         for (i=0; i<server_ids->num; i++) {
2052                 printf("Server id %d:%d:%d\n", 
2053                         server_ids->server_ids[i].pnn, 
2054                         server_ids->server_ids[i].type, 
2055                         server_ids->server_ids[i].server_id); 
2056         }
2057
2058         return -1;
2059 }
2060
2061 /*
2062   send a tcp tickle ack
2063  */
2064 static int tickle_tcp(struct ctdb_context *ctdb, int argc, const char **argv)
2065 {
2066         int ret;
2067         ctdb_sock_addr  src, dst;
2068
2069         if (argc < 2) {
2070                 usage();
2071         }
2072
2073         if (!parse_ip_port(argv[0], &src)) {
2074                 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[0]));
2075                 return -1;
2076         }
2077
2078         if (!parse_ip_port(argv[1], &dst)) {
2079                 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[1]));
2080                 return -1;
2081         }
2082
2083         ret = ctdb_sys_send_tcp(&src, &dst, 0, 0, 0);
2084         if (ret==0) {
2085                 return 0;
2086         }
2087         DEBUG(DEBUG_ERR, ("Error while sending tickle ack\n"));
2088
2089         return -1;
2090 }
2091
2092
2093 /*
2094   display public ip status
2095  */
2096 static int control_ip(struct ctdb_context *ctdb, int argc, const char **argv)
2097 {
2098         int i, ret;
2099         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2100         struct ctdb_all_public_ips *ips;
2101
2102         if (options.pnn == CTDB_BROADCAST_ALL) {
2103                 /* read the list of public ips from all nodes */
2104                 ret = control_get_all_public_ips(ctdb, tmp_ctx, &ips);
2105         } else {
2106                 /* read the public ip list from this node */
2107                 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ips);
2108         }
2109         if (ret != 0) {
2110                 DEBUG(DEBUG_ERR, ("Unable to get public ips from node %u\n", options.pnn));
2111                 talloc_free(tmp_ctx);
2112                 return ret;
2113         }
2114
2115         if (options.machinereadable){
2116                 printf(":Public IP:Node:");
2117                 if (options.verbose){
2118                         printf("ActiveInterface:AvailableInterfaces:ConfiguredInterfaces:");
2119                 }
2120                 printf("\n");
2121         } else {
2122                 if (options.pnn == CTDB_BROADCAST_ALL) {
2123                         printf("Public IPs on ALL nodes\n");
2124                 } else {
2125                         printf("Public IPs on node %u\n", options.pnn);
2126                 }
2127         }
2128
2129         for (i=1;i<=ips->num;i++) {
2130                 struct ctdb_control_public_ip_info *info = NULL;
2131                 int32_t pnn;
2132                 char *aciface = NULL;
2133                 char *avifaces = NULL;
2134                 char *cifaces = NULL;
2135
2136                 if (options.pnn == CTDB_BROADCAST_ALL) {
2137                         pnn = ips->ips[ips->num-i].pnn;
2138                 } else {
2139                         pnn = options.pnn;
2140                 }
2141
2142                 if (pnn != -1) {
2143                         ret = ctdb_ctrl_get_public_ip_info(ctdb, TIMELIMIT(), pnn, ctdb,
2144                                                    &ips->ips[ips->num-i].addr, &info);
2145                 } else {
2146                         ret = -1;
2147                 }
2148
2149                 if (ret == 0) {
2150                         int j;
2151                         for (j=0; j < info->num; j++) {
2152                                 if (cifaces == NULL) {
2153                                         cifaces = talloc_strdup(info,
2154                                                                 info->ifaces[j].name);
2155                                 } else {
2156                                         cifaces = talloc_asprintf_append(cifaces,
2157                                                                          ",%s",
2158                                                                          info->ifaces[j].name);
2159                                 }
2160
2161                                 if (info->active_idx == j) {
2162                                         aciface = info->ifaces[j].name;
2163                                 }
2164
2165                                 if (info->ifaces[j].link_state == 0) {
2166                                         continue;
2167                                 }
2168
2169                                 if (avifaces == NULL) {
2170                                         avifaces = talloc_strdup(info, info->ifaces[j].name);
2171                                 } else {
2172                                         avifaces = talloc_asprintf_append(avifaces,
2173                                                                           ",%s",
2174                                                                           info->ifaces[j].name);
2175                                 }
2176                         }
2177                 }
2178
2179                 if (options.machinereadable){
2180                         printf(":%s:%d:",
2181                                 ctdb_addr_to_str(&ips->ips[ips->num-i].addr),
2182                                 ips->ips[ips->num-i].pnn);
2183                         if (options.verbose){
2184                                 printf("%s:%s:%s:",
2185                                         aciface?aciface:"",
2186                                         avifaces?avifaces:"",
2187                                         cifaces?cifaces:"");
2188                         }
2189                         printf("\n");
2190                 } else {
2191                         if (options.verbose) {
2192                                 printf("%s node[%d] active[%s] available[%s] configured[%s]\n",
2193                                         ctdb_addr_to_str(&ips->ips[ips->num-i].addr),
2194                                         ips->ips[ips->num-i].pnn,
2195                                         aciface?aciface:"",
2196                                         avifaces?avifaces:"",
2197                                         cifaces?cifaces:"");
2198                         } else {
2199                                 printf("%s %d\n",
2200                                         ctdb_addr_to_str(&ips->ips[ips->num-i].addr),
2201                                         ips->ips[ips->num-i].pnn);
2202                         }
2203                 }
2204                 talloc_free(info);
2205         }
2206
2207         talloc_free(tmp_ctx);
2208         return 0;
2209 }
2210
2211 /*
2212   public ip info
2213  */
2214 static int control_ipinfo(struct ctdb_context *ctdb, int argc, const char **argv)
2215 {
2216         int i, ret;
2217         ctdb_sock_addr addr;
2218         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2219         struct ctdb_control_public_ip_info *info;
2220
2221         if (argc != 1) {
2222                 talloc_free(tmp_ctx);
2223                 usage();
2224         }
2225
2226         if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
2227                 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
2228                 return -1;
2229         }
2230
2231         /* read the public ip info from this node */
2232         ret = ctdb_ctrl_get_public_ip_info(ctdb, TIMELIMIT(), options.pnn,
2233                                            tmp_ctx, &addr, &info);
2234         if (ret != 0) {
2235                 DEBUG(DEBUG_ERR, ("Unable to get public ip[%s]info from node %u\n",
2236                                   argv[0], options.pnn));
2237                 talloc_free(tmp_ctx);
2238                 return ret;
2239         }
2240
2241         printf("Public IP[%s] info on node %u\n",
2242                ctdb_addr_to_str(&info->ip.addr),
2243                options.pnn);
2244
2245         printf("IP:%s\nCurrentNode:%d\nNumInterfaces:%u\n",
2246                ctdb_addr_to_str(&info->ip.addr),
2247                info->ip.pnn, info->num);
2248
2249         for (i=0; i<info->num; i++) {
2250                 info->ifaces[i].name[CTDB_IFACE_SIZE] = '\0';
2251
2252                 printf("Interface[%u]: Name:%s Link:%s References:%u%s\n",
2253                        i+1, info->ifaces[i].name,
2254                        info->ifaces[i].link_state?"up":"down",
2255                        (unsigned int)info->ifaces[i].references,
2256                        (i==info->active_idx)?" (active)":"");
2257         }
2258
2259         talloc_free(tmp_ctx);
2260         return 0;
2261 }
2262
2263 /*
2264   display interfaces status
2265  */
2266 static int control_ifaces(struct ctdb_context *ctdb, int argc, const char **argv)
2267 {
2268         int i, ret;
2269         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2270         struct ctdb_control_get_ifaces *ifaces;
2271
2272         /* read the public ip list from this node */
2273         ret = ctdb_ctrl_get_ifaces(ctdb, TIMELIMIT(), options.pnn,
2274                                    tmp_ctx, &ifaces);
2275         if (ret != 0) {
2276                 DEBUG(DEBUG_ERR, ("Unable to get interfaces from node %u\n",
2277                                   options.pnn));
2278                 talloc_free(tmp_ctx);
2279                 return ret;
2280         }
2281
2282         if (options.machinereadable){
2283                 printf(":Name:LinkStatus:References:\n");
2284         } else {
2285                 printf("Interfaces on node %u\n", options.pnn);
2286         }
2287
2288         for (i=0; i<ifaces->num; i++) {
2289                 if (options.machinereadable){
2290                         printf(":%s:%s:%u\n",
2291                                ifaces->ifaces[i].name,
2292                                ifaces->ifaces[i].link_state?"1":"0",
2293                                (unsigned int)ifaces->ifaces[i].references);
2294                 } else {
2295                         printf("name:%s link:%s references:%u\n",
2296                                ifaces->ifaces[i].name,
2297                                ifaces->ifaces[i].link_state?"up":"down",
2298                                (unsigned int)ifaces->ifaces[i].references);
2299                 }
2300         }
2301
2302         talloc_free(tmp_ctx);
2303         return 0;
2304 }
2305
2306
2307 /*
2308   set link status of an interface
2309  */
2310 static int control_setifacelink(struct ctdb_context *ctdb, int argc, const char **argv)
2311 {
2312         int ret;
2313         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2314         struct ctdb_control_iface_info info;
2315
2316         ZERO_STRUCT(info);
2317
2318         if (argc != 2) {
2319                 usage();
2320         }
2321
2322         if (strlen(argv[0]) > CTDB_IFACE_SIZE) {
2323                 DEBUG(DEBUG_ERR, ("interfaces name '%s' too long\n",
2324                                   argv[0]));
2325                 talloc_free(tmp_ctx);
2326                 return -1;
2327         }
2328         strcpy(info.name, argv[0]);
2329
2330         if (strcmp(argv[1], "up") == 0) {
2331                 info.link_state = 1;
2332         } else if (strcmp(argv[1], "down") == 0) {
2333                 info.link_state = 0;
2334         } else {
2335                 DEBUG(DEBUG_ERR, ("link state invalid '%s' should be 'up' or 'down'\n",
2336                                   argv[1]));
2337                 talloc_free(tmp_ctx);
2338                 return -1;
2339         }
2340
2341         /* read the public ip list from this node */
2342         ret = ctdb_ctrl_set_iface_link(ctdb, TIMELIMIT(), options.pnn,
2343                                    tmp_ctx, &info);
2344         if (ret != 0) {
2345                 DEBUG(DEBUG_ERR, ("Unable to set link state for interfaces %s node %u\n",
2346                                   argv[0], options.pnn));
2347                 talloc_free(tmp_ctx);
2348                 return ret;
2349         }
2350
2351         talloc_free(tmp_ctx);
2352         return 0;
2353 }
2354
2355 /*
2356   display pid of a ctdb daemon
2357  */
2358 static int control_getpid(struct ctdb_context *ctdb, int argc, const char **argv)
2359 {
2360         uint32_t pid;
2361         int ret;
2362
2363         ret = ctdb_ctrl_getpid(ctdb, TIMELIMIT(), options.pnn, &pid);
2364         if (ret != 0) {
2365                 DEBUG(DEBUG_ERR, ("Unable to get daemon pid from node %u\n", options.pnn));
2366                 return ret;
2367         }
2368         printf("Pid:%d\n", pid);
2369
2370         return 0;
2371 }
2372
2373 /*
2374   disable a remote node
2375  */
2376 static int control_disable(struct ctdb_context *ctdb, int argc, const char **argv)
2377 {
2378         int ret;
2379         struct ctdb_node_map *nodemap=NULL;
2380
2381         /* check if the node is already disabled */
2382         if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
2383                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2384                 exit(10);
2385         }
2386         if (nodemap->nodes[options.pnn].flags & NODE_FLAGS_PERMANENTLY_DISABLED) {
2387                 DEBUG(DEBUG_ERR,("Node %d is already disabled.\n", options.pnn));
2388                 return 0;
2389         }
2390
2391         do {
2392                 ret = ctdb_ctrl_modflags(ctdb, TIMELIMIT(), options.pnn, NODE_FLAGS_PERMANENTLY_DISABLED, 0);
2393                 if (ret != 0) {
2394                         DEBUG(DEBUG_ERR, ("Unable to disable node %u\n", options.pnn));
2395                         return ret;
2396                 }
2397
2398                 sleep(1);
2399
2400                 /* read the nodemap and verify the change took effect */
2401                 if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
2402                         DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2403                         exit(10);
2404                 }
2405
2406         } while (!(nodemap->nodes[options.pnn].flags & NODE_FLAGS_PERMANENTLY_DISABLED));
2407         ret = control_ipreallocate(ctdb, argc, argv);
2408         if (ret != 0) {
2409                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u\n", options.pnn));
2410                 return ret;
2411         }
2412
2413         return 0;
2414 }
2415
2416 /*
2417   enable a disabled remote node
2418  */
2419 static int control_enable(struct ctdb_context *ctdb, int argc, const char **argv)
2420 {
2421         int ret;
2422
2423         struct ctdb_node_map *nodemap=NULL;
2424
2425
2426         /* check if the node is already enabled */
2427         if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
2428                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2429                 exit(10);
2430         }
2431         if (!(nodemap->nodes[options.pnn].flags & NODE_FLAGS_PERMANENTLY_DISABLED)) {
2432                 DEBUG(DEBUG_ERR,("Node %d is already enabled.\n", options.pnn));
2433                 return 0;
2434         }
2435
2436         do {
2437                 ret = ctdb_ctrl_modflags(ctdb, TIMELIMIT(), options.pnn, 0, NODE_FLAGS_PERMANENTLY_DISABLED);
2438                 if (ret != 0) {
2439                         DEBUG(DEBUG_ERR, ("Unable to enable node %u\n", options.pnn));
2440                         return ret;
2441                 }
2442
2443                 sleep(1);
2444
2445                 /* read the nodemap and verify the change took effect */
2446                 if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
2447                         DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2448                         exit(10);
2449                 }
2450
2451         } while (nodemap->nodes[options.pnn].flags & NODE_FLAGS_PERMANENTLY_DISABLED);
2452
2453         ret = control_ipreallocate(ctdb, argc, argv);
2454         if (ret != 0) {
2455                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u\n", options.pnn));
2456                 return ret;
2457         }
2458
2459         return 0;
2460 }
2461
2462 /*
2463   stop a remote node
2464  */
2465 static int control_stop(struct ctdb_context *ctdb, int argc, const char **argv)
2466 {
2467         int ret;
2468         struct ctdb_node_map *nodemap=NULL;
2469
2470         do {
2471                 ret = ctdb_ctrl_stop_node(ctdb, TIMELIMIT(), options.pnn);
2472                 if (ret != 0) {
2473                         DEBUG(DEBUG_ERR, ("Unable to stop node %u   try again\n", options.pnn));
2474                 }
2475         
2476                 sleep(1);
2477
2478                 /* read the nodemap and verify the change took effect */
2479                 if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
2480                         DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2481                         exit(10);
2482                 }
2483
2484         } while (!(nodemap->nodes[options.pnn].flags & NODE_FLAGS_STOPPED));
2485         ret = control_ipreallocate(ctdb, argc, argv);
2486         if (ret != 0) {
2487                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u\n", options.pnn));
2488                 return ret;
2489         }
2490
2491         return 0;
2492 }
2493
2494 /*
2495   restart a stopped remote node
2496  */
2497 static int control_continue(struct ctdb_context *ctdb, int argc, const char **argv)
2498 {
2499         int ret;
2500
2501         struct ctdb_node_map *nodemap=NULL;
2502
2503         do {
2504                 ret = ctdb_ctrl_continue_node(ctdb, TIMELIMIT(), options.pnn);
2505                 if (ret != 0) {
2506                         DEBUG(DEBUG_ERR, ("Unable to continue node %u\n", options.pnn));
2507                         return ret;
2508                 }
2509         
2510                 sleep(1);
2511
2512                 /* read the nodemap and verify the change took effect */
2513                 if (ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap) != 0) {
2514                         DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2515                         exit(10);
2516                 }
2517
2518         } while (nodemap->nodes[options.pnn].flags & NODE_FLAGS_STOPPED);
2519         ret = control_ipreallocate(ctdb, argc, argv);
2520         if (ret != 0) {
2521                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u\n", options.pnn));
2522                 return ret;
2523         }
2524
2525         return 0;
2526 }
2527
2528 static uint32_t get_generation(struct ctdb_context *ctdb)
2529 {
2530         struct ctdb_vnn_map *vnnmap=NULL;
2531         int ret;
2532
2533         /* wait until the recmaster is not in recovery mode */
2534         while (1) {
2535                 uint32_t recmode, recmaster;
2536                 
2537                 if (vnnmap != NULL) {
2538                         talloc_free(vnnmap);
2539                         vnnmap = NULL;
2540                 }
2541
2542                 /* get the recmaster */
2543                 if (!ctdb_getrecmaster(ctdb_connection, CTDB_CURRENT_NODE, &recmaster)) {
2544                         DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
2545                         exit(10);
2546                 }
2547
2548                 /* get recovery mode */
2549                 if (!ctdb_getrecmode(ctdb_connection, recmaster, &recmode)) {
2550                         DEBUG(DEBUG_ERR, ("Unable to get recmode from node %u\n", options.pnn));
2551                         exit(10);
2552                 }
2553
2554                 /* get the current generation number */
2555                 ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), recmaster, ctdb, &vnnmap);
2556                 if (ret != 0) {
2557                         DEBUG(DEBUG_ERR, ("Unable to get vnnmap from recmaster (%u)\n", recmaster));
2558                         exit(10);
2559                 }
2560
2561                 if ((recmode == CTDB_RECOVERY_NORMAL)
2562                 &&  (vnnmap->generation != 1)){
2563                         return vnnmap->generation;
2564                 }
2565                 sleep(1);
2566         }
2567 }
2568
2569 /*
2570   ban a node from the cluster
2571  */
2572 static int control_ban(struct ctdb_context *ctdb, int argc, const char **argv)
2573 {
2574         int ret;
2575         struct ctdb_node_map *nodemap=NULL;
2576         struct ctdb_ban_time bantime;
2577
2578         if (argc < 1) {
2579                 usage();
2580         }
2581         
2582         /* verify the node exists */
2583         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
2584         if (ret != 0) {
2585                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2586                 return ret;
2587         }
2588
2589         if (nodemap->nodes[options.pnn].flags & NODE_FLAGS_BANNED) {
2590                 DEBUG(DEBUG_ERR,("Node %u is already banned.\n", options.pnn));
2591                 return -1;
2592         }
2593
2594         bantime.pnn  = options.pnn;
2595         bantime.time = strtoul(argv[0], NULL, 0);
2596
2597         ret = ctdb_ctrl_set_ban(ctdb, TIMELIMIT(), options.pnn, &bantime);
2598         if (ret != 0) {
2599                 DEBUG(DEBUG_ERR,("Banning node %d for %d seconds failed.\n", bantime.pnn, bantime.time));
2600                 return -1;
2601         }       
2602
2603         ret = control_ipreallocate(ctdb, argc, argv);
2604         if (ret != 0) {
2605                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u\n", options.pnn));
2606                 return ret;
2607         }
2608
2609         return 0;
2610 }
2611
2612
2613 /*
2614   unban a node from the cluster
2615  */
2616 static int control_unban(struct ctdb_context *ctdb, int argc, const char **argv)
2617 {
2618         int ret;
2619         struct ctdb_node_map *nodemap=NULL;
2620         struct ctdb_ban_time bantime;
2621
2622         /* verify the node exists */
2623         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
2624         if (ret != 0) {
2625                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2626                 return ret;
2627         }
2628
2629         if (!(nodemap->nodes[options.pnn].flags & NODE_FLAGS_BANNED)) {
2630                 DEBUG(DEBUG_ERR,("Node %u is not banned.\n", options.pnn));
2631                 return -1;
2632         }
2633
2634         bantime.pnn  = options.pnn;
2635         bantime.time = 0;
2636
2637         ret = ctdb_ctrl_set_ban(ctdb, TIMELIMIT(), options.pnn, &bantime);
2638         if (ret != 0) {
2639                 DEBUG(DEBUG_ERR,("Unbanning node %d failed.\n", bantime.pnn));
2640                 return -1;
2641         }       
2642
2643         ret = control_ipreallocate(ctdb, argc, argv);
2644         if (ret != 0) {
2645                 DEBUG(DEBUG_ERR, ("IP Reallocate failed on node %u\n", options.pnn));
2646                 return ret;
2647         }
2648
2649         return 0;
2650 }
2651
2652
2653 /*
2654   show ban information for a node
2655  */
2656 static int control_showban(struct ctdb_context *ctdb, int argc, const char **argv)
2657 {
2658         int ret;
2659         struct ctdb_node_map *nodemap=NULL;
2660         struct ctdb_ban_time *bantime;
2661
2662         /* verify the node exists */
2663         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
2664         if (ret != 0) {
2665                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
2666                 return ret;
2667         }
2668
2669         ret = ctdb_ctrl_get_ban(ctdb, TIMELIMIT(), options.pnn, ctdb, &bantime);
2670         if (ret != 0) {
2671                 DEBUG(DEBUG_ERR,("Showing ban info for node %d failed.\n", options.pnn));
2672                 return -1;
2673         }       
2674
2675         if (bantime->time == 0) {
2676                 printf("Node %u is not banned\n", bantime->pnn);
2677         } else {
2678                 printf("Node %u is banned banned for %d seconds\n", bantime->pnn, bantime->time);
2679         }
2680
2681         return 0;
2682 }
2683
2684 /*
2685   shutdown a daemon
2686  */
2687 static int control_shutdown(struct ctdb_context *ctdb, int argc, const char **argv)
2688 {
2689         int ret;
2690
2691         ret = ctdb_ctrl_shutdown(ctdb, TIMELIMIT(), options.pnn);
2692         if (ret != 0) {
2693                 DEBUG(DEBUG_ERR, ("Unable to shutdown node %u\n", options.pnn));
2694                 return ret;
2695         }
2696
2697         return 0;
2698 }
2699
2700 /*
2701   trigger a recovery
2702  */
2703 static int control_recover(struct ctdb_context *ctdb, int argc, const char **argv)
2704 {
2705         int ret;
2706         uint32_t generation, next_generation;
2707
2708         /* record the current generation number */
2709         generation = get_generation(ctdb);
2710
2711         ret = ctdb_ctrl_freeze_priority(ctdb, TIMELIMIT(), options.pnn, 1);
2712         if (ret != 0) {
2713                 DEBUG(DEBUG_ERR, ("Unable to freeze node\n"));
2714                 return ret;
2715         }
2716
2717         ret = ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
2718         if (ret != 0) {
2719                 DEBUG(DEBUG_ERR, ("Unable to set recovery mode\n"));
2720                 return ret;
2721         }
2722
2723         /* wait until we are in a new generation */
2724         while (1) {
2725                 next_generation = get_generation(ctdb);
2726                 if (next_generation != generation) {
2727                         return 0;
2728                 }
2729                 sleep(1);
2730         }
2731
2732         return 0;
2733 }
2734
2735
2736 /*
2737   display monitoring mode of a remote node
2738  */
2739 static int control_getmonmode(struct ctdb_context *ctdb, int argc, const char **argv)
2740 {
2741         uint32_t monmode;
2742         int ret;
2743
2744         ret = ctdb_ctrl_getmonmode(ctdb, TIMELIMIT(), options.pnn, &monmode);
2745         if (ret != 0) {
2746                 DEBUG(DEBUG_ERR, ("Unable to get monmode from node %u\n", options.pnn));
2747                 return ret;
2748         }
2749         if (!options.machinereadable){
2750                 printf("Monitoring mode:%s (%d)\n",monmode==CTDB_MONITORING_ACTIVE?"ACTIVE":"DISABLED",monmode);
2751         } else {
2752                 printf(":mode:\n");
2753                 printf(":%d:\n",monmode);
2754         }
2755         return 0;
2756 }
2757
2758
2759 /*
2760   display capabilities of a remote node
2761  */
2762 static int control_getcapabilities(struct ctdb_context *ctdb, int argc, const char **argv)
2763 {
2764         uint32_t capabilities;
2765         int ret;
2766
2767         ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), options.pnn, &capabilities);
2768         if (ret != 0) {
2769                 DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", options.pnn));
2770                 return ret;
2771         }
2772         
2773         if (!options.machinereadable){
2774                 printf("RECMASTER: %s\n", (capabilities&CTDB_CAP_RECMASTER)?"YES":"NO");
2775                 printf("LMASTER: %s\n", (capabilities&CTDB_CAP_LMASTER)?"YES":"NO");
2776                 printf("LVS: %s\n", (capabilities&CTDB_CAP_LVS)?"YES":"NO");
2777                 printf("NATGW: %s\n", (capabilities&CTDB_CAP_NATGW)?"YES":"NO");
2778         } else {
2779                 printf(":RECMASTER:LMASTER:LVS:NATGW:\n");
2780                 printf(":%d:%d:%d:%d:\n",
2781                         !!(capabilities&CTDB_CAP_RECMASTER),
2782                         !!(capabilities&CTDB_CAP_LMASTER),
2783                         !!(capabilities&CTDB_CAP_LVS),
2784                         !!(capabilities&CTDB_CAP_NATGW));
2785         }
2786         return 0;
2787 }
2788
2789 /*
2790   display lvs configuration
2791  */
2792 static int control_lvs(struct ctdb_context *ctdb, int argc, const char **argv)
2793 {
2794         uint32_t *capabilities;
2795         struct ctdb_node_map *nodemap=NULL;
2796         int i, ret;
2797         int healthy_count = 0;
2798
2799         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb, &nodemap);
2800         if (ret != 0) {
2801                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
2802                 return ret;
2803         }
2804
2805         capabilities = talloc_array(ctdb, uint32_t, nodemap->num);
2806         CTDB_NO_MEMORY(ctdb, capabilities);
2807         
2808         /* collect capabilities for all connected nodes */
2809         for (i=0; i<nodemap->num; i++) {
2810                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
2811                         continue;
2812                 }
2813                 if (nodemap->nodes[i].flags & NODE_FLAGS_PERMANENTLY_DISABLED) {
2814                         continue;
2815                 }
2816         
2817                 ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), i, &capabilities[i]);
2818                 if (ret != 0) {
2819                         DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", i));
2820                         return ret;
2821                 }
2822
2823                 if (!(capabilities[i] & CTDB_CAP_LVS)) {
2824                         continue;
2825                 }
2826
2827                 if (!(nodemap->nodes[i].flags & NODE_FLAGS_UNHEALTHY)) {
2828                         healthy_count++;
2829                 }
2830         }
2831
2832         /* Print all LVS nodes */
2833         for (i=0; i<nodemap->num; i++) {
2834                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
2835                         continue;
2836                 }
2837                 if (nodemap->nodes[i].flags & NODE_FLAGS_PERMANENTLY_DISABLED) {
2838                         continue;
2839                 }
2840                 if (!(capabilities[i] & CTDB_CAP_LVS)) {
2841                         continue;
2842                 }
2843
2844                 if (healthy_count != 0) {
2845                         if (nodemap->nodes[i].flags & NODE_FLAGS_UNHEALTHY) {
2846                                 continue;
2847                         }
2848                 }
2849
2850                 printf("%d:%s\n", i, 
2851                         ctdb_addr_to_str(&nodemap->nodes[i].addr));
2852         }
2853
2854         return 0;
2855 }
2856
2857 /*
2858   display who is the lvs master
2859  */
2860 static int control_lvsmaster(struct ctdb_context *ctdb, int argc, const char **argv)
2861 {
2862         uint32_t *capabilities;
2863         struct ctdb_node_map *nodemap=NULL;
2864         int i, ret;
2865         int healthy_count = 0;
2866
2867         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb, &nodemap);
2868         if (ret != 0) {
2869                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
2870                 return ret;
2871         }
2872
2873         capabilities = talloc_array(ctdb, uint32_t, nodemap->num);
2874         CTDB_NO_MEMORY(ctdb, capabilities);
2875         
2876         /* collect capabilities for all connected nodes */
2877         for (i=0; i<nodemap->num; i++) {
2878                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
2879                         continue;
2880                 }
2881                 if (nodemap->nodes[i].flags & NODE_FLAGS_PERMANENTLY_DISABLED) {
2882                         continue;
2883                 }
2884         
2885                 ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), i, &capabilities[i]);
2886                 if (ret != 0) {
2887                         DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", i));
2888                         return ret;
2889                 }
2890
2891                 if (!(capabilities[i] & CTDB_CAP_LVS)) {
2892                         continue;
2893                 }
2894
2895                 if (!(nodemap->nodes[i].flags & NODE_FLAGS_UNHEALTHY)) {
2896                         healthy_count++;
2897                 }
2898         }
2899
2900         /* find and show the lvsmaster */
2901         for (i=0; i<nodemap->num; i++) {
2902                 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
2903                         continue;
2904                 }
2905                 if (nodemap->nodes[i].flags & NODE_FLAGS_PERMANENTLY_DISABLED) {
2906                         continue;
2907                 }
2908                 if (!(capabilities[i] & CTDB_CAP_LVS)) {
2909                         continue;
2910                 }
2911
2912                 if (healthy_count != 0) {
2913                         if (nodemap->nodes[i].flags & NODE_FLAGS_UNHEALTHY) {
2914                                 continue;
2915                         }
2916                 }
2917
2918                 if (options.machinereadable){
2919                         printf("%d\n", i);
2920                 } else {
2921                         printf("Node %d is LVS master\n", i);
2922                 }
2923                 return 0;
2924         }
2925
2926         printf("There is no LVS master\n");
2927         return -1;
2928 }
2929
2930 /*
2931   disable monitoring on a  node
2932  */
2933 static int control_disable_monmode(struct ctdb_context *ctdb, int argc, const char **argv)
2934 {
2935         
2936         int ret;
2937
2938         ret = ctdb_ctrl_disable_monmode(ctdb, TIMELIMIT(), options.pnn);
2939         if (ret != 0) {
2940                 DEBUG(DEBUG_ERR, ("Unable to disable monmode on node %u\n", options.pnn));
2941                 return ret;
2942         }
2943         printf("Monitoring mode:%s\n","DISABLED");
2944
2945         return 0;
2946 }
2947
2948 /*
2949   enable monitoring on a  node
2950  */
2951 static int control_enable_monmode(struct ctdb_context *ctdb, int argc, const char **argv)
2952 {
2953         
2954         int ret;
2955
2956         ret = ctdb_ctrl_enable_monmode(ctdb, TIMELIMIT(), options.pnn);
2957         if (ret != 0) {
2958                 DEBUG(DEBUG_ERR, ("Unable to enable monmode on node %u\n", options.pnn));
2959                 return ret;
2960         }
2961         printf("Monitoring mode:%s\n","ACTIVE");
2962
2963         return 0;
2964 }
2965
2966 /*
2967   display remote list of keys/data for a db
2968  */
2969 static int control_catdb(struct ctdb_context *ctdb, int argc, const char **argv)
2970 {
2971         const char *db_name;
2972         struct ctdb_db_context *ctdb_db;
2973         int ret;
2974         bool persistent;
2975
2976         if (argc < 1) {
2977                 usage();
2978         }
2979
2980         db_name = argv[0];
2981
2982
2983         if (db_exists(ctdb, db_name, &persistent)) {
2984                 DEBUG(DEBUG_ERR,("Database '%s' does not exist\n", db_name));
2985                 return -1;
2986         }
2987
2988         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
2989
2990         if (ctdb_db == NULL) {
2991                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
2992                 return -1;
2993         }
2994
2995         /* traverse and dump the cluster tdb */
2996         ret = ctdb_dump_db(ctdb_db, stdout);
2997         if (ret == -1) {
2998                 DEBUG(DEBUG_ERR, ("Unable to dump database\n"));
2999                 DEBUG(DEBUG_ERR, ("Maybe try 'ctdb getdbstatus %s'"
3000                                   " and 'ctdb getvar AllowUnhealthyDBRead'\n",
3001                                   db_name));
3002                 return -1;
3003         }
3004         talloc_free(ctdb_db);
3005
3006         printf("Dumped %d records\n", ret);
3007         return 0;
3008 }
3009
3010 struct cattdb_data {
3011         struct ctdb_context *ctdb;
3012         uint32_t count;
3013 };
3014
3015 static int cattdb_traverse(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, void *private_data)
3016 {
3017         struct cattdb_data *d = private_data;
3018
3019         d->count++;
3020         
3021         return ctdb_dumpdb_record(d->ctdb, key, data, stdout);
3022 }
3023
3024 /*
3025   cat the local tdb database using same format as catdb
3026  */
3027 static int control_cattdb(struct ctdb_context *ctdb, int argc, const char **argv)
3028 {
3029         const char *db_name;
3030         struct ctdb_db_context *ctdb_db;
3031         struct cattdb_data d;
3032         bool persistent;
3033
3034         if (argc < 1) {
3035                 usage();
3036         }
3037
3038         db_name = argv[0];
3039
3040
3041         if (db_exists(ctdb, db_name, &persistent)) {
3042                 DEBUG(DEBUG_ERR,("Database '%s' does not exist\n", db_name));
3043                 return -1;
3044         }
3045
3046         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, false, 0);
3047
3048         if (ctdb_db == NULL) {
3049                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3050                 return -1;
3051         }
3052
3053         /* traverse the local tdb */
3054         d.count = 0;
3055         d.ctdb  = ctdb;
3056         if (tdb_traverse_read(ctdb_db->ltdb->tdb, cattdb_traverse, &d) == -1) {
3057                 printf("Failed to cattdb data\n");
3058                 exit(10);
3059         }
3060         talloc_free(ctdb_db);
3061
3062         printf("Dumped %d records\n", d.count);
3063         return 0;
3064 }
3065
3066 /*
3067   display the content of a database key
3068  */
3069 static int control_readkey(struct ctdb_context *ctdb, int argc, const char **argv)
3070 {
3071         const char *db_name;
3072         struct ctdb_db_context *ctdb_db;
3073         struct ctdb_record_handle *h;
3074         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3075         TDB_DATA key, data;
3076         bool persistent;
3077
3078         if (argc < 2) {
3079                 usage();
3080         }
3081
3082         db_name = argv[0];
3083
3084
3085         if (db_exists(ctdb, db_name, &persistent)) {
3086                 DEBUG(DEBUG_ERR,("Database '%s' does not exist\n", db_name));
3087                 return -1;
3088         }
3089
3090         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
3091
3092         if (ctdb_db == NULL) {
3093                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3094                 return -1;
3095         }
3096
3097         key.dptr  = discard_const(argv[1]);
3098         key.dsize = strlen((char *)key.dptr);
3099  
3100         h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data);
3101         if (h == NULL) {
3102                 printf("Failed to fetch record '%s' on node %d\n", 
3103                         (const char *)key.dptr, ctdb_get_pnn(ctdb));
3104                 talloc_free(tmp_ctx);
3105                 exit(10);
3106         }
3107
3108         printf("Data: size:%d ptr:[%s]\n", (int)data.dsize, data.dptr);
3109
3110         talloc_free(ctdb_db);
3111         talloc_free(tmp_ctx);
3112         return 0;
3113 }
3114
3115 /*
3116   display the content of a database key
3117  */
3118 static int control_writekey(struct ctdb_context *ctdb, int argc, const char **argv)
3119 {
3120         const char *db_name;
3121         struct ctdb_db_context *ctdb_db;
3122         struct ctdb_record_handle *h;
3123         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3124         TDB_DATA key, data;
3125         bool persistent;
3126
3127         if (argc < 3) {
3128                 usage();
3129         }
3130
3131         db_name = argv[0];
3132
3133
3134         if (db_exists(ctdb, db_name, &persistent)) {
3135                 DEBUG(DEBUG_ERR,("Database '%s' does not exist\n", db_name));
3136                 return -1;
3137         }
3138
3139         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
3140
3141         if (ctdb_db == NULL) {
3142                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3143                 return -1;
3144         }
3145
3146         key.dptr  = discard_const(argv[1]);
3147         key.dsize = strlen((char *)key.dptr);
3148  
3149         h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data);
3150         if (h == NULL) {
3151                 printf("Failed to fetch record '%s' on node %d\n", 
3152                         (const char *)key.dptr, ctdb_get_pnn(ctdb));
3153                 talloc_free(tmp_ctx);
3154                 exit(10);
3155         }
3156
3157         data.dptr  = discard_const(argv[2]);
3158         data.dsize = strlen((char *)data.dptr);
3159  
3160         if (ctdb_record_store(h, data) != 0) {
3161                 printf("Failed to store record\n");
3162         }
3163
3164         talloc_free(h);
3165         talloc_free(ctdb_db);
3166         talloc_free(tmp_ctx);
3167         return 0;
3168 }
3169
3170 /*
3171   fetch a record from a persistent database
3172  */
3173 static int control_pfetch(struct ctdb_context *ctdb, int argc, const char **argv)
3174 {
3175         const char *db_name;
3176         struct ctdb_db_context *ctdb_db;
3177         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3178         struct ctdb_transaction_handle *h;
3179         TDB_DATA key, data;
3180         int fd, ret;
3181         bool persistent;
3182
3183         if (argc < 2) {
3184                 talloc_free(tmp_ctx);
3185                 usage();
3186         }
3187
3188         db_name = argv[0];
3189
3190
3191         if (db_exists(ctdb, db_name, &persistent)) {
3192                 DEBUG(DEBUG_ERR,("Database '%s' does not exist\n", db_name));
3193                 talloc_free(tmp_ctx);
3194                 return -1;
3195         }
3196
3197         if (!persistent) {
3198                 DEBUG(DEBUG_ERR,("Database '%s' is not persistent\n", db_name));
3199                 talloc_free(tmp_ctx);
3200                 return -1;
3201         }
3202
3203         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
3204
3205         if (ctdb_db == NULL) {
3206                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3207                 talloc_free(tmp_ctx);
3208                 return -1;
3209         }
3210
3211         h = ctdb_transaction_start(ctdb_db, tmp_ctx);
3212         if (h == NULL) {
3213                 DEBUG(DEBUG_ERR,("Failed to start transaction on database %s\n", db_name));
3214                 talloc_free(tmp_ctx);
3215                 return -1;
3216         }
3217
3218         key.dptr  = discard_const(argv[1]);
3219         key.dsize = strlen(argv[1]);
3220         ret = ctdb_transaction_fetch(h, tmp_ctx, key, &data);
3221         if (ret != 0) {
3222                 DEBUG(DEBUG_ERR,("Failed to fetch record\n"));
3223                 talloc_free(tmp_ctx);
3224                 return -1;
3225         }
3226
3227         if (data.dsize == 0 || data.dptr == NULL) {
3228                 DEBUG(DEBUG_ERR,("Record is empty\n"));
3229                 talloc_free(tmp_ctx);
3230                 return -1;
3231         }
3232
3233         if (argc == 3) {
3234           fd = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0600);
3235                 if (fd == -1) {
3236                         DEBUG(DEBUG_ERR,("Failed to open output file %s\n", argv[2]));
3237                         talloc_free(tmp_ctx);
3238                         return -1;
3239                 }
3240                 write(fd, data.dptr, data.dsize);
3241                 close(fd);
3242         } else {
3243                 write(1, data.dptr, data.dsize);
3244         }
3245
3246         /* abort the transaction */
3247         talloc_free(h);
3248
3249
3250         talloc_free(tmp_ctx);
3251         return 0;
3252 }
3253
3254 /*
3255   fetch a record from a tdb-file
3256  */
3257 static int control_tfetch(struct ctdb_context *ctdb, int argc, const char **argv)
3258 {
3259         const char *tdb_file;
3260         TDB_CONTEXT *tdb;
3261         TDB_DATA key, data;
3262         int fd;
3263
3264         if (argc < 2) {
3265                 usage();
3266         }
3267
3268         tdb_file = argv[0];
3269
3270         tdb = tdb_open(tdb_file, 0, 0, O_RDONLY, 0);
3271         if (tdb == NULL) {
3272                 DEBUG(DEBUG_ERR,("Failed to open TDB file %s\n", tdb_file));
3273                 return -1;
3274         }
3275
3276         key.dptr  = discard_const(argv[1]);
3277         key.dsize = strlen(argv[1]);
3278         data = tdb_fetch(tdb, key);
3279         if (data.dptr == NULL || data.dsize < sizeof(struct ctdb_ltdb_header)) {
3280                 DEBUG(DEBUG_ERR,("Failed to read record %s from tdb %s\n", argv[1], tdb_file));
3281                 tdb_close(tdb);
3282                 return -1;
3283         }
3284
3285         tdb_close(tdb);
3286
3287         if (argc == 3) {
3288           fd = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0600);
3289                 if (fd == -1) {
3290                         DEBUG(DEBUG_ERR,("Failed to open output file %s\n", argv[2]));
3291                         return -1;
3292                 }
3293                 write(fd, data.dptr+sizeof(struct ctdb_ltdb_header), data.dsize-sizeof(struct ctdb_ltdb_header));
3294                 close(fd);
3295         } else {
3296                 write(1, data.dptr+sizeof(struct ctdb_ltdb_header), data.dsize-sizeof(struct ctdb_ltdb_header));
3297         }
3298
3299         return 0;
3300 }
3301
3302 /*
3303   write a record to a persistent database
3304  */
3305 static int control_pstore(struct ctdb_context *ctdb, int argc, const char **argv)
3306 {
3307         const char *db_name;
3308         struct ctdb_db_context *ctdb_db;
3309         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3310         struct ctdb_transaction_handle *h;
3311         struct stat st;
3312         TDB_DATA key, data;
3313         int fd, ret;
3314
3315         if (argc < 3) {
3316                 talloc_free(tmp_ctx);
3317                 usage();
3318         }
3319
3320         fd = open(argv[2], O_RDONLY);
3321         if (fd == -1) {
3322                 DEBUG(DEBUG_ERR,("Failed to open file containing record data : %s  %s\n", argv[2], strerror(errno)));
3323                 talloc_free(tmp_ctx);
3324                 return -1;
3325         }
3326         
3327         ret = fstat(fd, &st);
3328         if (ret == -1) {
3329                 DEBUG(DEBUG_ERR,("fstat of file %s failed: %s\n", argv[2], strerror(errno)));
3330                 close(fd);
3331                 talloc_free(tmp_ctx);
3332                 return -1;
3333         }
3334
3335         if (!S_ISREG(st.st_mode)) {
3336                 DEBUG(DEBUG_ERR,("Not a regular file %s\n", argv[2]));
3337                 close(fd);
3338                 talloc_free(tmp_ctx);
3339                 return -1;
3340         }
3341
3342         data.dsize = st.st_size;
3343         if (data.dsize == 0) {
3344                 data.dptr  = NULL;
3345         } else {
3346                 data.dptr = talloc_size(tmp_ctx, data.dsize);
3347                 if (data.dptr == NULL) {
3348                         DEBUG(DEBUG_ERR,("Failed to talloc %d of memory to store record data\n", (int)data.dsize));
3349                         close(fd);
3350                         talloc_free(tmp_ctx);
3351                         return -1;
3352                 }
3353                 ret = read(fd, data.dptr, data.dsize);
3354                 if (ret != data.dsize) {
3355                         DEBUG(DEBUG_ERR,("Failed to read %d bytes of record data\n", (int)data.dsize));
3356                         close(fd);
3357                         talloc_free(tmp_ctx);
3358                         return -1;
3359                 }
3360         }
3361         close(fd);
3362
3363
3364         db_name = argv[0];
3365
3366         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, true, 0);
3367         if (ctdb_db == NULL) {
3368                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3369                 talloc_free(tmp_ctx);
3370                 return -1;
3371         }
3372
3373         h = ctdb_transaction_start(ctdb_db, tmp_ctx);
3374         if (h == NULL) {
3375                 DEBUG(DEBUG_ERR,("Failed to start transaction on database %s\n", db_name));
3376                 talloc_free(tmp_ctx);
3377                 return -1;
3378         }
3379
3380         key.dptr  = discard_const(argv[1]);
3381         key.dsize = strlen(argv[1]);
3382         ret = ctdb_transaction_store(h, key, data);
3383         if (ret != 0) {
3384                 DEBUG(DEBUG_ERR,("Failed to store record\n"));
3385                 talloc_free(tmp_ctx);
3386                 return -1;
3387         }
3388
3389         ret = ctdb_transaction_commit(h);
3390         if (ret != 0) {
3391                 DEBUG(DEBUG_ERR,("Failed to commit transaction\n"));
3392                 talloc_free(tmp_ctx);
3393                 return -1;
3394         }
3395
3396
3397         talloc_free(tmp_ctx);
3398         return 0;
3399 }
3400
3401 /*
3402   check if a service is bound to a port or not
3403  */
3404 static int control_chktcpport(struct ctdb_context *ctdb, int argc, const char **argv)
3405 {
3406         int s, ret;
3407         unsigned v;
3408         int port;
3409         struct sockaddr_in sin;
3410
3411         if (argc != 1) {
3412                 printf("Use: ctdb chktcport <port>\n");
3413                 return EINVAL;
3414         }
3415
3416         port = atoi(argv[0]);
3417
3418         s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
3419         if (s == -1) {
3420                 printf("Failed to open local socket\n");
3421                 return errno;
3422         }
3423
3424         v = fcntl(s, F_GETFL, 0);
3425         fcntl(s, F_SETFL, v | O_NONBLOCK);
3426
3427         bzero(&sin, sizeof(sin));
3428         sin.sin_family = PF_INET;
3429         sin.sin_port   = htons(port);
3430         ret = bind(s, (struct sockaddr *)&sin, sizeof(sin));
3431         close(s);
3432         if (ret == -1) {
3433                 printf("Failed to bind to local socket: %d %s\n", errno, strerror(errno));
3434                 return errno;
3435         }
3436
3437         return 0;
3438 }
3439
3440
3441
3442 static void log_handler(struct ctdb_context *ctdb, uint64_t srvid, 
3443                              TDB_DATA data, void *private_data)
3444 {
3445         DEBUG(DEBUG_ERR,("Log data received\n"));
3446         if (data.dsize > 0) {
3447                 printf("%s", data.dptr);
3448         }
3449
3450         exit(0);
3451 }
3452
3453 /*
3454   display a list of log messages from the in memory ringbuffer
3455  */
3456 static int control_getlog(struct ctdb_context *ctdb, int argc, const char **argv)
3457 {
3458         int ret;
3459         int32_t res;
3460         struct ctdb_get_log_addr log_addr;
3461         TDB_DATA data;
3462         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3463         char *errmsg;
3464         struct timeval tv;
3465
3466         if (argc != 1) {
3467                 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
3468                 talloc_free(tmp_ctx);
3469                 return -1;
3470         }
3471
3472         log_addr.pnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE);
3473         log_addr.srvid = getpid();
3474         if (isalpha(argv[0][0]) || argv[0][0] == '-') { 
3475                 log_addr.level = get_debug_by_desc(argv[0]);
3476         } else {
3477                 log_addr.level = strtol(argv[0], NULL, 0);
3478         }
3479
3480
3481         data.dptr = (unsigned char *)&log_addr;
3482         data.dsize = sizeof(log_addr);
3483
3484         DEBUG(DEBUG_ERR, ("Pulling logs from node %u\n", options.pnn));
3485
3486         ctdb_client_set_message_handler(ctdb, log_addr.srvid, log_handler, NULL);
3487         sleep(1);
3488
3489         DEBUG(DEBUG_ERR,("Listen for response on %d\n", (int)log_addr.srvid));
3490
3491         ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_GET_LOG,
3492                            0, data, tmp_ctx, NULL, &res, NULL, &errmsg);
3493         if (ret != 0 || res != 0) {
3494                 DEBUG(DEBUG_ERR,("Failed to get logs - %s\n", errmsg));
3495                 talloc_free(tmp_ctx);
3496                 return -1;
3497         }
3498
3499
3500         tv = timeval_current();
3501         /* this loop will terminate when we have received the reply */
3502         while (timeval_elapsed(&tv) < 3.0) {    
3503                 event_loop_once(ctdb->ev);
3504         }
3505
3506         DEBUG(DEBUG_INFO,("Timed out waiting for log data.\n"));
3507
3508         talloc_free(tmp_ctx);
3509         return 0;
3510 }
3511
3512 /*
3513   clear the in memory log area
3514  */
3515 static int control_clearlog(struct ctdb_context *ctdb, int argc, const char **argv)
3516 {
3517         int ret;
3518         int32_t res;
3519         char *errmsg;
3520         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3521
3522         ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_CLEAR_LOG,
3523                            0, tdb_null, tmp_ctx, NULL, &res, NULL, &errmsg);
3524         if (ret != 0 || res != 0) {
3525                 DEBUG(DEBUG_ERR,("Failed to clear logs\n"));
3526                 talloc_free(tmp_ctx);
3527                 return -1;
3528         }
3529
3530         talloc_free(tmp_ctx);
3531         return 0;
3532 }
3533
3534
3535
3536 /*
3537   display a list of the databases on a remote ctdb
3538  */
3539 static int control_getdbmap(struct ctdb_context *ctdb, int argc, const char **argv)
3540 {
3541         int i, ret;
3542         struct ctdb_dbid_map *dbmap=NULL;
3543
3544         ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, ctdb, &dbmap);
3545         if (ret != 0) {
3546                 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n", options.pnn));
3547                 return ret;
3548         }
3549
3550         if(options.machinereadable){
3551                 printf(":ID:Name:Path:Persistent:Unhealthy:ReadOnly:\n");
3552                 for(i=0;i<dbmap->num;i++){
3553                         const char *path;
3554                         const char *name;
3555                         const char *health;
3556                         bool persistent;
3557                         bool readonly;
3558
3559                         ctdb_ctrl_getdbpath(ctdb, TIMELIMIT(), options.pnn,
3560                                             dbmap->dbs[i].dbid, ctdb, &path);
3561                         ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn,
3562                                             dbmap->dbs[i].dbid, ctdb, &name);
3563                         ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn,
3564                                               dbmap->dbs[i].dbid, ctdb, &health);
3565                         persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
3566                         readonly   = dbmap->dbs[i].flags & CTDB_DB_FLAGS_READONLY;
3567                         printf(":0x%08X:%s:%s:%d:%d:%d:\n",
3568                                dbmap->dbs[i].dbid, name, path,
3569                                !!(persistent), !!(health), !!(readonly));
3570                 }
3571                 return 0;
3572         }
3573
3574         printf("Number of databases:%d\n", dbmap->num);
3575         for(i=0;i<dbmap->num;i++){
3576                 const char *path;
3577                 const char *name;
3578                 const char *health;
3579                 bool persistent;
3580                 bool readonly;
3581
3582                 ctdb_ctrl_getdbpath(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &path);
3583                 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &name);
3584                 ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &health);
3585                 persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
3586                 readonly   = dbmap->dbs[i].flags & CTDB_DB_FLAGS_READONLY;
3587                 printf("dbid:0x%08x name:%s path:%s%s%s%s\n",
3588                        dbmap->dbs[i].dbid, name, path,
3589                        persistent?" PERSISTENT":"",
3590                        readonly?" READONLY":"",
3591                        health?" UNHEALTHY":"");
3592         }
3593
3594         return 0;
3595 }
3596
3597 /*
3598   display the status of a database on a remote ctdb
3599  */
3600 static int control_getdbstatus(struct ctdb_context *ctdb, int argc, const char **argv)
3601 {
3602         int i, ret;
3603         struct ctdb_dbid_map *dbmap=NULL;
3604         const char *db_name;
3605
3606         if (argc < 1) {
3607                 usage();
3608         }
3609
3610         db_name = argv[0];
3611
3612         ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, ctdb, &dbmap);
3613         if (ret != 0) {
3614                 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n", options.pnn));
3615                 return ret;
3616         }
3617
3618         for(i=0;i<dbmap->num;i++){
3619                 const char *path;
3620                 const char *name;
3621                 const char *health;
3622                 bool persistent;
3623                 bool readonly;
3624
3625                 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &name);
3626                 if (strcmp(name, db_name) != 0) {
3627                         continue;
3628                 }
3629
3630                 ctdb_ctrl_getdbpath(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &path);
3631                 ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &health);
3632                 persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
3633                 readonly   = dbmap->dbs[i].flags & CTDB_DB_FLAGS_READONLY;
3634                 printf("dbid: 0x%08x\nname: %s\npath: %s\nPERSISTENT: %s\nREADONLY: %s\nHEALTH: %s\n",
3635                        dbmap->dbs[i].dbid, name, path,
3636                        persistent?"yes":"no",
3637                        readonly?"yes":"no",
3638                        health?health:"OK");
3639                 return 0;
3640         }
3641
3642         DEBUG(DEBUG_ERR, ("db %s doesn't exist on node %u\n", db_name, options.pnn));
3643         return 0;
3644 }
3645
3646 /*
3647   check if the local node is recmaster or not
3648   it will return 1 if this node is the recmaster and 0 if it is not
3649   or if the local ctdb daemon could not be contacted
3650  */
3651 static int control_isnotrecmaster(struct ctdb_context *ctdb, int argc, const char **argv)
3652 {
3653         uint32_t mypnn, recmaster;
3654
3655         mypnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), options.pnn);
3656         if (mypnn == -1) {
3657                 printf("Failed to get pnn of node\n");
3658                 return 1;
3659         }
3660
3661         if (!ctdb_getrecmaster(ctdb_connection, options.pnn, &recmaster)) {
3662                 printf("Failed to get the recmaster\n");
3663                 return 1;
3664         }
3665
3666         if (recmaster != mypnn) {
3667                 printf("this node is not the recmaster\n");
3668                 return 1;
3669         }
3670
3671         printf("this node is the recmaster\n");
3672         return 0;
3673 }
3674
3675 /*
3676   ping a node
3677  */
3678 static int control_ping(struct ctdb_context *ctdb, int argc, const char **argv)
3679 {
3680         int ret;
3681         struct timeval tv = timeval_current();
3682         ret = ctdb_ctrl_ping(ctdb, options.pnn);
3683         if (ret == -1) {
3684                 printf("Unable to get ping response from node %u\n", options.pnn);
3685                 return -1;
3686         } else {
3687                 printf("response from %u time=%.6f sec  (%d clients)\n", 
3688                        options.pnn, timeval_elapsed(&tv), ret);
3689         }
3690         return 0;
3691 }
3692
3693
3694 /*
3695   get a tunable
3696  */
3697 static int control_getvar(struct ctdb_context *ctdb, int argc, const char **argv)
3698 {
3699         const char *name;
3700         uint32_t value;
3701         int ret;
3702
3703         if (argc < 1) {
3704                 usage();
3705         }
3706
3707         name = argv[0];
3708         ret = ctdb_ctrl_get_tunable(ctdb, TIMELIMIT(), options.pnn, name, &value);
3709         if (ret == -1) {
3710                 DEBUG(DEBUG_ERR, ("Unable to get tunable variable '%s'\n", name));
3711                 return -1;
3712         }
3713
3714         printf("%-19s = %u\n", name, value);
3715         return 0;
3716 }
3717
3718 /*
3719   set a tunable
3720  */
3721 static int control_setvar(struct ctdb_context *ctdb, int argc, const char **argv)
3722 {
3723         const char *name;
3724         uint32_t value;
3725         int ret;
3726
3727         if (argc < 2) {
3728                 usage();
3729         }
3730
3731         name = argv[0];
3732         value = strtoul(argv[1], NULL, 0);
3733
3734         ret = ctdb_ctrl_set_tunable(ctdb, TIMELIMIT(), options.pnn, name, value);
3735         if (ret == -1) {
3736                 DEBUG(DEBUG_ERR, ("Unable to set tunable variable '%s'\n", name));
3737                 return -1;
3738         }
3739         return 0;
3740 }
3741
3742 /*
3743   list all tunables
3744  */
3745 static int control_listvars(struct ctdb_context *ctdb, int argc, const char **argv)
3746 {
3747         uint32_t count;
3748         const char **list;
3749         int ret, i;
3750
3751         ret = ctdb_ctrl_list_tunables(ctdb, TIMELIMIT(), options.pnn, ctdb, &list, &count);
3752         if (ret == -1) {
3753                 DEBUG(DEBUG_ERR, ("Unable to list tunable variables\n"));
3754                 return -1;
3755         }
3756
3757         for (i=0;i<count;i++) {
3758                 control_getvar(ctdb, 1, &list[i]);
3759         }
3760
3761         talloc_free(list);
3762         
3763         return 0;
3764 }
3765
3766 /*
3767   display debug level on a node
3768  */
3769 static int control_getdebug(struct ctdb_context *ctdb, int argc, const char **argv)
3770 {
3771         int ret;
3772         int32_t level;
3773
3774         ret = ctdb_ctrl_get_debuglevel(ctdb, options.pnn, &level);
3775         if (ret != 0) {
3776                 DEBUG(DEBUG_ERR, ("Unable to get debuglevel response from node %u\n", options.pnn));
3777                 return ret;
3778         } else {
3779                 if (options.machinereadable){
3780                         printf(":Name:Level:\n");
3781                         printf(":%s:%d:\n",get_debug_by_level(level),level);
3782                 } else {
3783                         printf("Node %u is at debug level %s (%d)\n", options.pnn, get_debug_by_level(level), level);
3784                 }
3785         }
3786         return 0;
3787 }
3788
3789 /*
3790   display reclock file of a node
3791  */
3792 static int control_getreclock(struct ctdb_context *ctdb, int argc, const char **argv)
3793 {
3794         int ret;
3795         const char *reclock;
3796
3797         ret = ctdb_ctrl_getreclock(ctdb, TIMELIMIT(), options.pnn, ctdb, &reclock);
3798         if (ret != 0) {
3799                 DEBUG(DEBUG_ERR, ("Unable to get reclock file from node %u\n", options.pnn));
3800                 return ret;
3801         } else {
3802                 if (options.machinereadable){
3803                         if (reclock != NULL) {
3804                                 printf("%s", reclock);
3805                         }
3806                 } else {
3807                         if (reclock == NULL) {
3808                                 printf("No reclock file used.\n");
3809                         } else {
3810                                 printf("Reclock file:%s\n", reclock);
3811                         }
3812                 }
3813         }
3814         return 0;
3815 }
3816
3817 /*
3818   set the reclock file of a node
3819  */
3820 static int control_setreclock(struct ctdb_context *ctdb, int argc, const char **argv)
3821 {
3822         int ret;
3823         const char *reclock;
3824
3825         if (argc == 0) {
3826                 reclock = NULL;
3827         } else if (argc == 1) {
3828                 reclock = argv[0];
3829         } else {
3830                 usage();
3831         }
3832
3833         ret = ctdb_ctrl_setreclock(ctdb, TIMELIMIT(), options.pnn, reclock);
3834         if (ret != 0) {
3835                 DEBUG(DEBUG_ERR, ("Unable to get reclock file from node %u\n", options.pnn));
3836                 return ret;
3837         }
3838         return 0;
3839 }
3840
3841 /*
3842   set the natgw state on/off
3843  */
3844 static int control_setnatgwstate(struct ctdb_context *ctdb, int argc, const char **argv)
3845 {
3846         int ret;
3847         uint32_t natgwstate;
3848
3849         if (argc == 0) {
3850                 usage();
3851         }
3852
3853         if (!strcmp(argv[0], "on")) {
3854                 natgwstate = 1;
3855         } else if (!strcmp(argv[0], "off")) {
3856                 natgwstate = 0;
3857         } else {
3858                 usage();
3859         }
3860
3861         ret = ctdb_ctrl_setnatgwstate(ctdb, TIMELIMIT(), options.pnn, natgwstate);
3862         if (ret != 0) {
3863                 DEBUG(DEBUG_ERR, ("Unable to set the natgw state for node %u\n", options.pnn));
3864                 return ret;
3865         }
3866
3867         return 0;
3868 }
3869
3870 /*
3871   set the lmaster role on/off
3872  */
3873 static int control_setlmasterrole(struct ctdb_context *ctdb, int argc, const char **argv)
3874 {
3875         int ret;
3876         uint32_t lmasterrole;
3877
3878         if (argc == 0) {
3879                 usage();
3880         }
3881
3882         if (!strcmp(argv[0], "on")) {
3883                 lmasterrole = 1;
3884         } else if (!strcmp(argv[0], "off")) {
3885                 lmasterrole = 0;
3886         } else {
3887                 usage();
3888         }
3889
3890         ret = ctdb_ctrl_setlmasterrole(ctdb, TIMELIMIT(), options.pnn, lmasterrole);
3891         if (ret != 0) {
3892                 DEBUG(DEBUG_ERR, ("Unable to set the lmaster role for node %u\n", options.pnn));
3893                 return ret;
3894         }
3895
3896         return 0;
3897 }
3898
3899 /*
3900   set the recmaster role on/off
3901  */
3902 static int control_setrecmasterrole(struct ctdb_context *ctdb, int argc, const char **argv)
3903 {
3904         int ret;
3905         uint32_t recmasterrole;
3906
3907         if (argc == 0) {
3908                 usage();
3909         }
3910
3911         if (!strcmp(argv[0], "on")) {
3912                 recmasterrole = 1;
3913         } else if (!strcmp(argv[0], "off")) {
3914                 recmasterrole = 0;
3915         } else {
3916                 usage();
3917         }
3918
3919         ret = ctdb_ctrl_setrecmasterrole(ctdb, TIMELIMIT(), options.pnn, recmasterrole);
3920         if (ret != 0) {
3921                 DEBUG(DEBUG_ERR, ("Unable to set the recmaster role for node %u\n", options.pnn));
3922                 return ret;
3923         }
3924
3925         return 0;
3926 }
3927
3928 /*
3929   set debug level on a node or all nodes
3930  */
3931 static int control_setdebug(struct ctdb_context *ctdb, int argc, const char **argv)
3932 {
3933         int i, ret;
3934         int32_t level;
3935
3936         if (argc == 0) {
3937                 printf("You must specify the debug level. Valid levels are:\n");
3938                 for (i=0; debug_levels[i].description != NULL; i++) {
3939                         printf("%s (%d)\n", debug_levels[i].description, debug_levels[i].level);
3940                 }
3941
3942                 return 0;
3943         }
3944
3945         if (isalpha(argv[0][0]) || argv[0][0] == '-') { 
3946                 level = get_debug_by_desc(argv[0]);
3947         } else {
3948                 level = strtol(argv[0], NULL, 0);
3949         }
3950
3951         for (i=0; debug_levels[i].description != NULL; i++) {
3952                 if (level == debug_levels[i].level) {
3953                         break;
3954                 }
3955         }
3956         if (debug_levels[i].description == NULL) {
3957                 printf("Invalid debug level, must be one of\n");
3958                 for (i=0; debug_levels[i].description != NULL; i++) {
3959                         printf("%s (%d)\n", debug_levels[i].description, debug_levels[i].level);
3960                 }
3961                 return -1;
3962         }
3963
3964         ret = ctdb_ctrl_set_debuglevel(ctdb, options.pnn, level);
3965         if (ret != 0) {
3966                 DEBUG(DEBUG_ERR, ("Unable to set debug level on node %u\n", options.pnn));
3967         }
3968         return 0;
3969 }
3970
3971
3972 /*
3973   thaw a node
3974  */
3975 static int control_thaw(struct ctdb_context *ctdb, int argc, const char **argv)
3976 {
3977         int ret;
3978         uint32_t priority;
3979         
3980         if (argc == 1) {
3981                 priority = strtol(argv[0], NULL, 0);
3982         } else {
3983                 priority = 0;
3984         }
3985         DEBUG(DEBUG_ERR,("Thaw by priority %u\n", priority));
3986
3987         ret = ctdb_ctrl_thaw_priority(ctdb, TIMELIMIT(), options.pnn, priority);
3988         if (ret != 0) {
3989                 DEBUG(DEBUG_ERR, ("Unable to thaw node %u\n", options.pnn));
3990         }               
3991         return 0;
3992 }
3993
3994
3995 /*
3996   attach to a database
3997  */
3998 static int control_attach(struct ctdb_context *ctdb, int argc, const char **argv)
3999 {
4000         const char *db_name;
4001         struct ctdb_db_context *ctdb_db;
4002         bool persistent = false;
4003
4004         if (argc < 1) {
4005                 usage();
4006         }
4007         db_name = argv[0];
4008         if (argc > 2) {
4009                 usage();
4010         }
4011         if (argc == 2) {
4012                 if (strcmp(argv[1], "persistent") != 0) {
4013                         usage();
4014                 }
4015                 persistent = true;
4016         }
4017
4018         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
4019         if (ctdb_db == NULL) {
4020                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
4021                 return -1;
4022         }
4023
4024         return 0;
4025 }
4026
4027 /*
4028   set db priority
4029  */
4030 static int control_setdbprio(struct ctdb_context *ctdb, int argc, const char **argv)
4031 {
4032         struct ctdb_db_priority db_prio;
4033         int ret;
4034
4035         if (argc < 2) {
4036                 usage();
4037         }
4038
4039         db_prio.db_id    = strtoul(argv[0], NULL, 0);
4040         db_prio.priority = strtoul(argv[1], NULL, 0);
4041
4042         ret = ctdb_ctrl_set_db_priority(ctdb, TIMELIMIT(), options.pnn, &db_prio);
4043         if (ret != 0) {
4044                 DEBUG(DEBUG_ERR,("Unable to set db prio\n"));
4045                 return -1;
4046         }
4047
4048         return 0;
4049 }
4050
4051 /*
4052   get db priority
4053  */
4054 static int control_getdbprio(struct ctdb_context *ctdb, int argc, const char **argv)
4055 {
4056         uint32_t db_id, priority;
4057         int ret;
4058
4059         if (argc < 1) {
4060                 usage();
4061         }
4062
4063         db_id = strtoul(argv[0], NULL, 0);
4064
4065         ret = ctdb_ctrl_get_db_priority(ctdb, TIMELIMIT(), options.pnn, db_id, &priority);
4066         if (ret != 0) {
4067                 DEBUG(DEBUG_ERR,("Unable to get db prio\n"));
4068                 return -1;
4069         }
4070
4071         DEBUG(DEBUG_ERR,("Priority:%u\n", priority));
4072
4073         return 0;
4074 }
4075
4076 /*
4077   set the readonly capability for a database
4078  */
4079 static int control_setdbreadonly(struct ctdb_context *ctdb, int argc, const char **argv)
4080 {
4081         uint32_t db_id;
4082         int ret;
4083
4084         if (argc < 1) {
4085                 usage();
4086         }
4087
4088         db_id = strtoul(argv[0], NULL, 0);
4089
4090         ret = ctdb_ctrl_set_db_readonly(ctdb, options.pnn, db_id);
4091         if (ret != 0) {
4092                 DEBUG(DEBUG_ERR,("Unable to set db to support readonly\n"));
4093                 return -1;
4094         }
4095
4096         return 0;
4097 }
4098
4099 /*
4100   run an eventscript on a node
4101  */
4102 static int control_eventscript(struct ctdb_context *ctdb, int argc, const char **argv)
4103 {
4104         TDB_DATA data;
4105         int ret;
4106         int32_t res;
4107         char *errmsg;
4108         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4109
4110         if (argc != 1) {
4111                 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
4112                 return -1;
4113         }
4114
4115         data.dptr = (unsigned char *)discard_const(argv[0]);
4116         data.dsize = strlen((char *)data.dptr) + 1;
4117
4118         DEBUG(DEBUG_ERR, ("Running eventscripts with arguments \"%s\" on node %u\n", data.dptr, options.pnn));
4119
4120         ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_RUN_EVENTSCRIPTS,
4121                            0, data, tmp_ctx, NULL, &res, NULL, &errmsg);
4122         if (ret != 0 || res != 0) {
4123                 DEBUG(DEBUG_ERR,("Failed to run eventscripts - %s\n", errmsg));
4124                 talloc_free(tmp_ctx);
4125                 return -1;
4126         }
4127         talloc_free(tmp_ctx);
4128         return 0;
4129 }
4130
4131 #define DB_VERSION 1
4132 #define MAX_DB_NAME 64
4133 struct db_file_header {
4134         unsigned long version;
4135         time_t timestamp;
4136         unsigned long persistent;
4137         unsigned long size;
4138         const char name[MAX_DB_NAME];
4139 };
4140
4141 struct backup_data {
4142         struct ctdb_marshall_buffer *records;
4143         uint32_t len;
4144         uint32_t total;
4145         bool traverse_error;
4146 };
4147
4148 static int backup_traverse(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, void *private)
4149 {
4150         struct backup_data *bd = talloc_get_type(private, struct backup_data);
4151         struct ctdb_rec_data *rec;
4152
4153         /* add the record */
4154         rec = ctdb_marshall_record(bd->records, 0, key, NULL, data);
4155         if (rec == NULL) {
4156                 bd->traverse_error = true;
4157                 DEBUG(DEBUG_ERR,("Failed to marshall record\n"));
4158                 return -1;
4159         }
4160         bd->records = talloc_realloc_size(NULL, bd->records, rec->length + bd->len);
4161         if (bd->records == NULL) {
4162                 DEBUG(DEBUG_ERR,("Failed to expand marshalling buffer\n"));
4163                 bd->traverse_error = true;
4164                 return -1;
4165         }
4166         bd->records->count++;
4167         memcpy(bd->len+(uint8_t *)bd->records, rec, rec->length);
4168         bd->len += rec->length;
4169         talloc_free(rec);
4170
4171         bd->total++;
4172         return 0;
4173 }
4174
4175 /*
4176  * backup a database to a file 
4177  */
4178 static int control_backupdb(struct ctdb_context *ctdb, int argc, const char **argv)
4179 {
4180         int i, ret;
4181         struct ctdb_dbid_map *dbmap=NULL;
4182         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4183         struct db_file_header dbhdr;
4184         struct ctdb_db_context *ctdb_db;
4185         struct backup_data *bd;
4186         int fh = -1;
4187         int status = -1;
4188         const char *reason = NULL;
4189
4190         if (argc != 2) {
4191                 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
4192                 return -1;
4193         }
4194
4195         ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &dbmap);
4196         if (ret != 0) {
4197                 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n", options.pnn));
4198                 return ret;
4199         }
4200
4201         for(i=0;i<dbmap->num;i++){
4202                 const char *name;
4203
4204                 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, tmp_ctx, &name);
4205                 if(!strcmp(argv[0], name)){
4206                         talloc_free(discard_const(name));
4207                         break;
4208                 }
4209                 talloc_free(discard_const(name));
4210         }
4211         if (i == dbmap->num) {
4212                 DEBUG(DEBUG_ERR,("No database with name '%s' found\n", argv[0]));
4213                 talloc_free(tmp_ctx);
4214                 return -1;
4215         }
4216
4217         ret = ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn,
4218                                     dbmap->dbs[i].dbid, tmp_ctx, &reason);
4219         if (ret != 0) {
4220                 DEBUG(DEBUG_ERR,("Unable to get dbhealth for database '%s'\n",
4221                                  argv[0]));
4222                 talloc_free(tmp_ctx);
4223                 return -1;
4224         }
4225         if (reason) {
4226                 uint32_t allow_unhealthy = 0;
4227
4228                 ctdb_ctrl_get_tunable(ctdb, TIMELIMIT(), options.pnn,
4229                                       "AllowUnhealthyDBRead",
4230                                       &allow_unhealthy);
4231
4232                 if (allow_unhealthy != 1) {
4233                         DEBUG(DEBUG_ERR,("database '%s' is unhealthy: %s\n",
4234                                          argv[0], reason));
4235
4236                         DEBUG(DEBUG_ERR,("disallow backup : tunable AllowUnhealthyDBRead = %u\n",
4237                                          allow_unhealthy));
4238                         talloc_free(tmp_ctx);
4239                         return -1;
4240                 }
4241
4242                 DEBUG(DEBUG_WARNING,("WARNING database '%s' is unhealthy - see 'ctdb getdbstatus %s'\n",
4243                                      argv[0], argv[0]));
4244                 DEBUG(DEBUG_WARNING,("WARNING! allow backup of unhealthy database: "
4245                                      "tunnable AllowUnhealthyDBRead = %u\n",
4246                                      allow_unhealthy));
4247         }
4248
4249         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), argv[0], dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT, 0);
4250         if (ctdb_db == NULL) {
4251                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", argv[0]));
4252                 talloc_free(tmp_ctx);
4253                 return -1;
4254         }
4255
4256
4257         ret = tdb_transaction_start(ctdb_db->ltdb->tdb);
4258         if (ret == -1) {
4259                 DEBUG(DEBUG_ERR,("Failed to start transaction\n"));
4260                 talloc_free(tmp_ctx);
4261                 return -1;
4262         }
4263
4264
4265         bd = talloc_zero(tmp_ctx, struct backup_data);
4266         if (bd == NULL) {
4267                 DEBUG(DEBUG_ERR,("Failed to allocate backup_data\n"));
4268                 talloc_free(tmp_ctx);
4269                 return -1;
4270         }
4271
4272         bd->records = talloc_zero(bd, struct ctdb_marshall_buffer);
4273         if (bd->records == NULL) {
4274                 DEBUG(DEBUG_ERR,("Failed to allocate ctdb_marshall_buffer\n"));
4275                 talloc_free(tmp_ctx);
4276                 return -1;
4277         }
4278
4279         bd->len = offsetof(struct ctdb_marshall_buffer, data);
4280         bd->records->db_id = ctdb_db->db_id;
4281         /* traverse the database collecting all records */
4282         if (tdb_traverse_read(ctdb_db->ltdb->tdb, backup_traverse, bd) == -1 ||
4283             bd->traverse_error) {
4284                 DEBUG(DEBUG_ERR,("Traverse error\n"));
4285                 talloc_free(tmp_ctx);
4286                 return -1;              
4287         }
4288
4289         tdb_transaction_cancel(ctdb_db->ltdb->tdb);
4290
4291
4292         fh = open(argv[1], O_RDWR|O_CREAT, 0600);
4293         if (fh == -1) {
4294                 DEBUG(DEBUG_ERR,("Failed to open file '%s'\n", argv[1]));
4295                 talloc_free(tmp_ctx);
4296                 return -1;
4297         }
4298
4299         dbhdr.version = DB_VERSION;
4300         dbhdr.timestamp = time(NULL);
4301         dbhdr.persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
4302         dbhdr.size = bd->len;
4303         if (strlen(argv[0]) >= MAX_DB_NAME) {
4304                 DEBUG(DEBUG_ERR,("Too long dbname\n"));
4305                 goto done;
4306         }
4307         strncpy(discard_const(dbhdr.name), argv[0], MAX_DB_NAME);
4308         ret = write(fh, &dbhdr, sizeof(dbhdr));
4309         if (ret == -1) {
4310                 DEBUG(DEBUG_ERR,("write failed: %s\n", strerror(errno)));
4311                 goto done;
4312         }
4313         ret = write(fh, bd->records, bd->len);
4314         if (ret == -1) {
4315                 DEBUG(DEBUG_ERR,("write failed: %s\n", strerror(errno)));
4316                 goto done;
4317         }
4318
4319         status = 0;
4320 done:
4321         if (fh != -1) {
4322                 ret = close(fh);
4323                 if (ret == -1) {
4324                         DEBUG(DEBUG_ERR,("close failed: %s\n", strerror(errno)));
4325                 }
4326         }
4327
4328         DEBUG(DEBUG_ERR,("Database backed up to %s\n", argv[1]));
4329
4330         talloc_free(tmp_ctx);
4331         return status;
4332 }
4333
4334 /*
4335  * restore a database from a file 
4336  */
4337 static int control_restoredb(struct ctdb_context *ctdb, int argc, const char **argv)
4338 {
4339         int ret;
4340         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4341         TDB_DATA outdata;
4342         TDB_DATA data;
4343         struct db_file_header dbhdr;
4344         struct ctdb_db_context *ctdb_db;
4345         struct ctdb_node_map *nodemap=NULL;
4346         struct ctdb_vnn_map *vnnmap=NULL;
4347         int i, fh;
4348         struct ctdb_control_wipe_database w;
4349         uint32_t *nodes;
4350         uint32_t generation;
4351         struct tm *tm;
4352         char tbuf[100];
4353         char *dbname;
4354
4355         if (argc < 1 || argc > 2) {
4356                 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
4357                 return -1;
4358         }
4359
4360         fh = open(argv[0], O_RDONLY);
4361         if (fh == -1) {
4362                 DEBUG(DEBUG_ERR,("Failed to open file '%s'\n", argv[0]));
4363                 talloc_free(tmp_ctx);
4364                 return -1;
4365         }
4366
4367         read(fh, &dbhdr, sizeof(dbhdr));
4368         if (dbhdr.version != DB_VERSION) {
4369                 DEBUG(DEBUG_ERR,("Invalid version of database dump. File is version %lu but expected version was %u\n", dbhdr.version, DB_VERSION));
4370                 talloc_free(tmp_ctx);
4371                 return -1;
4372         }
4373
4374         dbname = discard_const(dbhdr.name);
4375         if (argc == 2) {
4376                 dbname = discard_const(argv[1]);
4377         }
4378
4379         outdata.dsize = dbhdr.size;
4380         outdata.dptr = talloc_size(tmp_ctx, outdata.dsize);
4381         if (outdata.dptr == NULL) {
4382                 DEBUG(DEBUG_ERR,("Failed to allocate data of size '%lu'\n", dbhdr.size));
4383                 close(fh);
4384                 talloc_free(tmp_ctx);
4385                 return -1;
4386         }               
4387         read(fh, outdata.dptr, outdata.dsize);
4388         close(fh);
4389
4390         tm = localtime(&dbhdr.timestamp);
4391         strftime(tbuf,sizeof(tbuf)-1,"%Y/%m/%d %H:%M:%S", tm);
4392         printf("Restoring database '%s' from backup @ %s\n",
4393                 dbname, tbuf);
4394
4395
4396         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), dbname, dbhdr.persistent, 0);
4397         if (ctdb_db == NULL) {
4398                 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", dbname));
4399                 talloc_free(tmp_ctx);
4400                 return -1;
4401         }
4402
4403         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb, &nodemap);
4404         if (ret != 0) {
4405                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
4406                 talloc_free(tmp_ctx);
4407                 return ret;
4408         }
4409
4410
4411         ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &vnnmap);
4412         if (ret != 0) {
4413                 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n", options.pnn));
4414                 talloc_free(tmp_ctx);
4415                 return ret;
4416         }
4417
4418         /* freeze all nodes */
4419         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4420         for (i=1; i<=NUM_DB_PRIORITIES; i++) {
4421                 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_FREEZE,
4422                                         nodes, i,
4423                                         TIMELIMIT(),
4424                                         false, tdb_null,
4425                                         NULL, NULL,
4426                                         NULL) != 0) {
4427                         DEBUG(DEBUG_ERR, ("Unable to freeze nodes.\n"));
4428                         ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4429                         talloc_free(tmp_ctx);
4430                         return -1;
4431                 }
4432         }
4433
4434         generation = vnnmap->generation;
4435         data.dptr = (void *)&generation;
4436         data.dsize = sizeof(generation);
4437
4438         /* start a cluster wide transaction */
4439         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4440         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_START,
4441                                         nodes, 0,
4442                                         TIMELIMIT(), false, data,
4443                                         NULL, NULL,
4444                                         NULL) != 0) {
4445                 DEBUG(DEBUG_ERR, ("Unable to start cluster wide transactions.\n"));
4446                 return -1;
4447         }
4448
4449
4450         w.db_id = ctdb_db->db_id;
4451         w.transaction_id = generation;
4452
4453         data.dptr = (void *)&w;
4454         data.dsize = sizeof(w);
4455
4456         /* wipe all the remote databases. */
4457         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4458         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_WIPE_DATABASE,
4459                                         nodes, 0,
4460                                         TIMELIMIT(), false, data,
4461                                         NULL, NULL,
4462                                         NULL) != 0) {
4463                 DEBUG(DEBUG_ERR, ("Unable to wipe database.\n"));
4464                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4465                 talloc_free(tmp_ctx);
4466                 return -1;
4467         }
4468         
4469         /* push the database */
4470         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4471         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_PUSH_DB,
4472                                         nodes, 0,
4473                                         TIMELIMIT(), false, outdata,
4474                                         NULL, NULL,
4475                                         NULL) != 0) {
4476                 DEBUG(DEBUG_ERR, ("Failed to push database.\n"));
4477                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4478                 talloc_free(tmp_ctx);
4479                 return -1;
4480         }
4481
4482         data.dptr = (void *)&ctdb_db->db_id;
4483         data.dsize = sizeof(ctdb_db->db_id);
4484
4485         /* mark the database as healthy */
4486         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4487         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_DB_SET_HEALTHY,
4488                                         nodes, 0,
4489                                         TIMELIMIT(), false, data,
4490                                         NULL, NULL,
4491                                         NULL) != 0) {
4492                 DEBUG(DEBUG_ERR, ("Failed to mark database as healthy.\n"));
4493                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4494                 talloc_free(tmp_ctx);
4495                 return -1;
4496         }
4497
4498         data.dptr = (void *)&generation;
4499         data.dsize = sizeof(generation);
4500
4501         /* commit all the changes */
4502         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_COMMIT,
4503                                         nodes, 0,
4504                                         TIMELIMIT(), false, data,
4505                                         NULL, NULL,
4506                                         NULL) != 0) {
4507                 DEBUG(DEBUG_ERR, ("Unable to commit databases.\n"));
4508                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4509                 talloc_free(tmp_ctx);
4510                 return -1;
4511         }
4512
4513
4514         /* thaw all nodes */
4515         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4516         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_THAW,
4517                                         nodes, 0,
4518                                         TIMELIMIT(),
4519                                         false, tdb_null,
4520                                         NULL, NULL,
4521                                         NULL) != 0) {
4522                 DEBUG(DEBUG_ERR, ("Unable to thaw nodes.\n"));
4523                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4524                 talloc_free(tmp_ctx);
4525                 return -1;
4526         }
4527
4528
4529         talloc_free(tmp_ctx);
4530         return 0;
4531 }
4532
4533 /*
4534  * dump a database backup from a file
4535  */
4536 static int control_dumpdbbackup(struct ctdb_context *ctdb, int argc, const char **argv)
4537 {
4538         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4539         TDB_DATA outdata;
4540         struct db_file_header dbhdr;
4541         int i, fh;
4542         struct tm *tm;
4543         char tbuf[100];
4544         struct ctdb_rec_data *rec = NULL;
4545         struct ctdb_marshall_buffer *m;
4546
4547         if (argc != 1) {
4548                 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
4549                 return -1;
4550         }
4551
4552         fh = open(argv[0], O_RDONLY);
4553         if (fh == -1) {
4554                 DEBUG(DEBUG_ERR,("Failed to open file '%s'\n", argv[0]));
4555                 talloc_free(tmp_ctx);
4556                 return -1;
4557         }
4558
4559         read(fh, &dbhdr, sizeof(dbhdr));
4560         if (dbhdr.version != DB_VERSION) {
4561                 DEBUG(DEBUG_ERR,("Invalid version of database dump. File is version %lu but expected version was %u\n", dbhdr.version, DB_VERSION));
4562                 talloc_free(tmp_ctx);
4563                 return -1;
4564         }
4565
4566         outdata.dsize = dbhdr.size;
4567         outdata.dptr = talloc_size(tmp_ctx, outdata.dsize);
4568         if (outdata.dptr == NULL) {
4569                 DEBUG(DEBUG_ERR,("Failed to allocate data of size '%lu'\n", dbhdr.size));
4570                 close(fh);
4571                 talloc_free(tmp_ctx);
4572                 return -1;
4573         }
4574         read(fh, outdata.dptr, outdata.dsize);
4575         close(fh);
4576         m = (struct ctdb_marshall_buffer *)outdata.dptr;
4577
4578         tm = localtime(&dbhdr.timestamp);
4579         strftime(tbuf,sizeof(tbuf)-1,"%Y/%m/%d %H:%M:%S", tm);
4580         printf("Backup of database name:'%s' dbid:0x%x08x from @ %s\n",
4581                 dbhdr.name, m->db_id, tbuf);
4582
4583         for (i=0; i < m->count; i++) {
4584                 uint32_t reqid = 0;
4585                 TDB_DATA key, data;
4586
4587                 /* we do not want the header splitted, so we pass NULL*/
4588                 rec = ctdb_marshall_loop_next(m, rec, &reqid,
4589                                               NULL, &key, &data);
4590
4591                 ctdb_dumpdb_record(ctdb, key, data, stdout);
4592         }
4593
4594         printf("Dumped %d records\n", i);
4595         talloc_free(tmp_ctx);
4596         return 0;
4597 }
4598
4599 /*
4600  * wipe a database from a file
4601  */
4602 static int control_wipedb(struct ctdb_context *ctdb, int argc,
4603                           const char **argv)
4604 {
4605         int ret;
4606         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4607         TDB_DATA data;
4608         struct ctdb_db_context *ctdb_db;
4609         struct ctdb_node_map *nodemap = NULL;
4610         struct ctdb_vnn_map *vnnmap = NULL;
4611         int i;
4612         struct ctdb_control_wipe_database w;
4613         uint32_t *nodes;
4614         uint32_t generation;
4615         struct ctdb_dbid_map *dbmap = NULL;
4616
4617         if (argc != 1) {
4618                 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
4619                 return -1;
4620         }
4621
4622         ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx,
4623                                  &dbmap);
4624         if (ret != 0) {
4625                 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n",
4626                                   options.pnn));
4627                 return ret;
4628         }
4629
4630         for(i=0;i<dbmap->num;i++){
4631                 const char *name;
4632
4633                 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn,
4634                                     dbmap->dbs[i].dbid, tmp_ctx, &name);
4635                 if(!strcmp(argv[0], name)){
4636                         talloc_free(discard_const(name));
4637                         break;
4638                 }
4639                 talloc_free(discard_const(name));
4640         }
4641         if (i == dbmap->num) {
4642                 DEBUG(DEBUG_ERR, ("No database with name '%s' found\n",
4643                                   argv[0]));
4644                 talloc_free(tmp_ctx);
4645                 return -1;
4646         }
4647
4648         ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), argv[0], dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT, 0);
4649         if (ctdb_db == NULL) {
4650                 DEBUG(DEBUG_ERR, ("Unable to attach to database '%s'\n",
4651                                   argv[0]));
4652                 talloc_free(tmp_ctx);
4653                 return -1;
4654         }
4655
4656         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb,
4657                                    &nodemap);
4658         if (ret != 0) {
4659                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n",
4660                                   options.pnn));
4661                 talloc_free(tmp_ctx);
4662                 return ret;
4663         }
4664
4665         ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx,
4666                                   &vnnmap);
4667         if (ret != 0) {
4668                 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n",
4669                                   options.pnn));
4670                 talloc_free(tmp_ctx);
4671                 return ret;
4672         }
4673
4674         /* freeze all nodes */
4675         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4676         for (i=1; i<=NUM_DB_PRIORITIES; i++) {
4677                 ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_FREEZE,
4678                                                 nodes, i,
4679                                                 TIMELIMIT(),
4680                                                 false, tdb_null,
4681                                                 NULL, NULL,
4682                                                 NULL);
4683                 if (ret != 0) {
4684                         DEBUG(DEBUG_ERR, ("Unable to freeze nodes.\n"));
4685                         ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn,
4686                                              CTDB_RECOVERY_ACTIVE);
4687                         talloc_free(tmp_ctx);
4688                         return -1;
4689                 }
4690         }
4691
4692         generation = vnnmap->generation;
4693         data.dptr = (void *)&generation;
4694         data.dsize = sizeof(generation);
4695
4696         /* start a cluster wide transaction */
4697         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4698         ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_START,
4699                                         nodes, 0,
4700                                         TIMELIMIT(), false, data,
4701                                         NULL, NULL,
4702                                         NULL);
4703         if (ret!= 0) {
4704                 DEBUG(DEBUG_ERR, ("Unable to start cluster wide "
4705                                   "transactions.\n"));
4706                 return -1;
4707         }
4708
4709         w.db_id = ctdb_db->db_id;
4710         w.transaction_id = generation;
4711
4712         data.dptr = (void *)&w;
4713         data.dsize = sizeof(w);
4714
4715         /* wipe all the remote databases. */
4716         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4717         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_WIPE_DATABASE,
4718                                         nodes, 0,
4719                                         TIMELIMIT(), false, data,
4720                                         NULL, NULL,
4721                                         NULL) != 0) {
4722                 DEBUG(DEBUG_ERR, ("Unable to wipe database.\n"));
4723                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4724                 talloc_free(tmp_ctx);
4725                 return -1;
4726         }
4727
4728         data.dptr = (void *)&ctdb_db->db_id;
4729         data.dsize = sizeof(ctdb_db->db_id);
4730
4731         /* mark the database as healthy */
4732         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4733         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_DB_SET_HEALTHY,
4734                                         nodes, 0,
4735                                         TIMELIMIT(), false, data,
4736                                         NULL, NULL,
4737                                         NULL) != 0) {
4738                 DEBUG(DEBUG_ERR, ("Failed to mark database as healthy.\n"));
4739                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4740                 talloc_free(tmp_ctx);
4741                 return -1;
4742         }
4743
4744         data.dptr = (void *)&generation;
4745         data.dsize = sizeof(generation);
4746
4747         /* commit all the changes */
4748         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_COMMIT,
4749                                         nodes, 0,
4750                                         TIMELIMIT(), false, data,
4751                                         NULL, NULL,
4752                                         NULL) != 0) {
4753                 DEBUG(DEBUG_ERR, ("Unable to commit databases.\n"));
4754                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4755                 talloc_free(tmp_ctx);
4756                 return -1;
4757         }
4758
4759         /* thaw all nodes */
4760         nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
4761         if (ctdb_client_async_control(ctdb, CTDB_CONTROL_THAW,
4762                                         nodes, 0,
4763                                         TIMELIMIT(),
4764                                         false, tdb_null,
4765                                         NULL, NULL,
4766                                         NULL) != 0) {
4767                 DEBUG(DEBUG_ERR, ("Unable to thaw nodes.\n"));
4768                 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
4769                 talloc_free(tmp_ctx);
4770                 return -1;
4771         }
4772
4773         DEBUG(DEBUG_ERR, ("Database wiped.\n"));
4774
4775         talloc_free(tmp_ctx);
4776         return 0;
4777 }
4778
4779 /*
4780   dump memory usage
4781  */
4782 static int control_dumpmemory(struct ctdb_context *ctdb, int argc, const char **argv)
4783 {
4784         TDB_DATA data;
4785         int ret;
4786         int32_t res;
4787         char *errmsg;
4788         TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4789         ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_DUMP_MEMORY,
4790                            0, tdb_null, tmp_ctx, &data, &res, NULL, &errmsg);
4791         if (ret != 0 || res != 0) {
4792                 DEBUG(DEBUG_ERR,("Failed to dump memory - %s\n", errmsg));
4793                 talloc_free(tmp_ctx);
4794                 return -1;
4795         }
4796         write(1, data.dptr, data.dsize);
4797         talloc_free(tmp_ctx);
4798         return 0;
4799 }
4800
4801 /*
4802   handler for memory dumps
4803 */
4804 static void mem_dump_handler(struct ctdb_context *ctdb, uint64_t srvid, 
4805                              TDB_DATA data, void *private_data)
4806 {
4807         write(1, data.dptr, data.dsize);
4808         exit(0);
4809 }
4810
4811 /*
4812   dump memory usage on the recovery daemon
4813  */
4814 static int control_rddumpmemory(struct ctdb_context *ctdb, int argc, const char **argv)
4815 {
4816         int ret;
4817         TDB_DATA data;
4818         struct rd_memdump_reply rd;
4819
4820         rd.pnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE);
4821         if (rd.pnn == -1) {
4822                 DEBUG(DEBUG_ERR, ("Failed to get pnn of local node\n"));
4823                 return -1;
4824         }
4825         rd.srvid = getpid();
4826
4827         /* register a message port for receiveing the reply so that we
4828            can receive the reply
4829         */
4830         ctdb_client_set_message_handler(ctdb, rd.srvid, mem_dump_handler, NULL);
4831
4832
4833         data.dptr = (uint8_t *)&rd;
4834         data.dsize = sizeof(rd);
4835
4836         ret = ctdb_client_send_message(ctdb, options.pnn, CTDB_SRVID_MEM_DUMP, data);
4837         if (ret != 0) {
4838                 DEBUG(DEBUG_ERR,("Failed to send memdump request message to %u\n", options.pnn));
4839                 return -1;
4840         }
4841
4842         /* this loop will terminate when we have received the reply */
4843         while (1) {     
4844                 event_loop_once(ctdb->ev);
4845         }
4846
4847         return 0;
4848 }
4849
4850 /*
4851   send a message to a srvid
4852  */
4853 static int control_msgsend(struct ctdb_context *ctdb, int argc, const char **argv)
4854 {
4855         unsigned long srvid;
4856         int ret;
4857         TDB_DATA data;
4858
4859         if (argc < 2) {
4860                 usage();
4861         }
4862
4863         srvid      = strtoul(argv[0], NULL, 0);
4864
4865         data.dptr = (uint8_t *)discard_const(argv[1]);
4866         data.dsize= strlen(argv[1]);
4867
4868         ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, srvid, data);
4869         if (ret != 0) {
4870                 DEBUG(DEBUG_ERR,("Failed to send memdump request message to %u\n", options.pnn));
4871                 return -1;
4872         }
4873
4874         return 0;
4875 }
4876
4877 /*
4878   handler for msglisten
4879 */
4880 static void msglisten_handler(struct ctdb_context *ctdb, uint64_t srvid, 
4881                              TDB_DATA data, void *private_data)
4882 {
4883         int i;
4884
4885         printf("Message received: ");
4886         for (i=0;i<data.dsize;i++) {
4887                 printf("%c", data.dptr[i]);
4888         }
4889         printf("\n");
4890 }
4891
4892 /*
4893   listen for messages on a messageport
4894  */
4895 static int control_msglisten(struct ctdb_context *ctdb, int argc, const char **argv)
4896 {
4897         uint64_t srvid;
4898
4899         srvid = getpid();
4900
4901         /* register a message port and listen for messages
4902         */
4903         ctdb_client_set_message_handler(ctdb, srvid, msglisten_handler, NULL);
4904         printf("Listening for messages on srvid:%d\n", (int)srvid);
4905
4906         while (1) {     
4907                 event_loop_once(ctdb->ev);
4908         }
4909
4910         return 0;
4911 }
4912
4913 /*
4914   list all nodes in the cluster
4915   we parse the nodes file directly
4916  */
4917 static int control_listnodes(struct ctdb_context *ctdb, int argc, const char **argv)
4918 {
4919         TALLOC_CTX *mem_ctx = talloc_new(NULL);
4920         struct pnn_node *pnn_nodes;
4921         struct pnn_node *pnn_node;
4922
4923         pnn_nodes = read_nodes_file(mem_ctx);
4924         if (pnn_nodes == NULL) {
4925                 DEBUG(DEBUG_ERR,("Failed to read nodes file\n"));
4926                 talloc_free(mem_ctx);
4927                 return -1;
4928         }
4929
4930         for(pnn_node=pnn_nodes;pnn_node;pnn_node=pnn_node->next) {
4931                 ctdb_sock_addr addr;
4932                 if (parse_ip(pnn_node->addr, NULL, 63999, &addr) == 0) {
4933                         DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s' in nodes file\n", pnn_node->addr));
4934                         talloc_free(mem_ctx);
4935                         return -1;
4936                 }
4937                 if (options.machinereadable){
4938                         printf(":%d:%s:\n", pnn_node->pnn, pnn_node->addr);
4939                 } else {
4940                         printf("%s\n", pnn_node->addr);
4941                 }
4942         }
4943         talloc_free(mem_ctx);
4944
4945         return 0;
4946 }
4947
4948 /*
4949   reload the nodes file on the local node
4950  */
4951 static int control_reload_nodes_file(struct ctdb_context *ctdb, int argc, const char **argv)
4952 {
4953         int i, ret;
4954         int mypnn;
4955         struct ctdb_node_map *nodemap=NULL;
4956
4957         mypnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE);
4958         if (mypnn == -1) {
4959                 DEBUG(DEBUG_ERR, ("Failed to read pnn of local node\n"));
4960                 return -1;
4961         }
4962
4963         ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
4964         if (ret != 0) {
4965                 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
4966                 return ret;
4967         }
4968
4969         /* reload the nodes file on all remote nodes */
4970         for (i=0;i<nodemap->num;i++) {
4971                 if (nodemap->nodes[i].pnn == mypnn) {
4972                         continue;
4973                 }
4974                 DEBUG(DEBUG_NOTICE, ("Reloading nodes file on node %u\n", nodemap->nodes[i].pnn));
4975                 ret = ctdb_ctrl_reload_nodes_file(ctdb, TIMELIMIT(),
4976                         nodemap->nodes[i].pnn);
4977                 if (ret != 0) {
4978                         DEBUG(DEBUG_ERR, ("ERROR: Failed to reload nodes file on node %u. You MUST fix that node manually!\n", nodemap->nodes[i].pnn));
4979                 }
4980         }
4981
4982         /* reload the nodes file on the local node */
4983         DEBUG(DEBUG_NOTICE, ("Reloading nodes file on node %u\n", mypnn));
4984         ret = ctdb_ctrl_reload_nodes_file(ctdb, TIMELIMIT(), mypnn);
4985         if (ret != 0) {
4986                 DEBUG(DEBUG_ERR, ("ERROR: Failed to reload nodes file on node %u. You MUST fix that node manually!\n", mypnn));
4987         }
4988
4989         /* initiate a recovery */
4990         control_recover(ctdb, argc, argv);
4991
4992         return 0;
4993 }
4994
4995
4996 static const struct {
4997         const char *name;
4998         int (*fn)(struct ctdb_context *, int, const char **);
4999         bool auto_all;
5000         bool without_daemon; /* can be run without daemon running ? */
5001         const char *msg;
5002         const char *args;
5003 } ctdb_commands[] = {
5004 #ifdef CTDB_VERS
5005         { "version",         control_version,           true,   false,  "show version of ctdb" },
5006 #endif
5007         { "status",          control_status,            true,   false,  "show node status" },
5008         { "uptime",          control_uptime,            true,   false,  "show node uptime" },
5009         { "ping",            control_ping,              true,   false,  "ping all nodes" },
5010         { "getvar",          control_getvar,            true,   false,  "get a tunable variable",               "<name>"},
5011         { "setvar",          control_setvar,            true,   false,  "set a tunable variable",               "<name> <value>"},
5012         { "listvars",        control_listvars,          true,   false,  "list tunable variables"},
5013         { "statistics",      control_statistics,        false,  false, "show statistics" },
5014         { "statisticsreset", control_statistics_reset,  true,   false,  "reset statistics"},
5015         { "stats",           control_stats,             false,  false,  "show rolling statistics", "[number of history records]" },
5016         { "ip",              control_ip,                false,  false,  "show which public ip's that ctdb manages" },
5017         { "ipinfo",          control_ipinfo,            true,   false,  "show details about a public ip that ctdb manages", "<ip>" },
5018         { "ifaces",          control_ifaces,            true,   false,  "show which interfaces that ctdb manages" },
5019         { "setifacelink",    control_setifacelink,      true,   false,  "set interface link status", "<iface> <status>" },
5020         { "process-exists",  control_process_exists,    true,   false,  "check if a process exists on a node",  "<pid>"},
5021         { "getdbmap",        control_getdbmap,          true,   false,  "show the database map" },
5022         { "getdbstatus",     control_getdbstatus,       true,   false,  "show the status of a database", "<dbname>" },
5023         { "catdb",           control_catdb,             true,   false,  "dump a ctdb database" ,                     "<dbname>"},
5024         { "cattdb",          control_cattdb,            true,   false,  "dump a local tdb database" ,                     "<dbname>"},
5025         { "getmonmode",      control_getmonmode,        true,   false,  "show monitoring mode" },
5026         { "getcapabilities", control_getcapabilities,   true,   false,  "show node capabilities" },
5027         { "pnn",             control_pnn,               true,   false,  "show the pnn of the currnet node" },
5028         { "lvs",             control_lvs,               true,   false,  "show lvs configuration" },
5029         { "lvsmaster",       control_lvsmaster,         true,   false,  "show which node is the lvs master" },
5030         { "disablemonitor",      control_disable_monmode,true,  false,  "set monitoring mode to DISABLE" },
5031         { "enablemonitor",      control_enable_monmode, true,   false,  "set monitoring mode to ACTIVE" },
5032         { "setdebug",        control_setdebug,          true,   false,  "set debug level",                      "<EMERG|ALERT|CRIT|ERR|WARNING|NOTICE|INFO|DEBUG>" },
5033         { "getdebug",        control_getdebug,          true,   false,  "get debug level" },
5034         { "getlog",          control_getlog,            true,   false,  "get the log data from the in memory ringbuffer", "<level>" },
5035         { "clearlog",          control_clearlog,        true,   false,  "clear the log data from the in memory ringbuffer" },
5036         { "attach",          control_attach,            true,   false,  "attach to a database",                 "<dbname> [persistent]" },
5037         { "dumpmemory",      control_dumpmemory,        true,   false,  "dump memory map to stdout" },
5038         { "rddumpmemory",    control_rddumpmemory,      true,   false,  "dump memory map from the recovery daemon to stdout" },
5039         { "getpid",          control_getpid,            true,   false,  "get ctdbd process ID" },
5040         { "disable",         control_disable,           true,   false,  "disable a nodes public IP" },
5041         { "enable",          control_enable,            true,   false,  "enable a nodes public IP" },
5042         { "stop",            control_stop,              true,   false,  "stop a node" },
5043         { "continue",        control_continue,          true,   false,  "re-start a stopped node" },
5044         { "ban",             control_ban,               true,   false,  "ban a node from the cluster",          "<bantime|0>"},
5045         { "unban",           control_unban,             true,   false,  "unban a node" },
5046         { "showban",         control_showban,           true,   false,  "show ban information"},
5047         { "shutdown",        control_shutdown,          true,   false,  "shutdown ctdbd" },
5048         { "recover",         control_recover,           true,   false,  "force recovery" },
5049         { "sync",            control_ipreallocate,      true,   false,  "wait until ctdbd has synced all state changes" },
5050         { "ipreallocate",    control_ipreallocate,      true,   false,  "force the recovery daemon to perform a ip reallocation procedure" },
5051         { "thaw",            control_thaw,              true,   false,  "thaw databases", "[priority:1-3]" },
5052         { "isnotrecmaster",  control_isnotrecmaster,    false,  false,  "check if the local node is recmaster or not" },
5053         { "killtcp",         kill_tcp,                  false,  false, "kill a tcp connection.", "<srcip:port> <dstip:port>" },
5054         { "gratiousarp",     control_gratious_arp,      false,  false, "send a gratious arp", "<ip> <interface>" },
5055         { "tickle",          tickle_tcp,                false,  false, "send a tcp tickle ack", "<srcip:port> <dstip:port>" },
5056         { "gettickles",      control_get_tickles,       false,  false, "get the list of tickles registered for this ip", "<ip> [<port>]" },
5057         { "addtickle",       control_add_tickle,        false,  false, "add a tickle for this ip", "<ip>:<port> <ip>:<port>" },
5058
5059         { "deltickle",       control_del_tickle,        false,  false, "delete a tickle from this ip", "<ip>:<port> <ip>:<port>" },
5060
5061         { "regsrvid",        regsrvid,                  false,  false, "register a server id", "<pnn> <type> <id>" },
5062         { "unregsrvid",      unregsrvid,                false,  false, "unregister a server id", "<pnn> <type> <id>" },
5063         { "chksrvid",        chksrvid,                  false,  false, "check if a server id exists", "<pnn> <type> <id>" },
5064         { "getsrvids",       getsrvids,                 false,  false, "get a list of all server ids"},
5065         { "vacuum",          ctdb_vacuum,               false,  false, "vacuum the databases of empty records", "[max_records]"},
5066         { "repack",          ctdb_repack,               false,  false, "repack all databases", "[max_freelist]"},
5067         { "listnodes",       control_listnodes,         false,  true, "list all nodes in the cluster"},
5068         { "reloadnodes",     control_reload_nodes_file, false,  false, "reload the nodes file and restart the transport on all nodes"},
5069         { "moveip",          control_moveip,            false,  false, "move/failover an ip address to another node", "<ip> <node>"},
5070         { "addip",           control_addip,             true,   false, "add a ip address to a node", "<ip/mask> <iface>"},
5071         { "delip",           control_delip,             false,  false, "delete an ip address from a node", "<ip>"},
5072         { "eventscript",     control_eventscript,       true,   false, "run the eventscript with the given parameters on a node", "<arguments>"},
5073         { "backupdb",        control_backupdb,          false,  false, "backup the database into a file.", "<database> <file>"},
5074         { "restoredb",        control_restoredb,        false,  false, "restore the database from a file.", "<file> [dbname]"},
5075         { "dumpdbbackup",    control_dumpdbbackup,      false,  true,  "dump database backup from a file.", "<file>"},
5076         { "wipedb",           control_wipedb,        false,     false, "wipe the contents of a database.", "<dbname>"},
5077         { "recmaster",        control_recmaster,        false,  false, "show the pnn for the recovery master."},
5078         { "scriptstatus",    control_scriptstatus,  false,      false, "show the status of the monitoring scripts (or all scripts)", "[all]"},
5079         { "enablescript",     control_enablescript,  false,     false, "enable an eventscript", "<script>"},
5080         { "disablescript",    control_disablescript,  false,    false, "disable an eventscript", "<script>"},
5081         { "natgwlist",        control_natgwlist,        false,  false, "show the nodes belonging to this natgw configuration"},
5082         { "xpnn",             control_xpnn,             true,   true,  "find the pnn of the local node without talking to the daemon (unreliable)" },
5083         { "getreclock",       control_getreclock,       false,  false, "Show the reclock file of a node"},
5084         { "setreclock",       control_setreclock,       false,  false, "Set/clear the reclock file of a node", "[filename]"},
5085         { "setnatgwstate",    control_setnatgwstate,    false,  false, "Set NATGW state to on/off", "{on|off}"},
5086         { "setlmasterrole",   control_setlmasterrole,   false,  false, "Set LMASTER role to on/off", "{on|off}"},
5087         { "setrecmasterrole", control_setrecmasterrole, false,  false, "Set RECMASTER role to on/off", "{on|off}"},
5088         { "setdbprio",        control_setdbprio,        false,  false, "Set DB priority", "<dbid> <prio:1-3>"},
5089         { "getdbprio",        control_getdbprio,        false,  false, "Get DB priority", "<dbid>"},
5090         { "setdbreadonly",    control_setdbreadonly,    false,  false, "Set DB readonly capable", "<dbid>"},
5091         { "msglisten",        control_msglisten,        false,  false, "Listen on a srvid port for messages", "<msg srvid>"},
5092         { "msgsend",          control_msgsend,  false,  false, "Send a message to srvid", "<srvid> <message>"},
5093         { "sync",            control_ipreallocate,      false,  false,  "wait until ctdbd has synced all state changes" },
5094         { "pfetch",          control_pfetch,            false,  false,  "fetch a record from a persistent database", "<db> <key> [<file>]" },
5095         { "pstore",          control_pstore,            false,  false,  "write a record to a persistent database", "<db> <key> <file containing record>" },
5096         { "tfetch",          control_tfetch,            false,  true,  "fetch a record from a [c]tdb-file", "<tdb-file> <key> [<file>]" },
5097         { "readkey",         control_readkey,           true,   false,  "read the content off a database key", "<tdb-file> <key>" },
5098         { "writekey",        control_writekey,          true,   false,  "write to a database key", "<tdb-file> <key> <value>" },
5099         { "checktcpport",    control_chktcpport,        false,  true,  "check if a service is bound to a specific tcp port or not", "<port>" },
5100 };
5101
5102 /*
5103   show usage message
5104  */
5105 static void usage(void)
5106 {
5107         int i;
5108         printf(
5109 "Usage: ctdb [options] <control>\n" \
5110 "Options:\n" \
5111 "   -n <node>          choose node number, or 'all' (defaults to local node)\n"
5112 "   -Y                 generate machinereadable output\n"
5113 "   -v                 generate verbose output\n"
5114 "   -t <timelimit>     set timelimit for control in seconds (default %u)\n", options.timelimit);
5115         printf("Controls:\n");
5116         for (i=0;i<ARRAY_SIZE(ctdb_commands);i++) {
5117                 printf("  %-15s %-27s  %s\n", 
5118                        ctdb_commands[i].name, 
5119                        ctdb_commands[i].args?ctdb_commands[i].args:"",
5120                        ctdb_commands[i].msg);
5121         }
5122         exit(1);
5123 }
5124
5125
5126 static void ctdb_alarm(int sig)
5127 {
5128         printf("Maximum runtime exceeded - exiting\n");
5129         _exit(ERR_TIMEOUT);
5130 }
5131
5132 /*
5133   main program
5134 */
5135 int main(int argc, const char *argv[])
5136 {
5137         struct ctdb_context *ctdb;
5138         char *nodestring = NULL;
5139         struct poptOption popt_options[] = {
5140                 POPT_AUTOHELP
5141                 POPT_CTDB_CMDLINE
5142                 { "timelimit", 't', POPT_ARG_INT, &options.timelimit, 0, "timelimit", "integer" },
5143                 { "node",      'n', POPT_ARG_STRING, &nodestring, 0, "node", "integer|all" },
5144                 { "machinereadable", 'Y', POPT_ARG_NONE, &options.machinereadable, 0, "enable machinereadable output", NULL },
5145                 { "verbose",    'v', POPT_ARG_NONE, &options.verbose, 0, "enable verbose output", NULL },
5146                 { "maxruntime", 'T', POPT_ARG_INT, &options.maxruntime, 0, "die if runtime exceeds this limit (in seconds)", "integer" },
5147                 POPT_TABLEEND
5148         };
5149         int opt;
5150         const char **extra_argv;
5151         int extra_argc = 0;
5152         int ret=-1, i;
5153         poptContext pc;
5154         struct event_context *ev;
5155         const char *control;
5156
5157         setlinebuf(stdout);
5158         
5159         /* set some defaults */
5160         options.maxruntime = 0;
5161         options.timelimit = 3;
5162         options.pnn = CTDB_CURRENT_NODE;
5163
5164         pc = poptGetContext(argv[0], argc, argv, popt_options, POPT_CONTEXT_KEEP_FIRST);
5165
5166         while ((opt = poptGetNextOpt(pc)) != -1) {
5167                 switch (opt) {
5168                 default:
5169                         DEBUG(DEBUG_ERR, ("Invalid option %s: %s\n", 
5170                                 poptBadOption(pc, 0), poptStrerror(opt)));
5171                         exit(1);
5172                 }
5173         }
5174
5175         /* setup the remaining options for the main program to use */
5176         extra_argv = poptGetArgs(pc);
5177         if (extra_argv) {
5178                 extra_argv++;
5179                 while (extra_argv[extra_argc]) extra_argc++;
5180         }
5181
5182         if (extra_argc < 1) {
5183                 usage();
5184         }
5185
5186         if (options.maxruntime == 0) {
5187                 const char *ctdb_timeout;
5188                 ctdb_timeout = getenv("CTDB_TIMEOUT");
5189                 if (ctdb_timeout != NULL) {
5190                         options.maxruntime = strtoul(ctdb_timeout, NULL, 0);
5191                 } else {
5192                         /* default timeout is 120 seconds */
5193                         options.maxruntime = 120;
5194                 }
5195         }
5196
5197         signal(SIGALRM, ctdb_alarm);
5198         alarm(options.maxruntime);
5199
5200         /* setup the node number to contact */
5201         if (nodestring != NULL) {
5202                 if (strcmp(nodestring, "all") == 0) {
5203                         options.pnn = CTDB_BROADCAST_ALL;
5204                 } else {
5205                         options.pnn = strtoul(nodestring, NULL, 0);
5206                 }
5207         }
5208
5209         control = extra_argv[0];
5210
5211         ev = event_context_init(NULL);
5212         if (!ev) {
5213                 DEBUG(DEBUG_ERR, ("Failed to initialize event system\n"));
5214                 exit(1);
5215         }
5216         tevent_loop_allow_nesting(ev);
5217
5218         for (i=0;i<ARRAY_SIZE(ctdb_commands);i++) {
5219                 if (strcmp(control, ctdb_commands[i].name) == 0) {
5220                         int j;
5221
5222                         if (ctdb_commands[i].without_daemon == true) {
5223                                 close(2);
5224                         }
5225
5226                         /* initialise ctdb */
5227                         ctdb = ctdb_cmdline_client(ev, TIMELIMIT());
5228
5229                         if (ctdb_commands[i].without_daemon == false) {
5230                                 const char *socket_name;
5231
5232                                 if (ctdb == NULL) {
5233                                         DEBUG(DEBUG_ERR, ("Failed to init ctdb\n"));
5234                                         exit(1);
5235                                 }
5236
5237                                 /* initialize a libctdb connection as well */
5238                                 socket_name = ctdb_get_socketname(ctdb);
5239                                 ctdb_connection = ctdb_connect(socket_name,
5240                                                        ctdb_log_file, stderr);
5241                                 if (ctdb_connection == NULL) {
5242                                         fprintf(stderr, "Failed to connect to daemon from libctdb\n");
5243                                         exit(1);
5244                                 }                               
5245                         
5246                                 /* verify the node exists */
5247                                 verify_node(ctdb);
5248
5249                                 if (options.pnn == CTDB_CURRENT_NODE) {
5250                                         int pnn;
5251                                         pnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), options.pnn);         
5252                                         if (pnn == -1) {
5253                                                 return -1;
5254                                         }
5255                                         options.pnn = pnn;
5256                                 }
5257                         }
5258
5259                         if (ctdb_commands[i].auto_all && 
5260                             options.pnn == CTDB_BROADCAST_ALL) {
5261                                 uint32_t *nodes;
5262                                 uint32_t num_nodes;
5263                                 ret = 0;
5264
5265                                 nodes = ctdb_get_connected_nodes(ctdb, TIMELIMIT(), ctdb, &num_nodes);
5266                                 CTDB_NO_MEMORY(ctdb, nodes);
5267         
5268                                 for (j=0;j<num_nodes;j++) {
5269                                         options.pnn = nodes[j];
5270                                         ret |= ctdb_commands[i].fn(ctdb, extra_argc-1, extra_argv+1);
5271                                 }
5272                                 talloc_free(nodes);
5273                         } else {
5274                                 ret = ctdb_commands[i].fn(ctdb, extra_argc-1, extra_argv+1);
5275                         }
5276                         break;
5277                 }
5278         }
5279
5280         if (i == ARRAY_SIZE(ctdb_commands)) {
5281                 DEBUG(DEBUG_ERR, ("Unknown control '%s'\n", control));
5282                 exit(1);
5283         }
5284
5285         return ret;
5286 }