blob: b6fc478595b523459ab0375b83e9c5bfb2867cf5 [file] [log] [blame]
Guido van Rossume15dee51995-07-18 14:12:02 +00001/* Abstract Object Interface (many thanks to Jim Fulton) */
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002
Guido van Rossume15dee51995-07-18 14:12:02 +00003#include "Python.h"
Guido van Rossumfa0b6ab1998-05-22 15:23:36 +00004#include <ctype.h>
Neil Schemenauer5a1f0152001-01-04 01:39:06 +00005#include "structmember.h" /* we need the offsetof() macro from there */
Tim Peters64b5ce32001-09-10 20:52:51 +00006#include "longintrepr.h"
Neil Schemenauer5a1f0152001-01-04 01:39:06 +00007
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00009
Guido van Rossumcea1c8c1998-05-22 00:47:05 +000010/* Shorthands to return certain errors */
Guido van Rossume15dee51995-07-18 14:12:02 +000011
12static PyObject *
Thomas Wouters0e3f5912006-08-11 14:57:12 +000013type_error(const char *msg, PyObject *obj)
Guido van Rossume15dee51995-07-18 14:12:02 +000014{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000015 PyErr_Format(PyExc_TypeError, msg, obj->ob_type->tp_name);
16 return NULL;
Guido van Rossume15dee51995-07-18 14:12:02 +000017}
18
Guido van Rossum052b7e11996-11-11 15:08:19 +000019static PyObject *
Fred Drake79912472000-07-09 04:06:11 +000020null_error(void)
Guido van Rossume15dee51995-07-18 14:12:02 +000021{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000022 if (!PyErr_Occurred())
23 PyErr_SetString(PyExc_SystemError,
24 "null argument to internal routine");
25 return NULL;
Guido van Rossume15dee51995-07-18 14:12:02 +000026}
27
Guido van Rossumcea1c8c1998-05-22 00:47:05 +000028/* Operations on any object */
29
Guido van Rossume15dee51995-07-18 14:12:02 +000030PyObject *
Fred Drake79912472000-07-09 04:06:11 +000031PyObject_Type(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +000032{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000033 PyObject *v;
Guido van Rossume15dee51995-07-18 14:12:02 +000034
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000035 if (o == NULL)
36 return null_error();
37 v = (PyObject *)o->ob_type;
38 Py_INCREF(v);
39 return v;
Guido van Rossume15dee51995-07-18 14:12:02 +000040}
41
Martin v. Löwis18e16552006-02-15 17:27:45 +000042Py_ssize_t
Jeremy Hylton6253f832000-07-12 12:56:19 +000043PyObject_Size(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +000044{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000045 PySequenceMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +000046
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000047 if (o == NULL) {
48 null_error();
49 return -1;
50 }
Guido van Rossume15dee51995-07-18 14:12:02 +000051
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000052 m = o->ob_type->tp_as_sequence;
53 if (m && m->sq_length)
54 return m->sq_length(o);
Guido van Rossume15dee51995-07-18 14:12:02 +000055
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000056 return PyMapping_Size(o);
Guido van Rossume15dee51995-07-18 14:12:02 +000057}
58
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +000059#undef PyObject_Length
Martin v. Löwis18e16552006-02-15 17:27:45 +000060Py_ssize_t
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +000061PyObject_Length(PyObject *o)
62{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000063 return PyObject_Size(o);
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +000064}
65#define PyObject_Length PyObject_Size
66
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020067int
68_PyObject_HasLen(PyObject *o) {
69 return (Py_TYPE(o)->tp_as_sequence && Py_TYPE(o)->tp_as_sequence->sq_length) ||
70 (Py_TYPE(o)->tp_as_mapping && Py_TYPE(o)->tp_as_mapping->mp_length);
71}
Raymond Hettinger6b27cda2005-09-24 21:23:05 +000072
Christian Heimes255f53b2007-12-08 15:33:56 +000073/* The length hint function returns a non-negative value from o.__len__()
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020074 or o.__length_hint__(). If those methods aren't found. If one of the calls
75 fails this function returns -1.
Christian Heimes255f53b2007-12-08 15:33:56 +000076*/
77
78Py_ssize_t
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020079PyObject_LengthHint(PyObject *o, Py_ssize_t defaultvalue)
Christian Heimes255f53b2007-12-08 15:33:56 +000080{
Benjamin Petersonce798522012-01-22 11:24:29 -050081 _Py_IDENTIFIER(__length_hint__);
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020082 Py_ssize_t res = PyObject_Length(o);
83 if (res < 0 && PyErr_Occurred()) {
84 if (!PyErr_ExceptionMatches(PyExc_TypeError)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000085 return -1;
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020086 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000087 PyErr_Clear();
88 }
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020089 else {
90 return res;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000091 }
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020092 PyObject *hint = _PyObject_LookupSpecial(o, &PyId___length_hint__);
93 if (hint == NULL) {
94 if (PyErr_Occurred()) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000095 return -1;
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020096 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000097 return defaultvalue;
98 }
Armin Ronacheraa9a79d2012-10-06 14:03:24 +020099 PyObject *result = PyObject_CallFunctionObjArgs(hint, NULL);
100 Py_DECREF(hint);
101 if (result == NULL) {
102 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
103 PyErr_Clear();
104 return defaultvalue;
105 }
106 return -1;
107 }
108 else if (result == Py_NotImplemented) {
109 Py_DECREF(result);
110 return defaultvalue;
111 }
112 if (!PyLong_Check(result)) {
113 PyErr_Format(PyExc_TypeError, "Length hint must be an integer, not %s",
114 Py_TYPE(result)->tp_name);
115 Py_DECREF(result);
116 return -1;
117 }
118 defaultvalue = PyLong_AsSsize_t(result);
119 Py_DECREF(result);
120 if (defaultvalue < 0 && PyErr_Occurred()) {
121 return -1;
122 }
123 if (defaultvalue < 0) {
124 PyErr_Format(PyExc_ValueError, "__length_hint__() should return >= 0");
125 return -1;
126 }
127 return defaultvalue;
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000128}
129
Guido van Rossume15dee51995-07-18 14:12:02 +0000130PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000131PyObject_GetItem(PyObject *o, PyObject *key)
Guido van Rossume15dee51995-07-18 14:12:02 +0000132{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000133 PyMappingMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +0000134
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000135 if (o == NULL || key == NULL)
136 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +0000137
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000138 m = o->ob_type->tp_as_mapping;
139 if (m && m->mp_subscript)
140 return m->mp_subscript(o, key);
Guido van Rossume15dee51995-07-18 14:12:02 +0000141
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000142 if (o->ob_type->tp_as_sequence) {
143 if (PyIndex_Check(key)) {
144 Py_ssize_t key_value;
145 key_value = PyNumber_AsSsize_t(key, PyExc_IndexError);
146 if (key_value == -1 && PyErr_Occurred())
147 return NULL;
148 return PySequence_GetItem(o, key_value);
149 }
150 else if (o->ob_type->tp_as_sequence->sq_item)
151 return type_error("sequence index must "
152 "be integer, not '%.200s'", key);
153 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +0000154
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000155 return type_error("'%.200s' object is not subscriptable", o);
Guido van Rossume15dee51995-07-18 14:12:02 +0000156}
157
158int
Fred Drake79912472000-07-09 04:06:11 +0000159PyObject_SetItem(PyObject *o, PyObject *key, PyObject *value)
Guido van Rossume15dee51995-07-18 14:12:02 +0000160{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000161 PyMappingMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +0000162
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000163 if (o == NULL || key == NULL || value == NULL) {
164 null_error();
165 return -1;
166 }
167 m = o->ob_type->tp_as_mapping;
168 if (m && m->mp_ass_subscript)
169 return m->mp_ass_subscript(o, key, value);
Guido van Rossume15dee51995-07-18 14:12:02 +0000170
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000171 if (o->ob_type->tp_as_sequence) {
172 if (PyIndex_Check(key)) {
173 Py_ssize_t key_value;
174 key_value = PyNumber_AsSsize_t(key, PyExc_IndexError);
175 if (key_value == -1 && PyErr_Occurred())
176 return -1;
177 return PySequence_SetItem(o, key_value, value);
178 }
179 else if (o->ob_type->tp_as_sequence->sq_ass_item) {
180 type_error("sequence index must be "
181 "integer, not '%.200s'", key);
182 return -1;
183 }
184 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +0000185
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000186 type_error("'%.200s' object does not support item assignment", o);
187 return -1;
Guido van Rossume15dee51995-07-18 14:12:02 +0000188}
189
Guido van Rossum6cdc6f41996-08-21 17:41:54 +0000190int
Fred Drake79912472000-07-09 04:06:11 +0000191PyObject_DelItem(PyObject *o, PyObject *key)
Guido van Rossum6cdc6f41996-08-21 17:41:54 +0000192{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000193 PyMappingMethods *m;
Guido van Rossum6cdc6f41996-08-21 17:41:54 +0000194
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000195 if (o == NULL || key == NULL) {
196 null_error();
197 return -1;
198 }
199 m = o->ob_type->tp_as_mapping;
200 if (m && m->mp_ass_subscript)
201 return m->mp_ass_subscript(o, key, (PyObject*)NULL);
Guido van Rossum6cdc6f41996-08-21 17:41:54 +0000202
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000203 if (o->ob_type->tp_as_sequence) {
204 if (PyIndex_Check(key)) {
205 Py_ssize_t key_value;
206 key_value = PyNumber_AsSsize_t(key, PyExc_IndexError);
207 if (key_value == -1 && PyErr_Occurred())
208 return -1;
209 return PySequence_DelItem(o, key_value);
210 }
211 else if (o->ob_type->tp_as_sequence->sq_ass_item) {
212 type_error("sequence index must be "
213 "integer, not '%.200s'", key);
214 return -1;
215 }
216 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +0000217
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000218 type_error("'%.200s' object does not support item deletion", o);
219 return -1;
Guido van Rossum6cdc6f41996-08-21 17:41:54 +0000220}
221
Martin v. Löwisb0d71d02002-01-05 10:50:30 +0000222int
223PyObject_DelItemString(PyObject *o, char *key)
224{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000225 PyObject *okey;
226 int ret;
Martin v. Löwisb0d71d02002-01-05 10:50:30 +0000227
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000228 if (o == NULL || key == NULL) {
229 null_error();
230 return -1;
231 }
232 okey = PyUnicode_FromString(key);
233 if (okey == NULL)
234 return -1;
235 ret = PyObject_DelItem(o, okey);
236 Py_DECREF(okey);
237 return ret;
Martin v. Löwisb0d71d02002-01-05 10:50:30 +0000238}
239
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000240/* We release the buffer right after use of this function which could
Guido van Rossum98297ee2007-11-06 21:34:58 +0000241 cause issues later on. Don't use these functions in new code.
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000242 */
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000243int
244PyObject_AsCharBuffer(PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000245 const char **buffer,
246 Py_ssize_t *buffer_len)
Guido van Rossum4c08d552000-03-10 22:55:18 +0000247{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000248 PyBufferProcs *pb;
249 Py_buffer view;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000250
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000251 if (obj == NULL || buffer == NULL || buffer_len == NULL) {
252 null_error();
253 return -1;
254 }
255 pb = obj->ob_type->tp_as_buffer;
256 if (pb == NULL || pb->bf_getbuffer == NULL) {
257 PyErr_SetString(PyExc_TypeError,
Victor Stinnerf2a94212011-05-30 23:21:30 +0200258 "expected bytes, bytearray "
259 "or buffer compatible object");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000260 return -1;
261 }
262 if ((*pb->bf_getbuffer)(obj, &view, PyBUF_SIMPLE)) return -1;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000263
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000264 *buffer = view.buf;
265 *buffer_len = view.len;
266 if (pb->bf_releasebuffer != NULL)
267 (*pb->bf_releasebuffer)(obj, &view);
268 Py_XDECREF(view.obj);
269 return 0;
Jeremy Hylton89c3a222001-11-09 21:59:42 +0000270}
Guido van Rossum4c08d552000-03-10 22:55:18 +0000271
Jeremy Hylton89c3a222001-11-09 21:59:42 +0000272int
273PyObject_CheckReadBuffer(PyObject *obj)
274{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000275 PyBufferProcs *pb = obj->ob_type->tp_as_buffer;
276 Py_buffer view;
Jeremy Hylton89c3a222001-11-09 21:59:42 +0000277
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000278 if (pb == NULL ||
279 pb->bf_getbuffer == NULL)
280 return 0;
281 if ((*pb->bf_getbuffer)(obj, &view, PyBUF_SIMPLE) == -1) {
282 PyErr_Clear();
283 return 0;
284 }
285 PyBuffer_Release(&view);
286 return 1;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000287}
288
289int PyObject_AsReadBuffer(PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000290 const void **buffer,
291 Py_ssize_t *buffer_len)
Guido van Rossum4c08d552000-03-10 22:55:18 +0000292{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000293 PyBufferProcs *pb;
294 Py_buffer view;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000295
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000296 if (obj == NULL || buffer == NULL || buffer_len == NULL) {
297 null_error();
298 return -1;
299 }
300 pb = obj->ob_type->tp_as_buffer;
301 if (pb == NULL ||
302 pb->bf_getbuffer == NULL) {
303 PyErr_SetString(PyExc_TypeError,
304 "expected an object with a buffer interface");
305 return -1;
306 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000307
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000308 if ((*pb->bf_getbuffer)(obj, &view, PyBUF_SIMPLE)) return -1;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000309
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000310 *buffer = view.buf;
311 *buffer_len = view.len;
312 if (pb->bf_releasebuffer != NULL)
313 (*pb->bf_releasebuffer)(obj, &view);
314 Py_XDECREF(view.obj);
315 return 0;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000316}
317
318int PyObject_AsWriteBuffer(PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000319 void **buffer,
320 Py_ssize_t *buffer_len)
Guido van Rossum4c08d552000-03-10 22:55:18 +0000321{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000322 PyBufferProcs *pb;
323 Py_buffer view;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000324
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000325 if (obj == NULL || buffer == NULL || buffer_len == NULL) {
326 null_error();
327 return -1;
328 }
329 pb = obj->ob_type->tp_as_buffer;
330 if (pb == NULL ||
331 pb->bf_getbuffer == NULL ||
332 ((*pb->bf_getbuffer)(obj, &view, PyBUF_WRITABLE) != 0)) {
333 PyErr_SetString(PyExc_TypeError,
334 "expected an object with a writable buffer interface");
335 return -1;
336 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000337
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000338 *buffer = view.buf;
339 *buffer_len = view.len;
340 if (pb->bf_releasebuffer != NULL)
341 (*pb->bf_releasebuffer)(obj, &view);
342 Py_XDECREF(view.obj);
343 return 0;
Guido van Rossum4c08d552000-03-10 22:55:18 +0000344}
345
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000346/* Buffer C-API for Python 3.0 */
347
348int
Travis E. Oliphant8ae62b62007-09-23 02:00:13 +0000349PyObject_GetBuffer(PyObject *obj, Py_buffer *view, int flags)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000350{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000351 if (!PyObject_CheckBuffer(obj)) {
352 PyErr_Format(PyExc_TypeError,
Victor Stinner0a5f65a2011-03-22 01:09:21 +0100353 "'%.100s' does not support the buffer interface",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000354 Py_TYPE(obj)->tp_name);
355 return -1;
356 }
357 return (*(obj->ob_type->tp_as_buffer->bf_getbuffer))(obj, view, flags);
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000358}
359
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000360static int
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100361_IsFortranContiguous(const Py_buffer *view)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000362{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000363 Py_ssize_t sd, dim;
364 int i;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000366 if (view->ndim == 0) return 1;
367 if (view->strides == NULL) return (view->ndim == 1);
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000368
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000369 sd = view->itemsize;
370 if (view->ndim == 1) return (view->shape[0] == 1 ||
371 sd == view->strides[0]);
372 for (i=0; i<view->ndim; i++) {
373 dim = view->shape[i];
374 if (dim == 0) return 1;
375 if (view->strides[i] != sd) return 0;
376 sd *= dim;
377 }
378 return 1;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000379}
380
381static int
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100382_IsCContiguous(const Py_buffer *view)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000383{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000384 Py_ssize_t sd, dim;
385 int i;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000386
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000387 if (view->ndim == 0) return 1;
388 if (view->strides == NULL) return 1;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000390 sd = view->itemsize;
391 if (view->ndim == 1) return (view->shape[0] == 1 ||
392 sd == view->strides[0]);
393 for (i=view->ndim-1; i>=0; i--) {
394 dim = view->shape[i];
395 if (dim == 0) return 1;
396 if (view->strides[i] != sd) return 0;
397 sd *= dim;
398 }
399 return 1;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000400}
401
402int
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100403PyBuffer_IsContiguous(const Py_buffer *view, char order)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000404{
405
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000406 if (view->suboffsets != NULL) return 0;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000407
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100408 if (order == 'C')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000409 return _IsCContiguous(view);
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100410 else if (order == 'F')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000411 return _IsFortranContiguous(view);
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100412 else if (order == 'A')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000413 return (_IsCContiguous(view) || _IsFortranContiguous(view));
414 return 0;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000415}
416
417
Guido van Rossum98297ee2007-11-06 21:34:58 +0000418void*
Travis E. Oliphant8ae62b62007-09-23 02:00:13 +0000419PyBuffer_GetPointer(Py_buffer *view, Py_ssize_t *indices)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000420{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000421 char* pointer;
422 int i;
423 pointer = (char *)view->buf;
424 for (i = 0; i < view->ndim; i++) {
425 pointer += view->strides[i]*indices[i];
426 if ((view->suboffsets != NULL) && (view->suboffsets[i] >= 0)) {
427 pointer = *((char**)pointer) + view->suboffsets[i];
428 }
429 }
430 return (void*)pointer;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000431}
432
433
Guido van Rossum98297ee2007-11-06 21:34:58 +0000434void
Antoine Pitrouf68c2a72010-09-01 12:58:21 +0000435_Py_add_one_to_index_F(int nd, Py_ssize_t *index, const Py_ssize_t *shape)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000436{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000437 int k;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000438
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000439 for (k=0; k<nd; k++) {
440 if (index[k] < shape[k]-1) {
441 index[k]++;
442 break;
443 }
444 else {
445 index[k] = 0;
446 }
447 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000448}
449
Guido van Rossum98297ee2007-11-06 21:34:58 +0000450void
Antoine Pitrouf68c2a72010-09-01 12:58:21 +0000451_Py_add_one_to_index_C(int nd, Py_ssize_t *index, const Py_ssize_t *shape)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000452{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000453 int k;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000454
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000455 for (k=nd-1; k>=0; k--) {
456 if (index[k] < shape[k]-1) {
457 index[k]++;
458 break;
459 }
460 else {
461 index[k] = 0;
462 }
463 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000464}
465
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000466int
Travis E. Oliphant8ae62b62007-09-23 02:00:13 +0000467PyBuffer_FromContiguous(Py_buffer *view, void *buf, Py_ssize_t len, char fort)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000468{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000469 int k;
Antoine Pitrouf68c2a72010-09-01 12:58:21 +0000470 void (*addone)(int, Py_ssize_t *, const Py_ssize_t *);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000471 Py_ssize_t *indices, elements;
472 char *src, *ptr;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000474 if (len > view->len) {
475 len = view->len;
476 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000478 if (PyBuffer_IsContiguous(view, fort)) {
479 /* simplest copy is all that is needed */
480 memcpy(view->buf, buf, len);
481 return 0;
482 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000483
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000484 /* Otherwise a more elaborate scheme is needed */
Guido van Rossum98297ee2007-11-06 21:34:58 +0000485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000486 /* XXX(nnorwitz): need to check for overflow! */
487 indices = (Py_ssize_t *)PyMem_Malloc(sizeof(Py_ssize_t)*(view->ndim));
488 if (indices == NULL) {
489 PyErr_NoMemory();
490 return -1;
491 }
492 for (k=0; k<view->ndim;k++) {
493 indices[k] = 0;
494 }
Guido van Rossum98297ee2007-11-06 21:34:58 +0000495
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000496 if (fort == 'F') {
Antoine Pitrouf68c2a72010-09-01 12:58:21 +0000497 addone = _Py_add_one_to_index_F;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000498 }
499 else {
Antoine Pitrouf68c2a72010-09-01 12:58:21 +0000500 addone = _Py_add_one_to_index_C;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000501 }
502 src = buf;
503 /* XXX : This is not going to be the fastest code in the world
504 several optimizations are possible.
505 */
506 elements = len / view->itemsize;
507 while (elements--) {
508 addone(view->ndim, indices, view->shape);
509 ptr = PyBuffer_GetPointer(view, indices);
510 memcpy(ptr, src, view->itemsize);
511 src += view->itemsize;
512 }
Guido van Rossum98297ee2007-11-06 21:34:58 +0000513
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000514 PyMem_Free(indices);
515 return 0;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000516}
517
Guido van Rossum98297ee2007-11-06 21:34:58 +0000518int PyObject_CopyData(PyObject *dest, PyObject *src)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000519{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000520 Py_buffer view_dest, view_src;
521 int k;
522 Py_ssize_t *indices, elements;
523 char *dptr, *sptr;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000524
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000525 if (!PyObject_CheckBuffer(dest) ||
526 !PyObject_CheckBuffer(src)) {
527 PyErr_SetString(PyExc_TypeError,
528 "both destination and source must have the "\
529 "buffer interface");
530 return -1;
531 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000532
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000533 if (PyObject_GetBuffer(dest, &view_dest, PyBUF_FULL) != 0) return -1;
534 if (PyObject_GetBuffer(src, &view_src, PyBUF_FULL_RO) != 0) {
535 PyBuffer_Release(&view_dest);
536 return -1;
537 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000539 if (view_dest.len < view_src.len) {
540 PyErr_SetString(PyExc_BufferError,
541 "destination is too small to receive data from source");
542 PyBuffer_Release(&view_dest);
543 PyBuffer_Release(&view_src);
544 return -1;
545 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000547 if ((PyBuffer_IsContiguous(&view_dest, 'C') &&
548 PyBuffer_IsContiguous(&view_src, 'C')) ||
549 (PyBuffer_IsContiguous(&view_dest, 'F') &&
550 PyBuffer_IsContiguous(&view_src, 'F'))) {
551 /* simplest copy is all that is needed */
552 memcpy(view_dest.buf, view_src.buf, view_src.len);
553 PyBuffer_Release(&view_dest);
554 PyBuffer_Release(&view_src);
555 return 0;
556 }
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000557
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000558 /* Otherwise a more elaborate copy scheme is needed */
Guido van Rossum98297ee2007-11-06 21:34:58 +0000559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000560 /* XXX(nnorwitz): need to check for overflow! */
561 indices = (Py_ssize_t *)PyMem_Malloc(sizeof(Py_ssize_t)*view_src.ndim);
562 if (indices == NULL) {
563 PyErr_NoMemory();
564 PyBuffer_Release(&view_dest);
565 PyBuffer_Release(&view_src);
566 return -1;
567 }
568 for (k=0; k<view_src.ndim;k++) {
569 indices[k] = 0;
570 }
571 elements = 1;
572 for (k=0; k<view_src.ndim; k++) {
573 /* XXX(nnorwitz): can this overflow? */
574 elements *= view_src.shape[k];
575 }
576 while (elements--) {
Antoine Pitrouf68c2a72010-09-01 12:58:21 +0000577 _Py_add_one_to_index_C(view_src.ndim, indices, view_src.shape);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000578 dptr = PyBuffer_GetPointer(&view_dest, indices);
579 sptr = PyBuffer_GetPointer(&view_src, indices);
580 memcpy(dptr, sptr, view_src.itemsize);
581 }
582 PyMem_Free(indices);
583 PyBuffer_Release(&view_dest);
584 PyBuffer_Release(&view_src);
585 return 0;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000586}
587
588void
589PyBuffer_FillContiguousStrides(int nd, Py_ssize_t *shape,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000590 Py_ssize_t *strides, int itemsize,
591 char fort)
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000592{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000593 int k;
594 Py_ssize_t sd;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000595
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000596 sd = itemsize;
597 if (fort == 'F') {
598 for (k=0; k<nd; k++) {
599 strides[k] = sd;
600 sd *= shape[k];
601 }
602 }
603 else {
604 for (k=nd-1; k>=0; k--) {
605 strides[k] = sd;
606 sd *= shape[k];
607 }
608 }
609 return;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000610}
611
612int
Martin v. Löwis423be952008-08-13 15:53:07 +0000613PyBuffer_FillInfo(Py_buffer *view, PyObject *obj, void *buf, Py_ssize_t len,
Stefan Krah4e141742012-03-06 15:27:31 +0100614 int readonly, int flags)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000615{
Stefan Krah9a2d99e2012-02-25 12:24:21 +0100616 if (view == NULL) return 0; /* XXX why not -1? */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000617 if (((flags & PyBUF_WRITABLE) == PyBUF_WRITABLE) &&
618 (readonly == 1)) {
619 PyErr_SetString(PyExc_BufferError,
620 "Object is not writable.");
621 return -1;
622 }
Guido van Rossum98297ee2007-11-06 21:34:58 +0000623
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000624 view->obj = obj;
625 if (obj)
626 Py_INCREF(obj);
627 view->buf = buf;
628 view->len = len;
629 view->readonly = readonly;
630 view->itemsize = 1;
631 view->format = NULL;
632 if ((flags & PyBUF_FORMAT) == PyBUF_FORMAT)
633 view->format = "B";
634 view->ndim = 1;
635 view->shape = NULL;
636 if ((flags & PyBUF_ND) == PyBUF_ND)
637 view->shape = &(view->len);
638 view->strides = NULL;
639 if ((flags & PyBUF_STRIDES) == PyBUF_STRIDES)
640 view->strides = &(view->itemsize);
641 view->suboffsets = NULL;
642 view->internal = NULL;
643 return 0;
Travis E. Oliphantb99f7622007-08-18 11:21:56 +0000644}
645
Martin v. Löwis423be952008-08-13 15:53:07 +0000646void
647PyBuffer_Release(Py_buffer *view)
648{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 PyObject *obj = view->obj;
650 if (obj && Py_TYPE(obj)->tp_as_buffer && Py_TYPE(obj)->tp_as_buffer->bf_releasebuffer)
651 Py_TYPE(obj)->tp_as_buffer->bf_releasebuffer(obj, view);
652 Py_XDECREF(obj);
653 view->obj = NULL;
Martin v. Löwis423be952008-08-13 15:53:07 +0000654}
655
Eric Smith8fd3eba2008-02-17 19:48:00 +0000656PyObject *
657PyObject_Format(PyObject *obj, PyObject *format_spec)
658{
Eric Smith8fd3eba2008-02-17 19:48:00 +0000659 PyObject *meth;
660 PyObject *empty = NULL;
661 PyObject *result = NULL;
Benjamin Petersonce798522012-01-22 11:24:29 -0500662 _Py_IDENTIFIER(__format__);
Eric Smith8fd3eba2008-02-17 19:48:00 +0000663
664 /* If no format_spec is provided, use an empty string */
665 if (format_spec == NULL) {
Victor Stinner9d3b93b2011-11-22 02:27:30 +0100666 empty = PyUnicode_New(0, 0);
Benjamin Petersonda2cf042010-06-05 00:45:37 +0000667 format_spec = empty;
Eric Smith8fd3eba2008-02-17 19:48:00 +0000668 }
669
Eric Smith8fd3eba2008-02-17 19:48:00 +0000670 /* Find the (unbound!) __format__ method (a borrowed reference) */
Benjamin Petersonce798522012-01-22 11:24:29 -0500671 meth = _PyObject_LookupSpecial(obj, &PyId___format__);
Eric Smith8fd3eba2008-02-17 19:48:00 +0000672 if (meth == NULL) {
Benjamin Petersonda2cf042010-06-05 00:45:37 +0000673 if (!PyErr_Occurred())
674 PyErr_Format(PyExc_TypeError,
675 "Type %.100s doesn't define __format__",
676 Py_TYPE(obj)->tp_name);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000677 goto done;
Eric Smith8fd3eba2008-02-17 19:48:00 +0000678 }
679
Benjamin Petersonda2cf042010-06-05 00:45:37 +0000680 /* And call it. */
681 result = PyObject_CallFunctionObjArgs(meth, format_spec, NULL);
Benjamin Peterson6f889ad32010-06-05 02:11:45 +0000682 Py_DECREF(meth);
Eric Smith8fd3eba2008-02-17 19:48:00 +0000683
684 if (result && !PyUnicode_Check(result)) {
Benjamin Petersonda2cf042010-06-05 00:45:37 +0000685 PyErr_SetString(PyExc_TypeError,
686 "__format__ method did not return string");
687 Py_DECREF(result);
688 result = NULL;
689 goto done;
Eric Smith8fd3eba2008-02-17 19:48:00 +0000690 }
691
692done:
693 Py_XDECREF(empty);
694 return result;
695}
Guido van Rossumcea1c8c1998-05-22 00:47:05 +0000696/* Operations on numbers */
697
698int
Fred Drake79912472000-07-09 04:06:11 +0000699PyNumber_Check(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +0000700{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000701 return o && o->ob_type->tp_as_number &&
702 (o->ob_type->tp_as_number->nb_int ||
703 o->ob_type->tp_as_number->nb_float);
Guido van Rossume15dee51995-07-18 14:12:02 +0000704}
705
Guido van Rossumcea1c8c1998-05-22 00:47:05 +0000706/* Binary operators */
Guido van Rossume15dee51995-07-18 14:12:02 +0000707
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000708#define NB_SLOT(x) offsetof(PyNumberMethods, x)
709#define NB_BINOP(nb_methods, slot) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 (*(binaryfunc*)(& ((char*)nb_methods)[slot]))
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000711#define NB_TERNOP(nb_methods, slot) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000712 (*(ternaryfunc*)(& ((char*)nb_methods)[slot]))
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000713
714/*
715 Calling scheme used for binary operations:
716
Neal Norwitz4886cc32006-08-21 17:06:07 +0000717 Order operations are tried until either a valid result or error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000718 w.op(v,w)[*], v.op(v,w), w.op(v,w)
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000719
Guido van Rossum4bb1e362001-09-28 23:49:48 +0000720 [*] only when v->ob_type != w->ob_type && w->ob_type is a subclass of
721 v->ob_type
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000722 */
723
724static PyObject *
725binary_op1(PyObject *v, PyObject *w, const int op_slot)
Guido van Rossume15dee51995-07-18 14:12:02 +0000726{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000727 PyObject *x;
728 binaryfunc slotv = NULL;
729 binaryfunc slotw = NULL;
Guido van Rossum4bb1e362001-09-28 23:49:48 +0000730
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000731 if (v->ob_type->tp_as_number != NULL)
732 slotv = NB_BINOP(v->ob_type->tp_as_number, op_slot);
733 if (w->ob_type != v->ob_type &&
734 w->ob_type->tp_as_number != NULL) {
735 slotw = NB_BINOP(w->ob_type->tp_as_number, op_slot);
736 if (slotw == slotv)
737 slotw = NULL;
738 }
739 if (slotv) {
740 if (slotw && PyType_IsSubtype(w->ob_type, v->ob_type)) {
741 x = slotw(v, w);
742 if (x != Py_NotImplemented)
743 return x;
744 Py_DECREF(x); /* can't do it */
745 slotw = NULL;
746 }
747 x = slotv(v, w);
748 if (x != Py_NotImplemented)
749 return x;
750 Py_DECREF(x); /* can't do it */
751 }
752 if (slotw) {
753 x = slotw(v, w);
754 if (x != Py_NotImplemented)
755 return x;
756 Py_DECREF(x); /* can't do it */
757 }
Brian Curtindfc80e32011-08-10 20:28:54 -0500758 Py_RETURN_NOTIMPLEMENTED;
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000759}
Guido van Rossum77660912002-04-16 16:32:50 +0000760
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000761static PyObject *
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000762binop_type_error(PyObject *v, PyObject *w, const char *op_name)
763{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000764 PyErr_Format(PyExc_TypeError,
765 "unsupported operand type(s) for %.100s: "
766 "'%.100s' and '%.100s'",
767 op_name,
768 v->ob_type->tp_name,
769 w->ob_type->tp_name);
770 return NULL;
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000771}
772
773static PyObject *
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000774binary_op(PyObject *v, PyObject *w, const int op_slot, const char *op_name)
775{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000776 PyObject *result = binary_op1(v, w, op_slot);
777 if (result == Py_NotImplemented) {
778 Py_DECREF(result);
779 return binop_type_error(v, w, op_name);
780 }
781 return result;
Guido van Rossume15dee51995-07-18 14:12:02 +0000782}
783
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000784
785/*
786 Calling scheme used for ternary operations:
787
Neal Norwitz4886cc32006-08-21 17:06:07 +0000788 Order operations are tried until either a valid result or error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000789 v.op(v,w,z), w.op(v,w,z), z.op(v,w,z)
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000790 */
791
792static PyObject *
793ternary_op(PyObject *v,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000794 PyObject *w,
795 PyObject *z,
796 const int op_slot,
797 const char *op_name)
Guido van Rossume15dee51995-07-18 14:12:02 +0000798{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000799 PyNumberMethods *mv, *mw, *mz;
800 PyObject *x = NULL;
801 ternaryfunc slotv = NULL;
802 ternaryfunc slotw = NULL;
803 ternaryfunc slotz = NULL;
Guido van Rossum77660912002-04-16 16:32:50 +0000804
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000805 mv = v->ob_type->tp_as_number;
806 mw = w->ob_type->tp_as_number;
807 if (mv != NULL)
808 slotv = NB_TERNOP(mv, op_slot);
809 if (w->ob_type != v->ob_type &&
810 mw != NULL) {
811 slotw = NB_TERNOP(mw, op_slot);
812 if (slotw == slotv)
813 slotw = NULL;
814 }
815 if (slotv) {
816 if (slotw && PyType_IsSubtype(w->ob_type, v->ob_type)) {
817 x = slotw(v, w, z);
818 if (x != Py_NotImplemented)
819 return x;
820 Py_DECREF(x); /* can't do it */
821 slotw = NULL;
822 }
823 x = slotv(v, w, z);
824 if (x != Py_NotImplemented)
825 return x;
826 Py_DECREF(x); /* can't do it */
827 }
828 if (slotw) {
829 x = slotw(v, w, z);
830 if (x != Py_NotImplemented)
831 return x;
832 Py_DECREF(x); /* can't do it */
833 }
834 mz = z->ob_type->tp_as_number;
835 if (mz != NULL) {
836 slotz = NB_TERNOP(mz, op_slot);
837 if (slotz == slotv || slotz == slotw)
838 slotz = NULL;
839 if (slotz) {
840 x = slotz(v, w, z);
841 if (x != Py_NotImplemented)
842 return x;
843 Py_DECREF(x); /* can't do it */
844 }
845 }
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000846
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000847 if (z == Py_None)
848 PyErr_Format(
849 PyExc_TypeError,
850 "unsupported operand type(s) for ** or pow(): "
851 "'%.100s' and '%.100s'",
852 v->ob_type->tp_name,
853 w->ob_type->tp_name);
854 else
855 PyErr_Format(
856 PyExc_TypeError,
857 "unsupported operand type(s) for pow(): "
858 "'%.100s', '%.100s', '%.100s'",
859 v->ob_type->tp_name,
860 w->ob_type->tp_name,
861 z->ob_type->tp_name);
862 return NULL;
Guido van Rossume15dee51995-07-18 14:12:02 +0000863}
864
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000865#define BINARY_FUNC(func, op, op_name) \
866 PyObject * \
867 func(PyObject *v, PyObject *w) { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000868 return binary_op(v, w, NB_SLOT(op), op_name); \
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000869 }
Guido van Rossume15dee51995-07-18 14:12:02 +0000870
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000871BINARY_FUNC(PyNumber_Or, nb_or, "|")
872BINARY_FUNC(PyNumber_Xor, nb_xor, "^")
873BINARY_FUNC(PyNumber_And, nb_and, "&")
874BINARY_FUNC(PyNumber_Lshift, nb_lshift, "<<")
875BINARY_FUNC(PyNumber_Rshift, nb_rshift, ">>")
876BINARY_FUNC(PyNumber_Subtract, nb_subtract, "-")
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000877BINARY_FUNC(PyNumber_Divmod, nb_divmod, "divmod()")
Guido van Rossume15dee51995-07-18 14:12:02 +0000878
879PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000880PyNumber_Add(PyObject *v, PyObject *w)
Guido van Rossume15dee51995-07-18 14:12:02 +0000881{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000882 PyObject *result = binary_op1(v, w, NB_SLOT(nb_add));
883 if (result == Py_NotImplemented) {
884 PySequenceMethods *m = v->ob_type->tp_as_sequence;
885 Py_DECREF(result);
886 if (m && m->sq_concat) {
887 return (*m->sq_concat)(v, w);
888 }
889 result = binop_type_error(v, w, "+");
890 }
891 return result;
Guido van Rossume15dee51995-07-18 14:12:02 +0000892}
893
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000894static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000895sequence_repeat(ssizeargfunc repeatfunc, PyObject *seq, PyObject *n)
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000896{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000897 Py_ssize_t count;
898 if (PyIndex_Check(n)) {
899 count = PyNumber_AsSsize_t(n, PyExc_OverflowError);
900 if (count == -1 && PyErr_Occurred())
901 return NULL;
902 }
903 else {
904 return type_error("can't multiply sequence by "
905 "non-int of type '%.200s'", n);
906 }
907 return (*repeatfunc)(seq, count);
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000908}
909
910PyObject *
911PyNumber_Multiply(PyObject *v, PyObject *w)
912{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000913 PyObject *result = binary_op1(v, w, NB_SLOT(nb_multiply));
914 if (result == Py_NotImplemented) {
915 PySequenceMethods *mv = v->ob_type->tp_as_sequence;
916 PySequenceMethods *mw = w->ob_type->tp_as_sequence;
917 Py_DECREF(result);
918 if (mv && mv->sq_repeat) {
919 return sequence_repeat(mv->sq_repeat, v, w);
920 }
921 else if (mw && mw->sq_repeat) {
922 return sequence_repeat(mw->sq_repeat, w, v);
923 }
924 result = binop_type_error(v, w, "*");
925 }
926 return result;
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000927}
928
Guido van Rossume15dee51995-07-18 14:12:02 +0000929PyObject *
Guido van Rossum4668b002001-08-08 05:00:18 +0000930PyNumber_FloorDivide(PyObject *v, PyObject *w)
931{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000932 return binary_op(v, w, NB_SLOT(nb_floor_divide), "//");
Guido van Rossum4668b002001-08-08 05:00:18 +0000933}
934
935PyObject *
936PyNumber_TrueDivide(PyObject *v, PyObject *w)
937{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000938 return binary_op(v, w, NB_SLOT(nb_true_divide), "/");
Guido van Rossum4668b002001-08-08 05:00:18 +0000939}
940
941PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000942PyNumber_Remainder(PyObject *v, PyObject *w)
Guido van Rossume15dee51995-07-18 14:12:02 +0000943{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000944 return binary_op(v, w, NB_SLOT(nb_remainder), "%");
Guido van Rossume15dee51995-07-18 14:12:02 +0000945}
946
947PyObject *
Fred Drake79912472000-07-09 04:06:11 +0000948PyNumber_Power(PyObject *v, PyObject *w, PyObject *z)
Guido van Rossume15dee51995-07-18 14:12:02 +0000949{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 return ternary_op(v, w, z, NB_SLOT(nb_power), "** or pow()");
Guido van Rossume15dee51995-07-18 14:12:02 +0000951}
952
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000953/* Binary in-place operators */
954
955/* The in-place operators are defined to fall back to the 'normal',
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000956 non in-place operations, if the in-place methods are not in place.
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000957
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000958 - If the left hand object has the appropriate struct members, and
959 they are filled, call the appropriate function and return the
960 result. No coercion is done on the arguments; the left-hand object
961 is the one the operation is performed on, and it's up to the
962 function to deal with the right-hand object.
Guido van Rossum77660912002-04-16 16:32:50 +0000963
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000964 - Otherwise, in-place modification is not supported. Handle it exactly as
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000965 a non in-place operation of the same kind.
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000966
967 */
968
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000969static PyObject *
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000970binary_iop1(PyObject *v, PyObject *w, const int iop_slot, const int op_slot)
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000971{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000972 PyNumberMethods *mv = v->ob_type->tp_as_number;
973 if (mv != NULL) {
974 binaryfunc slot = NB_BINOP(mv, iop_slot);
975 if (slot) {
976 PyObject *x = (slot)(v, w);
977 if (x != Py_NotImplemented) {
978 return x;
979 }
980 Py_DECREF(x);
981 }
982 }
983 return binary_op1(v, w, op_slot);
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000984}
985
986static PyObject *
987binary_iop(PyObject *v, PyObject *w, const int iop_slot, const int op_slot,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000988 const char *op_name)
Neil Schemenauerd4b0fea2002-12-30 20:18:15 +0000989{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000990 PyObject *result = binary_iop1(v, w, iop_slot, op_slot);
991 if (result == Py_NotImplemented) {
992 Py_DECREF(result);
993 return binop_type_error(v, w, op_name);
994 }
995 return result;
Thomas Wouterse289e0b2000-08-24 20:08:19 +0000996}
997
Neil Schemenauer5a1f0152001-01-04 01:39:06 +0000998#define INPLACE_BINOP(func, iop, op, op_name) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000999 PyObject * \
1000 func(PyObject *v, PyObject *w) { \
1001 return binary_iop(v, w, NB_SLOT(iop), NB_SLOT(op), op_name); \
1002 }
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001003
Neil Schemenauer5a1f0152001-01-04 01:39:06 +00001004INPLACE_BINOP(PyNumber_InPlaceOr, nb_inplace_or, nb_or, "|=")
1005INPLACE_BINOP(PyNumber_InPlaceXor, nb_inplace_xor, nb_xor, "^=")
1006INPLACE_BINOP(PyNumber_InPlaceAnd, nb_inplace_and, nb_and, "&=")
1007INPLACE_BINOP(PyNumber_InPlaceLshift, nb_inplace_lshift, nb_lshift, "<<=")
1008INPLACE_BINOP(PyNumber_InPlaceRshift, nb_inplace_rshift, nb_rshift, ">>=")
1009INPLACE_BINOP(PyNumber_InPlaceSubtract, nb_inplace_subtract, nb_subtract, "-=")
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001010
1011PyObject *
Guido van Rossum4668b002001-08-08 05:00:18 +00001012PyNumber_InPlaceFloorDivide(PyObject *v, PyObject *w)
1013{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001014 return binary_iop(v, w, NB_SLOT(nb_inplace_floor_divide),
1015 NB_SLOT(nb_floor_divide), "//=");
Guido van Rossum4668b002001-08-08 05:00:18 +00001016}
1017
1018PyObject *
1019PyNumber_InPlaceTrueDivide(PyObject *v, PyObject *w)
1020{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 return binary_iop(v, w, NB_SLOT(nb_inplace_true_divide),
1022 NB_SLOT(nb_true_divide), "/=");
Guido van Rossum4668b002001-08-08 05:00:18 +00001023}
1024
1025PyObject *
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001026PyNumber_InPlaceAdd(PyObject *v, PyObject *w)
1027{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001028 PyObject *result = binary_iop1(v, w, NB_SLOT(nb_inplace_add),
1029 NB_SLOT(nb_add));
1030 if (result == Py_NotImplemented) {
1031 PySequenceMethods *m = v->ob_type->tp_as_sequence;
1032 Py_DECREF(result);
1033 if (m != NULL) {
1034 binaryfunc f = NULL;
1035 f = m->sq_inplace_concat;
1036 if (f == NULL)
1037 f = m->sq_concat;
1038 if (f != NULL)
1039 return (*f)(v, w);
1040 }
1041 result = binop_type_error(v, w, "+=");
1042 }
1043 return result;
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001044}
1045
1046PyObject *
1047PyNumber_InPlaceMultiply(PyObject *v, PyObject *w)
1048{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001049 PyObject *result = binary_iop1(v, w, NB_SLOT(nb_inplace_multiply),
1050 NB_SLOT(nb_multiply));
1051 if (result == Py_NotImplemented) {
1052 ssizeargfunc f = NULL;
1053 PySequenceMethods *mv = v->ob_type->tp_as_sequence;
1054 PySequenceMethods *mw = w->ob_type->tp_as_sequence;
1055 Py_DECREF(result);
1056 if (mv != NULL) {
1057 f = mv->sq_inplace_repeat;
1058 if (f == NULL)
1059 f = mv->sq_repeat;
1060 if (f != NULL)
1061 return sequence_repeat(f, v, w);
1062 }
1063 else if (mw != NULL) {
1064 /* Note that the right hand operand should not be
1065 * mutated in this case so sq_inplace_repeat is not
1066 * used. */
1067 if (mw->sq_repeat)
1068 return sequence_repeat(mw->sq_repeat, w, v);
1069 }
1070 result = binop_type_error(v, w, "*=");
1071 }
1072 return result;
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001073}
1074
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001075PyObject *
1076PyNumber_InPlaceRemainder(PyObject *v, PyObject *w)
1077{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001078 return binary_iop(v, w, NB_SLOT(nb_inplace_remainder),
1079 NB_SLOT(nb_remainder), "%=");
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001080}
1081
1082PyObject *
1083PyNumber_InPlacePower(PyObject *v, PyObject *w, PyObject *z)
1084{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001085 if (v->ob_type->tp_as_number &&
1086 v->ob_type->tp_as_number->nb_inplace_power != NULL) {
1087 return ternary_op(v, w, z, NB_SLOT(nb_inplace_power), "**=");
1088 }
1089 else {
1090 return ternary_op(v, w, z, NB_SLOT(nb_power), "**=");
1091 }
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001092}
1093
1094
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001095/* Unary operators and functions */
Guido van Rossume15dee51995-07-18 14:12:02 +00001096
1097PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001098PyNumber_Negative(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001099{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001100 PyNumberMethods *m;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001101
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001102 if (o == NULL)
1103 return null_error();
1104 m = o->ob_type->tp_as_number;
1105 if (m && m->nb_negative)
1106 return (*m->nb_negative)(o);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001107
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001108 return type_error("bad operand type for unary -: '%.200s'", o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001109}
1110
1111PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001112PyNumber_Positive(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001113{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001114 PyNumberMethods *m;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001115
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001116 if (o == NULL)
1117 return null_error();
1118 m = o->ob_type->tp_as_number;
1119 if (m && m->nb_positive)
1120 return (*m->nb_positive)(o);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001121
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001122 return type_error("bad operand type for unary +: '%.200s'", o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001123}
1124
1125PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001126PyNumber_Invert(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001127{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001128 PyNumberMethods *m;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001130 if (o == NULL)
1131 return null_error();
1132 m = o->ob_type->tp_as_number;
1133 if (m && m->nb_invert)
1134 return (*m->nb_invert)(o);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001135
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001136 return type_error("bad operand type for unary ~: '%.200s'", o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001137}
1138
1139PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001140PyNumber_Absolute(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001141{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001142 PyNumberMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001143
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001144 if (o == NULL)
1145 return null_error();
1146 m = o->ob_type->tp_as_number;
1147 if (m && m->nb_absolute)
1148 return m->nb_absolute(o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001149
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001150 return type_error("bad operand type for abs(): '%.200s'", o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001151}
1152
Guido van Rossum98297ee2007-11-06 21:34:58 +00001153/* Return a Python Int or Long from the object item
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001154 Raise TypeError if the result is not an int-or-long
Guido van Rossum98297ee2007-11-06 21:34:58 +00001155 or if the object cannot be interpreted as an index.
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001156*/
1157PyObject *
Guido van Rossum38fff8c2006-03-07 18:50:55 +00001158PyNumber_Index(PyObject *item)
1159{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001160 PyObject *result = NULL;
1161 if (item == NULL)
1162 return null_error();
1163 if (PyLong_Check(item)) {
1164 Py_INCREF(item);
1165 return item;
1166 }
1167 if (PyIndex_Check(item)) {
1168 result = item->ob_type->tp_as_number->nb_index(item);
1169 if (result && !PyLong_Check(result)) {
1170 PyErr_Format(PyExc_TypeError,
1171 "__index__ returned non-int "
1172 "(type %.200s)",
1173 result->ob_type->tp_name);
1174 Py_DECREF(result);
1175 return NULL;
1176 }
1177 }
1178 else {
1179 PyErr_Format(PyExc_TypeError,
1180 "'%.200s' object cannot be interpreted "
1181 "as an integer", item->ob_type->tp_name);
1182 }
1183 return result;
Guido van Rossum38fff8c2006-03-07 18:50:55 +00001184}
1185
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001186/* Return an error on Overflow only if err is not NULL*/
1187
1188Py_ssize_t
1189PyNumber_AsSsize_t(PyObject *item, PyObject *err)
1190{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001191 Py_ssize_t result;
1192 PyObject *runerr;
1193 PyObject *value = PyNumber_Index(item);
1194 if (value == NULL)
1195 return -1;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001196
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001197 /* We're done if PyLong_AsSsize_t() returns without error. */
1198 result = PyLong_AsSsize_t(value);
1199 if (result != -1 || !(runerr = PyErr_Occurred()))
1200 goto finish;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 /* Error handling code -- only manage OverflowError differently */
1203 if (!PyErr_GivenExceptionMatches(runerr, PyExc_OverflowError))
1204 goto finish;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001205
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001206 PyErr_Clear();
1207 /* If no error-handling desired then the default clipping
1208 is sufficient.
1209 */
1210 if (!err) {
1211 assert(PyLong_Check(value));
1212 /* Whether or not it is less than or equal to
1213 zero is determined by the sign of ob_size
1214 */
1215 if (_PyLong_Sign(value) < 0)
1216 result = PY_SSIZE_T_MIN;
1217 else
1218 result = PY_SSIZE_T_MAX;
1219 }
1220 else {
1221 /* Otherwise replace the error with caller's error object. */
1222 PyErr_Format(err,
1223 "cannot fit '%.200s' into an index-sized integer",
1224 item->ob_type->tp_name);
1225 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00001226
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001227 finish:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001228 Py_DECREF(value);
1229 return result;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00001230}
1231
1232
Benjamin Petersonb7f1da52012-03-21 14:44:43 -04001233/*
1234 Returns the Integral instance converted to an int. The instance is expected
1235 to be an int or have an __int__ method. Steals integral's
1236 reference. error_format will be used to create the TypeError if integral
1237 isn't actually an Integral instance. error_format should be a format string
1238 that can accept a char* naming integral's type.
1239*/
1240static PyObject *
1241convert_integral_to_int(PyObject *integral, const char *error_format)
Christian Heimes15ebc882008-02-04 18:48:49 +00001242{
Benjamin Petersond614e702012-03-21 14:38:11 -04001243 PyNumberMethods *nb;
1244 if (PyLong_Check(integral))
1245 return integral;
1246 nb = Py_TYPE(integral)->tp_as_number;
1247 if (nb->nb_int) {
1248 PyObject *as_int = nb->nb_int(integral);
Mark Dickinson3d783852012-09-27 19:38:59 +01001249 if (!as_int || PyLong_Check(as_int)) {
1250 Py_DECREF(integral);
Benjamin Petersond614e702012-03-21 14:38:11 -04001251 return as_int;
Mark Dickinson3d783852012-09-27 19:38:59 +01001252 }
Benjamin Petersond614e702012-03-21 14:38:11 -04001253 Py_DECREF(as_int);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001254 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001255 PyErr_Format(PyExc_TypeError, error_format, Py_TYPE(integral)->tp_name);
1256 Py_DECREF(integral);
Benjamin Petersond614e702012-03-21 14:38:11 -04001257 return NULL;
Christian Heimes15ebc882008-02-04 18:48:49 +00001258}
1259
1260
Guido van Rossum9e896b32000-04-05 20:11:21 +00001261/* Add a check for embedded NULL-bytes in the argument. */
Barry Warsaw226ae6c1999-10-12 19:54:53 +00001262static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001263long_from_string(const char *s, Py_ssize_t len)
Barry Warsaw226ae6c1999-10-12 19:54:53 +00001264{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001265 char *end;
1266 PyObject *x;
Barry Warsaw226ae6c1999-10-12 19:54:53 +00001267
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001268 x = PyLong_FromString((char*)s, &end, 10);
1269 if (x == NULL)
1270 return NULL;
1271 if (end != s + len) {
1272 PyErr_SetString(PyExc_ValueError,
1273 "null byte in argument for int()");
1274 Py_DECREF(x);
1275 return NULL;
1276 }
1277 return x;
Barry Warsaw226ae6c1999-10-12 19:54:53 +00001278}
1279
Guido van Rossume15dee51995-07-18 14:12:02 +00001280PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001281PyNumber_Long(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001282{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 PyNumberMethods *m;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001284 PyObject *trunc_func;
1285 const char *buffer;
1286 Py_ssize_t buffer_len;
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04001287 _Py_IDENTIFIER(__trunc__);
Christian Heimes15ebc882008-02-04 18:48:49 +00001288
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001289 if (o == NULL)
1290 return null_error();
1291 if (PyLong_CheckExact(o)) {
1292 Py_INCREF(o);
1293 return o;
1294 }
1295 m = o->ob_type->tp_as_number;
1296 if (m && m->nb_int) { /* This should include subclasses of int */
1297 PyObject *res = m->nb_int(o);
1298 if (res && !PyLong_Check(res)) {
1299 PyErr_Format(PyExc_TypeError,
1300 "__int__ returned non-int (type %.200s)",
1301 res->ob_type->tp_name);
1302 Py_DECREF(res);
1303 return NULL;
1304 }
1305 return res;
1306 }
1307 if (PyLong_Check(o)) /* An int subclass without nb_int */
1308 return _PyLong_Copy((PyLongObject *)o);
Benjamin Peterson1b1a8e72012-03-20 23:48:11 -04001309 trunc_func = _PyObject_LookupSpecial(o, &PyId___trunc__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001310 if (trunc_func) {
1311 PyObject *truncated = PyEval_CallObject(trunc_func, NULL);
1312 PyObject *int_instance;
1313 Py_DECREF(trunc_func);
1314 /* __trunc__ is specified to return an Integral type,
Benjamin Peterson520e8502012-03-21 14:51:14 -04001315 but int() needs to return a int. */
Benjamin Petersonb7f1da52012-03-21 14:44:43 -04001316 int_instance = convert_integral_to_int(truncated,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001317 "__trunc__ returned non-Integral (type %.200s)");
1318 return int_instance;
1319 }
Benjamin Peterson1b1a8e72012-03-20 23:48:11 -04001320 if (PyErr_Occurred())
1321 return NULL;
Christian Heimes15ebc882008-02-04 18:48:49 +00001322
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001323 if (PyBytes_Check(o))
1324 /* need to do extra error checking that PyLong_FromString()
Benjamin Peterson520e8502012-03-21 14:51:14 -04001325 * doesn't do. In particular int('9.5') must raise an
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001326 * exception, not truncate the float.
1327 */
1328 return long_from_string(PyBytes_AS_STRING(o),
1329 PyBytes_GET_SIZE(o));
1330 if (PyUnicode_Check(o))
1331 /* The above check is done in PyLong_FromUnicode(). */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001332 return PyLong_FromUnicodeObject(o, 10);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001333 if (!PyObject_AsCharBuffer(o, &buffer, &buffer_len))
1334 return long_from_string(buffer, buffer_len);
Guido van Rossume15dee51995-07-18 14:12:02 +00001335
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001336 return type_error("int() argument must be a string or a "
1337 "number, not '%.200s'", o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001338}
1339
1340PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001341PyNumber_Float(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001342{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001343 PyNumberMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001344
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001345 if (o == NULL)
1346 return null_error();
1347 m = o->ob_type->tp_as_number;
1348 if (m && m->nb_float) { /* This should include subclasses of float */
1349 PyObject *res = m->nb_float(o);
1350 if (res && !PyFloat_Check(res)) {
1351 PyErr_Format(PyExc_TypeError,
1352 "__float__ returned non-float (type %.200s)",
1353 res->ob_type->tp_name);
1354 Py_DECREF(res);
1355 return NULL;
1356 }
1357 return res;
1358 }
1359 if (PyFloat_Check(o)) { /* A float subclass with nb_float == NULL */
1360 PyFloatObject *po = (PyFloatObject *)o;
1361 return PyFloat_FromDouble(po->ob_fval);
1362 }
1363 return PyFloat_FromString(o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001364}
1365
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001366
1367PyObject *
1368PyNumber_ToBase(PyObject *n, int base)
1369{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001370 PyObject *res = NULL;
1371 PyObject *index = PyNumber_Index(n);
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001372
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001373 if (!index)
1374 return NULL;
1375 if (PyLong_Check(index))
1376 res = _PyLong_Format(index, base);
1377 else
1378 /* It should not be possible to get here, as
1379 PyNumber_Index already has a check for the same
1380 condition */
1381 PyErr_SetString(PyExc_ValueError, "PyNumber_ToBase: index not "
1382 "int or long");
1383 Py_DECREF(index);
1384 return res;
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001385}
1386
1387
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001388/* Operations on sequences */
Guido van Rossume15dee51995-07-18 14:12:02 +00001389
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001390int
Fred Drake79912472000-07-09 04:06:11 +00001391PySequence_Check(PyObject *s)
Guido van Rossume15dee51995-07-18 14:12:02 +00001392{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001393 if (PyDict_Check(s))
1394 return 0;
1395 return s != NULL && s->ob_type->tp_as_sequence &&
1396 s->ob_type->tp_as_sequence->sq_item != NULL;
Guido van Rossume15dee51995-07-18 14:12:02 +00001397}
1398
Martin v. Löwis18e16552006-02-15 17:27:45 +00001399Py_ssize_t
Jeremy Hylton6253f832000-07-12 12:56:19 +00001400PySequence_Size(PyObject *s)
Guido van Rossume15dee51995-07-18 14:12:02 +00001401{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001402 PySequenceMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001403
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001404 if (s == NULL) {
1405 null_error();
1406 return -1;
1407 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001408
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001409 m = s->ob_type->tp_as_sequence;
1410 if (m && m->sq_length)
1411 return m->sq_length(s);
Guido van Rossume15dee51995-07-18 14:12:02 +00001412
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001413 type_error("object of type '%.200s' has no len()", s);
1414 return -1;
Guido van Rossume15dee51995-07-18 14:12:02 +00001415}
1416
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +00001417#undef PySequence_Length
Martin v. Löwis18e16552006-02-15 17:27:45 +00001418Py_ssize_t
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +00001419PySequence_Length(PyObject *s)
1420{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001421 return PySequence_Size(s);
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +00001422}
1423#define PySequence_Length PySequence_Size
1424
Guido van Rossume15dee51995-07-18 14:12:02 +00001425PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001426PySequence_Concat(PyObject *s, PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001427{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001428 PySequenceMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001430 if (s == NULL || o == NULL)
1431 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00001432
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001433 m = s->ob_type->tp_as_sequence;
1434 if (m && m->sq_concat)
1435 return m->sq_concat(s, o);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001436
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001437 /* Instances of user classes defining an __add__() method only
1438 have an nb_add slot, not an sq_concat slot. So we fall back
1439 to nb_add if both arguments appear to be sequences. */
1440 if (PySequence_Check(s) && PySequence_Check(o)) {
1441 PyObject *result = binary_op1(s, o, NB_SLOT(nb_add));
1442 if (result != Py_NotImplemented)
1443 return result;
1444 Py_DECREF(result);
1445 }
1446 return type_error("'%.200s' object can't be concatenated", s);
Guido van Rossume15dee51995-07-18 14:12:02 +00001447}
1448
1449PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001450PySequence_Repeat(PyObject *o, Py_ssize_t count)
Guido van Rossume15dee51995-07-18 14:12:02 +00001451{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001452 PySequenceMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001453
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001454 if (o == NULL)
1455 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00001456
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001457 m = o->ob_type->tp_as_sequence;
1458 if (m && m->sq_repeat)
1459 return m->sq_repeat(o, count);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001460
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001461 /* Instances of user classes defining a __mul__() method only
1462 have an nb_multiply slot, not an sq_repeat slot. so we fall back
1463 to nb_multiply if o appears to be a sequence. */
1464 if (PySequence_Check(o)) {
1465 PyObject *n, *result;
1466 n = PyLong_FromSsize_t(count);
1467 if (n == NULL)
1468 return NULL;
1469 result = binary_op1(o, n, NB_SLOT(nb_multiply));
1470 Py_DECREF(n);
1471 if (result != Py_NotImplemented)
1472 return result;
1473 Py_DECREF(result);
1474 }
1475 return type_error("'%.200s' object can't be repeated", o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001476}
1477
1478PyObject *
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001479PySequence_InPlaceConcat(PyObject *s, PyObject *o)
1480{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001481 PySequenceMethods *m;
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001482
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001483 if (s == NULL || o == NULL)
1484 return null_error();
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001486 m = s->ob_type->tp_as_sequence;
1487 if (m && m->sq_inplace_concat)
1488 return m->sq_inplace_concat(s, o);
1489 if (m && m->sq_concat)
1490 return m->sq_concat(s, o);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001491
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001492 if (PySequence_Check(s) && PySequence_Check(o)) {
1493 PyObject *result = binary_iop1(s, o, NB_SLOT(nb_inplace_add),
1494 NB_SLOT(nb_add));
1495 if (result != Py_NotImplemented)
1496 return result;
1497 Py_DECREF(result);
1498 }
1499 return type_error("'%.200s' object can't be concatenated", s);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001500}
1501
1502PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001503PySequence_InPlaceRepeat(PyObject *o, Py_ssize_t count)
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001504{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001505 PySequenceMethods *m;
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001506
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001507 if (o == NULL)
1508 return null_error();
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001510 m = o->ob_type->tp_as_sequence;
1511 if (m && m->sq_inplace_repeat)
1512 return m->sq_inplace_repeat(o, count);
1513 if (m && m->sq_repeat)
1514 return m->sq_repeat(o, count);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001515
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001516 if (PySequence_Check(o)) {
1517 PyObject *n, *result;
1518 n = PyLong_FromSsize_t(count);
1519 if (n == NULL)
1520 return NULL;
1521 result = binary_iop1(o, n, NB_SLOT(nb_inplace_multiply),
1522 NB_SLOT(nb_multiply));
1523 Py_DECREF(n);
1524 if (result != Py_NotImplemented)
1525 return result;
1526 Py_DECREF(result);
1527 }
1528 return type_error("'%.200s' object can't be repeated", o);
Thomas Wouterse289e0b2000-08-24 20:08:19 +00001529}
1530
1531PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001532PySequence_GetItem(PyObject *s, Py_ssize_t i)
Guido van Rossume15dee51995-07-18 14:12:02 +00001533{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001534 PySequenceMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001536 if (s == NULL)
1537 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00001538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001539 m = s->ob_type->tp_as_sequence;
1540 if (m && m->sq_item) {
1541 if (i < 0) {
1542 if (m->sq_length) {
1543 Py_ssize_t l = (*m->sq_length)(s);
1544 if (l < 0)
1545 return NULL;
1546 i += l;
1547 }
1548 }
1549 return m->sq_item(s, i);
1550 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001551
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001552 return type_error("'%.200s' object does not support indexing", s);
Guido van Rossume15dee51995-07-18 14:12:02 +00001553}
1554
1555PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00001556PySequence_GetSlice(PyObject *s, Py_ssize_t i1, Py_ssize_t i2)
Guido van Rossume15dee51995-07-18 14:12:02 +00001557{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001558 PyMappingMethods *mp;
Guido van Rossume15dee51995-07-18 14:12:02 +00001559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001560 if (!s) return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00001561
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001562 mp = s->ob_type->tp_as_mapping;
Benjamin Peterson568867a2010-09-11 16:02:03 +00001563 if (mp && mp->mp_subscript) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001564 PyObject *res;
1565 PyObject *slice = _PySlice_FromIndices(i1, i2);
1566 if (!slice)
1567 return NULL;
1568 res = mp->mp_subscript(s, slice);
1569 Py_DECREF(slice);
1570 return res;
1571 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001572
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001573 return type_error("'%.200s' object is unsliceable", s);
Guido van Rossume15dee51995-07-18 14:12:02 +00001574}
1575
1576int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001577PySequence_SetItem(PyObject *s, Py_ssize_t i, PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001578{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001579 PySequenceMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001580
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001581 if (s == NULL) {
1582 null_error();
1583 return -1;
1584 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001585
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001586 m = s->ob_type->tp_as_sequence;
1587 if (m && m->sq_ass_item) {
1588 if (i < 0) {
1589 if (m->sq_length) {
1590 Py_ssize_t l = (*m->sq_length)(s);
1591 if (l < 0)
1592 return -1;
1593 i += l;
1594 }
1595 }
1596 return m->sq_ass_item(s, i, o);
1597 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001599 type_error("'%.200s' object does not support item assignment", s);
1600 return -1;
Guido van Rossume15dee51995-07-18 14:12:02 +00001601}
1602
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001603int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001604PySequence_DelItem(PyObject *s, Py_ssize_t i)
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001605{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001606 PySequenceMethods *m;
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001608 if (s == NULL) {
1609 null_error();
1610 return -1;
1611 }
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001612
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001613 m = s->ob_type->tp_as_sequence;
1614 if (m && m->sq_ass_item) {
1615 if (i < 0) {
1616 if (m->sq_length) {
1617 Py_ssize_t l = (*m->sq_length)(s);
1618 if (l < 0)
1619 return -1;
1620 i += l;
1621 }
1622 }
1623 return m->sq_ass_item(s, i, (PyObject *)NULL);
1624 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001625
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001626 type_error("'%.200s' object doesn't support item deletion", s);
1627 return -1;
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001628}
1629
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001630int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001631PySequence_SetSlice(PyObject *s, Py_ssize_t i1, Py_ssize_t i2, PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001632{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001633 PyMappingMethods *mp;
Guido van Rossume15dee51995-07-18 14:12:02 +00001634
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001635 if (s == NULL) {
1636 null_error();
1637 return -1;
1638 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001639
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001640 mp = s->ob_type->tp_as_mapping;
Benjamin Peterson568867a2010-09-11 16:02:03 +00001641 if (mp && mp->mp_ass_subscript) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001642 int res;
1643 PyObject *slice = _PySlice_FromIndices(i1, i2);
1644 if (!slice)
1645 return -1;
1646 res = mp->mp_ass_subscript(s, slice, o);
1647 Py_DECREF(slice);
1648 return res;
1649 }
Thomas Wouters1d75a792000-08-17 22:37:32 +00001650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001651 type_error("'%.200s' object doesn't support slice assignment", s);
1652 return -1;
Guido van Rossume15dee51995-07-18 14:12:02 +00001653}
1654
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001655int
Martin v. Löwis18e16552006-02-15 17:27:45 +00001656PySequence_DelSlice(PyObject *s, Py_ssize_t i1, Py_ssize_t i2)
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001657{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001658 PyMappingMethods *mp;
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001659
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001660 if (s == NULL) {
1661 null_error();
1662 return -1;
1663 }
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001664
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001665 mp = s->ob_type->tp_as_mapping;
Benjamin Peterson568867a2010-09-11 16:02:03 +00001666 if (mp && mp->mp_ass_subscript) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001667 int res;
1668 PyObject *slice = _PySlice_FromIndices(i1, i2);
1669 if (!slice)
1670 return -1;
1671 res = mp->mp_ass_subscript(s, slice, NULL);
1672 Py_DECREF(slice);
1673 return res;
1674 }
1675 type_error("'%.200s' object doesn't support slice deletion", s);
1676 return -1;
Guido van Rossum6cdc6f41996-08-21 17:41:54 +00001677}
1678
Guido van Rossume15dee51995-07-18 14:12:02 +00001679PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001680PySequence_Tuple(PyObject *v)
Guido van Rossume15dee51995-07-18 14:12:02 +00001681{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001682 PyObject *it; /* iter(v) */
1683 Py_ssize_t n; /* guess for result tuple size */
1684 PyObject *result = NULL;
1685 Py_ssize_t j;
Guido van Rossume15dee51995-07-18 14:12:02 +00001686
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001687 if (v == NULL)
1688 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00001689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 /* Special-case the common tuple and list cases, for efficiency. */
1691 if (PyTuple_CheckExact(v)) {
1692 /* Note that we can't know whether it's safe to return
1693 a tuple *subclass* instance as-is, hence the restriction
1694 to exact tuples here. In contrast, lists always make
1695 a copy, so there's no need for exactness below. */
1696 Py_INCREF(v);
1697 return v;
1698 }
1699 if (PyList_Check(v))
1700 return PyList_AsTuple(v);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001701
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001702 /* Get iterator. */
1703 it = PyObject_GetIter(v);
1704 if (it == NULL)
1705 return NULL;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001706
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001707 /* Guess result size and allocate space. */
Armin Ronacheraa9a79d2012-10-06 14:03:24 +02001708 n = PyObject_LengthHint(v, 10);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001709 if (n == -1)
1710 goto Fail;
1711 result = PyTuple_New(n);
1712 if (result == NULL)
1713 goto Fail;
Tim Peters6912d4d2001-05-05 03:56:37 +00001714
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001715 /* Fill the tuple. */
1716 for (j = 0; ; ++j) {
1717 PyObject *item = PyIter_Next(it);
1718 if (item == NULL) {
1719 if (PyErr_Occurred())
1720 goto Fail;
1721 break;
1722 }
1723 if (j >= n) {
1724 Py_ssize_t oldn = n;
1725 /* The over-allocation strategy can grow a bit faster
1726 than for lists because unlike lists the
1727 over-allocation isn't permanent -- we reclaim
1728 the excess before the end of this routine.
1729 So, grow by ten and then add 25%.
1730 */
1731 n += 10;
1732 n += n >> 2;
1733 if (n < oldn) {
1734 /* Check for overflow */
1735 PyErr_NoMemory();
1736 Py_DECREF(item);
1737 goto Fail;
1738 }
1739 if (_PyTuple_Resize(&result, n) != 0) {
1740 Py_DECREF(item);
1741 goto Fail;
1742 }
1743 }
1744 PyTuple_SET_ITEM(result, j, item);
1745 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001746
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001747 /* Cut tuple back if guess was too large. */
1748 if (j < n &&
1749 _PyTuple_Resize(&result, j) != 0)
1750 goto Fail;
Tim Peters6912d4d2001-05-05 03:56:37 +00001751
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001752 Py_DECREF(it);
1753 return result;
Tim Peters6912d4d2001-05-05 03:56:37 +00001754
1755Fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001756 Py_XDECREF(result);
1757 Py_DECREF(it);
1758 return NULL;
Guido van Rossume15dee51995-07-18 14:12:02 +00001759}
1760
Guido van Rossum3c5936a1996-12-05 21:51:24 +00001761PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001762PySequence_List(PyObject *v)
Guido van Rossum3c5936a1996-12-05 21:51:24 +00001763{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001764 PyObject *result; /* result list */
1765 PyObject *rv; /* return value from PyList_Extend */
Guido van Rossum4669fb41997-04-02 05:31:09 +00001766
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001767 if (v == NULL)
1768 return null_error();
Guido van Rossum5dba9e81998-07-10 18:03:50 +00001769
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001770 result = PyList_New(0);
1771 if (result == NULL)
1772 return NULL;
Tim Petersf553f892001-05-01 20:45:31 +00001773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 rv = _PyList_Extend((PyListObject *)result, v);
1775 if (rv == NULL) {
1776 Py_DECREF(result);
1777 return NULL;
1778 }
1779 Py_DECREF(rv);
1780 return result;
Guido van Rossum3c5936a1996-12-05 21:51:24 +00001781}
1782
Andrew M. Kuchling74042d62000-06-18 18:43:14 +00001783PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001784PySequence_Fast(PyObject *v, const char *m)
Andrew M. Kuchling74042d62000-06-18 18:43:14 +00001785{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001786 PyObject *it;
Raymond Hettinger2fb70292004-01-11 23:26:51 +00001787
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001788 if (v == NULL)
1789 return null_error();
Andrew M. Kuchling74042d62000-06-18 18:43:14 +00001790
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001791 if (PyList_CheckExact(v) || PyTuple_CheckExact(v)) {
1792 Py_INCREF(v);
1793 return v;
1794 }
Andrew M. Kuchling74042d62000-06-18 18:43:14 +00001795
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001796 it = PyObject_GetIter(v);
1797 if (it == NULL) {
1798 if (PyErr_ExceptionMatches(PyExc_TypeError))
1799 PyErr_SetString(PyExc_TypeError, m);
1800 return NULL;
1801 }
Raymond Hettinger2fb70292004-01-11 23:26:51 +00001802
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001803 v = PySequence_List(it);
1804 Py_DECREF(it);
Andrew M. Kuchling74042d62000-06-18 18:43:14 +00001805
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001806 return v;
Andrew M. Kuchling74042d62000-06-18 18:43:14 +00001807}
1808
Tim Peters16a77ad2001-09-08 04:00:12 +00001809/* Iterate over seq. Result depends on the operation:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001810 PY_ITERSEARCH_COUNT: -1 if error, else # of times obj appears in seq.
1811 PY_ITERSEARCH_INDEX: 0-based index of first occurrence of obj in seq;
1812 set ValueError and return -1 if none found; also return -1 on error.
Tim Peters16a77ad2001-09-08 04:00:12 +00001813 Py_ITERSEARCH_CONTAINS: return 1 if obj in seq, else 0; -1 on error.
1814*/
Neal Norwitz1fc4b772006-03-04 18:49:58 +00001815Py_ssize_t
Tim Peters16a77ad2001-09-08 04:00:12 +00001816_PySequence_IterSearch(PyObject *seq, PyObject *obj, int operation)
Guido van Rossume15dee51995-07-18 14:12:02 +00001817{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001818 Py_ssize_t n;
1819 int wrapped; /* for PY_ITERSEARCH_INDEX, true iff n wrapped around */
1820 PyObject *it; /* iter(seq) */
Guido van Rossume15dee51995-07-18 14:12:02 +00001821
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001822 if (seq == NULL || obj == NULL) {
1823 null_error();
1824 return -1;
1825 }
Tim Peters75f8e352001-05-05 11:33:43 +00001826
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001827 it = PyObject_GetIter(seq);
1828 if (it == NULL) {
1829 type_error("argument of type '%.200s' is not iterable", seq);
1830 return -1;
1831 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001832
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001833 n = wrapped = 0;
1834 for (;;) {
1835 int cmp;
1836 PyObject *item = PyIter_Next(it);
1837 if (item == NULL) {
1838 if (PyErr_Occurred())
1839 goto Fail;
1840 break;
1841 }
Tim Peters16a77ad2001-09-08 04:00:12 +00001842
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001843 cmp = PyObject_RichCompareBool(obj, item, Py_EQ);
1844 Py_DECREF(item);
1845 if (cmp < 0)
1846 goto Fail;
1847 if (cmp > 0) {
1848 switch (operation) {
1849 case PY_ITERSEARCH_COUNT:
1850 if (n == PY_SSIZE_T_MAX) {
1851 PyErr_SetString(PyExc_OverflowError,
1852 "count exceeds C integer size");
1853 goto Fail;
1854 }
1855 ++n;
1856 break;
Tim Peters16a77ad2001-09-08 04:00:12 +00001857
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001858 case PY_ITERSEARCH_INDEX:
1859 if (wrapped) {
1860 PyErr_SetString(PyExc_OverflowError,
1861 "index exceeds C integer size");
1862 goto Fail;
1863 }
1864 goto Done;
Tim Peters16a77ad2001-09-08 04:00:12 +00001865
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001866 case PY_ITERSEARCH_CONTAINS:
1867 n = 1;
1868 goto Done;
Tim Peters16a77ad2001-09-08 04:00:12 +00001869
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001870 default:
1871 assert(!"unknown operation");
1872 }
1873 }
Tim Peters16a77ad2001-09-08 04:00:12 +00001874
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001875 if (operation == PY_ITERSEARCH_INDEX) {
1876 if (n == PY_SSIZE_T_MAX)
1877 wrapped = 1;
1878 ++n;
1879 }
1880 }
Tim Peters16a77ad2001-09-08 04:00:12 +00001881
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001882 if (operation != PY_ITERSEARCH_INDEX)
1883 goto Done;
Tim Peters16a77ad2001-09-08 04:00:12 +00001884
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001885 PyErr_SetString(PyExc_ValueError,
1886 "sequence.index(x): x not in sequence");
1887 /* fall into failure code */
Tim Peters16a77ad2001-09-08 04:00:12 +00001888Fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001889 n = -1;
1890 /* fall through */
Tim Peters16a77ad2001-09-08 04:00:12 +00001891Done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001892 Py_DECREF(it);
1893 return n;
Tim Peters75f8e352001-05-05 11:33:43 +00001894
Guido van Rossume15dee51995-07-18 14:12:02 +00001895}
1896
Tim Peters16a77ad2001-09-08 04:00:12 +00001897/* Return # of times o appears in s. */
Neal Norwitz1fc4b772006-03-04 18:49:58 +00001898Py_ssize_t
Tim Peters16a77ad2001-09-08 04:00:12 +00001899PySequence_Count(PyObject *s, PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001900{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001901 return _PySequence_IterSearch(s, o, PY_ITERSEARCH_COUNT);
Guido van Rossume15dee51995-07-18 14:12:02 +00001902}
1903
Tim Peterscb8d3682001-05-05 21:05:01 +00001904/* Return -1 if error; 1 if ob in seq; 0 if ob not in seq.
Tim Peters16a77ad2001-09-08 04:00:12 +00001905 * Use sq_contains if possible, else defer to _PySequence_IterSearch().
Tim Peterscb8d3682001-05-05 21:05:01 +00001906 */
1907int
1908PySequence_Contains(PyObject *seq, PyObject *ob)
1909{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001910 Py_ssize_t result;
1911 PySequenceMethods *sqm = seq->ob_type->tp_as_sequence;
1912 if (sqm != NULL && sqm->sq_contains != NULL)
1913 return (*sqm->sq_contains)(seq, ob);
1914 result = _PySequence_IterSearch(seq, ob, PY_ITERSEARCH_CONTAINS);
1915 return Py_SAFE_DOWNCAST(result, Py_ssize_t, int);
Tim Peterscb8d3682001-05-05 21:05:01 +00001916}
1917
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001918/* Backwards compatibility */
1919#undef PySequence_In
1920int
Fred Drake79912472000-07-09 04:06:11 +00001921PySequence_In(PyObject *w, PyObject *v)
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001922{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001923 return PySequence_Contains(w, v);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001924}
1925
Neal Norwitz1fc4b772006-03-04 18:49:58 +00001926Py_ssize_t
Fred Drake79912472000-07-09 04:06:11 +00001927PySequence_Index(PyObject *s, PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001928{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001929 return _PySequence_IterSearch(s, o, PY_ITERSEARCH_INDEX);
Guido van Rossume15dee51995-07-18 14:12:02 +00001930}
1931
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001932/* Operations on mappings */
1933
1934int
Fred Drake79912472000-07-09 04:06:11 +00001935PyMapping_Check(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001936{
Benjamin Peterson21992272011-12-28 12:01:31 -06001937 return o && o->ob_type->tp_as_mapping &&
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001938 o->ob_type->tp_as_mapping->mp_subscript;
Guido van Rossume15dee51995-07-18 14:12:02 +00001939}
1940
Martin v. Löwis18e16552006-02-15 17:27:45 +00001941Py_ssize_t
Jeremy Hylton6253f832000-07-12 12:56:19 +00001942PyMapping_Size(PyObject *o)
Guido van Rossume15dee51995-07-18 14:12:02 +00001943{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001944 PyMappingMethods *m;
Guido van Rossume15dee51995-07-18 14:12:02 +00001945
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001946 if (o == NULL) {
1947 null_error();
1948 return -1;
1949 }
Guido van Rossume15dee51995-07-18 14:12:02 +00001950
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001951 m = o->ob_type->tp_as_mapping;
1952 if (m && m->mp_length)
1953 return m->mp_length(o);
Guido van Rossume15dee51995-07-18 14:12:02 +00001954
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001955 type_error("object of type '%.200s' has no len()", o);
1956 return -1;
Guido van Rossume15dee51995-07-18 14:12:02 +00001957}
1958
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +00001959#undef PyMapping_Length
Martin v. Löwis18e16552006-02-15 17:27:45 +00001960Py_ssize_t
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +00001961PyMapping_Length(PyObject *o)
1962{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001963 return PyMapping_Size(o);
Marc-André Lemburgcf5f3582000-07-17 09:22:55 +00001964}
1965#define PyMapping_Length PyMapping_Size
1966
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001967PyObject *
Fred Drake79912472000-07-09 04:06:11 +00001968PyMapping_GetItemString(PyObject *o, char *key)
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001969{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001970 PyObject *okey, *r;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001971
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001972 if (key == NULL)
1973 return null_error();
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001974
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001975 okey = PyUnicode_FromString(key);
1976 if (okey == NULL)
1977 return NULL;
1978 r = PyObject_GetItem(o, okey);
1979 Py_DECREF(okey);
1980 return r;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001981}
1982
1983int
Fred Drake79912472000-07-09 04:06:11 +00001984PyMapping_SetItemString(PyObject *o, char *key, PyObject *value)
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001985{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001986 PyObject *okey;
1987 int r;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001988
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001989 if (key == NULL) {
1990 null_error();
1991 return -1;
1992 }
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00001993
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001994 okey = PyUnicode_FromString(key);
1995 if (okey == NULL)
1996 return -1;
1997 r = PyObject_SetItem(o, okey, value);
1998 Py_DECREF(okey);
1999 return r;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002000}
2001
2002int
Fred Drake79912472000-07-09 04:06:11 +00002003PyMapping_HasKeyString(PyObject *o, char *key)
Guido van Rossume15dee51995-07-18 14:12:02 +00002004{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002005 PyObject *v;
Guido van Rossume15dee51995-07-18 14:12:02 +00002006
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002007 v = PyMapping_GetItemString(o, key);
2008 if (v) {
2009 Py_DECREF(v);
2010 return 1;
2011 }
2012 PyErr_Clear();
2013 return 0;
Guido van Rossume15dee51995-07-18 14:12:02 +00002014}
2015
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002016int
Fred Drake79912472000-07-09 04:06:11 +00002017PyMapping_HasKey(PyObject *o, PyObject *key)
Guido van Rossume15dee51995-07-18 14:12:02 +00002018{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002019 PyObject *v;
Guido van Rossume15dee51995-07-18 14:12:02 +00002020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002021 v = PyObject_GetItem(o, key);
2022 if (v) {
2023 Py_DECREF(v);
2024 return 1;
2025 }
2026 PyErr_Clear();
2027 return 0;
Guido van Rossume15dee51995-07-18 14:12:02 +00002028}
2029
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002030PyObject *
2031PyMapping_Keys(PyObject *o)
2032{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002033 PyObject *keys;
2034 PyObject *fast;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02002035 _Py_IDENTIFIER(keys);
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002036
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002037 if (PyDict_CheckExact(o))
2038 return PyDict_Keys(o);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002039 keys = _PyObject_CallMethodId(o, &PyId_keys, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002040 if (keys == NULL)
2041 return NULL;
2042 fast = PySequence_Fast(keys, "o.keys() are not iterable");
2043 Py_DECREF(keys);
2044 return fast;
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002045}
2046
2047PyObject *
2048PyMapping_Items(PyObject *o)
2049{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002050 PyObject *items;
2051 PyObject *fast;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02002052 _Py_IDENTIFIER(items);
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002053
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002054 if (PyDict_CheckExact(o))
2055 return PyDict_Items(o);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002056 items = _PyObject_CallMethodId(o, &PyId_items, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002057 if (items == NULL)
2058 return NULL;
2059 fast = PySequence_Fast(items, "o.items() are not iterable");
2060 Py_DECREF(items);
2061 return fast;
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002062}
2063
2064PyObject *
2065PyMapping_Values(PyObject *o)
2066{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002067 PyObject *values;
2068 PyObject *fast;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +02002069 _Py_IDENTIFIER(values);
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002070
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002071 if (PyDict_CheckExact(o))
2072 return PyDict_Values(o);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002073 values = _PyObject_CallMethodId(o, &PyId_values, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002074 if (values == NULL)
2075 return NULL;
2076 fast = PySequence_Fast(values, "o.values() are not iterable");
2077 Py_DECREF(values);
2078 return fast;
Guido van Rossumcc2b0162007-02-11 06:12:03 +00002079}
2080
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002081/* Operations on callable objects */
2082
2083/* XXX PyCallable_Check() is in object.c */
2084
Guido van Rossume15dee51995-07-18 14:12:02 +00002085PyObject *
Fred Drake79912472000-07-09 04:06:11 +00002086PyObject_CallObject(PyObject *o, PyObject *a)
Guido van Rossume15dee51995-07-18 14:12:02 +00002087{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002088 return PyEval_CallObjectWithKeywords(o, a, NULL);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002089}
Guido van Rossume15dee51995-07-18 14:12:02 +00002090
2091PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +00002092PyObject_Call(PyObject *func, PyObject *arg, PyObject *kw)
2093{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002094 ternaryfunc call;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002095
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002096 if ((call = func->ob_type->tp_call) != NULL) {
2097 PyObject *result;
2098 if (Py_EnterRecursiveCall(" while calling a Python object"))
2099 return NULL;
2100 result = (*call)(func, arg, kw);
2101 Py_LeaveRecursiveCall();
2102 if (result == NULL && !PyErr_Occurred())
2103 PyErr_SetString(
2104 PyExc_SystemError,
2105 "NULL result without error in PyObject_Call");
2106 return result;
2107 }
2108 PyErr_Format(PyExc_TypeError, "'%.200s' object is not callable",
2109 func->ob_type->tp_name);
2110 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002111}
2112
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002113static PyObject*
2114call_function_tail(PyObject *callable, PyObject *args)
2115{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002116 PyObject *retval;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002117
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002118 if (args == NULL)
2119 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002120
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002121 if (!PyTuple_Check(args)) {
2122 PyObject *a;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002123
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002124 a = PyTuple_New(1);
2125 if (a == NULL) {
2126 Py_DECREF(args);
2127 return NULL;
2128 }
2129 PyTuple_SET_ITEM(a, 0, args);
2130 args = a;
2131 }
2132 retval = PyObject_Call(callable, args, NULL);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002133
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002134 Py_DECREF(args);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002135
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002136 return retval;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002137}
2138
Tim Peters6d6c1a32001-08-02 04:15:00 +00002139PyObject *
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002140PyObject_CallFunction(PyObject *callable, char *format, ...)
Guido van Rossume15dee51995-07-18 14:12:02 +00002141{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002142 va_list va;
2143 PyObject *args;
Guido van Rossume15dee51995-07-18 14:12:02 +00002144
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002145 if (callable == NULL)
2146 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00002147
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002148 if (format && *format) {
2149 va_start(va, format);
2150 args = Py_VaBuildValue(format, va);
2151 va_end(va);
2152 }
2153 else
2154 args = PyTuple_New(0);
Guido van Rossume15dee51995-07-18 14:12:02 +00002155
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002156 return call_function_tail(callable, args);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002157}
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002158
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002159PyObject *
2160_PyObject_CallFunction_SizeT(PyObject *callable, char *format, ...)
2161{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002162 va_list va;
2163 PyObject *args;
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002164
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002165 if (callable == NULL)
2166 return null_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002167
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002168 if (format && *format) {
2169 va_start(va, format);
2170 args = _Py_VaBuildValue_SizeT(format, va);
2171 va_end(va);
2172 }
2173 else
2174 args = PyTuple_New(0);
Guido van Rossumcea1c8c1998-05-22 00:47:05 +00002175
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002176 return call_function_tail(callable, args);
Guido van Rossume15dee51995-07-18 14:12:02 +00002177}
2178
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002179static PyObject*
2180callmethod(PyObject* func, char *format, va_list va, int is_size_t)
2181{
2182 PyObject *retval = NULL;
2183 PyObject *args;
2184
2185 if (!PyCallable_Check(func)) {
2186 type_error("attribute of type '%.200s' is not callable", func);
2187 goto exit;
2188 }
2189
2190 if (format && *format) {
2191 if (is_size_t)
2192 args = _Py_VaBuildValue_SizeT(format, va);
2193 else
2194 args = Py_VaBuildValue(format, va);
2195 }
2196 else
2197 args = PyTuple_New(0);
2198
2199 retval = call_function_tail(func, args);
2200
2201 exit:
2202 /* args gets consumed in call_function_tail */
2203 Py_XDECREF(func);
2204
2205 return retval;
2206}
2207
Guido van Rossume15dee51995-07-18 14:12:02 +00002208PyObject *
Guido van Rossume15dee51995-07-18 14:12:02 +00002209PyObject_CallMethod(PyObject *o, char *name, char *format, ...)
Guido van Rossume15dee51995-07-18 14:12:02 +00002210{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002211 va_list va;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002212 PyObject *func = NULL;
2213 PyObject *retval = NULL;
Guido van Rossume15dee51995-07-18 14:12:02 +00002214
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002215 if (o == NULL || name == NULL)
2216 return null_error();
Guido van Rossume15dee51995-07-18 14:12:02 +00002217
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002218 func = PyObject_GetAttrString(o, name);
2219 if (func == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002220 return 0;
2221 }
Guido van Rossume15dee51995-07-18 14:12:02 +00002222
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002223 va_start(va, format);
2224 retval = callmethod(func, format, va, 0);
2225 va_end(va);
2226 return retval;
2227}
2228
2229PyObject *
2230_PyObject_CallMethodId(PyObject *o, _Py_Identifier *name, char *format, ...)
2231{
2232 va_list va;
2233 PyObject *func = NULL;
2234 PyObject *retval = NULL;
2235
2236 if (o == NULL || name == NULL)
2237 return null_error();
2238
2239 func = _PyObject_GetAttrId(o, name);
2240 if (func == NULL) {
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002241 return 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002242 }
Guido van Rossume15dee51995-07-18 14:12:02 +00002243
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002244 va_start(va, format);
2245 retval = callmethod(func, format, va, 0);
2246 va_end(va);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002247 return retval;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002248}
2249
2250PyObject *
2251_PyObject_CallMethod_SizeT(PyObject *o, char *name, char *format, ...)
2252{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002253 va_list va;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002254 PyObject *func = NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002255 PyObject *retval;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002256
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002257 if (o == NULL || name == NULL)
2258 return null_error();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002259
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002260 func = PyObject_GetAttrString(o, name);
2261 if (func == NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002262 return 0;
2263 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002264 va_start(va, format);
2265 retval = callmethod(func, format, va, 1);
2266 va_end(va);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002267 return retval;
Guido van Rossume15dee51995-07-18 14:12:02 +00002268}
Guido van Rossum823649d2001-03-21 18:40:58 +00002269
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002270PyObject *
2271_PyObject_CallMethodId_SizeT(PyObject *o, _Py_Identifier *name, char *format, ...)
2272{
2273 va_list va;
2274 PyObject *func = NULL;
2275 PyObject *retval;
2276
2277 if (o == NULL || name == NULL)
2278 return null_error();
2279
2280 func = _PyObject_GetAttrId(o, name);
2281 if (func == NULL) {
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02002282 return NULL;
2283 }
2284 va_start(va, format);
2285 retval = callmethod(func, format, va, 1);
2286 va_end(va);
2287 return retval;
2288}
Guido van Rossum823649d2001-03-21 18:40:58 +00002289
Fred Drakeb421b8c2001-10-26 16:21:32 +00002290static PyObject *
Fred Drakeb0c079e2001-10-28 02:39:03 +00002291objargs_mktuple(va_list va)
Fred Drakeb421b8c2001-10-26 16:21:32 +00002292{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002293 int i, n = 0;
2294 va_list countva;
2295 PyObject *result, *tmp;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002296
Alexander Belopolskyf0f45142010-08-11 17:31:17 +00002297 Py_VA_COPY(countva, va);
Fred Drakeb421b8c2001-10-26 16:21:32 +00002298
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002299 while (((PyObject *)va_arg(countva, PyObject *)) != NULL)
2300 ++n;
2301 result = PyTuple_New(n);
2302 if (result != NULL && n > 0) {
2303 for (i = 0; i < n; ++i) {
2304 tmp = (PyObject *)va_arg(va, PyObject *);
2305 PyTuple_SET_ITEM(result, i, tmp);
2306 Py_INCREF(tmp);
2307 }
2308 }
2309 return result;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002310}
2311
2312PyObject *
Fred Drakeb0c079e2001-10-28 02:39:03 +00002313PyObject_CallMethodObjArgs(PyObject *callable, PyObject *name, ...)
Fred Drakeb421b8c2001-10-26 16:21:32 +00002314{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002315 PyObject *args, *tmp;
2316 va_list vargs;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002317
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002318 if (callable == NULL || name == NULL)
2319 return null_error();
Fred Drakeb421b8c2001-10-26 16:21:32 +00002320
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002321 callable = PyObject_GetAttr(callable, name);
2322 if (callable == NULL)
2323 return NULL;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002324
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002325 /* count the args */
2326 va_start(vargs, name);
2327 args = objargs_mktuple(vargs);
2328 va_end(vargs);
2329 if (args == NULL) {
2330 Py_DECREF(callable);
2331 return NULL;
2332 }
2333 tmp = PyObject_Call(callable, args, NULL);
2334 Py_DECREF(args);
2335 Py_DECREF(callable);
Fred Drakeb421b8c2001-10-26 16:21:32 +00002336
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002337 return tmp;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002338}
2339
2340PyObject *
Brett Cannonfd074152012-04-14 14:10:13 -04002341_PyObject_CallMethodObjIdArgs(PyObject *callable,
2342 struct _Py_Identifier *name, ...)
2343{
2344 PyObject *args, *tmp;
2345 va_list vargs;
2346
2347 if (callable == NULL || name == NULL)
2348 return null_error();
2349
2350 callable = _PyObject_GetAttrId(callable, name);
2351 if (callable == NULL)
2352 return NULL;
2353
2354 /* count the args */
2355 va_start(vargs, name);
2356 args = objargs_mktuple(vargs);
2357 va_end(vargs);
2358 if (args == NULL) {
2359 Py_DECREF(callable);
2360 return NULL;
2361 }
2362 tmp = PyObject_Call(callable, args, NULL);
2363 Py_DECREF(args);
2364 Py_DECREF(callable);
2365
2366 return tmp;
2367}
2368
2369PyObject *
Fred Drakeb0c079e2001-10-28 02:39:03 +00002370PyObject_CallFunctionObjArgs(PyObject *callable, ...)
Fred Drakeb421b8c2001-10-26 16:21:32 +00002371{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002372 PyObject *args, *tmp;
2373 va_list vargs;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002374
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002375 if (callable == NULL)
2376 return null_error();
Fred Drakeb421b8c2001-10-26 16:21:32 +00002377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002378 /* count the args */
2379 va_start(vargs, callable);
2380 args = objargs_mktuple(vargs);
2381 va_end(vargs);
2382 if (args == NULL)
2383 return NULL;
2384 tmp = PyObject_Call(callable, args, NULL);
2385 Py_DECREF(args);
Fred Drakeb421b8c2001-10-26 16:21:32 +00002386
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002387 return tmp;
Fred Drakeb421b8c2001-10-26 16:21:32 +00002388}
2389
2390
Guido van Rossum823649d2001-03-21 18:40:58 +00002391/* isinstance(), issubclass() */
2392
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04002393/* abstract_get_bases() has logically 4 return states:
Barry Warsawf16951c2002-04-23 22:45:44 +00002394 *
Barry Warsawf16951c2002-04-23 22:45:44 +00002395 * 1. getattr(cls, '__bases__') could raise an AttributeError
2396 * 2. getattr(cls, '__bases__') could raise some other exception
2397 * 3. getattr(cls, '__bases__') could return a tuple
2398 * 4. getattr(cls, '__bases__') could return something other than a tuple
2399 *
2400 * Only state #3 is a non-error state and only it returns a non-NULL object
2401 * (it returns the retrieved tuple).
2402 *
2403 * Any raised AttributeErrors are masked by clearing the exception and
2404 * returning NULL. If an object other than a tuple comes out of __bases__,
2405 * then again, the return value is NULL. So yes, these two situations
2406 * produce exactly the same results: NULL is returned and no error is set.
2407 *
2408 * If some exception other than AttributeError is raised, then NULL is also
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002409 * returned, but the exception is not cleared. That's because we want the
Barry Warsawf16951c2002-04-23 22:45:44 +00002410 * exception to be propagated along.
2411 *
2412 * Callers are expected to test for PyErr_Occurred() when the return value
2413 * is NULL to decide whether a valid exception should be propagated or not.
2414 * When there's no exception to propagate, it's customary for the caller to
2415 * set a TypeError.
2416 */
Neil Schemenauer6b471292001-10-18 03:18:43 +00002417static PyObject *
2418abstract_get_bases(PyObject *cls)
Guido van Rossum823649d2001-03-21 18:40:58 +00002419{
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04002420 _Py_IDENTIFIER(__bases__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002421 PyObject *bases;
Guido van Rossum823649d2001-03-21 18:40:58 +00002422
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002423 Py_ALLOW_RECURSION
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04002424 bases = _PyObject_GetAttrId(cls, &PyId___bases__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002425 Py_END_ALLOW_RECURSION
2426 if (bases == NULL) {
2427 if (PyErr_ExceptionMatches(PyExc_AttributeError))
2428 PyErr_Clear();
2429 return NULL;
2430 }
2431 if (!PyTuple_Check(bases)) {
2432 Py_DECREF(bases);
2433 return NULL;
2434 }
2435 return bases;
Neil Schemenauer6b471292001-10-18 03:18:43 +00002436}
2437
2438
2439static int
2440abstract_issubclass(PyObject *derived, PyObject *cls)
2441{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002442 PyObject *bases = NULL;
2443 Py_ssize_t i, n;
2444 int r = 0;
Neil Schemenauer6b471292001-10-18 03:18:43 +00002445
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002446 while (1) {
2447 if (derived == cls)
2448 return 1;
2449 bases = abstract_get_bases(derived);
2450 if (bases == NULL) {
2451 if (PyErr_Occurred())
2452 return -1;
2453 return 0;
2454 }
2455 n = PyTuple_GET_SIZE(bases);
2456 if (n == 0) {
2457 Py_DECREF(bases);
2458 return 0;
2459 }
2460 /* Avoid recursivity in the single inheritance case */
2461 if (n == 1) {
2462 derived = PyTuple_GET_ITEM(bases, 0);
2463 Py_DECREF(bases);
2464 continue;
2465 }
2466 for (i = 0; i < n; i++) {
2467 r = abstract_issubclass(PyTuple_GET_ITEM(bases, i), cls);
2468 if (r != 0)
2469 break;
2470 }
2471 Py_DECREF(bases);
2472 return r;
2473 }
Guido van Rossum823649d2001-03-21 18:40:58 +00002474}
2475
Walter Dörwaldd9a6ad32002-12-12 16:41:44 +00002476static int
2477check_class(PyObject *cls, const char *error)
2478{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002479 PyObject *bases = abstract_get_bases(cls);
2480 if (bases == NULL) {
2481 /* Do not mask errors. */
2482 if (!PyErr_Occurred())
2483 PyErr_SetString(PyExc_TypeError, error);
2484 return 0;
2485 }
2486 Py_DECREF(bases);
2487 return -1;
Walter Dörwaldd9a6ad32002-12-12 16:41:44 +00002488}
2489
Brett Cannon4f653312004-03-20 22:52:14 +00002490static int
Antoine Pitrouec569b72008-08-26 22:40:48 +00002491recursive_isinstance(PyObject *inst, PyObject *cls)
Guido van Rossum823649d2001-03-21 18:40:58 +00002492{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002493 PyObject *icls;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002494 int retval = 0;
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04002495 _Py_IDENTIFIER(__class__);
Guido van Rossum03bc7d32003-02-12 03:32:58 +00002496
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002497 if (PyType_Check(cls)) {
2498 retval = PyObject_TypeCheck(inst, (PyTypeObject *)cls);
2499 if (retval == 0) {
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04002500 PyObject *c = _PyObject_GetAttrId(inst, &PyId___class__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002501 if (c == NULL) {
Benjamin Peterson72288d42010-11-20 17:21:08 +00002502 if (PyErr_ExceptionMatches(PyExc_AttributeError))
R. David Murray6bb99892010-11-20 16:33:30 +00002503 PyErr_Clear();
Benjamin Peterson72288d42010-11-20 17:21:08 +00002504 else
R. David Murray6bb99892010-11-20 16:33:30 +00002505 retval = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002506 }
2507 else {
2508 if (c != (PyObject *)(inst->ob_type) &&
2509 PyType_Check(c))
2510 retval = PyType_IsSubtype(
2511 (PyTypeObject *)c,
2512 (PyTypeObject *)cls);
2513 Py_DECREF(c);
2514 }
2515 }
2516 }
2517 else {
2518 if (!check_class(cls,
Benjamin Petersone893af52010-06-28 19:43:42 +00002519 "isinstance() arg 2 must be a type or tuple of types"))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002520 return -1;
Benjamin Peterson9fc9bf42012-03-20 23:26:41 -04002521 icls = _PyObject_GetAttrId(inst, &PyId___class__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002522 if (icls == NULL) {
Benjamin Peterson72288d42010-11-20 17:21:08 +00002523 if (PyErr_ExceptionMatches(PyExc_AttributeError))
R. David Murray6bb99892010-11-20 16:33:30 +00002524 PyErr_Clear();
Benjamin Peterson72288d42010-11-20 17:21:08 +00002525 else
R. David Murray6bb99892010-11-20 16:33:30 +00002526 retval = -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002527 }
2528 else {
2529 retval = abstract_issubclass(icls, cls);
2530 Py_DECREF(icls);
2531 }
2532 }
Guido van Rossum823649d2001-03-21 18:40:58 +00002533
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002534 return retval;
Guido van Rossum823649d2001-03-21 18:40:58 +00002535}
2536
2537int
Brett Cannon4f653312004-03-20 22:52:14 +00002538PyObject_IsInstance(PyObject *inst, PyObject *cls)
2539{
Benjamin Petersonce798522012-01-22 11:24:29 -05002540 _Py_IDENTIFIER(__instancecheck__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002541 PyObject *checker;
Christian Heimesd5e2b6f2008-03-19 21:50:51 +00002542
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002543 /* Quick test for an exact match */
2544 if (Py_TYPE(inst) == (PyTypeObject *)cls)
2545 return 1;
Christian Heimesd5e2b6f2008-03-19 21:50:51 +00002546
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002547 if (PyTuple_Check(cls)) {
2548 Py_ssize_t i;
2549 Py_ssize_t n;
2550 int r = 0;
Antoine Pitrouec569b72008-08-26 22:40:48 +00002551
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002552 if (Py_EnterRecursiveCall(" in __instancecheck__"))
2553 return -1;
2554 n = PyTuple_GET_SIZE(cls);
2555 for (i = 0; i < n; ++i) {
2556 PyObject *item = PyTuple_GET_ITEM(cls, i);
2557 r = PyObject_IsInstance(inst, item);
2558 if (r != 0)
2559 /* either found it, or got an error */
2560 break;
2561 }
2562 Py_LeaveRecursiveCall();
2563 return r;
2564 }
Benjamin Peterson88fe5f92009-05-16 21:55:24 +00002565
Benjamin Petersonce798522012-01-22 11:24:29 -05002566 checker = _PyObject_LookupSpecial(cls, &PyId___instancecheck__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002567 if (checker != NULL) {
2568 PyObject *res;
2569 int ok = -1;
2570 if (Py_EnterRecursiveCall(" in __instancecheck__")) {
2571 Py_DECREF(checker);
2572 return ok;
2573 }
2574 res = PyObject_CallFunctionObjArgs(checker, inst, NULL);
2575 Py_LeaveRecursiveCall();
2576 Py_DECREF(checker);
2577 if (res != NULL) {
2578 ok = PyObject_IsTrue(res);
2579 Py_DECREF(res);
2580 }
2581 return ok;
2582 }
2583 else if (PyErr_Occurred())
2584 return -1;
2585 return recursive_isinstance(inst, cls);
Brett Cannon4f653312004-03-20 22:52:14 +00002586}
2587
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002588static int
Antoine Pitrouec569b72008-08-26 22:40:48 +00002589recursive_issubclass(PyObject *derived, PyObject *cls)
Guido van Rossum823649d2001-03-21 18:40:58 +00002590{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002591 if (PyType_Check(cls) && PyType_Check(derived)) {
2592 /* Fast path (non-recursive) */
2593 return PyType_IsSubtype((PyTypeObject *)derived, (PyTypeObject *)cls);
2594 }
2595 if (!check_class(derived,
2596 "issubclass() arg 1 must be a class"))
2597 return -1;
2598 if (!check_class(cls,
2599 "issubclass() arg 2 must be a class"
2600 " or tuple of classes"))
2601 return -1;
Guido van Rossum823649d2001-03-21 18:40:58 +00002602
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002603 return abstract_issubclass(derived, cls);
Guido van Rossum823649d2001-03-21 18:40:58 +00002604}
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002605
Brett Cannon4f653312004-03-20 22:52:14 +00002606int
2607PyObject_IsSubclass(PyObject *derived, PyObject *cls)
2608{
Benjamin Petersonce798522012-01-22 11:24:29 -05002609 _Py_IDENTIFIER(__subclasscheck__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002610 PyObject *checker;
Georg Brandldcfe8e42009-05-17 08:22:45 +00002611
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002612 if (PyTuple_Check(cls)) {
2613 Py_ssize_t i;
2614 Py_ssize_t n;
2615 int r = 0;
Antoine Pitrouec569b72008-08-26 22:40:48 +00002616
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002617 if (Py_EnterRecursiveCall(" in __subclasscheck__"))
2618 return -1;
2619 n = PyTuple_GET_SIZE(cls);
2620 for (i = 0; i < n; ++i) {
2621 PyObject *item = PyTuple_GET_ITEM(cls, i);
2622 r = PyObject_IsSubclass(derived, item);
2623 if (r != 0)
2624 /* either found it, or got an error */
2625 break;
2626 }
2627 Py_LeaveRecursiveCall();
2628 return r;
2629 }
Benjamin Peterson88fe5f92009-05-16 21:55:24 +00002630
Benjamin Petersonce798522012-01-22 11:24:29 -05002631 checker = _PyObject_LookupSpecial(cls, &PyId___subclasscheck__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002632 if (checker != NULL) {
2633 PyObject *res;
2634 int ok = -1;
2635 if (Py_EnterRecursiveCall(" in __subclasscheck__")) {
2636 Py_DECREF(checker);
2637 return ok;
2638 }
2639 res = PyObject_CallFunctionObjArgs(checker, derived, NULL);
2640 Py_LeaveRecursiveCall();
2641 Py_DECREF(checker);
2642 if (res != NULL) {
2643 ok = PyObject_IsTrue(res);
2644 Py_DECREF(res);
2645 }
2646 return ok;
2647 }
2648 else if (PyErr_Occurred())
2649 return -1;
2650 return recursive_issubclass(derived, cls);
Antoine Pitrouec569b72008-08-26 22:40:48 +00002651}
2652
2653int
2654_PyObject_RealIsInstance(PyObject *inst, PyObject *cls)
2655{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002656 return recursive_isinstance(inst, cls);
Antoine Pitrouec569b72008-08-26 22:40:48 +00002657}
2658
2659int
2660_PyObject_RealIsSubclass(PyObject *derived, PyObject *cls)
2661{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002662 return recursive_issubclass(derived, cls);
Brett Cannon4f653312004-03-20 22:52:14 +00002663}
2664
2665
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002666PyObject *
2667PyObject_GetIter(PyObject *o)
2668{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002669 PyTypeObject *t = o->ob_type;
2670 getiterfunc f = NULL;
2671 f = t->tp_iter;
2672 if (f == NULL) {
2673 if (PySequence_Check(o))
2674 return PySeqIter_New(o);
2675 return type_error("'%.200s' object is not iterable", o);
2676 }
2677 else {
2678 PyObject *res = (*f)(o);
2679 if (res != NULL && !PyIter_Check(res)) {
2680 PyErr_Format(PyExc_TypeError,
2681 "iter() returned non-iterator "
2682 "of type '%.100s'",
2683 res->ob_type->tp_name);
2684 Py_DECREF(res);
2685 res = NULL;
2686 }
2687 return res;
2688 }
Guido van Rossum213c7a62001-04-23 14:08:49 +00002689}
2690
Tim Petersf4848da2001-05-05 00:14:56 +00002691/* Return next item.
2692 * If an error occurs, return NULL. PyErr_Occurred() will be true.
2693 * If the iteration terminates normally, return NULL and clear the
2694 * PyExc_StopIteration exception (if it was set). PyErr_Occurred()
2695 * will be false.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002696 * Else return the next object. PyErr_Occurred() will be false.
Tim Petersf4848da2001-05-05 00:14:56 +00002697 */
Guido van Rossum213c7a62001-04-23 14:08:49 +00002698PyObject *
2699PyIter_Next(PyObject *iter)
2700{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002701 PyObject *result;
2702 result = (*iter->ob_type->tp_iternext)(iter);
2703 if (result == NULL &&
2704 PyErr_Occurred() &&
2705 PyErr_ExceptionMatches(PyExc_StopIteration))
2706 PyErr_Clear();
2707 return result;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002708}
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002709
2710
2711/*
2712 * Flatten a sequence of bytes() objects into a C array of
2713 * NULL terminated string pointers with a NULL char* terminating the array.
2714 * (ie: an argv or env list)
2715 *
2716 * Memory allocated for the returned list is allocated using malloc() and MUST
2717 * be freed by the caller using a free() loop or _Py_FreeCharPArray().
2718 */
2719char *const *
2720_PySequence_BytesToCharpArray(PyObject* self)
2721{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002722 char **array;
2723 Py_ssize_t i, argc;
2724 PyObject *item = NULL;
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002725
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002726 argc = PySequence_Size(self);
2727 if (argc == -1)
2728 return NULL;
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002729
Stefan Krah7cacd2e2012-08-21 08:16:09 +02002730 assert(argc >= 0);
2731
2732 if ((size_t)argc > (PY_SSIZE_T_MAX-sizeof(char *)) / sizeof(char *)) {
2733 PyErr_NoMemory();
2734 return NULL;
2735 }
2736
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002737 array = malloc((argc + 1) * sizeof(char *));
2738 if (array == NULL) {
2739 PyErr_NoMemory();
2740 return NULL;
2741 }
2742 for (i = 0; i < argc; ++i) {
2743 char *data;
2744 item = PySequence_GetItem(self, i);
Stefan Krahfd24f9e2012-08-20 11:04:24 +02002745 if (item == NULL) {
2746 /* NULL terminate before freeing. */
2747 array[i] = NULL;
2748 goto fail;
2749 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002750 data = PyBytes_AsString(item);
2751 if (data == NULL) {
2752 /* NULL terminate before freeing. */
2753 array[i] = NULL;
2754 goto fail;
2755 }
2756 array[i] = strdup(data);
2757 if (!array[i]) {
2758 PyErr_NoMemory();
2759 goto fail;
2760 }
2761 Py_DECREF(item);
2762 }
2763 array[argc] = NULL;
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002764
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002765 return array;
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002766
2767fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002768 Py_XDECREF(item);
2769 _Py_FreeCharPArray(array);
2770 return NULL;
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002771}
2772
2773
2774/* Free's a NULL terminated char** array of C strings. */
2775void
2776_Py_FreeCharPArray(char *const array[])
2777{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002778 Py_ssize_t i;
2779 for (i = 0; array[i] != NULL; ++i) {
2780 free(array[i]);
2781 }
2782 free((void*)array);
Gregory P. Smithfb94c5f2010-03-14 06:49:55 +00002783}