c++ - 使用 Windows API 获取用户的桌面文件夹？

Question

我正在尝试使用以下方法在 C++ 应用程序（通过 DLL）中获取用户的桌面文件夹SHGetSpecialFolderPath：

#define _WIN32_WINNT    0x0500
#define _WIN32_IE       0x0500
#define CSIDL_MYMUSIC   0x000D
#define CSIDL_MYVIDEO   0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

TCHAR path[MAX_PATH];

export LPSTR desktop_directory()
{

    if (SHGetSpecialFolderPath(HWND_DESKTOP, path, CSIDL_DESKTOP, FALSE)) { 
        return path;
    }

}

首先，我想返回一个 else 案例。我返回“错误”，但编译器警告我试图将 a 转换CHAR为LPSTR. 如果在那里，如果由于某种原因无法获取目录，DLL 可能会崩溃。

同样来自 MSDN 文档，它说“[SHGetSpecialFolderPath is not supported. instead, use ShGetFolderPath.]”，然后我导航到该页面并显示“ShGetFolderPath：已弃用。获取由 CSIDL 值标识的文件夹的路径。” 我应该改用什么？

所以：

我想添加一个 else 情况，我返回一个字符串说“错误”
我想知道我是否使用了正确的、未弃用的 API 函数，该函数适用于早在 Windows XP 的现代 Windows 操作系统。

编辑

这是根据要求更新的代码，

#ifndef UNICODE
#define UNICODE
#endif

#ifndef _UNICODE
#define _UNICODE
#endif

#define _WIN32_WINNT 0x0500
#define _WIN32_IE 0x0500
#define CSIDL_MYMUSIC 0x000D
#define CSIDL_MYVIDEO 0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

export LPCWSTR desktop_directory()
{

    static wchar_t path[MAX_PATH+1];

    if (SHGetFolderPath(NULL, CSIDL_DESKTOP, NULL, 0, path)) {
        MessageBox(NULL, path, L"TEST", MB_OK); //test
        return path;
    } else {
        return L"ERROR";
    }

}

使用 MinGW 编译：g++ "src\dll\main.cpp" -D UNICODE -D _UNICODE -O3 -DNDEBUG -s -shared -o "output\main.dll"

我需要使用将 DLL 中的字符串作为 UTF-8 传递WideCharToMultiByte(CP_UTF8, ...)，但我不知道该怎么做。

score 12 · Accepted Answer

SHGetFolderPath()返回 an HRESULT，其中 0 是S_OK，但您的代码期望它返回一个BOOLlike SHGetSpecialFolderPath()do ，其中 0 是FALSE。因此，您需要在代码中修复该错误，因为它当前将成功视为失败。

话虽如此，您正在LPSTR从您的函数中返回 a 。那是一个char*. 但是你正在使用TCHAR你的缓冲区。 TCHAR映射到char或者wchar_t取决于是否UNICODE定义。因此，您需要决定是要char*无条件返回 a ，还是要返回 a TCHAR*，或两者兼而有之。它有很大的不同，例如：

#define _WIN32_WINNT    0x0500
#define _WIN32_IE       0x0500
#define CSIDL_MYMUSIC   0x000D
#define CSIDL_MYVIDEO   0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

export LPSTR desktop_directory()
{
    static char path[MAX_PATH+1];
    if (SHGetSpecialFolderPathA(HWND_DESKTOP, path, CSIDL_DESKTOP, FALSE))
        return path;
    else
        return "ERROR";
}

相对：

#define _WIN32_WINNT    0x0500
#define _WIN32_IE       0x0500
#define CSIDL_MYMUSIC   0x000D
#define CSIDL_MYVIDEO   0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

export LPTSTR desktop_directory()
{
    static TCHAR path[MAX_PATH+1];
    if (SHGetSpecialFolderPath(HWND_DESKTOP, path, CSIDL_DESKTOP, FALSE))
        return path;
    else
        return TEXT("ERROR");
}

