WIP

Signed-off-by: Matthew A Johnson <matjoh@microsoft.com>

Author: matajoh

.gitignore

@@ -156,3 +156,6 @@ Python/frozen_modules/MANIFEST
 # Ignore ./python binary on Unix but still look into ./Python/ directory.
 /python
 !/Python/
+
+# Ignore the build directory.
+build*

Doc/c-api/typeobj.rst

@@ -2591,9 +2591,9 @@ Slot Type typedefs
    The purpose of this function is to separate memory allocation from memory
    initialization.  It should return a pointer to a block of memory of adequate
    length for the instance, suitably aligned, and initialized to zeros, but with
-   :c:member:`~PyObject.ob_refcnt` set to ``1`` and :c:member:`~PyObject.ob_type` set to the type argument.  If
+   :c:member:`~PyObject.ob_refcnt` set to ``2`` and :c:member:`~PyObject.ob_type` set to the type argument.  If
    the type's :c:member:`~PyTypeObject.tp_itemsize` is non-zero, the object's :c:member:`~PyVarObject.ob_size` field
-   should be initialized to *nitems* and the length of the allocated memory block
+   should be initialized to *nitems * 2* and the length of the allocated memory block
    should be ``tp_basicsize + nitems*tp_itemsize``, rounded up to a multiple of
    ``sizeof(void*)``; otherwise, *nitems* is not used and the length of the block
    should be :c:member:`~PyTypeObject.tp_basicsize`.

Doc/conf.py

@@ -215,6 +215,7 @@
     ('c:data', 'PyExc_UnicodeError'),
     ('c:data', 'PyExc_UnicodeTranslateError'),
     ('c:data', 'PyExc_ValueError'),
+    ('c:data', 'PyExc_NotWriteableError'),
     ('c:data', 'PyExc_ZeroDivisionError'),
     # C API: Standard Python warning classes
     ('c:data', 'PyExc_BytesWarning'),

Doc/data/stable_abi.dat

@@ -168,6 +168,13 @@ PyAPI_FUNC(void) _Py_NO_RETURN _Py_FatalErrorFormat(
     const char *format,
     ...);
 
