动手实现linux中的cp命令(可自行拓展)

63 阅读 0 评论 42 点赞

我是靠谱客的博主可靠小蚂蚁，最近开发中收集的这篇文章主要介绍动手实现linux中的cp命令(可自行拓展)，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

我们在学习系统编程的时候，一定会有这样的经历，让你动手实现一个简单的cp命令，也就是拷贝相应的文件到对应的目录，或者说是复制吧，当然，实现非常的简单，我们来看看源码吧:

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
int main(int argc ,char *argv[])
{
if(argc != 3)
{
fprintf(stderr , "input argument errorn");
return -1 ;
}
int
src_fd ;
int
dest_fd ;
src_fd = open(argv[1] , O_RDONLY);
if(-1 == src_fd)
{
perror("mycy srcfile error");
return -2 ;
}
char ch ;
dest_fd = open(argv[2] , O_WRONLY | O_CREAT | O_EXCL
, 0777);
if(dest_fd < 0)
{
//文件存在，判断是否覆盖
printf("Over write ? Y/Nn");
ch = getchar();
if(ch == 'Y')
{
close(dest_fd);
dest_fd = open(argv[2] , O_WRONLY | O_TRUNC);
}else
{
return
0 ;
}
}
//文件不存在，直接创建成功
char buffer[10] = {0} ;
int ret ;
while(1)
{
ret = read(src_fd , buffer , 10);
write(dest_fd , buffer , ret);
if(ret != 10)
break;
}
close(dest_fd);
close(src_fd);
return 0 ;
}