Skip to content

Commit

Permalink
Revert "pythonGH-78724: Initialize struct.Struct in __new__ (pythonGH…
Browse files Browse the repository at this point in the history
…-94532)"

This reverts commit c8c0afc.
  • Loading branch information
mdickinson committed Nov 26, 2023
1 parent 3faf8e5 commit 7190138
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 48 deletions.
14 changes: 0 additions & 14 deletions Lib/test/test_struct.py
Original file line number Diff line number Diff line change
Expand Up @@ -700,20 +700,6 @@ def test__struct_types_immutable(self):
with self.assertRaises(TypeError):
cls.x = 1

@support.cpython_only
def test__struct_Struct__new__initialized(self):
# See https://github.com/python/cpython/issues/78724

s = struct.Struct.__new__(struct.Struct, "b")
s.unpack_from(b"abcd")

@support.cpython_only
def test__struct_Struct_subclassing(self):
class Bob(struct.Struct):
pass

s = Bob("b")
s.unpack_from(b"abcd")

def test_issue35714(self):
# Embedded null characters should not be allowed in format strings.
Expand Down
59 changes: 33 additions & 26 deletions Modules/_struct.c
Original file line number Diff line number Diff line change
Expand Up @@ -1553,9 +1553,28 @@ prepare_s(PyStructObject *self)
return -1;
}

static PyObject *
s_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
PyObject *self;

assert(type != NULL);
allocfunc alloc_func = PyType_GetSlot(type, Py_tp_alloc);
assert(alloc_func != NULL);

self = alloc_func(type, 0);
if (self != NULL) {
PyStructObject *s = (PyStructObject*)self;
s->s_format = Py_NewRef(Py_None);
s->s_codes = NULL;
s->s_size = -1;
s->s_len = -1;
}
return self;
}

/*[clinic input]
@classmethod
Struct.__new__
Struct.__init__
format: object
Expand All @@ -1567,49 +1586,36 @@ the format string.
See help(struct) for more on format strings.
[clinic start generated code]*/

static PyObject *
Struct_impl(PyTypeObject *type, PyObject *format)
/*[clinic end generated code: output=49468b044e334308 input=8b91868eb1df0e28]*/
static int
Struct___init___impl(PyStructObject *self, PyObject *format)
/*[clinic end generated code: output=b8e80862444e92d0 input=192a4575a3dde802]*/
{
allocfunc alloc = PyType_GetSlot(type, Py_tp_alloc);
assert(alloc != NULL);
PyStructObject *self = (PyStructObject *)alloc(type, 0);

if (self == NULL) {
return NULL;
}
int ret = 0;

if (PyUnicode_Check(format)) {
format = PyUnicode_AsASCIIString(format);
if (format == NULL) {
Py_DECREF(self);
return NULL;
}
if (format == NULL)
return -1;
}
else {
Py_INCREF(format);
}

if (!PyBytes_Check(format)) {
Py_DECREF(format);
Py_DECREF(self);
PyErr_Format(PyExc_TypeError,
"Struct() argument 1 must be a str or bytes object, "
"not %.200s",
_PyType_Name(Py_TYPE(format)));
return NULL;
return -1;
}

self->s_format = format;
Py_SETREF(self->s_format, format);

if (prepare_s(self) < 0) {
Py_DECREF(self);
return NULL;
}
return (PyObject *)self;
ret = prepare_s(self);
return ret;
}


static int
s_clear(PyStructObject *s)
{
Expand Down Expand Up @@ -2219,8 +2225,9 @@ static PyType_Slot PyStructType_slots[] = {
{Py_tp_methods, s_methods},
{Py_tp_members, s_members},
{Py_tp_getset, s_getsetlist},
{Py_tp_new, Struct},
{Py_tp_init, Struct___init__},
{Py_tp_alloc, PyType_GenericAlloc},
{Py_tp_new, s_new},
{Py_tp_free, PyObject_GC_Del},
{0, 0},
};
Expand Down
16 changes: 8 additions & 8 deletions Modules/clinic/_struct.c.h

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 7190138

Please sign in to comment.