diff --git a/src/lib/decrypter.c b/src/lib/decrypter.c index 252ba9c..2e669f8 100644 --- a/src/lib/decrypter.c +++ b/src/lib/decrypter.c @@ -45,7 +45,7 @@ outputFlags decryptText(char *text, char *keyNew, inputFlags flags) char *textBuffer; int textKeyChainLength; int *textKeyChain; - char *key = malloc(strlen(keyNew)); + char *key = malloc(strlen(keyNew) + 1); //COPY keyNew TO key strcpy(key, keyNew); @@ -57,7 +57,7 @@ outputFlags decryptText(char *text, char *keyNew, inputFlags flags) } //SET LENGTH (numberBuffer) - returningText = malloc(numberBuffer); + returningText = malloc(numberBuffer + 1); textKeyChain = malloc(sizeof(int) * numberBuffer); int encryptedTextKeyChain[sizeof(int) * numberBuffer]; textKeyChainLength = numberBuffer; @@ -78,7 +78,7 @@ outputFlags decryptText(char *text, char *keyNew, inputFlags flags) numberBuffer++; } - textBuffer = malloc(numberBuffer); + textBuffer = malloc(numberBuffer + 1); //LOAD textBuffer for (int j = 0; j < strlen(text); j++) diff --git a/src/lib/encrypter.c b/src/lib/encrypter.c index 44dd8dd..58f1012 100644 --- a/src/lib/encrypter.c +++ b/src/lib/encrypter.c @@ -33,7 +33,7 @@ outputFlags encryptText(char *text, char *keyNew, inputFlags flags) } //VARIABLES - char *key = malloc(getKeyLength()); + char *key = malloc(getKeyLength() + 1); char *returningText; char *textBuffer; int *textKeyChain = malloc(sizeof(int) * strlen(text)); diff --git a/src/lib/misc.c b/src/lib/misc.c index 4280755..f76fdc7 100644 --- a/src/lib/misc.c +++ b/src/lib/misc.c @@ -118,7 +118,7 @@ unsigned char checkVersion(inputFlags flags) rewind(fileBuffer); //REWIND fileBuffer (NO SHIT) //SET LENGTH OF buffer - buffer = malloc(bufferSize); + buffer = malloc(bufferSize + 1); //FIX buffer strcpy(buffer, ""); diff --git a/src/lib/test/main.c b/src/lib/test/main.c index 3511c0b..b5282e1 100644 --- a/src/lib/test/main.c +++ b/src/lib/test/main.c @@ -22,7 +22,7 @@ int main(void) //SET KEY_LENGTH TO 100 setKeyLength(100); - keyBuffer = malloc(getKeyLength()); + keyBuffer = malloc(getKeyLength() + 1); //ENCRYPT outputFlags encrypted = encryptText(TEST_TEXT, NULL, flags);