修复级联注册bug

64850858

2021-06-28 b24be12405ac4db1e9874367ae27c81aae172495

 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java

@@ -7,7 +7,6 @@


import javax.sip.InvalidArgumentException;

import javax.sip.RequestEvent;

import javax.sip.ServerTransaction;

import javax.sip.SipException;

import javax.sip.header.AuthorizationHeader;

import javax.sip.header.ContactHeader;

@@ -17,19 +16,20 @@
import javax.sip.message.Request;

import javax.sip.message.Response;



import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import com.genersoft.iot.vmp.gb28181.bean.WvpSipDate;

import gov.nist.javax.sip.RequestEventExt;

import gov.nist.javax.sip.header.SIPDateHeader;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.util.StringUtils;



import com.genersoft.iot.vmp.common.VideoManagerConstants;

import com.genersoft.iot.vmp.conf.SipConfig;

import com.genersoft.iot.vmp.gb28181.SipLayer;

import com.genersoft.iot.vmp.gb28181.auth.DigestServerAuthenticationHelper;

import com.genersoft.iot.vmp.gb28181.auth.RegisterLogicHandler;

import com.genersoft.iot.vmp.gb28181.bean.Device;

import com.genersoft.iot.vmp.gb28181.bean.Host;

import com.genersoft.iot.vmp.gb28181.event.EventPublisher;

import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;

import com.genersoft.iot.vmp.gb28181.transmit.request.ISIPRequestProcessor;

import com.genersoft.iot.vmp.gb28181.transmit.request.SIPRequestAbstractProcessor;

import com.genersoft.iot.vmp.storager.IVideoManagerStorager;



import gov.nist.javax.sip.address.AddressImpl;

@@ -38,120 +38,138 @@


/**    

 * @Description:收到注册请求 处理 

 * @author: songww

 * @author: swwheihei

 * @date:   2020年5月3日 下午4:47:25     

 */

@Component

public class RegisterRequestProcessor implements ISIPRequestProcessor {

public class RegisterRequestProcessor extends SIPRequestAbstractProcessor {



   @Autowired

   private SipConfig config;

   private Logger logger = LoggerFactory.getLogger(RegisterRequestProcessor.class);



   private SipConfig sipConfig;

   

   @Autowired

   private SIPCommander cmder;

   private RegisterLogicHandler handler;

   

   @Autowired

   private IVideoManagerStorager storager;

   

   @Autowired

   private EventPublisher publisher;

	
   /***



   /**

    * 收到注册请求 处理

    * 
    * @param request

    *            请求消息

    */ 
     * @param evt

    */

   @Override

