Linux C++获取文件夹大小

最新推荐文章于 2024-07-23 11:13:08 发布

班公湖里洗过脚

最新推荐文章于 2024-07-23 11:13:08 发布

阅读量3.3k

点赞数 2

分类专栏： linux API 文章标签： linux

linux API 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

http://www.cnblogs.com/emituofo/p/6225403.html

项目中要计算指定文件夹的大小。
百度查到这篇文章，https://my.oschina.net/Tsybius2014/blog/330628
方法可行，运行正确。

拿到我们的项目中，却遇到一些问题：程序中一些读文件的代码，开始报异常，都不到文件。这些都是以前没有遇到过的问题。
到底是什么情况呢？排查了好久，终于发现使用该文章提供的计算文件夹大小的函数（暂且叫做GetDirectorySize），其中有改变当前目录的代码：

chdir(dir);

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

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

 1 #include <stdio.h>
 2 #include <sys/stat.h>
 3 #include <sys/types.h>
 4 #include <unistd.h>
 5 #include <stdlib.h>
 6 #include <dirent.h>
 7 #include <string.h>
 8 
 9 //计算某目录所占空间大小（包含本身的4096Byte）
10 long long int GetDirectorySize(char *dir)
11 {
12     DIR *dp;
13     struct dirent *entry;
14     struct stat statbuf;
15     long long int totalSize=0;
16 
17     if ((dp = opendir(dir)) == NULL)
18     {
19         fprintf(stderr, "Cannot open dir: %s\n", dir);
20         return -1; //可能是个文件，或者目录不存在
21     }
22     
23     //先加上自身目录的大小
24     lstat(dir, &statbuf);
25     totalSize+=statbuf.st_size;
26 
27     while ((entry = readdir(dp)) != NULL)
28     {
29         char subdir[256];
30         sprintf(subdir, "%s/%s", dir, entry->d_name);
31         lstat(subdir, &statbuf);
32         
33         if (S_ISDIR(statbuf.st_mode))
34         {
35             if (strcmp(".", entry->d_name) == 0 ||
36                 strcmp("..", entry->d_name) == 0)
37             {
38                 continue;
39             }
40 
41             long long int subDirSize = GetDirectorySize(subdir);
42             totalSize+=subDirSize;
43         }
44         else
45         {
46             totalSize+=statbuf.st_size;
47         }
48     }
49 
50     closedir(dp);    
51     return totalSize;
52 }
53 
54 int main(int argc, char* argv[])
55 {
56     char* dir = argv[1];
57     long long int  totalSize = GetDirectorySize(dir);    
58     printf("totalSize: %lld\n", totalSize);
59 
60     return 0;
61 }