Add asserts to PyBytes_AS_STRING and PyBytes_GET_SIZE.

2d28f592 · Guido van Rossum · 27ff74d8 · 2d28f592
Commit 2d28f592 authored Jun 14, 2007 by Guido van Rossum
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Include/bytesobject.h Include/bytesobject.h +2 -2

No files found.
--- a/Include/bytesobject.h
+++ b/Include/bytesobject.h
@@ -41,8 +41,8 @@ PyAPI_FUNC(char *) PyBytes_AsString(PyObject *);
 PyAPI_FUNC(int) PyBytes_Resize(PyObject *, Py_ssize_t);

 /* Macros, trading safety for speed */
-#define PyBytes_AS_STRING(self) (((PyBytesObject *)(self))->ob_bytes)
-#define PyBytes_GET_SIZE(self)  (((PyBytesObject *)(self))->ob_size)
+#define PyBytes_AS_STRING(self) (assert(PyBytes_Check(self)),((PyBytesObject *)(self))->ob_bytes)
+#define PyBytes_GET_SIZE(self)  (assert(PyBytes_Check(self)),((PyBytesObject *)(self))->ob_size)

 #ifdef __cplusplus
 }