Commit b50f9926 authored by Amaury Forgeot d'Arc's avatar Amaury Forgeot d'Arc

#3967: Correct a crash in count() and find() methods of string-like objects.

For example:
   "".count("xxxx", sys.maxint, 0)

Backport of r66631.
parent 4235e6f1
...@@ -120,6 +120,14 @@ class CommonTest(unittest.TestCase): ...@@ -120,6 +120,14 @@ class CommonTest(unittest.TestCase):
self.checkequal(2, 'aaa', 'count', '', -1) self.checkequal(2, 'aaa', 'count', '', -1)
self.checkequal(4, 'aaa', 'count', '', -10) self.checkequal(4, 'aaa', 'count', '', -10)
self.checkequal(1, '', 'count', '')
self.checkequal(0, '', 'count', '', 1, 1)
self.checkequal(0, '', 'count', '', sys.maxint, 0)
self.checkequal(0, '', 'count', 'xx')
self.checkequal(0, '', 'count', 'xx', 1, 1)
self.checkequal(0, '', 'count', 'xx', sys.maxint, 0)
self.checkraises(TypeError, 'hello', 'count') self.checkraises(TypeError, 'hello', 'count')
self.checkraises(TypeError, 'hello', 'count', 42) self.checkraises(TypeError, 'hello', 'count', 42)
...@@ -162,6 +170,14 @@ class CommonTest(unittest.TestCase): ...@@ -162,6 +170,14 @@ class CommonTest(unittest.TestCase):
self.checkraises(TypeError, 'hello', 'find') self.checkraises(TypeError, 'hello', 'find')
self.checkraises(TypeError, 'hello', 'find', 42) self.checkraises(TypeError, 'hello', 'find', 42)
self.checkequal(0, '', 'find', '')
self.checkequal(-1, '', 'find', '', 1, 1)
self.checkequal(-1, '', 'find', '', sys.maxint, 0)
self.checkequal(-1, '', 'find', 'xx')
self.checkequal(-1, '', 'find', 'xx', 1, 1)
self.checkequal(-1, '', 'find', 'xx', sys.maxint, 0)
# For a variety of combinations, # For a variety of combinations,
# verify that str.find() matches __contains__ # verify that str.find() matches __contains__
# and that the found substring is really at that location # and that the found substring is really at that location
......
...@@ -12,6 +12,9 @@ What's New in Python 2.5.3? ...@@ -12,6 +12,9 @@ What's New in Python 2.5.3?
Core and builtins Core and builtins
----------------- -----------------
- Issue #3967: Fixed a crash in the count() and find() methods of string-like
objects, when the "start" parameter is a huge value.
- Issue #3936: The parser warnings for using "as" and "with" as variable names - Issue #3936: The parser warnings for using "as" and "with" as variable names
didn't fire after import statements. didn't fire after import statements.
......
...@@ -13,11 +13,10 @@ stringlib_count(const STRINGLIB_CHAR* str, Py_ssize_t str_len, ...@@ -13,11 +13,10 @@ stringlib_count(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
{ {
Py_ssize_t count; Py_ssize_t count;
if (sub_len == 0) { if (str_len < 0)
if (str_len < 0) return 0; /* start > len(str) */
return 0; /* start > len(str) */ if (sub_len == 0)
return str_len + 1; return str_len + 1;
}
count = fastsearch(str, str_len, sub, sub_len, FAST_COUNT); count = fastsearch(str, str_len, sub, sub_len, FAST_COUNT);
......
...@@ -14,11 +14,10 @@ stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len, ...@@ -14,11 +14,10 @@ stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
{ {
Py_ssize_t pos; Py_ssize_t pos;
if (sub_len == 0) { if (str_len < 0)
if (str_len < 0) return -1;
return -1; if (sub_len == 0)
return offset; return offset;
}
pos = fastsearch(str, str_len, sub, sub_len, FAST_SEARCH); pos = fastsearch(str, str_len, sub, sub_len, FAST_SEARCH);
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment