Avoid crash when converting dict with circular reference

Fixes pybind#73
dbaston · Dec 6, 2024 · 0b1864c · 0b1864c
1 parent 482c9a2
commit 0b1864c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/include/pybind11_json/pybind11_json.hpp b/include/pybind11_json/pybind11_json.hpp
@@ -9,6 +9,7 @@
 #ifndef PYBIND11_JSON_HPP
 #define PYBIND11_JSON_HPP
 
+#include <set>
 #include <string>
 #include <vector>
 
@@ -67,8 +68,12 @@ namespace pyjson
         }
     }
 
-    inline nl::json to_json(const py::handle& obj)
+    inline nl::json to_json(const py::handle& obj, std::set<const PyObject*>& refs)
     {
+        if (auto [_, unique] = refs.insert(obj.ptr()); !unique) {
+            throw std::runtime_error("Circular reference detected");
+        }
+
         if (obj.ptr() == nullptr || obj.is_none())
         {
             return nullptr;
@@ -121,7 +126,7 @@ namespace pyjson
             auto out = nl::json::array();
             for (const py::handle value : obj)
             {
-                out.push_back(to_json(value));
+                out.push_back(to_json(value, refs));
             }
             return out;
         }
@@ -130,12 +135,19 @@ namespace pyjson
             auto out = nl::json::object();
             for (const py::handle key : obj)
             {
-                out[py::str(key).cast<std::string>()] = to_json(obj[key]);
+                out[py::str(key).cast<std::string>()] = to_json(obj[key], refs);
             }
             return out;
         }
         throw std::runtime_error("to_json not implemented for this type of object: " + py::repr(obj).cast<std::string>());
     }
+
+    inline nl::json to_json(const py::handle& obj)
+    {
+        std::set<const PyObject*> refs;
+        return to_json(obj, refs);
+    }
+
 }
 
 // nlohmann_json serializers

diff --git a/test/test_pybind11_json.cpp b/test/test_pybind11_json.cpp
@@ -481,3 +481,17 @@ TEST(pybind11_caster_fromjson, dict)
     ASSERT_EQ(j["number"].cast<int>(), 1234);
     ASSERT_EQ(j["hello"].cast<std::string>(), "world");
 }
+
+TEST(pybind11_caster_tojson, recursive_dict)
+{
+    py::scoped_interpreter guard;
+    py::module m = create_module("test");
+
+    m.def("to_json", &test_fromtojson);
+
+    // Simulate calling this binding from Python with a dictionary as argument
+    py::dict obj("number"_a=1234, "hello"_a="world");
+    obj["recur"] = obj;
+
+    ASSERT_ANY_THROW(m.attr("to_json")(obj));
+}