如何扩展Nginx Lua代码实现0-9数字与A-Z字母加密？

Hey there! Let's fix up your Lua code to add support for A-Z encryption properly. Your initial idea is on the right track, but there are a couple of Lua-specific details you missed that would break the code otherwise. Let's walk through the fixes and a clean implementation:

Key Issues in Your Initial Approach #

1. Table Index Syntax: In Lua, bf[a] tries to access a variable named a (which doesn't exist) instead of the string key "a". You need to use string literals like bf['A'] for letter keys.
2. Invalid Loop Logic: for i=0, 9 and a, z do isn't valid Lua syntax. We need separate (or combined) loops to iterate over digits and letters properly.

Corrected Implementation #

Here's a working version that supports both 0-9 and A-Z, with cleaner structure:

content_by_lua_block {
    local bf = {}
    -- 0-9 Digit Encryption Maps (kept as your original code)
    bf['0'] = '(((_<<_)<<_)'
    bf['1'] = '(({}>[])-(()>[]))'
    bf['2'] = '(({}>[])-(()>[]))])'
    bf['3'] = '(({}>[])-(()>[]))])*'
    bf['4'] = '(({}>[])-(()>[]))])*(()>[])'
    bf['5'] = '(({}>[])-(()>[]))])*(()>[])%'
    bf['6'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_'
    bf['7'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_)<<_'
    bf['8'] = '(()>[]))])*(()>[])%(((_<<_)<<_)+((_<<_)*_))'
    bf['9'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_)<<_)+((_<<_)*_))'
    
    -- A-Z Letter Encryption Maps (fill these with your desired values)
    bf['A'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_)<<_)+((_<<_)*_))(()>[])%'
    bf['B'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_)<<_)+((_<<_)*_))(()>[])%(()>[])%'
    bf['C'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_)<<_)+((_<<_)*_))(()>[])%(()>[])%(()>[])%'
    -- ... Add mappings for D through Y here ...
    bf['Z'] = '(({}>[])-(()>[]))])*(()>[])%(((_<<_)<<_)+((_<<_)*_))(()>[])%(()>[])%(()>[])%)%(()>[])%)%(()>[])%'

    local cookie = ngx.var.token
    
    -- Process 0-9 digits
    for i = 0, 9 do
        local digit = tostring(i)
        cookie = string.gsub(cookie, digit, "'+"..bf[digit].."+'")
    end
    
    -- Process A-Z letters (using ASCII byte range for cleaner iteration)
    for byte = string.byte('A'), string.byte('Z') do
        local letter = string.char(byte)
        cookie = string.gsub(cookie, letter, "'+"..bf[letter].."+'")
    end

    -- Output the processed cookie (your original code missed this step!)
    ngx.say(cookie)
}

Optional Optimization for Letter Mappings #

If your letter encryption strings follow a pattern (like adding (()>[])% for each subsequent letter), you can auto-generate them instead of writing 26 entries manually:

-- Auto-generate A-Z mappings based on a base string (e.g., bf['9'])
local base_letter_template = bf['9']
for idx = 1, 26 do
    local letter = string.char(string.byte('A') + idx - 1)
    -- Add `idx` copies of the repeating segment to the base template
    bf[letter] = base_letter_template .. string.rep('(()>[])%', idx)
end

Quick Notes #

• If you need to support lowercase letters too, just add a second loop for string.byte('a') to string.byte('z').
• Don't forget to output the processed cookie with ngx.say() (or another Nginx output method) — your original code didn't include this, so you wouldn't see the result otherwise.

内容的提问来源于stack exchange，提问作者anofl sogodi