diff options
author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2020-05-21 17:26:47 +0100 |
---|---|---|
committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2020-05-21 17:26:47 +0100 |
commit | 207e95fe7b6dcac15a00995c13a86bb748a2dcf4 (patch) | |
tree | b06f50395c47a3ec494d788efab793e596631f77 /lualib/lua_content | |
parent | 20f3cd1e223fed2bd7ea73ed90c1363b3f4d922e (diff) | |
download | rspamd-207e95fe7b6dcac15a00995c13a86bb748a2dcf4.tar.gz rspamd-207e95fe7b6dcac15a00995c13a86bb748a2dcf4.zip |
[Minor] Add small sanity check to avoid extensive computations
Diffstat (limited to 'lualib/lua_content')
-rw-r--r-- | lualib/lua_content/pdf.lua | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/lualib/lua_content/pdf.lua b/lualib/lua_content/pdf.lua index 0e3f024ae..fb3d07a93 100644 --- a/lualib/lua_content/pdf.lua +++ b/lualib/lua_content/pdf.lua @@ -838,6 +838,10 @@ end local function extract_outer_objects(task, input, pdf) local start_pos, end_pos = 1, 1 local obj_count = 0 + + lua_util.debugm(N, task, "pdf: extract objects from %s start positions and %s end positions", + #pdf.start_objects, #pdf.end_objects) + while start_pos <= #pdf.start_objects and end_pos <= #pdf.end_objects do local first = pdf.start_objects[start_pos] local last = pdf.end_objects[end_pos] @@ -1181,12 +1185,23 @@ end processors.trailer = function(input, task, positions, output) local last_pos = positions[#positions] + lua_util.debugm(N, task, 'pdf: process trailer at position %s (%s total length)', + last_pos, #input) + local last_span = input:span(last_pos[1]) + local lines_checked = 0 for line in last_span:lines(true) do if line:find('/Encrypt ') then lua_util.debugm(N, task, "pdf: found encrypted line in trailer: %s", line) output.encrypted = true + break + end + lines_checked = lines_checked + 1 + + if lines_checked > 100 then + lua_util.debugm(N, task, "pdf: trailer has too many lines, stop checking") + break end end end |