org.zaproxy.zap.extension.ascan.ScanProgressTableModel Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of zap Show documentation
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.
/*
* 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.ascan;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import javax.swing.table.AbstractTableModel;
import org.parosproxy.paros.Constant;
import org.parosproxy.paros.core.scanner.HostProcess;
import org.parosproxy.paros.core.scanner.Plugin;
public class ScanProgressTableModel extends AbstractTableModel {
private static final long serialVersionUID = 1L;
private static final String[] columnNames = {
"",
Constant.messages.getString("ascan.policy.table.strength"),
Constant.messages.getString("ascan.progress.table.progress"),
Constant.messages.getString("ascan.progress.table.time"),
Constant.messages.getString("ascan.progress.table.reqs"),
Constant.messages.getString("ascan.progress.table.alerts"),
Constant.messages.getString("ascan.progress.table.status"),
};
private HostProcess hp;
private List values;
private ScanProgressActionIcon focusedAction;
private String totRequests;
private String totTime;
public ScanProgressTableModel() {
super();
values = new ArrayList();
focusedAction = null;
}
/** @return */
@Override
public int getColumnCount() {
return columnNames.length;
}
/** @return */
@Override
public int getRowCount() {
if (values == null) {
return 0;
}
// Add other 5 rows for other info shown.
return values.size() + 5;
}
/**
* @param row
* @param col
* @return
*/
@Override
public Object getValueAt(int row, int col) {
// 1st row is for the Analyser, 2nd row is empty (for separation with the plugins), 3rd for
// Plugin label.
if (row == 0) {
switch (col) {
case 0:
return Constant.messages.getString("ascan.progress.table.analyser");
case 3:
return hp != null ? getElapsedTimeLabel(hp.getAnalyser().getRunningTime()) : "";
case 4:
return hp != null ? String.valueOf(hp.getAnalyser().getRequestCount()) : "";
default:
return null;
}
} else if (row == 1) {
return null;
} else if (row == 2) {
return col == 0 ? Constant.messages.getString("ascan.progress.table.name") : null;
}
// Adjust row for the plugin checks.
row -= 3;
// First check if we're showing the plugin status list
if (row < values.size()) {
// It's an entry so show the correct values
final ScanProgressItem item = values.get(row);
switch (col) {
case 0:
return item.getNameLabel();
case 1:
return item.getAttackStrenghtLabel();
case 2:
if (item.isCompleted() || item.isRunning() || item.isSkipped()) {
return item;
} else {
return null;
}
case 3:
return getElapsedTimeLabel(item.getElapsedTime());
case 4:
return item.getReqCount();
case 5:
return item.getAlertCount();
case 6:
return item.getProgressAction();
default:
return null;
}
// We're in the summary "region", first print an empty
// line and then two line of summary (tot time and to requests)
// Maybe could be done in a better way, for example using
// a dedicated panel positioned on top/bottom of the dialog
} else if (row == values.size()) {
// The first line after values should be empty
return null;
} else if (row == (values.size() + 1)) {
// The second line after values should contains the totals
switch (col) {
case 0:
return Constant.messages.getString("ascan.progress.label.totals");
case 3:
return totTime;
case 4:
return totRequests;
case 5:
return hp != null ? hp.getAlertCount() : 0;
default:
return null;
}
}
return null;
}
/**
* @param rowIndex
* @param columnIndex
* @return
*/
@Override
public boolean isCellEditable(int rowIndex, int columnIndex) {
return false;
}
/**
* @param col
* @return
*/
@Override
public String getColumnName(int col) {
return columnNames[col];
}
/**
* @param c
* @return
*/
@Override
public Class getColumnClass(int c) {
switch (c) {
case 0:
return String.class;
case 1:
return String.class;
case 2:
return ScanProgressItem.class;
case 3:
return String.class;
case 4:
return ScanProgressActionIcon.class;
}
return null;
}
/** @return */
public List getValues() {
return values;
}
/** @param actionIcon */
public void setFocusedAction(ScanProgressActionIcon actionIcon) {
focusedAction = actionIcon;
}
/** @return */
public ScanProgressActionIcon getFocusedAction() {
return focusedAction;
}
/** @param scan */
public void updateValues(ActiveScan scan, HostProcess hp) {
setHostProcess(hp);
// Update total elapsed time and request count
Date end = (scan.getTimeFinished() == null) ? new Date() : scan.getTimeFinished();
long elapsed = end.getTime() - scan.getTimeStarted().getTime();
totTime = getElapsedTimeLabel(elapsed);
totRequests = Integer.toString(scan.getTotalRequests());
this.fireTableDataChanged();
}
private void setHostProcess(HostProcess hp) {
if (this.hp == hp) {
for (ScanProgressItem spi : values) {
spi.refresh();
}
return;
}
this.hp = hp;
values.clear();
for (Plugin plugin : hp.getCompleted()) {
values.add(new ScanProgressItem(hp, plugin, ScanProgressItem.STATUS_COMPLETED));
}
for (Plugin plugin : hp.getRunning()) {
values.add(new ScanProgressItem(hp, plugin, ScanProgressItem.STATUS_RUNNING));
}
for (Plugin plugin : hp.getPending()) {
values.add(new ScanProgressItem(hp, plugin, ScanProgressItem.STATUS_PENDING));
}
}
/**
* Inner method for elapsed time label formatting
*
* @param elapsed the time in milliseconds
* @return the label with the elapsed time in readable format
*/
private String getElapsedTimeLabel(long elapsed) {
return (elapsed >= 0)
? String.format(
"%02d:%02d.%03d",
elapsed / 60000, (elapsed % 60000) / 1000, (elapsed % 1000))
: null;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy