648540858
2022-09-24 d7a1b94f905c5f28c9c8f2d48c3f9e28ebcf9cc4
src/main/java/com/genersoft/iot/vmp/gb28181/transmit/cmd/impl/SIPCommander.java
@@ -21,10 +21,12 @@
import com.genersoft.iot.vmp.service.bean.SSRCInfo;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
import com.genersoft.iot.vmp.utils.GitUtil;
import gov.nist.javax.sip.SipProviderImpl;
import gov.nist.javax.sip.SipStackImpl;
import gov.nist.javax.sip.message.MessageFactoryImpl;
import gov.nist.javax.sip.message.SIPRequest;
import gov.nist.javax.sip.stack.SIPClientTransaction;
import gov.nist.javax.sip.stack.SIPDialog;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -41,7 +43,9 @@
import javax.sip.message.Request;
import java.lang.reflect.Field;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
/**    
 * @description:设备能力接口,用于定义设备的控制、查询能力   
@@ -59,6 +63,9 @@
   @Autowired
   private SipFactory sipFactory;
   @Autowired
   private GitUtil gitUtil;
   @Autowired
   @Qualifier(value="tcpSipProvider")
@@ -285,7 +292,7 @@
         ptzXml.append("<ControlPriority>5</ControlPriority>\r\n");
         ptzXml.append("</Info>\r\n");
         ptzXml.append("</Control>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equalsIgnoreCase("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -320,7 +327,7 @@
         ptzXml.append("<ControlPriority>5</ControlPriority>\r\n");
         ptzXml.append("</Info>\r\n");
         ptzXml.append("</Control>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equalsIgnoreCase("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -429,7 +436,14 @@
         }), e ->{
            // 这里为例避免一个通道的点播只有一个callID这个参数使用一个固定值
            streamSession.put(device.getDeviceId(), channelId ,"play", stream, ssrcInfo.getSsrc(), mediaServerItem.getId(), ((ResponseEvent)e.event).getClientTransaction(), VideoStreamSessionManager.SessionType.play);
            streamSession.put(device.getDeviceId(), channelId ,"play", e.dialog);
            Dialog sipDialog = null;
            if (e.dialog == null) {
               SIPClientTransaction clientTransaction = (SIPClientTransaction)((ResponseEvent)e.event).getClientTransaction();
               sipDialog = new SIPDialog(clientTransaction, clientTransaction.getLastResponse());
            }else {
               sipDialog = e.dialog;
            }
            streamSession.put(device.getDeviceId(), channelId ,"play", sipDialog);
            okEvent.response(e);
         });
@@ -663,6 +677,38 @@
      streamByeCmd(deviceId, channelId, stream, callId, null);
   }
   @Override
   public void streamByeCmd(SIPDialog dialog, String channelId, SIPRequest request, SipSubscribe.Event okEvent) throws SipException, ParseException, InvalidArgumentException {
      Request byeRequest = dialog.createRequest(Request.BYE);
      SipURI byeURI = (SipURI) byeRequest.getRequestURI();
      byeURI.setHost(request.getRemoteAddress().getHostAddress());
      byeURI.setPort(request.getRemotePort());
      byeURI.setUser(channelId);
      ViaHeader viaHeader = (ViaHeader) byeRequest.getHeader(ViaHeader.NAME);
      String protocol = viaHeader.getTransport().toUpperCase();
      viaHeader.setRPort();
      // 增加Contact header
      Address concatAddress = sipFactory.createAddressFactory().createAddress(sipFactory.createAddressFactory().createSipURI(sipConfig.getId(), sipConfig.getIp()+":"+sipConfig.getPort()));
      byeRequest.addHeader(sipFactory.createHeaderFactory().createContactHeader(concatAddress));
      UserAgentHeader userAgentHeader = HeaderUtils.createUserAgentHeader(sipFactory);
      byeRequest.addHeader(userAgentHeader);
      ClientTransaction clientTransaction = null;
      if("TCP".equals(protocol)) {
         clientTransaction = tcpSipProvider.getNewClientTransaction(byeRequest);
      } else if("UDP".equals(protocol)) {
         clientTransaction = udpSipProvider.getNewClientTransaction(byeRequest);
      }
      CallIdHeader callIdHeader = (CallIdHeader) byeRequest.getHeader(CallIdHeader.NAME);
      if (okEvent != null) {
         sipSubscribe.addOkSubscribe(callIdHeader.getCallId(), okEvent);
      }
      CSeqHeader cSeqHeader = (CSeqHeader)byeRequest.getHeader(CSeqHeader.NAME);
      cSeqHeader.setSeqNumber(redisCatchStorage.getCSEQ());
      dialog.sendRequest(clientTransaction);
   }
   /**
    * 视频流停止
    */
