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