ZLMediaKit/src/Http/WebSocketSplitter.cpp

219 lines
7.1 KiB
C++
Raw Normal View History

2018-09-25 09:55:41 +08:00
/*
* MIT License
*
2019-05-08 15:40:07 +08:00
* Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
2018-09-25 09:55:41 +08:00
*
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
2018-09-21 16:11:09 +08:00
#include "WebSocketSplitter.h"
2018-09-21 18:48:35 +08:00
#include <sys/types.h>
2018-11-06 23:17:44 +08:00
#if !defined(_WIN32)
#include <sys/socket.h>
#include <arpa/inet.h>
#endif //!defined(_WIN32)
2018-09-21 16:11:09 +08:00
#include "Util/logger.h"
#include "Util/util.h"
2018-10-24 17:17:55 +08:00
using namespace toolkit;
2018-09-25 09:55:41 +08:00
2018-10-24 17:17:55 +08:00
namespace mediakit {
2018-09-25 09:55:41 +08:00
2018-09-21 16:11:09 +08:00
/**
*
0 1 2 3
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
+-+-+-+-+-------+-+-------------+-------------------------------+
|F|R|R|R| opcode|M| Payload len | Extended payload length |
|I|S|S|S| (4) |A| (7) | (16/64) |
|N|V|V|V| |S| | (if payload len==126/127) |
| |1|2|3| |K| | |
+-+-+-+-+-------+-+-------------+ - - - - - - - - - - - - - - - +
| Extended payload length continued, if payload len == 127 |
+ - - - - - - - - - - - - - - - +-------------------------------+
| |Masking-key, if MASK set to 1 |
+-------------------------------+-------------------------------+
| Masking-key (continued) | Payload Data |
+-------------------------------- - - - - - - - - - - - - - - - +
: Payload Data continued ... :
+ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - +
| Payload Data continued ... |
+---------------------------------------------------------------+
*/
#define CHECK_LEN(size) \
do{ \
if(len - (ptr - data) < size){ \
if(_remain_data.empty()){ \
_remain_data.assign((char *)data,len); \
} \
return ; \
} \
}while(0) \
void WebSocketSplitter::decode(uint8_t *data,uint64_t len) {
uint8_t *ptr = data;
if(!_got_header) {
//还没有获取数据头
if(!_remain_data.empty()){
_remain_data.append((char *)data,len);
data = ptr = (uint8_t *)_remain_data.data();
len = _remain_data.size();
}
begin_decode:
CHECK_LEN(1);
_fin = (*ptr & 0x80) >> 7;
_reserved = (*ptr & 0x70) >> 4;
_opcode = (WebSocketHeader::Type) (*ptr & 0x0F);
ptr += 1;
CHECK_LEN(1);
_mask_flag = (*ptr & 0x80) >> 7;
_playload_len = (*ptr & 0x7F);
ptr += 1;
if (_playload_len == 126) {
CHECK_LEN(2);
_playload_len = (*ptr << 8) | *(ptr + 1);
ptr += 2;
} else if (_playload_len == 127) {
CHECK_LEN(8);
_playload_len = ((uint64_t) ptr[0] << (8 * 7)) |
((uint64_t) ptr[1] << (8 * 6)) |
((uint64_t) ptr[2] << (8 * 5)) |
((uint64_t) ptr[3] << (8 * 4)) |
((uint64_t) ptr[4] << (8 * 3)) |
((uint64_t) ptr[5] << (8 * 2)) |
((uint64_t) ptr[6] << (8 * 1)) |
((uint64_t) ptr[7] << (8 * 0));
ptr += 8;
}
if (_mask_flag) {
CHECK_LEN(4);
_mask.assign(ptr, ptr + 4);
ptr += 4;
}
_got_header = true;
_mask_offset = 0;
_playload_offset = 0;
2018-09-21 18:48:35 +08:00
onWebSocketDecodeHeader(*this);
if(_playload_len == 0){
onWebSocketDecodeComplete(*this);
}
2018-09-21 16:11:09 +08:00
}
2018-09-23 00:55:00 +08:00
//进入后面逻辑代表已经获取到了webSocket协议头
2018-09-21 16:11:09 +08:00
uint64_t remain = len - (ptr - data);
if(remain > 0){
uint64_t playload_slice_len = remain;
if(playload_slice_len + _playload_offset > _playload_len){
playload_slice_len = _playload_len - _playload_offset;
}
_playload_offset += playload_slice_len;
2018-09-23 21:10:17 +08:00
onPlayloadData(ptr,playload_slice_len);
2018-09-21 16:11:09 +08:00
if(_playload_offset == _playload_len){
onWebSocketDecodeComplete(*this);
2018-09-21 16:11:09 +08:00
//这是下一个包
2018-09-23 00:55:00 +08:00
remain -= playload_slice_len;
ptr += playload_slice_len;
_got_header = false;
if(remain > 0){
//剩余数据是下一个包,把它的数据放置在缓存中
2018-09-28 21:18:58 +08:00
string str((char *)ptr,remain);
_remain_data = str;
2018-09-21 16:11:09 +08:00
2018-09-22 21:37:11 +08:00
data = ptr = (uint8_t *)_remain_data.data();
len = _remain_data.size();
2018-09-21 16:11:09 +08:00
goto begin_decode;
}
}
}
_remain_data.clear();
}
void WebSocketSplitter::onPlayloadData(uint8_t *ptr, uint64_t len) {
if(_mask_flag){
for(int i = 0; i < len ; ++i,++ptr){
*(ptr) ^= _mask[(i + _mask_offset) % 4];
}
_mask_offset = (_mask_offset + len) % 4;
}
2018-09-21 18:48:35 +08:00
onWebSocketDecodePlayload(*this, _mask_flag ? ptr - len : ptr, len, _playload_offset);
}
void WebSocketSplitter::encode(const WebSocketHeader &header,const Buffer::Ptr &buffer) {
2018-09-21 18:48:35 +08:00
string ret;
2019-08-27 11:54:27 +08:00
uint64_t len = buffer ? buffer->size() : 0;
uint8_t byte = header._fin << 7 | ((header._reserved & 0x07) << 4) | (header._opcode & 0x0F) ;
2018-09-21 18:48:35 +08:00
ret.push_back(byte);
auto mask_flag = (header._mask_flag && header._mask.size() >= 4);
byte = mask_flag << 7;
2018-09-21 18:48:35 +08:00
if(len < 126){
byte |= len;
2018-09-21 18:48:35 +08:00
ret.push_back(byte);
}else if(len <= 0xFFFF){
2018-09-21 18:48:35 +08:00
byte |= 126;
ret.push_back(byte);
2018-09-29 14:54:49 +08:00
auto len_low = htons(len);
ret.append((char *)&len_low,2);
2018-09-21 18:48:35 +08:00
}else{
byte |= 127;
ret.push_back(byte);
uint32_t len_high = htonl(len >> 32) ;
uint32_t len_low = htonl(len & 0xFFFFFFFF);
2018-09-21 18:48:35 +08:00
ret.append((char *)&len_high,4);
ret.append((char *)&len_low,4);
}
if(mask_flag){
ret.append((char *)header._mask.data(),4);
2018-09-21 18:48:35 +08:00
}
onWebSocketEncodeData(std::make_shared<BufferString>(std::move(ret)));
2018-09-21 18:48:35 +08:00
if(len > 0){
if(mask_flag){
uint8_t *ptr = (uint8_t*)buffer->data();
for(int i = 0; i < len ; ++i,++ptr){
*(ptr) ^= header._mask[i % 4];
}
2018-09-21 18:48:35 +08:00
}
onWebSocketEncodeData(buffer);
2018-09-21 18:48:35 +08:00
}
2018-09-21 16:11:09 +08:00
}
2018-09-21 18:48:35 +08:00
2018-10-24 17:17:55 +08:00
} /* namespace mediakit */
2018-09-21 18:48:35 +08:00