95 lines
1.9 KiB
C
95 lines
1.9 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <dirent.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
|
|
int main(int argc, char *argv[]) {
|
|
DIR *dir;
|
|
struct dirent *entry;
|
|
int pack_fd, file_fd;
|
|
char buffer[4096];
|
|
ssize_t bytes_read;
|
|
char filepath[PATH_MAX];
|
|
|
|
// 检查命令行参数
|
|
if (argc != 2) {
|
|
fprintf(stderr, "用法: %s <目录名>\n", argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
// 打开指定目录
|
|
dir = opendir(argv[1]);
|
|
if (dir == NULL) {
|
|
perror("无法打开指定目录");
|
|
return 1;
|
|
}
|
|
|
|
// 创建打包文件
|
|
pack_fd = open("a.pack", O_WRONLY | O_CREAT | O_TRUNC, 0644);
|
|
if (pack_fd == -1) {
|
|
perror("无法创建打包文件");
|
|
closedir(dir);
|
|
return 1;
|
|
}
|
|
|
|
// 遍历目录中的所有文件
|
|
while ((entry = readdir(dir)) != NULL) {
|
|
struct stat st;
|
|
|
|
// 跳过 . 和 .. 目录
|
|
if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) {
|
|
continue;
|
|
}
|
|
|
|
// 构建完整的文件路径
|
|
snprintf(filepath, PATH_MAX, "%s/%s", argv[1], entry->d_name);
|
|
|
|
// 获取文件状态
|
|
if (stat(filepath, &st) == -1) {
|
|
perror("获取文件状态失败");
|
|
continue;
|
|
}
|
|
|
|
// 只处理普通文件
|
|
if (!S_ISREG(st.st_mode)) {
|
|
continue;
|
|
}
|
|
|
|
printf("正在打包: %s\n", entry->d_name);
|
|
|
|
// 写入文件名长度
|
|
int name_len = strlen(entry->d_name);
|
|
write(pack_fd, &name_len, sizeof(int));
|
|
|
|
// 写入文件名
|
|
write(pack_fd, entry->d_name, name_len);
|
|
|
|
// 打开源文件
|
|
file_fd = open(filepath, O_RDONLY);
|
|
if (file_fd == -1) {
|
|
perror("无法打开源文件");
|
|
continue;
|
|
}
|
|
|
|
// 写入文件大小
|
|
int file_size = st.st_size;
|
|
write(pack_fd, &file_size, sizeof(int));
|
|
|
|
// 写入文件数据
|
|
while ((bytes_read = read(file_fd, buffer, sizeof(buffer))) > 0) {
|
|
write(pack_fd, buffer, bytes_read);
|
|
}
|
|
|
|
close(file_fd);
|
|
}
|
|
|
|
close(pack_fd);
|
|
closedir(dir);
|
|
|
|
printf("打包完成，所有文件已打包到 a.pack\n");
|
|
return 0;
|
|
}
|