513 lines
17 KiB
C++
513 lines
17 KiB
C++
/* Copyright (C) 2009, 2010, 2011 Curtis Gedak
|
|
*
|
|
* 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
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include "../include/DMRaid.h"
|
|
|
|
#include <stdlib.h> //atoi function
|
|
|
|
namespace GParted
|
|
{
|
|
|
|
//Initialize static data elements
|
|
bool DMRaid::dmraid_cache_initialized = false ;
|
|
bool DMRaid::dmraid_found = false ;
|
|
bool DMRaid::dmsetup_found = false ;
|
|
bool DMRaid::udevinfo_found = false ;
|
|
bool DMRaid::udevadm_found = false ;
|
|
std::vector<Glib::ustring> DMRaid::dmraid_devices ;
|
|
|
|
DMRaid::DMRaid()
|
|
{
|
|
//Ensure that cache has been loaded at least once
|
|
if ( ! dmraid_cache_initialized )
|
|
{
|
|
dmraid_cache_initialized = true ;
|
|
set_commands_found() ;
|
|
load_dmraid_cache() ;
|
|
}
|
|
}
|
|
|
|
DMRaid::DMRaid( const bool & do_refresh )
|
|
{
|
|
//Ensure that cache has been loaded at least once
|
|
if ( ! dmraid_cache_initialized )
|
|
{
|
|
dmraid_cache_initialized = true ;
|
|
set_commands_found() ;
|
|
if ( do_refresh == false )
|
|
load_dmraid_cache() ;
|
|
}
|
|
|
|
if ( do_refresh )
|
|
load_dmraid_cache() ;
|
|
}
|
|
|
|
DMRaid::~DMRaid()
|
|
{
|
|
}
|
|
|
|
void DMRaid::load_dmraid_cache()
|
|
{
|
|
//Load data into dmraid structures
|
|
Glib::ustring output, error ;
|
|
dmraid_devices .clear() ;
|
|
|
|
if ( dmraid_found )
|
|
{
|
|
if ( ! Utils::execute_command( "dmraid -sa -c", output, error, true ) )
|
|
{
|
|
Glib::ustring temp = Utils::regexp_label( output, "^(no raid disks).*" ) ;
|
|
if ( temp != "no raid disks" )
|
|
Utils::tokenize( output, dmraid_devices, "\n" ) ;
|
|
}
|
|
}
|
|
}
|
|
|
|
void DMRaid::set_commands_found()
|
|
{
|
|
//Set status of commands found
|
|
dmraid_found = (! Glib::find_program_in_path( "dmraid" ) .empty() ) ;
|
|
dmsetup_found = (! Glib::find_program_in_path( "dmsetup" ) .empty() ) ;
|
|
udevinfo_found = (! Glib::find_program_in_path( "udevinfo" ) .empty() ) ;
|
|
udevadm_found = (! Glib::find_program_in_path( "udevadm" ) .empty() ) ;
|
|
}
|
|
|
|
bool DMRaid::is_dmraid_supported()
|
|
{
|
|
//Determine if dmraid is supported on this computer
|
|
return ( dmraid_found && dmsetup_found ) ;
|
|
}
|
|
|
|
bool DMRaid::is_dmraid_device( const Glib::ustring & dev_path )
|
|
{
|
|
//Determine if the device is a dmraid device
|
|
bool device_found = false ; //assume not found
|
|
|
|
if ( is_dmraid_supported() )
|
|
{
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
if ( dev_path .find( dmraid_devices[k] ) != Glib::ustring::npos )
|
|
device_found = true ;
|
|
}
|
|
|
|
//Some distros appear to default to /dev/dm-# for device names, so
|
|
// also check with udev to see if they are in fact dmraid devices
|
|
if ( ! device_found && ( dev_path .find( "/dev/dm" ) != Glib::ustring::npos ) )
|
|
{
|
|
//Ensure we use the base device name if dev_path in form /dev/dm-#p#
|
|
Glib::ustring regexp = "(/dev/dm-[0-9]+)p[0-9]+" ;
|
|
Glib::ustring dev_path_only = Utils::regexp_label( dev_path, regexp ) ;
|
|
if ( dev_path_only .empty() )
|
|
dev_path_only = dev_path ;
|
|
|
|
Glib::ustring udev_name = get_udev_dm_name( dev_path_only ) ;
|
|
if ( ! udev_name .empty() )
|
|
{
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
if ( udev_name .find( dmraid_devices[k] ) != Glib::ustring::npos )
|
|
device_found = true ;
|
|
}
|
|
|
|
//Also check for a symbolic link if device not yet found
|
|
if ( ! device_found && file_test( dev_path, Glib::FILE_TEST_IS_SYMLINK ) )
|
|
{
|
|
//Path is a symbolic link so find real path
|
|
char c_str[4096+1] ;
|
|
//FIXME: it seems realpath is very unsafe to use (manpage)...
|
|
if ( realpath( dev_path .c_str(), c_str ) != NULL )
|
|
{
|
|
Glib::ustring tmp_path = c_str ;
|
|
if ( tmp_path .length() > 0 )
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
if ( tmp_path .find( dmraid_devices[k] ) != Glib::ustring::npos )
|
|
device_found = true ;
|
|
}
|
|
}
|
|
}
|
|
|
|
return device_found ;
|
|
}
|
|
|
|
//FIXME: Copy of code from FileSystem.cc.
|
|
// This should be replaced when a general execute_command with asynchronous updates is built.
|
|
int DMRaid::execute_command( const Glib::ustring & command, OperationDetail & operationdetail )
|
|
{
|
|
Glib::ustring output, error ;
|
|
|
|
operationdetail .add_child( OperationDetail( command, STATUS_NONE, FONT_BOLD_ITALIC ) ) ;
|
|
|
|
int exit_status = Utils::execute_command( "nice -n 19 " + command, output, error ) ;
|
|
|
|
if ( ! output .empty() )
|
|
operationdetail .get_last_child() .add_child( OperationDetail( output, STATUS_NONE, FONT_ITALIC ) ) ;
|
|
|
|
if ( ! error .empty() )
|
|
operationdetail .get_last_child() .add_child( OperationDetail( error, STATUS_NONE, FONT_ITALIC ) ) ;
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
void DMRaid::get_devices( std::vector<Glib::ustring> & device_list )
|
|
{
|
|
//Retrieve list of dmraid devices
|
|
device_list .clear() ;
|
|
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
device_list .push_back( DEV_MAP_PATH + dmraid_devices[k] ) ;
|
|
}
|
|
|
|
Glib::ustring DMRaid::get_dmraid_name( const Glib::ustring & dev_path )
|
|
{
|
|
//Retrieve name of dmraid device
|
|
Glib::ustring dmraid_name = "" ;
|
|
Glib::ustring regexp = "" ;
|
|
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
{
|
|
regexp = ".*(" + dmraid_devices[k] + ").*" ;
|
|
if ( Utils::regexp_label( dev_path, regexp ) == dmraid_devices[k] )
|
|
dmraid_name = dmraid_devices[k] ;
|
|
}
|
|
|
|
//Some distros appear to default to /dev/dm-# for device names, so
|
|
// also check with udev for dmraid name
|
|
if ( dmraid_name .empty() && ( dev_path .find( "/dev/dm" ) != Glib::ustring::npos ) )
|
|
{
|
|
Glib::ustring udev_name = get_udev_dm_name( dev_path ) ;
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
if ( udev_name .find( dmraid_devices[k] ) != Glib::ustring::npos )
|
|
dmraid_name = dmraid_devices[k] ;
|
|
|
|
//Also check for a symbolic link if dmraid_name not yet found
|
|
if ( dmraid_name .empty() && file_test( dev_path, Glib::FILE_TEST_IS_SYMLINK ) )
|
|
{
|
|
//Path is a symbolic link so find real path
|
|
char c_str[4096+1] ;
|
|
//FIXME: it seems realpath is very unsafe to use (manpage)...
|
|
if( realpath( dev_path .c_str(), c_str ) != NULL )
|
|
{
|
|
Glib::ustring tmp_path = c_str ;
|
|
if ( tmp_path .length() > 0 )
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
if ( tmp_path .find( dmraid_devices[k] ) != Glib::ustring::npos )
|
|
dmraid_name = dmraid_devices[k] ;
|
|
}
|
|
}
|
|
}
|
|
|
|
return dmraid_name ;
|
|
}
|
|
|
|
void DMRaid::get_dmraid_dir_entries( const Glib::ustring & dev_path, std::vector<Glib::ustring> & dir_list )
|
|
{
|
|
//Build list of all device entries matching device path
|
|
|
|
Glib::ustring dmraid_name = get_dmraid_name( dev_path ) ;
|
|
|
|
//Loop through the entries in the directory
|
|
Glib::ustring filename = "" ;
|
|
Glib::Dir dir( DEV_MAP_PATH) ;
|
|
while ( ( filename = dir .read_name() ) != "" )
|
|
{
|
|
if ( filename == "control" )
|
|
continue ;
|
|
|
|
if ( Utils::regexp_label( filename, "^(" + dmraid_name + ")" ) == dmraid_name )
|
|
dir_list .push_back( filename ) ;
|
|
}
|
|
}
|
|
|
|
int DMRaid::get_partition_number( const Glib::ustring & partition_name )
|
|
{
|
|
Glib::ustring dmraid_name = get_dmraid_name( partition_name ) ;
|
|
return std::atoi( Utils::regexp_label( partition_name, dmraid_name + "p?([0-9]+)" ) .c_str() ) ;
|
|
}
|
|
|
|
Glib::ustring DMRaid::get_udev_dm_name( const Glib::ustring & dev_path )
|
|
{
|
|
//Retrieve DM_NAME of device using udev information
|
|
Glib::ustring output = "" ;
|
|
Glib::ustring error = "" ;
|
|
Glib::ustring dm_name = "" ;
|
|
|
|
if ( udevinfo_found )
|
|
Utils::execute_command( "udevinfo --query=all --name=" + dev_path, output, error, true ) ;
|
|
else if ( udevadm_found )
|
|
Utils::execute_command( "udevadm info --query=all --name=" + dev_path, output, error, true ) ;
|
|
|
|
if ( ! output .empty() )
|
|
{
|
|
Glib::ustring regexp = "^E: DM_NAME=([^\n]*)$" ;
|
|
dm_name = Utils::regexp_label( output, regexp ) ;
|
|
}
|
|
|
|
return dm_name ;
|
|
}
|
|
|
|
Glib::ustring DMRaid::make_path_dmraid_compatible( Glib::ustring partition_path )
|
|
{
|
|
//The purpose of this method is to ensure that the partition name matches
|
|
// the name that the dmraid command would create.
|
|
//
|
|
//From my experience, the general rule of thumb naming convention for creating
|
|
// partition names is as follows:
|
|
//
|
|
// A) If the device name ends in a number, then append the letter "p" and
|
|
// the partition number to create the partition name.
|
|
// For Example:
|
|
// Device Name : /dev/mapper/isw_cjbdddajhi_Disk0
|
|
// Partition Name: /dev/mapper/isw_cjbdddajhi_Disk0p1
|
|
//
|
|
// B) If the device name does not end in a number, then append the
|
|
// partition number only to create the partition name.
|
|
// For Example:
|
|
// Device Name : /dev/mapper/isw_cjbdddajhi_Volume
|
|
// Partition Name: /dev/mapper/isw_cjbdddajhi_Volume1
|
|
//
|
|
//The dmraid command appears to never add the "p" into the partition name.
|
|
// Instead in all cases dmraid simply appends the partition number to the
|
|
// device name to create a partition name.
|
|
//
|
|
Glib::ustring reg_exp ;
|
|
Glib::ustring partition_number ;
|
|
|
|
for ( unsigned int k=0; k < dmraid_devices .size(); k++ )
|
|
{
|
|
reg_exp = DEV_MAP_PATH + dmraid_devices[ k ] + "p([0-9]+)" ;
|
|
partition_number = Utils::regexp_label( partition_path, reg_exp ) ;
|
|
if ( ! partition_number .empty() )
|
|
{
|
|
partition_path = DEV_MAP_PATH + dmraid_devices[ k ] + partition_number ;
|
|
return partition_path ;
|
|
}
|
|
}
|
|
|
|
//If parted is configured with --disable-device-mapper then the partition path
|
|
// could default to something like /dev/dm-#p# on some distros, so we need to
|
|
// convert these names back into /dev/mapper entries
|
|
reg_exp = "/dev/dm-[0-9]+p([0-9]+)" ;
|
|
partition_number = Utils::regexp_label( partition_path, reg_exp ) ;
|
|
if ( ! partition_number .empty() )
|
|
{
|
|
reg_exp = "(/dev/dm-[0-9]+)p[0-9]+" ;
|
|
Glib::ustring device_path = Utils::regexp_label( partition_path, reg_exp ) ;
|
|
if ( ! device_path .empty() )
|
|
{
|
|
device_path = get_udev_dm_name( device_path ) ;
|
|
partition_path = DEV_MAP_PATH + device_path + partition_number ;
|
|
}
|
|
}
|
|
|
|
return partition_path ;
|
|
}
|
|
|
|
bool DMRaid::create_dev_map_entries( const Partition & partition, OperationDetail & operationdetail )
|
|
{
|
|
//Create all missing dev mapper entries for a specified device.
|
|
|
|
Glib::ustring command ;
|
|
bool exit_status = true ;
|
|
|
|
/*TO TRANSLATORS: looks like create missing /dev/mapper entries */
|
|
Glib::ustring tmp = String::ucompose ( _("create missing %1 entries"), DEV_MAP_PATH ) ;
|
|
operationdetail .add_child( OperationDetail( tmp ) );
|
|
|
|
//Newer dmraid defaults to always inserting the letter 'p' between the device name
|
|
// and the partition number. Since the '-P' option is not valid for older dmraid
|
|
// versions, try with and without the '-P' option.
|
|
command = "dmraid -ay -P \"\" -v" ;
|
|
if ( execute_command( command, operationdetail .get_last_child() ) )
|
|
{
|
|
//Above command failed, so try without the '-P' option.
|
|
command = "dmraid -ay -v" ;
|
|
if ( execute_command( command, operationdetail .get_last_child() ) )
|
|
exit_status = false ; //command failed
|
|
}
|
|
|
|
operationdetail .get_last_child() .set_status( exit_status ? STATUS_SUCCES : STATUS_ERROR ) ;
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
bool DMRaid::create_dev_map_entries( const Glib::ustring & dev_path )
|
|
{
|
|
//Create all missing dev mapper entries for a specified device.
|
|
|
|
Glib::ustring command, output, error ;
|
|
bool exit_status = true ;
|
|
|
|
//Newer dmraid defaults to always inserting the letter 'p' between the device name
|
|
// and the partition number. Since the '-P' option is not valid for older dmraid
|
|
// versions, try with and without the '-P' option.
|
|
command = "dmraid -ay -P \"\" -v" ;
|
|
if ( Utils::execute_command( command, output, error, true ) )
|
|
{
|
|
//Above command failed, so try without the '-P' option.
|
|
command = "dmraid -ay -v" ;
|
|
if ( Utils::execute_command( command, output, error, true ) )
|
|
exit_status = false; //command failed
|
|
}
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
void DMRaid::get_affected_dev_map_entries( const Partition & partition, std::vector<Glib::ustring> & affected_entries )
|
|
{
|
|
//Build list of affected /dev/mapper entries when a partition is to be deleted.
|
|
|
|
//Retrieve list of matching directory entries
|
|
std::vector<Glib::ustring> dir_list ;
|
|
get_dmraid_dir_entries( partition .device_path, dir_list );
|
|
|
|
//All partition numbers equal to the number of the partition to be deleted
|
|
// will be affected.
|
|
// Also, if the partition is inside an extended partition, then all logical
|
|
// partition numbers greater than the number of the partition to be deleted
|
|
// will be affected.
|
|
Glib::ustring dmraid_name = get_dmraid_name( partition .device_path ) ;
|
|
for ( unsigned int k=0; k < dir_list .size(); k++ )
|
|
{
|
|
if ( Utils::regexp_label( dir_list[k], "^(" + dmraid_name + ")" ) == dmraid_name )
|
|
{
|
|
int dir_part_num = get_partition_number( dir_list[k] ) ;
|
|
if ( dir_part_num == partition .partition_number ||
|
|
( partition .inside_extended && dir_part_num > partition .partition_number )
|
|
)
|
|
affected_entries .push_back( dir_list[k] ) ;
|
|
}
|
|
}
|
|
}
|
|
|
|
void DMRaid::get_partition_dev_map_entries( const Partition & partition, std::vector<Glib::ustring> & partition_entries )
|
|
{
|
|
//Build list of all /dev/mapper entries for a partition.
|
|
|
|
//Retrieve list of matching directory entries
|
|
std::vector<Glib::ustring> dir_list ;
|
|
get_dmraid_dir_entries( partition .device_path, dir_list );
|
|
|
|
//Retrieve all partition numbers equal to the number of the partition.
|
|
Glib::ustring dmraid_name = get_dmraid_name( partition .device_path ) ;
|
|
for ( unsigned int k=0; k < dir_list .size(); k++ )
|
|
{
|
|
if ( Utils::regexp_label( dir_list[k], "^(" + dmraid_name + ")" ) == dmraid_name )
|
|
{
|
|
int dir_part_num = get_partition_number( dir_list[k] ) ;
|
|
if ( dir_part_num == partition .partition_number )
|
|
partition_entries .push_back( dir_list[k] ) ;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool DMRaid::delete_affected_dev_map_entries( const Partition & partition, OperationDetail & operationdetail )
|
|
{
|
|
//Delete all affected dev mapper entries (logical partitions >= specificied partition)
|
|
|
|
std::vector<Glib::ustring> affected_entries ;
|
|
Glib::ustring command ;
|
|
bool exit_status = true ;
|
|
|
|
/*TO TRANSLATORS: looks like delete affected /dev/mapper entries */
|
|
Glib::ustring tmp = String::ucompose ( _("delete affected %1 entries"), DEV_MAP_PATH ) ;
|
|
operationdetail .add_child( OperationDetail( tmp ) );
|
|
|
|
get_affected_dev_map_entries( partition, affected_entries ) ;
|
|
|
|
for ( unsigned int k=0; k < affected_entries .size(); k++ )
|
|
{
|
|
command = "dmsetup remove " + DEV_MAP_PATH + affected_entries[k] ;
|
|
if ( execute_command( command, operationdetail .get_last_child() ) )
|
|
exit_status = false ; //command failed
|
|
}
|
|
|
|
operationdetail .get_last_child() .set_status( exit_status ? STATUS_SUCCES : STATUS_ERROR ) ;
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
bool DMRaid::delete_dev_map_entry( const Partition & partition, OperationDetail & operationdetail )
|
|
{
|
|
//Delete a single partition which may be represented by multiple dev mapper entries
|
|
bool exit_status = true ;
|
|
|
|
/*TO TRANSLATORS: looks like delete /dev/mapper entry */
|
|
Glib::ustring tmp = String::ucompose ( _("delete %1 entry"), DEV_MAP_PATH ) ;
|
|
operationdetail .add_child( OperationDetail( tmp ) );
|
|
|
|
std::vector<Glib::ustring> partition_entries ;
|
|
get_partition_dev_map_entries( partition, partition_entries ) ;
|
|
|
|
for ( unsigned int k = 0; k < partition_entries .size(); k++ )
|
|
{
|
|
Glib::ustring command = "dmsetup remove " + partition_entries[k] ;
|
|
if ( execute_command( command, operationdetail .get_last_child() ) )
|
|
exit_status = false ; //command failed
|
|
}
|
|
|
|
operationdetail .get_last_child() .set_status( exit_status ? STATUS_SUCCES : STATUS_ERROR ) ;
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
bool DMRaid::purge_dev_map_entries( const Glib::ustring & dev_path )
|
|
{
|
|
//Delete all dev mapper entries for dmraid device
|
|
|
|
std::vector<Glib::ustring> dir_list ;
|
|
Glib::ustring command ;
|
|
Glib::ustring output, error ;
|
|
bool exit_status = true ;
|
|
|
|
get_dmraid_dir_entries( dev_path, dir_list ) ;
|
|
|
|
for ( unsigned int k=0; k < dir_list .size(); k++ )
|
|
{
|
|
command = "dmsetup remove " + DEV_MAP_PATH + dir_list[k] ;
|
|
if ( Utils::execute_command( command, output, error, true ) )
|
|
exit_status = false ; //command failed
|
|
}
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
bool DMRaid::update_dev_map_entry( const Partition & partition, OperationDetail & operationdetail )
|
|
{
|
|
//Update dmraid entry by first deleting the entry then recreating the entry
|
|
|
|
//Skip if extended partition because the only change is to the partition table.
|
|
if ( partition .type == GParted::TYPE_EXTENDED )
|
|
return true ;
|
|
|
|
bool exit_status = true ;
|
|
|
|
/*TO TRANSLATORS: looks like update /dev/mapper entry */
|
|
Glib::ustring tmp = String::ucompose ( _("update %1 entry"), DEV_MAP_PATH ) ;
|
|
operationdetail .add_child( OperationDetail( tmp ) );
|
|
|
|
if( ! delete_dev_map_entry( partition , operationdetail .get_last_child() ) )
|
|
exit_status = false ; //command failed
|
|
|
|
if( ! create_dev_map_entries( partition , operationdetail .get_last_child() ) )
|
|
exit_status = false ; //command failed
|
|
|
|
operationdetail .get_last_child() .set_status( exit_status ? STATUS_SUCCES : STATUS_ERROR ) ;
|
|
|
|
return exit_status ;
|
|
}
|
|
|
|
}//GParted
|