libreg/Key.h at main · OneBlue/libreg · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#pragma once

#include <vector>
#include <memory>
#include <optional>
#include "MultiString.h"
#include "Handle.h"
#include "Hive.h"
#include "Access.h"
#include "ValueType.h"
#include "SyscallFailure.h"
#include "Defs.h"

namespace libreg
{
    class Key
    {
    public:
        static Key Open(libreg::Hive hive, const MultiString& path, Access access);
        static Key Create(libreg::Hive hive, const MultiString& path, Access access, bool volatile_key = false);
        static Key OpenOrCreate(libreg::Hive hive, const MultiString& path, Access access, bool volatile_key = false);
        static void QueryInfo(const Handle<HKEY>& key, size_t* max_subkey_chars, size_t* max_value_name_chars, size_t* max_data_bytes);

        const Handle<HKEY>& NativeHandle() const;
        const MultiString& Path() const;
        MultiString Name() const;
        Hive Hive() const;

        std::vector<Key> SubKeys(Access access = Access::Read) const;
        std::vector<std::pair<MultiString, ValueType>> Values() const;
        Key OpenSubKey(const MultiString& name, Access access) const;
        std::optional<Key> TryOpenSubKey(const MultiString& name, Access access) const;
        Key CreateSubKey(const MultiString& name, bool volatile_key = false);
        Key OpenOrCreateSubkey(const MultiString& name, Access access, bool volatile_key);

        void SetValue(const MultiString& name, DWORD value, ValueType type);
        void SetValue(const MultiString& name, const std::vector<std::uint8_t>& value, ValueType type);
        void SetValue(const MultiString& name, const MultiString& value, ValueType type);

        template <typename T>
        T GetValue(const MultiString& name, ValueType expected_type = ValueType::None) const;

        void DeleteSubKey(const MultiString& name);
        void DeleteValue(const MultiString& name);
        void Delete();

        Access Access() const;

    private:
        Key(Handle<HKEY>&& handle, libreg::Hive hive, const MultiString& path, libreg::Access access);

        void SetValueImpl(const MultiString& name, const void* data, size_t size, ValueType type);
        void GetValueImpl(const MultiString& name, void* data, size_t& size, ValueType type) const;
        size_t GetValueSize(const MultiString& name) const;
        static void HandleException(const SyscallFailure& ex, libreg::Hive hive, const MultiString& path, bool valueOperation);

        std::shared_ptr<Handle<HKEY>> _handle;
        libreg::Hive _hive;
        MultiString _path;
        libreg::Access _access;
    };
}

#include "Key.hxx"