使用C语言实现文件拷贝并处理错误

使用C语言实现文件拷贝并处理错误

在这篇博客中，我将介绍如何使用C语言编写一个简单的文件拷贝程序，并且处理文件操作中的常见错误。这个程序会将源文件的内容复制到目标文件中，如果目标文件已存在，它还会提示用户是否覆盖该文件。

代码实现

以下是完整的代码实现：

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>int main(int argc, const char* argv[]) 
{if (argc != 3) {printf("用法: %s 源文件 目标文件\n", argv[0]);return 1;}int fd_src = open(argv[1], O_RDONLY);if (fd_src < 0) {perror("打开源文件出错");return 1;}int fd_dest = open(argv[2], O_WRONLY | O_CREAT | O_EXCL, 0644);if (fd_dest < 0) {printf("文件已存在！是否覆盖(y/n)？\n");char ch;ch = getchar();while (1) {if (ch == 'y' || ch == 'Y') {fd_dest = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0644);if (fd_dest < 0) {perror("open");close(fd_src);return 1;}break;} else if (ch == 'n' || ch == 'N') {close(fd_src);return 0;} else {printf("输入错误！请输入y或n:\n");ch = getchar();}}}char *buf = malloc(1024);if (buf == NULL) {perror("malloc");close(fd_src);close(fd_dest);return 1;}ssize_t ret;while ((ret = read(fd_src, buf, 1024)) > 0) {if (write(fd_dest, buf, ret) != ret) {perror("write");free(buf);close(fd_src);close(fd_dest);return 1;}}if (ret < 0) {perror("open");}free(buf);close(fd_src);close(fd_dest);printf("文件拷贝成功！\n");return 0;
}

代码分析

这个程序分为几个主要部分：命令行参数检查、文件打开、用户交互、文件读取与写入、错误处理。

1. 命令行参数检查

   • 程序首先检查命令行参数的数量。如果用户没有提供源文件和目标文件路径，程序会显示使用提示并退出。

2. 打开源文件

   • 使用open函数以只读模式(O_RDONLY)打开源文件。如果打开失败，perror函数会输出错误信息并退出程序。

3. 打开目标文件

   • 使用open函数以写入模式、创建模式(O_WRONLY | O_CREAT | O_EXCL)打开目标文件。如果目标文件已存在，程序会提示用户是否覆盖。
   • 如果用户选择覆盖文件，使用O_TRUNC标志重新打开文件。如果用户选择不覆盖，程序直接退出。

4. 读取与写入文件

   • 分配一个1024字节的缓冲区，并循环读取源文件内容到缓冲区，再将缓冲区的数据写入目标文件。
   • 如果读取或写入失败，程序会输出错误信息，并进行资源释放。

5. 错误处理与资源管理

   • 所有文件操作都经过错误检查，并在发生错误时释放已分配的资源，确保文件描述符正确关闭，内存释放。

运行示例

$ ./file_copy source.txt destination.txt
文件拷贝成功！

当destination.txt文件已存在时，程序会提示：

$ ./file_copy source.txt destination.txt
文件已存在！是否覆盖(y/n)？

总结

这个文件拷贝程序展示了如何在C语言中进行文件操作并处理常见的错误场景。通过细致的错误处理和用户交互，程序在保障功能正确性的同时提升了用户体验。