Fix core dump in an endcase of b.strip() that I missed.

gvanrossum · gvanrossum · commit eb29e9ab2b03 · 2007-08-08T21:55:33.000Z
diff --git a/Lib/test/test_bytes.py b/Lib/test/test_bytes.py
@@ -653,6 +653,7 @@ def test_strip(self):
         self.assertEqual(b.strip(b'pi'), b'mississ')
         self.assertEqual(b.strip(b'im'), b'ssissipp')
         self.assertEqual(b.strip(b'pim'), b'ssiss')
+        self.assertEqual(b.strip(b), b'')
 
     def test_lstrip(self):
         b = b'mississippi'
diff --git a/Objects/bytesobject.c b/Objects/bytesobject.c
@@ -2502,7 +2502,10 @@ bytes_strip(PyBytesObject *self, PyObject *arg)
     argptr = ((PyBytesObject *)arg)->ob_bytes;
     argsize = Py_Size(arg);
     left = lstrip_helper(myptr, mysize, argptr, argsize);
-    right = rstrip_helper(myptr, mysize, argptr, argsize);
+    if (left == mysize)
+        right = left;
+    else
+        right = rstrip_helper(myptr, mysize, argptr, argsize);
     return PyBytes_FromStringAndSize(self->ob_bytes + left, right - left);
 }
 
