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);