summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorMichael W. Hudson <mwh@python.net>2002-03-07 10:16:43 (GMT)
committerMichael W. Hudson <mwh@python.net>2002-03-07 10:16:43 (GMT)
commitaa3fd6f3dfdefb1b2efe553e8e9274eb177257fb (patch)
tree90d89be1a86a7c697e89a6b9f309c90daa763a14
parent38983bc5e3af22b5b923f809bcb06c0176123c0e (diff)
downloadcpython-aa3fd6f3dfdefb1b2efe553e8e9274eb177257fb.zip
cpython-aa3fd6f3dfdefb1b2efe553e8e9274eb177257fb.tar.gz
cpython-aa3fd6f3dfdefb1b2efe553e8e9274eb177257fb.tar.bz2
After some thinking, I decided to move all of this across onto
the branch. I don't think it can break code, so the only risk is introducing new bugs. In that vein, can I ask -checkins readers who have time to carefully look this over, check my refcounts &c? TIA. backport my checkin of revision 1.5 of structseq.c Apply (my) patch: [ 526072 ] pickling os.stat results round II structseq's constructors can now take "invisible" fields in a dict. Gave the constructors better error messages. their __reduce__ method puts these fields in a dict. (this is all in aid of getting os.stat_result's to pickle portably) Also fixes [ 526039 ] devious code can crash structseqs Thought needed about how much of this counts as a bugfix. Certainly #526039 needs to be fixed.
-rw-r--r--Objects/structseq.c99
1 files changed, 76 insertions, 23 deletions
diff --git a/Objects/structseq.c b/Objects/structseq.c
index e5f8e09..9228e0f 100644
--- a/Objects/structseq.c
+++ b/Objects/structseq.c
@@ -84,39 +84,79 @@ static PyObject *
structseq_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
PyObject *arg = NULL;
+ PyObject *dict = NULL;
+ PyObject *ob;
PyStructSequence *res = NULL;
- int len, required_len, i;
- static char *kwlist[] = {"sequence", 0};
- static char msgbuf[128];
+ int len, min_len, max_len, i;
+ static char *kwlist[] = {"sequence", "dict", 0};
- if (!PyArg_ParseTupleAndKeywords(args, kwds, "O:structseq",
- kwlist, &arg))
+ if (!PyArg_ParseTupleAndKeywords(args, kwds, "O|O:structseq",
+ kwlist, &arg, &dict))
return NULL;
- if (!PySequence_Check(arg)) {
- PyErr_SetString(PyExc_TypeError,
- "constructor requires a sequence");
+ arg = PySequence_Fast(arg, "constructor requires a sequence");
+
+ if (!arg) {
return NULL;
}
- len = PySequence_Length(arg);
- required_len = REAL_SIZE_TP(type);
- if (len != required_len) {
- PyOS_snprintf(
- msgbuf, sizeof(msgbuf),
- "constructor takes exactly %d arguments (%d given)",
- required_len,
- len);
- PyErr_SetString(PyExc_TypeError, msgbuf);
+ if (dict && !PyDict_Check(dict)) {
+ PyErr_Format(PyExc_TypeError,
+ "%.500s() takes a dict as second arg, if any",
+ type->tp_name);
+ Py_DECREF(arg);
return NULL;
}
+ len = PySequence_Fast_GET_SIZE(arg);
+ min_len = VISIBLE_SIZE_TP(type);
+ max_len = REAL_SIZE_TP(type);
+
+ if (min_len != max_len) {
+ if (len < min_len) {
+ PyErr_Format(PyExc_TypeError,
+ "%.500s() takes an at least %d-sequence (%d-sequence given)",
+ type->tp_name, min_len, len);
+ Py_DECREF(arg);
+ return NULL;
+ }
+
+ if (len > max_len) {
+ PyErr_Format(PyExc_TypeError,
+ "%.500s() takes an at most %d-sequence (%d-sequence given)",
+ type->tp_name, max_len, len);
+ Py_DECREF(arg);
+ return NULL;
+ }
+ }
+ else {
+ if (len != min_len) {
+ PyErr_Format(PyExc_TypeError,
+ "%.500s() takes a %d-sequence (%d-sequence given)",
+ type->tp_name, min_len, len);
+ Py_DECREF(arg);
+ return NULL;
+ }
+ }
+
res = (PyStructSequence*) PyStructSequence_New(type);
for (i = 0; i < len; ++i) {
- /* INCREF???? XXXX */
- res->ob_item[i] = PySequence_GetItem(arg, i);
+ PyObject *v = PySequence_Fast_GET_ITEM(arg, i);
+ Py_INCREF(v);
+ res->ob_item[i] = v;
+ }
+ for (; i < max_len; ++i) {
+ if (dict && (ob = PyDict_GetItemString(
+ dict, type->tp_members[i].name))) {
+ }
+ else {
+ ob = Py_None;
+ }
+ Py_INCREF(ob);
+ res->ob_item[i] = ob;
}
+ Py_DECREF(arg);
return (PyObject*) res;
}
@@ -192,21 +232,34 @@ static PyObject *
structseq_reduce(PyStructSequence* self)
{
PyObject* tup;
- long n_fields;
+ PyObject* dict;
+ long n_fields, n_visible_fields;
int i;
n_fields = REAL_SIZE(self);
- tup = PyTuple_New(n_fields);
+ n_visible_fields = VISIBLE_SIZE(self);
+ tup = PyTuple_New(n_visible_fields);
if (!tup) {
return NULL;
}
- for (i = 0; i < n_fields; i++) {
+ dict = PyDict_New();
+ if (!dict) {
+ Py_DECREF(tup);
+ return NULL;
+ }
+
+ for (i = 0; i < n_visible_fields; i++) {
Py_INCREF(self->ob_item[i]);
PyTuple_SET_ITEM(tup, i, self->ob_item[i]);
}
- return Py_BuildValue("(O(O))", self->ob_type, tup);
+ for (; i < n_fields; i++) {
+ PyDict_SetItemString(dict, self->ob_type->tp_members[i].name,
+ self->ob_item[i]);
+ }
+
+ return Py_BuildValue("(O(OO))", self->ob_type, tup, dict);
}
static PySequenceMethods structseq_as_sequence = {