/* Abstract Object Interface (many thanks to Jim Fulton) */ #include "Python.h" #include /* Shorthands to return certain errors */ static PyObject * type_error(const char *msg) { PyErr_SetString(PyExc_TypeError, msg); return NULL; } static PyObject * null_error(void) { if (!PyErr_Occurred()) PyErr_SetString(PyExc_SystemError, "null argument to internal routine"); return NULL; } /* Operations on any object */ int PyObject_Cmp(PyObject *o1, PyObject *o2, int *result) { int r; if (o1 == NULL || o2 == NULL) { null_error(); return -1; } r = PyObject_Compare(o1, o2); if (PyErr_Occurred()) return -1; *result = r; return 0; } PyObject * PyObject_Type(PyObject *o) { PyObject *v; if (o == NULL) return null_error(); v = (PyObject *)o->ob_type; Py_INCREF(v); return v; } int PyObject_Size(PyObject *o) { PySequenceMethods *m; if (o == NULL) { null_error(); return -1; } m = o->ob_type->tp_as_sequence; if (m && m->sq_length) return m->sq_length(o); return PyMapping_Size(o); } #undef PyObject_Length int PyObject_Length(PyObject *o) { return PyObject_Size(o); } #define PyObject_Length PyObject_Size PyObject * PyObject_GetItem(PyObject *o, PyObject *key) { PyMappingMethods *m; if (o == NULL || key == NULL) return null_error(); m = o->ob_type->tp_as_mapping; if (m && m->mp_subscript) return m->mp_subscript(o, key); if (o->ob_type->tp_as_sequence) { if (PyInt_Check(key)) return PySequence_GetItem(o, PyInt_AsLong(key)); else if (PyLong_Check(key)) { long key_value = PyLong_AsLong(key); if (key_value == -1 && PyErr_Occurred()) return NULL; return PySequence_GetItem(o, key_value); } return type_error("sequence index must be integer"); } return type_error("unsubscriptable object"); } int PyObject_SetItem(PyObject *o, PyObject *key, PyObject *value) { PyMappingMethods *m; if (o == NULL || key == NULL || value == NULL) { null_error(); return -1; } m = o->ob_type->tp_as_mapping; if (m && m->mp_ass_subscript) return m->mp_ass_subscript(o, key, value); if (o->ob_type->tp_as_sequence) { if (PyInt_Check(key)) return PySequence_SetItem(o, PyInt_AsLong(key), value); else if (PyLong_Check(key)) { long key_value = PyLong_AsLong(key); if (key_value == -1 && PyErr_Occurred()) return -1; return PySequence_SetItem(o, key_value, value); } type_error("sequence index must be integer"); return -1; } type_error("object does not support item assignment"); return -1; } int PyObject_DelItem(PyObject *o, PyObject *key) { PyMappingMethods *m; if (o == NULL || key == NULL) { null_error(); return -1; } m = o->ob_type->tp_as_mapping; if (m && m->mp_ass_subscript) return m->mp_ass_subscript(o, key, (PyObject*)NULL); if (o->ob_type->tp_as_sequence) { if (PyInt_Check(key)) return PySequence_DelItem(o, PyInt_AsLong(key)); else if (PyLong_Check(key)) { long key_value = PyLong_AsLong(key); if (key_value == -1 && PyErr_Occurred()) return -1; return PySequence_DelItem(o, key_value); } type_error("sequence index must be integer"); return -1; } type_error("object does not support item deletion"); return -1; } int PyObject_AsCharBuffer(PyObject *obj, const char **buffer, int *buffer_len) { PyBufferProcs *pb; const char *pp; int len; if (obj == NULL || buffer == NULL || buffer_len == NULL) { null_error(); return -1; } pb = obj->ob_type->tp_as_buffer; if ( pb == NULL || pb->bf_getcharbuffer == NULL || pb->bf_getsegcount == NULL ) { PyErr_SetString(PyExc_TypeError, "expected a character buffer object"); goto onError; } if ( (*pb->bf_getsegcount)(obj,NULL) != 1 ) { PyErr_SetString(PyExc_TypeError, "expected a single-segment buffer object"); goto onError; } len = (*pb->bf_getcharbuffer)(obj,0,&pp); if (len < 0) goto onError; *buffer = pp; *buffer_len = len; return 0; onError: return -1; } int PyObject_AsReadBuffer(PyObject *obj, const void **buffer, int *buffer_len) { PyBufferProcs *pb; void *pp; int len; if (obj == NULL || buffer == NULL || buffer_len == NULL) { null_error(); return -1; } pb = obj->ob_type->tp_as_buffer; if ( pb == NULL || pb->bf_getreadbuffer == NULL || pb->bf_getsegcount == NULL ) { PyErr_SetString(PyExc_TypeError, "expected a readable buffer object"); goto onError; } if ( (*pb->bf_getsegcount)(obj,NULL) != 1 ) { PyErr_SetString(PyExc_TypeError, "expected a single-segment buffer object"); goto onError; } len = (*pb->bf_getreadbuffer)(obj,0,&pp); if (len < 0) goto onError; *buffer = pp; *buffer_len = len; return 0; onError: return -1; } int PyObject_AsWriteBuffer(PyObject *obj, void **buffer, int *buffer_len) { PyBufferProcs *pb; void*pp; int len; if (obj == NULL || buffer == NULL || buffer_len == NULL) { null_error(); return -1; } pb = obj->ob_type->tp_as_buffer; if ( pb == NULL || pb->bf_getwritebuffer == NULL || pb->bf_getsegcount == NULL ) { PyErr_SetString(PyExc_TypeError, "expected a writeable buffer object"); goto onError; } if ( (*pb->bf_getsegcount)(obj,NULL) != 1 ) { PyErr_SetString(PyExc_TypeError, "expected a single-segment buffer object"); goto onError; } len = (*pb->bf_getwritebuffer)(obj,0,&pp); if (len < 0) goto onError; *buffer = pp; *buffer_len = len; return 0; onError: return -1; } /* Operations on numbers */ int PyNumber_Check(PyObject *o) { return o && o->ob_type->tp_as_number; } /* Binary operators */ #define BINOP(v, w, opname, ropname, thisfunc) \ if (PyInstance_Check(v) || PyInstance_Check(w)) \ return PyInstance_DoBinOp(v, w, opname, ropname, thisfunc) PyObject * PyNumber_Or(PyObject *v, PyObject *w) { BINOP(v, w, "__or__", "__ror__", PyNumber_Or); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_or) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for |"); } PyObject * PyNumber_Xor(PyObject *v, PyObject *w) { BINOP(v, w, "__xor__", "__rxor__", PyNumber_Xor); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_xor) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for ^"); } PyObject * PyNumber_And(PyObject *v, PyObject *w) { BINOP(v, w, "__and__", "__rand__", PyNumber_And); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_and) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for &"); } PyObject * PyNumber_Lshift(PyObject *v, PyObject *w) { BINOP(v, w, "__lshift__", "__rlshift__", PyNumber_Lshift); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_lshift) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for <<"); } PyObject * PyNumber_Rshift(PyObject *v, PyObject *w) { BINOP(v, w, "__rshift__", "__rrshift__", PyNumber_Rshift); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_rshift) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for >>"); } PyObject * PyNumber_Add(PyObject *v, PyObject *w) { PySequenceMethods *m; BINOP(v, w, "__add__", "__radd__", PyNumber_Add); m = v->ob_type->tp_as_sequence; if (m && m->sq_concat) return (*m->sq_concat)(v, w); else if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_add) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for +"); } PyObject * PyNumber_Subtract(PyObject *v, PyObject *w) { BINOP(v, w, "__sub__", "__rsub__", PyNumber_Subtract); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_subtract) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for -"); } PyObject * PyNumber_Multiply(PyObject *v, PyObject *w) { PyTypeObject *tp = v->ob_type; PySequenceMethods *m; BINOP(v, w, "__mul__", "__rmul__", PyNumber_Multiply); if (tp->tp_as_number != NULL && w->ob_type->tp_as_sequence != NULL) { /* number*sequence -- swap v and w */ PyObject *tmp = v; v = w; w = tmp; tp = v->ob_type; } if (tp->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_multiply) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } m = tp->tp_as_sequence; if (m && m->sq_repeat) { long mul_value; if (PyInt_Check(w)) { mul_value = PyInt_AsLong(w); } else if (PyLong_Check(w)) { mul_value = PyLong_AsLong(w); if (mul_value == -1 && PyErr_Occurred()) return NULL; } else { return type_error( "can't multiply sequence with non-int"); } return (*m->sq_repeat)(v, (int)mul_value); } return type_error("bad operand type(s) for *"); } PyObject * PyNumber_Divide(PyObject *v, PyObject *w) { BINOP(v, w, "__div__", "__rdiv__", PyNumber_Divide); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_divide) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for /"); } PyObject * PyNumber_Remainder(PyObject *v, PyObject *w) { if (PyString_Check(v)) return PyString_Format(v, w); else if (PyUnicode_Check(v)) return PyUnicode_Format(v, w); BINOP(v, w, "__mod__", "__rmod__", PyNumber_Remainder); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_remainder) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for %"); } PyObject * PyNumber_Divmod(PyObject *v, PyObject *w) { BINOP(v, w, "__divmod__", "__rdivmod__", PyNumber_Divmod); if (v->ob_type->tp_as_number != NULL) { PyObject *x = NULL; PyObject * (*f)(PyObject *, PyObject *) = NULL; if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_divmod) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for divmod()"); } /* Power (binary or ternary) */ static PyObject * do_pow(PyObject *v, PyObject *w) { PyObject *res; PyObject * (*f)(PyObject *, PyObject *, PyObject *); BINOP(v, w, "__pow__", "__rpow__", do_pow); if (v->ob_type->tp_as_number == NULL || w->ob_type->tp_as_number == NULL) { PyErr_SetString(PyExc_TypeError, "pow(x, y) requires numeric arguments"); return NULL; } if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_power) != NULL) res = (*f)(v, w, Py_None); else res = type_error("pow(x, y) not defined for these operands"); Py_DECREF(v); Py_DECREF(w); return res; } PyObject * PyNumber_Power(PyObject *v, PyObject *w, PyObject *z) { PyObject *res; PyObject *v1, *z1, *w2, *z2; PyObject * (*f)(PyObject *, PyObject *, PyObject *); if (z == Py_None) return do_pow(v, w); /* XXX The ternary version doesn't do class instance coercions */ if (PyInstance_Check(v)) return v->ob_type->tp_as_number->nb_power(v, w, z); if (v->ob_type->tp_as_number == NULL || z->ob_type->tp_as_number == NULL || w->ob_type->tp_as_number == NULL) { return type_error("pow(x, y, z) requires numeric arguments"); } if (PyNumber_Coerce(&v, &w) != 0) return NULL; res = NULL; v1 = v; z1 = z; if (PyNumber_Coerce(&v1, &z1) != 0) goto error2; w2 = w; z2 = z1; if (PyNumber_Coerce(&w2, &z2) != 0) goto error1; if (v->ob_type->tp_as_number != NULL && (f = v1->ob_type->tp_as_number->nb_power) != NULL) res = (*f)(v1, w2, z2); else res = type_error( "pow(x, y, z) not defined for these operands"); Py_DECREF(w2); Py_DECREF(z2); error1: Py_DECREF(v1); Py_DECREF(z1); error2: Py_DECREF(v); Py_DECREF(w); return res; } /* Binary in-place operators */ /* The in-place operators are defined to fall back to the 'normal', non in-place operations, if the in-place methods are not in place, and to take class instances into account. This is how it is supposed to work: - If the left-hand-side object (the first argument) is an instance object, try to let PyInstance_HalfBinOp() handle it. Pass the non in-place variant of the function as callback, because it will only be used if the left-hand object is changed by coercion. - Otherwise, if the left hand object is not an instance object, it has the appropriate struct members, and they are filled, call the appropriate function and return the result. No coercion is done on the arguments; the left-hand object is the one the operation is performed on, and it's up to the function to deal with the right-hand object. - Otherwise, in-place modification is not supported. Handle it exactly as a non in-place operation of the same kind: - If either object is an instance, let PyInstance_DoBinOp() handle it. - Otherwise, both arguments are C types. If the left-hand object has the appropriate struct members filled, coerce, call the appropriate function, and return the result. - Otherwise, we are out of options: raise a type error specific to augmented assignment. */ #define HASINPLACE(t) PyType_HasFeature((t)->ob_type, Py_TPFLAGS_HAVE_INPLACEOPS) PyObject * PyNumber_InPlaceOr(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__ior__", &x, PyNumber_Or, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_or) != NULL) return (*f)(v, w); BINOP(v, w, "__or__", "__ror__", PyNumber_Or); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_or) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for |="); } PyObject * PyNumber_InPlaceXor(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__ixor__", &x, PyNumber_Xor, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_xor) != NULL) return (*f)(v, w); BINOP(v, w, "__xor__", "__rxor__", PyNumber_Xor); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_xor) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for ^="); } PyObject * PyNumber_InPlaceAnd(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__iand__", &x, PyNumber_And, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_and) != NULL) return (*f)(v, w); BINOP(v, w, "__and__", "__rand__", PyNumber_And); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_and) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for &="); } PyObject * PyNumber_InPlaceLshift(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__ilshift__", &x, PyNumber_Lshift, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_lshift) != NULL) return (*f)(v, w); BINOP(v, w, "__lshift__", "__rlshift__", PyNumber_Lshift); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_lshift) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for <<="); } PyObject * PyNumber_InPlaceRshift(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__irshift__", &x, PyNumber_Rshift, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_rshift) != NULL) return (*f)(v, w); BINOP(v, w, "__rshift__", "__rrshift__", PyNumber_Rshift); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_rshift) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for >>="); } PyObject * PyNumber_InPlaceAdd(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__iadd__", &x, PyNumber_Add, 0) <= 0) return x; } else if (HASINPLACE(v) && (v->ob_type->tp_as_sequence != NULL && (f = v->ob_type->tp_as_sequence->sq_inplace_concat) != NULL) || (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_inplace_add) != NULL)) return (*f)(v, w); BINOP(v, w, "__add__", "__radd__", PyNumber_Add); if (v->ob_type->tp_as_sequence != NULL && (f = v->ob_type->tp_as_sequence->sq_concat) != NULL) return (*f)(v, w); else if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_add) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for +="); } PyObject * PyNumber_InPlaceSubtract(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__isub__", &x, PyNumber_Subtract, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_subtract) != NULL) return (*f)(v, w); BINOP(v, w, "__sub__", "__rsub__", PyNumber_Subtract); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_subtract) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for -="); } PyObject * PyNumber_InPlaceMultiply(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject * (*f2)(PyObject *, int) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__imul__", &x, PyNumber_Multiply, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_multiply) != NULL) return (*f)(v, w); else if (v->ob_type->tp_as_sequence != NULL && HASINPLACE(v) && (f2 = v->ob_type->tp_as_sequence->sq_inplace_repeat) != NULL) { long mul_value; if (PyInt_Check(w)) { mul_value = PyInt_AsLong(w); } else if (PyLong_Check(w)) { mul_value = PyLong_AsLong(w); if (mul_value == -1 && PyErr_Occurred()) return NULL; } else { return type_error( "can't multiply sequence with non-int"); } return (*f2)(v, (int)mul_value); } BINOP(v, w, "__mul__", "__rmul__", PyNumber_Multiply); /* if (tp->tp_as_number != NULL && w->ob_type->tp_as_sequence != NULL) { */ /* number*sequence -- swap v and w */ /* PyObject *tmp = v; v = w; w = tmp; tp = v->ob_type; } */ if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_multiply) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } else if (v->ob_type->tp_as_sequence != NULL && (f2 = v->ob_type->tp_as_sequence->sq_repeat) != NULL) { long mul_value; if (PyInt_Check(w)) { mul_value = PyInt_AsLong(w); } else if (PyLong_Check(w)) { mul_value = PyLong_AsLong(w); if (mul_value == -1 && PyErr_Occurred()) return NULL; } else { return type_error( "can't multiply sequence with non-int"); } return (*f2)(v, (int)mul_value); } return type_error("bad operand type(s) for *="); } PyObject * PyNumber_InPlaceDivide(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__idiv__", &x, PyNumber_Divide, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_divide) != NULL) return (*f)(v, w); BINOP(v, w, "__div__", "__rdiv__", PyNumber_Divide); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if (v->ob_type->tp_as_number != NULL && (f = v->ob_type->tp_as_number->nb_divide) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for /="); } PyObject * PyNumber_InPlaceRemainder(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__imod__", &x, PyNumber_Remainder, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_remainder) != NULL) return (*f)(v, w); if (PyString_Check(v)) return PyString_Format(v, w); else if (PyUnicode_Check(v)) return PyUnicode_Format(v, w); BINOP(v, w, "__mod__", "__rmod__", PyNumber_Remainder); if (v->ob_type->tp_as_number != NULL) { if (PyNumber_Coerce(&v, &w) != 0) return NULL; if ((f = v->ob_type->tp_as_number->nb_remainder) != NULL) x = (*f)(v, w); Py_DECREF(v); Py_DECREF(w); if (f != NULL) return x; } return type_error("bad operand type(s) for %="); } /* In-place Power (binary or ternary, for API consistency) */ static PyObject * do_inplace_pow(PyObject *v, PyObject *w) { PyObject * (*f)(PyObject *, PyObject *, PyObject *) = NULL; PyObject *x; if (PyInstance_Check(v)) { if (PyInstance_HalfBinOp(v, w, "__ipow__", &x, do_pow, 0) <= 0) return x; } else if (v->ob_type->tp_as_number != NULL && HASINPLACE(v) && (f = v->ob_type->tp_as_number->nb_inplace_power) != NULL) return (*f)(v, w, Py_None); BINOP(v, w, "__pow__", "__rpow__", do_pow); if (v->ob_type->tp_as_number == NULL || w->ob_type->tp_as_number == NULL) { return type_error("bad operand type(s) for **="); } if (PyNumber_Coerce(&v, &w) != 0) return NULL; if ((f = v->ob_type->tp_as_number->nb_power) != NULL) x = (*f)(v, w, Py_None); else x = type_error("bad operand type(s) for **="); Py_DECREF(v); Py_DECREF(w); return x; } PyObject * PyNumber_InPlacePower(PyObject *v, PyObject *w, PyObject *z) { PyObject *res; PyObject *v1, *z1, *w2, *z2, *oldv; PyObject * (*f)(PyObject *, PyObject *, PyObject *); if (z == Py_None) return do_inplace_pow(v, w); /* XXX The ternary version doesn't do class instance coercions */ if (PyInstance_Check(v)) return v->ob_type->tp_as_number->nb_inplace_power(v, w, z); if (v->ob_type->tp_as_number == NULL || z->ob_type->tp_as_number == NULL || w->ob_type->tp_as_number == NULL) { return type_error("(inplace) pow(x, y, z) requires numeric arguments"); } oldv = v; Py_INCREF(oldv); res = NULL; if (PyNumber_Coerce(&v, &w) != 0) goto error3; v1 = v; z1 = z; if (PyNumber_Coerce(&v1, &z1) != 0) goto error2; w2 = w; z2 = z1; if (PyNumber_Coerce(&w2, &z2) != 0) goto error1; if (oldv == v1 && HASINPLACE(v1) && v->ob_type->tp_as_number != NULL && (f = v1->ob_type->tp_as_number->nb_inplace_power) != NULL) res = (*f)(v1, w2, z2); else if (v1->ob_type->tp_as_number != NULL && (f = v1->ob_type->tp_as_number->nb_power) != NULL) res = (*f)(v1, w2, z2); else res = type_error( "(inplace) pow(x, y, z) not defined for these operands"); Py_DECREF(w2); Py_DECREF(z2); error1: Py_DECREF(v1); Py_DECREF(z1); error2: Py_DECREF(v); Py_DECREF(w); error3: Py_DECREF(oldv); return res; } /* Unary operators and functions */ PyObject * PyNumber_Negative(PyObject *o) { PyNumberMethods *m; if (o == NULL) return null_error(); m = o->ob_type->tp_as_number; if (m && m->nb_negative) return (*m->nb_negative)(o); return type_error("bad operand type for unary -"); } PyObject * PyNumber_Positive(PyObject *o) { PyNumberMethods *m; if (o == NULL) return null_error(); m = o->ob_type->tp_as_number; if (m && m->nb_positive) return (*m->nb_positive)(o); return type_error("bad operand type for unary +"); } PyObject * PyNumber_Invert(PyObject *o) { PyNumberMethods *m; if (o == NULL) return null_error(); m = o->ob_type->tp_as_number; if (m && m->nb_invert) return (*m->nb_invert)(o); return type_error("bad operand type for unary ~"); } PyObject * PyNumber_Absolute(PyObject *o) { PyNumberMethods *m; if (o == NULL) return null_error(); m = o->ob_type->tp_as_number; if (m && m->nb_absolute) return m->nb_absolute(o); return type_error("bad operand type for abs()"); } /* Add a check for embedded NULL-bytes in the argument. */ static PyObject * int_from_string(const char *s, int len) { char *end; PyObject *x; x = PyInt_FromString((char*)s, &end, 10); if (x == NULL) return NULL; if (end != s + len) { PyErr_SetString(PyExc_ValueError, "null byte in argument for int()"); Py_DECREF(x); return NULL; } return x; } PyObject * PyNumber_Int(PyObject *o) { PyNumberMethods *m; const char *buffer; int buffer_len; if (o == NULL) return null_error(); if (PyInt_Check(o)) { Py_INCREF(o); return o; } if (PyString_Check(o)) return int_from_string(PyString_AS_STRING(o), PyString_GET_SIZE(o)); if (PyUnicode_Check(o)) return PyInt_FromUnicode(PyUnicode_AS_UNICODE(o), PyUnicode_GET_SIZE(o), 10); m = o->ob_type->tp_as_number; if (m && m->nb_int) return m->nb_int(o); if (!PyObject_AsCharBuffer(o, &buffer, &buffer_len)) return int_from_string((char*)buffer, buffer_len); return type_error("object can't be converted to int"); } /* Add a check for embedded NULL-bytes in the argument. */ static PyObject * long_from_string(const char *s, int len) { char *end; PyObject *x; x = PyLong_FromString((char*)s, &end, 10); if (x == NULL) return NULL; if (end != s + len) { PyErr_SetString(PyExc_ValueError, "null byte in argument for long()"); Py_DECREF(x); return NULL; } return x; } PyObject * PyNumber_Long(PyObject *o) { PyNumberMethods *m; const char *buffer; int buffer_len; if (o == NULL) return null_error(); if (PyLong_Check(o)) { Py_INCREF(o); return o; } if (PyString_Check(o)) /* need to do extra error checking that PyLong_FromString() * doesn't do. In particular long('9.5') must raise an * exception, not truncate the float. */ return long_from_string(PyString_AS_STRING(o), PyString_GET_SIZE(o)); if (PyUnicode_Check(o)) /* The above check is done in PyLong_FromUnicode(). */ return PyLong_FromUnicode(PyUnicode_AS_UNICODE(o), PyUnicode_GET_SIZE(o), 10); m = o->ob_type->tp_as_number; if (m && m->nb_long) return m->nb_long(o); if (!PyObject_AsCharBuffer(o, &buffer, &buffer_len)) return long_from_string(buffer, buffer_len); return type_error("object can't be converted to long"); } PyObject * PyNumber_Float(PyObject *o) { PyNumberMethods *m; if (o == NULL) return null_error(); if (PyFloat_Check(o)) { Py_INCREF(o); return o; } if (!PyString_Check(o)) { m = o->ob_type->tp_as_number; if (m && m->nb_float) return m->nb_float(o); } return PyFloat_FromString(o, NULL); } /* Operations on sequences */ int PySequence_Check(PyObject *s) { return s != NULL && s->ob_type->tp_as_sequence; } int PySequence_Size(PyObject *s) { PySequenceMethods *m; if (s == NULL) { null_error(); return -1; } m = s->ob_type->tp_as_sequence; if (m && m->sq_length) return m->sq_length(s); type_error("len() of unsized object"); return -1; } #undef PySequence_Length int PySequence_Length(PyObject *s) { return PySequence_Size(s); } #define PySequence_Length PySequence_Size PyObject * PySequence_Concat(PyObject *s, PyObject *o) { PySequenceMethods *m; if (s == NULL || o == NULL) return null_error(); m = s->ob_type->tp_as_sequence; if (m && m->sq_concat) return m->sq_concat(s, o); return type_error("object can't be concatenated"); } PyObject * PySequence_Repeat(PyObject *o, int count) { PySequenceMethods *m; if (o == NULL) return null_error(); m = o->ob_type->tp_as_sequence; if (m && m->sq_repeat) return m->sq_repeat(o, count); return type_error("object can't be repeated"); } PyObject * PySequence_InPlaceConcat(PyObject *s, PyObject *o) { PySequenceMethods *m; if (s == NULL || o == NULL) return null_error(); m = s->ob_type->tp_as_sequence; if (m && HASINPLACE(s) && m->sq_inplace_concat) return m->sq_inplace_concat(s, o); if (m && m->sq_concat) return m->sq_concat(s, o); return type_error("object can't be concatenated"); } PyObject * PySequence_InPlaceRepeat(PyObject *o, int count) { PySequenceMethods *m; if (o == NULL) return null_error(); m = o->ob_type->tp_as_sequence; if (m && HASINPLACE(o) && m->sq_inplace_repeat) return m->sq_inplace_repeat(o, count); if (m && m->sq_repeat) return m->sq_repeat(o, count); return type_error("object can't be repeated"); } PyObject * PySequence_GetItem(PyObject *s, int i) { PySequenceMethods *m; if (s == NULL) return null_error(); m = s->ob_type->tp_as_sequence; if (m && m->sq_item) { if (i < 0) { if (m->sq_length) { int l = (*m->sq_length)(s); if (l < 0) return NULL; i += l; } } return m->sq_item(s, i); } return type_error("unindexable object"); } static PyObject * sliceobj_from_intint(int i, int j) { PyObject *start, *end, *slice; start = PyInt_FromLong((long)i); if (!start) return NULL; end = PyInt_FromLong((long)j); if (!end) { Py_DECREF(start); return NULL; } slice = PySlice_New(start, end, NULL); Py_DECREF(start); Py_DECREF(end); return slice; } PyObject * PySequence_GetSlice(PyObject *s, int i1, int i2) { PySequenceMethods *m; PyMappingMethods *mp; if (!s) return null_error(); m = s->ob_type->tp_as_sequence; if (m && m->sq_slice) { if (i1 < 0 || i2 < 0) { if (m->sq_length) { int l = (*m->sq_length)(s); if (l < 0) return NULL; if (i1 < 0) i1 += l; if (i2 < 0) i2 += l; } } return m->sq_slice(s, i1, i2); } else if ((mp = s->ob_type->tp_as_mapping) && mp->mp_subscript) { PyObject *res; PyObject *slice = sliceobj_from_intint(i1, i2); if (!slice) return NULL; res = mp->mp_subscript(s, slice); Py_DECREF(slice); return res; } return type_error("unsliceable object"); } int PySequence_SetItem(PyObject *s, int i, PyObject *o) { PySequenceMethods *m; if (s == NULL) { null_error(); return -1; } m = s->ob_type->tp_as_sequence; if (m && m->sq_ass_item) { if (i < 0) { if (m->sq_length) { int l = (*m->sq_length)(s); if (l < 0) return -1; i += l; } } return m->sq_ass_item(s, i, o); } type_error("object doesn't support item assignment"); return -1; } int PySequence_DelItem(PyObject *s, int i) { PySequenceMethods *m; if (s == NULL) { null_error(); return -1; } m = s->ob_type->tp_as_sequence; if (m && m->sq_ass_item) { if (i < 0) { if (m->sq_length) { int l = (*m->sq_length)(s); if (l < 0) return -1; i += l; } } return m->sq_ass_item(s, i, (PyObject *)NULL); } type_error("object doesn't support item deletion"); return -1; } int PySequence_SetSlice(PyObject *s, int i1, int i2, PyObject *o) { PySequenceMethods *m; PyMappingMethods *mp; if (s == NULL) { null_error(); return -1; } m = s->ob_type->tp_as_sequence; if (m && m->sq_ass_slice) { if (i1 < 0 || i2 < 0) { if (m->sq_length) { int l = (*m->sq_length)(s); if (l < 0) return -1; if (i1 < 0) i1 += l; if (i2 < 0) i2 += l; } } return m->sq_ass_slice(s, i1, i2, o); } else if ((mp = s->ob_type->tp_as_mapping) && mp->mp_ass_subscript) { int res; PyObject *slice = sliceobj_from_intint(i1, i2); if (!slice) return -1; res = mp->mp_ass_subscript(s, slice, o); Py_DECREF(slice); return res; } type_error("object doesn't support slice assignment"); return -1; } int PySequence_DelSlice(PyObject *s, int i1, int i2) { PySequenceMethods *m; if (s == NULL) { null_error(); return -1; } m = s->ob_type->tp_as_sequence; if (m && m->sq_ass_slice) { if (i1 < 0 || i2 < 0) { if (m->sq_length) { int l = (*m->sq_length)(s); if (l < 0) return -1; if (i1 < 0) i1 += l; if (i2 < 0) i2 += l; } } return m->sq_ass_slice(s, i1, i2, (PyObject *)NULL); } type_error("object doesn't support slice deletion"); return -1; } PyObject * PySequence_Tuple(PyObject *v) { PySequenceMethods *m; if (v == NULL) return null_error(); if (PyTuple_Check(v)) { Py_INCREF(v); return v; } if (PyList_Check(v)) return PyList_AsTuple(v); /* There used to be code for strings here, but tuplifying strings is not a common activity, so I nuked it. Down with code bloat! */ /* Generic sequence object */ m = v->ob_type->tp_as_sequence; if (m && m->sq_item) { int i; PyObject *t; int n = PySequence_Size(v); if (n < 0) return NULL; t = PyTuple_New(n); if (t == NULL) return NULL; for (i = 0; ; i++) { PyObject *item = (*m->sq_item)(v, i); if (item == NULL) { if (PyErr_ExceptionMatches(PyExc_IndexError)) PyErr_Clear(); else { Py_DECREF(t); t = NULL; } break; } if (i >= n) { if (n < 500) n += 10; else n += 100; if (_PyTuple_Resize(&t, n, 0) != 0) break; } PyTuple_SET_ITEM(t, i, item); } if (i < n && t != NULL) _PyTuple_Resize(&t, i, 0); return t; } /* None of the above */ return type_error("tuple() argument must be a sequence"); } PyObject * PySequence_List(PyObject *v) { PySequenceMethods *m; if (v == NULL) return null_error(); if (PyList_Check(v)) return PyList_GetSlice(v, 0, PyList_GET_SIZE(v)); m = v->ob_type->tp_as_sequence; if (m && m->sq_item) { int i; PyObject *l; int n = PySequence_Size(v); if (n < 0) return NULL; l = PyList_New(n); if (l == NULL) return NULL; for (i = 0; ; i++) { PyObject *item = (*m->sq_item)(v, i); if (item == NULL) { if (PyErr_ExceptionMatches(PyExc_IndexError)) PyErr_Clear(); else { Py_DECREF(l); l = NULL; } break; } if (i < n) PyList_SET_ITEM(l, i, item); else if (PyList_Append(l, item) < 0) { Py_DECREF(l); l = NULL; break; } } if (i < n && l != NULL) { if (PyList_SetSlice(l, i, n, (PyObject *)NULL) != 0) { Py_DECREF(l); l = NULL; } } return l; } return type_error("list() argument must be a sequence"); } PyObject * PySequence_Fast(PyObject *v, const char *m) { if (v == NULL) return null_error(); if (PyList_Check(v) || PyTuple_Check(v)) { Py_INCREF(v); return v; } v = PySequence_Tuple(v); if (v == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) return type_error(m); return v; } int PySequence_Count(PyObject *s, PyObject *o) { int l, i, n, cmp, err; PyObject *item; if (s == NULL || o == NULL) { null_error(); return -1; } l = PySequence_Size(s); if (l < 0) return -1; n = 0; for (i = 0; i < l; i++) { item = PySequence_GetItem(s, i); if (item == NULL) return -1; err = PyObject_Cmp(item, o, &cmp); Py_DECREF(item); if (err < 0) return err; if (cmp == 0) n++; } return n; } int PySequence_Contains(PyObject *w, PyObject *v) /* v in w */ { int i, cmp; PyObject *x; PySequenceMethods *sq; if(PyType_HasFeature(w->ob_type, Py_TPFLAGS_HAVE_SEQUENCE_IN)) { sq = w->ob_type->tp_as_sequence; if(sq != NULL && sq->sq_contains != NULL) return (*sq->sq_contains)(w, v); } /* If there is no better way to check whether an item is is contained, do it the hard way */ sq = w->ob_type->tp_as_sequence; if (sq == NULL || sq->sq_item == NULL) { PyErr_SetString(PyExc_TypeError, "'in' or 'not in' needs sequence right argument"); return -1; } for (i = 0; ; i++) { x = (*sq->sq_item)(w, i); if (x == NULL) { if (PyErr_ExceptionMatches(PyExc_IndexError)) { PyErr_Clear(); break; } return -1; } cmp = PyObject_Compare(v, x); Py_XDECREF(x); if (cmp == 0) return 1; if (PyErr_Occurred()) return -1; } return 0; } /* Backwards compatibility */ #undef PySequence_In int PySequence_In(PyObject *w, PyObject *v) { return PySequence_Contains(w, v); } int PySequence_Index(PyObject *s, PyObject *o) { int l, i, cmp, err; PyObject *item; if (s == NULL || o == NULL) { null_error(); return -1; } l = PySequence_Size(s); if (l < 0) return -1; for (i = 0; i < l; i++) { item = PySequence_GetItem(s, i); if (item == NULL) return -1; err = PyObject_Cmp(item, o, &cmp); Py_DECREF(item); if (err < 0) return err; if (cmp == 0) return i; } PyErr_SetString(PyExc_ValueError, "sequence.index(x): x not in list"); return -1; } /* Operations on mappings */ int PyMapping_Check(PyObject *o) { return o && o->ob_type->tp_as_mapping; } int PyMapping_Size(PyObject *o) { PyMappingMethods *m; if (o == NULL) { null_error(); return -1; } m = o->ob_type->tp_as_mapping; if (m && m->mp_length) return m->mp_length(o); type_error("len() of unsized object"); return -1; } #undef PyMapping_Length int PyMapping_Length(PyObject *o) { return PyMapping_Size(o); } #define PyMapping_Length PyMapping_Size PyObject * PyMapping_GetItemString(PyObject *o, char *key) { PyObject *okey, *r; if (key == NULL) return null_error(); okey = PyString_FromString(key); if (okey == NULL) return NULL; r = PyObject_GetItem(o, okey); Py_DECREF(okey); return r; } int PyMapping_SetItemString(PyObject *o, char *key, PyObject *value) { PyObject *okey; int r; if (key == NULL) { null_error(); return -1; } okey = PyString_FromString(key); if (okey == NULL) return -1; r = PyObject_SetItem(o, okey, value); Py_DECREF(okey); return r; } int PyMapping_HasKeyString(PyObject *o, char *key) { PyObject *v; v = PyMapping_GetItemString(o, key); if (v) { Py_DECREF(v); return 1; } PyErr_Clear(); return 0; } int PyMapping_HasKey(PyObject *o, PyObject *key) { PyObject *v; v = PyObject_GetItem(o, key); if (v) { Py_DECREF(v); return 1; } PyErr_Clear(); return 0; } /* Operations on callable objects */ /* XXX PyCallable_Check() is in object.c */ PyObject * PyObject_CallObject(PyObject *o, PyObject *a) { PyObject *r; PyObject *args = a; if (args == NULL) { args = PyTuple_New(0); if (args == NULL) return NULL; } r = PyEval_CallObject(o, args); if (args != a) { Py_DECREF(args); } return r; } PyObject * PyObject_CallFunction(PyObject *callable, char *format, ...) { va_list va; PyObject *args, *retval; va_start(va, format); if (callable == NULL) { va_end(va); return null_error(); } if (format) args = Py_VaBuildValue(format, va); else args = PyTuple_New(0); va_end(va); if (args == NULL) return NULL; if (!PyTuple_Check(args)) { PyObject *a; a = PyTuple_New(1); if (a == NULL) return NULL; if (PyTuple_SetItem(a, 0, args) < 0) return NULL; args = a; } retval = PyObject_CallObject(callable, args); Py_DECREF(args); return retval; } PyObject * PyObject_CallMethod(PyObject *o, char *name, char *format, ...) { va_list va; PyObject *args, *func = 0, *retval; va_start(va, format); if (o == NULL || name == NULL) { va_end(va); return null_error(); } func = PyObject_GetAttrString(o, name); if (func == NULL) { va_end(va); PyErr_SetString(PyExc_AttributeError, name); return 0; } if (!PyCallable_Check(func)) { va_end(va); return type_error("call of non-callable attribute"); } if (format && *format) args = Py_VaBuildValue(format, va); else args = PyTuple_New(0); va_end(va); if (!args) return NULL; if (!PyTuple_Check(args)) { PyObject *a; a = PyTuple_New(1); if (a == NULL) return NULL; if (PyTuple_SetItem(a, 0, args) < 0) return NULL; args = a; } retval = PyObject_CallObject(func, args); Py_DECREF(args); Py_DECREF(func); return retval; }