Cleanup mounted resources after application crash.

On initial state application cleans mounted resources which
were allocated by user befor application crashed.
Without that busy slot is not avaialble anymore for user.

Tested:
1. Mount CIFS share.
2. Send terminate signal to virtual-media (kill -9).
3. Mount CIFS share on the same slot as during step 1.

Change-Id: I7088e94832fb7bec171a56f73bd66cd29e9b246f
Signed-off-by: Krzysztof Richert <krzysztof.richert@intel.com>

1 files changed, 45 insertions, 0 deletions

diff --git a/src/state/initial_state.hpp b/src/state/initial_state.hpp
index 7fcc10b..02fe479 100644
--- a/src/state/initial_state.hpp
+++ b/src/state/initial_state.hpp
@@ -25,6 +25,10 @@ struct InitialState : public BasicStateT<InitialState>
                                                 "Legacy mode is not supported");
         }
 #endif
+        if (isLegacy)
+        {
+            cleanUpMountPoint();
+        }
         addMountPointInterface(event);
         addProcessInterface(event);
         addServiceInterface(event, isLegacy);
@@ -79,6 +83,47 @@ struct InitialState : public BasicStateT<InitialState>
         processIface->initialize();
     }
 
+    void cleanUpMountPoint()
+    {
+        if (UsbGadget::isConfigured(std::string(machine.getName())))
+        {
+            int result = UsbGadget::configure(std::string(machine.getName()),
+                                              machine.getConfig().nbdDevice,
+                                              StateChange::removed);
+            LogMsg(Logger::Info, "UsbGadget cleanup");
+
+            if (result != 0)
+            {
+                LogMsg(Logger::Critical, machine.getName(),
+                       "Some serious failure happened! Cleanup failed.");
+            }
+        }
+
+        auto localFile = std::filesystem::temp_directory_path() /
+                         std::string(machine.getName());
+
+        if (fs::exists(localFile))
+        {
+            if (0 == ::umount2(localFile.c_str(), MNT_FORCE))
+            {
+                LogMsg(Logger::Info, "Cleanup directory ", localFile);
+                std::error_code ec;
+                if (!std::filesystem::remove(localFile, ec))
+                {
+                    LogMsg(Logger::Error, ec,
+                           "Cleanup failed - unable to remove directory ",
+                           localFile);
+                }
+            }
+            else
+            {
+                LogMsg(Logger::Error,
+                       "Cleanup failed - unable to unmount directory ",
+                       localFile);
+            }
+        }
+    }
+
     void addMountPointInterface(const RegisterDbusEvent& event)
     {
         std::string objPath = getObjectPath(machine);