92 lines
2.6 KiB
C++
92 lines
2.6 KiB
C++
#include "output_h264.h"
|
|
#include <mist/mp4_generic.h>
|
|
#include <mist/bitfields.h>
|
|
|
|
namespace Mist{
|
|
OutH264::OutH264(Socket::Connection &conn) : HTTPOutput(conn){
|
|
if (targetParams.count("keysonly")){
|
|
keysOnly = 1;
|
|
}
|
|
if (config->getString("target").size()){
|
|
if (!streamName.size()){
|
|
WARN_MSG("Recording unconnected H264 output to file! Cancelled.");
|
|
conn.close();
|
|
return;
|
|
}
|
|
if (config->getString("target") == "-"){
|
|
parseData = true;
|
|
wantRequest = false;
|
|
INFO_MSG("Outputting %s to stdout in H264 format", streamName.c_str());
|
|
return;
|
|
}
|
|
if (connectToFile(config->getString("target"))){
|
|
parseData = true;
|
|
wantRequest = false;
|
|
INFO_MSG("Recording %s to %s in H264 format", streamName.c_str(), config->getString("target").c_str());
|
|
}else{
|
|
conn.close();
|
|
}
|
|
}
|
|
}
|
|
|
|
void OutH264::init(Util::Config *cfg){
|
|
HTTPOutput::init(cfg);
|
|
capa["name"] = "H264";
|
|
capa["desc"] = "Enables HTTP protocol H264 Annex B streaming";
|
|
capa["url_rel"] = "/$.h264";
|
|
capa["url_match"] = "/$.h264";
|
|
capa["codecs"][0u][0u].append("H264");
|
|
|
|
JSON::Value opt;
|
|
opt["arg"] = "string";
|
|
opt["default"] = "";
|
|
opt["arg_num"] = 1ll;
|
|
opt["help"] = "Target filename to store H264 file as, or - for stdout.";
|
|
cfg->addOption("target", opt);
|
|
}
|
|
|
|
bool OutH264::isRecording(){return config->getString("target").size();}
|
|
|
|
void OutH264::sendNext(){
|
|
if (keysOnly && !thisPacket.getFlag("keyframe")){return;}
|
|
char *dataPointer = 0;
|
|
size_t len = 0;
|
|
thisPacket.getString("data", dataPointer, len);
|
|
|
|
unsigned int i = 0;
|
|
while (i + 4 < len){
|
|
uint32_t ThisNaluSize = Bit::btohl(dataPointer+i);
|
|
myConn.SendNow("\000\000\000\001", 4);
|
|
myConn.SendNow(dataPointer + i + 4, ThisNaluSize);
|
|
i += ThisNaluSize+4;
|
|
}
|
|
}
|
|
|
|
void OutH264::sendHeader(){
|
|
MP4::AVCC avccbox;
|
|
unsigned int mainTrack = getMainSelectedTrack();
|
|
if (mainTrack && myMeta.tracks.count(mainTrack)){
|
|
avccbox.setPayload(myMeta.tracks[mainTrack].init);
|
|
myConn.SendNow(avccbox.asAnnexB());
|
|
}
|
|
sentHeader = true;
|
|
}
|
|
|
|
void OutH264::onHTTP(){
|
|
std::string method = H.method;
|
|
//Set mode to key frames only
|
|
keysOnly = (H.GetVar("keysonly") != "");
|
|
H.Clean();
|
|
H.SetHeader("Content-Type", "video/H264");
|
|
H.protocol = "HTTP/1.0";
|
|
H.setCORSHeaders();
|
|
if (method == "OPTIONS" || method == "HEAD"){
|
|
H.SendResponse("200", "OK", myConn);
|
|
return;
|
|
}
|
|
H.SendResponse("200", "OK", myConn);
|
|
parseData = true;
|
|
wantRequest = false;
|
|
}
|
|
}
|
|
|