github_mirrors/attachment
3
0
Fork 1
mirror of https://github.com/openappsec/attachment.git synced 2025-08-14 05:45:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

return the old functionality

Browse Source
This commit is contained in:
wiaamm 2025-07-06 10:44:36 +03:00
parent ba216839e6
commit 466abd73bc
1 changed files with 12 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
attachments/kong/nano_ffi.lua
View File

@ -85,10 +85,10 @@ end
function nano.handle_custom_response(session_data, response) function nano.handle_custom_response(session_data, response)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment then if not attachment then
kong.log.err("Cannot handle custom response: Attachment not available for worker ", worker_id, " - failing open") kong.log.warn("Cannot handle custom response: Attachment not available for worker ", worker_id, " - failing open")
return kong.response.exit(200, "Request allowed due to attachment unavailability") return kong.response.exit(200, "Request allowed due to attachment unavailability")
end end
@ -200,11 +200,9 @@ function nano.init_attachment()
kong.log.err("Worker ", worker_id, " failed to initialize attachment (attempt ", attempt, "/", retries, "): ", err) kong.log.err("Worker ", worker_id, " failed to initialize attachment (attempt ", attempt, "/", retries, "): ", err)
if attempt < retries then local delay = base_delay * (2 ^ (attempt - 1)) + math.random(0, 1000) / 1000
local delay = base_delay * (2 ^ (attempt - 1)) + math.random(0, 1000) / 1000 kong.log.info("Worker ", worker_id, " retrying attachment initialization in ", delay, " seconds...")
kong.log.info("Worker ", worker_id, " retrying attachment initialization in ", delay, " seconds...") ngx.sleep(delay)
ngx.sleep(delay)
end
end end
if not attachment then if not attachment then
@ -221,7 +219,7 @@ end
-- Initialize a session for a given request -- Initialize a session for a given request
function nano.init_session(session_id) function nano.init_session(session_id)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment then if not attachment then
kong.log.warn("Cannot initialize session: Attachment not available for worker ", worker_id, " - failing open") kong.log.warn("Cannot initialize session: Attachment not available for worker ", worker_id, " - failing open")
@ -328,7 +326,7 @@ end
-- Send data to NanoAttachment -- Send data to NanoAttachment
function nano.send_data(session_id, session_data, meta_data, header_data, contains_body, chunk_type) function nano.send_data(session_id, session_data, meta_data, header_data, contains_body, chunk_type)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment then if not attachment then
kong.log.warn("Attachment not available for worker ", worker_id, " - failing open") kong.log.warn("Attachment not available for worker ", worker_id, " - failing open")
@ -350,7 +348,7 @@ end
function nano.send_body(session_id, session_data, body_chunk, chunk_type) function nano.send_body(session_id, session_data, body_chunk, chunk_type)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment then if not attachment then
kong.log.warn("Attachment not available for worker ", worker_id, " - failing open") kong.log.warn("Attachment not available for worker ", worker_id, " - failing open")
@ -410,7 +408,7 @@ end
-- Finalize session cleanup -- Finalize session cleanup
function nano.fini_session(session_data) function nano.fini_session(session_data)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment or not session_data then if not attachment or not session_data then
kong.log.warn("Cannot finalize session: Invalid attachment or session_data for worker ", worker_id) kong.log.warn("Cannot finalize session: Invalid attachment or session_data for worker ", worker_id)
@ -425,7 +423,7 @@ end
-- Send response headers for inspection -- Send response headers for inspection
function nano.send_response_headers(session_id, session_data, headers, status_code, content_length) function nano.send_response_headers(session_id, session_data, headers, status_code, content_length)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment then if not attachment then
kong.log.warn("Attachment not available for worker ", worker_id, " - failing open") kong.log.warn("Attachment not available for worker ", worker_id, " - failing open")
@ -499,7 +497,7 @@ end
-- End inspection for a session -- End inspection for a session
function nano.end_inspection(session_id, session_data, chunk_type) function nano.end_inspection(session_id, session_data, chunk_type)
local worker_id = ngx.worker.id() local worker_id = ngx.worker.id()
local attachment = nano.ensure_attachment() local attachment = nano.attachments[worker_id]
if not attachment then if not attachment then
kong.log.warn("Attachment not available for worker ", worker_id, " - failing open during end_inspection") kong.log.warn("Attachment not available for worker ", worker_id, " - failing open during end_inspection")
@ -522,22 +520,4 @@ function nano.end_inspection(session_id, session_data, chunk_type)
end end
-- Helper function to ensure attachment is available -- Helper function to ensure attachment is available
function nano.ensure_attachment()
local worker_id = ngx.worker.id()
local attachment = nano.attachments[worker_id]
if not attachment then
kong.log.warn("Attachment not found for worker ", worker_id, ", attempting to reinitialize...")
if nano.init_attachment() then
attachment = nano.attachments[worker_id]
kong.log.info("Successfully reinitialized attachment for worker ", worker_id)
else
kong.log.err("Failed to reinitialize attachment for worker ", worker_id)
return nil
end
end
return attachment
end
return nano return nano