使用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语言中进行文件操作并处理常见的错误场景。通过细致的错误处理和用户交互，程序在保障功能正确性的同时提升了用户体验。