amass/ZLMediaKit
1
0
Fork 0
mirror of https://github.com/ZLMediaKit/ZLMediaKit.git synced 2024-11-26 04:31:37 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity
749fb98439
ZLMediaKit/server/main.cpp

319 lines
13 KiB
C++
Raw Normal View History

添加完整的服务器用例
2019-05-20 11:22:59 +08:00
/*
* MIT License
*
* Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
*
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include <map>
#include <signal.h>
#include <iostream>
#include "Util/MD5.h"
#include "Util/File.h"
#include "Util/logger.h"
#include "Util/SSLBox.h"
#include "Util/onceToken.h"
#include "Util/CMD.h"
#include "Network/TcpServer.h"
#include "Poller/EventPoller.h"
#include "Common/config.h"
#include "Rtsp/UDPServer.h"
#include "Rtsp/RtspSession.h"
#include "Rtmp/RtmpSession.h"
#include "Shell/ShellSession.h"
整理优化代码
2019-06-28 17:30:13 +08:00
#include "Rtmp/FlvMuxer.h"
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
#include "Player/PlayerProxy.h"
#include "Http/WebSocketSession.h"
添加FFmpeg拉流功能
2019-06-06 18:28:33 +08:00
#include "WebApi.h"
函数声明放置于头文件
2019-06-24 14:58:56 +08:00
#include "WebHook.h"
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
移植Windows
2019-06-15 17:07:10 +08:00
#if !defined(_WIN32)
#include "System.h"
#endif//!defined(_WIN32)
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
using namespace std;
using namespace toolkit;
using namespace mediakit;
namespace mediakit {
////////////HTTP配置///////////
namespace Http {
#define HTTP_FIELD "http."
#define HTTP_PORT 80
配置键名统一使用string
2019-06-19 10:32:54 +08:00
const string kPort = HTTP_FIELD"port";
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
#define HTTPS_PORT 443
配置键名统一使用string
2019-06-19 10:32:54 +08:00
const string kSSLPort = HTTP_FIELD"sslport";
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
onceToken token1([](){
mINI::Instance()[kPort] = HTTP_PORT;
mINI::Instance()[kSSLPort] = HTTPS_PORT;
},nullptr);
}//namespace Http
////////////SHELL配置///////////
namespace Shell {
#define SHELL_FIELD "shell."
#define SHELL_PORT 9000
配置键名统一使用string
2019-06-19 10:32:54 +08:00
const string kPort = SHELL_FIELD"port";
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
onceToken token1([](){
mINI::Instance()[kPort] = SHELL_PORT;
},nullptr);
} //namespace Shell
////////////RTSP服务器配置///////////
namespace Rtsp {
#define RTSP_FIELD "rtsp."
#define RTSP_PORT 554
#define RTSPS_PORT 322
配置键名统一使用string
2019-06-19 10:32:54 +08:00
const string kPort = RTSP_FIELD"port";
const string kSSLPort = RTSP_FIELD"sslport";
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
onceToken token1([](){
mINI::Instance()[kPort] = RTSP_PORT;
mINI::Instance()[kSSLPort] = RTSPS_PORT;
},nullptr);
} //namespace Rtsp
////////////RTMP服务器配置///////////
namespace Rtmp {
#define RTMP_FIELD "rtmp."
#define RTMP_PORT 1935
配置键名统一使用string
2019-06-19 10:32:54 +08:00
const string kPort = RTMP_FIELD"port";
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
onceToken token1([](){
mINI::Instance()[kPort] = RTMP_PORT;
},nullptr);
} //namespace RTMP
} // namespace mediakit
class CMD_main : public CMD {
public:
CMD_main() {
_parser.reset(new OptionParser(nullptr));
移植Windows
2019-06-15 17:07:10 +08:00
#if !defined(_WIN32)
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
(*_parser) << Option('d',/*该选项简称，如果是\x00则说明无简称*/
"daemon",/*该选项全称,每个选项必须有全称；不得为null或空字符串*/
Option::ArgNone,/*该选项后面必须跟值*/
nullptr,/*该选项默认值*/
false,/*该选项是否必须赋值，如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"是否以Daemon方式启动",/*该选项说明文字*/
nullptr);
移植Windows
2019-06-15 17:07:10 +08:00
#endif//!defined(_WIN32)
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
(*_parser) << Option('l',/*该选项简称，如果是\x00则说明无简称*/
"level",/*该选项全称,每个选项必须有全称；不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
to_string(LTrace).data(),/*该选项默认值*/
false,/*该选项是否必须赋值，如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"日志等级,LTrace~LError(0~4)",/*该选项说明文字*/
nullptr);
支持日志分天以及自动清理
2019-08-06 22:51:40 +08:00
(*_parser) << Option('m',/*该选项简称，如果是\x00则说明无简称*/
"max_day",/*该选项全称,每个选项必须有全称；不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
"7",/*该选项默认值*/
false,/*该选项是否必须赋值，如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"日志等级最多保存天数",/*该选项说明文字*/
nullptr);
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
(*_parser) << Option('c',/*该选项简称，如果是\x00则说明无简称*/
"config",/*该选项全称,每个选项必须有全称；不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
(exeDir() + "config.ini").data(),/*该选项默认值*/
false,/*该选项是否必须赋值，如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"配置文件路径",/*该选项说明文字*/
nullptr);
(*_parser) << Option('s',/*该选项简称，如果是\x00则说明无简称*/
"ssl",/*该选项全称,每个选项必须有全称；不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
(exeDir() + "ssl.p12").data(),/*该选项默认值*/
false,/*该选项是否必须赋值，如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"ssl证书路径,支持p12/pem类型",/*该选项说明文字*/
nullptr);
(*_parser) << Option('t',/*该选项简称，如果是\x00则说明无简称*/
"threads",/*该选项全称,每个选项必须有全称；不得为null或空字符串*/
Option::ArgRequired,/*该选项后面必须跟值*/
to_string(thread::hardware_concurrency()).data(),/*该选项默认值*/
false,/*该选项是否必须赋值，如果没有默认值且为ArgRequired时用户必须提供该参数否则将抛异常*/
"启动事件触发线程数",/*该选项说明文字*/
nullptr);
}
virtual ~CMD_main() {}
virtual const char *description() const {
return "主程序命令参数";
}
};
移植Windows
2019-06-15 17:07:10 +08:00
#if !defined(_WIN32)
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
static void inline listen_shell_input(){
cout << "> 欢迎进入命令模式，你可以输入\"help\"命令获取帮助" << endl;
cout << "> " << std::flush;
完整主进程
2019-05-20 22:22:05 +08:00
signal(SIGTTOU,SIG_IGN);
signal(SIGTTIN,SIG_IGN);
移植Windows
2019-06-15 17:07:10 +08:00
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
SockUtil::setNoBlocked(STDIN_FILENO);
auto oninput = [](int event) {
if (event & Event_Read) {
char buf[1024];
int n = read(STDIN_FILENO, buf, sizeof(buf));
if (n > 0) {
buf[n] = '\0';
try {
CMDRegister::Instance()(buf);
cout << "> " << std::flush;
} catch (ExitException &ex) {
InfoL << "ExitException";
kill(getpid(), SIGINT);
} catch (std::exception &ex) {
cout << ex.what() << endl;
}
} else {
DebugL << get_uv_errmsg();
EventPollerPool::Instance().getFirstPoller()->delEvent(STDIN_FILENO);
}
}
if (event & Event_Error) {
WarnL << "Event_Error";
EventPollerPool::Instance().getFirstPoller()->delEvent(STDIN_FILENO);
}
};
EventPollerPool::Instance().getFirstPoller()->addEvent(STDIN_FILENO, Event_Read | Event_Error | Event_LT,oninput);
}
移植Windows
2019-06-15 17:07:10 +08:00
#endif//!defined(_WIN32)
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
int main(int argc,char *argv[]) {
完善Hook与API
2019-05-20 16:26:04 +08:00
{
CMD_main cmd_main;
try {
cmd_main.operator()(argc, argv);
} catch (std::exception &ex) {
cout << ex.what() << endl;
return -1;
}
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
完善Hook与API
2019-05-20 16:26:04 +08:00
bool bDaemon = cmd_main.hasKey("daemon");
LogLevel logLevel = (LogLevel) cmd_main["level"].as<int>();
logLevel = MIN(MAX(logLevel, LTrace), LError);
修复热加载不同配置文件的bug
2019-05-31 15:49:52 +08:00
static string ini_file = cmd_main["config"];
完善Hook与API
2019-05-20 16:26:04 +08:00
string ssl_file = cmd_main["ssl"];
int threads = cmd_main["threads"];
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
完善Hook与API
2019-05-20 16:26:04 +08:00
//设置日志
Logger::Instance().add(std::make_shared<ConsoleChannel>("ConsoleChannel", logLevel));
支持日志分天以及自动清理
2019-08-06 22:51:40 +08:00
auto fileChannel = std::make_shared<FileChannel>("FileChannel", exeDir() + "log/", logLevel);
//日志最多保存天数
fileChannel->setMaxDay(cmd_main["max_day"]);
Logger::Instance().add(fileChannel);
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
移植Windows
2019-06-15 17:07:10 +08:00
#if !defined(_WIN32)
完善Hook与API
2019-05-20 16:26:04 +08:00
if (bDaemon) {
//启动守护进程
System::startDaemon();
}
开启崩溃捕获等
2019-05-20 16:34:46 +08:00
//开启崩溃捕获等
System::systemSetup();
移植Windows
2019-06-15 17:07:10 +08:00
#endif//!defined(_WIN32)
开启崩溃捕获等
2019-05-20 16:34:46 +08:00
完善Hook与API
2019-05-20 16:26:04 +08:00
//启动异步日志线程
Logger::Instance().setWriter(std::make_shared<AsyncLogWriter>());
//加载配置文件，如果配置文件不存在就创建一个
loadIniConfig(ini_file.data());
//加载证书，证书包含公钥和私钥
SSL_Initor::Instance().loadCertificate(ssl_file.data());
//信任某个自签名证书
SSL_Initor::Instance().trustCertificate(ssl_file.data());
//不忽略无效证书证书(例如自签名或过期证书)
忽略无效ssl证书
2019-05-30 09:21:45 +08:00
SSL_Initor::Instance().ignoreInvalidCertificate(true);
完善Hook与API
2019-05-20 16:26:04 +08:00
uint16_t shellPort = mINI::Instance()[Shell::kPort];
uint16_t rtspPort = mINI::Instance()[Rtsp::kPort];
uint16_t rtspsPort = mINI::Instance()[Rtsp::kSSLPort];
uint16_t rtmpPort = mINI::Instance()[Rtmp::kPort];
uint16_t httpPort = mINI::Instance()[Http::kPort];
uint16_t httpsPort = mINI::Instance()[Http::kSSLPort];
开启崩溃捕获等
2019-05-20 16:34:46 +08:00
//设置poller线程数,该函数必须在使用ZLToolKit网络相关对象之前调用才能生效
完善Hook与API
2019-05-20 16:26:04 +08:00
EventPollerPool::setPoolSize(threads);
//简单的telnet服务器，可用于服务器调试，但是不能使用23端口，否则telnet上了莫名其妙的现象
//测试方法:telnet 127.0.0.1 9000
TcpServer::Ptr shellSrv(new TcpServer());
TcpServer::Ptr rtspSrv(new TcpServer());
TcpServer::Ptr rtmpSrv(new TcpServer());
TcpServer::Ptr httpSrv(new TcpServer());
shellSrv->start<ShellSession>(shellPort);
rtspSrv->start<RtspSession>(rtspPort);//默认554
rtmpSrv->start<RtmpSession>(rtmpPort);//默认1935
//http服务器,支持websocket
httpSrv->start<EchoWebSocketSession>(httpPort);//默认80
//如果支持ssl，还可以开启https服务器
TcpServer::Ptr httpsSrv(new TcpServer());
//https服务器,支持websocket
httpsSrv->start<SSLEchoWebSocketSession>(httpsPort);//默认443
//支持ssl加密的rtsp服务器，可用于诸如亚马逊echo show这样的设备访问
TcpServer::Ptr rtspSSLSrv(new TcpServer());
rtspSSLSrv->start<RtspSessionWithSSL>(rtspsPort);//默认322
installWebApi();
InfoL << "已启动http api 接口";
installWebHook();
InfoL << "已启动http hook 接口";
移植Windows
2019-06-15 17:07:10 +08:00
#if !defined(_WIN32)
完善Hook与API
2019-05-20 16:26:04 +08:00
if (!bDaemon) {
//交互式shell输入
listen_shell_input();
}
移植Windows
2019-06-15 17:07:10 +08:00
#endif
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
完善Hook与API
2019-05-20 16:26:04 +08:00
//设置退出信号处理函数
static semaphore sem;
signal(SIGINT, [](int) {
InfoL << "SIGINT:exit";
解决程序退出时资源释放顺序问题
2019-05-20 17:03:04 +08:00
signal(SIGINT, SIG_IGN);// 设置退出信号
完善Hook与API
2019-05-20 16:26:04 +08:00
sem.post();
});// 设置退出信号
移植Windows
2019-06-15 17:07:10 +08:00
#if !defined(_WIN32)
修复热加载不同配置文件的bug
2019-05-31 15:49:52 +08:00
signal(SIGHUP, [](int) { mediakit::loadIniConfig(ini_file.data()); });
移植Windows
2019-06-15 17:07:10 +08:00
#endif
完善Hook与API
2019-05-20 16:26:04 +08:00
sem.wait();
}
解决程序退出时资源释放顺序问题
2019-05-20 17:03:04 +08:00
unInstallWebApi();
unInstallWebHook();
休眠1秒后退出
2019-05-28 09:50:05 +08:00
//休眠1秒再退出，防止资源释放顺序错误
解决程序退出时资源释放顺序问题
2019-05-20 17:03:04 +08:00
InfoL << "程序退出中,请等待...";
休眠1秒后退出
2019-05-28 09:50:05 +08:00
sleep(1);
解决程序退出时资源释放顺序问题
2019-05-20 17:03:04 +08:00
InfoL << "程序退出完毕!";
添加完整的服务器用例
2019-05-20 11:22:59 +08:00
return 0;
}
Reference in New Issue Copy Permalink