82 lines
1.9 KiB
C++
82 lines
1.9 KiB
C++
|
#include "Environment.h"
|
|||
|
|
#include <Psapi.h>
|
|||
|
|
#include <ShlObj.h>
|
|||
|
|
std::string Environment::CommandLine()
|
|||
|
|
{
|
|||
|
|
return GetCommandLineA();
|
|||
|
|
}
|
|||
|
|
std::string Environment::CurrentDirectory()
|
|||
|
|
{
|
|||
|
|
char buffer[MAX_PATH];
|
|||
|
|
GetCurrentDirectoryA(MAX_PATH, buffer);
|
|||
|
|
return std::string(buffer);
|
|||
|
|
}
|
|||
|
|
void Environment::CurrentDirectory(std::string path)
|
|||
|
|
{
|
|||
|
|
SetCurrentDirectoryA(path.c_str());
|
|||
|
|
}
|
|||
|
|
std::string Environment::SystemDirectory()
|
|||
|
|
{
|
|||
|
|
char buffer[MAX_PATH];
|
|||
|
|
GetSystemDirectoryA(buffer, MAX_PATH);
|
|||
|
|
return std::string(buffer);
|
|||
|
|
}
|
|||
|
|
std::string Environment::WindowsDirectory()
|
|||
|
|
{
|
|||
|
|
char buffer[MAX_PATH];
|
|||
|
|
GetWindowsDirectoryA(buffer, MAX_PATH);
|
|||
|
|
return std::string(buffer);
|
|||
|
|
}
|
|||
|
|
std::string Environment::MachineName()
|
|||
|
|
{
|
|||
|
|
char buffer[MAX_COMPUTERNAME_LENGTH + 1];
|
|||
|
|
DWORD size = MAX_COMPUTERNAME_LENGTH + 1;
|
|||
|
|
GetComputerNameA(buffer, &size);
|
|||
|
|
return std::string(buffer);
|
|||
|
|
}
|
|||
|
|
int Environment::ProcessorCount()
|
|||
|
|
{
|
|||
|
|
SYSTEM_INFO info;
|
|||
|
|
GetSystemInfo(&info);
|
|||
|
|
return info.dwNumberOfProcessors;
|
|||
|
|
}
|
|||
|
|
int Environment::SystemPageSize()
|
|||
|
|
{
|
|||
|
|
SYSTEM_INFO info;
|
|||
|
|
GetSystemInfo(&info);
|
|||
|
|
return info.dwPageSize;
|
|||
|
|
}
|
|||
|
|
long long Environment::WorkingSet()
|
|||
|
|
{
|
|||
|
|
PROCESS_MEMORY_COUNTERS pmc;
|
|||
|
|
GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc));
|
|||
|
|
return pmc.WorkingSetSize;
|
|||
|
|
}
|
|||
|
|
std::string Environment::UserName()
|
|||
|
|
{
|
|||
|
|
char buffer[MAX_PATH] = { 0 };
|
|||
|
|
DWORD size = MAX_PATH;
|
|||
|
|
GetUserNameA(buffer, &size);
|
|||
|
|
return std::string(buffer);
|
|||
|
|
}
|
|||
|
|
std::vector<std::string> Environment::LogicalDrives()
|
|||
|
|
{
|
|||
|
|
std::vector<std::string> result;
|
|||
|
|
char buffer[MAX_PATH];
|
|||
|
|
GetLogicalDriveStringsA(MAX_PATH, buffer);
|
|||
|
|
std::string str = std::string(buffer);
|
|||
|
|
int index = 0;
|
|||
|
|
while (index < str.size())
|
|||
|
|
{
|
|||
|
|
std::string drive = str.substr(index, 3);
|
|||
|
|
result.push_back(drive);
|
|||
|
|
index += 4;
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
std::string Environment::GetFolderPath(SpecialFolder folder, SpecialFolderOption option)
|
|||
|
|
{
|
|||
|
|
char buffer[MAX_PATH] = {};
|
|||
|
|
SHGetFolderPathA(NULL, (int)folder | (int)option, NULL, 0, buffer);
|
|||
|
|
return std::string(buffer);
|
|||
|
|
}
|