source: MondoRescue/branches/3.2/MondoRescue/lib/MondoRescue/Disk.pm@ 3553

#!/usr/bin/perl -w
#
# Disk subroutines brought by the MondoRescue project
#
# $Id$
#
# Copyright B. Cornec 2008-2015
# Provided under the GPL v2

package MondoRescue::Disk;

use strict 'vars';
use ProjectBuilder::Base;
use English;
use MondoRescue::File;

# Inherit from the "Exporter" module which handles exporting functions.

use Exporter;

# Export, by default, all the functions into the namespace of
# any code which uses this module.

our @ISA = qw(Exporter);
our @EXPORT = qw(mr_disk_list mr_disk_type mr_device_mounted);

=pod

=head1 NAME

MondoRescue::Disk, part of the mondorescue.org

=head1 DESCRIPTION

This modules provides disk related functions for the Mondorescue project

=head1 USAGE

=over 4

=item B<mr_disk_list>

This function uses fdisk to determine the list of all the disks on the system
It returns an array of strings giving all the devices

=cut

sub mr_disk_list {

my $fdisk = pb_check_req("fdisk",0);
my @disks;

open(FDISK, "LANG=C $fdisk -l 2>/dev/null |") || die "Unable to read from $fdisk";
while (<FDISK>) {
    chomp($_);
    my $i = $_;
    if ($i =~ /^Disk \//) {
        pb_log(2,"Found disk line: $i\n");
        $i =~ s|^Disk /([A-z0-9/_-]+):.*|/$1|;
        pb_log(2,"Pushing $i\n");
        push @disks,$i;
    }
}
close(FDISK);
pb_log(2,"Found the following disks: @disks\n");
return (@disks);
}

=item B<mr_disk_type>

This function uses fdisk to determine the type of disk whose device is passed as a parameter
It returns either msdos for MBR type or gpt for GPT type

=cut

sub mr_disk_type {

my $device = shift;
# MBR by default
my $type = "MBR";

my $fdisk = pb_check_req("fdisk",0);

open(FDISK, "LANG=C $fdisk -l $device 2>/dev/null |") || die "Unable to read from $fdisk";
while (<FDISK>) {
    if (($_ =~ /EFI GPT/) || ($_ =~ / GPT /) || ($_ =~ /Disk[\s]*label type: gpt/)) {
        $type= "GPT";
        last;
    }
}
close(FDISK);
pb_log(2,"Found a $type partition format\n");
return ($type);
}


=item B<mr_device_mounted>

This function returns the mount point if a device is mounted and undef if not.

=cut

sub mr_device_mounted {

my $device = shift;

return(1) if ((not defined $device) || ($device =~ "^/sys|^/proc"));

my $h = mr_file_read_all_link($device);
# Preserve that hash locally before earsing it to reuse another one
my $h2 = mr_file_copy_and_erase_hash($h);

foreach my $f (keys %$h2) {
    pb_log(2,"Working on device $f\n");
    # TODO: get them from a conf file - FreeBSD needs swapinfo for swap request
    foreach my $dev ("mount","cat /proc/swaps") {
        # Look for the device in the command result
        open(MOUNT,"$dev|") || die "Unable to execute the $dev command";
        while (<MOUNT>) {
            my ($mntp,$void) = split('\s',$_,2);
            $h = mr_file_read_all_link($mntp);
            foreach my $m (keys %$h) {
                pb_log(2,"== Working on mounted device $m\n");
                if ($m eq $f) {
                    # Find the mountpoint and return it
                    my ($void1,$mountpoint,$void2) = split('\s',$void);
                    pb_log(2,"*** Found $m on $mountpoint\n") if (defined $mountpoint);
                    return($mountpoint);
                }
            }
            my $h3 = mr_file_copy_and_erase_hash($h);
        }
        close(MOUNT);
    }
}

# Not found
return(undef);
}