winlin

reap the transcode process

... ... @@ -72,18 +72,19 @@ vhost __defaultVhost__ {
vhost dev {
enabled on;
gop_cache on;
hls on;
hls off;
hls_path ./objs/nginx/html;
hls_fragment 5;
hls_window 30;
#forward 127.0.0.1:19350;
forward 127.0.0.1:1936;
#forward 127.0.0.1:1936;
transcode {
enabled off;
enabled on;
ffmpeg ./objs/ffmpeg/bin/ffmpeg;
engine dev {
enabled on;
vfilter {
t 10;
}
vcodec libx264;
vbitrate 300;
... ...
... ... @@ -25,6 +25,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#include <stdlib.h>
#include <unistd.h>
#include <sys/wait.h>
#include <algorithm>
... ... @@ -356,6 +357,34 @@ int SrsFFMPEG::start()
return ret;
}
int SrsFFMPEG::cycle()
{
int ret = ERROR_SUCCESS;
if (!started) {
return ret;
}
int status = 0;
pid_t p = waitpid(pid, &status, WNOHANG);
if (p < 0) {
ret = ERROR_SYSTEM_WAITPID;
srs_error("transcode waitpid failed, pid=%d, ret=%d", pid, ret);
return ret;
}
if (p == 0) {
srs_info("transcode process pid=%d is running.", pid);
return ret;
}
srs_trace("transcode process pid=%d terminate, restart it.", pid);
started = false;
return ret;
}
void SrsFFMPEG::stop()
{
if (!started) {
... ... @@ -538,17 +567,22 @@ int SrsEncoder::cycle()
{
int ret = ERROR_SUCCESS;
// start all ffmpegs.
std::vector<SrsFFMPEG*>::iterator it;
for (it = ffmpegs.begin(); it != ffmpegs.end(); ++it) {
SrsFFMPEG* ffmpeg = *it;
// start all ffmpegs.
if ((ret = ffmpeg->start()) != ERROR_SUCCESS) {
srs_error("ffmpeg start failed. ret=%d", ret);
return ret;
}
}
// TODO: cycle all processes.
// check ffmpeg status.
if ((ret = ffmpeg->cycle()) != ERROR_SUCCESS) {
srs_error("ffmpeg cycle failed. ret=%d", ret);
return ret;
}
}
return ret;
}
... ...
... ... @@ -73,6 +73,7 @@ public:
public:
virtual int initialize(SrsRequest* req, SrsConfDirective* engine);
virtual int start();
virtual int cycle();
virtual void stop();
};
... ...
... ... @@ -86,6 +86,7 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#define ERROR_SYSTEM_IP_INVALID 411
#define ERROR_SYSTEM_CONFIG_TOO_LARGE 412
#define ERROR_SYSTEM_FORWARD_LOOP 413
#define ERROR_SYSTEM_WAITPID 414
// see librtmp.
// failed when open ssl create the dh
... ...
... ... @@ -502,6 +502,7 @@ SrsM3u8Muxer::SrsM3u8Muxer()
video_stream_dts = 0;
file_index = 0;
current = NULL;
_is_open = false;
}
SrsM3u8Muxer::~SrsM3u8Muxer()
... ... @@ -516,6 +517,11 @@ SrsM3u8Muxer::~SrsM3u8Muxer()
srs_freep(current);
}
bool SrsM3u8Muxer::is_open()
{
return _is_open;
}
int SrsM3u8Muxer::update_config(
std::string _app, std::string _stream,
std::string path, int fragment, int window
... ... @@ -571,6 +577,8 @@ int SrsM3u8Muxer::segment_open()
srs_info("open HLS muxer success. vhost=%s, path=%s",
vhost.c_str(), current->full_path.c_str());
_is_open = true;
return ret;
}
... ... @@ -709,6 +717,8 @@ int SrsM3u8Muxer::segment_close()
return ret;
}
_is_open = false;
return ret;
}
... ... @@ -1171,12 +1181,14 @@ void SrsHls::on_unpublish()
{
int ret = ERROR_SUCCESS;
if (muxer->is_open()) {
// close muxer when unpublish.
ret = ts_cache->flush_audio(muxer);
ret += muxer->segment_close();
if (ret != ERROR_SUCCESS) {
srs_error("ignore m3u8 muxer flush/close audio failed. ret=%d", ret);
return;
}
}
hls_enabled = false;
... ...
... ... @@ -131,6 +131,7 @@ private:
int hls_fragment;
int hls_window;
private:
bool _is_open;
int file_index;
std::string m3u8;
private:
... ... @@ -148,6 +149,7 @@ public:
SrsM3u8Muxer();
virtual ~SrsM3u8Muxer();
public:
virtual bool is_open();
virtual int update_config(std::string _app, std::string _stream, std::string path, int fragment, int window);
virtual int segment_open();
virtual int flush_audio(SrsMpegtsFrame* af, SrsCodecBuffer* ab);
... ...