[Fix] Clickhouse: Fix quoting

Issue: #2928
rspamd · Jun 17, 2019 · 5c81f0c · 5c81f0c
1 parent e39008c
commit 5c81f0c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lualib/lua_clickhouse.lua b/lualib/lua_clickhouse.lua
@@ -44,7 +44,12 @@ end
 
 local function clickhouse_quote(str)
   if str then
-    return str:gsub('[\'\\]', '\\%1'):lower()
+    return str:gsub('[\'\\\n\t]', {
+      ['\''] = '\\\'',
+      ['\\'] = '\\',
+      ['\n'] = '\\n',
+      ['\t'] = '\\t'
+    }):lower()
   end
 
   return ''
@@ -71,6 +76,8 @@ local function row_to_tsv(row)
       row[i] = '[' .. array_to_string(elt) .. ']'
     elseif type(elt) == 'number' then
       row[i] = ch_number(elt)
+    else
+      row[i] = clickhouse_quote(elt)
     end
   end