2007-05-03 05:06:24 +04:00
/*
2007-05-12 09:15:27 +04:00
ctdb recovery code
2007-05-03 05:06:24 +04:00
Copyright ( C ) Andrew Tridgell 2007
Copyright ( C ) Ronnie Sahlberg 2007
2007-05-31 07:50:53 +04:00
This program is free software ; you can redistribute it and / or modify
it under the terms of the GNU General Public License as published by
2007-07-10 09:29:31 +04:00
the Free Software Foundation ; either version 3 of the License , or
2007-05-31 07:50:53 +04:00
( at your option ) any later version .
This program is distributed in the hope that it will be useful ,
2007-05-03 05:06:24 +04:00
but WITHOUT ANY WARRANTY ; without even the implied warranty of
2007-05-31 07:50:53 +04:00
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the
GNU General Public License for more details .
You should have received a copy of the GNU General Public License
2007-07-10 09:29:31 +04:00
along with this program ; if not , see < http : //www.gnu.org/licenses/>.
2007-05-03 05:06:24 +04:00
*/
# include "includes.h"
# include "lib/events/events.h"
# include "lib/tdb/include/tdb.h"
# include "system/network.h"
# include "system/filesys.h"
# include "system/wait.h"
# include "../include/ctdb_private.h"
# include "lib/util/dlinklist.h"
# include "db_wrap.h"
2007-05-12 09:29:06 +04:00
/*
lock all databases - mark only
*/
static int ctdb_lock_all_databases_mark ( struct ctdb_context * ctdb )
{
struct ctdb_db_context * ctdb_db ;
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " Attempt to mark all databases locked when not frozen \n " ) ) ;
return - 1 ;
}
for ( ctdb_db = ctdb - > db_list ; ctdb_db ; ctdb_db = ctdb_db - > next ) {
if ( tdb_lockall_mark ( ctdb_db - > ltdb - > tdb ) ! = 0 ) {
return - 1 ;
}
}
return 0 ;
}
/*
lock all databases - unmark only
*/
static int ctdb_lock_all_databases_unmark ( struct ctdb_context * ctdb )
{
struct ctdb_db_context * ctdb_db ;
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " Attempt to unmark all databases locked when not frozen \n " ) ) ;
return - 1 ;
}
for ( ctdb_db = ctdb - > db_list ; ctdb_db ; ctdb_db = ctdb_db - > next ) {
if ( tdb_lockall_unmark ( ctdb_db - > ltdb - > tdb ) ! = 0 ) {
return - 1 ;
}
}
return 0 ;
}
2007-05-03 05:06:24 +04:00
int
ctdb_control_getvnnmap ( struct ctdb_context * ctdb , uint32_t opcode , TDB_DATA indata , TDB_DATA * outdata )
{
CHECK_CONTROL_DATA_SIZE ( 0 ) ;
2007-05-10 02:13:19 +04:00
struct ctdb_vnn_map_wire * map ;
size_t len ;
2007-05-03 05:06:24 +04:00
2007-05-10 02:13:19 +04:00
len = offsetof ( struct ctdb_vnn_map_wire , map ) + sizeof ( uint32_t ) * ctdb - > vnn_map - > size ;
map = talloc_size ( outdata , len ) ;
CTDB_NO_MEMORY_VOID ( ctdb , map ) ;
map - > generation = ctdb - > vnn_map - > generation ;
map - > size = ctdb - > vnn_map - > size ;
memcpy ( map - > map , ctdb - > vnn_map - > map , sizeof ( uint32_t ) * map - > size ) ;
outdata - > dsize = len ;
outdata - > dptr = ( uint8_t * ) map ;
2007-05-03 05:06:24 +04:00
return 0 ;
}
int
ctdb_control_setvnnmap ( struct ctdb_context * ctdb , uint32_t opcode , TDB_DATA indata , TDB_DATA * outdata )
{
2007-05-10 02:22:26 +04:00
struct ctdb_vnn_map_wire * map = ( struct ctdb_vnn_map_wire * ) indata . dptr ;
2007-05-14 07:48:40 +04:00
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " Attempt to set vnnmap when not frozen \n " ) ) ;
return - 1 ;
}
2007-05-10 02:22:26 +04:00
talloc_free ( ctdb - > vnn_map ) ;
ctdb - > vnn_map = talloc ( ctdb , struct ctdb_vnn_map ) ;
CTDB_NO_MEMORY ( ctdb , ctdb - > vnn_map ) ;
ctdb - > vnn_map - > generation = map - > generation ;
ctdb - > vnn_map - > size = map - > size ;
ctdb - > vnn_map - > map = talloc_array ( ctdb - > vnn_map , uint32_t , map - > size ) ;
CTDB_NO_MEMORY ( ctdb , ctdb - > vnn_map - > map ) ;
2007-05-03 05:06:24 +04:00
2007-05-10 02:22:26 +04:00
memcpy ( ctdb - > vnn_map - > map , map - > map , sizeof ( uint32_t ) * map - > size ) ;
2007-05-03 05:06:24 +04:00
return 0 ;
}
2007-05-03 07:07:34 +04:00
int
ctdb_control_getdbmap ( struct ctdb_context * ctdb , uint32_t opcode , TDB_DATA indata , TDB_DATA * outdata )
{
uint32_t i , len ;
struct ctdb_db_context * ctdb_db ;
struct ctdb_dbid_map * dbid_map ;
CHECK_CONTROL_DATA_SIZE ( 0 ) ;
len = 0 ;
for ( ctdb_db = ctdb - > db_list ; ctdb_db ; ctdb_db = ctdb_db - > next ) {
len + + ;
}
2007-09-21 06:24:02 +04:00
outdata - > dsize = offsetof ( struct ctdb_dbid_map , dbs ) + sizeof ( dbid_map - > dbs [ 0 ] ) * len ;
2007-05-03 07:07:34 +04:00
outdata - > dptr = ( unsigned char * ) talloc_zero_size ( outdata , outdata - > dsize ) ;
if ( ! outdata - > dptr ) {
2007-05-12 08:34:21 +04:00
DEBUG ( 0 , ( __location__ " Failed to allocate dbmap array \n " ) ) ;
2007-05-03 07:07:34 +04:00
exit ( 1 ) ;
}
dbid_map = ( struct ctdb_dbid_map * ) outdata - > dptr ;
dbid_map - > num = len ;
2007-09-21 06:24:02 +04:00
for ( i = 0 , ctdb_db = ctdb - > db_list ; ctdb_db ; i + + , ctdb_db = ctdb_db - > next ) {
dbid_map - > dbs [ i ] . dbid = ctdb_db - > db_id ;
dbid_map - > dbs [ i ] . persistent = ctdb_db - > persistent ;
2007-05-03 07:07:34 +04:00
}
return 0 ;
}
2007-05-03 07:30:38 +04:00
int
ctdb_control_getnodemap ( struct ctdb_context * ctdb , uint32_t opcode , TDB_DATA indata , TDB_DATA * outdata )
{
uint32_t i , num_nodes ;
struct ctdb_node_map * node_map ;
CHECK_CONTROL_DATA_SIZE ( 0 ) ;
2007-06-02 07:16:11 +04:00
num_nodes = ctdb - > num_nodes ;
2007-05-03 07:30:38 +04:00
outdata - > dsize = offsetof ( struct ctdb_node_map , nodes ) + num_nodes * sizeof ( struct ctdb_node_and_flags ) ;
outdata - > dptr = ( unsigned char * ) talloc_zero_size ( outdata , outdata - > dsize ) ;
if ( ! outdata - > dptr ) {
2007-05-12 08:34:21 +04:00
DEBUG ( 0 , ( __location__ " Failed to allocate nodemap array \n " ) ) ;
2007-05-03 07:30:38 +04:00
exit ( 1 ) ;
}
node_map = ( struct ctdb_node_map * ) outdata - > dptr ;
node_map - > num = num_nodes ;
for ( i = 0 ; i < num_nodes ; i + + ) {
2007-06-04 07:26:07 +04:00
inet_aton ( ctdb - > nodes [ i ] - > address . address , & node_map - > nodes [ i ] . sin . sin_addr ) ;
2007-09-04 03:50:07 +04:00
node_map - > nodes [ i ] . pnn = ctdb - > nodes [ i ] - > pnn ;
2007-05-03 07:30:38 +04:00
node_map - > nodes [ i ] . flags = ctdb - > nodes [ i ] - > flags ;
}
return 0 ;
}
2007-05-03 10:18:03 +04:00
2007-05-10 11:43:45 +04:00
struct getkeys_params {
struct ctdb_context * ctdb ;
uint32_t lmaster ;
uint32_t rec_count ;
struct getkeys_rec {
TDB_DATA key ;
TDB_DATA data ;
} * recs ;
} ;
static int traverse_getkeys ( struct tdb_context * tdb , TDB_DATA key , TDB_DATA data , void * p )
{
struct getkeys_params * params = ( struct getkeys_params * ) p ;
uint32_t lmaster ;
lmaster = ctdb_lmaster ( params - > ctdb , & key ) ;
/* only include this record if the lmaster matches or if
the wildcard lmaster ( - 1 ) was specified .
*/
if ( ( params - > lmaster ! = CTDB_LMASTER_ANY ) & & ( params - > lmaster ! = lmaster ) ) {
return 0 ;
}
params - > recs = talloc_realloc ( NULL , params - > recs , struct getkeys_rec , params - > rec_count + 1 ) ;
key . dptr = talloc_memdup ( params - > recs , key . dptr , key . dsize ) ;
data . dptr = talloc_memdup ( params - > recs , data . dptr , data . dsize ) ;
params - > recs [ params - > rec_count ] . key = key ;
params - > recs [ params - > rec_count ] . data = data ;
params - > rec_count + + ;
return 0 ;
}
/*
pul a bunch of records from a ltdb , filtering by lmaster
*/
int32_t ctdb_control_pull_db ( struct ctdb_context * ctdb , TDB_DATA indata , TDB_DATA * outdata )
2007-05-03 10:18:03 +04:00
{
2007-05-10 11:43:45 +04:00
struct ctdb_control_pulldb * pull ;
2007-05-03 10:18:03 +04:00
struct ctdb_db_context * ctdb_db ;
2007-05-10 11:43:45 +04:00
struct getkeys_params params ;
struct ctdb_control_pulldb_reply * reply ;
int i ;
size_t len = 0 ;
2007-05-12 09:29:06 +04:00
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " rejecting ctdb_control_pull_db when not frozen \n " ) ) ;
return - 1 ;
}
2007-05-10 11:43:45 +04:00
pull = ( struct ctdb_control_pulldb * ) indata . dptr ;
ctdb_db = find_ctdb_db ( ctdb , pull - > db_id ) ;
if ( ! ctdb_db ) {
DEBUG ( 0 , ( __location__ " Unknown db \n " ) ) ;
return - 1 ;
}
params . ctdb = ctdb ;
params . lmaster = pull - > lmaster ;
2007-05-12 13:57:12 +04:00
2007-05-10 11:43:45 +04:00
params . rec_count = 0 ;
params . recs = talloc_array ( outdata , struct getkeys_rec , 0 ) ;
CTDB_NO_MEMORY ( ctdb , params . recs ) ;
2007-05-12 09:29:06 +04:00
if ( ctdb_lock_all_databases_mark ( ctdb ) ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to get lock on entired db - failing \n " ) ) ;
2007-05-10 11:43:45 +04:00
return - 1 ;
}
tdb_traverse_read ( ctdb_db - > ltdb - > tdb , traverse_getkeys , & params ) ;
2007-05-03 10:18:03 +04:00
2007-05-12 09:29:06 +04:00
ctdb_lock_all_databases_unmark ( ctdb ) ;
2007-05-03 10:18:03 +04:00
2007-05-10 11:43:45 +04:00
reply = talloc ( outdata , struct ctdb_control_pulldb_reply ) ;
CTDB_NO_MEMORY ( ctdb , reply ) ;
2007-05-03 10:18:03 +04:00
2007-05-10 11:43:45 +04:00
reply - > db_id = pull - > db_id ;
reply - > count = params . rec_count ;
len = offsetof ( struct ctdb_control_pulldb_reply , data ) ;
for ( i = 0 ; i < reply - > count ; i + + ) {
struct ctdb_rec_data * rec ;
2007-09-21 06:24:02 +04:00
rec = ctdb_marshall_record ( outdata , 0 , params . recs [ i ] . key , NULL , params . recs [ i ] . data ) ;
2007-05-10 11:43:45 +04:00
reply = talloc_realloc_size ( outdata , reply , rec - > length + len ) ;
memcpy ( len + ( uint8_t * ) reply , rec , rec - > length ) ;
len + = rec - > length ;
talloc_free ( rec ) ;
}
talloc_free ( params . recs ) ;
outdata - > dptr = ( uint8_t * ) reply ;
outdata - > dsize = len ;
return 0 ;
}
/*
push a bunch of records into a ltdb , filtering by rsn
*/
int32_t ctdb_control_push_db ( struct ctdb_context * ctdb , TDB_DATA indata )
{
struct ctdb_control_pulldb_reply * reply = ( struct ctdb_control_pulldb_reply * ) indata . dptr ;
struct ctdb_db_context * ctdb_db ;
int i , ret ;
struct ctdb_rec_data * rec ;
2007-05-12 09:29:06 +04:00
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " rejecting ctdb_control_push_db when not frozen \n " ) ) ;
return - 1 ;
}
2007-05-10 11:43:45 +04:00
if ( indata . dsize < offsetof ( struct ctdb_control_pulldb_reply , data ) ) {
DEBUG ( 0 , ( __location__ " invalid data in pulldb reply \n " ) ) ;
return - 1 ;
}
ctdb_db = find_ctdb_db ( ctdb , reply - > db_id ) ;
2007-05-03 10:18:03 +04:00
if ( ! ctdb_db ) {
2007-05-10 11:43:45 +04:00
DEBUG ( 0 , ( __location__ " Unknown db 0x%08x \n " , reply - > db_id ) ) ;
2007-05-03 10:18:03 +04:00
return - 1 ;
}
2007-05-12 09:29:06 +04:00
if ( ctdb_lock_all_databases_mark ( ctdb ) ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to get lock on entired db - failing \n " ) ) ;
2007-05-10 11:43:45 +04:00
return - 1 ;
}
2007-05-03 10:18:03 +04:00
2007-05-10 11:43:45 +04:00
rec = ( struct ctdb_rec_data * ) & reply - > data [ 0 ] ;
2007-06-17 17:31:44 +04:00
DEBUG ( 3 , ( " starting push of %u records for dbid 0x%x \n " ,
reply - > count , reply - > db_id ) ) ;
2007-05-10 11:43:45 +04:00
for ( i = 0 ; i < reply - > count ; i + + ) {
TDB_DATA key , data ;
struct ctdb_ltdb_header * hdr , header ;
key . dptr = & rec - > data [ 0 ] ;
key . dsize = rec - > keylen ;
data . dptr = & rec - > data [ key . dsize ] ;
data . dsize = rec - > datalen ;
if ( data . dsize < sizeof ( struct ctdb_ltdb_header ) ) {
DEBUG ( 0 , ( __location__ " bad ltdb record \n " ) ) ;
2007-05-12 09:29:06 +04:00
goto failed ;
2007-05-10 11:43:45 +04:00
}
hdr = ( struct ctdb_ltdb_header * ) data . dptr ;
data . dptr + = sizeof ( * hdr ) ;
data . dsize - = sizeof ( * hdr ) ;
ret = ctdb_ltdb_fetch ( ctdb_db , key , & header , NULL , NULL ) ;
if ( ret ! = 0 ) {
2007-05-12 08:34:21 +04:00
DEBUG ( 0 , ( __location__ " Unable to fetch record \n " ) ) ;
2007-05-12 09:29:06 +04:00
goto failed ;
2007-05-10 11:43:45 +04:00
}
2007-05-12 13:57:12 +04:00
/* The check for dmaster gives priority to the dmaster
if the rsn values are equal */
if ( header . rsn < hdr - > rsn | |
2007-09-04 04:06:36 +04:00
( header . dmaster ! = ctdb - > pnn & & header . rsn = = hdr - > rsn ) ) {
2007-05-10 11:43:45 +04:00
ret = ctdb_ltdb_store ( ctdb_db , key , hdr , data ) ;
if ( ret ! = 0 ) {
2007-05-12 08:34:21 +04:00
DEBUG ( 0 , ( __location__ " Unable to store record \n " ) ) ;
2007-05-12 09:29:06 +04:00
goto failed ;
2007-05-10 11:43:45 +04:00
}
}
rec = ( struct ctdb_rec_data * ) ( rec - > length + ( uint8_t * ) rec ) ;
}
2007-06-17 17:31:44 +04:00
DEBUG ( 3 , ( " finished push of %u records for dbid 0x%x \n " ,
reply - > count , reply - > db_id ) ) ;
2007-05-12 09:29:06 +04:00
ctdb_lock_all_databases_unmark ( ctdb ) ;
2007-05-10 11:43:45 +04:00
return 0 ;
2007-05-12 09:29:06 +04:00
failed :
ctdb_lock_all_databases_unmark ( ctdb ) ;
return - 1 ;
2007-05-10 11:43:45 +04:00
}
static int traverse_setdmaster ( struct tdb_context * tdb , TDB_DATA key , TDB_DATA data , void * p )
{
uint32_t * dmaster = ( uint32_t * ) p ;
struct ctdb_ltdb_header * header = ( struct ctdb_ltdb_header * ) data . dptr ;
int ret ;
header - > dmaster = * dmaster ;
ret = tdb_store ( tdb , key , data , TDB_REPLACE ) ;
if ( ret ) {
2007-05-12 08:34:21 +04:00
DEBUG ( 0 , ( __location__ " failed to write tdb data back ret:%d \n " , ret ) ) ;
2007-05-10 11:43:45 +04:00
return ret ;
}
return 0 ;
}
int32_t ctdb_control_set_dmaster ( struct ctdb_context * ctdb , TDB_DATA indata )
{
struct ctdb_control_set_dmaster * p = ( struct ctdb_control_set_dmaster * ) indata . dptr ;
struct ctdb_db_context * ctdb_db ;
2007-05-03 10:18:03 +04:00
2007-05-12 09:29:06 +04:00
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " rejecting ctdb_control_set_dmaster when not frozen \n " ) ) ;
return - 1 ;
}
2007-05-10 11:43:45 +04:00
ctdb_db = find_ctdb_db ( ctdb , p - > db_id ) ;
if ( ! ctdb_db ) {
DEBUG ( 0 , ( __location__ " Unknown db 0x%08x \n " , p - > db_id ) ) ;
2007-05-03 10:18:03 +04:00
return - 1 ;
}
2007-05-10 11:43:45 +04:00
2007-05-12 09:29:06 +04:00
if ( ctdb_lock_all_databases_mark ( ctdb ) ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to get lock on entired db - failing \n " ) ) ;
2007-05-03 10:18:03 +04:00
return - 1 ;
2007-05-12 09:29:06 +04:00
}
2007-05-10 11:43:45 +04:00
tdb_traverse ( ctdb_db - > ltdb - > tdb , traverse_setdmaster , & p - > dmaster ) ;
2007-05-12 09:29:06 +04:00
ctdb_lock_all_databases_unmark ( ctdb ) ;
2007-05-10 11:43:45 +04:00
return 0 ;
}
2007-06-02 02:41:19 +04:00
struct ctdb_set_recmode_state {
2007-10-16 09:27:07 +04:00
struct ctdb_context * ctdb ;
2007-06-02 02:41:19 +04:00
struct ctdb_req_control * c ;
uint32_t recmode ;
2007-10-16 09:27:07 +04:00
int fd [ 2 ] ;
struct timed_event * te ;
struct fd_event * fde ;
pid_t child ;
2007-06-02 02:41:19 +04:00
} ;
/*
called when the ' recovered ' event script has finished
*/
static void ctdb_recovered_callback ( struct ctdb_context * ctdb , int status , void * p )
{
struct ctdb_set_recmode_state * state = talloc_get_type ( p , struct ctdb_set_recmode_state ) ;
2007-06-06 07:45:12 +04:00
ctdb_start_monitoring ( ctdb ) ;
2007-06-02 02:41:19 +04:00
if ( status = = 0 ) {
ctdb - > recovery_mode = state - > recmode ;
} else {
DEBUG ( 0 , ( __location__ " recovered event script failed (status %d) \n " , status ) ) ;
}
ctdb_request_control_reply ( ctdb , state - > c , NULL , status , NULL ) ;
talloc_free ( state ) ;
}
2007-10-16 09:27:07 +04:00
/*
called if our set_recmode child times out . this would happen if
ctdb_recovery_lock ( ) would block .
*/
static void ctdb_set_recmode_timeout ( struct event_context * ev , struct timed_event * te ,
struct timeval t , void * private_data )
{
struct ctdb_set_recmode_state * state = talloc_get_type ( private_data ,
struct ctdb_set_recmode_state ) ;
ctdb_request_control_reply ( state - > ctdb , state - > c , NULL , - 1 , " timeout in ctdb_set_recmode " ) ;
talloc_free ( state ) ;
}
/* when we free the recmode state we must kill any child process.
*/
static int set_recmode_destructor ( struct ctdb_set_recmode_state * state )
{
kill ( state - > child , SIGKILL ) ;
waitpid ( state - > child , NULL , 0 ) ;
return 0 ;
}
/* this is called when the client process has completed ctdb_recovery_lock()
and has written data back to us through the pipe .
*/
static void set_recmode_handler ( struct event_context * ev , struct fd_event * fde ,
uint16_t flags , void * private_data )
{
struct ctdb_set_recmode_state * state = talloc_get_type ( private_data ,
struct ctdb_set_recmode_state ) ;
2007-10-18 09:44:02 +04:00
char c = 0 ;
2007-10-16 09:27:07 +04:00
int ret ;
/* we got a response from our child process so we can abort the
timeout .
*/
talloc_free ( state - > te ) ;
state - > te = NULL ;
/* read the childs status when trying to lock the reclock file.
child wrote 0 if everything is fine and 1 if it did manage
to lock the file , which would be a problem since that means
we got a request to exit from recovery but we could still lock
the file which at this time SHOULD be locked by the recovery
daemon on the recmaster
*/
2007-10-18 09:44:02 +04:00
ret = read ( state - > fd [ 0 ] , & c , 1 ) ;
if ( ret ! = 1 | | c ! = 0 ) {
2007-10-16 09:27:07 +04:00
ctdb_request_control_reply ( state - > ctdb , state - > c , NULL , - 1 , " managed to lock reclock file from inside daemon " ) ;
talloc_free ( state ) ;
return ;
}
ctdb_stop_monitoring ( state - > ctdb ) ;
/* call the events script to tell all subsystems that we have recovered */
ret = ctdb_event_script_callback ( state - > ctdb ,
timeval_current_ofs ( state - > ctdb - > tunable . script_timeout , 0 ) ,
state ,
ctdb_recovered_callback ,
state , " recovered " ) ;
if ( ret ! = 0 ) {
ctdb_request_control_reply ( state - > ctdb , state - > c , NULL , - 1 , " failed to run eventscript from set_recmode " ) ;
talloc_free ( state ) ;
return ;
}
}
2007-05-12 08:34:21 +04:00
/*
set the recovery mode
*/
2007-06-02 02:41:19 +04:00
int32_t ctdb_control_set_recmode ( struct ctdb_context * ctdb ,
struct ctdb_req_control * c ,
TDB_DATA indata , bool * async_reply ,
2007-05-12 15:25:26 +04:00
const char * * errormsg )
2007-05-12 08:34:21 +04:00
{
2007-05-12 09:15:27 +04:00
uint32_t recmode = * ( uint32_t * ) indata . dptr ;
2007-06-02 05:36:42 +04:00
int ret ;
struct ctdb_set_recmode_state * state ;
2007-10-16 09:27:07 +04:00
pid_t parent = getpid ( ) ;
2007-06-02 05:36:42 +04:00
2007-05-12 09:15:27 +04:00
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " Attempt to change recovery mode to %u when not frozen \n " ,
recmode ) ) ;
2007-05-12 15:25:26 +04:00
( * errormsg ) = " Cannot change recovery mode while not frozen " ;
2007-05-12 09:15:27 +04:00
return - 1 ;
2007-05-12 08:34:21 +04:00
}
2007-06-02 05:36:42 +04:00
if ( recmode ! = CTDB_RECOVERY_NORMAL | |
ctdb - > recovery_mode ! = CTDB_RECOVERY_ACTIVE ) {
ctdb - > recovery_mode = recmode ;
return 0 ;
2007-06-02 02:41:19 +04:00
}
2007-06-02 05:36:42 +04:00
/* some special handling when ending recovery mode */
state = talloc ( ctdb , struct ctdb_set_recmode_state ) ;
CTDB_NO_MEMORY ( ctdb , state ) ;
2007-10-16 09:27:07 +04:00
/* For the rest of what needs to be done, we need to do this in
a child process since
1 , the call to ctdb_recovery_lock ( ) can block if the cluster
filesystem is in the process of recovery .
2 , running of the script may take a while .
*/
ret = pipe ( state - > fd ) ;
2007-06-02 05:36:42 +04:00
if ( ret ! = 0 ) {
2007-10-16 09:27:07 +04:00
talloc_free ( state ) ;
DEBUG ( 0 , ( __location__ " Failed to open pipe for set_recmode child \n " ) ) ;
2007-06-02 05:36:42 +04:00
return - 1 ;
2007-10-16 09:27:07 +04:00
}
2007-06-02 05:36:42 +04:00
2007-10-16 09:27:07 +04:00
state - > child = fork ( ) ;
if ( state - > child = = ( pid_t ) - 1 ) {
close ( state - > fd [ 0 ] ) ;
close ( state - > fd [ 1 ] ) ;
talloc_free ( state ) ;
return - 1 ;
2007-06-02 05:36:42 +04:00
}
2007-06-06 07:45:12 +04:00
2007-10-16 09:27:07 +04:00
if ( state - > child = = 0 ) {
char cc = 0 ;
close ( state - > fd [ 0 ] ) ;
/* we should not be able to get the lock on the nodes list,
as it should be held by the recovery master
*/
if ( ctdb_recovery_lock ( ctdb , false ) ) {
DEBUG ( 0 , ( " ERROR: recovery lock file %s not locked when recovering! \n " , ctdb - > recovery_lock_file ) ) ;
cc = 1 ;
}
write ( state - > fd [ 1 ] , & cc , 1 ) ;
/* make sure we die when our parent dies */
while ( kill ( parent , 0 ) = = 0 | | errno ! = ESRCH ) {
sleep ( 5 ) ;
}
_exit ( 0 ) ;
}
close ( state - > fd [ 1 ] ) ;
talloc_set_destructor ( state , set_recmode_destructor ) ;
state - > te = event_add_timed ( ctdb - > ev , state , timeval_current_ofs ( 3 , 0 ) ,
ctdb_set_recmode_timeout , state ) ;
state - > fde = event_add_fd ( ctdb - > ev , state , state - > fd [ 0 ] ,
EVENT_FD_READ | EVENT_FD_AUTOCLOSE ,
set_recmode_handler ,
( void * ) state ) ;
if ( state - > fde = = NULL ) {
talloc_free ( state ) ;
return - 1 ;
2007-06-02 05:36:42 +04:00
}
2007-10-16 09:27:07 +04:00
state - > ctdb = ctdb ;
state - > recmode = recmode ;
state - > c = talloc_steal ( state , c ) ;
2007-06-02 05:36:42 +04:00
* async_reply = true ;
2007-05-12 09:15:27 +04:00
return 0 ;
2007-05-12 08:34:21 +04:00
}
2007-05-14 00:25:15 +04:00
2007-05-23 11:21:14 +04:00
/*
callback for ctdb_control_max_rsn
*/
static int traverse_max_rsn ( struct tdb_context * tdb , TDB_DATA key , TDB_DATA data , void * p )
{
struct ctdb_ltdb_header * h = ( struct ctdb_ltdb_header * ) data . dptr ;
uint64_t * max_rsn = ( uint64_t * ) p ;
if ( data . dsize > = sizeof ( * h ) ) {
( * max_rsn ) = MAX ( * max_rsn , h - > rsn ) ;
}
return 0 ;
}
/*
get max rsn across an entire db
*/
int32_t ctdb_control_max_rsn ( struct ctdb_context * ctdb , TDB_DATA indata , TDB_DATA * outdata )
{
struct ctdb_db_context * ctdb_db ;
uint32_t db_id = * ( uint32_t * ) indata . dptr ;
uint64_t max_rsn = 0 ;
int ret ;
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " rejecting ctdb_control_max_rsn when not frozen \n " ) ) ;
return - 1 ;
}
ctdb_db = find_ctdb_db ( ctdb , db_id ) ;
if ( ! ctdb_db ) {
DEBUG ( 0 , ( __location__ " Unknown db \n " ) ) ;
return - 1 ;
}
if ( ctdb_lock_all_databases_mark ( ctdb ) ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to get lock on entired db - failing \n " ) ) ;
return - 1 ;
}
ret = tdb_traverse_read ( ctdb_db - > ltdb - > tdb , traverse_max_rsn , & max_rsn ) ;
if ( ret < 0 ) {
DEBUG ( 0 , ( __location__ " traverse failed in ctdb_control_max_rsn \n " ) ) ;
return - 1 ;
}
ctdb_lock_all_databases_unmark ( ctdb ) ;
outdata - > dptr = ( uint8_t * ) talloc ( outdata , uint64_t ) ;
if ( ! outdata - > dptr ) {
return - 1 ;
}
( * ( uint64_t * ) outdata - > dptr ) = max_rsn ;
outdata - > dsize = sizeof ( uint64_t ) ;
return 0 ;
}
/*
callback for ctdb_control_set_rsn_nonempty
*/
static int traverse_set_rsn_nonempty ( struct tdb_context * tdb , TDB_DATA key , TDB_DATA data , void * p )
{
struct ctdb_ltdb_header * h = ( struct ctdb_ltdb_header * ) data . dptr ;
uint64_t * rsn = ( uint64_t * ) p ;
if ( data . dsize > sizeof ( * h ) ) {
h - > rsn = * rsn ;
if ( tdb_store ( tdb , key , data , TDB_REPLACE ) ! = 0 ) {
return - 1 ;
}
}
return 0 ;
}
/*
set rsn for all non - empty records in a database to a given rsn
*/
int32_t ctdb_control_set_rsn_nonempty ( struct ctdb_context * ctdb , TDB_DATA indata , TDB_DATA * outdata )
{
struct ctdb_control_set_rsn_nonempty * p = ( struct ctdb_control_set_rsn_nonempty * ) indata . dptr ;
struct ctdb_db_context * ctdb_db ;
int ret ;
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " rejecting ctdb_control_set_rsn_nonempty when not frozen \n " ) ) ;
return - 1 ;
}
ctdb_db = find_ctdb_db ( ctdb , p - > db_id ) ;
if ( ! ctdb_db ) {
DEBUG ( 0 , ( __location__ " Unknown db \n " ) ) ;
return - 1 ;
}
if ( ctdb_lock_all_databases_mark ( ctdb ) ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to get lock on entired db - failing \n " ) ) ;
return - 1 ;
}
ret = tdb_traverse ( ctdb_db - > ltdb - > tdb , traverse_set_rsn_nonempty , & p - > rsn ) ;
if ( ret < 0 ) {
DEBUG ( 0 , ( __location__ " traverse failed in ctdb_control_set_rsn_nonempty \n " ) ) ;
return - 1 ;
}
ctdb_lock_all_databases_unmark ( ctdb ) ;
return 0 ;
}
/*
callback for ctdb_control_delete_low_rsn
*/
static int traverse_delete_low_rsn ( struct tdb_context * tdb , TDB_DATA key , TDB_DATA data , void * p )
{
struct ctdb_ltdb_header * h = ( struct ctdb_ltdb_header * ) data . dptr ;
uint64_t * rsn = ( uint64_t * ) p ;
if ( data . dsize < sizeof ( * h ) | | h - > rsn < * rsn ) {
if ( tdb_delete ( tdb , key ) ! = 0 ) {
return - 1 ;
}
}
return 0 ;
}
/*
delete any records with a rsn < the given rsn
*/
int32_t ctdb_control_delete_low_rsn ( struct ctdb_context * ctdb , TDB_DATA indata , TDB_DATA * outdata )
{
struct ctdb_control_delete_low_rsn * p = ( struct ctdb_control_delete_low_rsn * ) indata . dptr ;
struct ctdb_db_context * ctdb_db ;
int ret ;
if ( ctdb - > freeze_mode ! = CTDB_FREEZE_FROZEN ) {
DEBUG ( 0 , ( " rejecting ctdb_control_delete_low_rsn when not frozen \n " ) ) ;
return - 1 ;
}
ctdb_db = find_ctdb_db ( ctdb , p - > db_id ) ;
if ( ! ctdb_db ) {
DEBUG ( 0 , ( __location__ " Unknown db \n " ) ) ;
return - 1 ;
}
if ( ctdb_lock_all_databases_mark ( ctdb ) ! = 0 ) {
DEBUG ( 0 , ( __location__ " Failed to get lock on entired db - failing \n " ) ) ;
return - 1 ;
}
ret = tdb_traverse ( ctdb_db - > ltdb - > tdb , traverse_delete_low_rsn , & p - > rsn ) ;
if ( ret < 0 ) {
DEBUG ( 0 , ( __location__ " traverse failed in ctdb_control_delete_low_rsn \n " ) ) ;
return - 1 ;
}
ctdb_lock_all_databases_unmark ( ctdb ) ;
return 0 ;
}
2007-05-25 07:05:25 +04:00
2007-06-02 04:03:28 +04:00
/*
2007-06-02 05:36:42 +04:00
try and get the recovery lock in shared storage - should only work
on the recovery master recovery daemon . Anywhere else is a bug
2007-06-02 04:03:28 +04:00
*/
2007-06-02 05:36:42 +04:00
bool ctdb_recovery_lock ( struct ctdb_context * ctdb , bool keep )
2007-06-02 04:03:28 +04:00
{
struct flock lock ;
2007-06-02 05:36:42 +04:00
if ( ctdb - > recovery_lock_fd ! = - 1 ) {
close ( ctdb - > recovery_lock_fd ) ;
2007-06-02 04:03:28 +04:00
}
2007-06-02 05:36:42 +04:00
ctdb - > recovery_lock_fd = open ( ctdb - > recovery_lock_file , O_RDWR | O_CREAT , 0600 ) ;
if ( ctdb - > recovery_lock_fd = = - 1 ) {
2007-06-02 04:03:28 +04:00
DEBUG ( 0 , ( " Unable to open %s - (%s) \n " ,
2007-06-02 05:36:42 +04:00
ctdb - > recovery_lock_file , strerror ( errno ) ) ) ;
2007-06-02 04:03:28 +04:00
return false ;
}
2007-09-19 05:46:37 +04:00
set_close_on_exec ( ctdb - > recovery_lock_fd ) ;
2007-06-02 04:03:28 +04:00
lock . l_type = F_WRLCK ;
lock . l_whence = SEEK_SET ;
lock . l_start = 0 ;
lock . l_len = 1 ;
lock . l_pid = 0 ;
2007-06-02 05:36:42 +04:00
if ( fcntl ( ctdb - > recovery_lock_fd , F_SETLK , & lock ) ! = 0 ) {
2007-09-24 04:19:07 +04:00
close ( ctdb - > recovery_lock_fd ) ;
ctdb - > recovery_lock_fd = - 1 ;
2007-06-02 04:03:28 +04:00
return false ;
}
if ( ! keep ) {
2007-06-02 05:36:42 +04:00
close ( ctdb - > recovery_lock_fd ) ;
ctdb - > recovery_lock_fd = - 1 ;
2007-06-02 04:03:28 +04:00
}
return true ;
}