/// Entry point for DTSCFix, simply calls Converters::DTSCFix(). int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("filename", JSON::fromString("{\"arg_num\":1, \"arg\":\"string\", \"help\":\"Filename of the file to attempt to fix.\"}")); conf.addOption("force", JSON::fromString("{\"short\":\"f\", \"long\":\"force\", \"default\":0, \"help\":\"Force fixing.\"}")); conf.parseArgs(argc, argv); return Converters::DTSCFix(conf); } //main
int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("detail", JSON::fromString( "{\"arg_num\":1, \"arg\":\"integer\", \"default\":0, \"help\":\"Bitmask, 1 = Reconstruct, 2 = Explicit media info, 4 = Verbose chunks\"}")); conf.parseArgs(argc, argv); return Analysers::analyseRTMP(conf); } //main
///\brief Entry point for FLV2DTSC, simply calls Converters::FLV2DTSC(). int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("output", JSON::fromString( "{\"long\":\"output\", \"value\":[\"stdout\"], \"short\":\"o\", \"arg\":\"string\", \"help\":\"Name of the outputfile or stdout for standard output.\"}")); conf.parseArgs(argc, argv); if (conf.getString("output") == "stdout"){ return Converters::FLV2DTSC(std::cout); } std::ofstream oFile(conf.getString("output").c_str()); return Converters::FLV2DTSC(oFile); } //main
///\brief Debugging tool for DTSC data. /// /// Expects DTSC data in a file given on the command line, outputs human-readable information to stderr. ///\param conf The configuration parsed from the commandline. ///\return The return code of the analyser. int analyseDTSC(Util::Config conf){ DTSC::File F(conf.getString("filename")); if (!F){ std::cerr << "Not a valid DTSC file" << std::endl; return 1; } F.getMeta().toPrettyString(std::cout,0, 0x03); int bPos = 0; F.seek_bpos(0); F.parseNext(); while (F.getPacket()){ switch (F.getPacket().getVersion()){ case DTSC::DTSC_V1: { std::cout << "DTSCv1 packet: " << F.getPacket().getScan().toPrettyString() << std::endl; break; } case DTSC::DTSC_V2: { std::cout << "DTSCv2 packet (Track " << F.getPacket().getTrackId() << ", time " << F.getPacket().getTime() << "): " << F.getPacket().getScan().toPrettyString() << std::endl; break; } case DTSC::DTSC_HEAD: { std::cout << "DTSC header: " << F.getPacket().getScan().toPrettyString() << std::endl; break; } default: DEBUG_MSG(DLVL_WARN,"Invalid dtsc packet @ bpos %d", bPos); break; } bPos = F.getBytePos(); F.parseNext(); } return 0; }
/// Starts a loop, waiting for connections to send data to. int Start(int argc, char ** argv) { Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("stream_name", JSON::fromString("{\"arg_num\":1, \"arg\":\"string\", \"help\":\"Name of the stream this buffer will be providing.\"}")); conf.addOption("awaiting_ip", JSON::fromString("{\"arg_num\":2, \"arg\":\"string\", \"default\":\"\", \"help\":\"IP address to expect incoming data from. This will completely disable reading from standard input if used.\"}")); conf.addOption("reportstats", JSON::fromString("{\"default\":0, \"help\":\"Report stats to a controller process.\", \"short\":\"s\", \"long\":\"reportstats\"}")); conf.parseArgs(argc, argv); std::string name = conf.getString("stream_name"); SS = Util::Stream::makeLive(name); if (!SS.connected()) { perror("Could not create stream socket"); return 1; } conf.activate(); thisStream = Stream::get(); thisStream->setName(name); Socket::Connection incoming; Socket::Connection std_input(fileno(stdin)); tthread::thread * StatsThread = 0; if (conf.getBool("reportstats")){StatsThread = new tthread::thread(handleStats, 0);} tthread::thread * StdinThread = 0; std::string await_ip = conf.getString("awaiting_ip"); if (await_ip == ""){ StdinThread = new tthread::thread(handleStdin, 0); }else{ thisStream->setWaitingIP(await_ip); StdinThread = new tthread::thread(handlePushin, 0); } while (buffer_running && SS.connected() && conf.is_active){ //check for new connections, accept them if there are any //starts a thread for every accepted connection incoming = SS.accept(true); if (incoming.connected()){ user * usr_ptr = new user(incoming); thisStream->addUser(usr_ptr); usr_ptr->Thread = new tthread::thread(handleUser, (void *)usr_ptr); } }//main loop // disconnect listener buffer_running = false; std::cout << "End of input file - buffer shutting down" << std::endl; SS.close(); if (StatsThread){StatsThread->join();} StdinThread->join(); delete thisStream; return 0; }
/// Reads FLV from stdin and outputs human-readable information to stderr. int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.parseArgs(argc, argv); FLV::Tag FLV_in; // Temporary storage for incoming FLV data. std::ofstream vData("vData"); std::ofstream aData("aData"); while ( !feof(stdin)){ if (FLV_in.FileLoader(stdin)){ std::cout << "Tag: " << FLV_in.tagType() << "\n\tTime: " << FLV_in.tagTime() << std::endl; if (FLV_in.data[0] == 0x08){ //Audio aData.write(FLV_in.data + 13, FLV_in.len - 17); } if (FLV_in.data[0] == 0x09){ //Video vData.write(FLV_in.data + 16, FLV_in.len - 20); } } } vData.close(); aData.close(); return 0; }
int DTSC2OGG(Util::Config & conf){ DTSC::File DTSCFile(conf.getString("filename")); srand (Util::getMS());//randomising with milliseconds from boot std::vector<unsigned int> curSegTable; OGG::converter oggMeta; //Creating ID headers for theora and vorbis DTSC::readOnlyMeta fileMeta = DTSCFile.getMeta(); DTSC::Meta giveMeta(fileMeta); oggMeta.readDTSCHeader(giveMeta); std::cout << oggMeta.parsedPages;//outputting header pages //create DTSC in OGG pages DTSCFile.parseNext(); std::map< long long int, std::vector<JSON::Value> > DTSCBuffer; OGG::Page curOggPage; while(DTSCFile.getJSON()){ std::string tmpString; oggMeta.readDTSCVector(DTSCFile.getJSON(), tmpString); std::cout << tmpString; DTSCFile.parseNext(); } return 0; }
int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0]); conf.parseArgs(argc, argv); return Analysers::analyseAMF(); }
int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.parseArgs(argc,argv); return Converters::OGG2DTSC(); }
/// Debugging tool for MP4 data. /// Expects MP4 data through stdin, outputs human-readable information to stderr. int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.parseArgs(argc, argv); return Analysers::analyseMP4(); }
/// Reads an DTSC file and prints all readable data about it int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0]); conf.addOption("filename", JSON::fromString("{\"arg_num\":1, \"arg\":\"string\", \"help\":\"Filename of the DTSC file to analyse.\"}")); conf.parseArgs(argc, argv); return Analysers::analyseDTSC(conf); } //main
int DTSCMerge(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("output", JSON::fromString("{\"arg_num\":1, \"arg\":\"string\", \"help\":\"Filename of the output file.\"}")); conf.addOption("input", JSON::fromString("{\"arg_num\":2, \"arg\":\"string\", \"help\":\"Filename of the first input file.\"}")); conf.addOption("[additional_inputs ...]", JSON::fromString("{\"arg_num\":3, \"default\":\"\", \"arg\":\"string\", \"help\":\"Filenames of any number of aditional inputs.\"}")); conf.parseArgs(argc, argv); DTSC::File outFile; JSON::Value meta; DTSC::Meta newMeta; std::map<std::string,std::map<int, int> > trackMapping; bool fullSort = true; std::map<std::string, DTSC::File> inFiles; std::map<std::string, DTSC::Meta> metaData; std::string outFileName = argv[1]; std::string tmpFileName; for (int i = 2; i < argc; i++){ tmpFileName = argv[i]; if (tmpFileName == outFileName){ fullSort = false; }else{ DTSC::File F(tmpFileName); if (!F.getMeta().isFixed()){ std::cerr << tmpFileName << " has not been run through DTSCFix yet." << std::endl; return 1; } inFiles[tmpFileName] = F; } } if (fullSort){ outFile = DTSC::File(outFileName, true); }else{ outFile = DTSC::File(outFileName); if ( !outFile.getMeta().isFixed()){ std::cerr << outFileName << " has not been run through DTSCFix yet." << std::endl; return 1; } meta = outFile.getMeta().toJSON(); newMeta = DTSC::Meta(meta); if (meta.isMember("tracks") && meta["tracks"].size() > 0){ for (JSON::ObjIter trackIt = meta["tracks"].ObjBegin(); trackIt != meta["tracks"].ObjEnd(); trackIt++){ int nxtMap = getNextFree(trackMapping); trackMapping[argv[1]].insert(std::pair<int,int>(trackIt->second["trackid"].asInt(),nxtMap)); newMeta.tracks[nxtMap].trackID = nxtMap; } } } std::multimap<int,keyframeInfo> allSorted; for (std::map<std::string,DTSC::File>::iterator it = inFiles.begin(); it != inFiles.end(); it++){ DTSC::Meta tmpMeta(it->second.getMeta()); for (std::map<int,DTSC::Track>::iterator trackIt = tmpMeta.tracks.begin(); trackIt != tmpMeta.tracks.end(); trackIt++){ long long int oldID = trackIt->first; long long int mappedID = getNextFree(trackMapping); trackMapping[it->first].insert(std::pair<int,int>(oldID,mappedID)); for (std::deque<DTSC::Key>::iterator keyIt = trackIt->second.keys.begin(); keyIt != trackIt->second.keys.end(); keyIt++){ keyframeInfo tmpInfo; tmpInfo.fileName = it->first; tmpInfo.trackID = oldID; tmpInfo.keyTime = keyIt->getTime(); tmpInfo.keyBPos = keyIt->getBpos(); tmpInfo.keyNum = keyIt->getNumber(); tmpInfo.keyLen = keyIt->getLength(); if ((keyIt + 1) != trackIt->second.keys.end()){ tmpInfo.endBPos = (keyIt + 1)->getBpos(); }else{ tmpInfo.endBPos = it->second.getBytePosEOF(); } allSorted.insert(std::pair<int,keyframeInfo>(keyIt->getTime(),tmpInfo)); } newMeta.tracks[mappedID] = trackIt->second; newMeta.tracks[mappedID].trackID = mappedID; newMeta.tracks[mappedID].reset(); } } if (fullSort){ meta.null(); meta["moreheader"] = 0ll; std::string tmpWrite = meta.toPacked(); outFile.writeHeader(tmpWrite,true); } std::set<int> trackSelector; for (std::multimap<int,keyframeInfo>::iterator sortIt = allSorted.begin(); sortIt != allSorted.end(); sortIt++){ trackSelector.clear(); trackSelector.insert(sortIt->second.trackID); inFiles[sortIt->second.fileName].selectTracks(trackSelector); inFiles[sortIt->second.fileName].seek_time(sortIt->second.keyTime); inFiles[sortIt->second.fileName].seekNext(); while (inFiles[sortIt->second.fileName].getPacket() && inFiles[sortIt->second.fileName].getBytePos() <= sortIt->second.endBPos && !inFiles[sortIt->second.fileName].reachedEOF()){ if (inFiles[sortIt->second.fileName].getPacket().getTrackId() == sortIt->second.trackID){ JSON::Value tmp = inFiles[sortIt->second.fileName].getPacket().toJSON(); tmp["trackid"] = trackMapping[sortIt->second.fileName][sortIt->second.trackID]; outFile.writePacket(tmp); } inFiles[sortIt->second.fileName].seekNext(); } } if (fullSort || (meta.isMember("merged") && meta["merged"])){ newMeta.merged = 1; }else{ newMeta.merged = 0; } std::string writeMeta = newMeta.toJSON().toPacked(); meta["moreheader"] = outFile.addHeader(writeMeta); writeMeta = meta.toPacked(); outFile.writeHeader(writeMeta); return 0; }
///\brief Debugging tool for RTMP data. /// ///Expects RTMP data of one side of the conversation through stdin, outputs human-readable information to stderr. /// ///Will output FLV file to stdout, if available. /// ///Automatically skips the handshake data. ///\param conf The configuration parsed from the commandline. ///\return The return code of the analyser. int analyseRTMP(Util::Config conf){ int Detail = conf.getInteger("detail"); if (Detail > 0){ fprintf(stderr, "Detail level set:\n"); if (Detail & DETAIL_RECONSTRUCT){ fprintf(stderr, " - Will reconstuct FLV file to stdout\n"); std::cout.write(FLV::Header, 13); } if (Detail & DETAIL_EXPLICIT){ fprintf(stderr, " - Will list explicit video/audio data information\n"); } if (Detail & DETAIL_VERBOSE){ fprintf(stderr, " - Will list verbose chunk information\n"); } } std::string inbuffer; inbuffer.reserve(3073); while (std::cin.good() && inbuffer.size() < 3073){ inbuffer += std::cin.get(); } //read all of std::cin to temp inbuffer.erase(0, 3073); //strip the handshake part RTMPStream::Chunk next; FLV::Tag F; //FLV holder AMF::Object amfdata("empty", AMF::AMF0_DDV_CONTAINER); AMF::Object3 amf3data("empty", AMF::AMF3_DDV_CONTAINER); while (std::cin.good() || inbuffer.size()){ if (next.Parse(inbuffer)){ if (Detail & DETAIL_VERBOSE){ fprintf(stderr, "Chunk info: [%#2X] CS ID %u, timestamp %u, len %u, type ID %u, Stream ID %u\n", next.headertype, next.cs_id, next.timestamp, next.len, next.msg_type_id, next.msg_stream_id); } switch (next.msg_type_id){ case 0: //does not exist fprintf(stderr, "Error chunk - %i, %i, %i, %i, %i\n", next.cs_id, next.timestamp, next.real_len, next.len_left, next.msg_stream_id); //return 0; break; //happens when connection breaks unexpectedly case 1: //set chunk size RTMPStream::chunk_rec_max = ntohl(*(int*)next.data.c_str()); fprintf(stderr, "CTRL: Set chunk size: %i\n", RTMPStream::chunk_rec_max); break; case 2: //abort message - we ignore this one fprintf(stderr, "CTRL: Abort message: %i\n", ntohl(*(int*)next.data.c_str())); //4 bytes of stream id to drop break; case 3: //ack RTMPStream::snd_window_at = ntohl(*(int*)next.data.c_str()); fprintf(stderr, "CTRL: Acknowledgement: %i\n", RTMPStream::snd_window_at); break; case 4: { short int ucmtype = ntohs(*(short int*)next.data.c_str()); switch (ucmtype){ case 0: fprintf(stderr, "CTRL: User control message: stream begin %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; case 1: fprintf(stderr, "CTRL: User control message: stream EOF %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; case 2: fprintf(stderr, "CTRL: User control message: stream dry %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; case 3: fprintf(stderr, "CTRL: User control message: setbufferlen %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; case 4: fprintf(stderr, "CTRL: User control message: streamisrecorded %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; case 6: fprintf(stderr, "CTRL: User control message: pingrequest %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; case 7: fprintf(stderr, "CTRL: User control message: pingresponse %u\n", ntohl(*(unsigned int*)(next.data.c_str()+2))); break; default: fprintf(stderr, "CTRL: User control message: UNKNOWN %hu - %u\n", ucmtype, ntohl(*(unsigned int*)(next.data.c_str()+2))); break; } } break; case 5: //window size of other end RTMPStream::rec_window_size = ntohl(*(int*)next.data.c_str()); RTMPStream::rec_window_at = RTMPStream::rec_cnt; fprintf(stderr, "CTRL: Window size: %i\n", RTMPStream::rec_window_size); break; case 6: RTMPStream::snd_window_size = ntohl(*(int*)next.data.c_str()); //4 bytes window size, 1 byte limit type (ignored) fprintf(stderr, "CTRL: Set peer bandwidth: %i\n", RTMPStream::snd_window_size); break; case 8: if (Detail & (DETAIL_EXPLICIT | DETAIL_RECONSTRUCT)){ F.ChunkLoader(next); if (Detail & DETAIL_EXPLICIT){ fprintf(stderr, "Received %i bytes audio data\n", next.len); std::cerr << "Got a " << F.len << " bytes " << F.tagType() << " FLV tag of time " << F.tagTime() << "." << std::endl; } if (Detail & DETAIL_RECONSTRUCT){ std::cout.write(F.data, F.len); } } break; case 9: if (Detail & (DETAIL_EXPLICIT | DETAIL_RECONSTRUCT)){ F.ChunkLoader(next); if (Detail & DETAIL_EXPLICIT){ fprintf(stderr, "Received %i bytes video data\n", next.len); std::cerr << "Got a " << F.len << " bytes " << F.tagType() << " FLV tag of time " << F.tagTime() << "." << std::endl; } if (Detail & DETAIL_RECONSTRUCT){ std::cout.write(F.data, F.len); } } break; case 15: fprintf(stderr, "Received AFM3 data message\n"); break; case 16: fprintf(stderr, "Received AFM3 shared object\n"); break; case 17: { fprintf(stderr, "Received AFM3 command message:\n"); char soort = next.data[0]; next.data = next.data.substr(1); if (soort == 0){ amfdata = AMF::parse(next.data); std::cerr << amfdata.Print() << std::endl; }else{ amf3data = AMF::parse3(next.data); amf3data.Print(); } } break; case 18: { fprintf(stderr, "Received AFM0 data message (metadata):\n"); amfdata = AMF::parse(next.data); amfdata.Print(); if (Detail & DETAIL_RECONSTRUCT){ F.ChunkLoader(next); std::cout.write(F.data, F.len); } } break; case 19: fprintf(stderr, "Received AFM0 shared object\n"); break; case 20: { //AMF0 command message fprintf(stderr, "Received AFM0 command message:\n"); amfdata = AMF::parse(next.data); std::cerr << amfdata.Print() << std::endl; } break; case 22: fprintf(stderr, "Received aggregate message\n"); break; default: fprintf(stderr, "Unknown chunk received! Probably protocol corruption, stopping parsing of incoming data.\n"); return 1; break; } //switch for type of chunk }else{ //if chunk parsed if (std::cin.good()){ inbuffer += std::cin.get(); }else{ inbuffer.clear(); } } }//while std::cin.good() fprintf(stderr, "No more readable data\n"); return 0; }
///\brief Reads a DTSC file and attempts to fix the metadata in it. ///\param conf The current configuration of the program. ///\return The return code for the fixed program. int DTSCFix(Util::Config & conf){ DTSC::File F(conf.getString("filename")); F.seek_bpos(0); F.parseNext(); JSON::Value oriheader = F.getJSON(); JSON::Value meta = F.getMeta(); JSON::Value pack; if ( !oriheader.isMember("moreheader")){ std::cerr << "This file is too old to fix - please reconvert." << std::endl; return 1; } if (DTSC::isFixed(meta) && !conf.getBool("force")){ std::cerr << "This file was already fixed or doesn't need fixing - cancelling." << std::endl; return 0; } meta.removeMember("isFixed"); meta.removeMember("keytime"); meta.removeMember("keybpos"); meta.removeMember("moreheader"); std::map<std::string,int> trackIDs; std::map<std::string,HeaderEntryDTSC> trackData; long long int nowpack = 0; std::string currentID; int nextFreeID = 0; for (JSON::ObjIter it = meta["tracks"].ObjBegin(); it != meta["tracks"].ObjEnd(); it++){ trackIDs.insert(std::pair<std::string,int>(it->first,it->second["trackid"].asInt())); trackData[it->first].type = it->second["type"].asString(); trackData[it->first].trackID = it->second["trackid"].asInt(); trackData[it->first].type = it->second["type"].asString(); if (it->second["trackid"].asInt() >= nextFreeID){ nextFreeID = it->second["trackid"].asInt() + 1; } it->second.removeMember("keylen"); it->second.removeMember("keybpos"); it->second.removeMember("frags"); it->second.removeMember("keytime"); it->second.removeMember("keynum"); it->second.removeMember("keydata"); it->second.removeMember("keyparts"); it->second.removeMember("keys"); } F.parseNext(); while ( !F.getJSON().isNull()){ currentID = ""; if (F.getJSON()["trackid"].asInt() == 0){ if (F.getJSON()["datatype"].asString() == "video"){ currentID = "video0"; if (trackData[currentID].trackID == 0){ trackData[currentID].trackID = nextFreeID++; } if (meta.isMember("video")){ meta["tracks"][currentID] = meta["video"]; meta.removeMember("video"); } trackData[currentID].type = F.getJSON()["datatype"].asString(); }else{ if (F.getJSON()["datatype"].asString() == "audio"){ currentID = "audio0"; if (trackData[currentID].trackID == 0){ trackData[currentID].trackID = nextFreeID++; } if (meta.isMember("audio")){ meta["tracks"][currentID] = meta["audio"]; meta.removeMember("audio"); } trackData[currentID].type = F.getJSON()["datatype"].asString(); }else{ //fprintf(stderr, "Found an unknown package with packetid 0 and datatype %s\n",F.getJSON()["datatype"].asString().c_str()); F.parseNext(); continue; } } }else{ for( std::map<std::string,int>::iterator it = trackIDs.begin(); it != trackIDs.end(); it++ ) { if( it->second == F.getJSON()["trackid"].asInt() ) { currentID = it->first; break; } } if( currentID == "" ) { //fprintf(stderr, "Found an unknown v2 packet with id %d\n", F.getJSON()["trackid"].asInt()); F.parseNext(); continue; //should create new track but this shouldnt be needed... } } if (F.getJSON()["time"].asInt() < trackData[currentID].firstms){ trackData[currentID].firstms = F.getJSON()["time"].asInt(); } if (F.getJSON()["time"].asInt() >= nowpack){ nowpack = F.getJSON()["time"].asInt(); } if (trackData[currentID].type == "video"){ if (F.getJSON().isMember("keyframe")){ int newNum = meta["tracks"][currentID]["keys"].size(); meta["tracks"][currentID]["keys"][newNum]["num"] = ++trackData[currentID].keynum; meta["tracks"][currentID]["keys"][newNum]["time"] = F.getJSON()["time"]; meta["tracks"][currentID]["keys"][newNum]["bpos"] = F.getLastReadPos(); if (meta["tracks"][currentID]["keys"].size() > 1){ meta["tracks"][currentID]["keys"][newNum - 1]["len"] = F.getJSON()["time"].asInt() - meta["tracks"][currentID]["keys"][newNum - 1]["time"].asInt(); meta["tracks"][currentID]["keys"][newNum - 1]["size"] = trackData[currentID].totalSize; trackData[currentID].totalSize = 0; std::string encodeVec = JSON::encodeVector( trackData[currentID].parts.begin(), trackData[currentID].parts.end() ); meta["tracks"][currentID]["keys"][newNum - 1]["parts"] = encodeVec; meta["tracks"][currentID]["keys"][newNum - 1]["partsize"] = (long long int)trackData[currentID].parts.size(); trackData[currentID].parts.clear(); } } }else{ if ((F.getJSON()["time"].asInt() - trackData[currentID].lastKeyTime) > 1000){ trackData[currentID].lastKeyTime = F.getJSON()["time"].asInt(); int newNum = meta["tracks"][currentID]["keys"].size(); meta["tracks"][currentID]["keys"][newNum]["num"] = ++trackData[currentID].keynum; meta["tracks"][currentID]["keys"][newNum]["time"] = F.getJSON()["time"]; meta["tracks"][currentID]["keys"][newNum]["bpos"] = F.getLastReadPos(); if (meta["tracks"][currentID]["keys"].size() > 1){ meta["tracks"][currentID]["keys"][newNum - 1]["len"] = F.getJSON()["time"].asInt() - meta["tracks"][currentID]["keys"][newNum - 1]["time"].asInt(); meta["tracks"][currentID]["keys"][newNum - 1]["size"] = trackData[currentID].totalSize; trackData[currentID].totalSize = 0; std::string encodeVec = JSON::encodeVector( trackData[currentID].parts.begin(), trackData[currentID].parts.end() ); meta["tracks"][currentID]["keys"][newNum - 1]["parts"] = encodeVec; meta["tracks"][currentID]["keys"][newNum - 1]["partsize"] = (long long int)trackData[currentID].parts.size(); trackData[currentID].parts.clear(); } } } trackData[currentID].totalSize += F.getJSON()["data"].asString().size(); trackData[currentID].lastms = nowpack; trackData[currentID].parts.push_back(F.getJSON()["data"].asString().size()); F.parseNext(); } long long int firstms = 0x7fffffff; long long int lastms = -1; for (std::map<std::string,HeaderEntryDTSC>::iterator it = trackData.begin(); it != trackData.end(); it++){ if (it->second.firstms < firstms){ firstms = it->second.firstms; } if (it->second.lastms > lastms){ lastms = it->second.lastms; } meta["tracks"][it->first]["firstms"] = it->second.firstms; meta["tracks"][it->first]["lastms"] = it->second.lastms; meta["tracks"][it->first]["length"] = (it->second.lastms - it->second.firstms) / 1000; if ( !meta["tracks"][it->first].isMember("bps")){ meta["tracks"][it->first]["bps"] = (long long int)(it->second.lastms / ((it->second.lastms - it->second.firstms) / 1000)); } if (it->second.trackID != 0){ meta["tracks"][it->first]["trackid"] = trackIDs[it->first]; }else{ meta["tracks"][it->first]["trackid"] = nextFreeID ++; } meta["tracks"][it->first]["type"] = it->second.type; int tmp = meta["tracks"][it->first]["keys"].size(); if (tmp > 0){ if (tmp > 1){ meta["tracks"][it->first]["keys"][tmp - 1]["len"] = it->second.lastms - meta["tracks"][it->first]["keys"][tmp - 2]["time"].asInt(); }else{ meta["tracks"][it->first]["keys"][tmp - 1]["len"] = it->second.lastms; } meta["tracks"][it->first]["keys"][tmp - 1]["size"] = it->second.totalSize; std::string encodeVec = JSON::encodeVector( trackData[it->first].parts.begin(), trackData[it->first].parts.end() ); meta["tracks"][it->first]["keys"][tmp - 1]["parts"] = encodeVec; meta["tracks"][it->first]["keys"][tmp - 1]["partsize"] = (long long int)trackData[it->first].parts.size(); }else{ meta["tracks"][it->first]["keys"][tmp]["len"] = it->second.lastms; meta["tracks"][it->first]["keys"][tmp]["size"] = it->second.totalSize; std::string encodeVec = JSON::encodeVector( trackData[it->first].parts.begin(), trackData[it->first].parts.end() ); meta["tracks"][it->first]["keys"][tmp]["parts"] = encodeVec; meta["tracks"][it->first]["keys"][tmp]["partsize"] = (long long int)trackData[it->first].parts.size(); meta["tracks"][it->first]["keys"][tmp]["time"] = it->second.firstms; } //calculate fragments meta["tracks"][it->first]["frags"].null(); long long int currFrag = -1; long long int maxBps = 0; for (JSON::ArrIter arrIt = meta["tracks"][it->first]["keys"].ArrBegin(); arrIt != meta["tracks"][it->first]["keys"].ArrEnd(); arrIt++) { if ((*arrIt)["time"].asInt() / 10000 > currFrag){ currFrag = (*arrIt)["time"].asInt() / 10000; long long int fragLen = 1; long long int fragDur = (*arrIt)["len"].asInt(); long long int fragSize = (*arrIt)["size"].asInt(); for (JSON::ArrIter it2 = arrIt + 1; it2 != meta["tracks"][it->first]["keys"].ArrEnd(); it2++){ if ((*it2)["time"].asInt() / 10000 > currFrag || (it2 + 1) == meta["tracks"][it->first]["keys"].ArrEnd()){ JSON::Value thisFrag; thisFrag["num"] = (*arrIt)["num"].asInt(); thisFrag["time"] = (*arrIt)["time"].asInt(); thisFrag["len"] = fragLen; thisFrag["dur"] = fragDur; thisFrag["size"] = fragSize; if (fragDur / 1000){ thisFrag["bps"] = fragSize / (fragDur / 1000); if (maxBps < (fragSize / (fragDur / 1000))){ maxBps = (fragSize / (fragDur / 999)); } } else { thisFrag["bps"] = 1; } meta["tracks"][it->first]["frags"].append(thisFrag); break; } fragLen ++; fragDur += (*it2)["len"].asInt(); fragSize += (*it2)["size"].asInt(); } } } meta["tracks"][it->first]["maxbps"] = maxBps; } meta["firstms"] = firstms; meta["lastms"] = lastms; meta["length"] = (lastms - firstms) / 1000; //append the revised header std::string loader = meta.toPacked(); long long int newHPos = F.addHeader(loader); if ( !newHPos){ std::cerr << "Failure appending new header." << std::endl; return -1; } //re-write the original header with information about the location of the new one oriheader["moreheader"] = newHPos; loader = oriheader.toPacked(); if (F.writeHeader(loader)){ return 0; }else{ std::cerr << "Failure rewriting header." << std::endl; return -1; } } //DTSCFix
///\brief Debugging tool for DTSC data. /// /// Expects DTSC data in a file given on the command line, outputs human-readable information to stderr. ///\param conf The configuration parsed from the commandline. ///\return The return code of the analyser. int analyseDTSC(Util::Config conf){ DTSC::File F(conf.getString("filename")); JSON::Value meta = F.getMeta(); std::cout << meta.toPrettyString() << std::endl; JSON::Value pack; long long unsigned int firstpack = 0; long long unsigned int nowpack = 0; long long unsigned int lastaudio = 0; long long unsigned int lastvideo = 0; long long unsigned int lastkey = 0; long long unsigned int totalvideo = 0; long long unsigned int totalaudio = 0; long long unsigned int keyframes = 0; long long unsigned int key_min = 0xffffffff; long long unsigned int key_max = 0; long long unsigned int vid_min = 0xffffffff; long long unsigned int vid_max = 0; long long unsigned int aud_min = 0xffffffff; long long unsigned int aud_max = 0; long long unsigned int bfrm_min = 0xffffffff; long long unsigned int bfrm_max = 0; long long unsigned int bps = 0; F.seekNext(); while ( !F.getJSON().isNull()){ std::cout << F.getJSON().toPrettyString() << std::endl; nowpack = F.getJSON()["time"].asInt(); if (firstpack == 0){ firstpack = nowpack; } if (F.getJSON()["datatype"].asString() == "audio"){ if (lastaudio != 0 && (nowpack - lastaudio) != 0){ bps = F.getJSON()["data"].asString().size() / (nowpack - lastaudio); if (bps < aud_min){ aud_min = bps; } if (bps > aud_max){ aud_max = bps; } } totalaudio += F.getJSON()["data"].asString().size(); lastaudio = nowpack; } if (F.getJSON()["datatype"].asString() == "video"){ if (lastvideo != 0 && (nowpack - lastvideo) != 0){ bps = F.getJSON()["data"].asString().size() / (nowpack - lastvideo); if (bps < vid_min){ vid_min = bps; } if (bps > vid_max){ vid_max = bps; } } if (F.getJSON()["keyframe"].asInt() != 0){ if (lastkey != 0){ bps = nowpack - lastkey; if (bps < key_min){ key_min = bps; } if (bps > key_max){ key_max = bps; } } keyframes++; lastkey = nowpack; } if (F.getJSON()["offset"].asInt() != 0){ bps = F.getJSON()["offset"].asInt(); if (bps < bfrm_min){ bfrm_min = bps; } if (bps > bfrm_max){ bfrm_max = bps; } } totalvideo += F.getJSON()["data"].asString().size(); lastvideo = nowpack; } F.seekNext(); } std::cout << std::endl << "Summary:" << std::endl; meta["length"] = (long long int)((nowpack - firstpack) / 1000); if (meta.isMember("audio")){ meta["audio"]["bps"] = (long long int)(totalaudio / ((lastaudio - firstpack) / 1000)); std::cout << " Audio: " << meta["audio"]["codec"].asString() << std::endl; std::cout << " Bitrate: " << meta["audio"]["bps"].asInt() << std::endl; } if (meta.isMember("video")){ meta["video"]["bps"] = (long long int)(totalvideo / ((lastvideo - firstpack) / 1000)); meta["video"]["keyms"] = (long long int)((lastvideo - firstpack) / keyframes); if (meta["video"]["keyms"].asInt() - key_min > key_max - meta["video"]["keyms"].asInt()){ meta["video"]["keyvar"] = (long long int)(meta["video"]["keyms"].asInt() - key_min); }else{ meta["video"]["keyvar"] = (long long int)(key_max - meta["video"]["keyms"].asInt()); } std::cout << " Video: " << meta["video"]["codec"].asString() << std::endl; std::cout << " Bitrate: " << meta["video"]["bps"].asInt() << std::endl; std::cout << " Keyframes: " << meta["video"]["keyms"].asInt() << "~" << meta["video"]["keyvar"].asInt() << std::endl; std::cout << " B-frames: " << bfrm_min << " - " << bfrm_max << std::endl; } return 0; }
///\brief Starts a loop, waiting for connections to send data to. ///\param argc The number of arguments to the program. ///\param argv The arguments to the program. ///\return The return code of the buffer. int Start(int argc, char ** argv){ /*std::ofstream myfile ("/home/sharvanath/mistserver/test.txt",std::ios::app); //struct sockaddr add; //socklen_t add_length=sizeof(struct sockaddr); //getsockname(incoming.sock, &add,&add_length); myfile<<"hello1\n"; //myfile << "the family here is : "<<add.sa_family<<"\n"; myfile.close(); */ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("stream_name", JSON::fromString("{\"arg_num\":1, \"arg\":\"string\", \"help\":\"Name of the stream this buffer will be providing.\"}")); conf.addOption("awaiting_ip", JSON::fromString( "{\"arg_num\":2, \"arg\":\"string\", \"default\":\"\", \"help\":\"IP address to expect incoming data from. This will completely disable reading from standard input if used.\"}")); conf.addOption("reportstats", JSON::fromString("{\"default\":0, \"help\":\"Report stats to a controller process.\", \"short\":\"s\", \"long\":\"reportstats\"}")); conf.addOption("time", JSON::fromString( "{\"default\":0, \"arg\": \"integer\", \"help\":\"Buffer a specied amount of time in ms.\", \"short\":\"t\", \"long\":\"time\"}")); conf.parseArgs(argc, argv); std::string name = conf.getString("stream_name"); SS = Util::Stream::makeLive(name); if ( !SS.connected()){ perror("Could not create stream socket"); return 1; } SS.setBlocking(false); conf.activate(); thisStream = Stream::get(); thisStream->setName(name); thisStream->getStream()->setBufferTime(conf.getInteger("time")); Socket::Connection incoming; Socket::Connection std_input(fileno(stdin)); if (conf.getBool("reportstats")){ tthread::thread StatsThread(handleStats, 0); StatsThread.detach(); } std::string await_ip = conf.getString("awaiting_ip"); if (await_ip == ""){ tthread::thread StdinThread(handleStdin, 0); StdinThread.detach(); }else{ thisStream->setWaitingIP(await_ip); tthread::thread StdinThread(handlePushin, 0); StdinThread.detach(); } while (buffer_running && SS.connected() && conf.is_active){ //check for new connections, accept them if there are any //starts a thread for every accepted connection //sharva_mod incoming = SS.accept(true); if (incoming.connected()){ tthread::thread thisUser(handleUser, (void *)new user(incoming)); thisUser.detach(); }else{ Util::sleep(50);//sleep 50ms } } //main loop // disconnect listener buffer_running = false; std::cout << "Buffer shutting down" << std::endl; SS.close(); if (thisStream->getIPInput().connected()){ thisStream->getIPInput().close(); } delete thisStream; return 0; }
int main(int argc, char ** argv){ Util::Config conf = Util::Config(argv[0], PACKAGE_VERSION); conf.addOption("filename", JSON::fromString("{\"arg_num\":1, \"arg\":\"string\", \"help\":\"Filename of the DTSC file to analyse.\"}")); conf.parseArgs(argc, argv); return Converters::DTSC2OGG(conf); }