]> granicus.if.org Git - python/commitdiff
have a clear error when passing something > sys.maxsize to bytearray
authorBenjamin Peterson <benjamin@python.org>
Fri, 16 Apr 2010 22:35:38 +0000 (22:35 +0000)
committerBenjamin Peterson <benjamin@python.org>
Fri, 16 Apr 2010 22:35:38 +0000 (22:35 +0000)
Lib/test/test_bytes.py
Misc/NEWS
Objects/bytearrayobject.c

index dfd6a1e918a65c0b94b69f8609d059a02e5b002b..516d6d9782a8125e991b53229718a39cdc141208 100644 (file)
@@ -73,6 +73,7 @@ class BaseBytesTest(unittest.TestCase):
 
         self.assertEqual(self.type2test('0', 'ascii'), b'0')
         self.assertEqual(self.type2test(b'0'), b'0')
+        self.assertRaises(OverflowError, self.type2test, sys.maxsize + 1)
 
     def test_constructor_type_errors(self):
         self.assertRaises(TypeError, self.type2test, 0.0)
index 3c49738e886997e8cc514d2f31579d4c57f41383..b53beab88828885ce81c0a07b26749c34450a63e 100644 (file)
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -12,6 +12,9 @@ What's New in Python 2.7 beta 2?
 Core and Builtins
 -----------------
 
+- Issue #8417: Raise an OverflowError when an integer larger than sys.maxsize is
+  passed to bytearray.
+
 Library
 -------
 
index 699dec9d97214f1d4c30b1dc489c6eef48d3af4e..61fe8debb92dbee6e1c996bd54d24d341e1ff52a 100644 (file)
@@ -824,14 +824,18 @@ bytearray_init(PyByteArrayObject *self, PyObject *args, PyObject *kwds)
     }
 
     /* Is it an int? */
-    count = PyNumber_AsSsize_t(arg, PyExc_ValueError);
-    if (count == -1 && PyErr_Occurred())
-        PyErr_Clear();
-    else {
-        if (count < 0) {
-            PyErr_SetString(PyExc_ValueError, "negative count");
+    count = PyNumber_AsSsize_t(arg, PyExc_OverflowError);
+    if (count == -1 && PyErr_Occurred()) {
+        if (PyErr_ExceptionMatches(PyExc_OverflowError))
             return -1;
-        }
+        else
+            PyErr_Clear();
+    }
+    else if (count < 0) {
+        PyErr_SetString(PyExc_ValueError, "negative count");
+        return -1;
+    }
+    else {
         if (count > 0) {
             if (PyByteArray_Resize((PyObject *)self, count))
                 return -1;