java.org.jboss.security.config.parser.ApplicationPolicyParser Maven / Gradle / Ivy

Go to download
/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2008, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors. 
 *
 * This 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; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * 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, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jboss.security.config.parser;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import javax.security.auth.login.AppConfigurationEntry;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLEventReader;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.events.Attribute;
import javax.xml.stream.events.StartElement;
import javax.xml.stream.events.XMLEvent;

import org.jboss.security.acl.config.ACLProviderEntry;
import org.jboss.security.audit.config.AuditProviderEntry;
import org.jboss.security.auth.login.AuthenticationInfo;
import org.jboss.security.auth.login.JASPIAuthenticationInfo;
import org.jboss.security.authorization.config.AuthorizationModuleEntry;
import org.jboss.security.config.ACLInfo;
import org.jboss.security.config.ApplicationPolicy;
import org.jboss.security.config.AuditInfo;
import org.jboss.security.config.AuthorizationInfo;
import org.jboss.security.config.IdentityTrustInfo;
import org.jboss.security.config.MappingInfo;
import org.jboss.security.identitytrust.config.IdentityTrustModuleEntry;
import org.jboss.security.mapping.MappingType;
import org.jboss.security.mapping.config.MappingModuleEntry;

/**
 * Stax based Application Policy Parser
 * @author [email protected]
 * @since Jan 22, 2010
 */
public class ApplicationPolicyParser
{ 
   @SuppressWarnings("unchecked")
   public List parse(XMLEventReader xmlEventReader) throws XMLStreamException
   {
      List policies = new ArrayList();
      while(xmlEventReader.hasNext())
      { 
          XMLEvent xmlEvent = xmlEventReader.nextEvent();
          int eventType = xmlEvent.getEventType();
          switch(eventType)
          {
              case XMLStreamConstants.START_ELEMENT:  
                 StartElement appPolicyElement = (StartElement) xmlEvent;
                 if("application-policy".equals(StaxParserUtil.getStartElementName(appPolicyElement)) == false)
                    throw new RuntimeException(" element expected at " + 
                          StaxParserUtil.getLineColumnNumber(xmlEvent.getLocation()));
                 //We got the application-policy element. It just has one attribute "name"
                 Iterator attrs = appPolicyElement.getAttributes(); 
                 String extendsName = null;
                 String appPolicyName = null;
                 
                 while(attrs.hasNext())
                 {
                    Attribute attribute = attrs.next();
                    QName attributeName = attribute.getName();
                    String attributeValue = StaxParserUtil.getAttributeValue(attribute);
                    
                    if("name".equals(attributeName.getLocalPart()))
                       appPolicyName = attributeValue; 
                    else if("extends".equals(attributeName.getLocalPart()))
                       extendsName = attributeValue;  
                 }

                 ApplicationPolicy applicationPolicy = new ApplicationPolicy(appPolicyName); 
                 if(extendsName != null)
                    applicationPolicy.setBaseApplicationPolicyName(extendsName);
                  
                 route(xmlEventReader, applicationPolicy);
                 policies.add(applicationPolicy); 
          } 
      }
      return policies;
   }
   
   @SuppressWarnings("unchecked")
   private void route(XMLEventReader xmlEventReader, ApplicationPolicy appPolicy) throws XMLStreamException
   {
      while(true)
      {
         XMLEvent xmlEvent = xmlEventReader.peek();
         if(xmlEvent == null)
            return;
         StartElement startElement = xmlEvent.asStartElement();
         String elementName = StaxParserUtil.getStartElementName(startElement);
         if("authentication".equals(elementName))
         {
            xmlEvent = xmlEventReader.nextEvent();
            AuthenticationConfigParser parser = new AuthenticationConfigParser();
            Set entries = parser.parse(xmlEventReader);
            AuthenticationInfo authInfo = new AuthenticationInfo();
             
            authInfo.setAppConfigurationEntry(new ArrayList(entries));
            appPolicy.setAuthenticationInfo(authInfo); 
         }
         else if("authentication-jaspi".equals(elementName))
         {

            xmlEvent = xmlEventReader.nextEvent();
            AuthenticationJASPIConfigParser parser = new AuthenticationJASPIConfigParser();
            JASPIAuthenticationInfo authInfo = parser.parse(xmlEventReader); 
            appPolicy.setAuthenticationInfo(authInfo); 
         } 
         else if("authorization".equals(elementName))
         { 
            xmlEvent = xmlEventReader.nextEvent();
            AuthorizationConfigParser parser = new AuthorizationConfigParser();
            Set entries = parser.parse(xmlEventReader);
            AuthorizationInfo authInfo = new AuthorizationInfo(appPolicy.getName());
            authInfo.add(new ArrayList(entries));
            appPolicy.setAuthorizationInfo(authInfo); 
         } 
         else if("acl".equals(elementName))
         { 
            xmlEvent = xmlEventReader.nextEvent();
            AclConfigParser parser = new AclConfigParser();
            Set entries = parser.parse(xmlEventReader);
            ACLInfo aclInfo = new ACLInfo(appPolicy.getName());
            aclInfo.add(new ArrayList(entries));
            appPolicy.setAclInfo(aclInfo); 
         }  
         else if("rolemapping".equals(elementName))
         { 
            xmlEvent = xmlEventReader.nextEvent();
            MappingConfigParser parser = new MappingConfigParser();
            List entries = parser.parse(xmlEventReader);
            MappingInfo mappingInfo = new MappingInfo(appPolicy.getName()); 
            mappingInfo.add(entries);
            appPolicy.setMappingInfo(MappingType.ROLE.toString(), mappingInfo);
         }  
         else if("mapping".equals(elementName))
         { 
            xmlEvent = xmlEventReader.nextEvent();
            MappingConfigParser parser = new MappingConfigParser();
            List entries = parser.parse(xmlEventReader);
            for(MappingModuleEntry entry: entries)
            {
               MappingInfo mappingInfo = new MappingInfo(appPolicy.getName());
               mappingInfo.add(entry);
               String moduleType = entry.getMappingModuleType();
               appPolicy.setMappingInfo(moduleType, mappingInfo); 
            }
         } 
         else if("audit".equals(elementName))
         { 
            xmlEvent = xmlEventReader.nextEvent();
            AuditConfigParser parser = new AuditConfigParser();
            List entries = parser.parse(xmlEventReader);
            AuditInfo authInfo = new AuditInfo(appPolicy.getName());
            authInfo.add(entries);
            appPolicy.setAuditInfo(authInfo); 
         } 
         else if("identity-trust".equals(elementName))
         { 
            xmlEvent = xmlEventReader.nextEvent();
            IdentityTrustConfigParser parser = new IdentityTrustConfigParser();
            List entries = parser.parse(xmlEventReader);
            IdentityTrustInfo authInfo = new IdentityTrustInfo(appPolicy.getName());
            authInfo.add(entries);
            appPolicy.setIdentityTrustInfo(authInfo); 
         } 
         else if("application-policy".equals(elementName))
            break; 
         else throw new RuntimeException("Unknown element "  + elementName + " at location " + 
               StaxParserUtil.getLineColumnNumber(xmlEvent.getLocation())); 
      }
   }
}