RTSP调试代码

 

#ifdef _WIN32_WCE

#include "stdafx.h"

#endif





#ifndef _WIN32_WCE

#define WIN32_LEAN_AND_MEAN

#endif





#include <windows.h>

#include <commctrl.h>





#include <MMSystem.h>





#include <winsock2.h>





#include <stdio.h>

#include <stdlib.h>

#include <tchar.h>





#ifdef _WIN32_WCE

#pragma comment(lib, "ws2.lib")

#else

#pragma comment(lib, "ws2_32.lib")

#endif





#define  PP_NAME "User-Agent: rtsp client(v1.0)"

#define  PP_CRLF "\r\n"





//打开连接;

long InitSocket();





//关闭连接;

long DeInitSocket();





//初始化tcp socket;

long InitTCPSocket(int port);





//初始化udp socket;

long InitUDPSocket(const char *ip, int port);





//取socket端口号;

long GetSokcetPort(int sock, int *port);





//读取数据;

long ReadSocket(int sock, char *buf, int len, int timeout);





//发送命令数据;

long SendRTSPCmd(int sock, const char *cmd, const char *szparam);





//解析rtsp命令回应数据;

long PraseRTSPCmd();

long PraseOptionCmd(const char *sz);

long PraseDescribeCmd(const char *sz);

long PraseSetupCmd(const char *sz, char *sess);

long PrasePlayCmd(const char *sz);

long GetResponseCode(const char *sz);  //取返回值;





//////////////////////////////////////////////////////////////////////////

//字符串操作函数;

static char* getLine(char* startOfLine);









//生成rtsp发送命令;

char * GetRTSPCmd(const char *);

char * GetOptionCmd(char *url);

char * GetDescribeCmd(char *url);

char * GetPlayCmd(char *url, char *session, char *range);

char * GetSetupCmd(char *url, int port1, int port2);





char * GetReportCmd(char *);





//////////////////////////////////////////////////////////////////////////

//日志函数;

long logwr(void *, int len);





//全局变量定义区;

fd_set rfdsock;

//日志写入文件指针;

FILE *fp = NULL;

//





//////////////////////////////////////////////////////////////////////////

//rtsp请求解析;

long PraseURL(const char *url, char *szip, int *iport);

int _tmain(int argc, _TCHAR* argv[])

{

	int sockin, sc1, sc2;

	sockaddr_in addr;

	char *buf, *szcmd, *url;

	char szip[32];

	int nlen, iret, iport;

	int ip1, ip2;

	long lret;





	//初始化变量;

	FD_ZERO(&rfdsock);





	fp = fopen("rtsp_log.txt", "w+");

	//分配缓冲区;

	nlen = 10240;

	buf = (char*)malloc(nlen);





	//定义要连接的url;

	//url = "rtsp://192.168.1.43:2554/realmp3.mp3";

	url = "rtsp://192.168.10.177/bipbop-gear1-all.ts";

	//url = "rtsp://192.168.1.43/1.amr";

	//初始化sock;

	InitSocket();





	//分析url请求，取出ip，端口;

	lret = PraseURL(url, szip, &iport);

		

	//初始化与服务器连接的socket;

	sockin = InitTCPSocket(0);





	//与服务器连接;

	addr.sin_family = AF_INET;

	addr.sin_port = htons(iport);

	addr.sin_addr.s_addr = inet_addr(szip);

	iret = connect(sockin,(struct sockaddr*)&addr, sizeof addr);





	if(iret   ==   SOCKET_ERROR)   

	{   

		int   erro   =   WSAGetLastError();   





		printf("connect   fail   !");

		Sleep(3000);     





		closesocket(sockin);   

		WSACleanup();     

		return 0;     

	} 





	//发送option命令;

	szcmd = GetOptionCmd(url);

	lret = SendRTSPCmd(sockin, "OPTIONS",szcmd);

	free(szcmd);

	lret = ReadSocket(sockin, buf, nlen,100);

	

	//发送DESCRIBE命令;

	szcmd = GetDescribeCmd(url);

	lret = SendRTSPCmd(sockin, "DESCRIBE", szcmd);

	free(szcmd);

	lret = ReadSocket(sockin, buf,  nlen, 100);





	//解析Response;

	lret = PraseDescribeCmd((const char*)buf);

	

	//创建客户端接收端口;

	sc1 = InitUDPSocket(NULL, 6544);

	sc2 = InitUDPSocket(NULL, 6545);





	//将sock加入到要等待的队列;

	FD_SET(sc1, &rfdsock );

	FD_SET(sc2, &rfdsock);

	lret = GetSokcetPort(sc1, &ip1);

	lret = GetSokcetPort(sc2, &ip2);





	//发送Setup命令，告诉服务器客户端的接受数据的端口;

	szcmd = GetSetupCmd(url, ip1, ip2);

	//告诉服务器客户端的端口;

	lret = SendRTSPCmd(sockin, "SETUP", szcmd);

	free(szcmd);

	lret = ReadSocket(sockin, buf, nlen, 100);





	//解析Response返回的命令串;

	char szip2[9];

	lret = PraseSetupCmd(buf, szip2);





	char *session, *srange;

	session = szip2;

	//发送PLAY命令

	srange  = "Range: npt=0.000-39.471\r\n";

	szcmd = GetPlayCmd(url, session, srange);

	lret = SendRTSPCmd(sockin, "PLAY", szcmd);

	free(szcmd);

	lret = ReadSocket(sockin, buf, nlen, 100);





	timeval tv;

	fd_set fr;





	int i;





	tv.tv_sec = 20;

	tv.tv_usec = 0;

	struct sockaddr_in addr2;

	int addrlen;

	addrlen = sizeof addr;





	//将数据写到文件中去;

	FILE *ffp;

	ffp = fopen("bipbop-gear1-all.ts", "w+");





	//开始接受数据了;

	while(true)

	{

		fr = rfdsock;

		lret  = select(0, &fr, NULL, NULL, &tv);





		if(lret == SOCKET_ERROR)

		{

			break;

		}

		else if(lret >0)

		{

			//判断是哪个socket可以读取数据了

			for(i = 0; i< 2;i ++)

			{

				if(FD_ISSET(rfdsock.fd_array[i], &fr)

					&& FD_ISSET(rfdsock.fd_array[i], &rfdsock))

				{

					lret = recvfrom(rfdsock.fd_array[i], buf, nlen,0, (struct sockaddr*)&addr2, &addrlen );

					if(lret > 0 && ffp)

					{

						fwrite(buf, 1, lret, ffp);

					}

					else if(lret == SOCKET_ERROR)

					{

						break;

					}

				}

			}





		}

		else if(lret == 0)

			break;

	}

	fclose(ffp);





	//退出后的清理工作

	closesocket(sockin);

	closesocket(sc1);

	closesocket(sc2);





	fwrite("\r\nend", 1, 5, fp);

	fclose(fp);

	return 0;

}





