109 lines
2.4 KiB
C++
109 lines
2.4 KiB
C++
#include <Windows.h>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <shlobj.h>
|
|
#include "win.h"
|
|
#include "func.h"
|
|
|
|
bool put_data_on_clipboard(const char* text) {
|
|
size_t len = strlen(text);
|
|
if (len == 0) {
|
|
printf_s("Text is empty\n");
|
|
return false;
|
|
}
|
|
// Open the clipboard
|
|
if (!OpenClipboard(nullptr)) {
|
|
printf_s("Failed to open clipboard");
|
|
return false;
|
|
}
|
|
// Allocate global memory for the text
|
|
HGLOBAL hMem = GlobalAlloc(GMEM_MOVEABLE, strlen(text) + 1);
|
|
if (!hMem) {
|
|
CloseClipboard();
|
|
printf_s("Failed to allocate memory for text");
|
|
return false;
|
|
}
|
|
// Copy the text to the allocated memory
|
|
char* memData = static_cast<char*>(GlobalLock(hMem));
|
|
if (!memData) {
|
|
CloseClipboard();
|
|
printf_s("Failed to lock memory for text");
|
|
return false;
|
|
}
|
|
strcpy_s(memData, strlen(text) + 1, text);
|
|
GlobalUnlock(hMem);
|
|
// Set the data to the clipboard
|
|
EmptyClipboard();
|
|
SetClipboardData(CF_TEXT, hMem);
|
|
// Clean up and close the clipboard
|
|
CloseClipboard();
|
|
return true;
|
|
}
|
|
|
|
|
|
std::string get_user_password()
|
|
{
|
|
|
|
HANDLE hStdInput = GetStdHandle(STD_INPUT_HANDLE);
|
|
DWORD mode = 0;
|
|
|
|
// Create a restore point Mode
|
|
// is know 503
|
|
GetConsoleMode(hStdInput, &mode);
|
|
|
|
// Enable echo input
|
|
// set to 499
|
|
SetConsoleMode(hStdInput, mode & (~ENABLE_ECHO_INPUT));
|
|
|
|
// Take input
|
|
std::string ipt;
|
|
std::getline(std::cin, ipt);
|
|
|
|
// Otherwise next cout will print
|
|
// into the same line
|
|
std::cout << std::endl;
|
|
|
|
// Restore the mode
|
|
SetConsoleMode(hStdInput, mode);
|
|
|
|
return ipt;
|
|
}
|
|
|
|
std::optional<std::string> get_save_path()
|
|
{
|
|
PWSTR path = nullptr;
|
|
|
|
// Get the user's documents directory
|
|
HRESULT result = SHGetKnownFolderPath(FOLDERID_Profile, 0, nullptr, &path);
|
|
|
|
if (!SUCCEEDED(result)) {
|
|
CoTaskMemFree(path); // Free the allocated memory
|
|
return{};
|
|
}
|
|
|
|
std::wstring userDirectory(path);
|
|
CoTaskMemFree(path); // Free the allocated memory
|
|
|
|
userDirectory.append(L"\\password_manager");
|
|
|
|
DWORD attrib = GetFileAttributes(userDirectory.c_str());
|
|
|
|
//check if it exists
|
|
if(!(attrib != INVALID_FILE_ATTRIBUTES && (attrib & FILE_ATTRIBUTE_DIRECTORY)))
|
|
{
|
|
if (!CreateDirectory(userDirectory.c_str(), nullptr)) return {};
|
|
}
|
|
|
|
|
|
userDirectory.append(L"\\passwords.bin");
|
|
|
|
int size = WideCharToMultiByte(CP_UTF8, 0, userDirectory.c_str(), -1, nullptr, 0, nullptr, nullptr);
|
|
if (size > 0) {
|
|
std::string str(size, 0);
|
|
WideCharToMultiByte(CP_UTF8, 0, userDirectory.c_str(), -1, &str[0], size, nullptr, nullptr);
|
|
|
|
return str;
|
|
}
|
|
|
|
return {};
|
|
} |