/* * Zed Attack Proxy (ZAP) and its related class files. * * ZAP is an HTTP/HTTPS proxy for assessing web application security. * * Copyright 2014 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.view.popup; import java.util.List; import org.parosproxy.paros.Constant; import org.parosproxy.paros.db.DatabaseException; import org.parosproxy.paros.model.HistoryReference; import org.parosproxy.paros.model.Model; import org.parosproxy.paros.model.Session; import org.parosproxy.paros.model.SiteNode; import org.parosproxy.paros.view.View; import org.zaproxy.zap.model.Context; import org.zaproxy.zap.model.StructuralSiteNode; import org.zaproxy.zap.view.ContextIncludePanel; /** * @since 2.3.0 */ public class PopupMenuItemIncludeInContext extends PopupMenuItemSiteNodeContainer { private static final long serialVersionUID = 990419495607725846L; protected Context context; /** * This method initializes * */ public PopupMenuItemIncludeInContext() { super(Constant.messages.getString("context.new.title"), true); this.context = null; this.setPrecedeWithSeparator(true); } public PopupMenuItemIncludeInContext(Context context) { super(context.getName(), true); this.context = context; } @Override public String getParentMenuName() { return Constant.messages.getString("context.include.popup"); } @Override public boolean isSubMenu() { return true; } @Override public void performAction(SiteNode sn) { try { performAction(sn.getNodeName(), new StructuralSiteNode(sn).getRegexPattern()); } catch (DatabaseException e) { // Ignore } } protected void performAction(String name, String url) { if (context == null) { Session session = Model.getSingleton().getSession(); context = session.getNewContext(name); recreateUISharedContexts(session); } Context uiSharedContext = View.getSingleton().getSessionDialog().getUISharedContext(context.getIndex()); uiSharedContext.addIncludeInContextRegex(url); } @Override public void performHistoryReferenceActions(List<HistoryReference> hrefs) { Session session = Model.getSingleton().getSession(); if (context != null) { recreateUISharedContexts(session); } super.performHistoryReferenceActions(hrefs); // Show the session dialog without recreating UI Shared contexts View.getSingleton().showSessionDialog(session, ContextIncludePanel.getPanelName(context.getIndex()), false); } private void recreateUISharedContexts(Session session) { // Manually create the UI shared contexts so any modifications are done // on an UI shared Context, so changes can be undone by pressing Cancel View.getSingleton().getSessionDialog().recreateUISharedContexts(session); } @Override public boolean isButtonEnabledForSiteNode(SiteNode sn) { if (context == null) { // New context, if it doesn't already exist one with same name. return Model.getSingleton().getSession().getContext(sn.getNodeName()) == null; } if (context.isIncluded(sn) || context.isExcluded(sn)) { // Either explicitly included or excluded, so would have to change that regex in a non // trivial way to include! return false; } return true; } @Override public boolean isSafe() { return true; } }