   public void process(RequestEvent evt, SipLayer layer, ServerTransaction transaction) {

   public void process(RequestEvent evt) {

      try {

         System.out.println("收到注册请求，开始处理");

         RequestEventExt evtExt = (RequestEventExt)evt;

         String requestAddress = evtExt.getRemoteIpAddress() + ":" + evtExt.getRemotePort();

         logger.info("[{}] 收到注册请求，开始处理", requestAddress);

         Request request = evt.getRequest();



         Response response = null; 

         boolean passwordCorrect = false;

         // 注册标志  0：未携带授权头或者密码错误  1：注册成功   2：注销成功

         int registerFlag = 0;

         Device device = null;

         FromHeader fromHeader = (FromHeader) request.getHeader(FromHeader.NAME);

         AddressImpl address = (AddressImpl) fromHeader.getAddress();

         SipUri uri = (SipUri) address.getURI();

         String deviceId = uri.getUser();

         Device device = storager.queryVideoDevice(deviceId);

         AuthorizationHeader authorhead = (AuthorizationHeader) request.getHeader(AuthorizationHeader.NAME); 

         // 校验密码是否正确

         if (authorhead != null) {

            passwordCorrect = new DigestServerAuthenticationHelper().doAuthenticatePlainTextPassword(request,

                  config.getSipPassword());

                  sipConfig.getSipPassword());

         }

         if (StringUtils.isEmpty(sipConfig.getSipPassword())){

            passwordCorrect = true;

         }



         // 未携带授权头或者密码错误 回复401

         if (authorhead == null || !passwordCorrect) {

         if (authorhead == null ) {

            

            if (authorhead == null) {

               System.out.println("未携带授权头 回复401");

               logger.info("[{}] 未携带授权头 回复401", requestAddress);

            } else if (!passwordCorrect) {

               System.out.println("密码错误 回复401");

               logger.info("[{}] 密码错误 回复401", requestAddress);

            }

            response = layer.getMessageFactory().createResponse(Response.UNAUTHORIZED, request);

            new DigestServerAuthenticationHelper().generateChallenge(layer.getHeaderFactory(), response, config.getSipDomain());

         }

         // 携带授权头并且密码正确

         else if (passwordCorrect) {

            response = layer.getMessageFactory().createResponse(Response.OK, request);

            // 添加date头

            response.addHeader(layer.getHeaderFactory().createDateHeader(Calendar.getInstance(Locale.ENGLISH)));

            ExpiresHeader expiresHeader = (ExpiresHeader) request.getHeader(Expires.NAME);

            // 添加Contact头

            response.addHeader(request.getHeader(ContactHeader.NAME));

            // 添加Expires头

            response.addHeader(request.getExpires());

				
            // 1.获取到通信地址等信息，保存到Redis

            FromHeader fromHeader = (FromHeader) request.getHeader(FromHeader.NAME);

            ViaHeader viaHeader = (ViaHeader) request.getHeader(ViaHeader.NAME);

            String received = viaHeader.getReceived();

            int rPort = viaHeader.getRPort();

            // 本地模拟设备 received 为空 rPort 为 -1

            // 解析本地地址替代

            if (StringUtils.isEmpty(received) || rPort == -1) {

               received = viaHeader.getHost();

               rPort = viaHeader.getPort();

            }

            //

            Host host = new Host();

            host.setIp(received);

            host.setPort(rPort);

            host.setAddress(received.concat(":").concat(String.valueOf(rPort)));

            AddressImpl address = (AddressImpl) fromHeader.getAddress();

            SipUri uri = (SipUri) address.getURI();

            String deviceId = uri.getUser();

            device = new Device();

            device.setDeviceId(deviceId);

            device.setHost(host);

            // 注销成功

            if (expiresHeader != null && expiresHeader.getExpires() == 0) {

               registerFlag = 2;

            }

            // 注册成功

            else {

               registerFlag = 1;

               // 判断TCP还是UDP

               boolean isTcp = false;

               ViaHeader reqViaHeader = (ViaHeader) request.getHeader(ViaHeader.NAME);

               String transport = reqViaHeader.getTransport();

               if (transport.equals("TCP")) {

                  isTcp = true;

            response = getMessageFactory().createResponse(Response.UNAUTHORIZED, request);

            new DigestServerAuthenticationHelper().generateChallenge(getHeaderFactory(), response, sipConfig.getSipDomain());

         }else {

            if (!passwordCorrect){

               // 注册失败

               response = getMessageFactory().createResponse(Response.FORBIDDEN, request);

               response.setReasonPhrase("wrong password");

            }else {

               // 携带授权头并且密码正确

               response = getMessageFactory().createResponse(Response.OK, request);

               // 添加date头

               SIPDateHeader dateHeader = new SIPDateHeader();

               // 使用自己修改的

               WvpSipDate wvpSipDate = new WvpSipDate(Calendar.getInstance(Locale.ENGLISH).getTimeInMillis());

               dateHeader.setDate(wvpSipDate);

               response.addHeader(dateHeader);



               ExpiresHeader expiresHeader = (ExpiresHeader) request.getHeader(Expires.NAME);

               if (expiresHeader == null) {

                  response = getMessageFactory().createResponse(Response.BAD_REQUEST, request);

                  getServerTransaction(evt).sendResponse(response);

                  return;

               }

               device.setTransport(isTcp ? "TCP" : "UDP");

               // 添加Contact头

               response.addHeader(request.getHeader(ContactHeader.NAME));

               // 添加Expires头

               response.addHeader(request.getExpires());



               // 获取到通信地址等信息

               ViaHeader viaHeader = (ViaHeader) request.getHeader(ViaHeader.NAME);

               String received = viaHeader.getReceived();

               int rPort = viaHeader.getRPort();

               // 解析本地地址替代

               if (StringUtils.isEmpty(received) || rPort == -1) {

                  received = viaHeader.getHost();

                  rPort = viaHeader.getPort();

               }

               //



               if (device == null) {

                  device = new Device();

                  device.setStreamMode("UDP");

                  device.setDeviceId(deviceId);

               }

               device.setIp(received);

               device.setPort(rPort);

               device.setHostAddress(received.concat(":").concat(String.valueOf(rPort)));

               // 注销成功

               if (expiresHeader.getExpires() == 0) {

                  registerFlag = 2;

               }

               // 注册成功

               else {

                  device.setExpires(expiresHeader.getExpires());

                  registerFlag = 1;

                  // 判断TCP还是UDP

                  boolean isTcp = false;

                  ViaHeader reqViaHeader = (ViaHeader) request.getHeader(ViaHeader.NAME);

                  String transport = reqViaHeader.getTransport();

                  if (transport.equals("TCP")) {

                     isTcp = true;

                  }

                  device.setTransport(isTcp ? "TCP" : "UDP");

               }

            }

         }

         transaction.sendResponse(response);



         getServerTransaction(evt).sendResponse(response);

         // 注册成功

         // 保存到redis

         // 下发catelog查询目录

         if (registerFlag == 1 && device != null) {

            System.out.println("注册成功! deviceId:" + device.getDeviceId());

            storager.update(device);

            publisher.onlineEventPublish(device.getDeviceId(), VideoManagerConstants.EVENT_ONLINE_REGISTER);

            cmder.deviceInfoQuery(device);

         if (registerFlag == 1 ) {

            logger.info("[{}] 注册成功! deviceId:" + device.getDeviceId(), requestAddress);

            publisher.onlineEventPublish(device, VideoManagerConstants.EVENT_ONLINE_REGISTER);

            // 重新注册更新设备和通道，以免设备替换或更新后信息无法更新

            handler.onRegister(device);

         } else if (registerFlag == 2) {

            System.out.println("注销成功! deviceId:" + device.getDeviceId());

            logger.info("[{}] 注销成功! deviceId:" + device.getDeviceId(), requestAddress);

            publisher.outlineEventPublish(device.getDeviceId(), VideoManagerConstants.EVENT_OUTLINE_UNREGISTER);

         }

      } catch (SipException | InvalidArgumentException | NoSuchAlgorithmException | ParseException e) {

@@ -159,5 +177,21 @@
      }

      

   }

	
   public void setSipConfig(SipConfig sipConfig) {

      this.sipConfig = sipConfig;

   }



   public void setHandler(RegisterLogicHandler handler) {

      this.handler = handler;

   }



   public void setVideoManagerStorager(IVideoManagerStorager storager) {

      this.storager = storager;

   }



   public void setPublisher(EventPublisher publisher) {

      this.publisher = publisher;

   }



}