libdino/src/crypto/omemo/simple_pks.vala


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

using Gee;
using Omemo;

namespace Dino.Crypto.Omemo {

public class SimplePreKeyStore : PreKeyStore {
    public signal void pre_key_stored(Key key);
    public signal void pre_key_deleted(Key key);

    private Map<uint32, Key> pre_key_map = new HashMap<uint32, Key>();

    public override uint8[]? load_pre_key(uint32 pre_key_id) throws Error {
        if (contains_pre_key(pre_key_id)) {
            return pre_key_map[pre_key_id].record;
        }
        return null;
    }

    public override void store_pre_key(uint32 pre_key_id, uint8[] record) throws Error {
        Key key = new Key(pre_key_id, record);
        pre_key_map[pre_key_id] = key;
        pre_key_stored(key);
    }

    public override bool contains_pre_key(uint32 pre_key_id) throws Error {
        return pre_key_map.has_key(pre_key_id);
    }

    public override void delete_pre_key(uint32 pre_key_id) throws Error {
        Key key;
        if (pre_key_map.unset(pre_key_id, out key)) {
            pre_key_deleted(key);
        }
    }

    public class Key {
        public uint32 key_id { get; private set; }
        public uint8[] record { get; private set; }

        public Key(uint32 key_id, uint8[] record) {
            this.key_id = key_id;
            this.record = record;
        }
    }
}

}