server/ctdb_recover.c

   1 /*
   2    ctdb recovery code
   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 #include "includes.h"
  21 #include "lib/events/events.h"
  22 #include "lib/tdb/include/tdb.h"
  23 #include "system/time.h"
  24 #include "system/network.h"
  25 #include "system/filesys.h"
  26 #include "system/wait.h"
  27 #include "../include/ctdb_private.h"
  28 #include "lib/util/dlinklist.h"
  29 #include "db_wrap.h"
  30
  31 /*
  32   lock all databases - mark only
  33  */
  34 static int ctdb_lock_all_databases_mark(struct ctdb_context *ctdb, uint32_t priority)
  35 {
  36         struct ctdb_db_context *ctdb_db;
  37
  38         if ((priority < 1) || (priority > NUM_DB_PRIORITIES)) {
  39                 DEBUG(DEBUG_ERR,(__location__ " Illegal priority when trying to mark all databases Prio:%u\n", priority));
  40                 return -1;
  41         }
  42
  43         if (ctdb->freeze_mode[priority] != CTDB_FREEZE_FROZEN) {
  44                 DEBUG(DEBUG_ERR,("Attempt to mark all databases locked when not frozen\n"));
  45                 return -1;
  46         }
  47         /* The dual loop is a woraround for older versions of samba
  48            that does not yet support the set-db-priority/lock order
  49            call. So that we get basic deadlock avoiidance also for
  50            these old versions of samba.
  51            This code will be removed in the future.
  52         */
  53         for (ctdb_db=ctdb->db_list;ctdb_db;ctdb_db=ctdb_db->next) {
  54                 if (ctdb_db->priority != priority) {
  55                         continue;
  56                 }
  57                 if (strstr(ctdb_db->db_name, "notify") != NULL) {
  58                         continue;
  59                 }
  60                 if (tdb_lockall_mark(ctdb_db->ltdb->tdb) != 0) {
  61                         return -1;
  62                 }
  63         }
  64         for (ctdb_db=ctdb->db_list;ctdb_db;ctdb_db=ctdb_db->next) {
  65                 if (ctdb_db->priority != priority) {
  66                         continue;
  67                 }
  68                 if (strstr(ctdb_db->db_name, "notify") == NULL) {
  69                         continue;
  70                 }
  71                 if (tdb_lockall_mark(ctdb_db->ltdb->tdb) != 0) {
  72                         return -1;
  73                 }
  74         }
  75         return 0;
  76 }
  77
  78 /*
  79   lock all databases - unmark only
  80  */
  81 static int ctdb_lock_all_databases_unmark(struct ctdb_context *ctdb, uint32_t priority)
  82 {
  83         struct ctdb_db_context *ctdb_db;
  84
  85         if ((priority < 1) || (priority > NUM_DB_PRIORITIES)) {
  86                 DEBUG(DEBUG_ERR,(__location__ " Illegal priority when trying to mark all databases Prio:%u\n", priority));
  87                 return -1;
  88         }
  89
  90         if (ctdb->freeze_mode[priority] != CTDB_FREEZE_FROZEN) {
  91                 DEBUG(DEBUG_ERR,("Attempt to unmark all databases locked when not frozen\n"));
  92                 return -1;
  93         }
  94         for (ctdb_db=ctdb->db_list;ctdb_db;ctdb_db=ctdb_db->next) {
  95                 if (ctdb_db->priority != priority) {
  96                         continue;
  97                 }
  98                 if (tdb_lockall_unmark(ctdb_db->ltdb->tdb) != 0) {
  99                         return -1;
 100                 }
 101         }
 102         return 0;
 103 }
 104
 105
 106 int
 107 ctdb_control_getvnnmap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
 108 {
 109         CHECK_CONTROL_DATA_SIZE(0);
 110         struct ctdb_vnn_map_wire *map;
 111         size_t len;
 112
 113         len = offsetof(struct ctdb_vnn_map_wire, map) + sizeof(uint32_t)*ctdb->vnn_map->size;
 114         map = talloc_size(outdata, len);
 115         CTDB_NO_MEMORY(ctdb, map);
 116
 117         map->generation = ctdb->vnn_map->generation;
 118         map->size = ctdb->vnn_map->size;
 119         memcpy(map->map, ctdb->vnn_map->map, sizeof(uint32_t)*map->size);
 120
 121         outdata->dsize = len;
 122         outdata->dptr  = (uint8_t *)map;
 123
 124         return 0;
 125 }
 126
 127 int
 128 ctdb_control_setvnnmap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
 129 {
 130         struct ctdb_vnn_map_wire *map = (struct ctdb_vnn_map_wire *)indata.dptr;
 131         int i;
 132
 133         for(i=1; i<=NUM_DB_PRIORITIES; i++) {
 134                 if (ctdb->freeze_mode[i] != CTDB_FREEZE_FROZEN) {
 135                         DEBUG(DEBUG_ERR,("Attempt to set vnnmap when not frozen\n"));
 136                         return -1;
 137                 }
 138         }
 139
 140         talloc_free(ctdb->vnn_map);
 141
 142         ctdb->vnn_map = talloc(ctdb, struct ctdb_vnn_map);
 143         CTDB_NO_MEMORY(ctdb, ctdb->vnn_map);
 144
 145         ctdb->vnn_map->generation = map->generation;
 146         ctdb->vnn_map->size       = map->size;
 147         ctdb->vnn_map->map = talloc_array(ctdb->vnn_map, uint32_t, map->size);
 148         CTDB_NO_MEMORY(ctdb, ctdb->vnn_map->map);
 149
 150         memcpy(ctdb->vnn_map->map, map->map, sizeof(uint32_t)*map->size);
 151
 152         return 0;
 153 }
 154
 155 int
 156 ctdb_control_getdbmap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
 157 {
 158         uint32_t i, len;
 159         struct ctdb_db_context *ctdb_db;
 160         struct ctdb_dbid_map *dbid_map;
 161
 162         CHECK_CONTROL_DATA_SIZE(0);
 163
 164         len = 0;
 165         for(ctdb_db=ctdb->db_list;ctdb_db;ctdb_db=ctdb_db->next){
 166                 len++;
 167         }
 168
 169
 170         outdata->dsize = offsetof(struct ctdb_dbid_map, dbs) + sizeof(dbid_map->dbs[0])*len;
 171         outdata->dptr  = (unsigned char *)talloc_zero_size(outdata, outdata->dsize);
 172         if (!outdata->dptr) {
 173                 DEBUG(DEBUG_ALERT, (__location__ " Failed to allocate dbmap array\n"));
 174                 exit(1);
 175         }
 176
 177         dbid_map = (struct ctdb_dbid_map *)outdata->dptr;
 178         dbid_map->num = len;
 179         for (i=0,ctdb_db=ctdb->db_list;ctdb_db;i++,ctdb_db=ctdb_db->next){
 180                 dbid_map->dbs[i].dbid       = ctdb_db->db_id;
 181                 dbid_map->dbs[i].persistent = ctdb_db->persistent;
 182         }
 183
 184         return 0;
 185 }
 186
 187 int
 188 ctdb_control_getnodemap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
 189 {
 190         uint32_t i, num_nodes;
 191         struct ctdb_node_map *node_map;
 192
 193         CHECK_CONTROL_DATA_SIZE(0);
 194
 195         num_nodes = ctdb->num_nodes;
 196
 197         outdata->dsize = offsetof(struct ctdb_node_map, nodes) + num_nodes*sizeof(struct ctdb_node_and_flags);
 198         outdata->dptr  = (unsigned char *)talloc_zero_size(outdata, outdata->dsize);
 199         if (!outdata->dptr) {
 200                 DEBUG(DEBUG_ALERT, (__location__ " Failed to allocate nodemap array\n"));
 201                 exit(1);
 202         }
 203
 204         node_map = (struct ctdb_node_map *)outdata->dptr;
 205         node_map->num = num_nodes;
 206         for (i=0; i<num_nodes; i++) {
 207                 if (parse_ip(ctdb->nodes[i]->address.address,
 208                              NULL, /* TODO: pass in the correct interface here*/
 209                              0,
 210                              &node_map->nodes[i].addr) == 0)
 211                 {
 212                         DEBUG(DEBUG_ERR, (__location__ " Failed to parse %s into a sockaddr\n", ctdb->nodes[i]->address.address));
 213                 }
 214
 215                 node_map->nodes[i].pnn   = ctdb->nodes[i]->pnn;
 216                 node_map->nodes[i].flags = ctdb->nodes[i]->flags;
 217         }
 218
 219         return 0;
 220 }
 221
 222 /*
 223    get an old style ipv4-only nodemap
 224 */
 225 int
 226 ctdb_control_getnodemapv4(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
 227 {
 228         uint32_t i, num_nodes;
 229         struct ctdb_node_mapv4 *node_map;
 230
 231         CHECK_CONTROL_DATA_SIZE(0);
 232
 233         num_nodes = ctdb->num_nodes;
 234
 235         outdata->dsize = offsetof(struct ctdb_node_mapv4, nodes) + num_nodes*sizeof(struct ctdb_node_and_flagsv4);
 236         outdata->dptr  = (unsigned char *)talloc_zero_size(outdata, outdata->dsize);
 237         if (!outdata->dptr) {
 238                 DEBUG(DEBUG_ALERT, (__location__ " Failed to allocate nodemap array\n"));
 239                 exit(1);
 240         }
 241
 242         node_map = (struct ctdb_node_mapv4 *)outdata->dptr;
 243         node_map->num = num_nodes;
 244         for (i=0; i<num_nodes; i++) {
 245                 if (parse_ipv4(ctdb->nodes[i]->address.address, 0, &node_map->nodes[i].sin) == 0) {
 246                         DEBUG(DEBUG_ERR, (__location__ " Failed to parse %s into a sockaddr\n", ctdb->nodes[i]->address.address));
 247                         return -1;
 248                 }
 249
 250                 node_map->nodes[i].pnn   = ctdb->nodes[i]->pnn;
 251                 node_map->nodes[i].flags = ctdb->nodes[i]->flags;
 252         }
 253
 254         return 0;
 255 }
 256
 257 static void
 258 ctdb_reload_nodes_event(struct event_context *ev, struct timed_event *te,
 259                                struct timeval t, void *private_data)
 260 {
 261         int i, num_nodes;
 262         struct ctdb_context *ctdb = talloc_get_type(private_data, struct ctdb_context);
 263         TALLOC_CTX *tmp_ctx;
 264         struct ctdb_node **nodes;
 265
 266         tmp_ctx = talloc_new(ctdb);
 267
 268         /* steal the old nodes file for a while */
 269         talloc_steal(tmp_ctx, ctdb->nodes);
 270         nodes = ctdb->nodes;
 271         ctdb->nodes = NULL;
 272         num_nodes = ctdb->num_nodes;
 273         ctdb->num_nodes = 0;
 274
 275         /* load the new nodes file */
 276         ctdb_load_nodes_file(ctdb);
 277
 278         for (i=0; i<ctdb->num_nodes; i++) {
 279                 /* keep any identical pre-existing nodes and connections */
 280                 if ((i < num_nodes) && ctdb_same_address(&ctdb->nodes[i]->address, &nodes[i]->address)) {
 281                         talloc_free(ctdb->nodes[i]);
 282                         ctdb->nodes[i] = talloc_steal(ctdb->nodes, nodes[i]);
 283                         continue;
 284                 }
 285
 286                 if (ctdb->nodes[i]->flags & NODE_FLAGS_DELETED) {
 287                         continue;
 288                 }
 289
 290                 /* any new or different nodes must be added */
 291                 if (ctdb->methods->add_node(ctdb->nodes[i]) != 0) {
 292                         DEBUG(DEBUG_CRIT, (__location__ " methods->add_node failed at %d\n", i));
 293                         ctdb_fatal(ctdb, "failed to add node. shutting down\n");
 294                 }
 295                 if (ctdb->methods->connect_node(ctdb->nodes[i]) != 0) {
 296                         DEBUG(DEBUG_CRIT, (__location__ " methods->add_connect failed at %d\n", i));
 297                         ctdb_fatal(ctdb, "failed to connect to node. shutting down\n");
 298                 }
 299         }
 300
 301         /* tell the recovery daemon to reaload the nodes file too */
 302         ctdb_daemon_send_message(ctdb, ctdb->pnn, CTDB_SRVID_RELOAD_NODES, tdb_null);
 303
 304         talloc_free(tmp_ctx);
 305         return;
 306 }
 307
 308 /*
 309   reload the nodes file after a short delay (so that we can send the response
 310   back first
 311 */
 312 int
 313 ctdb_control_reload_nodes_file(struct ctdb_context *ctdb, uint32_t opcode)
 314 {
 315         event_add_timed(ctdb->ev, ctdb, timeval_current_ofs(1,0), ctdb_reload_nodes_event, ctdb);
 316
 317         return 0;
 318 }
 319
 320 /*
 321    a traverse function for pulling all relevent records from pulldb
 322  */
 323 struct pulldb_data {
 324         struct ctdb_context *ctdb;
 325         struct ctdb_marshall_buffer *pulldata;
 326         uint32_t len;
 327         bool failed;
 328 };
 329
 330 static int traverse_pulldb(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, void *p)
 331 {
 332         struct pulldb_data *params = (struct pulldb_data *)p;
 333         struct ctdb_rec_data *rec;
 334
 335         /* add the record to the blob */
 336         rec = ctdb_marshall_record(params->pulldata, 0, key, NULL, data);
 337         if (rec == NULL) {
 338                 params->failed = true;
 339                 return -1;
 340         }
 341         params->pulldata = talloc_realloc_size(NULL, params->pulldata, rec->length + params->len);
 342         if (params->pulldata == NULL) {
 343                 DEBUG(DEBUG_ERR,(__location__ " Failed to expand pulldb_data to %u (%u records)\n",
 344                          rec->length + params->len, params->pulldata->count));
 345                 params->failed = true;
 346                 return -1;
 347         }
 348         params->pulldata->count++;
 349         memcpy(params->len+(uint8_t *)params->pulldata, rec, rec->length);
 350         params->len += rec->length;
 351         talloc_free(rec);
 352
 353         return 0;
 354 }
 355
 356 /*
 357   pull a bunch of records from a ltdb, filtering by lmaster
 358  */
 359 int32_t ctdb_control_pull_db(struct ctdb_context *ctdb, TDB_DATA indata, TDB_DATA *outdata)
 360 {
 361         struct ctdb_control_pulldb *pull;
 362         struct ctdb_db_context *ctdb_db;
 363         struct pulldb_data params;
 364         struct ctdb_marshall_buffer *reply;
 365
 366         pull = (struct ctdb_control_pulldb *)indata.dptr;
 367
 368         ctdb_db = find_ctdb_db(ctdb, pull->db_id);
 369         if (!ctdb_db) {
 370                 DEBUG(DEBUG_ERR,(__location__ " Unknown db 0x%08x\n", pull->db_id));
 371                 return -1;
 372         }
 373
 374         if (ctdb->freeze_mode[ctdb_db->priority] != CTDB_FREEZE_FROZEN) {
 375                 DEBUG(DEBUG_DEBUG,("rejecting ctdb_control_pull_db when not frozen\n"));
 376                 return -1;
 377         }
 378
 379         reply = talloc_zero(outdata, struct ctdb_marshall_buffer);
 380         CTDB_NO_MEMORY(ctdb, reply);
 381
 382         reply->db_id = pull->db_id;
 383
 384         params.ctdb = ctdb;
 385         params.pulldata = reply;
 386         params.len = offsetof(struct ctdb_marshall_buffer, data);
 387         params.failed = false;
 388
 389         if (ctdb_db->unhealthy_reason) {
 390                 /* this is just a warning, as the tdb should be empty anyway */
 391                 DEBUG(DEBUG_WARNING,("db(%s) unhealty in ctdb_control_pull_db: %s\n",
 392                                      ctdb_db->db_name, ctdb_db->unhealthy_reason));
 393         }
 394
 395         if (ctdb_lock_all_databases_mark(ctdb, ctdb_db->priority) != 0) {
 396                 DEBUG(DEBUG_ERR,(__location__ " Failed to get lock on entired db - failing\n"));
 397                 return -1;
 398         }
 399
 400         if (tdb_traverse_read(ctdb_db->ltdb->tdb, traverse_pulldb, &params) == -1) {
 401                 DEBUG(DEBUG_ERR,(__location__ " Failed to get traverse db '%s'\n", ctdb_db->db_name));
 402                 ctdb_lock_all_databases_unmark(ctdb, ctdb_db->priority);
 403                 talloc_free(params.pulldata);
 404                 return -1;
 405         }
 406
 407         ctdb_lock_all_databases_unmark(ctdb, ctdb_db->priority);
 408
 409         outdata->dptr = (uint8_t *)params.pulldata;
 410         outdata->dsize = params.len;
 411
 412         return 0;
 413 }
 414
 415 /*
 416   push a bunch of records into a ltdb, filtering by rsn
 417  */
 418 int32_t ctdb_control_push_db(struct ctdb_context *ctdb, TDB_DATA indata)
 419 {
 420         struct ctdb_marshall_buffer *reply = (struct ctdb_marshall_buffer *)indata.dptr;
 421         struct ctdb_db_context *ctdb_db;
 422         int i, ret;
 423         struct ctdb_rec_data *rec;
 424
 425         if (indata.dsize < offsetof(struct ctdb_marshall_buffer, data)) {
 426                 DEBUG(DEBUG_ERR,(__location__ " invalid data in pulldb reply\n"));
 427                 return -1;
 428         }
 429
 430         ctdb_db = find_ctdb_db(ctdb, reply->db_id);
 431         if (!ctdb_db) {
 432                 DEBUG(DEBUG_ERR,(__location__ " Unknown db 0x%08x\n", reply->db_id));
 433                 return -1;
 434         }
 435
 436         if (ctdb->freeze_mode[ctdb_db->priority] != CTDB_FREEZE_FROZEN) {
 437                 DEBUG(DEBUG_DEBUG,("rejecting ctdb_control_push_db when not frozen\n"));
 438                 return -1;
 439         }
 440
 441         if (ctdb_lock_all_databases_mark(ctdb, ctdb_db->priority) != 0) {
 442                 DEBUG(DEBUG_ERR,(__location__ " Failed to get lock on entired db - failing\n"));
 443                 return -1;
 444         }
 445
 446         rec = (struct ctdb_rec_data *)&reply->data[0];
 447
 448         DEBUG(DEBUG_INFO,("starting push of %u records for dbid 0x%x\n",
 449                  reply->count, reply->db_id));
 450
 451         for (i=0;i<reply->count;i++) {
 452                 TDB_DATA key, data;
 453                 struct ctdb_ltdb_header *hdr;
 454
 455                 key.dptr = &rec->data[0];
 456                 key.dsize = rec->keylen;
 457                 data.dptr = &rec->data[key.dsize];
 458                 data.dsize = rec->datalen;
 459
 460                 if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
 461                         DEBUG(DEBUG_CRIT,(__location__ " bad ltdb record\n"));
 462                         goto failed;
 463                 }
 464                 hdr = (struct ctdb_ltdb_header *)data.dptr;
 465                 data.dptr += sizeof(*hdr);
 466                 data.dsize -= sizeof(*hdr);
 467
 468                 ret = ctdb_ltdb_store(ctdb_db, key, hdr, data);
 469                 if (ret != 0) {
 470                         DEBUG(DEBUG_CRIT, (__location__ " Unable to store record\n"));
 471                         goto failed;
 472                 }
 473
 474                 rec = (struct ctdb_rec_data *)(rec->length + (uint8_t *)rec);
 475         }
 476
 477         DEBUG(DEBUG_DEBUG,("finished push of %u records for dbid 0x%x\n",
 478                  reply->count, reply->db_id));
 479
 480         ctdb_lock_all_databases_unmark(ctdb, ctdb_db->priority);
 481         return 0;
 482
 483 failed:
 484         ctdb_lock_all_databases_unmark(ctdb, ctdb_db->priority);
 485         return -1;
 486 }
 487
 488 struct ctdb_set_recmode_state {
 489         struct ctdb_context *ctdb;
 490         struct ctdb_req_control *c;
 491         uint32_t recmode;
 492         int fd[2];
 493         struct timed_event *te;
 494         struct fd_event *fde;
 495         pid_t child;
 496         struct timeval start_time;
 497 };
 498
 499 /*
 500   called if our set_recmode child times out. this would happen if
 501   ctdb_recovery_lock() would block.
 502  */
 503 static void ctdb_set_recmode_timeout(struct event_context *ev, struct timed_event *te,
 504                                          struct timeval t, void *private_data)
 505 {
 506         struct ctdb_set_recmode_state *state = talloc_get_type(private_data,
 507                                            struct ctdb_set_recmode_state);
 508
 509         /* we consider this a success, not a failure, as we failed to
 510            set the recovery lock which is what we wanted.  This can be
 511            caused by the cluster filesystem being very slow to
 512            arbitrate locks immediately after a node failure.
 513          */
 514         DEBUG(DEBUG_ERR,(__location__ " set_recmode child process hung/timedout CFS slow to grant locks? (allowing recmode set anyway)\n"));
 515         state->ctdb->recovery_mode = state->recmode;
 516         ctdb_request_control_reply(state->ctdb, state->c, NULL, 0, NULL);
 517         talloc_free(state);
 518 }
 519
 520
 521 /* when we free the recmode state we must kill any child process.
 522 */
 523 static int set_recmode_destructor(struct ctdb_set_recmode_state *state)
 524 {
 525         double l = timeval_elapsed(&state->start_time);
 526
 527         ctdb_reclock_latency(state->ctdb, "daemon reclock", &state->ctdb->statistics.reclock.ctdbd, l);
 528
 529         if (state->fd[0] != -1) {
 530                 state->fd[0] = -1;
 531         }
 532         if (state->fd[1] != -1) {
 533                 state->fd[1] = -1;
 534         }
 535         kill(state->child, SIGKILL);
 536         return 0;
 537 }
 538
 539 /* this is called when the client process has completed ctdb_recovery_lock()
 540    and has written data back to us through the pipe.
 541 */
 542 static void set_recmode_handler(struct event_context *ev, struct fd_event *fde,
 543                              uint16_t flags, void *private_data)
 544 {
 545         struct ctdb_set_recmode_state *state= talloc_get_type(private_data,
 546                                              struct ctdb_set_recmode_state);
 547         char c = 0;
 548         int ret;
 549
 550         /* we got a response from our child process so we can abort the
 551            timeout.
 552         */
 553         talloc_free(state->te);
 554         state->te = NULL;
 555
 556
 557         /* read the childs status when trying to lock the reclock file.
 558            child wrote 0 if everything is fine and 1 if it did manage
 559            to lock the file, which would be a problem since that means
 560            we got a request to exit from recovery but we could still lock
 561            the file   which at this time SHOULD be locked by the recovery
 562            daemon on the recmaster
 563         */
 564         ret = read(state->fd[0], &c, 1);
 565         if (ret != 1 || c != 0) {
 566                 ctdb_request_control_reply(state->ctdb, state->c, NULL, -1, "managed to lock reclock file from inside daemon");
 567                 talloc_free(state);
 568                 return;
 569         }
 570
 571         state->ctdb->recovery_mode = state->recmode;
 572
 573         ctdb_request_control_reply(state->ctdb, state->c, NULL, 0, NULL);
 574         talloc_free(state);
 575         return;
 576 }
 577
 578 static void
 579 ctdb_drop_all_ips_event(struct event_context *ev, struct timed_event *te,
 580                                struct timeval t, void *private_data)
 581 {
 582         struct ctdb_context *ctdb = talloc_get_type(private_data, struct ctdb_context);
 583
 584         DEBUG(DEBUG_ERR,(__location__ " Been in recovery mode for too long. Dropping all IPS\n"));
 585         talloc_free(ctdb->release_ips_ctx);
 586         ctdb->release_ips_ctx = NULL;
 587
 588         ctdb_release_all_ips(ctdb);
 589 }
 590
 591 /*
 592   set the recovery mode
 593  */
 594 int32_t ctdb_control_set_recmode(struct ctdb_context *ctdb,
 595                                  struct ctdb_req_control *c,
 596                                  TDB_DATA indata, bool *async_reply,
 597                                  const char **errormsg)
 598 {
 599         uint32_t recmode = *(uint32_t *)indata.dptr;
 600         int i, ret;
 601         struct ctdb_set_recmode_state *state;
 602         pid_t parent = getpid();
 603
 604         /* if we enter recovery but stay in recovery for too long
 605            we will eventually drop all our ip addresses
 606         */
 607         if (recmode == CTDB_RECOVERY_NORMAL) {
 608                 talloc_free(ctdb->release_ips_ctx);
 609                 ctdb->release_ips_ctx = NULL;
 610         } else {
 611                 talloc_free(ctdb->release_ips_ctx);
 612                 ctdb->release_ips_ctx = talloc_new(ctdb);
 613                 CTDB_NO_MEMORY(ctdb, ctdb->release_ips_ctx);
 614
 615                 event_add_timed(ctdb->ev, ctdb->release_ips_ctx, timeval_current_ofs(ctdb->tunable.recovery_drop_all_ips, 0), ctdb_drop_all_ips_event, ctdb);
 616         }
 617
 618         if (recmode != ctdb->recovery_mode) {
 619                 DEBUG(DEBUG_NOTICE,(__location__ " Recovery mode set to %s\n",
 620                          recmode==CTDB_RECOVERY_NORMAL?"NORMAL":"ACTIVE"));
 621         }
 622
 623         if (recmode != CTDB_RECOVERY_NORMAL ||
 624             ctdb->recovery_mode != CTDB_RECOVERY_ACTIVE) {
 625                 ctdb->recovery_mode = recmode;
 626                 return 0;
 627         }
 628
 629         /* some special handling when ending recovery mode */
 630
 631         /* force the databases to thaw */
 632         for (i=1; i<=NUM_DB_PRIORITIES; i++) {
 633                 if (ctdb->freeze_handles[i] != NULL) {
 634                         ctdb_control_thaw(ctdb, i);
 635                 }
 636         }
 637
 638         state = talloc(ctdb, struct ctdb_set_recmode_state);
 639         CTDB_NO_MEMORY(ctdb, state);
 640
 641         state->start_time = timeval_current();
 642         state->fd[0] = -1;
 643         state->fd[1] = -1;
 644
 645         if (ctdb->tunable.verify_recovery_lock == 0) {
 646                 /* dont need to verify the reclock file */
 647                 ctdb->recovery_mode = recmode;
 648                 return 0;
 649         }
 650
 651         /* For the rest of what needs to be done, we need to do this in
 652            a child process since
 653            1, the call to ctdb_recovery_lock() can block if the cluster
 654               filesystem is in the process of recovery.
 655         */
 656         ret = pipe(state->fd);
 657         if (ret != 0) {
 658                 talloc_free(state);
 659                 DEBUG(DEBUG_CRIT,(__location__ " Failed to open pipe for set_recmode child\n"));
 660                 return -1;
 661         }
 662
 663         state->child = fork();
 664         if (state->child == (pid_t)-1) {
 665                 close(state->fd[0]);
 666                 close(state->fd[1]);
 667                 talloc_free(state);
 668                 return -1;
 669         }
 670
 671         if (state->child == 0) {
 672                 char cc = 0;
 673                 close(state->fd[0]);
 674
 675                 /* we should not be able to get the lock on the reclock file,
 676                   as it should  be held by the recovery master
 677                 */
 678                 if (ctdb_recovery_lock(ctdb, false)) {
 679                         DEBUG(DEBUG_CRIT,("ERROR: recovery lock file %s not locked when recovering!\n", ctdb->recovery_lock_file));
 680                         cc = 1;
 681                 }
 682
 683                 write(state->fd[1], &cc, 1);
 684                 /* make sure we die when our parent dies */
 685                 while (kill(parent, 0) == 0 || errno != ESRCH) {
 686                         sleep(5);
 687                         write(state->fd[1], &cc, 1);
 688                 }
 689                 _exit(0);
 690         }
 691         close(state->fd[1]);
 692         set_close_on_exec(state->fd[0]);
 693
 694         state->fd[1] = -1;
 695
 696         talloc_set_destructor(state, set_recmode_destructor);
 697
 698         DEBUG(DEBUG_DEBUG, (__location__ " Created PIPE FD:%d for setrecmode\n", state->fd[0]));
 699
 700         state->te = event_add_timed(ctdb->ev, state, timeval_current_ofs(5, 0),
 701                                     ctdb_set_recmode_timeout, state);
 702
 703         state->fde = event_add_fd(ctdb->ev, state, state->fd[0],
 704                                 EVENT_FD_READ|EVENT_FD_AUTOCLOSE,
 705                                 set_recmode_handler,
 706                                 (void *)state);
 707
 708         if (state->fde == NULL) {
 709                 talloc_free(state);
 710                 return -1;
 711         }
 712
 713         state->ctdb    = ctdb;
 714         state->recmode = recmode;
 715         state->c       = talloc_steal(state, c);
 716
 717         *async_reply = true;
 718
 719         return 0;
 720 }
 721
 722
 723 /*
 724   try and get the recovery lock in shared storage - should only work
 725   on the recovery master recovery daemon. Anywhere else is a bug
 726  */
 727 bool ctdb_recovery_lock(struct ctdb_context *ctdb, bool keep)
 728 {
 729         struct flock lock;
 730
 731         if (keep) {
 732                 DEBUG(DEBUG_ERR, ("Take the recovery lock\n"));
 733         }
 734         if (ctdb->recovery_lock_fd != -1) {
 735                 close(ctdb->recovery_lock_fd);
 736                 ctdb->recovery_lock_fd = -1;
 737         }
 738
 739         ctdb->recovery_lock_fd = open(ctdb->recovery_lock_file, O_RDWR|O_CREAT, 0600);
 740         if (ctdb->recovery_lock_fd == -1) {
 741                 DEBUG(DEBUG_ERR,("ctdb_recovery_lock: Unable to open %s - (%s)\n",
 742                          ctdb->recovery_lock_file, strerror(errno)));
 743                 return false;
 744         }
 745
 746         set_close_on_exec(ctdb->recovery_lock_fd);
 747
 748         lock.l_type = F_WRLCK;
 749         lock.l_whence = SEEK_SET;
 750         lock.l_start = 0;
 751         lock.l_len = 1;
 752         lock.l_pid = 0;
 753
 754         if (fcntl(ctdb->recovery_lock_fd, F_SETLK, &lock) != 0) {
 755                 close(ctdb->recovery_lock_fd);
 756                 ctdb->recovery_lock_fd = -1;
 757                 if (keep) {
 758                         DEBUG(DEBUG_CRIT,("ctdb_recovery_lock: Failed to get recovery lock on '%s'\n", ctdb->recovery_lock_file));
 759                 }
 760                 return false;
 761         }
 762
 763         if (!keep) {
 764                 close(ctdb->recovery_lock_fd);
 765                 ctdb->recovery_lock_fd = -1;
 766         }
 767
 768         if (keep) {
 769                 DEBUG(DEBUG_ERR, ("Recovery lock taken successfully\n"));
 770         }
 771
 772         DEBUG(DEBUG_NOTICE,("ctdb_recovery_lock: Got recovery lock on '%s'\n", ctdb->recovery_lock_file));
 773
 774         return true;
 775 }
 776
 777 /*
 778   delete a record as part of the vacuum process
 779   only delete if we are not lmaster or dmaster, and our rsn is <= the provided rsn
 780   use non-blocking locks
 781
 782   return 0 if the record was successfully deleted (i.e. it does not exist
 783   when the function returns)
 784   or !0 is the record still exists in the tdb after returning.
 785  */
 786 static int delete_tdb_record(struct ctdb_context *ctdb, struct ctdb_db_context *ctdb_db, struct ctdb_rec_data *rec)
 787 {
 788         TDB_DATA key, data;
 789         struct ctdb_ltdb_header *hdr, *hdr2;
 790
 791         /* these are really internal tdb functions - but we need them here for
 792            non-blocking lock of the freelist */
 793         int tdb_lock_nonblock(struct tdb_context *tdb, int list, int ltype);
 794         int tdb_unlock(struct tdb_context *tdb, int list, int ltype);
 795
 796
 797         key.dsize = rec->keylen;
 798         key.dptr  = &rec->data[0];
 799         data.dsize = rec->datalen;
 800         data.dptr = &rec->data[rec->keylen];
 801
 802         if (ctdb_lmaster(ctdb, &key) == ctdb->pnn) {
 803                 DEBUG(DEBUG_INFO,(__location__ " Called delete on record where we are lmaster\n"));
 804                 return -1;
 805         }
 806
 807         if (data.dsize != sizeof(struct ctdb_ltdb_header)) {
 808                 DEBUG(DEBUG_ERR,(__location__ " Bad record size\n"));
 809                 return -1;
 810         }
 811
 812         hdr = (struct ctdb_ltdb_header *)data.dptr;
 813
 814         /* use a non-blocking lock */
 815         if (tdb_chainlock_nonblock(ctdb_db->ltdb->tdb, key) != 0) {
 816                 return -1;
 817         }
 818
 819         data = tdb_fetch(ctdb_db->ltdb->tdb, key);
 820         if (data.dptr == NULL) {
 821                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 822                 return 0;
 823         }
 824
 825         if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
 826                 if (tdb_lock_nonblock(ctdb_db->ltdb->tdb, -1, F_WRLCK) == 0) {
 827                         tdb_delete(ctdb_db->ltdb->tdb, key);
 828                         tdb_unlock(ctdb_db->ltdb->tdb, -1, F_WRLCK);
 829                         DEBUG(DEBUG_CRIT,(__location__ " Deleted corrupt record\n"));
 830                 }
 831                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 832                 free(data.dptr);
 833                 return 0;
 834         }
 835
 836         hdr2 = (struct ctdb_ltdb_header *)data.dptr;
 837
 838         if (hdr2->rsn > hdr->rsn) {
 839                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 840                 DEBUG(DEBUG_INFO,(__location__ " Skipping record with rsn=%llu - called with rsn=%llu\n",
 841                          (unsigned long long)hdr2->rsn, (unsigned long long)hdr->rsn));
 842                 free(data.dptr);
 843                 return -1;
 844         }
 845
 846         if (hdr2->dmaster == ctdb->pnn) {
 847                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 848                 DEBUG(DEBUG_INFO,(__location__ " Attempted delete record where we are the dmaster\n"));
 849                 free(data.dptr);
 850                 return -1;
 851         }
 852
 853         if (tdb_lock_nonblock(ctdb_db->ltdb->tdb, -1, F_WRLCK) != 0) {
 854                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 855                 free(data.dptr);
 856                 return -1;
 857         }
 858
 859         if (tdb_delete(ctdb_db->ltdb->tdb, key) != 0) {
 860                 tdb_unlock(ctdb_db->ltdb->tdb, -1, F_WRLCK);
 861                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 862                 DEBUG(DEBUG_INFO,(__location__ " Failed to delete record\n"));
 863                 free(data.dptr);
 864                 return -1;
 865         }
 866
 867         tdb_unlock(ctdb_db->ltdb->tdb, -1, F_WRLCK);
 868         tdb_chainunlock(ctdb_db->ltdb->tdb, key);
 869         free(data.dptr);
 870         return 0;
 871 }
 872
 873
 874
 875 struct recovery_callback_state {
 876         struct ctdb_req_control *c;
 877 };
 878
 879
 880 /*
 881   called when the 'recovered' event script has finished
 882  */
 883 static void ctdb_end_recovery_callback(struct ctdb_context *ctdb, int status, void *p)
 884 {
 885         struct recovery_callback_state *state = talloc_get_type(p, struct recovery_callback_state);
 886
 887         ctdb_enable_monitoring(ctdb);
 888
 889         if (status != 0) {
 890                 DEBUG(DEBUG_ERR,(__location__ " recovered event script failed (status %d)\n", status));
 891                 if (status == -ETIME) {
 892                         ctdb_ban_self(ctdb);
 893                 }
 894         }
 895
 896         ctdb_request_control_reply(ctdb, state->c, NULL, status, NULL);
 897         talloc_free(state);
 898
 899         gettimeofday(&ctdb->last_recovery_finished, NULL);
 900 }
 901
 902 /*
 903   recovery has finished
 904  */
 905 int32_t ctdb_control_end_recovery(struct ctdb_context *ctdb,
 906                                 struct ctdb_req_control *c,
 907                                 bool *async_reply)
 908 {
 909         int ret;
 910         struct recovery_callback_state *state;
 911
 912         DEBUG(DEBUG_NOTICE,("Recovery has finished\n"));
 913
 914         ctdb_persistent_finish_trans3_commits(ctdb);
 915
 916         state = talloc(ctdb, struct recovery_callback_state);
 917         CTDB_NO_MEMORY(ctdb, state);
 918
 919         state->c    = c;
 920
 921         ctdb_disable_monitoring(ctdb);
 922
 923         ret = ctdb_event_script_callback(ctdb, state,
 924                                          ctdb_end_recovery_callback,
 925                                          state,
 926                                          false,
 927                                          CTDB_EVENT_RECOVERED, "%s", "");
 928
 929         if (ret != 0) {
 930                 ctdb_enable_monitoring(ctdb);
 931
 932                 DEBUG(DEBUG_ERR,(__location__ " Failed to end recovery\n"));
 933                 talloc_free(state);
 934                 return -1;
 935         }
 936
 937         /* tell the control that we will be reply asynchronously */
 938         state->c    = talloc_steal(state, c);
 939         *async_reply = true;
 940         return 0;
 941 }
 942
 943 /*
 944   called when the 'startrecovery' event script has finished
 945  */
 946 static void ctdb_start_recovery_callback(struct ctdb_context *ctdb, int status, void *p)
 947 {
 948         struct recovery_callback_state *state = talloc_get_type(p, struct recovery_callback_state);
 949
 950         if (status != 0) {
 951                 DEBUG(DEBUG_ERR,(__location__ " startrecovery event script failed (status %d)\n", status));
 952         }
 953
 954         ctdb_request_control_reply(ctdb, state->c, NULL, status, NULL);
 955         talloc_free(state);
 956 }
 957
 958 /*
 959   run the startrecovery eventscript
 960  */
 961 int32_t ctdb_control_start_recovery(struct ctdb_context *ctdb,
 962                                 struct ctdb_req_control *c,
 963                                 bool *async_reply)
 964 {
 965         int ret;
 966         struct recovery_callback_state *state;
 967
 968         DEBUG(DEBUG_NOTICE,(__location__ " startrecovery eventscript has been invoked\n"));
 969         gettimeofday(&ctdb->last_recovery_started, NULL);
 970
 971         state = talloc(ctdb, struct recovery_callback_state);
 972         CTDB_NO_MEMORY(ctdb, state);
 973
 974         state->c    = talloc_steal(state, c);
 975
 976         ctdb_disable_monitoring(ctdb);
 977
 978         ret = ctdb_event_script_callback(ctdb, state,
 979                                          ctdb_start_recovery_callback,
 980                                          state, false,
 981                                          CTDB_EVENT_START_RECOVERY,
 982                                          "%s", "");
 983
 984         if (ret != 0) {
 985                 DEBUG(DEBUG_ERR,(__location__ " Failed to start recovery\n"));
 986                 talloc_free(state);
 987                 return -1;
 988         }
 989
 990         /* tell the control that we will be reply asynchronously */
 991         *async_reply = true;
 992         return 0;
 993 }
 994
 995 /*
 996  try to delete all these records as part of the vacuuming process
 997  and return the records we failed to delete
 998 */
 999 int32_t ctdb_control_try_delete_records(struct ctdb_context *ctdb, TDB_DATA indata, TDB_DATA *outdata)
