diff --git a/example.cpp b/example.cpp index 8ec26e0..6006cc9 100644 --- a/example.cpp +++ b/example.cpp @@ -3,33 +3,35 @@ #include #include +using namespace std; + int main() { try { - std::cout << "=== Asymmetric Cryptography Example ===" << std::endl; - std::string generate_new_keys; - std::cout << "Generate new keys? (y/n): "; - std::cin >> generate_new_keys; + cout << "=== Asymmetric Cryptography Example ===" << endl; + string generate_new_keys; + cout << "Generate new keys? (y/n): "; + cin >> generate_new_keys; - std::string message; - std::vector encrypted; + string message; + vector encrypted; if (generate_new_keys == "y") { // 1. Generate keys - std::cout << "1. Generating key pair...\n\n"; + cout << "1. Generating key pair...\n\n"; auto keys = CryptoLib::generate_keys(2048); auto& private_key = keys.first; auto& public_key = keys.second; // 2. Save keys to files - std::cout << "2. Saving keys to files...\n\n"; + cout << "2. Saving keys to files...\n\n"; CryptoLib::save_private_key(*private_key, "private_key.pem"); CryptoLib::save_public_key(*public_key, "public_key.pem"); message = "Hello, this is a secret message!"; } else { - std::cout << "Enter message to decrypt (if exists): "; - std::cin.ignore(); - std::getline(std::cin, message); + cout << "Enter message to decrypt (if exists): "; + cin.ignore(); + getline(cin, message); if (!message.empty()) { encrypted = CryptoLib::base64_to_bytes(message); @@ -39,35 +41,35 @@ int main() { } // 3. Load keys from files - std::cout << "3. Loading keys from files...\n\n"; + cout << "3. Loading keys from files...\n\n"; auto loaded_private_key = CryptoLib::load_private_key("private_key.pem"); auto loaded_public_key = CryptoLib::load_public_key("public_key.pem"); // 4. Convert public key to string and back - std::cout << "4. Loaded keys:\n"; - std::string private_key_str = CryptoLib::private_key_to_string(*loaded_private_key); - std::string public_key_str = CryptoLib::public_key_to_string(*loaded_public_key); - std::cout << "Private key string as: " << private_key_str.substr(0, 50) << "...\n"; - std::cout << "Public key string as: " << public_key_str.substr(0, 50) << "...\n\n"; + cout << "4. Loaded keys:\n"; + string private_key_str = CryptoLib::private_key_to_string(*loaded_private_key); + string public_key_str = CryptoLib::public_key_to_string(*loaded_public_key); + cout << "Private key string as: " << private_key_str.substr(0, 50) << "...\n"; + cout << "Public key string as: " << public_key_str.substr(0, 50) << "...\n\n"; - std::cout << "5. Convert public key from string...\n\n"; + cout << "5. Convert public key from string...\n\n"; auto restored_public_key = CryptoLib::public_key_from_string(public_key_str); if (encrypted.empty()) { // 6. Encrypt data - std::cout << "6. Testing ecnryption..." << std::endl; - std::cout << "Original message: " << message << std::endl; + cout << "6. Testing ecnryption..." << endl; + cout << "Original message: " << message << endl; encrypted = CryptoLib::encrypt(*loaded_public_key, message); - std::cout << "Encrypted (base64): " << CryptoLib::bytes_to_base64(encrypted) << std::endl; - std::cout << "Encrypted data size: " << encrypted.size() << " bytes\n\n"; + cout << "Encrypted (base64): " << CryptoLib::bytes_to_base64(encrypted) << endl; + cout << "Encrypted data size: " << encrypted.size() << " bytes\n\n"; } // 7. Decrypt data - std::string decrypted_message = CryptoLib::decrypt(*loaded_private_key, encrypted); - std::cout << "7. Decrypted message: " << decrypted_message << "\n"; + string decrypted_message = CryptoLib::decrypt(*loaded_private_key, encrypted); + cout << "7. Decrypted message: " << decrypted_message << "\n"; - } catch (const std::exception& e) { - std::cerr << "Error: " << e.what() << std::endl; + } catch (const exception& e) { + cerr << "Error: " << e.what() << endl; return 1; }