added missing deallocation to why2-test

I forgot to deallocate encrypted text before decryption
This commit is contained in:
Václav Šmejkal 2022-05-29 17:00:43 +02:00
parent 4e0a0c7a27
commit 483194b52c

View File

@ -7,7 +7,8 @@
int main(void) int main(void)
{ {
//VARIABLES //VARIABLES
char *buffer = malloc(128); char *textBuffer = malloc(128);
char *keyBuffer;
int exitCode = 0; int exitCode = 0;
//FLAGS //FLAGS
@ -19,18 +20,23 @@ int main(void)
//SET KEY_LENGTH TO 100 //SET KEY_LENGTH TO 100
setKeyLength(100); setKeyLength(100);
keyBuffer = malloc(getKeyLength());
//ENCRYPT & DECRYPT //ENCRYPT & DECRYPT
outputFlags encrypted = encryptText(TEST_TEXT, NULL, flags); outputFlags encrypted = encryptText(TEST_TEXT, NULL, flags);
strcpy(buffer, encrypted.outputText); //GET ENCRYPTED TEXT strcpy(textBuffer, encrypted.outputText); //GET ENCRYPTED TEXT
strcpy(keyBuffer, encrypted.usedKey); //GET KEY
encrypted = decryptText(encrypted.outputText, encrypted.usedKey, flags); //DEALLOCATE BUFFER
deallocateOutput(encrypted);
encrypted = decryptText(textBuffer, keyBuffer, flags);
//COMPARE DIFFERENCE //COMPARE DIFFERENCE
if (strcmp(encrypted.outputText, TEST_TEXT) == 0) if (strcmp(encrypted.outputText, TEST_TEXT) == 0)
{ {
printf("Test successful!\n\nTEXT: %s\nOUTPUT: %s\nKEY: %s\n", TEST_TEXT, buffer, encrypted.usedKey); printf("Test successful!\n\nTEXT: %s\nOUTPUT: %s\nKEY: %s\n", TEST_TEXT, textBuffer, encrypted.usedKey);
} }
else else
{ {
@ -39,7 +45,8 @@ int main(void)
} }
//DEALLOCATION //DEALLOCATION
free(buffer); free(textBuffer);
free(keyBuffer);
deallocateOutput(encrypted); deallocateOutput(encrypted);
return exitCode; return exitCode;