Use Py_ssize_t for offset

Author: sergey-miryanov

Include/internal/pycore_code.h

@@ -540,7 +540,7 @@ typedef struct {
 PyAPI_FUNC(int) _Py_Instrument(PyCodeObject *co, PyInterpreterState *interp);
 
 // Export for '_testinternalcapi' shared extension
-PyAPI_FUNC(_Py_CODEUNIT) _Py_GetBaseCodeUnit(PyCodeObject *code, int offset);
+PyAPI_FUNC(_Py_CODEUNIT) _Py_GetBaseCodeUnit(PyCodeObject *code, Py_ssize_t offset);
 
 extern int _PyInstruction_GetLength(PyCodeObject *code, int offset);
 

Include/internal/pycore_instruments.h

@@ -122,7 +122,7 @@ extern int
 _Py_Instrumentation_GetLine(PyCodeObject *code, _PyCoLineInstrumentationData *line_data, int index);
 
 static inline uint8_t
-_PyCode_GetOriginalOpcode(_PyCoLineInstrumentationData *line_data, int index)
+_PyCode_GetOriginalOpcode(_PyCoLineInstrumentationData *line_data, Py_ssize_t index)
 {
     return line_data->data[index*line_data->bytes_per_entry];
 }

Modules/_testinternalcapi/test_cases.c.h

@@ -1867,7 +1867,8 @@ dummy_func(
             assert(INLINE_CACHE_ENTRIES_SEND == INLINE_CACHE_ENTRIES_FOR_ITER);
             #if TIER_ONE && defined(Py_DEBUG)
             if (!PyStackRef_IsNone(frame->f_executable)) {
-                int i = (int)(frame->instr_ptr - _PyFrame_GetBytecode(frame));
+                Py_ssize_t i = frame->instr_ptr - _PyFrame_GetBytecode(frame);
+                assert(i > 0);
                 int opcode = _Py_GetBaseCodeUnit(_PyFrame_GetCode(frame), i).op.code;
                 assert(opcode == SEND || opcode == FOR_ITER);
             }

Python/bytecodes.c

@@ -635,7 +635,7 @@ sanity_check_instrumentation(PyCodeObject *code)
 
 /* Get the underlying code unit, stripping instrumentation and ENTER_EXECUTOR */
 _Py_CODEUNIT
-_Py_GetBaseCodeUnit(PyCodeObject *code, int i)
+_Py_GetBaseCodeUnit(PyCodeObject *code, Py_ssize_t i)
 {
     _Py_CODEUNIT *src_instr = _PyCode_CODE(code) + i;
     _Py_CODEUNIT inst = {