将指针移交给C中的另一个函数

时间:2012-11-23 15:28:03

标签: c function pointers struct

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

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);
}

3 个答案:

答案 0 :(得分:5)

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

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

答案 1 :(得分:1)

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

答案 2 :(得分:1)

你这样做:

mirror = &origineel;
return mirror;

origineel是一个局部变量。以上相当于:

return &origineel;

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

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

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

struct filecrypt myStruct;
PrepareEncryption(&myStruct, "abc");