panlinlin
2020-12-27 6968839f21fb05bf6e3204c2040ee47130006cd0
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
@@ -4,23 +4,22 @@
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import javax.sip.ClientTransaction;
import javax.sip.Dialog;
import javax.sip.InvalidArgumentException;
import javax.sip.SipException;
import javax.sip.SipFactory;
import javax.sip.SipProvider;
import javax.sip.TransactionDoesNotExistException;
import javax.sip.*;
import javax.sip.address.SipURI;
import javax.sip.header.CallIdHeader;
import javax.sip.header.Header;
import javax.sip.header.ViaHeader;
import javax.sip.message.Request;
import com.alibaba.fastjson.JSONObject;
import com.genersoft.iot.vmp.common.StreamInfo;
import com.genersoft.iot.vmp.conf.MediaServerConfig;
import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
import com.genersoft.iot.vmp.media.zlm.ZLMUtils;
import com.genersoft.iot.vmp.gb28181.event.SipSubscribe;
import com.genersoft.iot.vmp.media.zlm.ZLMHttpHookSubscribe;
import com.genersoft.iot.vmp.media.zlm.ZLMRTPServerFactory;
import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
@@ -40,6 +39,8 @@
 */
@Component
public class SIPCommander implements ISIPCommander {
   private final Logger logger = LoggerFactory.getLogger(SIPCommander.class);
   
   @Autowired
   private SipConfig sipConfig;
@@ -62,10 +63,16 @@
   private SipProvider udpSipProvider;
   @Autowired
   private ZLMUtils zlmUtils;
   private ZLMRTPServerFactory zlmrtpServerFactory;
   @Value("${media.rtp.enable}")
   private boolean rtpEnable;
   @Autowired
   private ZLMHttpHookSubscribe subscribe;
   @Autowired
   private SipSubscribe sipSubscribe;
@@ -219,7 +226,7 @@
         
         Request request = headerProvider.createMessageRequest(device, ptzXml.toString(), "ViaPtzBranch", "FromPtzTag", "ToPtzTag");
         
         transmitRequest(device, request);
         transmitRequest(device, request, null);
         return true;
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
@@ -254,81 +261,100 @@
         ptzXml.append("</Control>\r\n");
         
