fix: wrapping context __exit__ leak [backport 2.21]

ddtrace/internal/wrapping/context.py

@@ -229,11 +229,14 @@
 
     CONTEXT_RETURN.parse(
         r"""
+            pop_block
             load_const                  {context}
             load_method                 $__return__
             rot_three
             rot_three
             call_method                 1
+            rot_two
+            pop_top
         """
     )
 
@@ -257,11 +260,14 @@
 
     CONTEXT_RETURN.parse(
         r"""
+            pop_block
             load_const                  {context}
             load_method                 $__return__
             rot_three
             rot_three
             call_method                 1
+            rot_two
+            pop_top
         """
     )
 

releasenotes/notes/fix-wrapping-context-memory-leak-ad3d8f27586ba662.yaml

@@ -0,0 +1,5 @@
+---
+fixes:
+  - |
+    code origin: fixed a potential memory leak when collecting entry span
+    location information.

tests/internal/test_wrapping.py

@@ -1,5 +1,6 @@
 import asyncio
 from contextlib import asynccontextmanager
+import gc
 import inspect
 import sys
 from types import CoroutineType
@@ -805,3 +806,21 @@ async def fibonacci(n):
     await asyncio.gather(*[fibonacci(n) for n in range(1, N)])
 
     assert set(values) == {(n, n) for n in range(0, N)}
+
+
+def test_wrapping_context_method_leaks():
+    def foo():
+        return 42
+
+    wc = DummyWrappingContext(foo)
+    wc.wrap()
+
+    method_count = len([_ for _ in gc.get_objects() if type(_).__name__ == "method"])
+
+    for _ in range(10000):
+        foo()
+
+    gc.collect()
+
+    new_method_count = len([_ for _ in gc.get_objects() if type(_).__name__ == "method"])
+    assert new_method_count <= method_count + 1