在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.因為是簡單的復制命令，所以並沒有參數可支持。
