[pve-devel] applied-series: [PATCH storage v2 1/3] Diskmanage: change parttype uuid detection

Fabian Grünbichler f.gruenbichler at proxmox.com
Fri Jul 12 16:02:06 CEST 2019


applied all three, thanks!

On Fri, Jul 12, 2019 at 12:13:49PM +0200, Dominik Csapak wrote:
> previously ceph included a udev rule to populate
> /dev/disk/by-parttypeuuid/
> 
> but not anymore, so we now use 'lsblk --json -o path,parttype' to
> get a mapping between parttype uuid and partition
> 
> fix the test by simulating empty lsblk output
> 
> Signed-off-by: Dominik Csapak <d.csapak at proxmox.com>
> ---
> changes from v1:
> * fix wrong output collection ($_ is undef in outfunc, use the first
>   parameter instead)
> * rename get_lsblk_info to get_parttype_info
> * reuse the 'uuidlist to result hash' code (this also now conserves
>   the behaviour that journals/db/wal etc. are now correctly saved,
>   this was broken in v1)
> * renames the *hash variables to *list
> * parameter/result renaming
> 
>  PVE/Diskmanage.pm     | 127 ++++++++++++++++++++++++++----------------
>  test/disklist_test.pm |   9 ++-
>  2 files changed, 87 insertions(+), 49 deletions(-)
> 
> diff --git a/PVE/Diskmanage.pm b/PVE/Diskmanage.pm
> index 0deb1a6..196eee5 100644
> --- a/PVE/Diskmanage.pm
> +++ b/PVE/Diskmanage.pm
> @@ -6,6 +6,7 @@ use PVE::ProcFSTools;
>  use Data::Dumper;
>  use Cwd qw(abs_path);
>  use Fcntl ':mode';
> +use JSON;
>  
>  use PVE::Tools qw(extract_param run_command file_get_contents file_read_firstline dir_glob_regex dir_glob_foreach trim);
>  
> @@ -15,6 +16,7 @@ my $SGDISK = "/sbin/sgdisk";
>  my $PVS = "/sbin/pvs";
>  my $LVS = "/sbin/lvs";
>  my $UDEVADM = "/bin/udevadm";
> +my $LSBLK = "/bin/lsblk";
>  
>  sub verify_blockdev_path {
>      my ($rel_path) = @_;
> @@ -154,8 +156,48 @@ sub get_smart_data {
>      return $smartdata;
>  }
>  
> +sub get_parttype_info() {
> +    my $cmd = [$LSBLK, '--json', '-o', 'path,parttype'];
> +    my $output = "";
> +    my $res = {};
> +    eval {
> +	run_command($cmd, outfunc => sub {
> +	    my ($line) = @_;
> +	    $output .= "$line\n";
> +	});
> +    };
> +    warn "$@\n" if $@;
> +    return $res if $output eq '';
> +
> +    my $parsed = eval { decode_json($output) };
> +    warn "$@\n" if $@;
> +    my $list = $parsed->{blockdevices} // [];
> +
> +    foreach my $dev (@$list) {
> +	next if !($dev->{parttype});
> +	my $type = $dev->{parttype};
> +	$res->{$type} = [] if !defined($res->{$type});
> +	push @{$res->{$type}}, $dev->{path};
> +    }
> +
> +    return $res;
> +}
> +
> +my $get_devices_by_partuuid = sub {
> +    my ($parttype_map, $uuids, $res) = @_;
> +
> +    $res = {} if !defined($res);
> +
> +    foreach my $uuid (sort keys %$uuids) {
> +       map { $res->{$_} = $uuids->{$uuid} } @{$parttype_map->{$uuid}};
> +    }
> +
> +    return $res;
> +};
> +
>  sub get_zfs_devices {
> -    my $list = {};
> +    my ($parttype_map) = @_;
> +    my $res = {};
>  
>      return {} if ! -x $ZPOOL;
>  
> @@ -167,7 +209,7 @@ sub get_zfs_devices {
>  	     my ($line) = @_;
>  
>  	     if ($line =~ m|^\t([^\t]+)\t|) {
> -		$list->{$1} = 1;
> +		$res->{$1} = 1;
>  	     }
>  	});
>      };
> @@ -176,26 +218,26 @@ sub get_zfs_devices {
>      # because maybe zfs tools are not installed
>      warn "$@\n" if $@;
>  
> -    my $applezfsuuid = "6a898cc3-1dd2-11b2-99a6-080020736631";
> -    my $bsdzfsuuid = "516e7cba-6ecf-11d6-8ff8-00022d09712b";
> +    my $uuids = {
> +	"6a898cc3-1dd2-11b2-99a6-080020736631" => 1, # apple
> +	"516e7cba-6ecf-11d6-8ff8-00022d09712b" => 1, # bsd
> +    };
> +
>  
> -    dir_glob_foreach('/dev/disk/by-parttypeuuid', "($applezfsuuid|$bsdzfsuuid)\..+", sub {
> -	my ($entry) = @_;
> -	my $real_dev = abs_path("/dev/disk/by-parttypeuuid/$entry");
> -	$list->{$real_dev} = 1;
> -    });
> +    $res = $get_devices_by_partuuid->($parttype_map, $uuids, $res);
>  
> -    return $list;
> +    return $res;
>  }
>  
>  sub get_lvm_devices {
> -    my $list = {};
> +    my ($parttype_map) = @_;
> +    my $res = {};
>      eval {
>  	run_command([$PVS, '--noheadings', '--readonly', '-o', 'pv_name'], outfunc => sub{
>  	    my ($line) = @_;
>  	    $line = trim($line);
>  	    if ($line =~ m|^/dev/|) {
> -		$list->{$line} = 1;
> +		$res->{$line} = 1;
>  	    }
>  	});
>      };
> @@ -204,40 +246,29 @@ sub get_lvm_devices {
>      # to give up, but indicate an error has occured
>      warn "$@\n" if $@;
>  
> -    my $lvmuuid = "e6d6d379-f507-44c2-a23c-238f2a3df928";
> +    my $uuids = {
> +	"e6d6d379-f507-44c2-a23c-238f2a3df928" => 1,
> +    };
>  
> -    dir_glob_foreach('/dev/disk/by-parttypeuuid', "$lvmuuid\..+", sub {
> -	my ($entry) = @_;
> -	my $real_dev = abs_path("/dev/disk/by-parttypeuuid/$entry");
> -	$list->{$real_dev} = 1;
> -    });
> +    $res = $get_devices_by_partuuid->($parttype_map, $uuids, $res);
>  
> -    return $list;
> +    return $res;
>  }
>  
>  sub get_ceph_journals {
> -    my $journalhash = {};
> -
> -    my $journal_uuid = '45b0969e-9b03-4f30-b4c6-b4b80ceff106';
> -    my $db_uuid = '30cd0809-c2b2-499c-8879-2d6b78529876';
> -    my $wal_uuid = '5ce17fce-4087-4169-b7ff-056cc58473f9';
> -    my $block_uuid = 'cafecafe-9b03-4f30-b4c6-b4b80ceff106';
> -
> -    dir_glob_foreach('/dev/disk/by-parttypeuuid', "($journal_uuid|$db_uuid|$wal_uuid|$block_uuid)\..+", sub {
> -	my ($entry, $type) = @_;
> -	my $real_dev = abs_path("/dev/disk/by-parttypeuuid/$entry");
> -	if ($type eq $journal_uuid) {
> -	    $journalhash->{$real_dev} = 1;
> -	} elsif ($type eq $db_uuid) {
> -	    $journalhash->{$real_dev} = 2;
> -	} elsif ($type eq $wal_uuid) {
> -	    $journalhash->{$real_dev} = 3;
> -	} elsif ($type eq $block_uuid) {
> -	    $journalhash->{$real_dev} = 4;
> -	}
> -    });
> +    my ($parttype_map) = @_;
> +    my $res = {};
> +
> +    my $uuids = {
> +	'45b0969e-9b03-4f30-b4c6-b4b80ceff106' => 1, # journal
> +	'30cd0809-c2b2-499c-8879-2d6b78529876' => 2, # db
> +	'5ce17fce-4087-4169-b7ff-056cc58473f9' => 3, # wal
> +	'cafecafe-9b03-4f30-b4c6-b4b80ceff106' => 4, # block
> +    };
> +
> +    $res = $get_devices_by_partuuid->($parttype_map, $uuids, $res);
>  
> -    return $journalhash;
> +    return $res;
>  }
>  
>  # reads the lv_tags and matches them with the devices
> @@ -442,12 +473,14 @@ sub get_disks {
>  	return $mounted->{$dev};
>      };
>  
> -    my $journalhash = get_ceph_journals();
> +    my $parttype_map = get_parttype_info();
> +
> +    my $journalhash = get_ceph_journals($parttype_map);
>      my $ceph_volume_infos = get_ceph_volume_infos();
>  
> -    my $zfslist = get_zfs_devices();
> +    my $zfshash = get_zfs_devices($parttype_map);
>  
> -    my $lvmlist = get_lvm_devices();
> +    my $lvmhash = get_lvm_devices($parttype_map);
>  
>      my $disk_regex = ".*";
>      if (defined($disks)) {
> @@ -519,11 +552,11 @@ sub get_disks {
>  
>  	my $used;
>  
> -	$used = 'LVM' if $lvmlist->{$devpath};
> +	$used = 'LVM' if $lvmhash->{$devpath};
>  
>  	$used = 'mounted' if &$dev_is_mounted($devpath);
>  
> -	$used = 'ZFS' if $zfslist->{$devpath};
> +	$used = 'ZFS' if $zfshash->{$devpath};
>  
>  	# we replaced cciss/ with cciss! above
>  	# but in the result we need cciss/ again
> @@ -578,11 +611,11 @@ sub get_disks {
>  		}
>  	    }
>  
> -	    if ($lvmlist->{"$partpath/$part"}) {
> +	    if ($lvmhash->{"$partpath/$part"}) {
>  		$found_lvm = 1;
>  	    }
>  
> -	    if ($zfslist->{"$partpath/$part"}) {
> +	    if ($zfshash->{"$partpath/$part"}) {
>  		$found_zfs = 1;
>  	    }
>  
> diff --git a/test/disklist_test.pm b/test/disklist_test.pm
> index 527e882..9cb6763 100644
> --- a/test/disklist_test.pm
> +++ b/test/disklist_test.pm
> @@ -54,9 +54,14 @@ sub mocked_run_command {
>  	    @$outputlines = split(/\n/, read_test_file('pvs'));
>  	} elsif ($cmd->[0] =~ m/lvs/i) {
>  	    @$outputlines = split(/\n/, read_test_file('lvs'));
> +	} elsif ($cmd->[0] =~ m/lsblk/i) {
> +	    my $content = read_test_file('lsblk');
> +	    if ($content eq '') {
> +		$content = '{}';
> +	    }
> +	    @$outputlines = split(/\n/, $content);
>  	} else {
> -	    print "unexpected run_command call: '@$cmd', aborting\n";
> -	    die;
> +	    die "unexpected run_command call: '@$cmd', aborting\n";
>  	}
>      } else {
>  	print "unexpected run_command call: '@$cmd', aborting\n";
> -- 
> 2.20.1
> 
> 
> _______________________________________________
> pve-devel mailing list
> pve-devel at pve.proxmox.com
> https://pve.proxmox.com/cgi-bin/mailman/listinfo/pve-devel




More information about the pve-devel mailing list