ZLKMedia/3rdpart/media-server/librtsp/source/server/rtsp-server.c

#include "rtsp-server.h"
#include "rtsp-reason.h"
#include "rfc822-datetime.h"
#include "rtsp-server-internal.h"
#include <stdlib.h>
#include <string.h>
#include <assert.h>

struct rtsp_server_t* rtsp_server_create(const char ip[65], unsigned short port, struct rtsp_handler_t* handler, void* ptr, void* ptr2)
{
	struct rtsp_server_t* rtsp;

	rtsp = (struct rtsp_server_t *)calloc(1, sizeof(struct rtsp_server_t));
	if (NULL == rtsp) return NULL;

	snprintf(rtsp->ip, sizeof(rtsp->ip), "%s", ip);
	rtsp->port = port;
	rtsp->param = ptr;
	rtsp->sendparam = ptr2;
	memcpy(&rtsp->handler, handler, sizeof(rtsp->handler));
	rtsp->parser = http_parser_create(HTTP_PARSER_REQUEST, NULL, NULL);
	return rtsp;
}

int rtsp_server_destroy(struct rtsp_server_t* rtsp)
{
	if (rtsp->handler.close)
		rtsp->handler.close(rtsp->sendparam);

	if (rtsp->parser)
	{
		http_parser_destroy(rtsp->parser);
		rtsp->parser = NULL;
	}

	free(rtsp);
	return 0;
}

int rtsp_server_input(struct rtsp_server_t* rtsp, const void* data, size_t* bytes)
{
	int r;
	r = http_parser_input(rtsp->parser, data, bytes);
	assert(r <= 2); // 1-need more data
	if (0 == r)
	{
		r = rtsp_server_handle(rtsp);
		http_parser_clear(rtsp->parser); // reset parser
	}
	return r;
}

const char* rtsp_server_get_header(struct rtsp_server_t *rtsp, const char* name)
{
	return http_get_header_by_name(rtsp->parser, name);
}

const char* rtsp_server_get_client(rtsp_server_t* rtsp, unsigned short* port)
{
	if (port) *port = rtsp->port;
	return rtsp->ip;
}

int rtsp_server_reply(struct rtsp_server_t *rtsp, int code)
{
	return rtsp_server_reply2(rtsp, code, NULL, NULL, 0);
}

int rtsp_server_reply2(struct rtsp_server_t *rtsp, int code, const char* header, const void* data, int bytes)
{
	int len;
	rfc822_datetime_t datetime;
	rfc822_datetime_format(time(NULL), datetime);

	// smpte=0:10:22-;time=19970123T153600Z
	len = snprintf(rtsp->reply, sizeof(rtsp->reply),
		"RTSP/1.0 %d %s\r\n"
		"CSeq: %u\r\n"
		"Date: %s\r\n"
		"User-Agent: %s\r\n",
		code, rtsp_reason_phrase(code), rtsp->cseq, datetime, USER_AGENT);

	// session header
	if (len > 0 && rtsp->session.session[0])
	{
		len += snprintf(rtsp->reply + len, sizeof(rtsp->reply) - len, "Session: %s;timeout=%d\r\n", rtsp->session.session, rtsp->session.timeout/1000);
	}

	// other headers
	if(len > 0 && header && *header)
	{
		len += snprintf(rtsp->reply + len, sizeof(rtsp->reply) - len, "%s", header);
	}

	// Last: Add Content-Length
	if (len > 0)
	{
		len += snprintf(rtsp->reply + len, sizeof(rtsp->reply) - len, "Content-Length: %d\r\n\r\n", bytes);
	}

	if (len < 0 || bytes < 0 || len + bytes >= sizeof(rtsp->reply))
		return rtsp_server_reply(rtsp, 513 /*Message Too Large*/);
	memcpy(rtsp->reply + len, data, bytes);
	return rtsp->handler.send(rtsp->sendparam, rtsp->reply, len + bytes);
}

int rtsp_server_send_interleaved_data(rtsp_server_t* rtsp, const void* data, size_t bytes)
{
	return rtsp->handler.send(rtsp->sendparam, data, bytes);
}