#if defined(_DEBUG) || defined(DEBUG) #include "sockutil.h" #include "rtsp-client.h" #include #include #include "sockpair.h" #include "cstringext.h" #include "sys/system.h" #include "cpm/unuse.h" #include "sdp.h" //#define UDP_MULTICAST_ADDR "239.0.0.2" extern "C" void rtp_receiver_tcp_input(uint8_t channel, const void* data, uint16_t bytes); extern "C" void rtp_receiver_test(socket_t rtp[2], const char* peer, int peerport[2], int payload, const char* encoding); extern "C" void* rtp_receiver_tcp_test(uint8_t interleave1, uint8_t interleave2, int payload, const char* encoding); extern "C" int rtsp_addr_is_multicast(const char* ip); struct rtsp_client_input_test_t { rtsp_client_t* rtsp; int transport; socket_t rtp[5][2]; unsigned short port[5][2]; }; static int rtsp_client_send(void* param, const char* uri, const void* req, size_t bytes) { //TODO: check uri and make socket //1. uri != rtsp describe uri(user input) //2. multi-uri if media_count > 1 struct rtsp_client_input_test_t* ctx = (struct rtsp_client_input_test_t*)param; return bytes; } static int rtpport(void* param, int media, const char* source, unsigned short rtp[2], char* ip, int len) { struct rtsp_client_input_test_t* ctx = (struct rtsp_client_input_test_t*)param; int m = rtsp_client_get_media_type(ctx->rtsp, media); //if (SDP_M_MEDIA_AUDIO != m && SDP_M_MEDIA_VIDEO != m) if (SDP_M_MEDIA_VIDEO != m) return 0; // ignore switch (ctx->transport) { case RTSP_TRANSPORT_RTP_UDP: // TODO: ipv6 assert(0 == sockpair_create("127.0.0.1", ctx->rtp[media], ctx->port[media])); rtp[0] = ctx->port[media][0]; rtp[1] = ctx->port[media][1]; if (rtsp_addr_is_multicast(ip)) { if (0 != socket_udp_multicast(ctx->rtp[media][0], ip, source, 16) || 0 != socket_udp_multicast(ctx->rtp[media][1], ip, source, 16)) return -1; } #if defined(UDP_MULTICAST_ADDR) else { if (0 != socket_udp_multicast(ctx->rtp[media][0], UDP_MULTICAST_ADDR, source, 16) || 0 != socket_udp_multicast(ctx->rtp[media][1], UDP_MULTICAST_ADDR, source, 16)) return -1; snprintf(ip, len, "%s", UDP_MULTICAST_ADDR); } #endif break; case RTSP_TRANSPORT_RTP_TCP: rtp[0] = 2 * media; rtp[1] = 2 * media + 1; break; default: assert(0); return -1; } return ctx->transport; } int rtsp_client_options(rtsp_client_t* rtsp, const char* commands); static void onrtp(void* param, uint8_t channel, const void* data, uint16_t bytes) { static int keepalive = 0; struct rtsp_client_input_test_t* ctx = (struct rtsp_client_input_test_t*)param; rtp_receiver_tcp_input(channel, data, bytes); if (++keepalive % 1000 == 0) { rtsp_client_play(ctx->rtsp, NULL, NULL); } } static int ondescribe(void* param, const char* sdp, int len) { struct rtsp_client_input_test_t* ctx = (struct rtsp_client_input_test_t*)param; return rtsp_client_setup(ctx->rtsp, sdp, len); } static int onsetup(void* param, int timeout, int64_t duration) { int i; uint64_t npt = 0; char ip[65]; u_short rtspport; struct rtsp_client_input_test_t* ctx = (struct rtsp_client_input_test_t*)param; assert(0 == rtsp_client_play(ctx->rtsp, &npt, NULL)); for (i = 0; i < rtsp_client_media_count(ctx->rtsp); i++) { int payload, port[2]; const char* encoding; const struct rtsp_header_transport_t* transport; transport = rtsp_client_get_media_transport(ctx->rtsp, i); encoding = rtsp_client_get_media_encoding(ctx->rtsp, i); payload = rtsp_client_get_media_payload(ctx->rtsp, i); if (RTSP_TRANSPORT_RTP_UDP == transport->transport) { //assert(RTSP_TRANSPORT_RTP_UDP == transport->transport); // udp only assert(0 == transport->multicast); // unicast only assert(transport->rtp.u.client_port1 == ctx->port[i][0]); assert(transport->rtp.u.client_port2 == ctx->port[i][1]); port[0] = transport->rtp.u.server_port1; port[1] = transport->rtp.u.server_port2; if (*transport->source) { rtp_receiver_test(ctx->rtp[i], transport->source, port, payload, encoding); } else { assert(0); //socket_getpeername(ctx->socket, ip, &rtspport); rtp_receiver_test(ctx->rtp[i], ip, port, payload, encoding); } } else if (RTSP_TRANSPORT_RTP_TCP == transport->transport) { //assert(transport->rtp.u.client_port1 == transport->interleaved1); //assert(transport->rtp.u.client_port2 == transport->interleaved2); rtp_receiver_tcp_test(transport->interleaved1, transport->interleaved2, payload, encoding); } else { assert(0); // TODO } } return 0; } static int onteardown(void* param) { return 0; } static int onplay(void* param, int media, const uint64_t* nptbegin, const uint64_t* nptend, const double* scale, const struct rtsp_rtp_info_t* rtpinfo, int count) { return 0; } static int onpause(void* param) { return 0; } void rtsp_client_input_test(const char* file) { int r; struct rtsp_client_input_test_t ctx; struct rtsp_client_handler_t handler; static char packet[2 * 1024 * 1024]; memset(&ctx, 0, sizeof(ctx)); handler.send = rtsp_client_send; handler.rtpport = rtpport; handler.ondescribe = ondescribe; handler.onsetup = onsetup; handler.onplay = onplay; handler.onpause = onpause; handler.onteardown = onteardown; handler.onrtp = onrtp; ctx.transport = RTSP_TRANSPORT_RTP_TCP; snprintf(packet, sizeof(packet), "rtsp://%s/%s", "127.0.0.1", file); // url //ctx.rtsp = rtsp_client_create(NULL, NULL, &handler, &ctx); ctx.rtsp = rtsp_client_create(packet, "username1", "password1", &handler, &ctx); assert(ctx.rtsp); assert(0 == rtsp_client_describe(ctx.rtsp)); FILE* fp = fopen(file, "rb"); while ( (r = fread(packet, 1, sizeof(packet), fp)) > 0) { assert(0 == rtsp_client_input(ctx.rtsp, packet, r)); } assert(0 == rtsp_client_teardown(ctx.rtsp)); rtsp_client_destroy(ctx.rtsp); fclose(fp); } #endif