Commit 93eff6fe authored by Fredrik Lundh's avatar Fredrik Lundh

changed find/rfind to return -1 for matches outside the source string

parent 9faa3eda
...@@ -154,6 +154,10 @@ class CommonTest(unittest.TestCase): ...@@ -154,6 +154,10 @@ class CommonTest(unittest.TestCase):
self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1) self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4) self.checkequal(-1, 'abcdefghiabc', 'find', 'def', 4)
self.checkequal(0, 'abc', 'find', '', 0)
self.checkequal(3, 'abc', 'find', '', 3)
self.checkequal(-1, 'abc', 'find', '', 4)
self.checkraises(TypeError, 'hello', 'find') self.checkraises(TypeError, 'hello', 'find')
self.checkraises(TypeError, 'hello', 'find', 42) self.checkraises(TypeError, 'hello', 'find', 42)
...@@ -188,6 +192,10 @@ class CommonTest(unittest.TestCase): ...@@ -188,6 +192,10 @@ class CommonTest(unittest.TestCase):
self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd') self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz') self.checkequal(-1, 'abcdefghiabc', 'rfind', 'abcz')
self.checkequal(3, 'abc', 'rfind', '', 0)
self.checkequal(3, 'abc', 'rfind', '', 3)
self.checkequal(-1, 'abc', 'rfind', '', 4)
self.checkraises(TypeError, 'hello', 'rfind') self.checkraises(TypeError, 'hello', 'rfind')
self.checkraises(TypeError, 'hello', 'rfind', 42) self.checkraises(TypeError, 'hello', 'rfind', 42)
......
...@@ -14,8 +14,11 @@ stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len, ...@@ -14,8 +14,11 @@ stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
{ {
Py_ssize_t pos; Py_ssize_t pos;
if (sub_len == 0) if (sub_len == 0) {
if (str_len < 0)
return -1;
return offset; return offset;
}
pos = fastsearch(str, str_len, sub, sub_len, FAST_SEARCH); pos = fastsearch(str, str_len, sub, sub_len, FAST_SEARCH);
...@@ -30,22 +33,20 @@ stringlib_rfind(const STRINGLIB_CHAR* str, Py_ssize_t str_len, ...@@ -30,22 +33,20 @@ stringlib_rfind(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
const STRINGLIB_CHAR* sub, Py_ssize_t sub_len, const STRINGLIB_CHAR* sub, Py_ssize_t sub_len,
Py_ssize_t offset) Py_ssize_t offset)
{ {
Py_ssize_t pos;
/* XXX - create reversefastsearch helper! */ /* XXX - create reversefastsearch helper! */
if (sub_len == 0) if (sub_len == 0) {
pos = str_len + offset; if (str_len < 0)
else { return -1;
Py_ssize_t j; return str_len + offset;
pos = -1; } else {
Py_ssize_t j, pos = -1;
for (j = str_len - sub_len; j >= 0; --j) for (j = str_len - sub_len; j >= 0; --j)
if (STRINGLIB_CMP(str+j, sub, sub_len) == 0) { if (STRINGLIB_CMP(str+j, sub, sub_len) == 0) {
pos = j + offset; pos = j + offset;
break; break;
} }
return pos;
} }
return pos;
} }
Py_LOCAL_INLINE(Py_ssize_t) Py_LOCAL_INLINE(Py_ssize_t)
......
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