commit 6d7735d7dca8a714d1d290805e638da742e22480
author Patrick Venture <venture@google.com> Fri Jun 21 10:03:19 2019 -0700
committer Patrick Venture <venture@google.com> Fri Jun 21 10:37:30 2019 -0700
tree e5ea7d07906dbd5db2df8faa1de3519a369f1786
parent 75c0c2740bc5ac3727553e65ea89aafbcb1d12e9

bmc: add preparation step before data is received

Add a preparation systemd trigger event before data is received.  On
systems under memory pressure, this'll trigger a service that can do
things like flush caches.

Signed-off-by: Patrick Venture <venture@google.com>
Change-Id: I175177f4a91b58d9f163098572a9d2614002b718

firmware_handler.hpp

diff --git a/firmware_handler.hpp b/firmware_handler.hpp
index de37e20..8c7cb04 100644
--- a/firmware_handler.hpp
+++ b/firmware_handler.hpp
@@ -104,6 +104,7 @@
         CreateFirmwareBlobHandler(
             const std::vector<HandlerPack>& firmwares,
             const std::vector<DataHandlerPack>& transports,
+            std::unique_ptr<TriggerableActionInterface> preparation,
             std::unique_ptr<TriggerableActionInterface> verification,
             std::unique_ptr<TriggerableActionInterface> update);
 
@@ -121,13 +122,14 @@
         const std::vector<HandlerPack>& firmwares,
         const std::vector<std::string>& blobs,
         const std::vector<DataHandlerPack>& transports, std::uint16_t bitmask,
+        std::unique_ptr<TriggerableActionInterface> preparation,
         std::unique_ptr<TriggerableActionInterface> verification,
         std::unique_ptr<TriggerableActionInterface> update) :
         handlers(firmwares),
         blobIDs(blobs), transports(transports), bitmask(bitmask),
         activeImage(activeImageBlobId), activeHash(activeHashBlobId),
         verifyImage(verifyBlobId), updateImage(updateBlobId), lookup(),
-        state(UpdateState::notYetStarted),
+        state(UpdateState::notYetStarted), preparation(std::move(preparation)),
         verification(std::move(verification)), update(std::move(update))
     {
     }
@@ -221,6 +223,12 @@
     /** The firmware update state. */
     UpdateState state;
 
+    /* preparation is triggered once we go into uploadInProgress(), but only
+     * once per full cycle, going back to notYetStarted resets this.
+     */
+    bool preparationTriggered = false;
+    std::unique_ptr<TriggerableActionInterface> preparation;
+
     std::unique_ptr<TriggerableActionInterface> verification;
 
     std::unique_ptr<TriggerableActionInterface> update;