init commit

yangys

2024-03-27 e48aa2ac8dea1be5db11c63edf0b912c4ad5ce65

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package com.qianwen.core.secure.interceptor;
 
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.qianwen.core.secure.constant.SecureConstant;
import com.qianwen.core.secure.props.BasicSecure;
import com.qianwen.core.secure.provider.HttpMethod;
import com.qianwen.core.secure.provider.ResponseProvider;
import com.qianwen.core.secure.utils.SecureUtil;
import com.qianwen.core.tool.jackson.JsonUtil;
import com.qianwen.core.tool.utils.WebUtil;
import org.springframework.lang.NonNull;
import org.springframework.util.AntPathMatcher;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
 
/* loaded from: blade-core-secure-9.3.0.0-SNAPSHOT.jar:org/springblade/core/secure/interceptor/BasicInterceptor.class */
public class BasicInterceptor extends HandlerInterceptorAdapter {
    private static final Logger log = LoggerFactory.getLogger(BasicInterceptor.class);
    private static final AntPathMatcher ANT_PATH_MATCHER = new AntPathMatcher();
    private final List<BasicSecure> basicSecures;
 
    public BasicInterceptor(final List<BasicSecure> basicSecures) {
        this.basicSecures = basicSecures;
    }
 
    public boolean preHandle(@NonNull HttpServletRequest request, @NonNull HttpServletResponse response, @NonNull Object handler) {
        boolean check = ((Boolean) this.basicSecures.stream().filter(basicSecure -> {
            return checkAuth(request, basicSecure);
        }).findFirst().map(authSecure -> {
            return Boolean.valueOf(checkBasic(authSecure.getUsername(), authSecure.getPassword()));
        }).orElse(Boolean.TRUE)).booleanValue();
        if (!check) {
            log.warn("授权认证失败，请求接口：{}，请求IP：{}，请求参数：{}", new Object[]{request.getRequestURI(), WebUtil.getIP(request), JsonUtil.toJson(request.getParameterMap())});
            response.setHeader(SecureConstant.BASIC_REALM_HEADER_KEY, SecureConstant.BASIC_REALM_HEADER_VALUE);
            ResponseProvider.write(response);
            return false;
        }
        return true;
    }
 
    private boolean checkAuth(HttpServletRequest request, BasicSecure basicSecure) {
        return checkMethod(request, basicSecure.getMethod()) && checkPath(request, basicSecure.getPattern());
    }
 
    private boolean checkMethod(HttpServletRequest request, HttpMethod method) {
        return method == HttpMethod.ALL || (method != null && method == HttpMethod.of(request.getMethod()));
    }
 
    private boolean checkPath(HttpServletRequest request, String pattern) {
        String servletPath = request.getServletPath();
        String pathInfo = request.getPathInfo();
        if (pathInfo != null && pathInfo.length() > 0) {
            servletPath = servletPath + pathInfo;
        }
        return ANT_PATH_MATCHER.match(pattern, servletPath);
    }
 
    private boolean checkBasic(String username, String password) {
        try {
            String[] tokens = SecureUtil.extractAndDecodeHeader();
            if (username.equals(tokens[0])) {
                if (password.equals(tokens[1])) {
                    return true;
                }
            }
            return false;
        } catch (Exception e) {
            log.warn("授权认证失败，错误信息：{}", e.getMessage());
            return false;
        }
    }
}