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