Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
Z
ZLMediaKit
概览
Overview
Details
Activity
Cycle Analytics
版本库
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
问题
0
Issues
0
列表
Board
标记
里程碑
合并请求
0
Merge Requests
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
Snippets
成员
Collapse sidebar
Close sidebar
活动
图像
聊天
创建新问题
作业
提交
Issue Boards
Open sidebar
张翔宇
ZLMediaKit
Commits
c8b192fa
Commit
c8b192fa
authored
May 30, 2019
by
xiongziliang
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
完善http错误提示
parent
0c460b39
隐藏空白字符变更
内嵌
并排
正在显示
2 个修改的文件
包含
25 行增加
和
20 行删除
+25
-20
src/Http/HttpSession.cpp
+23
-18
src/Http/HttpSession.h
+2
-2
没有找到文件。
src/Http/HttpSession.cpp
查看文件 @
c8b192fa
...
@@ -113,7 +113,7 @@ HttpSession::~HttpSession() {
...
@@ -113,7 +113,7 @@ HttpSession::~HttpSession() {
}
}
int64_t
HttpSession
::
onRecvHeader
(
const
char
*
header
,
uint64_t
len
)
{
int64_t
HttpSession
::
onRecvHeader
(
const
char
*
header
,
uint64_t
len
)
{
typedef
bool
(
HttpSession
::*
HttpCMDHandle
)(
int64_t
&
);
typedef
void
(
HttpSession
::*
HttpCMDHandle
)(
int64_t
&
);
static
unordered_map
<
string
,
HttpCMDHandle
>
g_mapCmdIndex
;
static
unordered_map
<
string
,
HttpCMDHandle
>
g_mapCmdIndex
;
static
onceToken
token
([]()
{
static
onceToken
token
([]()
{
g_mapCmdIndex
.
emplace
(
"GET"
,
&
HttpSession
::
Handle_Req_GET
);
g_mapCmdIndex
.
emplace
(
"GET"
,
&
HttpSession
::
Handle_Req_GET
);
...
@@ -133,9 +133,16 @@ int64_t HttpSession::onRecvHeader(const char *header,uint64_t len) {
...
@@ -133,9 +133,16 @@ int64_t HttpSession::onRecvHeader(const char *header,uint64_t len) {
//默认后面数据不是content而是header
//默认后面数据不是content而是header
int64_t
content_len
=
0
;
int64_t
content_len
=
0
;
auto
&
fun
=
it
->
second
;
auto
&
fun
=
it
->
second
;
if
(
!
(
this
->*
fun
)(
content_len
)){
try
{
shutdown
(
SockException
(
Err_shutdown
,
"Connection: close"
));
(
this
->*
fun
)(
content_len
);
}
}
catch
(
SockException
&
ex
){
if
(
ex
){
shutdown
(
ex
);
}
}
catch
(
exception
&
ex
){
shutdown
(
SockException
(
Err_shutdown
,
ex
.
what
()));
}
//清空解析器节省内存
//清空解析器节省内存
_parser
.
Clear
();
_parser
.
Clear
();
//返回content长度
//返回content长度
...
@@ -309,7 +316,7 @@ inline static string findIndexFile(const string &dir){
...
@@ -309,7 +316,7 @@ inline static string findIndexFile(const string &dir){
}
}
inline
bool
HttpSession
::
Handle_Req_GET
(
int64_t
&
content_len
)
{
inline
void
HttpSession
::
Handle_Req_GET
(
int64_t
&
content_len
)
{
//先看看是否为WebSocket请求
//先看看是否为WebSocket请求
if
(
checkWebSocket
()){
if
(
checkWebSocket
()){
content_len
=
-
1
;
content_len
=
-
1
;
...
@@ -319,17 +326,17 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
...
@@ -319,17 +326,17 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
//_contentCallBack是可持续的,后面还要处理后续数据
//_contentCallBack是可持续的,后面还要处理后续数据
return
true
;
return
true
;
};
};
return
true
;
return
;
}
}
//先看看该http事件是否被拦截
//先看看该http事件是否被拦截
if
(
emitHttpEvent
(
false
)){
if
(
emitHttpEvent
(
false
)){
return
true
;
return
;
}
}
//再看看是否为http-flv直播请求
//再看看是否为http-flv直播请求
if
(
checkLiveFlvStream
()){
if
(
checkLiveFlvStream
()){
return
true
;
return
;
}
}
//事件未被拦截,则认为是http下载请求
//事件未被拦截,则认为是http下载请求
...
@@ -357,10 +364,10 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
...
@@ -357,10 +364,10 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
if
(
!
makeMeun
(
_parser
.
Url
(),
strFile
,
strMeun
))
{
if
(
!
makeMeun
(
_parser
.
Url
(),
strFile
,
strMeun
))
{
//文件夹不存在
//文件夹不存在
sendNotFound
(
bClose
);
sendNotFound
(
bClose
);
return
!
bClose
;
throw
SockException
(
bClose
?
Err_shutdown
:
Err_success
,
"close connection after send 404 not found on folder"
)
;
}
}
sendResponse
(
"200 OK"
,
makeHttpHeader
(
bClose
,
strMeun
.
size
()
),
strMeun
);
sendResponse
(
"200 OK"
,
makeHttpHeader
(
bClose
,
strMeun
.
size
()
),
strMeun
);
return
!
bClose
;
throw
SockException
(
bClose
?
Err_shutdown
:
Err_success
,
"close connection after send 200 ok on folder"
)
;
}
}
}
while
(
0
);
}
while
(
0
);
...
@@ -369,7 +376,7 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
...
@@ -369,7 +376,7 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
if
(
0
!=
stat
(
strFile
.
data
(),
&
tFileStat
))
{
if
(
0
!=
stat
(
strFile
.
data
(),
&
tFileStat
))
{
//文件不存在
//文件不存在
sendNotFound
(
bClose
);
sendNotFound
(
bClose
);
return
!
bClose
;
throw
SockException
(
bClose
?
Err_shutdown
:
Err_success
,
"close connection after send 404 not found on file"
)
;
}
}
//文件智能指针,防止退出时未关闭
//文件智能指针,防止退出时未关闭
std
::
shared_ptr
<
FILE
>
pFilePtr
(
fopen
(
strFile
.
data
(),
"rb"
),
[](
FILE
*
pFile
)
{
std
::
shared_ptr
<
FILE
>
pFilePtr
(
fopen
(
strFile
.
data
(),
"rb"
),
[](
FILE
*
pFile
)
{
...
@@ -381,7 +388,7 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
...
@@ -381,7 +388,7 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
if
(
!
pFilePtr
)
{
if
(
!
pFilePtr
)
{
//打开文件失败
//打开文件失败
sendNotFound
(
bClose
);
sendNotFound
(
bClose
);
return
!
bClose
;
throw
SockException
(
bClose
?
Err_shutdown
:
Err_success
,
"close connection after send 404 not found on open file failed"
)
;
}
}
//判断是不是分节下载
//判断是不是分节下载
...
@@ -415,8 +422,8 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
...
@@ -415,8 +422,8 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
sendResponse
(
pcHttpResult
,
httpHeader
,
""
);
sendResponse
(
pcHttpResult
,
httpHeader
,
""
);
if
(
iRangeEnd
-
iRangeStart
<
0
)
{
if
(
iRangeEnd
-
iRangeStart
<
0
)
{
//文件是空的!
//文件是空的!
return
!
bClose
;
throw
SockException
(
bClose
?
Err_shutdown
:
Err_success
,
"close connection after send file partial range excpted"
)
;
}
}
//回复Content部分
//回复Content部分
std
::
shared_ptr
<
int64_t
>
piLeft
(
new
int64_t
(
iRangeEnd
-
iRangeStart
+
1
));
std
::
shared_ptr
<
int64_t
>
piLeft
(
new
int64_t
(
iRangeEnd
-
iRangeStart
+
1
));
...
@@ -475,7 +482,6 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
...
@@ -475,7 +482,6 @@ inline bool HttpSession::Handle_Req_GET(int64_t &content_len) {
onFlush
();
onFlush
();
_sock
->
setOnFlush
(
onFlush
);
_sock
->
setOnFlush
(
onFlush
);
return
true
;
}
}
inline
bool
makeMeun
(
const
string
&
httpPath
,
const
string
&
strFullPath
,
string
&
strRet
)
{
inline
bool
makeMeun
(
const
string
&
httpPath
,
const
string
&
strFullPath
,
string
&
strRet
)
{
...
@@ -665,7 +671,7 @@ inline bool HttpSession::emitHttpEvent(bool doInvoke){
...
@@ -665,7 +671,7 @@ inline bool HttpSession::emitHttpEvent(bool doInvoke){
}
}
return
consumed
;
return
consumed
;
}
}
inline
bool
HttpSession
::
Handle_Req_POST
(
int64_t
&
content_len
)
{
inline
void
HttpSession
::
Handle_Req_POST
(
int64_t
&
content_len
)
{
GET_CONFIG
(
uint64_t
,
maxReqSize
,
Http
::
kMaxReqSize
);
GET_CONFIG
(
uint64_t
,
maxReqSize
,
Http
::
kMaxReqSize
);
GET_CONFIG
(
int
,
maxReqCnt
,
Http
::
kMaxReqCount
);
GET_CONFIG
(
int
,
maxReqCnt
,
Http
::
kMaxReqCount
);
...
@@ -675,7 +681,7 @@ inline bool HttpSession::Handle_Req_POST(int64_t &content_len) {
...
@@ -675,7 +681,7 @@ inline bool HttpSession::Handle_Req_POST(int64_t &content_len) {
//content为空
//content为空
//emitHttpEvent内部会选择是否关闭连接
//emitHttpEvent内部会选择是否关闭连接
emitHttpEvent
(
true
);
emitHttpEvent
(
true
);
return
true
;
return
;
}
}
//根据Content-Length设置接收缓存大小
//根据Content-Length设置接收缓存大小
...
@@ -736,7 +742,6 @@ inline bool HttpSession::Handle_Req_POST(int64_t &content_len) {
...
@@ -736,7 +742,6 @@ inline bool HttpSession::Handle_Req_POST(int64_t &content_len) {
};
};
}
}
//有后续content数据要处理,暂时不关闭连接
//有后续content数据要处理,暂时不关闭连接
return
true
;
}
}
void
HttpSession
::
responseDelay
(
const
string
&
Origin
,
bool
bClose
,
void
HttpSession
::
responseDelay
(
const
string
&
Origin
,
bool
bClose
,
const
string
&
codeOut
,
const
KeyValue
&
headerOut
,
const
string
&
codeOut
,
const
KeyValue
&
headerOut
,
...
...
src/Http/HttpSession.h
查看文件 @
c8b192fa
...
@@ -95,8 +95,8 @@ protected:
...
@@ -95,8 +95,8 @@ protected:
}
}
private
:
private
:
inline
bool
Handle_Req_GET
(
int64_t
&
content_len
);
inline
void
Handle_Req_GET
(
int64_t
&
content_len
);
inline
bool
Handle_Req_POST
(
int64_t
&
content_len
);
inline
void
Handle_Req_POST
(
int64_t
&
content_len
);
inline
bool
checkLiveFlvStream
();
inline
bool
checkLiveFlvStream
();
inline
bool
checkWebSocket
();
inline
bool
checkWebSocket
();
inline
bool
emitHttpEvent
(
bool
doInvoke
);
inline
bool
emitHttpEvent
(
bool
doInvoke
);
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论