mirror of https://github.com/ossrs/srs.git
fix #367, support nginx-rtmp exec. 3.0.1
winlin
10 years ago
parent
7de181004f
commit
c34b0c86e1
@ -0,0 +1,12 @@
|
||||
# the config for srs to support nginx-rtmp exec.
|
||||
# @see https://github.com/simple-rtmp-server/srs/wiki/v2_CN_NgExec
|
||||
# @see full.conf for detail config.
|
||||
|
||||
listen 1935;
|
||||
max_connections 1000;
|
||||
vhost __defaultVhost__ {
|
||||
exec {
|
||||
enabled on;
|
||||
publish ./objs/ffmpeg/bin/ffmpeg -f flv -i [url] -c copy -y ./[stream].flv;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,214 @@
|
||||
/*
|
||||
The MIT License (MIT)
|
||||
|
||||
Copyright (c) 2013-2015 SRS(simple-rtmp-server)
|
||||
|
||||
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 <srs_app_ng_exec.hpp>
|
||||
|
||||
using namespace std;
|
||||
|
||||
#include <srs_kernel_error.hpp>
|
||||
#include <srs_kernel_log.hpp>
|
||||
#include <srs_app_config.hpp>
|
||||
#include <srs_rtmp_stack.hpp>
|
||||
#include <srs_app_pithy_print.hpp>
|
||||
#include <srs_app_process.hpp>
|
||||
#include <srs_kernel_utility.hpp>
|
||||
#include <srs_kernel_consts.hpp>
|
||||
#include <srs_rtmp_utility.hpp>
|
||||
|
||||
// when error, ng-exec sleep for a while and retry.
|
||||
#define SRS_RTMP_EXEC_SLEEP_US (int64_t)(3*1000*1000LL)
|
||||
|
||||
SrsNgExec::SrsNgExec()
|
||||
{
|
||||
pthread = new SrsReusableThread("encoder", this, SRS_RTMP_EXEC_SLEEP_US);
|
||||
pprint = SrsPithyPrint::create_exec();
|
||||
}
|
||||
|
||||
SrsNgExec::~SrsNgExec()
|
||||
{
|
||||
on_unpublish();
|
||||
|
||||
srs_freep(pthread);
|
||||
srs_freep(pprint);
|
||||
}
|
||||
|
||||
int SrsNgExec::on_publish(SrsRequest* req)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
// when publish, parse the exec_publish.
|
||||
if ((ret = parse_exec_publish(req)) != ERROR_SUCCESS) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
// start thread to run all processes.
|
||||
if ((ret = pthread->start()) != ERROR_SUCCESS) {
|
||||
srs_error("st_thread_create failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
srs_trace("exec thread cid=%d, current_cid=%d", pthread->cid(), _srs_context->get_id());
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void SrsNgExec::on_unpublish()
|
||||
{
|
||||
pthread->stop();
|
||||
clear_exec_publish();
|
||||
}
|
||||
|
||||
int SrsNgExec::cycle()
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
// ignore when no exec.
|
||||
if (exec_publishs.empty()) {
|
||||
return ret;
|
||||
}
|
||||
|
||||
std::vector<SrsProcess*>::iterator it;
|
||||
for (it = exec_publishs.begin(); it != exec_publishs.end(); ++it) {
|
||||
SrsProcess* process = *it;
|
||||
|
||||
// start all processes.
|
||||
if ((ret = process->start()) != ERROR_SUCCESS) {
|
||||
srs_error("exec publish start failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
// check process status.
|
||||
if ((ret = process->cycle()) != ERROR_SUCCESS) {
|
||||
srs_error("exec publish cycle failed. ret=%d", ret);
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
|
||||
// pithy print
|
||||
show_exec_log_message();
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void SrsNgExec::on_thread_stop()
|
||||
{
|
||||
std::vector<SrsProcess*>::iterator it;
|
||||
for (it = exec_publishs.begin(); it != exec_publishs.end(); ++it) {
|
||||
SrsProcess* ep = *it;
|
||||
ep->stop();
|
||||
}
|
||||
}
|
||||
|
||||
int SrsNgExec::parse_exec_publish(SrsRequest* req)
|
||||
{
|
||||
int ret = ERROR_SUCCESS;
|
||||
|
||||
if (!_srs_config->get_exec_enabled(req->vhost)) {
|
||||
srs_trace("ignore disabled exec for vhost=%s", req->vhost.c_str());
|
||||
return ret;
|
||||
}
|
||||
|
||||
// stream name: vhost/app/stream for print
|
||||
input_stream_name = req->vhost;
|
||||
input_stream_name += "/";
|
||||
input_stream_name += req->app;
|
||||
input_stream_name += "/";
|
||||
input_stream_name += req->stream;
|
||||
|
||||
std::vector<SrsConfDirective*> eps = _srs_config->get_exec_publishs(req->vhost);
|
||||
for (int i = 0; i < (int)eps.size(); i++) {
|
||||
SrsConfDirective* ep = eps.at(i);
|
||||
SrsProcess* process = new SrsProcess();
|
||||
|
||||
std::string binary = ep->arg0();
|
||||
std::vector<std::string> argv;
|
||||
for (int i = 0; i < (int)ep->args.size(); i++) {
|
||||
std::string epa = ep->args.at(i);
|
||||
|
||||
if (srs_string_contains(epa, ">")) {
|
||||
vector<string> epas = srs_string_split(epa, ">");
|
||||
for (int j = 0; j < (int)epas.size(); j++) {
|
||||
argv.push_back(parse(req, epas.at(j)));
|
||||
if (j == 0) {
|
||||
argv.push_back(">");
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
argv.push_back(parse(req, epa));
|
||||
}
|
||||
|
||||
if ((ret = process->initialize(binary, argv)) != ERROR_SUCCESS) {
|
||||
srs_freep(process);
|
||||
srs_error("initialize process failed, binary=%s, vhost=%s, ret=%d", binary.c_str(), req->vhost.c_str(), ret);
|
||||
return ret;
|
||||
}
|
||||
|
||||
exec_publishs.push_back(process);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void SrsNgExec::clear_exec_publish()
|
||||
{
|
||||
std::vector<SrsProcess*>::iterator it;
|
||||
for (it = exec_publishs.begin(); it != exec_publishs.end(); ++it) {
|
||||
SrsProcess* ep = *it;
|
||||
srs_freep(ep);
|
||||
}
|
||||
}
|
||||
|
||||
void SrsNgExec::show_exec_log_message()
|
||||
{
|
||||
pprint->elapse();
|
||||
|
||||
// reportable
|
||||
if (pprint->can_print()) {
|
||||
// TODO: FIXME: show more info.
|
||||
srs_trace("-> "SRS_CONSTS_LOG_EXEC" time=%"PRId64", publish=%d, input=%s",
|
||||
pprint->age(), (int)exec_publishs.size(), input_stream_name.c_str());
|
||||
}
|
||||
}
|
||||
|
||||
string SrsNgExec::parse(SrsRequest* req, string tmpl)
|
||||
{
|
||||
string output = tmpl;
|
||||
|
||||
output = srs_string_replace(output, "[vhost]", req->vhost);
|
||||
output = srs_string_replace(output, "[port]", req->port);
|
||||
output = srs_string_replace(output, "[app]", req->app);
|
||||
output = srs_string_replace(output, "[stream]", req->stream);
|
||||
|
||||
output = srs_string_replace(output, "[tcUrl]", req->tcUrl);
|
||||
output = srs_string_replace(output, "[swfUrl]", req->swfUrl);
|
||||
output = srs_string_replace(output, "[pageUrl]", req->pageUrl);
|
||||
|
||||
if (output.find("[url]") != string::npos) {
|
||||
string url = srs_generate_rtmp_url(req->host, ::atoi(req->port.c_str()), req->vhost, req->app, req->stream);
|
||||
output = srs_string_replace(output, "[url]", url);
|
||||
}
|
||||
|
||||
return output;
|
||||
}
|
||||
|
||||
@ -0,0 +1,71 @@
|
||||
/*
|
||||
The MIT License (MIT)
|
||||
|
||||
Copyright (c) 2013-2015 SRS(simple-rtmp-server)
|
||||
|
||||
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.
|
||||
*/
|
||||
|
||||
#ifndef SRS_APP_NG_EXEC_HPP
|
||||
#define SRS_APP_NG_EXEC_HPP
|
||||
|
||||
/*
|
||||
#include <srs_app_ng_exec.hpp>
|
||||
*/
|
||||
#include <srs_core.hpp>
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
|
||||
#include <srs_app_thread.hpp>
|
||||
|
||||
class SrsRequest;
|
||||
class SrsPithyPrint;
|
||||
class SrsProcess;
|
||||
|
||||
/**
|
||||
* the ng-exec is the exec feature introduced by nginx-rtmp,
|
||||
* @see https://github.com/arut/nginx-rtmp-module/wiki/Directives#exec_push
|
||||
* @see https://github.com/simple-rtmp-server/srs/issues/367
|
||||
*/
|
||||
class SrsNgExec : public ISrsReusableThreadHandler
|
||||
{
|
||||
private:
|
||||
SrsReusableThread* pthread;
|
||||
SrsPithyPrint* pprint;
|
||||
std::string input_stream_name;
|
||||
std::vector<SrsProcess*> exec_publishs;
|
||||
public:
|
||||
SrsNgExec();
|
||||
virtual ~SrsNgExec();
|
||||
public:
|
||||
virtual int on_publish(SrsRequest* req);
|
||||
virtual void on_unpublish();
|
||||
// interface ISrsReusableThreadHandler.
|
||||
public:
|
||||
virtual int cycle();
|
||||
virtual void on_thread_stop();
|
||||
private:
|
||||
virtual int parse_exec_publish(SrsRequest* req);
|
||||
virtual void clear_exec_publish();
|
||||
virtual void show_exec_log_message();
|
||||
virtual std::string parse(SrsRequest* req, std::string tmpl);
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
Loading…
Reference in New Issue