From: Bhargava Jandhyala (bjandhya) <bjandhya@cisco.com>
Date: Mon, 13 Dec 2021 11:32:18 +0000 (+0000)
Subject: Pull request #3213: file_api: Handling file_data
X-Git-Tag: 3.1.19.0~8
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=a2f557b1505d7f48c921a23ab32d5b2c7a2f5dba;p=thirdparty%2Fsnort3.git

Pull request #3213: file_api: Handling file_data

Merge in SNORT/snort3 from ~VKAMBALA/snort3:user_file_data to master

Squashed commit of the following:

commit c3eed73709c95f65054f1643ee2e0455e8d5717a
Author: krishnakanth <vkambala@cisco.com>
Date:   Fri Dec 10 17:36:29 2021 +0530

    file_api: Handling file_data
---

diff --git a/src/file_api/file_lib.cc b/src/file_api/file_lib.cc
index 78680f653..8e4604e07 100644
--- a/src/file_api/file_lib.cc
+++ b/src/file_api/file_lib.cc
@@ -358,9 +358,13 @@ void FileContext::log_file_event(Flow* flow, FilePolicyBase* policy)
             break;
         }
 
+        user_file_data_mutex.lock();
+
         if (policy and log_needed and user_file_data)
             policy->log_file_action(flow, this, FILE_ACTION_DEFAULT);
 
+        user_file_data_mutex.unlock();
+
         if ( config->trace_type )
             print(std::cout);
     }
diff --git a/src/file_api/file_lib.h b/src/file_api/file_lib.h
index 6a8cbebed..459aed616 100644
--- a/src/file_api/file_lib.h
+++ b/src/file_api/file_lib.h
@@ -24,6 +24,7 @@
 
 // This will be basis of file class
 
+#include <mutex>
 #include <ostream>
 #include <string>
 
@@ -108,6 +109,7 @@ protected:
     bool file_capture_enabled = false;
     FileState file_state = { FILE_CAPTURE_SUCCESS, FILE_SIG_PROCESSING };
     uint32_t policy_id = 0;
+    std::mutex user_file_data_mutex;
     UserFileDataBase* user_file_data = nullptr;
 
 private: