update

bit4woo · May 16, 2024 · ee02763 · ee02763
1 parent 0a28498
commit ee02763
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 7 deletions.
diff --git a/src/messageTab/Info/InfoTab.java b/src/messageTab/Info/InfoTab.java
@@ -1,9 +1,11 @@
 package messageTab.Info;
 
 import java.awt.Component;
+import java.util.ArrayList;
 import java.util.List;
 
 import javax.swing.JPanel;
+import javax.swing.JTable;
 import javax.swing.SwingWorker;
 
 import com.bit4woo.utilbox.utils.ByteArrayUtils;
@@ -115,11 +117,30 @@ public boolean isModified()
 	{
 		return false;
 	}
-
+
+	/**
+	 * ctrl+c复制数据逻辑会调用这个函数
+	 */
 	@Override
 	public byte[] getSelectedData()
-	{
-		return null;//TODO
+	{	
+		JTable table = ((InfoPanel)panel).getTable();
+		int[] rows = table.getSelectedRows();
+		int[] columns = table.getSelectedColumns();
+		List<String> result = new ArrayList<>();
+		for (int row:rows) {
+			List<String> line = new ArrayList<>();
+			for (int column:columns) {
+				try {
+					String value = (String)table.getValueAt(row, column);
+					line.add(value);
+				} catch (Exception e) {
+					//e.printStackTrace();
+				}
+			}
+			result.add(String.join(" ", line));
+		}
+		return String.join(" ", result).getBytes();
 	}
 
 

diff --git a/src/messageTab/Info/InfoTable.java b/src/messageTab/Info/InfoTable.java
@@ -1,7 +1,5 @@
 package messageTab.Info;
 
-import static config.ConfigTableModel.titles;
-
 import java.awt.Color;
 import java.awt.Font;
 import java.awt.FontMetrics;
@@ -28,8 +26,8 @@ public class InfoTable extends JTable {
      */
     private static final long serialVersionUID = 1L;
 
-    public static final String[] headers = {"Value", "Type"};
-
+    //public static final String[] headers = {"Value", "Type"};
+    public static final String[] headers = {"Value"};
 
     public InfoTable(InfoTableModel tableModel) {
         super(tableModel);
@@ -127,12 +125,17 @@ public void mouseClicked(MouseEvent e) {
                                 if (url.toLowerCase().startsWith("http://") || url.toLowerCase().startsWith("https://")) {
                                 	String browserPath = BurpExtender.getConfigTableModel().getConfigValueByKey("browserPath");
                                     SystemUtils.browserOpen(url, browserPath);
+                                    return;
                                 }
                         	}
                         } catch (Exception e1) {
                             e1.printStackTrace(BurpExtender.getStderr());
                         }
                     }
+
+                    //默认行为，复制到剪切板
+                    String value = (String) getValueAt(row, column);
+                    SystemUtils.writeToClipboard(value);
                 }
             }