nikola/password_manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add LoginInfoPointer Struct

Browse Source
This commit is contained in:
Nikola Petrov 2023-08-27 20:18:47 +02:00
parent a900289dbf
commit 934c68e6c2
5 changed files with 38 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Password_manager/include/arg_func.h
View File

@ -1,6 +1,6 @@
#pragma once #pragma once
struct Pass; struct LoginInfoPointer;
class Buffer; class Buffer;
class Cryptography; class Cryptography;
#include <string> #include <string>
@ -22,9 +22,11 @@ void print_args();
Arg get_args(int argc, char** argv, char** label); Arg get_args(int argc, char** argv, char** label);
const char* arg_get(Buffer& decrypted_buffer, const char* label); LoginInfoPointer arg_get(Buffer& decrypted_buffer, const char* label);
const char* arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto); LoginInfoPointer arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto);
LoginInfoPointer arg_input(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto);
void arg_list(Buffer& decrypted_buffer); void arg_list(Buffer& decrypted_buffer);
@ -32,8 +34,6 @@ void arg_delete(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char*
void arg_print_all_p(Buffer& decrypted_buffer, std::string& user_pass); void arg_print_all_p(Buffer& decrypted_buffer, std::string& user_pass);
const char* arg_input(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto);
void arg_change(Buffer& decrypted_buffer, Buffer& encrypted_buffer, std::string& user_pass, Cryptography& crypto); void arg_change(Buffer& decrypted_buffer, Buffer& encrypted_buffer, std::string& user_pass, Cryptography& crypto);
void arg_show(Buffer& decrypted_buffer, const char* label); void arg_show(Buffer& decrypted_buffer, const char* label);

10
Password_manager/include/func.h
View File

@ -9,6 +9,14 @@ struct LoginInfo
uint32_t password; uint32_t password;
}; };
struct LoginInfoPointer
{
const char* label;
const char* password;
LoginInfoPointer() : label(nullptr), password(nullptr) {}
};
struct Index struct Index
{ {
uint32_t count; uint32_t count;
@ -21,7 +29,7 @@ void delete_logininfo_from_buffer(Buffer& buffer, int index_of_pass);
void add_logininfo_to_buffer(Buffer& buffer, const char* label, const char* password); void add_logininfo_to_buffer(Buffer& buffer, const char* label, const char* password);
const char* get_logininfo_pointer_from_buffer(Buffer& buffer, int index_of_pass); LoginInfoPointer get_logininfo_pointer_from_buffer(Buffer& buffer, int index_of_pass);
void generate_password(std::string& password, int len); void generate_password(std::string& password, int len);

12
Password_manager/source/arg_func.cpp
View File

@ -56,17 +56,17 @@ Arg get_args(int argc, char** argv, char** label) {
return Arg::Error; return Arg::Error;
} }
const char* arg_get(Buffer& decrypted_buffer, const char* label) LoginInfoPointer arg_get(Buffer& decrypted_buffer, const char* label)
{ {
int pass = find_logininfo_in_buffer(decrypted_buffer, label); int pass = find_logininfo_in_buffer(decrypted_buffer, label);
if (pass < 0) { if (pass < 0) {
printf_s("Password not found\n"); printf_s("Password not found\n");
return nullptr; return LoginInfoPointer();
} }
return get_logininfo_pointer_from_buffer(decrypted_buffer, pass); return get_logininfo_pointer_from_buffer(decrypted_buffer, pass);
} }
const char* arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto) LoginInfoPointer arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto)
{ {
printf_s("Generating password for %s\n", label); printf_s("Generating password for %s\n", label);
@ -83,7 +83,7 @@ const char* arg_generate(Buffer& decrypted_buffer, Buffer& encrypted_buffer, con
return get_logininfo_pointer_from_buffer(decrypted_buffer, index->count - 1); return get_logininfo_pointer_from_buffer(decrypted_buffer, index->count - 1);
} }
const char* arg_input(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto) LoginInfoPointer arg_input(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const char* label, Cryptography& crypto)
{ {
printf_s("Input password for %s:", label); printf_s("Input password for %s:", label);
@ -91,7 +91,9 @@ const char* arg_input(Buffer& decrypted_buffer, Buffer& encrypted_buffer, const
if (new_string.empty()) if (new_string.empty())
{ {
printf_s("error getting password\n"); printf_s("error getting password\n");
return nullptr;
return LoginInfoPointer();
} }
int pass = find_logininfo_in_buffer(decrypted_buffer, label); int pass = find_logininfo_in_buffer(decrypted_buffer, label);

7
Password_manager/source/func.cpp
View File

@ -56,11 +56,14 @@ void add_logininfo_to_buffer(Buffer& buffer, const char* label, const char* pass
} }
const char* get_logininfo_pointer_from_buffer(Buffer& buffer, int index_of_pass) LoginInfoPointer get_logininfo_pointer_from_buffer(Buffer& buffer, int index_of_pass)
{ {
Index* index = (Index*)buffer.buffer; Index* index = (Index*)buffer.buffer;
LoginInfo* pass = (LoginInfo*)(buffer.buffer + sizeof(Index)); LoginInfo* pass = (LoginInfo*)(buffer.buffer + sizeof(Index));
return (const char*)buffer.buffer + pass[index_of_pass].password + index->offset; LoginInfoPointer ret;
ret.label = (const char*)buffer.buffer + pass[index_of_pass].label + index->offset;
ret.password = (const char*)buffer.buffer + pass[index_of_pass].password + index->offset;
return ret;
} }
void generate_password(std::string& password, int len) void generate_password(std::string& password, int len)

24
Password_manager/source/main.cpp
View File

@ -35,16 +35,16 @@ int main()
decrypted_buffer.add_end((uint8_t*)&index, sizeof(Index)); decrypted_buffer.add_end((uint8_t*)&index, sizeof(Index));
} }
const char* pass = nullptr; LoginInfoPointer login_info;
switch (args) switch (args)
{ {
case Arg::Get: case Arg::Get:
pass = arg_get(decrypted_buffer, label); login_info = arg_get(decrypted_buffer, label);
break; break;
case Arg::Generate: case Arg::Generate:
pass = arg_generate(decrypted_buffer, encrypted_buffer, label, crypto); login_info = arg_generate(decrypted_buffer, encrypted_buffer, label, crypto);
break; break;
case Arg::List: case Arg::List:
@ -60,7 +60,7 @@ int main()
break; break;
case Arg::Input: case Arg::Input:
pass = arg_input(decrypted_buffer, encrypted_buffer, label, crypto); login_info = arg_input(decrypted_buffer, encrypted_buffer, label, crypto);
break; break;
case Arg::Change: case Arg::Change:
@ -72,10 +72,10 @@ int main()
break; break;
} }
if (!pass) return 0; if (!login_info.password) return 0;
printf_s("Password copied to clipboard\n"); printf_s("Password copied to clipboard\n");
put_data_on_clipboard(pass); put_data_on_clipboard(login_info.password);
} }
@ -114,16 +114,16 @@ int main(int argc, char** argv)
decrypted_buffer.add_end((uint8_t*)&index, sizeof(Index)); decrypted_buffer.add_end((uint8_t*)&index, sizeof(Index));
} }
const char* pass = nullptr; LoginInfoPointer login_info;
switch (args) switch (args)
{ {
case Arg::Get: case Arg::Get:
pass = arg_get(decrypted_buffer, label); login_info = arg_get(decrypted_buffer, label);
break; break;
case Arg::Generate: case Arg::Generate:
pass = arg_generate(decrypted_buffer, encrypted_buffer, label, crypto); login_info = arg_generate(decrypted_buffer, encrypted_buffer, label, crypto);
break; break;
case Arg::List: case Arg::List:
@ -139,7 +139,7 @@ int main(int argc, char** argv)
break; break;
case Arg::Input: case Arg::Input:
pass = arg_input(decrypted_buffer, encrypted_buffer, label, crypto); login_info = arg_input(decrypted_buffer, encrypted_buffer, label, crypto);
break; break;
case Arg::Change: case Arg::Change:
@ -151,10 +151,10 @@ int main(int argc, char** argv)
break; break;
} }
if (!pass) return 1; if (!login_info.password) return 1;
printf_s("Password copied to clipboard\n"); printf_s("Password copied to clipboard\n");
put_data_on_clipboard(pass); put_data_on_clipboard(login_info.password);
} }