diff options
Diffstat (limited to 'Modules')
| -rw-r--r-- | Modules/_pickle.c | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/Modules/_pickle.c b/Modules/_pickle.c index 6f33c25..3ad55b5 100644 --- a/Modules/_pickle.c +++ b/Modules/_pickle.c @@ -3639,25 +3639,24 @@ load_binpersid(UnpicklerObject *self) static int load_pop(UnpicklerObject *self) { - int len; - - if ((len = self->stack->length) <= 0) - return stack_underflow(); + int len = self->stack->length; /* Note that we split the (pickle.py) stack into two stacks, * an object stack and a mark stack. We have to be clever and * pop the right one. We do this by looking at the top of the - * mark stack. + * mark stack first, and only signalling a stack underflow if + * the object stack is empty and the mark stack doesn't match + * our expectations. */ - - if ((self->num_marks > 0) && (self->marks[self->num_marks - 1] == len)) + if (self->num_marks > 0 && self->marks[self->num_marks - 1] == len) { self->num_marks--; - else { + } else if (len >= 0) { len--; Py_DECREF(self->stack->data[len]); self->stack->length = len; + } else { + return stack_underflow(); } - return 0; } |
