【Linux】利用文件操作的系统调用实现普通文件的cp命令

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>
#include<assert.h>

#include<fcntl.h>
#include<sys/stat.h>

//有可能的情况
// argv[0] = "./mycp"
//路经+源文件 argv[1] = "./main.c"
//目录路径 argv[2] = ".."
int main(int argc, char *argv[])
{
	if (argc < 3)
	{
		printf("mycp error, parameter is not enough\n");
		exit(0);
	}
	int fr = open(argv[1], O_RDONLY);//打开的是源文件
	if (fr == -1)
	{
		perror("open error ");
		exit(0);
	}
	//参考Linux下的stat函数
	struct stat st;
	int n = stat(argv[2], &st);
	char path[128] = { 0 };

	strcpy(path, argv[2]);//包含路径

	if (n != -1 && S_ISDIR(st.st_mode))//为目录文件，就可以直接加文件名
	{
		char *p = argv[1] + strlen(argv[1]);//从最后一个字符
		while (p != argv[1] && *p != '/')
		{
			p--;
		}
		strcat(path, "/");
		strcat(path, p);
	}

	int fw = open(path, O_WRONLY | O_CREAT | O_TRUNC, 0064);//open打开的是普通文件，如果给个目录路经，就会打开失败
	if (fw == -1)
	{
		perror("open2 error ");
		exit(0);
	}

	while (1)
	{
		char buff[128] = { 0 };
		int num = read(fr, buff, 127);
		if (num <= 0)
		{
			break;
		}

		int res = write(fw, buff, num);
		if (res <= 0)
		{
			break;
		}
	}
	close(fr);
	close(fw);
}