实现简单的cp命令

在Linux下实现简单的cp命令。这是《APUE》第四章的其中一道练习题。

其实思路很简单，弄清规则就行了。规则1：源文件必须得存在，否则出错；规则2：目的文件若不存在则创建，若存在，则提示是否覆盖，是就覆盖，不是就重新建一个。

下面给出代码：

 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 }

View Code

当目的文件存在时，提示是否覆盖：

当目的文件不存在时，则创建：

 

不足之处：

1.只能支持一对一的复制，不能支持一对多的复制，这个可以改进。

2.因为是简单的复制命令，所以并没有参数可支持。