From 805fa42a2c010d92b1c8fc29a60d32fd99fb4ea3 Mon Sep 17 00:00:00 2001 From: songww <songww@inspur.com> Date: 星期四, 14 五月 2020 14:19:42 +0800 Subject: [PATCH] 修复回放、停止预览bug,增加ZLM hook监听,长时间无人观看,停止摄像头推流。需开启ZLM的hook并配置url --- src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java | 249 +++++++++++++++++++++++++++++++++++-------------- 1 files changed, 178 insertions(+), 71 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java index 09f0934..82dcee8 100644 --- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java +++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java @@ -1,10 +1,13 @@ package com.genersoft.iot.vmp.gb28181.transmit.cmd.impl; import java.text.ParseException; -import java.util.Random; +import javax.sip.ClientTransaction; +import javax.sip.Dialog; import javax.sip.InvalidArgumentException; import javax.sip.SipException; +import javax.sip.TransactionDoesNotExistException; +import javax.sip.header.ViaHeader; import javax.sip.message.Request; import org.springframework.beans.factory.annotation.Autowired; @@ -13,9 +16,10 @@ import com.genersoft.iot.vmp.conf.SipConfig; import com.genersoft.iot.vmp.gb28181.SipLayer; import com.genersoft.iot.vmp.gb28181.bean.Device; +import com.genersoft.iot.vmp.gb28181.session.VideoStreamSessionManager; import com.genersoft.iot.vmp.gb28181.transmit.cmd.ISIPCommander; import com.genersoft.iot.vmp.gb28181.transmit.cmd.SIPRequestHeaderProvider; -import com.genersoft.iot.vmp.storager.IVideoManagerStorager; +import com.genersoft.iot.vmp.gb28181.utils.DateUtil; /** * @Description:璁惧鑳藉姏鎺ュ彛锛岀敤浜庡畾涔夎澶囩殑鎺у埗銆佹煡璇㈣兘鍔� @@ -26,7 +30,7 @@ public class SIPCommander implements ISIPCommander { @Autowired - private SipConfig config; + private SipConfig sipConfig; @Autowired private SIPRequestHeaderProvider headerProvider; @@ -35,65 +39,65 @@ private SipLayer sipLayer; @Autowired - private IVideoManagerStorager storager; - + private VideoStreamSessionManager streamSession; + /** * 浜戝彴鏂瑰悜鏀炬帶鍒讹紝浣跨敤閰嶇疆鏂囦欢涓殑榛樿闀滃ご绉诲姩閫熷害 * - * @param deviceId 鎺у埗璁惧 + * @param device 鎺у埗璁惧 * @param channelId 棰勮閫氶亾 * @param leftRight 闀滃ご宸︾Щ鍙崇Щ 0:鍋滄 1:宸︾Щ 2:鍙崇Щ * @param upDown 闀滃ご涓婄Щ涓嬬Щ 0:鍋滄 1:涓婄Щ 2:涓嬬Щ * @param moveSpeed 闀滃ご绉诲姩閫熷害 */ @Override - public boolean ptzdirectCmd(String deviceId, String channelId, int leftRight, int upDown) { - return ptzCmd(deviceId, channelId, leftRight, upDown, 0, config.getSpeed(), 0); + public boolean ptzdirectCmd(Device device, String channelId, int leftRight, int upDown) { + return ptzCmd(device, channelId, leftRight, upDown, 0, sipConfig.getSpeed(), 0); } /** * 浜戝彴鏂瑰悜鏀炬帶鍒� * - * @param deviceId 鎺у埗璁惧 + * @param device 鎺у埗璁惧 * @param channelId 棰勮閫氶亾 * @param leftRight 闀滃ご宸︾Щ鍙崇Щ 0:鍋滄 1:宸︾Щ 2:鍙崇Щ * @param upDown 闀滃ご涓婄Щ涓嬬Щ 0:鍋滄 1:涓婄Щ 2:涓嬬Щ * @param moveSpeed 闀滃ご绉诲姩閫熷害 */ @Override - public boolean ptzdirectCmd(String deviceId, String channelId, int leftRight, int upDown, int moveSpeed) { - return ptzCmd(deviceId, channelId, leftRight, upDown, 0, moveSpeed, 0); + public boolean ptzdirectCmd(Device device, String channelId, int leftRight, int upDown, int moveSpeed) { + return ptzCmd(device, channelId, leftRight, upDown, 0, moveSpeed, 0); } /** * 浜戝彴缂╂斁鎺у埗锛屼娇鐢ㄩ厤缃枃浠朵腑鐨勯粯璁ら暅澶寸缉鏀鹃�熷害 * - * @param deviceId 鎺у埗璁惧 + * @param device 鎺у埗璁惧 * @param channelId 棰勮閫氶亾 * @param inOut 闀滃ご鏀惧ぇ缂╁皬 0:鍋滄 1:缂╁皬 2:鏀惧ぇ */ @Override - public boolean ptzZoomCmd(String deviceId, String channelId, int inOut) { - return ptzCmd(deviceId, channelId, 0, 0, inOut, 0, config.getSpeed()); + public boolean ptzZoomCmd(Device device, String channelId, int inOut) { + return ptzCmd(device, channelId, 0, 0, inOut, 0, sipConfig.getSpeed()); } /** * 浜戝彴缂╂斁鎺у埗 * - * @param deviceId 鎺у埗璁惧 + * @param device 鎺у埗璁惧 * @param channelId 棰勮閫氶亾 * @param inOut 闀滃ご鏀惧ぇ缂╁皬 0:鍋滄 1:缂╁皬 2:鏀惧ぇ * @param zoomSpeed 闀滃ご缂╂斁閫熷害 */ @Override - public boolean ptzZoomCmd(String deviceId, String channelId, int inOut, int zoomSpeed) { - return ptzCmd(deviceId, channelId, 0, 0, inOut, 0, zoomSpeed); + public boolean ptzZoomCmd(Device device, String channelId, int inOut, int zoomSpeed) { + return ptzCmd(device, channelId, 0, 0, inOut, 0, zoomSpeed); } /** * 浜戝彴鎺у埗锛屾敮鎸佹柟鍚戜笌缂╂斁鎺у埗 * - * @param deviceId 鎺у埗璁惧 + * @param device 鎺у埗璁惧 * @param channelId 棰勮閫氶亾 * @param leftRight 闀滃ご宸︾Щ鍙崇Щ 0:鍋滄 1:宸︾Щ 2:鍙崇Щ * @param upDown 闀滃ご涓婄Щ涓嬬Щ 0:鍋滄 1:涓婄Щ 2:涓嬬Щ @@ -102,10 +106,9 @@ * @param zoomSpeed 闀滃ご缂╂斁閫熷害 */ @Override - public boolean ptzCmd(String deviceId, String channelId, int leftRight, int upDown, int inOut, int moveSpeed, + public boolean ptzCmd(Device device, String channelId, int leftRight, int upDown, int inOut, int moveSpeed, int zoomSpeed) { try { - Device device = storager.queryVideoDevice(deviceId); StringBuffer ptzXml = new StringBuffer(200); ptzXml.append("<?xml version=\"1.0\" ?>"); ptzXml.append("<Control>"); @@ -119,7 +122,7 @@ Request request = headerProvider.createMessageRequest(device, ptzXml.toString(), "ViaPtzBranch", "FromPtzTag", "ToPtzTag"); - transmitRequest(device.getTransport(), request); + transmitRequest(device, request); return true; } catch (SipException | ParseException | InvalidArgumentException e) { @@ -131,34 +134,77 @@ /** * 璇锋眰棰勮瑙嗛娴� * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 * @param channelId 棰勮閫氶亾 */ @Override - public String playStreamCmd(String deviceId, String channelId) { + public String playStreamCmd(Device device, String channelId) { try { - Device device = storager.queryVideoDevice(deviceId); - - //鐢熸垚ssrc鏍囪瘑鏁版嵁娴� 10浣嶆暟瀛� - String ssrc = ""; - Random random = new Random(); - // ZLMediaServer鏈�澶ц瘑鍒�7FFFFFFF鍗�2147483647锛屾墍浠ラ殢鏈烘暟涓嶈兘瓒呰繃杩欎釜鏁� - ssrc = String.valueOf(random.nextInt(2147483647)); + String ssrc = streamSession.createPlaySsrc(); + String transport = device.getTransport(); // StringBuffer content = new StringBuffer(200); content.append("v=0\r\n"); - content.append("o="+channelId+" 0 0 IN IP4 "+config.getSipIp()+"\r\n"); + content.append("o="+channelId+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n"); content.append("s=Play\r\n"); - content.append("c=IN IP4 "+config.getMediaIp()+"\r\n"); + content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n"); content.append("t=0 0\r\n"); + if("TCP".equals(transport)) { + content.append("m=video "+sipConfig.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n"); + } + if("UDP".equals(transport)) { + content.append("m=video "+sipConfig.getMediaPort()+" RTP/AVP 96 98 97\r\n"); + } + content.append("a=recvonly\r\n"); + content.append("a=rtpmap:96 PS/90000\r\n"); + content.append("a=rtpmap:98 H264/90000\r\n"); + content.append("a=rtpmap:97 MPEG4/90000\r\n"); + if("TCP".equals(transport)){ + content.append("a=setup:passive\r\n"); + content.append("a=connection:new\r\n"); + } + content.append("y="+ssrc+"\r\n");//ssrc + + Request request = headerProvider.createInviteRequest(device, channelId, content.toString(), null, "live", null); + + ClientTransaction transaction = transmitRequest(device, request); + streamSession.put(ssrc, transaction); + return ssrc; + } catch ( SipException | ParseException | InvalidArgumentException e) { + e.printStackTrace(); + return null; + } + } + + /** + * 璇锋眰鍥炴斁瑙嗛娴� + * + * @param device 瑙嗛璁惧 + * @param channelId 棰勮閫氶亾 + * @param startTime 寮�濮嬫椂闂�,鏍煎紡瑕佹眰锛歽yyy-MM-dd HH:mm:ss + * @param endTime 缁撴潫鏃堕棿,鏍煎紡瑕佹眰锛歽yyy-MM-dd HH:mm:ss + */ + @Override + public String playbackStreamCmd(Device device, String channelId, String startTime, String endTime) { + try { + + String ssrc = streamSession.createPlayBackSsrc(); + // + StringBuffer content = new StringBuffer(200); + content.append("v=0\r\n"); + content.append("o="+sipConfig.getSipId()+" 0 0 IN IP4 "+sipConfig.getSipIp()+"\r\n"); + content.append("s=Playback\r\n"); + content.append("u="+channelId+":0\r\n"); + content.append("c=IN IP4 "+sipConfig.getMediaIp()+"\r\n"); + content.append("t="+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime)+" "+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime) +"\r\n"); if(device.getTransport().equals("TCP")) { - content.append("m=video "+config.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n"); + content.append("m=video "+sipConfig.getMediaPort()+" TCP/RTP/AVP 96 98 97\r\n"); } if(device.getTransport().equals("UDP")) { - content.append("m=video "+config.getMediaPort()+" RTP/AVP 96 98 97\r\n"); + content.append("m=video "+sipConfig.getMediaPort()+" RTP/AVP 96 98 97\r\n"); } - content.append("a=sendrecv\r\n"); + content.append("a=recvonly\r\n"); content.append("a=rtpmap:96 PS/90000\r\n"); content.append("a=rtpmap:98 H264/90000\r\n"); content.append("a=rtpmap:97 MPEG4/90000\r\n"); @@ -168,94 +214,131 @@ } content.append("y="+ssrc+"\r\n");//ssrc - Request request = headerProvider.createInviteRequest(device, content.toString(), null, "live", null); + Request request = headerProvider.createPlaybackInviteRequest(device, channelId, content.toString(), null, "playback", null); - transmitRequest(device.getTransport(), request); + ClientTransaction transaction = transmitRequest(device, request); + streamSession.put(ssrc, transaction); return ssrc; } catch ( SipException | ParseException | InvalidArgumentException e) { e.printStackTrace(); return null; - } + } + } + + /** + * 瑙嗛娴佸仠姝� + * + * @param device 瑙嗛璁惧 + * @param channelId 棰勮閫氶亾 + */ + @Override + public void streamByeCmd(String ssrc) { + + try { + ClientTransaction transaction = streamSession.get(ssrc); + if (transaction == null) { + return; + } + + Dialog dialog = transaction.getDialog(); + if (dialog == null) { + return; + } + Request byeRequest = dialog.createRequest(Request.BYE); + ViaHeader viaHeader = (ViaHeader) byeRequest.getHeader(ViaHeader.NAME); + String protocol = viaHeader.getTransport().toUpperCase(); + ClientTransaction clientTransaction = null; + if("TCP".equals(protocol)) { + clientTransaction = sipLayer.getTcpSipProvider().getNewClientTransaction(byeRequest); + } else if("UDP".equals(protocol)) { + clientTransaction = sipLayer.getUdpSipProvider().getNewClientTransaction(byeRequest); + } + dialog.sendRequest(clientTransaction); + } catch (TransactionDoesNotExistException e) { + e.printStackTrace(); + } catch (SipException e) { + e.printStackTrace(); + } } /** * 璇煶骞挎挱 * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 * @param channelId 棰勮閫氶亾 */ @Override - public String audioBroadcastCmd(String deviceId, String channelId) { + public boolean audioBroadcastCmd(Device device, String channelId) { // TODO Auto-generated method stub - return null; + return false; } /** * 闊宠棰戝綍鍍忔帶鍒� * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 * @param channelId 棰勮閫氶亾 */ @Override - public String recordCmd(String deviceId, String channelId) { + public boolean recordCmd(Device device, String channelId) { // TODO Auto-generated method stub - return null; + return false; } /** * 鎶ヨ甯冮槻/鎾ら槻鍛戒护 * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 */ @Override - public String guardCmd(String deviceId) { + public boolean guardCmd(Device device) { // TODO Auto-generated method stub - return null; + return false; } /** * 鎶ヨ澶嶄綅鍛戒护 * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 */ @Override - public String alarmCmd(String deviceId) { + public boolean alarmCmd(Device device) { // TODO Auto-generated method stub - return null; + return false; } /** * 寮哄埗鍏抽敭甯у懡浠�,璁惧鏀跺埌姝ゅ懡浠ゅ簲绔嬪埢鍙戦�佷竴涓狪DR甯� * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 * @param channelId 棰勮閫氶亾 */ @Override - public String iFameCmd(String deviceId, String channelId) { + public boolean iFameCmd(Device device, String channelId) { // TODO Auto-generated method stub - return null; + return false; } /** * 鐪嬪畧浣嶆帶鍒跺懡浠� * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 */ @Override - public String homePositionCmd(String deviceId) { + public boolean homePositionCmd(Device device) { // TODO Auto-generated method stub - return null; + return false; } /** * 璁惧閰嶇疆鍛戒护 * - * @param deviceId 瑙嗛璁惧 + * @param device 瑙嗛璁惧 */ @Override - public String deviceConfigCmd(String deviceId) { + public boolean deviceConfigCmd(Device device) { // TODO Auto-generated method stub - return null; + return false; } /** @@ -286,8 +369,8 @@ catalogXml.append("</Query>"); Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaDeviceInfoBranch", "FromDeviceInfoTag", "ToDeviceInfoTag"); + transmitRequest(device, request); - transmitRequest(device.getTransport(), request); } catch (SipException | ParseException | InvalidArgumentException e) { e.printStackTrace(); return false; @@ -312,9 +395,7 @@ catalogXml.append("</Query>"); Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaCatalogBranch", "FromCatalogTag", "ToCatalogTag"); - - transmitRequest(device.getTransport(), request); - + transmitRequest(device, request); } catch (SipException | ParseException | InvalidArgumentException e) { e.printStackTrace(); return false; @@ -326,11 +407,33 @@ * 鏌ヨ褰曞儚淇℃伅 * * @param device 瑙嗛璁惧 + * @param startTime 寮�濮嬫椂闂�,鏍煎紡瑕佹眰锛歽yyy-MM-dd HH:mm:ss + * @param endTime 缁撴潫鏃堕棿,鏍煎紡瑕佹眰锛歽yyy-MM-dd HH:mm:ss */ @Override - public boolean recordInfoQuery(Device device) { - // TODO Auto-generated method stub - return false; + public boolean recordInfoQuery(Device device, String channelId, String startTime, String endTime) { + + try { + StringBuffer recordInfoXml = new StringBuffer(200); + recordInfoXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>"); + recordInfoXml.append("<Query>"); + recordInfoXml.append("<CmdType>RecordInfo</CmdType>"); + recordInfoXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>"); + recordInfoXml.append("<DeviceID>" + channelId + "</DeviceID>"); + recordInfoXml.append("<StartTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(startTime) + "</StartTime>"); + recordInfoXml.append("<EndTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(endTime) + "</EndTime>"); + recordInfoXml.append("<Secrecy>0</Secrecy>"); + // 澶у崕NVR瑕佹眰蹇呴』澧炲姞涓�涓�间负all鐨勬枃鏈厓绱犺妭鐐筎ype + recordInfoXml.append("<Type>all</Type>"); + recordInfoXml.append("</Query>"); + + Request request = headerProvider.createMessageRequest(device, recordInfoXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag"); + transmitRequest(device, request); + } catch (SipException | ParseException | InvalidArgumentException e) { + e.printStackTrace(); + return false; + } + return true; } /** @@ -377,11 +480,15 @@ return false; } - private void transmitRequest(String transport, Request request) throws SipException { - if(transport.equals("TCP")) { - sipLayer.getTcpSipProvider().sendRequest(request); - } else if(transport.equals("UDP")) { - sipLayer.getUdpSipProvider().sendRequest(request); + private ClientTransaction transmitRequest(Device device, Request request) throws SipException { + ClientTransaction clientTransaction = null; + if("TCP".equals(device.getTransport())) { + clientTransaction = sipLayer.getTcpSipProvider().getNewClientTransaction(request); + } else if("UDP".equals(device.getTransport())) { + clientTransaction = sipLayer.getUdpSipProvider().getNewClientTransaction(request); } + clientTransaction.sendRequest(); + return clientTransaction; } + } -- Gitblit v1.8.0