1 files changed, 102 insertions, 0 deletions
diff --git a/Source/WebKit2/UIProcess/ProcessAssertion.h b/Source/WebKit2/UIProcess/ProcessAssertion.h
new file mode 100644
index 000000000..e054affe8
--- /dev/null
+++ b/Source/WebKit2/UIProcess/ProcessAssertion.h
@@ -0,0 +1,102 @@
+/*
+ * Copyright (C) 2014 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
+ * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
+ * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+ * THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#ifndef ProcessAssertion_h
+#define ProcessAssertion_h
+
+#include <functional>
+#include <wtf/Function.h>
+
+#if PLATFORM(IOS) && !PLATFORM(IOS_SIMULATOR)
+#include <wtf/RetainPtr.h>
+#include <wtf/WeakPtr.h>
+OBJC_CLASS BKSProcessAssertion;
+#endif
+
+namespace WebKit {
+    
+enum class AssertionState {
+    Suspended,
+    Background,
+    Foreground
+};
+
+class ProcessAssertionClient {
+public:
+    virtual ~ProcessAssertionClient() { };
+    virtual void assertionWillExpireImminently() = 0;
+};
+
+class ProcessAssertion {
+public:
+    ProcessAssertion(pid_t, AssertionState, Function<void()>&& invalidationCallback = { });
+    ~ProcessAssertion();
+
+    void setClient(ProcessAssertionClient& client) { m_client = &client; }
+    ProcessAssertionClient* client() { return m_client; }
+
+    AssertionState state() const { return m_assertionState; }
+    void setState(AssertionState);
+
+#if PLATFORM(IOS) && !PLATFORM(IOS_SIMULATOR)
+protected:
+    enum class Validity { No, Yes, Unset };
+    Validity validity() const { return m_validity; }
+#endif
+
+private:
+#if PLATFORM(IOS) && !PLATFORM(IOS_SIMULATOR)
+    WeakPtr<ProcessAssertion> createWeakPtr() { return m_weakFactory.createWeakPtr(); }
+    void markAsInvalidated();
+
+    RetainPtr<BKSProcessAssertion> m_assertion;
+    Validity m_validity { Validity::Unset };
+    WeakPtrFactory<ProcessAssertion> m_weakFactory;
+    Function<void()> m_invalidationCallback;
+#endif
+    AssertionState m_assertionState;
+    ProcessAssertionClient* m_client { nullptr };
+};
+    
+class ProcessAndUIAssertion : public ProcessAssertion {
+public:
+    ProcessAndUIAssertion(pid_t, AssertionState);
+    ~ProcessAndUIAssertion();
+
+    void setClient(ProcessAssertionClient&);
+
+    void setState(AssertionState);
+
+#if PLATFORM(IOS) && !PLATFORM(IOS_SIMULATOR)
+private:
+    void updateRunInBackgroundCount();
+
+    bool m_isHoldingBackgroundAssertion { false };
+#endif
+};
+    
+}
+
+#endif // ProcessAssertion_h