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;
}