mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-25 12:11:36 +08:00
144 lines
3.8 KiB
C++
144 lines
3.8 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#ifndef ZLMEDIAKIT_PARSER_H
|
||
#define ZLMEDIAKIT_PARSER_H
|
||
|
||
#include <map>
|
||
#include <string>
|
||
#include "Util/util.h"
|
||
|
||
namespace mediakit {
|
||
|
||
// 从字符串中提取子字符串
|
||
std::string findSubString(const char *buf, const char *start, const char *end, size_t buf_size = 0);
|
||
// 把url解析为主机地址和端口号,兼容ipv4/ipv6/dns
|
||
void splitUrl(const std::string &url, std::string &host, uint16_t &port);
|
||
|
||
struct StrCaseCompare {
|
||
bool operator()(const std::string &__x, const std::string &__y) const { return strcasecmp(__x.data(), __y.data()) < 0; }
|
||
};
|
||
|
||
class StrCaseMap : public std::multimap<std::string, std::string, StrCaseCompare> {
|
||
public:
|
||
using Super = multimap<std::string, std::string, StrCaseCompare>;
|
||
StrCaseMap() = default;
|
||
~StrCaseMap() = default;
|
||
|
||
std::string &operator[](const std::string &k) {
|
||
auto it = find(k);
|
||
if (it == end()) {
|
||
it = Super::emplace(k, "");
|
||
}
|
||
return it->second;
|
||
}
|
||
|
||
template <typename K, typename V>
|
||
void emplace(K &&k, V &&v) {
|
||
auto it = find(k);
|
||
if (it != end()) {
|
||
return;
|
||
}
|
||
Super::emplace(std::forward<K>(k), std::forward<V>(v));
|
||
}
|
||
|
||
template <typename K, typename V>
|
||
void emplace_force(K &&k, V &&v) {
|
||
Super::emplace(std::forward<K>(k), std::forward<V>(v));
|
||
}
|
||
};
|
||
|
||
// rtsp/http/sip解析类
|
||
class Parser {
|
||
public:
|
||
Parser() = default;
|
||
~Parser() = default;
|
||
|
||
// 解析http/rtsp/sip请求，需要确保buf以\0结尾
|
||
void parse(const char *buf, size_t size);
|
||
|
||
// 获取命令字，如GET/POST
|
||
const std::string &method() const;
|
||
|
||
// 请求时，获取中间url，不包含?后面的参数
|
||
const std::string &url() const;
|
||
// 回复时，获取状态码，如200/404
|
||
const std::string &status() const;
|
||
|
||
// 获取中间url，包含?后面的参数
|
||
std::string fullUrl() const;
|
||
|
||
// 请求时，获取协议名，如HTTP/1.1
|
||
const std::string &protocol() const;
|
||
// 回复时，获取状态字符串，如 OK/Not Found
|
||
const std::string &statusStr() const;
|
||
|
||
// 根据header key名，获取请求header value值
|
||
const std::string &operator[](const char *name) const;
|
||
|
||
// 获取http body或sdp
|
||
const std::string &content() const;
|
||
|
||
// 清空，为了重用
|
||
void clear();
|
||
|
||
// 获取?后面的参数
|
||
const std::string ¶ms() const;
|
||
|
||
// 重新设置url
|
||
void setUrl(std::string url);
|
||
|
||
// 重新设置content
|
||
void setContent(std::string content);
|
||
|
||
// 获取header列表
|
||
StrCaseMap &getHeader() const;
|
||
|
||
// 获取url参数列表
|
||
StrCaseMap &getUrlArgs() const;
|
||
|
||
// 解析?后面的参数
|
||
static StrCaseMap parseArgs(const std::string &str, const char *pair_delim = "&", const char *key_delim = "=");
|
||
|
||
static std::string mergeUrl(const std::string &base_url, const std::string &path);
|
||
|
||
private:
|
||
std::string _method;
|
||
std::string _url;
|
||
std::string _protocol;
|
||
std::string _content;
|
||
std::string _params;
|
||
mutable StrCaseMap _headers;
|
||
mutable StrCaseMap _url_args;
|
||
};
|
||
|
||
// 解析rtsp url的工具类
|
||
class RtspUrl {
|
||
public:
|
||
bool _is_ssl;
|
||
uint16_t _port;
|
||
std::string _url;
|
||
std::string _user;
|
||
std::string _passwd;
|
||
std::string _host;
|
||
|
||
public:
|
||
RtspUrl() = default;
|
||
~RtspUrl() = default;
|
||
void parse(const std::string &url);
|
||
|
||
private:
|
||
void setup(bool, const std::string &, const std::string &, const std::string &);
|
||
};
|
||
|
||
} // namespace mediakit
|
||
|
||
#endif // ZLMEDIAKIT_PARSER_H
|