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