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