mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-30 15:36:57 +08:00
219 lines
7.1 KiB
C++
219 lines
7.1 KiB
C++
/*
|
||
* MIT License
|
||
*
|
||
* Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
|
||
*
|
||
* 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.
|
||
*/
|
||
|
||
#include "WebSocketSplitter.h"
|
||
#include <sys/types.h>
|
||
#if !defined(_WIN32)
|
||
#include <sys/socket.h>
|
||
#include <arpa/inet.h>
|
||
#endif //!defined(_WIN32)
|
||
|
||
#include "Util/logger.h"
|
||
#include "Util/util.h"
|
||
using namespace toolkit;
|
||
|
||
namespace mediakit {
|
||
|
||
/**
|
||
*
|
||
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;
|
||
onWebSocketDecodeHeader(*this);
|
||
if(_playload_len == 0){
|
||
onWebSocketDecodeComplete(*this);
|
||
}
|
||
}
|
||
|
||
//进入后面逻辑代表已经获取到了webSocket协议头，
|
||
|
||
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;
|
||
onPlayloadData(ptr,playload_slice_len);
|
||
|
||
if(_playload_offset == _playload_len){
|
||
onWebSocketDecodeComplete(*this);
|
||
|
||
//这是下一个包
|
||
remain -= playload_slice_len;
|
||
ptr += playload_slice_len;
|
||
_got_header = false;
|
||
|
||
if(remain > 0){
|
||
//剩余数据是下一个包，把它的数据放置在缓存中
|
||
string str((char *)ptr,remain);
|
||
_remain_data = str;
|
||
|
||
data = ptr = (uint8_t *)_remain_data.data();
|
||
len = _remain_data.size();
|
||
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;
|
||
}
|
||
onWebSocketDecodePlayload(*this, _mask_flag ? ptr - len : ptr, len, _playload_offset);
|
||
}
|
||
|
||
void WebSocketSplitter::encode(const WebSocketHeader &header,const Buffer::Ptr &buffer) {
|
||
string ret;
|
||
uint64_t len = buffer ? buffer->size() : 0;
|
||
uint8_t byte = header._fin << 7 | ((header._reserved & 0x07) << 4) | (header._opcode & 0x0F) ;
|
||
ret.push_back(byte);
|
||
|
||
auto mask_flag = (header._mask_flag && header._mask.size() >= 4);
|
||
byte = mask_flag << 7;
|
||
|
||
if(len < 126){
|
||
byte |= len;
|
||
ret.push_back(byte);
|
||
}else if(len <= 0xFFFF){
|
||
byte |= 126;
|
||
ret.push_back(byte);
|
||
|
||
auto len_low = htons(len);
|
||
ret.append((char *)&len_low,2);
|
||
}else{
|
||
byte |= 127;
|
||
ret.push_back(byte);
|
||
|
||
uint32_t len_high = htonl(len >> 32) ;
|
||
uint32_t len_low = htonl(len & 0xFFFFFFFF);
|
||
ret.append((char *)&len_high,4);
|
||
ret.append((char *)&len_low,4);
|
||
}
|
||
if(mask_flag){
|
||
ret.append((char *)header._mask.data(),4);
|
||
}
|
||
|
||
onWebSocketEncodeData(std::make_shared<BufferString>(std::move(ret)));
|
||
|
||
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];
|
||
}
|
||
}
|
||
onWebSocketEncodeData(buffer);
|
||
}
|
||
|
||
}
|
||
|
||
|
||
|
||
} /* namespace mediakit */
|
||
|
||
|
||
|
||
|