long InitSocket()

{

	WSADATA ws;

	long lret  = -1;

	lret = WSAStartup(MAKEWORD(2,2), &ws);

	return 0;

}





long DeInitSocket()

{





	WSACleanup();

	return 0;

}





long InitTCPSocket( int port )

{

	long lret ;

	int sock;

	sockaddr_in addr;

	sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);





	//int flag = 1;

	//lret = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (const char *)&flag, sizeof flag);





	//addr.sin_family = AF_INET;

	//addr.sin_port = 0;

	//addr.sin_addr.s_addr = INADDR_ANY;

	//lret = bind(sock, (struct sockaddr*)&addr, sizeof addr);

	lret = sock;

	return lret;

}

#define   SIO_RCVALL                         _WSAIOW(IOC_VENDOR,1)   

#define   SIO_RCVALL_MCAST             _WSAIOW(IOC_VENDOR,2)   

#define   SIO_RCVALL_IGMPMCAST     _WSAIOW(IOC_VENDOR,3)   

#define   SIO_KEEPALIVE_VALS         _WSAIOW(IOC_VENDOR,4)   

#define   SIO_ABSORB_RTRALERT       _WSAIOW(IOC_VENDOR,5)   

#define   SIO_UCAST_IF                     _WSAIOW(IOC_VENDOR,6)   

#define   SIO_LIMIT_BROADCASTS     _WSAIOW(IOC_VENDOR,7)   

#define   SIO_INDEX_BIND                 _WSAIOW(IOC_VENDOR,8)   

#define   SIO_INDEX_MCASTIF           _WSAIOW(IOC_VENDOR,9)   

#define   SIO_INDEX_ADD_MCAST       _WSAIOW(IOC_VENDOR,10)   

#define   SIO_INDEX_DEL_MCAST       _WSAIOW(IOC_VENDOR,11)   

long InitUDPSocket(const char *ip,  int port )

{

	long lret;

	int sock;

	sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);

	int flag = 1;

	lret = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (const char *)&flag, sizeof flag);





	sockaddr_in addr;





	addr.sin_family = AF_INET;

	addr.sin_port = port;

	if(ip)

		addr.sin_addr.s_addr = inet_addr(ip);

	else

		addr.sin_addr.s_addr = INADDR_ANY;

	lret = bind(sock, (struct sockaddr*)&addr, sizeof addr);

	//设置非堵塞通讯

	//u_long arg = 2; 

	//lret = ioctlsocket(sock, SIO_LIMIT_BROADCASTS, &arg);





	lret = sock;

	return lret;

}





long GetSokcetPort( int sock, int *port )

