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 package info.magnolia.cms.security;
35
36 import info.magnolia.cms.filters.OncePerRequestAbstractMgnlFilter;
37 import info.magnolia.cms.security.auth.callback.HttpClientCallback;
38
39 import java.io.IOException;
40 import java.util.ArrayList;
41 import java.util.List;
42
43 import javax.servlet.FilterChain;
44 import javax.servlet.ServletException;
45 import javax.servlet.http.HttpServletRequest;
46 import javax.servlet.http.HttpServletResponse;
47 import javax.servlet.http.HttpServletResponseWrapper;
48
49
50 import static info.magnolia.cms.util.ExceptionUtil.rethrow;
51 import static info.magnolia.cms.util.ExceptionUtil.wasCausedBy;
52 import static javax.servlet.http.HttpServletResponse.SC_FORBIDDEN;
53 import static javax.servlet.http.HttpServletResponse.SC_UNAUTHORIZED;
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70 public class SecurityCallbackFilter extends OncePerRequestAbstractMgnlFilter {
71 private static final org.slf4j.Logger log = org.slf4j.LoggerFactory.getLogger(SecurityCallbackFilter.class);
72
73
74
75
76 private final List<HttpClientCallback> clientCallbacks;
77
78 public SecurityCallbackFilter() {
79 this.clientCallbacks = new ArrayList<HttpClientCallback>();
80 }
81
82 @Override
83 public void doFilter(HttpServletRequest request, HttpServletResponse originalResponse, FilterChain chain) throws IOException, ServletException {
84 final StatusSniffingResponseWrapper response = new StatusSniffingResponseWrapper(originalResponse);
85 try {
86 chain.doFilter(request, response);
87 if (needsCallback(response)) {
88 selectAndHandleCallback(request, response);
89 }
90 } catch (Throwable e) {
91
92 if (wasCausedBy(e, javax.jcr.AccessDeniedException.class)) {
93 response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
94 selectAndHandleCallback(request, response);
95 } else {
96 rethrow(e, IOException.class, ServletException.class);
97 }
98 }
99 }
100
101 protected boolean needsCallback(StatusSniffingResponseWrapper response) {
102 final int status = response.getStatus();
103 return status == SC_FORBIDDEN || status == SC_UNAUTHORIZED;
104 }
105
106 protected void selectAndHandleCallback(HttpServletRequest request, StatusSniffingResponseWrapper response) {
107 selectClientCallback(request).handle(request, response);
108 }
109
110 protected HttpClientCallback selectClientCallback(HttpServletRequest request) {
111 for (HttpClientCallback clientCallback : clientCallbacks) {
112 if (clientCallback.accepts(request)) {
113 return clientCallback;
114 }
115 }
116 throw new IllegalStateException("No configured callback accepted this request " + request.toString());
117 }
118
119
120 public void addClientCallback(HttpClientCallback clientCallback) {
121 this.clientCallbacks.add(clientCallback);
122 }
123
124
125 public List<HttpClientCallback> getClientCallbacks() {
126 return clientCallbacks;
127 }
128
129
130
131
132
133
134
135
136 public static class StatusSniffingResponseWrapper extends HttpServletResponseWrapper {
137 private int status = SC_OK;
138
139 public StatusSniffingResponseWrapper(HttpServletResponse response) {
140 super(response);
141 }
142
143 public int getStatus() {
144 return status;
145 }
146
147 @Override
148 public void reset() {
149 super.reset();
150 status = SC_OK;
151 }
152
153 @Override
154 public void setStatus(int sc) {
155 super.setStatus(sc);
156 this.status = sc;
157 }
158
159 @Override
160 public void setStatus(int sc, String sm) {
161 super.setStatus(sc, sm);
162 this.status = sc;
163 }
164
165 @Override
166 public void sendRedirect(String location) throws IOException {
167 super.sendRedirect(location);
168 this.status = SC_MOVED_TEMPORARILY;
169 }
170
171 @Override
172 public void sendError(int sc) throws IOException {
173 super.sendError(sc);
174 this.status = sc;
175 }
176
177 @Override
178 public void sendError(int sc, String msg) throws IOException {
179 super.sendError(sc, msg);
180 this.status = sc;
181 }
182 }
183 }