修复级联注册bug

64850858

2021-06-28 b24be12405ac4db1e9874367ae27c81aae172495

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

@@ -17,6 +17,7 @@
import javax.sip.message.Response;



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;

@@ -59,106 +60,116 @@
   @Override

   public void process(RequestEvent evt) {

      try {

         logger.info("收到注册请求，开始处理");

         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,

                  sipConfig.getSipPassword());

         }

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

            passwordCorrect = true;

         }



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

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

         if (authorhead == null ) {

            

            if (authorhead == null) {

               logger.info("未携带授权头 回复401");

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

            } else if (!passwordCorrect) {

               logger.info("密码错误 回复401");

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

            }

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

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

         }

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

         else if (passwordCorrect) {

            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);

         }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);

            // 添加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();

            }

            //

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

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

            String deviceId = uri.getUser();

            device = storager.queryVideoDevice(deviceId);

            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 != 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;

               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");

               }

            }

         }



         getServerTransaction(evt).sendResponse(response);

         // 注册成功

         // 保存到redis

         // 下发catelog查询目录

         if (registerFlag == 1 ) {

            logger.info("注册成功! deviceId:" + device.getDeviceId());

            // boolean exists = storager.exists(device.getDeviceId());

            device.setRegisterTimeMillis(System.currentTimeMillis());

            storager.updateDevice(device);

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



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

            publisher.onlineEventPublish(device, VideoManagerConstants.EVENT_ONLINE_REGISTER);

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

            //if (!exists) {

               handler.onRegister(device);

            //}

            handler.onRegister(device);

         } else if (registerFlag == 2) {

            logger.info("注销成功! deviceId:" + device.getDeviceId());

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

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

         }

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