mirror of
https://github.com/sogou/workflow.git
synced 2026-10-22 23:39:54 +08:00
holmes1412
85 lines
2.0 KiB
C++
85 lines
2.0 KiB
C++
/*
|
|
Copyright (c) 2021 Sogou, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
|
|
Author: Li Yingxin (liyingxin@sogou-inc.com)
|
|
*/
|
|
|
|
#ifndef _WEBSOCKETMESSAGE_H_
|
|
#define _WEBSOCKETMESSAGE_H_
|
|
|
|
#include <string.h>
|
|
#include <string>
|
|
#include <stdint.h>
|
|
#include "ProtocolMessage.h"
|
|
#include "websocket_parser.h"
|
|
|
|
namespace protocol
|
|
{
|
|
|
|
#define WS_HANDSHAKE_TIMEOUT 10 * 1000
|
|
|
|
class WebSocketFrame : public ProtocolMessage
|
|
{
|
|
public:
|
|
bool set_opcode(int opcode);
|
|
int get_opcode() const;
|
|
|
|
void set_masking_key(uint32_t masking_key);
|
|
|
|
bool set_text_data(const char *data);
|
|
bool set_text_data(const char *data, size_t size, bool fin);
|
|
|
|
bool set_binary_data(const char *data, size_t size);
|
|
bool set_binary_data(const char *data, size_t size, bool fin);
|
|
|
|
bool get_data(const char **data, size_t *size) const;
|
|
|
|
bool finished() const;
|
|
|
|
public:
|
|
void set_client() { this->parser->is_server = 0; }
|
|
void set_server() { this->parser->is_server = 1; }
|
|
const websocket_parser_t *get_parser() { return this->parser; }
|
|
bool set_data(const websocket_parser_t *parser);
|
|
uint32_t get_masking_key() const;
|
|
|
|
private:
|
|
websocket_parser_t *parser;
|
|
|
|
protected:
|
|
virtual int encode(struct iovec vectors[], int max);
|
|
virtual int append(const void *buf, size_t *size);
|
|
|
|
public:
|
|
WebSocketFrame() : parser(new websocket_parser_t)
|
|
{
|
|
websocket_parser_init(this->parser);
|
|
}
|
|
|
|
virtual ~WebSocketFrame()
|
|
{
|
|
websocket_parser_deinit(this->parser);
|
|
delete this->parser;
|
|
}
|
|
|
|
WebSocketFrame(WebSocketFrame&& msg);
|
|
WebSocketFrame& operator = (WebSocketFrame&& msg);
|
|
};
|
|
|
|
} // end namespace protocol
|
|
|
|
#endif
|
|
|