/*************************************************************************************** * * 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 "http_flv_cln.h" #include "hqueue.h" #include "media_format.h" #include "http_parse.h" /**********************************************************/ HQUEUE * g_queue; int g_flag = 0; pthread_t g_tid = 0; typedef struct { int event; CHttpFlvClient* httpflv; } EVENT_PARAMS; /**********************************************************/ /** * @desc : http-flv notify callback * * @params : * event : event type * puser : user parameter */ int http_flv_notify_callback(int event, void * puser) { printf("%s, event = %d\r\n", __FUNCTION__, event); CHttpFlvClient * p_httpflv = (CHttpFlvClient *) puser; if (HTTP_FLV_EVE_VIDEOREADY == event) { int vcodec = p_httpflv->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); } } else if (HTTP_FLV_EVE_AUDIOREADY == event) { int acodec = p_httpflv->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_httpflv->get_audio_samplerate()); printf("audio channels is %d\r\n", p_httpflv->get_audio_channels()); } } EVENT_PARAMS params; params.event = event; params.httpflv = p_httpflv; if (!hqBufPut(g_queue, (char *) ¶ms)) { printf("hqBufPut failed\r\n"); } return 0; } /** * @desc : http-flv audio data callback * * @params : * pdata : audio data buffer * len : audio data buffer length * ts : timestamp * puser : user parameter */ int http_flv_audio_callback(uint8 * pdata, int len, uint32 ts, void * puser) { CHttpFlvClient * p_httpflv = (CHttpFlvClient *) puser; printf("%s, len = %d, ts = %u\r\n", __FUNCTION__, len, ts); return 0; } /** * @desc : http-flv video data callback * * @params : * pdata : video data buffer * len : video data buffer length * ts : timestamp * puser : user parameter */ int http_flv_video_callback(uint8 * pdata, int len, uint32 ts, void * puser) { CHttpFlvClient * p_httpflv = (CHttpFlvClient *) puser; printf("%s, len = %d, ts = %u\r\n", __FUNCTION__, len, ts); return 0; } void http_flv_setup(CHttpFlvClient * p_httpflv) { p_httpflv->set_notify_cb(http_flv_notify_callback, p_httpflv); p_httpflv->set_audio_cb(http_flv_audio_callback); p_httpflv->set_video_cb(http_flv_video_callback); } void http_flv_reconn(CHttpFlvClient * p_httpflv) { char url[512], user[64], pass[64]; strcpy(url, p_httpflv->get_url()); strcpy(user, p_httpflv->get_user()); strcpy(pass, p_httpflv->get_pass()); printf("http_flv_reconn, url = %s, user = %s, pass = %s\r\n", url, user, pass); p_httpflv->http_flv_close(); http_flv_setup(p_httpflv); p_httpflv->http_flv_start(url, user, pass); } void * http_flv_notify_handler(void * argv) { EVENT_PARAMS params; while (g_flag) { if (hqBufGet(g_queue, (char *) ¶ms)) { if (params.event == -1 || params.httpflv == NULL) { break; } if (HTTP_FLV_EVE_STOPPED == params.event || HTTP_FLV_EVE_CONNFAIL == params.event || HTTP_FLV_EVE_NOSIGNAL == params.event || HTTP_FLV_EVE_NODATA == params.event) { http_flv_reconn(params.httpflv); usleep(100*1000); } } } g_tid = 0; printf("%s exit\r\n", __FUNCTION__); return NULL; } #define HTTP_FLV_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("httpflvtest.log"); log_set_level(HT_LOG_DBG); network_init(); // allocate system BUFFER and http message BUFFER sys_buf_init(HTTP_FLV_CLN_NUM * 4); http_msg_buf_init(HTTP_FLV_CLN_NUM * 4); // create event queue g_queue = hqCreate(HTTP_FLV_CLN_NUM * 4, sizeof(EVENT_PARAMS), HQ_GET_WAIT | HQ_PUT_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 *)http_flv_notify_handler, NULL); if (g_tid == 0) { printf("create http flv notify handler thread failed\r\n"); return -1; } CHttpFlvClient * httpflv = new CHttpFlvClient[HTTP_FLV_CLN_NUM]; for (int i = 0; i < HTTP_FLV_CLN_NUM; i++) { http_flv_setup(&httpflv[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 = httpflv[i].http_flv_start(argv[1], p_user, p_pass); printf("http flv %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 < HTTP_FLV_CLN_NUM; i++) { httpflv[i].http_flv_close(); } delete[] httpflv; g_flag = 0; EVENT_PARAMS params; params.event = -1; params.httpflv = 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 http_msg_buf_deinit(); sys_buf_deinit(); // close log log_close(); return 0; }