alloc.hpp 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. /*
  2. * Copyright (c) 2014, Peter Thorson. 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 are met:
  6. * * Redistributions of source code must retain the above copyright
  7. * notice, this list of conditions and the following disclaimer.
  8. * * Redistributions in binary form must reproduce the above copyright
  9. * notice, this list of conditions and the following disclaimer in the
  10. * documentation and/or other materials provided with the distribution.
  11. * * Neither the name of the WebSocket++ Project nor the
  12. * names of its contributors may be used to endorse or promote products
  13. * derived from this software without specific prior written permission.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18. * ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
  19. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  20. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  21. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  22. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  24. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. *
  26. */
  27. #ifndef WEBSOCKETPP_MESSAGE_BUFFER_ALLOC_HPP
  28. #define WEBSOCKETPP_MESSAGE_BUFFER_ALLOC_HPP
  29. #include <websocketpp/common/memory.hpp>
  30. #include <websocketpp/frame.hpp>
  31. namespace websocketpp {
  32. namespace message_buffer {
  33. namespace alloc {
  34. /// A connection message manager that allocates a new message for each
  35. /// request.
  36. template <typename message>
  37. class con_msg_manager
  38. : public lib::enable_shared_from_this<con_msg_manager<message> >
  39. {
  40. public:
  41. typedef con_msg_manager<message> type;
  42. typedef lib::shared_ptr<con_msg_manager> ptr;
  43. typedef lib::weak_ptr<con_msg_manager> weak_ptr;
  44. typedef typename message::ptr message_ptr;
  45. /// Get an empty message buffer
  46. /**
  47. * @return A shared pointer to an empty new message
  48. */
  49. message_ptr get_message() {
  50. return message_ptr(lib::make_shared<message>(type::shared_from_this()));
  51. }
  52. /// Get a message buffer with specified size and opcode
  53. /**
  54. * @param op The opcode to use
  55. * @param size Minimum size in bytes to request for the message payload.
  56. *
  57. * @return A shared pointer to a new message with specified size.
  58. */
  59. message_ptr get_message(frame::opcode::value op,size_t size) {
  60. return message_ptr(lib::make_shared<message>(type::shared_from_this(),op,size));
  61. }
  62. /// Recycle a message
  63. /**
  64. * This method shouldn't be called. If it is, return false to indicate an
  65. * error. The rest of the method recycle chain should notice this and free
  66. * the memory.
  67. *
  68. * @param msg The message to be recycled.
  69. *
  70. * @return true if the message was successfully recycled, false otherwse.
  71. */
  72. bool recycle(message *) {
  73. return false;
  74. }
  75. };
  76. /// An endpoint message manager that allocates a new manager for each
  77. /// connection.
  78. template <typename con_msg_manager>
  79. class endpoint_msg_manager {
  80. public:
  81. typedef typename con_msg_manager::ptr con_msg_man_ptr;
  82. /// Get a pointer to a connection message manager
  83. /**
  84. * @return A pointer to the requested connection message manager.
  85. */
  86. con_msg_man_ptr get_manager() const {
  87. return con_msg_man_ptr(lib::make_shared<con_msg_manager>());
  88. }
  89. };
  90. } // namespace alloc
  91. } // namespace message_buffer
  92. } // namespace websocketpp
  93. #endif // WEBSOCKETPP_MESSAGE_BUFFER_ALLOC_HPP