]> git.proxmox.com Git - pve-storage.git/blame - PVE/Diskmanage.pm
Diskmanage: also detect BIOS boot, EFI and ZFS reserved type partitions
[pve-storage.git] / PVE / Diskmanage.pm
CommitLineData
cbba9b5b
DC
1package PVE::Diskmanage;
2
3use strict;
4use warnings;
d5c80a5b 5
cbba9b5b
DC
6use PVE::ProcFSTools;
7use Data::Dumper;
8use Cwd qw(abs_path);
3196c387 9use Fcntl ':mode';
92ae59df 10use File::stat;
8cd6d7e8 11use JSON;
cbba9b5b
DC
12
13use PVE::Tools qw(extract_param run_command file_get_contents file_read_firstline dir_glob_regex dir_glob_foreach trim);
14
15my $SMARTCTL = "/usr/sbin/smartctl";
16my $ZPOOL = "/sbin/zpool";
17my $SGDISK = "/sbin/sgdisk";
18my $PVS = "/sbin/pvs";
19dcd1ad 19my $LVS = "/sbin/lvs";
8cd6d7e8 20my $LSBLK = "/bin/lsblk";
cbba9b5b
DC
21
22sub verify_blockdev_path {
23 my ($rel_path) = @_;
24
25 die "missing path" if !$rel_path;
26 my $path = abs_path($rel_path);
27 die "failed to get absolute path to $rel_path\n" if !$path;
28
29 die "got unusual device path '$path'\n" if $path !~ m|^/dev/(.*)$|;
30
31 $path = "/dev/$1"; # untaint
32
33 assert_blockdev($path);
34
35 return $path;
36}
37
38sub assert_blockdev {
39 my ($dev, $noerr) = @_;
40
41 if ($dev !~ m|^/dev/| || !(-b $dev)) {
42 return undef if $noerr;
43 die "not a valid block device\n";
44 }
45
46 return 1;
47}
48
49sub init_disk {
50 my ($disk, $uuid) = @_;
51
52 assert_blockdev($disk);
53
54 # we should already have checked if it is in use in the api call
55 # but we check again for safety
56 die "disk $disk is already in use\n" if disk_is_used($disk);
57
58 my $id = $uuid || 'R';
59 run_command([$SGDISK, $disk, '-U', $id]);
60 return 1;
61}
62
63sub disk_is_used {
64 my ($disk) = @_;
65
66 my $dev = $disk;
67 $dev =~ s|^/dev/||;
68
7a98a62d 69 my $disklist = get_disks($dev, 1);
cbba9b5b
DC
70
71 die "'$disk' is not a valid local disk\n" if !defined($disklist->{$dev});
72 return 1 if $disklist->{$dev}->{used};
73
74 return 0;
75}
76
77sub get_smart_data {
dd902da7 78 my ($disk, $healthonly) = @_;
cbba9b5b
DC
79
80 assert_blockdev($disk);
81 my $smartdata = {};
dc1311cb 82 my $type;
cbba9b5b 83
9018a4e6 84 my $returncode = 0;
c9bd3d22 85
c3442aa5
WB
86 if ($disk =~ m!^/dev/(nvme\d+n\d+)$!) {
87 my $info = get_sysdir_info("/sys/block/$1");
88 $disk = "/dev/".($info->{device}
89 or die "failed to get nvme controller device for $disk\n");
90 }
c9bd3d22 91
dd902da7
DC
92 my $cmd = [$SMARTCTL, '-H'];
93 push @$cmd, '-A', '-f', 'brief' if !$healthonly;
94 push @$cmd, $disk;
95
cbba9b5b 96 eval {
dd902da7 97 $returncode = run_command($cmd, noerr => 1, outfunc => sub{
cbba9b5b
DC
98 my ($line) = @_;
99
1c999553
FG
100# ATA SMART attributes, e.g.:
101# ID# ATTRIBUTE_NAME FLAGS VALUE WORST THRESH FAIL RAW_VALUE
102# 1 Raw_Read_Error_Rate POSR-K 100 100 000 - 0
dc1311cb
FG
103#
104# SAS and NVME disks, e.g.:
105# Data Units Written: 5,584,952 [2.85 TB]
106# Accumulated start-stop cycles: 34
107
bd54091c 108 if (defined($type) && $type eq 'ata' && $line =~ m/^([ \d]{2}\d)\s+(\S+)\s+(\S{6})\s+(\d+)\s+(\d+)\s+(\S+)\s+(\S+)\s+(.*)$/) {
cbba9b5b 109 my $entry = {};
bd54091c 110
1c999553
FG
111 $entry->{name} = $2 if defined $2;
112 $entry->{flags} = $3 if defined $3;
cbba9b5b 113 # the +0 makes a number out of the strings
1c999553
FG
114 $entry->{value} = $4+0 if defined $4;
115 $entry->{worst} = $5+0 if defined $5;
bd54091c
DC
116 # some disks report the default threshold as --- instead of 000
117 if (defined($6) && $6 eq '---') {
118 $entry->{threshold} = 0;
119 } else {
120 $entry->{threshold} = $6+0 if defined $6;
121 }
1c999553
FG
122 $entry->{fail} = $7 if defined $7;
123 $entry->{raw} = $8 if defined $8;
124 $entry->{id} = $1 if defined $1;
cbba9b5b 125 push @{$smartdata->{attributes}}, $entry;
5db2d529 126 } elsif ($line =~ m/(?:Health Status|self\-assessment test result): (.*)$/ ) {
cbba9b5b
DC
127 $smartdata->{health} = $1;
128 } elsif ($line =~ m/Vendor Specific SMART Attributes with Thresholds:/) {
dc1311cb
FG
129 $type = 'ata';
130 delete $smartdata->{text};
131 } elsif ($line =~ m/=== START OF (READ )?SMART DATA SECTION ===/) {
132 $type = 'text';
133 } elsif (defined($type) && $type eq 'text') {
134 $smartdata->{text} = '' if !defined $smartdata->{text};
135 $smartdata->{text} .= "$line\n";
2c048efd
DC
136 # extract wearout from nvme/sas text, allow for decimal values
137 if ($line =~ m/Percentage Used(?: endurance indicator)?:\s*(\d+(?:\.\d+)?)\%/i) {
ea928fd4
DC
138 $smartdata->{wearout} = 100 - $1;
139 }
dd902da7
DC
140 } elsif ($line =~ m/SMART Disabled/) {
141 $smartdata->{health} = "SMART Disabled";
cbba9b5b
DC
142 }
143 });
144 };
9018a4e6
DC
145 my $err = $@;
146
147 # bit 0 and 1 mark an severe smartctl error
148 # all others are for disk status, so ignore them
149 # see smartctl(8)
150 if ((defined($returncode) && ($returncode & 0b00000011)) || $err) {
151 die "Error getting S.M.A.R.T. data: Exit code: $returncode\n";
152 }
dc1311cb
FG
153
154 $smartdata->{type} = $type;
155
cbba9b5b
DC
156 return $smartdata;
157}
158
b6bbc2ab 159sub get_lsblk_info() {
59c03cd9 160 my $cmd = [$LSBLK, '--json', '-o', 'path,parttype,fstype'];
8cd6d7e8
DC
161 my $output = "";
162 my $res = {};
163 eval {
164 run_command($cmd, outfunc => sub {
165 my ($line) = @_;
166 $output .= "$line\n";
167 });
168 };
169 warn "$@\n" if $@;
170 return $res if $output eq '';
171
172 my $parsed = eval { decode_json($output) };
173 warn "$@\n" if $@;
174 my $list = $parsed->{blockdevices} // [];
175
b6bbc2ab 176 $res = { map {
59c03cd9
FE
177 $_->{path} => {
178 parttype => $_->{parttype},
179 fstype => $_->{fstype}
180 }
b6bbc2ab 181 } @{$list} };
8cd6d7e8
DC
182
183 return $res;
184}
185
186my $get_devices_by_partuuid = sub {
b6bbc2ab 187 my ($lsblk_info, $uuids, $res) = @_;
8cd6d7e8
DC
188
189 $res = {} if !defined($res);
190
b6bbc2ab
FE
191 foreach my $dev (sort keys %{$lsblk_info}) {
192 my $uuid = $lsblk_info->{$dev}->{parttype};
193 next if !defined($uuid) || !defined($uuids->{$uuid});
194 $res->{$dev} = $uuids->{$uuid};
8cd6d7e8
DC
195 }
196
197 return $res;
198};
199
cbba9b5b 200sub get_zfs_devices {
b6bbc2ab 201 my ($lsblk_info) = @_;
8cd6d7e8 202 my $res = {};
cbba9b5b 203
501562d4 204 return {} if ! -x $ZPOOL;
4526dffa 205
cbba9b5b
DC
206 # use zpool and parttype uuid,
207 # because log and cache do not have
208 # zfs type uuid
209 eval {
210 run_command([$ZPOOL, 'list', '-HPLv'], outfunc => sub {
211 my ($line) = @_;
212
213 if ($line =~ m|^\t([^\t]+)\t|) {
8cd6d7e8 214 $res->{$1} = 1;
cbba9b5b
DC
215 }
216 });
217 };
218
219 # only warn here,
220 # because maybe zfs tools are not installed
221 warn "$@\n" if $@;
222
8cd6d7e8
DC
223 my $uuids = {
224 "6a898cc3-1dd2-11b2-99a6-080020736631" => 1, # apple
225 "516e7cba-6ecf-11d6-8ff8-00022d09712b" => 1, # bsd
226 };
cbba9b5b 227
cbba9b5b 228
b6bbc2ab 229 $res = $get_devices_by_partuuid->($lsblk_info, $uuids, $res);
8cd6d7e8
DC
230
231 return $res;
cbba9b5b
DC
232}
233
234sub get_lvm_devices {
b6bbc2ab 235 my ($lsblk_info) = @_;
8cd6d7e8 236 my $res = {};
cbba9b5b
DC
237 eval {
238 run_command([$PVS, '--noheadings', '--readonly', '-o', 'pv_name'], outfunc => sub{
239 my ($line) = @_;
240 $line = trim($line);
241 if ($line =~ m|^/dev/|) {
8cd6d7e8 242 $res->{$line} = 1;
cbba9b5b
DC
243 }
244 });
245 };
246
247 # if something goes wrong, we do not want
248 # to give up, but indicate an error has occured
249 warn "$@\n" if $@;
250
8cd6d7e8
DC
251 my $uuids = {
252 "e6d6d379-f507-44c2-a23c-238f2a3df928" => 1,
253 };
cbba9b5b 254
b6bbc2ab 255 $res = $get_devices_by_partuuid->($lsblk_info, $uuids, $res);
cbba9b5b 256
8cd6d7e8 257 return $res;
cbba9b5b
DC
258}
259
260sub get_ceph_journals {
b6bbc2ab 261 my ($lsblk_info) = @_;
8cd6d7e8
DC
262 my $res = {};
263
264 my $uuids = {
265 '45b0969e-9b03-4f30-b4c6-b4b80ceff106' => 1, # journal
266 '30cd0809-c2b2-499c-8879-2d6b78529876' => 2, # db
267 '5ce17fce-4087-4169-b7ff-056cc58473f9' => 3, # wal
268 'cafecafe-9b03-4f30-b4c6-b4b80ceff106' => 4, # block
269 };
270
b6bbc2ab 271 $res = $get_devices_by_partuuid->($lsblk_info, $uuids, $res);
cbba9b5b 272
8cd6d7e8 273 return $res;
cbba9b5b
DC
274}
275
19dcd1ad
DC
276# reads the lv_tags and matches them with the devices
277sub get_ceph_volume_infos {
278 my $result = {};
279
248f43f5
TL
280 my $cmd = [ $LVS, '-S', 'lv_name=~^osd-', '-o', 'devices,lv_name,lv_tags',
281 '--noheadings', '--readonly', '--separator', ';' ];
19dcd1ad
DC
282
283 run_command($cmd, outfunc => sub {
284 my $line = shift;
248f43f5
TL
285 $line =~ s/(?:^\s+)|(?:\s+$)//g; # trim whitespaces
286
287 my $fields = [ split(';', $line) ];
19dcd1ad
DC
288
289 # lvs syntax is /dev/sdX(Y) where Y is the start (which we do not need)
290 my ($dev) = $fields->[0] =~ m|^(/dev/[a-z]+)|;
291 if ($fields->[1] =~ m|^osd-([^-]+)-|) {
292 my $type = $1;
248f43f5 293 # $result autovivification is wanted, to not creating empty hashes
79f4a7bf 294 if (($type eq 'block' || $type eq 'data') && $fields->[2] =~ m/ceph.osd_id=([^,]+)/) {
19dcd1ad
DC
295 $result->{$dev}->{osdid} = $1;
296 $result->{$dev}->{bluestore} = ($type eq 'block');
bfb3d42d
DC
297 if ($fields->[2] =~ m/ceph\.encrypted=1/) {
298 $result->{$dev}->{encrypted} = 1;
299 }
19dcd1ad 300 } else {
248f43f5 301 # undef++ becomes '1' (see `perldoc perlop`: Auto-increment)
19dcd1ad
DC
302 $result->{$dev}->{$type}++;
303 }
304 }
305 });
306
307 return $result;
308}
309
cbba9b5b
DC
310sub get_udev_info {
311 my ($dev) = @_;
312
313 my $info = "";
314 my $data = {};
315 eval {
d3a5e309 316 run_command(['udevadm', 'info', '-p', $dev, '--query', 'all'], outfunc => sub {
cbba9b5b
DC
317 my ($line) = @_;
318 $info .= "$line\n";
319 });
320 };
321 warn $@ if $@;
322 return undef if !$info;
323
324 return undef if $info !~ m/^E: DEVTYPE=disk$/m;
325 return undef if $info =~ m/^E: ID_CDROM/m;
326
327 # we use this, because some disks are not simply in /dev
328 # e.g. /dev/cciss/c0d0
329 if ($info =~ m/^E: DEVNAME=(\S+)$/m) {
330 $data->{devpath} = $1;
331 }
332 return if !defined($data->{devpath});
333
334 $data->{serial} = 'unknown';
335 if ($info =~ m/^E: ID_SERIAL_SHORT=(\S+)$/m) {
336 $data->{serial} = $1;
337 }
338
339 $data->{gpt} = 0;
340 if ($info =~ m/^E: ID_PART_TABLE_TYPE=gpt$/m) {
341 $data->{gpt} = 1;
342 }
343
344 # detect SSD
345 $data->{rpm} = -1;
346 if ($info =~ m/^E: ID_ATA_ROTATION_RATE_RPM=(\d+)$/m) {
347 $data->{rpm} = $1;
348 }
349
350 if ($info =~ m/^E: ID_BUS=usb$/m) {
351 $data->{usb} = 1;
352 }
353
865bdbd9
DC
354 if ($info =~ m/^E: ID_MODEL=(.+)$/m) {
355 $data->{model} = $1;
356 }
357
cbba9b5b
DC
358 $data->{wwn} = 'unknown';
359 if ($info =~ m/^E: ID_WWN=(.*)$/m) {
360 $data->{wwn} = $1;
361 }
362
0f0d99a3
SI
363 if ($info =~ m/^E: DEVLINKS=(.+)$/m) {
364 my @devlinks = grep(m#^/dev/disk/by-id/(ata|scsi|nvme(?!-eui))#, split (/ /, $1));
365 $data->{by_id_link} = $devlinks[0] if defined($devlinks[0]);
366 }
367
cbba9b5b
DC
368 return $data;
369}
370
40be5c5c
FE
371sub get_sysdir_size {
372 my ($sysdir) = @_;
373
374 my $size = file_read_firstline("$sysdir/size");
375 return if !$size;
376
377 # linux always considers sectors to be 512 bytes,
378 # independently of real block size
379 return $size * 512;
380}
381
cbba9b5b
DC
382sub get_sysdir_info {
383 my ($sysdir) = @_;
384
461a9fd8
DC
385 return undef if ! -d "$sysdir/device";
386
cbba9b5b
DC
387 my $data = {};
388
40be5c5c 389 $data->{size} = get_sysdir_size($sysdir) or return;
cbba9b5b
DC
390
391 # dir/queue/rotational should be 1 for hdd, 0 for ssd
571b6f26 392 $data->{rotational} = file_read_firstline("$sysdir/queue/rotational") // -1;
cbba9b5b
DC
393
394 $data->{vendor} = file_read_firstline("$sysdir/device/vendor") || 'unknown';
395 $data->{model} = file_read_firstline("$sysdir/device/model") || 'unknown';
396
c3442aa5
WB
397 if (defined(my $device = readlink("$sysdir/device"))) {
398 # strip directory and untaint:
399 ($data->{device}) = $device =~ m!([^/]+)$!;
400 }
401
cbba9b5b
DC
402 return $data;
403}
404
6965a670 405sub get_wear_leveling_info {
dbad606d 406 my ($smartdata) = @_;
ea928fd4
DC
407 my $attributes = $smartdata->{attributes};
408
409 if (defined($smartdata->{wearout})) {
410 return $smartdata->{wearout};
411 }
6965a670
DC
412
413 my $wearout;
414
dbad606d
JJS
415 # Common register names that represent percentage values of potential
416 # failure indicators used in drivedb.h of smartmontool's. Order matters,
417 # as some drives may have multiple definitions
418 my @wearoutregisters = (
419 "Media_Wearout_Indicator",
420 "SSD_Life_Left",
421 "Wear_Leveling_Count",
422 "Perc_Write\/Erase_Ct_BC",
423 "Perc_Rated_Life_Remain",
424 "Remaining_Lifetime_Perc",
425 "Percent_Lifetime_Remain",
426 "Lifetime_Left",
427 "PCT_Life_Remaining",
428 "Lifetime_Remaining",
429 "Percent_Life_Remaining",
430 "Percent_Lifetime_Used",
431 "Perc_Rated_Life_Used"
432 );
433
434 # Search for S.M.A.R.T. attributes for known register
435 foreach my $register (@wearoutregisters) {
436 last if defined $wearout;
437 foreach my $attr (@$attributes) {
438 next if $attr->{name} !~ m/$register/;
439 $wearout = $attr->{value};
440 last;
6965a670
DC
441 }
442 }
443
6965a670
DC
444 return $wearout;
445}
446
10a48db5
DC
447sub dir_is_empty {
448 my ($dir) = @_;
449
450 my $dh = IO::Dir->new ($dir);
451 return 1 if !$dh;
452
453 while (defined(my $tmp = $dh->read)) {
454 next if $tmp eq '.' || $tmp eq '..';
455 $dh->close;
456 return 0;
457 }
458 $dh->close;
459 return 1;
460}
461
eebcdb11
DC
462sub is_iscsi {
463 my ($sysdir) = @_;
464
465 if (-l $sysdir && readlink($sysdir) =~ m|host[^/]*/session[^/]*|) {
466 return 1;
467 }
468
469 return 0;
470}
471
4731eb11
TL
472my sub is_ssdlike {
473 my ($type) = @_;
474 return $type eq 'ssd' || $type eq 'nvme';
475}
476
cbba9b5b 477sub get_disks {
52a064af 478 my ($disks, $nosmart) = @_;
cbba9b5b
DC
479 my $disklist = {};
480
481 my $mounted = {};
482
483 my $mounts = PVE::ProcFSTools::parse_proc_mounts();
484
485 foreach my $mount (@$mounts) {
486 next if $mount->[0] !~ m|^/dev/|;
487 $mounted->{abs_path($mount->[0])} = $mount->[1];
488 };
489
b6bbc2ab 490 my $lsblk_info = get_lsblk_info();
8cd6d7e8 491
b6bbc2ab 492 my $journalhash = get_ceph_journals($lsblk_info);
19dcd1ad 493 my $ceph_volume_infos = get_ceph_volume_infos();
cbba9b5b 494
b6bbc2ab 495 my $zfshash = get_zfs_devices($lsblk_info);
cbba9b5b 496
b6bbc2ab 497 my $lvmhash = get_lvm_devices($lsblk_info);
cbba9b5b 498
52a064af
DC
499 my $disk_regex = ".*";
500 if (defined($disks)) {
501 if (!ref($disks)) {
502 $disks = [ $disks ];
503 } elsif (ref($disks) ne 'ARRAY') {
504 die "disks is not a string or array reference\n";
505 }
506 # we get cciss/c0d0 but need cciss!c0d0
5045e0b7 507 $_ =~ s|cciss/|cciss!| for @$disks;
52a064af
DC
508
509 $disk_regex = "(?:" . join('|', @$disks) . ")";
1590fc13
DC
510 }
511
52a064af 512 dir_glob_foreach('/sys/block', $disk_regex, sub {
cbba9b5b 513 my ($dev) = @_;
cbba9b5b
DC
514 # whitelisting following devices
515 # hdX: ide block device
516 # sdX: sd block device
517 # vdX: virtual block device
518 # xvdX: xen virtual block device
519 # nvmeXnY: nvme devices
38ddd4ce 520 # cciss!cXnY: cciss devices
cbba9b5b
DC
521 return if $dev !~ m/^(h|s|x?v)d[a-z]+$/ &&
522 $dev !~ m/^nvme\d+n\d+$/ &&
38ddd4ce 523 $dev !~ m/^cciss\!c\d+d\d+$/;
cbba9b5b 524
532e89e7 525 my $data = get_udev_info("/sys/block/$dev");
cbba9b5b
DC
526 return if !defined($data);
527 my $devpath = $data->{devpath};
528
529 my $sysdir = "/sys/block/$dev";
530
cbba9b5b 531 # we do not want iscsi devices
eebcdb11 532 return if is_iscsi($sysdir);
cbba9b5b
DC
533
534 my $sysdata = get_sysdir_info($sysdir);
535 return if !defined($sysdata);
536
537 my $type = 'unknown';
538
539 if ($sysdata->{rotational} == 0) {
540 $type = 'ssd';
4731eb11 541 $type = 'nvme' if $dev =~ m/^nvme\d+n\d+$/;
cbba9b5b
DC
542 $data->{rpm} = 0;
543 } elsif ($sysdata->{rotational} == 1) {
544 if ($data->{rpm} != -1) {
545 $type = 'hdd';
546 } elsif ($data->{usb}) {
547 $type = 'usb';
548 $data->{rpm} = 0;
549 }
550 }
551
acd3d916 552 my $health = 'UNKNOWN';
6965a670 553 my $wearout = 'N/A';
7a98a62d
FG
554
555 if (!$nosmart) {
556 eval {
4731eb11 557 my $smartdata = get_smart_data($devpath, !is_ssdlike($type));
dd902da7
DC
558 $health = $smartdata->{health} if $smartdata->{health};
559
4731eb11 560 if (is_ssdlike($type)) {
7a98a62d 561 # if we have an ssd we try to get the wearout indicator
dbad606d 562 my $wearval = get_wear_leveling_info($smartdata);
9250ddfe 563 $wearout = $wearval if defined($wearval);
acd3d916 564 }
7a98a62d
FG
565 };
566 }
cbba9b5b 567
fc7c0e05
DC
568 # we replaced cciss/ with cciss! above
569 # but in the result we need cciss/ again
570 # because the caller might want to check the
571 # result again with the original parameter
572 if ($dev =~ m|^cciss!|) {
573 $dev =~ s|^cciss!|cciss/|;
574 }
575
cbba9b5b
DC
576 $disklist->{$dev} = {
577 vendor => $sysdata->{vendor},
865bdbd9 578 model => $data->{model} || $sysdata->{model},
cbba9b5b
DC
579 size => $sysdata->{size},
580 serial => $data->{serial},
581 gpt => $data->{gpt},
582 rpm => $data->{rpm},
583 type => $type,
584 wwn => $data->{wwn},
585 health => $health,
586 devpath => $devpath,
587 wearout => $wearout,
588 };
589
0f0d99a3
SI
590 my $by_id_link = $data->{by_id_link};
591 $disklist->{$dev}->{by_id_link} = $by_id_link if defined($by_id_link);
592
cbba9b5b 593 my $osdid = -1;
e2bd817c 594 my $bluestore = 0;
bfb3d42d 595 my $osdencrypted = 0;
cbba9b5b
DC
596
597 my $journal_count = 0;
e2bd817c
DC
598 my $db_count = 0;
599 my $wal_count = 0;
cbba9b5b 600
cbba9b5b
DC
601 my $partpath = $devpath;
602
603 # remove part after last / to
604 # get the base path for the partitions
605 # e.g. from /dev/cciss/c0d0 get /dev/cciss
606 $partpath =~ s/\/[^\/]+$//;
607
01aa7d75
FE
608 my $determine_usage = sub {
609 my ($devpath, $sysdir, $is_partition) = @_;
610
611 return 'LVM' if $lvmhash->{$devpath};
612 return 'ZFS' if $zfshash->{$devpath};
613
614 my $info = $lsblk_info->{$devpath} // {};
d3857eeb
FE
615
616 my $parttype = $info->{parttype};
617 if (defined($parttype)) {
618 return 'BIOS boot'
619 if $parttype eq '21686148-6449-6e6f-744e-656564454649';
620 return 'EFI'
621 if $parttype eq 'c12a7328-f81f-11d2-ba4b-00a0c93ec93b';
622 return 'ZFS reserved'
623 if $parttype eq '6a945a3b-1dd2-11b2-99a6-080020736631';
624 }
625
01aa7d75
FE
626 my $fstype = $info->{fstype};
627 if (defined($fstype)) {
628 return "${fstype} (mounted)" if $mounted->{$devpath};
629 return "${fstype}";
630 }
631 return 'mounted' if $mounted->{$devpath};
632
633 return if !$is_partition;
634
635 # for devices, this check is done explicitly later
636 return 'Device Mapper' if !dir_is_empty("$sysdir/holders");
637
638 return 'partition';
639 };
640
89c27ea8
FE
641 my $partitions = {};
642
cbba9b5b
DC
643 dir_glob_foreach("$sysdir", "$dev.+", sub {
644 my ($part) = @_;
645
89c27ea8
FE
646 $partitions->{$part}->{devpath} = "$partpath/$part";
647 $partitions->{$part}->{gpt} = $data->{gpt};
648 $partitions->{$part}->{size} =
649 get_sysdir_size("$sysdir/$part") // 0;
01aa7d75
FE
650 $partitions->{$part}->{used} =
651 $determine_usage->("$partpath/$part", "$sysdir/$part", 1);
cbba9b5b 652
0cca5356 653 if (my $mp = $mounted->{"$partpath/$part"}) {
cbba9b5b
DC
654 if ($mp =~ m|^/var/lib/ceph/osd/ceph-(\d+)$|) {
655 $osdid = $1;
656 }
657 }
658
0180fa42
TL
659 if (my $journal_part = $journalhash->{"$partpath/$part"}) {
660 $journal_count++ if $journal_part == 1;
661 $db_count++ if $journal_part == 2;
662 $wal_count++ if $journal_part == 3;
663 $bluestore = 1 if $journal_part == 4;
e2bd817c 664 }
cbba9b5b
DC
665 });
666
0180fa42
TL
667 if (my $ceph_volume = $ceph_volume_infos->{$devpath}) {
668 $journal_count += $ceph_volume->{journal} // 0;
669 $db_count += $ceph_volume->{db} // 0;
670 $wal_count += $ceph_volume->{wal} // 0;
35ccd3d0 671 if (defined($ceph_volume->{osdid})) {
0180fa42
TL
672 $osdid = $ceph_volume->{osdid};
673 $bluestore = 1 if $ceph_volume->{bluestore};
bfb3d42d 674 $osdencrypted = 1 if $ceph_volume->{encrypted};
19dcd1ad
DC
675 }
676 }
677
01aa7d75
FE
678 my $used = $determine_usage->($devpath, $sysdir, 0);
679 foreach my $part (sort keys %{$partitions}) {
680 next if $partitions->{$part}->{used} eq 'partition';
681 $used //= $partitions->{$part}->{used};
682 }
683 $used //= 'partitions' if scalar(keys %{$partitions});
cbba9b5b
DC
684 # multipath, software raid, etc.
685 # this check comes in last, to show more specific info
686 # if we have it
01aa7d75 687 $used //= 'Device Mapper' if !dir_is_empty("$sysdir/holders");
cbba9b5b
DC
688
689 $disklist->{$dev}->{used} = $used if $used;
690 $disklist->{$dev}->{osdid} = $osdid;
e2bd817c
DC
691 $disklist->{$dev}->{journals} = $journal_count if $journal_count;
692 $disklist->{$dev}->{bluestore} = $bluestore if $osdid != -1;
bfb3d42d 693 $disklist->{$dev}->{osdencrypted} = $osdencrypted if $osdid != -1;
e2bd817c
DC
694 $disklist->{$dev}->{db} = $db_count if $db_count;
695 $disklist->{$dev}->{wal} = $wal_count if $wal_count;
cbba9b5b
DC
696 });
697
698 return $disklist;
699
700}
701
3196c387
WL
702sub get_partnum {
703 my ($part_path) = @_;
704
92ae59df 705 my $st = stat($part_path);
3196c387 706
92ae59df
AA
707 next if !$st->mode || !S_ISBLK($st->mode) || !$st->rdev;
708 my $major = PVE::Tools::dev_t_major($st->rdev);
709 my $minor = PVE::Tools::dev_t_minor($st->rdev);
3196c387
WL
710 my $partnum_path = "/sys/dev/block/$major:$minor/";
711
712 my $partnum;
713
714 $partnum = file_read_firstline("${partnum_path}partition");
715
481f6177 716 die "Partition does not exist\n" if !defined($partnum);
3196c387
WL
717
718 #untaint and ensure it is a int
719 if ($partnum =~ m/(\d+)/) {
720 $partnum = $1;
721 die "Partition number $partnum is invalid\n" if $partnum > 128;
722 } else {
723 die "Failed to get partition number\n";
724 }
725
726 return $partnum;
727}
728
0d28307d
WL
729sub get_blockdev {
730 my ($part_path) = @_;
731
1207620c
TL
732 my ($dev, $block_dev);
733 if ($part_path =~ m|^/dev/(.*)$|) {
734 $dev = $1;
735 my $link = readlink "/sys/class/block/$dev";
736 $block_dev = $1 if $link =~ m|([^/]*)/$dev$|;
737 }
0d28307d
WL
738
739 die "Can't parse parent device\n" if !defined($block_dev);
740 die "No valid block device\n" if index($dev, $block_dev) == -1;
741
742 $block_dev = "/dev/$block_dev";
743 die "Block device does not exsists\n" if !(-b $block_dev);
744
745 return $block_dev;
746}
747
e39e8ee2
DC
748sub locked_disk_action {
749 my ($sub) = @_;
750 my $res = PVE::Tools::lock_file('/run/lock/pve-diskmanage.lck', undef, $sub);
751 die $@ if $@;
752 return $res;
753}
754
0370861c 755sub assert_disk_unused {
76c1e57b
DC
756 my ($dev) = @_;
757
0370861c 758 die "device '$dev' is already in use\n" if disk_is_used($dev);
76c1e57b
DC
759
760 return undef;
761}
762
1dc3038d
DC
763sub append_partition {
764 my ($dev, $size) = @_;
765
766 my $devname = $dev;
767 $devname =~ s|^/dev/||;
768
769 my $newpartid = 1;
770 dir_glob_foreach("/sys/block/$devname", qr/\Q$devname\E.*?(\d+)/, sub {
771 my ($part, $partid) = @_;
772
773 if ($partid >= $newpartid) {
774 $newpartid = $partid + 1;
775 }
776 });
777
778 $size = PVE::Tools::convert_size($size, 'b' => 'mb');
779
780 run_command([ $SGDISK, '-n', "$newpartid:0:+${size}M", $dev ],
781 errmsg => "error creating partition '$newpartid' on '$dev'");
782
783 my $partition;
784
785 # loop again to detect the real partiton device which does not always follow
786 # a strict $devname$partition scheme like /dev/nvme0n1 -> /dev/nvme0n1p1
787 dir_glob_foreach("/sys/block/$devname", qr/\Q$devname\E.*$newpartid/, sub {
788 my ($part) = @_;
789
790 $partition = "/dev/$part";
791 });
792
793 return $partition;
794}
795
cbba9b5b 7961;