         Request request = headerProvider.createMessageRequest(device, ptzXml.toString(), "ViaPtzBranch", "FromPtzTag", "ToPtzTag");
         transmitRequest(device, request);
         transmitRequest(device, request, null);
         return true;
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
      } 
      return false;
   }
   /**
    * 请求预览视频流
    *
    *    请求预览视频流
    * @param device  视频设备
    * @param channelId  预览通道
    */
    * @param event hook订阅
    * @param errorEvent sip错误订阅
    */
   @Override
   public StreamInfo playStreamCmd(Device device, String channelId) {
   public void playStreamCmd(Device device, String channelId, ZLMHttpHookSubscribe.Event event, SipSubscribe.Event errorEvent) {
      try {
         String ssrc = streamSession.createPlaySsrc();
         String streamId = null;
         if (rtpEnable) {
            streamId = String.format("gb_play_%s_%s", device.getDeviceId(), channelId);
         }else {
            streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase();
         }
         String streamMode = device.getStreamMode().toUpperCase();
         MediaServerConfig mediaInfo = storager.getMediaInfo();
         if (mediaInfo == null) {
            logger.warn("点播时发现ZLM尚未连接...");
            return;
         }
         String mediaPort = null;
         // 使用动态udp端口
         if (rtpEnable) {
            mediaPort = zlmUtils.getNewRTPPort(ssrc) + "";
            mediaPort = zlmrtpServerFactory.createRTPServer(streamId) + "";
         }else {
            mediaPort = mediaInfo.getRtpProxyPort();
         }
         // 添加订阅
         JSONObject subscribeKey = new JSONObject();
         subscribeKey.put("app", "rtp");
         subscribeKey.put("id", streamId);
         subscribe.addSubscribe(ZLMHttpHookSubscribe.HookType.on_publish, subscribeKey, event);
         //
         StringBuffer content = new StringBuffer(200);
           content.append("v=0\r\n");
           content.append("o="+channelId+" 0 0 IN IP4 "+mediaInfo.getLocalIP()+"\r\n");
           content.append("s=Play\r\n");
           content.append("c=IN IP4 "+mediaInfo.getLocalIP()+"\r\n");
           content.append("t=0 0\r\n");
           if("TCP-PASSIVE".equals(streamMode)) {
              content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
         content.append("v=0\r\n");
//         content.append("o="+channelId+" 0 0 IN IP4 "+mediaInfo.getWanIp()+"\r\n");
         content.append("o="+"00000"+" 0 0 IN IP4 "+mediaInfo.getWanIp()+"\r\n");
         content.append("s=Play\r\n");
         content.append("c=IN IP4 "+mediaInfo.getWanIp()+"\r\n");
         content.append("t=0 0\r\n");
         if("TCP-PASSIVE".equals(streamMode)) {
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 126 125 99 34 98 97 96\r\n");
         }else if ("TCP-ACTIVE".equals(streamMode)) {
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 126 125 99 34 98 97 96\r\n");
         }else if("UDP".equals(streamMode)) {
              content.append("m=video "+ mediaPort +" RTP/AVP 96 98 97\r\n");
            content.append("m=video "+ mediaPort +" RTP/AVP 126 125 99 34 98 97 96\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-PASSIVE".equals(streamMode)){ // tcp被动模式
              content.append("a=setup:passive\r\n");
         content.append("a=recvonly\r\n");
         content.append("a=fmtp:126 profile-level-id=42e01e\r\n");
         content.append("a=rtpmap:126 H264/90000\r\n");
         content.append("a=rtpmap:125 H264S/90000\r\n");
         content.append("a=fmtp:125 profile-level-id=42e01e\r\n");
         content.append("a=rtpmap:99 MP4V-ES/90000\r\n");
         content.append("a=fmtp:99 profile-level-id=3\r\n");
         content.append("a=rtpmap:98 H264/90000\r\n");
         content.append("a=rtpmap:97 MPEG4/90000\r\n");
         content.append("a=rtpmap:96 PS/90000\r\n");
         if("TCP-PASSIVE".equals(streamMode)){ // tcp被动模式
            content.append("a=setup:passive\r\n");
            content.append("a=connection:new\r\n");
           }else if ("TCP-ACTIVE".equals(streamMode)) { // tcp主动模式
         }else if ("TCP-ACTIVE".equals(streamMode)) { // tcp主动模式
            content.append("a=setup:active\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, ssrc);
           ClientTransaction transaction = transmitRequest(device, request);
           streamSession.put(ssrc, transaction);
         content.append("y="+ssrc+"\r\n");//ssrc
         Request request = headerProvider.createInviteRequest(device, channelId, content.toString(), null, "live", null, ssrc);
         ClientTransaction transaction = transmitRequest(device, request, errorEvent);
         streamSession.put(streamId, transaction);
         DeviceChannel deviceChannel = storager.queryChannel(device.getDeviceId(), channelId);
         if (deviceChannel != null) {
            deviceChannel.setSsrc(ssrc);
            deviceChannel.setStreamId(streamId);
            storager.updateChannel(device.getDeviceId(), deviceChannel);
         }
         StreamInfo streamInfo = new StreamInfo();
         streamInfo.setSsrc(ssrc);
         streamInfo.setCahnnelId(channelId);
         streamInfo.setDeviceID(device.getDeviceId());
         storager.startPlay(streamInfo);
         return streamInfo;
      } catch ( SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
         return null;
      }
      }
   }
   
   /**
@@ -340,38 +366,53 @@
    * @param endTime 结束时间,格式要求:yyyy-MM-dd HH:mm:ss
    */ 
   @Override
   public StreamInfo playbackStreamCmd(Device device, String channelId, String startTime, String endTime) {
   public void playbackStreamCmd(Device device, String channelId, String startTime, String endTime, ZLMHttpHookSubscribe.Event event
         , SipSubscribe.Event errorEvent) {
      try {
         MediaServerConfig mediaInfo = storager.getMediaInfo();
         String ssrc = streamSession.createPlayBackSsrc();
         String streamId = String.format("%08x", Integer.parseInt(ssrc)).toUpperCase();
         // 添加订阅
         JSONObject subscribeKey = new JSONObject();
         subscribeKey.put("app", "rtp");
         subscribeKey.put("id", streamId);
         subscribe.addSubscribe(ZLMHttpHookSubscribe.HookType.on_publish, subscribeKey, event);
         //
         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 "+mediaInfo.getLocalIP()+"\r\n");
           content.append("c=IN IP4 "+mediaInfo.getWanIp()+"\r\n");
           content.append("t="+DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(startTime)+" "
               +DateUtil.yyyy_MM_dd_HH_mm_ssToTimestamp(endTime) +"\r\n");
         String mediaPort = null;
         // 使用动态udp端口
         if (rtpEnable) {
            mediaPort = zlmUtils.getNewRTPPort(ssrc) + "";
            mediaPort = zlmrtpServerFactory.createRTPServer(streamId) + "";
         }else {
            mediaPort = mediaInfo.getRtpProxyPort();
         }
         String streamMode = device.getStreamMode().toUpperCase();
         if("TCP-PASSIVE".equals(streamMode)) {
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 126 125 99 34 98 97 96\r\n");
         }else if ("TCP-ACTIVE".equals(streamMode)) {
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 96 98 97\r\n");
            content.append("m=video "+ mediaPort +" TCP/RTP/AVP 126 125 99 34 98 97 96\r\n");
         }else if("UDP".equals(streamMode)) {
            content.append("m=video "+ mediaPort +" RTP/AVP 96 98 97\r\n");
            content.append("m=video "+ mediaPort +" RTP/AVP 126 125 99 34 98 97 96\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");
         content.append("a=recvonly\r\n");
         content.append("a=fmtp:126 profile-level-id=42e01e\r\n");
         content.append("a=rtpmap:126 H264/90000\r\n");
         content.append("a=rtpmap:125 H264S/90000\r\n");
         content.append("a=fmtp:125 profile-level-id=42e01e\r\n");
         content.append("a=rtpmap:99 MP4V-ES/90000\r\n");
         content.append("a=fmtp:99 profile-level-id=3\r\n");
         content.append("a=rtpmap:98 H264/90000\r\n");
         content.append("a=rtpmap:97 MPEG4/90000\r\n");
         content.append("a=rtpmap:96 PS/90000\r\n");
         if("TCP-PASSIVE".equals(streamMode)){ // tcp被动模式
            content.append("a=setup:passive\r\n");
            content.append("a=connection:new\r\n");
@@ -382,20 +423,12 @@
           content.append("y="+ssrc+"\r\n");//ssrc
           
           Request request = headerProvider.createPlaybackInviteRequest(device, channelId, content.toString(), null, "playback", null);
           ClientTransaction transaction = transmitRequest(device, request);
           streamSession.put(ssrc, transaction);
         StreamInfo streamInfo = new StreamInfo();
         streamInfo.setSsrc(ssrc);
         streamInfo.setCahnnelId(channelId);
         streamInfo.setDeviceID(device.getDeviceId());
         boolean b = storager.startPlayback(streamInfo);
         return streamInfo;
           ClientTransaction transaction = transmitRequest(device, request, errorEvent);
           streamSession.put(streamId, transaction);
      } catch ( SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
         return null;
      }
   }
   
@@ -404,10 +437,10 @@
    * 
    */
   @Override
   public void streamByeCmd(String ssrc) {
   public void streamByeCmd(String streamId) {
      
      try {
         ClientTransaction transaction = streamSession.get(ssrc);
         ClientTransaction transaction = streamSession.get(streamId);
         if (transaction == null) {
            return;
         }
@@ -433,6 +466,8 @@
            clientTransaction = udpSipProvider.getNewClientTransaction(byeRequest);
         }
         dialog.sendRequest(clientTransaction);
         streamSession.remove(streamId);
         zlmrtpServerFactory.closeRTPServer(streamId);
      } catch (TransactionDoesNotExistException e) {
         e.printStackTrace();
      } catch (SipException e) {
@@ -550,7 +585,8 @@
         catalogXml.append("</Query>\r\n");
         
         Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaDeviceInfoBranch", "FromDeviceInfoTag", "ToDeviceInfoTag");
         transmitRequest(device, request);
         transmitRequest(device, request, null);
         
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
@@ -565,7 +601,7 @@
    * @param device 视频设备
    */ 
   @Override
   public boolean catalogQuery(Device device) {
   public boolean catalogQuery(Device device, SipSubscribe.Event errorEvent) {
      // 清空通道
      storager.cleanChannelsForDevice(device.getDeviceId());
      try {
@@ -577,8 +613,9 @@
         catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>\r\n");
         catalogXml.append("</Query>\r\n");
         
         Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaCatalogBranch", "FromCatalogTag", "ToCatalogTag");
         transmitRequest(device, request);
         Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaCatalogBranch", "FromCatalogTag", null);
         transmitRequest(device, request, errorEvent);
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
         return false;
@@ -610,8 +647,10 @@
         recordInfoXml.append("<Type>all</Type>\r\n");
         recordInfoXml.append("</Query>\r\n");
         
         Request request = headerProvider.createMessageRequest(device, recordInfoXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag");
         transmitRequest(device, request);
         Request request = headerProvider.createMessageRequest(device, recordInfoXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", null);
         transmitRequest(device, request, null);
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
         return false;
@@ -663,16 +702,30 @@
      return false;
   }
   
   private ClientTransaction transmitRequest(Device device, Request request) throws SipException {
   private ClientTransaction transmitRequest(Device device, Request request, SipSubscribe.Event errorEvent) throws SipException {
      ClientTransaction clientTransaction = null;
      if("TCP".equals(device.getTransport())) {
         clientTransaction = tcpSipProvider.getNewClientTransaction(request);
      } else if("UDP".equals(device.getTransport())) {
         clientTransaction = udpSipProvider.getNewClientTransaction(request);
      }
      // 添加订阅
      if (errorEvent != null) {
         CallIdHeader callIdHeader = (CallIdHeader)request.getHeader(CallIdHeader.NAME);
         sipSubscribe.addSubscribe(callIdHeader.getCallId(), errorEvent);
      }
      clientTransaction.sendRequest();
      return clientTransaction;
   }
   @Override
   public void closeRTPServer(Device device, String channelId) {
      if (rtpEnable) {
         String streamId = String.format("gb_play_%s_%s", device.getDeviceId(), channelId);
         zlmrtpServerFactory.closeRTPServer(streamId);
      }
   }
}