WebSocketParser.h 883 B

12345678910111213141516171819202122232425262728293031323334353637
  1. #ifndef HV_WEBSOCKET_PARSER_H_
  2. #define HV_WEBSOCKET_PARSER_H_
  3. #include "hexport.h"
  4. #include <string>
  5. #include <memory>
  6. #include <functional>
  7. enum websocket_parser_state {
  8. WS_FRAME_BEGIN,
  9. WS_FRAME_HEADER,
  10. WS_FRAME_BODY,
  11. WS_FRAME_END,
  12. WS_FRAME_FIN,
  13. };
  14. struct websocket_parser_settings;
  15. struct websocket_parser;
  16. class HV_EXPORT WebSocketParser {
  17. public:
  18. static websocket_parser_settings cbs;
  19. websocket_parser* parser;
  20. websocket_parser_state state;
  21. int opcode;
  22. std::string message;
  23. std::function<void(int opcode, const std::string& msg)> onMessage;
  24. WebSocketParser();
  25. ~WebSocketParser();
  26. int FeedRecvData(const char* data, size_t len);
  27. };
  28. typedef std::shared_ptr<WebSocketParser> WebSocketParserPtr;
  29. #endif // HV_WEBSOCKET_PARSER_H_