lordgamez commented on a change in pull request #1283:
URL: https://github.com/apache/nifi-minifi-cpp/pull/1283#discussion_r839641779
##########
File path: libminifi/include/utils/file/FileUtils.h
##########
@@ -577,6 +616,146 @@ 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] = {};
+ 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);
+
+ // Check GetLastError for GlobalAlloc error condition.
+ if (domain_name == NULL) {
+ GlobalFree(account_name);
+ return std::nullopt;
+ }
+
+ // Second call to LookupAccountSid to get the account name.
+ bool_return = LookupAccountSid(
+ NULL, // name of local or remote computer
+ sid_owner, // security identifier
+ account_name, // account name buffer
+ (LPDWORD)&account_name_dword, // size of account name buffer
+ domain_name, // domain name
+ (LPDWORD)&domain_name_dword, // size of domain name buffer
+ &sid_type); // SID type
+
+ // Check GetLastError for LookupAccountSid error condition.
+ if (bool_return == FALSE) {
+ GlobalFree(account_name);
+ GlobalFree(domain_name);
+ return std::nullopt;
+ }
+
+ auto result = std::string(account_name);
+ GlobalFree(account_name);
+ GlobalFree(domain_name);
+ return result;
+#endif
+}
+
+#ifndef WIN32
+inline std::optional<std::string> get_file_group(const std::string& file_path)
{
+ struct stat info;
+ if (stat(file_path.c_str(), &info) != 0) {
+ return std::nullopt;
+ }
+
+ struct group gr;
+ gr.gr_name = 0;
+ struct group *result = nullptr;
+ char localbuf[1024] = {};
+ if ((getgrgid_r(info.st_uid, &gr, localbuf, sizeof(localbuf), &result) != 0)
|| gr.gr_name == 0) {
+ return std::nullopt;
+ }
+
+ return std::string(gr.gr_name);
+}
+#endif
+
+inline std::optional<std::string> get_relative_path(const std::string& path,
const std::string& base_path) {
+ if (!utils::StringUtils::startsWith(path, base_path)) {
+ return std::nullopt;
+ }
+
+ return std::filesystem::relative(path, base_path);
Review comment:
You are right, fixed 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]
