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