#include <Windows.h>
#include <tchar.h>
#include <strsafe.h> // 使用了StringCbCopyN等安全字符串处理函数
void DumpEnvStrings()
{
// 结尾有两个'\0',一个是最后一个环境变量的结尾,另一个是整个块的结尾
PTSTR pszEnvBlock = GetEnvironmentStrings();
TCHAR szName[MAX_PATH];
TCHAR szValue[MAX_PATH];
PTSTR pszCurrent = pszEnvBlock;
HRESULT hr = S_OK;
PCTSTR pszPos = NULL;
int current = 0;
while (pszCurrent != NULL)
{
// 当地一个字母不是=的时候 (一般环境变量字符串是这个样子[0]=::=::\[1]=c:\:=c:\system32 [2]=eistcode=0x0000 [0]这种是字符串数组的第一个字符串这个意思 每个字符串都是\0结束的)
if (*pszCurrent != TEXT('='))
{
// 找到第一个等号的位置
pszPos = _tcschr(pszCurrent, TEXT('='));
if (pszPos == NULL)
{
_tprintf(TEXT("not found = \n"));
}
// 位置指针指向环境变量值的第一个位置处
pszPos++;
// 获取环境变量名(去掉等号)(此处必须强制类型转换,否则变量名不完整)
size_t cbNameLength = (size_t)pszPos - (size_t)pszCurrent - sizeof(TCHAR);
hr = StringCbCopyN(szName, MAX_PATH, pszCurrent, cbNameLength);
if (FAILED(hr))
{
break;
}
// 获取环境变量值(包括最后的NULL字符,由于是让人看见的所以允许“截断”)
hr = StringCchCopyN(szValue, MAX_PATH, pszPos, _tcslen(pszPos)+1);
if (SUCCEEDED(hr))
{
_tprintf(TEXT("(%u) %s = %s\r\n"), current, szName, szValue);
}
else // 有错误发生,
{
// 检查“截断”
if (STRSAFE_E_INSUFFICIENT_BUFFER)
{
_tprintf(TEXT("(%u) %s = %s...\r\n"), current, szName, szValue);
}
else // 不应该发生的错误
{
_tprintf(TEXT("(%u) %s = %s???\r\n"), current, szName, szValue);
}
break;
}
}
else // 首个字符是等号
{
_tprintf(TEXT("(%u) %s\r\n"), current, pszCurrent);
}
// 下一个环境变量
current++;
// 指针指向字符串结尾 排除掉第一个字符是=号的字符串
while (*pszCurrent != TEXT('\0'))
{
pszCurrent++;
}
// 指针指向下一个环境变量
pszCurrent++;
// 检查是否还有环境变量
if (*pszCurrent == TEXT('\0'))
{
break;
}
}
// 释放内存
FreeEnvironmentStrings(pszEnvBlock);
}
int main()
{
DumpEnvStrings();
return 0;
}
DumpEnvStrings.cpp(获取环境变量)
最新推荐文章于 2023-03-21 00:12:34 发布