Line data Source code
1 : /*
2 : * Copyright (C) 2004-2025 Savoir-faire Linux Inc.
3 : *
4 : * This program is free software: you can redistribute it and/or modify
5 : * it under the terms of the GNU General Public License as published by
6 : * the Free Software Foundation, either version 3 of the License, or
7 : * (at your option) any later version.
8 : *
9 : * This program is distributed in the hope that it will be useful,
10 : * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 : * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 : * GNU General Public License for more details.
13 : *
14 : * You should have received a copy of the GNU General Public License
15 : * along with this program. If not, see <https://www.gnu.org/licenses/>.
16 : */
17 : #pragma once
18 :
19 : #include "jami_contact.h"
20 :
21 : #include <dhtnet/certstore.h>
22 : #include <opendht/infohash.h>
23 : #include <opendht/crypto.h>
24 :
25 : #include <map>
26 : #include <mutex>
27 : #include <chrono>
28 :
29 : namespace jami {
30 :
31 : class ContactList
32 : {
33 : public:
34 : using clock = std::chrono::system_clock;
35 : using time_point = clock::time_point;
36 : using VerifyResult = dht::crypto::TrustList::VerifyResult;
37 :
38 : using OnContactAdded = std::function<void(const std::string&, bool)>;
39 : using OnContactRemoved = std::function<void(const std::string&, bool)>;
40 : using OnIncomingTrustRequest = std::function<
41 : void(const std::string&, const std::string&, const std::vector<uint8_t>&, time_t)>;
42 : using OnAcceptConversation = std::function<void(const std::string&, const std::string&)>;
43 : using OnConfirmation = std::function<void(const std::string&, const std::string&)>;
44 : using OnDevicesChanged = std::function<void(const std::map<dht::PkId, KnownDevice>&)>;
45 :
46 : struct OnChangeCallback
47 : {
48 : OnContactAdded contactAdded;
49 : OnContactRemoved contactRemoved;
50 : OnIncomingTrustRequest trustRequest;
51 : OnDevicesChanged devicesChanged;
52 : OnAcceptConversation acceptConversation;
53 : OnConfirmation onConfirmation;
54 : };
55 :
56 : ContactList(const std::string& accountId,
57 : const std::shared_ptr<crypto::Certificate>& cert,
58 : const std::filesystem::path& path,
59 : OnChangeCallback cb);
60 : ~ContactList();
61 :
62 41957 : const std::string& accountId() const { return accountId_; }
63 :
64 : void load();
65 : void save();
66 :
67 : /* Contacts */
68 : std::map<std::string, std::string> getContactDetails(const dht::InfoHash&) const;
69 : std::optional<Contact> getContactInfo(const dht::InfoHash&) const;
70 :
71 : bool removeContact(const dht::InfoHash&, bool ban);
72 : bool removeContactConversation(const dht::InfoHash&);
73 : bool addContact(const dht::InfoHash&,
74 : bool confirmed = false,
75 : const std::string& conversationId = "");
76 : void updateConversation(
77 : const dht::InfoHash& h, const std::string& conversationId, bool added = false);
78 :
79 : bool setCertificateStatus(const std::string& cert_id,
80 : const dhtnet::tls::TrustStore::PermissionStatus status);
81 :
82 : bool setCertificateStatus(const std::shared_ptr<crypto::Certificate>& cert,
83 : dhtnet::tls::TrustStore::PermissionStatus status,
84 : bool local = true);
85 :
86 7939 : dhtnet::tls::TrustStore::PermissionStatus getCertificateStatus(const std::string& cert_id) const
87 : {
88 7939 : return trust_->getCertificateStatus(cert_id);
89 : }
90 :
91 0 : std::vector<std::string> getCertificatesByStatus(dhtnet::tls::TrustStore::PermissionStatus status) const
92 : {
93 0 : return trust_->getCertificatesByStatus(status);
94 : }
95 :
96 910 : bool isAllowed(const crypto::Certificate& crt, bool allowPublic)
97 : {
98 910 : return trust_->isAllowed(crt, allowPublic);
99 : }
100 :
101 797 : VerifyResult isValidAccountDevice(const crypto::Certificate& crt) const
102 : {
103 797 : return accountTrust_.verify(crt);
104 : }
105 :
106 : const std::map<dht::InfoHash, Contact>& getContacts() const;
107 : void setContacts(const std::map<dht::InfoHash, Contact>&);
108 : void updateContact(const dht::InfoHash&, const Contact&, bool emit = true);
109 :
110 : /** Should be called only after updateContact */
111 : void saveContacts() const;
112 :
113 : const std::filesystem::path& path() const { return path_; }
114 :
115 : /* Contact requests */
116 :
117 : /** Inform of a new contact request. Returns true if the request should be immediatly accepted
118 : * (already a contact) */
119 : bool onTrustRequest(const dht::InfoHash& peer_account,
120 : const std::shared_ptr<dht::crypto::PublicKey>& peer_device,
121 : time_t received,
122 : bool confirm,
123 : const std::string& conversationId,
124 : std::vector<uint8_t>&& payload);
125 : std::vector<std::map<std::string, std::string>> getTrustRequests() const;
126 : std::map<std::string, std::string> getTrustRequest(const dht::InfoHash& from) const;
127 : void acceptConversation(const std::string& convId, const std::string& deviceId = ""); // ToDO this is a bit dirty imho
128 : bool acceptTrustRequest(const dht::InfoHash& from);
129 : bool discardTrustRequest(const dht::InfoHash& from);
130 :
131 : /* Devices */
132 1574 : const std::map<dht::PkId, KnownDevice>& getKnownDevices() const { return knownDevices_; }
133 : void foundAccountDevice(const dht::PkId& device,
134 : const std::string& name = {},
135 0 : const time_point& last_sync = time_point::min());
136 : bool foundAccountDevice(const std::shared_ptr<dht::crypto::Certificate>& crt,
137 : const std::string& name = {},
138 2 : const time_point& last_sync = time_point::min(),
139 : bool notify = true);
140 : bool removeAccountDevice(const dht::PkId& device);
141 : void setAccountDeviceName(const dht::PkId& device, const std::string& name);
142 : std::string getAccountDeviceName(const dht::PkId& device) const;
143 :
144 : DeviceSync getSyncData() const;
145 : bool syncDevice(const dht::PkId& device, const time_point& syncDate);
146 :
147 : private:
148 : const std::string accountId_;
149 : const std::filesystem::path path_;
150 : mutable std::mutex mutex_;
151 : std::map<dht::InfoHash, Contact> contacts_;
152 : std::map<dht::InfoHash, TrustRequest> trustRequests_;
153 : std::map<dht::InfoHash, KnownDevice> knownDevicesLegacy_;
154 :
155 : std::map<dht::PkId, KnownDevice> knownDevices_;
156 :
157 : // Trust store with account main certificate as the only CA
158 : dht::crypto::TrustList accountTrust_;
159 : // Trust store for to match peer certificates
160 : std::unique_ptr<dhtnet::tls::TrustStore> trust_;
161 : std::string accountUri_;
162 :
163 : OnChangeCallback callbacks_;
164 :
165 : void loadContacts();
166 : void loadTrustRequests();
167 :
168 : void loadKnownDevices();
169 : void saveKnownDevices() const;
170 :
171 : /** Should be called only after onTrustRequest */
172 : void saveTrustRequests() const;
173 : };
174 :
175 : } // namespace jami
|
