lordgamez commented on a change in pull request #1283:
URL: https://github.com/apache/nifi-minifi-cpp/pull/1283#discussion_r839641485
##########
File path: libminifi/include/utils/file/FileUtils.h
##########
@@ -577,6 +614,147 @@ inline std::string get_file_content(const std::string
&file_name) {
bool contains(const std::filesystem::path& file_path, std::string_view
text_to_search);
+inline std::optional<std::string> get_file_owner(const std::string& file_path)
{
+#ifndef WIN32
+ struct stat info;
+ if (stat(file_path.c_str(), &info) != 0) {
+ return std::nullopt;
+ }
+
+ struct passwd pw;
+ pw.pw_name = 0;
+ struct passwd *result = nullptr;
+ char localbuf[1024];
+ std::fill(localbuf, localbuf + sizeof(localbuf), 0);
+ if (getpwuid_r(info.st_uid, &pw, localbuf, sizeof(localbuf), &result) != 0
|| pw.pw_name == 0) {
+ return std::nullopt;
+ }
+
+ return std::string(pw.pw_name);
+#else
+ DWORD return_code = 0;
+ PSID sid_owner = NULL;
+ BOOL bool_return = TRUE;
+ LPTSTR account_name = NULL;
+ LPTSTR domain_name = NULL;
+ DWORD account_name_dword = 1;
+ DWORD domain_name_dword = 1;
+ SID_NAME_USE sid_type = SidTypeUnknown;
+ HANDLE file_handle;
+ PSECURITY_DESCRIPTOR sec_descriptor = NULL;
+
+ // Get the handle of the file object.
+ file_handle = CreateFile(
+ TEXT(file_path.c_str()),
+ GENERIC_READ,
+ FILE_SHARE_READ,
+ NULL,
+ OPEN_EXISTING,
+ FILE_ATTRIBUTE_NORMAL,
+ NULL);
+
+ // Check GetLastError for CreateFile error code.
+ if (file_handle == INVALID_HANDLE_VALUE) {
+ return std::nullopt;
+ }
+
+ // Get the owner SID of the file.
+ return_code = GetSecurityInfo(
+ file_handle,
+ SE_FILE_OBJECT,
+ OWNER_SECURITY_INFORMATION,
+ &sid_owner,
+ NULL,
+ NULL,
+ NULL,
+ &sec_descriptor);
+
+ // Check GetLastError for GetSecurityInfo error condition.
+ if (return_code != ERROR_SUCCESS) {
+ return std::nullopt;
+ }
+
+ // First call to LookupAccountSid to get the buffer sizes.
+ bool_return = LookupAccountSid(
+ NULL,
+ sid_owner,
+ account_name,
+ (LPDWORD)&account_name_dword,
+ domain_name,
+ (LPDWORD)&domain_name_dword,
+ &sid_type);
+
+ // Reallocate memory for the buffers.
+ account_name = (LPTSTR)GlobalAlloc(
+ GMEM_FIXED,
+ account_name_dword);
+
+ // Check GetLastError for GlobalAlloc error condition.
+ if (account_name == NULL) {
+ return std::nullopt;
+ }
+
+ domain_name = (LPTSTR)GlobalAlloc(
+ GMEM_FIXED,
+ domain_name_dword);
Review comment:
Good call, didn't think of that updated in
e7fdcd5df853057195f37e932c836cf438928629
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
