RefererClientSecurity.java

/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2011, Geomatys
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library 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.
 */
package org.geotoolkit.security;

import java.net.URLConnection;
import org.apache.sis.util.ArgumentChecks;

/**
 * Client security which fill the referer for url connections.
 * 
 * @author Johann Sorel (Geomatys)
 * @module
 */
public class RefererClientSecurity extends DefaultClientSecurity{

    private final String referer;

    public RefererClientSecurity(String referer) {
        ArgumentChecks.ensureNonNull("referer", referer);
        this.referer = referer;
    }
    
    @Override
    public URLConnection secure(URLConnection cnx) {
        cnx = super.secure(cnx);
        cnx.setRequestProperty("referer",referer);
        return cnx;
    }
    
}