111 lines
3.7 KiB
C++
111 lines
3.7 KiB
C++
#include <mist/procs.h>
|
|
#include <mist/config.h>
|
|
#include <mist/timing.h>
|
|
#include "controller_streams.h"
|
|
#include "controller_storage.h"
|
|
#include <sys/stat.h>
|
|
|
|
namespace Controller {
|
|
|
|
std::map<std::string, int> lastBuffer; ///< Last moment of contact with all buffers.
|
|
|
|
bool streamsEqual(JSON::Value & one, JSON::Value & two){
|
|
if (one["channel"]["URL"] != two["channel"]["URL"]){
|
|
return false;
|
|
}
|
|
if (one["preset"]["cmd"] != two["preset"]["cmd"]){
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void startStream(std::string name, JSON::Value & data){
|
|
std::string URL = data["channel"]["URL"];
|
|
std::string preset = data["preset"]["cmd"];
|
|
std::string cmd1, cmd2, cmd3;
|
|
if (URL.substr(0, 4) == "push"){
|
|
std::string pusher = URL.substr(7);
|
|
cmd2 = "MistBuffer -s " + name + " " + pusher;
|
|
Util::Procs::Start(name, Util::getMyPath() + cmd2);
|
|
Log("BUFF", "(re)starting stream buffer " + name + " for push data from " + pusher);
|
|
}else{
|
|
if (URL.substr(0, 1) == "/"){
|
|
struct stat fileinfo;
|
|
if (stat(URL.c_str(), &fileinfo) != 0 || S_ISDIR(fileinfo.st_mode)){
|
|
Log("BUFF", "Warning for VoD stream " + name + "! File not found: " + URL);
|
|
data["error"] = "Not found: " + URL;
|
|
return;
|
|
}
|
|
cmd1 = "cat " + URL;
|
|
data["error"] = "Available";
|
|
return; //MistPlayer handles VoD
|
|
}else{
|
|
cmd1 = "ffmpeg -re -async 2 -i " + URL + " " + preset + " -f flv -";
|
|
cmd2 = "MistFLV2DTSC";
|
|
}
|
|
cmd3 = "MistBuffer -s " + name;
|
|
if (cmd2 != ""){
|
|
Util::Procs::Start(name, cmd1, Util::getMyPath() + cmd2, Util::getMyPath() + cmd3);
|
|
Log("BUFF", "(re)starting stream buffer " + name + " for ffmpeg data: " + cmd1);
|
|
}else{
|
|
Util::Procs::Start(name, cmd1, Util::getMyPath() + cmd3);
|
|
Log("BUFF", "(re)starting stream buffer " + name + " using input file " + URL);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CheckAllStreams(JSON::Value & data){
|
|
long long int currTime = Util::epoch();
|
|
for (JSON::ObjIter jit = data.ObjBegin(); jit != data.ObjEnd(); jit++){
|
|
if ( !Util::Procs::isActive(jit->first)){
|
|
startStream(jit->first, jit->second);
|
|
}
|
|
if (currTime - lastBuffer[jit->first] > 5){
|
|
if (jit->second.isMember("error") && jit->second["error"].asString() != ""){
|
|
jit->second["online"] = jit->second["error"];
|
|
}else{
|
|
jit->second["online"] = 0;
|
|
}
|
|
}else{
|
|
jit->second["online"] = 1;
|
|
}
|
|
}
|
|
static JSON::Value strlist;
|
|
bool changed = false;
|
|
if (strlist["config"] != Storage["config"]){
|
|
strlist["config"] = Storage["config"];
|
|
changed = true;
|
|
}
|
|
if (strlist["streams"] != Storage["streams"]){
|
|
strlist["streams"] = Storage["streams"];
|
|
changed = true;
|
|
}
|
|
if (changed){
|
|
WriteFile("/tmp/mist/streamlist", strlist.toString());
|
|
}
|
|
}
|
|
|
|
void CheckStreams(JSON::Value & in, JSON::Value & out){
|
|
bool changed = false;
|
|
for (JSON::ObjIter jit = in.ObjBegin(); jit != in.ObjEnd(); jit++){
|
|
if (out.isMember(jit->first)){
|
|
if ( !streamsEqual(jit->second, out[jit->first])){
|
|
Log("STRM", std::string("Updated stream ") + jit->first);
|
|
Util::Procs::Stop(jit->first);
|
|
startStream(jit->first, jit->second);
|
|
}
|
|
}else{
|
|
Log("STRM", std::string("New stream ") + jit->first);
|
|
startStream(jit->first, jit->second);
|
|
}
|
|
}
|
|
for (JSON::ObjIter jit = out.ObjBegin(); jit != out.ObjEnd(); jit++){
|
|
if ( !in.isMember(jit->first)){
|
|
Log("STRM", std::string("Deleted stream ") + jit->first);
|
|
Util::Procs::Stop(jit->first);
|
|
}
|
|
}
|
|
out = in;
|
|
}
|
|
|
|
} //Controller namespace
|