summaryrefslogtreecommitdiff
path: root/libs/tdlib/td/td/telegram/net/PublicRsaKeyShared.h
blob: 453bb7bda128b97be86aa9c94a47f894a733fcc7 (plain)
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
//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once

#include "td/telegram/net/DcId.h"

#include "td/mtproto/crypto.h"

#include "td/utils/common.h"
#include "td/utils/port/RwMutex.h"
#include "td/utils/Status.h"

#include <utility>

namespace td {

class PublicRsaKeyShared : public PublicRsaKeyInterface {
 public:
  explicit PublicRsaKeyShared(DcId dc_id);

  class Listener {
   public:
    Listener() = default;
    Listener(const Listener &) = delete;
    Listener &operator=(const Listener &) = delete;
    Listener(Listener &&) = delete;
    Listener &operator=(Listener &&) = delete;
    virtual ~Listener() = default;
    virtual bool notify() = 0;
  };

  void add_rsa(RSA rsa);
  Result<std::pair<RSA, int64>> get_rsa(const vector<int64> &fingerprints) override;
  void drop_keys() override;
  bool has_keys();

  void add_listener(std::unique_ptr<Listener> listener);

  DcId dc_id() const {
    return dc_id_;
  }

 private:
  DcId dc_id_;
  struct RsaOption {
    int64 fingerprint;
    RSA rsa;
  };
  std::vector<RsaOption> options_;
  std::vector<std::unique_ptr<Listener>> listeners_;
  RwMutex rw_mutex_;

  RSA *get_rsa_locked(int64 fingerprint);

  void notify();
};

}  // namespace td