From 6f458b5203cb0ec16df10f984f08761d226d0c8a Mon Sep 17 00:00:00 2001
From: Victor Hora <vhora@trustwave.com>
Date: Wed, 19 Sep 2018 19:41:49 -0400
Subject: [PATCH] Fix on top of jmx's m.setvar commit for USER collection in
 Lua scripts

---
 src/engine/lua.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/engine/lua.cc b/src/engine/lua.cc
index a16ac3dd..feecb551 100644
--- a/src/engine/lua.cc
+++ b/src/engine/lua.cc
@@ -399,7 +399,13 @@ int Lua::setvar(lua_State *L) {
                     t->m_rules->m_secWebAppId.m_value,
             var_value);
     }
+    else if (collection == "USER") {
+        t->m_collections.m_user_collection->storeOrUpdateFirst(
+            variableName, t->m_collections.m_user_collection_key,
+                    t->m_rules->m_secWebAppId.m_value,
+            var_value);
 
+    }
     return 0;
 }