相对：

#define _WIN32_WINNT    0x0500
#define _WIN32_IE       0x0500
#define CSIDL_MYMUSIC   0x000D
#define CSIDL_MYVIDEO   0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

export LPSTR desktop_directory_ansi()
{
    static char path[MAX_PATH+1];
    if (SHGetSpecialFolderPathA(HWND_DESKTOP, path, CSIDL_DESKTOP, FALSE))
        return path;
    else
        return "ERROR";
}

export LPWSTR desktop_directory_unicode()
{
    static wchar_t path[MAX_PATH+1];
    if (SHGetSpecialFolderPathW(HWND_DESKTOP, path, CSIDL_DESKTOP, FALSE))
        return path;
    else
        return L"ERROR";
}

更新：大多数 Win32 API 函数不支持 UTF-8，因此如果您希望函数返回 UTF-8 字符串，则必须调用函数的 Unicode 风格，然后使用WideCharToMultiByte()将输出转换为 UTF-8 . 但是你有一个问题 - 谁分配和释放 UTF-8 缓冲区？有几种不同的方法来处理它：

使用线程安全的静态缓冲区（但要注意这个问题）。如果您不需要担心多个线程访问该函数，则删除说明__declspec(thread)符：

#define _WIN32_WINNT 0x0500
#define _WIN32_IE 0x0500
#define CSIDL_MYMUSIC 0x000D
#define CSIDL_MYVIDEO 0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

__declspec(thread) char desktop_dir_buffer[((MAX_PATH*4)+1];

export LPCSTR desktop_directory()
{
    wchar_t path[MAX_PATH+1] = {0};

    if (SHGetFolderPathW(NULL, CSIDL_DESKTOPDIRECTORY, NULL, 0, path) != S_OK)
    {
        MessageBoxW(NULL, L"ERROR in SHGetFolderPathW", L"TEST", MB_OK);
        return NULL;
    }

    MessageBoxW(NULL, path, L"TEST", MB_OK);

    int buflen = WideCharToMultiByte(CP_UTF8, 0, path, lstrlenW(path), desktop_dir_buffer, MAX_PATH*4, NULL, NULL);
    if (buflen <= 0)
    {
        MessageBoxW(NULL, L"ERROR in WideCharToMultiByte", L"TEST", MB_OK);
        return NULL;
    }

    desktop_dir_buffer[buflen] = 0;

    return desktop_dir_buffer;
}

让 DLL 使用自己的内存管理器动态分配缓冲区并将其返回给调用者，然后要求调用者在使用完缓冲区后将缓冲区传递回 DLL，以便可以使用 DLL 的内存管理器释放它：

#define _WIN32_WINNT 0x0500
#define _WIN32_IE 0x0500
#define CSIDL_MYMUSIC 0x000D
#define CSIDL_MYVIDEO 0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

export LPCSTR desktop_directory()
{
    wchar_t path[MAX_PATH+1] = {0};

    if (SHGetFolderPathW(NULL, CSIDL_DESKTOP, NULL, 0, path) != S_OK)
    {
        MessageBoxW(NULL, L"ERROR in SHGetFolderPathW", L"TEST", MB_OK);
        return NULL;
    }

    MessageBoxW(NULL, path, L"TEST", MB_OK);

    int pathlen = lstrlenW(path);

    int buflen = WideCharToMultiByte(CP_UTF8, 0, path, pathlen, NULL, 0, NULL, NULL);
    if (buflen <= 0)
    {
        MessageBoxW(NULL, L"ERROR in WideCharToMultiByte", L"TEST", MB_OK);
        return NULL;
    }

    char *buffer = new char[buflen+1];
    buflen = WideCharToMultiByte(CP_UTF8, 0, path, pathlen, buffer, buflen, NULL, NULL);
    if (buflen <= 0)
    {
        delete[] buffer;
        MessageBoxW(NULL, L"ERROR in WideCharToMultiByte", L"TEST", MB_OK);
        return NULL;
    }
    buffer[buflen] = 0;

    return buffer;
}

export void free_buffer(LPVOID buffer)
{
    delete[] (char*) buffer;
}

让 DLL 使用 Win32 API 内存管理器动态分配缓冲区并将其返回给调用者，然后调用者可以使用相同的 Win32 API 内存管理器释放它，而无需将其传递回 DLL 以释放它：

#define _WIN32_WINNT 0x0500
#define _WIN32_IE 0x0500
#define CSIDL_MYMUSIC 0x000D
#define CSIDL_MYVIDEO 0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

export LPCSTR desktop_directory()
{
    wchar_t path[MAX_PATH+1] = {0};

    if (SHGetFolderPathW(NULL, CSIDL_DESKTOP, NULL, 0, path) != S_OK)
    {
        MessageBoxW(NULL, L"ERROR in SHGetFolderPathW", L"TEST", MB_OK);
        return NULL;
    }

    MessageBoxW(NULL, path, L"TEST", MB_OK);

    int pathlen = lstrlenW(path);

    int buflen = WideCharToMultiByte(CP_UTF8, 0, path, pathlen, NULL, 0, NULL, NULL);
    if (buflen <= 0)
    {
        MessageBoxW(NULL, L"ERROR in WideCharToMultiByte", L"TEST", MB_OK);
        return NULL;
    }

    char *buffer = (char*) LocalAlloc(LMEM_FIXED, buflen+1);
    if (!buffer)
    {
        MessageBoxW(NULL, L"ERROR in LocalAlloc", L"TEST", MB_OK);
        return NULL;
    }

    buflen = WideCharToMultiByte(CP_UTF8, 0, path, pathlen, buffer, buflen, NULL, NULL);
    if (buflen <= 0)
    {
        LocalFree(buffer);
        MessageBoxW(NULL, L"ERROR in WideCharToMultiByte", L"TEST", MB_OK);
        return NULL;
    }
    buffer[buflen] = 0;

    return buffer; // caller can use LocalFree() to free it
}

让调用者传入它自己的缓冲区，DLL 只是填充它。这样，调用者可以决定分配和释放它的最佳方式：

#define _WIN32_WINNT 0x0500
#define _WIN32_IE 0x0500
#define CSIDL_MYMUSIC 0x000D
#define CSIDL_MYVIDEO 0x000E

#include "dll.h"
#include <windows.h>
#include <shlobj.h>
#include <stdio.h>

// the caller can set buffer=NULL and buflen=0 to calculate the needed buffer size
export int desktop_directory(LPSTR buffer, int buflen)
{
    wchar_t path[MAX_PATH+1] = {0};

    if (SHGetFolderPathW(NULL, CSIDL_DESKTOP, NULL, 0, path) != S_OK)
    {
        MessageBoxW(NULL, L"ERROR in SHGetFolderPathW", L"TEST", MB_OK);
        return -1;
    }

    MessageBoxW(NULL, path, L"TEST", MB_OK);

    int pathlen = lstrlenW(path);

    int len = WideCharToMultiByte(CP_UTF8, 0, path, pathlen, buffer, buflen, NULL, NULL);
    if (len <= 0)
    {
        MessageBoxW(NULL, L"ERROR in WideCharToMultiByte", L"TEST", MB_OK);
        return -1;
    }

    if (!buffer)
        ++len;
    else if (len < buflen)
        buffer[len] = 0;

    return len;
}

score -1 · Accepted Answer

有比这更好的方法WidecharToMultibyte

wstring wcharthingy = wstring(widechar);
string convertedthingy = string(wcharthingy.Begin(),wcharthingy.end())

这是更多的伪代码，因为我不记得确切，但 Visual Studio 或 XCode 可能会纠正你反正。

c++ - 使用 Windows API 获取用户的桌面文件夹？

2 回答 2

Related

Reference