ConnectionStack.h 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. /*
  2. * Copyright (C) 2011, 2012 Apple Inc. All rights reserved.
  3. *
  4. * Redistribution and use in source and binary forms, with or without
  5. * modification, are permitted provided that the following conditions
  6. * are met:
  7. * 1. Redistributions of source code must retain the above copyright
  8. * notice, this list of conditions and the following disclaimer.
  9. * 2. Redistributions in binary form must reproduce the above copyright
  10. * notice, this list of conditions and the following disclaimer in the
  11. * documentation and/or other materials provided with the distribution.
  12. *
  13. * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15. * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16. * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17. * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23. * THE POSSIBILITY OF SUCH DAMAGE.
  24. */
  25. #ifndef ConnectionStack_h
  26. #define ConnectionStack_h
  27. #include <wtf/Vector.h>
  28. namespace CoreIPC {
  29. class Connection;
  30. }
  31. namespace WebKit {
  32. class ConnectionStack {
  33. public:
  34. static ConnectionStack& shared();
  35. CoreIPC::Connection* current()
  36. {
  37. return m_connectionStack.last();
  38. }
  39. class CurrentConnectionPusher {
  40. public:
  41. CurrentConnectionPusher(ConnectionStack& connectionStack, CoreIPC::Connection* connection)
  42. : m_connectionStack(connectionStack)
  43. #if !ASSERT_DISABLED
  44. , m_connection(connection)
  45. #endif
  46. {
  47. m_connectionStack.m_connectionStack.append(connection);
  48. }
  49. ~CurrentConnectionPusher()
  50. {
  51. ASSERT(m_connectionStack.current() == m_connection);
  52. m_connectionStack.m_connectionStack.removeLast();
  53. }
  54. private:
  55. ConnectionStack& m_connectionStack;
  56. #if !ASSERT_DISABLED
  57. CoreIPC::Connection* m_connection;
  58. #endif
  59. };
  60. private:
  61. // It's OK for these to be weak pointers because we only push object on the stack
  62. // from within didReceiveMessage and didReceiveSyncMessage and the Connection objects are
  63. // already ref'd for the duration of those functions.
  64. Vector<CoreIPC::Connection*, 4> m_connectionStack;
  65. };
  66. } // namespace WebKit
  67. #endif // ConnectionStack_h