111 lines
3.2 KiB
C++
111 lines
3.2 KiB
C++
#include "func.h"
|
|
#include "glob.h"
|
|
#include "Buffer.h"
|
|
|
|
int find_logininfo_in_buffer(Buffer& buffer, const char* label)
|
|
{
|
|
Index* index = (Index*)buffer.buffer;
|
|
LoginInfo* pass = (LoginInfo*)(buffer.buffer + sizeof(Index));
|
|
|
|
for (size_t i = 0; i < index->count; i++)
|
|
{
|
|
if (glob(label, (const char*)buffer.buffer + pass[i].label + index->offset) == Glob_Result::MATCHED)
|
|
return i;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
void delete_logininfo_from_buffer(Buffer& buffer, int index_of_pass)
|
|
{
|
|
Index* in_index = (Index*)buffer.buffer;
|
|
LoginInfo* in_pass = (LoginInfo*)(buffer.buffer + sizeof(Index));
|
|
|
|
int size_of_label = strlen((char*)buffer.buffer + in_pass[index_of_pass].label + in_index->offset) + 1;
|
|
int size_of_password = strlen((char*)buffer.buffer + in_pass[index_of_pass].password + in_index->offset) + 1;
|
|
int size_pass = size_of_label + size_of_password;
|
|
|
|
uint8_t* start_of_pass = buffer.buffer + in_pass[index_of_pass].label + in_index->offset;
|
|
buffer.remove_fast(start_of_pass, size_pass);
|
|
|
|
for (size_t i = index_of_pass; i < in_index->count; i++)
|
|
{
|
|
in_pass[i].label -= size_pass;
|
|
in_pass[i].password -= size_pass;
|
|
}
|
|
|
|
buffer.remove_fast((uint8_t*)&in_pass[index_of_pass], sizeof(LoginInfo));
|
|
in_index->count -= 1;
|
|
in_index->offset -= sizeof(LoginInfo);
|
|
}
|
|
|
|
void add_logininfo_to_buffer(Buffer& buffer, const char* label, const char* password)
|
|
{
|
|
int label_start = buffer.add_end((uint8_t*)label, strlen(label) + 1);
|
|
int password_start = buffer.add_end((uint8_t*)password, strlen(password) + 1);
|
|
|
|
Index* index = (Index*)buffer.buffer;
|
|
|
|
label_start = label_start - index->offset;
|
|
password_start = password_start - index->offset;
|
|
LoginInfo pass = { label_start, password_start };
|
|
index->count++;
|
|
index->offset += sizeof(LoginInfo);
|
|
|
|
buffer.add_middle((uint8_t*)&pass, sizeof(LoginInfo), index->offset - sizeof(LoginInfo));
|
|
|
|
}
|
|
|
|
LoginInfoPointer get_logininfo_pointer_from_buffer(Buffer& buffer, int index_of_pass)
|
|
{
|
|
Index* index = (Index*)buffer.buffer;
|
|
LoginInfo* pass = (LoginInfo*)(buffer.buffer + sizeof(Index));
|
|
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)
|
|
{
|
|
srand(time(NULL));
|
|
char characters[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789~!@#$%^&*()_-+={[}]|:;<,>.?";
|
|
for (int i = 0; i < len; i++)
|
|
{
|
|
password += characters[rand() % (sizeof(characters) - 1)];
|
|
}
|
|
}
|
|
|
|
bool save_buffer_to_file(Buffer& buffer)
|
|
{
|
|
std::ofstream file("passwords.bin", std::ios::binary | std::ios::out);
|
|
if (!file.is_open())
|
|
{
|
|
printf_s("Error saving file\n");
|
|
return false;
|
|
}
|
|
file.write((char*)buffer.buffer, buffer.taken);
|
|
file.close();
|
|
return true;
|
|
}
|
|
|
|
bool load_buffer_from_file(Buffer& buffer)
|
|
{
|
|
std::ifstream file("passwords.bin", std::ios::binary | std::ios::in);
|
|
if (!file.is_open()) return false;
|
|
|
|
file.seekg(0, std::ios::end);
|
|
size_t file_size = file.tellg();
|
|
buffer.resize(file_size);
|
|
file.seekg(0, std::ios::beg);
|
|
file.read((char*)buffer.buffer, buffer.size);
|
|
|
|
if (file)
|
|
buffer.taken = file_size;
|
|
else
|
|
buffer.taken = file.gcount();
|
|
|
|
file.close();
|
|
return true;
|
|
}
|