]> granicus.if.org Git - python/commitdiff
changed find/rfind to return -1 for matches outside the source string
authorFredrik Lundh <fredrik@pythonware.com>
Tue, 30 May 2006 17:11:48 +0000 (17:11 +0000)
committerFredrik Lundh <fredrik@pythonware.com>
Tue, 30 May 2006 17:11:48 +0000 (17:11 +0000)
Lib/test/string_tests.py
Objects/stringlib/find.h

index 5fe9fb92cfabe7602fdb9782722f19095bc0284d..c2e0875b768b446a64461cf35486f2c963f0f4f3 100644 (file)
@@ -154,6 +154,10 @@ class CommonTest(unittest.TestCase):
         self.checkequal(9, 'abcdefghiabc', 'find', 'abc', 1)
         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', 42)
 
@@ -188,6 +192,10 @@ class CommonTest(unittest.TestCase):
         self.checkequal(0, 'abcdefghiabc', 'rfind', 'abcd')
         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', 42)
 
index 4cea2db4b2379b9b4a0e6a40570fd7f5fa85adc2..4cdbb096da365465a60b86728699bd81b2ac2b97 100644 (file)
@@ -14,8 +14,11 @@ stringlib_find(const STRINGLIB_CHAR* str, Py_ssize_t str_len,
 {
     Py_ssize_t pos;
 
-    if (sub_len == 0)
+    if (sub_len == 0) {
+        if (str_len < 0)
+            return -1;
         return offset;
+    }
 
     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,
                 const STRINGLIB_CHAR* sub, Py_ssize_t sub_len,
                 Py_ssize_t offset)
 {
-    Py_ssize_t pos;
-
     /* XXX - create reversefastsearch helper! */
-    if (sub_len == 0)
-       pos = str_len + offset;
-    else {
-       Py_ssize_t j;
-        pos = -1;
+    if (sub_len == 0) {
+        if (str_len < 0)
+            return -1;
+       return str_len + offset;
+    } else {
+       Py_ssize_t j, pos = -1;
        for (j = str_len - sub_len; j >= 0; --j)
             if (STRINGLIB_CMP(str+j, sub, sub_len) == 0) {
                 pos = j + offset;
                 break;
             }
+        return pos;
     }
-
-    return pos;
 }
 
 Py_LOCAL_INLINE(Py_ssize_t)