/* Copyright (C) 2004, 2005, 2006, 2007, 2008 Bart Hakvoort * * 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/Win_GParted.h" #include "../include/GParted_Core.h" #include "../include/OperationCopy.h" #include "../include/OperationCreate.h" #include "../include/OperationDelete.h" #include "../include/OperationFormat.h" #include "../include/OperationResizeMove.h" #include "../include/OperationLabelPartition.h" #include "../include/ext2.h" #include "../include/ext3.h" #include "../include/fat16.h" #include "../include/fat32.h" #include "../include/linux_swap.h" #include "../include/reiserfs.h" #include "../include/ntfs.h" #include "../include/xfs.h" #include "../include/jfs.h" #include "../include/hfs.h" #include "../include/hfsplus.h" #include "../include/reiser4.h" #include "../include/ufs.h" #include #include std::vector libparted_messages ; //see ped_exception_handler() namespace GParted { GParted_Core::GParted_Core() { lp_device = NULL ; lp_disk = NULL ; lp_partition = NULL ; p_filesystem = NULL ; ped_exception_set_handler( ped_exception_handler ) ; //get valid flags ... for ( PedPartitionFlag flag = ped_partition_flag_next( static_cast( NULL ) ) ; flag ; flag = ped_partition_flag_next( flag ) ) flags .push_back( flag ) ; //throw libpartedversion to the stdout to see which version is actually used. std::cout << "======================" << std::endl ; std::cout << "libparted : " << ped_get_version() << std::endl ; std::cout << "======================" << std::endl ; //initialize filesystemlist find_supported_filesystems() ; } void GParted_Core::find_supported_filesystems() { FILESYSTEMS .clear() ; ext2 fs_ext2; FILESYSTEMS .push_back( fs_ext2 .get_filesystem_support() ) ; ext3 fs_ext3; FILESYSTEMS .push_back( fs_ext3 .get_filesystem_support() ) ; fat16 fs_fat16; FILESYSTEMS .push_back( fs_fat16 .get_filesystem_support() ) ; fat32 fs_fat32; FILESYSTEMS .push_back( fs_fat32 .get_filesystem_support() ) ; hfs fs_hfs; FILESYSTEMS .push_back( fs_hfs .get_filesystem_support() ) ; hfsplus fs_hfsplus; FILESYSTEMS .push_back( fs_hfsplus .get_filesystem_support() ) ; jfs fs_jfs; FILESYSTEMS .push_back( fs_jfs .get_filesystem_support() ) ; linux_swap fs_linux_swap; FILESYSTEMS .push_back( fs_linux_swap .get_filesystem_support() ) ; ntfs fs_ntfs; FILESYSTEMS .push_back( fs_ntfs .get_filesystem_support() ) ; reiser4 fs_reiser4; FILESYSTEMS .push_back( fs_reiser4 .get_filesystem_support() ) ; reiserfs fs_reiserfs; FILESYSTEMS .push_back( fs_reiserfs .get_filesystem_support() ) ; ufs fs_ufs; FILESYSTEMS .push_back( fs_ufs .get_filesystem_support() ) ; xfs fs_xfs; FILESYSTEMS .push_back( fs_xfs .get_filesystem_support() ) ; //unknown filesystem (default when no match is found) FS fs ; fs .filesystem = GParted::FS_UNKNOWN ; FILESYSTEMS .push_back( fs ) ; } void GParted_Core::set_user_devices( const std::vector & user_devices ) { this ->device_paths = user_devices ; this ->probe_devices = ! user_devices .size() ; } void GParted_Core::set_devices( std::vector & devices ) { devices .clear() ; Device temp_device ; init_maps() ; //only probe if no devices were specified as arguments.. if ( probe_devices ) { device_paths .clear() ; //try to find all available devices std::ifstream proc_partitions( "/proc/partitions" ) ; if ( proc_partitions ) { //parse device names from /proc/partitions std::string line ; std::string device ; while ( getline( proc_partitions, line ) ) { //Device names without a trailing digit refer to the whole disk. //These whole disk devices are the ones we want. device = Utils::regexp_label(line, "^[\t ]+[0-9]+[\t ]+[0-9]+[\t ]+[0-9]+[\t ]+([^0-9]+)$") ; if ( device != "" ) { //try to have libparted detect the device and add it to the list device = "/dev/" + device; ped_device_get( device .c_str() ) ; } } proc_partitions .close() ; } else { //file /proc/partitions doesn't exist so use libparted to probe devices ped_device_probe_all(); } lp_device = ped_device_get_next( NULL ); while ( lp_device ) { //only add this device if we can read the first sector (which means it's a real device) char * buf = static_cast( malloc( lp_device ->sector_size ) ) ; if ( buf ) { if ( ped_device_open( lp_device ) ) { if ( ped_device_read( lp_device, buf, 0, 1 ) ) device_paths .push_back( lp_device ->path ) ; ped_device_close( lp_device ) ; } free( buf ) ; } lp_device = ped_device_get_next( lp_device ) ; } close_device_and_disk() ; std::sort( device_paths .begin(), device_paths .end() ) ; } for ( unsigned int t = 0 ; t < device_paths .size() ; t++ ) { if ( open_device_and_disk( device_paths[ t ], false ) ) { temp_device .Reset() ; //device info.. temp_device .add_path( device_paths[ t ] ) ; temp_device .add_paths( get_alternate_paths( temp_device .get_path() ) ) ; temp_device .model = lp_device ->model ; temp_device .heads = lp_device ->bios_geom .heads ; temp_device .sectors = lp_device ->bios_geom .sectors ; temp_device .cylinders = lp_device ->bios_geom .cylinders ; temp_device .length = temp_device .heads * temp_device .sectors * temp_device .cylinders ; temp_device .cylsize = temp_device .heads * temp_device .sectors ; //make sure cylsize is at least 1 MiB if ( temp_device .cylsize < MEBIBYTE ) temp_device .cylsize = MEBIBYTE ; //normal harddisk if ( lp_disk ) { temp_device .disktype = lp_disk ->type ->name ; temp_device .max_prims = ped_disk_get_max_primary_partition_count( lp_disk ) ; set_device_partitions( temp_device ) ; set_mountpoints( temp_device .partitions ) ; set_used_sectors( temp_device .partitions ) ; if ( temp_device .highest_busy ) temp_device .readonly = ! commit_to_os( 1 ) ; } //harddisk without disklabel else { temp_device .disktype = _("unrecognized") ; temp_device .max_prims = -1 ; Partition partition_temp ; partition_temp .Set_Unallocated( temp_device .get_path(), 0, temp_device .length, false ); temp_device .partitions .push_back( partition_temp ); } devices .push_back( temp_device ) ; close_device_and_disk() ; } } //clear leftover information... //NOTE that we cannot clear mountinfo since it might be needed in get_all_mountpoints() alternate_paths .clear() ; fstab_info .clear() ; } bool GParted_Core::snap_to_cylinder( const Device & device, Partition & partition, Glib::ustring & error ) { if ( ! partition .strict ) { //FIXME: this doesn't work to well, when dealing with layouts which contain an extended partition //because of the metadata length % cylindersize isn't always 0 Sector diff = partition .sector_start % device .cylsize ; if ( diff ) { if ( diff < ( device .cylsize / 2 ) ) partition .sector_start -= diff ; else partition .sector_start += (device .cylsize - diff ) ; } diff = (partition .sector_end +1) % device .cylsize ; if ( diff ) { if ( diff < ( device .cylsize / 2 ) ) partition .sector_end -= diff ; else partition .sector_end += (device .cylsize - diff ) ; } if ( partition .sector_start < 0 ) partition .sector_start = 0 ; if ( partition .sector_end > device .length ) partition .sector_end = device .length -1 ; //ok, do some basic checks on the partition.. if ( partition .get_length() <= 0 ) { error = String::ucompose( _("A partition cannot have a length of %1 sectors"), partition .get_length() ) ; return false ; } if ( partition .get_length() < partition .sectors_used ) { error = String::ucompose( _("A partition with used sectors (%1) greater than it's length (%2) is not valid"), partition .sectors_used, partition .get_length() ) ; return false ; } //FIXME: it would be perfect if we could check for overlapping with adjacent partitions as well, //however, finding the adjacent partitions is not as easy as it seems and at this moment all the dialogs //already perform these checks. A perfect 'fixme-later' ;) } return true ; } bool GParted_Core::apply_operation_to_disk( Operation * operation ) { bool succes = false ; libparted_messages .clear() ; if ( calibrate_partition( operation ->partition_original, operation ->operation_detail ) ) switch ( operation ->type ) { case OPERATION_DELETE: succes = Delete( operation ->partition_original, operation ->operation_detail ) ; break ; case OPERATION_CHECK: succes = check_repair_filesystem( operation ->partition_original, operation ->operation_detail ) && maximize_filesystem( operation ->partition_original, operation ->operation_detail ) ; break ; case OPERATION_CREATE: succes = create( operation ->device, operation ->partition_new, operation ->operation_detail ) ; break ; case OPERATION_RESIZE_MOVE: //in case the to be resized/moved partition was a 'copy of..', we need a real path... operation ->partition_new .add_path( operation ->partition_original .get_path(), true ) ; succes = resize_move( operation ->device, operation ->partition_original, operation ->partition_new, operation ->operation_detail ) ; break ; case OPERATION_FORMAT: succes = format( operation ->partition_new, operation ->operation_detail ) ; break ; case OPERATION_COPY: //FIXME: in case of a new partition we should make sure the new partition is >= the source partition... //i think it's best to do this in the dialog_paste succes = ( operation ->partition_original .type == TYPE_UNALLOCATED || calibrate_partition( operation ->partition_new, operation ->operation_detail ) ) && calibrate_partition( static_cast( operation ) ->partition_copied, operation ->operation_detail ) && copy( static_cast( operation ) ->partition_copied, operation ->partition_new, static_cast( operation ) ->partition_copied .get_length(), operation ->operation_detail ) ; break ; case OPERATION_LABEL_PARTITION: succes = label_partition( operation ->partition_new, operation ->operation_detail ) ; break ; } if ( libparted_messages .size() > 0 ) { operation ->operation_detail .add_child( OperationDetail( _("libparted messages"), STATUS_INFO ) ) ; for ( unsigned int t = 0 ; t < libparted_messages .size() ; t++ ) operation ->operation_detail .get_last_child() .add_child( OperationDetail( libparted_messages[ t ], STATUS_NONE, FONT_ITALIC ) ) ; } return succes ; } bool GParted_Core::set_disklabel( const Glib::ustring & device_path, const Glib::ustring & disklabel ) { bool return_value = false ; if ( open_device_and_disk( device_path, false ) ) { PedDiskType *type = NULL ; type = ped_disk_type_get( disklabel .c_str() ) ; if ( type ) { lp_disk = ped_disk_new_fresh( lp_device, type ); return_value = commit() ; } close_device_and_disk() ; } return return_value ; } bool GParted_Core::toggle_flag( const Partition & partition, const Glib::ustring & flag, bool state ) { bool succes = false ; if ( open_device_and_disk( partition .device_path ) ) { lp_partition = NULL ; if ( partition .type == GParted::TYPE_EXTENDED ) lp_partition = ped_disk_extended_partition( lp_disk ) ; else lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; if ( lp_partition ) { PedPartitionFlag lp_flag = ped_partition_flag_get_by_name( flag .c_str() ) ; if ( lp_flag > 0 && ped_partition_set_flag( lp_partition, lp_flag, state ) ) succes = commit() ; } close_device_and_disk() ; } return succes ; } const std::vector & GParted_Core::get_filesystems() const { return FILESYSTEMS ; } const FS & GParted_Core::get_fs( GParted::FILESYSTEM filesystem ) const { for ( unsigned int t = 0 ; t < FILESYSTEMS .size() ; t++ ) if ( FILESYSTEMS[ t ] .filesystem == filesystem ) return FILESYSTEMS[ t ] ; return FILESYSTEMS .back() ; } std::vector GParted_Core::get_disklabeltypes() { std::vector disklabeltypes ; //msdos should be first in the list disklabeltypes .push_back( "msdos" ) ; PedDiskType *disk_type ; for ( disk_type = ped_disk_type_get_next( NULL ) ; disk_type ; disk_type = ped_disk_type_get_next( disk_type ) ) if ( Glib::ustring( disk_type->name ) != "msdos" ) disklabeltypes .push_back( disk_type->name ) ; return disklabeltypes ; } std::vector GParted_Core::get_all_mountpoints() { std::vector mountpoints ; for ( iter_mp = mount_info .begin() ; iter_mp != mount_info .end() ; ++iter_mp ) mountpoints .insert( mountpoints .end(), iter_mp ->second .begin(), iter_mp ->second .end() ) ; return mountpoints ; } std::map GParted_Core::get_available_flags( const Partition & partition ) { std::map flag_info ; if ( open_device_and_disk( partition .device_path ) ) { lp_partition = NULL ; if ( partition .type == GParted::TYPE_EXTENDED ) lp_partition = ped_disk_extended_partition( lp_disk ) ; else lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; if ( lp_partition ) { for ( unsigned int t = 0 ; t < flags .size() ; t++ ) if ( ped_partition_is_flag_available( lp_partition, flags[ t ] ) ) flag_info[ ped_partition_flag_get_name( flags[ t ] ) ] = ped_partition_get_flag( lp_partition, flags[ t ] ) ; } close_device_and_disk() ; } return flag_info ; } Glib::ustring GParted_Core::get_libparted_version() { return ped_get_version() ; } //private functions... void GParted_Core::init_maps() { alternate_paths .clear() ; mount_info .clear() ; fstab_info .clear() ; read_mountpoints_from_file( "/proc/mounts", mount_info ) ; read_mountpoints_from_file( "/etc/mtab", mount_info ) ; read_mountpoints_from_file( "/etc/fstab", fstab_info ) ; //sort the mountpoints and remove duplicates.. (no need to do this for fstab_info) for ( iter_mp = mount_info .begin() ; iter_mp != mount_info .end() ; ++iter_mp ) { std::sort( iter_mp ->second .begin(), iter_mp ->second .end() ) ; iter_mp ->second .erase( std::unique( iter_mp ->second .begin(), iter_mp ->second .end() ), iter_mp ->second .end() ) ; } //initialize alternate_paths... std::string line ; std::ifstream proc_partitions( "/proc/partitions" ) ; if ( proc_partitions ) { char c_str[4096+1] ; while ( getline( proc_partitions, line ) ) if ( sscanf( line .c_str(), "%*d %*d %*d %4096s", c_str ) == 1 ) { line = "/dev/" ; line += c_str ; //FIXME: it seems realpath is very unsafe to use (manpage)... if ( file_test( line, Glib::FILE_TEST_EXISTS ) && realpath( line .c_str(), c_str ) && line != c_str ) { //because we can make no assumption about which path libparted will detect //we add all combinations. alternate_paths[ c_str ] = line ; alternate_paths[ line ] = c_str ; } } proc_partitions .close() ; } } void GParted_Core::read_mountpoints_from_file( const Glib::ustring & filename, std::map< Glib::ustring, std::vector > & map ) { std::string line ; char node[4096+1], mountpoint[4096+1] ; unsigned int index ; std::ifstream file( filename .c_str() ) ; if ( file ) { while ( getline( file, line ) ) if ( Glib::str_has_prefix( line, "/" ) && sscanf( line .c_str(), "%4096s %4096s", node, mountpoint ) == 2 && Glib::ustring( node ) != "/dev/root" ) { line = mountpoint ; //see if mountpoint contains spaces and deal with it index = line .find( "\\040" ) ; if ( index < line .length() ) line .replace( index, 4, " " ) ; //only add this path if it exists if ( file_test( line, Glib::FILE_TEST_EXISTS ) ) map[ node ] .push_back( line ) ; } file .close() ; } } std::vector GParted_Core::get_alternate_paths( const Glib::ustring & path ) { std::vector paths ; iter = alternate_paths .find( path ) ; if ( iter != alternate_paths .end() ) paths .push_back( iter ->second ) ; return paths ; } void GParted_Core::set_device_partitions( Device & device ) { int EXT_INDEX = -1 ; char * lp_path ;//we have to free the result of ped_partition_get_path().. //clear partitions device .partitions .clear() ; lp_partition = ped_disk_next_partition( lp_disk, NULL ) ; while ( lp_partition ) { libparted_messages .clear() ; partition_temp .Reset() ; switch ( lp_partition ->type ) { case PED_PARTITION_NORMAL: case PED_PARTITION_LOGICAL: lp_path = ped_partition_get_path( lp_partition ) ; partition_temp .Set( device .get_path(), lp_path, lp_partition ->num, lp_partition ->type == 0 ? GParted::TYPE_PRIMARY : GParted::TYPE_LOGICAL, get_filesystem(), lp_partition ->geom .start, lp_partition ->geom .end, lp_partition ->type, ped_partition_is_busy( lp_partition ) ) ; free( lp_path ) ; partition_temp .add_paths( get_alternate_paths( partition_temp .get_path() ) ) ; set_flags( partition_temp ) ; if ( partition_temp .busy && partition_temp .partition_number > device .highest_busy ) device .highest_busy = partition_temp .partition_number ; break ; case PED_PARTITION_EXTENDED: lp_path = ped_partition_get_path( lp_partition ) ; partition_temp .Set( device .get_path(), lp_path, lp_partition ->num, GParted::TYPE_EXTENDED, GParted::FS_EXTENDED, lp_partition ->geom .start, lp_partition ->geom .end, false, ped_partition_is_busy( lp_partition ) ) ; free( lp_path ) ; partition_temp .add_paths( get_alternate_paths( partition_temp .get_path() ) ) ; set_flags( partition_temp ) ; EXT_INDEX = device .partitions .size() ; break ; default: break; } get_label( partition_temp ) ; partition_temp .messages .insert( partition_temp .messages .end(), libparted_messages. begin(), libparted_messages .end() ) ; //if there's an end, there's a partition ;) if ( partition_temp .sector_end > -1 ) { if ( ! partition_temp .inside_extended ) device .partitions .push_back( partition_temp ); else device .partitions[ EXT_INDEX ] .logicals .push_back( partition_temp ) ; } //next partition (if any) lp_partition = ped_disk_next_partition( lp_disk, lp_partition ) ; } if ( EXT_INDEX > -1 ) insert_unallocated( device .get_path(), device .partitions[ EXT_INDEX ] .logicals, device .partitions[ EXT_INDEX ] .sector_start, device .partitions[ EXT_INDEX ] .sector_end, true ) ; insert_unallocated( device .get_path(), device .partitions, 0, device .length -1, false ) ; } GParted::FILESYSTEM GParted_Core::get_filesystem() { //standard libparted filesystems.. if ( lp_partition && lp_partition ->fs_type ) { if ( Glib::ustring( lp_partition ->fs_type ->name ) == "extended" ) return GParted::FS_EXTENDED ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "ext2" ) return GParted::FS_EXT2 ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "ext3" ) return GParted::FS_EXT3 ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "linux-swap" ) return GParted::FS_LINUX_SWAP ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "fat16" ) return GParted::FS_FAT16 ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "fat32" ) return GParted::FS_FAT32 ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "ntfs" ) return GParted::FS_NTFS ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "reiserfs" ) return GParted::FS_REISERFS ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "xfs" ) return GParted::FS_XFS ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "jfs" ) return GParted::FS_JFS ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "hfs" ) return GParted::FS_HFS ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "hfs+" ) return GParted::FS_HFSPLUS ; else if ( Glib::ustring( lp_partition ->fs_type ->name ) == "ufs" ) return GParted::FS_UFS ; } //other filesystems libparted couldn't detect (i've send patches for these filesystems to the parted guys) char buf[512] ; ped_device_open( lp_device ); //reiser4 ped_geometry_read( & lp_partition ->geom, buf, 128, 1 ) ; ped_device_close( lp_device ); if ( Glib::ustring( buf ) == "ReIsEr4" ) return GParted::FS_REISER4 ; //no filesystem found.... temp = _( "Unable to detect filesystem! Possible reasons are:" ) ; temp += "\n-"; temp += _( "The filesystem is damaged" ) ; temp += "\n-" ; temp += _( "The filesystem is unknown to GParted" ) ; temp += "\n-"; temp += _( "There is no filesystem available (unformatted)" ) ; partition_temp .messages .push_back( temp ) ; return GParted::FS_UNKNOWN ; } void GParted_Core::get_label( Partition & partition ) { if ( partition .type != TYPE_EXTENDED ) { switch( get_fs( partition .filesystem ) .get_label ) { case FS::EXTERNAL: if ( set_proper_filesystem( partition .filesystem ) ) p_filesystem ->get_label( partition ) ; break ; case FS::LIBPARTED: break ; default: break ; } } } void GParted_Core::insert_unallocated( const Glib::ustring & device_path, std::vector & partitions, Sector start, Sector end, bool inside_extended ) { partition_temp .Reset() ; partition_temp .Set_Unallocated( device_path, 0, 0, inside_extended ) ; //if there are no partitions at all.. if ( partitions .empty() ) { partition_temp .sector_start = start ; partition_temp .sector_end = end ; partitions .push_back( partition_temp ); return ; } //start <---> first partition start if ( (partitions .front() .sector_start - start) >= MEBIBYTE ) { partition_temp .sector_start = start ; partition_temp .sector_end = partitions .front() .sector_start -1 ; partitions .insert( partitions .begin(), partition_temp ); } //look for gaps in between for ( unsigned int t =0 ; t < partitions .size() -1 ; t++ ) if ( ( partitions[ t +1 ] .sector_start - partitions[ t ] .sector_end ) >= MEBIBYTE ) { partition_temp .sector_start = partitions[ t ] .sector_end +1 ; partition_temp .sector_end = partitions[ t +1 ] .sector_start -1 ; partitions .insert( partitions .begin() + ++t, partition_temp ); } //last partition end <---> end if ( (end - partitions .back() .sector_end ) >= MEBIBYTE ) { partition_temp .sector_start = partitions .back() .sector_end +1 ; partition_temp .sector_end = end ; partitions .push_back( partition_temp ); } } void GParted_Core::set_mountpoints( std::vector & partitions ) { for ( unsigned int t = 0 ; t < partitions .size() ; t++ ) { if ( ( partitions[ t ] .type == GParted::TYPE_PRIMARY || partitions[ t ] .type == GParted::TYPE_LOGICAL ) && partitions[ t ] .filesystem != GParted::FS_LINUX_SWAP ) { if ( partitions[ t ] .busy ) { for ( unsigned int i = 0 ; i < partitions[ t ] .get_paths() .size() ; i++ ) { iter_mp = mount_info .find( partitions[ t ] .get_paths()[ i ] ) ; if ( iter_mp != mount_info .end() ) { partitions[ t ] .add_mountpoints( iter_mp ->second ) ; break ; } } if ( partitions[ t ] .get_mountpoints() .empty() ) partitions[ t ] .messages .push_back( _("Unable to find mountpoint") ) ; } else { iter_mp = fstab_info .find( partitions[ t ] .get_path() ); if ( iter_mp != fstab_info .end() ) partitions[ t ] .add_mountpoints( iter_mp ->second ) ; } } else if ( partitions[ t ] .type == GParted::TYPE_EXTENDED ) set_mountpoints( partitions[ t ] .logicals ) ; } } void GParted_Core::set_used_sectors( std::vector & partitions ) { struct statvfs sfs ; temp = _("Unable to read the contents of this filesystem!") ; temp += "\n" ; temp += _("Because of this some operations may be unavailable.") ; for ( unsigned int t = 0 ; t < partitions .size() ; t++ ) { if ( partitions[ t ] .filesystem != GParted::FS_LINUX_SWAP && partitions[ t ] .filesystem != GParted::FS_UNKNOWN ) { if ( partitions[ t ] .type == GParted::TYPE_PRIMARY || partitions[ t ] .type == GParted::TYPE_LOGICAL ) { if ( partitions[ t ] .busy ) { if ( partitions[ t ] .get_mountpoints() .size() > 0 ) { if ( statvfs( partitions[ t ] .get_mountpoint() .c_str(), &sfs ) == 0 ) partitions[ t ] .Set_Unused( sfs .f_bfree * (sfs .f_bsize / 512) ) ; else partitions[ t ] .messages .push_back( "statvfs (" + partitions[ t ] .get_mountpoint() + "): " + Glib::strerror( errno ) ) ; } } else { switch( get_fs( partitions[ t ] .filesystem ) .read ) { case GParted::FS::EXTERNAL : if ( set_proper_filesystem( partitions[ t ] .filesystem ) ) p_filesystem ->set_used_sectors( partitions[ t ] ) ; break ; case GParted::FS::LIBPARTED : LP_set_used_sectors( partitions[ t ] ) ; break ; default: break ; } } if ( partitions[ t ] .sectors_used == -1 ) partitions[ t ] .messages .push_back( temp ) ; } else if ( partitions[ t ] .type == GParted::TYPE_EXTENDED ) set_used_sectors( partitions[ t ] .logicals ) ; } } } void GParted_Core::LP_set_used_sectors( Partition & partition ) { PedFileSystem *fs = NULL; PedConstraint *constraint = NULL; if ( lp_disk ) { lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; if ( lp_partition ) { fs = ped_file_system_open( & lp_partition ->geom ); if ( fs ) { constraint = ped_file_system_get_resize_constraint( fs ) ; if ( constraint ) { partition .Set_Unused( partition .get_length() - constraint ->min_size ) ; ped_constraint_destroy( constraint ); } ped_file_system_close( fs ) ; } } } } void GParted_Core::set_flags( Partition & partition ) { for ( unsigned int t = 0 ; t < flags .size() ; t++ ) if ( ped_partition_is_flag_available( lp_partition, flags[ t ] ) && ped_partition_get_flag( lp_partition, flags[ t ] ) ) partition .flags .push_back( ped_partition_flag_get_name( flags[ t ] ) ) ; } bool GParted_Core::create( const Device & device, Partition & new_partition, OperationDetail & operationdetail ) { if ( new_partition .type == GParted::TYPE_EXTENDED ) { return create_partition( new_partition, operationdetail ) ; } else if ( create_partition( new_partition, operationdetail, get_fs( new_partition .filesystem ) .MIN ) ) { if ( new_partition .filesystem == GParted::FS_UNFORMATTED ) return true ; else return set_partition_type( new_partition, operationdetail ) && create_filesystem( new_partition, operationdetail ) ; } return false ; } bool GParted_Core::create_partition( Partition & new_partition, OperationDetail & operationdetail, Sector min_size ) { operationdetail .add_child( OperationDetail( _("create empty partition") ) ) ; new_partition .partition_number = 0 ; if ( open_device_and_disk( new_partition .device_path ) ) { PedPartitionType type; lp_partition = NULL ; PedConstraint *constraint = NULL ; PedFileSystemType* fs_type = NULL ; //create new partition switch ( new_partition .type ) { case GParted::TYPE_PRIMARY: type = PED_PARTITION_NORMAL ; break ; case GParted::TYPE_LOGICAL: type = PED_PARTITION_LOGICAL ; break ; case GParted::TYPE_EXTENDED: type = PED_PARTITION_EXTENDED ; break ; default : type = PED_PARTITION_FREESPACE; } if ( new_partition .type != GParted::TYPE_EXTENDED ) fs_type = ped_file_system_type_get( "ext2" ) ; lp_partition = ped_partition_new( lp_disk, type, fs_type, new_partition .sector_start, new_partition .sector_end ) ; if ( lp_partition ) { if ( new_partition .strict ) { PedGeometry *geom = ped_geometry_new( lp_device, new_partition .sector_start, new_partition .get_length() ) ; if ( geom ) constraint = ped_constraint_exact( geom ) ; } else constraint = ped_constraint_any( lp_device ); if ( constraint ) { if ( min_size > 0 ) constraint ->min_size = min_size ; if ( ped_disk_add_partition( lp_disk, lp_partition, constraint ) && commit() ) { //we have to free the result of ped_partition_get_path().. char * lp_path = ped_partition_get_path( lp_partition ) ; new_partition .add_path( lp_path, true ) ; free( lp_path ) ; new_partition .partition_number = lp_partition ->num ; new_partition .sector_start = lp_partition ->geom .start ; new_partition .sector_end = lp_partition ->geom .end ; operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("path: %1"), new_partition .get_path() ) + "\n" + String::ucompose( _("start: %1"), new_partition .sector_start ) + "\n" + String::ucompose( _("end: %1"), new_partition .sector_end ) + "\n" + String::ucompose( _("size: %1 (%2)"), new_partition .get_length(), Utils::format_size( new_partition .get_length() ) ), STATUS_NONE, FONT_ITALIC ) ) ; } ped_constraint_destroy( constraint ); } } close_device_and_disk() ; } if ( new_partition .partition_number > 0 && erase_filesystem_signatures( new_partition ) ) { operationdetail .get_last_child() .set_status( STATUS_SUCCES ) ; return true ; } else { operationdetail .get_last_child() .set_status( STATUS_ERROR ) ; return false ; } } bool GParted_Core::create_filesystem( const Partition & partition, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( String::ucompose( _("create new %1 filesystem"), Utils::get_filesystem_string( partition .filesystem ) ) ) ) ; bool succes = false ; switch ( get_fs( partition .filesystem ) .create ) { case GParted::FS::NONE: break ; case GParted::FS::GPARTED: break ; case GParted::FS::LIBPARTED: break ; case GParted::FS::EXTERNAL: succes = set_proper_filesystem( partition .filesystem ) && p_filesystem ->create( partition, operationdetail .get_last_child() ) ; break ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::format( const Partition & partition, OperationDetail & operationdetail ) { //remove all filesystem signatures... erase_filesystem_signatures( partition ) ; return set_partition_type( partition, operationdetail ) && create_filesystem( partition, operationdetail ) ; } bool GParted_Core::Delete( const Partition & partition, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( _("delete partition") ) ) ; bool succes = false ; if ( open_device_and_disk( partition .device_path ) ) { if ( partition .type == TYPE_EXTENDED ) lp_partition = ped_disk_extended_partition( lp_disk ) ; else lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; succes = ped_disk_delete_partition( lp_disk, lp_partition ) && commit() ; close_device_and_disk() ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::label_partition( const Partition & partition, OperationDetail & operationdetail ) { if( partition .label .empty() ) { operationdetail .add_child( OperationDetail( String::ucompose( _("Clear partition label on %1"), partition .get_path() ) ) ) ; } else { operationdetail .add_child( OperationDetail( String::ucompose( _("Set partition label to \"%1\" on %2"), partition .label, partition .get_path() ) ) ) ; } bool succes = false ; if ( partition .type != TYPE_EXTENDED ) { switch( get_fs( partition .filesystem ) .set_label ) { case FS::EXTERNAL: succes = set_proper_filesystem( partition .filesystem ) && p_filesystem ->set_label( partition, operationdetail .get_last_child() ) ; break ; case FS::LIBPARTED: break ; default: break ; } } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::resize_move( const Device & device, const Partition & partition_old, Partition & partition_new, OperationDetail & operationdetail ) { if ( partition_new .strict || calculate_exact_geom( partition_old, partition_new, operationdetail ) ) { if ( partition_old .type == TYPE_EXTENDED ) return resize_move_partition( partition_old, partition_new, operationdetail ) ; if ( partition_new .sector_start == partition_old .sector_start ) return resize( partition_old, partition_new, operationdetail ) ; if ( partition_new .get_length() == partition_old .get_length() ) return move( device, partition_old, partition_new, operationdetail ) ; Partition temp ; if ( partition_new .get_length() > partition_old .get_length() ) { //first move, then grow. Since old.length < new.length and new.start is valid, temp is valid. temp = partition_new ; temp .sector_end = temp .sector_start + partition_old .get_length() -1 ; } if ( partition_new .get_length() < partition_old .get_length() ) { //first shrink, then move. Since new.length < old.length and old.start is valid, temp is valid. temp = partition_old ; temp .sector_end = partition_old .sector_start + partition_new .get_length() -1 ; } temp .strict = true ; bool succes = resize_move( device, partition_old, temp, operationdetail ) ; temp .strict = false ; return succes && resize_move( device, temp, partition_new, operationdetail ) ; } return false ; } bool GParted_Core::move( const Device & device, const Partition & partition_old, const Partition & partition_new, OperationDetail & operationdetail ) { if ( partition_old .get_length() != partition_new .get_length() ) { operationdetail .add_child( OperationDetail( _("moving requires old and new length to be the same"), STATUS_ERROR, FONT_ITALIC ) ) ; return false ; } bool succes = false ; if ( check_repair_filesystem( partition_old, operationdetail ) ) { //NOTE: logical partitions are preceeded by metadata. To prevent this metadata from being //overwritten we move the partition first and only then the filesystem when moving to the left. //(maybe i should do some reading on how non-msdos disklabels deal with metadata....) if ( partition_new .sector_start < partition_old .sector_start ) { if ( resize_move_partition( partition_old, partition_new, operationdetail ) ) { if ( ! move_filesystem( partition_old, partition_new, operationdetail ) ) { operationdetail .add_child( OperationDetail( _("rollback last change to the partitiontable") ) ) ; if ( resize_move_partition( partition_new, partition_old, operationdetail .get_last_child() ) ) operationdetail .get_last_child() .set_status( STATUS_SUCCES ) ; else operationdetail .get_last_child() .set_status( STATUS_ERROR ) ; } else succes = true ; } } else succes = move_filesystem( partition_old, partition_new, operationdetail ) && resize_move_partition( partition_old, partition_new, operationdetail ) ; succes = succes && update_bootsector( partition_new, operationdetail ) && check_repair_filesystem( partition_new, operationdetail ) && maximize_filesystem( partition_new, operationdetail ) ; } return succes ; } bool GParted_Core::move_filesystem( const Partition & partition_old, const Partition & partition_new, OperationDetail & operationdetail ) { if ( partition_new .sector_start < partition_old .sector_start ) operationdetail .add_child( OperationDetail( _("move filesystem to the left") ) ) ; else if ( partition_new .sector_start > partition_old .sector_start ) operationdetail .add_child( OperationDetail( _("move filesystem to the right") ) ) ; else { operationdetail .add_child( OperationDetail( _("move filesystem") ) ) ; operationdetail .get_last_child() .add_child( OperationDetail( _("new and old filesystem have the same positition. skipping this operation"), STATUS_NONE, FONT_ITALIC ) ) ; operationdetail .get_last_child() .set_status( STATUS_SUCCES ) ; return true ; } bool succes = false ; switch ( get_fs( partition_old .filesystem ) .move ) { case GParted::FS::NONE: break ; case GParted::FS::GPARTED: succes = false ; if ( partition_new .test_overlap( partition_old ) ) { if ( copy_filesystem_simulation( partition_old, partition_new, operationdetail .get_last_child() ) ) { operationdetail .get_last_child() .add_child( OperationDetail( _("perform real move") ) ) ; Sector total_done ; succes = copy_filesystem( partition_old, partition_new, operationdetail .get_last_child() .get_last_child(), total_done ) ; operationdetail .get_last_child() .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; if ( ! succes ) { rollback_transaction( partition_old, partition_new, operationdetail .get_last_child(), total_done ) ; check_repair_filesystem( partition_old, operationdetail ) ; } } } else succes = copy_filesystem( partition_old, partition_new, operationdetail .get_last_child() ) ; break ; case GParted::FS::LIBPARTED: succes = resize_move_filesystem_using_libparted( partition_old, partition_new, operationdetail .get_last_child() ) ; break ; case GParted::FS::EXTERNAL: break ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::resize_move_filesystem_using_libparted( const Partition & partition_old, const Partition & partition_new, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( _("using libparted"), STATUS_NONE ) ) ; bool return_value = false ; if ( open_device_and_disk( partition_old .device_path ) ) { PedFileSystem * fs = NULL ; PedGeometry * lp_geom = NULL ; lp_geom = ped_geometry_new( lp_device, partition_old .sector_start, partition_old .get_length() ) ; if ( lp_geom ) { fs = ped_file_system_open( lp_geom ); if ( fs ) { lp_geom = NULL ; lp_geom = ped_geometry_new( lp_device, partition_new .sector_start, partition_new .get_length() ) ; if ( lp_geom ) return_value = ped_file_system_resize( fs, lp_geom, NULL ) && commit() ; ped_file_system_close( fs ); } } close_device_and_disk() ; } return return_value ; } bool GParted_Core::resize( const Partition & partition_old, const Partition & partition_new, OperationDetail & operationdetail ) { if ( partition_old .sector_start != partition_new .sector_start ) { operationdetail .add_child( OperationDetail( _("resizing requires old and new start to be the same"), STATUS_ERROR, FONT_ITALIC ) ) ; return false ; } bool succes = false ; if ( check_repair_filesystem( partition_new, operationdetail ) ) { succes = true ; if ( succes && partition_new .get_length() < partition_old .get_length() ) succes = resize_filesystem( partition_old, partition_new, operationdetail ) ; if ( succes ) succes = resize_move_partition( partition_old, partition_new, operationdetail ) ; //these 2 are always executed, however, if 1 of them fails the whole operation fails if ( ! check_repair_filesystem( partition_new, operationdetail ) ) succes = false ; //expand filesystem to fit exactly in partition if ( ! maximize_filesystem( partition_new, operationdetail ) ) succes = false ; return succes ; } return false ; } bool GParted_Core::resize_move_partition( const Partition & partition_old, const Partition & partition_new, OperationDetail & operationdetail ) { //i'm not too happy with this, but i think it is the correct way from a i18n POV enum Action { NONE = 0, MOVE_RIGHT = 1, MOVE_LEFT = 2, GROW = 3, SHRINK = 4, MOVE_RIGHT_GROW = 5, MOVE_RIGHT_SHRINK = 6, MOVE_LEFT_GROW = 7, MOVE_LEFT_SHRINK = 8 } ; Action action = NONE ; if ( partition_new .get_length() > partition_old .get_length() ) action = GROW ; else if ( partition_new .get_length() < partition_old .get_length() ) action = SHRINK ; if ( partition_new .sector_start > partition_old .sector_start && partition_new .sector_end > partition_old .sector_end ) action = action == GROW ? MOVE_RIGHT_GROW : action == SHRINK ? MOVE_RIGHT_SHRINK : MOVE_RIGHT ; else if ( partition_new .sector_start < partition_old .sector_start && partition_new .sector_end < partition_old .sector_end ) action = action == GROW ? MOVE_LEFT_GROW : action == SHRINK ? MOVE_LEFT_SHRINK : MOVE_LEFT ; Glib::ustring description ; switch ( action ) { case NONE : description = _("resize/move partition") ; break ; case MOVE_RIGHT : description = _("move partition to the right") ; break ; case MOVE_LEFT : description = _("move partition to the left") ; break ; case GROW : description = _("grow partition from %1 to %2") ; break ; case SHRINK : description = _("shrink partition from %1 to %2") ; break ; case MOVE_RIGHT_GROW : description = _("move partition to the right and grow it from %1 to %2") ; break ; case MOVE_RIGHT_SHRINK : description = _("move partition to the right and shrink it from %1 to %2") ; break ; case MOVE_LEFT_GROW : description = _("move partition to the left and grow it from %1 to %2") ; break ; case MOVE_LEFT_SHRINK : description = _("move partition to the left and shrink it from %1 to %2") ; break ; } if ( ! description .empty() && action != NONE && action != MOVE_LEFT && action != MOVE_RIGHT ) description = String::ucompose( description, Utils::format_size( partition_old .get_length() ), Utils::format_size( partition_new .get_length() ) ) ; operationdetail .add_child( OperationDetail( description ) ) ; if ( action == NONE ) { operationdetail .get_last_child() .add_child( OperationDetail( _("new and old partition have the same size and position. skipping this operation"), STATUS_NONE, FONT_ITALIC ) ) ; operationdetail .get_last_child() .set_status( STATUS_SUCCES ) ; return true ; } operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("old start: %1"), partition_old .sector_start ) + "\n" + String::ucompose( _("old end: %1"), partition_old .sector_end ) + "\n" + String::ucompose( _("old size: %1 (%2)"), partition_old .get_length(), Utils::format_size( partition_old .get_length() ) ), STATUS_NONE, FONT_ITALIC ) ) ; //finally the actual resize/move bool return_value = false ; PedConstraint *constraint = NULL ; lp_partition = NULL ; //sometimes the lp_partition ->geom .start,end and length values display random numbers //after going out of the 'if ( lp_partition)' scope. That's why we use some variables here. Sector new_start = -1, new_end = -1 ; if ( open_device_and_disk( partition_old .device_path ) ) { if ( partition_old .type == GParted::TYPE_EXTENDED ) lp_partition = ped_disk_extended_partition( lp_disk ) ; else lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition_old .get_sector() ) ; if ( lp_partition ) { PedGeometry *geom = ped_geometry_new( lp_device, partition_new .sector_start, partition_new .get_length() ) ; constraint = ped_constraint_exact( geom ) ; if ( constraint ) { if ( ped_disk_set_partition_geom( lp_disk, lp_partition, constraint, partition_new .sector_start, partition_new .sector_end ) ) { new_start = lp_partition ->geom .start ; new_end = lp_partition ->geom .end ; return_value = commit() ; } ped_constraint_destroy( constraint ); } } close_device_and_disk() ; } if ( return_value ) { operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("new start: %1"), new_start ) + "\n" + String::ucompose( _("new end: %1"), new_end ) + "\n" + String::ucompose( _("new size: %1 (%2)"), new_end - new_start + 1, Utils::format_size( new_end - new_start + 1 ) ), STATUS_NONE, FONT_ITALIC ) ) ; } operationdetail .get_last_child() .set_status( return_value ? STATUS_SUCCES : STATUS_ERROR ) ; return return_value ; } bool GParted_Core::resize_filesystem( const Partition & partition_old, const Partition & partition_new, OperationDetail & operationdetail, bool fill_partition ) { //by default 'grow' to accomodate expand_filesystem() GParted::FS::Support action = get_fs( partition_old .filesystem ) .grow ; if ( ! fill_partition ) { if ( partition_new .get_length() < partition_old .get_length() ) { operationdetail .add_child( OperationDetail( _("shrink filesystem") ) ) ; action = get_fs( partition_old .filesystem ) .shrink ; } else if ( partition_new .get_length() > partition_old .get_length() ) operationdetail .add_child( OperationDetail( _("grow filesystem") ) ) ; else { operationdetail .add_child( OperationDetail( _("resize filesystem") ) ) ; operationdetail .get_last_child() .add_child( OperationDetail( _("new and old filesystem have the same size. skipping this operation"), STATUS_NONE, FONT_ITALIC ) ) ; operationdetail .get_last_child() .set_status( STATUS_SUCCES ) ; return true ; } } bool succes = false ; switch ( action ) { case GParted::FS::NONE: break ; case GParted::FS::GPARTED: break ; case GParted::FS::LIBPARTED: succes = resize_move_filesystem_using_libparted( partition_old, partition_new, operationdetail .get_last_child() ) ; break ; case GParted::FS::EXTERNAL: succes = set_proper_filesystem( partition_new .filesystem ) && p_filesystem ->resize( partition_new, operationdetail .get_last_child(), fill_partition ) ; break ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::maximize_filesystem( const Partition & partition, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( _("grow filesystem to fill the partition") ) ) ; if ( get_fs( partition .filesystem ) .grow == GParted::FS::NONE ) { operationdetail .get_last_child() .add_child( OperationDetail( _("growing is not available for this filesystem"), STATUS_NONE, FONT_ITALIC ) ) ; operationdetail .get_last_child() .set_status( STATUS_N_A ) ; return true ; } return resize_filesystem( partition, partition, operationdetail, true ) ; } bool GParted_Core::copy( const Partition & partition_src, Partition & partition_dst, Sector min_size, OperationDetail & operationdetail ) { if ( partition_dst .get_length() < partition_src .get_length() ) { operationdetail .add_child( OperationDetail( _("the destination is smaller than the sourcepartition"), STATUS_ERROR, FONT_ITALIC ) ) ; return false ; } if ( check_repair_filesystem( partition_src, operationdetail ) ) { bool succes = true ; if ( partition_dst .status == GParted::STAT_COPY ) succes = create_partition( partition_dst, operationdetail, min_size ) ; if ( succes && set_partition_type( partition_dst, operationdetail ) ) { operationdetail .add_child( OperationDetail( String::ucompose( _("copy filesystem of %1 to %2"), partition_src .get_path(), partition_dst .get_path() ) ) ) ; switch ( get_fs( partition_dst .filesystem ) .copy ) { case GParted::FS::GPARTED : succes = copy_filesystem( partition_src, partition_dst, operationdetail .get_last_child() ) ; break ; case GParted::FS::LIBPARTED : //FIXME: see if copying through libparted has any advantages break ; case GParted::FS::EXTERNAL : succes = set_proper_filesystem( partition_dst .filesystem ) && p_filesystem ->copy( partition_src .get_path(), partition_dst .get_path(), operationdetail .get_last_child() ) ; break ; default : succes = false ; break ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes && update_bootsector( partition_dst, operationdetail ) && check_repair_filesystem( partition_dst, operationdetail ) && maximize_filesystem( partition_dst, operationdetail ) ; } } return false ; } bool GParted_Core::copy_filesystem_simulation( const Partition & partition_src, const Partition & partition_dst, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( _("perform readonly test") ) ) ; bool succes = copy_filesystem( partition_src, partition_dst, operationdetail .get_last_child(), true ) ; operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::copy_filesystem( const Partition & partition_src, const Partition & partition_dst, OperationDetail & operationdetail, bool readonly ) { Sector dummy ; return copy_filesystem( partition_src .device_path, partition_dst .device_path, partition_src .sector_start, partition_dst .sector_start, partition_src .get_length(), operationdetail, readonly, dummy ) ; } bool GParted_Core::copy_filesystem( const Partition & partition_src, const Partition & partition_dst, OperationDetail & operationdetail, Sector & total_done ) { return copy_filesystem( partition_src .device_path, partition_dst .device_path, partition_src .sector_start, partition_dst .sector_start, partition_src .get_length(), operationdetail, false, total_done ) ; } bool GParted_Core::copy_filesystem( const Glib::ustring & src_device, const Glib::ustring & dst_device, Sector src_start, Sector dst_start, Sector length, OperationDetail & operationdetail, bool readonly, Sector & total_done ) { operationdetail .add_child( OperationDetail( _("using internal algorithm"), STATUS_NONE ) ) ; operationdetail .add_child( OperationDetail( String::ucompose( readonly ? _("read %1 sectors") : _("copy %1 sectors"), length ), STATUS_NONE ) ) ; operationdetail .add_child( OperationDetail( _("finding optimal blocksize"), STATUS_NONE ) ) ; Sector optimal_blocksize = readonly ? 128 : 64, N = 32768 ; Sector offset_read = src_start, offset_write = dst_start ; if ( dst_start > src_start ) { offset_read += (length -N) ; offset_write += (length -N) ; } total_done = 0 ; Sector done = 0 ; Glib::Timer timer ; double smallest_time = 1000000 ; bool succes = true ; while ( succes && timer .elapsed() <= smallest_time && std::llabs( done ) + N <= length && optimal_blocksize * 2 < N ) { if ( done != 0 ) { smallest_time = timer .elapsed() ; optimal_blocksize *= 2 ; } timer .reset() ; succes = copy_blocks( src_device, dst_device, offset_read + done, offset_write + done, N, optimal_blocksize, operationdetail .get_last_child(), readonly, total_done ) ; timer.stop() ; operationdetail .get_last_child() .get_last_child() .add_child( OperationDetail( String::ucompose( _("%1 seconds"), timer .elapsed() ), STATUS_NONE, FONT_ITALIC ) ) ; if ( ( dst_start > src_start ) ) done -= N ; else done += N ; } if ( timer .elapsed() > smallest_time ) optimal_blocksize /= 2 ; if ( succes ) operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("optimal blocksize is %1 sectors (%2)"), optimal_blocksize, Utils::format_size( optimal_blocksize ) ), STATUS_NONE ) ) ; if ( succes ) succes = copy_blocks( src_device, dst_device, src_start + ( dst_start > src_start ? 0 : done ), dst_start + ( dst_start > src_start ? 0 : done ), length - std::llabs( done ), optimal_blocksize, operationdetail, readonly, total_done ) ; operationdetail .add_child( OperationDetail( String::ucompose( readonly ? _("%1 sectors read") : _("%1 sectors copied"), total_done ), STATUS_NONE ) ) ; return succes ; } void GParted_Core::rollback_transaction( const Partition & partition_src, const Partition & partition_dst, OperationDetail & operationdetail, Sector total_done ) { if ( total_done > 0 ) { operationdetail .add_child( OperationDetail( _("rollback last transaction") ) ) ; //find out exactly which part of the filesystem was copied (and to where it was copied).. Partition temp_src = partition_src ; Partition temp_dst = partition_dst ; if ( partition_dst .sector_start > partition_src .sector_start ) { temp_src .sector_start = temp_src .sector_end - (total_done-1) ; temp_dst .sector_start = temp_dst .sector_end - (total_done-1) ; } else { temp_src .sector_end = temp_src .sector_start + (total_done -1) ; temp_dst .sector_end = temp_dst .sector_start + (total_done -1) ; } //and copy it back (NOTE the reversed dst and src) bool succes = copy_filesystem( temp_dst, temp_src, operationdetail .get_last_child() ) ; operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; } } bool GParted_Core::check_repair_filesystem( const Partition & partition, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( String::ucompose( _("check filesystem on %1 for errors and (if possible) fix them"), partition .get_path() ) ) ) ; bool succes = false ; switch ( get_fs( partition .filesystem ) .check ) { case GParted::FS::NONE: operationdetail .get_last_child() .add_child( OperationDetail( _("checking is not available for this filesystem"), STATUS_NONE, FONT_ITALIC ) ) ; operationdetail .get_last_child() .set_status( STATUS_N_A ) ; return true ; break ; case GParted::FS::GPARTED: break ; case GParted::FS::LIBPARTED: break ; case GParted::FS::EXTERNAL: succes = set_proper_filesystem( partition .filesystem ) && p_filesystem ->check_repair( partition, operationdetail .get_last_child() ) ; break ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::set_partition_type( const Partition & partition, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( String::ucompose( _("set partitiontype on %1"), partition .get_path() ) ) ) ; bool return_value = false ; if ( open_device_and_disk( partition .device_path ) ) { PedFileSystemType * fs_type = ped_file_system_type_get( Utils::get_filesystem_string( partition .filesystem ) .c_str() ) ; //default is Linux (83) if ( ! fs_type ) fs_type = ped_file_system_type_get( "ext2" ) ; if ( fs_type ) { lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; if ( lp_partition && ped_partition_set_system( lp_partition, fs_type ) && commit() ) { operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("new partitiontype: %1"), lp_partition ->fs_type ->name ), STATUS_NONE, FONT_ITALIC ) ) ; return_value = true ; } } close_device_and_disk() ; } operationdetail .get_last_child() .set_status( return_value ? STATUS_SUCCES : STATUS_ERROR ) ; return return_value ; } void GParted_Core::set_progress_info( Sector total, Sector done, const Glib::Timer & timer, OperationDetail & operationdetail, bool readonly ) { operationdetail .fraction = done / static_cast( total ) ; std::time_t time_remaining = Utils::round( (total - done) / ( done / timer .elapsed() ) ) ; operationdetail .progress_text = String::ucompose( readonly ? _("%1 of %2 read (%3 remaining)") : _("%1 of %2 copied (%3 remaining)"), Utils::format_size( done ), Utils::format_size( total ), Utils::format_time( time_remaining) ) ; operationdetail .set_description( String::ucompose( readonly ? _("%1 of %2 read") : _("%1 of %2 copied"), done, total ), FONT_ITALIC ) ; } bool GParted_Core::copy_blocks( const Glib::ustring & src_device, const Glib::ustring & dst_device, Sector src_start, Sector dst_start, Sector length, Sector blocksize, OperationDetail & operationdetail, bool readonly, Sector & total_done ) { if ( blocksize > length ) blocksize = length ; if ( readonly ) operationdetail .add_child( OperationDetail( String::ucompose( _("read %1 sectors using a blocksize of %2 sectors"), length, blocksize ) ) ) ; else operationdetail .add_child( OperationDetail( String::ucompose( _("copy %1 sectors using a blocksize of %2 sectors"), length, blocksize ) ) ) ; Sector done = length % blocksize ; if ( dst_start > src_start ) { blocksize -= 2*blocksize ; done -= 2*done ; src_start += (length -1) ; dst_start += (length -1) ; } bool succes = false ; PedDevice *lp_device_src = ped_device_get( src_device .c_str() ); PedDevice *lp_device_dst = src_device != dst_device ? ped_device_get( dst_device .c_str() ) : lp_device_src ; if ( lp_device_src && lp_device_dst && ped_device_open( lp_device_src ) && ped_device_open( lp_device_dst ) ) { Glib::ustring error_message ; buf = static_cast( malloc( std::llabs( blocksize ) * 512 ) ) ; if ( buf ) { ped_device_sync( lp_device_dst ) ; succes = true ; if ( done != 0 ) succes = copy_block( lp_device_src, lp_device_dst, src_start, dst_start, done, error_message, readonly ) ; if ( ! succes ) done = 0 ; //add an empty sub which we will constantly update in the loop operationdetail .get_last_child() .add_child( OperationDetail( "", STATUS_NONE ) ) ; Glib::Timer timer_progress_timeout, timer_total ; while( succes && std::llabs( done ) < length ) { succes = copy_block( lp_device_src, lp_device_dst, src_start +done, dst_start +done, blocksize, error_message, readonly ) ; if ( succes ) done += blocksize ; if ( timer_progress_timeout .elapsed() >= 0.5 ) { set_progress_info( length, std::llabs( done + blocksize ), timer_total, operationdetail .get_last_child() .get_last_child(), readonly ) ; timer_progress_timeout .reset() ; } } free( buf ) ; } else error_message = Glib::strerror( errno ) ; //reset fraction to -1 to make room for a new one (or a pulsebar) operationdetail .get_last_child() .get_last_child() .fraction = -1 ; //final description operationdetail .get_last_child() .get_last_child() .set_description( String::ucompose( readonly ? _("%1 of %2 read") : _("%1 of %2 copied"), std::llabs( done ), length ), FONT_ITALIC ) ; if ( ! succes && ! error_message .empty() ) operationdetail .get_last_child() .add_child( OperationDetail( error_message, STATUS_NONE, FONT_ITALIC ) ) ; total_done += std::llabs( done ) ; //close and destroy the devices.. ped_device_close( lp_device_src ) ; ped_device_destroy( lp_device_src ) ; if ( src_device != dst_device ) { ped_device_close( lp_device_dst ) ; ped_device_destroy( lp_device_dst ) ; } } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::copy_block( PedDevice * lp_device_src, PedDevice * lp_device_dst, Sector offset_src, Sector offset_dst, Sector blocksize, Glib::ustring & error_message, bool readonly ) { if ( blocksize < 0 ) { blocksize = std::llabs( blocksize ) ; offset_src -= ( blocksize -1 ) ; offset_dst -= ( blocksize -1 ) ; } if ( blocksize != 0 ) { if ( ped_device_read( lp_device_src, buf, offset_src, blocksize ) ) { if ( readonly || ped_device_write( lp_device_dst, buf, offset_dst, blocksize ) ) return true ; else error_message = String::ucompose( _("Error while writing block at sector %1"), offset_dst ) ; } else error_message = String::ucompose( _("Error while reading block at sector %1"), offset_src ) ; } return false ; } bool GParted_Core::calibrate_partition( Partition & partition, OperationDetail & operationdetail ) { if ( partition .type == TYPE_PRIMARY || partition .type == TYPE_LOGICAL || partition .type == TYPE_EXTENDED ) { operationdetail .add_child( OperationDetail( String::ucompose( _("calibrate %1"), partition .get_path() ) ) ) ; bool succes = false ; if ( open_device_and_disk( partition .device_path ) ) { if ( partition .type == GParted::TYPE_EXTENDED ) lp_partition = ped_disk_extended_partition( lp_disk ) ; else lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; if ( lp_partition )//FIXME: add check to see if lp_partition ->type matches partition .type.. { char * lp_path = ped_partition_get_path( lp_partition ) ; partition .add_path( lp_path, true ) ; free( lp_path ) ; partition .sector_start = lp_partition ->geom .start ; partition .sector_end = lp_partition ->geom .end ; operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("path: %1"), partition .get_path() ) + "\n" + String::ucompose( _("start: %1"), partition .sector_start ) + "\n" + String::ucompose( _("end: %1"), partition .sector_end ) + "\n" + String::ucompose( _("size: %1 (%2)"), partition .get_length(), Utils::format_size( partition .get_length() ) ), STATUS_NONE, FONT_ITALIC ) ) ; succes = true ; } close_device_and_disk() ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } else //nothing to calibrate... return true ; } bool GParted_Core::calculate_exact_geom( const Partition & partition_old, Partition & partition_new, OperationDetail & operationdetail ) { operationdetail .add_child( OperationDetail( String::ucompose( _("calculate new size and position of %1"), partition_new .get_path() ) ) ) ; operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("requested start: %1"), partition_new .sector_start ) + "\n" + String::ucompose( _("requested end: %1"), partition_new .sector_end ) + "\n" + String::ucompose( _("requested size: %1 (%2)"), partition_new .get_length(), Utils::format_size( partition_new .get_length() ) ), STATUS_NONE, FONT_ITALIC ) ) ; bool succes = false ; if ( open_device_and_disk( partition_old .device_path ) ) { lp_partition = NULL ; if ( partition_old .type == GParted::TYPE_EXTENDED ) lp_partition = ped_disk_extended_partition( lp_disk ) ; else lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition_old .get_sector() ) ; if ( lp_partition ) { PedConstraint *constraint = NULL ; constraint = ped_constraint_any( lp_device ) ; if ( constraint ) { //FIXME: if we insert a weird partitionnew geom here (e.g. start > end) //ped_disk_set_partition_geom() will still return true (althoug an lp exception is written //to stdout.. see if this also affect create_partition and resize_move_partition //sended a patch to fix this to libparted list. will probably be in 1.7.2 if ( ped_disk_set_partition_geom( lp_disk, lp_partition, constraint, partition_new .sector_start, partition_new .sector_end ) ) { partition_new .sector_start = lp_partition ->geom .start ; partition_new .sector_end = lp_partition ->geom .end ; succes = true ; } ped_constraint_destroy( constraint ); } } close_device_and_disk() ; } if ( succes ) { operationdetail .get_last_child() .add_child( OperationDetail( String::ucompose( _("new start: %1"), partition_new .sector_start ) + "\n" + String::ucompose( _("new end: %1"), partition_new .sector_end ) + "\n" + String::ucompose( _("new size: %1 (%2)"), partition_new .get_length(), Utils::format_size( partition_new .get_length() ) ), STATUS_NONE, FONT_ITALIC ) ) ; } operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } bool GParted_Core::set_proper_filesystem( const FILESYSTEM & filesystem ) { if ( p_filesystem ) delete p_filesystem ; switch( filesystem ) { case FS_EXT2 : p_filesystem = new ext2() ; break ; case FS_EXT3 : p_filesystem = new ext3() ; break ; case FS_LINUX_SWAP : p_filesystem = new linux_swap() ; break ; case FS_FAT16 : p_filesystem = new fat16() ; break ; case FS_FAT32 : p_filesystem = new fat32() ; break ; case FS_NTFS : p_filesystem = new ntfs() ; break ; case FS_REISERFS : p_filesystem = new reiserfs() ; break ; case FS_REISER4 : p_filesystem = new reiser4() ; break ; case FS_XFS : p_filesystem = new xfs() ; break ; case FS_JFS : p_filesystem = new jfs() ; break ; case FS_HFS : p_filesystem = new hfs() ; break ; case FS_HFSPLUS : p_filesystem = new hfsplus() ; break ; case FS_UFS : p_filesystem = new ufs() ; break ; default : p_filesystem = NULL ; } return p_filesystem ; } bool GParted_Core::erase_filesystem_signatures( const Partition & partition ) { bool return_value = false ; if ( open_device_and_disk( partition .device_path ) ) { lp_partition = ped_disk_get_partition_by_sector( lp_disk, partition .get_sector() ) ; if ( lp_partition && ped_file_system_clobber( & lp_partition ->geom ) ) { //filesystems not yet supported by libparted if ( ped_device_open( lp_device ) ) { //reiser4 stores "ReIsEr4" at sector 128 return_value = ped_geometry_write( & lp_partition ->geom, "0000000", 128, 1 ) ; ped_device_close( lp_device ) ; } } close_device_and_disk() ; } return return_value ; } bool GParted_Core::update_bootsector( const Partition & partition, OperationDetail & operationdetail ) { //only for ntfs atm... //FIXME: 1) this should be done without relying on external commands // 2) this should probably be done in the fsclasses... if ( partition .filesystem == FS_NTFS ) { operationdetail .add_child( OperationDetail( String::ucompose( _("updating bootsector of %1 filesystem on %2"), Utils::get_filesystem_string( partition .filesystem ), partition .get_path() ) ) ) ; std::stringstream ss ; ss << std::hex << partition .sector_start ; Glib::ustring hex = ss .str() ; //fill with zeros and reverse... hex .insert( 0, 8 - hex .length(), '0' ) ; Glib::ustring reversed_hex ; for ( int t = 6 ; t >= 0 ; t -=2 ) reversed_hex .append( hex .substr( t, 2 ) ) ; Glib::ustring output, error, command ; command = "echo " + reversed_hex + " | /usr/bin/xxd -r -p | /bin/dd conv=notrunc of=" + partition .get_path() + " bs=1 seek=28" ; operationdetail .get_last_child() .add_child( OperationDetail( command, STATUS_NONE, FONT_BOLD_ITALIC ) ) ; bool succes = ! Utils::execute_command( command, output, error ) ; if ( ! output .empty() ) operationdetail .get_last_child() .get_last_child() .add_child( OperationDetail( output, STATUS_NONE, FONT_ITALIC ) ) ; if ( ! error .empty() ) operationdetail .get_last_child() .get_last_child() .add_child( OperationDetail( error, STATUS_NONE, FONT_ITALIC ) ) ; operationdetail .get_last_child() .set_status( succes ? STATUS_SUCCES : STATUS_ERROR ) ; return succes ; } return true ; } bool GParted_Core::open_device( const Glib::ustring & device_path ) { lp_device = ped_device_get( device_path .c_str() ); return lp_device ; } bool GParted_Core::open_device_and_disk( const Glib::ustring & device_path, bool strict ) { lp_device = NULL ; lp_disk = NULL ; if ( open_device( device_path ) ) { lp_disk = ped_disk_new( lp_device ); //if ! disk and writeable it's probably a HD without disklabel. //We return true here and deal with them in GParted_Core::get_devices if ( lp_disk || ( ! strict && ! lp_device ->read_only ) ) return true ; close_device_and_disk() ; } return false ; } void GParted_Core::close_disk() { if ( lp_disk ) ped_disk_destroy( lp_disk ) ; lp_disk = NULL ; } void GParted_Core::close_device_and_disk() { close_disk() ; if ( lp_device ) ped_device_destroy( lp_device ) ; lp_device = NULL ; } bool GParted_Core::commit() { bool succes = ped_disk_commit_to_dev( lp_disk ) ; commit_to_os( 10 ) ; return succes ; } bool GParted_Core::commit_to_os( std::time_t timeout ) { bool succes = ped_disk_commit_to_os( lp_disk ) ; if ( Glib::find_program_in_path( "udevsettle" ) .empty() ) sleep( timeout ) ; else Utils::execute_command( "udevsettle --timeout=" + Utils::num_to_str( timeout ) ) ; return succes ; } PedExceptionOption GParted_Core::ped_exception_handler( PedException * e ) { std::cout << e ->message << std::endl ; libparted_messages .push_back( e->message ) ; return PED_EXCEPTION_UNHANDLED ; } GParted_Core::~GParted_Core() { if ( p_filesystem ) delete p_filesystem ; } } //GParted