Kaydet (Commit) 0f30dbd9 authored tarafından Neal Norwitz's avatar Neal Norwitz

Fix SF # 626275, missing DECREF's in embedding example

Tested w/valgrind, all paths except the return on PyInt_AsLong() failure
I think I got all of these right.

Backport candidate.
üst 1bdca5e0
...@@ -17,6 +17,8 @@ main(int argc, char *argv[]) ...@@ -17,6 +17,8 @@ main(int argc, char *argv[])
/* Error checking of pName left out */ /* Error checking of pName left out */
pModule = PyImport_Import(pName); pModule = PyImport_Import(pName);
Py_DECREF(pName);
if (pModule != NULL) { if (pModule != NULL) {
pDict = PyModule_GetDict(pModule); pDict = PyModule_GetDict(pModule);
/* pDict is a borrowed reference */ /* pDict is a borrowed reference */
...@@ -29,6 +31,8 @@ main(int argc, char *argv[]) ...@@ -29,6 +31,8 @@ main(int argc, char *argv[])
for (i = 0; i < argc - 3; ++i) { for (i = 0; i < argc - 3; ++i) {
pValue = PyInt_FromLong(atoi(argv[i + 3])); pValue = PyInt_FromLong(atoi(argv[i + 3]));
if (!pValue) { if (!pValue) {
Py_DECREF(pArgs);
Py_DECREF(pModule);
fprintf(stderr, "Cannot convert argument\n"); fprintf(stderr, "Cannot convert argument\n");
return 1; return 1;
} }
...@@ -36,20 +40,22 @@ main(int argc, char *argv[]) ...@@ -36,20 +40,22 @@ main(int argc, char *argv[])
PyTuple_SetItem(pArgs, i, pValue); PyTuple_SetItem(pArgs, i, pValue);
} }
pValue = PyObject_CallObject(pFunc, pArgs); pValue = PyObject_CallObject(pFunc, pArgs);
Py_DECREF(pArgs);
if (pValue != NULL) { if (pValue != NULL) {
printf("Result of call: %ld\n", PyInt_AsLong(pValue)); printf("Result of call: %ld\n", PyInt_AsLong(pValue));
Py_DECREF(pValue); Py_DECREF(pValue);
} }
else { else {
Py_DECREF(pModule);
PyErr_Print(); PyErr_Print();
fprintf(stderr,"Call failed\n"); fprintf(stderr,"Call failed\n");
return 1; return 1;
} }
Py_DECREF(pArgs);
/* pDict and pFunc are borrowed and must not be Py_DECREF-ed */ /* pDict and pFunc are borrowed and must not be Py_DECREF-ed */
} }
else { else {
PyErr_Print(); if (PyErr_Occurred())
PyErr_Print();
fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]); fprintf(stderr, "Cannot find function \"%s\"\n", argv[2]);
} }
Py_DECREF(pModule); Py_DECREF(pModule);
...@@ -59,7 +65,6 @@ main(int argc, char *argv[]) ...@@ -59,7 +65,6 @@ main(int argc, char *argv[])
fprintf(stderr, "Failed to load \"%s\"\n", argv[1]); fprintf(stderr, "Failed to load \"%s\"\n", argv[1]);
return 1; return 1;
} }
Py_DECREF(pName);
Py_Finalize(); Py_Finalize();
return 0; return 0;
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment