一个分析“文件夹”选择框实现方法的过程

来源：互联网发布：淘宝客服一个月多少钱编辑：程序博客网时间：2024/05/22 03:33

在软件开发中，我们如果存在“导入导出”的场景时，难免会用到“文件夹”选择框。之前一直没有太关注过这个的实现过程。最近在工作中遇到了一些问题，我做了一些研究。在此记录下研究的过程。（转载请指明出于breaksoftware的csdn博客）

首先，我们发现我们的文件选择框，只能显示出本地文件夹，而不能显示设备虚拟出来的文件。比如

这样的设备，就不会在我们的文件选择框中出现。

我们看下我们代码中的设置

BROWSEINFOA bi;   bi.hwndOwner      = hWnd;bi.pidlRoot       = NULL;  bi.pszDisplayName = NULL;   bi.lpszTitle      = "请选择下载位置";   bi.ulFlags        = BIF_RETURNONLYFSDIRS | BIF_USENEWUI;  bi.lpfn           = BrowseCallbackProc;   g_defaultfolder = WinTools::GetSystemPath(CSIDL_DESKTOP);bi.lParam         = (LPARAM)(g_defaultfolder.c_str());bi.iImage         = 0;

当时我的第一直觉就是我们的ulFlags设置的不对，然后我翻阅了MSDN，去掉了BIF_RETURNONLYFSDIRS就好了。

但是问题接踵而至

当我们选择了这个设备下的文件夹后，我们并不能获取我们选择的文件夹路径。经调试发现是我们之后调用的获取文件夹路径的函数SHGetPathFromIDList返回失败。
那我们就让选择框对这类文件进行过滤。当时我还是认为是不是我们哪个ulflags没有设置。可是试了几个感觉可能的flags，还是不行。

后来，我寻找到一个该功能完善的软件A，它的展现是正确的。

最后我决定不再闭门造车，而是分析该软件A这块功能的具体实现。

首先我们要确认A软件使用的哪个函数打开文件选择框的。众所周知，我们使用的SHBrowseForFolderA属于SH类函数，即shell32.dll中的导出函数。SH类函数基本都是辅助类型函数，其是在windows原生API基础上做了一层封装。所以我们先要确定A软件使用的是不是SHBrowseForFolder函数。我们使用Windbg附加到A进程上

其次，使用bp shell32!SHBrowseForFolderA 和bp shell32!SHBrowseForFolderW下函数断点。一般来说，Windows平台的API都有的A版和一个W版（有特殊的函数只有一个版本），所以我们在分析时，往往给A版和W版都下断点。
最后运行挂起的A软件，点击“打开文件夹”。Windbg果然断住了，这个证明A软件使用的是SHBrowseForFolderW。

这样我们确定了软件A是使用的SHBrowseForFolderW，那么我们开始分析，看看它是如何个这个函数的。这儿涉及一个稍微有点复杂的过程，因为A软件很多地方是用.net写的。我调回到调用SHBrowseForFolderW的地方，仍然难以直观看到起参数的传递。我就改成分析SHBrowseForFolderW的实现，来查看其参数。我使用IDA，对Shell32.dll中的SHBrowseForFolderW进行逆向。以下列出其重要的代码

直到此时，我仍然认为我们的问题是出在flags设置不对上。所以我仍然只是关注了ulflags这个参数。我们看到我们可以在mov eax , [esi+10h]处看到ulflags的值。
回到windbg，使用u 7648dfae 7648dfff得到内存中的汇编代码和地址（7648dfae是中断下来后，得知SHBrowseForFolderW的入口地址）

我们在7648dff5处下断点（bp 7648dff5）。
中断后我们用r eax指令查看eax的值

我们终于拿到A软件的ulflags的设置，本来以为大功告成。于是在代码中将ulflags的值设置为0x40对应的宏。可是悲剧的是，问题依旧。看来并不是我们ulflags设置的不对。我们回到BROWSEINFO的参数说明。

typedef struct _browseinfo {  HWND              hwndOwner;  PCIDLIST_ABSOLUTE pidlRoot;  LPTSTR            pszDisplayName;  LPCTSTR           lpszTitle;  UINT              ulFlags;  BFFCALLBACK       lpfn;  LPARAM            lParam;  int               iImage;} BROWSEINFO, *PBROWSEINFO, *LPBROWSEINFO;

这些参数，除了ulflags对窗口的行为可能产生影响外，只能是lpfn了。我看了下我们的lpfn传递的是NULL，而A软件是否传了值呢？
我们将断点下在7648dffb，看看A软件是否传了值。

A软件传递了值！
那如何验证是否就是这个回调函数导致了我们之间的差异？
MSDN说明lpfn可以为NULL，那么我使用r eax=0来修改此处的eax，然后待7648dffb处指令执行完毕，就可以修改SHBrowseForFolderW内部使用了该回调地址的地方了。
修改好后，我们继续执行A软件，并选择之前出现“确定”按钮不可用的文件夹，可以看到这个时候的“确定”按钮可用了。

于是原因找到了，此时我们只要关注该回调如何实现便可以实现和A软件的功能。

那么这个回调如何实现呢？我们看个网上很普及的例子

int CALLBACK BrowseCallbackProcSetting(HWND hwnd, UINT uMsg, LPARAM lParam, LPARAM lpData)  {           if  ( BFFM_INITIALIZED == uMsg ) {                     ::SendMessage(hwnd,BFFM_SETSELECTION,TRUE,lpData);           }           else if ( BFFM_SELCHANGED == uMsg )         {                   char pszPath[MAX_PATH] = {0};                   LPITEMIDLIST pidl = (LPITEMIDLIST)(lParam);                   if ( NULL == pidl ) {                            return 0;                   }                   if (SHGetPathFromIDListA(pidl, pszPath)) {                                 ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 1 );                   }                   else {                            ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 0 );                   }         }         else if ( uMsg == BFFM_VALIDATEFAILED ){                   return 1;         }         return 0; }

