9a0544
From eb917d346bc8592924c5f6566b01841176c53c8c Mon Sep 17 00:00:00 2001
9a0544
From: Tomas Bzatek <tbzatek@redhat.com>
9a0544
Date: Mon, 22 Aug 2022 16:27:11 +0200
9a0544
Subject: [PATCH] udiskslinuxblock: Only permit ATA Secure Erase during
9a0544
 Format() on a whole block device
9a0544
9a0544
ATA Secure Erase requested as an option to the Format() method call used
9a0544
to perform the actual erase on a whole drive object it looked up. When
9a0544
Format() was called on a partition, this led to data loss on a whole drive.
9a0544
This commit adds a safeguard to check that the Format() is requested
9a0544
on a whole block device.
9a0544
9a0544
Severity of this issue was slightly lowered by a failure to submit
9a0544
the ATA Secure erase command in case some filesystem was mounted
9a0544
at that point.
9a0544
---
9a0544
 src/udiskslinuxblock.c | 16 ++++++++++++++++
9a0544
 1 file changed, 16 insertions(+)
9a0544
9a0544
diff --git a/src/udiskslinuxblock.c b/src/udiskslinuxblock.c
9a0544
index d1da94edf..db0ed2bf6 100644
9a0544
--- a/src/udiskslinuxblock.c
9a0544
+++ b/src/udiskslinuxblock.c
9a0544
@@ -2354,6 +2354,7 @@ erase_ata_device (UDisksBlock   *block,
9a0544
 {
9a0544
   gboolean ret = FALSE;
9a0544
   UDisksObject *drive_object = NULL;
9a0544
+  UDisksLinuxBlockObject *block_object = NULL;
9a0544
   UDisksDriveAta *ata = NULL;
9a0544
 
9a0544
   drive_object = udisks_daemon_find_object (daemon, udisks_block_get_drive (block));
9a0544
@@ -2369,6 +2370,20 @@ erase_ata_device (UDisksBlock   *block,
9a0544
       goto out;
9a0544
     }
9a0544
 
9a0544
+  /* Reverse check to ensure we're erasing whole block device and not a partition */
9a0544
+  block_object = udisks_linux_drive_object_get_block (UDISKS_LINUX_DRIVE_OBJECT (drive_object), FALSE /* get_hw */);
9a0544
+  if (block_object == NULL)
9a0544
+    {
9a0544
+      g_set_error (error, UDISKS_ERROR, UDISKS_ERROR_FAILED, "Couldn't find a block device for the drive to erase");
9a0544
+      goto out;
9a0544
+    }
9a0544
+  if (g_strcmp0 (g_dbus_object_get_object_path (G_DBUS_OBJECT (object)),
9a0544
+                 g_dbus_object_get_object_path (G_DBUS_OBJECT (block_object))) != 0)
9a0544
+    {
9a0544
+      g_set_error (error, UDISKS_ERROR, UDISKS_ERROR_FAILED, "ATA secure erase needs to be performed on a whole block device");
9a0544
+      goto out;
9a0544
+    }
9a0544
+
9a0544
   /* sleep a tiny bit here to avoid the secure erase code racing with
9a0544
    * programs spawned by udev
9a0544
    */
9a0544
@@ -2382,6 +2397,7 @@ erase_ata_device (UDisksBlock   *block,
9a0544
  out:
9a0544
   g_clear_object (&ata;;
9a0544
   g_clear_object (&drive_object);
9a0544
+  g_clear_object (&block_object);
9a0544
   return ret;
9a0544
 }
9a0544