修复单端口下历史回放暂停和恢复发送的流ID与zlm不一致

fine

2024-01-09 db493e262566c2ed2ac046e6f610b2da6cf61682

 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/event/request/SIPRequestProcessorParent.java

old mode 100644
new mode 100755
@@ -1,9 +1,12 @@
package com.genersoft.iot.vmp.gb28181.transmit.event.request;

import gov.nist.javax.sip.SipProviderImpl;
import gov.nist.javax.sip.SipStackImpl;
import com.genersoft.iot.vmp.gb28181.bean.ParentPlatform;
import com.genersoft.iot.vmp.gb28181.transmit.SIPSender;
import com.genersoft.iot.vmp.gb28181.utils.SipUtils;
import com.google.common.primitives.Bytes;
import gov.nist.javax.sip.message.SIPRequest;
import gov.nist.javax.sip.stack.SIPServerTransaction;
import gov.nist.javax.sip.message.SIPResponse;
import org.apache.commons.lang3.ArrayUtils;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
@@ -11,20 +14,21 @@
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;

import javax.sip.*;
import javax.sip.address.Address;
import javax.sip.address.AddressFactory;
import javax.sip.address.SipURI;
import javax.sip.header.ContentTypeHeader;
import javax.sip.header.ExpiresHeader;
import javax.sip.header.HeaderFactory;
import javax.sip.header.ViaHeader;
import javax.sip.message.MessageFactory;
import javax.sip.message.Request;
import javax.sip.message.Response;
import java.io.ByteArrayInputStream;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**    
 * @description:处理接收IPCamera发来的SIP协议请求消息
@@ -36,67 +40,13 @@
   private final static Logger logger = LoggerFactory.getLogger(SIPRequestProcessorParent.class);

   @Autowired
   @Qualifier(value="tcpSipProvider")
   private SipProviderImpl tcpSipProvider;

   @Autowired
   @Qualifier(value="udpSipProvider")
   private SipProviderImpl udpSipProvider;

   /**
    * 根据 RequestEvent 获取 ServerTransaction
    * @param evt
    * @return
    */
   public ServerTransaction getServerTransaction(RequestEvent evt) {
      Request request = evt.getRequest();
      ServerTransaction serverTransaction = evt.getServerTransaction();
      // 判断TCP还是UDP
      boolean isTcp = false;
      ViaHeader reqViaHeader = (ViaHeader) request.getHeader(ViaHeader.NAME);
      String transport = reqViaHeader.getTransport();
      if (transport.equals("TCP")) {
         isTcp = true;
      }

      if (serverTransaction == null) {
         try {
            if (isTcp) {
               SipStackImpl stack = (SipStackImpl)tcpSipProvider.getSipStack();
               serverTransaction = (SIPServerTransaction) stack.findTransaction((SIPRequest)request, true);
               if (serverTransaction == null) {
                  serverTransaction = tcpSipProvider.getNewServerTransaction(request);
               }
            } else {
               SipStackImpl stack = (SipStackImpl)udpSipProvider.getSipStack();
               serverTransaction = (SIPServerTransaction) stack.findTransaction((SIPRequest)request, true);
               if (serverTransaction == null) {
                  serverTransaction = udpSipProvider.getNewServerTransaction(request);
               }
            }
         } catch (TransactionAlreadyExistsException e) {
            logger.error(e.getMessage());
         } catch (TransactionUnavailableException e) {
            logger.error(e.getMessage());
         }
      }
      return serverTransaction;
   }
	
   public AddressFactory getAddressFactory() {
      try {
         return SipFactory.getInstance().createAddressFactory();
      } catch (PeerUnavailableException e) {
         e.printStackTrace();
      }
      return null;
   }
   private SIPSender sipSender;

   public HeaderFactory getHeaderFactory() {
      try {
         return SipFactory.getInstance().createHeaderFactory();
      } catch (PeerUnavailableException e) {
         e.printStackTrace();
         logger.error("未处理的异常 ", e);
      }
      return null;
   }
