IptablesChainComponent.java

/*
 * RHQ Management Platform
 * Copyright (C) 2005-2009 Red Hat, Inc.
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2, as
 * published by the Free Software Foundation, and/or the GNU Lesser
 * General Public License, version 2.1, also as published by the Free
 * Software Foundation.
 *
 * 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 and the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU General Public License
 * and the GNU Lesser General Public License along with this program;
 * if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

package org.rhq.plugins.iptables;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.rhq.augeas.AugeasProxy;
import org.rhq.augeas.node.AugeasNode;
import org.rhq.augeas.tree.AugeasTree;
import org.rhq.core.domain.configuration.Configuration;
import org.rhq.core.domain.configuration.definition.ConfigurationDefinition;
import org.rhq.core.domain.measurement.AvailabilityType;
import org.rhq.core.pluginapi.configuration.ConfigurationFacet;
import org.rhq.core.pluginapi.configuration.ConfigurationUpdateReport;
import org.rhq.core.pluginapi.inventory.InvalidPluginConfigurationException;
import org.rhq.core.pluginapi.inventory.ResourceContext;
/**
 * 
 * @author Filip Drabek
 *
 */
public class IptablesChainComponent implements AugeasRHQComponent<IptablesTableComponent>, ConfigurationFacet{

       private String tableName;
       private String chainName;
       private AugeasTree augeasTree;
       private ResourceContext<IptablesTableComponent> context;
       private IptablesTableComponent parentComponent;
       private final Log log = LogFactory.getLog(this.getClass());
       
       public void start(ResourceContext<IptablesTableComponent> context)
                     throws InvalidPluginConfigurationException, Exception {
              this.context = context;
              this.parentComponent = context.getParentResourceComponent();
              this.chainName = context.getResourceKey();
              this.tableName= context.getParentResourceComponent().getTableName();
       }

       public void stop() {
              
       }

       public AvailabilityType getAvailability() {
              return AvailabilityType.UP;
       }

       public Configuration loadResourceConfiguration() throws Exception {
              augeasTree = getAugeasTree();
              
               ConfigurationDefinition resourceConfigDef = this.context.getResourceType()
          .getResourceConfigurationDefinition();
        
              IptablesConfigTransform trans = new IptablesConfigTransform(augeasTree);
              List<AugeasNode> nodes = getIptablesChainNode(chainName);
              Configuration config= null;
              if (!nodes.isEmpty())
                config = trans.transform(nodes, resourceConfigDef);

              return config;
       }

       public void updateResourceConfiguration(ConfigurationUpdateReport config) {
              IptablesConfigTransform trans = new IptablesConfigTransform(augeasTree);
              trans.updateAugeas(config.getConfiguration(), new ArrayList<AugeasNode>());
       }
       
       public AugeasProxy getAugeasComponent() throws Exception {
              return parentComponent.getAugeasComponent();
       }

       public AugeasTree getAugeasTree() throws Exception {
              return parentComponent.getAugeasTree();
       }
       
       private List<AugeasNode> getIptablesChainNode(String chainName) throws Exception{
              if (augeasTree == null)
                     return null;
              String expr = File.separatorChar+tableName+File.separatorChar+chainName;
              
              List<AugeasNode> nodes = augeasTree.matchRelative(augeasTree.getRootNode(), expr);
              
          return nodes;
       }

}