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