org.parosproxy.paros.extension.ExtensionHook Maven / Gradle / Ivy
Show all versions of zap Show documentation
/*
*
* Paros and its related class files.
*
* Paros is an HTTP/HTTPS proxy for assessing web application security.
* Copyright (C) 2003-2004 Chinotec Technologies Company
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the Clarified Artistic License
* 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
* Clarified Artistic License for more details.
*
* You should have received a copy of the Clarified Artistic License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
// ZAP: 2012/12/19 Code Cleanup: Moved array brackets from variable name to type
// ZAP: 2012/12/20 Added listener setter for persistentConnectionListenerList.
// ZAP: 2013/01/16 Issue 453: Dynamic loading and unloading of add-ons
// ZAP: 2013/04/14 Issue 608: Rename the method ExtensionHook.addSiteMapListner to addSiteMapListener
// ZAP: 2013/05/02 Re-arranged all modifiers into Java coding standard order
// ZAP: 2014/03/23 Issue 1022: Proxy - Allow to override a proxied message
// ZAP: 2014/10/25 Issue 1062: Added scannerhook to be added by extensions.
// ZAP: 2016/04/08 Allow to add ContextDataFactory
// ZAP: 2016/05/30 Allow to add AddOnInstallationStatusListener
// ZAP: 2016/05/30 Issue 2494: ZAP Proxy is not showing the HTTP CONNECT Request in history tab
// ZAP: 2016/08/18 Allow to add ApiImplementor
// ZAP: 2017/07/25 Allow to add HttpSenderListener.
// ZAP: 2017/11/23 Add an add method for OverrideMessageProxyListener.
package org.parosproxy.paros.extension;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Vector;
import org.parosproxy.paros.common.AbstractParam;
import org.parosproxy.paros.core.proxy.ConnectRequestProxyListener;
import org.parosproxy.paros.core.proxy.OverrideMessageProxyListener;
import org.parosproxy.paros.core.proxy.ProxyListener;
import org.parosproxy.paros.core.scanner.ScannerHook;
import org.parosproxy.paros.model.Model;
import org.zaproxy.zap.PersistentConnectionListener;
import org.zaproxy.zap.extension.AddonFilesChangedListener;
import org.zaproxy.zap.extension.api.ApiImplementor;
import org.zaproxy.zap.extension.AddOnInstallationStatusListener;
import org.zaproxy.zap.model.ContextDataFactory;
import org.zaproxy.zap.network.HttpSenderListener;
import org.zaproxy.zap.view.SiteMapListener;
public class ExtensionHook {
private ExtensionHookMenu hookMenu = new ExtensionHookMenu();
private ExtensionHookView hookView = new ExtensionHookView();
private Model model = null;
private Vector optionsListenerList = new Vector<>();
private Vector proxyListenerList = new Vector<>();
/**
* The {@link OverrideMessageProxyListener}s added to this extension hook.
*
* Lazily initialised.
*
* @see #addOverrideMessageProxyListener(OverrideMessageProxyListener)
* @see #getOverrideMessageProxyListenerList()
*/
private List overrideMessageProxyListenersList;
/**
* The {@link ConnectRequestProxyListener}s added to this extension hook.
*
* Lazily initialised.
*
* @see #addConnectionRequestProxyListener(ConnectRequestProxyListener)
* @see #getConnectRequestProxyListeners()
*/
private List connectRequestProxyListeners;
private Vector sessionListenerList = new Vector<>();
private Vector optionsParamSetList = new Vector<>();
// ZAP: Added support for site map listeners
private Vector siteMapListenerList = new Vector<>();
// ZAP: Added support for Scanner Hooks
private Vector scannerHookList = new Vector<>();
private Vector persistentConnectionListenerList = new Vector<>();
private List addonFilesChangedListenerList = new ArrayList<>();
/**
* The {@link ContextDataFactory}s added to this extension hook.
*
* Lazily initialised.
*
* @see #addContextDataFactory(ContextDataFactory)
* @see #getContextDataFactories()
*/
private List contextDataFactories;
/**
* The {@link AddOnInstallationStatusListener}s added to this extension hook.
*
* Lazily initialised.
*
* @see #addAddOnInstallationStatusListener(AddOnInstallationStatusListener)
* @see #getAddOnInstallationStatusListeners()
*/
private List addOnInstallationStatusListeners;
/**
* The {@link ApiImplementor}s added to this extension hook.
*
* Lazily initialised.
*
* @see #addApiImplementor(ApiImplementor)
* @see #getApiImplementors()
*/
private List apiImplementors;
/**
* The {@link HttpSenderListener}s added to this extension hook.
*
* Lazily initialised.
*
* @see #addHttpSenderListener(HttpSenderListener)
* @see #getHttpSenderListeners()
*/
private List httpSenderListeners;
private ViewDelegate view = null;
private CommandLineArgument[] arg = new CommandLineArgument[0];
public ExtensionHook(Model model, ViewDelegate view) {
this.view = view;
this.model = model;
}
public void addOptionsChangedListener(OptionsChangedListener listener) {
optionsListenerList.add(listener);
}
public void addOptionsParamSet(AbstractParam paramSet) {
optionsParamSetList.add(paramSet);
}
public void addProxyListener(ProxyListener listener) {
proxyListenerList.add(listener);
}
/**
* Adds the given {@link ConnectRequestProxyListener} to the extension hook, to be later notified of CONNECT requests
* received by the local proxy.
*
* By default, the {@code ConnectRequestProxyListener}s added are removed from the local proxy when the extension is
* unloaded.
*
* @param listener the {@code ConnectRequestProxyListener} that will be added and then notified
* @throws IllegalArgumentException if the given {@code listener} is {@code null}.
* @since 2.5.0
*/
public void addConnectionRequestProxyListener(ConnectRequestProxyListener listener) {
if (listener == null) {
throw new IllegalArgumentException("Parameter listener must not be null.");
}
if (connectRequestProxyListeners == null) {
connectRequestProxyListeners = new ArrayList<>();
}
connectRequestProxyListeners.add(listener);
}
/**
* Gets the {@link ConnectRequestProxyListener}s added to this hook.
*
* @return an unmodifiable {@code List} containing the added {@code ConnectRequestProxyListener}s, never {@code null}.
* @since 2.5.0
*/
List getConnectRequestProxyListeners() {
if (connectRequestProxyListeners == null) {
return Collections.emptyList();
}
return Collections.unmodifiableList(connectRequestProxyListeners);
}
public void addSessionListener(SessionChangedListener listener) {
sessionListenerList.add(listener);
}
/**
* @deprecated Replaced by the method {@link #addSiteMapListener(SiteMapListener)}.
* It will be removed in a future release.
*/
@Deprecated
public void addSiteMapListner(SiteMapListener listener) {
siteMapListenerList.add(listener);
}
public void addSiteMapListener(SiteMapListener listener) {
siteMapListenerList.add(listener);
}
// ZAP: add a scanner hook
public void addScannerHook(ScannerHook hook) {
scannerHookList.add(hook);
}
public void addPersistentConnectionListener(PersistentConnectionListener listener) {
persistentConnectionListenerList.add(listener);
}
public void addCommandLine(CommandLineArgument[] arg) {
this.arg = arg;
}
public void addAddonFilesChangedListener(AddonFilesChangedListener listener) {
addonFilesChangedListenerList.add(listener);
}
/**
* Adds the given {@code listener} to the extension hook, to be later notified of changes in the installation status of the
* add-ons.
*
* @param listener the listener that will be added and then notified
* @throws IllegalArgumentException if the given {@code listener} is {@code null}.
* @since 2.5.0
*/
public void addAddOnInstallationStatusListener(AddOnInstallationStatusListener listener) {
if (listener == null) {
throw new IllegalArgumentException("Parameter listener must not be null.");
}
if (addOnInstallationStatusListeners == null) {
addOnInstallationStatusListeners = new ArrayList<>();
}
addOnInstallationStatusListeners.add(listener);
}
/**
* Gets the {@link AddOnInstallationStatusListener}s added to this hook.
*
* @return an unmodifiable {@code List} containing the added {@code AddOnInstallationStatusListener}s, never {@code null}.
* @since 2.5.0
*/
List getAddOnInstallationStatusListeners() {
if (addOnInstallationStatusListeners == null) {
return Collections.emptyList();
}
return Collections.unmodifiableList(addOnInstallationStatusListeners);
}
/**
* @return Returns the hookMenu.
*/
public ExtensionHookMenu getHookMenu() {
return hookMenu;
}
/**
* @return Returns the hookView.
*/
public ExtensionHookView getHookView() {
return hookView;
}
/**
* @return Returns the model.
*/
public Model getModel() {
return model;
}
/**
* @return Returns the optionsListenerList.
*/
public Vector getOptionsChangedListenerList() {
return optionsListenerList;
}
public Vector getOptionsParamSetList() {
return optionsParamSetList;
}
/**
* @return Returns the proxyListenerList.
*/
public Vector getProxyListenerList() {
return proxyListenerList;
}
/**
* @return Returns the sessionListenerList.
*/
public Vector getSessionListenerList() {
return sessionListenerList;
}
public Vector getSiteMapListenerList() {
return siteMapListenerList;
}
// ZAP: get all scannerhooks (used by extensionloader and the scanner)
public Vector getScannerHookList() {
return scannerHookList;
}
public Vector getPersistentConnectionListener() {
return persistentConnectionListenerList;
}
/**
* @return Returns the view.
*/
public ViewDelegate getView() {
return view;
}
public CommandLineArgument[] getCommandLineArgument() {
return arg;
}
public List getAddonFilesChangedListener() {
return addonFilesChangedListenerList;
}
/**
* Adds the given {@code overrideMessageProxyListener} to the extension hook, to be later added to the
* {@link org.parosproxy.paros.control.Proxy Proxy}.
*
* By default, the {@code OverrideMessageProxyListener}s added to this extension hook are removed from the {@code Proxy}
* when the extension is unloaded.
*
* @param overrideMessageProxyListener the {@code OverrideMessageProxyListener} that will be added to the {@code Proxy}
* @throws IllegalArgumentException if the given {@code overrideMessageProxyListener} is {@code null}.
* @since 2.7.0
*/
public void addOverrideMessageProxyListener(OverrideMessageProxyListener overrideMessageProxyListener) {
getOverrideMessageProxyListenerList().add(overrideMessageProxyListener);
}
/**
* Gets the {@link OverrideMessageProxyListener}s added to this hook.
*
* While it's possible to add the listener with this method it's not recommended, use
* {@link #addOverrideMessageProxyListener(OverrideMessageProxyListener)} whenever possible. The accessibility of this
* method might change in a future version (to package access).
*
* @return a {@code List} containing the added {@code OverrideMessageProxyListener}s, never {@code null}.
* @since 2.3.0
*/
public List getOverrideMessageProxyListenerList() {
if (overrideMessageProxyListenersList == null) {
overrideMessageProxyListenersList = new ArrayList<>();
}
return overrideMessageProxyListenersList;
}
/**
* Adds the given {@link ContextDataFactory} to the extension hook, to be later added to the {@link Model}.
*
* By default, the {@code ContextDataFactory}s added are removed from the {@code Model} when the extension is unloaded.
*
* @param contextDataFactory the {@code ContextDataFactory} that will be added to the {@code Model}
* @since 2.5.0
*/
public void addContextDataFactory(ContextDataFactory contextDataFactory) {
if (contextDataFactories == null) {
contextDataFactories = new ArrayList<>();
}
contextDataFactories.add(contextDataFactory);
}
/**
* Gets the {@link ContextDataFactory}s added to this hook.
*
* @return an unmodifiable {@code List} containing the added {@code ContextDataFactory}s, never {@code null}.
* @since 2.5.0
*/
List getContextDataFactories() {
if (contextDataFactories == null) {
return Collections.emptyList();
}
return Collections.unmodifiableList(contextDataFactories);
}
/**
* Adds the given {@code apiImplementor} to the extension hook, to be later added to the
* {@link org.zaproxy.zap.extension.api.API API}.
*
* By default, the {@code ApiImplementor}s added to this extension hook are removed from the {@code API} when the extension
* is unloaded.
*
* @param apiImplementor the ApiImplementor that will be added to the ZAP API
* @throws IllegalArgumentException if the given {@code apiImplementor} is {@code null}.
* @since 2.6.0
*/
public void addApiImplementor(ApiImplementor apiImplementor) {
if (apiImplementor == null) {
throw new IllegalArgumentException("Parameter apiImplementor must not be null.");
}
if (apiImplementors == null) {
apiImplementors = new ArrayList<>();
}
apiImplementors.add(apiImplementor);
}
/**
* Gets the {@link ApiImplementor}s added to this hook.
*
* @return an unmodifiable {@code List} containing the added {@code ApiImplementor}s, never {@code null}.
* @since 2.6.0
*/
List getApiImplementors() {
if (apiImplementors == null) {
return Collections.emptyList();
}
return Collections.unmodifiableList(apiImplementors);
}
/**
* Adds the given {@code httpSenderListener} to the extension hook, to be later added to the
* {@link org.parosproxy.paros.network.HttpSender HttpSender}.
*
* By default, the {@code HttpSenderListener}s added to this extension hook are removed from the {@code HttpSender} when the
* extension is unloaded.
*
* @param httpSenderListener the HttpSenderListener that will be added to the {@code HttpSender}
* @throws IllegalArgumentException if the given {@code httpSenderListener} is {@code null}.
* @since 2.7.0
*/
public void addHttpSenderListener(HttpSenderListener httpSenderListener) {
if (httpSenderListener == null) {
throw new IllegalArgumentException("Parameter httpSenderListener must not be null.");
}
if (httpSenderListeners == null) {
httpSenderListeners = new ArrayList<>();
}
httpSenderListeners.add(httpSenderListener);
}
/**
* Gets the {@link HttpSenderListener}s added to this hook.
*
* @return an unmodifiable {@code List} containing the added {@code HttpSenderListener}s, never {@code null}.
* @since 2.7.0
*/
List getHttpSenderListeners() {
if (httpSenderListeners == null) {
return Collections.emptyList();
}
return Collections.unmodifiableList(httpSenderListeners);
}
}