mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-23 03:10:04 +08:00
避免m3u8文件为chunked时崩溃 (#1543)
* Update HttpChunkedSplitter.h * 避免m3u8文件为chunked时崩溃 具体问题见 https://github.com/ZLMediaKit/ZLMediaKit/issues/1407 当数据最后小于2个字节时, 应该放弃回调. 做个保险, 避免导致溢出后崩溃. 这个bug很难出现, 但是的确存在. 一些特殊的服务器采用chunked返回的m3u8文件解析时, 有可能会遇到. * Update HttpChunkedSplitter.cpp Co-authored-by: 夏楚 <771730766@qq.com>
This commit is contained in:
parent
e4262222f4
commit
75b17b3a6a
@ -9,29 +9,32 @@
|
||||
*/
|
||||
|
||||
#include <string.h>
|
||||
#include "Common/macros.h"
|
||||
#include "HttpChunkedSplitter.h"
|
||||
|
||||
using namespace std;
|
||||
|
||||
//[chunk size][\r\n][chunk data][\r\n][chunk size][\r\n][chunk data][\r\n][chunk size = 0][\r\n][\r\n]
|
||||
|
||||
namespace mediakit{
|
||||
|
||||
const char *HttpChunkedSplitter::onSearchPacketTail(const char *data, size_t len) {
|
||||
auto pos = strstr(data,"\r\n");
|
||||
if(!pos){
|
||||
auto pos = strstr(data, "\r\n");
|
||||
if (!pos) {
|
||||
return nullptr;
|
||||
}
|
||||
return pos + 2;
|
||||
}
|
||||
|
||||
void HttpChunkedSplitter::onRecvContent(const char *data, size_t len) {
|
||||
onRecvChunk(data,len - 2);
|
||||
onRecvChunk(data, len - 2);
|
||||
}
|
||||
|
||||
ssize_t HttpChunkedSplitter::onRecvHeader(const char *data, size_t len) {
|
||||
string str(data,len - 2);
|
||||
int ret;
|
||||
sscanf(str.data(),"%X",&ret);
|
||||
return ret + 2;
|
||||
int size;
|
||||
CHECK(sscanf(data, "%X", &size) == 1 && size >= 0);
|
||||
//包括后面\r\n两个字节
|
||||
return size + 2;
|
||||
}
|
||||
|
||||
}//namespace mediakit
|
@ -21,12 +21,10 @@ public:
|
||||
/**
|
||||
* len == 0时代表结束
|
||||
*/
|
||||
typedef std::function<void (const char *data,size_t len)> onChunkData;
|
||||
using onChunkData = std::function<void(const char *data, size_t len)>;
|
||||
|
||||
HttpChunkedSplitter(const onChunkData &cb){
|
||||
_onChunkData = cb;
|
||||
};
|
||||
~HttpChunkedSplitter() override {} ;
|
||||
HttpChunkedSplitter(const onChunkData &cb) { _onChunkData = cb; };
|
||||
~HttpChunkedSplitter() override { _onChunkData = nullptr; };
|
||||
|
||||
protected:
|
||||
ssize_t onRecvHeader(const char *data,size_t len) override;
|
||||
|
Loading…
Reference in New Issue
Block a user