[Python-checkins] cpython (3.4): Raise more correct exception on overflow in setting buffer_size attribute of
serhiy.storchaka
python-checkins at python.org
Mon Sep 7 21:56:30 CEST 2015
https://hg.python.org/cpython/rev/ef9346be45d0
changeset: 97758:ef9346be45d0
branch: 3.4
parent: 97754:ff2c4f281720
user: Serhiy Storchaka <storchaka at gmail.com>
date: Mon Sep 07 22:51:56 2015 +0300
summary:
Raise more correct exception on overflow in setting buffer_size attribute of
expat parser.
files:
Lib/test/test_pyexpat.py | 3 +++
Modules/pyexpat.c | 13 +++++++------
2 files changed, 10 insertions(+), 6 deletions(-)
diff --git a/Lib/test/test_pyexpat.py b/Lib/test/test_pyexpat.py
--- a/Lib/test/test_pyexpat.py
+++ b/Lib/test/test_pyexpat.py
@@ -3,6 +3,7 @@
from io import BytesIO
import os
+import sys
import sysconfig
import unittest
import traceback
@@ -543,6 +544,8 @@
parser.buffer_size = -1
with self.assertRaises(ValueError):
parser.buffer_size = 0
+ with self.assertRaises((ValueError, OverflowError)):
+ parser.buffer_size = sys.maxsize + 1
with self.assertRaises(TypeError):
parser.buffer_size = 512.0
diff --git a/Modules/pyexpat.c b/Modules/pyexpat.c
--- a/Modules/pyexpat.c
+++ b/Modules/pyexpat.c
@@ -1403,17 +1403,18 @@
return -1;
}
- new_buffer_size=PyLong_AS_LONG(v);
+ new_buffer_size = PyLong_AsLong(v);
+ if (new_buffer_size <= 0) {
+ if (!PyErr_Occurred())
+ PyErr_SetString(PyExc_ValueError, "buffer_size must be greater than zero");
+ return -1;
+ }
+
/* trivial case -- no change */
if (new_buffer_size == self->buffer_size) {
return 0;
}
- if (new_buffer_size <= 0) {
- PyErr_SetString(PyExc_ValueError, "buffer_size must be greater than zero");
- return -1;
- }
-
/* check maximum */
if (new_buffer_size > INT_MAX) {
char errmsg[100];
--
Repository URL: https://hg.python.org/cpython
More information about the Python-checkins
mailing list