检测到堆损坏 | C

Question

当我从控制台调用程序时，我正在用 C 语言制作一个小程序来为我制作 .cpp 和 .h 文件。它正在工作，但是当我使用该程序时，在我按下回车键并生成文件后，我收到一个弹出错误：HEAP CORRUPTION DETECTED。

谁能帮我解决这个问题？

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <Windows.h>
#include <direct.h>

#pragma warning(disable : 4996)

int main(int argc, char*argv[]) {
    if (argc < 2 ) {
        printf("Not enough arguments given: usage [name]");
        return;
    }

    char* name=(char*)malloc(strlen(argv[1]));
    name = argv[1];
    char* name2 = (char*)malloc(strlen(name));
    strcpy(name2, name);

    strcat(name, ".cpp");
    strcat(name2, ".h");

    name[strlen(name) + 1] = '\0';
    name2[strlen(name2)+ 1] = '\0';

    FILE * fp = fopen(name, "w");
    FILE * fp2 = fopen(name2, "w");

    free(name);
    free(name2);

    return 0;
}

Answer 1

你没有分配足够长的时间

char* name=(char*)malloc(strlen(argv[1]));
name = argv[1];
char* name2 = (char*)malloc(strlen(name));
strcpy(name2, name);

strcat(name, ".cpp");
strcat(name2, ".h");

因为“.cpp”和“.h”需要空间而不忘记空终止字符，所以解决方案可以是：

size_t alen = strlen(argv[1]);
char * h = malloc(alen + 3);
char * cpp = malloc(alen + 5);

strcpy(h, argv[1]);
strcpy(h + alen, ".h");
strcpy(cpp, argv[1]);
strcpy(cpp + alen, ".cpp");

还有免费的h和cpp

在

name[strlen(name) + 1] = '\0';
name2[strlen(name2)+ 1] = '\0';

你已经假设有空终止字符能够使用 strlen。 （当然使用我上面的建议这些行不存在）

在

free(name);

你免费argv[1]，你不能。 （当然你的两个初始免费使用我上面的建议不存在）

在

if (argc < 2 ) {
    printf("Not enough arguments given: usage [name]");
    return;
}

你什么都不返回，一个 int 是预期的

您还需要检查 fopen 是否成功，以及 malloc 即使我们通常有足够的内存用于需要少量内存的程序

---

你的 main 可以是：

int main(int argc, char*argv[]) {
    if (argc < 2 ) {
        printf("Not enough arguments given: usage %s [name]\n", *argv);
        return -1;
    }

    size_t alen = strlen(argv[1]);
    char * h = malloc(alen + 3);
    char * cpp = malloc(alen + 5);

    if ((h == NULL) || (cpp == NULL)) {
      puts("not enough memory");
      return -1;
    }

    strcpy(h, argv[1]);
    strcpy(h + alen, ".h");
    strcpy(cpp, argv[1]);
    strcpy(cpp + alen, ".cpp");

    FILE * fph = fopen(h, "w");

    if (fph == NULL) {
      printf("cannot open %s\n", h);
      return -1;
    }

    FILE * fpcpp = fopen(cpp, "w");

    if (fcpp == NULL) {
      printf("cannot open %s\n", cpp);
      return -1;
    }


    free(h);
    free(cpp);

    /* write in files */

    fclose(fh);
    fclose(fcpp);

    return 0;
}