From abd569d79a346d9066c4b239dbc452861b26a6cb Mon Sep 17 00:00:00 2001
From: 64850858 <648540858@qq.com>
Date: 星期二, 08 六月 2021 14:43:03 +0800
Subject: [PATCH] 添加注册失败时回复403避免陷入401循环

---
 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java |  118 +++++++++++++++++++++++++++++++----------------------------
 1 files changed, 62 insertions(+), 56 deletions(-)

diff --git a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java
index 64be7ef..37cf1cd 100644
--- a/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java
+++ b/src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java
@@ -89,7 +89,7 @@
 			}
 
 			// 鏈惡甯︽巿鏉冨ご鎴栬�呭瘑鐮侀敊璇� 鍥炲401
-			if (authorhead == null || !passwordCorrect) {
+			if (authorhead == null ) {
 				
 				if (authorhead == null) {
 					logger.info("[{}] 鏈惡甯︽巿鏉冨ご 鍥炲401", requestAddress);
@@ -98,65 +98,71 @@
 				}
 				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);
-				if (expiresHeader == null) {
-					response = getMessageFactory().createResponse(Response.BAD_REQUEST, request);
-					getServerTransaction(evt).sendResponse(response);
-					return;
-				}
-				// 娣诲姞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();
-				// 瑙f瀽鏈湴鍦板潃鏇夸唬
-				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;
+					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();
+					// 瑙f瀽鏈湴鍦板潃鏇夸唬
+					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);
 			// 娉ㄥ唽鎴愬姛
 			// 淇濆瓨鍒皉edis

--
Gitblit v1.8.0