environmentpip.java

一个完整的XACML工程,学习XACML技术的好例子!

/*
* Copyright (c) 2006, University of Kent
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without 
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this 
* list of conditions and the following disclaimer.
* 
* Redistributions in binary form must reproduce the above copyright notice, 
* this list of conditions and the following disclaimer in the documentation 
* and/or other materials provided with the distribution. 
*
* 1. Neither the name of the University of Kent nor the names of its 
* contributors may be used to endorse or promote products derived from this 
* software without specific prior written permission. 
*
* 2. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS  
* IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 
* PURPOSE ARE DISCLAIMED. 
*
* 3. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
* POSSIBILITY OF SUCH DAMAGE.
*
* 4. YOU AGREE THAT THE EXCLUSIONS IN PARAGRAPHS 2 AND 3 ABOVE ARE REASONABLE
* IN THE CIRCUMSTANCES.  IN PARTICULAR, YOU ACKNOWLEDGE (1) THAT THIS
* SOFTWARE HAS BEEN MADE AVAILABLE TO YOU FREE OF CHARGE, (2) THAT THIS
* SOFTWARE IS NOT "PRODUCT" QUALITY, BUT HAS BEEN PRODUCED BY A RESEARCH
* GROUP WHO DESIRE TO MAKE THIS SOFTWARE FREELY AVAILABLE TO PEOPLE WHO WISH
* TO USE IT, AND (3) THAT BECAUSE THIS SOFTWARE IS NOT OF "PRODUCT" QUALITY
* IT IS INEVITABLE THAT THERE WILL BE BUGS AND ERRORS, AND POSSIBLY MORE
* SERIOUS FAULTS, IN THIS SOFTWARE.
*
* 5. This license is governed, except to the extent that local laws
* necessarily apply, by the laws of England and Wales.
*/
/*
 * ActionPIP.java
 *
 * Created on 02 November 2006, 12:19
 *
 * To change this template, choose Tools | Template Manager
 * and open the template in the editor.
 */

package uk.ac.kent.dpa.custom.pip;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import org.globus.wsrf.security.authorization.*;
import org.w3c.dom.*;
import javax.security.auth.Subject;
import javax.xml.rpc.handler.MessageContext;
import javax.xml.namespace.QName;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.globus.wsrf.impl.security.authorization.exceptions.*;
import java.util.*;
import javax.xml.parsers.*;
import uk.ac.kent.dpa.custom.authz.util.AuthzException;
import uk.ac.kent.dpa.custom.authz.util.Merge;
import issrg.web.service.EncodeXML;


/**
 *
 * @author Linying Su
 */

public class EnvironmentPIP implements PIP {
    
    private static Log logger = LogFactory.getLog(EnvironmentPIP.class.getName());
    /** Creates a new instance of ActionPIP */
    public EnvironmentPIP() {
    }
    
    public void collectAttributes(Subject subject, MessageContext ctx, QName operation) throws AttributeException {
        logger.info("the environment PIP is called");
        Calendar cal = new GregorianCalendar();
        int year = cal.get(Calendar.YEAR);
        int month = cal.get(Calendar.MONTH)+1;
        int day = cal.get(Calendar.DAY_OF_MONTH);
        int hour = cal.get(Calendar.HOUR_OF_DAY);
        int min = cal.get(Calendar.MINUTE);
        int sec = cal.get(Calendar.SECOND);
        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
        factory.setValidating(false);
        try {
            Document doc = factory.newDocumentBuilder().newDocument();
            Element request = doc.createElement("Request");
            request.setAttribute("xmlns","urn:oasis:names:tc:xacml:1.0:context");
            request.setAttribute("xmlns:xsi","http://www.w3.org/2001/XMLSchema-instance");
            Element action = doc.createElement("Environment");
            Element attribute = doc.createElement("Attribute");
            attribute.setAttribute("AttributeId","urn:oasis:names:tc:xacml:1.0:environment:current-date");
            attribute.setAttribute("DataType","http://www.w3.org/2001/XMLSchema#date");
            Element value = doc.createElement("AttributeValue");
            String y = ""+year;
            String m = month<10?"0"+month:""+month;
            String d = day<10?"0"+day:""+day;
            String date = y+"-"+m+"-"+d;
            Text text = doc.createTextNode(date);
            value.appendChild(text);
            attribute.appendChild(value);
            action.appendChild(attribute);
            attribute = doc.createElement("Attribute");
            attribute.setAttribute("AttributeId","urn:oasis:names:tc:xacml:1.0:environment:current-time");
            attribute.setAttribute("DataType","http://www.w3.org/2001/XMLSchema#time");
            value = doc.createElement("AttributeValue");
            String h = hour<10?"0"+hour:""+hour;
            m = min<10?"0"+min:""+min;
            String s = sec<10?"0"+sec:""+sec;
            String time = h+":"+m+":"+s;
            text = doc.createTextNode(time);
            value.appendChild(text);
            attribute.appendChild(value);
            action.appendChild(attribute);
            attribute = doc.createElement("Attribute");
            attribute.setAttribute("AttributeId","urn:oasis:names:tc:xacml:1.0:environment:current-dateTime");
            attribute.setAttribute("DataType","http://www.w3.org/2001/XMLSchema#dateTime");
            value = doc.createElement("AttributeValue");
            text = doc.createTextNode(date+"T"+time);
            value.appendChild(text);
            attribute.appendChild(value);
            action.appendChild(attribute);
            request.appendChild(action);

            Element pre = null;
            if (ctx!=null) pre = (Element)ctx.getProperty("request.context");
            if (pre!=null) {
                request = new Merge().merge(pre,request);
            }
            if (ctx!=null) ctx.setProperty("request.context",request);
            logger.debug("the current request context");
            logger.debug(new EncodeXML().encode(request,0));
        } catch (ParserConfigurationException pe) {
            throw new AttributeException("XML parser error:"+pe);
        } catch (AuthzException ae) {
            throw new AttributeException("processing error:"+ae);
        }
    }  
    
    public void initialize(PDPConfig pdpConfig, String name, String id) throws InitializeException {
        logger.debug("to initialise the custom enviromental PIP (scope:"+name+") for the service "+id);
        logger.debug("to get current time, date and dateTime");
    }
    
    public void close() throws CloseException {     
    }
}