🦄 refactor(C13, C14, C115): 更新了linux系统开发部分的结构
This commit is contained in:
125
C14-文件系统/mydu.c
Normal file
125
C14-文件系统/mydu.c
Normal file
@@ -0,0 +1,125 @@
|
||||
#include <glob.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/stat.h>
|
||||
#include <sys/types.h>
|
||||
|
||||
#define PATHSIZE 1024
|
||||
|
||||
static int path_noloop(const char *path)
|
||||
{
|
||||
char *pos;
|
||||
|
||||
/**
|
||||
* @brief strrchr
|
||||
* @details
|
||||
*
|
||||
* 找到 path 中最后一个 '/' 字符的位置
|
||||
*/
|
||||
pos = strrchr(path, '/');
|
||||
if (NULL == pos)
|
||||
exit(1);
|
||||
|
||||
if ((0 == strcmp(pos + 1, ".") || 0 == strcmp(pos + 1, "..")))
|
||||
return 0;
|
||||
|
||||
return 1;
|
||||
}
|
||||
|
||||
static int64_t mydu(const char *path)
|
||||
{
|
||||
// 优化递归
|
||||
// 将部分与递归无关的变量拿出栈,用 static 修饰
|
||||
static struct stat statres;
|
||||
static char nextpath[PATHSIZE];
|
||||
glob_t globres;
|
||||
int64_t sum;
|
||||
int i;
|
||||
|
||||
if (lstat(path, &statres) < 0)
|
||||
{
|
||||
perror("lstat()");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (!S_ISDIR(statres.st_mode))
|
||||
{
|
||||
/* 非目录 */
|
||||
return statres.st_blocks;
|
||||
/*
|
||||
* st_blocks 代表磁盘块数量,一个磁盘块大小为512字节
|
||||
* 所以除以2才表示多少个1k字节
|
||||
*/
|
||||
}
|
||||
|
||||
/* 目录 */
|
||||
/* 例如:/aaa/bbb/ccc/ddd/eee/fff */
|
||||
|
||||
strncpy(nextpath, path, PATHSIZE);
|
||||
strncat(nextpath, "/*", PATHSIZE); //[W]
|
||||
// if (glob(nextpath, 0, NULL, &globres))
|
||||
// {
|
||||
// perror("glob(), 1");
|
||||
// exit(1);
|
||||
// }
|
||||
glob(nextpath, 0, NULL, &globres);
|
||||
|
||||
// !!! 要考虑 . 和 .. 目录
|
||||
strncpy(nextpath, path, PATHSIZE);
|
||||
strncat(nextpath, "/.*", PATHSIZE); //[W]
|
||||
/*
|
||||
* 这个警告是因为 strncat 函数的第三个参数(即 PATHSIZE)等于目标字符串
|
||||
* nextpath 的大小。这可能会导致 nextpath 缓冲区溢出,因为 strncat
|
||||
* 函数会在目标字符串的末尾添加一个 null 字符('\0')。
|
||||
*/
|
||||
|
||||
// !!! APPEND
|
||||
// if (glob(nextpath, GLOB_APPEND, NULL, &globres))
|
||||
// {
|
||||
// perror("glob(), APPEND");
|
||||
// exit(1);
|
||||
// }
|
||||
glob(nextpath, GLOB_APPEND, NULL, &globres);
|
||||
|
||||
sum = statres.st_blocks;
|
||||
|
||||
for (i = 0; i < globres.gl_pathc; i++)
|
||||
{
|
||||
if (path_noloop(globres.gl_pathv[i]))
|
||||
{
|
||||
sum += mydu(globres.gl_pathv[i]);
|
||||
}
|
||||
}
|
||||
|
||||
// !!!
|
||||
globfree(&globres);
|
||||
return sum;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief mydu
|
||||
* @details
|
||||
* Usage: mydu [FILE]...
|
||||
*
|
||||
* 用到了递归的思想
|
||||
*
|
||||
* !!! 未考虑权限问题
|
||||
*
|
||||
* @param argc
|
||||
* @param argv
|
||||
*
|
||||
* @return int
|
||||
*/
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
if (argc < 2)
|
||||
{
|
||||
fprintf(stderr, "Usage: mydu [FILE]...\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
printf("%lld\n", mydu(argv[1]) / 2);
|
||||
|
||||
exit(0);
|
||||
}
|
||||
Reference in New Issue
Block a user