From 5650500f372d9fe5ace42fd3c324a3f71951154b Mon Sep 17 00:00:00 2001
From: James Kettle <albinowax@gmail.com>
Date: Fri, 8 Feb 2019 14:12:12 +0000
Subject: [PATCH] Fix numeric sorting & rare response-duplication with empty
 responses

---
 src/RequestTable.kt          |  2 +-
 src/RequestTableModel.kt     | 11 +++++++++++
 src/ThreadedRequestEngine.kt |  1 +
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/RequestTable.kt b/src/RequestTable.kt
index 15cd29b..e970a2a 100644
--- a/src/RequestTable.kt
+++ b/src/RequestTable.kt
@@ -148,7 +148,7 @@ class RequestTable(val service: IHttpService, val handler: AttackHandler): JPane
     }
 
 
-    override fun add(req: Request) {
+    @Synchronized override fun add(req: Request) {
         model.addRequest(req)
     }
 
diff --git a/src/RequestTableModel.kt b/src/RequestTableModel.kt
index c28569b..94813f4 100644
--- a/src/RequestTableModel.kt
+++ b/src/RequestTableModel.kt
@@ -19,6 +19,17 @@ class RequestTableModel : AbstractTableModel() {
         return columns[column]
     }
 
+    override fun getColumnClass(columnIndex: Int): Class<*> {
+        return when (columnIndex) {
+            0 -> java.lang.Integer::class.java
+            1 -> String::class.java
+            2 -> java.lang.Integer::class.java
+            3 -> java.lang.Integer::class.java
+            4 -> java.lang.Integer::class.java
+            else -> throw RuntimeException()
+        }
+    }
+
     override fun getValueAt(rowIndex: Int, columnIndex: Int): Any? {
         val request = requests[rowIndex]
 
diff --git a/src/ThreadedRequestEngine.kt b/src/ThreadedRequestEngine.kt
index 127d94c..74ffde5 100644
--- a/src/ThreadedRequestEngine.kt
+++ b/src/ThreadedRequestEngine.kt
@@ -187,6 +187,7 @@ open class ThreadedRequestEngine(url: String, val threads: Int, maxQueueSize: In
 
                         if (bodyStart+4 == buffer.length) {
                             // no need to read the body
+                            buffer = ""
                         }
                         else if (contentLength != -1) {
                             val responseLength = bodyStart + contentLength + 4