Lawrence
2020-06-10 76a75bf3d06b25cefee22245b4fb1eca0225f181
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
@@ -1,19 +1,22 @@
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;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.stereotype.Component;
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.gb28181.utils.DateUtil;
@@ -27,13 +30,16 @@
public class SIPCommander implements ISIPCommander {
   
   @Autowired
   private SipConfig config;
   private SipConfig sipConfig;
   
   @Autowired
   private SIPRequestHeaderProvider headerProvider;
   
   @Autowired
   private SipLayer sipLayer;
   @Autowired
   private VideoStreamSessionManager streamSession;
   
   /**
    * 云台方向放控制,使用配置文件中的默认镜头移动速度
@@ -46,7 +52,7 @@
    */
   @Override
   public boolean ptzdirectCmd(Device device, String channelId, int leftRight, int upDown) {
      return ptzCmd(device, channelId, leftRight, upDown, 0, config.getSpeed(), 0);
      return ptzCmd(device, channelId, leftRight, upDown, 0, sipConfig.getSpeed(), 0);
   }
   /**
@@ -72,7 +78,7 @@
    */  
   @Override
   public boolean ptzZoomCmd(Device device, String channelId, int inOut) {
      return ptzCmd(device, channelId, 0, 0, inOut, 0, config.getSpeed());
      return ptzCmd(device, channelId, 0, 0, inOut, 0, sipConfig.getSpeed());
   }
   /**
@@ -88,6 +94,51 @@
      return ptzCmd(device, channelId, 0, 0, inOut, 0, zoomSpeed);
   }
  
   /**
   * 云台指令码计算
   *
    * @param leftRight  镜头左移右移 0:停止 1:左移 2:右移
    * @param upDown     镜头上移下移 0:停止 1:上移 2:下移
    * @param inOut      镜头放大缩小 0:停止 1:缩小 2:放大
    * @param moveSpeed  镜头移动速度 默认 0XFF (0-255)
    * @param zoomSpeed  镜头缩放速度 默认 0X1 (0-15)
    */
    public static String cmdString(int leftRight, int upDown, int inOut, int moveSpeed, int zoomSpeed) {
      int cmdCode = 0;
      if (leftRight == 2) {
         cmdCode|=0x01;      // 右移
      } else if(leftRight == 1) {
         cmdCode|=0x02;      // 左移
      }
      if (upDown == 2) {
         cmdCode|=0x04;      // 下移
      } else if(upDown == 1) {
         cmdCode|=0x08;      // 上移
      }
      if (inOut == 2) {
         cmdCode |= 0x10;   // 放大
      } else if(inOut == 1) {
         cmdCode |= 0x20;   // 缩小
      }
      StringBuilder builder = new StringBuilder("A50F01");
      String strTmp;
      strTmp = String.format("%02X", cmdCode);
      builder.append(strTmp, 0, 2);
      strTmp = String.format("%02X", moveSpeed);
      builder.append(strTmp, 0, 2);
      builder.append(strTmp, 0, 2);
      // strTmp = String.format("%X", zoomSpeed);
      int zs=(zoomSpeed<<4 & 0XF0);
      strTmp = String.format("%X", zs);
      builder.append(strTmp, 0, 1).append("0");
      //计算校验码
      int checkCode = (0XA5 + 0X0F + 0X01 + cmdCode + moveSpeed + moveSpeed + zs/*(zoomSpeed << 4 & 0XF0)*/) % 0X100;
      strTmp = String.format("%02X", checkCode);
      builder.append(strTmp, 0, 2);
      return builder.toString();
}
   /**
    * 云台控制,支持方向与缩放控制
    * 
@@ -103,13 +154,14 @@
   public boolean ptzCmd(Device device, String channelId, int leftRight, int upDown, int inOut, int moveSpeed,
         int zoomSpeed) {
      try {
         String cmdStr= cmdString(leftRight, upDown, inOut, moveSpeed, zoomSpeed);
         StringBuffer ptzXml = new StringBuffer(200);
         ptzXml.append("<?xml version=\"1.0\" ?>");
         ptzXml.append("<Control>");
         ptzXml.append("<CmdType>DeviceControl</CmdType>");
         ptzXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");
         ptzXml.append("<DeviceID>" + channelId + "</DeviceID>");
         ptzXml.append("<PTZCmd>" + PtzCmdHelper.create(leftRight, upDown, inOut, moveSpeed, zoomSpeed) + "</PTZCmd>");
         ptzXml.append("<PTZCmd>" + cmdStr + "</PTZCmd>");
         ptzXml.append("<Info>");
         ptzXml.append("</Info>");
         ptzXml.append("</Control>");
@@ -117,7 +169,6 @@
         Request request = headerProvider.createMessageRequest(device, ptzXml.toString(), "ViaPtzBranch", "FromPtzTag", "ToPtzTag");
         
         transmitRequest(device, request);
         return true;
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
@@ -135,25 +186,70 @@
   public String playStreamCmd(Device device, String channelId) {
      try {
         
         //生成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 开始时间,格式要求:yyyy-MM-dd HH:mm:ss
    * @param endTime 结束时间,格式要求:yyyy-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");
@@ -163,14 +259,51 @@
           }
           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, 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();
      }
   }
   /**
@@ -323,22 +456,23 @@
    * @param endTime 结束时间,格式要求:yyyy-MM-dd HH:mm:ss
    */  
   @Override
   public boolean recordInfoQuery(Device device, String startTime, String endTime) {
   public boolean recordInfoQuery(Device device, String channelId, String startTime, String endTime) {
      
      try {
         StringBuffer catalogXml = new StringBuffer(200);
         catalogXml.append("<?xml version=\"1.0\" encoding=\"GB2312\"?>");
         catalogXml.append("<Query>");
         catalogXml.append("<CmdType>RecordInfo</CmdType>");
         catalogXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>");
         catalogXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>");
         catalogXml.append("<StartTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(startTime) + "</StartTime>");
         catalogXml.append("<EndTime>" + DateUtil.yyyy_MM_dd_HH_mm_ssToISO8601(endTime) + "</EndTime>");
         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的文本元素节点Type
         catalogXml.append("<Type>all</Type>");
         catalogXml.append("</Query>");
         recordInfoXml.append("<Type>all</Type>");
         recordInfoXml.append("</Query>");
         
         Request request = headerProvider.createMessageRequest(device, catalogXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag");
         Request request = headerProvider.createMessageRequest(device, recordInfoXml.toString(), "ViaRecordInfoBranch", "FromRecordInfoTag", "ToRecordInfoTag");
         transmitRequest(device, request);
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
@@ -391,11 +525,15 @@
      return false;
   }
   
   private void transmitRequest(Device device, Request request) throws SipException {
      if(device.getTransport().equals("TCP")) {
         sipLayer.getTcpSipProvider().sendRequest(request);
      } else if(device.getTransport().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;
   }
}