New input starting method based on JSON capabilities.
This commit is contained in:
parent
b4feaebbe8
commit
08953540f6
5 changed files with 159 additions and 119 deletions
|
@ -79,6 +79,7 @@ namespace DTSC {
|
||||||
Scan getMember(const char * indice);
|
Scan getMember(const char * indice);
|
||||||
Scan getMember(const char * indice, const unsigned int ind_len);
|
Scan getMember(const char * indice, const unsigned int ind_len);
|
||||||
Scan getIndice(unsigned int num);
|
Scan getIndice(unsigned int num);
|
||||||
|
std::string getIndiceName(unsigned int num);
|
||||||
unsigned int getSize();
|
unsigned int getSize();
|
||||||
|
|
||||||
char getType();
|
char getType();
|
||||||
|
|
|
@ -506,6 +506,33 @@ namespace DTSC {
|
||||||
return Scan();
|
return Scan();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Returns the name of the num-th member of this object.
|
||||||
|
/// Returns an empty string on error or when not an object.
|
||||||
|
std::string Scan::getIndiceName(unsigned int num) {
|
||||||
|
if (getType() == DTSC_OBJ || getType() == DTSC_CON) {
|
||||||
|
char * i = p + 1;
|
||||||
|
unsigned int arr_indice = 0;
|
||||||
|
//object, scan contents
|
||||||
|
while (i[0] + i[1] != 0 && i < p + len) { //while not encountering 0x0000 (we assume 0x0000EE)
|
||||||
|
if (i + 2 >= p + len) {
|
||||||
|
return "";//out of packet!
|
||||||
|
}
|
||||||
|
unsigned int strlen = i[0] * 256 + i[1];
|
||||||
|
i += 2;
|
||||||
|
if (arr_indice == num) {
|
||||||
|
return std::string(i, strlen);
|
||||||
|
} else {
|
||||||
|
arr_indice++;
|
||||||
|
i = skipDTSC(i + strlen, p + len);
|
||||||
|
if (!i) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
/// Returns the first byte of this DTSC value, or 0 on error.
|
/// Returns the first byte of this DTSC value, or 0 on error.
|
||||||
char Scan::getType() {
|
char Scan::getType() {
|
||||||
if (!p) {
|
if (!p) {
|
||||||
|
@ -555,13 +582,26 @@ namespace DTSC {
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Returns the string value of this DTSC string value.
|
/// Returns the string value of this DTSC string value.
|
||||||
/// Uses getString internally, does no conversion.
|
/// Uses getString internally, if a string.
|
||||||
|
/// Converts integer values to strings.
|
||||||
/// Returns an empty string on error.
|
/// Returns an empty string on error.
|
||||||
std::string Scan::asString() {
|
std::string Scan::asString() {
|
||||||
char * str;
|
switch (getType()) {
|
||||||
unsigned int strlen;
|
case DTSC_INT:{
|
||||||
getString(str, strlen);
|
std::stringstream st;
|
||||||
return std::string(str, strlen);
|
st << asInt();
|
||||||
|
return st.str();
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case DTSC_STR:{
|
||||||
|
char * str;
|
||||||
|
unsigned int strlen;
|
||||||
|
getString(str, strlen);
|
||||||
|
return std::string(str, strlen);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return "";
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Sets result to a pointer to the string, and strlen to the lenght of it.
|
/// Sets result to a pointer to the string, and strlen to the lenght of it.
|
||||||
|
@ -585,7 +625,7 @@ namespace DTSC {
|
||||||
JSON::Value Scan::asJSON(){
|
JSON::Value Scan::asJSON(){
|
||||||
JSON::Value result;
|
JSON::Value result;
|
||||||
unsigned int i = 0;
|
unsigned int i = 0;
|
||||||
JSON::fromDTMI2((const unsigned char*)p, len, i, result);
|
JSON::fromDTMI((const unsigned char*)p, len, i, result);
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
10
lib/ftp.cpp
10
lib/ftp.cpp
|
@ -22,16 +22,6 @@ FTP::User::User(Socket::Connection NewConnection, std::map<std::string, std::str
|
||||||
MyDir.SetVisibility("Converted", Filesystem::S_INACTIVE);
|
MyDir.SetVisibility("Converted", Filesystem::S_INACTIVE);
|
||||||
MyDir.SetVisibility("OnDemand", Filesystem::S_ACTIVE);
|
MyDir.SetVisibility("OnDemand", Filesystem::S_ACTIVE);
|
||||||
|
|
||||||
JSON::Value MyConfig = JSON::fromFile("/tmp/mist/streamlist");
|
|
||||||
for (JSON::ObjIter it = MyConfig["streams"].ObjBegin(); it != MyConfig["streams"].ObjEnd(); it++) {
|
|
||||||
std::string ThisStream = (*it).second["channel"]["URL"].toString();
|
|
||||||
ThisStream.erase(ThisStream.begin());
|
|
||||||
ThisStream.erase(ThisStream.end() - 1);
|
|
||||||
while (ThisStream.find('/') != std::string::npos) {
|
|
||||||
ThisStream.erase(0, ThisStream.find('/') + 1);
|
|
||||||
}
|
|
||||||
ActiveStreams.push_back(ThisStream);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
FTP::User::~User() {
|
FTP::User::~User() {
|
||||||
|
|
205
lib/stream.cpp
205
lib/stream.cpp
|
@ -13,6 +13,7 @@
|
||||||
#include "socket.h"
|
#include "socket.h"
|
||||||
#include "defines.h"
|
#include "defines.h"
|
||||||
#include "shared_memory.h"
|
#include "shared_memory.h"
|
||||||
|
#include "dtsc.h"
|
||||||
|
|
||||||
std::string Util::getTmpFolder() {
|
std::string Util::getTmpFolder() {
|
||||||
std::string dir;
|
std::string dir;
|
||||||
|
@ -46,7 +47,7 @@ std::string Util::getTmpFolder() {
|
||||||
/// letters to lowercase. If a '?' character is found, everything following
|
/// letters to lowercase. If a '?' character is found, everything following
|
||||||
/// that character is deleted. The original string is modified. If a '+'
|
/// that character is deleted. The original string is modified. If a '+'
|
||||||
/// exists, then only the part before the + is sanitized.
|
/// exists, then only the part before the + is sanitized.
|
||||||
void Util::Stream::sanitizeName(std::string & streamname) {
|
void Util::sanitizeName(std::string & streamname) {
|
||||||
//strip anything that isn't numbers, digits or underscores
|
//strip anything that isn't numbers, digits or underscores
|
||||||
size_t index = streamname.find('+');
|
size_t index = streamname.find('+');
|
||||||
if(index != std::string::npos){
|
if(index != std::string::npos){
|
||||||
|
@ -68,113 +69,127 @@ void Util::Stream::sanitizeName(std::string & streamname) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Util::Stream::getLive(std::string streamname) {
|
|
||||||
JSON::Value ServConf = JSON::fromFile(getTmpFolder() + "streamlist");
|
|
||||||
std::string bufferTime;
|
|
||||||
std::string debugLvl;
|
|
||||||
std::string player_bin = Util::getMyPath() + "MistInBuffer";
|
|
||||||
DEBUG_MSG(DLVL_WARN, "Starting %s -p -s %s", player_bin.c_str(), streamname.c_str());
|
|
||||||
char * argv[15] = {(char *)player_bin.c_str(), (char *)"-p", (char *)"-s", (char *)streamname.c_str()};
|
|
||||||
int argNum = 3;
|
|
||||||
if (ServConf["streams"][streamname].isMember("DVR")) {
|
|
||||||
bufferTime = ServConf["streams"][streamname]["DVR"].asString();
|
|
||||||
argv[++argNum] = (char *)"-b";
|
|
||||||
argv[++argNum] = (char *)bufferTime.c_str();
|
|
||||||
}
|
|
||||||
if (Util::Config::printDebugLevel != DEBUG){
|
|
||||||
debugLvl = JSON::Value((long long)Util::Config::printDebugLevel).asString();
|
|
||||||
argv[++argNum] = (char *)"--debug";
|
|
||||||
argv[++argNum] = (char *)debugLvl.c_str();
|
|
||||||
}
|
|
||||||
argv[++argNum] = (char *)0;
|
|
||||||
|
|
||||||
int pid = fork();
|
|
||||||
if (pid == -1) {
|
|
||||||
FAIL_MSG("Forking process for stream %s failed: %s", streamname.c_str(), strerror(errno));
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (pid == 0){
|
|
||||||
execvp(argv[0], argv);
|
|
||||||
FAIL_MSG("Starting process %s for stream %s failed: %s", argv[0], streamname.c_str(), strerror(errno));
|
|
||||||
_exit(42);
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Starts a process for a VoD stream.
|
/// Starts a process for a VoD stream.
|
||||||
bool Util::Stream::getVod(std::string filename, std::string streamname) {
|
bool Util::startInput(std::string streamname, std::string filename, bool forkFirst) {
|
||||||
std::string player_bin = Util::getMyPath() + "MistInDTSC";
|
IPC::sharedPage mistConfOut("!mistConfig", 4*1024*1024);
|
||||||
bool selected = false;
|
IPC::semaphore configLock("!mistConfLock", O_CREAT | O_RDWR, ACCESSPERMS, 1);
|
||||||
if (filename.substr(filename.size() - 5) == ".dtsc") {
|
configLock.wait();
|
||||||
player_bin = Util::getMyPath() + "MistInDTSC";
|
DTSC::Scan config = DTSC::Scan(mistConfOut.mapped, mistConfOut.len);
|
||||||
selected = true;
|
|
||||||
}
|
|
||||||
if (filename.substr(filename.size() - 4) == ".flv") {
|
|
||||||
player_bin = Util::getMyPath() + "MistInFLV";
|
|
||||||
selected = true;
|
|
||||||
}
|
|
||||||
INFO_MSG("Starting %s -p -s %s %s", player_bin.c_str(), streamname.c_str(), filename.c_str());
|
|
||||||
char * argv[15] = {(char *)player_bin.c_str(), (char *)"-p", (char *)"-s", (char *)streamname.c_str(), (char *)filename.c_str()};
|
|
||||||
int argNum = 4;
|
|
||||||
std::string debugLvl;
|
|
||||||
if (Util::Config::printDebugLevel != DEBUG){
|
|
||||||
debugLvl = JSON::Value((long long)Util::Config::printDebugLevel).asString();
|
|
||||||
argv[++argNum] = (char *)"--debug";
|
|
||||||
argv[++argNum] = (char *)debugLvl.c_str();
|
|
||||||
}
|
|
||||||
argv[++argNum] = (char *)0;
|
|
||||||
|
|
||||||
int pid = fork();
|
sanitizeName(streamname);
|
||||||
if (pid == -1) {
|
std::string smp = streamname.substr(0, streamname.find('+'));
|
||||||
FAIL_MSG("Forking process for stream %s failed: %s", streamname.c_str(), strerror(errno));
|
//check if smp (everything before +) exists
|
||||||
|
DTSC::Scan stream_cfg = config.getMember("streams").getMember(smp);
|
||||||
|
if (!stream_cfg){
|
||||||
|
DEBUG_MSG(DLVL_MEDIUM, "Stream %s not configured", streamname.c_str());
|
||||||
|
configLock.post();//unlock the config semaphore
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
if (pid == 0){
|
|
||||||
execvp(argv[0], argv);
|
//If starting without filename parameter, check if the stream is already active.
|
||||||
FAIL_MSG("Starting process %s for stream %s failed: %s", argv[0], streamname.c_str(), strerror(errno));
|
//If yes, don't activate again to prevent duplicate inputs.
|
||||||
_exit(42);
|
//It's still possible a duplicate starts anyway, this is caught in the inputs initializer.
|
||||||
}
|
//Note: this uses the _whole_ stream name, including + (if any).
|
||||||
return true;
|
//This means "test+a" and "test+b" have separate locks and do not interact with each other.
|
||||||
}
|
if (!filename.size()){
|
||||||
|
|
||||||
/// Probe for available streams. Currently first VoD, then Live.
|
|
||||||
bool Util::Stream::getStream(std::string streamname) {
|
|
||||||
sanitizeName(streamname);
|
|
||||||
JSON::Value ServConf = JSON::fromFile(getTmpFolder() + "streamlist");
|
|
||||||
std::string smp = streamname.substr(0,(streamname.find('+')));
|
|
||||||
//check if smp (everything before +) exists
|
|
||||||
///\todo Check if the input type used for this stream supports + syntax, if not, reject the request if smp != streamname.
|
|
||||||
if (ServConf["streams"].isMember(smp)){
|
|
||||||
//Check if the stream is already active, if yes, don't activate again.
|
|
||||||
//Note: this uses the _whole_ stream name, including + (if any).
|
|
||||||
//This means "test+a" and "test+b" have separate locks and do not interact with each other.
|
|
||||||
IPC::semaphore playerLock(std::string("/lock_" + streamname).c_str(), O_CREAT | O_RDWR, ACCESSPERMS, 1);
|
IPC::semaphore playerLock(std::string("/lock_" + streamname).c_str(), O_CREAT | O_RDWR, ACCESSPERMS, 1);
|
||||||
if (!playerLock.tryWait()) {
|
if (!playerLock.tryWait()) {
|
||||||
playerLock.close();
|
playerLock.close();
|
||||||
DEBUG_MSG(DLVL_MEDIUM, "Stream %s already active - not activating again", streamname.c_str());
|
DEBUG_MSG(DLVL_MEDIUM, "Stream %s already active - not activating again", streamname.c_str());
|
||||||
|
configLock.post();//unlock the config semaphore
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
playerLock.post();
|
playerLock.post();
|
||||||
playerLock.close();
|
playerLock.close();
|
||||||
if (ServConf["streams"][streamname]["source"].asString()[0] == '/') {
|
filename = stream_cfg.getMember("source").asString();
|
||||||
DEBUG_MSG(DLVL_MEDIUM, "Activating VoD stream %s", streamname.c_str());
|
}
|
||||||
return getVod(ServConf["streams"][streamname]["source"].asString(), streamname);
|
|
||||||
} else {
|
|
||||||
DEBUG_MSG(DLVL_MEDIUM, "Activating live stream %s", streamname.c_str());
|
std::string player_bin;
|
||||||
return getLive(streamname);
|
bool selected = false;
|
||||||
|
long long int curPrio = -1;
|
||||||
|
//check in curConf for capabilities-inputs-<naam>-priority/source_match
|
||||||
|
DTSC::Scan inputs = config.getMember("capabilities").getMember("inputs");
|
||||||
|
DTSC::Scan input;
|
||||||
|
unsigned int input_size = inputs.getSize();
|
||||||
|
for (unsigned int i = 0; i < input_size; ++i){
|
||||||
|
input = inputs.getIndice(i);
|
||||||
|
|
||||||
|
//if match voor current stream && priority is hoger dan wat we al hebben
|
||||||
|
if (curPrio < input.getMember("priority").asInt()){
|
||||||
|
std::string source = input.getMember("source_match").asString();
|
||||||
|
std::string front = source.substr(0,source.find('*'));
|
||||||
|
std::string back = source.substr(source.find('*')+1);
|
||||||
|
DEBUG_MSG(DLVL_MEDIUM, "Checking input %s: %s (%s)", inputs.getIndiceName(i).c_str(), input.getMember("name").asString().c_str(), source.c_str());
|
||||||
|
|
||||||
|
if (filename.substr(0,front.size()) == front && filename.substr(filename.size()-back.size()) == back){
|
||||||
|
player_bin = Util::getMyPath() + "MistIn" + input.getMember("name").asString();
|
||||||
|
curPrio = input.getMember("priority").asInt();
|
||||||
|
selected = true;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
DEBUG_MSG(DLVL_ERROR, "Stream not found: %s", streamname.c_str());
|
|
||||||
return false;
|
if (!selected){
|
||||||
}
|
configLock.post();//unlock the config semaphore
|
||||||
|
FAIL_MSG("No compatible input found for stream %s: %s", streamname.c_str(), filename.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
/// Create a stream on the system.
|
//copy the neccessary arguments to separate storage so we can unlock the config semaphore safely
|
||||||
/// Filters the streamname, removing invalid characters and
|
std::map<std::string, std::string> str_args;
|
||||||
/// converting all letters to lowercase.
|
//check required parameters
|
||||||
/// If a '?' character is found, everything following that character is deleted.
|
DTSC::Scan required = input.getMember("required");
|
||||||
Socket::Server Util::Stream::makeLive(std::string streamname) {
|
unsigned int req_size = required.getSize();
|
||||||
sanitizeName(streamname);
|
for (unsigned int i = 0; i < req_size; ++i){
|
||||||
std::string loc = getTmpFolder() + "stream_" + streamname;
|
std::string opt = required.getIndiceName(i);
|
||||||
//create and return the Socket::Server
|
if (!stream_cfg.getMember(opt)){
|
||||||
return Socket::Server(loc);
|
configLock.post();//unlock the config semaphore
|
||||||
|
FAIL_MSG("Required parameter %s for stream %s missing", opt.c_str(), streamname.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
str_args[required.getIndice(i).getMember("option").asString()] = stream_cfg.getMember(opt).asString();
|
||||||
|
}
|
||||||
|
//check optional parameters
|
||||||
|
DTSC::Scan optional = input.getMember("optional");
|
||||||
|
unsigned int opt_size = optional.getSize();
|
||||||
|
for (unsigned int i = 0; i < opt_size; ++i){
|
||||||
|
std::string opt = optional.getIndiceName(i);
|
||||||
|
if (stream_cfg.getMember(opt)){
|
||||||
|
str_args[optional.getIndice(i).getMember("option").asString()] = stream_cfg.getMember(opt).asString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//finally, unlock the config semaphore
|
||||||
|
configLock.post();
|
||||||
|
|
||||||
|
INFO_MSG("Starting %s -p -s %s %s", player_bin.c_str(), streamname.c_str(), filename.c_str());
|
||||||
|
char * argv[30] = {(char *)player_bin.c_str(), (char *)"-s", (char *)streamname.c_str(), (char *)filename.c_str()};
|
||||||
|
int argNum = 3;
|
||||||
|
std::string debugLvl;
|
||||||
|
if (Util::Config::printDebugLevel != DEBUG && !str_args.count("--debug")){
|
||||||
|
debugLvl = JSON::Value((long long)Util::Config::printDebugLevel).asString();
|
||||||
|
argv[++argNum] = (char *)"--debug";
|
||||||
|
argv[++argNum] = (char *)debugLvl.c_str();
|
||||||
|
}
|
||||||
|
for (std::map<std::string, std::string>::iterator it = str_args.begin(); it != str_args.end(); ++it){
|
||||||
|
argv[++argNum] = (char *)it->first.c_str();
|
||||||
|
argv[++argNum] = (char *)it->second.c_str();
|
||||||
|
}
|
||||||
|
argv[++argNum] = (char *)0;
|
||||||
|
|
||||||
|
int pid = 1;
|
||||||
|
if (forkFirst){
|
||||||
|
pid = fork();
|
||||||
|
if (pid == -1) {
|
||||||
|
FAIL_MSG("Forking process for stream %s failed: %s", streamname.c_str(), strerror(errno));
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (pid == 0){
|
||||||
|
execvp(argv[0], argv);
|
||||||
|
FAIL_MSG("Starting process %s for stream %s failed: %s", argv[0], streamname.c_str(), strerror(errno));
|
||||||
|
_exit(42);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
10
lib/stream.h
10
lib/stream.h
|
@ -7,12 +7,6 @@
|
||||||
|
|
||||||
namespace Util {
|
namespace Util {
|
||||||
std::string getTmpFolder();
|
std::string getTmpFolder();
|
||||||
class Stream {
|
void sanitizeName(std::string & streamname);
|
||||||
public:
|
bool startInput(std::string streamname, std::string filename = "", bool forkFirst = true);
|
||||||
static void sanitizeName(std::string & streamname);
|
|
||||||
static bool getLive(std::string streamname);
|
|
||||||
static bool getVod(std::string filename, std::string streamname);
|
|
||||||
static bool getStream(std::string streamname);
|
|
||||||
static Socket::Server makeLive(std::string streamname);
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue