diff --git a/src/conn_http_smooth.cpp b/src/conn_http_smooth.cpp new file mode 100644 index 00000000..c92d15e8 --- /dev/null +++ b/src/conn_http_smooth.cpp @@ -0,0 +1,435 @@ +/// \file conn_http_dynamic.cpp +/// Contains the main code for the HTTP Dynamic Connector + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +/// Holds everything unique to HTTP Dynamic Connector. +namespace Connector_HTTP{ + + std::string GenerateBootstrap(std::string & MovieId, JSON::Value & metadata, int fragnum, int starttime, int endtime){ + std::string empty; + + MP4::ASRT asrt; + if (starttime == 0){ + asrt.setUpdate(false); + }else{ + asrt.setUpdate(true); + } + asrt.setVersion(1); + asrt.setQualityEntry(empty, 0); + if (!metadata.isMember("keytime") || metadata["keytime"].size() == 0){ + asrt.setSegmentRun(1, 20000, 0); + }else{ + asrt.setSegmentRun(1, metadata["keytime"].size(), 0); + } + + + MP4::AFRT afrt; + if (starttime == 0){ + afrt.setUpdate(false); + }else{ + afrt.setUpdate(true); + } + afrt.setVersion(1); + afrt.setTimeScale(1000); + afrt.setQualityEntry(empty, 0); + MP4::afrt_runtable afrtrun; + if (!metadata.isMember("keytime") || metadata["keytime"].size() == 0){ + afrtrun.firstFragment = 1; + afrtrun.firstTimestamp = 0; + if (!metadata.isMember("video") || !metadata["video"].isMember("keyms") || metadata["video"]["keyms"].asInt() == 0){ + afrtrun.duration = 2000; + }else{ + afrtrun.duration = metadata["video"]["keyms"].asInt(); + } + afrt.setFragmentRun(afrtrun, 0); + }else{ + for (int i = 0; i < metadata["keytime"].size(); i++){ + afrtrun.firstFragment = i+1; + afrtrun.firstTimestamp = metadata["keytime"][i].asInt(); + if (i+1 < metadata["keytime"].size()){ + afrtrun.duration = metadata["keytime"][i+1].asInt() - metadata["keytime"][i].asInt(); + }else{ + if (metadata["lastms"].asInt()){ + afrtrun.duration = metadata["lastms"].asInt() - metadata["keytime"][i].asInt(); + }else{ + afrtrun.duration = 3000;//guess 3 seconds if unknown + } + } + afrt.setFragmentRun(afrtrun, i); + } + } + + MP4::ABST abst; + abst.setVersion(1); + abst.setBootstrapinfoVersion(1); + abst.setProfile(0); + if (starttime == 0){ + abst.setUpdate(false); + }else{ + abst.setUpdate(true); + } + abst.setTimeScale(1000); + if (metadata.isMember("length") && metadata["length"].asInt() > 0){ + abst.setLive(false); + if (metadata["lastms"].asInt()){ + abst.setCurrentMediaTime(metadata["lastms"].asInt()); + }else{ + abst.setCurrentMediaTime(1000*metadata["length"].asInt()); + } + }else{ + abst.setLive(true); + abst.setCurrentMediaTime(0xFFFFFFFF); + } + abst.setSmpteTimeCodeOffset(0); + abst.setMovieIdentifier(MovieId); + abst.setServerEntry(empty, 0); + abst.setQualityEntry(empty, 0); + abst.setDrmData(empty); + abst.setMetaData(empty); + abst.setSegmentRunTable(asrt, 0); + abst.setFragmentRunTable(afrt, 0); + + #if DEBUG >= 8 + std::cout << "Sending bootstrap:" << std::endl << abst.toPrettyString(0) << std::endl; + #endif + return std::string((char*)abst.asBox(), (int)abst.boxedSize()); + } + + + /// Returns a F4M-format manifest file + std::string BuildManifest(std::string & MovieId, JSON::Value & metadata){ + std::string Result; + if (metadata.isMember("length") && metadata["length"].asInt() > 0){ + Result="\n" + "\n" + "" + MovieId + "\n" + "" + metadata["video"]["width"].asString() + "\n" + "" + metadata["video"]["height"].asString() + "\n" + "" + metadata["length"].asString() + ".000\n" + "video/mp4\n" + "recorded\n" + "streaming\n" + "" + Base64::encode(GenerateBootstrap(MovieId, metadata, 1, 0, 0)) + "\n" + "\n" + "AgAKb25NZXRhRGF0YQgAAAAAAAl0cmFja2luZm8KAAAAAgMACXRpbWVzY2FsZQBA+GoAAAAAAAAGbGVuZ3RoAEGMcHoQAAAAAAhsYW5ndWFnZQIAA2VuZwARc2FtcGxlZGVzY3JpcHRpb24KAAAAAQMACnNhbXBsZXR5cGUCAARhdmMxAAAJAAAJAwAJdGltZXNjYWxlAEDncAAAAAAAAAZsZW5ndGgAQXtNvTAAAAAACGxhbmd1YWdlAgADZW5nABFzYW1wbGVkZXNjcmlwdGlvbgoAAAABAwAKc2FtcGxldHlwZQIABG1wNGEAAAkAAAkADWF1ZGlvY2hhbm5lbHMAQAAAAAAAAAAAD2F1ZGlvc2FtcGxlcmF0ZQBA53AAAAAAAAAOdmlkZW9mcmFtZXJhdGUAQDf/gi5SciUABmFhY2FvdABAAAAAAAAAAAAIYXZjbGV2ZWwAQD8AAAAAAAAACmF2Y3Byb2ZpbGUAQFNAAAAAAAAADGF1ZGlvY29kZWNpZAIABG1wNGEADHZpZGVvY29kZWNpZAIABGF2YzEABXdpZHRoAECQ4AAAAAAAAAZoZWlnaHQAQIMAAAAAAAAACmZyYW1lV2lkdGgAQJDgAAAAAAAAC2ZyYW1lSGVpZ2h0AECDAAAAAAAAAAxkaXNwbGF5V2lkdGgAQJDgAAAAAAAADWRpc3BsYXlIZWlnaHQAQIMAAAAAAAAADG1vb3Zwb3NpdGlvbgBBmxq2uAAAAAAIZHVyYXRpb24AQIKjqW3oyhIAAAk=\n" + "\n" + "\n"; + }else{ + Result="\n" + "\n" + "" + MovieId + "\n" + "video/mp4\n" + "live\n" + "streaming\n" + "" + Base64::encode(GenerateBootstrap(MovieId, metadata, 1, 0, 0)) + "\n" + "\n" + "\n"; + } + #if DEBUG >= 8 + std::cerr << "Sending this manifest:" << std::endl << Result << std::endl; + #endif + return Result; + }//BuildManifest + + /// Main function for Connector_HTTP_Dynamic + int Connector_HTTP_Dynamic(Socket::Connection conn){ + std::deque FlashBuf; + int FlashBufSize = 0; + long long int FlashBufTime = 0; + FLV::Tag tmp;//temporary tag + + DTSC::Stream Strm;//Incoming stream buffer. + HTTP::Parser HTTP_R, HTTP_S;//HTTP Receiver en HTTP Sender. + + bool ready4data = false;//Set to true when streaming is to begin. + bool pending_manifest = false; + bool receive_marks = false;//when set to true, this stream will ignore keyframes and instead use pause marks + bool inited = false; + Socket::Connection ss(-1); + std::string streamname; + std::string recBuffer = ""; + + std::string Quality; + int Segment = -1; + int ReqFragment = -1; + int temp; + int Flash_RequestPending = 0; + unsigned int lastStats = 0; + conn.setBlocking(false);//do not block on conn.spool() when no data is available + + while (conn.connected()){ + if (conn.spool() || conn.Received().size()){ + //make sure it ends in a \n + if (*(conn.Received().get().rbegin()) != '\n'){ + std::string tmp = conn.Received().get(); + conn.Received().get().clear(); + if (conn.Received().size()){ + conn.Received().get().insert(0, tmp); + }else{ + conn.Received().append(tmp); + } + } + if (HTTP_R.Read(conn.Received().get())){ + #if DEBUG >= 4 + std::cout << "Received request: " << HTTP_R.getUrl() << std::endl; + #endif + conn.setHost(HTTP_R.GetHeader("X-Origin")); + if (HTTP_R.url.find("f4m") == std::string::npos){ + streamname = HTTP_R.url.substr(1,HTTP_R.url.find("/",1)-1); + if (!ss){ + ss = Util::Stream::getStream(streamname); + if (!ss.connected()){ + #if DEBUG >= 1 + fprintf(stderr, "Could not connect to server!\n"); + #endif + ss.close(); + HTTP_S.Clean(); + HTTP_S.SetBody("No such stream is available on the system. Please try again.\n"); + conn.SendNow(HTTP_S.BuildResponse("404", "Not found")); + ready4data = false; + continue; + } + ss.setBlocking(false); + inited = true; + } + Quality = HTTP_R.url.substr( HTTP_R.url.find("/",1)+1 ); + Quality = Quality.substr(0, Quality.find("Seg")); + temp = HTTP_R.url.find("Seg") + 3; + Segment = atoi( HTTP_R.url.substr(temp,HTTP_R.url.find("-",temp)-temp).c_str()); + temp = HTTP_R.url.find("Frag") + 4; + ReqFragment = atoi( HTTP_R.url.substr(temp).c_str() ); + #if DEBUG >= 4 + printf( "Quality: %s, Seg %d Frag %d\n", Quality.c_str(), Segment, ReqFragment); + #endif + std::stringstream sstream; + sstream << "f " << ReqFragment << "\no \n"; + ss.SendNow(sstream.str().c_str()); + Flash_RequestPending++; + }else{ + streamname = HTTP_R.url.substr(1,HTTP_R.url.find("/",1)-1); + if (!Strm.metadata.isNull()){ + HTTP_S.Clean(); + HTTP_S.SetHeader("Content-Type","text/xml"); + HTTP_S.SetHeader("Cache-Control","no-cache"); + if (Strm.metadata.isMember("length")){receive_marks = true;} + std::string manifest = BuildManifest(streamname, Strm.metadata); + HTTP_S.SetBody(manifest); + conn.SendNow(HTTP_S.BuildResponse("200", "OK")); + #if DEBUG >= 3 + printf("Sent manifest\n"); + #endif + pending_manifest = false; + }else{ + pending_manifest = true; + } + } + ready4data = true; + HTTP_R.Clean(); //clean for any possible next requests + } + }else{ + if (Flash_RequestPending){ + usleep(1000);//sleep 1ms + }else{ + usleep(10000);//sleep 10ms + } + } + if (ready4data){ + if (!inited){ + //we are ready, connect the socket! + ss = Util::Stream::getStream(streamname); + if (!ss.connected()){ + #if DEBUG >= 1 + fprintf(stderr, "Could not connect to server!\n"); + #endif + ss.close(); + HTTP_S.Clean(); + HTTP_S.SetBody("No such stream is available on the system. Please try again.\n"); + conn.SendNow(HTTP_S.BuildResponse("404", "Not found")); + ready4data = false; + continue; + } + ss.setBlocking(false); + #if DEBUG >= 3 + fprintf(stderr, "Everything connected, starting to send video data...\n"); + #endif + inited = true; + } + unsigned int now = Util::epoch(); + if (now != lastStats){ + lastStats = now; + ss.SendNow(conn.getStats("HTTP_Dynamic").c_str()); + } + if (ss.spool()){ + while (Strm.parsePacket(ss.Received())){ + if (Strm.getPacket(0).isMember("time")){ + if (!Strm.metadata.isMember("firsttime")){ + Strm.metadata["firsttime"] = Strm.getPacket(0)["time"]; + }else{ + if (!Strm.metadata.isMember("length") || Strm.metadata["length"].asInt() == 0){ + Strm.getPacket(0)["time"] = Strm.getPacket(0)["time"].asInt() - Strm.metadata["firsttime"].asInt(); + } + } + Strm.metadata["lasttime"] = Strm.getPacket(0)["time"]; + } + if (pending_manifest){ + HTTP_S.Clean(); + HTTP_S.SetHeader("Content-Type","text/xml"); + HTTP_S.SetHeader("Cache-Control","no-cache"); + if (Strm.metadata.isMember("length")){receive_marks = true;} + std::string manifest = BuildManifest(streamname, Strm.metadata); + HTTP_S.SetBody(manifest); + conn.SendNow(HTTP_S.BuildResponse("200", "OK")); + #if DEBUG >= 3 + printf("Sent manifest\n"); + #endif + pending_manifest = false; + } + if (!receive_marks && Strm.metadata.isMember("length")){receive_marks = true;} + if ((Strm.getPacket(0).isMember("keyframe") && !receive_marks) || Strm.lastType() == DTSC::PAUSEMARK){ + #if DEBUG >= 4 + fprintf(stderr, "Received a %s fragment of %i bytes.\n", Strm.getPacket(0)["datatype"].asString().c_str(), FlashBufSize); + #endif + if (Flash_RequestPending > 0 && FlashBufSize){ + #if DEBUG >= 3 + fprintf(stderr, "Sending a fragment..."); + #endif + //static std::string btstrp; + //btstrp = GenerateBootstrap(streamname, Strm.metadata, ReqFragment, FlashBufTime, Strm.getPacket(0)["time"]); + HTTP_S.Clean(); + HTTP_S.SetHeader("Content-Type", "video/mp4"); + HTTP_S.SetBody(""); + HTTP_S.SetHeader("Content-Length", FlashBufSize+8);//32+33+btstrp.size()); + conn.SendNow(HTTP_S.BuildResponse("200", "OK")); + //conn.SendNow("\x00\x00\x00\x21" "afra\x00\x00\x00\x00\x00\x00\x00\x03\xE8\x00\x00\x00\x01", 21); + //unsigned long tmptime = htonl(FlashBufTime << 32); + //conn.SendNow((char*)&tmptime, 4); + //tmptime = htonl(FlashBufTime & 0xFFFFFFFF); + //conn.SendNow((char*)&tmptime, 4); + //tmptime = htonl(65); + //conn.SendNow((char*)&tmptime, 4); + + //conn.SendNow(btstrp); + + //conn.SendNow("\x00\x00\x00\x18moof\x00\x00\x00\x10mfhd\x00\x00\x00\x00", 20); + //unsigned long fragno = htonl(ReqFragment); + //conn.SendNow((char*)&fragno, 4); + unsigned long size = htonl(FlashBufSize+8); + conn.SendNow((char*)&size, 4); + conn.SendNow("mdat", 4); + while (FlashBuf.size() > 0){ + conn.SendNow(FlashBuf.front()); + FlashBuf.pop_front(); + } + Flash_RequestPending--; + #if DEBUG >= 3 + fprintf(stderr, "Done\n"); + #endif + } + FlashBuf.clear(); + FlashBufSize = 0; + } + if (Strm.lastType() == DTSC::VIDEO || Strm.lastType() == DTSC::AUDIO){ + if (FlashBufSize == 0){ + //fill buffer with init data, if needed. + if (Strm.metadata.isMember("audio") && Strm.metadata["audio"].isMember("init")){ + tmp.DTSCAudioInit(Strm); + tmp.tagTime(Strm.getPacket(0)["time"].asInt()); + FlashBuf.push_back(std::string(tmp.data, tmp.len)); + FlashBufSize += tmp.len; + } + if (Strm.metadata.isMember("video") && Strm.metadata["video"].isMember("init")){ + tmp.DTSCVideoInit(Strm); + tmp.tagTime(Strm.getPacket(0)["time"].asInt()); + FlashBuf.push_back(std::string(tmp.data, tmp.len)); + FlashBufSize += tmp.len; + } + FlashBufTime = Strm.getPacket(0)["time"].asInt(); + } + tmp.DTSCLoader(Strm); + FlashBuf.push_back(std::string(tmp.data, tmp.len)); + FlashBufSize += tmp.len; + } + } + if (pending_manifest && !Strm.metadata.isNull()){ + HTTP_S.Clean(); + HTTP_S.SetHeader("Content-Type","text/xml"); + HTTP_S.SetHeader("Cache-Control","no-cache"); + if (Strm.metadata.isMember("length")){receive_marks = true;} + std::string manifest = BuildManifest(streamname, Strm.metadata); + HTTP_S.SetBody(manifest); + conn.SendNow(HTTP_S.BuildResponse("200", "OK")); + #if DEBUG >= 3 + printf("Sent manifest\n"); + #endif + pending_manifest = false; + } + } + if (!ss.connected()){break;} + } + } + conn.close(); + ss.SendNow(conn.getStats("HTTP_Dynamic").c_str()); + ss.close(); + #if DEBUG >= 1 + if (FLV::Parse_Error){fprintf(stderr, "FLV Parser Error: %s\n", FLV::Error_Str.c_str());} + fprintf(stderr, "User %i disconnected.\n", conn.getSocket()); + if (inited){ + fprintf(stderr, "Status was: inited\n"); + }else{ + if (ready4data){ + fprintf(stderr, "Status was: ready4data\n"); + }else{ + fprintf(stderr, "Status was: connected\n"); + } + } + #endif + return 0; + }//Connector_HTTP_Dynamic main function + +};//Connector_HTTP_Dynamic namespace + +int main(int argc, char ** argv){ + Util::Config conf(argv[0], PACKAGE_VERSION); + conf.addConnectorOptions(1935); + conf.parseArgs(argc, argv); + Socket::Server server_socket = Socket::Server("/tmp/mist/http_dynamic"); + if (!server_socket.connected()){return 1;} + conf.activate(); + + while (server_socket.connected() && conf.is_active){ + Socket::Connection S = server_socket.accept(); + if (S.connected()){//check if the new connection is valid + pid_t myid = fork(); + if (myid == 0){//if new child, start MAINHANDLER + return Connector_HTTP::Connector_HTTP_Dynamic(S); + }else{//otherwise, do nothing or output debugging text + #if DEBUG >= 3 + fprintf(stderr, "Spawned new process %i for socket %i\n", (int)myid, S.getSocket()); + #endif + } + } + }//while connected + server_socket.close(); + return 0; +}//main