From c2cf65a8032485eb1142102d744eb5a1ccb4d754 Mon Sep 17 00:00:00 2001 From: bit4woo Date: Fri, 3 May 2024 15:52:27 +0800 Subject: [PATCH] Update LineEntryMenu.java --- src/PoC/LineEntryMenu.java | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/src/PoC/LineEntryMenu.java b/src/PoC/LineEntryMenu.java index a5b8983..53892bb 100644 --- a/src/PoC/LineEntryMenu.java +++ b/src/PoC/LineEntryMenu.java @@ -10,6 +10,7 @@ import java.net.URLEncoder; import java.nio.charset.StandardCharsets; import java.util.ArrayList; +import java.util.Arrays; import java.util.List; import javax.swing.AbstractAction; @@ -185,6 +186,15 @@ public void actionPerformed(ActionEvent actionEvent) { public void actionPerformed(ActionEvent actionEvent) { LineEntry entry = lineTable.getLineTabelModel().getLineEntries().getValueAtIndex(rows[0]); String tags = entry.getTags(); + try {//用户搜索什么内容，很有可能就是要用什么PoC + String searchWord = PoCPanel.getTextFieldSearch().getText(); + List tagList = Arrays.asList(tags.split(",")); + if (tagList.contains(searchWord)) { + tags = searchWord; + } + } catch (Exception e) { + e.printStackTrace(); + } tags = getTags(tags); List targets = Commons.getLinesFromTextArea(PoCPanel.getTitleTable().getTextAreaTarget()); String Command = RunNucleiAction.genTagsCommand(targets, tags); @@ -206,6 +216,15 @@ public String getTags(String tags) { public void actionPerformed(ActionEvent actionEvent) { LineEntry entry = lineTable.getLineTabelModel().getLineEntries().getValueAtIndex(rows[0]); String tags = entry.getTags(); + try {//用户搜索什么内容，很有可能就是要用什么PoC + String searchWord = PoCPanel.getTextFieldSearch().getText(); + List tagList = Arrays.asList(tags.split(",")); + if (tagList.contains(searchWord)) { + tags = searchWord; + } + } catch (Exception e) { + e.printStackTrace(); + } tags = getTags(tags); List targets = Commons.getLinesFromTextArea(PoCPanel.getTitleTable().getTextAreaTarget()); String Command = RunNucleiAction.genTagsCommand(targets, tags);