Search in sources :

Example 1 with ServletResponseWrapper

use of javax.servlet.ServletResponseWrapper in project jetty.project by eclipse.

the class Dump method doGet.

/* ------------------------------------------------------------ */
@Override
public void doGet(final HttpServletRequest request, final HttpServletResponse response) throws ServletException, IOException {
    if (request.getRemoteUser() == null) {
        try {
            request.login("user", "password");
        } catch (ServletException se) {
            getServletContext().log(se.toString());
        }
    }
    // Handle a dump of data
    final String data = request.getParameter("data");
    final String chars = request.getParameter("chars");
    final String block = request.getParameter("block");
    final String dribble = request.getParameter("dribble");
    final boolean flush = request.getParameter("flush") != null ? Boolean.parseBoolean(request.getParameter("flush")) : false;
    if (request.getPathInfo() != null && request.getPathInfo().toLowerCase(Locale.ENGLISH).indexOf("script") != -1) {
        response.sendRedirect(response.encodeRedirectURL(getServletContext().getContextPath() + "/dump/info"));
        return;
    }
    request.setCharacterEncoding("UTF-8");
    if (request.getParameter("busy") != null) {
        long end = System.currentTimeMillis() + Long.parseLong(request.getParameter("busy"));
        while (System.currentTimeMillis() < end) {
        }
    }
    if (request.getParameter("empty") != null) {
        response.setStatus(200);
        response.flushBuffer();
        return;
    }
    if (request.getParameter("sleep") != null) {
        try {
            long s = Long.parseLong(request.getParameter("sleep"));
            if (request.getHeader("Expect") != null && request.getHeader("Expect").indexOf("102") >= 0) {
                Thread.sleep(s / 2);
                response.sendError(102);
                Thread.sleep(s / 2);
            } else
                Thread.sleep(s);
        } catch (InterruptedException e) {
            return;
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }
    if (request.getParameter("startAsync") != null && request.getAttribute("ASYNC") != Boolean.TRUE) {
        request.setAttribute("ASYNC", Boolean.TRUE);
        try {
            final AsyncContext async = request.startAsync(request, response);
            async.setTimeout(Long.parseLong(request.getParameter("startAsync")));
            async.addListener(new AsyncListener() {

                @Override
                public void onTimeout(AsyncEvent event) throws IOException {
                    response.addHeader("Dump", "onTimeout");
                    try {
                        if (!dump(response, data, chars, block, dribble, flush)) {
                            response.setContentType("text/plain");
                            response.getOutputStream().println("EXPIRED");
                        }
                        async.complete();
                    } catch (IOException e) {
                        getServletContext().log("", e);
                    }
                }

                @Override
                public void onStartAsync(AsyncEvent event) throws IOException {
                    response.addHeader("Dump", "onStartAsync");
                }

                @Override
                public void onError(AsyncEvent event) throws IOException {
                    response.addHeader("Dump", "onError");
                }

                @Override
                public void onComplete(AsyncEvent event) throws IOException {
                    response.addHeader("Dump", "onComplete");
                }
            });
            if (request.getParameter("dispatch") != null) {
                request.setAttribute("RESUME", Boolean.TRUE);
                final long resume = Long.parseLong(request.getParameter("dispatch"));
                _timer.schedule(new TimerTask() {

                    @Override
                    public void run() {
                        async.dispatch();
                    }
                }, resume);
            }
            if (request.getParameter("complete") != null) {
                final long complete = Long.parseLong(request.getParameter("complete"));
                _timer.schedule(new TimerTask() {

                    @Override
                    public void run() {
                        try {
                            response.setContentType("text/html");
                            response.getOutputStream().println("<h1>COMPLETED</h1>");
                            async.complete();
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
                }, complete);
            }
            return;
        } catch (Exception e) {
            throw new ServletException(e);
        }
    }
    request.setAttribute("Dump", this);
    getServletContext().setAttribute("Dump", this);
    // getServletContext().log("dump "+request.getRequestURI());
    // Force a content length response
    String length = request.getParameter("length");
    if (length != null && length.length() > 0) {
        response.setContentLength(Integer.parseInt(length));
    }
    // Handle a dump of data
    if (dump(response, data, chars, block, dribble, flush))
        return;
    // handle an exception
    String info = request.getPathInfo();
    if (info != null && info.endsWith("Exception")) {
        try {
            throw (Throwable) Thread.currentThread().getContextClassLoader().loadClass(info.substring(1)).newInstance();
        } catch (Throwable th) {
            throw new ServletException(th);
        }
    }
    // test a reset
    String reset = request.getParameter("reset");
    if (reset != null && reset.length() > 0) {
        response.getOutputStream().println("THIS SHOULD NOT BE SEEN!");
        response.setHeader("SHOULD_NOT", "BE SEEN");
        response.reset();
    }
    // handle an redirect
    String redirect = request.getParameter("redirect");
    if (redirect != null && redirect.length() > 0) {
        response.getOutputStream().println("THIS SHOULD NOT BE SEEN!");
        response.sendRedirect(response.encodeRedirectURL(redirect));
        try {
            response.getOutputStream().println("THIS SHOULD NOT BE SEEN!");
        } catch (IOException e) {
        // ignored as stream is closed.
        }
        return;
    }
    // handle an error
    String error = request.getParameter("error");
    if (error != null && error.length() > 0 && request.getAttribute(RequestDispatcher.ERROR_STATUS_CODE) == null) {
        response.getOutputStream().println("THIS SHOULD NOT BE SEEN!");
        response.sendError(Integer.parseInt(error));
        try {
            response.getOutputStream().println("THIS SHOULD NOT BE SEEN!");
        } catch (IllegalStateException e) {
            try {
                response.getWriter().println("NOR THIS!!");
            } catch (IOException e2) {
            }
        } catch (IOException e) {
        }
        return;
    }
    // Handle a extra headers
    String headers = request.getParameter("headers");
    if (headers != null && headers.length() > 0) {
        long h = Long.parseLong(headers);
        for (int i = 0; i < h; i++) response.addHeader("Header" + i, "Value" + i);
    }
    String buffer = request.getParameter("buffer");
    if (buffer != null && buffer.length() > 0)
        response.setBufferSize(Integer.parseInt(buffer));
    String charset = request.getParameter("charset");
    if (charset == null)
        charset = "UTF-8";
    response.setCharacterEncoding(charset);
    response.setContentType("text/html");
    if (info != null && info.indexOf("Locale/") >= 0) {
        try {
            String locale_name = info.substring(info.indexOf("Locale/") + 7);
            Field f = java.util.Locale.class.getField(locale_name);
            response.setLocale((Locale) f.get(null));
        } catch (Exception e) {
            e.printStackTrace();
            response.setLocale(Locale.getDefault());
        }
    }
    String cn = request.getParameter("cookie");
    String cv = request.getParameter("cookiev");
    if (cn != null && cv != null) {
        Cookie cookie = new Cookie(cn, cv);
        if (request.getParameter("version") != null)
            cookie.setVersion(Integer.parseInt(request.getParameter("version")));
        cookie.setComment("Cookie from dump servlet");
        response.addCookie(cookie);
    }
    String pi = request.getPathInfo();
    if (pi != null && pi.startsWith("/ex")) {
        OutputStream out = response.getOutputStream();
        out.write("</H1>This text should be reset</H1>".getBytes());
        if ("/ex0".equals(pi))
            throw new ServletException("test ex0", new Throwable());
        else if ("/ex1".equals(pi))
            throw new IOException("test ex1");
        else if ("/ex2".equals(pi))
            throw new UnavailableException("test ex2");
        else if (pi.startsWith("/ex3/"))
            throw new UnavailableException("test ex3", Integer.parseInt(pi.substring(5)));
        throw new RuntimeException("test");
    }
    if ("true".equals(request.getParameter("close")))
        response.setHeader("Connection", "close");
    String buffered = request.getParameter("buffered");
    PrintWriter pout = null;
    try {
        pout = response.getWriter();
    } catch (IllegalStateException e) {
        pout = new PrintWriter(new OutputStreamWriter(response.getOutputStream(), charset));
    }
    if (buffered != null)
        pout = new PrintWriter(new BufferedWriter(pout, Integer.parseInt(buffered)));
    try {
        pout.write("<html>\n<body>\n");
        pout.write("<h1>Dump Servlet</h1>\n");
        pout.write("<table width=\"95%\">");
        pout.write("<tr>\n");
        pout.write("<th align=\"right\">getContentLength:&nbsp;</th>");
        pout.write("<td>" + Integer.toString(request.getContentLength()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getContentType:&nbsp;</th>");
        pout.write("<td>" + notag(request.getContentType()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getContextPath:&nbsp;</th>");
        pout.write("<td>" + request.getContextPath() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getDispatcherType:&nbsp;</th>");
        pout.write("<td>" + request.getDispatcherType() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getLocale:&nbsp;</th>");
        pout.write("<td>" + request.getLocale() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getLocalName:&nbsp;</th>");
        pout.write("<td>" + request.getLocalName() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getLocalAddr:&nbsp;</th>");
        pout.write("<td>" + request.getLocalAddr() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getLocalPort:&nbsp;</th>");
        pout.write("<td>" + Integer.toString(request.getLocalPort()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getMethod:&nbsp;</th>");
        pout.write("<td>" + notag(request.getMethod()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getPathInfo:&nbsp;</th>");
        pout.write("<td>" + notag(request.getPathInfo()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getPathTranslated:&nbsp;</th>");
        pout.write("<td>" + notag(request.getPathTranslated()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getProtocol:&nbsp;</th>");
        pout.write("<td>" + request.getProtocol() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getQueryString:&nbsp;</th>");
        pout.write("<td>" + notag(request.getQueryString()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRemoteAddr:&nbsp;</th>");
        pout.write("<td>" + request.getRemoteAddr() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRemoteHost:&nbsp;</th>");
        pout.write("<td>" + request.getRemoteHost() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRemotePort:&nbsp;</th>");
        pout.write("<td>" + request.getRemotePort() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRemoteUser:&nbsp;</th>");
        pout.write("<td>" + request.getRemoteUser() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRequestedSessionId:&nbsp;</th>");
        pout.write("<td>" + request.getRequestedSessionId() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRequestURI:&nbsp;</th>");
        pout.write("<td>" + notag(request.getRequestURI()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getRequestURL:&nbsp;</th>");
        pout.write("<td>" + notag(request.getRequestURL().toString()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getScheme:&nbsp;</th>");
        pout.write("<td>" + request.getScheme() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getServerName:&nbsp;</th>");
        pout.write("<td>" + notag(request.getServerName()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getServletPath:&nbsp;</th>");
        pout.write("<td>" + notag(request.getServletPath()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getServerPort:&nbsp;</th>");
        pout.write("<td>" + Integer.toString(request.getServerPort()) + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">getUserPrincipal:&nbsp;</th>");
        pout.write("<td>" + request.getUserPrincipal() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">isAsyncStarted():&nbsp;</th>");
        pout.write("<td>" + request.isAsyncStarted() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">isAsyncSupported():&nbsp;</th>");
        pout.write("<td>" + request.isAsyncSupported() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">isSecure():&nbsp;</th>");
        pout.write("<td>" + request.isSecure() + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">isUserInRole(admin):&nbsp;</th>");
        pout.write("<td>" + request.isUserInRole("admin") + "</td>");
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"right\">encodeRedirectURL(/foo?bar):&nbsp;</th>");
        pout.write("<td>" + response.encodeRedirectURL("/foo?bar") + "</td>");
        Enumeration<Locale> locales = request.getLocales();
        while (locales.hasMoreElements()) {
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">getLocales:&nbsp;</th>");
            pout.write("<td>" + locales.nextElement() + "</td>");
        }
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Other HTTP Headers:</big></th>");
        Enumeration<String> h = request.getHeaderNames();
        String name;
        while (h.hasMoreElements()) {
            name = (String) h.nextElement();
            Enumeration<String> h2 = request.getHeaders(name);
            while (h2.hasMoreElements()) {
                String hv = (String) h2.nextElement();
                pout.write("</tr><tr>\n");
                pout.write("<th align=\"right\">" + notag(name) + ":&nbsp;</th>");
                pout.write("<td>" + notag(hv) + "</td>");
            }
        }
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Request Parameters:</big></th>");
        h = request.getParameterNames();
        while (h.hasMoreElements()) {
            name = (String) h.nextElement();
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">" + notag(name) + ":&nbsp;</th>");
            pout.write("<td>" + notag(request.getParameter(name)) + "</td>");
            String[] values = request.getParameterValues(name);
            if (values == null) {
                pout.write("</tr><tr>\n");
                pout.write("<th align=\"right\">" + notag(name) + " Values:&nbsp;</th>");
                pout.write("<td>" + "NULL!" + "</td>");
            } else if (values.length > 1) {
                for (int i = 0; i < values.length; i++) {
                    pout.write("</tr><tr>\n");
                    pout.write("<th align=\"right\">" + notag(name) + "[" + i + "]:&nbsp;</th>");
                    pout.write("<td>" + notag(values[i]) + "</td>");
                }
            }
        }
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Cookies:</big></th>");
        Cookie[] cookies = request.getCookies();
        for (int i = 0; cookies != null && i < cookies.length; i++) {
            Cookie cookie = cookies[i];
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">" + notag(cookie.getName()) + ":&nbsp;</th>");
            pout.write("<td>" + notag(cookie.getValue()) + "</td>");
        }
        String content_type = request.getContentType();
        if (content_type != null && !content_type.startsWith("application/x-www-form-urlencoded") && !content_type.startsWith("multipart/form-data")) {
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"left\" valign=\"top\" colspan=\"2\"><big><br/>Content:</big></th>");
            pout.write("</tr><tr>\n");
            pout.write("<td><pre>");
            char[] content = new char[4096];
            int len;
            try {
                Reader in = request.getReader();
                while ((len = in.read(content)) >= 0) pout.write(notag(new String(content, 0, len)));
            } catch (IOException e) {
                pout.write(e.toString());
            }
            pout.write("</pre></td>");
        }
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Request Attributes:</big></th>");
        Enumeration<String> a = request.getAttributeNames();
        while (a.hasMoreElements()) {
            name = (String) a.nextElement();
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\" valign=\"top\">" + name.replace(".", " .") + ":&nbsp;</th>");
            Object value = request.getAttribute(name);
            if (value instanceof File) {
                File file = (File) value;
                pout.write("<td>" + "<pre>" + file.getName() + " (" + file.length() + " " + new Date(file.lastModified()) + ")</pre>" + "</td>");
            } else
                pout.write("<td>" + "<pre>" + toString(request.getAttribute(name)) + "</pre>" + "</td>");
        }
        request.setAttribute("org.eclipse.jetty.servlet.MultiPartFilter.files", null);
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Servlet InitParameters:</big></th>");
        a = getInitParameterNames();
        while (a.hasMoreElements()) {
            name = (String) a.nextElement();
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">" + name + ":&nbsp;</th>");
            pout.write("<td>" + toString(getInitParameter(name)) + "</td>");
        }
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Context InitParameters:</big></th>");
        a = getServletContext().getInitParameterNames();
        while (a.hasMoreElements()) {
            name = (String) a.nextElement();
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\" valign=\"top\">" + name.replace(".", " .") + ":&nbsp;</th>");
            pout.write("<td>" + toString(getServletContext().getInitParameter(name)) + "</td>");
        }
        pout.write("</tr><tr>\n");
        pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Context Attributes:</big></th>");
        a = getServletContext().getAttributeNames();
        while (a.hasMoreElements()) {
            name = (String) a.nextElement();
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\" valign=\"top\">" + name.replace(".", " .") + ":&nbsp;</th>");
            pout.write("<td>" + "<pre>" + toString(getServletContext().getAttribute(name)) + "</pre>" + "</td>");
        }
        String res = request.getParameter("resource");
        if (res != null && res.length() > 0) {
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"left\" colspan=\"2\"><big><br/>Get Resource: \"" + res + "\"</big></th>");
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">getServletContext().getResource(...):&nbsp;</th>");
            try {
                pout.write("<td>" + getServletContext().getResource(res) + "</td>");
            } catch (Exception e) {
                pout.write("<td>" + "" + e + "</td>");
            }
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">getServletContext().getResourcePaths(...):&nbsp;</th>");
            try {
                pout.write("<td>" + getServletContext().getResourcePaths(res) + "</td>");
            } catch (Exception e) {
                pout.write("<td>" + "" + e + "</td>");
            }
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">getServletContext().getRealPath(...):&nbsp;</th>");
            try {
                pout.write("<td>" + getServletContext().getRealPath(res) + "</td>");
            } catch (Exception e) {
                pout.write("<td>" + "" + e + "</td>");
            }
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">getServletContext().getContext(...):&nbsp;</th>");
            ServletContext context = getServletContext().getContext(res);
            pout.write("<td>" + context + "</td>");
            if (context != null) {
                pout.write("</tr><tr>\n");
                pout.write("<th align=\"right\">getServletContext().getContext(...).getResource(...):&nbsp;</th>");
                try {
                    pout.write("<td>" + context.getResource(res) + "</td>");
                } catch (Exception e) {
                    pout.write("<td>" + "" + e + "</td>");
                }
                pout.write("</tr><tr>\n");
                pout.write("<th align=\"right\">getServletContext().getContext(...).getResourcePaths(...):&nbsp;</th>");
                try {
                    pout.write("<td>" + context.getResourcePaths(res) + "</td>");
                } catch (Exception e) {
                    pout.write("<td>" + "" + e + "</td>");
                }
                String cp = context.getContextPath();
                if (cp == null || "/".equals(cp))
                    cp = "";
                pout.write("</tr><tr>\n");
                pout.write("<th align=\"right\">getServletContext().getContext(...).getRequestDispatcher(...):&nbsp;</th>");
                pout.write("<td>" + context.getRequestDispatcher(res.substring(cp.length())) + "</td>");
                pout.write("</tr><tr>\n");
                pout.write("<th align=\"right\">getServletContext().getContext(...).getRealPath(...):&nbsp;</th>");
                pout.write("<td>" + context.getRealPath(res.substring(cp.length())) + "</td>");
            }
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">this.getClass().getResource(...):&nbsp;</th>");
            pout.write("<td>" + this.getClass().getResource(res) + "</td>");
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">this.getClass().getClassLoader().getResource(...):&nbsp;</th>");
            pout.write("<td>" + this.getClass().getClassLoader().getResource(res) + "</td>");
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">Thread.currentThread().getContextClassLoader().getResource(...):&nbsp;</th>");
            pout.write("<td>" + Thread.currentThread().getContextClassLoader().getResource(res) + "</td>");
            pout.write("</tr><tr>\n");
            pout.write("<th align=\"right\">Thread.currentThread().getContextClassLoader().getResources(...):&nbsp;</th>");
            Enumeration<URL> urls = Thread.currentThread().getContextClassLoader().getResources(res);
            if (urls == null)
                pout.write("<td>null</td>");
            else
                pout.write("<td>" + Collections.list(urls) + "</td>");
        }
        pout.write("</tr></table>\n");
        /* ------------------------------------------------------------ */
        pout.write("<h2>Request Wrappers</h2>\n");
        ServletRequest rw = request;
        int w = 0;
        while (rw != null) {
            pout.write((w++) + ": " + rw.getClass().getName() + "<br/>");
            if (rw instanceof HttpServletRequestWrapper)
                rw = ((HttpServletRequestWrapper) rw).getRequest();
            else if (rw instanceof ServletRequestWrapper)
                rw = ((ServletRequestWrapper) rw).getRequest();
            else
                rw = null;
        }
        /* ------------------------------------------------------------ */
        pout.write("<h2>Response Wrappers</h2>\n");
        ServletResponse rsw = response;
        w = 0;
        while (rsw != null) {
            pout.write((w++) + ": " + rsw.getClass().getName() + "<br/>");
            if (rsw instanceof HttpServletResponseWrapper)
                rsw = ((HttpServletResponseWrapper) rsw).getResponse();
            else if (rsw instanceof ServletResponseWrapper)
                rsw = ((ServletResponseWrapper) rsw).getResponse();
            else
                rsw = null;
        }
        pout.write("<br/>");
        pout.write("<h2>International Characters (UTF-8)</h2>");
        pout.write("LATIN LETTER SMALL CAPITAL AE<br/>\n");
        pout.write("Directly uni encoded(\\u1d01): ᴁ<br/>");
        pout.write("HTML reference (&amp;AElig;): &AElig;<br/>");
        pout.write("Decimal (&amp;#7425;): &#7425;<br/>");
        pout.write("Javascript unicode (\\u1d01) : <script language='javascript'>document.write(\"ᴁ\");</script><br/>");
        pout.write("<br/>");
        pout.write("<h2>Form to generate GET content</h2>");
        pout.write("<form method=\"GET\" action=\"" + response.encodeURL(getURI(request)) + "\">");
        pout.write("TextField: <input type=\"text\" name=\"TextField\" value=\"value\"/><br/>\n");
        pout.write("<input type=\"submit\" name=\"Action\" value=\"Submit\">");
        pout.write("</form>");
        pout.write("<br/>");
        pout.write("<h2>Form to generate POST content</h2>");
        pout.write("<form method=\"POST\" accept-charset=\"utf-8\" action=\"" + response.encodeURL(getURI(request)) + "\">");
        pout.write("TextField: <input type=\"text\" name=\"TextField\" value=\"value\"/><br/>\n");
        pout.write("Select: <select multiple name=\"Select\">\n");
        pout.write("<option>ValueA</option>");
        pout.write("<option>ValueB1,ValueB2</option>");
        pout.write("<option>ValueC</option>");
        pout.write("</select><br/>");
        pout.write("<input type=\"submit\" name=\"Action\" value=\"Submit\"><br/>");
        pout.write("</form>");
        pout.write("<br/>");
        pout.write("<h2>Form to generate UPLOAD content</h2>");
        pout.write("<form method=\"POST\" enctype=\"multipart/form-data\" accept-charset=\"utf-8\" action=\"" + response.encodeURL(getURI(request)) + (request.getQueryString() == null ? "" : ("?" + request.getQueryString())) + "\">");
        pout.write("TextField: <input type=\"text\" name=\"TextField\" value=\"comment\"/><br/>\n");
        pout.write("File 1: <input type=\"file\" name=\"file1\" /><br/>\n");
        pout.write("File 2: <input type=\"file\" name=\"file2\" /><br/>\n");
        pout.write("<input type=\"submit\" name=\"Action\" value=\"Submit\"><br/>");
        pout.write("</form>");
        pout.write("<h2>Form to set Cookie</h2>");
        pout.write("<form method=\"POST\" action=\"" + response.encodeURL(getURI(request)) + "\">");
        pout.write("cookie: <input type=\"text\" name=\"cookie\" /><br/>\n");
        pout.write("value: <input type=\"text\" name=\"cookiev\" /><br/>\n");
        pout.write("<input type=\"submit\" name=\"Action\" value=\"setCookie\">");
        pout.write("</form>\n");
        pout.write("<h2>Form to get Resource</h2>");
        pout.write("<form method=\"POST\" action=\"" + response.encodeURL(getURI(request)) + "\">");
        pout.write("resource: <input type=\"text\" name=\"resource\" /><br/>\n");
        pout.write("<input type=\"submit\" name=\"Action\" value=\"getResource\">");
        pout.write("</form>\n");
    } catch (Exception e) {
        getServletContext().log("dump " + e);
    }
    String lines = request.getParameter("lines");
    if (lines != null) {
        char[] line = "<span>A line of characters. Blah blah blah blah.  blooble blooble</span></br>\n".toCharArray();
        for (int l = Integer.parseInt(lines); l-- > 0; ) {
            pout.write("<span>" + l + " </span>");
            pout.write(line);
        }
    }
    pout.write("</body>\n</html>\n");
    pout.close();
    if (pi != null) {
        if ("/ex4".equals(pi))
            throw new ServletException("test ex4", new Throwable());
        if ("/ex5".equals(pi))
            throw new IOException("test ex5");
        if ("/ex6".equals(pi))
            throw new UnavailableException("test ex6");
    }
}
Also used : Locale(java.util.Locale) HttpServletRequest(javax.servlet.http.HttpServletRequest) ServletRequest(javax.servlet.ServletRequest) OutputStream(java.io.OutputStream) UnavailableException(javax.servlet.UnavailableException) HttpServletResponseWrapper(javax.servlet.http.HttpServletResponseWrapper) Reader(java.io.Reader) AsyncContext(javax.servlet.AsyncContext) URL(java.net.URL) BufferedWriter(java.io.BufferedWriter) ServletException(javax.servlet.ServletException) Field(java.lang.reflect.Field) TimerTask(java.util.TimerTask) HttpServletRequestWrapper(javax.servlet.http.HttpServletRequestWrapper) ServletContext(javax.servlet.ServletContext) ServletResponseWrapper(javax.servlet.ServletResponseWrapper) HttpServletResponseWrapper(javax.servlet.http.HttpServletResponseWrapper) PrintWriter(java.io.PrintWriter) Cookie(javax.servlet.http.Cookie) HttpServletResponse(javax.servlet.http.HttpServletResponse) ServletResponse(javax.servlet.ServletResponse) HttpServletRequestWrapper(javax.servlet.http.HttpServletRequestWrapper) ServletRequestWrapper(javax.servlet.ServletRequestWrapper) IOException(java.io.IOException) AsyncEvent(javax.servlet.AsyncEvent) ServletException(javax.servlet.ServletException) IOException(java.io.IOException) UnavailableException(javax.servlet.UnavailableException) Date(java.util.Date) AsyncListener(javax.servlet.AsyncListener) OutputStreamWriter(java.io.OutputStreamWriter) File(java.io.File)

Example 2 with ServletResponseWrapper

use of javax.servlet.ServletResponseWrapper in project sling by apache.

the class ExternalServletContextWrapperTest method testUnwrappingDoubleWrappedResponse.

@Test
public void testUnwrappingDoubleWrappedResponse() {
    final ServletResponse resp = context.mock(ServletResponse.class);
    final ServletResponseWrapper wrapper = new ServletResponseWrapper(resp);
    final ServletResponseWrapper wrapper2 = new ServletResponseWrapper(wrapper);
    ServletResponse unwrapped = ExternalServletContextWrapper.RequestDispatcherWrapper.unwrapServletResponse(wrapper2);
    assertEquals(resp, unwrapped);
}
Also used : HttpServletResponse(javax.servlet.http.HttpServletResponse) ServletResponse(javax.servlet.ServletResponse) ServletResponseWrapper(javax.servlet.ServletResponseWrapper) HttpServletResponseWrapper(javax.servlet.http.HttpServletResponseWrapper) Test(org.junit.Test)

Example 3 with ServletResponseWrapper

use of javax.servlet.ServletResponseWrapper in project OpenOLAT by OpenOLAT.

the class DefaultDispatcher method serveResource.

/**
 * Serve the specified resource, optionally including the data content.
 *
 * @param request  The servlet request we are processing
 * @param response The servlet response we are creating
 * @param content  Should the content be included?
 * @param encoding The encoding to use if it is necessary to access the
 *                 source as characters rather than as bytes
 *
 * @exception IOException if an input/output error occurs
 * @exception ServletException if a servlet-specified error occurs
 */
protected void serveResource(HttpServletRequest request, HttpServletResponse response, boolean content, String encoding) throws IOException, ServletException {
    boolean serveContent = content;
    boolean debug = log.isDebug();
    // Identify the requested resource path
    String path = getRelativePath(request);
    if (debug) {
        if (serveContent)
            log.debug("DefaultServlet.serveResource:  Serving resource '" + path + "' headers and data");
        else
            log.debug("DefaultServlet.serveResource:  Serving resource '" + path + "' headers only");
    }
    WebResourceRoot resources = getResources(request);
    WebResource resource = resources.getResource(path);
    if (!resource.exists()) {
        // Check if we're included so we can return the appropriate
        // missing resource name in the error
        String requestUri = (String) request.getAttribute(RequestDispatcher.INCLUDE_REQUEST_URI);
        if (requestUri == null) {
            requestUri = request.getRequestURI();
        } else {
            // SRV.9.3 says we must throw a FNFE
            throw new FileNotFoundException("defaultServlet.missingResource");
        }
        response.sendError(HttpServletResponse.SC_NOT_FOUND, requestUri);
        return;
    }
    // ends with "/" or "\", return NOT FOUND
    if (resource.isFile() && path.endsWith("/") || path.endsWith("\\")) {
        // Check if we're included so we can return the appropriate
        // missing resource name in the error
        String requestUri = (String) request.getAttribute(RequestDispatcher.INCLUDE_REQUEST_URI);
        if (requestUri == null) {
            requestUri = request.getRequestURI();
        }
        response.sendError(HttpServletResponse.SC_NOT_FOUND, requestUri);
        return;
    }
    boolean isError = response.getStatus() >= HttpServletResponse.SC_BAD_REQUEST;
    boolean included = false;
    // satisfied.
    if (resource.isFile()) {
        // Checking If headers
        included = (request.getAttribute(RequestDispatcher.INCLUDE_CONTEXT_PATH) != null);
        if (!included && !isError && !checkIfHeaders(request, response, resource)) {
            return;
        }
    }
    // Find content type.
    String contentType = resource.getMimeType();
    if (contentType == null) {
        contentType = request.getServletContext().getMimeType(resource.getName());
        resource.setMimeType(contentType);
    }
    // These need to reflect the original resource, not the potentially
    // gzip'd version of the resource so get them now if they are going to
    // be needed later
    String eTag = null;
    String lastModifiedHttp = null;
    if (resource.isFile() && !isError) {
        eTag = resource.getETag();
        lastModifiedHttp = resource.getLastModifiedHttp();
    }
    // Serve a gzipped version of the file if present
    boolean usingGzippedVersion = false;
    if (gzip && !included && resource.isFile() && !path.endsWith(".gz")) {
        WebResource gzipResource = resources.getResource(path + ".gz");
        if (gzipResource.exists() && gzipResource.isFile()) {
            Collection<String> varyHeaders = response.getHeaders("Vary");
            boolean addRequired = true;
            for (String varyHeader : varyHeaders) {
                if ("*".equals(varyHeader) || "accept-encoding".equalsIgnoreCase(varyHeader)) {
                    addRequired = false;
                    break;
                }
            }
            if (addRequired) {
                response.addHeader("Vary", "accept-encoding");
            }
            if (checkIfGzip(request)) {
                response.addHeader("Content-Encoding", "gzip");
                resource = gzipResource;
                usingGzippedVersion = true;
            }
        }
    }
    ArrayList<Range> ranges = null;
    long contentLength = -1L;
    if (resource.isDirectory()) {
        contentType = "text/html;charset=UTF-8";
    } else {
        if (!isError) {
            if (useAcceptRanges) {
                // Accept ranges header
                response.setHeader("Accept-Ranges", "bytes");
            }
            // Parse range specifier
            ranges = parseRange(request, response, resource);
            // ETag header
            response.setHeader("ETag", eTag);
            // Last-Modified header
            response.setHeader("Last-Modified", lastModifiedHttp);
        }
        // Get content length
        contentLength = resource.getContentLength();
        // (silent) ISE when setting the output buffer size
        if (contentLength == 0L) {
            serveContent = false;
        }
    }
    ServletOutputStream ostream = null;
    PrintWriter writer = null;
    if (serveContent) {
        // Trying to retrieve the servlet output stream
        try {
            ostream = response.getOutputStream();
        } catch (IllegalStateException e) {
            // trying to serve a text file
            if (!usingGzippedVersion && ((contentType == null) || (contentType.startsWith("text")) || (contentType.endsWith("xml")) || (contentType.contains("/javascript")))) {
                writer = response.getWriter();
                // Cannot reliably serve partial content with a Writer
                ranges = FULL;
            } else {
                throw e;
            }
        }
    }
    // Check to see if a Filter, Valve of wrapper has written some content.
    // If it has, disable range requests and setting of a content length
    // since neither can be done reliably.
    ServletResponse r = response;
    long contentWritten = 0;
    while (r instanceof ServletResponseWrapper) {
        r = ((ServletResponseWrapper) r).getResponse();
    }
    if (contentWritten > 0) {
        ranges = FULL;
    }
    if (resource.isDirectory() || isError || ((ranges == null || ranges.isEmpty()) && request.getHeader("Range") == null) || ranges == FULL) {
        // Set the appropriate output headers
        if (contentType != null) {
            if (debug)
                log.debug("DefaultServlet.serveFile:  contentType='" + contentType + "'");
            response.setContentType(contentType);
        }
        if (resource.isFile() && contentLength >= 0 && (!serveContent || ostream != null)) {
            if (debug)
                log.debug("DefaultServlet.serveFile:  contentLength=" + contentLength);
            // written to the response.
            if (contentWritten == 0) {
                response.setContentLengthLong(contentLength);
            }
        }
        InputStream renderResult = null;
        if (resource.isDirectory()) {
            if (serveContent) {
                // Serve the directory browser
                // TODO tomcat render(getPathPrefix(request), resource);
                renderResult = null;
            }
        }
        // Copy the input stream to our output stream (if requested)
        if (serveContent) {
            resource.increaseDownloadCount();
            try {
                response.setBufferSize(output);
            } catch (IllegalStateException e) {
            // Silent catch
            }
            if (ostream == null) {
                // content directly.
                if (resource.isDirectory()) {
                    // render(getPathPrefix(request), resource);
                    renderResult = null;
                } else {
                    renderResult = resource.getInputStream();
                }
                copy(resource, renderResult, writer, encoding);
            } else {
                // Output is via an InputStream
                if (resource.isDirectory()) {
                    // render(getPathPrefix(request), resource);
                    renderResult = null;
                } else {
                    renderResult = resource.getInputStream();
                }
                // the output (this method closes the stream)
                if (renderResult != null) {
                    copy(renderResult, ostream);
                }
            }
        }
    } else {
        // download counter
        resource.increaseDownloadCount();
        if ((ranges == null) || (ranges.isEmpty()))
            return;
        // Partial content response.
        response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
        if (ranges.size() == 1) {
            Range range = ranges.get(0);
            response.addHeader("Content-Range", "bytes " + range.start + "-" + range.end + "/" + range.length);
            long length = range.end - range.start + 1;
            response.setContentLengthLong(length);
            if (contentType != null) {
                if (debug)
                    log.debug("DefaultServlet.serveFile:  contentType='" + contentType + "'");
                response.setContentType(contentType);
            }
            if (serveContent) {
                try {
                    response.setBufferSize(output);
                } catch (IllegalStateException e) {
                // Silent catch
                }
                if (ostream != null) {
                    copy(resource, ostream, range);
                } else {
                    // we should not get here
                    throw new IllegalStateException();
                }
            }
        } else {
            response.setContentType("multipart/byteranges; boundary=" + mimeSeparation);
            if (serveContent) {
                try {
                    response.setBufferSize(output);
                } catch (IllegalStateException e) {
                // Silent catch
                }
                if (ostream != null) {
                    copy(resource, ostream, ranges.iterator(), contentType);
                } else {
                    // we should not get here
                    throw new IllegalStateException();
                }
            }
        }
    }
}
Also used : HttpServletResponse(javax.servlet.http.HttpServletResponse) ServletResponse(javax.servlet.ServletResponse) ServletOutputStream(javax.servlet.ServletOutputStream) BufferedInputStream(java.io.BufferedInputStream) InputStream(java.io.InputStream) FileNotFoundException(java.io.FileNotFoundException) ServletResponseWrapper(javax.servlet.ServletResponseWrapper) PrintWriter(java.io.PrintWriter)

Example 4 with ServletResponseWrapper

use of javax.servlet.ServletResponseWrapper in project tomcat70 by apache.

the class ApplicationDispatcher method wrapResponse.

/**
 * Create and return a response wrapper that has been inserted in the
 * appropriate spot in the response chain.
 */
private ServletResponse wrapResponse(State state) {
    // Locate the response we should insert in front of
    ServletResponse previous = null;
    ServletResponse current = state.outerResponse;
    while (current != null) {
        if (state.hresponse == null && (current instanceof HttpServletResponse)) {
            state.hresponse = (HttpServletResponse) current;
            if (// Forward only needs hresponse
            !state.including)
                return null;
        }
        if (!(current instanceof ServletResponseWrapper))
            break;
        if (current instanceof ApplicationHttpResponse)
            break;
        if (current instanceof ApplicationResponse)
            break;
        previous = current;
        current = ((ServletResponseWrapper) current).getResponse();
    }
    // Instantiate a new wrapper at this point and insert it in the chain
    ServletResponse wrapper = null;
    if ((current instanceof ApplicationHttpResponse) || (current instanceof Response) || (current instanceof HttpServletResponse))
        wrapper = new ApplicationHttpResponse((HttpServletResponse) current, state.including);
    else
        wrapper = new ApplicationResponse(current, state.including);
    if (previous == null)
        state.outerResponse = wrapper;
    else
        ((ServletResponseWrapper) previous).setResponse(wrapper);
    state.wrapResponse = wrapper;
    return (wrapper);
}
Also used : HttpServletResponse(javax.servlet.http.HttpServletResponse) Response(org.apache.catalina.connector.Response) ServletResponse(javax.servlet.ServletResponse) HttpServletResponse(javax.servlet.http.HttpServletResponse) ServletResponse(javax.servlet.ServletResponse) HttpServletResponse(javax.servlet.http.HttpServletResponse) ServletResponseWrapper(javax.servlet.ServletResponseWrapper)

Example 5 with ServletResponseWrapper

use of javax.servlet.ServletResponseWrapper in project tomcat70 by apache.

the class ApplicationDispatcher method unwrapResponse.

/**
 * Unwrap the response if we have wrapped it.
 */
private void unwrapResponse(State state) {
    if (state.wrapResponse == null)
        return;
    if (state.outerRequest.isAsyncStarted()) {
        if (!state.outerRequest.getAsyncContext().hasOriginalRequestAndResponse()) {
            return;
        }
    }
    ServletResponse previous = null;
    ServletResponse current = state.outerResponse;
    while (current != null) {
        // If we run into the container response we are done
        if ((current instanceof Response) || (current instanceof ResponseFacade))
            break;
        // Remove the current response if it is our wrapper
        if (current == state.wrapResponse) {
            ServletResponse next = ((ServletResponseWrapper) current).getResponse();
            if (previous == null)
                state.outerResponse = next;
            else
                ((ServletResponseWrapper) previous).setResponse(next);
            break;
        }
        // Advance to the next response in the chain
        previous = current;
        current = ((ServletResponseWrapper) current).getResponse();
    }
}
Also used : HttpServletResponse(javax.servlet.http.HttpServletResponse) Response(org.apache.catalina.connector.Response) ServletResponse(javax.servlet.ServletResponse) HttpServletResponse(javax.servlet.http.HttpServletResponse) ServletResponse(javax.servlet.ServletResponse) ResponseFacade(org.apache.catalina.connector.ResponseFacade) ServletResponseWrapper(javax.servlet.ServletResponseWrapper)

Aggregations

ServletResponseWrapper (javax.servlet.ServletResponseWrapper)15 ServletResponse (javax.servlet.ServletResponse)13 HttpServletResponse (javax.servlet.http.HttpServletResponse)13 PrintWriter (java.io.PrintWriter)6 ServletOutputStream (javax.servlet.ServletOutputStream)6 ServletRequestWrapper (javax.servlet.ServletRequestWrapper)6 ServletException (javax.servlet.ServletException)5 ServletRequest (javax.servlet.ServletRequest)5 HttpServletRequest (javax.servlet.http.HttpServletRequest)5 BufferedInputStream (java.io.BufferedInputStream)4 FileNotFoundException (java.io.FileNotFoundException)4 IOException (java.io.IOException)4 InputStream (java.io.InputStream)4 PrivilegedActionException (java.security.PrivilegedActionException)3 Deque (java.util.Deque)3 HttpServletResponseWrapper (javax.servlet.http.HttpServletResponseWrapper)3 ResponseFacade (org.apache.catalina.connector.ResponseFacade)3 ByteArrayInputStream (java.io.ByteArrayInputStream)2 FileInputStream (java.io.FileInputStream)2 OutputStreamWriter (java.io.OutputStreamWriter)2