在Linux下实现简单的cp命令。这是《APUE》第四章的其中一道练习题。其实思路很简单,弄清规则就行了。规则1:源文件必须得存在,否则出错;规则2:目的文件若不存在则创建,若存在,则提示是否覆盖,是就覆盖,不是就重新建一个。下面给出代码:/* 实现简单的cp命令 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>int my_cp(char *argv[]); int main(int argc, char *argv[]) /* argv[1]和argv[2]是文件的路径 */
{
my_cp(argv); return 0;
}int my_cp(char *argv[]) /* 传进来文件的路径 */
{
FILE *fp1, *fp2;
char ch;
char flag; /* 提示是否要覆盖 */
char filename[256]; if ((fp1 = fopen(argv[1], "r")) == NULL) /* 源文件必须得存在,否则出错 */
{
printf("error:the %s doesn"t exist.", argv[1]);
exit(1);
} if ((fp2 = fopen(argv[2], "r")) != NULL) /* 若文件2已存在 */
{
printf("The file %s has been exist, cover? Y/N:", argv[2]);
scanf(" %c", &flag);
fclose(fp2); /* 因为下面无论是if还是else,都要重新打开文件,所以这里先关掉 */ if (flag == "y" || flag == "Y") /* 覆盖文件2 */
{
if ((fp2 = fopen(argv[2], "w")) == NULL)
{
printf("Cannot rewrite!");
exit(1);
}
while ((ch = fgetc(fp1)) != EOF)/* 将文件1的内容复制到文件2上 */
fputc(ch, fp2);
}
else /* 不覆盖文件2,新建一个文件2 */
{
/* 命名文件2为filename(1) */
strcpy(filename, argv[2]);
strcat(filename, "-copy"); if ((fp2 = fopen(filename, "a")) == NULL)
{
printf("Cannot build the file %s.", filename);
exit(1);
}
while ((ch = fgetc(fp1)) != EOF)/* 将文件1的内容复制到文件2上 */
fputc(ch, fp2);
}
}
else /* 文件2不存在,则创建 */
{
//fclose(fp2); /* 不应该要这句,因为前面用r模式打开,要是不存在则会返回NULL指针,fclose(NULL)会出错 */
if ((fp2 = fopen(argv[2], "w")) == NULL)
{
printf("Cannot rewrite!");
exit(1);
}
while ((ch = fgetc(fp1)) != EOF)/* 将文件1的内容复制到文件2上 */
fputc(ch, fp2);
}
fclose(fp1);
fclose(fp2);
}当目的文件存在时,提示是否覆盖:当目的文件不存在时,则创建:不足之处:1.只能支持一对一的复制,不能支持一对多的复制,这个可以改进。2.因为是简单的复制命令,所以并没有参数可支持。APUE习题4.6源代码----实现自己的简易 cp 命令