?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
ctdb-vacuum: Track time for vacuuming in database statistics
[Samba.git] / ctdb / tools / ctdb.c
blobf0c87d91e5b6b7a163e4f60b92bc15a5c48c1971
1 /*
2 ctdb control tool
3
4 Copyright (C) Andrew Tridgell 2007
5 Copyright (C) Ronnie Sahlberg 2007
6
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
11
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, see <http://www.gnu.org/licenses/>.
19 */
20
21 #include "includes.h"
22 #include "system/time.h"
23 #include "system/filesys.h"
24 #include "system/network.h"
25 #include "system/locale.h"
26 #include "popt.h"
27 #include "cmdline.h"
28 #include "../include/ctdb_version.h"
29 #include "../include/ctdb_client.h"
30 #include "../include/ctdb_private.h"
31 #include "../common/rb_tree.h"
32 #include "lib/tdb_wrap/tdb_wrap.h"
33 #include "lib/util/dlinklist.h"
34
35 #define ERR_TIMEOUT 20 /* timed out trying to reach node */
36 #define ERR_NONODE 21 /* node does not exist */
37 #define ERR_DISNODE 22 /* node is disconnected */
38
39 static void usage(void);
40
41 static struct {
42 int timelimit;
43 uint32_t pnn;
44 uint32_t *nodes;
45 int machinereadable;
46 int verbose;
47 int maxruntime;
48 int printemptyrecords;
49 int printdatasize;
50 int printlmaster;
51 int printhash;
52 int printrecordflags;
53 } options;
54
55 #define LONGTIMEOUT options.timelimit*10
56
57 #define TIMELIMIT() timeval_current_ofs(options.timelimit, 0)
58 #define LONGTIMELIMIT() timeval_current_ofs(LONGTIMEOUT, 0)
59
60 static double timeval_delta(struct timeval *tv2, struct timeval *tv)
61 {
62 return (tv2->tv_sec - tv->tv_sec) +
63 (tv2->tv_usec - tv->tv_usec)*1.0e-6;
64 }
65
66 static int control_version(struct ctdb_context *ctdb, int argc, const char **argv)
67 {
68 printf("CTDB version: %s\n", CTDB_VERSION_STRING);
69 return 0;
70 }
71
72 #define CTDB_NOMEM_ABORT(p) do { if (!(p)) { \
73 DEBUG(DEBUG_ALERT,("ctdb fatal error: %s\n", \
74 "Out of memory in " __location__ )); \
75 abort(); \
76 }} while (0)
77
78 static uint32_t getpnn(struct ctdb_context *ctdb)
79 {
80 if ((options.pnn == CTDB_BROADCAST_ALL) ||
81 (options.pnn == CTDB_MULTICAST)) {
82 DEBUG(DEBUG_ERR,
83 ("Cannot get PNN for node %u\n", options.pnn));
84 exit(1);
85 }
86
87 if (options.pnn == CTDB_CURRENT_NODE) {
88 return ctdb_get_pnn(ctdb);
89 } else {
90 return options.pnn;
91 }
92 }
93
94 static void assert_single_node_only(void)
95 {
96 if ((options.pnn == CTDB_BROADCAST_ALL) ||
97 (options.pnn == CTDB_MULTICAST)) {
98 DEBUG(DEBUG_ERR,
99 ("This control can not be applied to multiple PNNs\n"));
100 exit(1);
101 }
102 }
103
104 /* Pretty print the flags to a static buffer in human-readable format.
105 * This never returns NULL!
106 */
107 static const char *pretty_print_flags(uint32_t flags)
108 {
109 int j;
110 static const struct {
111 uint32_t flag;
112 const char *name;
113 } flag_names[] = {
114 { NODE_FLAGS_DISCONNECTED, "DISCONNECTED" },
115 { NODE_FLAGS_PERMANENTLY_DISABLED, "DISABLED" },
116 { NODE_FLAGS_BANNED, "BANNED" },
117 { NODE_FLAGS_UNHEALTHY, "UNHEALTHY" },
118 { NODE_FLAGS_DELETED, "DELETED" },
119 { NODE_FLAGS_STOPPED, "STOPPED" },
120 { NODE_FLAGS_INACTIVE, "INACTIVE" },
121 };
122 static char flags_str[512]; /* Big enough to contain all flag names */
123
124 flags_str[0] = '\0';
125 for (j=0;j<ARRAY_SIZE(flag_names);j++) {
126 if (flags & flag_names[j].flag) {
127 if (flags_str[0] == '\0') {
128 (void) strcpy(flags_str, flag_names[j].name);
129 } else {
130 (void) strncat(flags_str, "|", sizeof(flags_str)-1);
131 (void) strncat(flags_str, flag_names[j].name,
132 sizeof(flags_str)-1);
133 }
134 }
135 }
136 if (flags_str[0] == '\0') {
137 (void) strcpy(flags_str, "OK");
138 }
139
140 return flags_str;
141 }
142
143 static int h2i(char h)
144 {
145 if (h >= 'a' && h <= 'f') return h - 'a' + 10;
146 if (h >= 'A' && h <= 'F') return h - 'f' + 10;
147 return h - '0';
148 }
149
150 static TDB_DATA hextodata(TALLOC_CTX *mem_ctx, const char *str)
151 {
152 int i, len;
153 TDB_DATA key = {NULL, 0};
154
155 len = strlen(str);
156 if (len & 0x01) {
157 DEBUG(DEBUG_ERR,("Key specified with odd number of hexadecimal digits\n"));
158 return key;
159 }
160
161 key.dsize = len>>1;
162 key.dptr = talloc_size(mem_ctx, key.dsize);
163
164 for (i=0; i < len/2; i++) {
165 key.dptr[i] = h2i(str[i*2]) << 4 | h2i(str[i*2+1]);
166 }
167 return key;
168 }
169
170 /* Parse a nodestring. Parameter dd_ok controls what happens to nodes
171 * that are disconnected or deleted. If dd_ok is true those nodes are
172 * included in the output list of nodes. If dd_ok is false, those
173 * nodes are filtered from the "all" case and cause an error if
174 * explicitly specified.
175 */
176 static bool parse_nodestring(struct ctdb_context *ctdb,
177 TALLOC_CTX *mem_ctx,
178 const char * nodestring,
179 uint32_t current_pnn,
180 bool dd_ok,
181 uint32_t **nodes,
182 uint32_t *pnn_mode)
183 {
184 TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
185 int n;
186 uint32_t i;
187 struct ctdb_node_map *nodemap;
188 int ret;
189
190 *nodes = NULL;
191
192 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
193 if (ret != 0) {
194 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
195 talloc_free(tmp_ctx);
196 exit(10);
197 }
198
199 if (nodestring != NULL) {
200 *nodes = talloc_array(mem_ctx, uint32_t, 0);
201 if (*nodes == NULL) {
202 goto failed;
203 }
204
205 n = 0;
206
207 if (strcmp(nodestring, "all") == 0) {
208 *pnn_mode = CTDB_BROADCAST_ALL;
209
210 /* all */
211 for (i = 0; i < nodemap->num; i++) {
212 if ((nodemap->nodes[i].flags &
213 (NODE_FLAGS_DISCONNECTED |
214 NODE_FLAGS_DELETED)) && !dd_ok) {
215 continue;
216 }
217 *nodes = talloc_realloc(mem_ctx, *nodes,
218 uint32_t, n+1);
219 if (*nodes == NULL) {
220 goto failed;
221 }
222 (*nodes)[n] = i;
223 n++;
224 }
225 } else {
226 /* x{,y...} */
227 char *ns, *tok;
228
229 ns = talloc_strdup(tmp_ctx, nodestring);
230 tok = strtok(ns, ",");
231 while (tok != NULL) {
232 uint32_t pnn;
233 char *endptr;
234 i = (uint32_t)strtoul(tok, &endptr, 0);
235 if (i == 0 && tok == endptr) {
236 DEBUG(DEBUG_ERR,
237 ("Invalid node %s\n", tok));
238 talloc_free(tmp_ctx);
239 exit(ERR_NONODE);
240 }
241 if (i >= nodemap->num) {
242 DEBUG(DEBUG_ERR, ("Node %u does not exist\n", i));
243 talloc_free(tmp_ctx);
244 exit(ERR_NONODE);
245 }
246 if ((nodemap->nodes[i].flags &
247 (NODE_FLAGS_DISCONNECTED |
248 NODE_FLAGS_DELETED)) && !dd_ok) {
249 DEBUG(DEBUG_ERR, ("Node %u has status %s\n", i, pretty_print_flags(nodemap->nodes[i].flags)));
250 talloc_free(tmp_ctx);
251 exit(ERR_DISNODE);
252 }
253 if ((pnn = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), i)) < 0) {
254 DEBUG(DEBUG_ERR, ("Can not access node %u. Node is not operational.\n", i));
255 talloc_free(tmp_ctx);
256 exit(10);
257 }
258
259 *nodes = talloc_realloc(mem_ctx, *nodes,
260 uint32_t, n+1);
261 if (*nodes == NULL) {
262 goto failed;
263 }
264
265 (*nodes)[n] = i;
266 n++;
267
268 tok = strtok(NULL, ",");
269 }
270 talloc_free(ns);
271
272 if (n == 1) {
273 *pnn_mode = (*nodes)[0];
274 } else {
275 *pnn_mode = CTDB_MULTICAST;
276 }
277 }
278 } else {
279 /* default - no nodes specified */
280 *nodes = talloc_array(mem_ctx, uint32_t, 1);
281 if (*nodes == NULL) {
282 goto failed;
283 }
284 *pnn_mode = CTDB_CURRENT_NODE;
285
286 if (((*nodes)[0] = ctdb_ctrl_getpnn(ctdb, TIMELIMIT(), current_pnn)) < 0) {
287 goto failed;
288 }
289 }
290
291 talloc_free(tmp_ctx);
292 return true;
293
294 failed:
295 talloc_free(tmp_ctx);
296 return false;
297 }
298
299 /*
300 check if a database exists
301 */
302 static bool db_exists(struct ctdb_context *ctdb, const char *dbarg,
303 uint32_t *dbid, const char **dbname, uint8_t *flags)
304 {
305 int i, ret;
306 struct ctdb_dbid_map *dbmap=NULL;
307 bool dbid_given = false, found = false;
308 uint32_t id;
309 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
310 const char *name;
311
312 ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &dbmap);
313 if (ret != 0) {
314 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n", options.pnn));
315 goto fail;
316 }
317
318 if (strncmp(dbarg, "0x", 2) == 0) {
319 id = strtoul(dbarg, NULL, 0);
320 dbid_given = true;
321 }
322
323 for(i=0; i<dbmap->num; i++) {
324 if (dbid_given) {
325 if (id == dbmap->dbs[i].dbid) {
326 found = true;
327 break;
328 }
329 } else {
330 ret = ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, tmp_ctx, &name);
331 if (ret != 0) {
332 DEBUG(DEBUG_ERR, ("Unable to get dbname from dbid %u\n", dbmap->dbs[i].dbid));
333 goto fail;
334 }
335
336 if (strcmp(name, dbarg) == 0) {
337 id = dbmap->dbs[i].dbid;
338 found = true;
339 break;
340 }
341 }
342 }
343
344 if (found && dbid_given && dbname != NULL) {
345 ret = ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, tmp_ctx, &name);
346 if (ret != 0) {
347 DEBUG(DEBUG_ERR, ("Unable to get dbname from dbid %u\n", dbmap->dbs[i].dbid));
348 found = false;
349 goto fail;
350 }
351 }
352
353 if (found) {
354 if (dbid) *dbid = id;
355 if (dbname) *dbname = talloc_strdup(ctdb, name);
356 if (flags) *flags = dbmap->dbs[i].flags;
357 } else {
358 DEBUG(DEBUG_ERR,("No database matching '%s' found\n", dbarg));
359 }
360
361 fail:
362 talloc_free(tmp_ctx);
363 return found;
364 }
365
366 /*
367 see if a process exists
368 */
369 static int control_process_exists(struct ctdb_context *ctdb, int argc, const char **argv)
370 {
371 uint32_t pnn, pid;
372 int ret;
373 if (argc < 1) {
374 usage();
375 }
376
377 if (sscanf(argv[0], "%u:%u", &pnn, &pid) != 2) {
378 DEBUG(DEBUG_ERR, ("Badly formed pnn:pid\n"));
379 return -1;
380 }
381
382 ret = ctdb_ctrl_process_exists(ctdb, pnn, pid);
383 if (ret == 0) {
384 printf("%u:%u exists\n", pnn, pid);
385 } else {
386 printf("%u:%u does not exist\n", pnn, pid);
387 }
388 return ret;
389 }
390
391 /*
392 display statistics structure
393 */
394 static void show_statistics(struct ctdb_statistics *s, int show_header)
395 {
396 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
397 int i;
398 const char *prefix=NULL;
399 int preflen=0;
400 int tmp, days, hours, minutes, seconds;
401 const struct {
402 const char *name;
403 uint32_t offset;
404 } fields[] = {
405 #define STATISTICS_FIELD(n) { #n, offsetof(struct ctdb_statistics, n) }
406 STATISTICS_FIELD(num_clients),
407 STATISTICS_FIELD(frozen),
408 STATISTICS_FIELD(recovering),
409 STATISTICS_FIELD(num_recoveries),
410 STATISTICS_FIELD(client_packets_sent),
411 STATISTICS_FIELD(client_packets_recv),
412 STATISTICS_FIELD(node_packets_sent),
413 STATISTICS_FIELD(node_packets_recv),
414 STATISTICS_FIELD(keepalive_packets_sent),
415 STATISTICS_FIELD(keepalive_packets_recv),
416 STATISTICS_FIELD(node.req_call),
417 STATISTICS_FIELD(node.reply_call),
418 STATISTICS_FIELD(node.req_dmaster),
419 STATISTICS_FIELD(node.reply_dmaster),
420 STATISTICS_FIELD(node.reply_error),
421 STATISTICS_FIELD(node.req_message),
422 STATISTICS_FIELD(node.req_control),
423 STATISTICS_FIELD(node.reply_control),
424 STATISTICS_FIELD(client.req_call),
425 STATISTICS_FIELD(client.req_message),
426 STATISTICS_FIELD(client.req_control),
427 STATISTICS_FIELD(timeouts.call),
428 STATISTICS_FIELD(timeouts.control),
429 STATISTICS_FIELD(timeouts.traverse),
430 STATISTICS_FIELD(locks.num_calls),
431 STATISTICS_FIELD(locks.num_current),
432 STATISTICS_FIELD(locks.num_pending),
433 STATISTICS_FIELD(locks.num_failed),
434 STATISTICS_FIELD(total_calls),
435 STATISTICS_FIELD(pending_calls),
436 STATISTICS_FIELD(childwrite_calls),
437 STATISTICS_FIELD(pending_childwrite_calls),
438 STATISTICS_FIELD(memory_used),
439 STATISTICS_FIELD(max_hop_count),
440 STATISTICS_FIELD(total_ro_delegations),
441 STATISTICS_FIELD(total_ro_revokes),
442 };
443
444 tmp = s->statistics_current_time.tv_sec - s->statistics_start_time.tv_sec;
445 seconds = tmp%60;
446 tmp /= 60;
447 minutes = tmp%60;
448 tmp /= 60;
449 hours = tmp%24;
450 tmp /= 24;
451 days = tmp;
452
453 if (options.machinereadable){
454 if (show_header) {
455 printf("CTDB version:");
456 printf("Current time of statistics:");
457 printf("Statistics collected since:");
458 for (i=0;i<ARRAY_SIZE(fields);i++) {
459 printf("%s:", fields[i].name);
460 }
461 printf("num_reclock_ctdbd_latency:");
462 printf("min_reclock_ctdbd_latency:");
463 printf("avg_reclock_ctdbd_latency:");
464 printf("max_reclock_ctdbd_latency:");
465
466 printf("num_reclock_recd_latency:");
467 printf("min_reclock_recd_latency:");
468 printf("avg_reclock_recd_latency:");
469 printf("max_reclock_recd_latency:");
470
471 printf("num_call_latency:");
472 printf("min_call_latency:");
473 printf("avg_call_latency:");
474 printf("max_call_latency:");
475
476 printf("num_lockwait_latency:");
477 printf("min_lockwait_latency:");
478 printf("avg_lockwait_latency:");
479 printf("max_lockwait_latency:");
480
481 printf("num_childwrite_latency:");
482 printf("min_childwrite_latency:");
483 printf("avg_childwrite_latency:");
484 printf("max_childwrite_latency:");
485 printf("\n");
486 }
487 printf("%d:", CTDB_PROTOCOL);
488 printf("%d:", (int)s->statistics_current_time.tv_sec);
489 printf("%d:", (int)s->statistics_start_time.tv_sec);
490 for (i=0;i<ARRAY_SIZE(fields);i++) {
491 printf("%d:", *(uint32_t *)(fields[i].offset+(uint8_t *)s));
492 }
493 printf("%d:", s->reclock.ctdbd.num);
494 printf("%.6f:", s->reclock.ctdbd.min);
495 printf("%.6f:", s->reclock.ctdbd.num?s->reclock.ctdbd.total/s->reclock.ctdbd.num:0.0);
496 printf("%.6f:", s->reclock.ctdbd.max);
497
498 printf("%d:", s->reclock.recd.num);
499 printf("%.6f:", s->reclock.recd.min);
500 printf("%.6f:", s->reclock.recd.num?s->reclock.recd.total/s->reclock.recd.num:0.0);
501 printf("%.6f:", s->reclock.recd.max);
502
503 printf("%d:", s->call_latency.num);
504 printf("%.6f:", s->call_latency.min);
505 printf("%.6f:", s->call_latency.num?s->call_latency.total/s->call_latency.num:0.0);
506 printf("%.6f:", s->call_latency.max);
507
508 printf("%d:", s->childwrite_latency.num);
509 printf("%.6f:", s->childwrite_latency.min);
510 printf("%.6f:", s->childwrite_latency.num?s->childwrite_latency.total/s->childwrite_latency.num:0.0);
511 printf("%.6f:", s->childwrite_latency.max);
512 printf("\n");
513 } else {
514 printf("CTDB version %u\n", CTDB_PROTOCOL);
515 printf("Current time of statistics : %s", ctime(&s->statistics_current_time.tv_sec));
516 printf("Statistics collected since : (%03d %02d:%02d:%02d) %s", days, hours, minutes, seconds, ctime(&s->statistics_start_time.tv_sec));
517
518 for (i=0;i<ARRAY_SIZE(fields);i++) {
519 if (strchr(fields[i].name, '.')) {
520 preflen = strcspn(fields[i].name, ".")+1;
521 if (!prefix || strncmp(prefix, fields[i].name, preflen) != 0) {
522 prefix = fields[i].name;
523 printf(" %*.*s\n", preflen-1, preflen-1, fields[i].name);
524 }
525 } else {
526 preflen = 0;
527 }
528 printf(" %*s%-22s%*s%10u\n",
529 preflen?4:0, "",
530 fields[i].name+preflen,
531 preflen?0:4, "",
532 *(uint32_t *)(fields[i].offset+(uint8_t *)s));
533 }
534 printf(" hop_count_buckets:");
535 for (i=0;i<MAX_COUNT_BUCKETS;i++) {
536 printf(" %d", s->hop_count_bucket[i]);
537 }
538 printf("\n");
539 printf(" lock_buckets:");
540 for (i=0; i<MAX_COUNT_BUCKETS; i++) {
541 printf(" %d", s->locks.buckets[i]);
542 }
543 printf("\n");
544 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n", "locks_latency MIN/AVG/MAX", s->locks.latency.min, s->locks.latency.num?s->locks.latency.total/s->locks.latency.num:0.0, s->locks.latency.max, s->locks.latency.num);
545
546 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n", "reclock_ctdbd MIN/AVG/MAX", s->reclock.ctdbd.min, s->reclock.ctdbd.num?s->reclock.ctdbd.total/s->reclock.ctdbd.num:0.0, s->reclock.ctdbd.max, s->reclock.ctdbd.num);
547
548 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n", "reclock_recd MIN/AVG/MAX", s->reclock.recd.min, s->reclock.recd.num?s->reclock.recd.total/s->reclock.recd.num:0.0, s->reclock.recd.max, s->reclock.recd.num);
549
550 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n", "call_latency MIN/AVG/MAX", s->call_latency.min, s->call_latency.num?s->call_latency.total/s->call_latency.num:0.0, s->call_latency.max, s->call_latency.num);
551 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n", "childwrite_latency MIN/AVG/MAX", s->childwrite_latency.min, s->childwrite_latency.num?s->childwrite_latency.total/s->childwrite_latency.num:0.0, s->childwrite_latency.max, s->childwrite_latency.num);
552 }
553
554 talloc_free(tmp_ctx);
555 }
556
557 /*
558 display remote ctdb statistics combined from all nodes
559 */
560 static int control_statistics_all(struct ctdb_context *ctdb)
561 {
562 int ret, i;
563 struct ctdb_statistics statistics;
564 uint32_t *nodes;
565 uint32_t num_nodes;
566
567 nodes = ctdb_get_connected_nodes(ctdb, TIMELIMIT(), ctdb, &num_nodes);
568 CTDB_NO_MEMORY(ctdb, nodes);
569
570 ZERO_STRUCT(statistics);
571
572 for (i=0;i<num_nodes;i++) {
573 struct ctdb_statistics s1;
574 int j;
575 uint32_t *v1 = (uint32_t *)&s1;
576 uint32_t *v2 = (uint32_t *)&statistics;
577 uint32_t num_ints =
578 offsetof(struct ctdb_statistics, __last_counter) / sizeof(uint32_t);
579 ret = ctdb_ctrl_statistics(ctdb, nodes[i], &s1);
580 if (ret != 0) {
581 DEBUG(DEBUG_ERR, ("Unable to get statistics from node %u\n", nodes[i]));
582 return ret;
583 }
584 for (j=0;j<num_ints;j++) {
585 v2[j] += v1[j];
586 }
587 statistics.max_hop_count =
588 MAX(statistics.max_hop_count, s1.max_hop_count);
589 statistics.call_latency.max =
590 MAX(statistics.call_latency.max, s1.call_latency.max);
591 }
592 talloc_free(nodes);
593 printf("Gathered statistics for %u nodes\n", num_nodes);
594 show_statistics(&statistics, 1);
595 return 0;
596 }
597
598 /*
599 display remote ctdb statistics
600 */
601 static int control_statistics(struct ctdb_context *ctdb, int argc, const char **argv)
602 {
603 int ret;
604 struct ctdb_statistics statistics;
605
606 if (options.pnn == CTDB_BROADCAST_ALL) {
607 return control_statistics_all(ctdb);
608 }
609
610 ret = ctdb_ctrl_statistics(ctdb, options.pnn, &statistics);
611 if (ret != 0) {
612 DEBUG(DEBUG_ERR, ("Unable to get statistics from node %u\n", options.pnn));
613 return ret;
614 }
615 show_statistics(&statistics, 1);
616 return 0;
617 }
618
619
620 /*
621 reset remote ctdb statistics
622 */
623 static int control_statistics_reset(struct ctdb_context *ctdb, int argc, const char **argv)
624 {
625 int ret;
626
627 ret = ctdb_statistics_reset(ctdb, options.pnn);
628 if (ret != 0) {
629 DEBUG(DEBUG_ERR, ("Unable to reset statistics on node %u\n", options.pnn));
630 return ret;
631 }
632 return 0;
633 }
634
635
636 /*
637 display remote ctdb rolling statistics
638 */
639 static int control_stats(struct ctdb_context *ctdb, int argc, const char **argv)
640 {
641 int ret;
642 struct ctdb_statistics_wire *stats;
643 int i, num_records = -1;
644
645 assert_single_node_only();
646
647 if (argc ==1) {
648 num_records = atoi(argv[0]) - 1;
649 }
650
651 ret = ctdb_ctrl_getstathistory(ctdb, TIMELIMIT(), options.pnn, ctdb, &stats);
652 if (ret != 0) {
653 DEBUG(DEBUG_ERR, ("Unable to get rolling statistics from node %u\n", options.pnn));
654 return ret;
655 }
656 for (i=0;i<stats->num;i++) {
657 if (stats->stats[i].statistics_start_time.tv_sec == 0) {
658 continue;
659 }
660 show_statistics(&stats->stats[i], i==0);
661 if (i == num_records) {
662 break;
663 }
664 }
665 return 0;
666 }
667
668
669 /*
670 display remote ctdb db statistics
671 */
672 static int control_dbstatistics(struct ctdb_context *ctdb, int argc, const char **argv)
673 {
674 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
675 struct ctdb_db_statistics *dbstat;
676 int i;
677 uint32_t db_id;
678 int num_hot_keys;
679 int ret;
680
681 if (argc < 1) {
682 usage();
683 }
684
685 if (!db_exists(ctdb, argv[0], &db_id, NULL, NULL)) {
686 return -1;
687 }
688
689 ret = ctdb_ctrl_dbstatistics(ctdb, options.pnn, db_id, tmp_ctx, &dbstat);
690 if (ret != 0) {
691 DEBUG(DEBUG_ERR,("Failed to read db statistics from node\n"));
692 talloc_free(tmp_ctx);
693 return -1;
694 }
695
696 printf("DB Statistics: %s\n", argv[0]);
697 printf(" %*s%-22s%*s%10u\n", 0, "", "ro_delegations", 4, "",
698 dbstat->db_ro_delegations);
699 printf(" %*s%-22s%*s%10u\n", 0, "", "ro_revokes", 4, "",
700 dbstat->db_ro_delegations);
701 printf(" %s\n", "locks");
702 printf(" %*s%-22s%*s%10u\n", 4, "", "total", 0, "",
703 dbstat->locks.num_calls);
704 printf(" %*s%-22s%*s%10u\n", 4, "", "failed", 0, "",
705 dbstat->locks.num_failed);
706 printf(" %*s%-22s%*s%10u\n", 4, "", "current", 0, "",
707 dbstat->locks.num_current);
708 printf(" %*s%-22s%*s%10u\n", 4, "", "pending", 0, "",
709 dbstat->locks.num_pending);
710 printf(" %s", "hop_count_buckets:");
711 for (i=0; i<MAX_COUNT_BUCKETS; i++) {
712 printf(" %d", dbstat->hop_count_bucket[i]);
713 }
714 printf("\n");
715 printf(" %s", "lock_buckets:");
716 for (i=0; i<MAX_COUNT_BUCKETS; i++) {
717 printf(" %d", dbstat->locks.buckets[i]);
718 }
719 printf("\n");
720 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n",
721 "locks_latency MIN/AVG/MAX",
722 dbstat->locks.latency.min,
723 (dbstat->locks.latency.num ?
724 dbstat->locks.latency.total /dbstat->locks.latency.num :
725 0.0),
726 dbstat->locks.latency.max,
727 dbstat->locks.latency.num);
728 printf(" %-30s %.6f/%.6f/%.6f sec out of %d\n",
729 "vacuum_latency MIN/AVG/MAX",
730 dbstat->vacuum.latency.min,
731 (dbstat->vacuum.latency.num ?
732 dbstat->vacuum.latency.total /dbstat->vacuum.latency.num :
733 0.0),
734 dbstat->vacuum.latency.max,
735 dbstat->vacuum.latency.num);
736 num_hot_keys = 0;
737 for (i=0; i<dbstat->num_hot_keys; i++) {
738 if (dbstat->hot_keys[i].count > 0) {
739 num_hot_keys++;
740 }
741 }
742 dbstat->num_hot_keys = num_hot_keys;
743
744 printf(" Num Hot Keys: %d\n", dbstat->num_hot_keys);
745 for (i = 0; i < dbstat->num_hot_keys; i++) {
746 int j;
747 printf(" Count:%d Key:", dbstat->hot_keys[i].count);
748 for (j = 0; j < dbstat->hot_keys[i].key.dsize; j++) {
749 printf("%02x", dbstat->hot_keys[i].key.dptr[j]&0xff);
750 }
751 printf("\n");
752 }
753
754 talloc_free(tmp_ctx);
755 return 0;
756 }
757
758 /*
759 display uptime of remote node
760 */
761 static int control_uptime(struct ctdb_context *ctdb, int argc, const char **argv)
762 {
763 int ret;
764 struct ctdb_uptime *uptime = NULL;
765 int tmp, days, hours, minutes, seconds;
766
767 ret = ctdb_ctrl_uptime(ctdb, ctdb, TIMELIMIT(), options.pnn, &uptime);
768 if (ret != 0) {
769 DEBUG(DEBUG_ERR, ("Unable to get uptime from node %u\n", options.pnn));
770 return ret;
771 }
772
773 if (options.machinereadable){
774 printf(":Current Node Time:Ctdb Start Time:Last Recovery/Failover Time:Last Recovery/IPFailover Duration:\n");
775 printf(":%u:%u:%u:%lf\n",
776 (unsigned int)uptime->current_time.tv_sec,
777 (unsigned int)uptime->ctdbd_start_time.tv_sec,
778 (unsigned int)uptime->last_recovery_finished.tv_sec,
779 timeval_delta(&uptime->last_recovery_finished,
780 &uptime->last_recovery_started)
781 );
782 return 0;
783 }
784
785 printf("Current time of node : %s", ctime(&uptime->current_time.tv_sec));
786
787 tmp = uptime->current_time.tv_sec - uptime->ctdbd_start_time.tv_sec;
788 seconds = tmp%60;
789 tmp /= 60;
790 minutes = tmp%60;
791 tmp /= 60;
792 hours = tmp%24;
793 tmp /= 24;
794 days = tmp;
795 printf("Ctdbd start time : (%03d %02d:%02d:%02d) %s", days, hours, minutes, seconds, ctime(&uptime->ctdbd_start_time.tv_sec));
796
797 tmp = uptime->current_time.tv_sec - uptime->last_recovery_finished.tv_sec;
798 seconds = tmp%60;
799 tmp /= 60;
800 minutes = tmp%60;
801 tmp /= 60;
802 hours = tmp%24;
803 tmp /= 24;
804 days = tmp;
805 printf("Time of last recovery/failover: (%03d %02d:%02d:%02d) %s", days, hours, minutes, seconds, ctime(&uptime->last_recovery_finished.tv_sec));
806
807 printf("Duration of last recovery/failover: %lf seconds\n",
808 timeval_delta(&uptime->last_recovery_finished,
809 &uptime->last_recovery_started));
810
811 return 0;
812 }
813
814 /*
815 show the PNN of the current node
816 */
817 static int control_pnn(struct ctdb_context *ctdb, int argc, const char **argv)
818 {
819 uint32_t mypnn;
820
821 mypnn = getpnn(ctdb);
822
823 printf("PNN:%d\n", mypnn);
824 return 0;
825 }
826
827
828 struct pnn_node {
829 struct pnn_node *next, *prev;
830 ctdb_sock_addr addr;
831 int pnn;
832 };
833
834 static struct pnn_node *read_pnn_node_file(TALLOC_CTX *mem_ctx,
835 const char *file)
836 {
837 int nlines;
838 char **lines;
839 int i, pnn;
840 struct pnn_node *pnn_nodes = NULL;
841 struct pnn_node *pnn_node;
842
843 lines = file_lines_load(file, &nlines, 0, mem_ctx);
844 if (lines == NULL) {
845 return NULL;
846 }
847 for (i=0, pnn=0; i<nlines; i++) {
848 char *node;
849
850 node = lines[i];
851 /* strip leading spaces */
852 while((*node == ' ') || (*node == '\t')) {
853 node++;
854 }
855 if (*node == '#') {
856 pnn++;
857 continue;
858 }
859 if (strcmp(node, "") == 0) {
860 continue;
861 }
862 pnn_node = talloc(mem_ctx, struct pnn_node);
863 pnn_node->pnn = pnn++;
864
865 if (!parse_ip(node, NULL, 0, &pnn_node->addr)) {
866 DEBUG(DEBUG_ERR,
867 ("Invalid IP address '%s' in file %s\n",
868 node, file));
869 /* Caller will free mem_ctx */
870 return NULL;
871 }
872
873 DLIST_ADD_END(pnn_nodes, pnn_node, NULL);
874 }
875
876 return pnn_nodes;
877 }
878
879 static struct pnn_node *read_nodes_file(TALLOC_CTX *mem_ctx)
880 {
881 const char *nodes_list;
882
883 /* read the nodes file */
884 nodes_list = getenv("CTDB_NODES");
885 if (nodes_list == NULL) {
886 nodes_list = talloc_asprintf(mem_ctx, "%s/nodes",
887 getenv("CTDB_BASE"));
888 if (nodes_list == NULL) {
889 DEBUG(DEBUG_ALERT,(__location__ " Out of memory\n"));
890 exit(1);
891 }
892 }
893
894 return read_pnn_node_file(mem_ctx, nodes_list);
895 }
896
897 /*
898 show the PNN of the current node
899 discover the pnn by loading the nodes file and try to bind to all
900 addresses one at a time until the ip address is found.
901 */
902 static int find_node_xpnn(void)
903 {
904 TALLOC_CTX *mem_ctx = talloc_new(NULL);
905 struct pnn_node *pnn_nodes;
906 struct pnn_node *pnn_node;
907 int pnn;
908
909 pnn_nodes = read_nodes_file(mem_ctx);
910 if (pnn_nodes == NULL) {
911 DEBUG(DEBUG_ERR,("Failed to read nodes file\n"));
912 talloc_free(mem_ctx);
913 return -1;
914 }
915
916 for(pnn_node=pnn_nodes;pnn_node;pnn_node=pnn_node->next) {
917 if (ctdb_sys_have_ip(&pnn_node->addr)) {
918 pnn = pnn_node->pnn;
919 talloc_free(mem_ctx);
920 return pnn;
921 }
922 }
923
924 printf("Failed to detect which PNN this node is\n");
925 talloc_free(mem_ctx);
926 return -1;
927 }
928
929 static int control_xpnn(struct ctdb_context *ctdb, int argc, const char **argv)
930 {
931 uint32_t pnn;
932
933 assert_single_node_only();
934
935 pnn = find_node_xpnn();
936 if (pnn == -1) {
937 return -1;
938 }
939
940 printf("PNN:%d\n", pnn);
941 return 0;
942 }
943
944 /* Helpers for ctdb status
945 */
946 static bool is_partially_online(struct ctdb_context *ctdb, struct ctdb_node_and_flags *node)
947 {
948 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
949 int j;
950 bool ret = false;
951
952 if (node->flags == 0) {
953 struct ctdb_control_get_ifaces *ifaces;
954
955 if (ctdb_ctrl_get_ifaces(ctdb, TIMELIMIT(), node->pnn,
956 tmp_ctx, &ifaces) == 0) {
957 for (j=0; j < ifaces->num; j++) {
958 if (ifaces->ifaces[j].link_state != 0) {
959 continue;
960 }
961 ret = true;
962 break;
963 }
964 }
965 }
966 talloc_free(tmp_ctx);
967
968 return ret;
969 }
970
971 static void control_status_header_machine(void)
972 {
973 printf(":Node:IP:Disconnected:Banned:Disabled:Unhealthy:Stopped"
974 ":Inactive:PartiallyOnline:ThisNode:\n");
975 }
976
977 static int control_status_1_machine(struct ctdb_context *ctdb, int mypnn,
978 struct ctdb_node_and_flags *node)
979 {
980 printf(":%d:%s:%d:%d:%d:%d:%d:%d:%d:%c:\n", node->pnn,
981 ctdb_addr_to_str(&node->addr),
982 !!(node->flags&NODE_FLAGS_DISCONNECTED),
983 !!(node->flags&NODE_FLAGS_BANNED),
984 !!(node->flags&NODE_FLAGS_PERMANENTLY_DISABLED),
985 !!(node->flags&NODE_FLAGS_UNHEALTHY),
986 !!(node->flags&NODE_FLAGS_STOPPED),
987 !!(node->flags&NODE_FLAGS_INACTIVE),
988 is_partially_online(ctdb, node) ? 1 : 0,
989 (node->pnn == mypnn)?'Y':'N');
990
991 return node->flags;
992 }
993
994 static int control_status_1_human(struct ctdb_context *ctdb, int mypnn,
995 struct ctdb_node_and_flags *node)
996 {
997 printf("pnn:%d %-16s %s%s\n", node->pnn,
998 ctdb_addr_to_str(&node->addr),
999 is_partially_online(ctdb, node) ? "PARTIALLYONLINE" : pretty_print_flags(node->flags),
1000 node->pnn == mypnn?" (THIS NODE)":"");
1001
1002 return node->flags;
1003 }
1004
1005 /*
1006 display remote ctdb status
1007 */
1008 static int control_status(struct ctdb_context *ctdb, int argc, const char **argv)
1009 {
1010 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1011 int i;
1012 struct ctdb_vnn_map *vnnmap=NULL;
1013 struct ctdb_node_map *nodemap=NULL;
1014 uint32_t recmode, recmaster, mypnn;
1015 int num_deleted_nodes = 0;
1016 int ret;
1017
1018 mypnn = getpnn(ctdb);
1019
1020 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &nodemap);
1021 if (ret != 0) {
1022 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1023 talloc_free(tmp_ctx);
1024 return -1;
1025 }
1026
1027 if (options.machinereadable) {
1028 control_status_header_machine();
1029 for (i=0;i<nodemap->num;i++) {
1030 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
1031 continue;
1032 }
1033 (void) control_status_1_machine(ctdb, mypnn,
1034 &nodemap->nodes[i]);
1035 }
1036 talloc_free(tmp_ctx);
1037 return 0;
1038 }
1039
1040 for (i=0; i<nodemap->num; i++) {
1041 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
1042 num_deleted_nodes++;
1043 }
1044 }
1045 if (num_deleted_nodes == 0) {
1046 printf("Number of nodes:%d\n", nodemap->num);
1047 } else {
1048 printf("Number of nodes:%d (including %d deleted nodes)\n",
1049 nodemap->num, num_deleted_nodes);
1050 }
1051 for(i=0;i<nodemap->num;i++){
1052 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
1053 continue;
1054 }
1055 (void) control_status_1_human(ctdb, mypnn, &nodemap->nodes[i]);
1056 }
1057
1058 ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &vnnmap);
1059 if (ret != 0) {
1060 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n", options.pnn));
1061 talloc_free(tmp_ctx);
1062 return -1;
1063 }
1064 if (vnnmap->generation == INVALID_GENERATION) {
1065 printf("Generation:INVALID\n");
1066 } else {
1067 printf("Generation:%d\n",vnnmap->generation);
1068 }
1069 printf("Size:%d\n",vnnmap->size);
1070 for(i=0;i<vnnmap->size;i++){
1071 printf("hash:%d lmaster:%d\n", i, vnnmap->map[i]);
1072 }
1073
1074 ret = ctdb_ctrl_getrecmode(ctdb, tmp_ctx, TIMELIMIT(), options.pnn, &recmode);
1075 if (ret != 0) {
1076 DEBUG(DEBUG_ERR, ("Unable to get recmode from node %u\n", options.pnn));
1077 talloc_free(tmp_ctx);
1078 return -1;
1079 }
1080 printf("Recovery mode:%s (%d)\n",recmode==CTDB_RECOVERY_NORMAL?"NORMAL":"RECOVERY",recmode);
1081
1082 ret = ctdb_ctrl_getrecmaster(ctdb, tmp_ctx, TIMELIMIT(), options.pnn, &recmaster);
1083 if (ret != 0) {
1084 DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
1085 talloc_free(tmp_ctx);
1086 return -1;
1087 }
1088 printf("Recovery master:%d\n",recmaster);
1089
1090 talloc_free(tmp_ctx);
1091 return 0;
1092 }
1093
1094 static int control_nodestatus(struct ctdb_context *ctdb, int argc, const char **argv)
1095 {
1096 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1097 int i, ret;
1098 struct ctdb_node_map *nodemap=NULL;
1099 uint32_t * nodes;
1100 uint32_t pnn_mode, mypnn;
1101
1102 if (argc > 1) {
1103 usage();
1104 }
1105
1106 if (!parse_nodestring(ctdb, tmp_ctx, argc == 1 ? argv[0] : NULL,
1107 options.pnn, true, &nodes, &pnn_mode)) {
1108 return -1;
1109 }
1110
1111 if (options.machinereadable) {
1112 control_status_header_machine();
1113 } else if (pnn_mode == CTDB_BROADCAST_ALL) {
1114 printf("Number of nodes:%d\n", (int) talloc_array_length(nodes));
1115 }
1116
1117 mypnn = getpnn(ctdb);
1118
1119 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &nodemap);
1120 if (ret != 0) {
1121 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1122 talloc_free(tmp_ctx);
1123 return -1;
1124 }
1125
1126 ret = 0;
1127
1128 for (i = 0; i < talloc_array_length(nodes); i++) {
1129 if (options.machinereadable) {
1130 ret |= control_status_1_machine(ctdb, mypnn,
1131 &nodemap->nodes[nodes[i]]);
1132 } else {
1133 ret |= control_status_1_human(ctdb, mypnn,
1134 &nodemap->nodes[nodes[i]]);
1135 }
1136 }
1137
1138 talloc_free(tmp_ctx);
1139 return ret;
1140 }
1141
1142 static struct pnn_node *read_natgw_nodes_file(struct ctdb_context *ctdb,
1143 TALLOC_CTX *mem_ctx)
1144 {
1145 const char *natgw_list;
1146 struct pnn_node *natgw_nodes = NULL;
1147
1148 natgw_list = getenv("CTDB_NATGW_NODES");
1149 if (natgw_list == NULL) {
1150 natgw_list = talloc_asprintf(mem_ctx, "%s/natgw_nodes",
1151 getenv("CTDB_BASE"));
1152 if (natgw_list == NULL) {
1153 DEBUG(DEBUG_ALERT,(__location__ " Out of memory\n"));
1154 exit(1);
1155 }
1156 }
1157 /* The PNNs will be junk but they're not used */
1158 natgw_nodes = read_pnn_node_file(mem_ctx, natgw_list);
1159 if (natgw_nodes == NULL) {
1160 DEBUG(DEBUG_ERR,
1161 ("Failed to load natgw node list '%s'\n", natgw_list));
1162 }
1163 return natgw_nodes;
1164 }
1165
1166
1167 /* talloc off the existing nodemap... */
1168 static struct ctdb_node_map *talloc_nodemap(struct ctdb_node_map *nodemap)
1169 {
1170 return talloc_zero_size(nodemap,
1171 offsetof(struct ctdb_node_map, nodes) +
1172 nodemap->num * sizeof(struct ctdb_node_and_flags));
1173 }
1174
1175 static struct ctdb_node_map *
1176 filter_nodemap_by_addrs(struct ctdb_context *ctdb,
1177 struct ctdb_node_map *nodemap,
1178 struct pnn_node *nodes)
1179 {
1180 int i;
1181 struct pnn_node *n;
1182 struct ctdb_node_map *ret;
1183
1184 ret = talloc_nodemap(nodemap);
1185 CTDB_NO_MEMORY_NULL(ctdb, ret);
1186
1187 ret->num = 0;
1188
1189 for (i = 0; i < nodemap->num; i++) {
1190 for(n = nodes; n != NULL ; n = n->next) {
1191 if (ctdb_same_ip(&n->addr,
1192 &nodemap->nodes[i].addr)) {
1193 break;
1194 }
1195 }
1196 if (n == NULL) {
1197 continue;
1198 }
1199
1200 ret->nodes[ret->num] = nodemap->nodes[i];
1201 ret->num++;
1202 }
1203
1204 return ret;
1205 }
1206
1207 static struct ctdb_node_map *
1208 filter_nodemap_by_capabilities(struct ctdb_context *ctdb,
1209 struct ctdb_node_map *nodemap,
1210 uint32_t required_capabilities,
1211 bool first_only)
1212 {
1213 int i;
1214 uint32_t capabilities;
1215 struct ctdb_node_map *ret;
1216
1217 ret = talloc_nodemap(nodemap);
1218 CTDB_NO_MEMORY_NULL(ctdb, ret);
1219
1220 ret->num = 0;
1221
1222 for (i = 0; i < nodemap->num; i++) {
1223 int res;
1224
1225 /* Disconnected nodes have no capabilities! */
1226 if (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) {
1227 continue;
1228 }
1229
1230 res = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(),
1231 nodemap->nodes[i].pnn,
1232 &capabilities);
1233 if (res != 0) {
1234 DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n",
1235 nodemap->nodes[i].pnn));
1236 talloc_free(ret);
1237 return NULL;
1238 }
1239 if (!(capabilities & required_capabilities)) {
1240 continue;
1241 }
1242
1243 ret->nodes[ret->num] = nodemap->nodes[i];
1244 ret->num++;
1245 if (first_only) {
1246 break;
1247 }
1248 }
1249
1250 return ret;
1251 }
1252
1253 static struct ctdb_node_map *
1254 filter_nodemap_by_flags(struct ctdb_context *ctdb,
1255 struct ctdb_node_map *nodemap,
1256 uint32_t flags_mask)
1257 {
1258 int i;
1259 struct ctdb_node_map *ret;
1260
1261 ret = talloc_nodemap(nodemap);
1262 CTDB_NO_MEMORY_NULL(ctdb, ret);
1263
1264 ret->num = 0;
1265
1266 for (i = 0; i < nodemap->num; i++) {
1267 if (nodemap->nodes[i].flags & flags_mask) {
1268 continue;
1269 }
1270
1271 ret->nodes[ret->num] = nodemap->nodes[i];
1272 ret->num++;
1273 }
1274
1275 return ret;
1276 }
1277
1278 /*
1279 display the list of nodes belonging to this natgw configuration
1280 */
1281 static int control_natgwlist(struct ctdb_context *ctdb, int argc, const char **argv)
1282 {
1283 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1284 int i, ret;
1285 struct pnn_node *natgw_nodes = NULL;
1286 struct ctdb_node_map *orig_nodemap=NULL;
1287 struct ctdb_node_map *nodemap;
1288 uint32_t mypnn, pnn;
1289 const char *ip;
1290
1291 /* When we have some nodes that could be the NATGW, make a
1292 * series of attempts to find the first node that doesn't have
1293 * certain status flags set.
1294 */
1295 uint32_t exclude_flags[] = {
1296 /* Look for a nice healthy node */
1297 NODE_FLAGS_DISCONNECTED|NODE_FLAGS_STOPPED|NODE_FLAGS_DELETED|NODE_FLAGS_BANNED|NODE_FLAGS_UNHEALTHY,
1298 /* If not found, an UNHEALTHY/BANNED node will do */
1299 NODE_FLAGS_DISCONNECTED|NODE_FLAGS_STOPPED|NODE_FLAGS_DELETED,
1300 /* If not found, a STOPPED node will do */
1301 NODE_FLAGS_DISCONNECTED|NODE_FLAGS_DELETED,
1302 0,
1303 };
1304
1305 /* read the natgw nodes file into a linked list */
1306 natgw_nodes = read_natgw_nodes_file(ctdb, tmp_ctx);
1307 if (natgw_nodes == NULL) {
1308 ret = -1;
1309 goto done;
1310 }
1311
1312 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE,
1313 tmp_ctx, &orig_nodemap);
1314 if (ret != 0) {
1315 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node.\n"));
1316 talloc_free(tmp_ctx);
1317 return -1;
1318 }
1319
1320 /* Get a nodemap that includes only the nodes in the NATGW
1321 * group */
1322 nodemap = filter_nodemap_by_addrs(ctdb, orig_nodemap, natgw_nodes);
1323 if (nodemap == NULL) {
1324 ret = -1;
1325 goto done;
1326 }
1327
1328 ret = 2; /* matches ENOENT */
1329 pnn = -1;
1330 ip = "0.0.0.0";
1331 /* For each flag mask... */
1332 for (i = 0; exclude_flags[i] != 0; i++) {
1333 /* ... get a nodemap that excludes nodes with with
1334 * masked flags... */
1335 struct ctdb_node_map *t =
1336 filter_nodemap_by_flags(ctdb, nodemap,
1337 exclude_flags[i]);
1338 if (t == NULL) {
1339 /* No memory */
1340 ret = -1;
1341 goto done;
1342 }
1343 if (t->num > 0) {
1344 /* ... and find the first node with the NATGW
1345 * capability */
1346 struct ctdb_node_map *n;
1347 n = filter_nodemap_by_capabilities(ctdb, t,
1348 CTDB_CAP_NATGW,
1349 true);
1350 if (n == NULL) {
1351 /* No memory */
1352 ret = -1;
1353 goto done;
1354 }
1355 if (n->num > 0) {
1356 ret = 0;
1357 pnn = n->nodes[0].pnn;
1358 ip = ctdb_addr_to_str(&n->nodes[0].addr);
1359 break;
1360 }
1361 }
1362 talloc_free(t);
1363 }
1364
1365 if (options.machinereadable) {
1366 printf(":Node:IP:\n");
1367 printf(":%d:%s:\n", pnn, ip);
1368 } else {
1369 printf("%d %s\n", pnn, ip);
1370 }
1371
1372 /* print the pruned list of nodes belonging to this natgw list */
1373 mypnn = getpnn(ctdb);
1374 if (options.machinereadable) {
1375 control_status_header_machine();
1376 } else {
1377 printf("Number of nodes:%d\n", nodemap->num);
1378 }
1379 for(i=0;i<nodemap->num;i++){
1380 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
1381 continue;
1382 }
1383 if (options.machinereadable) {
1384 control_status_1_machine(ctdb, mypnn, &(nodemap->nodes[i]));
1385 } else {
1386 control_status_1_human(ctdb, mypnn, &(nodemap->nodes[i]));
1387 }
1388 }
1389
1390 done:
1391 talloc_free(tmp_ctx);
1392 return ret;
1393 }
1394
1395 /*
1396 display the status of the scripts for monitoring (or other events)
1397 */
1398 static int control_one_scriptstatus(struct ctdb_context *ctdb,
1399 enum ctdb_eventscript_call type)
1400 {
1401 struct ctdb_scripts_wire *script_status;
1402 int ret, i;
1403
1404 ret = ctdb_ctrl_getscriptstatus(ctdb, TIMELIMIT(), options.pnn, ctdb, type, &script_status);
1405 if (ret != 0) {
1406 DEBUG(DEBUG_ERR, ("Unable to get script status from node %u\n", options.pnn));
1407 return ret;
1408 }
1409
1410 if (script_status == NULL) {
1411 if (!options.machinereadable) {
1412 printf("%s cycle never run\n",
1413 ctdb_eventscript_call_names[type]);
1414 }
1415 return 0;
1416 }
1417
1418 if (!options.machinereadable) {
1419 int num_run = 0;
1420 for (i=0; i<script_status->num_scripts; i++) {
1421 if (script_status->scripts[i].status != -ENOEXEC) {
1422 num_run++;
1423 }
1424 }
1425 printf("%d scripts were executed last %s cycle\n",
1426 num_run,
1427 ctdb_eventscript_call_names[type]);
1428 }
1429 for (i=0; i<script_status->num_scripts; i++) {
1430 const char *status = NULL;
1431
1432 switch (script_status->scripts[i].status) {
1433 case -ETIME:
1434 status = "TIMEDOUT";
1435 break;
1436 case -ENOEXEC:
1437 status = "DISABLED";
1438 break;
1439 case 0:
1440 status = "OK";
1441 break;
1442 default:
1443 if (script_status->scripts[i].status > 0)
1444 status = "ERROR";
1445 break;
1446 }
1447 if (options.machinereadable) {
1448 printf(":%s:%s:%i:%s:%lu.%06lu:%lu.%06lu:%s:\n",
1449 ctdb_eventscript_call_names[type],
1450 script_status->scripts[i].name,
1451 script_status->scripts[i].status,
1452 status,
1453 (long)script_status->scripts[i].start.tv_sec,
1454 (long)script_status->scripts[i].start.tv_usec,
1455 (long)script_status->scripts[i].finished.tv_sec,
1456 (long)script_status->scripts[i].finished.tv_usec,
1457 script_status->scripts[i].output);
1458 continue;
1459 }
1460 if (status)
1461 printf("%-20s Status:%s ",
1462 script_status->scripts[i].name, status);
1463 else
1464 /* Some other error, eg from stat. */
1465 printf("%-20s Status:CANNOT RUN (%s)",
1466 script_status->scripts[i].name,
1467 strerror(-script_status->scripts[i].status));
1468
1469 if (script_status->scripts[i].status >= 0) {
1470 printf("Duration:%.3lf ",
1471 timeval_delta(&script_status->scripts[i].finished,
1472 &script_status->scripts[i].start));
1473 }
1474 if (script_status->scripts[i].status != -ENOEXEC) {
1475 printf("%s",
1476 ctime(&script_status->scripts[i].start.tv_sec));
1477 if (script_status->scripts[i].status != 0) {
1478 printf(" OUTPUT:%s\n",
1479 script_status->scripts[i].output);
1480 }
1481 } else {
1482 printf("\n");
1483 }
1484 }
1485 return 0;
1486 }
1487
1488
1489 static int control_scriptstatus(struct ctdb_context *ctdb,
1490 int argc, const char **argv)
1491 {
1492 int ret;
1493 enum ctdb_eventscript_call type, min, max;
1494 const char *arg;
1495
1496 if (argc > 1) {
1497 DEBUG(DEBUG_ERR, ("Unknown arguments to scriptstatus\n"));
1498 return -1;
1499 }
1500
1501 if (argc == 0)
1502 arg = ctdb_eventscript_call_names[CTDB_EVENT_MONITOR];
1503 else
1504 arg = argv[0];
1505
1506 for (type = 0; type < CTDB_EVENT_MAX; type++) {
1507 if (strcmp(arg, ctdb_eventscript_call_names[type]) == 0) {
1508 min = type;
1509 max = type+1;
1510 break;
1511 }
1512 }
1513 if (type == CTDB_EVENT_MAX) {
1514 if (strcmp(arg, "all") == 0) {
1515 min = 0;
1516 max = CTDB_EVENT_MAX;
1517 } else {
1518 DEBUG(DEBUG_ERR, ("Unknown event type %s\n", argv[0]));
1519 return -1;
1520 }
1521 }
1522
1523 if (options.machinereadable) {
1524 printf(":Type:Name:Code:Status:Start:End:Error Output...:\n");
1525 }
1526
1527 for (type = min; type < max; type++) {
1528 ret = control_one_scriptstatus(ctdb, type);
1529 if (ret != 0) {
1530 return ret;
1531 }
1532 }
1533
1534 return 0;
1535 }
1536
1537 /*
1538 enable an eventscript
1539 */
1540 static int control_enablescript(struct ctdb_context *ctdb, int argc, const char **argv)
1541 {
1542 int ret;
1543
1544 if (argc < 1) {
1545 usage();
1546 }
1547
1548 ret = ctdb_ctrl_enablescript(ctdb, TIMELIMIT(), options.pnn, argv[0]);
1549 if (ret != 0) {
1550 DEBUG(DEBUG_ERR, ("Unable to enable script %s on node %u\n", argv[0], options.pnn));
1551 return ret;
1552 }
1553
1554 return 0;
1555 }
1556
1557 /*
1558 disable an eventscript
1559 */
1560 static int control_disablescript(struct ctdb_context *ctdb, int argc, const char **argv)
1561 {
1562 int ret;
1563
1564 if (argc < 1) {
1565 usage();
1566 }
1567
1568 ret = ctdb_ctrl_disablescript(ctdb, TIMELIMIT(), options.pnn, argv[0]);
1569 if (ret != 0) {
1570 DEBUG(DEBUG_ERR, ("Unable to disable script %s on node %u\n", argv[0], options.pnn));
1571 return ret;
1572 }
1573
1574 return 0;
1575 }
1576
1577 /*
1578 display the pnn of the recovery master
1579 */
1580 static int control_recmaster(struct ctdb_context *ctdb, int argc, const char **argv)
1581 {
1582 uint32_t recmaster;
1583 int ret;
1584
1585 ret = ctdb_ctrl_getrecmaster(ctdb, ctdb, TIMELIMIT(), options.pnn, &recmaster);
1586 if (ret != 0) {
1587 DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
1588 return -1;
1589 }
1590 printf("%d\n",recmaster);
1591
1592 return 0;
1593 }
1594
1595 /*
1596 add a tickle to a public address
1597 */
1598 static int control_add_tickle(struct ctdb_context *ctdb, int argc, const char **argv)
1599 {
1600 struct ctdb_tcp_connection t;
1601 TDB_DATA data;
1602 int ret;
1603
1604 assert_single_node_only();
1605
1606 if (argc < 2) {
1607 usage();
1608 }
1609
1610 if (parse_ip_port(argv[0], &t.src_addr) == 0) {
1611 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1612 return -1;
1613 }
1614 if (parse_ip_port(argv[1], &t.dst_addr) == 0) {
1615 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[1]));
1616 return -1;
1617 }
1618
1619 data.dptr = (uint8_t *)&t;
1620 data.dsize = sizeof(t);
1621
1622 /* tell all nodes about this tcp connection */
1623 ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_TCP_ADD_DELAYED_UPDATE,
1624 0, data, ctdb, NULL, NULL, NULL, NULL);
1625 if (ret != 0) {
1626 DEBUG(DEBUG_ERR,("Failed to add tickle\n"));
1627 return -1;
1628 }
1629
1630 return 0;
1631 }
1632
1633
1634 /*
1635 delete a tickle from a node
1636 */
1637 static int control_del_tickle(struct ctdb_context *ctdb, int argc, const char **argv)
1638 {
1639 struct ctdb_tcp_connection t;
1640 TDB_DATA data;
1641 int ret;
1642
1643 assert_single_node_only();
1644
1645 if (argc < 2) {
1646 usage();
1647 }
1648
1649 if (parse_ip_port(argv[0], &t.src_addr) == 0) {
1650 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1651 return -1;
1652 }
1653 if (parse_ip_port(argv[1], &t.dst_addr) == 0) {
1654 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[1]));
1655 return -1;
1656 }
1657
1658 data.dptr = (uint8_t *)&t;
1659 data.dsize = sizeof(t);
1660
1661 /* tell all nodes about this tcp connection */
1662 ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_TCP_REMOVE,
1663 0, data, ctdb, NULL, NULL, NULL, NULL);
1664 if (ret != 0) {
1665 DEBUG(DEBUG_ERR,("Failed to remove tickle\n"));
1666 return -1;
1667 }
1668
1669 return 0;
1670 }
1671
1672
1673 /*
1674 get a list of all tickles for this pnn
1675 */
1676 static int control_get_tickles(struct ctdb_context *ctdb, int argc, const char **argv)
1677 {
1678 struct ctdb_control_tcp_tickle_list *list;
1679 ctdb_sock_addr addr;
1680 int i, ret;
1681 unsigned port = 0;
1682
1683 assert_single_node_only();
1684
1685 if (argc < 1) {
1686 usage();
1687 }
1688
1689 if (argc == 2) {
1690 port = atoi(argv[1]);
1691 }
1692
1693 if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
1694 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1695 return -1;
1696 }
1697
1698 ret = ctdb_ctrl_get_tcp_tickles(ctdb, TIMELIMIT(), options.pnn, ctdb, &addr, &list);
1699 if (ret == -1) {
1700 DEBUG(DEBUG_ERR, ("Unable to list tickles\n"));
1701 return -1;
1702 }
1703
1704 if (options.machinereadable){
1705 printf(":source ip:port:destination ip:port:\n");
1706 for (i=0;i<list->tickles.num;i++) {
1707 if (port && port != ntohs(list->tickles.connections[i].dst_addr.ip.sin_port)) {
1708 continue;
1709 }
1710 printf(":%s:%u", ctdb_addr_to_str(&list->tickles.connections[i].src_addr), ntohs(list->tickles.connections[i].src_addr.ip.sin_port));
1711 printf(":%s:%u:\n", ctdb_addr_to_str(&list->tickles.connections[i].dst_addr), ntohs(list->tickles.connections[i].dst_addr.ip.sin_port));
1712 }
1713 } else {
1714 printf("Tickles for ip:%s\n", ctdb_addr_to_str(&list->addr));
1715 printf("Num tickles:%u\n", list->tickles.num);
1716 for (i=0;i<list->tickles.num;i++) {
1717 if (port && port != ntohs(list->tickles.connections[i].dst_addr.ip.sin_port)) {
1718 continue;
1719 }
1720 printf("SRC: %s:%u ", ctdb_addr_to_str(&list->tickles.connections[i].src_addr), ntohs(list->tickles.connections[i].src_addr.ip.sin_port));
1721 printf("DST: %s:%u\n", ctdb_addr_to_str(&list->tickles.connections[i].dst_addr), ntohs(list->tickles.connections[i].dst_addr.ip.sin_port));
1722 }
1723 }
1724
1725 talloc_free(list);
1726
1727 return 0;
1728 }
1729
1730
1731 static int move_ip(struct ctdb_context *ctdb, ctdb_sock_addr *addr, uint32_t pnn)
1732 {
1733 struct ctdb_all_public_ips *ips;
1734 struct ctdb_public_ip ip;
1735 int i, ret;
1736 uint32_t *nodes;
1737 uint32_t disable_time;
1738 TDB_DATA data;
1739 struct ctdb_node_map *nodemap=NULL;
1740 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1741
1742 disable_time = 30;
1743 data.dptr = (uint8_t*)&disable_time;
1744 data.dsize = sizeof(disable_time);
1745 ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, CTDB_SRVID_DISABLE_IP_CHECK, data);
1746 if (ret != 0) {
1747 DEBUG(DEBUG_ERR,("Failed to send message to disable ipcheck\n"));
1748 return -1;
1749 }
1750
1751
1752
1753 /* read the public ip list from the node */
1754 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), pnn, ctdb, &ips);
1755 if (ret != 0) {
1756 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", pnn));
1757 talloc_free(tmp_ctx);
1758 return -1;
1759 }
1760
1761 for (i=0;i<ips->num;i++) {
1762 if (ctdb_same_ip(addr, &ips->ips[i].addr)) {
1763 break;
1764 }
1765 }
1766 if (i==ips->num) {
1767 DEBUG(DEBUG_ERR, ("Node %u can not host ip address '%s'\n",
1768 pnn, ctdb_addr_to_str(addr)));
1769 talloc_free(tmp_ctx);
1770 return -1;
1771 }
1772
1773 ip.pnn = pnn;
1774 ip.addr = *addr;
1775
1776 data.dptr = (uint8_t *)&ip;
1777 data.dsize = sizeof(ip);
1778
1779 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &nodemap);
1780 if (ret != 0) {
1781 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1782 talloc_free(tmp_ctx);
1783 return ret;
1784 }
1785
1786 nodes = list_of_nodes(ctdb, nodemap, tmp_ctx, NODE_FLAGS_INACTIVE, pnn);
1787 ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_RELEASE_IP,
1788 nodes, 0,
1789 LONGTIMELIMIT(),
1790 false, data,
1791 NULL, NULL,
1792 NULL);
1793 if (ret != 0) {
1794 DEBUG(DEBUG_ERR,("Failed to release IP on nodes\n"));
1795 talloc_free(tmp_ctx);
1796 return -1;
1797 }
1798
1799 ret = ctdb_ctrl_takeover_ip(ctdb, LONGTIMELIMIT(), pnn, &ip);
1800 if (ret != 0) {
1801 DEBUG(DEBUG_ERR,("Failed to take over IP on node %d\n", pnn));
1802 talloc_free(tmp_ctx);
1803 return -1;
1804 }
1805
1806 /* update the recovery daemon so it now knows to expect the new
1807 node assignment for this ip.
1808 */
1809 ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, CTDB_SRVID_RECD_UPDATE_IP, data);
1810 if (ret != 0) {
1811 DEBUG(DEBUG_ERR,("Failed to send message to update the ip on the recovery master.\n"));
1812 return -1;
1813 }
1814
1815 talloc_free(tmp_ctx);
1816 return 0;
1817 }
1818
1819
1820 /*
1821 * scans all other nodes and returns a pnn for another node that can host this
1822 * ip address or -1
1823 */
1824 static int
1825 find_other_host_for_public_ip(struct ctdb_context *ctdb, ctdb_sock_addr *addr)
1826 {
1827 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1828 struct ctdb_all_public_ips *ips;
1829 struct ctdb_node_map *nodemap=NULL;
1830 int i, j, ret;
1831 int pnn;
1832
1833 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
1834 if (ret != 0) {
1835 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
1836 talloc_free(tmp_ctx);
1837 return ret;
1838 }
1839
1840 for(i=0;i<nodemap->num;i++){
1841 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
1842 continue;
1843 }
1844 if (nodemap->nodes[i].pnn == options.pnn) {
1845 continue;
1846 }
1847
1848 /* read the public ip list from this node */
1849 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &ips);
1850 if (ret != 0) {
1851 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", nodemap->nodes[i].pnn));
1852 return -1;
1853 }
1854
1855 for (j=0;j<ips->num;j++) {
1856 if (ctdb_same_ip(addr, &ips->ips[j].addr)) {
1857 pnn = nodemap->nodes[i].pnn;
1858 talloc_free(tmp_ctx);
1859 return pnn;
1860 }
1861 }
1862 talloc_free(ips);
1863 }
1864
1865 talloc_free(tmp_ctx);
1866 return -1;
1867 }
1868
1869 /* If pnn is -1 then try to find a node to move IP to... */
1870 static bool try_moveip(struct ctdb_context *ctdb, ctdb_sock_addr *addr, uint32_t pnn)
1871 {
1872 bool pnn_specified = (pnn == -1 ? false : true);
1873 int retries = 0;
1874
1875 while (retries < 5) {
1876 if (!pnn_specified) {
1877 pnn = find_other_host_for_public_ip(ctdb, addr);
1878 if (pnn == -1) {
1879 return false;
1880 }
1881 DEBUG(DEBUG_NOTICE,
1882 ("Trying to move public IP to node %u\n", pnn));
1883 }
1884
1885 if (move_ip(ctdb, addr, pnn) == 0) {
1886 return true;
1887 }
1888
1889 sleep(3);
1890 retries++;
1891 }
1892
1893 return false;
1894 }
1895
1896
1897 /*
1898 move/failover an ip address to a specific node
1899 */
1900 static int control_moveip(struct ctdb_context *ctdb, int argc, const char **argv)
1901 {
1902 uint32_t pnn;
1903 ctdb_sock_addr addr;
1904
1905 assert_single_node_only();
1906
1907 if (argc < 2) {
1908 usage();
1909 return -1;
1910 }
1911
1912 if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
1913 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
1914 return -1;
1915 }
1916
1917
1918 if (sscanf(argv[1], "%u", &pnn) != 1) {
1919 DEBUG(DEBUG_ERR, ("Badly formed pnn\n"));
1920 return -1;
1921 }
1922
1923 if (!try_moveip(ctdb, &addr, pnn)) {
1924 DEBUG(DEBUG_ERR,("Failed to move IP to node %d.\n", pnn));
1925 return -1;
1926 }
1927
1928 return 0;
1929 }
1930
1931 static int rebalance_node(struct ctdb_context *ctdb, uint32_t pnn)
1932 {
1933 TDB_DATA data;
1934
1935 data.dptr = (uint8_t *)&pnn;
1936 data.dsize = sizeof(uint32_t);
1937 if (ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, CTDB_SRVID_REBALANCE_NODE, data) != 0) {
1938 DEBUG(DEBUG_ERR,
1939 ("Failed to send message to force node %u to be a rebalancing target\n",
1940 pnn));
1941 return -1;
1942 }
1943
1944 return 0;
1945 }
1946
1947
1948 /*
1949 rebalance a node by setting it to allow failback and triggering a
1950 takeover run
1951 */
1952 static int control_rebalancenode(struct ctdb_context *ctdb, int argc, const char **argv)
1953 {
1954 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1955 uint32_t *nodes;
1956 uint32_t pnn_mode;
1957 int i, ret;
1958
1959 assert_single_node_only();
1960
1961 if (argc > 1) {
1962 usage();
1963 }
1964
1965 /* Determine the nodes where IPs need to be reloaded */
1966 if (!parse_nodestring(ctdb, tmp_ctx, argc == 1 ? argv[0] : NULL,
1967 options.pnn, true, &nodes, &pnn_mode)) {
1968 ret = -1;
1969 goto done;
1970 }
1971
1972 for (i = 0; i < talloc_array_length(nodes); i++) {
1973 if (!rebalance_node(ctdb, nodes[i])) {
1974 ret = -1;
1975 }
1976 }
1977
1978 done:
1979 talloc_free(tmp_ctx);
1980 return ret;
1981 }
1982
1983 static int rebalance_ip(struct ctdb_context *ctdb, ctdb_sock_addr *addr)
1984 {
1985 struct ctdb_public_ip ip;
1986 int ret;
1987 uint32_t *nodes;
1988 uint32_t disable_time;
1989 TDB_DATA data;
1990 struct ctdb_node_map *nodemap=NULL;
1991 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
1992
1993 disable_time = 30;
1994 data.dptr = (uint8_t*)&disable_time;
1995 data.dsize = sizeof(disable_time);
1996 ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, CTDB_SRVID_DISABLE_IP_CHECK, data);
1997 if (ret != 0) {
1998 DEBUG(DEBUG_ERR,("Failed to send message to disable ipcheck\n"));
1999 return -1;
2000 }
2001
2002 ip.pnn = -1;
2003 ip.addr = *addr;
2004
2005 data.dptr = (uint8_t *)&ip;
2006 data.dsize = sizeof(ip);
2007
2008 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &nodemap);
2009 if (ret != 0) {
2010 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
2011 talloc_free(tmp_ctx);
2012 return ret;
2013 }
2014
2015 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
2016 ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_RELEASE_IP,
2017 nodes, 0,
2018 LONGTIMELIMIT(),
2019 false, data,
2020 NULL, NULL,
2021 NULL);
2022 if (ret != 0) {
2023 DEBUG(DEBUG_ERR,("Failed to release IP on nodes\n"));
2024 talloc_free(tmp_ctx);
2025 return -1;
2026 }
2027
2028 talloc_free(tmp_ctx);
2029 return 0;
2030 }
2031
2032 /*
2033 release an ip form all nodes and have it re-assigned by recd
2034 */
2035 static int control_rebalanceip(struct ctdb_context *ctdb, int argc, const char **argv)
2036 {
2037 ctdb_sock_addr addr;
2038
2039 assert_single_node_only();
2040
2041 if (argc < 1) {
2042 usage();
2043 return -1;
2044 }
2045
2046 if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
2047 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
2048 return -1;
2049 }
2050
2051 if (rebalance_ip(ctdb, &addr) != 0) {
2052 DEBUG(DEBUG_ERR,("Error when trying to reassign ip\n"));
2053 return -1;
2054 }
2055
2056 return 0;
2057 }
2058
2059 static int getips_store_callback(void *param, void *data)
2060 {
2061 struct ctdb_public_ip *node_ip = (struct ctdb_public_ip *)data;
2062 struct ctdb_all_public_ips *ips = param;
2063 int i;
2064
2065 i = ips->num++;
2066 ips->ips[i].pnn = node_ip->pnn;
2067 ips->ips[i].addr = node_ip->addr;
2068 return 0;
2069 }
2070
2071 static int getips_count_callback(void *param, void *data)
2072 {
2073 uint32_t *count = param;
2074
2075 (*count)++;
2076 return 0;
2077 }
2078
2079 #define IP_KEYLEN 4
2080 static uint32_t *ip_key(ctdb_sock_addr *ip)
2081 {
2082 static uint32_t key[IP_KEYLEN];
2083
2084 bzero(key, sizeof(key));
2085
2086 switch (ip->sa.sa_family) {
2087 case AF_INET:
2088 key[0] = ip->ip.sin_addr.s_addr;
2089 break;
2090 case AF_INET6: {
2091 uint32_t *s6_a32 = (uint32_t *)&(ip->ip6.sin6_addr.s6_addr);
2092 key[0] = s6_a32[3];
2093 key[1] = s6_a32[2];
2094 key[2] = s6_a32[1];
2095 key[3] = s6_a32[0];
2096 break;
2097 }
2098 default:
2099 DEBUG(DEBUG_ERR, (__location__ " ERROR, unknown family passed :%u\n", ip->sa.sa_family));
2100 return key;
2101 }
2102
2103 return key;
2104 }
2105
2106 static void *add_ip_callback(void *parm, void *data)
2107 {
2108 return parm;
2109 }
2110
2111 static int
2112 control_get_all_public_ips(struct ctdb_context *ctdb, TALLOC_CTX *tmp_ctx, struct ctdb_all_public_ips **ips)
2113 {
2114 struct ctdb_all_public_ips *tmp_ips;
2115 struct ctdb_node_map *nodemap=NULL;
2116 trbt_tree_t *ip_tree;
2117 int i, j, len, ret;
2118 uint32_t count;
2119
2120 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
2121 if (ret != 0) {
2122 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
2123 return ret;
2124 }
2125
2126 ip_tree = trbt_create(tmp_ctx, 0);
2127
2128 for(i=0;i<nodemap->num;i++){
2129 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
2130 continue;
2131 }
2132 if (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) {
2133 continue;
2134 }
2135
2136 /* read the public ip list from this node */
2137 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &tmp_ips);
2138 if (ret != 0) {
2139 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", nodemap->nodes[i].pnn));
2140 return -1;
2141 }
2142
2143 for (j=0; j<tmp_ips->num;j++) {
2144 struct ctdb_public_ip *node_ip;
2145
2146 node_ip = talloc(tmp_ctx, struct ctdb_public_ip);
2147 node_ip->pnn = tmp_ips->ips[j].pnn;
2148 node_ip->addr = tmp_ips->ips[j].addr;
2149
2150 trbt_insertarray32_callback(ip_tree,
2151 IP_KEYLEN, ip_key(&tmp_ips->ips[j].addr),
2152 add_ip_callback,
2153 node_ip);
2154 }
2155 talloc_free(tmp_ips);
2156 }
2157
2158 /* traverse */
2159 count = 0;
2160 trbt_traversearray32(ip_tree, IP_KEYLEN, getips_count_callback, &count);
2161
2162 len = offsetof(struct ctdb_all_public_ips, ips) +
2163 count*sizeof(struct ctdb_public_ip);
2164 tmp_ips = talloc_zero_size(tmp_ctx, len);
2165 trbt_traversearray32(ip_tree, IP_KEYLEN, getips_store_callback, tmp_ips);
2166
2167 *ips = tmp_ips;
2168
2169 return 0;
2170 }
2171
2172
2173 static void ctdb_every_second(struct event_context *ev, struct timed_event *te, struct timeval t, void *p)
2174 {
2175 struct ctdb_context *ctdb = talloc_get_type(p, struct ctdb_context);
2176
2177 event_add_timed(ctdb->ev, ctdb,
2178 timeval_current_ofs(1, 0),
2179 ctdb_every_second, ctdb);
2180 }
2181
2182 struct srvid_reply_handler_data {
2183 bool done;
2184 bool wait_for_all;
2185 uint32_t *nodes;
2186 const char *srvid_str;
2187 };
2188
2189 static void srvid_broadcast_reply_handler(struct ctdb_context *ctdb,
2190 uint64_t srvid,
2191 TDB_DATA data,
2192 void *private_data)
2193 {
2194 struct srvid_reply_handler_data *d =
2195 (struct srvid_reply_handler_data *)private_data;
2196 int i;
2197 int32_t ret;
2198
2199 if (data.dsize != sizeof(ret)) {
2200 DEBUG(DEBUG_ERR, (__location__ " Wrong reply size\n"));
2201 return;
2202 }
2203
2204 /* ret will be a PNN (i.e. >=0) on success, or negative on error */
2205 ret = *(int32_t *)data.dptr;
2206 if (ret < 0) {
2207 DEBUG(DEBUG_ERR,
2208 ("%s failed with result %d\n", d->srvid_str, ret));
2209 return;
2210 }
2211
2212 if (!d->wait_for_all) {
2213 d->done = true;
2214 return;
2215 }
2216
2217 /* Wait for all replies */
2218 d->done = true;
2219 for (i = 0; i < talloc_array_length(d->nodes); i++) {
2220 if (d->nodes[i] == ret) {
2221 DEBUG(DEBUG_INFO,
2222 ("%s reply received from node %u\n",
2223 d->srvid_str, ret));
2224 d->nodes[i] = -1;
2225 }
2226 if (d->nodes[i] != -1) {
2227 /* Found a node that hasn't yet replied */
2228 d->done = false;
2229 }
2230 }
2231 }
2232
2233 /* Broadcast the given SRVID to all connected nodes. Wait for 1 reply
2234 * or replies from all connected nodes. arg is the data argument to
2235 * pass in the srvid_request structure - pass 0 if this isn't needed.
2236 */
2237 static int srvid_broadcast(struct ctdb_context *ctdb,
2238 uint64_t srvid, uint32_t *arg,
2239 const char *srvid_str, bool wait_for_all)
2240 {
2241 int ret;
2242 TDB_DATA data;
2243 uint32_t pnn;
2244 uint64_t reply_srvid;
2245 struct srvid_request request;
2246 struct srvid_request_data request_data;
2247 struct srvid_reply_handler_data reply_data;
2248 struct timeval tv;
2249
2250 ZERO_STRUCT(request);
2251
2252 /* Time ticks to enable timeouts to be processed */
2253 event_add_timed(ctdb->ev, ctdb,
2254 timeval_current_ofs(1, 0),
2255 ctdb_every_second, ctdb);
2256
2257 pnn = ctdb_get_pnn(ctdb);
2258 reply_srvid = getpid();
2259
2260 if (arg == NULL) {
2261 request.pnn = pnn;
2262 request.srvid = reply_srvid;
2263
2264 data.dptr = (uint8_t *)&request;
2265 data.dsize = sizeof(request);
2266 } else {
2267 request_data.pnn = pnn;
2268 request_data.srvid = reply_srvid;
2269 request_data.data = *arg;
2270
2271 data.dptr = (uint8_t *)&request_data;
2272 data.dsize = sizeof(request_data);
2273 }
2274
2275 /* Register message port for reply from recovery master */
2276 ctdb_client_set_message_handler(ctdb, reply_srvid,
2277 srvid_broadcast_reply_handler,
2278 &reply_data);
2279
2280 reply_data.wait_for_all = wait_for_all;
2281 reply_data.nodes = NULL;
2282 reply_data.srvid_str = srvid_str;
2283
2284 again:
2285 reply_data.done = false;
2286
2287 if (wait_for_all) {
2288 struct ctdb_node_map *nodemap;
2289
2290 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(),
2291 CTDB_CURRENT_NODE, ctdb, &nodemap);
2292 if (ret != 0) {
2293 DEBUG(DEBUG_ERR,
2294 ("Unable to get nodemap from current node, try again\n"));
2295 sleep(1);
2296 goto again;
2297 }
2298
2299 if (reply_data.nodes != NULL) {
2300 talloc_free(reply_data.nodes);
2301 }
2302 reply_data.nodes = list_of_connected_nodes(ctdb, nodemap,
2303 NULL, true);
2304
2305 talloc_free(nodemap);
2306 }
2307
2308 /* Send to all connected nodes. Only recmaster replies */
2309 ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED,
2310 srvid, data);
2311 if (ret != 0) {
2312 /* This can only happen if the socket is closed and
2313 * there's no way to recover from that, so don't try
2314 * again.
2315 */
2316 DEBUG(DEBUG_ERR,
2317 ("Failed to send %s request to connected nodes\n",
2318 srvid_str));
2319 return -1;
2320 }
2321
2322 tv = timeval_current();
2323 /* This loop terminates the reply is received */
2324 while (timeval_elapsed(&tv) < 5.0 && !reply_data.done) {
2325 event_loop_once(ctdb->ev);
2326 }
2327
2328 if (!reply_data.done) {
2329 DEBUG(DEBUG_NOTICE,
2330 ("Still waiting for confirmation of %s\n", srvid_str));
2331 sleep(1);
2332 goto again;
2333 }
2334
2335 ctdb_client_remove_message_handler(ctdb, reply_srvid, &reply_data);
2336
2337 talloc_free(reply_data.nodes);
2338
2339 return 0;
2340 }
2341
2342 static int ipreallocate(struct ctdb_context *ctdb)
2343 {
2344 return srvid_broadcast(ctdb, CTDB_SRVID_TAKEOVER_RUN, NULL,
2345 "IP reallocation", false);
2346 }
2347
2348
2349 static int control_ipreallocate(struct ctdb_context *ctdb, int argc, const char **argv)
2350 {
2351 return ipreallocate(ctdb);
2352 }
2353
2354 /*
2355 add a public ip address to a node
2356 */
2357 static int control_addip(struct ctdb_context *ctdb, int argc, const char **argv)
2358 {
2359 int i, ret;
2360 int len, retries = 0;
2361 unsigned mask;
2362 ctdb_sock_addr addr;
2363 struct ctdb_control_ip_iface *pub;
2364 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2365 struct ctdb_all_public_ips *ips;
2366
2367
2368 if (argc != 2) {
2369 talloc_free(tmp_ctx);
2370 usage();
2371 }
2372
2373 if (!parse_ip_mask(argv[0], argv[1], &addr, &mask)) {
2374 DEBUG(DEBUG_ERR, ("Badly formed ip/mask : %s\n", argv[0]));
2375 talloc_free(tmp_ctx);
2376 return -1;
2377 }
2378
2379 /* read the public ip list from the node */
2380 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ips);
2381 if (ret != 0) {
2382 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %u\n", options.pnn));
2383 talloc_free(tmp_ctx);
2384 return -1;
2385 }
2386 for (i=0;i<ips->num;i++) {
2387 if (ctdb_same_ip(&addr, &ips->ips[i].addr)) {
2388 DEBUG(DEBUG_ERR,("Can not add ip to node. Node already hosts this ip\n"));
2389 return 0;
2390 }
2391 }
2392
2393
2394
2395 /* Dont timeout. This command waits for an ip reallocation
2396 which sometimes can take wuite a while if there has
2397 been a recent recovery
2398 */
2399 alarm(0);
2400
2401 len = offsetof(struct ctdb_control_ip_iface, iface) + strlen(argv[1]) + 1;
2402 pub = talloc_size(tmp_ctx, len);
2403 CTDB_NO_MEMORY(ctdb, pub);
2404
2405 pub->addr = addr;
2406 pub->mask = mask;
2407 pub->len = strlen(argv[1])+1;
2408 memcpy(&pub->iface[0], argv[1], strlen(argv[1])+1);
2409
2410 do {
2411 ret = ctdb_ctrl_add_public_ip(ctdb, TIMELIMIT(), options.pnn, pub);
2412 if (ret != 0) {
2413 DEBUG(DEBUG_ERR, ("Unable to add public ip to node %u. Wait 3 seconds and try again.\n", options.pnn));
2414 sleep(3);
2415 retries++;
2416 }
2417 } while (retries < 5 && ret != 0);
2418 if (ret != 0) {
2419 DEBUG(DEBUG_ERR, ("Unable to add public ip to node %u. Giving up.\n", options.pnn));
2420 talloc_free(tmp_ctx);
2421 return ret;
2422 }
2423
2424 if (rebalance_node(ctdb, options.pnn) != 0) {
2425 DEBUG(DEBUG_ERR,("Error when trying to rebalance node\n"));
2426 return ret;
2427 }
2428
2429 talloc_free(tmp_ctx);
2430 return 0;
2431 }
2432
2433 /*
2434 add a public ip address to a node
2435 */
2436 static int control_ipiface(struct ctdb_context *ctdb, int argc, const char **argv)
2437 {
2438 ctdb_sock_addr addr;
2439
2440 if (argc != 1) {
2441 usage();
2442 }
2443
2444 if (!parse_ip(argv[0], NULL, 0, &addr)) {
2445 printf("Badly formed ip : %s\n", argv[0]);
2446 return -1;
2447 }
2448
2449 printf("IP on interface %s\n", ctdb_sys_find_ifname(&addr));
2450
2451 return 0;
2452 }
2453
2454 static int control_delip(struct ctdb_context *ctdb, int argc, const char **argv);
2455
2456 static int control_delip_all(struct ctdb_context *ctdb, int argc, const char **argv, ctdb_sock_addr *addr)
2457 {
2458 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2459 struct ctdb_node_map *nodemap=NULL;
2460 struct ctdb_all_public_ips *ips;
2461 int ret, i, j;
2462
2463 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, tmp_ctx, &nodemap);
2464 if (ret != 0) {
2465 DEBUG(DEBUG_ERR, ("Unable to get nodemap from current node\n"));
2466 return ret;
2467 }
2468
2469 /* remove it from the nodes that are not hosting the ip currently */
2470 for(i=0;i<nodemap->num;i++){
2471 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
2472 continue;
2473 }
2474 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &ips);
2475 if (ret != 0) {
2476 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %d\n", nodemap->nodes[i].pnn));
2477 continue;
2478 }
2479
2480 for (j=0;j<ips->num;j++) {
2481 if (ctdb_same_ip(addr, &ips->ips[j].addr)) {
2482 break;
2483 }
2484 }
2485 if (j==ips->num) {
2486 continue;
2487 }
2488
2489 if (ips->ips[j].pnn == nodemap->nodes[i].pnn) {
2490 continue;
2491 }
2492
2493 options.pnn = nodemap->nodes[i].pnn;
2494 control_delip(ctdb, argc, argv);
2495 }
2496
2497
2498 /* remove it from every node (also the one hosting it) */
2499 for(i=0;i<nodemap->num;i++){
2500 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
2501 continue;
2502 }
2503 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), nodemap->nodes[i].pnn, tmp_ctx, &ips);
2504 if (ret != 0) {
2505 DEBUG(DEBUG_ERR, ("Unable to get public ip list from node %d\n", nodemap->nodes[i].pnn));
2506 continue;
2507 }
2508
2509 for (j=0;j<ips->num;j++) {
2510 if (ctdb_same_ip(addr, &ips->ips[j].addr)) {
2511 break;
2512 }
2513 }
2514 if (j==ips->num) {
2515 continue;
2516 }
2517
2518 options.pnn = nodemap->nodes[i].pnn;
2519 control_delip(ctdb, argc, argv);
2520 }
2521
2522 talloc_free(tmp_ctx);
2523 return 0;
2524 }
2525
2526 /*
2527 delete a public ip address from a node
2528 */
2529 static int control_delip(struct ctdb_context *ctdb, int argc, const char **argv)
2530 {
2531 int i, ret;
2532 ctdb_sock_addr addr;
2533 struct ctdb_control_ip_iface pub;
2534 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2535 struct ctdb_all_public_ips *ips;
2536
2537 if (argc != 1) {
2538 talloc_free(tmp_ctx);
2539 usage();
2540 }
2541
2542 if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
2543 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
2544 return -1;
2545 }
2546
2547 if (options.pnn == CTDB_BROADCAST_ALL) {
2548 return control_delip_all(ctdb, argc, argv, &addr);
2549 }
2550
2551 pub.addr = addr;
2552 pub.mask = 0;
2553 pub.len = 0;
2554
2555 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ips);
2556 if (ret != 0) {
2557 DEBUG(DEBUG_ERR, ("Unable to get public ip list from cluster\n"));
2558 talloc_free(tmp_ctx);
2559 return ret;
2560 }
2561
2562 for (i=0;i<ips->num;i++) {
2563 if (ctdb_same_ip(&addr, &ips->ips[i].addr)) {
2564 break;
2565 }
2566 }
2567
2568 if (i==ips->num) {
2569 DEBUG(DEBUG_ERR, ("This node does not support this public address '%s'\n",
2570 ctdb_addr_to_str(&addr)));
2571 talloc_free(tmp_ctx);
2572 return -1;
2573 }
2574
2575 /* This is an optimisation. If this node is hosting the IP
2576 * then try to move it somewhere else without invoking a full
2577 * takeover run. We don't care if this doesn't work!
2578 */
2579 if (ips->ips[i].pnn == options.pnn) {
2580 (void) try_moveip(ctdb, &addr, -1);
2581 }
2582
2583 ret = ctdb_ctrl_del_public_ip(ctdb, TIMELIMIT(), options.pnn, &pub);
2584 if (ret != 0) {
2585 DEBUG(DEBUG_ERR, ("Unable to del public ip from node %u\n", options.pnn));
2586 talloc_free(tmp_ctx);
2587 return ret;
2588 }
2589
2590 talloc_free(tmp_ctx);
2591 return 0;
2592 }
2593
2594 static int kill_tcp_from_file(struct ctdb_context *ctdb,
2595 int argc, const char **argv)
2596 {
2597 struct ctdb_control_killtcp *killtcp;
2598 int max_entries, current, i;
2599 struct timeval timeout;
2600 char line[128], src[128], dst[128];
2601 int linenum;
2602 TDB_DATA data;
2603 struct client_async_data *async_data;
2604 struct ctdb_client_control_state *state;
2605
2606 if (argc != 0) {
2607 usage();
2608 }
2609
2610 linenum = 1;
2611 killtcp = NULL;
2612 max_entries = 0;
2613 current = 0;
2614 while (!feof(stdin)) {
2615 if (fgets(line, sizeof(line), stdin) == NULL) {
2616 continue;
2617 }
2618
2619 /* Silently skip empty lines */
2620 if (line[0] == '\n') {
2621 continue;
2622 }
2623
2624 if (sscanf(line, "%s %s\n", src, dst) != 2) {
2625 DEBUG(DEBUG_ERR, ("Bad line [%d]: '%s'\n",
2626 linenum, line));
2627 talloc_free(killtcp);
2628 return -1;
2629 }
2630
2631 if (current >= max_entries) {
2632 max_entries += 1024;
2633 killtcp = talloc_realloc(ctdb, killtcp,
2634 struct ctdb_control_killtcp,
2635 max_entries);
2636 CTDB_NO_MEMORY(ctdb, killtcp);
2637 }
2638
2639 if (!parse_ip_port(src, &killtcp[current].src_addr)) {
2640 DEBUG(DEBUG_ERR, ("Bad IP:port on line [%d]: '%s'\n",
2641 linenum, src));
2642 talloc_free(killtcp);
2643 return -1;
2644 }
2645
2646 if (!parse_ip_port(dst, &killtcp[current].dst_addr)) {
2647 DEBUG(DEBUG_ERR, ("Bad IP:port on line [%d]: '%s'\n",
2648 linenum, dst));
2649 talloc_free(killtcp);
2650 return -1;
2651 }
2652
2653 current++;
2654 }
2655
2656 async_data = talloc_zero(ctdb, struct client_async_data);
2657 if (async_data == NULL) {
2658 talloc_free(killtcp);
2659 return -1;
2660 }
2661
2662 for (i = 0; i < current; i++) {
2663
2664 data.dsize = sizeof(struct ctdb_control_killtcp);
2665 data.dptr = (unsigned char *)&killtcp[i];
2666
2667 timeout = TIMELIMIT();
2668 state = ctdb_control_send(ctdb, options.pnn, 0,
2669 CTDB_CONTROL_KILL_TCP, 0, data,
2670 async_data, &timeout, NULL);
2671
2672 if (state == NULL) {
2673 DEBUG(DEBUG_ERR,
2674 ("Failed to call async killtcp control to node %u\n",
2675 options.pnn));
2676 talloc_free(killtcp);
2677 return -1;
2678 }
2679
2680 ctdb_client_async_add(async_data, state);
2681 }
2682
2683 if (ctdb_client_async_wait(ctdb, async_data) != 0) {
2684 DEBUG(DEBUG_ERR,("killtcp failed\n"));
2685 talloc_free(killtcp);
2686 return -1;
2687 }
2688
2689 talloc_free(killtcp);
2690 return 0;
2691 }
2692
2693
2694 /*
2695 kill a tcp connection
2696 */
2697 static int kill_tcp(struct ctdb_context *ctdb, int argc, const char **argv)
2698 {
2699 int ret;
2700 struct ctdb_control_killtcp killtcp;
2701
2702 assert_single_node_only();
2703
2704 if (argc == 0) {
2705 return kill_tcp_from_file(ctdb, argc, argv);
2706 }
2707
2708 if (argc < 2) {
2709 usage();
2710 }
2711
2712 if (!parse_ip_port(argv[0], &killtcp.src_addr)) {
2713 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[0]));
2714 return -1;
2715 }
2716
2717 if (!parse_ip_port(argv[1], &killtcp.dst_addr)) {
2718 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[1]));
2719 return -1;
2720 }
2721
2722 ret = ctdb_ctrl_killtcp(ctdb, TIMELIMIT(), options.pnn, &killtcp);
2723 if (ret != 0) {
2724 DEBUG(DEBUG_ERR, ("Unable to killtcp from node %u\n", options.pnn));
2725 return ret;
2726 }
2727
2728 return 0;
2729 }
2730
2731
2732 /*
2733 send a gratious arp
2734 */
2735 static int control_gratious_arp(struct ctdb_context *ctdb, int argc, const char **argv)
2736 {
2737 int ret;
2738 ctdb_sock_addr addr;
2739
2740 assert_single_node_only();
2741
2742 if (argc < 2) {
2743 usage();
2744 }
2745
2746 if (!parse_ip(argv[0], NULL, 0, &addr)) {
2747 DEBUG(DEBUG_ERR, ("Bad IP '%s'\n", argv[0]));
2748 return -1;
2749 }
2750
2751 ret = ctdb_ctrl_gratious_arp(ctdb, TIMELIMIT(), options.pnn, &addr, argv[1]);
2752 if (ret != 0) {
2753 DEBUG(DEBUG_ERR, ("Unable to send gratious_arp from node %u\n", options.pnn));
2754 return ret;
2755 }
2756
2757 return 0;
2758 }
2759
2760 /*
2761 register a server id
2762 */
2763 static int regsrvid(struct ctdb_context *ctdb, int argc, const char **argv)
2764 {
2765 int ret;
2766 struct ctdb_server_id server_id;
2767
2768 if (argc < 3) {
2769 usage();
2770 }
2771
2772 server_id.pnn = strtoul(argv[0], NULL, 0);
2773 server_id.type = strtoul(argv[1], NULL, 0);
2774 server_id.server_id = strtoul(argv[2], NULL, 0);
2775
2776 ret = ctdb_ctrl_register_server_id(ctdb, TIMELIMIT(), &server_id);
2777 if (ret != 0) {
2778 DEBUG(DEBUG_ERR, ("Unable to register server_id from node %u\n", options.pnn));
2779 return ret;
2780 }
2781 DEBUG(DEBUG_ERR,("Srvid registered. Sleeping for 999 seconds\n"));
2782 sleep(999);
2783 return -1;
2784 }
2785
2786 /*
2787 unregister a server id
2788 */
2789 static int unregsrvid(struct ctdb_context *ctdb, int argc, const char **argv)
2790 {
2791 int ret;
2792 struct ctdb_server_id server_id;
2793
2794 if (argc < 3) {
2795 usage();
2796 }
2797
2798 server_id.pnn = strtoul(argv[0], NULL, 0);
2799 server_id.type = strtoul(argv[1], NULL, 0);
2800 server_id.server_id = strtoul(argv[2], NULL, 0);
2801
2802 ret = ctdb_ctrl_unregister_server_id(ctdb, TIMELIMIT(), &server_id);
2803 if (ret != 0) {
2804 DEBUG(DEBUG_ERR, ("Unable to unregister server_id from node %u\n", options.pnn));
2805 return ret;
2806 }
2807 return -1;
2808 }
2809
2810 /*
2811 check if a server id exists
2812 */
2813 static int chksrvid(struct ctdb_context *ctdb, int argc, const char **argv)
2814 {
2815 uint32_t status;
2816 int ret;
2817 struct ctdb_server_id server_id;
2818
2819 if (argc < 3) {
2820 usage();
2821 }
2822
2823 server_id.pnn = strtoul(argv[0], NULL, 0);
2824 server_id.type = strtoul(argv[1], NULL, 0);
2825 server_id.server_id = strtoul(argv[2], NULL, 0);
2826
2827 ret = ctdb_ctrl_check_server_id(ctdb, TIMELIMIT(), options.pnn, &server_id, &status);
2828 if (ret != 0) {
2829 DEBUG(DEBUG_ERR, ("Unable to check server_id from node %u\n", options.pnn));
2830 return ret;
2831 }
2832
2833 if (status) {
2834 printf("Server id %d:%d:%d EXISTS\n", server_id.pnn, server_id.type, server_id.server_id);
2835 } else {
2836 printf("Server id %d:%d:%d does NOT exist\n", server_id.pnn, server_id.type, server_id.server_id);
2837 }
2838 return 0;
2839 }
2840
2841 /*
2842 get a list of all server ids that are registered on a node
2843 */
2844 static int getsrvids(struct ctdb_context *ctdb, int argc, const char **argv)
2845 {
2846 int i, ret;
2847 struct ctdb_server_id_list *server_ids;
2848
2849 ret = ctdb_ctrl_get_server_id_list(ctdb, ctdb, TIMELIMIT(), options.pnn, &server_ids);
2850 if (ret != 0) {
2851 DEBUG(DEBUG_ERR, ("Unable to get server_id list from node %u\n", options.pnn));
2852 return ret;
2853 }
2854
2855 for (i=0; i<server_ids->num; i++) {
2856 printf("Server id %d:%d:%d\n",
2857 server_ids->server_ids[i].pnn,
2858 server_ids->server_ids[i].type,
2859 server_ids->server_ids[i].server_id);
2860 }
2861
2862 return -1;
2863 }
2864
2865 /*
2866 check if a server id exists
2867 */
2868 static int check_srvids(struct ctdb_context *ctdb, int argc, const char **argv)
2869 {
2870 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
2871 uint64_t *ids;
2872 uint8_t *result;
2873 int i;
2874
2875 if (argc < 1) {
2876 talloc_free(tmp_ctx);
2877 usage();
2878 }
2879
2880 ids = talloc_array(tmp_ctx, uint64_t, argc);
2881 result = talloc_array(tmp_ctx, uint8_t, argc);
2882
2883 for (i = 0; i < argc; i++) {
2884 ids[i] = strtoull(argv[i], NULL, 0);
2885 }
2886
2887 if (!ctdb_client_check_message_handlers(ctdb, ids, argc, result)) {
2888 DEBUG(DEBUG_ERR, ("Unable to check server_id from node %u\n",
2889 options.pnn));
2890 talloc_free(tmp_ctx);
2891 return -1;
2892 }
2893
2894 for (i=0; i < argc; i++) {
2895 printf("Server id %d:%llu %s\n", options.pnn, (long long)ids[i],
2896 result[i] ? "exists" : "does not exist");
2897 }
2898
2899 talloc_free(tmp_ctx);
2900 return 0;
2901 }
2902
2903 /*
2904 send a tcp tickle ack
2905 */
2906 static int tickle_tcp(struct ctdb_context *ctdb, int argc, const char **argv)
2907 {
2908 int ret;
2909 ctdb_sock_addr src, dst;
2910
2911 if (argc < 2) {
2912 usage();
2913 }
2914
2915 if (!parse_ip_port(argv[0], &src)) {
2916 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[0]));
2917 return -1;
2918 }
2919
2920 if (!parse_ip_port(argv[1], &dst)) {
2921 DEBUG(DEBUG_ERR, ("Bad IP:port '%s'\n", argv[1]));
2922 return -1;
2923 }
2924
2925 ret = ctdb_sys_send_tcp(&src, &dst, 0, 0, 0);
2926 if (ret==0) {
2927 return 0;
2928 }
2929 DEBUG(DEBUG_ERR, ("Error while sending tickle ack\n"));
2930
2931 return -1;
2932 }
2933
2934
2935 /*
2936 display public ip status
2937 */
2938 static int control_ip(struct ctdb_context *ctdb, int argc, const char **argv)
2939 {
2940 int i, ret;
2941 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
2942 struct ctdb_all_public_ips *ips;
2943
2944 if (options.pnn == CTDB_BROADCAST_ALL) {
2945 /* read the list of public ips from all nodes */
2946 ret = control_get_all_public_ips(ctdb, tmp_ctx, &ips);
2947 } else {
2948 /* read the public ip list from this node */
2949 ret = ctdb_ctrl_get_public_ips(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ips);
2950 }
2951 if (ret != 0) {
2952 DEBUG(DEBUG_ERR, ("Unable to get public ips from node %u\n", options.pnn));
2953 talloc_free(tmp_ctx);
2954 return ret;
2955 }
2956
2957 if (options.machinereadable){
2958 printf(":Public IP:Node:");
2959 if (options.verbose){
2960 printf("ActiveInterface:AvailableInterfaces:ConfiguredInterfaces:");
2961 }
2962 printf("\n");
2963 } else {
2964 if (options.pnn == CTDB_BROADCAST_ALL) {
2965 printf("Public IPs on ALL nodes\n");
2966 } else {
2967 printf("Public IPs on node %u\n", options.pnn);
2968 }
2969 }
2970
2971 for (i=1;i<=ips->num;i++) {
2972 struct ctdb_control_public_ip_info *info = NULL;
2973 int32_t pnn;
2974 char *aciface = NULL;
2975 char *avifaces = NULL;
2976 char *cifaces = NULL;
2977
2978 if (options.pnn == CTDB_BROADCAST_ALL) {
2979 pnn = ips->ips[ips->num-i].pnn;
2980 } else {
2981 pnn = options.pnn;
2982 }
2983
2984 if (pnn != -1) {
2985 ret = ctdb_ctrl_get_public_ip_info(ctdb, TIMELIMIT(), pnn, ctdb,
2986 &ips->ips[ips->num-i].addr, &info);
2987 } else {
2988 ret = -1;
2989 }
2990
2991 if (ret == 0) {
2992 int j;
2993 for (j=0; j < info->num; j++) {
2994 if (cifaces == NULL) {
2995 cifaces = talloc_strdup(info,
2996 info->ifaces[j].name);
2997 } else {
2998 cifaces = talloc_asprintf_append(cifaces,
2999 ",%s",
3000 info->ifaces[j].name);
3001 }
3002
3003 if (info->active_idx == j) {
3004 aciface = info->ifaces[j].name;
3005 }
3006
3007 if (info->ifaces[j].link_state == 0) {
3008 continue;
3009 }
3010
3011 if (avifaces == NULL) {
3012 avifaces = talloc_strdup(info, info->ifaces[j].name);
3013 } else {
3014 avifaces = talloc_asprintf_append(avifaces,
3015 ",%s",
3016 info->ifaces[j].name);
3017 }
3018 }
3019 }
3020
3021 if (options.machinereadable){
3022 printf(":%s:%d:",
3023 ctdb_addr_to_str(&ips->ips[ips->num-i].addr),
3024 ips->ips[ips->num-i].pnn);
3025 if (options.verbose){
3026 printf("%s:%s:%s:",
3027 aciface?aciface:"",
3028 avifaces?avifaces:"",
3029 cifaces?cifaces:"");
3030 }
3031 printf("\n");
3032 } else {
3033 if (options.verbose) {
3034 printf("%s node[%d] active[%s] available[%s] configured[%s]\n",
3035 ctdb_addr_to_str(&ips->ips[ips->num-i].addr),
3036 ips->ips[ips->num-i].pnn,
3037 aciface?aciface:"",
3038 avifaces?avifaces:"",
3039 cifaces?cifaces:"");
3040 } else {
3041 printf("%s %d\n",
3042 ctdb_addr_to_str(&ips->ips[ips->num-i].addr),
3043 ips->ips[ips->num-i].pnn);
3044 }
3045 }
3046 talloc_free(info);
3047 }
3048
3049 talloc_free(tmp_ctx);
3050 return 0;
3051 }
3052
3053 /*
3054 public ip info
3055 */
3056 static int control_ipinfo(struct ctdb_context *ctdb, int argc, const char **argv)
3057 {
3058 int i, ret;
3059 ctdb_sock_addr addr;
3060 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3061 struct ctdb_control_public_ip_info *info;
3062
3063 if (argc != 1) {
3064 talloc_free(tmp_ctx);
3065 usage();
3066 }
3067
3068 if (parse_ip(argv[0], NULL, 0, &addr) == 0) {
3069 DEBUG(DEBUG_ERR,("Wrongly formed ip address '%s'\n", argv[0]));
3070 return -1;
3071 }
3072
3073 /* read the public ip info from this node */
3074 ret = ctdb_ctrl_get_public_ip_info(ctdb, TIMELIMIT(), options.pnn,
3075 tmp_ctx, &addr, &info);
3076 if (ret != 0) {
3077 DEBUG(DEBUG_ERR, ("Unable to get public ip[%s]info from node %u\n",
3078 argv[0], options.pnn));
3079 talloc_free(tmp_ctx);
3080 return ret;
3081 }
3082
3083 printf("Public IP[%s] info on node %u\n",
3084 ctdb_addr_to_str(&info->ip.addr),
3085 options.pnn);
3086
3087 printf("IP:%s\nCurrentNode:%d\nNumInterfaces:%u\n",
3088 ctdb_addr_to_str(&info->ip.addr),
3089 info->ip.pnn, info->num);
3090
3091 for (i=0; i<info->num; i++) {
3092 info->ifaces[i].name[CTDB_IFACE_SIZE] = '\0';
3093
3094 printf("Interface[%u]: Name:%s Link:%s References:%u%s\n",
3095 i+1, info->ifaces[i].name,
3096 info->ifaces[i].link_state?"up":"down",
3097 (unsigned int)info->ifaces[i].references,
3098 (i==info->active_idx)?" (active)":"");
3099 }
3100
3101 talloc_free(tmp_ctx);
3102 return 0;
3103 }
3104
3105 /*
3106 display interfaces status
3107 */
3108 static int control_ifaces(struct ctdb_context *ctdb, int argc, const char **argv)
3109 {
3110 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3111 int i;
3112 struct ctdb_control_get_ifaces *ifaces;
3113 int ret;
3114
3115 /* read the public ip list from this node */
3116 ret = ctdb_ctrl_get_ifaces(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &ifaces);
3117 if (ret != 0) {
3118 DEBUG(DEBUG_ERR, ("Unable to get interfaces from node %u\n",
3119 options.pnn));
3120 talloc_free(tmp_ctx);
3121 return -1;
3122 }
3123
3124 if (options.machinereadable){
3125 printf(":Name:LinkStatus:References:\n");
3126 } else {
3127 printf("Interfaces on node %u\n", options.pnn);
3128 }
3129
3130 for (i=0; i<ifaces->num; i++) {
3131 if (options.machinereadable){
3132 printf(":%s:%s:%u\n",
3133 ifaces->ifaces[i].name,
3134 ifaces->ifaces[i].link_state?"1":"0",
3135 (unsigned int)ifaces->ifaces[i].references);
3136 } else {
3137 printf("name:%s link:%s references:%u\n",
3138 ifaces->ifaces[i].name,
3139 ifaces->ifaces[i].link_state?"up":"down",
3140 (unsigned int)ifaces->ifaces[i].references);
3141 }
3142 }
3143
3144 talloc_free(tmp_ctx);
3145 return 0;
3146 }
3147
3148
3149 /*
3150 set link status of an interface
3151 */
3152 static int control_setifacelink(struct ctdb_context *ctdb, int argc, const char **argv)
3153 {
3154 int ret;
3155 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3156 struct ctdb_control_iface_info info;
3157
3158 ZERO_STRUCT(info);
3159
3160 if (argc != 2) {
3161 usage();
3162 }
3163
3164 if (strlen(argv[0]) > CTDB_IFACE_SIZE) {
3165 DEBUG(DEBUG_ERR, ("interfaces name '%s' too long\n",
3166 argv[0]));
3167 talloc_free(tmp_ctx);
3168 return -1;
3169 }
3170 strcpy(info.name, argv[0]);
3171
3172 if (strcmp(argv[1], "up") == 0) {
3173 info.link_state = 1;
3174 } else if (strcmp(argv[1], "down") == 0) {
3175 info.link_state = 0;
3176 } else {
3177 DEBUG(DEBUG_ERR, ("link state invalid '%s' should be 'up' or 'down'\n",
3178 argv[1]));
3179 talloc_free(tmp_ctx);
3180 return -1;
3181 }
3182
3183 /* read the public ip list from this node */
3184 ret = ctdb_ctrl_set_iface_link(ctdb, TIMELIMIT(), options.pnn,
3185 tmp_ctx, &info);
3186 if (ret != 0) {
3187 DEBUG(DEBUG_ERR, ("Unable to set link state for interfaces %s node %u\n",
3188 argv[0], options.pnn));
3189 talloc_free(tmp_ctx);
3190 return ret;
3191 }
3192
3193 talloc_free(tmp_ctx);
3194 return 0;
3195 }
3196
3197 /*
3198 display pid of a ctdb daemon
3199 */
3200 static int control_getpid(struct ctdb_context *ctdb, int argc, const char **argv)
3201 {
3202 uint32_t pid;
3203 int ret;
3204
3205 ret = ctdb_ctrl_getpid(ctdb, TIMELIMIT(), options.pnn, &pid);
3206 if (ret != 0) {
3207 DEBUG(DEBUG_ERR, ("Unable to get daemon pid from node %u\n", options.pnn));
3208 return ret;
3209 }
3210 printf("Pid:%d\n", pid);
3211
3212 return 0;
3213 }
3214
3215 typedef bool update_flags_handler_t(struct ctdb_context *ctdb, void *data);
3216
3217 static int update_flags_and_ipreallocate(struct ctdb_context *ctdb,
3218 void *data,
3219 update_flags_handler_t handler,
3220 uint32_t flag,
3221 const char *desc,
3222 bool set_flag)
3223 {
3224 struct ctdb_node_map *nodemap = NULL;
3225 bool flag_is_set;
3226 int ret;
3227
3228 /* Check if the node is already in the desired state */
3229 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
3230 if (ret != 0) {
3231 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
3232 exit(10);
3233 }
3234 flag_is_set = nodemap->nodes[options.pnn].flags & flag;
3235 if (set_flag == flag_is_set) {
3236 DEBUG(DEBUG_NOTICE, ("Node %d is %s %s\n", options.pnn,
3237 (set_flag ? "already" : "not"), desc));
3238 return 0;
3239 }
3240
3241 do {
3242 if (!handler(ctdb, data)) {
3243 DEBUG(DEBUG_WARNING,
3244 ("Failed to send control to set state %s on node %u, try again\n",
3245 desc, options.pnn));
3246 }
3247
3248 sleep(1);
3249
3250 /* Read the nodemap and verify the change took effect.
3251 * Even if the above control/hanlder timed out then it
3252 * could still have worked!
3253 */
3254 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE,
3255 ctdb, &nodemap);
3256 if (ret != 0) {
3257 DEBUG(DEBUG_WARNING,
3258 ("Unable to get nodemap from local node, try again\n"));
3259 }
3260 flag_is_set = nodemap->nodes[options.pnn].flags & flag;
3261 } while (nodemap == NULL || (set_flag != flag_is_set));
3262
3263 return ipreallocate(ctdb);
3264 }
3265
3266 /* Administratively disable a node */
3267 static bool update_flags_disabled(struct ctdb_context *ctdb, void *data)
3268 {
3269 int ret;
3270
3271 ret = ctdb_ctrl_modflags(ctdb, TIMELIMIT(), options.pnn,
3272 NODE_FLAGS_PERMANENTLY_DISABLED, 0);
3273 return ret == 0;
3274 }
3275
3276 static int control_disable(struct ctdb_context *ctdb, int argc, const char **argv)
3277 {
3278 return update_flags_and_ipreallocate(ctdb, NULL,
3279 update_flags_disabled,
3280 NODE_FLAGS_PERMANENTLY_DISABLED,
3281 "disabled",
3282 true /* set_flag*/);
3283 }
3284
3285 /* Administratively re-enable a node */
3286 static bool update_flags_not_disabled(struct ctdb_context *ctdb, void *data)
3287 {
3288 int ret;
3289
3290 ret = ctdb_ctrl_modflags(ctdb, TIMELIMIT(), options.pnn,
3291 0, NODE_FLAGS_PERMANENTLY_DISABLED);
3292 return ret == 0;
3293 }
3294
3295 static int control_enable(struct ctdb_context *ctdb, int argc, const char **argv)
3296 {
3297 return update_flags_and_ipreallocate(ctdb, NULL,
3298 update_flags_not_disabled,
3299 NODE_FLAGS_PERMANENTLY_DISABLED,
3300 "disabled",
3301 false /* set_flag*/);
3302 }
3303
3304 /* Stop a node */
3305 static bool update_flags_stopped(struct ctdb_context *ctdb, void *data)
3306 {
3307 int ret;
3308
3309 ret = ctdb_ctrl_stop_node(ctdb, TIMELIMIT(), options.pnn);
3310
3311 return ret == 0;
3312 }
3313
3314 static int control_stop(struct ctdb_context *ctdb, int argc, const char **argv)
3315 {
3316 return update_flags_and_ipreallocate(ctdb, NULL,
3317 update_flags_stopped,
3318 NODE_FLAGS_STOPPED,
3319 "stopped",
3320 true /* set_flag*/);
3321 }
3322
3323 /* Continue a stopped node */
3324 static bool update_flags_not_stopped(struct ctdb_context *ctdb, void *data)
3325 {
3326 int ret;
3327
3328 ret = ctdb_ctrl_continue_node(ctdb, TIMELIMIT(), options.pnn);
3329
3330 return ret == 0;
3331 }
3332
3333 static int control_continue(struct ctdb_context *ctdb, int argc, const char **argv)
3334 {
3335 return update_flags_and_ipreallocate(ctdb, NULL,
3336 update_flags_not_stopped,
3337 NODE_FLAGS_STOPPED,
3338 "stopped",
3339 false /* set_flag */);
3340 }
3341
3342 static uint32_t get_generation(struct ctdb_context *ctdb)
3343 {
3344 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3345 struct ctdb_vnn_map *vnnmap=NULL;
3346 int ret;
3347 uint32_t generation;
3348
3349 /* wait until the recmaster is not in recovery mode */
3350 while (1) {
3351 uint32_t recmode, recmaster;
3352
3353 if (vnnmap != NULL) {
3354 talloc_free(vnnmap);
3355 vnnmap = NULL;
3356 }
3357
3358 /* get the recmaster */
3359 ret = ctdb_ctrl_getrecmaster(ctdb, tmp_ctx, TIMELIMIT(), CTDB_CURRENT_NODE, &recmaster);
3360 if (ret != 0) {
3361 DEBUG(DEBUG_ERR, ("Unable to get recmaster from node %u\n", options.pnn));
3362 talloc_free(tmp_ctx);
3363 exit(10);
3364 }
3365
3366 /* get recovery mode */
3367 ret = ctdb_ctrl_getrecmode(ctdb, tmp_ctx, TIMELIMIT(), recmaster, &recmode);
3368 if (ret != 0) {
3369 DEBUG(DEBUG_ERR, ("Unable to get recmode from node %u\n", options.pnn));
3370 talloc_free(tmp_ctx);
3371 exit(10);
3372 }
3373
3374 /* get the current generation number */
3375 ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), recmaster, tmp_ctx, &vnnmap);
3376 if (ret != 0) {
3377 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from recmaster (%u)\n", recmaster));
3378 talloc_free(tmp_ctx);
3379 exit(10);
3380 }
3381
3382 if ((recmode == CTDB_RECOVERY_NORMAL) && (vnnmap->generation != 1)) {
3383 generation = vnnmap->generation;
3384 talloc_free(tmp_ctx);
3385 return generation;
3386 }
3387 sleep(1);
3388 }
3389 }
3390
3391 /* Ban a node */
3392 static bool update_state_banned(struct ctdb_context *ctdb, void *data)
3393 {
3394 struct ctdb_ban_time *bantime = (struct ctdb_ban_time *)data;
3395 int ret;
3396
3397 ret = ctdb_ctrl_set_ban(ctdb, TIMELIMIT(), options.pnn, bantime);
3398
3399 return ret == 0;
3400 }
3401
3402 static int control_ban(struct ctdb_context *ctdb, int argc, const char **argv)
3403 {
3404 struct ctdb_ban_time bantime;
3405
3406 if (argc < 1) {
3407 usage();
3408 }
3409
3410 bantime.pnn = options.pnn;
3411 bantime.time = strtoul(argv[0], NULL, 0);
3412
3413 if (bantime.time == 0) {
3414 DEBUG(DEBUG_ERR, ("Invalid ban time specified - must be >0\n"));
3415 return -1;
3416 }
3417
3418 return update_flags_and_ipreallocate(ctdb, &bantime,
3419 update_state_banned,
3420 NODE_FLAGS_BANNED,
3421 "banned",
3422 true /* set_flag*/);
3423 }
3424
3425
3426 /* Unban a node */
3427 static int control_unban(struct ctdb_context *ctdb, int argc, const char **argv)
3428 {
3429 struct ctdb_ban_time bantime;
3430
3431 bantime.pnn = options.pnn;
3432 bantime.time = 0;
3433
3434 return update_flags_and_ipreallocate(ctdb, &bantime,
3435 update_state_banned,
3436 NODE_FLAGS_BANNED,
3437 "banned",
3438 false /* set_flag*/);
3439 }
3440
3441 /*
3442 show ban information for a node
3443 */
3444 static int control_showban(struct ctdb_context *ctdb, int argc, const char **argv)
3445 {
3446 int ret;
3447 struct ctdb_node_map *nodemap=NULL;
3448 struct ctdb_ban_time *bantime;
3449
3450 /* verify the node exists */
3451 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
3452 if (ret != 0) {
3453 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
3454 return ret;
3455 }
3456
3457 ret = ctdb_ctrl_get_ban(ctdb, TIMELIMIT(), options.pnn, ctdb, &bantime);
3458 if (ret != 0) {
3459 DEBUG(DEBUG_ERR,("Showing ban info for node %d failed.\n", options.pnn));
3460 return -1;
3461 }
3462
3463 if (bantime->time == 0) {
3464 printf("Node %u is not banned\n", bantime->pnn);
3465 } else {
3466 printf("Node %u is banned, %d seconds remaining\n",
3467 bantime->pnn, bantime->time);
3468 }
3469
3470 return 0;
3471 }
3472
3473 /*
3474 shutdown a daemon
3475 */
3476 static int control_shutdown(struct ctdb_context *ctdb, int argc, const char **argv)
3477 {
3478 int ret;
3479
3480 ret = ctdb_ctrl_shutdown(ctdb, TIMELIMIT(), options.pnn);
3481 if (ret != 0) {
3482 DEBUG(DEBUG_ERR, ("Unable to shutdown node %u\n", options.pnn));
3483 return ret;
3484 }
3485
3486 return 0;
3487 }
3488
3489 /*
3490 trigger a recovery
3491 */
3492 static int control_recover(struct ctdb_context *ctdb, int argc, const char **argv)
3493 {
3494 int ret;
3495 uint32_t generation, next_generation;
3496
3497 /* record the current generation number */
3498 generation = get_generation(ctdb);
3499
3500 ret = ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
3501 if (ret != 0) {
3502 DEBUG(DEBUG_ERR, ("Unable to set recovery mode\n"));
3503 return ret;
3504 }
3505
3506 /* wait until we are in a new generation */
3507 while (1) {
3508 next_generation = get_generation(ctdb);
3509 if (next_generation != generation) {
3510 return 0;
3511 }
3512 sleep(1);
3513 }
3514
3515 return 0;
3516 }
3517
3518
3519 /*
3520 display monitoring mode of a remote node
3521 */
3522 static int control_getmonmode(struct ctdb_context *ctdb, int argc, const char **argv)
3523 {
3524 uint32_t monmode;
3525 int ret;
3526
3527 ret = ctdb_ctrl_getmonmode(ctdb, TIMELIMIT(), options.pnn, &monmode);
3528 if (ret != 0) {
3529 DEBUG(DEBUG_ERR, ("Unable to get monmode from node %u\n", options.pnn));
3530 return ret;
3531 }
3532 if (!options.machinereadable){
3533 printf("Monitoring mode:%s (%d)\n",monmode==CTDB_MONITORING_ACTIVE?"ACTIVE":"DISABLED",monmode);
3534 } else {
3535 printf(":mode:\n");
3536 printf(":%d:\n",monmode);
3537 }
3538 return 0;
3539 }
3540
3541
3542 /*
3543 display capabilities of a remote node
3544 */
3545 static int control_getcapabilities(struct ctdb_context *ctdb, int argc, const char **argv)
3546 {
3547 uint32_t capabilities;
3548 int ret;
3549
3550 ret = ctdb_ctrl_getcapabilities(ctdb, TIMELIMIT(), options.pnn, &capabilities);
3551 if (ret != 0) {
3552 DEBUG(DEBUG_ERR, ("Unable to get capabilities from node %u\n", options.pnn));
3553 return -1;
3554 }
3555
3556 if (!options.machinereadable){
3557 printf("RECMASTER: %s\n", (capabilities&CTDB_CAP_RECMASTER)?"YES":"NO");
3558 printf("LMASTER: %s\n", (capabilities&CTDB_CAP_LMASTER)?"YES":"NO");
3559 printf("LVS: %s\n", (capabilities&CTDB_CAP_LVS)?"YES":"NO");
3560 printf("NATGW: %s\n", (capabilities&CTDB_CAP_NATGW)?"YES":"NO");
3561 } else {
3562 printf(":RECMASTER:LMASTER:LVS:NATGW:\n");
3563 printf(":%d:%d:%d:%d:\n",
3564 !!(capabilities&CTDB_CAP_RECMASTER),
3565 !!(capabilities&CTDB_CAP_LMASTER),
3566 !!(capabilities&CTDB_CAP_LVS),
3567 !!(capabilities&CTDB_CAP_NATGW));
3568 }
3569 return 0;
3570 }
3571
3572 /*
3573 display lvs configuration
3574 */
3575
3576 static uint32_t lvs_exclude_flags[] = {
3577 /* Look for a nice healthy node */
3578 NODE_FLAGS_INACTIVE|NODE_FLAGS_DISABLED,
3579 /* If not found, an UNHEALTHY node will do */
3580 NODE_FLAGS_INACTIVE|NODE_FLAGS_PERMANENTLY_DISABLED,
3581 0,
3582 };
3583
3584 static int control_lvs(struct ctdb_context *ctdb, int argc, const char **argv)
3585 {
3586 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3587 struct ctdb_node_map *orig_nodemap=NULL;
3588 struct ctdb_node_map *nodemap;
3589 int i, ret;
3590
3591 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn,
3592 tmp_ctx, &orig_nodemap);
3593 if (ret != 0) {
3594 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
3595 talloc_free(tmp_ctx);
3596 return -1;
3597 }
3598
3599 nodemap = filter_nodemap_by_capabilities(ctdb, orig_nodemap,
3600 CTDB_CAP_LVS, false);
3601 if (nodemap == NULL) {
3602 /* No memory */
3603 ret = -1;
3604 goto done;
3605 }
3606
3607 ret = 0;
3608
3609 for (i = 0; lvs_exclude_flags[i] != 0; i++) {
3610 struct ctdb_node_map *t =
3611 filter_nodemap_by_flags(ctdb, nodemap,
3612 lvs_exclude_flags[i]);
3613 if (t == NULL) {
3614 /* No memory */
3615 ret = -1;
3616 goto done;
3617 }
3618 if (t->num > 0) {
3619 /* At least 1 node without excluded flags */
3620 int j;
3621 for (j = 0; j < t->num; j++) {
3622 printf("%d:%s\n", t->nodes[j].pnn,
3623 ctdb_addr_to_str(&t->nodes[j].addr));
3624 }
3625 goto done;
3626 }
3627 talloc_free(t);
3628 }
3629 done:
3630 talloc_free(tmp_ctx);
3631 return ret;
3632 }
3633
3634 /*
3635 display who is the lvs master
3636 */
3637 static int control_lvsmaster(struct ctdb_context *ctdb, int argc, const char **argv)
3638 {
3639 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3640 struct ctdb_node_map *nodemap=NULL;
3641 int i, ret;
3642
3643 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn,
3644 tmp_ctx, &nodemap);
3645 if (ret != 0) {
3646 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
3647 talloc_free(tmp_ctx);
3648 return -1;
3649 }
3650
3651 for (i = 0; lvs_exclude_flags[i] != 0; i++) {
3652 struct ctdb_node_map *t =
3653 filter_nodemap_by_flags(ctdb, nodemap,
3654 lvs_exclude_flags[i]);
3655 if (t == NULL) {
3656 /* No memory */
3657 ret = -1;
3658 goto done;
3659 }
3660 if (t->num > 0) {
3661 struct ctdb_node_map *n;
3662 n = filter_nodemap_by_capabilities(ctdb,
3663 t,
3664 CTDB_CAP_LVS,
3665 true);
3666 if (n == NULL) {
3667 /* No memory */
3668 ret = -1;
3669 goto done;
3670 }
3671 if (n->num > 0) {
3672 ret = 0;
3673 printf(options.machinereadable ?
3674 "%d\n" : "Node %d is LVS master\n",
3675 n->nodes[0].pnn);
3676 goto done;
3677 }
3678 }
3679 talloc_free(t);
3680 }
3681
3682 printf("There is no LVS master\n");
3683 ret = 255;
3684 done:
3685 talloc_free(tmp_ctx);
3686 return ret;
3687 }
3688
3689 /*
3690 disable monitoring on a node
3691 */
3692 static int control_disable_monmode(struct ctdb_context *ctdb, int argc, const char **argv)
3693 {
3694
3695 int ret;
3696
3697 ret = ctdb_ctrl_disable_monmode(ctdb, TIMELIMIT(), options.pnn);
3698 if (ret != 0) {
3699 DEBUG(DEBUG_ERR, ("Unable to disable monmode on node %u\n", options.pnn));
3700 return ret;
3701 }
3702 printf("Monitoring mode:%s\n","DISABLED");
3703
3704 return 0;
3705 }
3706
3707 /*
3708 enable monitoring on a node
3709 */
3710 static int control_enable_monmode(struct ctdb_context *ctdb, int argc, const char **argv)
3711 {
3712
3713 int ret;
3714
3715 ret = ctdb_ctrl_enable_monmode(ctdb, TIMELIMIT(), options.pnn);
3716 if (ret != 0) {
3717 DEBUG(DEBUG_ERR, ("Unable to enable monmode on node %u\n", options.pnn));
3718 return ret;
3719 }
3720 printf("Monitoring mode:%s\n","ACTIVE");
3721
3722 return 0;
3723 }
3724
3725 /*
3726 display remote list of keys/data for a db
3727 */
3728 static int control_catdb(struct ctdb_context *ctdb, int argc, const char **argv)
3729 {
3730 const char *db_name;
3731 struct ctdb_db_context *ctdb_db;
3732 int ret;
3733 struct ctdb_dump_db_context c;
3734 uint8_t flags;
3735
3736 if (argc < 1) {
3737 usage();
3738 }
3739
3740 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
3741 return -1;
3742 }
3743
3744 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, flags & CTDB_DB_FLAGS_PERSISTENT, 0);
3745 if (ctdb_db == NULL) {
3746 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3747 return -1;
3748 }
3749
3750 if (options.printlmaster) {
3751 ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn,
3752 ctdb, &ctdb->vnn_map);
3753 if (ret != 0) {
3754 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n",
3755 options.pnn));
3756 return ret;
3757 }
3758 }
3759
3760 ZERO_STRUCT(c);
3761 c.f = stdout;
3762 c.printemptyrecords = (bool)options.printemptyrecords;
3763 c.printdatasize = (bool)options.printdatasize;
3764 c.printlmaster = (bool)options.printlmaster;
3765 c.printhash = (bool)options.printhash;
3766 c.printrecordflags = (bool)options.printrecordflags;
3767
3768 /* traverse and dump the cluster tdb */
3769 ret = ctdb_dump_db(ctdb_db, &c);
3770 if (ret == -1) {
3771 DEBUG(DEBUG_ERR, ("Unable to dump database\n"));
3772 DEBUG(DEBUG_ERR, ("Maybe try 'ctdb getdbstatus %s'"
3773 " and 'ctdb getvar AllowUnhealthyDBRead'\n",
3774 db_name));
3775 return -1;
3776 }
3777 talloc_free(ctdb_db);
3778
3779 printf("Dumped %d records\n", ret);
3780 return 0;
3781 }
3782
3783 struct cattdb_data {
3784 struct ctdb_context *ctdb;
3785 uint32_t count;
3786 };
3787
3788 static int cattdb_traverse(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, void *private_data)
3789 {
3790 struct cattdb_data *d = private_data;
3791 struct ctdb_dump_db_context c;
3792
3793 d->count++;
3794
3795 ZERO_STRUCT(c);
3796 c.f = stdout;
3797 c.printemptyrecords = (bool)options.printemptyrecords;
3798 c.printdatasize = (bool)options.printdatasize;
3799 c.printlmaster = false;
3800 c.printhash = (bool)options.printhash;
3801 c.printrecordflags = true;
3802
3803 return ctdb_dumpdb_record(d->ctdb, key, data, &c);
3804 }
3805
3806 /*
3807 cat the local tdb database using same format as catdb
3808 */
3809 static int control_cattdb(struct ctdb_context *ctdb, int argc, const char **argv)
3810 {
3811 const char *db_name;
3812 struct ctdb_db_context *ctdb_db;
3813 struct cattdb_data d;
3814 uint8_t flags;
3815
3816 if (argc < 1) {
3817 usage();
3818 }
3819
3820 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
3821 return -1;
3822 }
3823
3824 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, flags & CTDB_DB_FLAGS_PERSISTENT, 0);
3825 if (ctdb_db == NULL) {
3826 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3827 return -1;
3828 }
3829
3830 /* traverse the local tdb */
3831 d.count = 0;
3832 d.ctdb = ctdb;
3833 if (tdb_traverse_read(ctdb_db->ltdb->tdb, cattdb_traverse, &d) == -1) {
3834 printf("Failed to cattdb data\n");
3835 exit(10);
3836 }
3837 talloc_free(ctdb_db);
3838
3839 printf("Dumped %d records\n", d.count);
3840 return 0;
3841 }
3842
3843 /*
3844 display the content of a database key
3845 */
3846 static int control_readkey(struct ctdb_context *ctdb, int argc, const char **argv)
3847 {
3848 const char *db_name;
3849 struct ctdb_db_context *ctdb_db;
3850 struct ctdb_record_handle *h;
3851 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3852 TDB_DATA key, data;
3853 uint8_t flags;
3854
3855 if (argc < 2) {
3856 usage();
3857 }
3858
3859 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
3860 return -1;
3861 }
3862
3863 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, flags & CTDB_DB_FLAGS_PERSISTENT, 0);
3864 if (ctdb_db == NULL) {
3865 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3866 return -1;
3867 }
3868
3869 key.dptr = discard_const(argv[1]);
3870 key.dsize = strlen((char *)key.dptr);
3871
3872 h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data);
3873 if (h == NULL) {
3874 printf("Failed to fetch record '%s' on node %d\n",
3875 (const char *)key.dptr, ctdb_get_pnn(ctdb));
3876 talloc_free(tmp_ctx);
3877 exit(10);
3878 }
3879
3880 printf("Data: size:%d ptr:[%.*s]\n", (int)data.dsize, (int)data.dsize, data.dptr);
3881
3882 talloc_free(tmp_ctx);
3883 talloc_free(ctdb_db);
3884 return 0;
3885 }
3886
3887 /*
3888 display the content of a database key
3889 */
3890 static int control_writekey(struct ctdb_context *ctdb, int argc, const char **argv)
3891 {
3892 const char *db_name;
3893 struct ctdb_db_context *ctdb_db;
3894 struct ctdb_record_handle *h;
3895 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3896 TDB_DATA key, data;
3897 uint8_t flags;
3898
3899 if (argc < 3) {
3900 usage();
3901 }
3902
3903 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
3904 return -1;
3905 }
3906
3907 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, flags & CTDB_DB_FLAGS_PERSISTENT, 0);
3908 if (ctdb_db == NULL) {
3909 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3910 return -1;
3911 }
3912
3913 key.dptr = discard_const(argv[1]);
3914 key.dsize = strlen((char *)key.dptr);
3915
3916 h = ctdb_fetch_lock(ctdb_db, tmp_ctx, key, &data);
3917 if (h == NULL) {
3918 printf("Failed to fetch record '%s' on node %d\n",
3919 (const char *)key.dptr, ctdb_get_pnn(ctdb));
3920 talloc_free(tmp_ctx);
3921 exit(10);
3922 }
3923
3924 data.dptr = discard_const(argv[2]);
3925 data.dsize = strlen((char *)data.dptr);
3926
3927 if (ctdb_record_store(h, data) != 0) {
3928 printf("Failed to store record\n");
3929 }
3930
3931 talloc_free(h);
3932 talloc_free(tmp_ctx);
3933 talloc_free(ctdb_db);
3934 return 0;
3935 }
3936
3937 /*
3938 fetch a record from a persistent database
3939 */
3940 static int control_pfetch(struct ctdb_context *ctdb, int argc, const char **argv)
3941 {
3942 const char *db_name;
3943 struct ctdb_db_context *ctdb_db;
3944 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
3945 struct ctdb_transaction_handle *h;
3946 TDB_DATA key, data;
3947 int fd, ret;
3948 bool persistent;
3949 uint8_t flags;
3950
3951 if (argc < 2) {
3952 talloc_free(tmp_ctx);
3953 usage();
3954 }
3955
3956 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
3957 talloc_free(tmp_ctx);
3958 return -1;
3959 }
3960
3961 persistent = flags & CTDB_DB_FLAGS_PERSISTENT;
3962 if (!persistent) {
3963 DEBUG(DEBUG_ERR,("Database '%s' is not persistent\n", db_name));
3964 talloc_free(tmp_ctx);
3965 return -1;
3966 }
3967
3968 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
3969 if (ctdb_db == NULL) {
3970 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
3971 talloc_free(tmp_ctx);
3972 return -1;
3973 }
3974
3975 h = ctdb_transaction_start(ctdb_db, tmp_ctx);
3976 if (h == NULL) {
3977 DEBUG(DEBUG_ERR,("Failed to start transaction on database %s\n", db_name));
3978 talloc_free(tmp_ctx);
3979 return -1;
3980 }
3981
3982 key.dptr = discard_const(argv[1]);
3983 key.dsize = strlen(argv[1]);
3984 ret = ctdb_transaction_fetch(h, tmp_ctx, key, &data);
3985 if (ret != 0) {
3986 DEBUG(DEBUG_ERR,("Failed to fetch record\n"));
3987 talloc_free(tmp_ctx);
3988 return -1;
3989 }
3990
3991 if (data.dsize == 0 || data.dptr == NULL) {
3992 DEBUG(DEBUG_ERR,("Record is empty\n"));
3993 talloc_free(tmp_ctx);
3994 return -1;
3995 }
3996
3997 if (argc == 3) {
3998 fd = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0600);
3999 if (fd == -1) {
4000 DEBUG(DEBUG_ERR,("Failed to open output file %s\n", argv[2]));
4001 talloc_free(tmp_ctx);
4002 return -1;
4003 }
4004 sys_write(fd, data.dptr, data.dsize);
4005 close(fd);
4006 } else {
4007 sys_write(1, data.dptr, data.dsize);
4008 }
4009
4010 /* abort the transaction */
4011 talloc_free(h);
4012
4013
4014 talloc_free(tmp_ctx);
4015 return 0;
4016 }
4017
4018 /*
4019 fetch a record from a tdb-file
4020 */
4021 static int control_tfetch(struct ctdb_context *ctdb, int argc, const char **argv)
4022 {
4023 const char *tdb_file;
4024 TDB_CONTEXT *tdb;
4025 TDB_DATA key, data;
4026 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
4027 int fd;
4028
4029 if (argc < 2) {
4030 usage();
4031 }
4032
4033 tdb_file = argv[0];
4034
4035 tdb = tdb_open(tdb_file, 0, 0, O_RDONLY, 0);
4036 if (tdb == NULL) {
4037 printf("Failed to open TDB file %s\n", tdb_file);
4038 return -1;
4039 }
4040
4041 if (!strncmp(argv[1], "0x", 2)) {
4042 key = hextodata(tmp_ctx, argv[1] + 2);
4043 if (key.dsize == 0) {
4044 printf("Failed to convert \"%s\" into a TDB_DATA\n", argv[1]);
4045 return -1;
4046 }
4047 } else {
4048 key.dptr = discard_const(argv[1]);
4049 key.dsize = strlen(argv[1]);
4050 }
4051
4052 data = tdb_fetch(tdb, key);
4053 if (data.dptr == NULL || data.dsize < sizeof(struct ctdb_ltdb_header)) {
4054 printf("Failed to read record %s from tdb %s\n", argv[1], tdb_file);
4055 tdb_close(tdb);
4056 return -1;
4057 }
4058
4059 tdb_close(tdb);
4060
4061 if (argc == 3) {
4062 fd = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0600);
4063 if (fd == -1) {
4064 printf("Failed to open output file %s\n", argv[2]);
4065 return -1;
4066 }
4067 if (options.verbose){
4068 sys_write(fd, data.dptr, data.dsize);
4069 } else {
4070 sys_write(fd, data.dptr+sizeof(struct ctdb_ltdb_header), data.dsize-sizeof(struct ctdb_ltdb_header));
4071 }
4072 close(fd);
4073 } else {
4074 if (options.verbose){
4075 sys_write(1, data.dptr, data.dsize);
4076 } else {
4077 sys_write(1, data.dptr+sizeof(struct ctdb_ltdb_header), data.dsize-sizeof(struct ctdb_ltdb_header));
4078 }
4079 }
4080
4081 talloc_free(tmp_ctx);
4082 return 0;
4083 }
4084
4085 /*
4086 store a record and header to a tdb-file
4087 */
4088 static int control_tstore(struct ctdb_context *ctdb, int argc, const char **argv)
4089 {
4090 const char *tdb_file;
4091 TDB_CONTEXT *tdb;
4092 TDB_DATA key, value, data;
4093 TALLOC_CTX *tmp_ctx = talloc_new(NULL);
4094 struct ctdb_ltdb_header header;
4095
4096 if (argc < 3) {
4097 usage();
4098 }
4099
4100 tdb_file = argv[0];
4101
4102 tdb = tdb_open(tdb_file, 0, 0, O_RDWR, 0);
4103 if (tdb == NULL) {
4104 printf("Failed to open TDB file %s\n", tdb_file);
4105 return -1;
4106 }
4107
4108 if (!strncmp(argv[1], "0x", 2)) {
4109 key = hextodata(tmp_ctx, argv[1] + 2);
4110 if (key.dsize == 0) {
4111 printf("Failed to convert \"%s\" into a TDB_DATA\n", argv[1]);
4112 return -1;
4113 }
4114 } else {
4115 key.dptr = discard_const(argv[1]);
4116 key.dsize = strlen(argv[1]);
4117 }
4118
4119 if (!strncmp(argv[2], "0x", 2)) {
4120 value = hextodata(tmp_ctx, argv[2] + 2);
4121 if (value.dsize == 0) {
4122 printf("Failed to convert \"%s\" into a TDB_DATA\n", argv[2]);
4123 return -1;
4124 }
4125 } else {
4126 value.dptr = discard_const(argv[2]);
4127 value.dsize = strlen(argv[2]);
4128 }
4129
4130 ZERO_STRUCT(header);
4131 if (argc > 3) {
4132 header.rsn = atoll(argv[3]);
4133 }
4134 if (argc > 4) {
4135 header.dmaster = atoi(argv[4]);
4136 }
4137 if (argc > 5) {
4138 header.flags = atoi(argv[5]);
4139 }
4140
4141 data.dsize = sizeof(struct ctdb_ltdb_header) + value.dsize;
4142 data.dptr = talloc_size(tmp_ctx, data.dsize);
4143 if (data.dptr == NULL) {
4144 printf("Failed to allocate header+value\n");
4145 return -1;
4146 }
4147
4148 *(struct ctdb_ltdb_header *)data.dptr = header;
4149 memcpy(data.dptr + sizeof(struct ctdb_ltdb_header), value.dptr, value.dsize);
4150
4151 if (tdb_store(tdb, key, data, TDB_REPLACE) != 0) {
4152 printf("Failed to write record %s to tdb %s\n", argv[1], tdb_file);
4153 tdb_close(tdb);
4154 return -1;
4155 }
4156
4157 tdb_close(tdb);
4158
4159 talloc_free(tmp_ctx);
4160 return 0;
4161 }
4162
4163 /*
4164 write a record to a persistent database
4165 */
4166 static int control_pstore(struct ctdb_context *ctdb, int argc, const char **argv)
4167 {
4168 const char *db_name;
4169 struct ctdb_db_context *ctdb_db;
4170 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4171 struct ctdb_transaction_handle *h;
4172 struct stat st;
4173 TDB_DATA key, data;
4174 int fd, ret;
4175
4176 if (argc < 3) {
4177 talloc_free(tmp_ctx);
4178 usage();
4179 }
4180
4181 fd = open(argv[2], O_RDONLY);
4182 if (fd == -1) {
4183 DEBUG(DEBUG_ERR,("Failed to open file containing record data : %s %s\n", argv[2], strerror(errno)));
4184 talloc_free(tmp_ctx);
4185 return -1;
4186 }
4187
4188 ret = fstat(fd, &st);
4189 if (ret == -1) {
4190 DEBUG(DEBUG_ERR,("fstat of file %s failed: %s\n", argv[2], strerror(errno)));
4191 close(fd);
4192 talloc_free(tmp_ctx);
4193 return -1;
4194 }
4195
4196 if (!S_ISREG(st.st_mode)) {
4197 DEBUG(DEBUG_ERR,("Not a regular file %s\n", argv[2]));
4198 close(fd);
4199 talloc_free(tmp_ctx);
4200 return -1;
4201 }
4202
4203 data.dsize = st.st_size;
4204 if (data.dsize == 0) {
4205 data.dptr = NULL;
4206 } else {
4207 data.dptr = talloc_size(tmp_ctx, data.dsize);
4208 if (data.dptr == NULL) {
4209 DEBUG(DEBUG_ERR,("Failed to talloc %d of memory to store record data\n", (int)data.dsize));
4210 close(fd);
4211 talloc_free(tmp_ctx);
4212 return -1;
4213 }
4214 ret = sys_read(fd, data.dptr, data.dsize);
4215 if (ret != data.dsize) {
4216 DEBUG(DEBUG_ERR,("Failed to read %d bytes of record data\n", (int)data.dsize));
4217 close(fd);
4218 talloc_free(tmp_ctx);
4219 return -1;
4220 }
4221 }
4222 close(fd);
4223
4224
4225 db_name = argv[0];
4226
4227 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, true, 0);
4228 if (ctdb_db == NULL) {
4229 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
4230 talloc_free(tmp_ctx);
4231 return -1;
4232 }
4233
4234 h = ctdb_transaction_start(ctdb_db, tmp_ctx);
4235 if (h == NULL) {
4236 DEBUG(DEBUG_ERR,("Failed to start transaction on database %s\n", db_name));
4237 talloc_free(tmp_ctx);
4238 return -1;
4239 }
4240
4241 key.dptr = discard_const(argv[1]);
4242 key.dsize = strlen(argv[1]);
4243 ret = ctdb_transaction_store(h, key, data);
4244 if (ret != 0) {
4245 DEBUG(DEBUG_ERR,("Failed to store record\n"));
4246 talloc_free(tmp_ctx);
4247 return -1;
4248 }
4249
4250 ret = ctdb_transaction_commit(h);
4251 if (ret != 0) {
4252 DEBUG(DEBUG_ERR,("Failed to commit transaction\n"));
4253 talloc_free(tmp_ctx);
4254 return -1;
4255 }
4256
4257
4258 talloc_free(tmp_ctx);
4259 return 0;
4260 }
4261
4262 /*
4263 * delete a record from a persistent database
4264 */
4265 static int control_pdelete(struct ctdb_context *ctdb, int argc, const char **argv)
4266 {
4267 const char *db_name;
4268 struct ctdb_db_context *ctdb_db;
4269 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4270 struct ctdb_transaction_handle *h;
4271 TDB_DATA key;
4272 int ret;
4273 bool persistent;
4274 uint8_t flags;
4275
4276 if (argc < 2) {
4277 talloc_free(tmp_ctx);
4278 usage();
4279 }
4280
4281 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
4282 talloc_free(tmp_ctx);
4283 return -1;
4284 }
4285
4286 persistent = flags & CTDB_DB_FLAGS_PERSISTENT;
4287 if (!persistent) {
4288 DEBUG(DEBUG_ERR, ("Database '%s' is not persistent\n", db_name));
4289 talloc_free(tmp_ctx);
4290 return -1;
4291 }
4292
4293 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
4294 if (ctdb_db == NULL) {
4295 DEBUG(DEBUG_ERR, ("Unable to attach to database '%s'\n", db_name));
4296 talloc_free(tmp_ctx);
4297 return -1;
4298 }
4299
4300 h = ctdb_transaction_start(ctdb_db, tmp_ctx);
4301 if (h == NULL) {
4302 DEBUG(DEBUG_ERR, ("Failed to start transaction on database %s\n", db_name));
4303 talloc_free(tmp_ctx);
4304 return -1;
4305 }
4306
4307 key.dptr = discard_const(argv[1]);
4308 key.dsize = strlen(argv[1]);
4309 ret = ctdb_transaction_store(h, key, tdb_null);
4310 if (ret != 0) {
4311 DEBUG(DEBUG_ERR, ("Failed to delete record\n"));
4312 talloc_free(tmp_ctx);
4313 return -1;
4314 }
4315
4316 ret = ctdb_transaction_commit(h);
4317 if (ret != 0) {
4318 DEBUG(DEBUG_ERR, ("Failed to commit transaction\n"));
4319 talloc_free(tmp_ctx);
4320 return -1;
4321 }
4322
4323 talloc_free(tmp_ctx);
4324 return 0;
4325 }
4326
4327 static const char *ptrans_parse_string(TALLOC_CTX *mem_ctx, const char *s,
4328 TDB_DATA *data)
4329 {
4330 const char *t;
4331 size_t n;
4332 const char *ret; /* Next byte after successfully parsed value */
4333
4334 /* Error, unless someone says otherwise */
4335 ret = NULL;
4336 /* Indicates no value to parse */
4337 *data = tdb_null;
4338
4339 /* Skip whitespace */
4340 n = strspn(s, " \t");
4341 t = s + n;
4342
4343 if (t[0] == '"') {
4344 /* Quoted ASCII string - no wide characters! */
4345 t++;
4346 n = strcspn(t, "\"");
4347 if (t[n] == '"') {
4348 if (n > 0) {
4349 data->dsize = n;
4350 data->dptr = talloc_memdup(mem_ctx, t, n);
4351 CTDB_NOMEM_ABORT(data->dptr);
4352 }
4353 ret = t + n + 1;
4354 } else {
4355 DEBUG(DEBUG_WARNING,("Unmatched \" in input %s\n", s));
4356 }
4357 } else {
4358 DEBUG(DEBUG_WARNING,("Unsupported input format in %s\n", s));
4359 }
4360
4361 return ret;
4362 }
4363
4364 static bool ptrans_get_key_value(TALLOC_CTX *mem_ctx, FILE *file,
4365 TDB_DATA *key, TDB_DATA *value)
4366 {
4367 char line [1024]; /* FIXME: make this more flexible? */
4368 const char *t;
4369 char *ptr;
4370
4371 ptr = fgets(line, sizeof(line), file);
4372
4373 if (ptr == NULL) {
4374 return false;
4375 }
4376
4377 /* Get key */
4378 t = ptrans_parse_string(mem_ctx, line, key);
4379 if (t == NULL || key->dptr == NULL) {
4380 /* Line Ignored but not EOF */
4381 return true;
4382 }
4383
4384 /* Get value */
4385 t = ptrans_parse_string(mem_ctx, t, value);
4386 if (t == NULL) {
4387 /* Line Ignored but not EOF */
4388 talloc_free(key->dptr);
4389 *key = tdb_null;
4390 return true;
4391 }
4392
4393 return true;
4394 }
4395
4396 /*
4397 * Update a persistent database as per file/stdin
4398 */
4399 static int control_ptrans(struct ctdb_context *ctdb,
4400 int argc, const char **argv)
4401 {
4402 const char *db_name;
4403 struct ctdb_db_context *ctdb_db;
4404 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4405 struct ctdb_transaction_handle *h;
4406 TDB_DATA key, value;
4407 FILE *file;
4408 int ret;
4409
4410 if (argc < 1) {
4411 talloc_free(tmp_ctx);
4412 usage();
4413 }
4414
4415 file = stdin;
4416 if (argc == 2) {
4417 file = fopen(argv[1], "r");
4418 if (file == NULL) {
4419 DEBUG(DEBUG_ERR,("Unable to open file for reading '%s'\n", argv[1]));
4420 talloc_free(tmp_ctx);
4421 return -1;
4422 }
4423 }
4424
4425 db_name = argv[0];
4426
4427 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, true, 0);
4428 if (ctdb_db == NULL) {
4429 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
4430 goto error;
4431 }
4432
4433 h = ctdb_transaction_start(ctdb_db, tmp_ctx);
4434 if (h == NULL) {
4435 DEBUG(DEBUG_ERR,("Failed to start transaction on database %s\n", db_name));
4436 goto error;
4437 }
4438
4439 while (ptrans_get_key_value(tmp_ctx, file, &key, &value)) {
4440 if (key.dsize != 0) {
4441 ret = ctdb_transaction_store(h, key, value);
4442 /* Minimise memory use */
4443 talloc_free(key.dptr);
4444 if (value.dptr != NULL) {
4445 talloc_free(value.dptr);
4446 }
4447 if (ret != 0) {
4448 DEBUG(DEBUG_ERR,("Failed to store record\n"));
4449 ctdb_transaction_cancel(h);
4450 goto error;
4451 }
4452 }
4453 }
4454
4455 ret = ctdb_transaction_commit(h);
4456 if (ret != 0) {
4457 DEBUG(DEBUG_ERR,("Failed to commit transaction\n"));
4458 goto error;
4459 }
4460
4461 if (file != stdin) {
4462 fclose(file);
4463 }
4464 talloc_free(tmp_ctx);
4465 return 0;
4466
4467 error:
4468 if (file != stdin) {
4469 fclose(file);
4470 }
4471
4472 talloc_free(tmp_ctx);
4473 return -1;
4474 }
4475
4476 /*
4477 check if a service is bound to a port or not
4478 */
4479 static int control_chktcpport(struct ctdb_context *ctdb, int argc, const char **argv)
4480 {
4481 int s, ret;
4482 int v;
4483 int port;
4484 struct sockaddr_in sin;
4485
4486 if (argc != 1) {
4487 printf("Use: ctdb chktcport <port>\n");
4488 return EINVAL;
4489 }
4490
4491 port = atoi(argv[0]);
4492
4493 s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
4494 if (s == -1) {
4495 printf("Failed to open local socket\n");
4496 return errno;
4497 }
4498
4499 v = fcntl(s, F_GETFL, 0);
4500 if (v == -1 || fcntl(s, F_SETFL, v | O_NONBLOCK) != 0) {
4501 printf("Unable to set socket non-blocking: %s\n", strerror(errno));
4502 }
4503
4504 bzero(&sin, sizeof(sin));
4505 sin.sin_family = PF_INET;
4506 sin.sin_port = htons(port);
4507 ret = bind(s, (struct sockaddr *)&sin, sizeof(sin));
4508 close(s);
4509 if (ret == -1) {
4510 printf("Failed to bind to local socket: %d %s\n", errno, strerror(errno));
4511 return errno;
4512 }
4513
4514 return 0;
4515 }
4516
4517
4518 /* Reload public IPs on a specified nodes */
4519 static int control_reloadips(struct ctdb_context *ctdb, int argc, const char **argv)
4520 {
4521 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
4522 uint32_t *nodes;
4523 uint32_t pnn_mode;
4524 uint32_t timeout;
4525 int ret;
4526
4527 assert_single_node_only();
4528
4529 if (argc > 1) {
4530 usage();
4531 }
4532
4533 /* Determine the nodes where IPs need to be reloaded */
4534 if (!parse_nodestring(ctdb, tmp_ctx, argc == 1 ? argv[0] : NULL,
4535 options.pnn, true, &nodes, &pnn_mode)) {
4536 ret = -1;
4537 goto done;
4538 }
4539
4540 again:
4541 /* Disable takeover runs on all connected nodes. A reply
4542 * indicating success is needed from each node so all nodes
4543 * will need to be active. This will retry until maxruntime
4544 * is exceeded, hence no error handling.
4545 *
4546 * A check could be added to not allow reloading of IPs when
4547 * there are disconnected nodes. However, this should
4548 * probably be left up to the administrator.
4549 */
4550 timeout = LONGTIMEOUT;
4551 srvid_broadcast(ctdb, CTDB_SRVID_DISABLE_TAKEOVER_RUNS, &timeout,
4552 "Disable takeover runs", true);
4553
4554 /* Now tell all the desired nodes to reload their public IPs.
4555 * Keep trying this until it succeeds. This assumes all
4556 * failures are transient, which might not be true...
4557 */
4558 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_RELOAD_PUBLIC_IPS,
4559 nodes, 0, LONGTIMELIMIT(),
4560 false, tdb_null,
4561 NULL, NULL, NULL) != 0) {
4562 DEBUG(DEBUG_ERR,
4563 ("Unable to reload IPs on some nodes, try again.\n"));
4564 goto again;
4565 }
4566
4567 /* It isn't strictly necessary to wait until takeover runs are
4568 * re-enabled but doing so can't hurt.
4569 */
4570 timeout = 0;
4571 srvid_broadcast(ctdb, CTDB_SRVID_DISABLE_TAKEOVER_RUNS, &timeout,
4572 "Enable takeover runs", true);
4573
4574 ipreallocate(ctdb);
4575
4576 ret = 0;
4577 done:
4578 talloc_free(tmp_ctx);
4579 return ret;
4580 }
4581
4582 /*
4583 display a list of the databases on a remote ctdb
4584 */
4585 static int control_getdbmap(struct ctdb_context *ctdb, int argc, const char **argv)
4586 {
4587 int i, ret;
4588 struct ctdb_dbid_map *dbmap=NULL;
4589
4590 ret = ctdb_ctrl_getdbmap(ctdb, TIMELIMIT(), options.pnn, ctdb, &dbmap);
4591 if (ret != 0) {
4592 DEBUG(DEBUG_ERR, ("Unable to get dbids from node %u\n", options.pnn));
4593 return ret;
4594 }
4595
4596 if(options.machinereadable){
4597 printf(":ID:Name:Path:Persistent:Sticky:Unhealthy:ReadOnly:\n");
4598 for(i=0;i<dbmap->num;i++){
4599 const char *path;
4600 const char *name;
4601 const char *health;
4602 bool persistent;
4603 bool readonly;
4604 bool sticky;
4605
4606 ctdb_ctrl_getdbpath(ctdb, TIMELIMIT(), options.pnn,
4607 dbmap->dbs[i].dbid, ctdb, &path);
4608 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn,
4609 dbmap->dbs[i].dbid, ctdb, &name);
4610 ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn,
4611 dbmap->dbs[i].dbid, ctdb, &health);
4612 persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
4613 readonly = dbmap->dbs[i].flags & CTDB_DB_FLAGS_READONLY;
4614 sticky = dbmap->dbs[i].flags & CTDB_DB_FLAGS_STICKY;
4615 printf(":0x%08X:%s:%s:%d:%d:%d:%d:\n",
4616 dbmap->dbs[i].dbid, name, path,
4617 !!(persistent), !!(sticky),
4618 !!(health), !!(readonly));
4619 }
4620 return 0;
4621 }
4622
4623 printf("Number of databases:%d\n", dbmap->num);
4624 for(i=0;i<dbmap->num;i++){
4625 const char *path;
4626 const char *name;
4627 const char *health;
4628 bool persistent;
4629 bool readonly;
4630 bool sticky;
4631
4632 ctdb_ctrl_getdbpath(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &path);
4633 ctdb_ctrl_getdbname(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &name);
4634 ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn, dbmap->dbs[i].dbid, ctdb, &health);
4635 persistent = dbmap->dbs[i].flags & CTDB_DB_FLAGS_PERSISTENT;
4636 readonly = dbmap->dbs[i].flags & CTDB_DB_FLAGS_READONLY;
4637 sticky = dbmap->dbs[i].flags & CTDB_DB_FLAGS_STICKY;
4638 printf("dbid:0x%08x name:%s path:%s%s%s%s%s\n",
4639 dbmap->dbs[i].dbid, name, path,
4640 persistent?" PERSISTENT":"",
4641 sticky?" STICKY":"",
4642 readonly?" READONLY":"",
4643 health?" UNHEALTHY":"");
4644 }
4645
4646 return 0;
4647 }
4648
4649 /*
4650 display the status of a database on a remote ctdb
4651 */
4652 static int control_getdbstatus(struct ctdb_context *ctdb, int argc, const char **argv)
4653 {
4654 const char *db_name;
4655 uint32_t db_id;
4656 uint8_t flags;
4657 const char *path;
4658 const char *health;
4659
4660 if (argc < 1) {
4661 usage();
4662 }
4663
4664 if (!db_exists(ctdb, argv[0], &db_id, &db_name, &flags)) {
4665 return -1;
4666 }
4667
4668 ctdb_ctrl_getdbpath(ctdb, TIMELIMIT(), options.pnn, db_id, ctdb, &path);
4669 ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn, db_id, ctdb, &health);
4670 printf("dbid: 0x%08x\nname: %s\npath: %s\nPERSISTENT: %s\nSTICKY: %s\nREADONLY: %s\nHEALTH: %s\n",
4671 db_id, db_name, path,
4672 (flags & CTDB_DB_FLAGS_PERSISTENT ? "yes" : "no"),
4673 (flags & CTDB_DB_FLAGS_STICKY ? "yes" : "no"),
4674 (flags & CTDB_DB_FLAGS_READONLY ? "yes" : "no"),
4675 (health ? health : "OK"));
4676
4677 return 0;
4678 }
4679
4680 /*
4681 check if the local node is recmaster or not
4682 it will return 1 if this node is the recmaster and 0 if it is not
4683 or if the local ctdb daemon could not be contacted
4684 */
4685 static int control_isnotrecmaster(struct ctdb_context *ctdb, int argc, const char **argv)
4686 {
4687 uint32_t mypnn, recmaster;
4688 int ret;
4689
4690 assert_single_node_only();
4691
4692 mypnn = getpnn(ctdb);
4693
4694 ret = ctdb_ctrl_getrecmaster(ctdb, ctdb, TIMELIMIT(), options.pnn, &recmaster);
4695 if (ret != 0) {
4696 printf("Failed to get the recmaster\n");
4697 return 1;
4698 }
4699
4700 if (recmaster != mypnn) {
4701 printf("this node is not the recmaster\n");
4702 return 1;
4703 }
4704
4705 printf("this node is the recmaster\n");
4706 return 0;
4707 }
4708
4709 /*
4710 ping a node
4711 */
4712 static int control_ping(struct ctdb_context *ctdb, int argc, const char **argv)
4713 {
4714 int ret;
4715 struct timeval tv = timeval_current();
4716 ret = ctdb_ctrl_ping(ctdb, options.pnn);
4717 if (ret == -1) {
4718 printf("Unable to get ping response from node %u\n", options.pnn);
4719 return -1;
4720 } else {
4721 printf("response from %u time=%.6f sec (%d clients)\n",
4722 options.pnn, timeval_elapsed(&tv), ret);
4723 }
4724 return 0;
4725 }
4726
4727
4728 /*
4729 get a node's runstate
4730 */
4731 static int control_runstate(struct ctdb_context *ctdb, int argc, const char **argv)
4732 {
4733 int ret;
4734 enum ctdb_runstate runstate;
4735
4736 ret = ctdb_ctrl_get_runstate(ctdb, TIMELIMIT(), options.pnn, &runstate);
4737 if (ret == -1) {
4738 printf("Unable to get runstate response from node %u\n",
4739 options.pnn);
4740 return -1;
4741 } else {
4742 bool found = true;
4743 enum ctdb_runstate t;
4744 int i;
4745 for (i=0; i<argc; i++) {
4746 found = false;
4747 t = runstate_from_string(argv[i]);
4748 if (t == CTDB_RUNSTATE_UNKNOWN) {
4749 printf("Invalid run state (%s)\n", argv[i]);
4750 return -1;
4751 }
4752
4753 if (t == runstate) {
4754 found = true;
4755 break;
4756 }
4757 }
4758
4759 if (!found) {
4760 printf("CTDB not in required run state (got %s)\n",
4761 runstate_to_string((enum ctdb_runstate)runstate));
4762 return -1;
4763 }
4764 }
4765
4766 printf("%s\n", runstate_to_string(runstate));
4767 return 0;
4768 }
4769
4770
4771 /*
4772 get a tunable
4773 */
4774 static int control_getvar(struct ctdb_context *ctdb, int argc, const char **argv)
4775 {
4776 const char *name;
4777 uint32_t value;
4778 int ret;
4779
4780 if (argc < 1) {
4781 usage();
4782 }
4783
4784 name = argv[0];
4785 ret = ctdb_ctrl_get_tunable(ctdb, TIMELIMIT(), options.pnn, name, &value);
4786 if (ret != 0) {
4787 DEBUG(DEBUG_ERR, ("Unable to get tunable variable '%s'\n", name));
4788 return -1;
4789 }
4790
4791 printf("%-23s = %u\n", name, value);
4792 return 0;
4793 }
4794
4795 /*
4796 set a tunable
4797 */
4798 static int control_setvar(struct ctdb_context *ctdb, int argc, const char **argv)
4799 {
4800 const char *name;
4801 uint32_t value;
4802 int ret;
4803
4804 if (argc < 2) {
4805 usage();
4806 }
4807
4808 name = argv[0];
4809 value = strtoul(argv[1], NULL, 0);
4810
4811 ret = ctdb_ctrl_set_tunable(ctdb, TIMELIMIT(), options.pnn, name, value);
4812 if (ret == -1) {
4813 DEBUG(DEBUG_ERR, ("Unable to set tunable variable '%s'\n", name));
4814 return -1;
4815 }
4816 return 0;
4817 }
4818
4819 /*
4820 list all tunables
4821 */
4822 static int control_listvars(struct ctdb_context *ctdb, int argc, const char **argv)
4823 {
4824 uint32_t count;
4825 const char **list;
4826 int ret, i;
4827
4828 ret = ctdb_ctrl_list_tunables(ctdb, TIMELIMIT(), options.pnn, ctdb, &list, &count);
4829 if (ret == -1) {
4830 DEBUG(DEBUG_ERR, ("Unable to list tunable variables\n"));
4831 return -1;
4832 }
4833
4834 for (i=0;i<count;i++) {
4835 control_getvar(ctdb, 1, &list[i]);
4836 }
4837
4838 talloc_free(list);
4839
4840 return 0;
4841 }
4842
4843 /*
4844 display debug level on a node
4845 */
4846 static int control_getdebug(struct ctdb_context *ctdb, int argc, const char **argv)
4847 {
4848 int ret;
4849 int32_t level;
4850
4851 ret = ctdb_ctrl_get_debuglevel(ctdb, options.pnn, &level);
4852 if (ret != 0) {
4853 DEBUG(DEBUG_ERR, ("Unable to get debuglevel response from node %u\n", options.pnn));
4854 return ret;
4855 } else {
4856 const char *desc = get_debug_by_level(level);
4857 if (desc == NULL) {
4858 /* This should never happen */
4859 desc = "Unknown";
4860 }
4861 if (options.machinereadable){
4862 printf(":Name:Level:\n");
4863 printf(":%s:%d:\n", desc, level);
4864 } else {
4865 printf("Node %u is at debug level %s (%d)\n",
4866 options.pnn, desc, level);
4867 }
4868 }
4869 return 0;
4870 }
4871
4872 /*
4873 display reclock file of a node
4874 */
4875 static int control_getreclock(struct ctdb_context *ctdb, int argc, const char **argv)
4876 {
4877 int ret;
4878 const char *reclock;
4879
4880 ret = ctdb_ctrl_getreclock(ctdb, TIMELIMIT(), options.pnn, ctdb, &reclock);
4881 if (ret != 0) {
4882 DEBUG(DEBUG_ERR, ("Unable to get reclock file from node %u\n", options.pnn));
4883 return ret;
4884 } else {
4885 if (options.machinereadable){
4886 if (reclock != NULL) {
4887 printf("%s", reclock);
4888 }
4889 } else {
4890 if (reclock == NULL) {
4891 printf("No reclock file used.\n");
4892 } else {
4893 printf("Reclock file:%s\n", reclock);
4894 }
4895 }
4896 }
4897 return 0;
4898 }
4899
4900 /*
4901 set the reclock file of a node
4902 */
4903 static int control_setreclock(struct ctdb_context *ctdb, int argc, const char **argv)
4904 {
4905 int ret;
4906 const char *reclock;
4907
4908 if (argc == 0) {
4909 reclock = NULL;
4910 } else if (argc == 1) {
4911 reclock = argv[0];
4912 } else {
4913 usage();
4914 }
4915
4916 ret = ctdb_ctrl_setreclock(ctdb, TIMELIMIT(), options.pnn, reclock);
4917 if (ret != 0) {
4918 DEBUG(DEBUG_ERR, ("Unable to get reclock file from node %u\n", options.pnn));
4919 return ret;
4920 }
4921 return 0;
4922 }
4923
4924 /*
4925 set the natgw state on/off
4926 */
4927 static int control_setnatgwstate(struct ctdb_context *ctdb, int argc, const char **argv)
4928 {
4929 int ret;
4930 uint32_t natgwstate;
4931
4932 if (argc == 0) {
4933 usage();
4934 }
4935
4936 if (!strcmp(argv[0], "on")) {
4937 natgwstate = 1;
4938 } else if (!strcmp(argv[0], "off")) {
4939 natgwstate = 0;
4940 } else {
4941 usage();
4942 }
4943
4944 ret = ctdb_ctrl_setnatgwstate(ctdb, TIMELIMIT(), options.pnn, natgwstate);
4945 if (ret != 0) {
4946 DEBUG(DEBUG_ERR, ("Unable to set the natgw state for node %u\n", options.pnn));
4947 return ret;
4948 }
4949
4950 return 0;
4951 }
4952
4953 /*
4954 set the lmaster role on/off
4955 */
4956 static int control_setlmasterrole(struct ctdb_context *ctdb, int argc, const char **argv)
4957 {
4958 int ret;
4959 uint32_t lmasterrole;
4960
4961 if (argc == 0) {
4962 usage();
4963 }
4964
4965 if (!strcmp(argv[0], "on")) {
4966 lmasterrole = 1;
4967 } else if (!strcmp(argv[0], "off")) {
4968 lmasterrole = 0;
4969 } else {
4970 usage();
4971 }
4972
4973 ret = ctdb_ctrl_setlmasterrole(ctdb, TIMELIMIT(), options.pnn, lmasterrole);
4974 if (ret != 0) {
4975 DEBUG(DEBUG_ERR, ("Unable to set the lmaster role for node %u\n", options.pnn));
4976 return ret;
4977 }
4978
4979 return 0;
4980 }
4981
4982 /*
4983 set the recmaster role on/off
4984 */
4985 static int control_setrecmasterrole(struct ctdb_context *ctdb, int argc, const char **argv)
4986 {
4987 int ret;
4988 uint32_t recmasterrole;
4989
4990 if (argc == 0) {
4991 usage();
4992 }
4993
4994 if (!strcmp(argv[0], "on")) {
4995 recmasterrole = 1;
4996 } else if (!strcmp(argv[0], "off")) {
4997 recmasterrole = 0;
4998 } else {
4999 usage();
5000 }
5001
5002 ret = ctdb_ctrl_setrecmasterrole(ctdb, TIMELIMIT(), options.pnn, recmasterrole);
5003 if (ret != 0) {
5004 DEBUG(DEBUG_ERR, ("Unable to set the recmaster role for node %u\n", options.pnn));
5005 return ret;
5006 }
5007
5008 return 0;
5009 }
5010
5011 /*
5012 set debug level on a node or all nodes
5013 */
5014 static int control_setdebug(struct ctdb_context *ctdb, int argc, const char **argv)
5015 {
5016 int ret;
5017 int32_t level;
5018
5019 if (argc == 0) {
5020 printf("You must specify the debug level. Valid levels are:\n");
5021 print_debug_levels(stdout);
5022 return 0;
5023 }
5024
5025 if (!parse_debug(argv[0], &level)) {
5026 printf("Invalid debug level, must be one of\n");
5027 print_debug_levels(stdout);
5028 return -1;
5029 }
5030
5031 ret = ctdb_ctrl_set_debuglevel(ctdb, options.pnn, level);
5032 if (ret != 0) {
5033 DEBUG(DEBUG_ERR, ("Unable to set debug level on node %u\n", options.pnn));
5034 }
5035 return 0;
5036 }
5037
5038
5039 /*
5040 thaw a node
5041 */
5042 static int control_thaw(struct ctdb_context *ctdb, int argc, const char **argv)
5043 {
5044 int ret;
5045 uint32_t priority;
5046
5047 if (argc == 1) {
5048 priority = strtol(argv[0], NULL, 0);
5049 } else {
5050 priority = 0;
5051 }
5052 DEBUG(DEBUG_ERR,("Thaw by priority %u\n", priority));
5053
5054 ret = ctdb_ctrl_thaw_priority(ctdb, TIMELIMIT(), options.pnn, priority);
5055 if (ret != 0) {
5056 DEBUG(DEBUG_ERR, ("Unable to thaw node %u\n", options.pnn));
5057 }
5058 return 0;
5059 }
5060
5061
5062 /*
5063 attach to a database
5064 */
5065 static int control_attach(struct ctdb_context *ctdb, int argc, const char **argv)
5066 {
5067 const char *db_name;
5068 struct ctdb_db_context *ctdb_db;
5069 bool persistent = false;
5070
5071 if (argc < 1) {
5072 usage();
5073 }
5074 db_name = argv[0];
5075 if (argc > 2) {
5076 usage();
5077 }
5078 if (argc == 2) {
5079 if (strcmp(argv[1], "persistent") != 0) {
5080 usage();
5081 }
5082 persistent = true;
5083 }
5084
5085 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, persistent, 0);
5086 if (ctdb_db == NULL) {
5087 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", db_name));
5088 return -1;
5089 }
5090
5091 return 0;
5092 }
5093
5094 /*
5095 * detach from a database
5096 */
5097 static int control_detach(struct ctdb_context *ctdb, int argc,
5098 const char **argv)
5099 {
5100 uint32_t db_id;
5101 uint8_t flags;
5102 int ret, i, status = 0;
5103 struct ctdb_node_map *nodemap = NULL;
5104 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5105 uint32_t recmode;
5106
5107 if (argc < 1) {
5108 usage();
5109 }
5110
5111 assert_single_node_only();
5112
5113 ret = ctdb_ctrl_getrecmode(ctdb, tmp_ctx, TIMELIMIT(), options.pnn,
5114 &recmode);
5115 if (ret != 0) {
5116 DEBUG(DEBUG_ERR, ("Database cannot be detached "
5117 "when recovery is active\n"));
5118 talloc_free(tmp_ctx);
5119 return -1;
5120 }
5121
5122 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx,
5123 &nodemap);
5124 if (ret != 0) {
5125 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n",
5126 options.pnn));
5127 talloc_free(tmp_ctx);
5128 return -1;
5129 }
5130
5131 for (i=0; i<nodemap->num; i++) {
5132 uint32_t value;
5133
5134 if (nodemap->nodes[i].flags & NODE_FLAGS_DISCONNECTED) {
5135 continue;
5136 }
5137
5138 if (nodemap->nodes[i].flags & NODE_FLAGS_DELETED) {
5139 continue;
5140 }
5141
5142 if (nodemap->nodes[i].flags & NODE_FLAGS_INACTIVE) {
5143 DEBUG(DEBUG_ERR, ("Database cannot be detached on "
5144 "inactive (stopped or banned) node "
5145 "%u\n", nodemap->nodes[i].pnn));
5146 talloc_free(tmp_ctx);
5147 return -1;
5148 }
5149
5150 ret = ctdb_ctrl_get_tunable(ctdb, TIMELIMIT(),
5151 nodemap->nodes[i].pnn,
5152 "AllowClientDBAttach",
5153 &value);
5154 if (ret != 0) {
5155 DEBUG(DEBUG_ERR, ("Unable to get tunable "
5156 "AllowClientDBAttach from node %u\n",
5157 nodemap->nodes[i].pnn));
5158 talloc_free(tmp_ctx);
5159 return -1;
5160 }
5161
5162 if (value == 1) {
5163 DEBUG(DEBUG_ERR, ("Database access is still active on "
5164 "node %u. Set AllowClientDBAttach=0 "
5165 "on all nodes.\n",
5166 nodemap->nodes[i].pnn));
5167 talloc_free(tmp_ctx);
5168 return -1;
5169 }
5170 }
5171
5172 talloc_free(tmp_ctx);
5173
5174 for (i=0; i<argc; i++) {
5175 if (!db_exists(ctdb, argv[i], &db_id, NULL, &flags)) {
5176 continue;
5177 }
5178
5179 if (flags & CTDB_DB_FLAGS_PERSISTENT) {
5180 DEBUG(DEBUG_ERR, ("Persistent database '%s' "
5181 "cannot be detached\n", argv[i]));
5182 status = -1;
5183 continue;
5184 }
5185
5186 ret = ctdb_detach(ctdb, db_id);
5187 if (ret != 0) {
5188 DEBUG(DEBUG_ERR, ("Database '%s' detach failed\n",
5189 argv[i]));
5190 status = ret;
5191 }
5192 }
5193
5194 return status;
5195 }
5196
5197 /*
5198 set db priority
5199 */
5200 static int control_setdbprio(struct ctdb_context *ctdb, int argc, const char **argv)
5201 {
5202 struct ctdb_db_priority db_prio;
5203 int ret;
5204
5205 if (argc < 2) {
5206 usage();
5207 }
5208
5209 db_prio.db_id = strtoul(argv[0], NULL, 0);
5210 db_prio.priority = strtoul(argv[1], NULL, 0);
5211
5212 ret = ctdb_ctrl_set_db_priority(ctdb, TIMELIMIT(), options.pnn, &db_prio);
5213 if (ret != 0) {
5214 DEBUG(DEBUG_ERR,("Unable to set db prio\n"));
5215 return -1;
5216 }
5217
5218 return 0;
5219 }
5220
5221 /*
5222 get db priority
5223 */
5224 static int control_getdbprio(struct ctdb_context *ctdb, int argc, const char **argv)
5225 {
5226 uint32_t db_id, priority;
5227 int ret;
5228
5229 if (argc < 1) {
5230 usage();
5231 }
5232
5233 if (!db_exists(ctdb, argv[0], &db_id, NULL, NULL)) {
5234 return -1;
5235 }
5236
5237 ret = ctdb_ctrl_get_db_priority(ctdb, TIMELIMIT(), options.pnn, db_id, &priority);
5238 if (ret != 0) {
5239 DEBUG(DEBUG_ERR,("Unable to get db prio\n"));
5240 return -1;
5241 }
5242
5243 DEBUG(DEBUG_ERR,("Priority:%u\n", priority));
5244
5245 return 0;
5246 }
5247
5248 /*
5249 set the sticky records capability for a database
5250 */
5251 static int control_setdbsticky(struct ctdb_context *ctdb, int argc, const char **argv)
5252 {
5253 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5254 uint32_t db_id;
5255 int ret;
5256
5257 if (argc < 1) {
5258 usage();
5259 }
5260
5261 if (!db_exists(ctdb, argv[0], &db_id, NULL, NULL)) {
5262 return -1;
5263 }
5264
5265 ret = ctdb_ctrl_set_db_sticky(ctdb, options.pnn, db_id);
5266 if (ret != 0) {
5267 DEBUG(DEBUG_ERR,("Unable to set db to support sticky records\n"));
5268 talloc_free(tmp_ctx);
5269 return -1;
5270 }
5271
5272 talloc_free(tmp_ctx);
5273 return 0;
5274 }
5275
5276 /*
5277 set the readonly capability for a database
5278 */
5279 static int control_setdbreadonly(struct ctdb_context *ctdb, int argc, const char **argv)
5280 {
5281 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5282 uint32_t db_id;
5283 int ret;
5284
5285 if (argc < 1) {
5286 usage();
5287 }
5288
5289 if (!db_exists(ctdb, argv[0], &db_id, NULL, NULL)) {
5290 return -1;
5291 }
5292
5293 ret = ctdb_ctrl_set_db_readonly(ctdb, options.pnn, db_id);
5294 if (ret != 0) {
5295 DEBUG(DEBUG_ERR,("Unable to set db to support readonly\n"));
5296 talloc_free(tmp_ctx);
5297 return -1;
5298 }
5299
5300 talloc_free(tmp_ctx);
5301 return 0;
5302 }
5303
5304 /*
5305 get db seqnum
5306 */
5307 static int control_getdbseqnum(struct ctdb_context *ctdb, int argc, const char **argv)
5308 {
5309 uint32_t db_id;
5310 uint64_t seqnum;
5311 int ret;
5312
5313 if (argc < 1) {
5314 usage();
5315 }
5316
5317 if (!db_exists(ctdb, argv[0], &db_id, NULL, NULL)) {
5318 return -1;
5319 }
5320
5321 ret = ctdb_ctrl_getdbseqnum(ctdb, TIMELIMIT(), options.pnn, db_id, &seqnum);
5322 if (ret != 0) {
5323 DEBUG(DEBUG_ERR, ("Unable to get seqnum from node."));
5324 return -1;
5325 }
5326
5327 printf("Sequence number:%lld\n", (long long)seqnum);
5328
5329 return 0;
5330 }
5331
5332 /*
5333 run an eventscript on a node
5334 */
5335 static int control_eventscript(struct ctdb_context *ctdb, int argc, const char **argv)
5336 {
5337 TDB_DATA data;
5338 int ret;
5339 int32_t res;
5340 char *errmsg;
5341 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5342
5343 if (argc != 1) {
5344 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
5345 return -1;
5346 }
5347
5348 data.dptr = (unsigned char *)discard_const(argv[0]);
5349 data.dsize = strlen((char *)data.dptr) + 1;
5350
5351 DEBUG(DEBUG_ERR, ("Running eventscripts with arguments \"%s\" on node %u\n", data.dptr, options.pnn));
5352
5353 ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_RUN_EVENTSCRIPTS,
5354 0, data, tmp_ctx, NULL, &res, NULL, &errmsg);
5355 if (ret != 0 || res != 0) {
5356 DEBUG(DEBUG_ERR,("Failed to run eventscripts - %s\n", errmsg));
5357 talloc_free(tmp_ctx);
5358 return -1;
5359 }
5360 talloc_free(tmp_ctx);
5361 return 0;
5362 }
5363
5364 #define DB_VERSION 1
5365 #define MAX_DB_NAME 64
5366 struct db_file_header {
5367 unsigned long version;
5368 time_t timestamp;
5369 unsigned long persistent;
5370 unsigned long size;
5371 const char name[MAX_DB_NAME];
5372 };
5373
5374 struct backup_data {
5375 struct ctdb_marshall_buffer *records;
5376 uint32_t len;
5377 uint32_t total;
5378 bool traverse_error;
5379 };
5380
5381 static int backup_traverse(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, void *private)
5382 {
5383 struct backup_data *bd = talloc_get_type(private, struct backup_data);
5384 struct ctdb_rec_data *rec;
5385
5386 /* add the record */
5387 rec = ctdb_marshall_record(bd->records, 0, key, NULL, data);
5388 if (rec == NULL) {
5389 bd->traverse_error = true;
5390 DEBUG(DEBUG_ERR,("Failed to marshall record\n"));
5391 return -1;
5392 }
5393 bd->records = talloc_realloc_size(NULL, bd->records, rec->length + bd->len);
5394 if (bd->records == NULL) {
5395 DEBUG(DEBUG_ERR,("Failed to expand marshalling buffer\n"));
5396 bd->traverse_error = true;
5397 return -1;
5398 }
5399 bd->records->count++;
5400 memcpy(bd->len+(uint8_t *)bd->records, rec, rec->length);
5401 bd->len += rec->length;
5402 talloc_free(rec);
5403
5404 bd->total++;
5405 return 0;
5406 }
5407
5408 /*
5409 * backup a database to a file
5410 */
5411 static int control_backupdb(struct ctdb_context *ctdb, int argc, const char **argv)
5412 {
5413 const char *db_name;
5414 int ret;
5415 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5416 struct db_file_header dbhdr;
5417 struct ctdb_db_context *ctdb_db;
5418 struct backup_data *bd;
5419 int fh = -1;
5420 int status = -1;
5421 const char *reason = NULL;
5422 uint32_t db_id;
5423 uint8_t flags;
5424
5425 assert_single_node_only();
5426
5427 if (argc != 2) {
5428 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
5429 return -1;
5430 }
5431
5432 if (!db_exists(ctdb, argv[0], &db_id, &db_name, &flags)) {
5433 return -1;
5434 }
5435
5436 ret = ctdb_ctrl_getdbhealth(ctdb, TIMELIMIT(), options.pnn,
5437 db_id, tmp_ctx, &reason);
5438 if (ret != 0) {
5439 DEBUG(DEBUG_ERR,("Unable to get dbhealth for database '%s'\n",
5440 argv[0]));
5441 talloc_free(tmp_ctx);
5442 return -1;
5443 }
5444 if (reason) {
5445 uint32_t allow_unhealthy = 0;
5446
5447 ctdb_ctrl_get_tunable(ctdb, TIMELIMIT(), options.pnn,
5448 "AllowUnhealthyDBRead",
5449 &allow_unhealthy);
5450
5451 if (allow_unhealthy != 1) {
5452 DEBUG(DEBUG_ERR,("database '%s' is unhealthy: %s\n",
5453 argv[0], reason));
5454
5455 DEBUG(DEBUG_ERR,("disallow backup : tunable AllowUnhealthyDBRead = %u\n",
5456 allow_unhealthy));
5457 talloc_free(tmp_ctx);
5458 return -1;
5459 }
5460
5461 DEBUG(DEBUG_WARNING,("WARNING database '%s' is unhealthy - see 'ctdb getdbstatus %s'\n",
5462 argv[0], argv[0]));
5463 DEBUG(DEBUG_WARNING,("WARNING! allow backup of unhealthy database: "
5464 "tunnable AllowUnhealthyDBRead = %u\n",
5465 allow_unhealthy));
5466 }
5467
5468 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, flags & CTDB_DB_FLAGS_PERSISTENT, 0);
5469 if (ctdb_db == NULL) {
5470 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", argv[0]));
5471 talloc_free(tmp_ctx);
5472 return -1;
5473 }
5474
5475
5476 ret = tdb_transaction_start(ctdb_db->ltdb->tdb);
5477 if (ret == -1) {
5478 DEBUG(DEBUG_ERR,("Failed to start transaction\n"));
5479 talloc_free(tmp_ctx);
5480 return -1;
5481 }
5482
5483
5484 bd = talloc_zero(tmp_ctx, struct backup_data);
5485 if (bd == NULL) {
5486 DEBUG(DEBUG_ERR,("Failed to allocate backup_data\n"));
5487 talloc_free(tmp_ctx);
5488 return -1;
5489 }
5490
5491 bd->records = talloc_zero(bd, struct ctdb_marshall_buffer);
5492 if (bd->records == NULL) {
5493 DEBUG(DEBUG_ERR,("Failed to allocate ctdb_marshall_buffer\n"));
5494 talloc_free(tmp_ctx);
5495 return -1;
5496 }
5497
5498 bd->len = offsetof(struct ctdb_marshall_buffer, data);
5499 bd->records->db_id = ctdb_db->db_id;
5500 /* traverse the database collecting all records */
5501 if (tdb_traverse_read(ctdb_db->ltdb->tdb, backup_traverse, bd) == -1 ||
5502 bd->traverse_error) {
5503 DEBUG(DEBUG_ERR,("Traverse error\n"));
5504 talloc_free(tmp_ctx);
5505 return -1;
5506 }
5507
5508 tdb_transaction_cancel(ctdb_db->ltdb->tdb);
5509
5510
5511 fh = open(argv[1], O_RDWR|O_CREAT, 0600);
5512 if (fh == -1) {
5513 DEBUG(DEBUG_ERR,("Failed to open file '%s'\n", argv[1]));
5514 talloc_free(tmp_ctx);
5515 return -1;
5516 }
5517
5518 ZERO_STRUCT(dbhdr);
5519 dbhdr.version = DB_VERSION;
5520 dbhdr.timestamp = time(NULL);
5521 dbhdr.persistent = flags & CTDB_DB_FLAGS_PERSISTENT;
5522 dbhdr.size = bd->len;
5523 if (strlen(argv[0]) >= MAX_DB_NAME) {
5524 DEBUG(DEBUG_ERR,("Too long dbname\n"));
5525 goto done;
5526 }
5527 strncpy(discard_const(dbhdr.name), argv[0], MAX_DB_NAME-1);
5528 ret = sys_write(fh, &dbhdr, sizeof(dbhdr));
5529 if (ret == -1) {
5530 DEBUG(DEBUG_ERR,("write failed: %s\n", strerror(errno)));
5531 goto done;
5532 }
5533 ret = sys_write(fh, bd->records, bd->len);
5534 if (ret == -1) {
5535 DEBUG(DEBUG_ERR,("write failed: %s\n", strerror(errno)));
5536 goto done;
5537 }
5538
5539 status = 0;
5540 done:
5541 if (fh != -1) {
5542 ret = close(fh);
5543 if (ret == -1) {
5544 DEBUG(DEBUG_ERR,("close failed: %s\n", strerror(errno)));
5545 }
5546 }
5547
5548 DEBUG(DEBUG_ERR,("Database backed up to %s\n", argv[1]));
5549
5550 talloc_free(tmp_ctx);
5551 return status;
5552 }
5553
5554 /*
5555 * restore a database from a file
5556 */
5557 static int control_restoredb(struct ctdb_context *ctdb, int argc, const char **argv)
5558 {
5559 int ret;
5560 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5561 TDB_DATA outdata;
5562 TDB_DATA data;
5563 struct db_file_header dbhdr;
5564 struct ctdb_db_context *ctdb_db;
5565 struct ctdb_node_map *nodemap=NULL;
5566 struct ctdb_vnn_map *vnnmap=NULL;
5567 int i, fh;
5568 struct ctdb_control_wipe_database w;
5569 uint32_t *nodes;
5570 uint32_t generation;
5571 struct tm *tm;
5572 char tbuf[100];
5573 char *dbname;
5574
5575 assert_single_node_only();
5576
5577 if (argc < 1 || argc > 2) {
5578 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
5579 return -1;
5580 }
5581
5582 fh = open(argv[0], O_RDONLY);
5583 if (fh == -1) {
5584 DEBUG(DEBUG_ERR,("Failed to open file '%s'\n", argv[0]));
5585 talloc_free(tmp_ctx);
5586 return -1;
5587 }
5588
5589 sys_read(fh, &dbhdr, sizeof(dbhdr));
5590 if (dbhdr.version != DB_VERSION) {
5591 DEBUG(DEBUG_ERR,("Invalid version of database dump. File is version %lu but expected version was %u\n", dbhdr.version, DB_VERSION));
5592 close(fh);
5593 talloc_free(tmp_ctx);
5594 return -1;
5595 }
5596
5597 dbname = discard_const(dbhdr.name);
5598 if (argc == 2) {
5599 dbname = discard_const(argv[1]);
5600 }
5601
5602 outdata.dsize = dbhdr.size;
5603 outdata.dptr = talloc_size(tmp_ctx, outdata.dsize);
5604 if (outdata.dptr == NULL) {
5605 DEBUG(DEBUG_ERR,("Failed to allocate data of size '%lu'\n", dbhdr.size));
5606 close(fh);
5607 talloc_free(tmp_ctx);
5608 return -1;
5609 }
5610 sys_read(fh, outdata.dptr, outdata.dsize);
5611 close(fh);
5612
5613 tm = localtime(&dbhdr.timestamp);
5614 strftime(tbuf,sizeof(tbuf)-1,"%Y/%m/%d %H:%M:%S", tm);
5615 printf("Restoring database '%s' from backup @ %s\n",
5616 dbname, tbuf);
5617
5618
5619 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), dbname, dbhdr.persistent, 0);
5620 if (ctdb_db == NULL) {
5621 DEBUG(DEBUG_ERR,("Unable to attach to database '%s'\n", dbname));
5622 talloc_free(tmp_ctx);
5623 return -1;
5624 }
5625
5626 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb, &nodemap);
5627 if (ret != 0) {
5628 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n", options.pnn));
5629 talloc_free(tmp_ctx);
5630 return ret;
5631 }
5632
5633
5634 ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx, &vnnmap);
5635 if (ret != 0) {
5636 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n", options.pnn));
5637 talloc_free(tmp_ctx);
5638 return ret;
5639 }
5640
5641 /* freeze all nodes */
5642 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5643 for (i=1; i<=NUM_DB_PRIORITIES; i++) {
5644 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_FREEZE,
5645 nodes, i,
5646 TIMELIMIT(),
5647 false, tdb_null,
5648 NULL, NULL,
5649 NULL) != 0) {
5650 DEBUG(DEBUG_ERR, ("Unable to freeze nodes.\n"));
5651 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5652 talloc_free(tmp_ctx);
5653 return -1;
5654 }
5655 }
5656
5657 generation = vnnmap->generation;
5658 data.dptr = (void *)&generation;
5659 data.dsize = sizeof(generation);
5660
5661 /* start a cluster wide transaction */
5662 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5663 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_START,
5664 nodes, 0,
5665 TIMELIMIT(), false, data,
5666 NULL, NULL,
5667 NULL) != 0) {
5668 DEBUG(DEBUG_ERR, ("Unable to start cluster wide transactions.\n"));
5669 return -1;
5670 }
5671
5672
5673 w.db_id = ctdb_db->db_id;
5674 w.transaction_id = generation;
5675
5676 data.dptr = (void *)&w;
5677 data.dsize = sizeof(w);
5678
5679 /* wipe all the remote databases. */
5680 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5681 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_WIPE_DATABASE,
5682 nodes, 0,
5683 TIMELIMIT(), false, data,
5684 NULL, NULL,
5685 NULL) != 0) {
5686 DEBUG(DEBUG_ERR, ("Unable to wipe database.\n"));
5687 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5688 talloc_free(tmp_ctx);
5689 return -1;
5690 }
5691
5692 /* push the database */
5693 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5694 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_PUSH_DB,
5695 nodes, 0,
5696 TIMELIMIT(), false, outdata,
5697 NULL, NULL,
5698 NULL) != 0) {
5699 DEBUG(DEBUG_ERR, ("Failed to push database.\n"));
5700 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5701 talloc_free(tmp_ctx);
5702 return -1;
5703 }
5704
5705 data.dptr = (void *)&ctdb_db->db_id;
5706 data.dsize = sizeof(ctdb_db->db_id);
5707
5708 /* mark the database as healthy */
5709 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5710 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_DB_SET_HEALTHY,
5711 nodes, 0,
5712 TIMELIMIT(), false, data,
5713 NULL, NULL,
5714 NULL) != 0) {
5715 DEBUG(DEBUG_ERR, ("Failed to mark database as healthy.\n"));
5716 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5717 talloc_free(tmp_ctx);
5718 return -1;
5719 }
5720
5721 data.dptr = (void *)&generation;
5722 data.dsize = sizeof(generation);
5723
5724 /* commit all the changes */
5725 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_COMMIT,
5726 nodes, 0,
5727 TIMELIMIT(), false, data,
5728 NULL, NULL,
5729 NULL) != 0) {
5730 DEBUG(DEBUG_ERR, ("Unable to commit databases.\n"));
5731 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5732 talloc_free(tmp_ctx);
5733 return -1;
5734 }
5735
5736
5737 /* thaw all nodes */
5738 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5739 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_THAW,
5740 nodes, 0,
5741 TIMELIMIT(),
5742 false, tdb_null,
5743 NULL, NULL,
5744 NULL) != 0) {
5745 DEBUG(DEBUG_ERR, ("Unable to thaw nodes.\n"));
5746 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5747 talloc_free(tmp_ctx);
5748 return -1;
5749 }
5750
5751
5752 talloc_free(tmp_ctx);
5753 return 0;
5754 }
5755
5756 /*
5757 * dump a database backup from a file
5758 */
5759 static int control_dumpdbbackup(struct ctdb_context *ctdb, int argc, const char **argv)
5760 {
5761 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5762 TDB_DATA outdata;
5763 struct db_file_header dbhdr;
5764 int i, fh;
5765 struct tm *tm;
5766 char tbuf[100];
5767 struct ctdb_rec_data *rec = NULL;
5768 struct ctdb_marshall_buffer *m;
5769 struct ctdb_dump_db_context c;
5770
5771 assert_single_node_only();
5772
5773 if (argc != 1) {
5774 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
5775 return -1;
5776 }
5777
5778 fh = open(argv[0], O_RDONLY);
5779 if (fh == -1) {
5780 DEBUG(DEBUG_ERR,("Failed to open file '%s'\n", argv[0]));
5781 talloc_free(tmp_ctx);
5782 return -1;
5783 }
5784
5785 sys_read(fh, &dbhdr, sizeof(dbhdr));
5786 if (dbhdr.version != DB_VERSION) {
5787 DEBUG(DEBUG_ERR,("Invalid version of database dump. File is version %lu but expected version was %u\n", dbhdr.version, DB_VERSION));
5788 close(fh);
5789 talloc_free(tmp_ctx);
5790 return -1;
5791 }
5792
5793 outdata.dsize = dbhdr.size;
5794 outdata.dptr = talloc_size(tmp_ctx, outdata.dsize);
5795 if (outdata.dptr == NULL) {
5796 DEBUG(DEBUG_ERR,("Failed to allocate data of size '%lu'\n", dbhdr.size));
5797 close(fh);
5798 talloc_free(tmp_ctx);
5799 return -1;
5800 }
5801 sys_read(fh, outdata.dptr, outdata.dsize);
5802 close(fh);
5803 m = (struct ctdb_marshall_buffer *)outdata.dptr;
5804
5805 tm = localtime(&dbhdr.timestamp);
5806 strftime(tbuf,sizeof(tbuf)-1,"%Y/%m/%d %H:%M:%S", tm);
5807 printf("Backup of database name:'%s' dbid:0x%x08x from @ %s\n",
5808 dbhdr.name, m->db_id, tbuf);
5809
5810 ZERO_STRUCT(c);
5811 c.f = stdout;
5812 c.printemptyrecords = (bool)options.printemptyrecords;
5813 c.printdatasize = (bool)options.printdatasize;
5814 c.printlmaster = false;
5815 c.printhash = (bool)options.printhash;
5816 c.printrecordflags = (bool)options.printrecordflags;
5817
5818 for (i=0; i < m->count; i++) {
5819 uint32_t reqid = 0;
5820 TDB_DATA key, data;
5821
5822 /* we do not want the header splitted, so we pass NULL*/
5823 rec = ctdb_marshall_loop_next(m, rec, &reqid,
5824 NULL, &key, &data);
5825
5826 ctdb_dumpdb_record(ctdb, key, data, &c);
5827 }
5828
5829 printf("Dumped %d records\n", i);
5830 talloc_free(tmp_ctx);
5831 return 0;
5832 }
5833
5834 /*
5835 * wipe a database from a file
5836 */
5837 static int control_wipedb(struct ctdb_context *ctdb, int argc,
5838 const char **argv)
5839 {
5840 const char *db_name;
5841 int ret;
5842 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
5843 TDB_DATA data;
5844 struct ctdb_db_context *ctdb_db;
5845 struct ctdb_node_map *nodemap = NULL;
5846 struct ctdb_vnn_map *vnnmap = NULL;
5847 int i;
5848 struct ctdb_control_wipe_database w;
5849 uint32_t *nodes;
5850 uint32_t generation;
5851 uint8_t flags;
5852
5853 assert_single_node_only();
5854
5855 if (argc != 1) {
5856 DEBUG(DEBUG_ERR,("Invalid arguments\n"));
5857 return -1;
5858 }
5859
5860 if (!db_exists(ctdb, argv[0], NULL, &db_name, &flags)) {
5861 return -1;
5862 }
5863
5864 ctdb_db = ctdb_attach(ctdb, TIMELIMIT(), db_name, flags & CTDB_DB_FLAGS_PERSISTENT, 0);
5865 if (ctdb_db == NULL) {
5866 DEBUG(DEBUG_ERR, ("Unable to attach to database '%s'\n",
5867 argv[0]));
5868 talloc_free(tmp_ctx);
5869 return -1;
5870 }
5871
5872 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), options.pnn, ctdb,
5873 &nodemap);
5874 if (ret != 0) {
5875 DEBUG(DEBUG_ERR, ("Unable to get nodemap from node %u\n",
5876 options.pnn));
5877 talloc_free(tmp_ctx);
5878 return ret;
5879 }
5880
5881 ret = ctdb_ctrl_getvnnmap(ctdb, TIMELIMIT(), options.pnn, tmp_ctx,
5882 &vnnmap);
5883 if (ret != 0) {
5884 DEBUG(DEBUG_ERR, ("Unable to get vnnmap from node %u\n",
5885 options.pnn));
5886 talloc_free(tmp_ctx);
5887 return ret;
5888 }
5889
5890 /* freeze all nodes */
5891 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5892 for (i=1; i<=NUM_DB_PRIORITIES; i++) {
5893 ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_FREEZE,
5894 nodes, i,
5895 TIMELIMIT(),
5896 false, tdb_null,
5897 NULL, NULL,
5898 NULL);
5899 if (ret != 0) {
5900 DEBUG(DEBUG_ERR, ("Unable to freeze nodes.\n"));
5901 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn,
5902 CTDB_RECOVERY_ACTIVE);
5903 talloc_free(tmp_ctx);
5904 return -1;
5905 }
5906 }
5907
5908 generation = vnnmap->generation;
5909 data.dptr = (void *)&generation;
5910 data.dsize = sizeof(generation);
5911
5912 /* start a cluster wide transaction */
5913 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5914 ret = ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_START,
5915 nodes, 0,
5916 TIMELIMIT(), false, data,
5917 NULL, NULL,
5918 NULL);
5919 if (ret!= 0) {
5920 DEBUG(DEBUG_ERR, ("Unable to start cluster wide "
5921 "transactions.\n"));
5922 return -1;
5923 }
5924
5925 w.db_id = ctdb_db->db_id;
5926 w.transaction_id = generation;
5927
5928 data.dptr = (void *)&w;
5929 data.dsize = sizeof(w);
5930
5931 /* wipe all the remote databases. */
5932 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5933 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_WIPE_DATABASE,
5934 nodes, 0,
5935 TIMELIMIT(), false, data,
5936 NULL, NULL,
5937 NULL) != 0) {
5938 DEBUG(DEBUG_ERR, ("Unable to wipe database.\n"));
5939 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5940 talloc_free(tmp_ctx);
5941 return -1;
5942 }
5943
5944 data.dptr = (void *)&ctdb_db->db_id;
5945 data.dsize = sizeof(ctdb_db->db_id);
5946
5947 /* mark the database as healthy */
5948 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5949 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_DB_SET_HEALTHY,
5950 nodes, 0,
5951 TIMELIMIT(), false, data,
5952 NULL, NULL,
5953 NULL) != 0) {
5954 DEBUG(DEBUG_ERR, ("Failed to mark database as healthy.\n"));
5955 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5956 talloc_free(tmp_ctx);
5957 return -1;
5958 }
5959
5960 data.dptr = (void *)&generation;
5961 data.dsize = sizeof(generation);
5962
5963 /* commit all the changes */
5964 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_TRANSACTION_COMMIT,
5965 nodes, 0,
5966 TIMELIMIT(), false, data,
5967 NULL, NULL,
5968 NULL) != 0) {
5969 DEBUG(DEBUG_ERR, ("Unable to commit databases.\n"));
5970 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5971 talloc_free(tmp_ctx);
5972 return -1;
5973 }
5974
5975 /* thaw all nodes */
5976 nodes = list_of_active_nodes(ctdb, nodemap, tmp_ctx, true);
5977 if (ctdb_client_async_control(ctdb, CTDB_CONTROL_THAW,
5978 nodes, 0,
5979 TIMELIMIT(),
5980 false, tdb_null,
5981 NULL, NULL,
5982 NULL) != 0) {
5983 DEBUG(DEBUG_ERR, ("Unable to thaw nodes.\n"));
5984 ctdb_ctrl_setrecmode(ctdb, TIMELIMIT(), options.pnn, CTDB_RECOVERY_ACTIVE);
5985 talloc_free(tmp_ctx);
5986 return -1;
5987 }
5988
5989 DEBUG(DEBUG_ERR, ("Database wiped.\n"));
5990
5991 talloc_free(tmp_ctx);
5992 return 0;
5993 }
5994
5995 /*
5996 dump memory usage
5997 */
5998 static int control_dumpmemory(struct ctdb_context *ctdb, int argc, const char **argv)
5999 {
6000 TDB_DATA data;
6001 int ret;
6002 int32_t res;
6003 char *errmsg;
6004 TALLOC_CTX *tmp_ctx = talloc_new(ctdb);
6005 ret = ctdb_control(ctdb, options.pnn, 0, CTDB_CONTROL_DUMP_MEMORY,
6006 0, tdb_null, tmp_ctx, &data, &res, NULL, &errmsg);
6007 if (ret != 0 || res != 0) {
6008 DEBUG(DEBUG_ERR,("Failed to dump memory - %s\n", errmsg));
6009 talloc_free(tmp_ctx);
6010 return -1;
6011 }
6012 sys_write(1, data.dptr, data.dsize);
6013 talloc_free(tmp_ctx);
6014 return 0;
6015 }
6016
6017 /*
6018 handler for memory dumps
6019 */
6020 static void mem_dump_handler(struct ctdb_context *ctdb, uint64_t srvid,
6021 TDB_DATA data, void *private_data)
6022 {
6023 sys_write(1, data.dptr, data.dsize);
6024 exit(0);
6025 }
6026
6027 /*
6028 dump memory usage on the recovery daemon
6029 */
6030 static int control_rddumpmemory(struct ctdb_context *ctdb, int argc, const char **argv)
6031 {
6032 int ret;
6033 TDB_DATA data;
6034 struct srvid_request rd;
6035
6036 rd.pnn = ctdb_get_pnn(ctdb);
6037 rd.srvid = getpid();
6038
6039 /* register a message port for receiveing the reply so that we
6040 can receive the reply
6041 */
6042 ctdb_client_set_message_handler(ctdb, rd.srvid, mem_dump_handler, NULL);
6043
6044
6045 data.dptr = (uint8_t *)&rd;
6046 data.dsize = sizeof(rd);
6047
6048 ret = ctdb_client_send_message(ctdb, options.pnn, CTDB_SRVID_MEM_DUMP, data);
6049 if (ret != 0) {
6050 DEBUG(DEBUG_ERR,("Failed to send memdump request message to %u\n", options.pnn));
6051 return -1;
6052 }
6053
6054 /* this loop will terminate when we have received the reply */
6055 while (1) {
6056 event_loop_once(ctdb->ev);
6057 }
6058
6059 return 0;
6060 }
6061
6062 /*
6063 send a message to a srvid
6064 */
6065 static int control_msgsend(struct ctdb_context *ctdb, int argc, const char **argv)
6066 {
6067 unsigned long srvid;
6068 int ret;
6069 TDB_DATA data;
6070
6071 if (argc < 2) {
6072 usage();
6073 }
6074
6075 srvid = strtoul(argv[0], NULL, 0);
6076
6077 data.dptr = (uint8_t *)discard_const(argv[1]);
6078 data.dsize= strlen(argv[1]);
6079
6080 ret = ctdb_client_send_message(ctdb, CTDB_BROADCAST_CONNECTED, srvid, data);
6081 if (ret != 0) {
6082 DEBUG(DEBUG_ERR,("Failed to send memdump request message to %u\n", options.pnn));
6083 return -1;
6084 }
6085
6086 return 0;
6087 }
6088
6089 /*
6090 handler for msglisten
6091 */
6092 static void msglisten_handler(struct ctdb_context *ctdb, uint64_t srvid,
6093 TDB_DATA data, void *private_data)
6094 {
6095 int i;
6096
6097 printf("Message received: ");
6098 for (i=0;i<data.dsize;i++) {
6099 printf("%c", data.dptr[i]);
6100 }
6101 printf("\n");
6102 }
6103
6104 /*
6105 listen for messages on a messageport
6106 */
6107 static int control_msglisten(struct ctdb_context *ctdb, int argc, const char **argv)
6108 {
6109 uint64_t srvid;
6110
6111 srvid = getpid();
6112
6113 /* register a message port and listen for messages
6114 */
6115 ctdb_client_set_message_handler(ctdb, srvid, msglisten_handler, NULL);
6116 printf("Listening for messages on srvid:%d\n", (int)srvid);
6117
6118 while (1) {
6119 event_loop_once(ctdb->ev);
6120 }
6121
6122 return 0;
6123 }
6124
6125 /*
6126 list all nodes in the cluster
6127 we parse the nodes file directly
6128 */
6129 static int control_listnodes(struct ctdb_context *ctdb, int argc, const char **argv)
6130 {
6131 TALLOC_CTX *mem_ctx = talloc_new(NULL);
6132 struct pnn_node *pnn_nodes;
6133 struct pnn_node *pnn_node;
6134
6135 assert_single_node_only();
6136
6137 pnn_nodes = read_nodes_file(mem_ctx);
6138 if (pnn_nodes == NULL) {
6139 DEBUG(DEBUG_ERR,("Failed to read nodes file\n"));
6140 talloc_free(mem_ctx);
6141 return -1;
6142 }
6143
6144 for(pnn_node=pnn_nodes;pnn_node;pnn_node=pnn_node->next) {
6145 const char *addr = ctdb_addr_to_str(&pnn_node->addr);
6146 if (options.machinereadable){
6147 printf(":%d:%s:\n", pnn_node->pnn, addr);
6148 } else {
6149 printf("%s\n", addr);
6150 }
6151 }
6152 talloc_free(mem_ctx);
6153
6154 return 0;
6155 }
6156
6157 /*
6158 reload the nodes file on the local node
6159 */
6160 static int control_reload_nodes_file(struct ctdb_context *ctdb, int argc, const char **argv)
6161 {
6162 int i, ret;
6163 int mypnn;
6164 struct ctdb_node_map *nodemap=NULL;
6165
6166 assert_single_node_only();
6167
6168 mypnn = ctdb_get_pnn(ctdb);
6169
6170 ret = ctdb_ctrl_getnodemap(ctdb, TIMELIMIT(), CTDB_CURRENT_NODE, ctdb, &nodemap);
6171 if (ret != 0) {
6172 DEBUG(DEBUG_ERR, ("Unable to get nodemap from local node\n"));
6173 return ret;
6174 }
6175
6176 /* reload the nodes file on all remote nodes */
6177 for (i=0;i<nodemap->num;i++) {
6178 if (nodemap->nodes[i].pnn == mypnn) {
6179 continue;
6180 }
6181 DEBUG(DEBUG_NOTICE, ("Reloading nodes file on node %u\n", nodemap->nodes[i].pnn));
6182 ret = ctdb_ctrl_reload_nodes_file(ctdb, TIMELIMIT(),
6183 nodemap->nodes[i].pnn);
6184 if (ret != 0) {
6185 DEBUG(DEBUG_ERR, ("ERROR: Failed to reload nodes file on node %u. You MUST fix that node manually!\n", nodemap->nodes[i].pnn));
6186 }
6187 }
6188
6189 /* reload the nodes file on the local node */
6190 DEBUG(DEBUG_NOTICE, ("Reloading nodes file on node %u\n", mypnn));
6191 ret = ctdb_ctrl_reload_nodes_file(ctdb, TIMELIMIT(), mypnn);
6192 if (ret != 0) {
6193 DEBUG(DEBUG_ERR, ("ERROR: Failed to reload nodes file on node %u. You MUST fix that node manually!\n", mypnn));
6194 }
6195
6196 /* initiate a recovery */
6197 control_recover(ctdb, argc, argv);
6198
6199 return 0;
6200 }
6201
6202
6203 static const struct {
6204 const char *name;
6205 int (*fn)(struct ctdb_context *, int, const char **);
6206 bool auto_all;
6207 bool without_daemon; /* can be run without daemon running ? */
6208 const char *msg;
6209 const char *args;
6210 } ctdb_commands[] = {
6211 { "version", control_version, true, true, "show version of ctdb" },
6212 { "status", control_status, true, false, "show node status" },
6213 { "uptime", control_uptime, true, false, "show node uptime" },
6214 { "ping", control_ping, true, false, "ping all nodes" },
6215 { "runstate", control_runstate, true, false, "get/check runstate of a node", "[setup|first_recovery|startup|running]" },
6216 { "getvar", control_getvar, true, false, "get a tunable variable", "<name>"},
6217 { "setvar", control_setvar, true, false, "set a tunable variable", "<name> <value>"},
6218 { "listvars", control_listvars, true, false, "list tunable variables"},
6219 { "statistics", control_statistics, false, false, "show statistics" },
6220 { "statisticsreset", control_statistics_reset, true, false, "reset statistics"},
6221 { "stats", control_stats, false, false, "show rolling statistics", "[number of history records]" },
6222 { "ip", control_ip, false, false, "show which public ip's that ctdb manages" },
6223 { "ipinfo", control_ipinfo, true, false, "show details about a public ip that ctdb manages", "<ip>" },
6224 { "ifaces", control_ifaces, true, false, "show which interfaces that ctdb manages" },
6225 { "setifacelink", control_setifacelink, true, false, "set interface link status", "<iface> <status>" },
6226 { "process-exists", control_process_exists, true, false, "check if a process exists on a node", "<pid>"},
6227 { "getdbmap", control_getdbmap, true, false, "show the database map" },
6228 { "getdbstatus", control_getdbstatus, true, false, "show the status of a database", "<dbname|dbid>" },
6229 { "catdb", control_catdb, true, false, "dump a ctdb database" , "<dbname|dbid>"},
6230 { "cattdb", control_cattdb, true, false, "dump a local tdb database" , "<dbname|dbid>"},
6231 { "getmonmode", control_getmonmode, true, false, "show monitoring mode" },
6232 { "getcapabilities", control_getcapabilities, true, false, "show node capabilities" },
6233 { "pnn", control_pnn, true, false, "show the pnn of the currnet node" },
6234 { "lvs", control_lvs, true, false, "show lvs configuration" },
6235 { "lvsmaster", control_lvsmaster, true, false, "show which node is the lvs master" },
6236 { "disablemonitor", control_disable_monmode,true, false, "set monitoring mode to DISABLE" },
6237 { "enablemonitor", control_enable_monmode, true, false, "set monitoring mode to ACTIVE" },
6238 { "setdebug", control_setdebug, true, false, "set debug level", "<EMERG|ALERT|CRIT|ERR|WARNING|NOTICE|INFO|DEBUG>" },
6239 { "getdebug", control_getdebug, true, false, "get debug level" },
6240 { "attach", control_attach, true, false, "attach to a database", "<dbname> [persistent]" },
6241 { "detach", control_detach, false, false, "detach from a database", "<dbname|dbid> [<dbname|dbid> ...]" },
6242 { "dumpmemory", control_dumpmemory, true, false, "dump memory map to stdout" },
6243 { "rddumpmemory", control_rddumpmemory, true, false, "dump memory map from the recovery daemon to stdout" },
6244 { "getpid", control_getpid, true, false, "get ctdbd process ID" },
6245 { "disable", control_disable, true, false, "disable a nodes public IP" },
6246 { "enable", control_enable, true, false, "enable a nodes public IP" },
6247 { "stop", control_stop, true, false, "stop a node" },
6248 { "continue", control_continue, true, false, "re-start a stopped node" },
6249 { "ban", control_ban, true, false, "ban a node from the cluster", "<bantime>"},
6250 { "unban", control_unban, true, false, "unban a node" },
6251 { "showban", control_showban, true, false, "show ban information"},
6252 { "shutdown", control_shutdown, true, false, "shutdown ctdbd" },
6253 { "recover", control_recover, true, false, "force recovery" },
6254 { "sync", control_ipreallocate, false, false, "wait until ctdbd has synced all state changes" },
6255 { "ipreallocate", control_ipreallocate, false, false, "force the recovery daemon to perform a ip reallocation procedure" },
6256 { "thaw", control_thaw, true, false, "thaw databases", "[priority:1-3]" },
6257 { "isnotrecmaster", control_isnotrecmaster, false, false, "check if the local node is recmaster or not" },
6258 { "killtcp", kill_tcp, false, false, "kill a tcp connection.", "[<srcip:port> <dstip:port>]" },
6259 { "gratiousarp", control_gratious_arp, false, false, "send a gratious arp", "<ip> <interface>" },
6260 { "tickle", tickle_tcp, false, false, "send a tcp tickle ack", "<srcip:port> <dstip:port>" },
6261 { "gettickles", control_get_tickles, false, false, "get the list of tickles registered for this ip", "<ip> [<port>]" },
6262 { "addtickle", control_add_tickle, false, false, "add a tickle for this ip", "<ip>:<port> <ip>:<port>" },
6263
6264 { "deltickle", control_del_tickle, false, false, "delete a tickle from this ip", "<ip>:<port> <ip>:<port>" },
6265
6266 { "regsrvid", regsrvid, false, false, "register a server id", "<pnn> <type> <id>" },
6267 { "unregsrvid", unregsrvid, false, false, "unregister a server id", "<pnn> <type> <id>" },
6268 { "chksrvid", chksrvid, false, false, "check if a server id exists", "<pnn> <type> <id>" },
6269 { "getsrvids", getsrvids, false, false, "get a list of all server ids"},
6270 { "check_srvids", check_srvids, false, false, "check if a srvid exists", "<id>+" },
6271 { "repack", ctdb_repack, false, false, "repack all databases", "[max_freelist]"},
6272 { "listnodes", control_listnodes, false, true, "list all nodes in the cluster"},
6273 { "reloadnodes", control_reload_nodes_file, false, false, "reload the nodes file and restart the transport on all nodes"},
6274 { "moveip", control_moveip, false, false, "move/failover an ip address to another node", "<ip> <node>"},
6275 { "rebalanceip", control_rebalanceip, false, false, "release an ip from the node and let recd rebalance it", "<ip>"},
6276 { "addip", control_addip, true, false, "add a ip address to a node", "<ip/mask> <iface>"},
6277 { "delip", control_delip, false, false, "delete an ip address from a node", "<ip>"},
6278 { "eventscript", control_eventscript, true, false, "run the eventscript with the given parameters on a node", "<arguments>"},
6279 { "backupdb", control_backupdb, false, false, "backup the database into a file.", "<dbname|dbid> <file>"},
6280 { "restoredb", control_restoredb, false, false, "restore the database from a file.", "<file> [dbname]"},
6281 { "dumpdbbackup", control_dumpdbbackup, false, true, "dump database backup from a file.", "<file>"},
6282 { "wipedb", control_wipedb, false, false, "wipe the contents of a database.", "<dbname|dbid>"},
6283 { "recmaster", control_recmaster, true, false, "show the pnn for the recovery master."},
6284 { "scriptstatus", control_scriptstatus, true, false, "show the status of the monitoring scripts (or all scripts)", "[all]"},
6285 { "enablescript", control_enablescript, true, false, "enable an eventscript", "<script>"},
6286 { "disablescript", control_disablescript, true, false, "disable an eventscript", "<script>"},
6287 { "natgwlist", control_natgwlist, true, false, "show the nodes belonging to this natgw configuration"},
6288 { "xpnn", control_xpnn, false, true, "find the pnn of the local node without talking to the daemon (unreliable)" },
6289 { "getreclock", control_getreclock, true, false, "Show the reclock file of a node"},
6290 { "setreclock", control_setreclock, true, false, "Set/clear the reclock file of a node", "[filename]"},
6291 { "setnatgwstate", control_setnatgwstate, false, false, "Set NATGW state to on/off", "{on|off}"},
6292 { "setlmasterrole", control_setlmasterrole, false, false, "Set LMASTER role to on/off", "{on|off}"},
6293 { "setrecmasterrole", control_setrecmasterrole, false, false, "Set RECMASTER role to on/off", "{on|off}"},
6294 { "setdbprio", control_setdbprio, false, false, "Set DB priority", "<dbname|dbid> <prio:1-3>"},
6295 { "getdbprio", control_getdbprio, false, false, "Get DB priority", "<dbname|dbid>"},
6296 { "setdbreadonly", control_setdbreadonly, false, false, "Set DB readonly capable", "<dbname|dbid>"},
6297 { "setdbsticky", control_setdbsticky, false, false, "Set DB sticky-records capable", "<dbname|dbid>"},
6298 { "msglisten", control_msglisten, false, false, "Listen on a srvid port for messages", "<msg srvid>"},
6299 { "msgsend", control_msgsend, false, false, "Send a message to srvid", "<srvid> <message>"},
6300 { "pfetch", control_pfetch, false, false, "fetch a record from a persistent database", "<dbname|dbid> <key> [<file>]" },
6301 { "pstore", control_pstore, false, false, "write a record to a persistent database", "<dbname|dbid> <key> <file containing record>" },
6302 { "pdelete", control_pdelete, false, false, "delete a record from a persistent database", "<dbname|dbid> <key>" },
6303 { "ptrans", control_ptrans, false, false, "update a persistent database (from stdin)", "<dbname|dbid>" },
6304 { "tfetch", control_tfetch, false, true, "fetch a record from a [c]tdb-file [-v]", "<tdb-file> <key> [<file>]" },
6305 { "tstore", control_tstore, false, true, "store a record (including ltdb header)", "<tdb-file> <key> <data> [<rsn> <dmaster> <flags>]" },
6306 { "readkey", control_readkey, true, false, "read the content off a database key", "<dbname|dbid> <key>" },
6307 { "writekey", control_writekey, true, false, "write to a database key", "<dbname|dbid> <key> <value>" },
6308 { "checktcpport", control_chktcpport, false, true, "check if a service is bound to a specific tcp port or not", "<port>" },
6309 { "rebalancenode", control_rebalancenode, false, false, "mark nodes as forced IP rebalancing targets", "[<pnn-list>]"},
6310 { "getdbseqnum", control_getdbseqnum, false, false, "get the sequence number off a database", "<dbname|dbid>" },
6311 { "nodestatus", control_nodestatus, true, false, "show and return node status", "[<pnn-list>]" },
6312 { "dbstatistics", control_dbstatistics, false, false, "show db statistics", "<dbname|dbid>" },
6313 { "reloadips", control_reloadips, false, false, "reload the public addresses file on specified nodes" , "[<pnn-list>]" },
6314 { "ipiface", control_ipiface, false, true, "Find which interface an ip address is hosted on", "<ip>" },
6315 };
6316
6317 /*
6318 show usage message
6319 */
6320 static void usage(void)
6321 {
6322 int i;
6323 printf(
6324 "Usage: ctdb [options] <control>\n" \
6325 "Options:\n" \
6326 " -n <node> choose node number, or 'all' (defaults to local node)\n"
6327 " -Y generate machinereadable output\n"
6328 " -v generate verbose output\n"
6329 " -t <timelimit> set timelimit for control in seconds (default %u)\n", options.timelimit);
6330 printf("Controls:\n");
6331 for (i=0;i<ARRAY_SIZE(ctdb_commands);i++) {
6332 printf(" %-15s %-27s %s\n",
6333 ctdb_commands[i].name,
6334 ctdb_commands[i].args?ctdb_commands[i].args:"",
6335 ctdb_commands[i].msg);
6336 }
6337 exit(1);
6338 }
6339
6340
6341 static void ctdb_alarm(int sig)
6342 {
6343 printf("Maximum runtime exceeded - exiting\n");
6344 _exit(ERR_TIMEOUT);
6345 }
6346
6347 /*
6348 main program
6349 */
6350 int main(int argc, const char *argv[])
6351 {
6352 struct ctdb_context *ctdb;
6353 char *nodestring = NULL;
6354 struct poptOption popt_options[] = {
6355 POPT_AUTOHELP
6356 POPT_CTDB_CMDLINE
6357 { "timelimit", 't', POPT_ARG_INT, &options.timelimit, 0, "timelimit", "integer" },
6358 { "node", 'n', POPT_ARG_STRING, &nodestring, 0, "node", "integer|all" },
6359 { "machinereadable", 'Y', POPT_ARG_NONE, &options.machinereadable, 0, "enable machinereadable output", NULL },
6360 { "verbose", 'v', POPT_ARG_NONE, &options.verbose, 0, "enable verbose output", NULL },
6361 { "maxruntime", 'T', POPT_ARG_INT, &options.maxruntime, 0, "die if runtime exceeds this limit (in seconds)", "integer" },
6362 { "print-emptyrecords", 0, POPT_ARG_NONE, &options.printemptyrecords, 0, "print the empty records when dumping databases (catdb, cattdb, dumpdbbackup)", NULL },
6363 { "print-datasize", 0, POPT_ARG_NONE, &options.printdatasize, 0, "do not print record data when dumping databases, only the data size", NULL },
6364 { "print-lmaster", 0, POPT_ARG_NONE, &options.printlmaster, 0, "print the record's lmaster in catdb", NULL },
6365 { "print-hash", 0, POPT_ARG_NONE, &options.printhash, 0, "print the record's hash when dumping databases", NULL },
6366 { "print-recordflags", 0, POPT_ARG_NONE, &options.printrecordflags, 0, "print the record flags in catdb and dumpdbbackup", NULL },
6367 POPT_TABLEEND
6368 };
6369 int opt;
6370 const char **extra_argv;
6371 int extra_argc = 0;
6372 int ret=-1, i;
6373 poptContext pc;
6374 struct event_context *ev;
6375 const char *control;
6376
6377 setlinebuf(stdout);
6378
6379 /* set some defaults */
6380 options.maxruntime = 0;
6381 options.timelimit = 10;
6382 options.pnn = CTDB_CURRENT_NODE;
6383
6384 pc = poptGetContext(argv[0], argc, argv, popt_options, POPT_CONTEXT_KEEP_FIRST);
6385
6386 while ((opt = poptGetNextOpt(pc)) != -1) {
6387 switch (opt) {
6388 default:
6389 DEBUG(DEBUG_ERR, ("Invalid option %s: %s\n",
6390 poptBadOption(pc, 0), poptStrerror(opt)));
6391 exit(1);
6392 }
6393 }
6394
6395 /* setup the remaining options for the main program to use */
6396 extra_argv = poptGetArgs(pc);
6397 if (extra_argv) {
6398 extra_argv++;
6399 while (extra_argv[extra_argc]) extra_argc++;
6400 }
6401
6402 if (extra_argc < 1) {
6403 usage();
6404 }
6405
6406 if (options.maxruntime == 0) {
6407 const char *ctdb_timeout;
6408 ctdb_timeout = getenv("CTDB_TIMEOUT");
6409 if (ctdb_timeout != NULL) {
6410 options.maxruntime = strtoul(ctdb_timeout, NULL, 0);
6411 } else {
6412 /* default timeout is 120 seconds */
6413 options.maxruntime = 120;
6414 }
6415 }
6416
6417 signal(SIGALRM, ctdb_alarm);
6418 alarm(options.maxruntime);
6419
6420 control = extra_argv[0];
6421
6422 /* Default value for CTDB_BASE - don't override */
6423 setenv("CTDB_BASE", CTDB_ETCDIR, 0);
6424
6425 ev = event_context_init(NULL);
6426 if (!ev) {
6427 DEBUG(DEBUG_ERR, ("Failed to initialize event system\n"));
6428 exit(1);
6429 }
6430
6431 for (i=0;i<ARRAY_SIZE(ctdb_commands);i++) {
6432 if (strcmp(control, ctdb_commands[i].name) == 0) {
6433 break;
6434 }
6435 }
6436
6437 if (i == ARRAY_SIZE(ctdb_commands)) {
6438 DEBUG(DEBUG_ERR, ("Unknown control '%s'\n", control));
6439 exit(1);
6440 }
6441
6442 if (ctdb_commands[i].without_daemon == true) {
6443 if (nodestring != NULL) {
6444 DEBUG(DEBUG_ERR, ("Can't specify node(s) with \"ctdb %s\"\n", control));
6445 exit(1);
6446 }
6447 return ctdb_commands[i].fn(NULL, extra_argc-1, extra_argv+1);
6448 }
6449
6450 /* initialise ctdb */
6451 ctdb = ctdb_cmdline_client(ev, TIMELIMIT());
6452
6453 if (ctdb == NULL) {
6454 uint32_t pnn;
6455 DEBUG(DEBUG_ERR, ("Failed to init ctdb\n"));
6456
6457 pnn = find_node_xpnn();
6458 if (pnn == -1) {
6459 DEBUG(DEBUG_ERR,
6460 ("Is this node part of a CTDB cluster?\n"));
6461 }
6462 exit(1);
6463 }
6464
6465 /* setup the node number(s) to contact */
6466 if (!parse_nodestring(ctdb, ctdb, nodestring, CTDB_CURRENT_NODE, false,
6467 &options.nodes, &options.pnn)) {
6468 usage();
6469 }
6470
6471 if (options.pnn == CTDB_CURRENT_NODE) {
6472 options.pnn = options.nodes[0];
6473 }
6474
6475 if (ctdb_commands[i].auto_all &&
6476 ((options.pnn == CTDB_BROADCAST_ALL) ||
6477 (options.pnn == CTDB_MULTICAST))) {
6478 int j;
6479
6480 ret = 0;
6481 for (j = 0; j < talloc_array_length(options.nodes); j++) {
6482 options.pnn = options.nodes[j];
6483 ret |= ctdb_commands[i].fn(ctdb, extra_argc-1, extra_argv+1);
6484 }
6485 } else {
6486 ret = ctdb_commands[i].fn(ctdb, extra_argc-1, extra_argv+1);
6487 }
6488
6489 talloc_free(ctdb);
6490 talloc_free(ev);
6491 (void)poptFreeContext(pc);
6492
6493 return ret;
6494
6495 }
Samba GIT mirror