c - 为什么复制目录的 C 程序代码在逻辑上是错误的？

Question

我正在尝试创建一个复制程序，该程序获取源目录和目标目录并将目录从源复制到目标。在两个目录名都是文件路径的情况下，它可以工作。对于源是文件而目标是目录的情况，它可以工作。然而，对于源和目录都是目录的情况，它不能正常工作，甚至是非多级目录的简单情况。 。请建议版本。

#include<stdio.h>
#include<string.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<dirent.h>
#include<stdlib.h>
#include<unistd.h>

void copyfile(char source[],char destination[]);
void traverse(char source[],char destination[]);
int isdir(char path[]);

int main(int argc,char *argv[])
{
    if(argc<3)
    {
        printf("ERROR: Improper syntax\n");
        printf("\t./copy [source path] [destination path]\n");
        exit(0);
    }

    char *source=(char *)malloc(sizeof(char)*strlen(argv[1])+1);
    char *destination=(char *)malloc(sizeof(char)*strlen(argv[2])+1);

    strcpy(source,argv[1]);
    strcpy(destination,argv[2]);

    if(isdir(source)&&!isdir(destination))
    {
        printf("Destination cannot be a file\n");
        exit(0);
    }   
    if(isdir(source)&&isdir(destination))
        traverse(source,destination);
    else copyfile(source,destination);

    return 0;
}

void copyfile(char source[],char destination[])
{
    int c;

    if(!isdir(source)&&isdir(destination))
    {
        char *fname=strrchr(source,'/');
        if(realloc(destination,strlen(destination)+strlen(source)+2)==NULL)
        {
            printf("Memory reallocation error\n");
            exit(0);
        }
        strcat(destination,fname);  
    }

    FILE *f1,*f2;
    f1=fopen(source,"r");
    f2=fopen(destination,"w");

    if(f1==NULL)
    {
        printf("File does not exists");
        return;
    }

    if(f2==NULL)
    {
        printf("File copying error: Directory may or may not exist\n");
        return;
    }

    while((c=fgetc(f1))!=EOF)
        fputc(c,f2);

    fclose(f1);
    fclose(f2);
    return;
}

int isdir(char path[])
{
    struct stat dir;
    stat(path,&dir);
    if(S_ISREG(dir.st_mode))
        return 0;
    if(S_ISDIR(dir.st_mode))
        return 1;
    printf("The source is not a file, neither a dir\n");
    exit(0);
}

void traverse(char source[],char destination[])
{   
    char *fname=strrchr(source,'/');

    char *tdest=(char *)malloc(strlen(destination)+strlen(fname)+1+6);

    strcpy(tdest,"mkdir ");
    strcat(tdest,destination);
    strcat(tdest,fname);
    system(tdest);
    if(realloc(destination,(strlen(destination)+strlen(fname))*sizeof(char)+1)==NULL)
    {
        printf("Destination rellocaion pointer inturrupted\n");
        exit(0);
    }
    strcat(destination,fname);  

    DIR *d1;
    if((d1=opendir(source))==NULL)
    {
        printf("Source access inturrupted\n");
        exit(0);
    }

    struct dirent *cwd;

    struct stat dir;
    while((cwd=readdir(d1))!=NULL)
    {   
        char *tsource=(char *)malloc((strlen(source)+strlen(cwd->d_name))*sizeof(char)+2);
        stat(cwd->d_name,&dir);
        if(S_ISDIR(dir.st_mode))
        {
            if(strcmp(cwd->d_name,".")==0||strcmp(cwd->d_name,"..")==0)continue;        

            traverse(source,destination);
        }

        strcpy(tsource,source);
        strcat(tsource,"/");
        strcat(tsource,cwd->d_name);
        copyfile(tsource,destination);
    }

    closedir(d1);       
    return;
}

Answer 1

您失败了 realloc :它的返回值必须分配(通常是重新分配)给一个指针。

destination = realloc(destination, (strlen(destination)+strlen(fname))*sizeof(char)+1);

if (destination == NULL)
{
    printf("Destination rellocaion pointer inturrupted\n");
    exit(0);
}

请注意，所有对 malloc 的调用都应避免强制转换返回值，并且所有返回值必须进行 NULL 检查。

<小时/>

正如@Quentin正确评论的那样，请注意，如果realloc失败，则目标指向的内存将被泄漏:该内存的地址将丢失。因此，为了非常安全地做到这一点，您可以使用临时变量:

char *temp = realloc(destination, (strlen(destination)+strlen(fname))*sizeof(char)+1);

if (temp == NULL)
{
    printf("Destination rellocaion pointer inturrupted\n");
    exit(0);
}

destination = temp;

<小时/>

还请注意，所有malloc分配的 block 都必须被释放，即使操作系统会在程序终止时为您释放它。这是编码和避免 future 内存泄漏问题的最佳方法:释放内存，只要该特定内存不再被使用。