org.parosproxy.paros.extension.history.PopupMenuExportResponse 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.
/*
*
* Paros and its related class files.
*
* Paros is an HTTP/HTTPS proxy for assessing web application security.
* Copyright (C) 2005 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/01/12 Reflected the rename of the class ExtensionPopupMenu to
// ExtensionPopupMenuItem.
// ZAP: 2012/04/25 Added @Override annotation to all appropriate methods.
// ZAP: 2012/07/29 Issue 43: Cleaned up access to ExtensionHistory UI
// ZAP: 2013/03/03 Issue 546: Remove all template Javadoc comments
// ZAP: 2014/03/23 Changed to a JMenuItem.
// ZAP: 2016/07/25 Remove String constructor (unused/unnecessary)
// ZAP: 2018/08/15 Added null check
// ZAP: 2019/06/01 Normalise line endings.
// ZAP: 2019/06/05 Normalise format/style.
// ZAP: 2019/11/05 Use WritableFileChooser for saves.
package org.parosproxy.paros.extension.history;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.util.List;
import javax.swing.JFileChooser;
import javax.swing.JMenuItem;
import javax.swing.JOptionPane;
import org.apache.log4j.Logger;
import org.parosproxy.paros.Constant;
import org.parosproxy.paros.model.HistoryReference;
import org.parosproxy.paros.network.HttpMessage;
import org.zaproxy.zap.view.widgets.WritableFileChooser;
public class PopupMenuExportResponse extends JMenuItem {
private static final long serialVersionUID = 1L;
private static final String NEWLINE = System.lineSeparator();
// ZAP: Added logger
private static Logger log = Logger.getLogger(PopupMenuExportResponse.class);
private ExtensionHistory extension = null;
public PopupMenuExportResponse() {
super(Constant.messages.getString("history.export.response.popup")); // ZAP: i18n
this.addActionListener(
new java.awt.event.ActionListener() {
@Override
public void actionPerformed(java.awt.event.ActionEvent e) {
List hrefs = extension.getSelectedHistoryReferences();
if (hrefs.size() == 0) {
extension
.getView()
.showWarningDialog(
Constant.messages.getString(
"history.export.response.select.warning")); // ZAP: i18n
return;
}
File file = getOutputFile();
if (file == null) {
return;
}
boolean isAppend = true;
if (file.exists()) {
int rc =
extension
.getView()
.showYesNoCancelDialog(
Constant.messages.getString(
"file.overwrite.warning"));
if (rc == JOptionPane.CANCEL_OPTION) {
return;
} else if (rc == JOptionPane.YES_OPTION) {
isAppend = false;
}
}
try (BufferedOutputStream bos =
new BufferedOutputStream(new FileOutputStream(file, isAppend)); ) {
for (HistoryReference href : hrefs) {
HttpMessage msg = null;
msg = href.getHttpMessage();
exportHistory(msg, bos);
}
} catch (Exception e1) {
extension
.getView()
.showWarningDialog(
Constant.messages.getString("file.save.error")
+ file.getAbsolutePath()
+ ".");
// ZAP: Log exceptions
log.warn(e1.getMessage(), e1);
}
}
});
}
void setExtension(ExtensionHistory extension) {
this.extension = extension;
}
private void exportHistory(HttpMessage msg, BufferedOutputStream bos) {
try {
if (!msg.getResponseHeader().isEmpty()) {
String boundary =
NEWLINE
+ "==== "
+ msg.getHistoryRef().getHistoryId()
+ " =========="
+ NEWLINE;
bos.write(boundary.getBytes());
bos.write(msg.getResponseBody().getBytes());
}
} catch (Exception e) {
// ZAP: Log exceptions
log.warn(e.getMessage(), e);
}
}
private File getOutputFile() {
String filename = "untitled.txt";
JFileChooser chooser =
new WritableFileChooser(extension.getModel().getOptionsParam().getUserDirectory());
if (filename.length() > 0) {
chooser.setSelectedFile(new File(filename));
}
File file = null;
int rc = chooser.showSaveDialog(extension.getView().getMainFrame());
if (rc == JFileChooser.APPROVE_OPTION) {
file = chooser.getSelectedFile();
if (file == null) {
return file;
}
return file;
}
return file;
}
}