| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * $Id: _psutil_posix.c 1142 2011-10-05 18:45:49Z g.rodola $ | |
| 3 * | |
| 4 * Copyright (c) 2009, Jay Loden, Giampaolo Rodola'. All rights reserved. | |
| 5 * Use of this source code is governed by a BSD-style license that can be | |
| 6 * found in the LICENSE file. | |
| 7 * | |
| 8 * Functions specific to all POSIX compliant platforms. | |
| 9 */ | |
| 10 | |
| 11 #include <Python.h> | |
| 12 #include <errno.h> | |
| 13 #include <stdlib.h> | |
| 14 #include <sys/resource.h> | |
| 15 | |
| 16 #include "_psutil_posix.h" | |
| 17 | |
| 18 | |
| 19 /* | |
| 20 * Given a PID return process priority as a Python integer. | |
| 21 */ | |
| 22 static PyObject* | |
| 23 posix_getpriority(PyObject* self, PyObject* args) | |
| 24 { | |
| 25 long pid; | |
| 26 int priority; | |
| 27 errno = 0; | |
| 28 if (! PyArg_ParseTuple(args, "l", &pid)) { | |
| 29 return NULL; | |
| 30 } | |
| 31 priority = getpriority(PRIO_PROCESS, pid); | |
| 32 if (errno != 0) { | |
| 33 return PyErr_SetFromErrno(PyExc_OSError); | |
| 34 } | |
| 35 return Py_BuildValue("i", priority); | |
| 36 } | |
| 37 | |
| 38 /* | |
| 39 * Given a PID and a value change process priority. | |
| 40 */ | |
| 41 static PyObject* | |
| 42 posix_setpriority(PyObject* self, PyObject* args) | |
| 43 { | |
| 44 long pid; | |
| 45 int priority; | |
| 46 int retval; | |
| 47 if (! PyArg_ParseTuple(args, "li", &pid, &priority)) { | |
| 48 return NULL; | |
| 49 } | |
| 50 retval = setpriority(PRIO_PROCESS, pid, priority); | |
| 51 if (retval == -1) { | |
| 52 return PyErr_SetFromErrno(PyExc_OSError); | |
| 53 } | |
| 54 Py_INCREF(Py_None); | |
| 55 return Py_None; | |
| 56 } | |
| 57 | |
| 58 | |
| 59 /* | |
| 60 * define the psutil C module methods and initialize the module. | |
| 61 */ | |
| 62 static PyMethodDef | |
| 63 PsutilMethods[] = | |
| 64 { | |
| 65 {"getpriority", posix_getpriority, METH_VARARGS, | |
| 66 "Return process priority"}, | |
| 67 {"setpriority", posix_setpriority, METH_VARARGS, | |
| 68 "Set process priority"}, | |
| 69 {NULL, NULL, 0, NULL} | |
| 70 }; | |
| 71 | |
| 72 struct module_state { | |
| 73 PyObject *error; | |
| 74 }; | |
| 75 | |
| 76 #if PY_MAJOR_VERSION >= 3 | |
| 77 #define GETSTATE(m) ((struct module_state*)PyModule_GetState(m)) | |
| 78 #else | |
| 79 #define GETSTATE(m) (&_state) | |
| 80 #endif | |
| 81 | |
| 82 #if PY_MAJOR_VERSION >= 3 | |
| 83 | |
| 84 static int | |
| 85 psutil_posix_traverse(PyObject *m, visitproc visit, void *arg) { | |
| 86 Py_VISIT(GETSTATE(m)->error); | |
| 87 return 0; | |
| 88 } | |
| 89 | |
| 90 static int | |
| 91 psutil_posix_clear(PyObject *m) { | |
| 92 Py_CLEAR(GETSTATE(m)->error); | |
| 93 return 0; | |
| 94 } | |
| 95 | |
| 96 static struct PyModuleDef | |
| 97 moduledef = { | |
| 98 PyModuleDef_HEAD_INIT, | |
| 99 "psutil_posix", | |
| 100 NULL, | |
| 101 sizeof(struct module_state), | |
| 102 PsutilMethods, | |
| 103 NULL, | |
| 104 psutil_posix_traverse, | |
| 105 psutil_posix_clear, | |
| 106 NULL | |
| 107 }; | |
| 108 | |
| 109 #define INITERROR return NULL | |
| 110 | |
| 111 PyObject * | |
| 112 PyInit__psutil_posix(void) | |
| 113 | |
| 114 #else | |
| 115 #define INITERROR return | |
| 116 | |
| 117 void init_psutil_posix(void) | |
| 118 #endif | |
| 119 { | |
| 120 #if PY_MAJOR_VERSION >= 3 | |
| 121 PyObject *module = PyModule_Create(&moduledef); | |
| 122 #else | |
| 123 PyObject *module = Py_InitModule("_psutil_posix", PsutilMethods); | |
| 124 #endif | |
| 125 if (module == NULL) { | |
| 126 INITERROR; | |
| 127 } | |
| 128 #if PY_MAJOR_VERSION >= 3 | |
| 129 return module; | |
| 130 #endif | |
| 131 } | |
| 132 | |
| 133 | |
| 134 | |
| OLD | NEW |