org.parosproxy.paros.extension.history.PopupMenuExportMessage 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/03/15 Changed the method initialize to check if "fw" is not null before closing.
// Made the "log" final.
// ZAP: 2012/04/25 Added @Override annotation to all appropriate methods.
// ZAP: 2012/07/29 Issue 43: Cleaned up access to ExtensionHistory UI
// ZAP: 2012/11/01 Changed to load the HttpMessage from the database only once.
// ZAP: 2013/03/03 Issue 546: Remove all template Javadoc comments
// ZAP: 2014/03/23 Changed to a JMenuItem.
// ZAP: 2016/04/05 Issue 2458: Fix xlint warning messages
// ZAP: 2016/07/25 Remove String constructor (unused/unnecessary)
// ZAP: 2018/03/29 Use FileNameExtensionFilter.
// 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.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.Writer;
import java.util.List;
import java.util.Locale;
import javax.swing.JFileChooser;
import javax.swing.JMenuItem;
import javax.swing.JOptionPane;
import javax.swing.filechooser.FileNameExtensionFilter;
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 PopupMenuExportMessage extends JMenuItem {
private static final long serialVersionUID = 1L;
private static final Logger log = Logger.getLogger(PopupMenuExportMessage.class);
private static final String CRLF = "\r\n";
private ExtensionHistory extension = null;
public PopupMenuExportMessage() {
super(Constant.messages.getString("history.export.messages.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.messages.select.warning"));
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;
}
}
BufferedWriter fw = null;
try {
fw = new BufferedWriter(new FileWriter(file, isAppend));
for (HistoryReference href : hrefs) {
exportHistory(href, fw);
}
} catch (Exception e1) {
extension
.getView()
.showWarningDialog(
Constant.messages.getString("file.save.error")
+ file.getAbsolutePath()
+ ".");
// ZAP: Log exceptions
log.warn(e1.getMessage(), e1);
} finally {
try {
if (fw != null) {
fw.close();
}
} catch (Exception e2) {
// ZAP: Log exceptions
log.warn(e2.getMessage(), e2);
}
}
}
});
}
void setExtension(ExtensionHistory extension) {
this.extension = extension;
}
private void exportHistory(HistoryReference ref, Writer writer) {
if (ref == null) {
return;
}
String s = null;
try {
// ZAP: Changed to load the HttpMessage from the database only once.
HttpMessage msg = ref.getHttpMessage();
writer.write("==== " + ref.getHistoryId() + " ==========" + CRLF);
s = msg.getRequestHeader().toString();
writer.write(s);
s = msg.getRequestBody().toString();
writer.write(s);
if (!s.endsWith(CRLF)) {
writer.write(CRLF);
}
if (!msg.getResponseHeader().isEmpty()) {
s = msg.getResponseHeader().toString();
writer.write(s);
s = msg.getResponseBody().toString();
writer.write(s);
if (!s.endsWith(CRLF)) {
writer.write(CRLF);
}
}
} catch (Exception e) {
// ZAP: Log exceptions
log.warn(e.getMessage(), e);
}
}
private File getOutputFile() {
JFileChooser chooser =
new WritableFileChooser(extension.getModel().getOptionsParam().getUserDirectory());
chooser.setFileFilter(
new FileNameExtensionFilter(
Constant.messages.getString("file.format.ascii"), "txt"));
File file = null;
int rc = chooser.showSaveDialog(extension.getView().getMainFrame());
if (rc == JFileChooser.APPROVE_OPTION) {
file = chooser.getSelectedFile();
if (file == null) {
return file;
}
String fileName = file.getAbsolutePath();
if (!fileName.toLowerCase(Locale.ROOT).endsWith(".txt")) {
fileName += ".txt";
file = new File(fileName);
}
return file;
}
return file;
}
}