首页 > 代码库 > 实现简单的cp命令

实现简单的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.因为是简单的复制命令,所以并没有参数可支持。

实现简单的cp命令