Update str_ops.c

BobTheBuidler · web-flow · commit badf077d0e98 · 2025-06-10T03:14:58.000-04:00
diff --git a/mypyc/lib-rt/str_ops.c b/mypyc/lib-rt/str_ops.c
@@ -511,7 +511,12 @@ PyObject *CPy_Encode(PyObject *obj, PyObject *encoding, PyObject *errors) {
     }
 }
 
-Py_ssize_t CPyStr_Count(PyObject *unicode, PyObject *substring, Py_ssize_t start) {
+Py_ssize_t CPyStr_Count(PyObject *unicode, PyObject *substring, CPyTagged start) {
+    Py_ssize_t temp_start = CPyTagged_AsSsize_t(start);
+    if (temp_start == -1 && PyErr_Occurred()) {
+        PyErr_SetString(PyExc_OverflowError, CPYTHON_LARGE_INT_ERRMSG);
+        return NULL;
+    }
     Py_ssize_t end = PyUnicode_GET_LENGTH(unicode);
     return PyUnicode_Count(unicode, substring, start, end);
 }

Original file line number	Diff line number	Diff line change
`@@ -511,7 +511,12 @@ PyObject CPy_Encode(PyObject obj, PyObject encoding, PyObject errors) {`
`511`	`511`	`}`
`512`	`512`	`}`
`513`	`513`
`514`		`-Py_ssize_t CPyStr_Count(PyObject unicode, PyObject substring, Py_ssize_t start) {`
	`514`	`+Py_ssize_t CPyStr_Count(PyObject unicode, PyObject substring, CPyTagged start) {`
	`515`	`+ Py_ssize_t temp_start = CPyTagged_AsSsize_t(start);`
	`516`	`+ if (temp_start == -1 && PyErr_Occurred()) {`
	`517`	`+ PyErr_SetString(PyExc_OverflowError, CPYTHON_LARGE_INT_ERRMSG);`
	`518`	`+ return NULL;`
	`519`	`+ }`
`515`	`520`	`Py_ssize_t end = PyUnicode_GET_LENGTH(unicode);`
`516`	`521`	`return PyUnicode_Count(unicode, substring, start, end);`
`517`	`522`	`}`