{

	long lret = -1;

	sockaddr_in addr;

	int nlen;

	nlen = sizeof addr;





	addr.sin_port = 0;





	*port = 0;





	if(getsockname(sock, (struct sockaddr*)&addr, &nlen) < 0)

		lret = -1;

	else

	{

		lret = 0;

		*port = addr.sin_port;

	}





	return lret;

}













long ReadSocket(int sock,  char *buf, int len, int timeout )

{

	long lret ;

	int iret;

	fd_set fr;

	timeval tm;

	tm.tv_sec = timeout;

	tm.tv_usec = 0;





	FD_ZERO(&fr);

	fr.fd_count = 1;

	fr.fd_array[0] = sock;





	lret = select(sock, &fr, NULL, NULL, &tm);





	if(lret > 0)

	{

		lret = recv(sock, buf, len, 0);

		if(lret == SOCKET_ERROR)

		{





		}

		else if(lret > 0)

		{

			logwr((void*)"***Recive:\r\n", 12);

			logwr((void*)buf, lret);

		}

	}





	return lret;

}





long SendRTSPCmd( int sock, const char *cmd, const char *szparam )

{

	long lret;

	int ilen;

	ilen = strlen(szparam);

	lret = send(sock, szparam, ilen,0);

	if(lret == SOCKET_ERROR)

	{

		lret = WSAGetLastError();

	}

	logwr((void*)"***Send:\r\n", 10);

	logwr((void *)szparam, ilen);

	return lret;

}





char * GetRTSPCmd( const char * szName)

{

	char *str = NULL;

	char const*  cmdFmt = NULL;

	if(!strcmp(szName, "OPTIONS"))

	{

		cmdFmt =

			"OPTIONS %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"%s"

			"%s"

#ifdef SUPPORT_REAL_RTSP

			REAL_OPTIONS_HEADERS

#endif

			"\r\n";

	}

	else if(!strcmp(szName, "ANNOUNCE"))

	{

		cmdFmt =

			"ANNOUNCE %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Content-Type: application/sdp\r\n"

			"%s"

			"Content-length: %d\r\n\r\n"

			"%s";

	}

	else if(!strcmp(szName, "PLAY"))

	{

		cmdFmt ="PLAY %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Session: %s\r\n"

			"%s"

			"%s"

			"%s"

			"%s"

			"\r\n";





	}

	else if(!strcmp(szName, "PAUSE"))

	{

		cmdFmt =

			"PAUSE %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Session: %s\r\n"

			"%s"

			"%s"

			"\r\n";

	}

	else if(!strcmp(szName, "RECORD"))

	{

		cmdFmt =

			"RECORD %s%s%s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Session: %s\r\n"

			"Range: npt=0-\r\n"

			"%s"

			"%s"

			"\r\n";

	}

	else if(!strcmp(szName, "SET_PARAMETER"))

	{

		cmdFmt =

			"SET_PARAMETER %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Session: %s\r\n"

			"%s"

			"%s"

			"Content-length: %d\r\n\r\n"

			"%s: %s\r\n";

	}

	else if(!strcmp(szName, "GET_PARAMETER"))

	{

		cmdFmt =

			"GET_PARAMETER %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Session: %s\r\n"

			"%s"

			"%s"

			"Content-type: text/parameters\r\n"

			"Content-length: %d\r\n\r\n"

			"%s\r\n";

	}

	else if(!strcmp(szName, "TEARDOWN"))

	{

		cmdFmt =

			"TEARDOWN %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Session: %s\r\n"

			"%s"

			"%s"

			"\r\n";

	}

	else if(!strcmp(szName, "DESCRIBE"))

	{

		cmdFmt =

			"DESCRIBE %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"%s"

			"%s"

			"%s"

#ifdef SUPPORT_REAL_RTSP

			REAL_DESCRIBE_HEADERS

#endif

			"\r\n";

	}

	else if(!strcmp(szName, "ANNOUNCE"))

	{

		cmdFmt =

			"ANNOUNCE %s RTSP/1.0\r\n"

			"CSeq: %d\r\n"

			"Content-Type: application/sdp\r\n"

			"%s"

			"Content-length: %d\r\n\r\n"

			"%s";

	}

	else if(!strcmp(szName, "SETUP"))

	{

		cmdFmt ="SETUP %s"

			"CSeq: %d\r\n"

			"%s"

			"%s"

			"%s"

			"%s"

			"\r\n";

	}

	str = (char*)cmdFmt;

	return str;

}





char * GetOptionCmd( char *url )

{

	int nlen, iret;

	char *ss;

	char *s = GetRTSPCmd("OPTIONS");





	nlen = strlen(s);

	iret = nlen + strlen(url) + strlen(PP_NAME) + 200;

	ss = (char*)malloc(iret);





	sprintf(ss, s, url, 1, PP_NAME, PP_CRLF);









	return ss;

}





