|
@@ -0,0 +1,82 @@
|
|
|
+/*
|
|
|
+ * Copyright (c) 2018-2025, lengleng All rights reserved.
|
|
|
+ *
|
|
|
+ * Redistribution and use in source and binary forms, with or without
|
|
|
+ * modification, are permitted provided that the following conditions are met:
|
|
|
+ *
|
|
|
+ * Redistributions of source code must retain the above copyright notice,
|
|
|
+ * this list of conditions and the following disclaimer.
|
|
|
+ * Redistributions in binary form must reproduce the above copyright
|
|
|
+ * notice, this list of conditions and the following disclaimer in the
|
|
|
+ * documentation and/or other materials provided with the distribution.
|
|
|
+ * Neither the name of the pig4cloud.com developer nor the names of its
|
|
|
+ * contributors may be used to endorse or promote products derived from
|
|
|
+ * this software without specific prior written permission.
|
|
|
+ * Author: lengleng (wangiegie@gmail.com)
|
|
|
+ */
|
|
|
+
|
|
|
+package com.pig4cloud.pigx.common.security.mobile;
|
|
|
+
|
|
|
+import org.springframework.http.HttpHeaders;
|
|
|
+import org.springframework.http.HttpStatus;
|
|
|
+import org.springframework.http.ResponseEntity;
|
|
|
+import org.springframework.security.core.AuthenticationException;
|
|
|
+import org.springframework.security.oauth2.common.OAuth2AccessToken;
|
|
|
+import org.springframework.security.oauth2.common.exceptions.OAuth2Exception;
|
|
|
+import org.springframework.security.oauth2.provider.error.AbstractOAuth2SecurityExceptionHandler;
|
|
|
+import org.springframework.security.web.AuthenticationEntryPoint;
|
|
|
+import org.springframework.util.StringUtils;
|
|
|
+
|
|
|
+import javax.servlet.ServletException;
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
+import java.io.IOException;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @author lengleng
|
|
|
+ * @date 2018/8/16
|
|
|
+ * 手机号登录异常处理
|
|
|
+ */
|
|
|
+public class MobileAuthenticationEntryPoint extends AbstractOAuth2SecurityExceptionHandler implements
|
|
|
+ AuthenticationEntryPoint {
|
|
|
+ private static final String WWW_AUTHENTICATE = "WWW-Authenticate";
|
|
|
+ private static final String SUFFIX = ",";
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException authException)
|
|
|
+ throws IOException, ServletException {
|
|
|
+ doHandle(request, response, authException);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ protected ResponseEntity<OAuth2Exception> enhanceResponse(ResponseEntity<OAuth2Exception> response, Exception exception) {
|
|
|
+ HttpHeaders headers = response.getHeaders();
|
|
|
+ String existing = null;
|
|
|
+ if (headers.containsKey(WWW_AUTHENTICATE)) {
|
|
|
+ existing = extractTypePrefix(headers.getFirst("WWW-Authenticate"));
|
|
|
+ }
|
|
|
+ StringBuilder builder = new StringBuilder();
|
|
|
+ String typeName = OAuth2AccessToken.BEARER_TYPE;
|
|
|
+ builder.append(typeName + " ");
|
|
|
+ String realmName = "oauth";
|
|
|
+ builder.append("realm=\"" + realmName + "\"");
|
|
|
+ if (existing != null) {
|
|
|
+ builder.append(", " + existing);
|
|
|
+ }
|
|
|
+ HttpHeaders update = new HttpHeaders();
|
|
|
+ update.putAll(response.getHeaders());
|
|
|
+ update.set("WWW-Authenticate", builder.toString());
|
|
|
+ return new ResponseEntity<>(response.getBody(), update, HttpStatus.EXPECTATION_FAILED);
|
|
|
+ }
|
|
|
+
|
|
|
+ private String extractTypePrefix(String header) {
|
|
|
+ String existing = header;
|
|
|
+ String[] tokens = existing.split(" +");
|
|
|
+ if (tokens.length > 1 && !tokens[0].endsWith(SUFFIX)) {
|
|
|
+ existing = StringUtils.arrayToDelimitedString(tokens, " ").substring(existing.indexOf(" ") + 1);
|
|
|
+ }
|
|
|
+ return existing;
|
|
|
+ }
|
|
|
+
|
|
|
+}
|
|
|
+
|