commit a6e3b99df73312c465b4b174b167bb785e886a71
author John Edward Broadbent <jebr@google.com> Thu Mar 17 14:33:15 2022 -0700
committer John Edward Broadbent <jebr@google.com> Tue Mar 22 18:09:04 2022 -0700
tree 7cf91ed1dcdf1fa6c20a9e261984b66bfb52313b
parent 86dfb244ee46ff5199c166da7f856e7b6b16d614

Refactor findsize into util away from erase

At first the size was only needed for erase, now it is needed for the
drives interface. The code needed to be refactored to make drive
interface implementation possible.

Signed-off-by: John Edward Broadbent <jebr@google.com>
Change-Id: I0a23ee23a0de3fd89a9e776b4854e8da9a8ff2e4

include/verifyDriveGeometry.hpp

diff --git a/include/verifyDriveGeometry.hpp b/include/verifyDriveGeometry.hpp
index 457297b..ea54998 100644
--- a/include/verifyDriveGeometry.hpp
+++ b/include/verifyDriveGeometry.hpp
@@ -1,6 +1,7 @@
 #pragma once
 
 #include "erase.hpp"
+#include "util.hpp"
 
 #include <string_view>
 
@@ -22,6 +23,10 @@
      *
      *  @param[in] bytes - Size of the block device
      */
+    void geometryOkay()
+    {
+        geometryOkay(util::Util::findSizeOfBlockDevice(devPath));
+    }
     void geometryOkay(uint64_t bytes);
 };