From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from firstgate.proxmox.com (firstgate.proxmox.com [212.224.123.68]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by lists.proxmox.com (Postfix) with ESMTPS id 93ACEA1152 for ; Fri, 10 Nov 2023 14:51:02 +0100 (CET) Received: from firstgate.proxmox.com (localhost [127.0.0.1]) by firstgate.proxmox.com (Proxmox) with ESMTP id 7A04A17AE for ; Fri, 10 Nov 2023 14:50:32 +0100 (CET) Received: from proxmox-new.maurer-it.com (proxmox-new.maurer-it.com [94.136.29.106]) (using TLSv1.3 with cipher TLS_AES_256_GCM_SHA384 (256/256 bits) key-exchange X25519 server-signature RSA-PSS (2048 bits)) (No client certificate requested) by firstgate.proxmox.com (Proxmox) with ESMTPS for ; Fri, 10 Nov 2023 14:50:31 +0100 (CET) Received: from proxmox-new.maurer-it.com (localhost.localdomain [127.0.0.1]) by proxmox-new.maurer-it.com (Proxmox) with ESMTP id 4ED0747B20 for ; Fri, 10 Nov 2023 14:50:31 +0100 (CET) From: Markus Frank To: pbs-devel@lists.proxmox.com Date: Fri, 10 Nov 2023 14:50:06 +0100 Message-Id: <20231110135010.106015-2-m.frank@proxmox.com> X-Mailer: git-send-email 2.39.2 In-Reply-To: <20231110135010.106015-1-m.frank@proxmox.com> References: <20231110135010.106015-1-m.frank@proxmox.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit X-SPAM-LEVEL: Spam detection results: 0 AWL -0.037 Adjusted score from AWL reputation of From: address BAYES_00 -1.9 Bayes spam probability is 0 to 1% DMARC_MISSING 0.1 Missing DMARC policy KAM_DMARC_STATUS 0.01 Test Rule for DKIM or SPF Failure with Strict Alignment SPF_HELO_NONE 0.001 SPF: HELO does not publish an SPF Record SPF_PASS -0.001 SPF: sender matches SPF record T_SCC_BODY_TEXT_LINE -0.01 - Subject: [pbs-devel] [PATCH proxmox-backup 2/6] feature #3690: add wipe_blockdev & change_parttype rust equivalent X-BeenThere: pbs-devel@lists.proxmox.com X-Mailman-Version: 2.1.29 Precedence: list List-Id: Proxmox Backup Server development discussion List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-List-Received-Date: Fri, 10 Nov 2023 13:51:02 -0000 The wipe_blockdev & change_parttype functions are similar to PVE::Diskmanage's wipe_blockdev & change_parttype functions. The partition_by_name & complete_partition_name functions are modified disk_by_name & complete_disk_name functions for partitions. Signed-off-by: Markus Frank --- src/tools/disks/mod.rs | 90 +++++++++++++++++++++++++++++++++++++++++- 1 file changed, 89 insertions(+), 1 deletion(-) diff --git a/src/tools/disks/mod.rs b/src/tools/disks/mod.rs index 72e374ca..75d84696 100644 --- a/src/tools/disks/mod.rs +++ b/src/tools/disks/mod.rs @@ -19,7 +19,7 @@ use proxmox_lang::{io_bail, io_format_err}; use proxmox_schema::api; use proxmox_sys::linux::procfs::{mountinfo::Device, MountInfo}; -use pbs_api_types::BLOCKDEVICE_NAME_REGEX; +use pbs_api_types::{BLOCKDEVICE_NAME_REGEX, BLOCKDEVICE_PARTITION_NAME_REGEX}; mod zfs; pub use zfs::*; @@ -111,6 +111,12 @@ impl DiskManage { self.disk_by_sys_path(syspath) } + /// Get a `Disk` for a name in `/sys/class/block/`. + pub fn partition_by_name(self: Arc, name: &str) -> io::Result { + let syspath = format!("/sys/class/block/{}", name); + self.disk_by_sys_path(syspath) + } + /// Gather information about mounted disks: fn mounted_devices(&self) -> Result<&HashSet, Error> { self.mounted_devices @@ -1056,6 +1062,72 @@ pub fn inititialize_gpt_disk(disk: &Disk, uuid: Option<&str>) -> Result<(), Erro Ok(()) } +/// Wipes all labels and the first 200 MiB of a disk/partition (or the whole if it is smaller). +/// If called with a partition, also sets the partition type to 0x83 'Linux filesystem'. +pub fn wipe_blockdev(disk: &Disk) -> Result<(), Error> { + let disk_path = disk.device_path().unwrap(); + + let mut is_partition = false; + for disk_i in get_lsblk_info()?.iter() { + if disk_i.path == disk_path.to_str().unwrap() && disk_i.partition_type.is_some() { + is_partition = true; + } + } + + let mut to_wipe: Vec = Vec::new(); + + let partitions_map = disk.partitions()?; + for (_, part_disk) in partitions_map.iter() { + let part_path = part_disk.device_path().unwrap(); + to_wipe.push(part_path.to_path_buf()); + } + + to_wipe.push(disk_path.to_path_buf()); + + println!("Wiping block device {}", disk_path.display()); + + let mut wipefs_command = std::process::Command::new("wipefs"); + wipefs_command.arg("--all").args(&to_wipe); + + proxmox_sys::command::run_command(wipefs_command, None)?; + + let size = disk.size().map(|size| size / 1024 / 1024)?; + let count = size.min(200); + + let mut dd_command = std::process::Command::new("dd"); + let args = [ + "if=/dev/zero", + &format!("of={}", disk_path.display()), + "bs=1M", + "conv=fdatasync", + &format!("count={}", count), + ]; + dd_command.args(args); + + proxmox_sys::command::run_command(dd_command, None)?; + + if is_partition { + change_parttype(&disk, "8300")?; + } + + Ok(()) +} + +pub fn change_parttype(part_disk: &Disk, part_type: &str) -> Result<(), Error> { + let part_path = part_disk.device_path().unwrap(); + if let Ok(stat) = nix::sys::stat::stat(part_path) { + let mut sgdisk_command = std::process::Command::new("sgdisk"); + let major = unsafe { libc::major(stat.st_rdev) }; + let minor = unsafe { libc::minor(stat.st_rdev) }; + let partnum_path = &format!("/sys/dev/block/{}:{}/partition", major, minor); + let partnum: u32 = std::fs::read_to_string(partnum_path)?.trim_end().parse()?; + sgdisk_command.arg(&format!("-t{}:{}", partnum, part_type)); + sgdisk_command.arg(&part_disk.parent().unwrap().device_path().unwrap()); + proxmox_sys::command::run_command(sgdisk_command, None)?; + } + Ok(()) +} + /// Create a single linux partition using the whole available space pub fn create_single_linux_partition(disk: &Disk) -> Result { let disk_path = match disk.device_path() { @@ -1136,6 +1208,22 @@ pub fn complete_disk_name(_arg: &str, _param: &HashMap) -> Vec) -> Vec { + let dir = match proxmox_sys::fs::scan_subdir( + libc::AT_FDCWD, + "/sys/class/block", + &BLOCKDEVICE_PARTITION_NAME_REGEX, + ) { + Ok(dir) => dir, + Err(_) => return vec![], + }; + + dir.flatten() + .map(|item| item.file_name().to_str().unwrap().to_string()) + .collect() +} + /// Read the FS UUID (parse blkid output) /// /// Note: Calling blkid is more reliable than using the udev ID_FS_UUID property. -- 2.39.2