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