@@ -105,9 +55,16 @@
      try {
         return SipFactory.getInstance().createMessageFactory();
      } catch (PeerUnavailableException e) {
         e.printStackTrace();
         logger.error("未处理的异常 ", e);
      }
      return null;
   }

   class ResponseAckExtraParam{
      String content;
      ContentTypeHeader contentTypeHeader;
      SipURI sipURI;
      int expires = -1;
   }

   /***
@@ -116,62 +73,150 @@
    * 200 OK
    * 400
    * 404
    * @param evt
    * @throws SipException
    * @throws InvalidArgumentException
    * @throws ParseException
    */
   public void responseAck(RequestEvent evt, int statusCode) throws SipException, InvalidArgumentException, ParseException {
      Response response = getMessageFactory().createResponse(statusCode, evt.getRequest());
      ServerTransaction serverTransaction = getServerTransaction(evt);
      serverTransaction.sendResponse(response);
      if (statusCode >= 200) {
         if (serverTransaction.getDialog() != null) serverTransaction.getDialog().delete();
      }
   public SIPResponse responseAck(SIPRequest sipRequest, int statusCode) throws SipException, InvalidArgumentException, ParseException {
      return responseAck(sipRequest, statusCode, null);
   }

   public void responseAck(RequestEvent evt, int statusCode, String msg) throws SipException, InvalidArgumentException, ParseException {
      Response response = getMessageFactory().createResponse(statusCode, evt.getRequest());
      response.setReasonPhrase(msg);
      ServerTransaction serverTransaction = getServerTransaction(evt);
      serverTransaction.sendResponse(response);
      if (statusCode >= 200) {
         if (serverTransaction.getDialog() != null) serverTransaction.getDialog().delete();
   public SIPResponse responseAck(SIPRequest sipRequest, int statusCode, String msg) throws SipException, InvalidArgumentException, ParseException {
      return responseAck(sipRequest, statusCode, msg, null);
   }

   public SIPResponse responseAck(SIPRequest sipRequest, int statusCode, String msg, ResponseAckExtraParam responseAckExtraParam) throws SipException, InvalidArgumentException, ParseException {
      if (sipRequest.getToHeader().getTag() == null) {
         sipRequest.getToHeader().setTag(SipUtils.getNewTag());
      }
      SIPResponse response = (SIPResponse)getMessageFactory().createResponse(statusCode, sipRequest);
      response.setStatusCode(statusCode);
      if (msg != null) {
         response.setReasonPhrase(msg);
      }

      if (responseAckExtraParam != null) {
         if (responseAckExtraParam.sipURI != null && sipRequest.getMethod().equals(Request.INVITE)) {
            logger.debug("responseSdpAck SipURI: {}:{}", responseAckExtraParam.sipURI.getHost(), responseAckExtraParam.sipURI.getPort());
            Address concatAddress = SipFactory.getInstance().createAddressFactory().createAddress(
                  SipFactory.getInstance().createAddressFactory().createSipURI(responseAckExtraParam.sipURI.getUser(),  responseAckExtraParam.sipURI.getHost()+":"+responseAckExtraParam.sipURI.getPort()
                  ));
            response.addHeader(SipFactory.getInstance().createHeaderFactory().createContactHeader(concatAddress));
         }
         if (responseAckExtraParam.contentTypeHeader != null) {
            response.setContent(responseAckExtraParam.content, responseAckExtraParam.contentTypeHeader);
         }

         if (sipRequest.getMethod().equals(Request.SUBSCRIBE)) {
            if (responseAckExtraParam.expires == -1) {
               logger.error("[参数不全] 2xx的SUBSCRIBE回复，必须设置Expires header");
            }else {
               ExpiresHeader expiresHeader = SipFactory.getInstance().createHeaderFactory().createExpiresHeader(responseAckExtraParam.expires);
               response.addHeader(expiresHeader);
            }
         }
      }else {
         if (sipRequest.getMethod().equals(Request.SUBSCRIBE)) {
            logger.error("[参数不全] 2xx的SUBSCRIBE回复，必须设置Expires header");
         }
      }

      // 发送response
      sipSender.transmitRequest(sipRequest.getLocalAddress().getHostAddress(), response);

      return response;
   }

   /**
    * 回复带sdp的200
    * @param evt
    * @param sdp
    * @throws SipException
    * @throws InvalidArgumentException
    * @throws ParseException
    */
   public void responseAck(RequestEvent evt, String sdp) throws SipException, InvalidArgumentException, ParseException {
      Response response = getMessageFactory().createResponse(Response.OK, evt.getRequest());
      SipFactory sipFactory = SipFactory.getInstance();
      ContentTypeHeader contentTypeHeader = sipFactory.createHeaderFactory().createContentTypeHeader("APPLICATION", "SDP");
      response.setContent(sdp, contentTypeHeader);
   public SIPResponse responseSdpAck(SIPRequest request, String sdp, ParentPlatform platform) throws SipException, InvalidArgumentException, ParseException {

      SipURI sipURI = (SipURI)evt.getRequest().getRequestURI();
      ContentTypeHeader contentTypeHeader = SipFactory.getInstance().createHeaderFactory().createContentTypeHeader("APPLICATION", "SDP");

      Address concatAddress = sipFactory.createAddressFactory().createAddress(
            sipFactory.createAddressFactory().createSipURI(sipURI.getUser(),  sipURI.getHost()+":"+sipURI.getPort()
            ));
      response.addHeader(sipFactory.createHeaderFactory().createContactHeader(concatAddress));
      getServerTransaction(evt).sendResponse(response);
      // 兼容国标中的使用编码@域名作为RequestURI的情况
      SipURI sipURI = (SipURI)request.getRequestURI();
      if (sipURI.getPort() == -1) {
         sipURI = SipFactory.getInstance().createAddressFactory().createSipURI(platform.getServerGBId(),  platform.getServerIP()+":"+platform.getServerPort());
      }
      ResponseAckExtraParam responseAckExtraParam = new ResponseAckExtraParam();
      responseAckExtraParam.contentTypeHeader = contentTypeHeader;
      responseAckExtraParam.content = sdp;
      responseAckExtraParam.sipURI = sipURI;

      return responseAck(request, Response.OK, null, responseAckExtraParam);
   }

   /**
    * 回复带xml的200
    */
   public SIPResponse responseXmlAck(SIPRequest request, String xml, ParentPlatform platform, Integer expires) throws SipException, InvalidArgumentException, ParseException {
      ContentTypeHeader contentTypeHeader = SipFactory.getInstance().createHeaderFactory().createContentTypeHeader("Application", "MANSCDP+xml");

      SipURI sipURI = (SipURI)request.getRequestURI();
      if (sipURI.getPort() == -1) {
         sipURI = SipFactory.getInstance().createAddressFactory().createSipURI(platform.getServerGBId(),  platform.getServerIP()+":"+platform.getServerPort());
      }
      ResponseAckExtraParam responseAckExtraParam = new ResponseAckExtraParam();
      responseAckExtraParam.contentTypeHeader = contentTypeHeader;
      responseAckExtraParam.content = xml;
      responseAckExtraParam.sipURI = sipURI;
      responseAckExtraParam.expires = expires;
      return responseAck(request, Response.OK, null, responseAckExtraParam);
   }

   public Element getRootElement(RequestEvent evt) throws DocumentException {
      return getRootElement(evt, "gb2312");
   }
   public Element getRootElement(RequestEvent evt, String charset) throws DocumentException {
      if (charset == null) charset = "gb2312";
      if (charset == null) {
         charset = "gb2312";
      }
      Request request = evt.getRequest();
      SAXReader reader = new SAXReader();
      reader.setEncoding(charset);
      Document xml = reader.read(new ByteArrayInputStream(request.getRawContent()));
      // 对海康出现的未转义字符做处理。
      String[] destStrArray = new String[]{"<",">","&","'","""};
      char despChar = '&'; // 或许可扩展兼容其他字符
      byte destBye = (byte) despChar;
      List<Byte> result = new ArrayList<>();
      byte[] rawContent = request.getRawContent();
      if (rawContent == null) {
         return null;
      }
      for (int i = 0; i < rawContent.length; i++) {
         if (rawContent[i] == destBye) {
            boolean resul = false;
            for (String destStr : destStrArray) {
               if (i + destStr.length() <= rawContent.length) {
                  byte[] bytes = Arrays.copyOfRange(rawContent, i, i + destStr.length());
                  resul = resul || (Arrays.equals(bytes,destStr.getBytes()));
               }
            }
            if (resul) {
               result.add(rawContent[i]);
            }
         }else {
            result.add(rawContent[i]);
         }
      }
      byte[] bytesResult = Bytes.toArray(result);

      Document xml;
      try {
         xml = reader.read(new ByteArrayInputStream(bytesResult));
      }catch (DocumentException e) {
         logger.warn("[xml解析异常]： 原文如下： \r\n{}", new String(bytesResult));
         logger.warn("[xml解析异常]： 原文如下： 尝试兼容性处理");
         String[] xmlLineArray = new String(bytesResult).split("\\r?\\n");

         // 兼容海康的address字段带有<破换xml结构导致无法解析xml的问题
         StringBuilder stringBuilder = new StringBuilder();
         for (String s : xmlLineArray) {
            if (s.startsWith("<Address")) {
               continue;
            }
            stringBuilder.append(s);
         }
         xml = reader.read(new ByteArrayInputStream(stringBuilder.toString().getBytes()));
      }
      return xml.getRootElement();
   }