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