Cavemanon/haproxy-protection
8
Fork 0
mirror of https://gitgud.io/fatchan/haproxy-protection.git synced 2025-05-09 02:05:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjust lua get_server_names

Browse Source
This commit is contained in:
Thomas Lynch
2023-09-18 23:14:32 +10:00
parent bf3e2571c1
commit 2f9823bf51
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/lua/scripts/register-bot-check.lua
View File

@ -1,13 +1,16 @@
package.path = package.path .. "./?.lua;/etc/haproxy/scripts/?.lua;/etc/haproxy/libs/?.lua" package.path = package.path .. "./?.lua;/etc/haproxy/scripts/?.lua;/etc/haproxy/libs/?.lua"
local bot_check = require("bot-check") local bot_check = require("bot-check")
local utils = require("utils")
local backends_map = Map.new('/etc/haproxy/map/backends.map', Map._str) local backends_map = Map.new('/etc/haproxy/map/backends.map', Map._str)
function get_server_names(txn) function get_server_names(txn)
local key = txn.sf:hdr("Host") local key = txn.sf:hdr("Host")
local value = backends_map:lookup(key or "") local value = backends_map:lookup(key or "")
if value ~= nil then if value ~= nil then
return value local vals = utils.split(value, ",")
-- todo: something smarter?
return vals[math.random(#vals)]
else else
return "" return ""
end end