基于pjsip实现GBT28181协议

/** * @file SipStreamHandler.cpp * @brief 流媒体处理者类实现文件. * @see SipStreamHandler.h * * Change History : \n * -linxuan 创建 0000/0/0 */ #include "SipStreamHandler.h" #include "TNLog.h" namespace SIP { //! RTP负载类型定义 static std::string RTPMAP_PS = "96 PS/90000"; static std::string RTPMAP_H264 = "98 H264/90000"; static std::string RTPMAP_MPEG4 = "97 MPEG4/90000"; //! RTP范围定义[19000,19199] static const int START_RTP_PORT = 19000; static const int END_RTP_PORT = 19199; //! 因rtp、rtcp端口连续，故此处步长置为2 static int STEP_RTP_PORT = 2; //! 当前rtp端口 static std::atomic_int CUR_RTP_PORT(START_RTP_PORT); //! 流处理对象统计，也用来做gbt28181协议中y字段值组成部分 static std::atomic<int> g_streamCount(0); CSipStreamHandler::CSipStreamHandler(pjsip_endpoint *endpoint, PISipStreamNotify notifyObj, pjmedia_endpt *meidaEndpit, bool tcp/*=false*/) : CSipSendBaseHandler(endpoint, tcp) , m_stop(true) , m_meidaEndpit(meidaEndpit) , m_streamNotifyObj(notifyObj) { g_streamCount++; } CSipStreamHandler::~CSipStreamHandler() { m_stop = true; if (m_mediaStreamSdpInfo.isPlay) { pjsip_tx_data *tdata; pj_status_t status = pjsip_inv_end_session(m_mediaStreamSdpInfo.inv, PJSIP_SC_DECLINE, NULL, &tdata); if (status == PJ_SUCCESS && tdata != NULL) pjsip_inv_send_msg(m_mediaStreamSdpInfo.inv, tdata); else LOG_WARN_FMT("stream handler[%d] end session failed. code[%d]", m_handlerId, status); WaitMsg(SIP_COM_TIMEOUT); // 传输断开 pjmedia_transport_detach(m_mediaStreamSdpInfo.transport, this); pjmedia_transport_close(m_mediaStreamSdpInfo.transport); } } bool CSipStreamHandler::OnMessage(pjsip_rx_data *rdata) { if (NULL == rdata || NULL == rdata->msg_info.msg) { LOG_WARN("parameter error"); return false; } std::string srcIP = rdata->pkt_info.src_name; if (srcIP != m_sipProFillInfo.targetIP) { //LOG_WARN_FMT("stream handler[%d] parameter src ip error. srcIP[%s] targetIP[%s]", m_handlerId, srcIP.c_str(), m_sipProFillInfo.targetIP.c_str()); return false; } bool res = false; pjsip_method_e methodID = rdata->msg_info.msg->line.req.method.id; LOG_INFO_FMT("stream handler[%d] receive method[%d].", m_handlerId, methodID); switch (methodID) { case PJSIP_INVITE_METHOD: case PJSIP_CANCEL_METHOD: case PJSIP_ACK_METHOD: case PJSIP_BYE_METHOD: // 如上消息,接收到即认为处理完成 res = true; break; default: break; } return res; } // rtp流传输预留端口 [19000,19199] static int GetRtpPort() { int res = CUR_RTP_PORT; CUR_RTP_PORT += STEP_RTP_PORT; if (END_RTP_PORT <= CUR_RTP_PORT) CUR_RTP_PORT = START_RTP_PORT; return res; } bool CSipStreamHandler::StartStream(MediaStreamType type /*= SMST_PLAY*/ , time_t start /*= 0 */ , time_t end /*= 0 */ ) { // 启动后，不在操作 if (!m_stop) { LOG_INFO_FMT("stream handler[%d] stream started.", m_handlerId); return true; } m_stop = false; LOG_INFO_FMT("stream handler[%d] starting stream. type[%d]", m_handlerId, type); // sdp字段信息收集 int format = 0; switch (type) { case SMST_PLAY: format = 0; break; case SMST_PLAYBACK: case SMST_DOWNLOAD: format = 1; m_mediaStreamSdpInfo.u = "u="+ m_sipProFillInfo.targetID +":3"; break; default: LOG_ERROR_FMT("stream handler[%d] start stream failed. meidia stream type error. type[%d]", m_handlerId, type); return false; } m_mediaStreamSdpInfo.type = type; char temp[32] = { 0 }; char from[128] = { 0 }; char to[128] = { 0 }; char target[128] = { 0 }; char contact[128] = { 0 }; sprintf(temp, "y=%1d%s%04d", format, m_sipProFillInfo.srcID.substr(3, 5).c_str(), g_streamCount % 10000); m_mediaStreamSdpInfo.y = temp; m_mediaStreamSdpInfo.port = GetRtpPort(); LOG_INFO_FMT("stream handler[%d] rtp port[%d]", m_handlerId, m_mediaStreamSdpInfo.port); sprintf(from, "sip:%s@%s", m_sipProFillInfo.srcID.c_str(), m_sipProFillInfo.srcRealm.c_str()); sprintf(to, "sip:%s@%s", m_sipProFillInfo.targetID.c_str(), m_sipProFillInfo.srcRealm.c_str()); if (m_tcpFlg) sprintf(target, "sip:%s@%s:%d;transport=tcp", m_sipProFillInfo.targetID.c_str(), m_sipProFillInfo.targetIP.c_str(), m_sipProFillInfo.targetPort); else sprintf(target, "sip:%s@%s:%d", m_sipProFillInfo.targetID.c_str(), m_sipProFillInfo.targetIP.c_str(), m_sipProFillInfo.targetPort); if (m_tcpFlg) sprintf(contact, "sip:%s@%s:%d;transport=tcp", m_sipProFillInfo.srcID.c_str(), m_sipProFillInfo.srcIP.c_str(), m_sipProFillInfo.srcPort); else sprintf(contact, "sip:%s@%s:%d", m_sipProFillInfo.srcID.c_str(), m_sipProFillInfo.srcIP.c_str(), m_sipProFillInfo.srcPort); // 创建会话 pj_str_t tmpFrom = pj_str(from); pj_str_t tmpContact = pj_str(contact); pj_str_t tmpTo = pj_str(to); pj_str_t tmpTarget = pj_str(target); pj_status_t status = pjsip_dlg_create_uac(pjsip_ua_instance(), &tmpFrom, &tmpContact, &tmpTo, &tmpTarget, &m_mediaStreamSdpInfo.dlg); if (status != PJ_SUCCESS) { LOG_ERROR_FMT("stream handler[%d] start stream failed. pjsip_dlg_create_uac failed. code[%d]", m_handlerId, status); return false; } //创建SDP m_mediaStreamSdpInfo.start = start; m_mediaStreamSdpInfo.end = end; pjmedia_sdp_session *sdp; CreateSDP(&sdp, start, end); // INVITE 会话建立 status = pjsip_inv_create_uac(m_mediaStreamSdpInfo.dlg, sdp, 0, &m_mediaStreamSdpInfo.inv); if (status != PJ_SUCCESS) { pjsip_dlg_terminate(m_mediaStreamSdpInfo.dlg); LOG_ERROR_FMT("stream handler[%d] start stream failed. pjsip_inv_create_uac failed. code[%d]", m_handlerId, status); return false; } m_mediaStreamSdpInfo.inv->mod_data[0] = this; // invite请求信息填充 pjsip_tx_data *tdata; status = pjsip_inv_invite(m_mediaStreamSdpInfo.inv, &tdata); if (status != PJ_SUCCESS) { LOG_ERROR_FMT("stream handler[%d] start stream failed. pjsip_inv_invite failed. code[%d]", m_handlerId, status); return false; } //Subject填充 pj_str_t subjectName = pj_str("Subject"); char subjectBuf[128] = { 0 }; sprintf(subjectBuf, "%s:01,%s:01", m_sipProFillInfo.targetID.c_str(), m_sipProFillInfo.srcID.c_str()); pj_str_t subjectValue = pj_str(subjectBuf); pjsip_generic_string_hdr* hdr=pjsip_generic_string_hdr_create(m_pPool, &subjectName, &subjectValue); pjsip_msg_add_hdr(tdata->msg,(pjsip_hdr *)hdr); status = pjsip_inv_send_msg(m_mediaStreamSdpInfo.inv, tdata); if (status != PJ_SUCCESS) { LOG_ERROR_FMT("stream handler[%d] start stream failed. pjsip_inv_send_msg failed. code[%d]", m_handlerId, status); return false; } if (!WaitMsg(SIP_COM_TIMEOUT)) { LOG_ERROR_FMT("stream handler[%d] start stream failed. start stream r