utils.h File Reference

#include <Windows.h>
#include <string>
#include <ctime>

Go to the source code of this file.

Functions
bool	IsRunningAsAdmin ()
bool	FileExists (const char *filename)
std::string	ToString (const std::wstring &wstr)
std::string	ToLower (std::string str)
std::string	GetCurrentPath ()
std::string	GetBaseName (const std::string &path)
std::string	FormatTime (const std::time_t &time)

Function Documentation

FileExists()

bool FileExists

(

const char *

filename

)

Definition at line 33 of file utils.cpp.

                                      {
    HANDLE hFile = CreateFileA(filename,
        GENERIC_READ,          // Open for reading
        FILE_SHARE_READ,       // Share for reading
        NULL,                  // Default security
        OPEN_EXISTING,         // Open only if exists
        FILE_ATTRIBUTE_NORMAL, // Normal file
        NULL);                 // No template
 
    if (hFile == INVALID_HANDLE_VALUE) {
        return false;  // File does not exist
    }
 
    CloseHandle(hFile);
    return true;  // File exists
}

FormatTime()

std::string FormatTime

(

const std::time_t &

time

)

Definition at line 87 of file utils.cpp.

                                            {
    std::tm timeinfo;
    localtime_s(&timeinfo, &time);
 
    char buffer[80];
    std::strftime(buffer, sizeof(buffer), "%a %b %d %H:%M:%S %Y", &timeinfo);
 
    return std::string(buffer);
}

Referenced by CleanUpProtobufMessage().

GetBaseName()

std::string GetBaseName

(

const std::string &

path

)

Definition at line 82 of file utils.cpp.

                                             {
    const char* fileName = PathFindFileNameA(path.c_str());
    return std::string(fileName);
}

GetCurrentPath()

std::string GetCurrentPath

(

)

Definition at line 68 of file utils.cpp.

{
    char buffer[MAX_PATH];
    DWORD length = GetCurrentDirectoryA(MAX_PATH, buffer);
 
    if (length == 0)
    {
        // Handle error - you might want to throw an exception or return an error code
        return "";
    }
 
    return std::string(buffer);
}

References MAX_PATH.

Referenced by BuildContainerCommandLine(), and WinMain().

IsRunningAsAdmin()

bool IsRunningAsAdmin

(

)

Definition at line 9 of file utils.cpp.

                        {
    BOOL isAdmin = FALSE;
    PSID adminGroup = NULL;
 
    // Allocate and initialize a SID of the administrators group.
    SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
    if (AllocateAndInitializeSid(
        &NtAuthority,
        2,
        SECURITY_BUILTIN_DOMAIN_RID,
        DOMAIN_ALIAS_RID_ADMINS,
        0, 0, 0, 0, 0, 0,
        &adminGroup)) {
        // Check whether the SID of administrators group is enabled in 
        // the primary access token of the process.
        if (!CheckTokenMembership(NULL, adminGroup, &isAdmin)) {
            isAdmin = FALSE;
        }
        FreeSid(adminGroup);
    }
 
    return isAdmin != FALSE;
}

Referenced by WinMain().

ToLower()

std::string ToLower

(

std::string

str

)

Definition at line 62 of file utils.cpp.

                                 {
    std::transform(str.begin(), str.end(), str.begin(),
        [](unsigned char c) { return std::tolower(c); });
    return str;
}

ToString()

std::string ToString

(

const std::wstring &

wstr

)

Definition at line 50 of file utils.cpp.

{
    if (wstr.empty())
    {
        return std::string();
    }
    int size_needed = WideCharToMultiByte(CP_UTF8, 0, &wstr[0], (int)wstr.size(), NULL, 0, NULL, NULL);
    std::string str(size_needed, 0);
    WideCharToMultiByte(CP_UTF8, 0, &wstr[0], (int)wstr.size(), &str[0], size_needed, NULL, NULL);
    return str;
}

Referenced by DisplayEventInfo().