当前位置：首页 > 行业动态 > 正文

c实现 linux du

admin
行业动态
2025-02-05
3

C实现Linux du：该程序通过C语言模拟Linux的du命令，用于查看目录或文件的磁盘使用情况，支持多种选项如-h（以人类可读方式显示）、-s（只显示总大小）等，方便用户了解磁盘空间占用。

在Linux中，du（disk usage）命令用于估算文件和目录的磁盘使用情况，虽然可以使用现有的du命令，但实现一个简化版的du功能可以帮助理解其工作原理，下面是一个用C语言实现简化版du功能的示例代码。

实现步骤

1、获取命令行参数：解析输入的目录路径。

2、递归遍历目录：使用opendir()、readdir()和closedir()函数递归遍历目录结构。

3、计算文件大小：使用stat()系统调用获取文件大小。

4、输出结果：格式化输出每个文件或目录的大小。

c实现 linux du

示例代码

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <dirent.h>
#include <string.h>
void print_usage(const char *prog_name) {
    fprintf(stderr, "Usage: %s [OPTIONS] DIRECTORY
", prog_name);
    fprintf(stderr, "Options:
");
    fprintf(stderr, "  -h, --human-readable   Print sizes in human readable format (e.g., 1K, 234M, 2G).
");
    exit(EXIT_FAILURE);
}
long get_file_size(const char *path) {
    struct stat st;
    if (stat(path, &st) != 0) {
        perror("stat");
        return -1;
    }
    return st.st_size;
}
void du_recursive(const char *basePath, const char *path, int level, int human_readable) {
    struct dirent *entry;
    DIR *dir = opendir(path);
    if (!dir) {
        perror("opendir");
        return;
    }
    while ((entry = readdir(dir)) != NULL) {
        if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0)
            continue;
        char newPath[PATH_MAX];
        snprintf(newPath, sizeof(newPath), "%s/%s", path, entry->d_name);
        struct stat st;
        if (stat(newPath, &st) == -1) {
            perror("stat");
            continue;
        }
        if (S_ISDIR(st.st_mode)) {
            du_recursive(basePath, newPath, level + 1, human_readable);
        } else {
            long size = get_file_size(newPath);
            if (size == -1)
                continue;
            for (int i = 0; i < level; ++i) printf("  ");
            if (human_readable) {
                char human_readable_size[20];
                snprintf(human_readable_size, sizeof(human_readable_size), "%ld", size);
                printf("%s: %s bytes
", newPath + strlen(basePath), human_readable_size);
            } else {
                printf("%s: %ld bytes
", newPath + strlen(basePath), size);
            }
        }
    }
    closedir(dir);
}
int main(int argc, char *argv[]) {
    if (argc < 2) {
        print_usage(argv[0]);
    }
    int human_readable = 0;
    for (int i = 1; i < argc; ++i) {
        if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--human-readable") == 0) {
            human_readable = 1;
        } else {
            const char *path = argv[i];
            du_recursive(path, path, 0, human_readable);
        }
    }
    return 0;
}