initial commit

2026-04-29 15:37:57 +00:00 · 2011-09-07 20:27:29 -05:00
commit fa73c478e9
34 changed files with 3581 additions and 0 deletions
--- a/src/websocket_frame.hpp
+++ b/src/websocket_frame.hpp
@@ -0,0 +1,113 @@
+#ifndef WEBSOCKET_FRAME_HPP
+#define WEBSOCKET_FRAME_HPP
+
+#include "network_utilities.hpp"
+
+#include <string>
+#include <vector>
+#include <cstring>
+
+namespace websocketpp {
+
+class frame {
+public:	
+	enum opcode_s {
+		CONTINUATION_FRAME = 0x00,
+		TEXT_FRAME = 0x01,
+		BINARY_FRAME = 0x02,
+		CONNECTION_CLOSE = 0x08,
+		PING = 0x09,
+		PONG = 0x0A
+	};
+	
+	typedef enum opcode_s opcode;
+	
+	static const uint8_t MAX_FRAME_OPCODE = 0x07;
+	
+	// basic payload byte flags
+	static const uint8_t BPB0_OPCODE = 0x0F;
+	static const uint8_t BPB0_RSV3 = 0x10;
+	static const uint8_t BPB0_RSV2 = 0x20;
+	static const uint8_t BPB0_RSV1 = 0x40;
+	static const uint8_t BPB0_FIN = 0x80;
+	static const uint8_t BPB1_PAYLOAD = 0x7F;
+	static const uint8_t BPB1_MASK = 0x80;
+	
+	static const uint8_t BASIC_PAYLOAD_LIMIT = 0x7D; // 125
+	static const uint8_t BASIC_PAYLOAD_16BIT_CODE = 0x7E; // 126
+	static const uint16_t PAYLOAD_16BIT_LIMIT = 0xFFFF; // 2^16, 65535 
+	static const uint8_t BASIC_PAYLOAD_64BIT_CODE = 0x7F; // 127
+	static const uint64_t PAYLOAD_64BIT_LIMIT = 0x7FFFFFFFFFFFFFFF; // 2^63
+	
+	static const unsigned int BASIC_HEADER_LENGTH = 2;		
+	static const unsigned int MAX_HEADER_LENGTH = 14;
+	static const uint8_t extended_header_length = 12;
+	static const uint64_t max_payload_size = 100000000; // 100MB
+	
+	// create an empty frame for writing into
+	frame() {
+		// not sure if these are necessary with c++ but putting in just in case
+		memset(m_header,0,MAX_HEADER_LENGTH);
+	}
+	
+	// get pointers to underlying buffers
+	char* get_header();
+	char* get_extended_header();
+	unsigned int get_header_len() const;
+	
+	char* get_masking_key();
+	
+	// get and set header bits
+	bool get_fin() const;
+	void set_fin(bool fin);
+	
+	bool get_rsv1() const;
+	void set_rsv1(bool b);
+	
+	bool get_rsv2() const;
+	void set_rsv2(bool b);
+	
+	bool get_rsv3() const;
+	void set_rsv3(bool b);
+	
+	opcode get_opcode() const;
+	void set_opcode(opcode op);
+	
+	bool get_masked() const;
+	void set_masked(bool masked);
+	
+	uint8_t get_basic_size() const;
+	size_t get_payload_size() const;
+	
+	std::vector<unsigned char> &get_payload();
+	
+	void set_payload(const std::vector<unsigned char> source);
+	void set_payload(const std::string source);
+	void set_payload_helper(size_t s);
+	
+	bool is_control() const;
+	
+	void print_frame() const;
+	
+	// reads basic header, sets and returns m_header_bits_needed
+	unsigned int process_basic_header();
+	void process_extended_header();
+	void process_payload();
+	void process_payload2(); // experiment with more efficient masking code.
+	
+	bool validate_basic_header() const;
+	
+	void generate_masking_key();
+	void clear_masking_key();
+	
+private:
+	char m_header[MAX_HEADER_LENGTH];
+	std::vector<unsigned char> m_payload;
+	
+	char m_masking_key[4];	
+	unsigned int m_extended_header_bytes_needed;
+};
+
+}
+
+#endif // WEBSOCKET_FRAME_HPP