利用main函数参数来实现cp命令

利用main函数参数来实现cp命令

#include<stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<fcntl.h>
#include<unistd.h>
#include<getch.h>

int main(int argc,char *argv[])
{
	int in_fd,out_fd,n_chars;
	char buf[4096];

	if(argc!=3)//判断传入主函数参数是否为3个参数
	{
		fprintf(stderr,"格式错误，应为cp x.x x.x\n",*argv);
		return -1;
	}
	if((in_fd=open(argv[1],O_RDONLY))==-1)//判断是否能打开文件，能打开继续，打不开返回-1
		{
			puts("无法打开文件。\n");
			return -1;
		}
	if((out_fd=open(argv[2],O_WRONLY|O_CREAT|O_EXCL,0644))==-1)//判断文件是否存在，若存在则选择
	{
		char cmd=0;
		printf("目标已经存在，是否覆盖y/n");
		cmd=getch();
		if('y'!=cmd&&'Y'!=cmd)
			return 0;
		out_fd=open(argv[2],O_WRONLY|O_CREAT);//文件覆盖
		if(out_fd<0)
		{
			puts("创建文件失败\n");
			return -1;
		}
	}
	while((n_chars=read(in_fd,buf,4096))>0)//每次读4096个字节就写4096字节进目标文件
		if(write(out_fd,buf,n_chars)!=n_chars)
			puts("无法写入\n");
		if(n_chars==-1)
			puts("Read error from\n");
		if(close(in_fd)==-1||close(out_fd)==-1)
			puts("error closing\n");
}