如何尽可能干净地使用 VS C++ GetEnvironmentVariable？

Question

（这与其说是学究气的练习，倒不如说是个问题。）

我已经制作了一个不错的小程序，它是我的 linux 操作系统的原生程序，但我认为它也足以存在于我的 Windows 机器上。因此，我想访问 Windows 的环境变量，MSDN 引用了这样一个示例：

const DWORD buff_size = 50;
LPTSTR buff = new TCHAR[buff_size];

const DWORD var_size = GetEnvironmentVariable("HOME",buff,buff_size);

if (var_size==0) { /* fine, some failure or no HOME */ }
else if (var_size>buff_size) {

    // OK, so 50 isn't big enough.
    if (buff) delete [] buff;
    buff = new TCHAR[var_size];
    
    const DWORD new_size = GetEnvironmentVariable("HOME",buff,var_size);

    if (new_size==0 || new_size>var_size) { /* *Sigh* */ }
    else { /* great, we're done */ }
}
else { /* in one go! */ }

这（对我而言）不如使用 getenv 并仅检查空指针那么好。我也不想动态分配内存，因为我只是想让程序在 Windows 和我的 linux 操作系统上运行，这意味着这个 MS 代码必须与 nix 代码很好地配合。更具体地说：

template <class T> // let the compiler sort out between char* and TCHAR*
inline bool get_home(T& val) { // return true if OK, false otherwise
#if defined (__linux) || (__unix)
    val = getenv("HOME");
    if (val) return true;
    else return false;
#elif defined (WINDOWS) || defined (_WIN32) || defined (WIN32)
    // something like the MS Code above
#else
    // probably I'll just return false here.
#endif
}

所以，我必须在堆上进行普遍分配，或者在调用函数中使用#ifdef 来释放内存。不是很漂亮。

当然，我本来可以一开始就在堆栈上分配“buff”，但是如果在第一次调用 GetEnvironmentVariable 时“buff_size”不够大，我就必须创建一个新的TCHAR[]。更好，但是如果我是一个书呆子并且不想到处创建多余的数组怎么办？有什么更美观的想法吗？

我不是很了解，所以有人会嫉妒我故意强迫 GetEnvironmentVariable 失败以获取字符串大小吗？有没有人看到以下问题：

const DWORD buff_size = GetEnvironmentVariable("HOME",0,0);
TCHAR buff[buff_size];
const DWORD ret = GetEnvironmentVariable("HOME",buff,buff_size);
// ...

还有其他想法或建议吗？ （或者更正明显的错误？）

更新： 下面有很多有用的信息。我认为最好的办法是使用static char[]，例如：

inline const char* get_home(void) { // inline not required, but what the hell.
#if defined (__linux) || (__unix)
    return getenv("HOME");
#elif defined (WINDOWS) || defined (WIN32) || defined (_WIN32)
    static char buff[MAX_PATH];
    const DWORD ret = GetEnvironmentVariableA("USERPROFILE",buff,MAX_PATH);
    if (ret==0 || ret>MAX_PATH)
        return 0;
    else
        return buff;
 #else
        return 0;
 #endif
 }

也许这不是最优雅的方式，但它可能是在 *nix 和 Windows 之间同步我想做的最简单的方式。 （稍后我还会担心 Unicode 支持。）

感谢所有帮助过的人。

Answer 1

DWORD bufferSize = 65535; //Limit according to http://msdn.microsoft.com/en-us/library/ms683188.aspx
std::wstring buff;
buff.resize(bufferSize);
bufferSize = GetEnvironmentVariableW(L"Name", &buff[0], bufferSize);
if (!bufferSize)
    //error
buff.resize(bufferSize);

当然，如果您想要 ASCII，请将 wstring 替换为 string 并将 GetEnvironmentVariableW 替换为 GetEnvironmentVariableA。

编辑：您也可以自己创建getenv。这是因为

后续调用 getenv 时可能会使用相同的内存位置，覆盖之前的内容。

const char * WinGetEnv(const char * name)
{
    const DWORD buffSize = 65535;
    static char buffer[buffSize];
    if (GetEnvironmentVariableA(name, buffer, buffSize))
    {
        return buffer;
    }
    else
    {
        return 0;
    }
}

当然，如果您想保持对 unicode 的支持，最好使用所有这些的宽字符版本。

Answer 2

这不是最初的问题，但可能值得将 MFC 方式添加到此线程以供参考：

CString strComSpec;
if (strComSpec.GetEnvironmentVariable(_T("COMSPEC")))
{
    //Do your stuff here
}

Answer 3

VC++ 在 stdlib.h 中实现 getenv，例如，参见 here。

Answer 4

您在帖子末尾提出的建议是正确的方法 - 调用一次以获取所需的缓冲区大小，然后再次调用以实际获取数据。许多 Win32 API 以这种方式工作，起初令人困惑，但很常见。

您可以做的一件事是在第一次调用时传入一个最佳猜测缓冲区及其大小，并且只有在失败时才再次调用。

Answer 5

不要打扰。 %HOME% 是 Windows 上的路径，应该可以被所有合理的程序使用。因此，它将适合WCHAR[MAX_PATH]。你不需要处理比这更长的边缘情况 - 如果它更长，大多数文件函数无论如何都会拒绝它，所以你最好早点失败。

但是，不要假设您可以使用TCHAR[MAX_PATH] 或char[MAX_PATH]。您无法控制%HOME% 的内容；它将包含用户名。如果那是“André”（即不是 ASCII），则必须将 %HOME% 存储在 WCHAR[MAX_PATH] 中。