2 files changed, 4 insertions, 1 deletions
diff --git a/edify/include/edify/updater_interface.h b/edify/include/edify/updater_interface.h
index a4d581eec..aa977e3c8 100644
--- a/edify/include/edify/updater_interface.h
+++ b/edify/include/edify/updater_interface.h
@@ -44,4 +44,5 @@ class UpdaterInterface {
   virtual ZipArchiveHandle GetPackageHandle() const = 0;
   virtual std::string GetResult() const = 0;
   virtual uint8_t* GetMappedPackageAddress() const = 0;
+  virtual size_t GetMappedPackageLength() const = 0;
 };
diff --git a/updater/include/updater/updater.h b/updater/include/updater/updater.h
index 08816bf36..8676b6038 100644
--- a/updater/include/updater/updater.h
+++ b/updater/include/updater/updater.h
@@ -62,10 +62,12 @@ class Updater : public UpdaterInterface {
   std::string GetResult() const override {
     return result_;
   }
-
   uint8_t* GetMappedPackageAddress() const override {
     return mapped_package_.addr;
   }
+  size_t GetMappedPackageLength() const override {
+    return mapped_package_.length;
+  }
 
  private:
   friend class UpdaterTestBase;