1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package edu.internet2.middleware.shibboleth.common.profile.provider;
18
19 import java.io.IOException;
20 import java.io.OutputStreamWriter;
21
22 import org.apache.velocity.Template;
23 import org.apache.velocity.VelocityContext;
24 import org.apache.velocity.app.VelocityEngine;
25 import org.apache.velocity.runtime.resource.util.StringResourceRepository;
26 import org.opensaml.ws.transport.InTransport;
27 import org.opensaml.ws.transport.OutTransport;
28 import org.opensaml.ws.transport.http.HttpServletRequestAdapter;
29 import org.opensaml.xml.util.DatatypeHelper;
30 import org.owasp.esapi.ESAPI;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34 import edu.internet2.middleware.shibboleth.common.profile.AbstractErrorHandler;
35 import edu.internet2.middleware.shibboleth.common.util.StringResourceLoader;
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55 public class VelocityErrorHandler extends AbstractErrorHandler {
56
57
58 private final Logger log = LoggerFactory.getLogger(VelocityErrorHandler.class);
59
60
61 private VelocityEngine velocityEngine;
62
63
64 private String templatePath;
65
66
67
68
69
70
71
72 public VelocityErrorHandler(VelocityEngine engine, String template) {
73 if (engine == null) {
74 log.error("Velocity engine may not be null");
75 throw new IllegalArgumentException("Velocity engine may not be null");
76 }
77 velocityEngine = engine;
78
79 templatePath = DatatypeHelper.safeTrimOrNullString(template);
80 if (templatePath == null) {
81 log.error("Velocity template path may not be null or empty");
82 throw new IllegalArgumentException("Velocity template path may not be null or empty");
83 }
84 }
85
86
87
88
89
90
91 public void initialize() throws IOException {
92 String templateString = DatatypeHelper.inputstreamToString(getClass().getResourceAsStream(templatePath), null);
93 StringResourceRepository repository = StringResourceLoader.getRepository();
94
95 repository.putStringResource(templatePath, templateString);
96 }
97
98
99 public void processRequest(InTransport in, OutTransport out) {
100 VelocityContext context = new VelocityContext();
101 context.put("request", ((HttpServletRequestAdapter) in).getWrappedRequest());
102 context.put("requestError", in.getAttribute(AbstractErrorHandler.ERROR_KEY));
103 context.put("encoder", ESAPI.encoder());
104
105 try {
106 OutputStreamWriter responseWriter = new OutputStreamWriter(out.getOutgoingStream());
107 Template template = velocityEngine.getTemplate(templatePath);
108 template.merge(context, responseWriter);
109 responseWriter.flush();
110 } catch (Throwable t) {
111 log.error("Unable to evaluate velocity error template", t);
112 }
113
114 return;
115 }
116 }