1 files changed, 56 insertions, 0 deletions
diff --git a/dom/system/gonk/nsVolumeMountLock.h b/dom/system/gonk/nsVolumeMountLock.h
new file mode 100644
index 000000000..caf5b2ad5
--- /dev/null
+++ b/dom/system/gonk/nsVolumeMountLock.h
@@ -0,0 +1,56 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef mozilla_system_nsvolumemountlock_h__
+#define mozilla_system_nsvolumemountlock_h__
+
+#include "nsIVolumeMountLock.h"
+
+#include "mozilla/dom/WakeLock.h"
+#include "nsIObserver.h"
+#include "nsString.h"
+#include "nsTArray.h"
+#include "nsWeakReference.h"
+
+class nsIVolume;
+
+namespace mozilla {
+namespace system {
+
+/* The VolumeMountLock is designed so that it can be used in the Child or
+ * Parent process. While the VolumeMountLock object exists, then the
+ * VolumeManager/AutoMounter will prevent a mounted volume from being
+ * shared with the PC.
+ */
+
+class nsVolumeMountLock final : public nsIVolumeMountLock,
+                                public nsIObserver,
+                                public nsSupportsWeakReference
+{
+public:
+  NS_DECL_ISUPPORTS
+  NS_DECL_NSIOBSERVER
+  NS_DECL_NSIVOLUMEMOUNTLOCK
+
+  static already_AddRefed<nsVolumeMountLock> Create(const nsAString& volumeName);
+
+  const nsString& VolumeName() const  { return mVolumeName; }
+
+private:
+  nsVolumeMountLock(const nsAString& aVolumeName);
+  ~nsVolumeMountLock();
+
+  nsresult Init();
+  nsresult Lock(nsIVolume* aVolume);
+
+  RefPtr<dom::WakeLock>  mWakeLock;
+  nsString                 mVolumeName;
+  int32_t                  mVolumeGeneration;
+  bool                     mUnlocked;
+};
+
+} // namespace system
+} // namespace mozilla
+
+#endif  // mozilla_system_nsvolumemountlock_h__