From 5230c3081ab6a4f2f1c68e29e97a1f5839df824b Mon Sep 17 00:00:00 2001
From: "Kirill A. Korinsky" <kirill@korins.ky>
Date: Thu, 5 Jan 2023 17:37:31 +0100
Subject: [PATCH] Allow to send chunked request from lua

Let assume that we have a lua script which calls `wrk.format` with both
headers and body, with`headers["Transfer-Encoding"] = "chunked"`.

That attempt fails with error like: `unexpected content-length header at`

Closes: https://github.com/wg/wrk/issues/503
---
 src/wrk.lua | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/wrk.lua b/src/wrk.lua
index 0229fdfe..587c3084 100644
--- a/src/wrk.lua
+++ b/src/wrk.lua
@@ -58,7 +58,9 @@ function wrk.format(method, path, headers, body)
       headers["Host"] = wrk.headers["Host"]
    end
 
-   headers["Content-Length"] = body and string.len(body)
+   if not headers["Transfer-Encoding"] then
+      headers["Content-Length"] = body and string.len(body)
+   end
 
    s[1] = string.format("%s %s HTTP/1.1", method, path)
    for name, value in pairs(headers) do
@@ -66,7 +68,15 @@ function wrk.format(method, path, headers, body)
    end
 
    s[#s+1] = ""
-   s[#s+1] = body or ""
+   if headers["Transfer-Encoding"] and body then
+      s[#s+1] = string.format("%x", string.len(body))
+      s[#s+1] = body
+      s[#s+1] = "0"
+      s[#s+1] = ""
+      s[#s+1] = ""
+   else
+      s[#s+1] = body or ""
+   end
 
    return table.concat(s, "\r\n")
 end