• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

realm / realm-core / 1707

25 Sep 2023 06:02PM UTC coverage: 91.192% (-0.006%) from 91.198%
1707

push

Evergreen

web-flow
Fixed double delete in CAPITimer and add documentation for platform networking CAPI integration (#6994)

* Fixed double delete in CAPITimer and added individual functions for handling callbacks.
* Added realm_sync_socket_write_complete() fcn and other updates from review
* Don't call handler functions more than once, esp for timers; updated documentation
* Updated some comments
* Updated changelog after release

95848 of 175718 branches covered (0.0%)

20 of 33 new or added lines in 2 files covered. (60.61%)

65 existing lines in 13 files now uncovered.

232403 of 254849 relevant lines covered (91.19%)

6713313.75 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

85.53
/src/realm/sync/noinst/client_impl_base.cpp
1
#include <system_error>
2
#include <sstream>
3

4
#include <realm/util/assert.hpp>
5
#include <realm/util/safe_int_ops.hpp>
6
#include <realm/util/random.hpp>
7
#include <realm/util/memory_stream.hpp>
8
#include <realm/util/basic_system_errors.hpp>
9
#include <realm/util/scope_exit.hpp>
10
#include <realm/util/to_string.hpp>
11
#include <realm/util/uri.hpp>
12
#include <realm/util/platform_info.hpp>
13
#include <realm/sync/impl/clock.hpp>
14
#include <realm/impl/simulated_failure.hpp>
15
#include <realm/sync/network/http.hpp>
16
#include <realm/sync/network/websocket.hpp>
17
#include <realm/sync/noinst/client_history_impl.hpp>
18
#include <realm/sync/noinst/client_impl_base.hpp>
19
#include <realm/sync/noinst/compact_changesets.hpp>
20
#include <realm/sync/protocol.hpp>
21
#include <realm/version.hpp>
22
#include <realm/sync/changeset_parser.hpp>
23

24
#include <realm/sync/network/websocket.hpp> // Only for websocket::Error TODO remove
25

26
// NOTE: The protocol specification is in `/doc/protocol.md`
27

28
using namespace realm;
29
using namespace _impl;
30
using namespace realm::util;
31
using namespace realm::sync;
32
using namespace realm::sync::websocket;
33

34
// clang-format off
35
using Connection      = ClientImpl::Connection;
36
using Session         = ClientImpl::Session;
37
using UploadChangeset = ClientHistory::UploadChangeset;
38

39
// These are a work-around for a bug in MSVC. It cannot find in-class types
40
// mentioned in signature of out-of-line member function definitions.
41
using ConnectionTerminationReason = ClientImpl::ConnectionTerminationReason;
42
using OutputBuffer                = ClientImpl::OutputBuffer;
43
using ReceivedChangesets          = ClientProtocol::ReceivedChangesets;
44
// clang-format on
45

46
void ClientImpl::ReconnectInfo::reset() noexcept
47
{
2,004✔
48
    m_backoff_state.reset();
2,004✔
49
    scheduled_reset = false;
2,004✔
50
}
2,004✔
51

52

53
void ClientImpl::ReconnectInfo::update(ConnectionTerminationReason new_reason,
54
                                       std::optional<ResumptionDelayInfo> new_delay_info)
55
{
3,418✔
56
    m_backoff_state.update(new_reason, new_delay_info);
3,418✔
57
}
3,418✔
58

59

60
std::chrono::milliseconds ClientImpl::ReconnectInfo::delay_interval()
61
{
5,704✔
62
    if (scheduled_reset) {
5,704✔
63
        reset();
4✔
64
    }
4✔
65

2,938✔
66
    if (!m_backoff_state.triggering_error) {
5,704✔
67
        return std::chrono::milliseconds::zero();
4,352✔
68
    }
4,352✔
69

724✔
70
    switch (*m_backoff_state.triggering_error) {
1,352✔
71
        case ConnectionTerminationReason::closed_voluntarily:
74✔
72
            return std::chrono::milliseconds::zero();
74✔
73
        case ConnectionTerminationReason::server_said_do_not_reconnect:
18✔
74
            return std::chrono::milliseconds::max();
18✔
75
        default:
1,256✔
76
            if (m_reconnect_mode == ReconnectMode::testing) {
1,256✔
77
                return std::chrono::milliseconds::max();
1,032✔
78
            }
1,032✔
79

112✔
80
            REALM_ASSERT(m_reconnect_mode == ReconnectMode::normal);
224✔
81
            return m_backoff_state.delay_interval();
224✔
82
    }
1,352✔
83
}
1,352✔
84

85

86
bool ClientImpl::decompose_server_url(const std::string& url, ProtocolEnvelope& protocol, std::string& address,
87
                                      port_type& port, std::string& path) const
88
{
3,322✔
89
    util::Uri uri(url); // Throws
3,322✔
90
    uri.canonicalize(); // Throws
3,322✔
91
    std::string userinfo, address_2, port_2;
3,322✔
92
    bool realm_scheme = (uri.get_scheme() == "realm:" || uri.get_scheme() == "realms:");
3,322✔
93
    bool ws_scheme = (uri.get_scheme() == "ws:" || uri.get_scheme() == "wss:");
3,322✔
94
    bool good = ((realm_scheme || ws_scheme) && uri.get_auth(userinfo, address_2, port_2) && userinfo.empty() &&
3,322✔
95
                 !address_2.empty() && uri.get_query().empty() && uri.get_frag().empty()); // Throws
3,322✔
96
    if (REALM_UNLIKELY(!good))
3,322✔
97
        return false;
1,486✔
98
    ProtocolEnvelope protocol_2;
3,322✔
99
    port_type port_3;
3,322✔
100
    if (realm_scheme) {
3,322✔
101
        if (uri.get_scheme() == "realm:") {
×
102
            protocol_2 = ProtocolEnvelope::realm;
×
103
            port_3 = (m_enable_default_port_hack ? 80 : 7800);
×
104
        }
×
105
        else {
×
106
            protocol_2 = ProtocolEnvelope::realms;
×
107
            port_3 = (m_enable_default_port_hack ? 443 : 7801);
×
108
        }
×
109
    }
×
110
    else {
3,322✔
111
        REALM_ASSERT(ws_scheme);
3,322✔
112
        if (uri.get_scheme() == "ws:") {
3,322✔
113
            protocol_2 = ProtocolEnvelope::ws;
3,318✔
114
            port_3 = 80;
3,318✔
115
        }
3,318✔
116
        else {
4✔
117
            protocol_2 = ProtocolEnvelope::wss;
4✔
118
            port_3 = 443;
4✔
119
        }
4✔
120
    }
3,322✔
121
    if (!port_2.empty()) {
3,322✔
122
        std::istringstream in(port_2);    // Throws
3,322✔
123
        in.imbue(std::locale::classic()); // Throws
3,322✔
124
        in >> port_3;
3,322✔
125
        if (REALM_UNLIKELY(!in || !in.eof() || port_3 < 1))
3,322✔
126
            return false;
1,486✔
127
    }
3,322✔
128
    std::string path_2 = uri.get_path(); // Throws (copy)
3,322✔
129

1,486✔
130
    protocol = protocol_2;
3,322✔
131
    address = std::move(address_2);
3,322✔
132
    port = port_3;
3,322✔
133
    path = std::move(path_2);
3,322✔
134
    return true;
3,322✔
135
}
3,322✔
136

137

138
ClientImpl::ClientImpl(ClientConfig config)
139
    : logger_ptr{config.logger ? std::move(config.logger) : std::make_shared<util::StderrLogger>()}
140
    , logger{*logger_ptr}
141
    , m_reconnect_mode{config.reconnect_mode}
142
    , m_connect_timeout{config.connect_timeout}
143
    , m_connection_linger_time{config.one_connection_per_session ? 0 : config.connection_linger_time}
144
    , m_ping_keepalive_period{config.ping_keepalive_period}
145
    , m_pong_keepalive_timeout{config.pong_keepalive_timeout}
146
    , m_fast_reconnect_limit{config.fast_reconnect_limit}
147
    , m_reconnect_backoff_info{config.reconnect_backoff_info}
148
    , m_disable_upload_activation_delay{config.disable_upload_activation_delay}
149
    , m_dry_run{config.dry_run}
150
    , m_enable_default_port_hack{config.enable_default_port_hack}
151
    , m_disable_upload_compaction{config.disable_upload_compaction}
152
    , m_fix_up_object_ids{config.fix_up_object_ids}
153
    , m_roundtrip_time_handler{std::move(config.roundtrip_time_handler)}
154
    , m_socket_provider{std::move(config.socket_provider)}
155
    , m_client_protocol{} // Throws
156
    , m_one_connection_per_session{config.one_connection_per_session}
157
    , m_random{}
158
{
8,950✔
159
    // FIXME: Would be better if seeding was up to the application.
4,406✔
160
    util::seed_prng_nondeterministically(m_random); // Throws
8,950✔
161

4,406✔
162
    logger.info("Realm sync client (%1)", REALM_VER_CHUNK); // Throws
8,950✔
163
    logger.debug("Supported protocol versions: %1-%2", get_oldest_supported_protocol_version(),
8,950✔
164
                 get_current_protocol_version()); // Throws
8,950✔
165
    logger.info("Platform: %1", util::get_platform_info());
8,950✔
166
    const char* build_mode;
8,950✔
167
#if REALM_DEBUG
8,950✔
168
    build_mode = "Debug";
8,950✔
169
#else
170
    build_mode = "Release";
171
#endif
172
    logger.debug("Build mode: %1", build_mode);
8,950✔
173
    logger.debug("Config param: one_connection_per_session = %1",
8,950✔
174
                 config.one_connection_per_session); // Throws
8,950✔
175
    logger.debug("Config param: connect_timeout = %1 ms",
8,950✔
176
                 config.connect_timeout); // Throws
8,950✔
177
    logger.debug("Config param: connection_linger_time = %1 ms",
8,950✔
178
                 config.connection_linger_time); // Throws
8,950✔
179
    logger.debug("Config param: ping_keepalive_period = %1 ms",
8,950✔
180
                 config.ping_keepalive_period); // Throws
8,950✔
181
    logger.debug("Config param: pong_keepalive_timeout = %1 ms",
8,950✔
182
                 config.pong_keepalive_timeout); // Throws
8,950✔
183
    logger.debug("Config param: fast_reconnect_limit = %1 ms",
8,950✔
184
                 config.fast_reconnect_limit); // Throws
8,950✔
185
    logger.debug("Config param: disable_upload_compaction = %1",
8,950✔
186
                 config.disable_upload_compaction); // Throws
8,950✔
187
    logger.debug("Config param: disable_sync_to_disk = %1",
8,950✔
188
                 config.disable_sync_to_disk); // Throws
8,950✔
189
    logger.debug("Config param: reconnect backoff info: max_delay: %1 ms, initial_delay: %2 ms, multiplier: %3",
8,950✔
190
                 m_reconnect_backoff_info.max_resumption_delay_interval.count(),
8,950✔
191
                 m_reconnect_backoff_info.resumption_delay_interval.count(),
8,950✔
192
                 m_reconnect_backoff_info.resumption_delay_backoff_multiplier);
8,950✔
193

4,406✔
194
    if (config.reconnect_mode != ReconnectMode::normal) {
8,950✔
195
        logger.warn("Testing/debugging feature 'nonnormal reconnect mode' enabled - "
768✔
196
                    "never do this in production!");
768✔
197
    }
768✔
198

4,406✔
199
    if (config.dry_run) {
8,950✔
200
        logger.warn("Testing/debugging feature 'dry run' enabled - "
×
201
                    "never do this in production!");
×
202
    }
×
203

4,406✔
204
    REALM_ASSERT_EX(m_socket_provider, "Must provide socket provider in sync Client config");
8,950✔
205

4,406✔
206
    if (m_one_connection_per_session) {
8,950✔
207
        // FIXME: Re-enable this warning when the load balancer is able to handle
2✔
208
        // multiplexing.
2✔
209
        //        logger.warn("Testing/debugging feature 'one connection per session' enabled - "
2✔
210
        //            "never do this in production");
2✔
211
    }
4✔
212

4,406✔
213
    if (config.disable_upload_activation_delay) {
8,950✔
214
        logger.warn("Testing/debugging feature 'disable_upload_activation_delay' enabled - "
×
215
                    "never do this in production");
×
216
    }
×
217

4,406✔
218
    if (config.disable_sync_to_disk) {
8,950✔
219
        logger.warn("Testing/debugging feature 'disable_sync_to_disk' enabled - "
×
220
                    "never do this in production");
×
221
    }
×
222

4,406✔
223
    m_actualize_and_finalize = create_trigger([this](Status status) {
14,362✔
224
        if (status == ErrorCodes::OperationAborted)
14,362✔
225
            return;
×
226
        else if (!status.is_ok())
14,362✔
227
            throw Exception(status);
×
228
        actualize_and_finalize_session_wrappers(); // Throws
14,362✔
229
    });
14,362✔
230
}
8,950✔
231

232

233
void ClientImpl::post(SyncSocketProvider::FunctionHandler&& handler)
234
{
98,864✔
235
    REALM_ASSERT(m_socket_provider);
98,864✔
236
    {
98,864✔
237
        std::lock_guard lock(m_drain_mutex);
98,864✔
238
        ++m_outstanding_posts;
98,864✔
239
        m_drained = false;
98,864✔
240
    }
98,864✔
241
    m_socket_provider->post([handler = std::move(handler), this](Status status) {
98,864✔
242
        auto decr_guard = util::make_scope_exit([&]() noexcept {
98,864✔
243
            std::lock_guard lock(m_drain_mutex);
98,864✔
244
            REALM_ASSERT(m_outstanding_posts);
98,864✔
245
            --m_outstanding_posts;
98,864✔
246
            m_drain_cv.notify_all();
98,864✔
247
        });
98,864✔
248
        handler(status);
98,858✔
249
    });
98,858✔
250
}
98,864✔
251

252

253
void ClientImpl::drain_connections()
254
{
8,950✔
255
    logger.debug("Draining connections during sync client shutdown");
8,950✔
256
    for (auto& server_slot_pair : m_server_slots) {
5,640✔
257
        auto& server_slot = server_slot_pair.second;
2,330✔
258

1,096✔
259
        if (server_slot.connection) {
2,330✔
260
            auto& conn = server_slot.connection;
2,232✔
261
            conn->force_close();
2,232✔
262
        }
2,232✔
263
        else {
98✔
264
            for (auto& conn_pair : server_slot.alt_connections) {
48✔
265
                conn_pair.second->force_close();
6✔
266
            }
6✔
267
        }
98✔
268
    }
2,330✔
269
}
8,950✔
270

271

272
SyncSocketProvider::SyncTimer ClientImpl::create_timer(std::chrono::milliseconds delay,
273
                                                       SyncSocketProvider::FunctionHandler&& handler)
274
{
16,550✔
275
    REALM_ASSERT(m_socket_provider);
16,550✔
276
    {
16,550✔
277
        std::lock_guard lock(m_drain_mutex);
16,550✔
278
        ++m_outstanding_posts;
16,550✔
279
        m_drained = false;
16,550✔
280
    }
16,550✔
281
    return m_socket_provider->create_timer(delay, [handler = std::move(handler), this](Status status) {
16,552✔
282
        handler(status);
16,552✔
283

8,088✔
284
        std::lock_guard lock(m_drain_mutex);
16,552✔
285
        REALM_ASSERT(m_outstanding_posts);
16,552✔
286
        --m_outstanding_posts;
16,552✔
287
        m_drain_cv.notify_all();
16,552✔
288
    });
16,552✔
289
}
16,550✔
290

291

292
ClientImpl::SyncTrigger ClientImpl::create_trigger(SyncSocketProvider::FunctionHandler&& handler)
293
{
11,394✔
294
    REALM_ASSERT(m_socket_provider);
11,394✔
295
    return std::make_unique<Trigger<ClientImpl>>(this, std::move(handler));
11,394✔
296
}
11,394✔
297

298
Connection::~Connection()
299
{
2,444✔
300
    if (m_websocket_sentinel) {
2,444✔
301
        m_websocket_sentinel->destroyed = true;
×
302
        m_websocket_sentinel.reset();
×
303
    }
×
304
}
2,444✔
305

306
void Connection::activate()
307
{
2,444✔
308
    REALM_ASSERT(m_on_idle);
2,444✔
309
    m_activated = true;
2,444✔
310
    if (m_num_active_sessions == 0)
2,444✔
311
        m_on_idle->trigger();
×
312
    // We cannot in general connect immediately, because a prior failure to
1,152✔
313
    // connect may require a delay before reconnecting (see `m_reconnect_info`).
1,152✔
314
    initiate_reconnect_wait(); // Throws
2,444✔
315
}
2,444✔
316

317

318
void Connection::activate_session(std::unique_ptr<Session> sess)
319
{
9,600✔
320
    REALM_ASSERT(sess);
9,600✔
321
    REALM_ASSERT(&sess->m_conn == this);
9,600✔
322
    REALM_ASSERT(!m_force_closed);
9,600✔
323
    Session& sess_2 = *sess;
9,600✔
324
    session_ident_type ident = sess->m_ident;
9,600✔
325
    auto p = m_sessions.emplace(ident, std::move(sess)); // Throws
9,600✔
326
    bool was_inserted = p.second;
9,600✔
327
    REALM_ASSERT(was_inserted);
9,600✔
328
    // Save the session ident to the historical list of session idents
4,622✔
329
    m_session_history.insert(ident);
9,600✔
330
    sess_2.activate(); // Throws
9,600✔
331
    if (m_state == ConnectionState::connected) {
9,600✔
332
        bool fast_reconnect = false;
6,758✔
333
        sess_2.connection_established(fast_reconnect); // Throws
6,758✔
334
    }
6,758✔
335
    ++m_num_active_sessions;
9,600✔
336
}
9,600✔
337

338

339
void Connection::initiate_session_deactivation(Session* sess)
340
{
9,600✔
341
    REALM_ASSERT(sess);
9,600✔
342
    REALM_ASSERT(&sess->m_conn == this);
9,600✔
343
    REALM_ASSERT(m_num_active_sessions);
9,600✔
344
    // Since the client may be waiting for m_num_active_sessions to reach 0
4,622✔
345
    // in stop_and_wait() (on a separate thread), deactivate Session before
4,622✔
346
    // decrementing the num active sessions value.
4,622✔
347
    sess->initiate_deactivation(); // Throws
9,600✔
348
    if (sess->m_state == Session::Deactivated) {
9,600✔
349
        finish_session_deactivation(sess);
914✔
350
    }
914✔
351
    if (REALM_UNLIKELY(--m_num_active_sessions == 0)) {
9,600✔
352
        if (m_activated && m_state == ConnectionState::disconnected)
3,946✔
353
            m_on_idle->trigger();
330✔
354
    }
3,946✔
355
}
9,600✔
356

357

358
void Connection::cancel_reconnect_delay()
359
{
2,208✔
360
    REALM_ASSERT(m_activated);
2,208✔
361

1,210✔
362
    if (m_reconnect_delay_in_progress) {
2,208✔
363
        if (m_nonzero_reconnect_delay)
1,996✔
364
            logger.detail("Canceling reconnect delay"); // Throws
1,002✔
365

1,104✔
366
        // Cancel the in-progress wait operation by destroying the timer
1,104✔
367
        // object. Destruction is needed in this case, because a new wait
1,104✔
368
        // operation might have to be initiated before the previous one
1,104✔
369
        // completes (its completion handler starts to execute), so the new wait
1,104✔
370
        // operation must be done on a new timer object.
1,104✔
371
        m_reconnect_disconnect_timer.reset();
1,996✔
372
        m_reconnect_delay_in_progress = false;
1,996✔
373
        m_reconnect_info.reset();
1,996✔
374
        initiate_reconnect_wait(); // Throws
1,996✔
375
        return;
1,996✔
376
    }
1,996✔
377

106✔
378
    // If we are not disconnected, then we need to make sure the next time we get disconnected
106✔
379
    // that we are allowed to re-connect as quickly as possible.
106✔
380
    //
106✔
381
    // Setting m_reconnect_info.scheduled_reset will cause initiate_reconnect_wait to reset the
106✔
382
    // backoff/delay state before calculating the next delay, unless a PONG message is received
106✔
383
    // for the urgent PING message we send below.
106✔
384
    //
106✔
385
    // If we get a PONG message for the urgent PING message sent below, then the connection is
106✔
386
    // healthy and we can calculate the next delay normally.
106✔
387
    if (m_state != ConnectionState::disconnected) {
212✔
388
        m_reconnect_info.scheduled_reset = true;
212✔
389
        m_ping_after_scheduled_reset_of_reconnect_info = false;
212✔
390

106✔
391
        schedule_urgent_ping(); // Throws
212✔
392
        return;
212✔
393
    }
212✔
394
    // Nothing to do in this case. The next reconnect attemp will be made as
106✔
395
    // soon as there are any sessions that are both active and unsuspended.
106✔
396
}
212✔
397

398
void ClientImpl::Connection::finish_session_deactivation(Session* sess)
399
{
8,034✔
400
    REALM_ASSERT(sess->m_state == Session::Deactivated);
8,034✔
401
    auto ident = sess->m_ident;
8,034✔
402
    m_sessions.erase(ident);
8,034✔
403
    m_session_history.erase(ident);
8,034✔
404
}
8,034✔
405

406
void Connection::force_close()
407
{
2,240✔
408
    if (m_force_closed) {
2,240✔
409
        return;
×
410
    }
×
411

1,054✔
412
    m_force_closed = true;
2,240✔
413

1,054✔
414
    if (m_state != ConnectionState::disconnected) {
2,240✔
415
        voluntary_disconnect();
2,154✔
416
    }
2,154✔
417

1,054✔
418
    REALM_ASSERT_EX(m_state == ConnectionState::disconnected, m_state);
2,240✔
419
    if (m_reconnect_delay_in_progress || m_disconnect_delay_in_progress) {
2,240✔
420
        m_reconnect_disconnect_timer.reset();
86✔
421
        m_reconnect_delay_in_progress = false;
86✔
422
        m_disconnect_delay_in_progress = false;
86✔
423
    }
86✔
424

1,054✔
425
    // We must copy any session pointers we want to close to a vector because force_closing
1,054✔
426
    // the session may remove it from m_sessions and invalidate the iterator uses to loop
1,054✔
427
    // through the map. By copying to a separate vector we ensure our iterators remain valid.
1,054✔
428
    std::vector<Session*> to_close;
2,240✔
429
    for (auto& session_pair : m_sessions) {
1,130✔
430
        if (session_pair.second->m_state == Session::State::Active) {
150✔
431
            to_close.push_back(session_pair.second.get());
150✔
432
        }
150✔
433
    }
150✔
434

1,054✔
435
    for (auto& sess : to_close) {
1,130✔
436
        sess->force_close();
150✔
437
    }
150✔
438

1,054✔
439
    logger.debug("Force closed idle connection");
2,240✔
440
}
2,240✔
441

442

443
void Connection::websocket_connected_handler(const std::string& protocol)
444
{
3,306✔
445
    if (!protocol.empty()) {
3,306✔
446
        std::string_view expected_prefix =
3,306✔
447
            is_flx_sync_connection() ? get_flx_websocket_protocol_prefix() : get_pbs_websocket_protocol_prefix();
3,044✔
448
        // FIXME: Use std::string_view::begins_with() in C++20.
1,636✔
449
        auto prefix_matches = [&](std::string_view other) {
3,306✔
450
            return protocol.size() >= other.size() && (protocol.substr(0, other.size()) == other);
3,306✔
451
        };
3,306✔
452
        if (prefix_matches(expected_prefix)) {
3,306✔
453
            util::MemoryInputStream in;
3,306✔
454
            in.set_buffer(protocol.data() + expected_prefix.size(), protocol.data() + protocol.size());
3,306✔
455
            in.imbue(std::locale::classic());
3,306✔
456
            in.unsetf(std::ios_base::skipws);
3,306✔
457
            int value_2 = 0;
3,306✔
458
            in >> value_2;
3,306✔
459
            if (in && in.eof() && value_2 >= 0) {
3,306✔
460
                bool good_version =
3,306✔
461
                    (value_2 >= get_oldest_supported_protocol_version() && value_2 <= get_current_protocol_version());
3,306✔
462
                if (good_version) {
3,306✔
463
                    logger.detail("Negotiated protocol version: %1", value_2);
3,306✔
464
                    // For now, grab the connection ID from the websocket if it supports it. In the future, the server
1,636✔
465
                    // will provide the appservices connection ID via a log message.
1,636✔
466
                    // TODO: Remove once the server starts sending the connection ID
1,636✔
467
                    receive_appservices_request_id(m_websocket->get_appservices_request_id());
3,306✔
468
                    m_negotiated_protocol_version = value_2;
3,306✔
469
                    handle_connection_established(); // Throws
3,306✔
470
                    return;
3,306✔
471
                }
3,306✔
472
            }
×
473
        }
3,306✔
474
        close_due_to_client_side_error({ErrorCodes::SyncProtocolNegotiationFailed,
×
475
                                        util::format("Bad protocol info from server: '%1'", protocol)},
×
476
                                       IsFatal{true}, ConnectionTerminationReason::bad_headers_in_http_response);
×
477
    }
×
478
    else {
×
479
        close_due_to_client_side_error(
×
480
            {ErrorCodes::SyncProtocolNegotiationFailed, "Missing protocol info from server"}, IsFatal{true},
×
481
            ConnectionTerminationReason::bad_headers_in_http_response);
×
482
    }
×
483
}
3,306✔
484

485

486
bool Connection::websocket_binary_message_received(util::Span<const char> data)
487
{
77,722✔
488
    if (m_force_closed) {
77,722✔
489
        logger.debug("Received binary message after connection was force closed");
×
490
        return false;
×
491
    }
×
492

38,852✔
493
    using sf = SimulatedFailure;
77,722✔
494
    if (sf::check_trigger(sf::sync_client__read_head)) {
77,722✔
495
        close_due_to_client_side_error(
456✔
496
            {ErrorCodes::RuntimeError, "Simulated failure during sync client websocket read"}, IsFatal{false},
456✔
497
            ConnectionTerminationReason::read_or_write_error);
456✔
498
        return bool(m_websocket);
456✔
499
    }
456✔
500

38,586✔
501
    handle_message_received(data);
77,266✔
502
    return bool(m_websocket);
77,266✔
503
}
77,266✔
504

505

506
void Connection::websocket_error_handler()
507
{
632✔
508
    m_websocket_error_received = true;
632✔
509
}
632✔
510

511
bool Connection::websocket_closed_handler(bool was_clean, WebSocketError error_code, std::string_view msg)
512
{
760✔
513
    if (m_force_closed) {
760✔
514
        logger.debug("Received websocket close message after connection was force closed");
×
515
        return false;
×
516
    }
×
517
    logger.info("Closing the websocket with error code=%1, message='%2', was_clean=%3", error_code, msg, was_clean);
760✔
518

396✔
519
    switch (error_code) {
760✔
520
        case WebSocketError::websocket_ok:
76✔
521
            break;
76✔
522
        case WebSocketError::websocket_resolve_failed:
4✔
523
            [[fallthrough]];
4✔
524
        case WebSocketError::websocket_connection_failed: {
4✔
525
            SessionErrorInfo error_info(
4✔
526
                {ErrorCodes::SyncConnectFailed, util::format("Failed to connect to sync: %1", msg)}, IsFatal{false});
4✔
527
            involuntary_disconnect(std::move(error_info), ConnectionTerminationReason::connect_operation_failed);
4✔
528
            break;
4✔
529
        }
4✔
530
        case WebSocketError::websocket_read_error:
618✔
531
            [[fallthrough]];
618✔
532
        case WebSocketError::websocket_write_error: {
618✔
533
            close_due_to_transient_error({ErrorCodes::ConnectionClosed, msg},
618✔
534
                                         ConnectionTerminationReason::read_or_write_error);
618✔
535
            break;
618✔
536
        }
618✔
537
        case WebSocketError::websocket_going_away:
324✔
538
            [[fallthrough]];
×
539
        case WebSocketError::websocket_protocol_error:
✔
540
            [[fallthrough]];
×
541
        case WebSocketError::websocket_unsupported_data:
✔
542
            [[fallthrough]];
×
543
        case WebSocketError::websocket_invalid_payload_data:
✔
544
            [[fallthrough]];
×
545
        case WebSocketError::websocket_policy_violation:
✔
546
            [[fallthrough]];
×
547
        case WebSocketError::websocket_reserved:
✔
548
            [[fallthrough]];
×
549
        case WebSocketError::websocket_no_status_received:
✔
550
            [[fallthrough]];
×
551
        case WebSocketError::websocket_invalid_extension: {
✔
552
            close_due_to_client_side_error({ErrorCodes::SyncProtocolInvariantFailed, msg}, IsFatal{false},
×
553
                                           ConnectionTerminationReason::websocket_protocol_violation); // Throws
×
554
            break;
×
555
        }
×
556
        case WebSocketError::websocket_message_too_big: {
4✔
557
            auto message = util::format(
4✔
558
                "Sync websocket closed because the server received a message that was too large: %1", msg);
4✔
559
            SessionErrorInfo error_info(Status(ErrorCodes::LimitExceeded, std::move(message)), IsFatal{false});
4✔
560
            error_info.server_requests_action = ProtocolErrorInfo::Action::ClientReset;
4✔
561
            involuntary_disconnect(std::move(error_info),
4✔
562
                                   ConnectionTerminationReason::websocket_protocol_violation); // Throws
4✔
563
            break;
4✔
564
        }
×
565
        case WebSocketError::websocket_tls_handshake_failed: {
10✔
566
            close_due_to_client_side_error(
10✔
567
                Status(ErrorCodes::TlsHandshakeFailed, util::format("TLS handshake failed: %1", msg)), IsFatal{false},
10✔
568
                ConnectionTerminationReason::ssl_certificate_rejected); // Throws
10✔
569
            break;
10✔
570
        }
×
571
        case WebSocketError::websocket_client_too_old:
✔
572
            [[fallthrough]];
×
573
        case WebSocketError::websocket_client_too_new:
✔
574
            [[fallthrough]];
×
575
        case WebSocketError::websocket_protocol_mismatch: {
✔
576
            close_due_to_client_side_error({ErrorCodes::SyncProtocolNegotiationFailed, msg}, IsFatal{true},
×
577
                                           ConnectionTerminationReason::http_response_says_fatal_error); // Throws
×
578
            break;
×
579
        }
×
580
        case WebSocketError::websocket_fatal_error: {
✔
581
            involuntary_disconnect(SessionErrorInfo({ErrorCodes::ConnectionClosed, msg}, IsFatal{true}),
×
582
                                   ConnectionTerminationReason::http_response_says_fatal_error);
×
583
            break;
×
584
        }
×
585
        case WebSocketError::websocket_forbidden: {
✔
586
            SessionErrorInfo error_info({ErrorCodes::AuthError, msg}, IsFatal{true});
×
587
            error_info.server_requests_action = ProtocolErrorInfo::Action::LogOutUser;
×
588
            involuntary_disconnect(std::move(error_info),
×
589
                                   ConnectionTerminationReason::http_response_says_fatal_error);
×
590
            break;
×
591
        }
×
592
        case WebSocketError::websocket_unauthorized: {
36✔
593
            SessionErrorInfo error_info(
36✔
594
                {ErrorCodes::AuthError,
36✔
595
                 util::format("Websocket was closed because of an authentication issue: %1", msg)},
36✔
596
                IsFatal{false});
36✔
597
            error_info.server_requests_action = ProtocolErrorInfo::Action::RefreshUser;
36✔
598
            involuntary_disconnect(std::move(error_info),
36✔
599
                                   ConnectionTerminationReason::http_response_says_nonfatal_error);
36✔
600
            break;
36✔
601
        }
×
602
        case WebSocketError::websocket_moved_permanently: {
12✔
603
            SessionErrorInfo error_info({ErrorCodes::ConnectionClosed, msg}, IsFatal{false});
12✔
604
            error_info.server_requests_action = ProtocolErrorInfo::Action::RefreshLocation;
12✔
605
            involuntary_disconnect(std::move(error_info),
12✔
606
                                   ConnectionTerminationReason::http_response_says_nonfatal_error);
12✔
607
            break;
12✔
608
        }
×
609
        case WebSocketError::websocket_abnormal_closure: {
✔
610
            SessionErrorInfo error_info({ErrorCodes::ConnectionClosed, msg}, IsFatal{false});
×
611
            error_info.server_requests_action = ProtocolErrorInfo::Action::RefreshUser;
×
612
            involuntary_disconnect(std::move(error_info),
×
613
                                   ConnectionTerminationReason::http_response_says_nonfatal_error);
×
614
            break;
×
615
        }
×
616
        case WebSocketError::websocket_internal_server_error:
✔
617
            [[fallthrough]];
×
618
        case WebSocketError::websocket_retry_error: {
✔
619
            involuntary_disconnect(SessionErrorInfo({ErrorCodes::ConnectionClosed, msg}, IsFatal{false}),
×
620
                                   ConnectionTerminationReason::http_response_says_nonfatal_error);
×
621
            break;
×
622
        }
760✔
623
    }
760✔
624

396✔
625
    return bool(m_websocket);
760✔
626
}
760✔
627

628
// Guarantees that handle_reconnect_wait() is never called from within the
629
// execution of initiate_reconnect_wait() (no callback reentrance).
630
void Connection::initiate_reconnect_wait()
631
{
7,858✔
632
    REALM_ASSERT(m_activated);
7,858✔
633
    REALM_ASSERT(!m_reconnect_delay_in_progress);
7,858✔
634
    REALM_ASSERT(!m_disconnect_delay_in_progress);
7,858✔
635

3,950✔
636
    // If we've been force closed then we don't need/want to reconnect. Just return early here.
3,950✔
637
    if (m_force_closed) {
7,858✔
638
        return;
2,154✔
639
    }
2,154✔
640

2,938✔
641
    m_reconnect_delay_in_progress = true;
5,704✔
642
    auto delay = m_reconnect_info.delay_interval();
5,704✔
643
    if (delay == std::chrono::milliseconds::max()) {
5,704✔
644
        logger.detail("Reconnection delayed indefinitely"); // Throws
1,050✔
645
        // Not actually starting a timer corresponds to an infinite wait
578✔
646
        m_nonzero_reconnect_delay = true;
1,050✔
647
        return;
1,050✔
648
    }
1,050✔
649

2,360✔
650
    if (delay == std::chrono::milliseconds::zero()) {
4,654✔
651
        m_nonzero_reconnect_delay = false;
4,428✔
652
    }
4,428✔
653
    else {
226✔
654
        logger.detail("Allowing reconnection in %1 milliseconds", delay.count()); // Throws
226✔
655
        m_nonzero_reconnect_delay = true;
226✔
656
    }
226✔
657

2,360✔
658
    // We create a timer for the reconnect_disconnect timer even if the delay is zero because
2,360✔
659
    // we need it to be cancelable in case the connection is terminated before the timer
2,360✔
660
    // callback is run.
2,360✔
661
    m_reconnect_disconnect_timer = m_client.create_timer(delay, [this](Status status) {
4,650✔
662
        // If the operation is aborted, the connection object may have been
2,360✔
663
        // destroyed.
2,360✔
664
        if (status != ErrorCodes::OperationAborted)
4,650✔
665
            handle_reconnect_wait(status); // Throws
3,414✔
666
    });                                    // Throws
4,650✔
667
}
4,654✔
668

669

670
void Connection::handle_reconnect_wait(Status status)
671
{
3,416✔
672
    if (!status.is_ok()) {
3,416✔
673
        REALM_ASSERT(status != ErrorCodes::OperationAborted);
×
674
        throw Exception(status);
×
675
    }
×
676

1,694✔
677
    REALM_ASSERT(m_reconnect_delay_in_progress);
3,416✔
678
    m_reconnect_delay_in_progress = false;
3,416✔
679

1,694✔
680
    if (m_num_active_unsuspended_sessions > 0)
3,416✔
681
        initiate_reconnect(); // Throws
3,416✔
682
}
3,416✔
683

684
struct Connection::WebSocketObserverShim : public sync::WebSocketObserver {
685
    explicit WebSocketObserverShim(Connection* conn)
686
        : conn(conn)
687
        , sentinel(conn->m_websocket_sentinel)
688
    {
3,418✔
689
    }
3,418✔
690

691
    Connection* conn;
692
    util::bind_ptr<LifecycleSentinel> sentinel;
693

694
    void websocket_connected_handler(const std::string& protocol) override
695
    {
3,306✔
696
        if (sentinel->destroyed) {
3,306✔
697
            return;
×
698
        }
×
699

1,636✔
700
        return conn->websocket_connected_handler(protocol);
3,306✔
701
    }
3,306✔
702

703
    void websocket_error_handler() override
704
    {
632✔
705
        if (sentinel->destroyed) {
632✔
706
            return;
×
707
        }
×
708

332✔
709
        conn->websocket_error_handler();
632✔
710
    }
632✔
711

712
    bool websocket_binary_message_received(util::Span<const char> data) override
713
    {
77,720✔
714
        if (sentinel->destroyed) {
77,720✔
715
            return false;
×
716
        }
×
717

38,852✔
718
        return conn->websocket_binary_message_received(data);
77,720✔
719
    }
77,720✔
720

721
    bool websocket_closed_handler(bool was_clean, WebSocketError error_code, std::string_view msg) override
722
    {
760✔
723
        if (sentinel->destroyed) {
760✔
724
            return true;
×
725
        }
×
726

396✔
727
        return conn->websocket_closed_handler(was_clean, error_code, msg);
760✔
728
    }
760✔
729
};
730

731
void Connection::initiate_reconnect()
732
{
3,416✔
733
    REALM_ASSERT(m_activated);
3,416✔
734

1,694✔
735
    m_state = ConnectionState::connecting;
3,416✔
736
    report_connection_state_change(ConnectionState::connecting); // Throws
3,416✔
737
    if (m_websocket_sentinel) {
3,416✔
738
        m_websocket_sentinel->destroyed = true;
×
739
    }
×
740
    m_websocket_sentinel = util::make_bind<LifecycleSentinel>();
3,416✔
741
    m_websocket.reset();
3,416✔
742

1,694✔
743
    // Watchdog
1,694✔
744
    initiate_connect_wait(); // Throws
3,416✔
745

1,694✔
746
    std::vector<std::string> sec_websocket_protocol;
3,416✔
747
    {
3,416✔
748
        auto protocol_prefix =
3,416✔
749
            is_flx_sync_connection() ? get_flx_websocket_protocol_prefix() : get_pbs_websocket_protocol_prefix();
3,150✔
750
        int min = get_oldest_supported_protocol_version();
3,416✔
751
        int max = get_current_protocol_version();
3,416✔
752
        REALM_ASSERT_3(min, <=, max);
3,416✔
753
        // List protocol version in descending order to ensure that the server
1,694✔
754
        // selects the highest possible version.
1,694✔
755
        for (int version = max; version >= min; --version) {
34,168✔
756
            sec_websocket_protocol.push_back(util::format("%1%2", protocol_prefix, version)); // Throws
30,752✔
757
        }
30,752✔
758
    }
3,416✔
759

1,694✔
760
    logger.info("Connecting to '%1%2:%3%4'", to_string(m_server_endpoint.envelope), m_server_endpoint.address,
3,416✔
761
                m_server_endpoint.port, m_http_request_path_prefix);
3,416✔
762

1,694✔
763
    m_websocket_error_received = false;
3,416✔
764
    m_websocket =
3,416✔
765
        m_client.m_socket_provider->connect(std::make_unique<WebSocketObserverShim>(this),
3,416✔
766
                                            WebSocketEndpoint{
3,416✔
767
                                                m_server_endpoint.address,
3,416✔
768
                                                m_server_endpoint.port,
3,416✔
769
                                                get_http_request_path(),
3,416✔
770
                                                std::move(sec_websocket_protocol),
3,416✔
771
                                                is_ssl(m_server_endpoint.envelope),
3,416✔
772
                                                /// DEPRECATED - The following will be removed in a future release
1,694✔
773
                                                {m_custom_http_headers.begin(), m_custom_http_headers.end()},
3,416✔
774
                                                m_verify_servers_ssl_certificate,
3,416✔
775
                                                m_ssl_trust_certificate_path,
3,416✔
776
                                                m_ssl_verify_callback,
3,416✔
777
                                                m_proxy_config,
3,416✔
778
                                            });
3,416✔
779
}
3,416✔
780

781

782
void Connection::initiate_connect_wait()
783
{
3,418✔
784
    // Deploy a watchdog to enforce an upper bound on the time it can take to
1,694✔
785
    // fully establish the connection (including SSL and WebSocket
1,694✔
786
    // handshakes). Without such a watchdog, connect operations could take very
1,694✔
787
    // long, or even indefinite time.
1,694✔
788
    milliseconds_type time = m_client.m_connect_timeout;
3,418✔
789

1,694✔
790
    m_connect_timer = m_client.create_timer(std::chrono::milliseconds(time), [this](Status status) {
3,418✔
791
        // If the operation is aborted, the connection object may have been
1,694✔
792
        // destroyed.
1,694✔
793
        if (status != ErrorCodes::OperationAborted)
3,418✔
794
            handle_connect_wait(status); // Throws
×
795
    });                                  // Throws
3,418✔
796
}
3,418✔
797

798

799
void Connection::handle_connect_wait(Status status)
800
{
×
801
    if (!status.is_ok()) {
×
802
        REALM_ASSERT(status != ErrorCodes::OperationAborted);
×
803
        throw Exception(status);
×
804
    }
×
805

806
    REALM_ASSERT_EX(m_state == ConnectionState::connecting, m_state);
×
807
    logger.info("Connect timeout"); // Throws
×
808
    involuntary_disconnect(
×
809
        SessionErrorInfo{Status{ErrorCodes::SyncConnectTimeout, "Sync connection was not fully established in time"},
×
810
                         IsFatal{false}},
×
811
        ConnectionTerminationReason::sync_connect_timeout); // Throws
×
812
}
×
813

814

815
void Connection::handle_connection_established()
816
{
3,306✔
817
    // Cancel connect timeout watchdog
1,636✔
818
    m_connect_timer.reset();
3,306✔
819

1,636✔
820
    m_state = ConnectionState::connected;
3,306✔
821

1,636✔
822
    milliseconds_type now = monotonic_clock_now();
3,306✔
823
    m_pong_wait_started_at = now; // Initially, no time was spent waiting for a PONG message
3,306✔
824
    initiate_ping_delay(now);     // Throws
3,306✔
825

1,636✔
826
    bool fast_reconnect = false;
3,306✔
827
    if (m_disconnect_has_occurred) {
3,306✔
828
        milliseconds_type time = now - m_disconnect_time;
1,058✔
829
        if (time <= m_client.m_fast_reconnect_limit)
1,058✔
830
            fast_reconnect = true;
1,058✔
831
    }
1,058✔
832

1,636✔
833
    for (auto& p : m_sessions) {
4,478✔
834
        Session& sess = *p.second;
4,478✔
835
        sess.connection_established(fast_reconnect); // Throws
4,478✔
836
    }
4,478✔
837

1,636✔
838
    report_connection_state_change(ConnectionState::connected); // Throws
3,306✔
839
}
3,306✔
840

841

842
void Connection::schedule_urgent_ping()
843
{
212✔
844
    REALM_ASSERT_EX(m_state != ConnectionState::disconnected, m_state);
212✔
845
    if (m_ping_delay_in_progress) {
212✔
846
        m_heartbeat_timer.reset();
168✔
847
        m_ping_delay_in_progress = false;
168✔
848
        m_minimize_next_ping_delay = true;
168✔
849
        milliseconds_type now = monotonic_clock_now();
168✔
850
        initiate_ping_delay(now); // Throws
168✔
851
        return;
168✔
852
    }
168✔
853
    REALM_ASSERT_EX(m_state == ConnectionState::connecting || m_waiting_for_pong, m_state);
44✔
854
    if (!m_send_ping)
44✔
855
        m_minimize_next_ping_delay = true;
44✔
856
}
44✔
857

858

859
void Connection::initiate_ping_delay(milliseconds_type now)
860
{
3,646✔
861
    REALM_ASSERT(!m_ping_delay_in_progress);
3,646✔
862
    REALM_ASSERT(!m_waiting_for_pong);
3,646✔
863
    REALM_ASSERT(!m_send_ping);
3,646✔
864

1,762✔
865
    milliseconds_type delay = 0;
3,646✔
866
    if (!m_minimize_next_ping_delay) {
3,646✔
867
        delay = m_client.m_ping_keepalive_period;
3,462✔
868
        // Make a randomized deduction of up to 10%, or up to 100% if this is
1,684✔
869
        // the first PING message to be sent since the connection was
1,684✔
870
        // established. The purpose of this randomized deduction is to reduce
1,684✔
871
        // the risk of many connections sending PING messages simultaneously to
1,684✔
872
        // the server.
1,684✔
873
        milliseconds_type max_deduction = (m_ping_sent ? delay / 10 : delay);
3,352✔
874
        auto distr = std::uniform_int_distribution<milliseconds_type>(0, max_deduction);
3,462✔
875
        milliseconds_type randomized_deduction = distr(m_client.get_random());
3,462✔
876
        delay -= randomized_deduction;
3,462✔
877
        // Deduct the time spent waiting for PONG
1,684✔
878
        REALM_ASSERT_3(now, >=, m_pong_wait_started_at);
3,462✔
879
        milliseconds_type spent_time = now - m_pong_wait_started_at;
3,462✔
880
        if (spent_time < delay) {
3,462✔
881
            delay -= spent_time;
3,454✔
882
        }
3,454✔
883
        else {
8✔
884
            delay = 0;
8✔
885
        }
8✔
886
    }
3,462✔
887
    else {
184✔
888
        m_minimize_next_ping_delay = false;
184✔
889
    }
184✔
890

1,762✔
891

1,762✔
892
    m_ping_delay_in_progress = true;
3,646✔
893

1,762✔
894
    m_heartbeat_timer = m_client.create_timer(std::chrono::milliseconds(delay), [this](Status status) {
3,644✔
895
        if (status == ErrorCodes::OperationAborted)
3,644✔
896
            return;
3,456✔
897
        else if (!status.is_ok())
188✔
898
            throw Exception(status);
×
899

62✔
900
        handle_ping_delay();                                    // Throws
188✔
901
    });                                                         // Throws
188✔
902
    logger.debug("Will emit a ping in %1 milliseconds", delay); // Throws
3,646✔
903
}
3,646✔
904

905

906
void Connection::handle_ping_delay()
907
{
188✔
908
    REALM_ASSERT(m_ping_delay_in_progress);
188✔
909
    m_ping_delay_in_progress = false;
188✔
910
    m_send_ping = true;
188✔
911

62✔
912
    initiate_pong_timeout(); // Throws
188✔
913

62✔
914
    if (m_state == ConnectionState::connected && !m_sending)
188✔
915
        send_next_message(); // Throws
156✔
916
}
188✔
917

918

919
void Connection::initiate_pong_timeout()
920
{
188✔
921
    REALM_ASSERT(!m_ping_delay_in_progress);
188✔
922
    REALM_ASSERT(!m_waiting_for_pong);
188✔
923
    REALM_ASSERT(m_send_ping);
188✔
924

62✔
925
    m_waiting_for_pong = true;
188✔
926
    m_pong_wait_started_at = monotonic_clock_now();
188✔
927

62✔
928
    milliseconds_type time = m_client.m_pong_keepalive_timeout;
188✔
929
    m_heartbeat_timer = m_client.create_timer(std::chrono::milliseconds(time), [this](Status status) {
188✔
930
        if (status == ErrorCodes::OperationAborted)
188✔
931
            return;
176✔
932
        else if (!status.is_ok())
12✔
933
            throw Exception(status);
×
934

6✔
935
        handle_pong_timeout(); // Throws
12✔
936
    });                        // Throws
12✔
937
}
188✔
938

939

940
void Connection::handle_pong_timeout()
941
{
12✔
942
    REALM_ASSERT(m_waiting_for_pong);
12✔
943
    logger.debug("Timeout on reception of PONG message"); // Throws
12✔
944
    close_due_to_transient_error({ErrorCodes::ConnectionClosed, "Timed out waiting for PONG response from server"},
12✔
945
                                 ConnectionTerminationReason::pong_timeout);
12✔
946
}
12✔
947

948

949
void Connection::initiate_write_message(const OutputBuffer& out, Session* sess)
950
{
96,834✔
951
    // Stop sending messages if an websocket error was received.
46,370✔
952
    if (m_websocket_error_received)
96,834✔
953
        return;
×
954

46,370✔
955
    m_websocket->async_write_binary(out.as_span(), [this, sentinel = m_websocket_sentinel](Status status) {
96,834✔
956
        if (sentinel->destroyed) {
95,252✔
957
            return;
×
958
        }
×
959
        if (status == ErrorCodes::OperationAborted)
95,252✔
960
            return;
×
961
        else if (!status.is_ok())
95,252✔
962
            throw Exception(status);
×
963

45,416✔
964
        handle_write_message(); // Throws
95,252✔
965
    });                         // Throws
95,252✔
966
    m_sending_session = sess;
96,834✔
967
    m_sending = true;
96,834✔
968
}
96,834✔
969

970

971
void Connection::handle_write_message()
972
{
95,254✔
973
    m_sending_session->message_sent(); // Throws
95,254✔
974
    if (m_sending_session->m_state == Session::Deactivated) {
95,254✔
975
        finish_session_deactivation(m_sending_session);
102✔
976
    }
102✔
977
    m_sending_session = nullptr;
95,254✔
978
    m_sending = false;
95,254✔
979
    send_next_message(); // Throws
95,254✔
980
}
95,254✔
981

982

983
void Connection::send_next_message()
984
{
153,816✔
985
    REALM_ASSERT_EX(m_state == ConnectionState::connected, m_state);
153,816✔
986
    REALM_ASSERT(!m_sending_session);
153,816✔
987
    REALM_ASSERT(!m_sending);
153,816✔
988
    if (m_send_ping) {
153,816✔
989
        send_ping(); // Throws
176✔
990
        return;
176✔
991
    }
176✔
992
    while (!m_sessions_enlisted_to_send.empty()) {
213,976✔
993
        // The state of being connected is not supposed to be able to change
74,912✔
994
        // across this loop thanks to the "no callback reentrance" guarantee
74,912✔
995
        // provided by Websocket::async_write_text(), and friends.
74,912✔
996
        REALM_ASSERT_EX(m_state == ConnectionState::connected, m_state);
157,486✔
997

74,912✔
998
        Session& sess = *m_sessions_enlisted_to_send.front();
157,486✔
999
        m_sessions_enlisted_to_send.pop_front();
157,486✔
1000
        sess.send_message(); // Throws
157,486✔
1001

74,912✔
1002
        if (sess.m_state == Session::Deactivated) {
157,486✔
1003
            finish_session_deactivation(&sess);
2,294✔
1004
        }
2,294✔
1005

74,912✔
1006
        // An enlisted session may choose to not send a message. In that case,
74,912✔
1007
        // we should pass the opportunity to the next enlisted session.
74,912✔
1008
        if (m_sending)
157,486✔
1009
            break;
97,150✔
1010
    }
157,486✔
1011
}
153,640✔
1012

1013

1014
void Connection::send_ping()
1015
{
176✔
1016
    REALM_ASSERT(!m_ping_delay_in_progress);
176✔
1017
    REALM_ASSERT(m_waiting_for_pong);
176✔
1018
    REALM_ASSERT(m_send_ping);
176✔
1019

56✔
1020
    m_send_ping = false;
176✔
1021
    if (m_reconnect_info.scheduled_reset)
176✔
1022
        m_ping_after_scheduled_reset_of_reconnect_info = true;
148✔
1023

56✔
1024
    m_last_ping_sent_at = monotonic_clock_now();
176✔
1025
    logger.debug("Sending: PING(timestamp=%1, rtt=%2)", m_last_ping_sent_at,
176✔
1026
                 m_previous_ping_rtt); // Throws
176✔
1027

56✔
1028
    ClientProtocol& protocol = get_client_protocol();
176✔
1029
    OutputBuffer& out = get_output_buffer();
176✔
1030
    protocol.make_ping(out, m_last_ping_sent_at, m_previous_ping_rtt); // Throws
176✔
1031
    initiate_write_ping(out);                                          // Throws
176✔
1032
    m_ping_sent = true;
176✔
1033
}
176✔
1034

1035

1036
void Connection::initiate_write_ping(const OutputBuffer& out)
1037
{
176✔
1038
    m_websocket->async_write_binary(out.as_span(), [this, sentinel = m_websocket_sentinel](Status status) {
176✔
1039
        if (sentinel->destroyed) {
176✔
1040
            return;
×
1041
        }
×
1042
        if (status == ErrorCodes::OperationAborted)
176✔
1043
            return;
×
1044
        else if (!status.is_ok())
176✔
1045
            throw Exception(status);
×
1046

56✔
1047
        handle_write_ping(); // Throws
176✔
1048
    });                      // Throws
176✔
1049
    m_sending = true;
176✔
1050
}
176✔
1051

1052

1053
void Connection::handle_write_ping()
1054
{
176✔
1055
    REALM_ASSERT(m_sending);
176✔
1056
    REALM_ASSERT(!m_sending_session);
176✔
1057
    m_sending = false;
176✔
1058
    send_next_message(); // Throws
176✔
1059
}
176✔
1060

1061

1062
void Connection::handle_message_received(util::Span<const char> data)
1063
{
77,266✔
1064
    // parse_message_received() parses the message and calls the proper handler
38,586✔
1065
    // on the Connection object (this).
38,586✔
1066
    get_client_protocol().parse_message_received<Connection>(*this, std::string_view(data.data(), data.size()));
77,266✔
1067
}
77,266✔
1068

1069

1070
void Connection::initiate_disconnect_wait()
1071
{
4,268✔
1072
    REALM_ASSERT(!m_reconnect_delay_in_progress);
4,268✔
1073

2,012✔
1074
    if (m_disconnect_delay_in_progress) {
4,268✔
1075
        m_reconnect_disconnect_timer.reset();
2,060✔
1076
        m_disconnect_delay_in_progress = false;
2,060✔
1077
    }
2,060✔
1078

2,012✔
1079
    milliseconds_type time = m_client.m_connection_linger_time;
4,268✔
1080

2,012✔
1081
    m_reconnect_disconnect_timer = m_client.create_timer(std::chrono::milliseconds(time), [this](Status status) {
4,268✔
1082
        // If the operation is aborted, the connection object may have been
2,012✔
1083
        // destroyed.
2,012✔
1084
        if (status != ErrorCodes::OperationAborted)
4,268✔
1085
            handle_disconnect_wait(status); // Throws
6✔
1086
    });                                     // Throws
4,268✔
1087
    m_disconnect_delay_in_progress = true;
4,268✔
1088
}
4,268✔
1089

1090

1091
void Connection::handle_disconnect_wait(Status status)
1092
{
6✔
1093
    if (!status.is_ok()) {
6✔
1094
        REALM_ASSERT(status != ErrorCodes::OperationAborted);
×
1095
        throw Exception(status);
×
1096
    }
×
1097

1098
    m_disconnect_delay_in_progress = false;
6✔
1099

1100
    REALM_ASSERT_EX(m_state != ConnectionState::disconnected, m_state);
6✔
1101
    if (m_num_active_unsuspended_sessions == 0) {
6✔
1102
        if (m_client.m_connection_linger_time > 0)
6✔
1103
            logger.detail("Linger time expired"); // Throws
×
1104
        voluntary_disconnect();                   // Throws
6✔
1105
        logger.info("Disconnected");              // Throws
6✔
1106
    }
6✔
1107
}
6✔
1108

1109

1110
void Connection::close_due_to_protocol_error(Status status)
1111
{
4✔
1112
    SessionErrorInfo error_info(std::move(status), IsFatal{true});
4✔
1113
    error_info.server_requests_action = ProtocolErrorInfo::Action::ProtocolViolation;
4✔
1114
    involuntary_disconnect(std::move(error_info),
4✔
1115
                           ConnectionTerminationReason::sync_protocol_violation); // Throws
4✔
1116
}
4✔
1117

1118

1119
void Connection::close_due_to_client_side_error(Status status, IsFatal is_fatal, ConnectionTerminationReason reason)
1120
{
466✔
1121
    logger.info("Connection closed due to error: %1", status); // Throws
466✔
1122

272✔
1123
    involuntary_disconnect(SessionErrorInfo{std::move(status), is_fatal}, reason); // Throw
466✔
1124
}
466✔
1125

1126

1127
void Connection::close_due_to_transient_error(Status status, ConnectionTerminationReason reason)
1128
{
630✔
1129
    logger.info("Connection closed due to transient error: %1", status); // Throws
630✔
1130
    SessionErrorInfo error_info{std::move(status), IsFatal{false}};
630✔
1131
    error_info.server_requests_action = ProtocolErrorInfo::Action::Transient;
630✔
1132

330✔
1133
    involuntary_disconnect(std::move(error_info), reason); // Throw
630✔
1134
}
630✔
1135

1136

1137
// Close connection due to error discovered on the server-side, and then
1138
// reported to the client by way of a connection-level ERROR message.
1139
void Connection::close_due_to_server_side_error(ProtocolError error_code, const ProtocolErrorInfo& info)
1140
{
66✔
1141
    logger.info("Connection closed due to error reported by server: %1 (%2)", info.message,
66✔
1142
                int(error_code)); // Throws
66✔
1143

32✔
1144
    const auto reason = info.is_fatal ? ConnectionTerminationReason::server_said_do_not_reconnect
42✔
1145
                                      : ConnectionTerminationReason::server_said_try_again_later;
56✔
1146
    involuntary_disconnect(SessionErrorInfo{info, protocol_error_to_status(error_code, info.message)},
66✔
1147
                           reason); // Throws
66✔
1148
}
66✔
1149

1150

1151
void Connection::disconnect(const SessionErrorInfo& info)
1152
{
3,418✔
1153
    // Cancel connect timeout watchdog
1,694✔
1154
    m_connect_timer.reset();
3,418✔
1155

1,694✔
1156
    if (m_state == ConnectionState::connected) {
3,418✔
1157
        m_disconnect_time = monotonic_clock_now();
3,306✔
1158
        m_disconnect_has_occurred = true;
3,306✔
1159

1,636✔
1160
        // Sessions that are in the Deactivating state at this time can be
1,636✔
1161
        // immediately discarded, in part because they are no longer enlisted to
1,636✔
1162
        // send. Such sessions will be taken to the Deactivated state by
1,636✔
1163
        // Session::connection_lost(), and then they will be removed from
1,636✔
1164
        // `m_sessions`.
1,636✔
1165
        auto i = m_sessions.begin(), end = m_sessions.end();
3,306✔
1166
        while (i != end) {
7,108✔
1167
            // Prevent invalidation of the main iterator when erasing elements
1,992✔
1168
            auto j = i++;
3,802✔
1169
            Session& sess = *j->second;
3,802✔
1170
            sess.connection_lost(); // Throws
3,802✔
1171
            if (sess.m_state == Session::Unactivated || sess.m_state == Session::Deactivated)
3,802✔
1172
                m_sessions.erase(j);
1,566✔
1173
        }
3,802✔
1174
    }
3,306✔
1175

1,694✔
1176
    change_state_to_disconnected();
3,418✔
1177

1,694✔
1178
    m_ping_delay_in_progress = false;
3,418✔
1179
    m_waiting_for_pong = false;
3,418✔
1180
    m_send_ping = false;
3,418✔
1181
    m_minimize_next_ping_delay = false;
3,418✔
1182
    m_ping_after_scheduled_reset_of_reconnect_info = false;
3,418✔
1183
    m_ping_sent = false;
3,418✔
1184
    m_heartbeat_timer.reset();
3,418✔
1185
    m_previous_ping_rtt = 0;
3,418✔
1186

1,694✔
1187
    m_websocket_sentinel->destroyed = true;
3,418✔
1188
    m_websocket_sentinel.reset();
3,418✔
1189
    m_websocket.reset();
3,418✔
1190
    m_input_body_buffer.reset();
3,418✔
1191
    m_sending_session = nullptr;
3,418✔
1192
    m_sessions_enlisted_to_send.clear();
3,418✔
1193
    m_sending = false;
3,418✔
1194

1,694✔
1195
    report_connection_state_change(ConnectionState::disconnected, info); // Throws
3,418✔
1196
    initiate_reconnect_wait();                                           // Throws
3,418✔
1197
}
3,418✔
1198

1199
bool Connection::is_flx_sync_connection() const noexcept
1200
{
105,484✔
1201
    return m_server_endpoint.server_mode != SyncServerMode::PBS;
105,484✔
1202
}
105,484✔
1203

1204
void Connection::receive_pong(milliseconds_type timestamp)
1205
{
172✔
1206
    logger.debug("Received: PONG(timestamp=%1)", timestamp);
172✔
1207

54✔
1208
    bool legal_at_this_time = (m_waiting_for_pong && !m_send_ping);
172✔
1209
    if (REALM_UNLIKELY(!legal_at_this_time)) {
172✔
1210
        close_due_to_protocol_error(
×
1211
            {ErrorCodes::SyncProtocolInvariantFailed, "Received PONG message when it was not valid"}); // Throws
×
1212
        return;
×
1213
    }
×
1214

54✔
1215
    if (REALM_UNLIKELY(timestamp != m_last_ping_sent_at)) {
172✔
1216
        close_due_to_protocol_error(
×
1217
            {ErrorCodes::SyncProtocolInvariantFailed,
×
1218
             util::format("Received PONG message with an invalid timestamp (expected %1, received %2)",
×
1219
                          m_last_ping_sent_at, timestamp)}); // Throws
×
1220
        return;
×
1221
    }
×
1222

54✔
1223
    milliseconds_type now = monotonic_clock_now();
172✔
1224
    milliseconds_type round_trip_time = now - timestamp;
172✔
1225
    logger.debug("Round trip time was %1 milliseconds", round_trip_time);
172✔
1226
    m_previous_ping_rtt = round_trip_time;
172✔
1227

54✔
1228
    // If this PONG message is a response to a PING mesage that was sent after
54✔
1229
    // the last invocation of cancel_reconnect_delay(), then the connection is
54✔
1230
    // still good, and we do not have to skip the next reconnect delay.
54✔
1231
    if (m_ping_after_scheduled_reset_of_reconnect_info) {
172✔
1232
        REALM_ASSERT(m_reconnect_info.scheduled_reset);
132✔
1233
        m_ping_after_scheduled_reset_of_reconnect_info = false;
132✔
1234
        m_reconnect_info.scheduled_reset = false;
132✔
1235
    }
132✔
1236

54✔
1237
    m_heartbeat_timer.reset();
172✔
1238
    m_waiting_for_pong = false;
172✔
1239

54✔
1240
    initiate_ping_delay(now); // Throws
172✔
1241

54✔
1242
    if (m_client.m_roundtrip_time_handler)
172✔
1243
        m_client.m_roundtrip_time_handler(m_previous_ping_rtt); // Throws
×
1244
}
172✔
1245

1246
Session* Connection::find_and_validate_session(session_ident_type session_ident, std::string_view message) noexcept
1247
{
70,082✔
1248
    if (session_ident == 0) {
70,082✔
1249
        return nullptr;
×
1250
    }
×
1251

35,454✔
1252
    auto* sess = get_session(session_ident);
70,082✔
1253
    if (REALM_LIKELY(sess)) {
70,082✔
1254
        return sess;
70,080✔
1255
    }
70,080✔
1256
    // Check the history to see if the message received was for a previous session
1257
    if (auto it = m_session_history.find(session_ident); it == m_session_history.end()) {
2✔
1258
        logger.error("Bad session identifier in %1 message, session_ident = %2", message, session_ident);
×
1259
        close_due_to_protocol_error(
×
1260
            {ErrorCodes::SyncProtocolInvariantFailed,
×
1261
             util::format("Received message %1 for session iden %2 when that session never existed", message,
×
1262
                          session_ident)});
×
1263
    }
×
1264
    else {
2✔
1265
        logger.error("Received %1 message for closed session, session_ident = %2", message,
2✔
1266
                     session_ident); // Throws
2✔
1267
    }
2✔
1268
    return nullptr;
2✔
1269
}
2✔
1270

1271
void Connection::receive_error_message(const ProtocolErrorInfo& info, session_ident_type session_ident)
1272
{
954✔
1273
    Session* sess = nullptr;
954✔
1274
    if (session_ident != 0) {
954✔
1275
        sess = find_and_validate_session(session_ident, "ERROR");
884✔
1276
        if (REALM_UNLIKELY(!sess)) {
884✔
1277
            return;
×
1278
        }
×
1279
        if (auto status = sess->receive_error_message(info); !status.is_ok()) {
884✔
1280
            close_due_to_protocol_error(std::move(status)); // Throws
×
1281
            return;
×
1282
        }
×
1283

448✔
1284
        if (sess->m_state == Session::Deactivated) {
884✔
1285
            finish_session_deactivation(sess);
×
1286
        }
×
1287
        return;
884✔
1288
    }
884✔
1289

34✔
1290
    logger.info("Received: ERROR \"%1\" (error_code=%2, is_fatal=%3, session_ident=%4, error_action=%5)",
70✔
1291
                info.message, info.raw_error_code, info.is_fatal, session_ident,
70✔
1292
                info.server_requests_action); // Throws
70✔
1293

34✔
1294
    bool known_error_code = bool(get_protocol_error_message(info.raw_error_code));
70✔
1295
    if (REALM_LIKELY(known_error_code)) {
70✔
1296
        ProtocolError error_code = ProtocolError(info.raw_error_code);
66✔
1297
        if (REALM_LIKELY(!is_session_level_error(error_code))) {
66✔
1298
            close_due_to_server_side_error(error_code, info); // Throws
66✔
1299
            return;
66✔
1300
        }
66✔
1301
        close_due_to_protocol_error(
×
1302
            {ErrorCodes::SyncProtocolInvariantFailed,
×
1303
             util::format("Received ERROR message with a non-connection-level error code %1 without a session ident",
×
1304
                          info.raw_error_code)});
×
1305
    }
×
1306
    else {
4✔
1307
        close_due_to_protocol_error(
4✔
1308
            {ErrorCodes::SyncProtocolInvariantFailed,
4✔
1309
             util::format("Received ERROR message with unknown error code %1", info.raw_error_code)});
4✔
1310
    }
4✔
1311
}
70✔
1312

1313

1314
void Connection::receive_query_error_message(int raw_error_code, std::string_view message, int64_t query_version,
1315
                                             session_ident_type session_ident)
1316
{
16✔
1317
    if (session_ident == 0) {
16✔
1318
        return close_due_to_protocol_error(
×
1319
            {ErrorCodes::SyncProtocolInvariantFailed, "Received query error message for session ident 0"});
×
1320
    }
×
1321

8✔
1322
    if (!is_flx_sync_connection()) {
16✔
1323
        return close_due_to_protocol_error({ErrorCodes::SyncProtocolInvariantFailed,
×
1324
                                            "Received a FLX query error message on a non-FLX sync connection"});
×
1325
    }
×
1326

8✔
1327
    Session* sess = find_and_validate_session(session_ident, "QUERY_ERROR");
16✔
1328
    if (REALM_UNLIKELY(!sess)) {
16✔
1329
        return;
×
1330
    }
×
1331

8✔
1332
    if (auto status = sess->receive_query_error_message(raw_error_code, message, query_version); !status.is_ok()) {
16✔
1333
        close_due_to_protocol_error(std::move(status));
×
1334
    }
×
1335
}
16✔
1336

1337

1338
void Connection::receive_ident_message(session_ident_type session_ident, SaltedFileIdent client_file_ident)
1339
{
3,244✔
1340
    Session* sess = find_and_validate_session(session_ident, "IDENT");
3,244✔
1341
    if (REALM_UNLIKELY(!sess)) {
3,244✔
1342
        return;
×
1343
    }
×
1344

1,530✔
1345
    if (auto status = sess->receive_ident_message(client_file_ident); !status.is_ok())
3,244✔
1346
        close_due_to_protocol_error(std::move(status)); // Throws
×
1347
}
3,244✔
1348

1349
void Connection::receive_download_message(session_ident_type session_ident, const SyncProgress& progress,
1350
                                          std::uint_fast64_t downloadable_bytes, int64_t query_version,
1351
                                          DownloadBatchState batch_state,
1352
                                          const ReceivedChangesets& received_changesets)
1353
{
44,570✔
1354
    Session* sess = find_and_validate_session(session_ident, "DOWNLOAD");
44,570✔
1355
    if (REALM_UNLIKELY(!sess)) {
44,570✔
1356
        return;
×
1357
    }
×
1358

23,656✔
1359
    if (auto status = sess->receive_download_message(progress, downloadable_bytes, batch_state, query_version,
44,570✔
1360
                                                     received_changesets);
44,570✔
1361
        !status.is_ok()) {
44,570✔
1362
        close_due_to_protocol_error(std::move(status));
×
1363
    }
×
1364
}
44,570✔
1365

1366
void Connection::receive_mark_message(session_ident_type session_ident, request_ident_type request_ident)
1367
{
16,600✔
1368
    Session* sess = find_and_validate_session(session_ident, "MARK");
16,600✔
1369
    if (REALM_UNLIKELY(!sess)) {
16,600✔
1370
        return;
×
1371
    }
×
1372

7,758✔
1373
    if (auto status = sess->receive_mark_message(request_ident); !status.is_ok())
16,600✔
1374
        close_due_to_protocol_error(std::move(status)); // Throws
×
1375
}
16,600✔
1376

1377

1378
void Connection::receive_unbound_message(session_ident_type session_ident)
1379
{
4,724✔
1380
    Session* sess = find_and_validate_session(session_ident, "UNBOUND");
4,724✔
1381
    if (REALM_UNLIKELY(!sess)) {
4,724✔
1382
        return;
×
1383
    }
×
1384

2,032✔
1385
    if (auto status = sess->receive_unbound_message(); !status.is_ok()) {
4,724✔
1386
        close_due_to_protocol_error(std::move(status)); // Throws
×
1387
        return;
×
1388
    }
×
1389

2,032✔
1390
    if (sess->m_state == Session::Deactivated) {
4,724✔
1391
        finish_session_deactivation(sess);
4,724✔
1392
    }
4,724✔
1393
}
4,724✔
1394

1395

1396
void Connection::receive_test_command_response(session_ident_type session_ident, request_ident_type request_ident,
1397
                                               std::string_view body)
1398
{
44✔
1399
    Session* sess = find_and_validate_session(session_ident, "TEST_COMMAND");
44✔
1400
    if (REALM_UNLIKELY(!sess)) {
44✔
1401
        return;
×
1402
    }
×
1403

22✔
1404
    if (auto status = sess->receive_test_command_response(request_ident, body); !status.is_ok()) {
44✔
1405
        close_due_to_protocol_error(std::move(status));
×
1406
    }
×
1407
}
44✔
1408

1409

1410
void Connection::receive_server_log_message(session_ident_type session_ident, util::Logger::Level level,
1411
                                            std::string_view message)
1412
{
6,942✔
1413
    std::string prefix;
6,942✔
1414
    if (REALM_LIKELY(!m_appservices_coid.empty())) {
6,942✔
1415
        prefix = util::format("Server[%1]", m_appservices_coid);
6,942✔
1416
    }
6,942✔
1417
    else {
×
1418
        prefix = "Server";
×
1419
    }
×
1420

3,042✔
1421
    if (session_ident != 0) {
6,942✔
1422
        if (auto sess = get_session(session_ident)) {
4,954✔
1423
            sess->logger.log(level, "%1 log: %2", prefix, message);
4,954✔
1424
            return;
4,954✔
1425
        }
4,954✔
1426

1427
        logger.log(level, "%1 log for unknown session %2: %3", prefix, session_ident, message);
×
1428
        return;
×
1429
    }
×
1430

990✔
1431
    logger.log(level, "%1 log: %2", prefix, message);
1,988✔
1432
}
1,988✔
1433

1434

1435
void Connection::receive_appservices_request_id(std::string_view coid)
1436
{
5,294✔
1437
    // Only set once per connection
2,626✔
1438
    if (!coid.empty() && m_appservices_coid.empty()) {
5,294✔
1439
        m_appservices_coid = coid;
2,224✔
1440
        logger.info("Connected to app services with request id: \"%1\"", m_appservices_coid);
2,224✔
1441
    }
2,224✔
1442
}
5,294✔
1443

1444

1445
void Connection::handle_protocol_error(Status status)
1446
{
×
1447
    close_due_to_protocol_error(std::move(status));
×
1448
}
×
1449

1450

1451
// Sessions are guaranteed to be granted the opportunity to send a message in
1452
// the order that they enlist. Note that this is important to ensure
1453
// nonoverlapping communication with the server for consecutive sessions
1454
// associated with the same Realm file.
1455
//
1456
// CAUTION: The specified session may get destroyed before this function
1457
// returns, but only if its Session::send_message() puts it into the Deactivated
1458
// state.
1459
void Connection::enlist_to_send(Session* sess)
1460
{
159,158✔
1461
    REALM_ASSERT_EX(m_state == ConnectionState::connected, m_state);
159,158✔
1462
    m_sessions_enlisted_to_send.push_back(sess); // Throws
159,158✔
1463
    if (!m_sending)
159,158✔
1464
        send_next_message(); // Throws
58,234✔
1465
}
159,158✔
1466

1467

1468
std::string Connection::get_active_appservices_connection_id()
1469
{
72✔
1470
    return m_appservices_coid;
72✔
1471
}
72✔
1472

1473
void Session::cancel_resumption_delay()
1474
{
4,568✔
1475
    REALM_ASSERT_EX(m_state == Active, m_state);
4,568✔
1476

2,504✔
1477
    if (!m_suspended)
4,568✔
1478
        return;
4,184✔
1479

200✔
1480
    m_suspended = false;
384✔
1481

200✔
1482
    logger.debug("Resumed"); // Throws
384✔
1483

200✔
1484
    if (unbind_process_complete())
384✔
1485
        initiate_rebind(); // Throws
376✔
1486

200✔
1487
    m_conn.one_more_active_unsuspended_session(); // Throws
384✔
1488

200✔
1489
    on_resumed(); // Throws
384✔
1490
}
384✔
1491

1492

1493
void Session::gather_pending_compensating_writes(util::Span<Changeset> changesets,
1494
                                                 std::vector<ProtocolErrorInfo>* out)
1495
{
21,164✔
1496
    if (m_pending_compensating_write_errors.empty() || changesets.empty()) {
21,164✔
1497
        return;
21,122✔
1498
    }
21,122✔
1499

22✔
1500
#ifdef REALM_DEBUG
42✔
1501
    REALM_ASSERT_DEBUG(
42✔
1502
        std::is_sorted(m_pending_compensating_write_errors.begin(), m_pending_compensating_write_errors.end(),
42✔
1503
                       [](const ProtocolErrorInfo& lhs, const ProtocolErrorInfo& rhs) {
42✔
1504
                           return lhs.compensating_write_server_version < rhs.compensating_write_server_version;
42✔
1505
                       }));
42✔
1506
#endif
42✔
1507

22✔
1508
    while (!m_pending_compensating_write_errors.empty() &&
82✔
1509
           m_pending_compensating_write_errors.front().compensating_write_server_version <=
62✔
1510
               changesets.back().version) {
40✔
1511
        auto& cur_error = m_pending_compensating_write_errors.front();
40✔
1512
        REALM_ASSERT_3(cur_error.compensating_write_server_version, >=, changesets.front().version);
40✔
1513
        out->push_back(std::move(cur_error));
40✔
1514
        m_pending_compensating_write_errors.pop_front();
40✔
1515
    }
40✔
1516
}
42✔
1517

1518

1519
void Session::integrate_changesets(ClientReplication& repl, const SyncProgress& progress,
1520
                                   std::uint_fast64_t downloadable_bytes,
1521
                                   const ReceivedChangesets& received_changesets, VersionInfo& version_info,
1522
                                   DownloadBatchState download_batch_state)
1523
{
42,484✔
1524
    auto& history = repl.get_history();
42,484✔
1525
    if (received_changesets.empty()) {
42,484✔
1526
        if (download_batch_state == DownloadBatchState::MoreToCome) {
21,296✔
1527
            throw IntegrationException(ErrorCodes::SyncProtocolInvariantFailed,
×
1528
                                       "received empty download message that was not the last in batch",
×
1529
                                       ProtocolError::bad_progress);
×
1530
        }
×
1531
        history.set_sync_progress(progress, &downloadable_bytes, version_info); // Throws
21,296✔
1532
        return;
21,296✔
1533
    }
21,296✔
1534

11,700✔
1535
    std::vector<ProtocolErrorInfo> pending_compensating_write_errors;
21,188✔
1536
    auto transact = get_db()->start_read();
21,188✔
1537
    history.integrate_server_changesets(
21,188✔
1538
        progress, &downloadable_bytes, received_changesets, version_info, download_batch_state, logger, transact,
21,188✔
1539
        [&](const TransactionRef&, util::Span<Changeset> changesets) {
21,176✔
1540
            gather_pending_compensating_writes(changesets, &pending_compensating_write_errors);
21,164✔
1541
        },
21,164✔
1542
        get_transact_reporter()); // Throws
21,188✔
1543
    if (received_changesets.size() == 1) {
21,188✔
1544
        logger.debug("1 remote changeset integrated, producing client version %1",
14,594✔
1545
                     version_info.sync_version.version); // Throws
14,594✔
1546
    }
14,594✔
1547
    else {
6,594✔
1548
        logger.debug("%2 remote changesets integrated, producing client version %1",
6,594✔
1549
                     version_info.sync_version.version, received_changesets.size()); // Throws
6,594✔
1550
    }
6,594✔
1551

11,700✔
1552
    for (const auto& pending_error : pending_compensating_write_errors) {
11,720✔
1553
        logger.info("Reporting compensating write for client version %1 in server version %2: %3",
40✔
1554
                    pending_error.compensating_write_rejected_client_version,
40✔
1555
                    pending_error.compensating_write_server_version, pending_error.message);
40✔
1556
        try {
40✔
1557
            on_connection_state_changed(
40✔
1558
                m_conn.get_state(),
40✔
1559
                SessionErrorInfo{pending_error,
40✔
1560
                                 protocol_error_to_status(static_cast<ProtocolError>(pending_error.raw_error_code),
40✔
1561
                                                          pending_error.message)});
40✔
1562
        }
40✔
1563
        catch (...) {
20✔
1564
            logger.error("Exception thrown while reporting compensating write: %1", exception_to_status());
×
1565
        }
×
1566
    }
40✔
1567
}
21,188✔
1568

1569

1570
void Session::on_integration_failure(const IntegrationException& error)
1571
{
32✔
1572
    REALM_ASSERT_EX(m_state == Active, m_state);
32✔
1573
    REALM_ASSERT(!m_client_error && !m_error_to_send);
32✔
1574
    logger.error("Failed to integrate downloaded changesets: %1", error.to_status());
32✔
1575

16✔
1576
    m_client_error = util::make_optional<IntegrationException>(error);
32✔
1577
    m_error_to_send = true;
32✔
1578

16✔
1579
    // Surface the error to the user otherwise is lost.
16✔
1580
    on_connection_state_changed(m_conn.get_state(), SessionErrorInfo{error.to_status(), IsFatal{false}});
32✔
1581

16✔
1582
    // Since the deactivation process has not been initiated, the UNBIND
16✔
1583
    // message cannot have been sent unless an ERROR message was received.
16✔
1584
    REALM_ASSERT(m_suspended || m_error_message_received || !m_unbind_message_sent);
32✔
1585
    if (m_ident_message_sent && !m_error_message_received && !m_suspended) {
32✔
1586
        ensure_enlisted_to_send(); // Throws
32✔
1587
    }
32✔
1588
}
32✔
1589

1590
void Session::on_changesets_integrated(version_type client_version, const SyncProgress& progress)
1591
{
43,846✔
1592
    REALM_ASSERT_EX(m_state == Active, m_state);
43,846✔
1593
    REALM_ASSERT_3(progress.download.server_version, >=, m_download_progress.server_version);
43,846✔
1594
    m_download_progress = progress.download;
43,846✔
1595
    bool upload_progressed = (progress.upload.client_version > m_progress.upload.client_version);
43,846✔
1596
    m_progress = progress;
43,846✔
1597
    if (upload_progressed) {
43,846✔
1598
        if (progress.upload.client_version > m_last_version_selected_for_upload) {
32,296✔
1599
            if (progress.upload.client_version > m_upload_progress.client_version)
13,884✔
1600
                m_upload_progress = progress.upload;
1,570✔
1601
            m_last_version_selected_for_upload = progress.upload.client_version;
13,884✔
1602
        }
13,884✔
1603

16,592✔
1604
        check_for_upload_completion();
32,296✔
1605
    }
32,296✔
1606

23,250✔
1607
    do_recognize_sync_version(client_version); // Allows upload process to resume
43,846✔
1608
    check_for_download_completion();           // Throws
43,846✔
1609

23,250✔
1610
    // If the client migrated from PBS to FLX, create subscriptions when new tables are received from server.
23,250✔
1611
    if (auto migration_store = get_migration_store(); migration_store && m_is_flx_sync_session) {
43,846✔
1612
        auto& flx_subscription_store = *get_flx_subscription_store();
2,398✔
1613
        get_migration_store()->create_subscriptions(flx_subscription_store);
2,398✔
1614
    }
2,398✔
1615

23,250✔
1616
    // Since the deactivation process has not been initiated, the UNBIND
23,250✔
1617
    // message cannot have been sent unless an ERROR message was received.
23,250✔
1618
    REALM_ASSERT(m_suspended || m_error_message_received || !m_unbind_message_sent);
43,846✔
1619
    if (m_ident_message_sent && !m_error_message_received && !m_suspended) {
43,846✔
1620
        ensure_enlisted_to_send(); // Throws
43,842✔
1621
    }
43,842✔
1622
}
43,846✔
1623

1624

1625
Session::~Session()
1626
{
9,600✔
1627
    //    REALM_ASSERT_EX(m_state == Unactivated || m_state == Deactivated, m_state);
4,622✔
1628
}
9,600✔
1629

1630

1631
std::string Session::make_logger_prefix(session_ident_type ident)
1632
{
9,600✔
1633
    std::ostringstream out;
9,600✔
1634
    out.imbue(std::locale::classic());
9,600✔
1635
    out << "Session[" << ident << "]: "; // Throws
9,600✔
1636
    return out.str();                    // Throws
9,600✔
1637
}
9,600✔
1638

1639

1640
void Session::activate()
1641
{
9,598✔
1642
    REALM_ASSERT_EX(m_state == Unactivated, m_state);
9,598✔
1643

4,622✔
1644
    logger.debug("Activating"); // Throws
9,598✔
1645

4,622✔
1646
    bool has_pending_client_reset = false;
9,598✔
1647
    if (REALM_LIKELY(!get_client().is_dry_run())) {
9,600✔
1648
        // The reason we need a mutable reference from get_client_reset_config() is because we
4,620✔
1649
        // don't want the session to keep a strong reference to the client_reset_config->fresh_copy
4,620✔
1650
        // DB. If it did, then the fresh DB would stay alive for the duration of this sync session
4,620✔
1651
        // and we want to clean it up once the reset is finished. Additionally, the fresh copy will
4,620✔
1652
        // be set to a new copy on every reset so there is no reason to keep a reference to it.
4,620✔
1653
        // The modification to the client reset config happens via std::move(client_reset_config->fresh_copy).
4,620✔
1654
        // If the client reset config were a `const &` then this std::move would create another strong
4,620✔
1655
        // reference which we don't want to happen.
4,620✔
1656
        util::Optional<ClientReset>& client_reset_config = get_client_reset_config();
9,598✔
1657

4,620✔
1658
        bool file_exists = util::File::exists(get_realm_path());
9,598✔
1659

4,620✔
1660
        logger.info("client_reset_config = %1, Realm exists = %2, "
9,598✔
1661
                    "client reset = %3",
9,598✔
1662
                    client_reset_config ? "true" : "false", file_exists ? "true" : "false",
9,594✔
1663
                    (client_reset_config && file_exists) ? "true" : "false"); // Throws
9,598✔
1664
        if (client_reset_config && !m_client_reset_operation) {
9,598✔
1665
            m_client_reset_operation = std::make_unique<_impl::ClientResetOperation>(
336✔
1666
                logger, get_db(), std::move(client_reset_config->fresh_copy), client_reset_config->mode,
336✔
1667
                std::move(client_reset_config->notify_before_client_reset),
336✔
1668
                std::move(client_reset_config->notify_after_client_reset),
336✔
1669
                client_reset_config->recovery_is_allowed); // Throws
336✔
1670
        }
336✔
1671

4,620✔
1672
        if (!m_client_reset_operation) {
9,598✔
1673
            const ClientReplication& repl = access_realm(); // Throws
9,262✔
1674
            repl.get_history().get_status(m_last_version_available, m_client_file_ident, m_progress,
9,262✔
1675
                                          &has_pending_client_reset); // Throws
9,262✔
1676
        }
9,262✔
1677
    }
9,598✔
1678
    logger.debug("client_file_ident = %1, client_file_ident_salt = %2", m_client_file_ident.ident,
9,598✔
1679
                 m_client_file_ident.salt); // Throws
9,598✔
1680
    m_upload_target_version = m_last_version_available;
9,598✔
1681
    m_upload_progress = m_progress.upload;
9,598✔
1682
    m_last_version_selected_for_upload = m_upload_progress.client_version;
9,598✔
1683
    m_download_progress = m_progress.download;
9,598✔
1684
    REALM_ASSERT_3(m_last_version_available, >=, m_progress.upload.client_version);
9,598✔
1685

4,622✔
1686
    logger.debug("last_version_available  = %1", m_last_version_available);           // Throws
9,598✔
1687
    logger.debug("progress_download_server_version = %1", m_progress.download.server_version); // Throws
9,598✔
1688
    logger.debug("progress_download_client_version = %1",
9,598✔
1689
                 m_progress.download.last_integrated_client_version);                                      // Throws
9,598✔
1690
    logger.debug("progress_upload_server_version = %1", m_progress.upload.last_integrated_server_version); // Throws
9,598✔
1691
    logger.debug("progress_upload_client_version = %1", m_progress.upload.client_version);                 // Throws
9,598✔
1692

4,622✔
1693
    reset_protocol_state();
9,598✔
1694
    m_state = Active;
9,598✔
1695

4,622✔
1696
    REALM_ASSERT(!m_suspended);
9,598✔
1697
    m_conn.one_more_active_unsuspended_session(); // Throws
9,598✔
1698

4,622✔
1699
    try {
9,598✔
1700
        process_pending_flx_bootstrap();
9,598✔
1701
    }
9,598✔
1702
    catch (const IntegrationException& error) {
4,624✔
1703
        logger.error("Error integrating bootstrap changesets: %1", error.what());
4✔
1704
        m_suspended = true;
4✔
1705
        m_conn.one_less_active_unsuspended_session(); // Throws
4✔
1706
        on_suspended(SessionErrorInfo{Status{error.code(), error.what()}, IsFatal{true}});
4✔
1707
    }
4✔
1708

4,622✔
1709
    if (has_pending_client_reset) {
9,598✔
1710
        handle_pending_client_reset_acknowledgement();
24✔
1711
    }
24✔
1712
}
9,598✔
1713

1714

1715
// The caller (Connection) must discard the session if the session has become
1716
// deactivated upon return.
1717
void Session::initiate_deactivation()
1718
{
9,600✔
1719
    REALM_ASSERT_EX(m_state == Active, m_state);
9,600✔
1720

4,622✔
1721
    logger.debug("Initiating deactivation"); // Throws
9,600✔
1722

4,622✔
1723
    m_state = Deactivating;
9,600✔
1724

4,622✔
1725
    if (!m_suspended)
9,600✔
1726
        m_conn.one_less_active_unsuspended_session(); // Throws
9,076✔
1727

4,622✔
1728
    if (m_enlisted_to_send) {
9,600✔
1729
        REALM_ASSERT(!unbind_process_complete());
5,098✔
1730
        return;
5,098✔
1731
    }
5,098✔
1732

2,284✔
1733
    // Deactivate immediately if the BIND message has not yet been sent and the
2,284✔
1734
    // session is not enlisted to send, or if the unbinding process has already
2,284✔
1735
    // completed.
2,284✔
1736
    if (!m_bind_message_sent || unbind_process_complete()) {
4,502✔
1737
        complete_deactivation(); // Throws
914✔
1738
        // Life cycle state is now Deactivated
408✔
1739
        return;
914✔
1740
    }
914✔
1741

1,876✔
1742
    // Ready to send the UNBIND message, if it has not already been sent
1,876✔
1743
    if (!m_unbind_message_sent) {
3,588✔
1744
        enlist_to_send(); // Throws
3,414✔
1745
        return;
3,414✔
1746
    }
3,414✔
1747
}
3,588✔
1748

1749

1750
void Session::complete_deactivation()
1751
{
9,600✔
1752
    REALM_ASSERT_EX(m_state == Deactivating, m_state);
9,600✔
1753
    m_state = Deactivated;
9,600✔
1754

4,622✔
1755
    logger.debug("Deactivation completed"); // Throws
9,600✔
1756
}
9,600✔
1757

1758

1759
// Called by the associated Connection object when this session is granted an
1760
// opportunity to send a message.
1761
//
1762
// The caller (Connection) must discard the session if the session has become
1763
// deactivated upon return.
1764
void Session::send_message()
1765
{
157,482✔
1766
    REALM_ASSERT_EX(m_state == Active || m_state == Deactivating, m_state);
157,482✔
1767
    REALM_ASSERT(m_enlisted_to_send);
157,482✔
1768
    m_enlisted_to_send = false;
157,482✔
1769
    if (m_state == Deactivating || m_error_message_received || m_suspended) {
157,482✔
1770
        // Deactivation has been initiated. If the UNBIND message has not been
4,328✔
1771
        // sent yet, there is no point in sending it. Instead, we can let the
4,328✔
1772
        // deactivation process complete.
4,328✔
1773
        if (!m_bind_message_sent) {
8,982✔
1774
            return complete_deactivation(); // Throws
2,294✔
1775
            // Life cycle state is now Deactivated
1,368✔
1776
        }
2,294✔
1777

2,960✔
1778
        // Session life cycle state is Deactivating or the unbinding process has
2,960✔
1779
        // been initiated by a session specific ERROR message
2,960✔
1780
        if (!m_unbind_message_sent)
6,688✔
1781
            send_unbind_message(); // Throws
6,688✔
1782
        return;
6,688✔
1783
    }
6,688✔
1784

70,584✔
1785
    // Session life cycle state is Active and the unbinding process has
70,584✔
1786
    // not been initiated
70,584✔
1787
    REALM_ASSERT(!m_unbind_message_sent);
148,500✔
1788

70,584✔
1789
    if (!m_bind_message_sent)
148,500✔
1790
        return send_bind_message(); // Throws
9,272✔
1791

66,198✔
1792
    if (!m_ident_message_sent) {
139,228✔
1793
        if (have_client_file_ident())
7,746✔
1794
            send_ident_message(); // Throws
7,746✔
1795
        return;
7,746✔
1796
    }
7,746✔
1797

62,762✔
1798
    const auto has_pending_test_command = std::any_of(m_pending_test_commands.begin(), m_pending_test_commands.end(),
131,482✔
1799
                                                      [](const PendingTestCommand& command) {
62,834✔
1800
                                                          return command.pending;
144✔
1801
                                                      });
144✔
1802
    if (has_pending_test_command) {
131,482✔
1803
        return send_test_command_message();
44✔
1804
    }
44✔
1805

62,740✔
1806
    if (m_error_to_send)
131,438✔
1807
        return send_json_error_message(); // Throws
26✔
1808

62,728✔
1809
    // Stop sending upload, mark and query messages when the client detects an error.
62,728✔
1810
    if (m_client_error) {
131,412✔
1811
        return;
16✔
1812
    }
16✔
1813

62,720✔
1814
    if (m_target_download_mark > m_last_download_mark_sent)
131,396✔
1815
        return send_mark_message(); // Throws
17,162✔
1816

54,676✔
1817
    auto is_upload_allowed = [&]() -> bool {
114,242✔
1818
        if (!m_is_flx_sync_session) {
114,242✔
1819
            return true;
105,004✔
1820
        }
105,004✔
1821

4,722✔
1822
        auto migration_store = get_migration_store();
9,238✔
1823
        if (!migration_store) {
9,238✔
1824
            return true;
×
1825
        }
×
1826

4,722✔
1827
        auto sentinel_query_version = migration_store->get_sentinel_subscription_set_version();
9,238✔
1828
        if (!sentinel_query_version) {
9,238✔
1829
            return true;
9,212✔
1830
        }
9,212✔
1831

14✔
1832
        // Do not allow upload if the last query sent is the sentinel one used by the migration store.
14✔
1833
        return m_last_sent_flx_query_version != *sentinel_query_version;
26✔
1834
    };
26✔
1835

54,676✔
1836
    if (!is_upload_allowed()) {
114,234✔
1837
        return;
12✔
1838
    }
12✔
1839

54,670✔
1840
    auto check_pending_flx_version = [&]() -> bool {
114,230✔
1841
        if (!m_is_flx_sync_session) {
114,230✔
1842
            return false;
105,004✔
1843
        }
105,004✔
1844

4,716✔
1845
        if (!m_allow_upload) {
9,226✔
1846
            return false;
1,562✔
1847
        }
1,562✔
1848

3,930✔
1849
        m_pending_flx_sub_set = get_flx_subscription_store()->get_next_pending_version(
7,664✔
1850
            m_last_sent_flx_query_version, m_upload_progress.client_version);
7,664✔
1851

3,930✔
1852
        if (!m_pending_flx_sub_set) {
7,664✔
1853
            return false;
6,066✔
1854
        }
6,066✔
1855

800✔
1856
        return m_upload_progress.client_version >= m_pending_flx_sub_set->snapshot_version;
1,598✔
1857
    };
1,598✔
1858

54,670✔
1859
    if (check_pending_flx_version()) {
114,222✔
1860
        return send_query_change_message(); // throws
836✔
1861
    }
836✔
1862

54,252✔
1863
    REALM_ASSERT_3(m_upload_progress.client_version, <=, m_upload_target_version);
113,386✔
1864
    REALM_ASSERT_3(m_upload_target_version, <=, m_last_version_available);
113,386✔
1865
    if (m_allow_upload && (m_upload_target_version > m_upload_progress.client_version)) {
113,386✔
1866
        return send_upload_message(); // Throws
55,382✔
1867
    }
55,382✔
1868
}
113,386✔
1869

1870

1871
void Session::send_bind_message()
1872
{
9,272✔
1873
    REALM_ASSERT_EX(m_state == Active, m_state);
9,272✔
1874

4,386✔
1875
    session_ident_type session_ident = m_ident;
9,272✔
1876
    bool need_client_file_ident = !have_client_file_ident();
9,272✔
1877
    const bool is_subserver = false;
9,272✔
1878

4,386✔
1879

4,386✔
1880
    ClientProtocol& protocol = m_conn.get_client_protocol();
9,272✔
1881
    int protocol_version = m_conn.get_negotiated_protocol_version();
9,272✔
1882
    OutputBuffer& out = m_conn.get_output_buffer();
9,272✔
1883
    // Discard the token since it's ignored by the server.
4,386✔
1884
    std::string empty_access_token;
9,272✔
1885
    if (m_is_flx_sync_session) {
9,272✔
1886
        nlohmann::json bind_json_data;
1,348✔
1887
        if (auto migrated_partition = get_migration_store()->get_migrated_partition()) {
1,348✔
1888
            bind_json_data["migratedPartition"] = *migrated_partition;
60✔
1889
        }
60✔
1890
        bind_json_data["sessionReason"] = static_cast<uint64_t>(get_session_reason());
1,348✔
1891
        if (logger.would_log(util::Logger::Level::debug)) {
1,348✔
1892
            std::string json_data_dump;
1,348✔
1893
            if (!bind_json_data.empty()) {
1,348✔
1894
                json_data_dump = bind_json_data.dump();
1,348✔
1895
            }
1,348✔
1896
            logger.debug(
1,348✔
1897
                "Sending: BIND(session_ident=%1, need_client_file_ident=%2, is_subserver=%3, json_data=\"%4\")",
1,348✔
1898
                session_ident, need_client_file_ident, is_subserver, json_data_dump);
1,348✔
1899
        }
1,348✔
1900
        protocol.make_flx_bind_message(protocol_version, out, session_ident, bind_json_data, empty_access_token,
1,348✔
1901
                                       need_client_file_ident, is_subserver); // Throws
1,348✔
1902
    }
1,348✔
1903
    else {
7,924✔
1904
        std::string server_path = get_virt_path();
7,924✔
1905
        logger.debug("Sending: BIND(session_ident=%1, need_client_file_ident=%2, is_subserver=%3, server_path=%4)",
7,924✔
1906
                     session_ident, need_client_file_ident, is_subserver, server_path);
7,924✔
1907
        protocol.make_pbs_bind_message(protocol_version, out, session_ident, server_path, empty_access_token,
7,924✔
1908
                                       need_client_file_ident, is_subserver); // Throws
7,924✔
1909
    }
7,924✔
1910
    m_conn.initiate_write_message(out, this); // Throws
9,272✔
1911

4,386✔
1912
    m_bind_message_sent = true;
9,272✔
1913

4,386✔
1914
    // Ready to send the IDENT message if the file identifier pair is already
4,386✔
1915
    // available.
4,386✔
1916
    if (!need_client_file_ident)
9,272✔
1917
        enlist_to_send(); // Throws
4,668✔
1918
}
9,272✔
1919

1920

1921
void Session::send_ident_message()
1922
{
7,746✔
1923
    REALM_ASSERT_EX(m_state == Active, m_state);
7,746✔
1924
    REALM_ASSERT(m_bind_message_sent);
7,746✔
1925
    REALM_ASSERT(!m_unbind_message_sent);
7,746✔
1926
    REALM_ASSERT(have_client_file_ident());
7,746✔
1927

3,436✔
1928

3,436✔
1929
    ClientProtocol& protocol = m_conn.get_client_protocol();
7,746✔
1930
    OutputBuffer& out = m_conn.get_output_buffer();
7,746✔
1931
    session_ident_type session_ident = m_ident;
7,746✔
1932

3,436✔
1933
    if (m_is_flx_sync_session) {
7,746✔
1934
        const auto active_query_set = get_flx_subscription_store()->get_active();
954✔
1935
        const auto active_query_body = active_query_set.to_ext_json();
954✔
1936
        logger.debug("Sending: IDENT(client_file_ident=%1, client_file_ident_salt=%2, "
954✔
1937
                     "scan_server_version=%3, scan_client_version=%4, latest_server_version=%5, "
954✔
1938
                     "latest_server_version_salt=%6, query_version=%7, query_size=%8, query=\"%9\")",
954✔
1939
                     m_client_file_ident.ident, m_client_file_ident.salt, m_progress.download.server_version,
954✔
1940
                     m_progress.download.last_integrated_client_version, m_progress.latest_server_version.version,
954✔
1941
                     m_progress.latest_server_version.salt, active_query_set.version(), active_query_body.size(),
954✔
1942
                     active_query_body); // Throws
954✔
1943
        protocol.make_flx_ident_message(out, session_ident, m_client_file_ident, m_progress,
954✔
1944
                                        active_query_set.version(), active_query_body); // Throws
954✔
1945
        m_last_sent_flx_query_version = active_query_set.version();
954✔
1946
    }
954✔
1947
    else {
6,792✔
1948
        logger.debug("Sending: IDENT(client_file_ident=%1, client_file_ident_salt=%2, "
6,792✔
1949
                     "scan_server_version=%3, scan_client_version=%4, latest_server_version=%5, "
6,792✔
1950
                     "latest_server_version_salt=%6)",
6,792✔
1951
                     m_client_file_ident.ident, m_client_file_ident.salt, m_progress.download.server_version,
6,792✔
1952
                     m_progress.download.last_integrated_client_version, m_progress.latest_server_version.version,
6,792✔
1953
                     m_progress.latest_server_version.salt);                                  // Throws
6,792✔
1954
        protocol.make_pbs_ident_message(out, session_ident, m_client_file_ident, m_progress); // Throws
6,792✔
1955
    }
6,792✔
1956
    m_conn.initiate_write_message(out, this); // Throws
7,746✔
1957

3,436✔
1958
    m_ident_message_sent = true;
7,746✔
1959

3,436✔
1960
    // Other messages may be waiting to be sent
3,436✔
1961
    enlist_to_send(); // Throws
7,746✔
1962
}
7,746✔
1963

1964
void Session::send_query_change_message()
1965
{
836✔
1966
    REALM_ASSERT_EX(m_state == Active, m_state);
836✔
1967
    REALM_ASSERT(m_ident_message_sent);
836✔
1968
    REALM_ASSERT(!m_unbind_message_sent);
836✔
1969
    REALM_ASSERT(m_pending_flx_sub_set);
836✔
1970
    REALM_ASSERT_3(m_pending_flx_sub_set->query_version, >, m_last_sent_flx_query_version);
836✔
1971

418✔
1972
    if (REALM_UNLIKELY(get_client().is_dry_run())) {
836✔
1973
        return;
×
1974
    }
×
1975

418✔
1976
    auto sub_store = get_flx_subscription_store();
836✔
1977
    auto latest_sub_set = sub_store->get_by_version(m_pending_flx_sub_set->query_version);
836✔
1978
    auto latest_queries = latest_sub_set.to_ext_json();
836✔
1979
    logger.debug("Sending: QUERY(query_version=%1, query_size=%2, query=\"%3\", snapshot_version=%4)",
836✔
1980
                 latest_sub_set.version(), latest_queries.size(), latest_queries, latest_sub_set.snapshot_version());
836✔
1981

418✔
1982
    OutputBuffer& out = m_conn.get_output_buffer();
836✔
1983
    session_ident_type session_ident = get_ident();
836✔
1984
    ClientProtocol& protocol = m_conn.get_client_protocol();
836✔
1985
    protocol.make_query_change_message(out, session_ident, latest_sub_set.version(), latest_queries);
836✔
1986
    m_conn.initiate_write_message(out, this);
836✔
1987

418✔
1988
    m_last_sent_flx_query_version = latest_sub_set.version();
836✔
1989

418✔
1990
    request_download_completion_notification();
836✔
1991
}
836✔
1992

1993
void Session::send_upload_message()
1994
{
55,382✔
1995
    REALM_ASSERT_EX(m_state == Active, m_state);
55,382✔
1996
    REALM_ASSERT(m_ident_message_sent);
55,382✔
1997
    REALM_ASSERT(!m_unbind_message_sent);
55,382✔
1998
    REALM_ASSERT_3(m_upload_target_version, >, m_upload_progress.client_version);
55,382✔
1999

27,254✔
2000
    if (REALM_UNLIKELY(get_client().is_dry_run()))
55,382✔
2001
        return;
27,254✔
2002

27,254✔
2003
    auto target_upload_version = m_upload_target_version;
55,382✔
2004
    if (m_is_flx_sync_session) {
55,382✔
2005
        if (!m_pending_flx_sub_set || m_pending_flx_sub_set->snapshot_version < m_upload_progress.client_version) {
3,922✔
2006
            m_pending_flx_sub_set = get_flx_subscription_store()->get_next_pending_version(
3,160✔
2007
                m_last_sent_flx_query_version, m_upload_progress.client_version);
3,160✔
2008
        }
3,160✔
2009
        if (m_pending_flx_sub_set && m_pending_flx_sub_set->snapshot_version < m_upload_target_version) {
3,922✔
2010
            target_upload_version = m_pending_flx_sub_set->snapshot_version;
762✔
2011
        }
762✔
2012
    }
3,922✔
2013

27,254✔
2014
    const ClientReplication& repl = access_realm(); // Throws
55,382✔
2015

27,254✔
2016
    std::vector<UploadChangeset> uploadable_changesets;
55,382✔
2017
    version_type locked_server_version = 0;
55,382✔
2018
    repl.get_history().find_uploadable_changesets(m_upload_progress, target_upload_version, uploadable_changesets,
55,382✔
2019
                                                  locked_server_version); // Throws
55,382✔
2020

27,254✔
2021
    if (uploadable_changesets.empty()) {
55,382✔
2022
        // Nothing more to upload right now
13,232✔
2023
        check_for_upload_completion(); // Throws
27,790✔
2024
        // If we need to limit upload up to some version other than the last client version available and there are no
13,232✔
2025
        // changes to upload, then there is no need to send an empty message.
13,232✔
2026
        if (target_upload_version != m_upload_target_version) {
27,790✔
2027
            logger.debug("Empty UPLOAD was skipped (progress_client_version=%1, progress_server_version=%2)",
324✔
2028
                         m_upload_progress.client_version, m_upload_progress.last_integrated_server_version);
324✔
2029
            // Other messages may be waiting to be sent
162✔
2030
            return enlist_to_send(); // Throws
324✔
2031
        }
324✔
2032
    }
27,592✔
2033
    else {
27,592✔
2034
        m_last_version_selected_for_upload = uploadable_changesets.back().progress.client_version;
27,592✔
2035
    }
27,592✔
2036

27,254✔
2037
    if (m_is_flx_sync_session && m_pending_flx_sub_set && target_upload_version != m_upload_target_version) {
55,220✔
2038
        logger.trace("Limiting UPLOAD message up to version %1 to send QUERY version %2",
438✔
2039
                     m_pending_flx_sub_set->snapshot_version, m_pending_flx_sub_set->query_version);
438✔
2040
    }
438✔
2041

27,092✔
2042
    version_type progress_client_version = m_upload_progress.client_version;
55,058✔
2043
    version_type progress_server_version = m_upload_progress.last_integrated_server_version;
55,058✔
2044

27,092✔
2045
    logger.debug("Sending: UPLOAD(progress_client_version=%1, progress_server_version=%2, "
55,058✔
2046
                 "locked_server_version=%3, num_changesets=%4)",
55,058✔
2047
                 progress_client_version, progress_server_version, locked_server_version,
55,058✔
2048
                 uploadable_changesets.size()); // Throws
55,058✔
2049

27,092✔
2050
    ClientProtocol& protocol = m_conn.get_client_protocol();
55,058✔
2051
    ClientProtocol::UploadMessageBuilder upload_message_builder = protocol.make_upload_message_builder(); // Throws
55,058✔
2052

27,092✔
2053
    for (const UploadChangeset& uc : uploadable_changesets) {
49,306✔
2054
        logger.debug("Fetching changeset for upload (client_version=%1, server_version=%2, "
42,768✔
2055
                     "changeset_size=%3, origin_timestamp=%4, origin_file_ident=%5)",
42,768✔
2056
                     uc.progress.client_version, uc.progress.last_integrated_server_version, uc.changeset.size(),
42,768✔
2057
                     uc.origin_timestamp, uc.origin_file_ident); // Throws
42,768✔
2058
        if (logger.would_log(util::Logger::Level::trace)) {
42,768✔
2059
            BinaryData changeset_data = uc.changeset.get_first_chunk();
×
2060
            if (changeset_data.size() < 1024) {
×
2061
                logger.trace("Changeset: %1",
×
2062
                             _impl::clamped_hex_dump(changeset_data)); // Throws
×
2063
            }
×
2064
            else {
×
2065
                logger.trace("Changeset(comp): %1 %2", changeset_data.size(),
×
2066
                             protocol.compressed_hex_dump(changeset_data));
×
2067
            }
×
2068

2069
#if REALM_DEBUG
×
2070
            ChunkedBinaryInputStream in{changeset_data};
×
2071
            Changeset log;
×
2072
            try {
×
2073
                parse_changeset(in, log);
×
2074
                std::stringstream ss;
×
2075
                log.print(ss);
×
2076
                logger.trace("Changeset (parsed):\n%1", ss.str());
×
2077
            }
×
2078
            catch (const BadChangesetError& err) {
×
2079
                logger.error("Unable to parse changeset: %1", err.what());
×
2080
            }
×
2081
#endif
×
2082
        }
×
2083

20,554✔
2084
#if 0 // Upload log compaction is currently not implemented
2085
        if (!get_client().m_disable_upload_compaction) {
2086
            ChangesetEncoder::Buffer encode_buffer;
2087

2088
            {
2089
                // Upload compaction only takes place within single changesets to
2090
                // avoid another client seeing inconsistent snapshots.
2091
                ChunkedBinaryInputStream stream{uc.changeset};
2092
                Changeset changeset;
2093
                parse_changeset(stream, changeset); // Throws
2094
                // FIXME: What is the point of setting these? How can compaction care about them?
2095
                changeset.version = uc.progress.client_version;
2096
                changeset.last_integrated_remote_version = uc.progress.last_integrated_server_version;
2097
                changeset.origin_timestamp = uc.origin_timestamp;
2098
                changeset.origin_file_ident = uc.origin_file_ident;
2099

2100
                compact_changesets(&changeset, 1);
2101
                encode_changeset(changeset, encode_buffer);
2102

2103
                logger.debug("Upload compaction: original size = %1, compacted size = %2", uc.changeset.size(),
2104
                             encode_buffer.size()); // Throws
2105
            }
2106

2107
            upload_message_builder.add_changeset(
2108
                uc.progress.client_version, uc.progress.last_integrated_server_version, uc.origin_timestamp,
2109
                uc.origin_file_ident, BinaryData{encode_buffer.data(), encode_buffer.size()}); // Throws
2110
        }
2111
        else
2112
#endif
2113
        {
42,768✔
2114
            upload_message_builder.add_changeset(uc.progress.client_version,
42,768✔
2115
                                                 uc.progress.last_integrated_server_version, uc.origin_timestamp,
42,768✔
2116
                                                 uc.origin_file_ident,
42,768✔
2117
                                                 uc.changeset); // Throws
42,768✔
2118
        }
42,768✔
2119
    }
42,768✔
2120

27,092✔
2121
    int protocol_version = m_conn.get_negotiated_protocol_version();
55,058✔
2122
    OutputBuffer& out = m_conn.get_output_buffer();
55,058✔
2123
    session_ident_type session_ident = get_ident();
55,058✔
2124
    upload_message_builder.make_upload_message(protocol_version, out, session_ident, progress_client_version,
55,058✔
2125
                                               progress_server_version,
55,058✔
2126
                                               locked_server_version); // Throws
55,058✔
2127
    m_conn.initiate_write_message(out, this);                          // Throws
55,058✔
2128

27,092✔
2129
    // Other messages may be waiting to be sent
27,092✔
2130
    enlist_to_send(); // Throws
55,058✔
2131
}
55,058✔
2132

2133

2134
void Session::send_mark_message()
2135
{
17,162✔
2136
    REALM_ASSERT_EX(m_state == Active, m_state);
17,162✔
2137
    REALM_ASSERT(m_ident_message_sent);
17,162✔
2138
    REALM_ASSERT(!m_unbind_message_sent);
17,162✔
2139
    REALM_ASSERT_3(m_target_download_mark, >, m_last_download_mark_sent);
17,162✔
2140

8,044✔
2141
    request_ident_type request_ident = m_target_download_mark;
17,162✔
2142
    logger.debug("Sending: MARK(request_ident=%1)", request_ident); // Throws
17,162✔
2143

8,044✔
2144
    ClientProtocol& protocol = m_conn.get_client_protocol();
17,162✔
2145
    OutputBuffer& out = m_conn.get_output_buffer();
17,162✔
2146
    session_ident_type session_ident = get_ident();
17,162✔
2147
    protocol.make_mark_message(out, session_ident, request_ident); // Throws
17,162✔
2148
    m_conn.initiate_write_message(out, this);                      // Throws
17,162✔
2149

8,044✔
2150
    m_last_download_mark_sent = request_ident;
17,162✔
2151

8,044✔
2152
    // Other messages may be waiting to be sent
8,044✔
2153
    enlist_to_send(); // Throws
17,162✔
2154
}
17,162✔
2155

2156

2157
void Session::send_unbind_message()
2158
{
6,688✔
2159
    REALM_ASSERT_EX(m_state == Deactivating || m_error_message_received || m_suspended, m_state);
6,688✔
2160
    REALM_ASSERT(m_bind_message_sent);
6,688✔
2161
    REALM_ASSERT(!m_unbind_message_sent);
6,688✔
2162

2,960✔
2163
    logger.debug("Sending: UNBIND"); // Throws
6,688✔
2164

2,960✔
2165
    ClientProtocol& protocol = m_conn.get_client_protocol();
6,688✔
2166
    OutputBuffer& out = m_conn.get_output_buffer();
6,688✔
2167
    session_ident_type session_ident = get_ident();
6,688✔
2168
    protocol.make_unbind_message(out, session_ident); // Throws
6,688✔
2169
    m_conn.initiate_write_message(out, this);         // Throws
6,688✔
2170

2,960✔
2171
    m_unbind_message_sent = true;
6,688✔
2172
}
6,688✔
2173

2174

2175
void Session::send_json_error_message()
2176
{
26✔
2177
    REALM_ASSERT_EX(m_state == Active, m_state);
26✔
2178
    REALM_ASSERT(m_ident_message_sent);
26✔
2179
    REALM_ASSERT(!m_unbind_message_sent);
26✔
2180
    REALM_ASSERT(m_error_to_send);
26✔
2181
    REALM_ASSERT(m_client_error);
26✔
2182

12✔
2183
    ClientProtocol& protocol = m_conn.get_client_protocol();
26✔
2184
    OutputBuffer& out = m_conn.get_output_buffer();
26✔
2185
    session_ident_type session_ident = get_ident();
26✔
2186
    auto protocol_error = m_client_error->error_for_server;
26✔
2187

12✔
2188
    auto message = util::format("%1", m_client_error->to_status());
26✔
2189
    logger.info("Sending: ERROR \"%1\" (error_code=%2, session_ident=%3)", message, static_cast<int>(protocol_error),
26✔
2190
                session_ident); // Throws
26✔
2191

12✔
2192
    nlohmann::json error_body_json;
26✔
2193
    error_body_json["message"] = std::move(message);
26✔
2194
    protocol.make_json_error_message(out, session_ident, static_cast<int>(protocol_error),
26✔
2195
                                     error_body_json.dump()); // Throws
26✔
2196
    m_conn.initiate_write_message(out, this);                 // Throws
26✔
2197

12✔
2198
    m_error_to_send = false;
26✔
2199
    enlist_to_send(); // Throws
26✔
2200
}
26✔
2201

2202

2203
void Session::send_test_command_message()
2204
{
44✔
2205
    REALM_ASSERT_EX(m_state == Active, m_state);
44✔
2206

22✔
2207
    auto it = std::find_if(m_pending_test_commands.begin(), m_pending_test_commands.end(),
44✔
2208
                           [](const PendingTestCommand& command) {
44✔
2209
                               return command.pending;
44✔
2210
                           });
44✔
2211
    REALM_ASSERT(it != m_pending_test_commands.end());
44✔
2212

22✔
2213
    ClientProtocol& protocol = m_conn.get_client_protocol();
44✔
2214
    OutputBuffer& out = m_conn.get_output_buffer();
44✔
2215
    auto session_ident = get_ident();
44✔
2216

22✔
2217
    logger.info("Sending: TEST_COMMAND \"%1\" (session_ident=%2, request_ident=%3)", it->body, session_ident, it->id);
44✔
2218
    protocol.make_test_command_message(out, session_ident, it->id, it->body);
44✔
2219

22✔
2220
    m_conn.initiate_write_message(out, this); // Throws;
44✔
2221
    it->pending = false;
44✔
2222

22✔
2223
    enlist_to_send();
44✔
2224
}
44✔
2225

2226

2227
Status Session::receive_ident_message(SaltedFileIdent client_file_ident)
2228
{
3,244✔
2229
    logger.debug("Received: IDENT(client_file_ident=%1, client_file_ident_salt=%2)", client_file_ident.ident,
3,244✔
2230
                 client_file_ident.salt); // Throws
3,244✔
2231

1,530✔
2232
    // Ignore the message if the deactivation process has been initiated,
1,530✔
2233
    // because in that case, the associated Realm and SessionWrapper must
1,530✔
2234
    // not be accessed any longer.
1,530✔
2235
    if (m_state != Active)
3,244✔
2236
        return Status::OK(); // Success
76✔
2237

1,500✔
2238
    bool legal_at_this_time = (m_bind_message_sent && !have_client_file_ident() && !m_error_message_received &&
3,168✔
2239
                               !m_unbound_message_received);
3,168✔
2240
    if (REALM_UNLIKELY(!legal_at_this_time)) {
3,168✔
2241
        return {ErrorCodes::SyncProtocolInvariantFailed, "Received IDENT message when it was not legal"};
×
2242
    }
×
2243
    if (REALM_UNLIKELY(client_file_ident.ident < 1)) {
3,168✔
2244
        return {ErrorCodes::SyncProtocolInvariantFailed, "Bad client file identifier in IDENT message"};
×
2245
    }
×
2246
    if (REALM_UNLIKELY(client_file_ident.salt == 0)) {
3,168✔
2247
        return {ErrorCodes::SyncProtocolInvariantFailed, "Bad client file identifier salt in IDENT message"};
×
2248
    }
×
2249

1,500✔
2250
    m_client_file_ident = client_file_ident;
3,168✔
2251

1,500✔
2252
    if (REALM_UNLIKELY(get_client().is_dry_run())) {
3,168✔
2253
        // Ready to send the IDENT message
2254
        ensure_enlisted_to_send(); // Throws
×
2255
        return Status::OK();       // Success
×
2256
    }
×
2257

1,500✔
2258
    // access before the client reset (if applicable) because
1,500✔
2259
    // the reset can take a while and the sync session might have died
1,500✔
2260
    // by the time the reset finishes.
1,500✔
2261
    ClientReplication& repl = access_realm(); // Throws
3,168✔
2262

1,500✔
2263
    auto client_reset_if_needed = [&]() -> bool {
3,168✔
2264
        if (!m_client_reset_operation) {
3,168✔
2265
            return false;
2,832✔
2266
        }
2,832✔
2267

168✔
2268
        // ClientResetOperation::finalize() will return true only if the operation actually did
168✔
2269
        // a client reset. It may choose not to do a reset if the local Realm does not exist
168✔
2270
        // at this point (in that case there is nothing to reset). But in any case, we must
168✔
2271
        // clean up m_client_reset_operation at this point as sync should be able to continue from
168✔
2272
        // this point forward.
168✔
2273
        auto client_reset_operation = std::move(m_client_reset_operation);
336✔
2274
        util::UniqueFunction<void(int64_t)> on_flx_subscription_complete = [this](int64_t version) {
220✔
2275
            this->on_flx_sync_version_complete(version);
104✔
2276
        };
104✔
2277
        if (!client_reset_operation->finalize(client_file_ident, get_flx_subscription_store(),
336✔
2278
                                              std::move(on_flx_subscription_complete))) {
168✔
2279
            return false;
×
2280
        }
×
2281
        realm::VersionID client_reset_old_version = client_reset_operation->get_client_reset_old_version();
336✔
2282
        realm::VersionID client_reset_new_version = client_reset_operation->get_client_reset_new_version();
336✔
2283

168✔
2284
        // The fresh Realm has been used to reset the state
168✔
2285
        logger.debug("Client reset is completed, path=%1", get_realm_path()); // Throws
336✔
2286

168✔
2287
        SaltedFileIdent client_file_ident;
336✔
2288
        bool has_pending_client_reset = false;
336✔
2289
        repl.get_history().get_status(m_last_version_available, client_file_ident, m_progress,
336✔
2290
                                      &has_pending_client_reset); // Throws
336✔
2291
        REALM_ASSERT_3(m_client_file_ident.ident, ==, client_file_ident.ident);
336✔
2292
        REALM_ASSERT_3(m_client_file_ident.salt, ==, client_file_ident.salt);
336✔
2293
        REALM_ASSERT_EX(m_progress.download.last_integrated_client_version == 0,
336✔
2294
                        m_progress.download.last_integrated_client_version);
336✔
2295
        REALM_ASSERT_EX(m_progress.upload.client_version == 0, m_progress.upload.client_version);
336✔
2296
        REALM_ASSERT_EX(m_progress.upload.last_integrated_server_version == 0,
336✔
2297
                        m_progress.upload.last_integrated_server_version);
336✔
2298
        logger.trace("last_version_available  = %1", m_last_version_available); // Throws
336✔
2299

168✔
2300
        m_upload_target_version = m_last_version_available;
336✔
2301
        m_upload_progress = m_progress.upload;
336✔
2302
        m_download_progress = m_progress.download;
336✔
2303
        // In recovery mode, there may be new changesets to upload and nothing left to download.
168✔
2304
        // In FLX DiscardLocal mode, there may be new commits due to subscription handling.
168✔
2305
        // For both, we want to allow uploads again without needing external changes to download first.
168✔
2306
        m_allow_upload = true;
336✔
2307
        REALM_ASSERT_EX(m_last_version_selected_for_upload == 0, m_last_version_selected_for_upload);
336✔
2308

168✔
2309
        get_transact_reporter()->report_sync_transact(client_reset_old_version, client_reset_new_version);
336✔
2310

168✔
2311
        if (has_pending_client_reset) {
336✔
2312
            handle_pending_client_reset_acknowledgement();
268✔
2313
        }
268✔
2314

168✔
2315
        // If a migration or rollback is in progress, mark it complete when client reset is completed.
168✔
2316
        if (auto migration_store = get_migration_store()) {
336✔
2317
            migration_store->complete_migration_or_rollback();
240✔
2318
        }
240✔
2319

168✔
2320
        return true;
336✔
2321
    };
336✔
2322
    // if a client reset happens, it will take care of setting the file ident
1,500✔
2323
    // and if not, we do it here
1,500✔
2324
    bool did_client_reset = false;
3,168✔
2325
    try {
3,168✔
2326
        did_client_reset = client_reset_if_needed();
3,168✔
2327
    }
3,168✔
2328
    catch (const std::exception& e) {
1,534✔
2329
        auto err_msg = util::format("A fatal error occurred during client reset: '%1'", e.what());
68✔
2330
        logger.error(err_msg.c_str());
68✔
2331
        SessionErrorInfo err_info(Status{ErrorCodes::AutoClientResetFailed, err_msg}, IsFatal{true});
68✔
2332
        suspend(err_info);
68✔
2333
        return Status::OK();
68✔
2334
    }
68✔
2335
    if (!did_client_reset) {
3,100✔
2336
        repl.get_history().set_client_file_ident(client_file_ident,
2,832✔
2337
                                                 m_fix_up_object_ids); // Throws
2,832✔
2338
        m_progress.download.last_integrated_client_version = 0;
2,832✔
2339
        m_progress.upload.client_version = 0;
2,832✔
2340
        m_last_version_selected_for_upload = 0;
2,832✔
2341
    }
2,832✔
2342

1,466✔
2343
    // Ready to send the IDENT message
1,466✔
2344
    ensure_enlisted_to_send(); // Throws
3,100✔
2345
    return Status::OK();       // Success
3,100✔
2346
}
3,100✔
2347

2348
Status Session::receive_download_message(const SyncProgress& progress, std::uint_fast64_t downloadable_bytes,
2349
                                         DownloadBatchState batch_state, int64_t query_version,
2350
                                         const ReceivedChangesets& received_changesets)
2351
{
44,568✔
2352
    REALM_ASSERT_EX(query_version >= 0, query_version);
44,568✔
2353
    // Ignore the message if the deactivation process has been initiated,
23,656✔
2354
    // because in that case, the associated Realm and SessionWrapper must
23,656✔
2355
    // not be accessed any longer.
23,656✔
2356
    if (m_state != Active)
44,568✔
2357
        return Status::OK();
518✔
2358

23,352✔
2359
    if (is_steady_state_download_message(batch_state, query_version)) {
44,050✔
2360
        batch_state = DownloadBatchState::SteadyState;
42,488✔
2361
    }
42,488✔
2362

23,352✔
2363
    logger.debug("Received: DOWNLOAD(download_server_version=%1, download_client_version=%2, "
44,050✔
2364
                 "latest_server_version=%3, latest_server_version_salt=%4, "
44,050✔
2365
                 "upload_client_version=%5, upload_server_version=%6, downloadable_bytes=%7, "
44,050✔
2366
                 "last_in_batch=%8, query_version=%9, num_changesets=%10, ...)",
44,050✔
2367
                 progress.download.server_version, progress.download.last_integrated_client_version,
44,050✔
2368
                 progress.latest_server_version.version, progress.latest_server_version.salt,
44,050✔
2369
                 progress.upload.client_version, progress.upload.last_integrated_server_version, downloadable_bytes,
44,050✔
2370
                 batch_state != DownloadBatchState::MoreToCome, query_version, received_changesets.size()); // Throws
44,050✔
2371

23,352✔
2372
    // Ignore download messages when the client detects an error. This is to prevent transforming the same bad
23,352✔
2373
    // changeset over and over again.
23,352✔
2374
    if (m_client_error) {
44,050✔
2375
        logger.debug("Ignoring download message because the client detected an integration error");
×
2376
        return Status::OK();
×
2377
    }
×
2378

23,352✔
2379
    bool legal_at_this_time = (m_ident_message_sent && !m_error_message_received && !m_unbound_message_received);
44,052✔
2380
    if (REALM_UNLIKELY(!legal_at_this_time)) {
44,050✔
2381
        return {ErrorCodes::SyncProtocolInvariantFailed, "Received DOWNLOAD message when it was not legal"};
×
2382
    }
×
2383
    if (auto status = check_received_sync_progress(progress); REALM_UNLIKELY(!status.is_ok())) {
44,050✔
2384
        logger.error("Bad sync progress received (%1)", status);
×
2385
        return status;
×
2386
    }
×
2387

23,352✔
2388
    version_type server_version = m_progress.download.server_version;
44,050✔
2389
    version_type last_integrated_client_version = m_progress.download.last_integrated_client_version;
44,050✔
2390
    for (const Transformer::RemoteChangeset& changeset : received_changesets) {
44,458✔
2391
        // Check that per-changeset server version is strictly increasing, except in FLX sync where the server
22,334✔
2392
        // version must be increasing, but can stay the same during bootstraps.
22,334✔
2393
        bool good_server_version = m_is_flx_sync_session ? (changeset.remote_version >= server_version)
23,150✔
2394
                                                         : (changeset.remote_version > server_version);
42,624✔
2395
        // Each server version cannot be greater than the one in the header of the download message.
22,334✔
2396
        good_server_version = good_server_version && (changeset.remote_version <= progress.download.server_version);
43,440✔
2397
        if (!good_server_version) {
43,440✔
2398
            return {ErrorCodes::SyncProtocolInvariantFailed,
×
2399
                    util::format("Bad server version in changeset header (DOWNLOAD) (%1, %2, %3)",
×
2400
                                 changeset.remote_version, server_version, progress.download.server_version)};
×
2401
        }
×
2402
        server_version = changeset.remote_version;
43,440✔
2403
        // Check that per-changeset last integrated client version is "weakly"
22,334✔
2404
        // increasing.
22,334✔
2405
        bool good_client_version =
43,440✔
2406
            (changeset.last_integrated_local_version >= last_integrated_client_version &&
43,440✔
2407
             changeset.last_integrated_local_version <= progress.download.last_integrated_client_version);
43,440✔
2408
        if (!good_client_version) {
43,440✔
2409
            return {ErrorCodes::SyncProtocolInvariantFailed,
×
2410
                    util::format("Bad last integrated client version in changeset header (DOWNLOAD) "
×
2411
                                 "(%1, %2, %3)",
×
2412
                                 changeset.last_integrated_local_version, last_integrated_client_version,
×
2413
                                 progress.download.last_integrated_client_version)};
×
2414
        }
×
2415
        last_integrated_client_version = changeset.last_integrated_local_version;
43,440✔
2416
        // Server shouldn't send our own changes, and zero is not a valid client
22,334✔
2417
        // file identifier.
22,334✔
2418
        bool good_file_ident =
43,440✔
2419
            (changeset.origin_file_ident > 0 && changeset.origin_file_ident != m_client_file_ident.ident);
43,440✔
2420
        if (!good_file_ident) {
43,440✔
2421
            return {ErrorCodes::SyncProtocolInvariantFailed,
×
2422
                    util::format("Bad origin file identifier in changeset header (DOWNLOAD)",
×
2423
                                 changeset.origin_file_ident)};
×
2424
        }
×
2425
    }
43,440✔
2426

23,352✔
2427
    auto hook_action = call_debug_hook(SyncClientHookEvent::DownloadMessageReceived, progress, query_version,
44,050✔
2428
                                       batch_state, received_changesets.size());
44,050✔
2429
    if (hook_action == SyncClientHookAction::EarlyReturn) {
44,050✔
2430
        return Status::OK();
12✔
2431
    }
12✔
2432
    REALM_ASSERT_EX(hook_action == SyncClientHookAction::NoAction, hook_action);
44,038✔
2433

23,346✔
2434
    if (process_flx_bootstrap_message(progress, batch_state, query_version, received_changesets)) {
44,038✔
2435
        clear_resumption_delay_state();
1,558✔
2436
        return Status::OK();
1,558✔
2437
    }
1,558✔
2438

22,566✔
2439
    initiate_integrate_changesets(downloadable_bytes, batch_state, progress, received_changesets); // Throws
42,480✔
2440

22,566✔
2441
    hook_action = call_debug_hook(SyncClientHookEvent::DownloadMessageIntegrated, progress, query_version,
42,480✔
2442
                                  batch_state, received_changesets.size());
42,480✔
2443
    if (hook_action == SyncClientHookAction::EarlyReturn) {
42,480✔
2444
        return Status::OK();
×
2445
    }
×
2446
    REALM_ASSERT_EX(hook_action == SyncClientHookAction::NoAction, hook_action);
42,480✔
2447

22,566✔
2448
    // When we receive a DOWNLOAD message successfully, we can clear the backoff timer value used to reconnect
22,566✔
2449
    // after a retryable session error.
22,566✔
2450
    clear_resumption_delay_state();
42,480✔
2451
    return Status::OK();
42,480✔
2452
}
42,480✔
2453

2454
Status Session::receive_mark_message(request_ident_type request_ident)
2455
{
16,600✔
2456
    logger.debug("Received: MARK(request_ident=%1)", request_ident); // Throws
16,600✔
2457

7,758✔
2458
    // Ignore the message if the deactivation process has been initiated,
7,758✔
2459
    // because in that case, the associated Realm and SessionWrapper must
7,758✔
2460
    // not be accessed any longer.
7,758✔
2461
    if (m_state != Active)
16,600✔
2462
        return Status::OK(); // Success
84✔
2463

7,726✔
2464
    bool legal_at_this_time = (m_ident_message_sent && !m_error_message_received && !m_unbound_message_received);
16,516✔
2465
    if (REALM_UNLIKELY(!legal_at_this_time)) {
16,516✔
2466
        return {ErrorCodes::SyncProtocolInvariantFailed, "Received MARK message when it was not legal"};
×
2467
    }
×
2468
    bool good_request_ident =
16,516✔
2469
        (request_ident <= m_last_download_mark_sent && request_ident > m_last_download_mark_received);
16,516✔
2470
    if (REALM_UNLIKELY(!good_request_ident)) {
16,516✔
2471
        return {
×
2472
            ErrorCodes::SyncProtocolInvariantFailed,
×
2473
            util::format(
×
2474
                "Received MARK message with invalid request identifer (last mark sent: %1 last mark received: %2)",
×
2475
                m_last_download_mark_sent, m_last_download_mark_received)};
×
2476
    }
×
2477

7,726✔
2478
    m_server_version_at_last_download_mark = m_progress.download.server_version;
16,516✔
2479
    m_last_download_mark_received = request_ident;
16,516✔
2480
    check_for_download_completion(); // Throws
16,516✔
2481

7,726✔
2482
    return Status::OK(); // Success
16,516✔
2483
}
16,516✔
2484

2485

2486
// The caller (Connection) must discard the session if the session has become
2487
// deactivated upon return.
2488
Status Session::receive_unbound_message()
2489
{
4,724✔
2490
    logger.debug("Received: UNBOUND");
4,724✔
2491

2,032✔
2492
    bool legal_at_this_time = (m_unbind_message_sent && !m_error_message_received && !m_unbound_message_received);
4,724✔
2493
    if (REALM_UNLIKELY(!legal_at_this_time)) {
4,724✔
2494
        return {ErrorCodes::SyncProtocolInvariantFailed, "Received UNBOUND message when it was not legal"};
×
2495
    }
×
2496

2,032✔
2497
    // The fact that the UNBIND message has been sent, but an ERROR message has
2,032✔
2498
    // not been received, implies that the deactivation process must have been
2,032✔
2499
    // initiated, so this session must be in the Deactivating state or the session
2,032✔
2500
    // has been suspended because of a client side error.
2,032✔
2501
    REALM_ASSERT_EX(m_state == Deactivating || m_suspended, m_state);
4,724!
2502

2,032✔
2503
    m_unbound_message_received = true;
4,724✔
2504

2,032✔
2505
    // Detect completion of the unbinding process
2,032✔
2506
    if (m_unbind_message_send_complete && m_state == Deactivating) {
4,724✔
2507
        // The deactivation process completes when the unbinding process
2,032✔
2508
        // completes.
2,032✔
2509
        complete_deactivation(); // Throws
4,724✔
2510
        // Life cycle state is now Deactivated
2,032✔
2511
    }
4,724✔
2512

2,032✔
2513
    return Status::OK(); // Success
4,724✔
2514
}
4,724✔
2515

2516

2517
Status Session::receive_query_error_message(int error_code, std::string_view message, int64_t query_version)
2518
{
16✔
2519
    logger.info("Received QUERY_ERROR \"%1\" (error_code=%2, query_version=%3)", message, error_code, query_version);
16✔
2520
    // Ignore the message if the deactivation process has been initiated,
8✔
2521
    // because in that case, the associated Realm and SessionWrapper must
8✔
2522
    // not be accessed any longer.
8✔
2523
    if (m_state == Active) {
16✔
2524
        on_flx_sync_error(query_version, std::string_view(message.data(), message.size())); // throws
16✔
2525
    }
16✔
2526
    return Status::OK();
16✔
2527
}
16✔
2528

2529
// The caller (Connection) must discard the session if the session has become
2530
// deactivated upon return.
2531
Status Session::receive_error_message(const ProtocolErrorInfo& info)
2532
{
884✔
2533
    logger.info("Received: ERROR \"%1\" (error_code=%2, is_fatal=%3, error_action=%4)", info.message,
884✔
2534
                info.raw_error_code, info.is_fatal, info.server_requests_action); // Throws
884✔
2535

448✔
2536
    bool legal_at_this_time = (m_bind_message_sent && !m_error_message_received && !m_unbound_message_received);
884✔
2537
    if (REALM_UNLIKELY(!legal_at_this_time)) {
884✔
2538
        return {ErrorCodes::SyncProtocolInvariantFailed, "Received ERROR message when it was not legal"};
×
2539
    }
×
2540

448✔
2541
    auto protocol_error = static_cast<ProtocolError>(info.raw_error_code);
884✔
2542
    auto status = protocol_error_to_status(protocol_error, info.message);
884✔
2543
    if (status != ErrorCodes::UnknownError && REALM_UNLIKELY(!is_session_level_error(protocol_error))) {
884✔
2544
        return {ErrorCodes::SyncProtocolInvariantFailed,
×
2545
                util::format("Received ERROR message for session with non-session-level error code %1",
×
2546
                             info.raw_error_code)};
×
2547
    }
×
2548

448✔
2549
    // Can't process debug hook actions once the Session is undergoing deactivation, since
448✔
2550
    // the SessionWrapper may not be available
448✔
2551
    if (m_state == Active) {
884✔
2552
        auto debug_action = call_debug_hook(SyncClientHookEvent::ErrorMessageReceived, info);
880✔
2553
        if (debug_action == SyncClientHookAction::EarlyReturn) {
880✔
2554
            return Status::OK();
4✔
2555
        }
4✔
2556
    }
880✔
2557

446✔
2558
    // For compensating write errors, we need to defer raising them to the SDK until after the server version
446✔
2559
    // containing the compensating write has appeared in a download message.
446✔
2560
    if (status == ErrorCodes::SyncCompensatingWrite) {
880✔
2561
        // If the client is not active, the compensating writes will not be processed now, but will be
20✔
2562
        // sent again the next time the client connects
20✔
2563
        if (m_state == Active) {
40✔
2564
            m_pending_compensating_write_errors.push_back(info);
40✔
2565
        }
40✔
2566
        return Status::OK();
40✔
2567
    }
40✔
2568

426✔
2569
    m_error_message_received = true;
840✔
2570
    suspend(SessionErrorInfo{info, std::move(status)});
840✔
2571
    return Status::OK();
840✔
2572
}
840✔
2573

2574
void Session::suspend(const SessionErrorInfo& info)
2575
{
908✔
2576
    REALM_ASSERT(!m_suspended);
908✔
2577
    REALM_ASSERT_EX(m_state == Active || m_state == Deactivating, m_state);
908✔
2578
    logger.debug("Suspended"); // Throws
908✔
2579

460✔
2580
    m_suspended = true;
908✔
2581

460✔
2582
    // Detect completion of the unbinding process
460✔
2583
    if (m_unbind_message_send_complete && m_error_message_received) {
908!
2584
        // The fact that the UNBIND message has been sent, but we are not being suspended because
2585
        // we received an ERROR message implies that the deactivation process must
2586
        // have been initiated, so this session must be in the Deactivating state.
2587
        REALM_ASSERT_EX(m_state == Deactivating, m_state);
×
2588

2589
        // The deactivation process completes when the unbinding process
2590
        // completes.
2591
        complete_deactivation(); // Throws
×
2592
        // Life cycle state is now Deactivated
2593
    }
×
2594

460✔
2595
    // Notify the application of the suspension of the session if the session is
460✔
2596
    // still in the Active state
460✔
2597
    if (m_state == Active) {
908✔
2598
        m_conn.one_less_active_unsuspended_session(); // Throws
904✔
2599
        on_suspended(info);                           // Throws
904✔
2600
    }
904✔
2601

460✔
2602
    if (!info.is_fatal) {
908✔
2603
        begin_resumption_delay(info);
380✔
2604
    }
380✔
2605

460✔
2606
    // Ready to send the UNBIND message, if it has not been sent already
460✔
2607
    if (!m_unbind_message_sent)
908✔
2608
        ensure_enlisted_to_send(); // Throws
904✔
2609
}
908✔
2610

2611
Status Session::receive_test_command_response(request_ident_type ident, std::string_view body)
2612
{
44✔
2613
    logger.info("Received: TEST_COMMAND \"%1\" (session_ident=%2, request_ident=%3)", body, m_ident, ident);
44✔
2614
    auto it = std::find_if(m_pending_test_commands.begin(), m_pending_test_commands.end(),
44✔
2615
                           [&](const PendingTestCommand& command) {
44✔
2616
                               return command.id == ident;
44✔
2617
                           });
44✔
2618
    if (it == m_pending_test_commands.end()) {
44✔
2619
        return {ErrorCodes::SyncProtocolInvariantFailed,
×
2620
                util::format("Received test command response for a non-existent ident %1", ident)};
×
2621
    }
×
2622

22✔
2623
    it->promise.emplace_value(std::string{body});
44✔
2624
    m_pending_test_commands.erase(it);
44✔
2625

22✔
2626
    return Status::OK();
44✔
2627
}
44✔
2628

2629
void Session::begin_resumption_delay(const ProtocolErrorInfo& error_info)
2630
{
380✔
2631
    REALM_ASSERT(!m_try_again_activation_timer);
380✔
2632

198✔
2633
    m_try_again_delay_info.update(static_cast<sync::ProtocolError>(error_info.raw_error_code),
380✔
2634
                                  error_info.resumption_delay_interval);
380✔
2635
    auto try_again_interval = m_try_again_delay_info.delay_interval();
380✔
2636
    if (ProtocolError(error_info.raw_error_code) == ProtocolError::session_closed) {
380✔
2637
        // FIXME With compensating writes the server sends this error after completing a bootstrap. Doing the
10✔
2638
        // normal backoff behavior would result in waiting up to 5 minutes in between each query change which is
10✔
2639
        // not acceptable latency. So for this error code alone, we hard-code a 1 second retry interval.
10✔
2640
        try_again_interval = std::chrono::milliseconds{1000};
20✔
2641
    }
20✔
2642
    logger.debug("Will attempt to resume session after %1 milliseconds", try_again_interval.count());
380✔
2643
    m_try_again_activation_timer = get_client().create_timer(try_again_interval, [this](Status status) {
380✔
2644
        if (status == ErrorCodes::OperationAborted)
380✔
2645
            return;
4✔
2646
        else if (!status.is_ok())
376✔
2647
            throw Exception(status);
×
2648

196✔
2649
        m_try_again_activation_timer.reset();
376✔
2650
        cancel_resumption_delay();
376✔
2651
    });
376✔
2652
}
380✔
2653

2654
void Session::clear_resumption_delay_state()
2655
{
44,040✔
2656
    if (m_try_again_activation_timer) {
44,040✔
2657
        logger.debug("Clearing resumption delay state after successful download");
×
2658
        m_try_again_delay_info.reset();
×
2659
    }
×
2660
}
44,040✔
2661

2662
Status ClientImpl::Session::check_received_sync_progress(const SyncProgress& progress) noexcept
2663
{
44,054✔
2664
    const SyncProgress& a = m_progress;
44,054✔
2665
    const SyncProgress& b = progress;
44,054✔
2666
    std::string message;
44,054✔
2667
    if (b.latest_server_version.version < a.latest_server_version.version) {
44,054✔
2668
        message = util::format("Latest server version in download messages must be weakly increasing throughout a "
×
2669
                               "session (current: %1, received: %2)",
×
2670
                               a.latest_server_version.version, b.latest_server_version.version);
×
2671
    }
×
2672
    if (b.upload.client_version < a.upload.client_version) {
44,054✔
2673
        message = util::format("Last integrated client version in download messages must be weakly increasing "
×
2674
                               "throughout a session (current: %1, received: %2)",
×
2675
                               a.upload.client_version, b.upload.client_version);
×
2676
    }
×
2677
    if (b.upload.client_version > m_last_version_available) {
44,054✔
2678
        message = util::format("Last integrated client version on server cannot be greater than the latest client "
×
2679
                               "version in existence (current: %1, received: %2)",
×
2680
                               m_last_version_available, b.upload.client_version);
×
2681
    }
×
2682
    if (b.download.server_version < a.download.server_version) {
44,054✔
2683
        message =
×
2684
            util::format("Download cursor must be weakly increasing throughout a session (current: %1, received: %2)",
×
2685
                         a.download.server_version, b.download.server_version);
×
2686
    }
×
2687
    if (b.download.server_version > b.latest_server_version.version) {
44,054✔
2688
        message = util::format(
×
2689
            "Download cursor cannot be greater than the latest server version in existence (cursor: %1, latest: %2)",
×
2690
            b.download.server_version, b.latest_server_version.version);
×
2691
    }
×
2692
    if (b.download.last_integrated_client_version < a.download.last_integrated_client_version) {
44,054✔
2693
        message = util::format(
×
2694
            "Last integrated client version on the server at the position in the server's history of the download "
×
2695
            "cursor must be weakly increasing throughout a session (current: %1, received: %2)",
×
2696
            a.download.last_integrated_client_version, b.download.last_integrated_client_version);
×
2697
    }
×
2698
    if (b.download.last_integrated_client_version > b.upload.client_version) {
44,054✔
2699
        message = util::format("Last integrated client version on the server in the position at the server's history "
×
2700
                               "of the download cursor cannot be greater than the latest client version integrated "
×
2701
                               "on the server (download: %1, upload: %2)",
×
2702
                               b.download.last_integrated_client_version, b.upload.client_version);
×
2703
    }
×
2704
    if (b.download.server_version < b.upload.last_integrated_server_version) {
44,054✔
2705
        message = util::format(
×
2706
            "The server version of the download cursor cannot be less than the server version integrated in the "
×
2707
            "latest client version acknowledged by the server (download: %1, upload: %2)",
×
2708
            b.download.server_version, b.upload.last_integrated_server_version);
×
2709
    }
×
2710

23,354✔
2711
    if (message.empty()) {
44,054✔
2712
        return Status::OK();
44,054✔
2713
    }
44,054✔
UNCOV
2714
    return {ErrorCodes::SyncProtocolInvariantFailed, std::move(message)};
×
UNCOV
2715
}
×
2716

2717

2718
void Session::check_for_upload_completion()
2719
{
75,332✔
2720
    REALM_ASSERT_EX(m_state == Active, m_state);
75,332✔
2721
    if (!m_upload_completion_notification_requested) {
75,332✔
2722
        return;
43,384✔
2723
    }
43,384✔
2724

15,074✔
2725
    // during an ongoing client reset operation, we never upload anything
15,074✔
2726
    if (m_client_reset_operation)
31,948✔
2727
        return;
232✔
2728

14,958✔
2729
    // Upload process must have reached end of history
14,958✔
2730
    REALM_ASSERT_3(m_upload_progress.client_version, <=, m_last_version_available);
31,716✔
2731
    bool scan_complete = (m_upload_progress.client_version == m_last_version_available);
31,716✔
2732
    if (!scan_complete)
31,716✔
2733
        return;
6,050✔
2734

12,516✔
2735
    // All uploaded changesets must have been acknowledged by the server
12,516✔
2736
    REALM_ASSERT_3(m_progress.upload.client_version, <=, m_last_version_selected_for_upload);
25,666✔
2737
    bool all_uploads_accepted = (m_progress.upload.client_version == m_last_version_selected_for_upload);
25,666✔
2738
    if (!all_uploads_accepted)
25,666✔
2739
        return;
11,010✔
2740

7,242✔
2741
    m_upload_completion_notification_requested = false;
14,656✔
2742
    on_upload_completion(); // Throws
14,656✔
2743
}
14,656✔
2744

2745

2746
void Session::check_for_download_completion()
2747
{
60,358✔
2748
    REALM_ASSERT_3(m_target_download_mark, >=, m_last_download_mark_received);
60,358✔
2749
    REALM_ASSERT_3(m_last_download_mark_received, >=, m_last_triggering_download_mark);
60,358✔
2750
    if (m_last_download_mark_received == m_last_triggering_download_mark)
60,358✔
2751
        return;
43,642✔
2752
    if (m_last_download_mark_received < m_target_download_mark)
16,716✔
2753
        return;
428✔
2754
    if (m_download_progress.server_version < m_server_version_at_last_download_mark)
16,288✔
2755
        return;
×
2756
    m_last_triggering_download_mark = m_target_download_mark;
16,288✔
2757
    if (REALM_UNLIKELY(!m_allow_upload)) {
16,288✔
2758
        // Activate the upload process now, and enable immediate reactivation
1,988✔
2759
        // after a subsequent fast reconnect.
1,988✔
2760
        m_allow_upload = true;
5,034✔
2761
        ensure_enlisted_to_send(); // Throws
5,034✔
2762
    }
5,034✔
2763
    on_download_completion(); // Throws
16,288✔
2764
}
16,288✔
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2025 Coveralls, Inc