@@ -690,7 +736,7 @@
            dialog = streamSession.getDialogByStream(ssrcTransaction.getDeviceId(), ssrcTransaction.getChannelId(), ssrcTransaction.getStream());
         }
         mediaServerService.releaseSsrc(ssrcTransaction.getMediaServerId(), ssrcTransaction.getSsrc());
         mediaServerService.closeRTPServer(ssrcTransaction.getDeviceId(), ssrcTransaction.getChannelId(), ssrcTransaction.getStream());
         mediaServerService.closeRTPServer(ssrcTransaction.getMediaServerId(), ssrcTransaction.getStream());
         streamSession.remove(ssrcTransaction.getDeviceId(), ssrcTransaction.getChannelId(), ssrcTransaction.getStream());
         if (dialog == null) {
@@ -727,7 +773,7 @@
         // 增加Contact header
         Address concatAddress = sipFactory.createAddressFactory().createAddress(sipFactory.createAddressFactory().createSipURI(sipConfig.getId(), sipConfig.getIp()+":"+sipConfig.getPort()));
         byeRequest.addHeader(sipFactory.createHeaderFactory().createContactHeader(concatAddress));
         UserAgentHeader userAgentHeader = SipUtils.createUserAgentHeader(sipFactory);
         UserAgentHeader userAgentHeader = SipUtils.createUserAgentHeader(sipFactory, gitUtil);
         byeRequest.addHeader(userAgentHeader);
         ClientTransaction clientTransaction = null;
         if("TCP".equals(protocol)) {
@@ -755,21 +801,9 @@
    * 语音广播
    * 
    * @param device  视频设备
    * @param channelId  预览通道
    */
   @Override
   public boolean audioBroadcastCmd(Device device, String channelId) {
      // 改为新的实现
      return false;
   }
   /**
    * 语音广播
    *
    * @param device  视频设备
    */
   @Override
   public boolean audioBroadcastCmd(Device device) {
   public boolean audioBroadcastCmd(Device device,String channelId, SipSubscribe.Event okEvent, SipSubscribe.Event errorEvent) {
      try {
         StringBuffer broadcastXml = new StringBuffer(200);
         String charset = device.getCharset();
@@ -778,7 +812,7 @@
         broadcastXml.append("<CmdType>Broadcast</CmdType>\r\n");
         broadcastXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>\r\n");
         broadcastXml.append("<SourceID>" + sipConfig.getId() + "</SourceID>\r\n");
         broadcastXml.append("<TargetID>" + device.getDeviceId() + "</TargetID>\r\n");
         broadcastXml.append("<TargetID>" + channelId + "</TargetID>\r\n");
         broadcastXml.append("</Notify>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
@@ -807,15 +841,15 @@
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
         Request request = headerProvider.createMessageRequest(device, broadcastXml.toString(), SipUtils.getNewViaTag(), SipUtils.getNewFromTag(), null, callIdHeader);
         transmitRequest(device, request, errorEvent);
      } catch (SipException | ParseException | InvalidArgumentException e) {
         e.printStackTrace();
      }
   }
      }
   }
   /**
    * 音视频录像控制
    * 
@@ -1242,7 +1276,7 @@
            recordInfoXml.append("<Type>" + type+"</Type>\r\n");
         }
         recordInfoXml.append("</Query>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -1299,7 +1333,7 @@
            cmdXml.append("<EndAlarmTime>" + endTime + "</EndAlarmTime>\r\n");
         }
         cmdXml.append("</Query>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -1335,7 +1369,7 @@
         }
         cmdXml.append("<ConfigType>" + configType + "</ConfigType>\r\n");
         cmdXml.append("</Query>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -1368,7 +1402,7 @@
            cmdXml.append("<DeviceID>" + channelId + "</DeviceID>\r\n");
         }
         cmdXml.append("</Query>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -1398,7 +1432,7 @@
         mobilePostitionXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>\r\n");
         mobilePostitionXml.append("<Interval>60</Interval>\r\n");
         mobilePostitionXml.append("</Query>\r\n");
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
@@ -1420,7 +1454,7 @@
    * @return         true = 命令发送成功
    */
   @Override
   public boolean mobilePositionSubscribe(Device device, Dialog dialog, SipSubscribe.Event okEvent ,SipSubscribe.Event errorEvent) {
   public SIPRequest mobilePositionSubscribe(Device device, SIPRequest requestOld, SipSubscribe.Event okEvent ,SipSubscribe.Event errorEvent) {
      try {
         StringBuffer subscribePostitionXml = new StringBuffer(200);
         String charset = device.getCharset();
@@ -1430,38 +1464,27 @@
         subscribePostitionXml.append("<SN>" + (int)((Math.random()*9+1)*100000) + "</SN>\r\n");
         subscribePostitionXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>\r\n");
         if (device.getSubscribeCycleForMobilePosition() > 0) {
            subscribePostitionXml.append("<Interval>" + String.valueOf(device.getMobilePositionSubmissionInterval()) + "</Interval>\r\n");
            subscribePostitionXml.append("<Interval>" + device.getMobilePositionSubmissionInterval() + "</Interval>\r\n");
         }
         subscribePostitionXml.append("</Query>\r\n");
         Request request;
         if (dialog != null) {
            SipURI requestURI = sipFactory.createAddressFactory().createSipURI(device.getDeviceId(), device.getHostAddress());
            request = dialog.createRequest(Request.SUBSCRIBE);
            ExpiresHeader expiresHeader = sipFactory.createHeaderFactory().createExpiresHeader(device.getSubscribeCycleForCatalog());
            request.setExpires(expiresHeader);
         CallIdHeader callIdHeader;
            request.setRequestURI(requestURI);
            ContentTypeHeader contentTypeHeader = sipFactory.createHeaderFactory().createContentTypeHeader("Application", "MANSCDP+xml");
            request.setContent(subscribePostitionXml.toString(), contentTypeHeader);
            CSeqHeader cSeqHeader = (CSeqHeader)request.getHeader(CSeqHeader.NAME);
            cSeqHeader.setSeqNumber(redisCatchStorage.getCSEQ());
            request.removeHeader(CSeqHeader.NAME);
            request.addHeader(cSeqHeader);
         if (requestOld != null) {
            callIdHeader = sipFactory.createHeaderFactory().createCallIdHeader(requestOld.getCallIdHeader().getCallId());
         }else {
            CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
            callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
                  : udpSipProvider.getNewCallId();
            request = headerProvider.createSubscribeRequest(device, subscribePostitionXml.toString(), SipUtils.getNewViaTag(), SipUtils.getNewFromTag(), null, device.getSubscribeCycleForMobilePosition(), "presence" ,callIdHeader); //Position;id=" + tm.substring(tm.length() - 4));
         }
         SIPRequest request = (SIPRequest)headerProvider.createSubscribeRequest(device, subscribePostitionXml.toString(), requestOld, device.getSubscribeCycleForMobilePosition(), "presence" ,callIdHeader); //Position;id=" + tm.substring(tm.length() - 4));
         transmitRequest(device, request, errorEvent, okEvent);
         return true;
         return request;
      } catch ( NumberFormatException | ParseException | InvalidArgumentException   | SipException e) {
         e.printStackTrace();
         return false;
         return null;
      }
   }
@@ -1511,7 +1534,7 @@
         CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
               : udpSipProvider.getNewCallId();
         Request request = headerProvider.createSubscribeRequest(device, cmdXml.toString(), SipUtils.getNewViaTag(), SipUtils.getNewFromTag(), null, expires, "presence" , callIdHeader);
         Request request = headerProvider.createSubscribeRequest(device, cmdXml.toString(), null, expires, "presence" , callIdHeader);
         transmitRequest(device, request);
         return true;
@@ -1523,7 +1546,7 @@
   }
   @Override
   public boolean catalogSubscribe(Device device, Dialog dialog, SipSubscribe.Event okEvent, SipSubscribe.Event errorEvent) {
   public SIPRequest catalogSubscribe(Device device, SIPRequest requestOld, SipSubscribe.Event okEvent, SipSubscribe.Event errorEvent) {
      try {
         StringBuffer cmdXml = new StringBuffer(200);
         String charset = device.getCharset();
@@ -1534,40 +1557,24 @@
         cmdXml.append("<DeviceID>" + device.getDeviceId() + "</DeviceID>\r\n");
         cmdXml.append("</Query>\r\n");
         CallIdHeader callIdHeader ;
         Request request;
         if (dialog != null) {
            SipURI requestURI = sipFactory.createAddressFactory().createSipURI(device.getDeviceId(), device.getHostAddress());
            request = dialog.createRequest(Request.SUBSCRIBE);
            ExpiresHeader expiresHeader = sipFactory.createHeaderFactory().createExpiresHeader(device.getSubscribeCycleForCatalog());
            request.setExpires(expiresHeader);
            request.setRequestURI(requestURI);
            ContentTypeHeader contentTypeHeader = sipFactory.createHeaderFactory().createContentTypeHeader("Application", "MANSCDP+xml");
            request.setContent(cmdXml.toString(), contentTypeHeader);
            CSeqHeader cSeqHeader = (CSeqHeader)request.getHeader(CSeqHeader.NAME);
            cSeqHeader.setSeqNumber(redisCatchStorage.getCSEQ());
            request.removeHeader(CSeqHeader.NAME);
            request.addHeader(cSeqHeader);
         if (requestOld != null) {
            callIdHeader = sipFactory.createHeaderFactory().createCallIdHeader(requestOld.getCallIdHeader().getCallId());
         }else {
            CallIdHeader callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
            callIdHeader = device.getTransport().equals("TCP") ? tcpSipProvider.getNewCallId()
                  : udpSipProvider.getNewCallId();
            // 有效时间默认为60秒以上
            request = headerProvider.createSubscribeRequest(device, cmdXml.toString(), SipUtils.getNewViaTag(),
                  SipUtils.getNewFromTag(), null, device.getSubscribeCycleForCatalog(), "Catalog" ,
                  callIdHeader);
         }
         // 有效时间默认为60秒以上
         SIPRequest request = (SIPRequest)headerProvider.createSubscribeRequest(device, cmdXml.toString(), requestOld,  device.getSubscribeCycleForCatalog(), "Catalog" ,
               callIdHeader);
         transmitRequest(device, request, errorEvent, okEvent);
         return true;
         return request;
      } catch ( NumberFormatException | ParseException | InvalidArgumentException   | SipException e) {
         e.printStackTrace();
         return false;
         return null;
      }
   }
@@ -1616,13 +1623,11 @@
         clientTransaction = udpSipProvider.getNewClientTransaction(request);
      }
      if (request.getHeader(UserAgentHeader.NAME) == null) {
         UserAgentHeader userAgentHeader = null;
         try {
            userAgentHeader = SipUtils.createUserAgentHeader(sipFactory);
            request.addHeader(SipUtils.createUserAgentHeader(sipFactory, gitUtil));
         } catch (ParseException e) {
            logger.error("添加UserAgentHeader失败", e);
         }
         request.addHeader(userAgentHeader);
      }
      CallIdHeader callIdHeader = (CallIdHeader)request.getHeader(CallIdHeader.NAME);
      // 添加错误订阅
@@ -1768,7 +1773,7 @@
   private int getInfoCseq() {
      return (int) ((Math.random() * 9 + 1) * Math.pow(10, 8));
   }
   @Override
   public void playbackControlCmd(Device device, StreamInfo streamInfo, String content,SipSubscribe.Event errorEvent, SipSubscribe.Event okEvent) {
      try {
@@ -1844,62 +1849,5 @@
         throw new RuntimeException(e);
      }
      return true;
   }
   private void sendNotify(Device device, String catalogXmlContent,
                     SubscribeInfo subscribeInfo, SipSubscribe.Event errorEvent,  SipSubscribe.Event okEvent )
         throws SipException, ParseException {
      MessageFactoryImpl messageFactory = (MessageFactoryImpl) sipFactory.createMessageFactory();
      String characterSet = device.getCharset();
      // 设置编码, 防止中文乱码
      messageFactory.setDefaultContentEncodingCharset(characterSet);
      Dialog dialog  = subscribeInfo.getDialog();
      if (dialog == null || !dialog.getState().equals(DialogState.CONFIRMED)) {
         return;
      }
      SIPRequest notifyRequest = (SIPRequest)dialog.createRequest(Request.NOTIFY);
      ContentTypeHeader contentTypeHeader = sipFactory.createHeaderFactory().createContentTypeHeader("Application", "MANSCDP+xml");
      notifyRequest.setContent(catalogXmlContent, contentTypeHeader);
      SubscriptionStateHeader subscriptionState = sipFactory.createHeaderFactory()
            .createSubscriptionStateHeader(SubscriptionStateHeader.ACTIVE);
      notifyRequest.addHeader(subscriptionState);
      EventHeader event = sipFactory.createHeaderFactory().createEventHeader(subscribeInfo.getEventType());
      if (subscribeInfo.getEventId() != null) {
         event.setEventId(subscribeInfo.getEventId());
      }
      notifyRequest.addHeader(event);
      SipURI sipURI = (SipURI) notifyRequest.getRequestURI();
      if (subscribeInfo.getTransaction() != null) {
         SIPRequest request = (SIPRequest) subscribeInfo.getTransaction().getRequest();
         sipURI.setHost(request.getRemoteAddress().getHostAddress());
         sipURI.setPort(request.getRemotePort());
      }else {
         sipURI.setHost(device.getIp());
         sipURI.setPort(device.getPort());
      }
      ClientTransaction transaction = null;
      if ("TCP".equals(device.getTransport())) {
         transaction = tcpSipProvider.getNewClientTransaction(notifyRequest);
      } else if ("UDP".equals(device.getTransport())) {
         transaction = udpSipProvider.getNewClientTransaction(notifyRequest);
      }
      // 添加错误订阅
      if (errorEvent != null) {
         sipSubscribe.addErrorSubscribe(subscribeInfo.getCallId(), errorEvent);
      }
      // 添加订阅
      if (okEvent != null) {
         sipSubscribe.addOkSubscribe(subscribeInfo.getCallId(), okEvent);
      }
      if (transaction == null) {
         logger.error("平台{}的Transport错误:{}",device.getDeviceId(), device.getTransport());
         return;
      }
      dialog.sendRequest(transaction);
   }
}