ctdb: Use ctdb_wait_for_process_to_exit()
[obnox/samba/samba-obnox.git] / ctdb / server / ctdb_recover.c
1 /* 
2    ctdb recovery code
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 #include "replace.h"
21 #include "system/time.h"
22 #include "system/network.h"
23 #include "system/filesys.h"
24 #include "system/wait.h"
25
26 #include <talloc.h>
27 #include <tevent.h>
28 #include <tdb.h>
29
30 #include "lib/tdb_wrap/tdb_wrap.h"
31 #include "lib/util/dlinklist.h"
32 #include "lib/util/debug.h"
33 #include "lib/util/time.h"
34 #include "lib/util/util_process.h"
35
36 #include "ctdb_private.h"
37 #include "ctdb_client.h"
38
39 #include "common/system.h"
40 #include "common/common.h"
41 #include "common/logging.h"
42
43 int 
44 ctdb_control_getvnnmap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
45 {
46         struct ctdb_vnn_map_wire *map;
47         size_t len;
48
49         CHECK_CONTROL_DATA_SIZE(0);
50
51         len = offsetof(struct ctdb_vnn_map_wire, map) + sizeof(uint32_t)*ctdb->vnn_map->size;
52         map = talloc_size(outdata, len);
53         CTDB_NO_MEMORY(ctdb, map);
54
55         map->generation = ctdb->vnn_map->generation;
56         map->size = ctdb->vnn_map->size;
57         memcpy(map->map, ctdb->vnn_map->map, sizeof(uint32_t)*map->size);
58
59         outdata->dsize = len;
60         outdata->dptr  = (uint8_t *)map;
61
62         return 0;
63 }
64
65 int
66 ctdb_control_setvnnmap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
67 {
68         struct ctdb_vnn_map_wire *map = (struct ctdb_vnn_map_wire *)indata.dptr;
69
70         if (ctdb->recovery_mode != CTDB_RECOVERY_ACTIVE) {
71                 DEBUG(DEBUG_ERR, ("Attempt to set vnnmap when not in recovery\n"));
72                 return -1;
73         }
74
75         talloc_free(ctdb->vnn_map);
76
77         ctdb->vnn_map = talloc(ctdb, struct ctdb_vnn_map);
78         CTDB_NO_MEMORY(ctdb, ctdb->vnn_map);
79
80         ctdb->vnn_map->generation = map->generation;
81         ctdb->vnn_map->size       = map->size;
82         ctdb->vnn_map->map = talloc_array(ctdb->vnn_map, uint32_t, map->size);
83         CTDB_NO_MEMORY(ctdb, ctdb->vnn_map->map);
84
85         memcpy(ctdb->vnn_map->map, map->map, sizeof(uint32_t)*map->size);
86
87         return 0;
88 }
89
90 int 
91 ctdb_control_getdbmap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
92 {
93         uint32_t i, len;
94         struct ctdb_db_context *ctdb_db;
95         struct ctdb_dbid_map_old *dbid_map;
96
97         CHECK_CONTROL_DATA_SIZE(0);
98
99         len = 0;
100         for(ctdb_db=ctdb->db_list;ctdb_db;ctdb_db=ctdb_db->next){
101                 len++;
102         }
103
104
105         outdata->dsize = offsetof(struct ctdb_dbid_map_old, dbs) + sizeof(dbid_map->dbs[0])*len;
106         outdata->dptr  = (unsigned char *)talloc_zero_size(outdata, outdata->dsize);
107         if (!outdata->dptr) {
108                 DEBUG(DEBUG_ALERT, (__location__ " Failed to allocate dbmap array\n"));
109                 exit(1);
110         }
111
112         dbid_map = (struct ctdb_dbid_map_old *)outdata->dptr;
113         dbid_map->num = len;
114         for (i=0,ctdb_db=ctdb->db_list;ctdb_db;i++,ctdb_db=ctdb_db->next){
115                 dbid_map->dbs[i].db_id       = ctdb_db->db_id;
116                 if (ctdb_db->persistent != 0) {
117                         dbid_map->dbs[i].flags |= CTDB_DB_FLAGS_PERSISTENT;
118                 }
119                 if (ctdb_db->readonly != 0) {
120                         dbid_map->dbs[i].flags |= CTDB_DB_FLAGS_READONLY;
121                 }
122                 if (ctdb_db->sticky != 0) {
123                         dbid_map->dbs[i].flags |= CTDB_DB_FLAGS_STICKY;
124                 }
125         }
126
127         return 0;
128 }
129
130 int
131 ctdb_control_getnodemap(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata, TDB_DATA *outdata)
132 {
133         CHECK_CONTROL_DATA_SIZE(0);
134
135         outdata->dptr  = (unsigned char *)ctdb_node_list_to_map(ctdb->nodes,
136                                                                 ctdb->num_nodes,
137                                                                 outdata);
138         if (outdata->dptr == NULL) {
139                 return -1;
140         }
141
142         outdata->dsize = talloc_get_size(outdata->dptr);
143
144         return 0;
145 }
146
147 /*
148   reload the nodes file
149 */
150 int
151 ctdb_control_reload_nodes_file(struct ctdb_context *ctdb, uint32_t opcode)
152 {
153         int i, num_nodes;
154         TALLOC_CTX *tmp_ctx;
155         struct ctdb_node **nodes;
156
157         tmp_ctx = talloc_new(ctdb);
158
159         /* steal the old nodes file for a while */
160         talloc_steal(tmp_ctx, ctdb->nodes);
161         nodes = ctdb->nodes;
162         ctdb->nodes = NULL;
163         num_nodes = ctdb->num_nodes;
164         ctdb->num_nodes = 0;
165
166         /* load the new nodes file */
167         ctdb_load_nodes_file(ctdb);
168
169         for (i=0; i<ctdb->num_nodes; i++) {
170                 /* keep any identical pre-existing nodes and connections */
171                 if ((i < num_nodes) && ctdb_same_address(&ctdb->nodes[i]->address, &nodes[i]->address)) {
172                         talloc_free(ctdb->nodes[i]);
173                         ctdb->nodes[i] = talloc_steal(ctdb->nodes, nodes[i]);
174                         continue;
175                 }
176
177                 if (ctdb->nodes[i]->flags & NODE_FLAGS_DELETED) {
178                         continue;
179                 }
180
181                 /* any new or different nodes must be added */
182                 if (ctdb->methods->add_node(ctdb->nodes[i]) != 0) {
183                         DEBUG(DEBUG_CRIT, (__location__ " methods->add_node failed at %d\n", i));
184                         ctdb_fatal(ctdb, "failed to add node. shutting down\n");
185                 }
186                 if (ctdb->methods->connect_node(ctdb->nodes[i]) != 0) {
187                         DEBUG(DEBUG_CRIT, (__location__ " methods->add_connect failed at %d\n", i));
188                         ctdb_fatal(ctdb, "failed to connect to node. shutting down\n");
189                 }
190         }
191
192         /* tell the recovery daemon to reaload the nodes file too */
193         ctdb_daemon_send_message(ctdb, ctdb->pnn, CTDB_SRVID_RELOAD_NODES, tdb_null);
194
195         talloc_free(tmp_ctx);
196
197         return 0;
198 }
199
200 /* 
201    a traverse function for pulling all relevent records from pulldb
202  */
203 struct pulldb_data {
204         struct ctdb_context *ctdb;
205         struct ctdb_db_context *ctdb_db;
206         struct ctdb_marshall_buffer *pulldata;
207         uint32_t len;
208         uint32_t allocated_len;
209         bool failed;
210 };
211
212 static int traverse_pulldb(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, void *p)
213 {
214         struct pulldb_data *params = (struct pulldb_data *)p;
215         struct ctdb_rec_data_old *rec;
216         struct ctdb_context *ctdb = params->ctdb;
217         struct ctdb_db_context *ctdb_db = params->ctdb_db;
218
219         /* add the record to the blob */
220         rec = ctdb_marshall_record(params->pulldata, 0, key, NULL, data);
221         if (rec == NULL) {
222                 params->failed = true;
223                 return -1;
224         }
225         if (params->len + rec->length >= params->allocated_len) {
226                 params->allocated_len = rec->length + params->len + ctdb->tunable.pulldb_preallocation_size;
227                 params->pulldata = talloc_realloc_size(NULL, params->pulldata, params->allocated_len);
228         }
229         if (params->pulldata == NULL) {
230                 DEBUG(DEBUG_CRIT,(__location__ " Failed to expand pulldb_data to %u\n", rec->length + params->len));
231                 ctdb_fatal(params->ctdb, "failed to allocate memory for recovery. shutting down\n");
232         }
233         params->pulldata->count++;
234         memcpy(params->len+(uint8_t *)params->pulldata, rec, rec->length);
235         params->len += rec->length;
236
237         if (ctdb->tunable.db_record_size_warn != 0 && rec->length > ctdb->tunable.db_record_size_warn) {
238                 DEBUG(DEBUG_ERR,("Data record in %s is big. Record size is %d bytes\n", ctdb_db->db_name, (int)rec->length));
239         }
240
241         talloc_free(rec);
242
243         return 0;
244 }
245
246 /*
247   pull a bunch of records from a ltdb, filtering by lmaster
248  */
249 int32_t ctdb_control_pull_db(struct ctdb_context *ctdb, TDB_DATA indata, TDB_DATA *outdata)
250 {
251         struct ctdb_pulldb *pull;
252         struct ctdb_db_context *ctdb_db;
253         struct pulldb_data params;
254         struct ctdb_marshall_buffer *reply;
255
256         pull = (struct ctdb_pulldb *)indata.dptr;
257
258         ctdb_db = find_ctdb_db(ctdb, pull->db_id);
259         if (!ctdb_db) {
260                 DEBUG(DEBUG_ERR,(__location__ " Unknown db 0x%08x\n", pull->db_id));
261                 return -1;
262         }
263
264         if (!ctdb_db_frozen(ctdb_db)) {
265                 DEBUG(DEBUG_ERR,
266                       ("rejecting ctdb_control_pull_db when not frozen\n"));
267                 return -1;
268         }
269
270         reply = talloc_zero(outdata, struct ctdb_marshall_buffer);
271         CTDB_NO_MEMORY(ctdb, reply);
272
273         reply->db_id = pull->db_id;
274
275         params.ctdb = ctdb;
276         params.ctdb_db = ctdb_db;
277         params.pulldata = reply;
278         params.len = offsetof(struct ctdb_marshall_buffer, data);
279         params.allocated_len = params.len;
280         params.failed = false;
281
282         if (ctdb_db->unhealthy_reason) {
283                 /* this is just a warning, as the tdb should be empty anyway */
284                 DEBUG(DEBUG_WARNING,("db(%s) unhealty in ctdb_control_pull_db: %s\n",
285                                      ctdb_db->db_name, ctdb_db->unhealthy_reason));
286         }
287
288         if (ctdb_lockdb_mark(ctdb_db) != 0) {
289                 DEBUG(DEBUG_ERR,(__location__ " Failed to get lock on entire db - failing\n"));
290                 return -1;
291         }
292
293         if (tdb_traverse_read(ctdb_db->ltdb->tdb, traverse_pulldb, &params) == -1) {
294                 DEBUG(DEBUG_ERR,(__location__ " Failed to get traverse db '%s'\n", ctdb_db->db_name));
295                 ctdb_lockdb_unmark(ctdb_db);
296                 talloc_free(params.pulldata);
297                 return -1;
298         }
299
300         ctdb_lockdb_unmark(ctdb_db);
301
302         outdata->dptr = (uint8_t *)params.pulldata;
303         outdata->dsize = params.len;
304
305         if (ctdb->tunable.db_record_count_warn != 0 && params.pulldata->count > ctdb->tunable.db_record_count_warn) {
306                 DEBUG(DEBUG_ERR,("Database %s is big. Contains %d records\n", ctdb_db->db_name, params.pulldata->count));
307         }
308         if (ctdb->tunable.db_size_warn != 0 && outdata->dsize > ctdb->tunable.db_size_warn) {
309                 DEBUG(DEBUG_ERR,("Database %s is big. Contains %d bytes\n", ctdb_db->db_name, (int)outdata->dsize));
310         }
311
312
313         return 0;
314 }
315
316 /*
317   push a bunch of records into a ltdb, filtering by rsn
318  */
319 int32_t ctdb_control_push_db(struct ctdb_context *ctdb, TDB_DATA indata)
320 {
321         struct ctdb_marshall_buffer *reply = (struct ctdb_marshall_buffer *)indata.dptr;
322         struct ctdb_db_context *ctdb_db;
323         int i, ret;
324         struct ctdb_rec_data_old *rec;
325
326         if (indata.dsize < offsetof(struct ctdb_marshall_buffer, data)) {
327                 DEBUG(DEBUG_ERR,(__location__ " invalid data in pulldb reply\n"));
328                 return -1;
329         }
330
331         ctdb_db = find_ctdb_db(ctdb, reply->db_id);
332         if (!ctdb_db) {
333                 DEBUG(DEBUG_ERR,(__location__ " Unknown db 0x%08x\n", reply->db_id));
334                 return -1;
335         }
336
337         if (!ctdb_db_frozen(ctdb_db)) {
338                 DEBUG(DEBUG_ERR,
339                       ("rejecting ctdb_control_push_db when not frozen\n"));
340                 return -1;
341         }
342
343         if (ctdb_lockdb_mark(ctdb_db) != 0) {
344                 DEBUG(DEBUG_ERR,(__location__ " Failed to get lock on entire db - failing\n"));
345                 return -1;
346         }
347
348         rec = (struct ctdb_rec_data_old *)&reply->data[0];
349
350         DEBUG(DEBUG_INFO,("starting push of %u records for dbid 0x%x\n",
351                  reply->count, reply->db_id));
352
353         for (i=0;i<reply->count;i++) {
354                 TDB_DATA key, data;
355                 struct ctdb_ltdb_header *hdr;
356
357                 key.dptr = &rec->data[0];
358                 key.dsize = rec->keylen;
359                 data.dptr = &rec->data[key.dsize];
360                 data.dsize = rec->datalen;
361
362                 if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
363                         DEBUG(DEBUG_CRIT,(__location__ " bad ltdb record\n"));
364                         goto failed;
365                 }
366                 hdr = (struct ctdb_ltdb_header *)data.dptr;
367                 /* strip off any read only record flags. All readonly records
368                    are revoked implicitely by a recovery
369                 */
370                 hdr->flags &= ~CTDB_REC_RO_FLAGS;
371
372                 data.dptr += sizeof(*hdr);
373                 data.dsize -= sizeof(*hdr);
374
375                 ret = ctdb_ltdb_store(ctdb_db, key, hdr, data);
376                 if (ret != 0) {
377                         DEBUG(DEBUG_CRIT, (__location__ " Unable to store record\n"));
378                         goto failed;
379                 }
380
381                 rec = (struct ctdb_rec_data_old *)(rec->length + (uint8_t *)rec);
382         }           
383
384         DEBUG(DEBUG_DEBUG,("finished push of %u records for dbid 0x%x\n",
385                  reply->count, reply->db_id));
386
387         if (ctdb_db->readonly) {
388                 DEBUG(DEBUG_CRIT,("Clearing the tracking database for dbid 0x%x\n",
389                                   ctdb_db->db_id));
390                 if (tdb_wipe_all(ctdb_db->rottdb) != 0) {
391                         DEBUG(DEBUG_ERR,("Failed to wipe tracking database for 0x%x. Dropping read-only delegation support\n", ctdb_db->db_id));
392                         ctdb_db->readonly = false;
393                         tdb_close(ctdb_db->rottdb);
394                         ctdb_db->rottdb = NULL;
395                         ctdb_db->readonly = false;
396                 }
397                 while (ctdb_db->revokechild_active != NULL) {
398                         talloc_free(ctdb_db->revokechild_active);
399                 }
400         }
401
402         ctdb_lockdb_unmark(ctdb_db);
403         return 0;
404
405 failed:
406         ctdb_lockdb_unmark(ctdb_db);
407         return -1;
408 }
409
410 struct ctdb_set_recmode_state {
411         struct ctdb_context *ctdb;
412         struct ctdb_req_control_old *c;
413         uint32_t recmode;
414         int fd[2];
415         struct tevent_timer *te;
416         struct tevent_fd *fde;
417         pid_t child;
418         struct timeval start_time;
419 };
420
421 /*
422   called if our set_recmode child times out. this would happen if
423   ctdb_recovery_lock() would block.
424  */
425 static void ctdb_set_recmode_timeout(struct tevent_context *ev,
426                                      struct tevent_timer *te,
427                                      struct timeval t, void *private_data)
428 {
429         struct ctdb_set_recmode_state *state = talloc_get_type(private_data, 
430                                            struct ctdb_set_recmode_state);
431
432         /* we consider this a success, not a failure, as we failed to
433            set the recovery lock which is what we wanted.  This can be
434            caused by the cluster filesystem being very slow to
435            arbitrate locks immediately after a node failure.       
436          */
437         DEBUG(DEBUG_ERR,(__location__ " set_recmode child process hung/timedout CFS slow to grant locks? (allowing recmode set anyway)\n"));
438         state->ctdb->recovery_mode = state->recmode;
439         ctdb_request_control_reply(state->ctdb, state->c, NULL, 0, NULL);
440         talloc_free(state);
441 }
442
443
444 /* when we free the recmode state we must kill any child process.
445 */
446 static int set_recmode_destructor(struct ctdb_set_recmode_state *state)
447 {
448         double l = timeval_elapsed(&state->start_time);
449
450         CTDB_UPDATE_RECLOCK_LATENCY(state->ctdb, "daemon reclock", reclock.ctdbd, l);
451
452         if (state->fd[0] != -1) {
453                 state->fd[0] = -1;
454         }
455         if (state->fd[1] != -1) {
456                 state->fd[1] = -1;
457         }
458         ctdb_kill(state->ctdb, state->child, SIGKILL);
459         return 0;
460 }
461
462 /* this is called when the client process has completed ctdb_recovery_lock()
463    and has written data back to us through the pipe.
464 */
465 static void set_recmode_handler(struct tevent_context *ev,
466                                 struct tevent_fd *fde,
467                                 uint16_t flags, void *private_data)
468 {
469         struct ctdb_set_recmode_state *state= talloc_get_type(private_data, 
470                                              struct ctdb_set_recmode_state);
471         char c = 0;
472         int ret;
473
474         /* we got a response from our child process so we can abort the
475            timeout.
476         */
477         talloc_free(state->te);
478         state->te = NULL;
479
480
481         /* If, as expected, the child was unable to take the recovery
482          * lock then it will have written 0 into the pipe, so
483          * continue.  However, any other value (e.g. 1) indicates that
484          * it was able to take the recovery lock when it should have
485          * been held by the recovery daemon on the recovery master.
486         */
487         ret = sys_read(state->fd[0], &c, 1);
488         if (ret != 1 || c != 0) {
489                 ctdb_request_control_reply(
490                         state->ctdb, state->c, NULL, -1,
491                         "Took recovery lock from daemon during recovery - probably a cluster filesystem lock coherence problem");
492                 talloc_free(state);
493                 return;
494         }
495
496         state->ctdb->recovery_mode = state->recmode;
497
498         /* release any deferred attach calls from clients */
499         if (state->recmode == CTDB_RECOVERY_NORMAL) {
500                 ctdb_process_deferred_attach(state->ctdb);
501         }
502
503         ctdb_request_control_reply(state->ctdb, state->c, NULL, 0, NULL);
504         talloc_free(state);
505         return;
506 }
507
508 static void
509 ctdb_drop_all_ips_event(struct tevent_context *ev, struct tevent_timer *te,
510                         struct timeval t, void *private_data)
511 {
512         struct ctdb_context *ctdb = talloc_get_type(private_data, struct ctdb_context);
513
514         DEBUG(DEBUG_ERR,(__location__ " Been in recovery mode for too long. Dropping all IPS\n"));
515         talloc_free(ctdb->release_ips_ctx);
516         ctdb->release_ips_ctx = NULL;
517
518         ctdb_release_all_ips(ctdb);
519 }
520
521 /*
522  * Set up an event to drop all public ips if we remain in recovery for too
523  * long
524  */
525 int ctdb_deferred_drop_all_ips(struct ctdb_context *ctdb)
526 {
527         if (ctdb->release_ips_ctx != NULL) {
528                 talloc_free(ctdb->release_ips_ctx);
529         }
530         ctdb->release_ips_ctx = talloc_new(ctdb);
531         CTDB_NO_MEMORY(ctdb, ctdb->release_ips_ctx);
532
533         tevent_add_timer(ctdb->ev, ctdb->release_ips_ctx,
534                          timeval_current_ofs(ctdb->tunable.recovery_drop_all_ips, 0),
535                          ctdb_drop_all_ips_event, ctdb);
536         return 0;
537 }
538
539 /*
540   set the recovery mode
541  */
542 int32_t ctdb_control_set_recmode(struct ctdb_context *ctdb, 
543                                  struct ctdb_req_control_old *c,
544                                  TDB_DATA indata, bool *async_reply,
545                                  const char **errormsg)
546 {
547         uint32_t recmode = *(uint32_t *)indata.dptr;
548         int i, ret;
549         struct ctdb_set_recmode_state *state;
550         pid_t parent = getpid();
551         struct ctdb_db_context *ctdb_db;
552
553         /* if we enter recovery but stay in recovery for too long
554            we will eventually drop all our ip addresses
555         */
556         if (recmode == CTDB_RECOVERY_NORMAL) {
557                 talloc_free(ctdb->release_ips_ctx);
558                 ctdb->release_ips_ctx = NULL;
559         } else {
560                 if (ctdb_deferred_drop_all_ips(ctdb) != 0) {
561                         DEBUG(DEBUG_ERR,("Failed to set up deferred drop all ips\n"));
562                 }
563         }
564
565         if (recmode != ctdb->recovery_mode) {
566                 DEBUG(DEBUG_NOTICE,(__location__ " Recovery mode set to %s\n", 
567                          recmode==CTDB_RECOVERY_NORMAL?"NORMAL":"ACTIVE"));
568         }
569
570         if (recmode != CTDB_RECOVERY_NORMAL ||
571             ctdb->recovery_mode != CTDB_RECOVERY_ACTIVE) {
572                 ctdb->recovery_mode = recmode;
573                 return 0;
574         }
575
576         /* some special handling when ending recovery mode */
577
578         for (ctdb_db = ctdb->db_list; ctdb_db != NULL; ctdb_db = ctdb_db->next) {
579                 if (ctdb_db->generation != ctdb->vnn_map->generation) {
580                         DEBUG(DEBUG_ERR,
581                               ("Inconsistent DB generation %u for %s\n",
582                                ctdb_db->generation, ctdb_db->db_name));
583                         DEBUG(DEBUG_ERR, ("Recovery mode set to ACTIVE\n"));
584                         return -1;
585                 }
586         }
587
588         /* force the databases to thaw */
589         for (i=1; i<=NUM_DB_PRIORITIES; i++) {
590                 if (ctdb_db_prio_frozen(ctdb, i)) {
591                         ctdb_control_thaw(ctdb, i, false);
592                 }
593         }
594
595         /* release any deferred attach calls from clients */
596         if (recmode == CTDB_RECOVERY_NORMAL) {
597                 ctdb_process_deferred_attach(ctdb);
598         }
599
600         if (ctdb->recovery_lock_file == NULL) {
601                 /* Not using recovery lock file */
602                 ctdb->recovery_mode = recmode;
603                 return 0;
604         }
605
606         state = talloc(ctdb, struct ctdb_set_recmode_state);
607         CTDB_NO_MEMORY(ctdb, state);
608
609         state->start_time = timeval_current();
610         state->fd[0] = -1;
611         state->fd[1] = -1;
612
613         /* For the rest of what needs to be done, we need to do this in
614            a child process since 
615            1, the call to ctdb_recovery_lock() can block if the cluster
616               filesystem is in the process of recovery.
617         */
618         ret = pipe(state->fd);
619         if (ret != 0) {
620                 talloc_free(state);
621                 DEBUG(DEBUG_CRIT,(__location__ " Failed to open pipe for set_recmode child\n"));
622                 return -1;
623         }
624
625         state->child = ctdb_fork(ctdb);
626         if (state->child == (pid_t)-1) {
627                 close(state->fd[0]);
628                 close(state->fd[1]);
629                 talloc_free(state);
630                 return -1;
631         }
632
633         if (state->child == 0) {
634                 char cc = 0;
635                 close(state->fd[0]);
636
637                 prctl_set_comment("ctdb_recmode");
638                 debug_extra = talloc_asprintf(NULL, "set_recmode:");
639                 /* Daemon should not be able to get the recover lock,
640                  * as it should be held by the recovery master */
641                 if (ctdb_recovery_lock(ctdb)) {
642                         DEBUG(DEBUG_ERR,
643                               ("ERROR: Daemon able to take recovery lock on \"%s\" during recovery\n",
644                                ctdb->recovery_lock_file));
645                         ctdb_recovery_unlock(ctdb);
646                         cc = 1;
647                 }
648
649                 sys_write(state->fd[1], &cc, 1);
650                 ctdb_wait_for_process_to_exit(parent);
651                 _exit(0);
652         }
653         close(state->fd[1]);
654         set_close_on_exec(state->fd[0]);
655
656         state->fd[1] = -1;
657
658         talloc_set_destructor(state, set_recmode_destructor);
659
660         DEBUG(DEBUG_DEBUG, (__location__ " Created PIPE FD:%d for setrecmode\n", state->fd[0]));
661
662         state->te = tevent_add_timer(ctdb->ev, state, timeval_current_ofs(5, 0),
663                                      ctdb_set_recmode_timeout, state);
664
665         state->fde = tevent_add_fd(ctdb->ev, state, state->fd[0], TEVENT_FD_READ,
666                                    set_recmode_handler, (void *)state);
667
668         if (state->fde == NULL) {
669                 talloc_free(state);
670                 return -1;
671         }
672         tevent_fd_set_auto_close(state->fde);
673
674         state->ctdb    = ctdb;
675         state->recmode = recmode;
676         state->c       = talloc_steal(state, c);
677
678         *async_reply = true;
679
680         return 0;
681 }
682
683
684 bool ctdb_recovery_have_lock(struct ctdb_context *ctdb)
685 {
686         return ctdb->recovery_lock_fd != -1;
687 }
688
689 /*
690   try and get the recovery lock in shared storage - should only work
691   on the recovery master recovery daemon. Anywhere else is a bug
692  */
693 bool ctdb_recovery_lock(struct ctdb_context *ctdb)
694 {
695         struct flock lock;
696
697         ctdb->recovery_lock_fd = open(ctdb->recovery_lock_file,
698                                       O_RDWR|O_CREAT, 0600);
699         if (ctdb->recovery_lock_fd == -1) {
700                 DEBUG(DEBUG_ERR,
701                       ("ctdb_recovery_lock: Unable to open %s - (%s)\n",
702                        ctdb->recovery_lock_file, strerror(errno)));
703                 return false;
704         }
705
706         set_close_on_exec(ctdb->recovery_lock_fd);
707
708         lock.l_type = F_WRLCK;
709         lock.l_whence = SEEK_SET;
710         lock.l_start = 0;
711         lock.l_len = 1;
712         lock.l_pid = 0;
713
714         if (fcntl(ctdb->recovery_lock_fd, F_SETLK, &lock) != 0) {
715                 int saved_errno = errno;
716                 close(ctdb->recovery_lock_fd);
717                 ctdb->recovery_lock_fd = -1;
718                 /* Fail silently on these errors, since they indicate
719                  * lock contention, but log an error for any other
720                  * failure. */
721                 if (saved_errno != EACCES &&
722                     saved_errno != EAGAIN) {
723                         DEBUG(DEBUG_ERR,("ctdb_recovery_lock: Failed to get "
724                                          "recovery lock on '%s' - (%s)\n",
725                                          ctdb->recovery_lock_file,
726                                          strerror(saved_errno)));
727                 }
728                 return false;
729         }
730
731         return true;
732 }
733
734 void ctdb_recovery_unlock(struct ctdb_context *ctdb)
735 {
736         if (ctdb->recovery_lock_fd != -1) {
737                 DEBUG(DEBUG_NOTICE, ("Releasing recovery lock\n"));
738                 close(ctdb->recovery_lock_fd);
739                 ctdb->recovery_lock_fd = -1;
740         }
741 }
742
743 /*
744   delete a record as part of the vacuum process
745   only delete if we are not lmaster or dmaster, and our rsn is <= the provided rsn
746   use non-blocking locks
747
748   return 0 if the record was successfully deleted (i.e. it does not exist
749   when the function returns)
750   or !0 is the record still exists in the tdb after returning.
751  */
752 static int delete_tdb_record(struct ctdb_context *ctdb, struct ctdb_db_context *ctdb_db, struct ctdb_rec_data_old *rec)
753 {
754         TDB_DATA key, data, data2;
755         struct ctdb_ltdb_header *hdr, *hdr2;
756         
757         /* these are really internal tdb functions - but we need them here for
758            non-blocking lock of the freelist */
759         int tdb_lock_nonblock(struct tdb_context *tdb, int list, int ltype);
760         int tdb_unlock(struct tdb_context *tdb, int list, int ltype);
761
762
763         key.dsize = rec->keylen;
764         key.dptr  = &rec->data[0];
765         data.dsize = rec->datalen;
766         data.dptr = &rec->data[rec->keylen];
767
768         if (ctdb_lmaster(ctdb, &key) == ctdb->pnn) {
769                 DEBUG(DEBUG_INFO,(__location__ " Called delete on record where we are lmaster\n"));
770                 return -1;
771         }
772
773         if (data.dsize != sizeof(struct ctdb_ltdb_header)) {
774                 DEBUG(DEBUG_ERR,(__location__ " Bad record size\n"));
775                 return -1;
776         }
777
778         hdr = (struct ctdb_ltdb_header *)data.dptr;
779
780         /* use a non-blocking lock */
781         if (tdb_chainlock_nonblock(ctdb_db->ltdb->tdb, key) != 0) {
782                 return -1;
783         }
784
785         data2 = tdb_fetch(ctdb_db->ltdb->tdb, key);
786         if (data2.dptr == NULL) {
787                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
788                 return 0;
789         }
790
791         if (data2.dsize < sizeof(struct ctdb_ltdb_header)) {
792                 if (tdb_lock_nonblock(ctdb_db->ltdb->tdb, -1, F_WRLCK) == 0) {
793                         if (tdb_delete(ctdb_db->ltdb->tdb, key) != 0) {
794                                 DEBUG(DEBUG_CRIT,(__location__ " Failed to delete corrupt record\n"));
795                         }
796                         tdb_unlock(ctdb_db->ltdb->tdb, -1, F_WRLCK);
797                         DEBUG(DEBUG_CRIT,(__location__ " Deleted corrupt record\n"));
798                 }
799                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
800                 free(data2.dptr);
801                 return 0;
802         }
803         
804         hdr2 = (struct ctdb_ltdb_header *)data2.dptr;
805
806         if (hdr2->rsn > hdr->rsn) {
807                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
808                 DEBUG(DEBUG_INFO,(__location__ " Skipping record with rsn=%llu - called with rsn=%llu\n",
809                          (unsigned long long)hdr2->rsn, (unsigned long long)hdr->rsn));
810                 free(data2.dptr);
811                 return -1;
812         }
813
814         /* do not allow deleting record that have readonly flags set. */
815         if (hdr->flags & CTDB_REC_RO_FLAGS) {
816                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
817                 DEBUG(DEBUG_INFO,(__location__ " Skipping record with readonly flags set\n"));
818                 free(data2.dptr);
819                 return -1;
820         }
821         if (hdr2->flags & CTDB_REC_RO_FLAGS) {
822                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
823                 DEBUG(DEBUG_INFO,(__location__ " Skipping record with readonly flags set\n"));
824                 free(data2.dptr);
825                 return -1;
826         }
827
828         if (hdr2->dmaster == ctdb->pnn) {
829                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
830                 DEBUG(DEBUG_INFO,(__location__ " Attempted delete record where we are the dmaster\n"));
831                 free(data2.dptr);
832                 return -1;
833         }
834
835         if (tdb_lock_nonblock(ctdb_db->ltdb->tdb, -1, F_WRLCK) != 0) {
836                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
837                 free(data2.dptr);
838                 return -1;
839         }
840
841         if (tdb_delete(ctdb_db->ltdb->tdb, key) != 0) {
842                 tdb_unlock(ctdb_db->ltdb->tdb, -1, F_WRLCK);
843                 tdb_chainunlock(ctdb_db->ltdb->tdb, key);
844                 DEBUG(DEBUG_INFO,(__location__ " Failed to delete record\n"));
845                 free(data2.dptr);
846                 return -1;
847         }
848
849         tdb_unlock(ctdb_db->ltdb->tdb, -1, F_WRLCK);
850         tdb_chainunlock(ctdb_db->ltdb->tdb, key);
851         free(data2.dptr);
852         return 0;
853 }
854
855
856
857 struct recovery_callback_state {
858         struct ctdb_req_control_old *c;
859 };
860
861
862 /*
863   called when the 'recovered' event script has finished
864  */
865 static void ctdb_end_recovery_callback(struct ctdb_context *ctdb, int status, void *p)
866 {
867         struct recovery_callback_state *state = talloc_get_type(p, struct recovery_callback_state);
868
869         ctdb_enable_monitoring(ctdb);
870         CTDB_INCREMENT_STAT(ctdb, num_recoveries);
871
872         if (status != 0) {
873                 DEBUG(DEBUG_ERR,(__location__ " recovered event script failed (status %d)\n", status));
874                 if (status == -ETIME) {
875                         ctdb_ban_self(ctdb);
876                 }
877         }
878
879         ctdb_request_control_reply(ctdb, state->c, NULL, status, NULL);
880         talloc_free(state);
881
882         gettimeofday(&ctdb->last_recovery_finished, NULL);
883
884         if (ctdb->runstate == CTDB_RUNSTATE_FIRST_RECOVERY) {
885                 ctdb_set_runstate(ctdb, CTDB_RUNSTATE_STARTUP);
886         }
887 }
888
889 /*
890   recovery has finished
891  */
892 int32_t ctdb_control_end_recovery(struct ctdb_context *ctdb, 
893                                 struct ctdb_req_control_old *c,
894                                 bool *async_reply)
895 {
896         int ret;
897         struct recovery_callback_state *state;
898
899         DEBUG(DEBUG_NOTICE,("Recovery has finished\n"));
900
901         ctdb_persistent_finish_trans3_commits(ctdb);
902
903         state = talloc(ctdb, struct recovery_callback_state);
904         CTDB_NO_MEMORY(ctdb, state);
905
906         state->c    = c;
907
908         ctdb_disable_monitoring(ctdb);
909
910         ret = ctdb_event_script_callback(ctdb, state,
911                                          ctdb_end_recovery_callback, 
912                                          state, 
913                                          CTDB_EVENT_RECOVERED, "%s", "");
914
915         if (ret != 0) {
916                 ctdb_enable_monitoring(ctdb);
917
918                 DEBUG(DEBUG_ERR,(__location__ " Failed to end recovery\n"));
919                 talloc_free(state);
920                 return -1;
921         }
922
923         /* tell the control that we will be reply asynchronously */
924         state->c    = talloc_steal(state, c);
925         *async_reply = true;
926         return 0;
927 }
928
929 /*
930   called when the 'startrecovery' event script has finished
931  */
932 static void ctdb_start_recovery_callback(struct ctdb_context *ctdb, int status, void *p)
933 {
934         struct recovery_callback_state *state = talloc_get_type(p, struct recovery_callback_state);
935
936         if (status != 0) {
937                 DEBUG(DEBUG_ERR,(__location__ " startrecovery event script failed (status %d)\n", status));
938         }
939
940         ctdb_request_control_reply(ctdb, state->c, NULL, status, NULL);
941         talloc_free(state);
942 }
943
944 /*
945   run the startrecovery eventscript
946  */
947 int32_t ctdb_control_start_recovery(struct ctdb_context *ctdb, 
948                                 struct ctdb_req_control_old *c,
949                                 bool *async_reply)
950 {
951         int ret;
952         struct recovery_callback_state *state;
953
954         DEBUG(DEBUG_NOTICE,(__location__ " startrecovery eventscript has been invoked\n"));
955         gettimeofday(&ctdb->last_recovery_started, NULL);
956
957         state = talloc(ctdb, struct recovery_callback_state);
958         CTDB_NO_MEMORY(ctdb, state);
959
960         state->c    = talloc_steal(state, c);
961
962         ctdb_disable_monitoring(ctdb);
963
964         ret = ctdb_event_script_callback(ctdb, state,
965                                          ctdb_start_recovery_callback, 
966                                          state,
967                                          CTDB_EVENT_START_RECOVERY,
968                                          "%s", "");
969
970         if (ret != 0) {
971                 DEBUG(DEBUG_ERR,(__location__ " Failed to start recovery\n"));
972                 talloc_free(state);
973                 return -1;
974         }
975
976         /* tell the control that we will be reply asynchronously */
977         *async_reply = true;
978         return 0;
979 }
980
981 /*
982  try to delete all these records as part of the vacuuming process
983  and return the records we failed to delete
984 */
985 int32_t ctdb_control_try_delete_records(struct ctdb_context *ctdb, TDB_DATA indata, TDB_DATA *outdata)
986 {
987         struct ctdb_marshall_buffer *reply = (struct ctdb_marshall_buffer *)indata.dptr;
988         struct ctdb_db_context *ctdb_db;
989         int i;
990         struct ctdb_rec_data_old *rec;
991         struct ctdb_marshall_buffer *records;
992
993         if (indata.dsize < offsetof(struct ctdb_marshall_buffer, data)) {
994                 DEBUG(DEBUG_ERR,(__location__ " invalid data in try_delete_records\n"));
995                 return -1;
996         }
997
998         ctdb_db = find_ctdb_db(ctdb, reply->db_id);
999         if (!ctdb_db) {
1000                 DEBUG(DEBUG_ERR,(__location__ " Unknown db 0x%08x\n", reply->db_id));
1001                 return -1;
1002         }
1003
1004
1005         DEBUG(DEBUG_DEBUG,("starting try_delete_records of %u records for dbid 0x%x\n",
1006                  reply->count, reply->db_id));
1007
1008
1009         /* create a blob to send back the records we couldnt delete */  
1010         records = (struct ctdb_marshall_buffer *)
1011                         talloc_zero_size(outdata, 
1012                                     offsetof(struct ctdb_marshall_buffer, data));
1013         if (records == NULL) {
1014                 DEBUG(DEBUG_ERR,(__location__ " Out of memory\n"));
1015                 return -1;
1016         }
1017         records->db_id = ctdb_db->db_id;
1018
1019
1020         rec = (struct ctdb_rec_data_old *)&reply->data[0];
1021         for (i=0;i<reply->count;i++) {
1022                 TDB_DATA key, data;
1023
1024                 key.dptr = &rec->data[0];
1025                 key.dsize = rec->keylen;
1026                 data.dptr = &rec->data[key.dsize];
1027                 data.dsize = rec->datalen;
1028
1029                 if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
1030                         DEBUG(DEBUG_CRIT,(__location__ " bad ltdb record in indata\n"));
1031                         return -1;
1032                 }
1033
1034                 /* If we cant delete the record we must add it to the reply
1035                    so the lmaster knows it may not purge this record
1036                 */
1037                 if (delete_tdb_record(ctdb, ctdb_db, rec) != 0) {
1038                         size_t old_size;
1039                         struct ctdb_ltdb_header *hdr;
1040
1041                         hdr = (struct ctdb_ltdb_header *)data.dptr;
1042                         data.dptr += sizeof(*hdr);
1043                         data.dsize -= sizeof(*hdr);
1044
1045                         DEBUG(DEBUG_INFO, (__location__ " Failed to vacuum delete record with hash 0x%08x\n", ctdb_hash(&key)));
1046
1047                         old_size = talloc_get_size(records);
1048                         records = talloc_realloc_size(outdata, records, old_size + rec->length);
1049                         if (records == NULL) {
1050                                 DEBUG(DEBUG_ERR,(__location__ " Failed to expand\n"));
1051                                 return -1;
1052                         }
1053                         records->count++;
1054                         memcpy(old_size+(uint8_t *)records, rec, rec->length);
1055                 } 
1056
1057                 rec = (struct ctdb_rec_data_old *)(rec->length + (uint8_t *)rec);
1058         }           
1059
1060
1061         *outdata = ctdb_marshall_finish(records);
1062
1063         return 0;
1064 }
1065
1066 /**
1067  * Store a record as part of the vacuum process:
1068  * This is called from the RECEIVE_RECORD control which
1069  * the lmaster uses to send the current empty copy
1070  * to all nodes for storing, before it lets the other
1071  * nodes delete the records in the second phase with
1072  * the TRY_DELETE_RECORDS control.
1073  *
1074  * Only store if we are not lmaster or dmaster, and our
1075  * rsn is <= the provided rsn. Use non-blocking locks.
1076  *
1077  * return 0 if the record was successfully stored.
1078  * return !0 if the record still exists in the tdb after returning.
1079  */
1080 static int store_tdb_record(struct ctdb_context *ctdb,
1081                             struct ctdb_db_context *ctdb_db,
1082                             struct ctdb_rec_data_old *rec)
1083 {
1084         TDB_DATA key, data, data2;
1085         struct ctdb_ltdb_header *hdr, *hdr2;
1086         int ret;
1087
1088         key.dsize = rec->keylen;
1089         key.dptr = &rec->data[0];
1090         data.dsize = rec->datalen;
1091         data.dptr = &rec->data[rec->keylen];
1092
1093         if (ctdb_lmaster(ctdb, &key) == ctdb->pnn) {
1094                 DEBUG(DEBUG_INFO, (__location__ " Called store_tdb_record "
1095                                    "where we are lmaster\n"));
1096                 return -1;
1097         }
1098
1099         if (data.dsize != sizeof(struct ctdb_ltdb_header)) {
1100                 DEBUG(DEBUG_ERR, (__location__ " Bad record size\n"));
1101                 return -1;
1102         }
1103
1104         hdr = (struct ctdb_ltdb_header *)data.dptr;
1105
1106         /* use a non-blocking lock */
1107         if (tdb_chainlock_nonblock(ctdb_db->ltdb->tdb, key) != 0) {
1108                 DEBUG(DEBUG_INFO, (__location__ " Failed to lock chain in non-blocking mode\n"));
1109                 return -1;
1110         }
1111
1112         data2 = tdb_fetch(ctdb_db->ltdb->tdb, key);
1113         if (data2.dptr == NULL || data2.dsize < sizeof(struct ctdb_ltdb_header)) {
1114                 if (tdb_store(ctdb_db->ltdb->tdb, key, data, 0) == -1) {
1115                         DEBUG(DEBUG_ERR, (__location__ "Failed to store record\n"));
1116                         ret = -1;
1117                         goto done;
1118                 }
1119                 DEBUG(DEBUG_INFO, (__location__ " Stored record\n"));
1120                 ret = 0;
1121                 goto done;
1122         }
1123
1124         hdr2 = (struct ctdb_ltdb_header *)data2.dptr;
1125
1126         if (hdr2->rsn > hdr->rsn) {
1127                 DEBUG(DEBUG_INFO, (__location__ " Skipping record with "
1128                                    "rsn=%llu - called with rsn=%llu\n",
1129                                    (unsigned long long)hdr2->rsn,
1130                                    (unsigned long long)hdr->rsn));
1131                 ret = -1;
1132                 goto done;
1133         }
1134
1135         /* do not allow vacuuming of records that have readonly flags set. */
1136         if (hdr->flags & CTDB_REC_RO_FLAGS) {
1137                 DEBUG(DEBUG_INFO,(__location__ " Skipping record with readonly "
1138                                   "flags set\n"));
1139                 ret = -1;
1140                 goto done;
1141         }
1142         if (hdr2->flags & CTDB_REC_RO_FLAGS) {
1143                 DEBUG(DEBUG_INFO,(__location__ " Skipping record with readonly "
1144                                   "flags set\n"));
1145                 ret = -1;
1146                 goto done;
1147         }
1148
1149         if (hdr2->dmaster == ctdb->pnn) {
1150                 DEBUG(DEBUG_INFO, (__location__ " Attempted to store record "
1151                                    "where we are the dmaster\n"));
1152                 ret = -1;
1153                 goto done;
1154         }
1155
1156         if (tdb_store(ctdb_db->ltdb->tdb, key, data, 0) != 0) {
1157                 DEBUG(DEBUG_INFO,(__location__ " Failed to store record\n"));
1158                 ret = -1;
1159                 goto done;
1160         }
1161
1162         ret = 0;
1163
1164 done:
1165         tdb_chainunlock(ctdb_db->ltdb->tdb, key);
1166         free(data2.dptr);
1167         return  ret;
1168 }
1169
1170
1171
1172 /**
1173  * Try to store all these records as part of the vacuuming process
1174  * and return the records we failed to store.
1175  */
1176 int32_t ctdb_control_receive_records(struct ctdb_context *ctdb,
1177                                      TDB_DATA indata, TDB_DATA *outdata)
1178 {
1179         struct ctdb_marshall_buffer *reply = (struct ctdb_marshall_buffer *)indata.dptr;
1180         struct ctdb_db_context *ctdb_db;
1181         int i;
1182         struct ctdb_rec_data_old *rec;
1183         struct ctdb_marshall_buffer *records;
1184
1185         if (indata.dsize < offsetof(struct ctdb_marshall_buffer, data)) {
1186                 DEBUG(DEBUG_ERR,
1187                       (__location__ " invalid data in receive_records\n"));
1188                 return -1;
1189         }
1190
1191         ctdb_db = find_ctdb_db(ctdb, reply->db_id);
1192         if (!ctdb_db) {
1193                 DEBUG(DEBUG_ERR, (__location__ " Unknown db 0x%08x\n",
1194                                   reply->db_id));
1195                 return -1;
1196         }
1197
1198         DEBUG(DEBUG_DEBUG, ("starting receive_records of %u records for "
1199                             "dbid 0x%x\n", reply->count, reply->db_id));
1200
1201         /* create a blob to send back the records we could not store */
1202         records = (struct ctdb_marshall_buffer *)
1203                         talloc_zero_size(outdata,
1204                                 offsetof(struct ctdb_marshall_buffer, data));
1205         if (records == NULL) {
1206                 DEBUG(DEBUG_ERR, (__location__ " Out of memory\n"));
1207                 return -1;
1208         }
1209         records->db_id = ctdb_db->db_id;
1210
1211         rec = (struct ctdb_rec_data_old *)&reply->data[0];
1212         for (i=0; i<reply->count; i++) {
1213                 TDB_DATA key, data;
1214
1215                 key.dptr = &rec->data[0];
1216                 key.dsize = rec->keylen;
1217                 data.dptr = &rec->data[key.dsize];
1218                 data.dsize = rec->datalen;
1219
1220                 if (data.dsize < sizeof(struct ctdb_ltdb_header)) {
1221                         DEBUG(DEBUG_CRIT, (__location__ " bad ltdb record "
1222                                            "in indata\n"));
1223                         return -1;
1224                 }
1225
1226                 /*
1227                  * If we can not store the record we must add it to the reply
1228                  * so the lmaster knows it may not purge this record.
1229                  */
1230                 if (store_tdb_record(ctdb, ctdb_db, rec) != 0) {
1231                         size_t old_size;
1232                         struct ctdb_ltdb_header *hdr;
1233
1234                         hdr = (struct ctdb_ltdb_header *)data.dptr;
1235                         data.dptr += sizeof(*hdr);
1236                         data.dsize -= sizeof(*hdr);
1237
1238                         DEBUG(DEBUG_INFO, (__location__ " Failed to store "
1239                                            "record with hash 0x%08x in vacuum "
1240                                            "via RECEIVE_RECORDS\n",
1241                                            ctdb_hash(&key)));
1242
1243                         old_size = talloc_get_size(records);
1244                         records = talloc_realloc_size(outdata, records,
1245                                                       old_size + rec->length);
1246                         if (records == NULL) {
1247                                 DEBUG(DEBUG_ERR, (__location__ " Failed to "
1248                                                   "expand\n"));
1249                                 return -1;
1250                         }
1251                         records->count++;
1252                         memcpy(old_size+(uint8_t *)records, rec, rec->length);
1253                 }
1254
1255                 rec = (struct ctdb_rec_data_old *)(rec->length + (uint8_t *)rec);
1256         }
1257
1258         *outdata = ctdb_marshall_finish(records);
1259
1260         return 0;
1261 }
1262
1263
1264 /*
1265   report capabilities
1266  */
1267 int32_t ctdb_control_get_capabilities(struct ctdb_context *ctdb, TDB_DATA *outdata)
1268 {
1269         uint32_t *capabilities = NULL;
1270
1271         capabilities = talloc(outdata, uint32_t);
1272         CTDB_NO_MEMORY(ctdb, capabilities);
1273         *capabilities = ctdb->capabilities;
1274
1275         outdata->dsize = sizeof(uint32_t);
1276         outdata->dptr = (uint8_t *)capabilities;
1277
1278         return 0;       
1279 }
1280
1281 /* The recovery daemon will ping us at regular intervals.
1282    If we havent been pinged for a while we assume the recovery
1283    daemon is inoperable and we restart.
1284 */
1285 static void ctdb_recd_ping_timeout(struct tevent_context *ev,
1286                                    struct tevent_timer *te,
1287                                    struct timeval t, void *p)
1288 {
1289         struct ctdb_context *ctdb = talloc_get_type(p, struct ctdb_context);
1290         uint32_t *count = talloc_get_type(ctdb->recd_ping_count, uint32_t);
1291
1292         DEBUG(DEBUG_ERR, ("Recovery daemon ping timeout. Count : %u\n", *count));
1293
1294         if (*count < ctdb->tunable.recd_ping_failcount) {
1295                 (*count)++;
1296                 tevent_add_timer(ctdb->ev, ctdb->recd_ping_count,
1297                                  timeval_current_ofs(ctdb->tunable.recd_ping_timeout, 0),
1298                                  ctdb_recd_ping_timeout, ctdb);
1299                 return;
1300         }
1301
1302         DEBUG(DEBUG_ERR, ("Final timeout for recovery daemon ping. Restarting recovery daemon. (This can be caused if the cluster filesystem has hung)\n"));
1303
1304         ctdb_stop_recoverd(ctdb);
1305         ctdb_start_recoverd(ctdb);
1306 }
1307
1308 int32_t ctdb_control_recd_ping(struct ctdb_context *ctdb)
1309 {
1310         talloc_free(ctdb->recd_ping_count);
1311
1312         ctdb->recd_ping_count = talloc_zero(ctdb, uint32_t);
1313         CTDB_NO_MEMORY(ctdb, ctdb->recd_ping_count);
1314
1315         if (ctdb->tunable.recd_ping_timeout != 0) {
1316                 tevent_add_timer(ctdb->ev, ctdb->recd_ping_count,
1317                                  timeval_current_ofs(ctdb->tunable.recd_ping_timeout, 0),
1318                                  ctdb_recd_ping_timeout, ctdb);
1319         }
1320
1321         return 0;
1322 }
1323
1324
1325
1326 int32_t ctdb_control_set_recmaster(struct ctdb_context *ctdb, uint32_t opcode, TDB_DATA indata)
1327 {
1328         uint32_t new_recmaster;
1329
1330         CHECK_CONTROL_DATA_SIZE(sizeof(uint32_t));
1331         new_recmaster = ((uint32_t *)(&indata.dptr[0]))[0];
1332
1333         if (ctdb->pnn != new_recmaster && ctdb->recovery_master == ctdb->pnn) {
1334                 DEBUG(DEBUG_NOTICE,
1335                       ("This node (%u) is no longer the recovery master\n", ctdb->pnn));
1336         }
1337
1338         if (ctdb->pnn == new_recmaster && ctdb->recovery_master != new_recmaster) {
1339                 DEBUG(DEBUG_NOTICE,
1340                       ("This node (%u) is now the recovery master\n", ctdb->pnn));
1341         }
1342
1343         ctdb->recovery_master = new_recmaster;
1344         return 0;
1345 }
1346
1347
1348 int32_t ctdb_control_stop_node(struct ctdb_context *ctdb)
1349 {
1350         DEBUG(DEBUG_NOTICE, ("Stopping node\n"));
1351         ctdb_disable_monitoring(ctdb);
1352         ctdb->nodes[ctdb->pnn]->flags |= NODE_FLAGS_STOPPED;
1353
1354         return 0;
1355 }
1356
1357 int32_t ctdb_control_continue_node(struct ctdb_context *ctdb)
1358 {
1359         DEBUG(DEBUG_NOTICE, ("Continue node\n"));
1360         ctdb->nodes[ctdb->pnn]->flags &= ~NODE_FLAGS_STOPPED;
1361
1362         return 0;
1363 }
1364