on_demand_rtsp_server.cpp

/**********
This library is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the
Free Software Foundation; either version 3 of the License, or (at your
option) any later version. (See <http://www.gnu.org/copyleft/lesser.html>.)

This library is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for
more details.

You should have received a copy of the GNU Lesser General Public License
along with this library; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
**********/
// Copyright (c) 1996-2018, Live Networks, Inc.  All rights reserved
// A test program that demonstrates how to stream - via unicast RTP
// - various kinds of file on demand, using a built-in RTSP server.
// main program

#include "liveMedia.hh"
#include "BasicUsageEnvironment.hh"
#include <sys/types.h>  
#include <sys/stat.h>
#include <fcntl.h>
 
#include "capture_and_encoding.h"
#include "version.h"

UsageEnvironment* env;

// To make the second and subsequent client for each stream reuse the same
// input stream as the first client (rather than playing the file from the
// start for each client), change the following "False" to "True":
Boolean reuseFirstSource = False;

// To stream *only* MPEG-1 or 2 video "I" frames
// (e.g., to reduce network bandwidth),
// change the following "False" to "True":
Boolean iFramesOnly = False;

char const* inputFileName = "/tmp/h264_fifo";

static void announceStream(RTSPServer* rtspServer, ServerMediaSession* sms,
			   char const* streamName, char const* inputFileName); // fwd

static char newDemuxWatchVariable;

static MatroskaFileServerDemux* matroskaDemux;
static void onMatroskaDemuxCreation(MatroskaFileServerDemux* newDemux, void* /*clientData*/) {
	matroskaDemux = newDemux;
	newDemuxWatchVariable = 1;
}

static OggFileServerDemux* oggDemux;
static void onOggDemuxCreation(OggFileServerDemux* newDemux, void* /*clientData*/) {
	oggDemux = newDemux;
	newDemuxWatchVariable = 1;
}

int main(int argc, char** argv) {
	// Begin by setting up our usage environment:
	char const* versionFileName = "/tmp/version";
	TaskScheduler* scheduler = BasicTaskScheduler::createNew();
	env = BasicUsageEnvironment::createNew(*scheduler);

	int ver_fd;
	int ret;
	ver_fd = open(versionFileName, O_RDWR | O_CREAT | O_TRUNC, 0777);
	if (ver_fd < 0) {
		  *env << "Failed open /tmp/version\n";
		  exit(1);
	}

	ret = write(ver_fd, VERSION, sizeof(VERSION));
	if (ret != sizeof(VERSION)) {
		*env << "write version failed!\n";
	}

	close(ver_fd);

	*env << "my-carrier-server version: " << VERSION << "\n";

	UserAuthenticationDatabase* authDB = NULL;
#ifdef ACCESS_CONTROL
	// To implement client access control to the RTSP server, do the following:
	authDB = new UserAuthenticationDatabase;
	authDB->addUserRecord("username1", "password1"); // replace these with real strings
	// Repeat the above with each <username>, <password> that you wish to allow
	// access to the server.
#endif
	int fd;

	capture_and_encoding();//基于君正提供的API初始化实现采集和编码模块
	unlink(inputFileName);
		
	if (mkfifo(inputFileName, 0777) < 0) {
			  *env << "mkfifo Failed\n";
			  exit(1);; 
	}

	if (fork() > 0) {
		fd = open(inputFileName, O_RDWR | O_CREAT | O_TRUNC, 0777);
		if (fd < 0) {
			  *env << "Failed open fifo\n";
			  exit(1);; 
		}   
		while (1) {
			  get_stream(fd ,0);//基于君正提供的API实现采集和编码，并将编码后的数据保存到fifo中。
		}
	} else {
		// Create a 'H264 Video RTP' sink from the RTP 'groupsock':
		OutPacketBuffer::maxSize = 600000;
		
        // Create the RTSP server:
		RTSPServer* rtspServer = RTSPServer::createNew(*env, 8554, authDB);
		if (rtspServer == NULL) {
		  *env << "Failed to create RTSP server: " << env->getResultMsg() << "\n";
		  exit(1);
		}

		char const* descriptionString
		  = "Session streamed by \"testOnDemandRTSPServer\"";

		// Set up each of the possible streams that can be served by the
		// RTSP server.  Each such stream is implemented using a
		// "ServerMediaSession" object, plus one or more
		// "ServerMediaSubsession" objects for each audio/video substream.

		// A H.264 video elementary stream:
		char const* streamName = "stream";
		ServerMediaSession* sms
		  = ServerMediaSession::createNew(*env, streamName, streamName,
					      descriptionString);
		sms->addSubsession(H264VideoFileServerMediaSubsession
			       ::createNew(*env, inputFileName, reuseFirstSource));
		rtspServer->addServerMediaSession(sms);

		announceStream(rtspServer, sms, streamName, inputFileName);
	}

	env->taskScheduler().doEventLoop(); // does not return

	return 0; // only to prevent compiler warning
}

static void announceStream(RTSPServer* rtspServer, ServerMediaSession* sms,
			   char const* streamName, char const* inputFileName) {
	char* url = rtspServer->rtspURL(sms);
	UsageEnvironment& env = rtspServer->envir();
	env << "\n\"" << streamName << "\" stream, from the file \""
	    << inputFileName << "\"\n";
	env << "Play this stream using the URL \"" << url << "\"\n";
	delete[] url;
	system("/system/init/tcp_server_daemon.sh &");
}