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