随手写个非递归枚举所有文件的程序, 大家评评

三月廿七

void EnumFiles(const char* directory, list<string>& fileList)
{
     char path[256];
     sprintf(path, "%s\\%s", directory, "*.*");
   
     WIN32_FIND_DATA finddata;
     HANDLE handle = FindFirstFile(path, &finddata);
     
     do
     {
          if (strcmp(finddata.cFileName, ".") &&
              strcmp(finddata.cFileName, "..")
            )
          {
               string s = finddata.cFileName;
               fileList.push_back(s);
          }
   
     }while(FindNextFile(handle, &finddata));\

     FcloseHandle(handle );
}

EnumFilesI(const char* directory)
{
     stack<string> direstoryStack;
     direstoryStack.push(directory);

     while (!direstoryStack.empty())
     {
          list<string> fileList;
          string s = direstoryStack.top();
          direstoryStack.pop();

          EnumFiles(s.c_str(), fileList);

          for (list<string>::iterator it = fileList.begin(); it != fileList.end(); it++)
          {
               string s2 = *it;
               char path[256];

               sprintf(path, "%s\\%s", s.c_str(), s2.c_str());
               WIN32_FIND_DATA finddata;
               HANDLE handle = FindFirstFile(path, &finddata);
               
               if (handle.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
               {
                    direstoryStack.push(path);
               }
               else
               {
                    OutputDebugString(path);
               }
          }
     }
}

AD8018

char path[256];
------------------------

MAX_PATH一般是260吧，用256并且不加长度限制，
貌似可以构造特殊的目录， 成为一个缓冲区溢出的安全漏洞

OwnWaterloo

回复 1# 三月廿七

1. 记得就最近LZ你就有个帖子，标题是算法，内容是STL搓

1.1 看见标题中的非递归再看LZ的ID就猜到应该会有这样的代码：

     direstoryStack.push(directory);
     while (!direstoryStack.empty())
     {
...
          string s = direstoryStack.top();
          direstoryStack.pop();
...

1.2. char*与std :: string的混用太喜感了


2. A版本的sprintf,strcmp,string与T版本的WIN32_FIND_DATA,FindFirstFile,FindNextFile混用也很喜感
好奇LZ平时编译代码的首选编译器是什么？VC6还是mingw？

nketc

char* and std::string tegether may be to avoid memory management.

sprintf and FindFirstFile's "T" version, if you not define UNICODE, it's not  a problem. Just as OW say, VC6 usually not define UNICODE.

tempname2

试试这个：有一颗源码树，复制这个目录结构，把其中的.c文件重命名为.txt文件,其余的文件不用管。

三月廿七

nketc 发表于 2012-05-12 10:08
char* and std::string tegether may be to avoid memory management.
sprintf and FindFirstFile's "T" ...

这就是 工业代码 和 学术代码的区别，

三月廿七

OwnWaterloo 发表于 2012-05-11 20:34
回复 1# 三月廿七

1. 记得就最近LZ你就有个帖子，标题是算法，内容是STL搓

1. 我一开始用 stack<char*>， 结果 stack.pop() 没有真正的pop， 所以我就换了一个pop之后没有作用的 string
    谁知道 std::stack里面的干的什么事？

2, 很多第三方库不支持 UNICODE， 如果想要集成第三方库，就得使用char
   

三月廿七

这段代码我在 vs2003、2005、2008里面跑的很正常， 并且经过强力测试
我不管他什么 char* / string,  A版/T版， 只要这段代码能够保证产品速度和质量，就是好代码

我也很好奇，你平时用的是什么编译器？ 这点常识都没有

OwnWaterloo

老规矩，拍照先：

三月廿七 发表于 2012-05-12 13:22
这就是 工业代码 和 学术代码的区别，

三月廿七 发表于 2012-05-12 13:29
1. 我一开始用 stack<char*>， 结果 stack.pop() 没有真正的pop， 所以我就换了一个pop之后没有作用的 string
    谁知道 std::stack里面的干的什么事？

2, 很多第三方库不支持 UNICODE， 如果想要集成第三方库，就得使用char

三月廿七 发表于 2012-05-12 13:31
这段代码我在 vs2003、2005、2008里面跑的很正常， 并且经过强力测试

我不管他什么 char* / string,  A版/T版， 只要这段代码能够保证产品速度和质量，就是好代码

速度归速度，但质量？笑翻了

OwnWaterloo

回复 4# nketc

还是回复你吧。LZ那种工业代码我写不来


>> char* and std::string tegether may be to avoid memory management.

是avoid内存管理，还是avoid无知。你自己看~


>> sprintf and FindFirstFile's "T" version, if you not define UNICODE, it's not  a problem. Just as OW say, VC6 usually not define UNICODE.

it's not  a problem有个前提，就是if you not define UNICODE。
如果用一致的，比如sprintf+FindFirstFileA，那正确性就与这个前提无关。

其实很早之前，对这种问题我自己知道保持一致，对其他不一致的代码也当作没看见。
反正编译器能检查出来，会报告错误（C里面貌似是警告），到时候写不一致代码的人自己就知道去找相关信息。
直到大二，一个学弟的不一致代码出现了问题，因为那个API（记得是注册表相关的）的无论A还是W版的参数都是void*，编译器没法检查出来。
才觉得这问题必须引起重视，否则遇见这种API又出了问题很难查清原由。