From 09548dabd892d30f56b7ee551c5d5892655814d3 Mon Sep 17 00:00:00 2001
From: Nikola Petrov <nikolape7@gmail.com>
Date: Sun, 4 Feb 2024 20:47:07 +0100
Subject: [PATCH] fix arg_change

---
 include/cryptography.hpp | 1 +
 source/arg_func.cpp      | 2 +-
 source/cryptography.cpp  | 5 +++++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/include/cryptography.hpp b/include/cryptography.hpp
index a3f41f4..672aa0d 100644
--- a/include/cryptography.hpp
+++ b/include/cryptography.hpp
@@ -10,6 +10,7 @@ class Cryptography
 public:
 	Cryptography(std::string password);
 	~Cryptography();
+	void change_pass(std::string password);
 	bool encrypt(Buffer *plain, Buffer *encrypted);
 	bool decrypt(Buffer *encrypted, Buffer *decrypted);
 
diff --git a/source/arg_func.cpp b/source/arg_func.cpp
index c531b10..0a2ed3c 100644
--- a/source/arg_func.cpp
+++ b/source/arg_func.cpp
@@ -287,7 +287,7 @@ void arg_change(Buffer &decrypted_buffer, Buffer &encrypted_buffer, std::string
 		return;
 	}
 
-	crypto.generate_key_and_iv_from_password(new_string.c_str(), new_string.size());
+	crypto.change_pass(new_string);
 	crypto.encrypt(&decrypted_buffer, &encrypted_buffer);
 	encrypted_buffer.save_to_file();
 	printf("Password changed\n");
diff --git a/source/cryptography.cpp b/source/cryptography.cpp
index 84fe3a8..a024f2f 100644
--- a/source/cryptography.cpp
+++ b/source/cryptography.cpp
@@ -16,6 +16,11 @@ Cryptography::~Cryptography()
 {
 }
 
+void Cryptography::change_pass(std::string password)
+{
+	key = password;
+}
+
 bool Cryptography::encrypt(Buffer *plain, Buffer *encrypted)
 {
 	ByteArray in(plain->buffer, plain->buffer + plain->taken);