1000 {
1001         struct ctdb_marshall_buffer *reply = (struct ctdb_marshall_buffer *)indata.dptr;
1002         struct ctdb_db_context *ctdb_db;
1003         int i;
1004         struct ctdb_rec_data *rec;
1005         struct ctdb_marshall_buffer *records;
1006
1007         if (indata.dsize < offsetof(struct ctdb_marshall_buffer, data)) {
1008                 DEBUG(DEBUG_ERR,(__location__ " invalid data in try_delete_records\n"));
1009                 return -1;
1010         }
1011
1012         ctdb_db = find_ctdb_db(ctdb, reply->db_id);
1013         if (!ctdb_db) {
1014                 DEBUG(DEBUG_ERR,(__location__ " Unknown db 0x%08x\n", reply->db_id));
1015                 return -1;
1016         }
1017
1018
1019         DEBUG(DEBUG_DEBUG,("starting try_delete_records of %u records for dbid 0x%x\n",
1020                  reply->count, reply->db_id));
1021
1022
1023         /* create a blob to send back the records we couldnt delete */
1024         records = (struct ctdb_marshall_buffer *)
1025                         talloc_zero_size(outdata,
1026                                     offsetof(struct ctdb_marshall_buffer, data));
1027         if (records == NULL) {
1028                 DEBUG(DEBUG_ERR,(__location__ " Out of memory\n"));
1029                 return -1;
1030         }
1031         records->db_id = ctdb_db->db_id;
1032
1033
1034         rec = (struct ctdb_rec_data *)&reply->data[0];
1035         for (i=0;i<reply->count;i++) {
1036                 TDB_DATA key, data;
1037
1038                 key.dptr = &rec->data[0];
1039                 key.dsize = rec->keylen;
1040                 data.dptr = &rec->data[key.dsize];
1041                 data.dsize = rec->datalen;
1042
1043                 if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
1044                         DEBUG(DEBUG_CRIT,(__location__ " bad ltdb record in indata\n"));
1045                         return -1;
1046                 }
1047
1048                 /* If we cant delete the record we must add it to the reply
1049                    so the lmaster knows it may not purge this record
1050                 */
1051                 if (delete_tdb_record(ctdb, ctdb_db, rec) != 0) {
1052                         size_t old_size;
1053                         struct ctdb_ltdb_header *hdr;
1054
1055                         hdr = (struct ctdb_ltdb_header *)data.dptr;
1056                         data.dptr += sizeof(*hdr);
1057                         data.dsize -= sizeof(*hdr);
1058
1059                         DEBUG(DEBUG_INFO, (__location__ " Failed to vacuum delete record with hash 0x%08x\n", ctdb_hash(&key)));
1060
1061                         old_size = talloc_get_size(records);
1062                         records = talloc_realloc_size(outdata, records, old_size + rec->length);
1063                         if (records == NULL) {
1064                                 DEBUG(DEBUG_ERR,(__location__ " Failed to expand\n"));
1065                                 return -1;
1066                         }
1067                         records->count++;
1068                         memcpy(old_size+(uint8_t *)records, rec, rec->length);
1069                 }
1070
1071                 rec = (struct ctdb_rec_data *)(rec->length + (uint8_t *)rec);
1072         }
1073
1074
1075         outdata->dptr = (uint8_t *)records;
1076         outdata->dsize = talloc_get_size(records);
1077
1078         return 0;
1079 }
1080
1081 /**
1082  * Store a record as part of the vacuum process:
1083  * This is called from the RECEIVE_RECORD control which
1084  * the lmaster uses to send the current empty copy
1085  * to all nodes for storing, before it lets the other
1086  * nodes delete the records in the second phase with
1087  * the TRY_DELETE_RECORDS control.
1088  *
1089  * Only store if we are not lmaster or dmaster, and our
1090  * rsn is <= the provided rsn. Use non-blocking locks.
1091  *
1092  * return 0 if the record was successfully stored.
1093  * return !0 if the record still exists in the tdb after returning.
1094  */
1095 static int store_tdb_record(struct ctdb_context *ctdb,
1096                             struct ctdb_db_context *ctdb_db,
1097                             struct ctdb_rec_data *rec)
1098 {
1099         TDB_DATA key, data, data2;
1100         struct ctdb_ltdb_header *hdr, *hdr2;
1101         int ret;
1102
1103         key.dsize = rec->keylen;
1104         key.dptr = &rec->data[0];
1105         data.dsize = rec->datalen;
1106         data.dptr = &rec->data[rec->keylen];
1107
1108         if (ctdb_lmaster(ctdb, &key) == ctdb->pnn) {
1109                 DEBUG(DEBUG_INFO, (__location__ " Called store_tdb_record "
1110                                    "where we are lmaster\n"));
1111                 return -1;
1112         }
1113
1114         if (data.dsize != sizeof(struct ctdb_ltdb_header)) {
1115                 DEBUG(DEBUG_ERR, (__location__ " Bad record size\n"));
1116                 return -1;
1117         }
1118
1119         hdr = (struct ctdb_ltdb_header *)data.dptr;
1120
1121         /* use a non-blocking lock */
1122         if (tdb_chainlock_nonblock(ctdb_db->ltdb->tdb, key) != 0) {
1123                 DEBUG(DEBUG_ERR, (__location__ " Failed to lock chain\n"));
1124                 return -1;
1125         }
1126
1127         data2 = tdb_fetch(ctdb_db->ltdb->tdb, key);
1128         if (data2.dptr == NULL || data2.dsize < sizeof(struct ctdb_ltdb_header)) {
1129                 tdb_store(ctdb_db->ltdb->tdb, key, data, 0);
1130                 DEBUG(DEBUG_INFO, (__location__ " Stored record\n"));
1131                 ret = 0;
1132                 goto done;
1133         }
1134
1135         hdr2 = (struct ctdb_ltdb_header *)data.dptr;
1136
1137         if (hdr2->rsn > hdr->rsn) {
1138                 DEBUG(DEBUG_INFO, (__location__ " Skipping record with "
1139                                    "rsn=%llu - called with rsn=%llu\n",
1140                                    (unsigned long long)hdr2->rsn,
1141                                    (unsigned long long)hdr->rsn));
1142                 ret = -1;
1143                 goto done;
1144         }
1145
1146         if (hdr2->dmaster == ctdb->pnn) {
1147                 DEBUG(DEBUG_INFO, (__location__ " Attempted to store record "
1148                                    "where we are the dmaster\n"));
1149                 ret = -1;
1150                 goto done;
1151         }
1152
1153         if (tdb_store(ctdb_db->ltdb->tdb, key, data, 0) != 0) {
1154                 DEBUG(DEBUG_INFO,(__location__ " Failed to store record\n"));
1155                 ret = -1;
1156                 goto done;
1157         }
1158
1159         ret = 0;
1160
1161 done:
1162         tdb_chainunlock(ctdb_db->ltdb->tdb, key);
1163         free(data2.dptr);
1164         return  ret;
1165 }
1166
1167
1168
1169 /**
1170  * Try to store all these records as part of the vacuuming process
1171  * and return the records we failed to store.
1172  */
1173 int32_t ctdb_control_receive_records(struct ctdb_context *ctdb,
1174                                      TDB_DATA indata, TDB_DATA *outdata)
1175 {
1176         struct ctdb_marshall_buffer *reply = (struct ctdb_marshall_buffer *)indata.dptr;
1177         struct ctdb_db_context *ctdb_db;
1178         int i;
1179         struct ctdb_rec_data *rec;
1180         struct ctdb_marshall_buffer *records;
1181
1182         if (indata.dsize < offsetof(struct ctdb_marshall_buffer, data)) {
1183                 DEBUG(DEBUG_ERR,
1184                       (__location__ " invalid data in receive_records\n"));
1185                 return -1;
1186         }
1187
1188         ctdb_db = find_ctdb_db(ctdb, reply->db_id);
1189         if (!ctdb_db) {
1190                 DEBUG(DEBUG_ERR, (__location__ " Unknown db 0x%08x\n",
1191                                   reply->db_id));
1192                 return -1;
1193         }
1194
1195         DEBUG(DEBUG_DEBUG, ("starting receive_records of %u records for "
1196                             "dbid 0x%x\n", reply->count, reply->db_id));
1197
1198         /* create a blob to send back the records we could not store */
1199         records = (struct ctdb_marshall_buffer *)
1200                         talloc_zero_size(outdata,
1201                                 offsetof(struct ctdb_marshall_buffer, data));
1202         if (records == NULL) {
1203                 DEBUG(DEBUG_ERR, (__location__ " Out of memory\n"));
1204                 return -1;
1205         }
1206         records->db_id = ctdb_db->db_id;
1207
1208         rec = (struct ctdb_rec_data *)&reply->data[0];
1209         for (i=0; i<reply->count; i++) {
1210                 TDB_DATA key, data;
1211
1212                 key.dptr = &rec->data[0];
1213                 key.dsize = rec->keylen;
1214                 data.dptr = &rec->data[key.dsize];
1215                 data.dsize = rec->datalen;
1216
1217                 if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
1218                         DEBUG(DEBUG_CRIT, (__location__ " bad ltdb record "
1219                                            "in indata\n"));
1220                         return -1;
1221                 }
1222
1223                 /*
1224                  * If we can not store the record we must add it to the reply
1225                  * so the lmaster knows it may not purge this record.
1226                  */
1227                 if (store_tdb_record(ctdb, ctdb_db, rec) != 0) {
1228                         size_t old_size;
1229                         struct ctdb_ltdb_header *hdr;
1230
1231                         hdr = (struct ctdb_ltdb_header *)data.dptr;
1232                         data.dptr += sizeof(*hdr);
1233                         data.dsize -= sizeof(*hdr);
1234
1235                         DEBUG(DEBUG_INFO, (__location__ " Failed to store "
1236                                            "record with hash 0x%08x in vacuum "
1237                                            "via RECEIVE_RECORDS\n",
1238                                            ctdb_hash(&key)));
1239
1240                         old_size = talloc_get_size(records);
1241                         records = talloc_realloc_size(outdata, records,
1242                                                       old_size + rec->length);
1243                         if (records == NULL) {
1244                                 DEBUG(DEBUG_ERR, (__location__ " Failed to "
1245                                                   "expand\n"));
1246                                 return -1;
1247                         }
1248                         records->count++;
1249                         memcpy(old_size+(uint8_t *)records, rec, rec->length);
1250                 }
1251
1252                 rec = (struct ctdb_rec_data *)(rec->length + (uint8_t *)rec);
1253         }
1254
1255
1256         outdata->dptr = (uint8_t *)records;
1257         outdata->dsize = talloc_get_size(records);
1258
1259         return 0;
1260 }
1261
1262
1263 /*
1264   report capabilities
1265  */
1266 int32_t ctdb_control_get_capabilities(struct ctdb_context *ctdb, TDB_DATA *outdata)
1267 {
1268         uint32_t *capabilities = NULL;
1269
1270         capabilities = talloc(outdata, uint32_t);
1271         CTDB_NO_MEMORY(ctdb, capabilities);
1272         *capabilities = ctdb->capabilities;
1273
1274         outdata->dsize = sizeof(uint32_t);
1275         outdata->dptr = (uint8_t *)capabilities;
1276
1277         return 0;
1278 }
1279
1280 static void ctdb_recd_ping_timeout(struct event_context *ev, struct timed_event *te, struct timeval t, void *p)
1281 {
1282         struct ctdb_context *ctdb = talloc_get_type(p, struct ctdb_context);
1283         uint32_t *count = talloc_get_type(ctdb->recd_ping_count, uint32_t);
1284
1285         DEBUG(DEBUG_ERR, ("Recovery daemon ping timeout. Count : %u\n", *count));
1286
1287         if (*count < ctdb->tunable.recd_ping_failcount) {
1288                 (*count)++;
1289                 event_add_timed(ctdb->ev, ctdb->recd_ping_count,
1290                         timeval_current_ofs(ctdb->tunable.recd_ping_timeout, 0),
1291                         ctdb_recd_ping_timeout, ctdb);
1292                 return;
1293         }
1294
1295         DEBUG(DEBUG_ERR, ("Final timeout for recovery daemon ping. Shutting down ctdb daemon. (This can be caused if the cluster filesystem has hung)\n"));
1296
1297         ctdb_stop_recoverd(ctdb);
1298         ctdb_stop_keepalive(ctdb);
1299         ctdb_stop_monitoring(ctdb);
1300         ctdb_release_all_ips(ctdb);
1301         if (ctdb->methods != NULL) {
1302                 ctdb->methods->shutdown(ctdb);
1303         }
1304         ctdb_event_script(ctdb, CTDB_EVENT_SHUTDOWN);
1305         DEBUG(DEBUG_ERR, ("Recovery daemon ping timeout. Daemon has been shut down.\n"));
1306         exit(0);
1307 }
1308
1309 /* The recovery daemon will ping us at regular intervals.
1310    If we havent been pinged for a while we assume the recovery
1311    daemon is inoperable and we shut down.
1312 */
1313 int32_t ctdb_control_recd_ping(struct ctdb_context *ctdb)
1314 {
1315         talloc_free(ctdb->recd_ping_count);
1316
1317         ctdb->recd_ping_count = talloc_zero(ctdb, uint32_t);
1318         CTDB_NO_MEMORY(ctdb, ctdb->recd_ping_count);
1319
1320         if (ctdb->tunable.recd_ping_timeout != 0) {
1321                 event_add_timed(ctdb->ev, ctdb->recd_ping_count,
1322                         timeval_current_ofs(ctdb->tunable.recd_ping_timeout, 0),
1323                         ctdb_recd_ping_timeout, ctdb);
1324         }
1325
1326         return 0;
1327 }
1328
1329
1330
1331 int32_t ctdb_control_set_recmaster(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata)
1332 {
1333         CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
1334
1335         ctdb->recovery_master = ((uint32_t *)(&indata.dptr[0]))[0];
1336         return 0;
1337 }
1338
1339
1340 struct stop_node_callback_state {
1341         struct ctdb_req_control *c;
1342 };
1343
1344 /*
1345   called when the 'stopped' event script has finished
1346  */
1347 static void ctdb_stop_node_callback(struct ctdb_context *ctdb, int status, void *p)
1348 {
1349         struct stop_node_callback_state *state = talloc_get_type(p, struct stop_node_callback_state);
1350
1351         if (status != 0) {
1352                 DEBUG(DEBUG_ERR,(__location__ " stopped event script failed (status %d)\n", status));
1353                 ctdb->nodes[ctdb->pnn]->flags &= ~NODE_FLAGS_STOPPED;
1354                 if (status == -ETIME) {
1355                         ctdb_ban_self(ctdb);
1356                 }
1357         }
1358
1359         ctdb_request_control_reply(ctdb, state->c, NULL, status, NULL);
1360         talloc_free(state);
1361 }
1362
1363 int32_t ctdb_control_stop_node(struct ctdb_context *ctdb, struct ctdb_req_control *c, bool *async_reply)
1364 {
1365         int ret;
1366         struct stop_node_callback_state *state;
1367
1368         DEBUG(DEBUG_INFO,(__location__ " Stopping node\n"));
1369
1370         state = talloc(ctdb, struct stop_node_callback_state);
1371         CTDB_NO_MEMORY(ctdb, state);
1372
1373         state->c    = talloc_steal(state, c);
1374
1375         ctdb_disable_monitoring(ctdb);
1376
1377         ret = ctdb_event_script_callback(ctdb, state,
1378                                          ctdb_stop_node_callback,
1379                                          state, false,
1380                                          CTDB_EVENT_STOPPED, "%s", "");
1381
1382         if (ret != 0) {
1383                 ctdb_enable_monitoring(ctdb);
1384
1385                 DEBUG(DEBUG_ERR,(__location__ " Failed to stop node\n"));
1386                 talloc_free(state);
1387                 return -1;
1388         }
1389
1390         ctdb->nodes[ctdb->pnn]->flags |= NODE_FLAGS_STOPPED;
1391
1392         *async_reply = true;
1393
1394         return 0;
1395 }
1396
1397 int32_t ctdb_control_continue_node(struct ctdb_context *ctdb)
1398 {
1399         DEBUG(DEBUG_INFO,(__location__ " Continue node\n"));
1400         ctdb->nodes[ctdb->pnn]->flags &= ~NODE_FLAGS_STOPPED;
1401
1402         return 0;
1403 }