获取目录的大小函数linux,Linux C++获取文件夹大小1（通过lstat实现）

拿到我们的项目中，却遇到一些问题：程序中一些读文件的代码，开始报异常，读不到文件。这些问题是以前没有遇到过的。

到底是什么情况呢？排查了好久，终于发现该文章提供的计算文件夹大小的函数(暂且叫做GetDirectorySize)，其中有改变当前目录的代码：

chdir(dir);

我们的项目是多线程的，一个线程调用GetDirectorySize，调用的过程中改变了当前目录，而此时另一个线程使用相对路径去读文件，原来能读到的，现在就读不到了。特别提示chdir改变的是，当前进程(当然包括其下所有线程)的工作目录！！！(具体可以查看线程共享进程的那些资源？)

为了去掉GetDirectorySize的副作用，我重新实现了该函数：

#include

#include

#include

#include

#include

#include

#include

//计算某目录所占空间大小(包含本身的4096Byte)

long long int GetDirectorySize(char *dir)

{

DIR *dp;

struct dirent *entry;

struct stat statbuf;

long long int totalSize=0;

if ((dp = opendir(dir)) == NULL)

{

fprintf(stderr, "Cannot open dir: %s\n", dir);

return -1; //可能是个文件，或者目录不存在

}

//先加上自身目录的大小

lstat(dir, &statbuf);

totalSize+=statbuf.st_size;

while ((entry = readdir(dp)) != NULL)

{

char subdir[256];

sprintf(subdir, "%s/%s", dir, entry->d_name);

lstat(subdir, &statbuf);

if (S_ISDIR(statbuf.st_mode))

{

if (strcmp(".", entry->d_name) == 0 ||

strcmp("..", entry->d_name) == 0)

{

continue;

}

long long int subDirSize = GetDirectorySize(subdir);

totalSize+=subDirSize;

}

else

{

totalSize+=statbuf.st_size;

}

}

closedir(dp);

return totalSize;

}

int main(int argc, char* argv[])

{

char* dir = argv[1];

long long int totalSize = GetDirectorySize(dir);

printf("totalSize: %lld\n", totalSize);

return 0;

}