diff options
| -rw-r--r-- | utils/aes128ecb_decrypt_file.cpp | 4 | ||||
| -rw-r--r-- | utils/aes128ecb_encrypt_file.cpp | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/utils/aes128ecb_decrypt_file.cpp b/utils/aes128ecb_decrypt_file.cpp index 788118b..60b1546 100644 --- a/utils/aes128ecb_decrypt_file.cpp +++ b/utils/aes128ecb_decrypt_file.cpp @@ -25,7 +25,9 @@ namespace std::ifstream::pos_type get_file_size(const std::string& path) { - std::ifstream ifs(path, std::ifstream::binary | std::ifstream::ate); + std::ifstream ifs; + ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit); + ifs.open(path, std::ifstream::binary | std::ifstream::ate); return ifs.tellg(); } } diff --git a/utils/aes128ecb_encrypt_file.cpp b/utils/aes128ecb_encrypt_file.cpp index 407dd44..2e24f04 100644 --- a/utils/aes128ecb_encrypt_file.cpp +++ b/utils/aes128ecb_encrypt_file.cpp @@ -25,7 +25,9 @@ namespace std::ifstream::pos_type get_file_size(const std::string& path) { - std::ifstream ifs(path, std::ifstream::binary | std::ifstream::ate); + std::ifstream ifs; + ifs.exceptions(std::ifstream::badbit | std::ifstream::failbit); + ifs.open(path, std::ifstream::binary | std::ifstream::ate); return ifs.tellg(); } } |