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