0

我在处理这段代码时遇到了问题。当我执行“PrepareEncryption”(例如)时,代码返回一个有效的指针,但是当我将它传递给“EncryptBig”时它不再有效(指向随机数)。我最好的选择是删除原始结构。那么,如果这是问题所在,我该如何保存它呢?而且我知道顺便说一句有内存泄漏。

struct filecrypt
{
    FILE* bestand;
    FILE* nieuwbstnd;
    unsigned int positie;
    unsigned int size;
    unsigned int huidig;
    float procentum;
};

struct filecrypt *PrepareEncryption(char* locatie)
{
    struct stat file_status;
    struct filecrypt origineel, *mirror;
    int error;
    char* nieuw;

    if (stat(locatie, &file_status) != 0)
        return NULL;

    error = fopen_s(&origineel.bestand, locatie, "rb");
    if (error != 0)
        return NULL;

    error = strlen(locatie)+5;
    nieuw = (char*)malloc(error);
    if (nieuw == NULL)
        return NULL;
    strcpy_s(nieuw, error-3, locatie);
    strcat_s(nieuw, error, ".cpt");

    error = fopen_s(&origineel.nieuwbstnd, nieuw, "wb+");
    if (error != 0)
        return NULL;

    origineel.huidig = 0;
    origineel.positie = 0;
    origineel.procentum = 0.0f;
    origineel.size = file_status.st_size;
    mirror = &origineel;
    return mirror;
}

float EncryptBig(struct filecrypt *handle)
{
    int i, index = 0;
    float calc;
    char buf, *bytes = (char*)malloc(10485760); // 10 MB
    if (bytes == NULL)
    {
        handle = NULL;
        fcloseall();
        return -1.0f;
    }

    for (i = handle->huidig; i < (handle->huidig+10485760); i++)
    {
        if (i > handle->size)
            break;

        fseek(handle->bestand, i, SEEK_SET);
        fread_s(&buf, 1, 1, 1, handle->bestand);

        __asm
        {
            mov         eax, dword ptr [bytes]  
            add         eax, dword ptr [index]  
            mov         cl, byte ptr [buf]
            xor         cl, 18
            xor         cl, 75
            not         cl
            mov         byte ptr [eax], cl 
            mov         eax, dword ptr [index]  
            add         eax, 1  
            mov         dword ptr [index], eax
        }
    }

    fwrite(bytes, 1, i, handle->nieuwbstnd);
    fseek(handle->nieuwbstnd, i, SEEK_SET);

    handle->huidig += i;
    calc = (float)handle->huidig;
    calc /= (float)handle->size;
    calc *= 100.0f;

    if (calc == 100.0)
    {
        // GEHEUGEN LEK!
        // MOET NOG BIJGEWERKT WORDEN!
        fcloseall();
        handle = NULL;
    }
    return calc;
}

void example(char* path)
{
    float progress;
    struct filecrypt* handle;

    handle = PrepareEncryption(path);
    do
    {
        progress = EncryptBig(handle);
        printf_s("%f", progress);
    }
    while (handle != NULL);
}
4

3 回答 3

4

这是因为您返回一个指向局部变量的指针。

局部变量存储在堆栈中,当一个函数返回时,堆栈的该区域被其他函数重用,并且您留下一个指针,该指针现在指向未使用的内存或现在被其他东西占用的内存。这是未定义的行为,有时可能会起作用,有时可能会给您“垃圾”数据,有时可能会崩溃。

于 2012-11-23T15:32:12.143 回答
0

在 PrepareEncryption 中,您将返回指向 的指针struct filecrypt origineel,该指针分配在堆栈(本地对象)上。这就是问题。就在函数返回(结束执行)之后,占用的内存origineel变得无效。您需要通过调用 malloc 在堆上分配它。

于 2012-11-23T15:32:46.070 回答
0

你做:

mirror = &origineel;
return mirror;

origineel是一个局部变量。以上等价于:

return &origineel;

...您正在返回一个指向局部变量的指针,该指针在函数末尾超出范围。它有时似乎返回一个有效指针的事实只是机会。

使用malloc,或者更好的是,将指向目标位置的指针地址作为参数传递给您的函数并且不返回它:

int *PrepareEncryption(struct filecrypt *origineel, char* locatie);

struct filecrypt myStruct;
PrepareEncryption(&myStruct, "abc");
于 2012-11-23T15:33:59.913 回答