From 15b293a7ec1914950dea4b14810f61f831423499 Mon Sep 17 00:00:00 2001
From: Borys <borys@dragonflydb.io>
Date: Wed, 18 Dec 2024 08:44:24 +0200
Subject: [PATCH] fix: crash during getting info about replication (#4328)

---
 src/server/rdb_save.cc | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/server/rdb_save.cc b/src/server/rdb_save.cc
index 516b5d67f..0500d411d 100644
--- a/src/server/rdb_save.cc
+++ b/src/server/rdb_save.cc
@@ -1333,6 +1333,10 @@ size_t RdbSaver::Impl::GetTotalBuffersSize() const {
 
   auto cb = [this, &channel_bytes, &serializer_bytes](ShardId sid) {
     auto& snapshot = shard_snapshots_[sid];
+    // before create a snapshot we save header so shard_snapshots_ are vector of nullptr until we
+    // start snapshots saving
+    if (!snapshot)
+      return;
     if (channel_.has_value())
       channel_bytes.fetch_add(channel_->GetSize(), memory_order_relaxed);
     serializer_bytes.store(snapshot->GetBufferCapacity() + snapshot->GetTempBuffersSize(),
@@ -1355,6 +1359,10 @@ RdbSaver::SnapshotStats RdbSaver::Impl::GetCurrentSnapshotProgress() const {
 
   auto cb = [this, &results](ShardId sid) {
     auto& snapshot = shard_snapshots_[sid];
+    // before create a snapshot we save header so shard_snapshots_ are vector of nullptr until we
+    // start snapshots saving
+    if (!snapshot)
+      return;
     results[sid] = snapshot->GetCurrentSnapshotProgress();
   };