46 lines
1.1 KiB
C++
46 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include "common.h"
|
|
#include <botan/secmem.h>
|
|
#include <fstream>
|
|
#include <optional>
|
|
#include <unordered_map>
|
|
|
|
constexpr u16 VERSION = 1;
|
|
constexpr u64 AFTER_HEADER_OFFSET = 68;
|
|
|
|
struct FileHeader {
|
|
u64 offset;
|
|
bool deleted;
|
|
u64 name_ciphertext_size;
|
|
std::string name;
|
|
u64 content_ciphertext_size;
|
|
u64 content_offset;
|
|
};
|
|
|
|
class Vault {
|
|
public:
|
|
explicit Vault(std::string path, const std::string &password);
|
|
|
|
static void create(const std::string &path, const std::string &password);
|
|
|
|
std::string read_file(const std::string &name);
|
|
void create_file(const std::string &name, const std::string &content);
|
|
void delete_file(const std::string &name);
|
|
void update_file(const std::string &name, const std::string &content);
|
|
|
|
const std::string &path() const { return m_path; }
|
|
const std::unordered_map<std::string, FileHeader> &file_map() const {
|
|
return m_file_map;
|
|
}
|
|
|
|
private:
|
|
std::string m_path;
|
|
std::fstream m_file;
|
|
Botan::secure_vector<u8> m_key;
|
|
std::unordered_map<std::string, FileHeader> m_file_map;
|
|
|
|
void build_file_map();
|
|
std::optional<FileHeader> read_file_header();
|
|
};
|