C中的内存管理命令行参数

时间:2015-03-05 11:38:26

标签: c command-line-arguments

我正在编写一个简单的程序,它接受命令行参数并将它们存储到char **中。我正在尝试更多地了解内存管理,但无法通过这个简单的绊脚石。我的程序应该将命令行argumetns复制到动态分配的char **中。但是我阵列中的第一个位置总是腐败。下面是代码及其打印内容:

if (strcmp(argv[1], "test") ==0)
{
    test();
}
else
{
    char ** file_names = malloc(10);

    for(int i =0; i < argc-1; ++i)
    {
        file_names[i] = malloc(10);
        strcpy(file_names[i], argv[i+1]);

        printf("%s, %s\n", argv[i+1], file_names[i]);
    }

    printf("____________\n");

    for(int i =0; i < argc-1; ++i)
    {
        printf("%s\n", file_names[i]);
    }
}

出来的是:

what, what
test, test
again, again
wow, wow
____________
pK@??
test
again
wow

有人可以解释为什么会这样吗?感谢

2 个答案:

答案 0 :(得分:7)

此:

char ** file_names = malloc(10);

是一个错误。它试图分配10个字节,这与你需要多少字节完全没有关系。欠分配然后覆盖会给你不明确的行为。

应该是这样的:

char **file_names = malloc(argc * sizeof *file_names);

这通过乘以参数的数量来计算分配的大小(argc,如果你不想存储argv[0],那么当然这应该是(argc - 1)字符指针的大小,表示为sizeof *file_names。由于file_names的类型为char * *,因此*file_names的类型为char *,这是您想要的。这是一般模式,可以经常应用,它可以让您停止重复类型名称。它可以保护您免受错误。

例如比较:

double *floats = malloc(1024 * sizeof(float));  /* BAD CODE */

double *floats = malloc(1024 * sizeof *floats);

如果您认为最初它是float *floats(如命名所示),则第一个变体包含另一个分配不足的错误,而第二个变体“幸免于”类型的变化而没有错误。

然后你需要在假设它之前检查它是否成功。

答案 1 :(得分:1)

您希望为file_names分配适当的内存量,可能更像是:

char ** file_names = malloc(sizeof(char*) * (argc - 1));