be.fedict.eid.idp.sp.PhotoServlet.java Source code

Introduction

Here is the source code for be.fedict.eid.idp.sp.PhotoServlet.java

Source

/*
 * eID Identity Provider Project.
 * Copyright (C) 2010 FedICT.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version
 * 3.0 as published by the Free Software Foundation.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, see
 * http://www.gnu.org/licenses/.
 */

package be.fedict.eid.idp.sp;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.IOException;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

public class PhotoServlet extends HttpServlet {

    private static final long serialVersionUID = 1L;

    private static final Log LOG = LogFactory.getLog(PhotoServlet.class);

    public static final String PHOTO_SESSION_ATTRIBUTE = "Photo";

    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        LOG.debug("doGet");
        response.setContentType("image/jpg");
        response.setHeader("Cache-Control", "no-cache, no-store, must-revalidate, max-age=-1"); // http 1.1
        response.setHeader("Pragma", "no-cache, no-store"); // http 1.0
        response.setDateHeader("Expires", -1);
        ServletOutputStream out = response.getOutputStream();
        HttpSession session = request.getSession();

        byte[] photoData = (byte[]) session.getAttribute(PHOTO_SESSION_ATTRIBUTE);

        if (null != photoData) {
            BufferedImage photo = ImageIO.read(new ByteArrayInputStream(photoData));
            if (null == photo) {
                /*
                 * In this case we render a photo containing some error message.
                 */
                photo = new BufferedImage(140, 200, BufferedImage.TYPE_INT_RGB);
                Graphics2D graphics = (Graphics2D) photo.getGraphics();
                RenderingHints renderingHints = new RenderingHints(RenderingHints.KEY_TEXT_ANTIALIASING,
                        RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
                graphics.setRenderingHints(renderingHints);
                graphics.setColor(Color.WHITE);
                graphics.fillRect(1, 1, 140 - 1 - 1, 200 - 1 - 1);
                graphics.setColor(Color.RED);
                graphics.setFont(new Font("Dialog", Font.BOLD, 20));
                graphics.drawString("Photo Error", 0, 200 / 2);
                graphics.dispose();
                ImageIO.write(photo, "jpg", out);
            } else {
                out.write(photoData);
            }
        }
        out.close();
    }
}