PassiveScannerList.java

/*
 * Zed Attack Proxy (ZAP) and its related class files.
 * 
 * ZAP is an HTTP/HTTPS proxy for assessing web application security.
 * 
 * Copyright 2013 the 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.pscan;

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

import org.apache.log4j.Logger;
import org.zaproxy.zap.extension.pscan.scanner.RegexAutoTagScanner;

public class PassiveScannerList {

	private static final Logger logger = Logger.getLogger(PassiveScannerList.class);

	private List<PassiveScanner> passiveScanners = new ArrayList<>();
	private Set<String> scannerNames = new HashSet<>();

	protected boolean add (PassiveScanner scanner) {
		if (scannerNames.contains(scanner.getName())) {
			// Prevent duplicates, log error?
			return false;
		}
		passiveScanners.add(scanner);
		scannerNames.add(scanner.getName());

		return true;
	}
	
	protected List<PassiveScanner> list () {
		return this.passiveScanners;
	}
	
	public void setAutoTagScanners(List<RegexAutoTagScanner> autoTagScanners) {
		List<PassiveScanner> tempScanners = new ArrayList<>(passiveScanners.size() + autoTagScanners.size());
        
        for (Iterator<PassiveScanner> it = passiveScanners.iterator(); it.hasNext();) {
            PassiveScanner scanner = it.next();
            if (!(scanner instanceof RegexAutoTagScanner)) {
                tempScanners.add(scanner);
            } else {
                this.scannerNames.remove(scanner.getName());
            }
        }
        
        for (Iterator<RegexAutoTagScanner> it = autoTagScanners.iterator(); it.hasNext();) {
            PassiveScanner scanner = it.next();
            if (scannerNames.contains(scanner.getName())) {
            	logger.error("Duplicate passive scanner name: " + scanner.getName());
            } else {
            	tempScanners.add(scanner);
            	scannerNames.add(scanner.getName());
            }
        }
        
        this.passiveScanners = tempScanners;
    }

	public PassiveScanner removeScanner(String className) {
		for (PassiveScanner scanner : passiveScanners) {
			if (scanner.getClass().getName().equals(className)) {
				scannerNames.remove(scanner.getName());
				passiveScanners.remove(scanner);
				return scanner;
			}
		}
		return null;
	}

}