1

zj

2024-10-21 a54e43c6d57c82660d46b24b0720175314960d78

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
package security.filter;
 
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
 
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
 
import org.apache.commons.io.IOUtils;
import org.springframework.security.Authentication;
import org.springframework.security.AuthenticationException;
import org.springframework.security.ui.webapp.AuthenticationProcessingFilter;
import org.springframework.security.util.RedirectUtils;
 
import com.alibaba.fastjson.JSON;
 
import kernel.web.ResultObject;
 
public class AjaxableAuthenticationProcessingFilter extends AuthenticationProcessingFilter {
 
    /**
     * If true, causes any redirection URLs to be calculated minus the protocol and
     * context path (defaults to false).
     */
    private boolean useRelativeContext = false;
 
    public void setUseRelativeContext(boolean useRelativeContext) {
        this.useRelativeContext = useRelativeContext;
    }
 
    protected void onSuccessfulAuthentication(HttpServletRequest request, HttpServletResponse response,
            Authentication authResult) throws IOException {
        super.onSuccessfulAuthentication(request, response, authResult);
        if ("XMLHttpRequest".equals(request.getHeader("X-Requested-With"))) {
            ResultObject resultObject = new ResultObject();
            sendResponse(response, JSON.toJSONString(resultObject));
        }
    }
 
    private void sendResponse(HttpServletResponse response, String jsonStr)
            throws UnsupportedEncodingException, IOException {
        response.setContentType("application/json");
        response.setHeader("Access-Control-Allow-Origin", "*");
        ByteArrayInputStream input = new ByteArrayInputStream(jsonStr.getBytes("UTF-8"));
        ServletOutputStream output = response.getOutputStream();
        IOUtils.copy(input, output);
        IOUtils.closeQuietly(input);
    }
 
    protected void onUnsuccessfulAuthentication(HttpServletRequest request, HttpServletResponse response,
            AuthenticationException failed) throws IOException {
        super.onUnsuccessfulAuthentication(request, response, failed);
        if ("XMLHttpRequest".equals(request.getHeader("X-Requested-With"))) {
            ResultObject resultObject = new ResultObject();
            resultObject.setCode("1");
            resultObject.setMsg(failed.getMessage());
            sendResponse(response, JSON.toJSONString(resultObject));
        }
    }
 
    protected void sendRedirect(HttpServletRequest request, HttpServletResponse response, String url)
            throws IOException {
        // ignore redirect when request via ajax
        if (!"XMLHttpRequest".equals(request.getHeader("X-Requested-With"))) {
            RedirectUtils.sendRedirect(request, response, url, useRelativeContext);
        }
    }
}