source: vendor/current/ctdb/server/ctdb_control.c

Last change on this file was 989, checked in by Silvan Scherrer, 9 years ago

Samba Server: update vendor to version 4.4.7
File size: 27.4 KB
Line 
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
36#include "common/reqid.h"
37#include "common/common.h"
38#include "common/logging.h"
39
40
41struct 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 */
53int32_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
75static 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 */
93static 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_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_GRATUITOUS_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_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_state));
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_iface));
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_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_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 case CTDB_CONTROL_DB_PULL:
719 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_pulldb_ext));
720 return ctdb_control_db_pull(ctdb, c, indata, outdata);
721
722 case CTDB_CONTROL_DB_PUSH_START:
723 CHECK_CONTROL_DATA_SIZE(sizeof(struct ctdb_pulldb_ext));
724 return ctdb_control_db_push_start(ctdb, indata);
725
726 case CTDB_CONTROL_DB_PUSH_CONFIRM:
727 CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
728 return ctdb_control_db_push_confirm(ctdb, indata, outdata);
729
730 default:
731 DEBUG(DEBUG_CRIT,(__location__ " Unknown CTDB control opcode %u\n", opcode));
732 return -1;
733 }
734}
735
736/*
737 send a reply for a ctdb control
738 */
739void ctdb_request_control_reply(struct ctdb_context *ctdb, struct ctdb_req_control_old *c,
740 TDB_DATA *outdata, int32_t status, const char *errormsg)
741{
742 struct ctdb_reply_control_old *r;
743 size_t len;
744
745 /* some controls send no reply */
746 if (c->flags & CTDB_CTRL_FLAG_NOREPLY) {
747 return;
748 }
749
750 len = offsetof(struct ctdb_reply_control_old, data) + (outdata?outdata->dsize:0);
751 if (errormsg) {
752 len += strlen(errormsg);
753 }
754 r = ctdb_transport_allocate(ctdb, ctdb, CTDB_REPLY_CONTROL, len, struct ctdb_reply_control_old);
755 if (r == NULL) {
756 DEBUG(DEBUG_ERR,(__location__ "Unable to allocate transport - OOM or transport is down\n"));
757 return;
758 }
759
760 r->hdr.destnode = c->hdr.srcnode;
761 r->hdr.reqid = c->hdr.reqid;
762 r->status = status;
763 r->datalen = outdata?outdata->dsize:0;
764 if (outdata && outdata->dsize) {
765 memcpy(&r->data[0], outdata->dptr, outdata->dsize);
766 }
767 if (errormsg) {
768 r->errorlen = strlen(errormsg);
769 memcpy(&r->data[r->datalen], errormsg, r->errorlen);
770 }
771
772 ctdb_queue_packet_opcode(ctdb, &r->hdr, c->opcode);
773
774 talloc_free(r);
775}
776
777/*
778 called when a CTDB_REQ_CONTROL packet comes in
779*/
780void ctdb_request_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
781{
782 struct ctdb_req_control_old *c = (struct ctdb_req_control_old *)hdr;
783 TDB_DATA data, *outdata;
784 int32_t status;
785 bool async_reply = false;
786 const char *errormsg = NULL;
787
788 data.dptr = &c->data[0];
789 data.dsize = c->datalen;
790
791 outdata = talloc_zero(c, TDB_DATA);
792
793 status = ctdb_control_dispatch(ctdb, c, data, outdata, hdr->srcnode,
794 &errormsg, &async_reply);
795
796 if (!async_reply) {
797 ctdb_request_control_reply(ctdb, c, outdata, status, errormsg);
798 }
799}
800
801/*
802 called when a CTDB_REPLY_CONTROL packet comes in
803*/
804void ctdb_reply_control(struct ctdb_context *ctdb, struct ctdb_req_header *hdr)
805{
806 struct ctdb_reply_control_old *c = (struct ctdb_reply_control_old *)hdr;
807 TDB_DATA data;
808 struct ctdb_control_state *state;
809 const char *errormsg = NULL;
810
811 state = reqid_find(ctdb->idr, hdr->reqid, struct ctdb_control_state);
812 if (state == NULL) {
813 DEBUG(DEBUG_ERR,("pnn %u Invalid reqid %u in ctdb_reply_control\n",
814 ctdb->pnn, hdr->reqid));
815 return;
816 }
817
818 if (hdr->reqid != state->reqid) {
819 /* we found a record but it was the wrong one */
820 DEBUG(DEBUG_ERR, ("Dropped orphaned control reply with reqid:%u\n", hdr->reqid));
821 return;
822 }
823
824 data.dptr = &c->data[0];
825 data.dsize = c->datalen;
826 if (c->errorlen) {
827 errormsg = talloc_strndup(state,
828 (char *)&c->data[c->datalen], c->errorlen);
829 }
830
831 /* make state a child of the packet, so it goes away when the packet
832 is freed. */
833 talloc_steal(hdr, state);
834
835 state->callback(ctdb, c->status, data, errormsg, state->private_data);
836}
837
838static int ctdb_control_destructor(struct ctdb_control_state *state)
839{
840 reqid_remove(state->ctdb->idr, state->reqid);
841 return 0;
842}
843
844/*
845 handle a timeout of a control
846 */
847static void ctdb_control_timeout(struct tevent_context *ev,
848 struct tevent_timer *te,
849 struct timeval t, void *private_data)
850{
851 struct ctdb_control_state *state = talloc_get_type(private_data, struct ctdb_control_state);
852 TALLOC_CTX *tmp_ctx = talloc_new(ev);
853
854 CTDB_INCREMENT_STAT(state->ctdb, timeouts.control);
855
856 talloc_steal(tmp_ctx, state);
857
858 state->callback(state->ctdb, -1, tdb_null,
859 "ctdb_control timed out",
860 state->private_data);
861 talloc_free(tmp_ctx);
862}
863
864
865/*
866 send a control message to a node
867 */
868int ctdb_daemon_send_control(struct ctdb_context *ctdb, uint32_t destnode,
869 uint64_t srvid, uint32_t opcode, uint32_t client_id,
870 uint32_t flags,
871 TDB_DATA data,
872 ctdb_control_callback_fn_t callback,
873 void *private_data)
874{
875 struct ctdb_req_control_old *c;
876 struct ctdb_control_state *state;
877 size_t len;
878
879 if (ctdb->methods == NULL) {
880 DEBUG(DEBUG_INFO,(__location__ " Failed to send control. Transport is DOWN\n"));
881 return -1;
882 }
883
884 if (((destnode == CTDB_BROADCAST_VNNMAP) ||
885 (destnode == CTDB_BROADCAST_ALL) ||
886 (destnode == CTDB_BROADCAST_CONNECTED)) &&
887 !(flags & CTDB_CTRL_FLAG_NOREPLY)) {
888 DEBUG(DEBUG_CRIT,("Attempt to broadcast control without NOREPLY\n"));
889 return -1;
890 }
891
892 if (destnode != CTDB_BROADCAST_VNNMAP &&
893 destnode != CTDB_BROADCAST_ALL &&
894 destnode != CTDB_BROADCAST_CONNECTED &&
895 (!ctdb_validate_pnn(ctdb, destnode) ||
896 (ctdb->nodes[destnode]->flags & NODE_FLAGS_DISCONNECTED))) {
897 if (!(flags & CTDB_CTRL_FLAG_NOREPLY)) {
898 callback(ctdb, -1, tdb_null, "ctdb_control to disconnected node", private_data);
899 }
900 return 0;
901 }
902
903 /* the state is made a child of private_data if possible. This means any reply
904 will be discarded if the private_data goes away */
905 state = talloc(private_data?private_data:ctdb, struct ctdb_control_state);
906 CTDB_NO_MEMORY(ctdb, state);
907
908 state->reqid = reqid_new(ctdb->idr, state);
909 state->callback = callback;
910 state->private_data = private_data;
911 state->ctdb = ctdb;
912 state->flags = flags;
913
914 talloc_set_destructor(state, ctdb_control_destructor);
915
916 len = offsetof(struct ctdb_req_control_old, data) + data.dsize;
917 c = ctdb_transport_allocate(ctdb, state, CTDB_REQ_CONTROL, len,
918 struct ctdb_req_control_old);
919 CTDB_NO_MEMORY(ctdb, c);
920 talloc_set_name_const(c, "ctdb_req_control packet");
921
922 c->hdr.destnode = destnode;
923 c->hdr.reqid = state->reqid;
924 c->opcode = opcode;
925 c->client_id = client_id;
926 c->flags = flags;
927 c->srvid = srvid;
928 c->datalen = data.dsize;
929 if (data.dsize) {
930 memcpy(&c->data[0], data.dptr, data.dsize);
931 }
932
933 ctdb_queue_packet(ctdb, &c->hdr);
934
935 if (flags & CTDB_CTRL_FLAG_NOREPLY) {
936 talloc_free(state);
937 return 0;
938 }
939
940 if (ctdb->tunable.control_timeout) {
941 tevent_add_timer(ctdb->ev, state,
942 timeval_current_ofs(ctdb->tunable.control_timeout, 0),
943 ctdb_control_timeout, state);
944 }
945
946 talloc_free(c);
947 return 0;
948}
Note: See TracBrowser for help on using the repository browser.