Java tutorial
/** * ************************************************************************************************* * * * This file is part of WebGoat, an Open Web Application Security Project * utility. For details, please see http://www.owasp.org/ * * Copyright (c) 2002 - 20014 Bruce Mayhew * * This program is free software; you can redistribute it and/or modify it under * the terms of the GNU General Public License as published by the Free Software * Foundation; either version 2 of the License, or (at your option) any later * version. * * This program is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more * details. * * You should have received a copy of the GNU General Public License along with * this program; if not, write to the Free Software Foundation, Inc., 59 Temple * Place - Suite 330, Boston, MA 02111-1307, USA. * * Getting Source ============== * * Source for this application is maintained at https://github.com/WebGoat/WebGoat, a repository * for free software projects. * * For details, please see http://webgoat.github.io */ package org.owasp.webgoat.service; import java.io.PrintWriter; import java.io.StringWriter; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpSession; import org.owasp.webgoat.session.WebSession; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.http.HttpStatus; import org.springframework.web.bind.annotation.ExceptionHandler; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.bind.annotation.ResponseStatus; /** * <p>Abstract BaseService class.</p> * * @author rlawson * @version $Id: $Id */ @RequestMapping("/service") public abstract class BaseService { private static final Logger logger = LoggerFactory.getLogger(BaseService.class); /** * <p>handleException.</p> * * @param request a {@link javax.servlet.http.HttpServletRequest} object. * @param ex a {@link java.lang.Exception} object. * @return a {@link org.owasp.webgoat.service.ExceptionInfo} object. */ @ExceptionHandler(Exception.class) @ResponseStatus(value = HttpStatus.I_AM_A_TEAPOT) public @ResponseBody ExceptionInfo handleException(HttpServletRequest request, Exception ex) { String url = request.getRequestURL().toString(); logger.error("Exception handler for service caught exception when processing: " + url, ex); ExceptionInfo response = new ExceptionInfo(); response.setUrl(url); response.setMessage(getStringStackTrace(ex)); return response; } /** * <p>getWebSession.</p> * * @param session a {@link javax.servlet.http.HttpSession} object. * @return a {@link org.owasp.webgoat.session.WebSession} object. */ public WebSession getWebSession(HttpSession session) { WebSession ws; Object o = session.getAttribute(WebSession.SESSION); if (o == null) { throw new IllegalArgumentException( "No valid WebSession object found, has session timed out? [" + session.getId() + "]"); } if (!(o instanceof WebSession)) { throw new IllegalArgumentException("Invalid WebSession object found, this is probably a bug! [" + o.getClass() + " | " + session.getId() + "]"); } ws = (WebSession) o; return ws; } /** * <p>getStringStackTrace.</p> * * @param t a {@link java.lang.Throwable} object. * @return a {@link java.lang.String} object. */ public String getStringStackTrace(Throwable t) { StringWriter sw = new StringWriter(); PrintWriter pw = new PrintWriter(sw); t.printStackTrace(pw); return sw.toString(); } }