嵌入式开发C语言学习day25-华清作业7.31

思维导图

作业

// 使用父子进程完成两个文件的拷贝
//       父进程拷贝前一半内容
//       子进程拷贝后一半内容
//       子进程结束后退出
//       父进程回收子进程的资源
#include<myhead.h>

#include<myhead.h>

// 父进程执行的文件拷贝部分
void parentCopy(int source_fd, int dest_fd, off_t splitPosition) {
    lseek(source_fd, 0, SEEK_SET);  // 将文件指针重置到文件开头
    char temp[1024];
    off_t bytesRead = 0;
    ssize_t numRead;
    while ((numRead = read(source_fd, temp, sizeof(temp))) > 0 && bytesRead < splitPosition) {
        ssize_t numWritten = write(dest_fd, temp, numRead);
        if (numWritten!= numRead) {
            perror("父进程拷贝文件失败");
            exit(-1);
        }
        bytesRead += numRead;
    }
    printf("父进程拷贝前半文件成功\n");
}

// 子进程执行的文件拷贝部分
void childCopy(int source_fd, int dest_fd, off_t splitPosition) {
    lseek(source_fd, splitPosition, SEEK_SET);
    char temp[1024];
    ssize_t numRead;
    while ((numRead = read(source_fd, temp, sizeof(temp))) > 0) {
        ssize_t numWritten = write(dest_fd, temp, numRead);
        if (numWritten!= numRead) {
            perror("子进程拷贝文件失败");
            exit(-1);
        }
    }
    printf("子进程拷贝后半文件成功\n");
}

int main(int argc, char *argv[]) {
    // 判断输入文件个数是否合规
    if (argc < 3) {
        printf("输入文件名不合法\n");
        return -1;
    }

    // 打开源文件
    int source_fd = open(argv[1], O_RDONLY);
    if (source_fd == -1) {
        perror("打开源文件失败");
        return -1;
    }
    printf("打开源文件成功\n");

    // 打开目标文件
    int dest_fd = open(argv[2], O_WRONLY | O_CREAT | O_TRUNC, 0666);
    if (dest_fd == -1) {
        printf("打开目标文件失败\n");
        close(source_fd);
        return -1;
    }
    printf("打开目标文件成功\n");

    // 获取源文件大小
    struct stat fileStat;
    if (fstat(source_fd, &fileStat) == -1) {
        perror("获取文件状态失败");
        close(source_fd);
        close(dest_fd);
        return -1;
    }

    // 计算分割位置
    off_t splitPosition = fileStat.st_size / 2;

    // 创建子进程
    pid_t pid = fork();

    if (pid == -1) {
        perror("创建子进程失败");
        close(source_fd);
        close(dest_fd);
        return -1;
    }

    // 父进程
    if (pid > 0) {
        printf("我是父进程 我的 pid 为 %d\n", getpid());
        parentCopy(source_fd, dest_fd, splitPosition);
    }

    // 子进程
    if (pid == 0) {
        printf("我是子进程 我的 pid 为 %d\n", getpid());
        childCopy(source_fd, dest_fd, splitPosition);
        _exit(0);
    }

    // 等待子进程结束
    pid_t pidchild = wait(NULL);
    printf("子进程结束\n");

    // 关闭文件
    close(source_fd);
    close(dest_fd);
    printf("关闭文件成功\n");

    return 0;
}