math.factorial depends on PyLong_AsLong correctly converting floats; rewrite

it to do the conversion explicitly instead.  See issue #7550.
This commit is contained in:
Mark Dickinson 2009-12-20 13:58:18 +00:00
parent 0732fd952b
commit 5698977186
1 changed files with 9 additions and 2 deletions

View File

@ -1082,15 +1082,22 @@ math_factorial(PyObject *self, PyObject *arg)
PyObject *result, *iobj, *newresult;
if (PyFloat_Check(arg)) {
PyObject *lx;
double dx = PyFloat_AS_DOUBLE((PyFloatObject *)arg);
if (dx != floor(dx)) {
if (!(Py_IS_FINITE(dx) && dx == floor(dx))) {
PyErr_SetString(PyExc_ValueError,
"factorial() only accepts integral values");
return NULL;
}
lx = PyLong_FromDouble(dx);
if (lx == NULL)
return NULL;
x = PyLong_AsLong(lx);
Py_DECREF(lx);
}
else
x = PyInt_AsLong(arg);
x = PyInt_AsLong(arg);
if (x == -1 && PyErr_Occurred())
return NULL;
if (x < 0) {