org.zaproxy.zap.extension.ext.OptionsExtensionTableModel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zap Show documentation

The Zed Attack Proxy (ZAP) is an easy to use integrated penetration testing tool for finding vulnerabilities in web applications. It is designed to be used by people with a wide range of security experience and as such is ideal for developers and functional testers who are new to penetration testing. ZAP provides automated scanners as well as a set of tools that allow you to find security vulnerabilities manually.

There is a newer version: 2.15.0

Show newest version

/*
 * Zed Attack Proxy (ZAP) and its related class files.
 * 
 * ZAP is an HTTP/HTTPS proxy for assessing web application security.
 * 
 * Copyright 2011 ZAP development team
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); 
 * you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at 
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0 
 *   
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
 * See the License for the specific language governing permissions and 
 * limitations under the License. 
 */

package org.zaproxy.zap.extension.ext;

import java.util.Collections;
import java.util.Comparator;
import java.util.List;

import javax.swing.table.AbstractTableModel;

import org.apache.log4j.Logger;
import org.parosproxy.paros.Constant;
import org.parosproxy.paros.extension.Extension;
import org.zaproxy.zap.control.ExtensionFactory;

public class OptionsExtensionTableModel extends AbstractTableModel {

	private static final long serialVersionUID = 1L;

	private static final String[] columnNames = {
				Constant.messages.getString("options.ext.label.enabled"),
				Constant.messages.getString("options.ext.label.core"),
				Constant.messages.getString("options.ext.label.author"),
				Constant.messages.getString("options.ext.label.extension")};
    
    private List extensions = ExtensionFactory.getAllExtensions();
    
    private static Logger log = Logger.getLogger(OptionsExtensionTableModel.class);

    public OptionsExtensionTableModel() {
        super();
        // Sort extensions by name
        Collections.sort(extensions, new Comparator() {

			@Override
			public int compare(Extension ext0, Extension ext1) {
				if (ext0 == null || ext1 == null) {
					return 0;
				}
				return ext0.getUIName().compareTo(ext1.getUIName());
			}});
    }

    @Override
    public int getColumnCount() {
        return columnNames.length;
    }

    @Override
    public int getRowCount() {
        return extensions.size();
    }

    @Override
    public Object getValueAt(int row, int col) {
        Extension ext = getExtension(row);
        if (ext != null) {
        	try {
				switch (col) {
				case 0:	return ext.isEnabled();
				case 1:
					if (ext.isCore()) {
						return Constant.messages.getString("options.ext.label.iscore");
					}
					return "";
				case 2: return ext.getAuthor();
				case 3: return ext.getUIName();
				}
			} catch (Exception e) {
				log.error("Failed on extension " + ext.getName(), e);
			}
        }
        return null;
    }
    
    @Override
    public boolean isCellEditable(int rowIndex, int columnIndex) {
    	if (columnIndex == 0) {
    		// Dont allow enabled core extensions to be edited via the UI (can edit the config file directly;)
    		if (getExtension(rowIndex).isCore() && getExtension(rowIndex).isEnabled()) {
    			return false;
    		}
    		// Check dependencies
    		List> deps = getExtension(rowIndex).getDependencies();
    		for (Classdep : deps) {
    			Extension ext = getExtension(dep);
    			if (ext == null || ! ext.isEnabled()) {
    				return false;
    			}
    		}
    		return true;
    	}
        return false;
    }
    
    private Extension getExtension(Class c) {
		for (Extension ext: extensions) {
			if (ext.getClass().equals(c)) {
				return ext;
			}
		}
    	return null;
    }
    
    @Override
    public void setValueAt(Object value, int row, int col) {
    	if (col == 0) {
    		getExtension(row).setEnabled((Boolean) value);
            fireTableCellUpdated(row, col);
    		// En/Disable dependencies
    		enableDependants(getExtension(row), (Boolean) value);
    	}
    }

    private void enableDependants(Extension extension, Boolean enabled) {
    	int row = 0;
		for (Extension ext: extensions) {
			if (ext.getDependencies().contains(extension.getClass())) {
				ext.setEnabled(enabled);
				this.fireTableCellUpdated(row, 0);
				enableDependants(ext, enabled); 
			}
			row++;
		}
	}

	@Override
	public String getColumnName(int col) {
        return columnNames[col];
    }
    
	@Override
	public Class getColumnClass(int c) {
    	if (c == 0) {
    		return Boolean.class;
    	}
        return String.class;
        
    }
	
	protected Extension getExtension (int row) {
        return  extensions.get(row);
	}

	protected Extension getExtension (String name) {
		for (Extension ext : extensions) {
			if (ext.getName().equals(name)) {
				return ext;
			}
		}
        return  null;
	}

	protected List getExtensions() {
		return extensions;
	}
    
}