io.undertow.servlet.handlers.ServletDebugPageHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of undertow-servlet
There is a newer version: 2.3.18.Final
/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2014 Red Hat, Inc., and individual contributors
 * as indicated by the @author tags.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package io.undertow.servlet.handlers;

import io.undertow.server.HttpServerExchange;
import io.undertow.servlet.spec.HttpServletRequestImpl;

import java.io.IOException;
import java.nio.charset.StandardCharsets;

/**
 * generates a servlet error page with a stack trace
 *
 * @author Stuart Douglas
 */
public class ServletDebugPageHandler {


    public static final String ERROR_CSS =
            "";


    public static void handleRequest(HttpServerExchange exchange, final ServletRequestContext servletRequestContext, final Throwable exception) throws IOException {
        HttpServletRequestImpl req = servletRequestContext.getOriginalRequest();
        StringBuilder sb = new StringBuilder();
        //todo: make this good
        sb.append("ERROR");
        sb.append(ERROR_CSS);
        sb.append("
Error processing request");
        writeLabel(sb, "Context Path", req.getContextPath());
        writeLabel(sb, "Servlet Path", req.getServletPath());
        writeLabel(sb, "Path Info", req.getPathInfo());
        writeLabel(sb, "Query String", req.getQueryString());
        sb.append("Stack Trace
");
        sb.append(escapeBodyText(exception.toString()));
        sb.append("
");
        for(StackTraceElement element : exception.getStackTrace()) {
            sb.append(escapeBodyText(element.toString()));
            sb.append("
");
        }
        sb.append("");
        servletRequestContext.getOriginalResponse().setContentType("text/html");
        servletRequestContext.getOriginalResponse().setCharacterEncoding("UTF-8");
        try {
            servletRequestContext.getOriginalResponse().getOutputStream().write(sb.toString().getBytes(StandardCharsets.UTF_8));
        } catch (IllegalStateException e) {
            servletRequestContext.getOriginalResponse().getWriter().write(sb.toString());
        }
    }

    private static void writeLabel(StringBuilder sb, String label, String value) {
        sb.append("");
        sb.append(escapeBodyText(label));
        sb.append(":
");
        sb.append(escapeBodyText(value));
        sb.append("

");

    }


    public static String escapeBodyText(final String bodyText) {
        if(bodyText == null) {
            return "null";
        }
        return bodyText.replace("&", "&").replace("<", "<").replace(">", ">");
    }
}