ctdb/server/ctdb_control.c

   1 /*
   2    ctdb_control protocol code
   3
   4    Copyright (C) Andrew Tridgell  2007
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, see <http://www.gnu.org/licenses/>.
  18 */
  19 #include "includes.h"
  20 #include "tdb.h"
  21 #include "system/network.h"
  22 #include "system/filesys.h"
  23 #include "system/wait.h"
  24 #include "../include/ctdb_private.h"
  25 #include "lib/util/dlinklist.h"
  26 #include "lib/tdb_wrap/tdb_wrap.h"
  27 #include "lib/util/talloc_report.h"
  28
  29
  30 struct ctdb_control_state {
  31         struct ctdb_context *ctdb;
  32         uint32_t reqid;
  33         ctdb_control_callback_fn_t callback;
  34         void *private_data;
  35         unsigned flags;
  36 };
  37
  38
  39 /*
  40   dump talloc memory hierarchy, returning it as a blob to the client
  41  */
  42 int32_t ctdb_dump_memory(struct ctdb_context *ctdb, TDB_DATA *outdata)
  43 {
  44         char *report;
  45         size_t reportlen;
  46
  47         report = talloc_report_str(outdata, NULL);
  48         if (report == NULL) {
  49                 DEBUG(DEBUG_ERR,
  50                       (__location__ " talloc_report_str failed\n"));
  51                 return -1;
  52         }
  53         reportlen = talloc_get_size(report);
  54
  55         if (reportlen > 0) {
  56                 reportlen -= 1; /* strip trailing zero */
  57         }
  58
  59         outdata->dptr = (uint8_t *)report;
  60         outdata->dsize = reportlen;
  61         return 0;
  62 }
  63
  64 static int32_t control_not_implemented(const char *unsupported,
  65                                        const char *alternate)
  66 {
  67         if (alternate == NULL) {
  68                 DEBUG(DEBUG_ERR,
  69                       ("Control %s is not implemented any more\n",
  70                        unsupported));
  71         } else {
  72                 DEBUG(DEBUG_ERR,
  73                       ("Control %s is not implemented any more, use %s instead\n",
  74                        unsupported, alternate));
  75         }
  76         return -1;
  77 }
  78
  79 /*
  80   process a control request
  81  */
  82 static int32_t ctdb_control_dispatch(struct ctdb_context *ctdb,
  83                                      struct ctdb_req_control *c,
  84                                      TDB_DATA indata,
  85                                      TDB_DATA *outdata, uint32_t srcnode,
  86                                      const char **errormsg,
  87                                      bool *async_reply)
  88 {
  89         uint32_t opcode = c->opcode;
  90         uint64_t srvid = c->srvid;
  91         uint32_t client_id = c->client_id;
  92
  93         switch (opcode) {
  94         case CTDB_CONTROL_PROCESS_EXISTS: {
  95                 CHECK_CONTROL_DATA_SIZE(sizeof(pid_t));
  96                 return ctdb_control_process_exists(ctdb, *(pid_t *)indata.dptr);
  97         }
  98
  99         case CTDB_CONTROL_SET_DEBUG: {
 100                 CHECK_CONTROL_DATA_SIZE(sizeof(int32_t));
 101                 DEBUGLEVEL = *(int32_t *)indata.dptr;
 102                 return 0;
 103         }
 104
 105         case CTDB_CONTROL_GET_DEBUG: {
 106                 CHECK_CONTROL_DATA_SIZE(0);
 107                 outdata->dptr = (uint8_t *)&(DEBUGLEVEL);
 108                 outdata->dsize = sizeof(DEBUGLEVEL);
 109                 return 0;
 110         }
 111
 112         case CTDB_CONTROL_STATISTICS: {
 113                 CHECK_CONTROL_DATA_SIZE(0);
 114                 ctdb->statistics.memory_used = talloc_total_size(NULL);
 115                 ctdb->statistics.num_clients = ctdb->num_clients;
 116                 ctdb->statistics.frozen = (ctdb_db_all_frozen(ctdb) ? 1 : 0);
 117                 ctdb->statistics.recovering = (ctdb->recovery_mode == CTDB_RECOVERY_ACTIVE);
 118                 ctdb->statistics.statistics_current_time = timeval_current();
 119
 120                 outdata->dptr = (uint8_t *)&ctdb->statistics;
 121                 outdata->dsize = sizeof(ctdb->statistics);
 122                 return 0;
 123         }
 124
 125         case CTDB_CONTROL_GET_ALL_TUNABLES: {
 126                 CHECK_CONTROL_DATA_SIZE(0);
 127                 outdata->dptr = (uint8_t *)&ctdb->tunable;
 128                 outdata->dsize = sizeof(ctdb->tunable);
 129                 return 0;
 130         }
 131
 132         case CTDB_CONTROL_DUMP_MEMORY: {
 133                 CHECK_CONTROL_DATA_SIZE(0);
 134                 return ctdb_dump_memory(ctdb, outdata);
 135         }
 136
 137         case CTDB_CONTROL_STATISTICS_RESET: {
 138                 struct ctdb_db_context *ctdb_db;
 139
 140                 CHECK_CONTROL_DATA_SIZE(0);
 141                 ZERO_STRUCT(ctdb->statistics);
 142                 for (ctdb_db = ctdb->db_list;
 143                      ctdb_db != NULL;
 144                      ctdb_db = ctdb_db->next) {
 145                         ctdb_db_statistics_reset(ctdb_db);
 146                 }
 147                 ctdb->statistics.statistics_start_time = timeval_current();
 148                 return 0;
 149         }
 150
 151         case CTDB_CONTROL_GETVNNMAP:
 152                 return ctdb_control_getvnnmap(ctdb, opcode, indata, outdata);
 153
 154         case CTDB_CONTROL_GET_DBMAP:
 155                 return ctdb_control_getdbmap(ctdb, opcode, indata, outdata);
 156
 157         case CTDB_CONTROL_GET_NODEMAPv4:
 158                 return control_not_implemented("GET_NODEMAPv4", "GET_NODEMAP");
 159
 160         case CTDB_CONTROL_GET_NODEMAP:
 161                 return ctdb_control_getnodemap(ctdb, opcode, indata, outdata);
 162
 163         case CTDB_CONTROL_GET_NODES_FILE:
 164                 return ctdb_control_getnodesfile(ctdb, opcode, indata, outdata);
 165
 166         case CTDB_CONTROL_RELOAD_NODES_FILE:
 167                 CHECK_CONTROL_DATA_SIZE(0);
 168                 return ctdb_control_reload_nodes_file(ctdb, opcode);
 169
 170         case CTDB_CONTROL_SET_DB_STICKY: {
 171                 uint32_t db_id;
 172                 struct ctdb_db_context *ctdb_db;
 173
 174                 CHECK_CONTROL_DATA_SIZE(sizeof(db_id));
 175                 db_id = *(uint32_t *)indata.dptr;
 176                 ctdb_db = find_ctdb_db(ctdb, db_id);
 177                 if (ctdb_db == NULL) return -1;
 178                 return ctdb_set_db_sticky(ctdb, ctdb_db);
 179         }
 180
 181         case CTDB_CONTROL_SETVNNMAP:
 182                 return ctdb_control_setvnnmap(ctdb, opcode, indata, outdata);
 183
 184         case CTDB_CONTROL_PULL_DB:
 185                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_pulldb));
 186                 return ctdb_control_pull_db(ctdb, indata, outdata);
 187
 188         case CTDB_CONTROL_SET_DMASTER:
 189                 return control_not_implemented("SET_DMASTER", NULL);
 190
 191         case CTDB_CONTROL_PUSH_DB:
 192                 return ctdb_control_push_db(ctdb, indata);
 193
 194         case CTDB_CONTROL_GET_RECMODE: {
 195                 return ctdb->recovery_mode;
 196         }
 197
 198         case CTDB_CONTROL_SET_RECMASTER: {
 199                 return ctdb_control_set_recmaster(ctdb, opcode, indata);
 200         }
 201
 202         case CTDB_CONTROL_GET_RECMASTER:
 203                 return ctdb->recovery_master;
 204
 205         case CTDB_CONTROL_GET_PID:
 206                 return getpid();
 207
 208         case CTDB_CONTROL_GET_PNN:
 209                 return ctdb->pnn;
 210
 211         case CTDB_CONTROL_PING:
 212                 CHECK_CONTROL_DATA_SIZE(0);
 213                 return ctdb->num_clients;
 214
 215         case CTDB_CONTROL_GET_RUNSTATE:
 216                 CHECK_CONTROL_DATA_SIZE(0);
 217                 outdata->dptr = (uint8_t *)&ctdb->runstate;
 218                 outdata->dsize = sizeof(uint32_t);
 219                 return 0;
 220
 221
 222         case CTDB_CONTROL_SET_DB_READONLY: {
 223                 uint32_t db_id;
 224                 struct ctdb_db_context *ctdb_db;
 225
 226                 CHECK_CONTROL_DATA_SIZE(sizeof(db_id));
 227                 db_id = *(uint32_t *)indata.dptr;
 228                 ctdb_db = find_ctdb_db(ctdb, db_id);
 229                 if (ctdb_db == NULL) return -1;
 230                 return ctdb_set_db_readonly(ctdb, ctdb_db);
 231         }
 232         case CTDB_CONTROL_GET_DBNAME: {
 233                 uint32_t db_id;
 234                 struct ctdb_db_context *ctdb_db;
 235
 236                 CHECK_CONTROL_DATA_SIZE(sizeof(db_id));
 237                 db_id = *(uint32_t *)indata.dptr;
 238                 ctdb_db = find_ctdb_db(ctdb, db_id);
 239                 if (ctdb_db == NULL) return -1;
 240                 outdata->dptr = discard_const(ctdb_db->db_name);
 241                 outdata->dsize = strlen(ctdb_db->db_name)+1;
 242                 return 0;
 243         }
 244
 245         case CTDB_CONTROL_GETDBPATH: {
 246                 uint32_t db_id;
 247                 struct ctdb_db_context *ctdb_db;
 248
 249                 CHECK_CONTROL_DATA_SIZE(sizeof(db_id));
 250                 db_id = *(uint32_t *)indata.dptr;
 251                 ctdb_db = find_ctdb_db(ctdb, db_id);
 252                 if (ctdb_db == NULL) return -1;
 253                 outdata->dptr = discard_const(ctdb_db->db_path);
 254                 outdata->dsize = strlen(ctdb_db->db_path)+1;
 255                 return 0;
 256         }
 257
 258         case CTDB_CONTROL_DB_ATTACH:
 259           return ctdb_control_db_attach(ctdb, indata, outdata, srvid, false, client_id, c, async_reply);
 260
 261         case CTDB_CONTROL_DB_ATTACH_PERSISTENT:
 262           return ctdb_control_db_attach(ctdb, indata, outdata, srvid, true, client_id, c, async_reply);
 263
 264         case CTDB_CONTROL_SET_CALL:
 265                 return control_not_implemented("SET_CALL", NULL);
 266
 267         case CTDB_CONTROL_TRAVERSE_START:
 268                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_traverse_start));
 269                 return ctdb_control_traverse_start(ctdb, indata, outdata, srcnode, client_id);
 270
 271         case CTDB_CONTROL_TRAVERSE_START_EXT:
 272                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_traverse_start_ext));
 273                 return ctdb_control_traverse_start_ext(ctdb, indata, outdata, srcnode, client_id);
 274
 275         case CTDB_CONTROL_TRAVERSE_ALL:
 276                 return ctdb_control_traverse_all(ctdb, indata, outdata);
 277
 278         case CTDB_CONTROL_TRAVERSE_ALL_EXT:
 279                 return ctdb_control_traverse_all_ext(ctdb, indata, outdata);
 280
 281         case CTDB_CONTROL_TRAVERSE_DATA:
 282                 return ctdb_control_traverse_data(ctdb, indata, outdata);
 283
 284         case CTDB_CONTROL_TRAVERSE_KILL:
 285                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_traverse_start));
 286                 return ctdb_control_traverse_kill(ctdb, indata, outdata, srcnode);
 287
 288         case CTDB_CONTROL_REGISTER_SRVID:
 289                 return daemon_register_message_handler(ctdb, client_id, srvid);
 290
 291         case CTDB_CONTROL_DEREGISTER_SRVID:
 292                 return daemon_deregister_message_handler(ctdb, client_id, srvid);
 293
 294         case CTDB_CONTROL_CHECK_SRVIDS:
 295                 return daemon_check_srvids(ctdb, indata, outdata);
 296
 297         case CTDB_CONTROL_ENABLE_SEQNUM:
 298                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 299                 return ctdb_ltdb_enable_seqnum(ctdb, *(uint32_t *)indata.dptr);
 300
 301         case CTDB_CONTROL_UPDATE_SEQNUM:
 302                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 303                 return ctdb_ltdb_update_seqnum(ctdb, *(uint32_t *)indata.dptr, srcnode);
 304
 305         case CTDB_CONTROL_FREEZE:
 306                 CHECK_CONTROL_DATA_SIZE(0);
 307                 return ctdb_control_freeze(ctdb, c, async_reply);
 308
 309         case CTDB_CONTROL_THAW:
 310                 CHECK_CONTROL_DATA_SIZE(0);
 311                 return ctdb_control_thaw(ctdb, (uint32_t)c->srvid, true);
 312
 313         case CTDB_CONTROL_SET_RECMODE:
 314                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 315                 return ctdb_control_set_recmode(ctdb, c, indata, async_reply, errormsg);
 316
 317         case CTDB_CONTROL_GET_MONMODE:
 318                 CHECK_CONTROL_DATA_SIZE(0);
 319                 return ctdb_monitoring_mode(ctdb);
 320
 321         case CTDB_CONTROL_ENABLE_MONITOR:
 322                 CHECK_CONTROL_DATA_SIZE(0);
 323                 ctdb_enable_monitoring(ctdb);
 324                 return 0;
 325
 326         case CTDB_CONTROL_RUN_EVENTSCRIPTS:
 327                 return ctdb_run_eventscripts(ctdb, c, indata, async_reply);
 328
 329         case CTDB_CONTROL_DISABLE_MONITOR:
 330                 CHECK_CONTROL_DATA_SIZE(0);
 331                 ctdb_disable_monitoring(ctdb);
 332                 return 0;
 333
 334         case CTDB_CONTROL_SHUTDOWN:
 335                 DEBUG(DEBUG_NOTICE,("Received SHUTDOWN command.\n"));
 336                 ctdb_shutdown_sequence(ctdb, 0);
 337                 /* In case above returns due to duplicate shutdown */
 338                 return 0;
 339
 340         case CTDB_CONTROL_TAKEOVER_IPv4:
 341                 return control_not_implemented("TAKEOVER_IPv4", "TAKEOVER_IP");
 342
 343         case CTDB_CONTROL_TAKEOVER_IP:
 344                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_public_ip));
 345                 return ctdb_control_takeover_ip(ctdb, c, indata, async_reply);
 346
 347         case CTDB_CONTROL_RELEASE_IPv4:
 348                 return control_not_implemented("RELEASE_IPv4", "RELEASE_IP");
 349
 350         case CTDB_CONTROL_RELEASE_IP:
 351                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_public_ip));
 352                 return ctdb_control_release_ip(ctdb, c, indata, async_reply);
 353
 354         case CTDB_CONTROL_IPREALLOCATED:
 355                 CHECK_CONTROL_DATA_SIZE(0);
 356                 return ctdb_control_ipreallocated(ctdb, c, async_reply);
 357
 358         case CTDB_CONTROL_GET_PUBLIC_IPSv4:
 359                 return control_not_implemented("GET_PUBLIC_IPSv4",
 360                                                "GET_PUBLIC_IPS");
 361
 362         case CTDB_CONTROL_GET_PUBLIC_IPS:
 363                 CHECK_CONTROL_DATA_SIZE(0);
 364                 return ctdb_control_get_public_ips(ctdb, c, outdata);
 365
 366         case CTDB_CONTROL_TCP_CLIENT:
 367                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_tcp_addr));
 368                 return ctdb_control_tcp_client(ctdb, client_id, indata);
 369
 370         case CTDB_CONTROL_STARTUP:
 371                 CHECK_CONTROL_DATA_SIZE(0);
 372                 return ctdb_control_startup(ctdb, srcnode);
 373
 374         case CTDB_CONTROL_TCP_ADD:
 375                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
 376                 return ctdb_control_tcp_add(ctdb, indata, false);
 377
 378         case CTDB_CONTROL_TCP_ADD_DELAYED_UPDATE:
 379                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
 380                 return ctdb_control_tcp_add(ctdb, indata, true);
 381
 382         case CTDB_CONTROL_TCP_REMOVE:
 383                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
 384                 return ctdb_control_tcp_remove(ctdb, indata);
 385
 386         case CTDB_CONTROL_SET_TUNABLE:
 387                 return ctdb_control_set_tunable(ctdb, indata);
 388
 389         case CTDB_CONTROL_GET_TUNABLE:
 390                 return ctdb_control_get_tunable(ctdb, indata, outdata);
 391
 392         case CTDB_CONTROL_LIST_TUNABLES:
 393                 return ctdb_control_list_tunables(ctdb, outdata);
 394
 395         case CTDB_CONTROL_MODIFY_FLAGS:
 396                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_node_flag_change));
 397                 return ctdb_control_modflags(ctdb, indata);
 398
 399         case CTDB_CONTROL_KILL_TCP:
 400                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_tcp_connection));
 401                 return ctdb_control_kill_tcp(ctdb, indata);
 402
 403         case CTDB_CONTROL_GET_TCP_TICKLE_LIST:
 404                 CHECK_CONTROL_DATA_SIZE(sizeof(ctdb_sock_addr));
 405                 return ctdb_control_get_tcp_tickle_list(ctdb, indata, outdata);
 406
 407         case CTDB_CONTROL_SET_TCP_TICKLE_LIST:
 408                 /* data size is verified in the called function */
 409                 return ctdb_control_set_tcp_tickle_list(ctdb, indata);
 410
 411         case CTDB_CONTROL_REGISTER_SERVER_ID:
 412                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_server_id));
 413                 return ctdb_control_register_server_id(ctdb, client_id, indata);
 414
 415         case CTDB_CONTROL_UNREGISTER_SERVER_ID:
 416                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_server_id));
 417                 return ctdb_control_unregister_server_id(ctdb, indata);
 418
 419         case CTDB_CONTROL_CHECK_SERVER_ID:
 420                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_server_id));
 421                 return ctdb_control_check_server_id(ctdb, indata);
 422
 423         case CTDB_CONTROL_GET_SERVER_ID_LIST:
 424                 CHECK_CONTROL_DATA_SIZE(0);
 425                 return ctdb_control_get_server_id_list(ctdb, outdata);
 426
 427         case CTDB_CONTROL_PERSISTENT_STORE:
 428                 return control_not_implemented("PERSISTENT_STORE", NULL);
 429
 430         case CTDB_CONTROL_UPDATE_RECORD:
 431                 return ctdb_control_update_record(ctdb, c, indata, async_reply);
 432
 433         case CTDB_CONTROL_SEND_GRATIOUS_ARP:
 434                 return ctdb_control_send_gratious_arp(ctdb, indata);
 435
 436         case CTDB_CONTROL_TRANSACTION_START:
 437                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 438                 return ctdb_control_transaction_start(ctdb, *(uint32_t *)indata.dptr);
 439
 440         case CTDB_CONTROL_TRANSACTION_COMMIT:
 441                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 442                 return ctdb_control_transaction_commit(ctdb, *(uint32_t *)indata.dptr);
 443
 444         case CTDB_CONTROL_WIPE_DATABASE:
 445                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_transdb));
 446                 return ctdb_control_wipe_database(ctdb, indata);
 447
 448         case CTDB_CONTROL_UPTIME:
 449                 return ctdb_control_uptime(ctdb, outdata);
 450
 451         case CTDB_CONTROL_START_RECOVERY:
 452                 return ctdb_control_start_recovery(ctdb, c, async_reply);
 453
 454         case CTDB_CONTROL_END_RECOVERY:
 455                 return ctdb_control_end_recovery(ctdb, c, async_reply);
 456
 457         case CTDB_CONTROL_TRY_DELETE_RECORDS:
 458                 return ctdb_control_try_delete_records(ctdb, indata, outdata);
 459
 460         case CTDB_CONTROL_ADD_PUBLIC_IP:
 461                 return ctdb_control_add_public_address(ctdb, indata);
 462
 463         case CTDB_CONTROL_DEL_PUBLIC_IP:
 464                 return ctdb_control_del_public_address(ctdb, c, indata,
 465                                                        async_reply);
 466
 467         case CTDB_CONTROL_GET_CAPABILITIES:
 468                 return ctdb_control_get_capabilities(ctdb, outdata);
 469
 470         case CTDB_CONTROL_START_PERSISTENT_UPDATE:
 471                 return ctdb_control_start_persistent_update(ctdb, c, indata);
 472
 473         case CTDB_CONTROL_CANCEL_PERSISTENT_UPDATE:
 474                 return ctdb_control_cancel_persistent_update(ctdb, c, indata);
 475
 476         case CTDB_CONTROL_TRANS2_COMMIT:
 477         case CTDB_CONTROL_TRANS2_COMMIT_RETRY:
 478                 return control_not_implemented("TRANS2_COMMIT", "TRANS3_COMMIT");
 479
 480         case CTDB_CONTROL_TRANS2_ERROR:
 481                 return control_not_implemented("TRANS2_ERROR", NULL);
 482
 483         case CTDB_CONTROL_TRANS2_FINISHED:
 484                 return control_not_implemented("TRANS2_FINISHED", NULL);
 485
 486         case CTDB_CONTROL_TRANS2_ACTIVE:
 487                 return control_not_implemented("TRANS2_ACTIVE", NULL);
 488
 489         case CTDB_CONTROL_TRANS3_COMMIT:
 490                 return ctdb_control_trans3_commit(ctdb, c, indata, async_reply);
 491
 492         case CTDB_CONTROL_RECD_PING:
 493                 CHECK_CONTROL_DATA_SIZE(0);
 494                 return ctdb_control_recd_ping(ctdb);
 495
 496         case CTDB_CONTROL_GET_EVENT_SCRIPT_STATUS:
 497                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 498                 return ctdb_control_get_event_script_status(ctdb, *(uint32_t *)indata.dptr, outdata);
 499
 500         case CTDB_CONTROL_RECD_RECLOCK_LATENCY:
 501                 CHECK_CONTROL_DATA_SIZE(sizeof(double));
 502                 CTDB_UPDATE_RECLOCK_LATENCY(ctdb, "recd reclock", reclock.recd, *((double *)indata.dptr));
 503                 return 0;
 504         case CTDB_CONTROL_GET_RECLOCK_FILE:
 505                 CHECK_CONTROL_DATA_SIZE(0);
 506                 if (ctdb->recovery_lock_file != NULL) {
 507                         outdata->dptr  = discard_const(ctdb->recovery_lock_file);
 508                         outdata->dsize = strlen(ctdb->recovery_lock_file) + 1;
 509                 }
 510                 return 0;
 511         case CTDB_CONTROL_SET_RECLOCK_FILE: {
 512                 char *t;
 513
 514                 if (indata.dsize == 0) {
 515                         TALLOC_FREE(ctdb->recovery_lock_file);
 516                         return 0;
 517                 }
 518
 519                 /* Return silent success if unchanged.  Recovery
 520                  * master updates all nodes on each recovery - we
 521                  * don't need the extra memory allocation or log
 522                  * message each time. */
 523                 if (ctdb->recovery_lock_file != NULL &&
 524                     strcmp(discard_const(indata.dptr),
 525                            ctdb->recovery_lock_file) == 0) {
 526                         return 0;
 527                 }
 528
 529                 t = talloc_strdup(ctdb, discard_const(indata.dptr));
 530                 if (t == NULL) {
 531                         DEBUG(DEBUG_ERR, ("Out of memory in SET_RECLOCK_FILE\n"));
 532                         return -1;
 533                 }
 534
 535                 talloc_free(ctdb->recovery_lock_file);
 536                 ctdb->recovery_lock_file = t;
 537                 DEBUG(DEBUG_NOTICE, ("Updated recovery lock file to %s\n", t));
 538
 539                 return 0;
 540         }
 541
 542         case CTDB_CONTROL_STOP_NODE:
 543                 CHECK_CONTROL_DATA_SIZE(0);
 544                 return ctdb_control_stop_node(ctdb);
 545
 546         case CTDB_CONTROL_CONTINUE_NODE:
 547                 CHECK_CONTROL_DATA_SIZE(0);
 548                 return ctdb_control_continue_node(ctdb);
 549
 550         case CTDB_CONTROL_SET_NATGWSTATE: {
 551                 uint32_t natgwstate;
 552
 553                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 554                 natgwstate = *(uint32_t *)indata.dptr;
 555                 if (natgwstate == 0) {
 556                         ctdb->capabilities &= ~CTDB_CAP_NATGW;
 557                 } else {
 558                         ctdb->capabilities |= CTDB_CAP_NATGW;
 559                 }
 560                 return 0;
 561         }
 562
 563         case CTDB_CONTROL_SET_LMASTERROLE: {
 564                 uint32_t lmasterrole;
 565
 566                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 567                 lmasterrole = *(uint32_t *)indata.dptr;
 568                 if (lmasterrole == 0) {
 569                         ctdb->capabilities &= ~CTDB_CAP_LMASTER;
 570                 } else {
 571                         ctdb->capabilities |= CTDB_CAP_LMASTER;
 572                 }
 573                 return 0;
 574         }
 575
 576         case CTDB_CONTROL_SET_RECMASTERROLE: {
 577                 uint32_t recmasterrole;
 578
 579                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 580                 recmasterrole = *(uint32_t *)indata.dptr;
 581                 if (recmasterrole == 0) {
 582                         ctdb->capabilities &= ~CTDB_CAP_RECMASTER;
 583                 } else {
 584                         ctdb->capabilities |= CTDB_CAP_RECMASTER;
 585                 }
 586                 return 0;
 587         }
 588
 589         case CTDB_CONTROL_ENABLE_SCRIPT:
 590                 return ctdb_control_enable_script(ctdb, indata);
 591
 592         case CTDB_CONTROL_DISABLE_SCRIPT:
 593                 return ctdb_control_disable_script(ctdb, indata);
 594
 595         case CTDB_CONTROL_SET_BAN_STATE:
 596                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_ban_time));
 597                 return ctdb_control_set_ban_state(ctdb, indata);
 598
 599         case CTDB_CONTROL_GET_BAN_STATE:
 600                 CHECK_CONTROL_DATA_SIZE(0);
 601                 return ctdb_control_get_ban_state(ctdb, outdata);
 602
 603         case CTDB_CONTROL_SET_DB_PRIORITY:
 604                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_db_priority));
 605                 return ctdb_control_set_db_priority(ctdb, indata, client_id);
 606
 607         case CTDB_CONTROL_GET_DB_PRIORITY: {
 608                 uint32_t db_id;
 609                 struct ctdb_db_context *ctdb_db;
 610
 611                 CHECK_CONTROL_DATA_SIZE(sizeof(db_id));
 612                 db_id = *(uint32_t *)indata.dptr;
 613                 ctdb_db = find_ctdb_db(ctdb, db_id);
 614                 if (ctdb_db == NULL) return -1;
 615                 return ctdb_db->priority;
 616         }
 617
 618         case CTDB_CONTROL_TRANSACTION_CANCEL:
 619                 CHECK_CONTROL_DATA_SIZE(0);
 620                 return ctdb_control_transaction_cancel(ctdb);
 621
 622         case CTDB_CONTROL_REGISTER_NOTIFY:
 623                 return ctdb_control_register_notify(ctdb, client_id, indata);
 624
 625         case CTDB_CONTROL_DEREGISTER_NOTIFY:
 626                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_client_notify_deregister));
 627                 return ctdb_control_deregister_notify(ctdb, client_id, indata);
 628
 629         case CTDB_CONTROL_GET_LOG:
 630                 return control_not_implemented("GET_LOG", NULL);
 631
 632         case CTDB_CONTROL_CLEAR_LOG:
 633                 return control_not_implemented("CLEAR_LOG", NULL);
 634
 635         case CTDB_CONTROL_GET_DB_SEQNUM:
 636                 CHECK_CONTROL_DATA_SIZE(sizeof(uint64_t));
 637                 return ctdb_control_get_db_seqnum(ctdb, indata, outdata);
 638
 639         case CTDB_CONTROL_DB_SET_HEALTHY:
 640                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 641                 return ctdb_control_db_set_healthy(ctdb, indata);
 642
 643         case CTDB_CONTROL_DB_GET_HEALTH:
 644                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 645                 return ctdb_control_db_get_health(ctdb, indata, outdata);
 646
 647         case CTDB_CONTROL_GET_PUBLIC_IP_INFO:
 648                 CHECK_CONTROL_DATA_SIZE(sizeof(ctdb_sock_addr));
 649                 return ctdb_control_get_public_ip_info(ctdb, c, indata, outdata);
 650
 651         case CTDB_CONTROL_GET_IFACES:
 652                 CHECK_CONTROL_DATA_SIZE(0);
 653                 return ctdb_control_get_ifaces(ctdb, c, outdata);
 654
 655         case CTDB_CONTROL_SET_IFACE_LINK_STATE:
 656                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_iface_info));
 657                 return ctdb_control_set_iface_link(ctdb, c, indata);
 658
 659         case CTDB_CONTROL_GET_STAT_HISTORY:
 660                 CHECK_CONTROL_DATA_SIZE(0);
 661                 return ctdb_control_get_stat_history(ctdb, c, outdata);
 662
 663         case CTDB_CONTROL_SCHEDULE_FOR_DELETION: {
 664                 struct ctdb_control_schedule_for_deletion *d;
 665                 size_t size = offsetof(struct ctdb_control_schedule_for_deletion, key);
 666                 CHECK_CONTROL_MIN_DATA_SIZE(size);
 667                 d = (struct ctdb_control_schedule_for_deletion *)indata.dptr;
 668                 size += d->keylen;
 669                 CHECK_CONTROL_DATA_SIZE(size);
 670                 return ctdb_control_schedule_for_deletion(ctdb, indata);
 671         }
 672         case CTDB_CONTROL_GET_DB_STATISTICS:
 673                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 674                 return ctdb_control_get_db_statistics(ctdb, *(uint32_t *)indata.dptr, outdata);
 675
 676         case CTDB_CONTROL_RELOAD_PUBLIC_IPS:
 677                 CHECK_CONTROL_DATA_SIZE(0);
 678                 return ctdb_control_reload_public_ips(ctdb, c, async_reply);
 679
 680         case CTDB_CONTROL_RECEIVE_RECORDS:
 681                 return ctdb_control_receive_records(ctdb, indata, outdata);
 682
 683         case CTDB_CONTROL_DB_DETACH:
 684                 return ctdb_control_db_detach(ctdb, indata, client_id);
 685
 686         case CTDB_CONTROL_DB_FREEZE:
 687                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 688                 return ctdb_control_db_freeze(ctdb, c, *(uint32_t *)indata.dptr,
 689                                               async_reply);
 690
 691         case CTDB_CONTROL_DB_THAW:
 692                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 693                 return ctdb_control_db_thaw(ctdb, *(uint32_t *)indata.dptr);
 694
 695         case CTDB_CONTROL_DB_TRANSACTION_START:
 696                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_transdb));
 697                 return ctdb_control_db_transaction_start(ctdb, indata);
 698
 699         case CTDB_CONTROL_DB_TRANSACTION_COMMIT:
 700                 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_control_transdb));
 701                 return ctdb_control_db_transaction_commit(ctdb, indata);
 702
 703         case CTDB_CONTROL_DB_TRANSACTION_CANCEL:
 704                 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
 705                 return ctdb_control_db_transaction_cancel(ctdb, indata);
 706
 707         default:
 708                 DEBUG(DEBUG_CRIT,(__location__ " Unknown CTDB control opcode %u\n", opcode));
 709                 return -1;
 710         }
 711 }
 712
 713 /*
 714   send a reply for a ctdb control
 715  */
 716 void ctdb_request_control_reply(struct ctdb_context *ctdb, struct ctdb_req_control *c,
 717                                 TDB_DATA *outdata, int32_t status, const char *errormsg)
 718 {
 719         struct ctdb_reply_control *r;
 720         size_t len;
 721
 722         /* some controls send no reply */
 723         if (c->flags & CTDB_CTRL_FLAG_NOREPLY) {
 724                 return;
 725         }
 726
 727         len = offsetof(struct ctdb_reply_control, data) + (outdata?outdata->dsize:0);
 728         if (errormsg) {
 729                 len += strlen(errormsg);
 730         }
 731         r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REPLY_CONTROL, len, struct ctdb_reply_control);
 732         if (r == NULL) {
 733                 DEBUG(DEBUG_ERR,(__location__ "Unable to allocate transport - OOM or transport is down\n"));
 734                 return;
 735         }
 736
 737         r->hdr.destnode     = c->hdr.srcnode;
 738         r->hdr.reqid        = c->hdr.reqid;
 739         r->status           = status;
 740         r->datalen          = outdata?outdata->dsize:0;
 741         if (outdata && outdata->dsize) {
 742                 memcpy(&r->data[0], outdata->dptr, outdata->dsize);
 743         }
 744         if (errormsg) {
 745                 r->errorlen = strlen(errormsg);
 746                 memcpy(&r->data[r->datalen], errormsg, r->errorlen);
 747         }
 748
 749         ctdb_queue_packet_opcode(ctdb, &r->hdr, c->opcode);
 750
 751         talloc_free(r);
 752 }
 753
 754 /*
 755   called when a CTDB_REQ_CONTROL packet comes in
 756 */
 757 void ctdb_request_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 758 {
 759         struct ctdb_req_control *c = (struct ctdb_req_control *)hdr;
 760         TDB_DATA data, *outdata;
 761         int32_t status;
 762         bool async_reply = false;
 763         const char *errormsg = NULL;
 764
 765         data.dptr = &c->data[0];
 766         data.dsize = c->datalen;
 767
 768         outdata = talloc_zero(c, TDB_DATA);
 769
 770         status = ctdb_control_dispatch(ctdb, c, data, outdata, hdr->srcnode,
 771                                        &errormsg, &async_reply);
 772
 773         if (!async_reply) {
 774                 ctdb_request_control_reply(ctdb, c, outdata, status, errormsg);
 775         }
 776 }
 777
 778 /*
 779   called when a CTDB_REPLY_CONTROL packet comes in
 780 */
 781 void ctdb_reply_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
 782 {
 783         struct ctdb_reply_control *c = (struct ctdb_reply_control *)hdr;
 784         TDB_DATA data;
 785         struct ctdb_control_state *state;
 786         const char *errormsg = NULL;
 787
 788         state = ctdb_reqid_find(ctdb, hdr->reqid, struct ctdb_control_state);
 789         if (state == NULL) {
 790                 DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_reply_control\n",
 791                          ctdb->pnn, hdr->reqid));
 792                 return;
 793         }
 794
 795         if (hdr->reqid != state->reqid) {
 796                 /* we found a record  but it was the wrong one */
 797                 DEBUG(DEBUG_ERR, ("Dropped orphaned control reply with reqid:%u\n", hdr->reqid));
 798                 return;
 799         }
 800
 801         data.dptr = &c->data[0];
 802         data.dsize = c->datalen;
 803         if (c->errorlen) {
 804                 errormsg = talloc_strndup(state,
 805                                           (char *)&c->data[c->datalen], c->errorlen);
 806         }
 807
 808         /* make state a child of the packet, so it goes away when the packet
 809            is freed. */
 810         talloc_steal(hdr, state);
 811
 812         state->callback(ctdb, c->status, data, errormsg, state->private_data);
 813 }
 814
 815 static int ctdb_control_destructor(struct ctdb_control_state *state)
 816 {
 817         ctdb_reqid_remove(state->ctdb, state->reqid);
 818         return 0;
 819 }
 820
 821 /*
 822   handle a timeout of a control
 823  */
 824 static void ctdb_control_timeout(struct event_context *ev, struct timed_event *te,
 825                        struct timeval t, void *private_data)
 826 {
 827         struct ctdb_control_state *state = talloc_get_type(private_data, struct ctdb_control_state);
 828         TALLOC_CTX *tmp_ctx = talloc_new(ev);
 829
 830         CTDB_INCREMENT_STAT(state->ctdb, timeouts.control);
 831
 832         talloc_steal(tmp_ctx, state);
 833
 834         state->callback(state->ctdb, -1, tdb_null,
 835                         "ctdb_control timed out",
 836                         state->private_data);
 837         talloc_free(tmp_ctx);
 838 }
 839
 840
 841 /*
 842   send a control message to a node
 843  */
 844 int ctdb_daemon_send_control(struct ctdb_context *ctdb, uint32_t destnode,
 845                              uint64_t srvid, uint32_t opcode, uint32_t client_id,
 846                              uint32_t flags,
 847                              TDB_DATA data,
 848                              ctdb_control_callback_fn_t callback,
 849                              void *private_data)
 850 {
 851         struct ctdb_req_control *c;
 852         struct ctdb_control_state *state;
 853         size_t len;
 854
 855         if (ctdb->methods == NULL) {
 856                 DEBUG(DEBUG_INFO,(__location__ " Failed to send control. Transport is DOWN\n"));
 857                 return -1;
 858         }
 859
 860         if (((destnode == CTDB_BROADCAST_VNNMAP) ||
 861              (destnode == CTDB_BROADCAST_ALL) ||
 862              (destnode == CTDB_BROADCAST_CONNECTED)) &&
 863             !(flags & CTDB_CTRL_FLAG_NOREPLY)) {
 864                 DEBUG(DEBUG_CRIT,("Attempt to broadcast control without NOREPLY\n"));
 865                 return -1;
 866         }
 867
 868         if (destnode != CTDB_BROADCAST_VNNMAP &&
 869             destnode != CTDB_BROADCAST_ALL &&
 870             destnode != CTDB_BROADCAST_CONNECTED &&
 871             (!ctdb_validate_pnn(ctdb, destnode) ||
 872              (ctdb->nodes[destnode]->flags & NODE_FLAGS_DISCONNECTED))) {
 873                 if (!(flags & CTDB_CTRL_FLAG_NOREPLY)) {
 874                         callback(ctdb, -1, tdb_null, "ctdb_control to disconnected node", private_data);
 875                 }
 876                 return 0;
 877         }
 878
 879         /* the state is made a child of private_data if possible. This means any reply
 880            will be discarded if the private_data goes away */
 881         state = talloc(private_data?private_data:ctdb, struct ctdb_control_state);
 882         CTDB_NO_MEMORY(ctdb, state);
 883
 884         state->reqid = ctdb_reqid_new(ctdb, state);
 885         state->callback = callback;
 886         state->private_data = private_data;
 887         state->ctdb = ctdb;
 888         state->flags = flags;
 889
 890         talloc_set_destructor(state, ctdb_control_destructor);
 891
 892         len = offsetof(struct ctdb_req_control, data) + data.dsize;
 893         c = ctdb_transport_allocate(ctdb, state, CTDB_REQ_CONTROL, len,
 894                                     struct ctdb_req_control);
 895         CTDB_NO_MEMORY(ctdb, c);
 896         talloc_set_name_const(c, "ctdb_req_control packet");
 897
 898         c->hdr.destnode     = destnode;
 899         c->hdr.reqid        = state->reqid;
 900         c->opcode           = opcode;
 901         c->client_id        = client_id;
 902         c->flags            = flags;
 903         c->srvid            = srvid;
 904         c->datalen          = data.dsize;
 905         if (data.dsize) {
 906                 memcpy(&c->data[0], data.dptr, data.dsize);
 907         }
 908
 909         ctdb_queue_packet(ctdb, &c->hdr);
 910
 911         if (flags & CTDB_CTRL_FLAG_NOREPLY) {
 912                 talloc_free(state);
 913                 return 0;
 914         }
 915
 916         if (ctdb->tunable.control_timeout) {
 917                 event_add_timed(ctdb->ev, state,
 918                                 timeval_current_ofs(ctdb->tunable.control_timeout, 0),
 919                                 ctdb_control_timeout, state);
 920         }
 921
 922         talloc_free(c);
 923         return 0;
 924 }