python / python/cpython

[C-API Docs] Clarify Constraints on the Arena Allocator

Open
#101,655 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

3.11 3.12 3.13 docs topic-C-API
Dominant language
Python
Stars
77.2k
Forks
36k
PR merge metrics
PR metrics pending

Description

The "arena" allocator is used by obmalloc, as well as the code for allocating frames. It is exposed by the following public C-API:

  • PyObject_GetArenaAllocator()
  • PyObject_SetArenaAllocator()

(It is also exposed/used internally as _PyObject_VirtualAlloc() and _PyObject_VirtualFree(). The implementation is exposed internally as _PyMem_ArenaAlloc() and _PyMem_ArenaFree().)

First of all, the docs for PyObject_GetArenaAllocator() are not clear at all about what the function should be used for. I expect it is only meant for use in combination with PyObject_SetArenaAllocator() (e.g. to wrap the old one; to check what's set). That should be clarified.

Second, the C-API docs make it clear that the two public functions may be called before the runtime is initialized. However, the following should be clarified, as there are no documented restrictions AFAICS:

  • may PyObject_SetArenaAllocator() be called after runtime init?
    • status quo: presumably not, since obmalloc would break (similar to the restrictions on PyMem_SetAllocator())
    • internally: we do not ever change the arena allocator
    • recommendation: explicitly document that it must not be called after runtime init (a la PyMem_SetAllocator())
  • may the allocator returned by PyObject_GetArenaAllocator() be used directly?
    • status quo: possibly yes, since people may be doing so at this point (unlikely though)
    • internally: we use it directly (but that's kind of the point)
    • recommendation: explicitly state that it should not be used directly (point to object allocator API instead)
  • ...before runtime init?
    • (relevant only if direct use is okay)
    • status quo: possibly yes, since people may be doing so at this point (unlikely though)
    • internally: we do not use it before runtime init
    • recommendation: at the least document that it shouldn't be used after runtime init
    • better: state that it's only allowed when PyObject_SetArenaAllocator() is allowed (i.e. only before init)
  • must PyObject_GetArenaAllocator() be called with the GIL held after runtime init?
    • (relevant only if post-init PyObject_SetArenaAllocator() is okay)
    • status quo: possibly not, since people haven't needed to before (unlikely though)
    • internally: we always hold the GIL
    • recommendation: recommend (or require) that the GIL be held
  • must the allocator passed to PyObject_SetArenaAllocator() be thread-safe (and reentrant)?
    • status quo: if the allocator may be used directly then possibly yes (since it seems it can be used before runtime init, as well as later without the GIL held); otherwise no (since the PyMem "object" allocator is documented as protected by the GIL)
    • internally: the default arena allocator is thread-safe (and reentrant)
    • recommendation: recommend that the arena allocator be thread-safe (just like the PyMem "raw" allocator)
    • better: explicitly require it (so we can later drop the restriction on the PyMem "mem" and "object" allocators)

If we do specify tighter restrictions than were already documented then we'll need to consider who might be using the public functions in unexpected ways. However, I'd be surprised if there were more than a few (if any) using them at all.


Regardless, ISTM it would be better to deprecate the two public functions and replace them with a new PyConfig field. That would align a lot better with the intention. If there is a need to see if the currently set allocator matches some expected one, then it would be better to have an explicit function for that (e.g. PyObject_IsArenaAllocator()) rather than exposing PyObject_GetArenaAllocator() for that purpose.

The same goes for PyMem_SetAllocator(), where we'd have an additional PyMem_SetAllocatorWrapper() for the after-init case.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the linked C-API documentation for PyObject_GetArenaAllocator(), PyObject_SetArenaAllocator(), and the before-Python-initialization guidance, then compare the linked PyMem_SetAllocator restrictions. Determine and document the allowed initialization, GIL, direct-use, and thread-safety constraints; separately assess whether deprecation and a PyConfig replacement should be specified.

Written by the indexing model from the issue text.

Assessment

Tech stack
c, python
Domain
api, documentation
Issue type
Documentation
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.