char * GetDescribeCmd( char *url )

{

	int nlen, iret;

	char *ss;

	char *s = GetRTSPCmd("DESCRIBE");





	nlen = strlen(s);

	iret = nlen + strlen(url) + strlen(PP_NAME) + 200;

	ss = (char*)malloc(iret);





	sprintf(ss, s, url, 2, PP_NAME, PP_CRLF, PP_CRLF);





	return ss;

}

char * GetPlayCmd( char *url , char *session, char *range)

{

	int nlen, iret;

	char *ss;

	char *s = GetRTSPCmd("PLAY");





	nlen = strlen(s);

	iret = nlen + strlen(url) + strlen(PP_NAME) + 200;

	ss = (char*)malloc(iret);





	//char buf[128] = {0};

	//sprintf(buf, "Session: %s\r\n", session);





	sprintf(ss, s, url, 4, session, range, PP_NAME, PP_CRLF, PP_CRLF);





	return ss;

}





char * GetSetupCmd( char *url , int port1, int port2)

{

	int nlen, iret;

	char *ss;

	char *s = GetRTSPCmd("SETUP");





	nlen = strlen(s);

	iret = nlen + strlen(url) + strlen(PP_NAME) + 200;

	ss = (char*)malloc(iret);

	char buf[128] = {0};

	char buf2[128] = {0};

	if(port1 == 0)

		strcpy(buf, "\r\nTransport: RTP/AVP/TCP;unicast;interleaved=0-1");

	else

		sprintf(buf, "\r\nTransport: RTP/AVP;unicast;client_port=%d-%d", ntohs(port2), ntohs(port1));

	//sprintf(buf2, "SETUP %s/streamid=0 RTSP/1.0\r\n", url);

	//sprintf(ss, s, buf2, 3,  buf, PP_NAME, PP_CRLF, PP_CRLF);

	sprintf(buf2, "%s/track1 RTSP/1.0\r\n", url);

	sprintf(ss, s, buf2, 3,  PP_NAME, buf, PP_CRLF, PP_CRLF);





	return ss;

}





long PraseURL( const char *url, char *szip, int *iport )

{

	long lret = -1;

	if(url)

	{ 

		//找到了rtsp这个标识符

		if(!_strnicmp(url, "rtsp://", 7))

		{

			//找ip

			char *s, *ss;

			s = (char*)url + strlen("rtsp://");

			ss = strchr(s, '/');





			strncpy(szip, s,  ss- s);

			szip[ss -s] = '\0';





			//查找下是否是有端口设置

			s = strchr(szip, ':');

			//有端口设置

			if(s)

			{

				ss = s;

				s ++;

				*iport = atoi(s);





				//同时修正ip地址

				szip[ss - szip] = '/0';

			}

			else

				*iport = 554;





			lret = 0;

		}

	}









	return lret;

}





long logwr( void *data, int len )

{

	long lret = -1;

	if(fp)

		lret = fwrite(data, 1, len, fp);

	

	return lret;

}









static char* getLine(char* startOfLine) {

	// returns the start of the next line, or NULL if none

	for (char* ptr = startOfLine; *ptr != '\0'; ++ptr) {

		// Check for the end of line: \r\n (but also accept \r or \n by itself):

		if (*ptr == '\r' || *ptr == '\n') {

			// We found the end of the line

			if (*ptr == '\r') {

				*ptr++ = '\0';

				if (*ptr == '\n') ++ptr;

			} else {

				*ptr++ = '\0';

			}

			return ptr;

		}

	}





	return NULL;

}





long GetResponseCode( const char *sz )

{

	long lret = -1;

	if(sz)

	{

		if(sscanf(sz, "%*s%u", &lret) != 1)

			;

	}





	return lret;

}





long PraseDescribeCmd( const char *sz )

{

	long lret = -1;

	char *ss, *szst;

	szst = (char *)sz;

	int contentLength = -1;





	if(GetResponseCode(sz )== 200)

	{

		ss= getLine(szst);

		while(1)

		{

			ss = getLine(ss);

			if(ss == NULL)

				break;

			if (sscanf(ss, "Content-Length: %d", &contentLength) == 1

				|| sscanf(ss, "Content-length: %d", &contentLength) == 1) {

					if (contentLength < 0) {

						//....

					}

			}

		}

	}





	return lret;

}





long PraseSetupCmd( const char *sz , char *sess)

{

	long lret = -1;

	char *ss, *szst;

	szst = (char *)sz;

	int contentLength = -1;





	if(GetResponseCode(sz )== 200)

	{

		ss= getLine(szst);

		while(1)

		{

			ss = getLine(ss);

			if(ss == NULL)

				break;

			if (sscanf(ss, "Session: %sP4 192.168.10.177", sess) == 1)

			{

				lret = 1;

				return lret;

			}

		}





	}

	return lret;

}