+PyAPI_FUNC(PyObject *) _PyErr_WriteToImmutable(const char* file, int line, PyObject *obj);
+#define PyErr_WriteToImmutable(obj) _PyErr_WriteToImmutable(__FILE__, __LINE__, _PyObject_CAST(obj))
+
+PyAPI_FUNC(PyObject *) _PyErr_WriteToImmutableKey(const char* file, int line, PyObject *key);
+#define PyErr_WriteToImmutableKey(key) _PyErr_WriteToImmutableKey(__FILE__, __LINE__, _PyObject_CAST(key))
+
+
 extern PyObject *_PyErr_SetImportErrorWithNameFrom(
         PyObject *,
         PyObject *,

Include/cpython/pyerrors.h

@@ -24,14 +24,24 @@ typedef struct {
     /* Cached hash code of me_key. */
     Py_hash_t me_hash;
     PyObject *me_key;
-    PyObject *me_value; /* This field is only meaningful for combined tables */
+    PyObject *_me_value; /* This field is only meaningful for combined tables */
 } PyDictKeyEntry;
 
 typedef struct {
     PyObject *me_key;   /* The key must be Unicode and have hash. */
-    PyObject *me_value; /* This field is only meaningful for combined tables */
+    PyObject *_me_value; /* This field is only meaningful for combined tables */
 } PyDictUnicodeEntry;
 
+#define _PyDictEntry_IsImmutable(entry) (((uintptr_t)((entry)->_me_value)) & 0x1)
+#define _PyDictEntry_SetImmutable(entry) ((entry)->_me_value = (PyObject*)((uintptr_t)(entry)->_me_value | 0x1))
+#define _PyDictEntry_Hash(entry) ((entry)->me_hash)
+#define _PyDictEntry_Key(entry) ((entry)->me_key)
+#define _PyDictEntry_Value(entry) ((PyObject*)((((uintptr_t)((entry)->_me_value)) >> 1) << 1))
+#define _PyDictEntry_SetValue(entry, value) ((entry)->_me_value = value)
+#define _PyDictEntry_IsEmpty(entry) ((entry)->_me_value == NULL)
+
+extern PyObject *_PyDict_IsKeyImmutable(PyObject* op, PyObject* key);
+
 extern PyDictKeysObject *_PyDict_NewKeysForClass(void);
 extern PyObject *_PyDict_FromKeys(PyObject *, PyObject *, PyObject *);
 
@@ -50,6 +60,7 @@ extern Py_ssize_t _Py_dict_lookup(PyDictObject *mp, PyObject *key, Py_hash_t has
 extern Py_ssize_t _PyDict_LookupIndex(PyDictObject *, PyObject *);
 extern Py_ssize_t _PyDictKeys_StringLookup(PyDictKeysObject* dictkeys, PyObject *key);
 extern PyObject *_PyDict_LoadGlobal(PyDictObject *, PyDictObject *, PyObject *);
+extern PyObject *_PyDict_SetKeyImmutable(PyDictObject *mp, PyObject *key);
 
 /* Consumes references to key and value */
 extern int _PyDict_SetItem_Take2(PyDictObject *op, PyObject *key, PyObject *value);

Include/internal/pycore_dict.h

@@ -198,7 +198,7 @@ _PyFrame_MakeAndSetFrameObject(_PyInterpreterFrame *frame);
 
 /* Gets the PyFrameObject for this frame, lazily
  * creating it if necessary.
- * Returns a borrowed referennce */
+ * Returns a borrowed reference */
 static inline PyFrameObject *
 _PyFrame_GetFrameObject(_PyInterpreterFrame *frame)
 {

Include/internal/pycore_frame.h

@@ -0,0 +1,22 @@
+#ifndef Py_FREEZE_H
+#define Py_FREEZE_H
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+#ifndef Py_BUILD_CORE
+#  error "this header requires Py_BUILD_CORE define"
+#endif
+
+PyObject* _Py_Freeze(PyObject*);
+#define Py_Freeze(op) _Py_Freeze(_PyObject_CAST(op))
+
+PyObject* _Py_FreezeGlobals(void);
+
+bool _Py_GlobalsImmutable_Check(void);
+
+#ifdef __cplusplus
+}
+#endif
+#endif /* !Py_FREEZE_H */

Include/internal/pycore_global_objects_fini_generated.h

@@ -64,7 +64,8 @@ static inline void _Py_RefcntAdd(PyObject* op, Py_ssize_t n)
 #ifdef Py_REF_DEBUG
     _Py_AddRefTotal(_PyInterpreterState_GET(), n);
 #endif
-    op->ob_refcnt += n;
+    op->ob_refcnt += n * 2;
+    assert(op->ob_refcnt % 2 == 0);
 }
 #define _Py_RefcntAdd(op, n) _Py_RefcntAdd(_PyObject_CAST(op), n)
 
@@ -81,7 +82,7 @@ static inline void _Py_ClearImmortal(PyObject *op)
 {
     if (op) {
         assert(op->ob_refcnt == _Py_IMMORTAL_REFCNT);
-        op->ob_refcnt = 1;
+        op->ob_refcnt = 2;
         Py_DECREF(op);
     }
 }
@@ -91,6 +92,19 @@ static inline void _Py_ClearImmortal(PyObject *op)
         op = NULL; \
     } while (0)
 
+#define _Py_IMMUTABLE_FLAG 0x1
+
+static inline void _Py_SetImmutable(PyObject *op)
+{
+    if(op) {
+        op->ob_refcnt |= _Py_IMMUTABLE_FLAG;
+    }
+}
+#define _Py_SetImmutable(op) _Py_SetImmutable(_PyObject_CAST(op))
+
+#define Py_CHECKWRITE(op) ((op) && (!_Py_IsImmutable(op)))
+#define Py_REQUIREWRITE(op, msg) {if (Py_CHECKWRITE(op)) { _PyObject_ASSERT_FAILED_MSG(op, msg); }}
+
 static inline void
 _Py_DECREF_SPECIALIZED(PyObject *op, const destructor destruct)
 {
@@ -101,7 +115,8 @@ _Py_DECREF_SPECIALIZED(PyObject *op, const destructor destruct)
 #ifdef Py_REF_DEBUG
     _Py_DEC_REFTOTAL(_PyInterpreterState_GET());
 #endif
-    if (--op->ob_refcnt != 0) {
+    op->ob_refcnt -= 2;
+    if (op->ob_refcnt != 0) {
         assert(op->ob_refcnt > 0);
     }
     else {
@@ -122,7 +137,7 @@ _Py_DECREF_NO_DEALLOC(PyObject *op)
 #ifdef Py_REF_DEBUG
     _Py_DEC_REFTOTAL(_PyInterpreterState_GET());
 #endif
-    op->ob_refcnt--;
+    op->ob_refcnt -= 2;
 #ifdef Py_DEBUG
     if (op->ob_refcnt <= 0) {
         _Py_FatalRefcountError("Expected a positive remaining refcount");