diff --git a/src/caesarcrypt.c b/src/caesarcrypt.c index 10def7f..3d2d6fd 100644 --- a/src/caesarcrypt.c +++ b/src/caesarcrypt.c @@ -12,6 +12,7 @@ #include "dyninput.h" #include +#include #include @@ -24,11 +25,12 @@ int main() ,stdout ); char buf[4]; + int64_t maxsize = INT64_MAX; fgets(buf, 3, stdin); int choice = strtol(buf, NULL, 10); fputs("Enter your message text: ", stdout); - char* msg = dyninput_str(INT64_MAX); + char* msg = dyninput_str(maxsize); if (!msg) { puts("ERROR: String input failed! Exiting..."); @@ -41,18 +43,18 @@ int main() if (choice == 1) { - caesar_encrypt(msg, 128, key); + caesar_encrypt(msg, maxsize, key); puts("\nYour encrypted message is:"); - for (int i = 0; i < 128 && msg[i]; i++) + for (int i = 0; i < maxsize && msg[i]; i++) { putchar(msg[i]); } } else if (choice == 2) { - caesar_decrypt(msg, 128, key); + caesar_decrypt(msg, maxsize, key); puts("\nYour decrypted message is:"); - for (int i = 0; i < 128 && msg[i]; i++) + for (int i = 0; i < maxsize && msg[i]; i++) { putchar(msg[i]); } diff --git a/src/crypt.c b/src/crypt.c index aa15d0b..adba553 100644 --- a/src/crypt.c +++ b/src/crypt.c @@ -9,7 +9,9 @@ */ #include "crypt.h" + #include +#include /* @@ -43,9 +45,9 @@ inline bool isupper(char c) } -void caesar_encrypt(char* str, int size, int key) +void caesar_encrypt(char* str, int64_t size, int key) { - for (int i = 0; i < size && str[i] != '\n'; i++) + for (int64_t i = 0; i < size && str[i] != '\n'; i++) if (str[i] != ' ') { bool upper = isupper(str[i]); @@ -56,9 +58,9 @@ void caesar_encrypt(char* str, int size, int key) } -void caesar_decrypt(char* str, int size, int key) +void caesar_decrypt(char* str, int64_t size, int key) { - for (int i = 0; i < size && str[i] != '\n'; i++) + for (int64_t i = 0; i < size && str[i] != '\n'; i++) if (str[i] != ' ') { bool upper = isupper(str[i]); diff --git a/src/crypt.h b/src/crypt.h index cfe42dd..3ce733b 100644 --- a/src/crypt.h +++ b/src/crypt.h @@ -9,8 +9,9 @@ */ #pragma once +#include -void caesar_encrypt(char* str, int size, int key); +void caesar_encrypt(char* str, int64_t size, int key); -void caesar_decrypt(char* str, int size, int key); +void caesar_decrypt(char* str, int64_t size, int key);