2001-09-25 16:49:28 +04:00
/*
2001-12-17 19:58:17 +03:00
* Copyright ( C ) 2001 Sistina Software ( UK ) Limited .
2001-09-25 16:49:28 +04:00
*
2001-12-17 19:58:17 +03:00
* This file is released under the GPL .
2001-09-25 16:49:28 +04:00
*/
# include "tools.h"
# include <assert.h>
# include <getopt.h>
# include <signal.h>
# include <syslog.h>
# include <libgen.h>
# include <sys/stat.h>
2001-10-12 14:32:06 +04:00
# include <ctype.h>
2001-09-25 16:49:28 +04:00
# include "stub.h"
# ifdef READLINE_SUPPORT
# include <readline/readline.h>
# include <readline/history.h>
2001-12-31 18:20:18 +03:00
# define MAX_HISTORY 100
# ifndef rl_completion_matches
# define rl_completion_matches(a, b) completion_matches((char *)a, b)
# endif
2001-09-25 16:49:28 +04:00
# endif
2001-11-13 17:17:50 +03:00
/* define exported table of valid switches */
2001-09-25 16:49:28 +04:00
struct arg the_args [ ARG_COUNT + 1 ] = {
2001-12-17 19:58:17 +03:00
# define arg(a, b, c, d) {b, "--" c, d, 0, NULL},
2001-09-25 16:49:28 +04:00
# include "args.h"
2001-12-17 19:58:17 +03:00
# undef arg
2001-09-25 16:49:28 +04:00
} ;
static int _array_size ;
static int _num_commands ;
static struct command * _commands ;
2001-11-12 18:10:01 +03:00
/* Exported LVM1 disk format */
struct format_instance * fid ;
2001-11-13 17:17:50 +03:00
/* Export command being processed */
struct command * the_command ;
2001-11-12 18:10:01 +03:00
struct cmd_context * cmd ;
2001-10-08 22:44:22 +04:00
2001-10-24 21:53:50 +04:00
/* Whether or not to dump persistent filter state */
2001-11-14 13:44:14 +03:00
static int _dump_filter ;
2001-09-25 16:49:28 +04:00
static int _interactive ;
static FILE * _log ;
2001-12-17 19:58:17 +03:00
/*
* Both verbose have a global setting which comes
* from the command line that invoked the shell ,
* or the config file . These are the ' default '
* variables . In addition people may set a level
* for a single command .
*/
static int _default_debug ;
static int _debug ;
static int _default_verbose ;
static int _verbose ;
2002-01-01 00:27:39 +03:00
static int _default_test ;
static int _test ;
2001-12-17 15:01:09 +03:00
/*
2001-12-31 18:20:18 +03:00
* The lvm_sys_dir contains :
2001-12-17 15:01:09 +03:00
*
* o The lvm configuration ( lvm . conf )
* o The persistent filter cache ( . cache )
2001-12-31 22:09:51 +03:00
* o Volume group backups ( backup / )
2001-12-17 15:01:09 +03:00
*
*/
2001-12-31 18:20:18 +03:00
static char _sys_dir [ PATH_MAX ] = " /etc/lvm " ;
static char _backup_dir [ PATH_MAX ] ;
static char _dev_dir [ PATH_MAX ] ;
2001-12-31 22:09:51 +03:00
static int _backup_days = 14 ; /* Keep at least 14 days */
static int _backup_number = 10 ; /* Keep at least 10 backups */
static int _backup_auto = 1 ; /* Autobackups enabled by default */
2001-12-31 18:20:18 +03:00
# define DEFAULT_DEV_DIR " / dev"
2001-12-17 15:01:09 +03:00
2001-12-31 22:09:51 +03:00
2001-09-25 16:49:28 +04:00
/* static functions */
static void register_commands ( void ) ;
static struct command * find_command ( const char * name ) ;
static void register_command ( const char * name , command_fn fn ,
const char * desc , const char * usage , . . . ) ;
static void create_new_command ( const char * name , command_fn command ,
const char * desc , const char * usage ,
int nargs , int * args ) ;
static void alloc_command ( void ) ;
static void add_getopt_arg ( int arg , char * * ptr , struct option * * o ) ;
static int process_command_line ( struct command * com , int * argc , char * * * argv ) ;
static struct arg * find_arg ( struct command * com , int a ) ;
static int process_common_commands ( struct command * com ) ;
static int run_command ( int argc , char * * argv ) ;
static int init ( void ) ;
static void fin ( void ) ;
static int run_script ( int argc , char * * argv ) ;
# ifdef READLINE_SUPPORT
static int shell ( void ) ;
# endif
static void display_help ( void ) ;
int main ( int argc , char * * argv )
{
char * namebase , * base ;
int ret , alias = 0 ;
if ( ! init ( ) )
return - 1 ;
namebase = strdup ( argv [ 0 ] ) ;
base = basename ( namebase ) ;
if ( strcmp ( base , " lvm " ) )
alias = 1 ;
free ( namebase ) ;
register_commands ( ) ;
# ifdef READLINE_SUPPORT
if ( ! alias & & argc = = 1 ) {
ret = shell ( ) ;
goto out ;
}
# endif
if ( ! alias ) {
if ( argc < 2 ) {
log_fatal ( " Please supply an LVM command. " ) ;
display_help ( ) ;
2001-10-06 01:39:30 +04:00
ret = EINVALID_CMD_LINE ;
2001-09-25 16:49:28 +04:00
goto out ;
}
argc - - ;
argv + + ;
}
ret = run_command ( argc , argv ) ;
2001-10-06 01:39:30 +04:00
if ( ( ret = = ENO_SUCH_CMD ) & & ( ! alias ) )
2001-09-25 16:49:28 +04:00
ret = run_script ( argc , argv ) ;
2001-10-06 01:39:30 +04:00
if ( ret = = ENO_SUCH_CMD )
2001-10-15 16:49:58 +04:00
log_error ( " No such command. Try 'help'. " ) ;
2001-09-25 16:49:28 +04:00
out :
fin ( ) ;
return ret ;
}
void usage ( const char * name )
{
struct command * com = find_command ( name ) ;
if ( ! com )
return ;
log_error ( " %s: %s \n \n %s " , com - > name , com - > desc , com - > usage ) ;
}
int yes_no_arg ( struct arg * a )
{
2001-11-10 01:01:04 +03:00
a - > sign = SIGN_NONE ;
2001-09-25 16:49:28 +04:00
if ( ! strcmp ( a - > value , " y " ) )
a - > i_value = 1 ;
else if ( ! strcmp ( a - > value , " n " ) )
a - > i_value = 0 ;
else
return 0 ;
return 1 ;
}
2001-11-10 01:01:04 +03:00
int _get_int_arg ( struct arg * a , char * * ptr )
2001-09-25 16:49:28 +04:00
{
2001-11-10 01:01:04 +03:00
char * val ;
long v ;
val = a - > value ;
switch ( * val ) {
case ' + ' :
a - > sign = SIGN_PLUS ;
val + + ;
break ;
case ' - ' :
a - > sign = SIGN_MINUS ;
val + + ;
break ;
default :
a - > sign = SIGN_NONE ;
}
if ( ! isdigit ( * val ) )
return 0 ;
v = strtol ( val , ptr , 10 ) ;
2001-09-25 16:49:28 +04:00
2001-11-10 01:01:04 +03:00
if ( * ptr = = val )
return 0 ;
a - > i_value = ( uint32_t ) v ;
return 1 ;
}
int size_arg ( struct arg * a )
{
2001-09-25 16:49:28 +04:00
char * ptr ;
int i ;
2001-11-10 01:01:04 +03:00
static char * suffixes = " kmgt " ;
2001-09-25 16:49:28 +04:00
2001-11-10 01:01:04 +03:00
if ( ! _get_int_arg ( a , & ptr ) )
2001-09-25 16:49:28 +04:00
return 0 ;
if ( * ptr ) {
for ( i = strlen ( suffixes ) - 1 ; i > = 0 ; i - - )
if ( suffixes [ i ] = = tolower ( ( int ) * ptr ) )
break ;
if ( i < 0 )
return 0 ;
while ( i - - > 0 )
2001-11-10 01:01:04 +03:00
a - > i_value * = 1024 ;
2001-09-25 16:49:28 +04:00
}
return 1 ;
}
int int_arg ( struct arg * a )
{
char * ptr ;
2001-11-10 01:01:04 +03:00
if ( ! _get_int_arg ( a , & ptr ) | | ( * ptr ) | | ( a - > sign = = SIGN_MINUS ) )
return 0 ;
return 1 ;
}
int int_arg_with_sign ( struct arg * a )
{
char * ptr ;
if ( ! _get_int_arg ( a , & ptr ) | | ( * ptr ) )
2001-09-25 16:49:28 +04:00
return 0 ;
return 1 ;
}
int string_arg ( struct arg * a )
{
return 1 ;
}
int permission_arg ( struct arg * a )
{
2001-11-10 01:01:04 +03:00
a - > sign = SIGN_NONE ;
2001-09-25 16:49:28 +04:00
if ( ( ! strcmp ( a - > value , " rw " ) ) | | ( ! strcmp ( a - > value , " wr " ) ) )
2001-10-06 01:39:30 +04:00
a - > i_value = LVM_READ | LVM_WRITE ;
2001-09-25 16:49:28 +04:00
else if ( ! strcmp ( a - > value , " r " ) )
2001-10-06 01:39:30 +04:00
a - > i_value = LVM_READ ;
2001-09-25 16:49:28 +04:00
else
return 0 ;
return 1 ;
}
2001-10-12 14:32:06 +04:00
char yes_no_prompt ( const char * prompt , . . . )
2001-09-25 16:49:28 +04:00
{
int c = 0 ;
va_list ap ;
while ( c ! = ' y ' & & c ! = ' n ' ) {
if ( c = = ' \n ' | | c = = 0 ) {
va_start ( ap , prompt ) ;
vprintf ( prompt , ap ) ;
va_end ( ap ) ;
}
c = tolower ( getchar ( ) ) ;
}
2001-10-12 14:32:06 +04:00
2001-11-10 01:01:04 +03:00
while ( getchar ( ) ! = ' \n ' ) ;
2001-10-12 14:32:06 +04:00
2001-09-25 16:49:28 +04:00
return c ;
}
static void register_commands ( )
{
# define xx(a, b, c...) register_command(# a, a, b, ## c, \
debug_ARG , help_ARG , suspend_ARG , \
2001-12-17 19:58:17 +03:00
version_ARG , verbose_ARG , \
quiet_ARG , - 1 ) ;
2001-09-25 16:49:28 +04:00
# include "commands.h"
# undef xx
}
static void register_command ( const char * name , command_fn fn ,
const char * desc , const char * usage , . . . )
{
int nargs = 0 , i ;
int * args ;
va_list ap ;
/* count how many arguments we have */
va_start ( ap , usage ) ;
while ( va_arg ( ap , int ) > = 0 )
nargs + + ;
va_end ( ap ) ;
/* allocate space for them */
2001-11-10 01:01:04 +03:00
if ( ! ( args = dbg_malloc ( sizeof ( * args ) * nargs ) ) ) {
2001-09-25 16:49:28 +04:00
log_fatal ( " Out of memory. " ) ;
2001-10-06 01:39:30 +04:00
exit ( ECMD_FAILED ) ;
2001-09-25 16:49:28 +04:00
}
/* fill them in */
va_start ( ap , usage ) ;
for ( i = 0 ; i < nargs ; i + + )
args [ i ] = va_arg ( ap , int ) ;
va_end ( ap ) ;
/* enter the command in the register */
create_new_command ( name , fn , desc , usage , nargs , args ) ;
}
static struct command * find_command ( const char * name )
{
int i ;
char * namebase , * base ;
namebase = strdup ( name ) ;
base = basename ( namebase ) ;
for ( i = 0 ; i < _num_commands ; i + + ) {
if ( ! strcmp ( base , _commands [ i ] . name ) )
break ;
}
free ( namebase ) ;
if ( i > = _num_commands )
return 0 ;
return _commands + i ;
}
static void create_new_command ( const char * name , command_fn command ,
const char * desc , const char * usage ,
int nargs , int * args )
{
struct command * nc ;
alloc_command ( ) ;
nc = _commands + _num_commands + + ;
nc - > name = name ;
nc - > desc = desc ;
nc - > usage = usage ;
nc - > fn = command ;
nc - > num_args = nargs ;
nc - > valid_args = args ;
}
static void __alloc ( int size )
{
2001-11-10 01:01:04 +03:00
if ( ! ( _commands = dbg_realloc ( _commands , sizeof ( * _commands ) * size ) ) ) {
2001-09-25 16:49:28 +04:00
log_fatal ( " Couldn't allocate memory. " ) ;
2001-10-06 01:39:30 +04:00
exit ( ECMD_FAILED ) ;
2001-09-25 16:49:28 +04:00
}
_array_size = size ;
}
static void alloc_command ( void )
{
if ( ! _array_size )
__alloc ( 32 ) ;
if ( _array_size < = _num_commands )
__alloc ( 2 * _array_size ) ;
}
2001-12-17 13:08:27 +03:00
/*
* Sets up the short and long argument . If there
* is no short argument then the index of the
* argument in the the_args array is set as the
* long opt value . Yuck . Of course this means we
* can ' t have more than ' a ' long arguments . Since
* we have only 1 ATM ( - - version ) I think we can
* live with this restriction .
*/
2001-09-25 16:49:28 +04:00
static void add_getopt_arg ( int arg , char * * ptr , struct option * * o )
{
struct arg * a = the_args + arg ;
if ( a - > short_arg ) {
* ( * ptr ) + + = a - > short_arg ;
if ( a - > fn )
* ( * ptr ) + + = ' : ' ;
}
if ( * ( a - > long_arg + 2 ) ) {
( * o ) - > name = a - > long_arg + 2 ;
( * o ) - > has_arg = a - > fn ? 1 : 0 ;
( * o ) - > flag = NULL ;
2001-12-17 13:08:27 +03:00
( * o ) - > val = arg ;
2001-09-25 16:49:28 +04:00
( * o ) + + ;
}
}
static int process_command_line ( struct command * com , int * argc , char * * * argv )
{
int i , opt ;
char str [ ( ( ARG_COUNT + 1 ) * 2 ) + 1 ] , * ptr = str ;
struct option opts [ ARG_COUNT + 1 ] , * o = opts ;
struct arg * a ;
for ( i = 0 ; i < ARG_COUNT ; i + + ) {
struct arg * a = the_args + i ;
/* zero the count and arg */
a - > count = 0 ;
a - > value = 0 ;
a - > i_value = 0 ;
}
/* fill in the short and long opts */
for ( i = 0 ; i < com - > num_args ; i + + )
add_getopt_arg ( com - > valid_args [ i ] , & ptr , & o ) ;
* ptr = ' \0 ' ;
2001-11-10 01:01:04 +03:00
memset ( o , 0 , sizeof ( * o ) ) ;
2001-09-25 16:49:28 +04:00
/* initialise getopt_long & scan for command line switches */
optarg = 0 ;
optind = 0 ;
while ( ( opt = getopt_long ( * argc , * argv , str , opts , NULL ) ) > = 0 ) {
a = find_arg ( com , opt ) ;
if ( ! a ) {
log_fatal ( " Unrecognised option. " ) ;
return 0 ;
}
if ( a - > fn ) {
2001-12-17 19:58:17 +03:00
2001-09-25 16:49:28 +04:00
if ( ! optarg ) {
log_error ( " Option requires argument. " ) ;
return 0 ;
}
a - > value = optarg ;
if ( ! a - > fn ( a ) ) {
2001-12-31 18:20:18 +03:00
log_error ( " Invalid argument %s " , optarg ) ;
2001-09-25 16:49:28 +04:00
return 0 ;
}
}
a - > count + + ;
}
* argc - = optind ;
* argv + = optind ;
return 1 ;
}
static struct arg * find_arg ( struct command * com , int opt )
{
struct arg * a ;
2001-12-17 13:08:27 +03:00
int i , arg ;
2001-09-25 16:49:28 +04:00
for ( i = 0 ; i < com - > num_args ; i + + ) {
2001-12-17 13:08:27 +03:00
arg = com - > valid_args [ i ] ;
a = the_args + arg ;
/*
* opt should equal either the
* short arg , or the index into
* ' the_args ' .
*/
if ( ( a - > short_arg & & ( opt = = a - > short_arg ) ) | | ( opt = = arg ) )
2001-09-25 16:49:28 +04:00
return a ;
}
return 0 ;
}
static int process_common_commands ( struct command * com )
{
int l ;
if ( arg_count ( suspend_ARG ) )
kill ( getpid ( ) , SIGSTOP ) ;
2001-12-17 19:58:17 +03:00
_debug = _default_debug ;
if ( arg_count ( debug_ARG ) )
_debug = arg_count ( debug_ARG ) ;
_verbose = _default_verbose ;
if ( arg_count ( verbose_ARG ) )
_verbose = arg_count ( verbose_ARG ) ;
if ( arg_count ( quiet_ARG ) ) {
_debug = 0 ;
_verbose = 0 ;
}
2001-09-25 16:49:28 +04:00
2002-01-01 00:27:39 +03:00
_test = _default_test ;
if ( arg_count ( test_ARG ) )
_test = arg_count ( test_ARG ) ;
2001-09-25 16:49:28 +04:00
if ( arg_count ( help_ARG ) ) {
usage ( com - > name ) ;
2001-10-06 01:39:30 +04:00
return ECMD_PROCESSED ;
2001-09-25 16:49:28 +04:00
}
if ( arg_count ( version_ARG ) ) {
2001-10-02 02:12:10 +04:00
/* FIXME: Add driver and software version */
log_error ( " %s: " , com - > name ) ;
2001-10-06 01:39:30 +04:00
return ECMD_PROCESSED ;
2001-09-25 16:49:28 +04:00
}
/* Set autobackup if command takes this option */
for ( l = 0 ; l < com - > num_args ; l + + )
2001-12-20 19:05:14 +03:00
if ( com - > valid_args [ l ] = = autobackup_ARG )
2001-12-31 22:09:51 +03:00
if ( ! autobackup_init ( _backup_dir , _backup_days ,
_backup_number , _backup_auto ) )
2001-10-06 01:39:30 +04:00
return EINVALID_CMD_LINE ;
2001-09-25 16:49:28 +04:00
2001-12-31 18:20:18 +03:00
/* Zero indicates it's OK to continue processing this command */
2001-09-25 16:49:28 +04:00
return 0 ;
}
int help ( int argc , char * * argv )
{
if ( ! argc )
display_help ( ) ;
else {
int i ;
for ( i = 0 ; i < argc ; i + + )
usage ( argv [ i ] ) ;
}
return 0 ;
}
2001-12-17 15:01:09 +03:00
static void display_help ( void )
2001-09-25 16:49:28 +04:00
{
int i ;
log_error ( " Available lvm commands: " ) ;
log_error ( " Use 'lvm help <command>' for more information " ) ;
2001-10-08 22:44:22 +04:00
log_error ( " " ) ;
2001-09-25 16:49:28 +04:00
for ( i = 0 ; i < _num_commands ; i + + ) {
struct command * com = _commands + i ;
log_error ( " %-16.16s%s " , com - > name , com - > desc ) ;
}
}
static int run_command ( int argc , char * * argv )
{
int ret = 0 ;
2001-11-13 17:17:50 +03:00
if ( ! ( the_command = find_command ( argv [ 0 ] ) ) )
2001-10-06 01:39:30 +04:00
return ENO_SUCH_CMD ;
2001-09-25 16:49:28 +04:00
2001-11-13 17:17:50 +03:00
if ( ! process_command_line ( the_command , & argc , & argv ) ) {
2001-09-25 16:49:28 +04:00
log_error ( " Error during parsing of command line. " ) ;
2001-10-06 01:39:30 +04:00
return EINVALID_CMD_LINE ;
2001-09-25 16:49:28 +04:00
}
2001-11-13 17:17:50 +03:00
if ( ( ret = process_common_commands ( the_command ) ) )
2001-09-25 16:49:28 +04:00
return ret ;
2001-12-17 19:58:17 +03:00
init_debug ( _debug ) ;
init_verbose ( _verbose ) ;
2002-01-01 00:27:39 +03:00
init_test ( _test ) ;
2001-12-17 19:58:17 +03:00
2001-11-13 17:17:50 +03:00
ret = the_command - > fn ( argc , argv ) ;
2001-09-25 16:49:28 +04:00
2001-12-17 19:58:17 +03:00
/*
* set the debug and verbose levels back
* to the global default .
*/
init_debug ( _default_debug ) ;
init_verbose ( _default_verbose ) ;
2002-01-01 00:27:39 +03:00
init_test ( _default_test ) ;
2001-12-17 19:58:17 +03:00
2001-10-12 18:25:53 +04:00
/*
* free off any memory the command used .
*/
2001-11-12 18:10:01 +03:00
pool_empty ( cmd - > mem ) ;
2001-10-12 18:25:53 +04:00
2001-10-06 01:39:30 +04:00
if ( ret = = EINVALID_CMD_LINE & & ! _interactive )
2001-11-13 17:17:50 +03:00
usage ( the_command - > name ) ;
2001-09-25 16:49:28 +04:00
2001-12-17 19:58:17 +03:00
2001-09-25 16:49:28 +04:00
return ret ;
}
static int split ( char * str , int * argc , char * * argv , int max )
{
char * b = str , * e ;
* argc = 0 ;
while ( * b ) {
while ( * b & & isspace ( * b ) )
b + + ;
if ( ( ! * b ) | | ( * b = = ' # ' ) )
break ;
e = b ;
while ( * e & & ! isspace ( * e ) )
e + + ;
argv [ ( * argc ) + + ] = b ;
if ( ! * e )
break ;
* e + + = ' \0 ' ;
b = e ;
if ( * argc = = max )
break ;
}
return * argc ;
}
static void __init_log ( struct config_file * cf )
{
2001-10-25 21:25:48 +04:00
char * open_mode = " a " ;
2001-09-25 16:49:28 +04:00
const char * log_file = find_config_str ( cf - > root , " log/file " , ' / ' , 0 ) ;
2001-10-25 21:25:48 +04:00
if ( find_config_int ( cf - > root , " log/overwrite " , ' / ' , 0 ) )
open_mode = " w " ;
2001-09-25 16:49:28 +04:00
if ( log_file ) {
/* set up the logging */
2001-10-25 21:25:48 +04:00
if ( ! ( _log = fopen ( log_file , open_mode ) ) )
2001-10-23 15:50:49 +04:00
log_error ( " Couldn't open log file %s " , log_file ) ;
2001-09-25 16:49:28 +04:00
else
init_log ( _log ) ;
}
2001-12-17 19:58:17 +03:00
_default_debug = find_config_int ( cf - > root , " log/level " , ' / ' , 0 ) ;
init_debug ( _default_debug ) ;
2001-10-23 22:20:27 +04:00
2001-12-17 19:58:17 +03:00
_default_verbose = find_config_int ( cf - > root , " log/verbose " , ' / ' , 0 ) ;
init_verbose ( _default_verbose ) ;
2002-01-01 00:27:39 +03:00
_default_test = find_config_int ( cf - > root , " log/test " , ' / ' , 0 ) ;
init_test ( _default_test ) ;
2001-09-25 16:49:28 +04:00
}
2001-11-13 17:17:50 +03:00
static int dev_cache_setup ( struct config_file * cf )
2001-10-23 15:50:49 +04:00
{
struct config_node * cn ;
struct config_value * cv ;
if ( ! dev_cache_init ( ) ) {
stack ;
return 0 ;
}
2001-11-13 17:17:50 +03:00
if ( ! ( cn = find_config_node ( cf - > root , " devices/scan " , ' / ' ) ) ) {
2001-10-23 15:50:49 +04:00
if ( ! dev_cache_add_dir ( " /dev " ) ) {
log_error ( " Failed to add /dev to internal "
" device cache " ) ;
return 0 ;
}
2001-11-10 01:01:04 +03:00
log_verbose
( " device/scan not in config file: Defaulting to /dev " ) ;
2001-10-23 22:20:27 +04:00
return 1 ;
2001-10-23 15:50:49 +04:00
}
for ( cv = cn - > v ; cv ; cv = cv - > next ) {
if ( cv - > type ! = CFG_STRING ) {
log_error ( " Invalid string in config file: "
" devices/scan " ) ;
return 0 ;
}
2001-11-10 01:01:04 +03:00
2001-10-23 15:50:49 +04:00
if ( ! dev_cache_add_dir ( cv - > v . str ) ) {
2001-11-10 01:01:04 +03:00
log_error ( " Failed to add %s to internal device cache " ,
2001-10-23 15:50:49 +04:00
cv - > v . str ) ;
return 0 ;
}
2001-11-10 01:01:04 +03:00
}
2001-10-23 15:50:49 +04:00
return 1 ;
}
2001-11-13 17:17:50 +03:00
static struct dev_filter * filter_components_setup ( struct config_file * cf )
2001-10-23 15:50:49 +04:00
{
struct config_node * cn ;
2001-10-23 22:20:27 +04:00
struct dev_filter * f1 , * f2 , * f3 ;
2001-10-23 15:50:49 +04:00
2001-10-23 16:33:57 +04:00
if ( ! ( f2 = lvm_type_filter_create ( ) ) )
2001-10-23 15:50:49 +04:00
return 0 ;
2001-11-13 17:17:50 +03:00
if ( ! ( cn = find_config_node ( cf - > root , " devices/filter " , ' / ' ) ) ) {
2001-10-23 22:20:27 +04:00
log_debug ( " devices/filter not found in config file: no regex "
" filter installed " ) ;
2001-10-23 16:33:57 +04:00
return f2 ;
2001-10-23 15:50:49 +04:00
}
2001-10-23 16:33:57 +04:00
if ( ! ( f1 = regex_filter_create ( cn - > v ) ) ) {
2001-10-23 15:50:49 +04:00
log_error ( " Failed to create regex device filter " ) ;
2001-10-23 16:33:57 +04:00
return f2 ;
2001-10-23 15:50:49 +04:00
}
2001-10-23 22:20:27 +04:00
if ( ! ( f3 = composite_filter_create ( 2 , f1 , f2 ) ) ) {
2001-10-23 15:50:49 +04:00
log_error ( " Failed to create composite device filter " ) ;
2001-10-23 16:33:57 +04:00
return f2 ;
2001-10-23 15:50:49 +04:00
}
2001-10-23 22:20:27 +04:00
return f3 ;
}
2001-11-13 17:17:50 +03:00
static struct dev_filter * filter_setup ( struct config_file * cf )
2001-10-23 22:20:27 +04:00
{
const char * lvm_cache ;
struct dev_filter * f3 , * f4 ;
struct stat st ;
2001-12-17 15:01:09 +03:00
char cache_file [ PATH_MAX ] ;
2001-10-23 22:20:27 +04:00
2001-11-14 13:44:14 +03:00
_dump_filter = 0 ;
2001-10-24 21:53:50 +04:00
2001-11-13 17:17:50 +03:00
if ( ! ( f3 = filter_components_setup ( cmd - > cf ) ) )
2001-10-23 22:20:27 +04:00
return 0 ;
2001-12-17 17:05:43 +03:00
if ( lvm_snprintf ( cache_file , sizeof ( cache_file ) ,
2001-12-31 18:20:18 +03:00
" %s/.cache " , _sys_dir ) < 0 ) {
log_error ( " Persistent cache filename too long ('%s/.cache'). " ,
_sys_dir ) ;
2001-12-17 15:01:09 +03:00
return 0 ;
}
2001-11-13 17:17:50 +03:00
lvm_cache = find_config_str ( cf - > root , " devices/cache " , ' / ' ,
2001-12-17 15:01:09 +03:00
cache_file ) ;
2001-10-23 22:20:27 +04:00
if ( ! ( f4 = persistent_filter_create ( f3 , lvm_cache ) ) ) {
log_error ( " Failed to create persistent device filter " ) ;
2001-10-24 21:53:50 +04:00
return 0 ;
2001-10-23 22:20:27 +04:00
}
2001-10-24 21:53:50 +04:00
/* Should we ever dump persistent filter state? */
2001-11-13 17:17:50 +03:00
if ( find_config_int ( cf - > root , " devices/write_cache_state " , ' / ' , 1 ) )
2001-11-14 13:44:14 +03:00
_dump_filter = 1 ;
2001-11-10 01:01:04 +03:00
2002-01-01 00:27:39 +03:00
if ( ! * _sys_dir )
_dump_filter = 0 ;
2001-10-23 22:20:27 +04:00
if ( ! stat ( lvm_cache , & st ) & & ! persistent_filter_load ( f4 ) )
log_verbose ( " Failed to load existing device cache from %s " ,
lvm_cache ) ;
2001-10-23 15:50:49 +04:00
return f4 ;
}
2001-12-17 15:01:09 +03:00
static int _get_env_vars ( void )
{
const char * e ;
2002-01-01 00:27:39 +03:00
/* Set to "" to avoid using any system directory */
2001-12-17 15:01:09 +03:00
if ( ( e = getenv ( " LVM_SYSTEM_DIR " ) ) ) {
2001-12-31 18:20:18 +03:00
if ( lvm_snprintf ( _sys_dir , sizeof ( _sys_dir ) , " %s " , e ) < 0 ) {
log_error ( " LVM_SYSTEM_DIR environment variable "
" is too long. " ) ;
2001-12-17 15:01:09 +03:00
return 0 ;
}
}
return 1 ;
}
2001-09-25 16:49:28 +04:00
static int init ( void )
{
struct stat info ;
2002-01-01 00:27:39 +03:00
char config_file [ PATH_MAX ] = " " ;
2001-12-17 15:01:09 +03:00
if ( ! _get_env_vars ( ) )
return 0 ;
2001-11-12 18:10:01 +03:00
2002-01-01 00:27:39 +03:00
/* Create system directory if it doesn't already exist */
if ( ! create_dir ( _sys_dir ) )
return 0 ;
2001-11-12 18:10:01 +03:00
if ( ! ( cmd = dbg_malloc ( sizeof ( * cmd ) ) ) ) {
log_error ( " Failed to allocate command context " ) ;
2001-12-17 15:01:09 +03:00
return 0 ;
2001-11-12 18:10:01 +03:00
}
2001-10-08 22:44:22 +04:00
2001-11-13 17:17:50 +03:00
if ( ! ( cmd - > cf = create_config_file ( ) ) ) {
2001-09-25 16:49:28 +04:00
stack ;
2001-12-17 15:01:09 +03:00
return 0 ;
2001-09-25 16:49:28 +04:00
}
/* Use LOG_USER for syslog messages by default */
init_syslog ( LOG_USER ) ;
/* send log messages to stderr for now */
init_log ( stderr ) ;
2002-01-01 00:27:39 +03:00
if ( * _sys_dir & & lvm_snprintf ( config_file , sizeof ( config_file ) ,
" %s/lvm.conf " , _sys_dir ) < 0 ) {
2001-12-31 18:20:18 +03:00
log_error ( " lvm_sys_dir was too long " ) ;
2001-12-17 15:01:09 +03:00
return 0 ;
}
if ( stat ( config_file , & info ) ! = - 1 ) {
2001-09-25 16:49:28 +04:00
/* we've found a config file */
2001-12-17 15:01:09 +03:00
if ( ! read_config ( cmd - > cf , config_file ) ) {
log_error ( " Failed to load config file %s " ,
config_file ) ;
return 0 ;
2001-09-25 16:49:28 +04:00
}
2001-11-13 17:17:50 +03:00
__init_log ( cmd - > cf ) ;
2001-09-25 16:49:28 +04:00
}
2001-12-31 18:20:18 +03:00
if ( lvm_snprintf ( _dev_dir , sizeof ( _dev_dir ) , " %s/ " ,
find_config_str ( cmd - > cf - > root , " devices/dir " ,
' / ' , DEFAULT_DEV_DIR ) ) < 0 ) {
log_error ( " Device directory given in config file too long " ) ;
return 0 ;
}
cmd - > dev_dir = _dev_dir ;
dm_set_dev_dir ( cmd - > dev_dir ) ;
2001-11-21 22:32:35 +03:00
dm_log_init ( print_log ) ;
2002-01-01 00:27:39 +03:00
if ( ! * strncpy ( _backup_dir ,
find_config_str ( cmd - > cf - > root , " backup/dir " , ' / ' , " " ) ,
sizeof ( _backup_dir ) ) & &
* _sys_dir & &
lvm_snprintf ( _backup_dir , sizeof ( _backup_dir ) , " %s/backup " ,
_sys_dir ) < 0 ) {
2001-12-31 18:20:18 +03:00
log_error ( " Backup directory given in config file too long " ) ;
return 0 ;
}
2002-01-01 00:27:39 +03:00
if ( ! create_dir ( _backup_dir ) )
return 0 ;
2001-12-31 22:09:51 +03:00
_backup_days = find_config_int ( cmd - > cf - > root , " backup/days " , ' / ' ,
_backup_days ) ;
_backup_number = find_config_int ( cmd - > cf - > root , " backup/keep " , ' / ' ,
_backup_number ) ;
_backup_auto = find_config_int ( cmd - > cf - > root , " backup/auto " , ' / ' ,
_backup_auto ) ;
2001-12-17 15:01:09 +03:00
if ( ! dev_cache_setup ( cmd - > cf ) )
return 0 ;
2001-10-08 22:44:22 +04:00
2001-11-13 17:17:50 +03:00
if ( ! ( cmd - > filter = filter_setup ( cmd - > cf ) ) ) {
2001-10-23 15:50:49 +04:00
log_error ( " Failed to set up internal device filters " ) ;
2001-12-17 15:01:09 +03:00
return 0 ;
2001-10-08 22:44:22 +04:00
}
2001-11-12 18:10:01 +03:00
if ( ! ( cmd - > mem = pool_create ( 4 * 1024 ) ) ) {
log_error ( " Command pool creation failed " ) ;
2001-12-17 15:01:09 +03:00
return 0 ;
2001-10-02 02:12:10 +04:00
}
2001-12-17 15:01:09 +03:00
if ( ! ( fid = create_lvm1_format ( cmd ) ) )
return 0 ;
2001-09-25 16:49:28 +04:00
2001-12-17 15:01:09 +03:00
return 1 ;
2001-09-25 16:49:28 +04:00
}
static void __fin_commands ( void )
{
int i ;
for ( i = 0 ; i < _num_commands ; i + + )
dbg_free ( _commands [ i ] . valid_args ) ;
dbg_free ( _commands ) ;
}
static void fin ( void )
{
2001-11-14 13:44:14 +03:00
if ( _dump_filter )
2001-11-12 18:10:01 +03:00
persistent_filter_dump ( cmd - > filter ) ;
2001-10-24 21:53:50 +04:00
2001-11-12 18:10:01 +03:00
fid - > ops - > destroy ( fid ) ;
cmd - > filter - > destroy ( cmd - > filter ) ;
pool_destroy ( cmd - > mem ) ;
2001-12-13 03:07:29 +03:00
vgcache_destroy ( ) ;
2001-10-01 23:36:06 +04:00
dev_cache_exit ( ) ;
2001-11-13 17:17:50 +03:00
destroy_config_file ( cmd - > cf ) ;
2001-11-12 18:10:01 +03:00
dbg_free ( cmd ) ;
2001-09-25 16:49:28 +04:00
__fin_commands ( ) ;
dump_memory ( ) ;
fin_log ( ) ;
if ( _log )
fclose ( _log ) ;
}
static int run_script ( int argc , char * * argv )
{
FILE * script ;
char buffer [ CMD_LEN ] ;
int ret = 0 ;
int magic_number = 0 ;
if ( ( script = fopen ( argv [ 0 ] , " r " ) ) = = NULL )
2001-10-06 01:39:30 +04:00
return ENO_SUCH_CMD ;
2001-09-25 16:49:28 +04:00
2001-11-10 01:01:04 +03:00
while ( fgets ( buffer , sizeof ( buffer ) , script ) ! = NULL ) {
2001-09-25 16:49:28 +04:00
if ( ! magic_number ) {
if ( buffer [ 0 ] = = ' # ' & & buffer [ 1 ] = = ' ! ' )
magic_number = 1 ;
else
2001-10-06 01:39:30 +04:00
return ENO_SUCH_CMD ;
2001-09-25 16:49:28 +04:00
}
2001-11-10 01:01:04 +03:00
if ( ( strlen ( buffer ) = = sizeof ( buffer ) - 1 )
& & ( buffer [ sizeof ( buffer ) - 1 ] - 2 ! = ' \n ' ) ) {
2001-09-25 16:49:28 +04:00
buffer [ 50 ] = ' \0 ' ;
log_error ( " Line too long (max 255) beginning: %s " ,
buffer ) ;
2001-10-06 01:39:30 +04:00
ret = EINVALID_CMD_LINE ;
2001-09-25 16:49:28 +04:00
break ;
}
if ( split ( buffer , & argc , argv , MAX_ARGS ) = = MAX_ARGS ) {
buffer [ 50 ] = ' \0 ' ;
log_error ( " Too many arguments: %s " , buffer ) ;
2001-10-06 01:39:30 +04:00
ret = EINVALID_CMD_LINE ;
2001-09-25 16:49:28 +04:00
break ;
}
if ( ! argc )
continue ;
if ( ! strcmp ( argv [ 0 ] , " quit " ) )
break ;
run_command ( argc , argv ) ;
}
fclose ( script ) ;
return ret ;
}
# ifdef READLINE_SUPPORT
/* List matching commands */
2001-12-17 20:18:47 +03:00
static char * _list_cmds ( const char * text , int state )
2001-09-25 16:49:28 +04:00
{
static int i = 0 ;
static int len = 0 ;
/* Initialise if this is a new completion attempt */
if ( ! state ) {
i = 0 ;
len = strlen ( text ) ;
}
while ( i < _num_commands )
if ( ! strncmp ( text , _commands [ i + + ] . name , len ) )
return strdup ( _commands [ i - 1 ] . name ) ;
return NULL ;
}
/* List matching arguments */
2001-12-17 20:18:47 +03:00
static char * _list_args ( const char * text , int state )
2001-09-25 16:49:28 +04:00
{
static int match_no = 0 ;
static int len = 0 ;
static struct command * com ;
/* Initialise if this is a new completion attempt */
if ( ! state ) {
char * s = rl_line_buffer ;
int j = 0 ;
match_no = 0 ;
com = NULL ;
len = strlen ( text ) ;
/* Find start of first word in line buffer */
while ( isspace ( * s ) )
s + + ;
/* Look for word in list of commands */
for ( j = 0 ; j < _num_commands ; j + + ) {
char * p ;
char * q = s ;
p = ( char * ) _commands [ j ] . name ;
while ( * p = = * q ) {
p + + ;
q + + ;
}
if ( ( ! * p ) & & * q = = ' ' ) {
com = _commands + j ;
break ;
}
}
if ( ! com )
return NULL ;
}
/* Short form arguments */
if ( len < 3 ) {
while ( match_no < com - > num_args ) {
char s [ 3 ] ;
char c ;
if ( ! ( c = ( the_args +
com - > valid_args [ match_no + + ] ) - > short_arg ) )
2001-11-10 01:01:04 +03:00
continue ;
2001-09-25 16:49:28 +04:00
sprintf ( s , " -%c " , c ) ;
if ( ! strncmp ( text , s , len ) )
return strdup ( s ) ;
}
}
/* Long form arguments */
if ( match_no < com - > num_args )
match_no = com - > num_args ;
while ( match_no - com - > num_args < com - > num_args ) {
char * l ;
l = ( the_args +
com - > valid_args [ match_no + + - com - > num_args ] ) - > long_arg ;
if ( ! strncmp ( text , l , len ) )
return strdup ( l ) ;
}
return NULL ;
}
2001-12-17 20:18:47 +03:00
/* Custom completion function */
static char * * _completion ( const char * text , int start_pos , int end_pos )
{
char * * match_list = NULL ;
int p = 0 ;
while ( isspace ( ( int ) * ( rl_line_buffer + p ) ) )
p + + ;
/* First word should be one of our commands */
if ( start_pos = = p )
match_list = rl_completion_matches ( text , _list_cmds ) ;
else if ( * text = = ' - ' )
match_list = rl_completion_matches ( text , _list_args ) ;
/* else other args */
/* No further completion */
rl_attempted_completion_over = 1 ;
return match_list ;
}
2001-12-17 20:59:58 +03:00
static int _hist_file ( char * buffer , size_t size )
{
char * e = getenv ( " HOME " ) ;
if ( lvm_snprintf ( buffer , size , " %s/.lvm_history " , e ) < 0 ) {
2001-12-31 18:20:18 +03:00
log_error ( " $HOME/.lvm_history: path too long " ) ;
2001-12-17 20:59:58 +03:00
return 0 ;
}
return 1 ;
}
2001-12-31 18:20:18 +03:00
2001-12-17 20:59:58 +03:00
static void _read_history ( void )
{
char hist_file [ PATH_MAX ] ;
if ( ! _hist_file ( hist_file , sizeof ( hist_file ) ) )
return ;
if ( read_history ( hist_file ) )
log_very_verbose ( " Couldn't read history from %s. " , hist_file ) ;
2001-12-31 18:20:18 +03:00
stifle_history ( find_config_int ( cmd - > cf - > root , " shell/history_size " ,
' / ' , MAX_HISTORY ) ) ;
2001-12-17 20:59:58 +03:00
}
static void _write_history ( void )
{
char hist_file [ PATH_MAX ] ;
if ( ! _hist_file ( hist_file , sizeof ( hist_file ) ) )
return ;
if ( write_history ( hist_file ) )
log_very_verbose ( " Couldn't write history to %s. " , hist_file ) ;
}
2001-09-25 16:49:28 +04:00
static int shell ( void )
{
2001-10-15 16:49:58 +04:00
int argc , ret ;
char * input = NULL , * args [ MAX_ARGS ] , * * argv ;
2001-09-25 16:49:28 +04:00
rl_readline_name = " lvm " ;
2001-12-17 20:18:47 +03:00
rl_attempted_completion_function = ( CPPFunction * ) _completion ;
2001-09-25 16:49:28 +04:00
2001-12-17 20:59:58 +03:00
_read_history ( ) ;
2001-09-25 16:49:28 +04:00
_interactive = 1 ;
while ( 1 ) {
2001-10-15 16:49:58 +04:00
free ( input ) ;
2001-09-25 16:49:28 +04:00
input = readline ( " lvm> " ) ;
/* EOF */
if ( ! input ) {
printf ( " \n " ) ;
break ;
}
/* empty line */
if ( ! * input )
continue ;
add_history ( input ) ;
2001-10-15 16:49:58 +04:00
argv = args ;
2001-09-25 16:49:28 +04:00
if ( split ( input , & argc , argv , MAX_ARGS ) = = MAX_ARGS ) {
log_error ( " Too many arguments, sorry. " ) ;
continue ;
}
2001-10-15 16:49:58 +04:00
if ( ! strcmp ( argv [ 0 ] , " lvm " ) ) {
argv + + ;
argc - - ;
}
2001-09-25 16:49:28 +04:00
if ( ! argc )
continue ;
2001-10-15 16:49:58 +04:00
if ( ! strcmp ( argv [ 0 ] , " quit " ) ) {
2001-12-17 20:59:58 +03:00
remove_history ( history_length - 1 ) ;
2001-10-15 16:49:58 +04:00
log_error ( " Exiting. " ) ;
2001-09-25 16:49:28 +04:00
break ;
2001-10-15 16:49:58 +04:00
}
2001-09-25 16:49:28 +04:00
2001-10-15 16:49:58 +04:00
ret = run_command ( argc , argv ) ;
if ( ret = = ENO_SUCH_CMD )
log_error ( " No such command '%s'. Try 'help'. " ,
argv [ 0 ] ) ;
2001-09-25 16:49:28 +04:00
}
2001-12-17 20:59:58 +03:00
_write_history ( ) ;
2001-09-25 16:49:28 +04:00
free ( input ) ;
return 0 ;
}
# endif