293 lines
12 KiB
C++
293 lines
12 KiB
C++
#ifndef SRCCTL_RTSP_DATA_CHANNEL_H
|
||
#define SRCCTL_RTSP_DATA_CHANNEL_H
|
||
|
||
#include "Socket/CCServerSocket.h"
|
||
#include "Socket/CCDatagramSocket.h"
|
||
#include "CCSDP.h"
|
||
|
||
#define CC_R_RTSP_BUFFER_SIZE 1
|
||
|
||
namespace CTL{
|
||
typedef struct RTSP_Message{
|
||
int statusCode = 0; // 状态码 (响应时使用)
|
||
int port = 0; // 客户端端口号
|
||
bool isRequest = true; // 是否为请求消息(true)还是响应消息(false)
|
||
String client_ip; // 客户端IP
|
||
String url_data; // 请求URL
|
||
String method; // RTSP方法 (OPTIONS, DESCRIBE, SETUP, PLAY等)
|
||
String uri; // 请求URI
|
||
String uri_path; // 请求URI路径
|
||
String version; // RTSP版本 (如"RTSP/1.0")
|
||
String reasonPhrase; // 原因短语 (响应时使用)
|
||
std::map<String, String> headers; // 头部字段
|
||
String body; // 消息体
|
||
String sdp_content; // SDP内容
|
||
} RTSP_Request;
|
||
class RTSPDataChannel{
|
||
ClientSocket* socket_t = nullptr;
|
||
public:
|
||
explicit RTSPDataChannel(ClientSocket* socket){
|
||
socket_t = socket;
|
||
}
|
||
~RTSPDataChannel(){
|
||
if (socket_t) {
|
||
socket_t->close();
|
||
socket_t = nullptr;
|
||
}
|
||
}
|
||
static void MessagePrint(const String& message,...){
|
||
System::Println(message);
|
||
}
|
||
int receive(char *buffer, const size_t size) const {
|
||
if (socket_t && socket_t->isConnect()) {
|
||
const auto input = socket_t->getInputStream();
|
||
if (input->available()) {
|
||
return input->read(buffer, size);
|
||
}
|
||
return -2;
|
||
}
|
||
return -1;
|
||
}
|
||
int write(const char *buffer, const size_t size) const {
|
||
if (socket_t && socket_t->isConnect()) {
|
||
const auto output = socket_t->getOutputStream();
|
||
return output->write(buffer,size);
|
||
}
|
||
return 0;
|
||
}
|
||
String readHeader() const{
|
||
if (this->socket_t) {
|
||
String data;
|
||
data.reserve(4096); // 预分配内存
|
||
int ET = 1;
|
||
while (true){
|
||
char buffer[CC_R_RTSP_BUFFER_SIZE] = {0};
|
||
if(const auto length = receive(buffer,CC_R_RTSP_BUFFER_SIZE); length > 0){
|
||
data.append(buffer, length); // 块追加代替逐个字符处理
|
||
if (const auto pos = data.find("\r\n\r\n"); pos != CTL::String::npos) {
|
||
data.resize(pos + 4); // 截断到header结束位置
|
||
break;
|
||
}
|
||
}
|
||
else if (length == -2) {
|
||
if (ET >= 1000 * 1000) {
|
||
this->MessagePrint("RTSPDataChannel readHeader -> Timeout");
|
||
break;
|
||
}
|
||
ET++;
|
||
Thread::Sleep(1);
|
||
}
|
||
else{
|
||
return "close";
|
||
}
|
||
}
|
||
return data;
|
||
}
|
||
return "Error_10030";
|
||
}
|
||
String readData(const uint64_t length) const{
|
||
if (this->socket_t) {
|
||
String data;
|
||
data.reserve(length); // 预分配内存
|
||
int ET = 1;
|
||
while (true){
|
||
char buffer[CC_R_RTSP_BUFFER_SIZE] = {0};
|
||
if(const auto len = receive(buffer,CC_R_RTSP_BUFFER_SIZE); len > 0){
|
||
data.append(buffer, len);
|
||
if (data.length() >= length) {
|
||
break;
|
||
}
|
||
}
|
||
else if (len == -2) {
|
||
if (ET >= 1000 * 300) {
|
||
this->MessagePrint("RTSPDataChannel readData -> Timeout");
|
||
break;
|
||
}
|
||
ET++;
|
||
Thread::Sleep(1);
|
||
}
|
||
else{
|
||
return "close";
|
||
}
|
||
}
|
||
return data;
|
||
}
|
||
return "Error_10030";
|
||
}
|
||
static RTSP_Message ParseRTSP(const String& data){
|
||
RTSP_Message msg;
|
||
if (data.empty()) {
|
||
return msg;
|
||
}
|
||
msg.url_data = data;
|
||
// 查找空行(\r\n\r\n)来分离头部和消息体
|
||
size_t headerEndPos = data.find("\r\n\r\n");
|
||
String headersPart;
|
||
String bodyPart;
|
||
if (headerEndPos != CTL::String::npos) {
|
||
headersPart = data.substr(0, headerEndPos);
|
||
bodyPart = data.substr(headerEndPos + 4); // 跳过"\r\n\r\n"
|
||
}
|
||
else {
|
||
headersPart = data;
|
||
}
|
||
// 分割头部行
|
||
std::vector<String> lines;
|
||
String currentLine;
|
||
for (size_t i = 0; i < headersPart.length(); ++i) {
|
||
if (headersPart[i] == '\r' && i + 1 < headersPart.length() && headersPart[i + 1] == '\n') {
|
||
lines.push_back(currentLine);
|
||
currentLine.clear();
|
||
i++; // 跳过'\n'
|
||
} else {
|
||
currentLine += headersPart[i];
|
||
}
|
||
}
|
||
if (!currentLine.empty()) {
|
||
lines.push_back(currentLine);
|
||
}
|
||
if (!lines.empty()) {
|
||
// 解析第一行 (请求行或状态行)
|
||
const String& firstLine = lines[0];
|
||
|
||
// 检查是请求还是响应
|
||
if (firstLine.find("RTSP/") == 0) {
|
||
// 这是响应行: "RTSP/1.0 200 OK"
|
||
msg.isRequest = false;
|
||
|
||
size_t firstSpace = firstLine.find(' ');
|
||
if (firstSpace != String::npos) {
|
||
size_t secondSpace = firstLine.find(' ', firstSpace + 1);
|
||
if (secondSpace != String::npos) {
|
||
msg.version = firstLine.substr(0, firstSpace);
|
||
try {
|
||
msg.statusCode = std::stoi(firstLine.substr(firstSpace + 1,
|
||
secondSpace - firstSpace - 1));
|
||
}
|
||
catch (...) {
|
||
msg.statusCode = 500; // 默认错误状态
|
||
}
|
||
msg.reasonPhrase = firstLine.substr(secondSpace + 1);
|
||
}
|
||
}
|
||
}
|
||
else {
|
||
// 这是请求行: "OPTIONS rtsp://... RTSP/1.0"
|
||
size_t firstSpace = firstLine.find(' ');
|
||
size_t lastSpace = firstLine.rfind(' ');
|
||
|
||
if (firstSpace != String::npos && lastSpace != String::npos && firstSpace < lastSpace) {
|
||
msg.method = firstLine.substr(0, firstSpace);
|
||
msg.uri = firstLine.substr(firstSpace + 1, lastSpace - firstSpace - 1);
|
||
msg.version = firstLine.substr(lastSpace + 1);
|
||
|
||
// 从URI中解析路径部分
|
||
size_t schemeEnd = msg.uri.find("://");
|
||
if (schemeEnd != String::npos) {
|
||
// 跳过协议和主机部分，找到路径
|
||
size_t hostStart = schemeEnd + 3;
|
||
size_t pathStart = msg.uri.find('/', hostStart);
|
||
if (pathStart != String::npos) {
|
||
msg.uri_path = msg.uri.substr(pathStart);
|
||
} else {
|
||
// 如果没有路径，则设置为根路径
|
||
msg.uri_path = "/";
|
||
}
|
||
} else {
|
||
// 如果没有协议前缀，整个URI可能是路径
|
||
if (msg.uri.empty() || msg.uri[0] != '/') {
|
||
msg.uri_path = "/" + msg.uri;
|
||
} else {
|
||
msg.uri_path = msg.uri;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 解析头部字段
|
||
for (size_t i = 1; i < lines.size(); ++i) {
|
||
const String& line = lines[i];
|
||
size_t colonPos = line.find(':');
|
||
if (colonPos != String::npos) {
|
||
String key = line.substr(0, colonPos);
|
||
String value = line.substr(colonPos + 1);
|
||
|
||
// 移除键和值前后的空白字符
|
||
size_t startKey = key.find_first_not_of(" \t");
|
||
size_t endKey = key.find_last_not_of(" \t");
|
||
if (startKey != String::npos && endKey != String::npos) {
|
||
key = key.substr(startKey, endKey - startKey + 1);
|
||
} else if (startKey == String::npos) {
|
||
continue; // 跳过无效行
|
||
}
|
||
|
||
size_t startPos = value.find_first_not_of(" \t");
|
||
size_t endPos = value.find_last_not_of(" \t");
|
||
if (startPos != String::npos && endPos != String::npos) {
|
||
value = value.substr(startPos, endPos - startPos + 1);
|
||
} else if (startPos == String::npos) {
|
||
value.clear();
|
||
}
|
||
|
||
// 将头部字段转换为小写作为键
|
||
String lowerKey = key;
|
||
std::transform(lowerKey.begin(), lowerKey.end(), lowerKey.begin(), ::tolower);
|
||
|
||
msg.headers[lowerKey] = value;
|
||
}
|
||
}
|
||
}
|
||
// 设置消息体
|
||
msg.body = bodyPart;
|
||
return msg;
|
||
}
|
||
};
|
||
class RTSP_Response{
|
||
RTSPDataChannel* socket_t = nullptr;
|
||
std::map<String, String> headers;
|
||
SDPBuilder sdp_builder;
|
||
public:
|
||
//-----------------------------------------------------------------------------------------
|
||
int statusCode = 200;
|
||
String reasonPhrase = "OK";
|
||
bool auto_send = true;
|
||
//-----------------------------------------------------------------------------------------
|
||
void setSocket(RTSPDataChannel* socket){
|
||
this->socket_t = socket;
|
||
}
|
||
RTSPDataChannel* getSocket() const{
|
||
return socket_t;
|
||
}
|
||
String getHeader(const String& key) const{
|
||
auto it = headers.find(key);
|
||
if (it != headers.end()) {
|
||
return it->second;
|
||
}
|
||
return "";
|
||
}
|
||
void setHeader(const String& key, const String& value){
|
||
headers[key] = value;
|
||
}
|
||
String getBuild() const{
|
||
String response = String::format("RTSP/1.0 {} {}\r\n", statusCode, reasonPhrase.c_str());
|
||
for (const auto& header : headers) {
|
||
response.append(String::format("{}: {}\r\n",header.first.c_str(),header.second.c_str()));
|
||
}
|
||
response.append("\r\n");
|
||
return response;
|
||
}
|
||
bool send(){
|
||
if(socket_t != nullptr){
|
||
int ret = socket_t->write(getBuild().c_str(), getBuild().length());
|
||
if (ret > 0) {
|
||
auto_send = false;
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
//-----------------------------------------------------------------------------------------
|
||
};
|
||
}
|
||
|
||
#endif |