1 #ifndef SRC_WEBSOCKETHANDLER_HPP_

2 #define SRC_WEBSOCKETHANDLER_HPP_

3 

4 #include <Arduino.h>

5 #include <lwip/def.h>

6 

7 #include <string>

8 #undef min

9 #undef max

10 

11 #include <sstream>

12 

13 #include "HTTPSServerConstants.hpp"

14 #include "ConnectionContext.hpp"

15 #include "WebsocketInputStreambuf.hpp"

16 

18 

19 

21 {

22 

23  uint8_t opCode : 4;

24  uint8_t rsv3 : 1;

25  uint8_t rsv2 : 1;

26  uint8_t rsv1 : 1;

27  uint8_t fin : 1;

28 

29 

30  uint8_t len : 7;

31  uint8_t mask : 1;

32 };

33 

35 {

36 public:

37 

38  static const int OPCODE_CONTINUE = 0x00;

39  static const int OPCODE_TEXT = 0x01;

40  static const int OPCODE_BINARY = 0x02;

41  static const int OPCODE_CLOSE = 0x08;

42  static const int OPCODE_PING = 0x09;

43  static const int OPCODE_PONG = 0x0a;

44 

45  static const uint16_t CLOSE_NORMAL_CLOSURE = 1000;

46  static const uint16_t CLOSE_GOING_AWAY = 1001;

47  static const uint16_t CLOSE_PROTOCOL_ERROR = 1002;

48  static const uint16_t CLOSE_CANNOT_ACCEPT = 1003;

49  static const uint16_t CLOSE_NO_STATUS_CODE = 1005;

50  static const uint16_t CLOSE_CLOSED_ABNORMALLY = 1006;

51  static const uint16_t CLOSE_NOT_CONSISTENT = 1007;

52  static const uint16_t CLOSE_VIOLATED_POLICY = 1008;

53  static const uint16_t CLOSE_TOO_BIG = 1009;

54  static const uint16_t CLOSE_NO_EXTENSION = 1010;

55  static const uint16_t CLOSE_UNEXPECTED_CONDITION = 1011;

56  static const uint16_t CLOSE_SERVICE_RESTART = 1012;

57  static const uint16_t CLOSE_TRY_AGAIN_LATER = 1013;

58  static const uint16_t CLOSE_TLS_HANDSHAKE_FAILURE = 1015;

59 

60  static const uint8_t SEND_TYPE_BINARY = 0x01;

61  static const uint8_t SEND_TYPE_TEXT = 0x02;

62 

65  virtual void onClose();

67  virtual void onError(std::string error);

68 

69  void close(uint16_t status = CLOSE_NORMAL_CLOSURE, std::string message = "");

70  void send(std::string data, uint8_t sendType = SEND_TYPE_BINARY);

71  void send(uint8_t *data, uint16_t length, uint8_t sendType = SEND_TYPE_BINARY);

72  bool closed();

73 

74  void loop();

76 

77 private:

78  int read();

79 

81  bool _receivedClose;

82  bool _sentClose;

83 };

84 

85 }

86 

87 #endif