github_mirrors/attachment
3
0
Fork 1
mirror of https://github.com/openappsec/attachment.git synced 2025-11-19 18:44:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix response body

Browse Source
This commit is contained in:
wiaamm
2025-11-18 00:43:08 +02:00
parent 908e210af0
commit 72fdbd6501
1 changed files with 29 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
attachments/kong/plugins/open-appsec-waf-kong-plugin/handler.lua
View File

@@ -193,54 +193,46 @@ function NanoHandler.body_filter(conf)
return return
end end
-- Collect response body chunks -- Process response body chunks incrementally
local chunk = ngx.arg[1] local chunk = ngx.arg[1]
local eof = ngx.arg[2] local eof = ngx.arg[2]
kong.log.debug("[body_filter] Session: ", session_id, " | Chunk size: ", chunk and #chunk or 0, " | EOF: ", tostring(eof)) kong.log.debug("[body_filter] Session: ", session_id, " | Chunk size: ", chunk and #chunk or 0, " | EOF: ", tostring(eof))
-- Initialize buffer on first call -- Send each chunk to nano service as it arrives
if not ctx.response_body_buffer then
ctx.response_body_buffer = {}
kong.log.debug("[body_filter] Session: ", session_id, " | Initialized response body buffer")
end
-- Collect chunks
if chunk and #chunk > 0 then if chunk and #chunk > 0 then
table.insert(ctx.response_body_buffer, chunk) kong.log.debug("[body_filter] Session: ", session_id, " | Sending chunk to nano, size: ", #chunk, " bytes")
kong.log.debug("[body_filter] Session: ", session_id, " | Buffered chunk, total chunks: ", #ctx.response_body_buffer)
end
-- Process complete body at EOF -- Initialize chunk counter
if eof then ctx.body_buffer_chunk = ctx.body_buffer_chunk or 0
kong.log.debug("[body_filter] Session: ", session_id, " | EOF reached, buffer has ", #ctx.response_body_buffer, " chunks")
-- Send body to nano service if we collected chunks local verdict, response, modifications = nano.send_body(session_id, session_data, chunk, nano.HttpChunkType.HTTP_RESPONSE_BODY)
if #ctx.response_body_buffer > 0 then
local complete_body = table.concat(ctx.response_body_buffer)
kong.log.debug("[body_filter] Session: ", session_id, " | Sending complete body, size: ", #complete_body, " bytes")
local verdict, response, modifications = nano.send_body(session_id, session_data, complete_body, nano.HttpChunkType.HTTP_RESPONSE_BODY) kong.log.debug("[body_filter] Session: ", session_id, " | Verdict after chunk #", ctx.body_buffer_chunk, ": ", verdict)
if modifications then if modifications then
kong.log.debug("[body_filter] Session: ", session_id, " | Applying body modifications") kong.log.debug("[body_filter] Session: ", session_id, " | Applying body modifications to chunk")
complete_body = nano.handle_body_modifications(complete_body, modifications, 0) chunk = nano.handle_body_modifications(chunk, modifications, ctx.body_buffer_chunk)
ngx.arg[1] = complete_body ngx.arg[1] = chunk
end end
ctx.body_buffer_chunk = ctx.body_buffer_chunk + 1
ctx.body_seen = true
if verdict == nano.AttachmentVerdict.DROP then if verdict == nano.AttachmentVerdict.DROP then
kong.log.warn("[body_filter] Body verdict DROP for session: ", session_id) kong.log.warn("[body_filter] Body chunk verdict DROP for session: ", session_id)
nano.fini_session(session_data) nano.fini_session(session_data)
ctx.session_finalized = true ctx.session_finalized = true
local result = nano.handle_custom_response(session_data, response) local result = nano.handle_custom_response(session_data, response)
nano.cleanup_all() nano.cleanup_all()
return result return result
end end
else
kong.log.debug("[body_filter] Session: ", session_id, " | No body chunks collected (empty response or no body)")
end end
-- End inspection -- End inspection at EOF
if eof then
kong.log.debug("[body_filter] Session: ", session_id, " | EOF reached, body_seen: ", tostring(ctx.body_seen))
kong.log.debug("[body_filter] Session: ", session_id, " | Ending inspection") kong.log.debug("[body_filter] Session: ", session_id, " | Ending inspection")
local verdict, response = nano.end_inspection(session_id, session_data, nano.HttpChunkType.HTTP_RESPONSE_END) local verdict, response = nano.end_inspection(session_id, session_data, nano.HttpChunkType.HTTP_RESPONSE_END)