Fix vulnerabilities in yql_plan.cpp reported by coverity (#9306)

ydb-platform · Sep 16, 2024 · ab3fea9 · ab3fea9
1 parent e7ab6ea
commit ab3fea9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ydb/library/yql/core/services/yql_plan.cpp b/ydb/library/yql/core/services/yql_plan.cpp
@@ -127,7 +127,9 @@ void WriteProviders(const TString& tag, const TProviderInfoMap& providers, NYson
             writer.OnListItem();
             writer.OnBeginMap();
             writer.OnKeyedItem("Id");
-            writer.OnUint64Scalar(p.second.Pin.find(pin)->second);
+            const auto found = p.second.Pin.find(pin);
+            YQL_ENSURE(found != p.second.Pin.cend());
+            writer.OnUint64Scalar(found->second);
             p.second.Provider->GetPlanFormatter().WritePinDetails(*pin, writer);
             writer.OnEndMap();
         }
@@ -412,7 +414,9 @@ class TPlanBuilder : public IPlanBuilder {
         THashMap<TPinKey, ui32, TPinKey::THash> allInputs;
         THashMap<TPinKey, ui32, TPinKey::THash> allOutputs;
         for (auto node : order) {
-            auto& info = nodes.find(node.Get())->second;
+            const auto found = nodes.find(node.Get());
+            YQL_ENSURE(found != nodes.cend());
+            auto& info = found->second;
             if (!info.IsVisible) {
                 continue;
             }