74 lines
1.8 KiB
C++
74 lines
1.8 KiB
C++
#include "TcpRequest.h"
|
||
#include "Log.h"
|
||
#include <Json.h>
|
||
|
||
QString TcpRequest::getUrl()
|
||
{
|
||
return url;
|
||
}
|
||
|
||
/**
|
||
* @brief 获取请求体参数
|
||
*/
|
||
QVariantMap TcpRequest::getBodyParams()
|
||
{
|
||
QVariantMap params = this->parseParams();
|
||
return params;
|
||
}
|
||
|
||
QString TcpRequest::getErrorString()
|
||
{
|
||
return errorString;
|
||
}
|
||
|
||
/**
|
||
* @brief 从缓冲区读取数据
|
||
*/
|
||
bool TcpRequest::readFromSocket(QTcpSocket* socket)
|
||
{
|
||
// 读取第一行
|
||
QString str = socket->readLine().trimmed();
|
||
if (!str.contains("Url: ")) {
|
||
Log::error(QString("receive error headers: %1").arg(str).toStdString());
|
||
socket->readAll();
|
||
socket->flush();
|
||
errorString = "未知请求路径参数：" + str;
|
||
return false;
|
||
} else {
|
||
url = str.split(": ")[1];
|
||
}
|
||
// 读取第二行
|
||
str = socket->readLine().trimmed();
|
||
if (!str.contains("Content-Length: ")) {
|
||
contentLength = 0;
|
||
} else {
|
||
contentLength = str.split(": ")[1].toLongLong();
|
||
}
|
||
// 如果请求体长度为0就不再往下读取
|
||
if (contentLength == 0) {
|
||
socket->readAll();
|
||
socket->flush();
|
||
return true;
|
||
} else {
|
||
body.clear();
|
||
// 读取空行并丢弃
|
||
socket->readLine();
|
||
// 循环读取数据，直到读取了content-length长度数据
|
||
qint64 readSize = 0;
|
||
while (socket->bytesAvailable() > 0) {
|
||
char bytes[contentLength];
|
||
qint64 size = socket->read(bytes, contentLength - readSize);
|
||
if (size == -1) {
|
||
errorString = "socket读取数据出错";
|
||
return false;
|
||
}
|
||
readSize += size;
|
||
body += QString::fromLocal8Bit(bytes, size);
|
||
if (readSize >= contentLength) {
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
return true;
|
||
}
|