这段代码中，我们主要用的是BFFM_SELCHANGED == uMsg这段。
这段的主要思想是：用户点击的那个文件夹，我们可以获取pidl，但是如果之后我们不能获取pidl对应的文件夹路径，我们的逻辑还是有问题。所有，在用户点击了一个文件夹后，我们在会立即检查该文件夹的pidl是否可以拿到。如果可以拿到，那么我们就让选择框的OK按钮置成可用，否则不可用。这种思想是预防于未来，我觉的还是很赞的。
但是这段代码还是不健壮的。在win32位机子上，我们发现了一个特殊的场景：就是pidl可以获得文件夹路径，但是该文件夹不可访问。导致我们设置后，无法打开这个文件夹，导致之后要将文件保存到该目录下失败。这个是个非常严重的问题。其实这个问题还是很常见的，我们永远无法预测神奇的用户诡异的行为：比如他把A目录设置为只读，然后通过我们程序去选择这个目录，导致我们无法成功在该文件夹下新建文件——因为该文件夹只读。那么这个时候，我们需要做到:在用户选择时，判断该文件夹我们是否可以写入，如果可以写入，则OK按钮置为可用，否则置为不可用。
所以要将

if (SHGetPathFromIDListA(pidl, pszPath)) {          ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 1 );}else {    ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 0 );}

改成

if (SHGetPathFromIDListA(pidl, pszPath)) {         HANDLE hFile = CreateFileA(pszPath, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL  );    if ( INVALID_HANDLE_VALUE == h )  {                   ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 0 );    }    else {        CloseHandle(hFile);        hFile = NULL;        ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 1 );    }}else {    ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 0 );}

假如你认为一切已经大功告成，那就错了。后来我们又发现，“新建文件夹”按钮无法和“确定”按钮同步。

我目前还没找到一个优雅的控制“新建文件夹”按钮的方法，只能通过枚举子窗口，同时在子窗口中寻找“（”和“）”来识别和控制“新建文件夹”按钮。于是整套完成的流程是

int CALLBACK BrowseCallbackProcSetting(HWND hwnd, UINT uMsg, LPARAM lParam, LPARAM lpData)  {  if  ( BFFM_INITIALIZED == uMsg ) {  ::SendMessage(hwnd,BFFM_SETSELECTION,TRUE,lpData);  }  else if ( BFFM_SELCHANGED == uMsg ){        HWND hNewFloderButton = NULL;         HWND hChild = GetWindow(hwnd, GW_CHILD);        int nMaxCount = 64;        while ( hChild && nMaxCount > 0 ) {            // 控制循环次数，以免死循环，保守性编程            nMaxCount--;            WCHAR wszBuffer[MAX_PATH] = {0};            int nCount = GetClassName( hChild, wszBuffer, MAX_PATH );            std::wstring wstrClassName( wszBuffer, nCount);             if ( 0 == wstrClassName.compare(L"Button") ) {                memset(wszBuffer, 0, MAX_PATH);                nCount = GetWindowText(hChild, wszBuffer, MAX_PATH);                std::wstring wstrText(wszBuffer, nCount);                // 不同操作系统上，显示不一样，比如Win7 64bit是(&M)                if ( -1 != wstrText.find(L"(")                    && -1 != wstrText.find(L")") ) {                    // 新建文件夹按钮                    hNewFloderButton = hChild;                    break;                }            }            hChild = GetNextWindow(hChild, GW_HWNDNEXT);        }char pszPath[MAX_PATH] = {0};LPITEMIDLIST pidl = (LPITEMIDLIST)(lParam);if ( NULL == pidl ) {return 0;}if (SHGetPathFromIDListA(pidl, pszPath)) {     HANDLE hFile = CreateFileA(pszPath, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_FLAG_BACKUP_SEMANTICS, NULL  );if ( INVALID_HANDLE_VALUE == hFile )  {                if ( NULL != hNewFloderButton ) {                    ::EnableWindow(hNewFloderButton, FALSE);                }                ::SendMessage(hwnd, BFFM_ENABLEOK, 0, 0 );}else {CloseHandle(hFile);hFile = NULL;                if ( NULL != hNewFloderButton ) {                    ::EnableWindow(hNewFloderButton, TRUE);                }::SendMessage(hwnd, BFFM_ENABLEOK, 0, 1 );}}else {            if ( NULL != hNewFloderButton ) {                ::EnableWindow(hNewFloderButton, FALSE);            }::SendMessage(hwnd, BFFM_ENABLEOK, 0, 0 );}}else if ( uMsg == BFFM_VALIDATEFAILED ){return 1;}return 0; }

std::string ExportFodler(){char pszPath[MAX_PATH] = {0};  BROWSEINFOA bi;   bi.hwndOwner      = GetHwnd();  bi.pidlRoot       = NULL;  bi.pszDisplayName = NULL;   bi.lpszTitle      = "请选择下载位置";   bi.ulFlags        = BIF_NEWDIALOGSTYLE;bi.lpfn           = BrowseCallbackProcSetting;bi.lParam         = 0;  bi.iImage         = 0;   LPITEMIDLIST pidl = SHBrowseForFolderA(&bi);  if ( NULL == pidl )  {      return "";}if ( SHGetPathFromIDListA( pidl, pszPath ) )  {                strcat(pszPath,"\\");return std::string(pszPath);}return "";}