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);
|
2018-09-27 16:03:53 +08:00
|
|
|
|
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){
|
2018-09-27 16:03:53 +08:00
|
|
|
|
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);
|
|
|
|
|
}
|
|
|
|
|
|
2019-08-13 12:00:21 +08:00
|
|
|
|
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;
|
2018-09-27 16:03:53 +08:00
|
|
|
|
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);
|
|
|
|
|
|
2018-09-27 16:03:53 +08:00
|
|
|
|
auto mask_flag = (header._mask_flag && header._mask.size() >= 4);
|
|
|
|
|
byte = mask_flag << 7;
|
2018-09-21 18:48:35 +08:00
|
|
|
|
|
2018-09-27 16:03:53 +08:00
|
|
|
|
if(len < 126){
|
|
|
|
|
byte |= len;
|
2018-09-21 18:48:35 +08:00
|
|
|
|
ret.push_back(byte);
|
2018-09-27 16:03:53 +08:00
|
|
|
|
}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);
|
|
|
|
|
|
2018-09-27 16:03:53 +08:00
|
|
|
|
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);
|
|
|
|
|
}
|
2018-09-27 16:03:53 +08:00
|
|
|
|
if(mask_flag){
|
|
|
|
|
ret.append((char *)header._mask.data(),4);
|
2018-09-21 18:48:35 +08:00
|
|
|
|
}
|
|
|
|
|
|
2019-08-13 12:00:21 +08:00
|
|
|
|
onWebSocketEncodeData(std::make_shared<BufferString>(std::move(ret)));
|
2018-09-21 18:48:35 +08:00
|
|
|
|
|
2018-09-27 16:03:53 +08:00
|
|
|
|
if(len > 0){
|
|
|
|
|
if(mask_flag){
|
2019-08-13 12:00:21 +08:00
|
|
|
|
uint8_t *ptr = (uint8_t*)buffer->data();
|
2018-09-27 16:03:53 +08:00
|
|
|
|
for(int i = 0; i < len ; ++i,++ptr){
|
|
|
|
|
*(ptr) ^= header._mask[i % 4];
|
|
|
|
|
}
|
2018-09-21 18:48:35 +08:00
|
|
|
|
}
|
2019-08-13 12:00:21 +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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|