409 lines
9.3 KiB
C++
409 lines
9.3 KiB
C++
/***************************************************************************************
|
|
*
|
|
* IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
|
*
|
|
* By downloading, copying, installing or using the software you agree to this license.
|
|
* If you do not agree to this license, do not download, install,
|
|
* copy or use the software.
|
|
*
|
|
* Copyright (C) 2014-2024, Happytimesoft Corporation, all rights reserved.
|
|
*
|
|
* Redistribution and use in binary forms, with or without modification, are permitted.
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software distributed
|
|
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
* CONDITIONS OF ANY KIND, either express or implied. See the License for the specific
|
|
* language governing permissions and limitations under the License.
|
|
*
|
|
****************************************************************************************/
|
|
|
|
#include "sys_inc.h"
|
|
#include "rtsp_cln.h"
|
|
#include "hqueue.h"
|
|
#include "http.h"
|
|
#include "http_parse.h"
|
|
|
|
|
|
/**********************************************************/
|
|
HQUEUE * g_queue;
|
|
|
|
int g_flag = 0;
|
|
pthread_t g_tid = 0;
|
|
|
|
typedef struct
|
|
{
|
|
int event;
|
|
CRtspClient * rtsp;
|
|
} EVENT_PARAMS;
|
|
|
|
/**********************************************************/
|
|
|
|
/**
|
|
* @desc : rtsp notify callback
|
|
*
|
|
* @params :
|
|
* event : event type
|
|
* puser : user parameter
|
|
*/
|
|
int rtsp_notify_cb(int event, void * puser)
|
|
{
|
|
printf("%s, event = %d\r\n", __FUNCTION__, event);
|
|
|
|
CRtspClient * p_rtsp = (CRtspClient *) puser;
|
|
|
|
if (RTSP_EVE_CONNSUCC == event)
|
|
{
|
|
int vcodec = p_rtsp->video_codec();
|
|
if (vcodec != VIDEO_CODEC_NONE)
|
|
{
|
|
char codec_str[20] = {'\0'};
|
|
|
|
switch (vcodec)
|
|
{
|
|
case VIDEO_CODEC_H264:
|
|
strcpy(codec_str, "H264");
|
|
break;
|
|
|
|
case VIDEO_CODEC_H265:
|
|
strcpy(codec_str, "H265");
|
|
break;
|
|
|
|
case VIDEO_CODEC_MP4:
|
|
strcpy(codec_str, "MP4");
|
|
break;
|
|
|
|
case VIDEO_CODEC_JPEG:
|
|
strcpy(codec_str, "JPEG");
|
|
break;
|
|
}
|
|
|
|
printf("video codec is %s\r\n", codec_str);
|
|
}
|
|
|
|
int acodec = p_rtsp->audio_codec();
|
|
if (acodec != AUDIO_CODEC_NONE)
|
|
{
|
|
char codec_str[20] = {'\0'};
|
|
|
|
switch (acodec)
|
|
{
|
|
case AUDIO_CODEC_G711A:
|
|
strcpy(codec_str, "G711A");
|
|
break;
|
|
|
|
case AUDIO_CODEC_G711U:
|
|
strcpy(codec_str, "G711U");
|
|
break;
|
|
|
|
case AUDIO_CODEC_G722:
|
|
strcpy(codec_str, "G722");
|
|
break;
|
|
|
|
case AUDIO_CODEC_G726:
|
|
strcpy(codec_str, "G726");
|
|
break;
|
|
|
|
case AUDIO_CODEC_OPUS:
|
|
strcpy(codec_str, "OPUS");
|
|
break;
|
|
|
|
case AUDIO_CODEC_AAC:
|
|
strcpy(codec_str, "AAC");
|
|
break;
|
|
}
|
|
|
|
printf("audio codec is %s\r\n", codec_str);
|
|
printf("audio sample rate is %d\r\n", p_rtsp->get_audio_samplerate());
|
|
printf("audio channels is %d\r\n", p_rtsp->get_audio_channels());
|
|
}
|
|
}
|
|
|
|
EVENT_PARAMS params;
|
|
|
|
params.event = event;
|
|
params.rtsp = p_rtsp;
|
|
|
|
if (!hqBufPut(g_queue, (char *) ¶ms))
|
|
{
|
|
printf("hqBufPut failed\r\n");
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @desc : rtsp audio data callback
|
|
*
|
|
* @params :
|
|
* pdata : audio data buffer
|
|
* len : audio data buffer length
|
|
* ts : timestamp
|
|
* seq : sequential
|
|
* puser : user parameter
|
|
*/
|
|
int rtsp_audio_cb(uint8 * pdata, int len, uint32 ts, uint16 seq, void * puser)
|
|
{
|
|
CRtspClient * p_rtsp = (CRtspClient *) puser;
|
|
|
|
printf("%s, len = %d, ts = %u, seq = %d\r\n", __FUNCTION__, len, ts, seq);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @desc : rtsp video data callback
|
|
*
|
|
* @params :
|
|
* pdata : video data buffer
|
|
* len : video data buffer length
|
|
* ts : timestamp
|
|
* seq : sequential
|
|
* puser : user parameter
|
|
*/
|
|
int rtsp_video_cb(uint8 * pdata, int len, uint32 ts, uint16 seq, void * puser)
|
|
{
|
|
CRtspClient * p_rtsp = (CRtspClient *) puser;
|
|
|
|
printf("%s, len = %d, ts = %u, seq = %d\r\n", __FUNCTION__, len, ts, seq);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @desc : RTCP raw data callback
|
|
*
|
|
* @params :
|
|
* pdata : RTCP raw data
|
|
* len : data length
|
|
* type : AV_VIDEO_CH, AV_AUDIO_CH, AV_METADATA_CH, AV_BACK_CH
|
|
* puser : user parameter
|
|
*/
|
|
int rtsp_rtcp_cb(uint8 * pdata, int len, int type, void * puser)
|
|
{
|
|
CRtspClient * p_rtsp = (CRtspClient *) puser;
|
|
|
|
printf("%s, len = %d, type = %d\r\n", __FUNCTION__, len, type);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @desc : rtsp meta data callback
|
|
*
|
|
* @params :
|
|
* pdata : meta data buffer
|
|
* len : meta data buffer length
|
|
* ts : timestamp
|
|
* seq : sequential
|
|
* puser : user parameter
|
|
*/
|
|
int rtsp_metadata_cb(uint8 * pdata, int len, uint32 ts, uint16 seq, void * puser)
|
|
{
|
|
CRtspClient * p_rtsp = (CRtspClient *) puser;
|
|
|
|
printf("%s, len = %d, ts = %u, seq = %d\r\n", __FUNCTION__, len, ts, seq);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void rtsp_setup(CRtspClient * p_rtsp)
|
|
{
|
|
p_rtsp->set_notify_cb(rtsp_notify_cb, p_rtsp);
|
|
p_rtsp->set_audio_cb(rtsp_audio_cb);
|
|
p_rtsp->set_video_cb(rtsp_video_cb);
|
|
p_rtsp->set_rtcp_cb(rtsp_rtcp_cb);
|
|
#ifdef METADATA
|
|
p_rtsp->set_metadata_cb(rtsp_metadata_cb);
|
|
#endif
|
|
// p_rtsp->set_rtp_over_udp(1); // rtp over udp
|
|
// p_rtsp->set_rtp_multicast(1); // force multicast rtp via rtsp
|
|
#ifdef BACKCHANNEL
|
|
// p_rtsp->set_bc_flag(1); // enable audio backchannel
|
|
// p_rtsp->set_bc_data_flag(1); // enable send audio data
|
|
#endif
|
|
#ifdef REPLAY
|
|
// p_rtsp->set_replay_flag(1); // enable replay
|
|
// p_rtsp->set_replay_range(time(NULL) - 3600, time(NULL)); // set the replay timestamp range
|
|
#endif
|
|
#ifdef OVER_HTTP
|
|
// p_rtsp->set_rtsp_over_http(1, 80); // rtsp over http
|
|
#endif
|
|
#ifdef OVER_WEBSOCKET
|
|
// p_rtsp->set_rtsp_over_ws(1, 80); // rtsp over websocket
|
|
#endif
|
|
|
|
p_rtsp->set_rx_timeout(10); // No data within 10s, receiving timeout
|
|
// p_rtsp->set_channel(AV_VIDEO_CH, 0); // not setup video channel
|
|
// p_rtsp->set_channel(AV_AUDIO_CH, 0); // not setup audio channel
|
|
// p_rtsp->set_channel(AV_METADATA_CH, 0); // not setup metadata channel
|
|
// p_rtsp->set_channel(AV_BACK_CH, 0); // not setup audio back channel
|
|
}
|
|
|
|
void rtsp_reconn(CRtspClient * p_rtsp)
|
|
{
|
|
char url[256], user[64], pass[64];
|
|
|
|
strcpy(url, p_rtsp->get_url());
|
|
strcpy(user, p_rtsp->get_user());
|
|
strcpy(pass, p_rtsp->get_pass());
|
|
|
|
printf("rtsp_reconn, url = %s, user = %s, pass = %s\r\n", url, user, pass);
|
|
|
|
p_rtsp->rtsp_close();
|
|
|
|
rtsp_setup(p_rtsp);
|
|
|
|
p_rtsp->rtsp_start(url, user, pass);
|
|
}
|
|
|
|
void * rtsp_notify_handler(void * argv)
|
|
{
|
|
EVENT_PARAMS params;
|
|
|
|
while (g_flag)
|
|
{
|
|
if (hqBufGet(g_queue, (char *) ¶ms))
|
|
{
|
|
if (params.event == -1 || params.rtsp == NULL)
|
|
{
|
|
break;
|
|
}
|
|
|
|
if (RTSP_EVE_STOPPED == params.event ||
|
|
RTSP_EVE_CONNFAIL == params.event ||
|
|
RTSP_EVE_NOSIGNAL == params.event ||
|
|
RTSP_EVE_NODATA == params.event)
|
|
{
|
|
rtsp_reconn(params.rtsp);
|
|
|
|
usleep(100*1000);
|
|
}
|
|
}
|
|
}
|
|
|
|
g_tid = 0;
|
|
|
|
printf("%s exit\r\n", __FUNCTION__);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
#define RTSP_CLN_NUM 1
|
|
|
|
int main(int argc, char * argv[])
|
|
{
|
|
/*if (argc < 2)
|
|
{
|
|
printf("usage: %s url {user} {pass}\r\n", argv[0]);
|
|
return -1;
|
|
}*/
|
|
|
|
log_init("rtsptest.log");
|
|
log_set_level(HT_LOG_DBG);
|
|
|
|
network_init();
|
|
|
|
// allocate system BUFFER and rtsp parser BUFFER
|
|
sys_buf_init(RTSP_CLN_NUM * 8);
|
|
rtsp_parse_buf_init(RTSP_CLN_NUM * 8);
|
|
|
|
#if defined(OVER_HTTP) || defined(OVER_WEBSOCKET)
|
|
// allocate http message parser BUFFER
|
|
http_msg_buf_init(RTSP_CLN_NUM * 8);
|
|
#endif
|
|
|
|
// create event queue
|
|
g_queue = hqCreate(RTSP_CLN_NUM * 4, sizeof(EVENT_PARAMS), HQ_PUT_WAIT | HQ_GET_WAIT);
|
|
if (NULL == g_queue)
|
|
{
|
|
printf("create queue failed\r\n");
|
|
return -1;
|
|
}
|
|
|
|
// create event handler thread
|
|
g_flag = 1;
|
|
g_tid = sys_os_create_thread((void *)rtsp_notify_handler, NULL);
|
|
if (g_tid == 0)
|
|
{
|
|
printf("create rtsp notify handler thread failed\r\n");
|
|
return -1;
|
|
}
|
|
|
|
CRtspClient * rtsp = new CRtspClient[RTSP_CLN_NUM];
|
|
|
|
for (int i = 0; i < RTSP_CLN_NUM; i++)
|
|
{
|
|
rtsp_setup(&rtsp[i]);
|
|
|
|
char * p_user = NULL;
|
|
char * p_pass = NULL;
|
|
|
|
/*if (argc >= 3)
|
|
{
|
|
p_user = argv[2];
|
|
}
|
|
|
|
if (argc >= 4)
|
|
{
|
|
p_pass = argv[3];
|
|
}*/
|
|
|
|
BOOL ret = rtsp[i].rtsp_start("rtsp://118.70.125.152:5656/cam/realmonitor?channel=1&subtype=0", "root", "abcd1234");
|
|
|
|
printf("rtsp %d start ret = %d\r\n", i, ret);
|
|
|
|
usleep(100 * 1000);
|
|
}
|
|
|
|
for (;;)
|
|
{
|
|
if (getchar() == 'q')
|
|
{
|
|
break;
|
|
}
|
|
|
|
usleep(1000*1000); // 1s
|
|
}
|
|
|
|
for (int i = 0; i < RTSP_CLN_NUM; i++)
|
|
{
|
|
rtsp[i].rtsp_close();
|
|
}
|
|
|
|
delete[] rtsp;
|
|
|
|
g_flag = 0;
|
|
|
|
EVENT_PARAMS params;
|
|
|
|
params.event = -1;
|
|
params.rtsp = NULL;
|
|
|
|
hqBufPut(g_queue, (char *) ¶ms);
|
|
|
|
// waiting for event handler thread to exit
|
|
while (g_tid)
|
|
{
|
|
usleep(10*1000);
|
|
}
|
|
|
|
hqDelete(g_queue);
|
|
g_queue = NULL;
|
|
|
|
// free memory resources
|
|
rtsp_parse_buf_deinit();
|
|
sys_buf_deinit();
|
|
|
|
#if defined(OVER_HTTP) || defined(OVER_WEBSOCKET)
|
|
http_msg_buf_deinit();
|
|
#endif
|
|
|
|
// close log
|
|
log_close();
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|