Examples with HistoryReference org.parosproxy.paros.model.HistoryReference used on opensource projects

Example 21 with HistoryReference

use of org.parosproxy.paros.model.HistoryReference in project zaproxy by zaproxy.

the class PopupMenuHistoryReference method isEnableForComponent.

@Override
public boolean isEnableForComponent(Component invoker) {
    boolean display = false;
    if (invoker.getName() == null) {
        return false;
    }
    if (invoker.getName().equals("History Table")) {
        this.lastInvoker = Invoker.history;
        this.hrefsTableInvoker = (HistoryReferencesTable) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else if (invoker instanceof JTree && invoker.getName().equals("treeSite")) {
        this.lastInvoker = Invoker.sites;
        this.treeInvoker = (JTree) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else if (invoker.getName().equals("treeAlert")) {
        this.lastInvoker = Invoker.alerts;
        this.treeInvoker = (JTree) invoker;
        JTree tree = (JTree) invoker;
        if (tree.getLastSelectedPathComponent() != null) {
            if (tree.getSelectionCount() > 1) {
                // Note - the Alerts tree only supports single selections
                this.setEnabled(false);
            } else {
                DefaultMutableTreeNode node = (DefaultMutableTreeNode) tree.getLastSelectedPathComponent();
                if (!node.isRoot() && node.getUserObject() != null) {
                    this.setEnabled(isEnabledForHistoryReference(getSelectedHistoryReferences().get(0)));
                } else {
                    this.setEnabled(false);
                }
            }
        }
        display = true;
    } else if (invoker.getName().equals(SearchPanel.HTTP_MESSAGE_CONTAINER_NAME)) {
        this.lastInvoker = Invoker.search;
        this.hrefsTableInvoker = (HistoryReferencesTable) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else if (invoker.getName().equals(ActiveScanPanel.MESSAGE_CONTAINER_NAME)) {
        this.lastInvoker = Invoker.ascan;
        this.hrefsTableInvoker = (HistoryReferencesTable) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else if (invoker.getName().equals("HttpFuzzerResultsContentPanel")) {
        this.lastInvoker = Invoker.fuzz;
        this.hrefsTableInvoker = (HistoryReferencesTable) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else if (invoker.getName().equals("ForcedBrowseMessageContainer")) {
        this.lastInvoker = Invoker.bruteforce;
        this.hrefsTableInvoker = (HistoryReferencesTable) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else if (invoker instanceof HistoryReferencesTable) {
        this.lastInvoker = Invoker.hreftable;
        this.hrefTableInvoker = (HistoryReferencesTable) invoker;
        this.setEnabled(isEnabledForHistoryReferences(getSelectedHistoryReferences()));
        display = true;
    } else {
    // Only enable this while testing
    /*
        	log.debug("Popup " + this.getName() + 
        			" not enabled for panel " + invoker.getName() + 
        			" class " + invoker.getClass().getName());
        	*/
    }
    if (display) {
        if (this.isEnabled() && !this.isSafe() && Control.getSingleton().getMode().equals(Mode.protect)) {
            boolean inScope = true;
            Session session = Model.getSingleton().getSession();
            for (HistoryReference href : getSelectedHistoryReferences()) {
                if (!session.isInScope(href)) {
                    inScope = false;
                    break;
                }
            }
            if (!inScope) {
                // Not safe and not in scope while in protected mode
                this.setEnabled(false);
            }
        }
        return this.isEnableForInvoker(lastInvoker);
    }
    return false;
}

Example 22 with HistoryReference

use of org.parosproxy.paros.model.HistoryReference in project zaproxy by zaproxy.

the class AlertPanel method getTreeAlert.

/**
	 * This method initializes treeAlert	
	 * 	
	 * @return javax.swing.JTree	
	 */
JTree getTreeAlert() {
    if (treeAlert == null) {
        treeAlert = new JTree() {

            private static final long serialVersionUID = 1L;

            @Override
            public Point getPopupLocation(final MouseEvent event) {
                if (event != null) {
                    // Select item on right click
                    TreePath tp = treeAlert.getPathForLocation(event.getX(), event.getY());
                    if (tp != null) {
                        // to be selected
                        if (!treeAlert.getSelectionModel().isPathSelected(tp)) {
                            treeAlert.getSelectionModel().setSelectionPath(tp);
                        }
                    }
                }
                return super.getPopupLocation(event);
            }
        };
        treeAlert.setName(ALERT_TREE_PANEL_NAME);
        treeAlert.setShowsRootHandles(true);
        treeAlert.setBorder(javax.swing.BorderFactory.createEmptyBorder(0, 0, 0, 0));
        treeAlert.setComponentPopupMenu(new JPopupMenu() {

            private static final long serialVersionUID = 1L;

            @Override
            public void show(Component invoker, int x, int y) {
                final int countSelectedNodes = treeAlert.getSelectionCount();
                final ArrayList<HistoryReference> uniqueHistoryReferences = new ArrayList<>(countSelectedNodes);
                if (countSelectedNodes > 0) {
                    SortedSet<Integer> historyReferenceIdsAdded = new TreeSet<>();
                    for (TreePath path : treeAlert.getSelectionPaths()) {
                        final AlertNode node = (AlertNode) path.getLastPathComponent();
                        final Object userObject = node.getUserObject();
                        if (userObject instanceof Alert) {
                            HistoryReference historyReference = ((Alert) userObject).getHistoryRef();
                            if (historyReference != null && !historyReferenceIdsAdded.contains(Integer.valueOf(historyReference.getHistoryId()))) {
                                historyReferenceIdsAdded.add(Integer.valueOf(historyReference.getHistoryId()));
                                uniqueHistoryReferences.add(historyReference);
                            }
                        }
                    }
                    uniqueHistoryReferences.trimToSize();
                }
                SelectableHistoryReferencesContainer messageContainer = new DefaultSelectableHistoryReferencesContainer(treeAlert.getName(), treeAlert, Collections.<HistoryReference>emptyList(), uniqueHistoryReferences);
                view.getPopupMenu().show(messageContainer, x, y);
            }
        });
        treeAlert.addMouseListener(new java.awt.event.MouseAdapter() {

            @Override
            public void mouseClicked(java.awt.event.MouseEvent e) {
                if (SwingUtilities.isLeftMouseButton(e) && e.getClickCount() > 1) {
                    // Its a double click - edit the alert
                    editSelectedAlert();
                }
            }
        });
        treeAlert.addTreeSelectionListener(new javax.swing.event.TreeSelectionListener() {

            @Override
            public void valueChanged(javax.swing.event.TreeSelectionEvent e) {
                DefaultMutableTreeNode node = (DefaultMutableTreeNode) treeAlert.getLastSelectedPathComponent();
                if (node != null && node.getUserObject() != null) {
                    Object obj = node.getUserObject();
                    if (obj instanceof Alert) {
                        Alert alert = (Alert) obj;
                        setMessage(alert.getMessage(), alert.getEvidence());
                        treeAlert.requestFocusInWindow();
                        getAlertViewPanel().displayAlert(alert);
                    } else {
                        getAlertViewPanel().clearAlert();
                    }
                } else {
                    getAlertViewPanel().clearAlert();
                }
            }
        });
        treeAlert.setCellRenderer(new AlertTreeCellRenderer());
        treeAlert.setExpandsSelectedPaths(true);
    }
    return treeAlert;
}

Example 23 with HistoryReference

use of org.parosproxy.paros.model.HistoryReference in project zaproxy by zaproxy.

the class ExtensionAlert method publishAlertEvent.

private void publishAlertEvent(Alert alert, String event) {
    HistoryReference historyReference = hrefs.get(alert.getSourceHistoryId());
    if (historyReference == null) {
        historyReference = Control.getSingleton().getExtensionLoader().getExtension(ExtensionHistory.class).getHistoryReference(alert.getSourceHistoryId());
    }
    Map<String, String> map = new HashMap<>();
    map.put(AlertEventPublisher.ALERT_ID, Integer.toString(alert.getAlertId()));
    map.put(AlertEventPublisher.HISTORY_REFERENCE_ID, Integer.toString(alert.getSourceHistoryId()));
    ZAP.getEventBus().publishSyncEvent(AlertEventPublisher.getPublisher(), new Event(AlertEventPublisher.getPublisher(), event, new Target(historyReference.getSiteNode()), map));
}

Example 24 with HistoryReference

use of org.parosproxy.paros.model.HistoryReference in project zaproxy by zaproxy.

the class ExtensionAntiCSRF method hook.

@Override
public void hook(ExtensionHook extensionHook) {
    super.hook(extensionHook);
    final ExtensionHistory extensionHistory = (ExtensionHistory) Control.getSingleton().getExtensionLoader().getExtension(ExtensionHistory.NAME);
    if (extensionHistory != null) {
        historyReferenceFactory = new HistoryReferenceFactory() {

            @Override
            public HistoryReference createHistoryReference(int id) {
                return extensionHistory.getHistoryReference(id);
            }
        };
    } else {
        historyReferenceFactory = new HistoryReferenceFactory() {

            @Override
            public HistoryReference createHistoryReference(int id) throws HttpMalformedHeaderException, DatabaseException {
                return new HistoryReference(id);
            }
        };
    }
    AntiCsrfToken.setHistoryReferenceFactory(historyReferenceFactory);
    extensionHook.addSessionListener(this);
    if (getView() != null) {
        extensionHook.getHookView().addOptionPanel(getOptionsAntiCsrfPanel());
        extensionHook.getHookMenu().addPopupMenuItem(this.getPopupMenuGenerateForm());
    }
    ExtensionPassiveScan extensionPassiveScan = (ExtensionPassiveScan) Control.getSingleton().getExtensionLoader().getExtension(ExtensionPassiveScan.NAME);
    if (extensionPassiveScan != null) {
        extensionPassiveScan.addPassiveScanner(antiCsrfDetectScanner);
    }
    AntiCsrfAPI api = new AntiCsrfAPI(this);
    api.addApiOptions(getParam());
    extensionHook.addApiImplementor(api);
}

Example 25 with HistoryReference

use of org.parosproxy.paros.model.HistoryReference in project zaproxy by zaproxy.

the class ExtensionAlert method refreshAlert.

private void refreshAlert(Session session) throws DatabaseException {
    if (Constant.isLowMemoryOptionSet()) {
        return;
    }
    SiteMap siteTree = this.getModel().getSession().getSiteTree();
    TableAlert tableAlert = getModel().getDb().getTableAlert();
    // TODO this doesnt work, but should be used when its fixed :/
    //Vector<Integer> v = tableAlert.getAlertListBySession(Model.getSingleton().getSession().getSessionId());
    Vector<Integer> v = tableAlert.getAlertList();
    final ExtensionHistory extensionHistory = (ExtensionHistory) Control.getSingleton().getExtensionLoader().getExtension(ExtensionHistory.NAME);
    for (int i = 0; i < v.size(); i++) {
        int alertId = v.get(i).intValue();
        RecordAlert recAlert = tableAlert.read(alertId);
        int historyId = recAlert.getHistoryId();
        HistoryReference historyReference = null;
        if (extensionHistory != null) {
            historyReference = extensionHistory.getHistoryReference(historyId);
        }
        if (historyReference == null) {
            historyReference = this.hrefs.get(Integer.valueOf(historyId));
        }
        Alert alert;
        if (historyReference != null) {
            alert = new Alert(recAlert, historyReference);
        } else {
            alert = new Alert(recAlert);
        }
        historyReference = alert.getHistoryRef();
        if (historyReference != null) {
            // The ref can be null if hrefs are purged
            addAlertToTree(alert);
            Integer key = Integer.valueOf(historyId);
            if (!hrefs.containsKey(key)) {
                this.hrefs.put(key, alert.getHistoryRef());
            }
        }
    }
    siteTree.nodeStructureChanged((SiteNode) siteTree.getRoot());
}