实现简单的cp命令
在Linux下实现简单的cp命令。这是《APUE》第四章的其中一道练习题。
其实思路很简单,弄清规则就行了。规则1:源文件必须得存在,否则出错;规则2:目的文件若不存在则创建,若存在,则提示是否覆盖,是就覆盖,不是就重新建一个。
下面给出代码:
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
1 /* 实现简单的cp命令 */ 2 #include <stdio.h> 3 #include <stdlib.h> 4 #include <string.h> 5 6 int my_cp(char *argv[]); 7 8 int main(int argc, char *argv[]) /* argv[1]和argv[2]是文件的路径 */ 9 { 10 11 my_cp(argv); 12 13 return 0; 14 } 15 16 int my_cp(char *argv[]) /* 传进来文件的路径 */ 17 { 18 FILE *fp1, *fp2; 19 char ch; 20 char flag; /* 提示是否要覆盖 */ 21 char filename[256]; 22 23 if ((fp1 = fopen(argv[1], "r")) == NULL) /* 源文件必须得存在,否则出错 */ 24 { 25 printf("error:the %s doesn't exist.", argv[1]); 26 exit(1); 27 } 28 29 if ((fp2 = fopen(argv[2], "r")) != NULL) /* 若文件2已存在 */ 30 { 31 printf("The file %s has been exist, cover? Y/N:", argv[2]); 32 scanf(" %c", &flag); 33 34 35 fclose(fp2); /* 因为下面无论是if还是else,都要重新打开文件,所以这里先关掉 */ 36 37 if (flag == 'y' || flag == 'Y') /* 覆盖文件2 */ 38 { 39 if ((fp2 = fopen(argv[2], "w")) == NULL) 40 { 41 printf("Cannot rewrite!"); 42 exit(1); 43 } 44 while ((ch = fgetc(fp1)) != EOF)/* 将文件1的内容复制到文件2上 */ 45 fputc(ch, fp2); 46 } 47 else /* 不覆盖文件2,新建一个文件2 */ 48 { 49 /* 命名文件2为filename(1) */ 50 strcpy(filename, argv[2]); 51 strcat(filename, "-copy"); 52 53 if ((fp2 = fopen(filename, "a")) == NULL) 54 { 55 printf("Cannot build the file %s.", filename); 56 exit(1); 57 } 58 while ((ch = fgetc(fp1)) != EOF)/* 将文件1的内容复制到文件2上 */ 59 fputc(ch, fp2); 60 } 61 } 62 else /* 文件2不存在,则创建 */ 63 { 64 //fclose(fp2); /* 不应该要这句,因为前面用r模式打开,要是不存在则会返回NULL指针,fclose(NULL)会出错 */ 65 if ((fp2 = fopen(argv[2], "w")) == NULL) 66 { 67 printf("Cannot rewrite!"); 68 exit(1); 69 } 70 while ((ch = fgetc(fp1)) != EOF)/* 将文件1的内容复制到文件2上 */ 71 fputc(ch, fp2); 72 73 } 74 75 fclose(fp1); 76 fclose(fp2); 77 }
当目的文件存在时,提示是否覆盖:
当目的文件不存在时,则创建:
不足之处:
1.只能支持一对一的复制,不能支持一对多的复制,这个可以改进。
2.因为是简单的复制命令,所以并没有参数可支持。