Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
71 changes: 52 additions & 19 deletions tests/wrk_benchmark_announce.lua
Original file line number Diff line number Diff line change
@@ -1,34 +1,67 @@
-- else the randomness would be the same every run
math.randomseed(os.time())

local charset = "0123456789ABCDEF"

function hex_to_char(hex)
local n = tonumber(hex, 16)
local f = string.char(n)
return f
end

function hex_string_to_char_string(hex)
local ret = {}
local r
for i = 0, 19 do
local x = i * 2
r = hex:sub(x+1, x+2)
local f = hex_to_char(r)
table.insert(ret, f)
end
return table.concat(ret)
end

function url_encode(str)
str = string.gsub (str, "([^0-9a-zA-Z !'()*._~-])", -- locale independent
function (c) return string.format ("%%%02X", string.byte(c)) end)
str = string.gsub (str, " ", "+")
return str
end

function gen_hex_string(length)
local ret = {}
local r
for i = 1, length do
r = math.random(1, #charset)
table.insert(ret, charset:sub(r, r))
end
return table.concat(ret)
end

function random_info_hash()
local hexString = gen_hex_string(40)
local str = hex_string_to_char_string(hexString)
return url_encode(str)
end

function generate_unique_info_hashes(size)
local result = {}
local seen = {}

for i = 0, size - 1 do
local bytes = {}
bytes[1] = i & 0xFF
bytes[2] = (i >> 8) & 0xFF
bytes[3] = (i >> 16) & 0xFF
bytes[4] = (i >> 24) & 0xFF

local info_hash = bytes
local key = table.concat(info_hash, ",")

if not seen[key] then
table.insert(result, info_hash)
seen[key] = true
end

for i = 1, size do
result[i] = random_info_hash()
end

return result
end

info_hashes = generate_unique_info_hashes(10000000)
info_hashes = generate_unique_info_hashes(5000000)

index = 0
index = 1

-- the request function that will run at each request
request = function()
path = "/announce?info_hash=" .. info_hashes[index] .. "&peer_id=-lt0D80-a%D4%10%19%99%A6yh%9A%E1%CD%96&port=54434&uploaded=885&downloaded=0&left=0&corrupt=0&key=A78381BD&numwant=200&compact=1&no_peer_id=1&supportcrypto=1&redundant=0"
index += 1
index = index + 1
headers = {}
headers["X-Forwarded-For"] = "1.1.1.1"
return wrk.format("GET", path, headers)
Expand Down