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